FEAT: Added fixed UI mode name

This commit is contained in:
Somashekar B R 2024-09-20 14:49:02 +05:30
parent a38675b844
commit 2d5865e773
6 changed files with 2292 additions and 2002 deletions

View file

@ -115,7 +115,7 @@ class LLMSettings(BaseModel):
"mock",
"ollama",
"gemini",
"fireworks"
"fireworks",
]
max_new_tokens: int = Field(
256,
@ -198,7 +198,7 @@ class HuggingFaceSettings(BaseModel):
class EmbeddingSettings(BaseModel):
mode: Literal[
"huggingface", "openai", "azopenai", "sagemaker", "ollama", "mock", "gemini"
"huggingface", "openai", "azopenai", "sagemaker", "ollama", "mock", "gemini","fireworks"
]
ingest_mode: Literal["simple", "batch", "parallel", "pipeline"] = Field(
"simple",