bilalb/Philosophy-RAG-demo
1
0
Fork 0
forked from AI_team/Philosophy-RAG-demo
Code Pull Requests Activity

Makes backend enum pretty printable

Browse Source
This commit is contained in:
Nielson Janné 2025-03-12 14:45:25 +01:00
parent 562c9a2ae7
commit 32c8774a0a
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
generic_rag/backend/models.py
View File

@ -19,6 +19,10 @@ class BackendType(Enum):
aws = "aws" aws = "aws"
local = "local" local = "local"
# make the enum pretty printable for argparse
def __str__(self):
return self.value
def get_chat_model(backend_type: BackendType) -> BaseChatModel: def get_chat_model(backend_type: BackendType) -> BaseChatModel:
if backend_type == BackendType.azure: if backend_type == BackendType.azure: