fix: code styling

applied "make check" changes
This commit is contained in:
olederle 2024-03-15 14:46:25 +01:00
parent 71e44ce738
commit e3c0c7fbfc
3 changed files with 11 additions and 3 deletions

View File

@ -74,7 +74,9 @@ class EmbeddingComponent:
)
case "azopenai":
try:
from llama_index.embeddings.azure_openai import AzureOpenAIEmbedding # type: ignore
from llama_index.embeddings.azure_openai import ( # type: ignore
AzureOpenAIEmbedding,
)
except ImportError as e:
raise ImportError(
"Azure OpenAI dependencies not found, install with `poetry install --extras embeddings-azopenai`"

View File

@ -113,7 +113,9 @@ class LLMComponent:
)
case "azopenai":
try:
from llama_index.llms.azure_openai import AzureOpenAI # type: ignore
from llama_index.llms.azure_openai import ( # type: ignore
AzureOpenAI,
)
except ImportError as e:
raise ImportError(
"Azure OpenAI dependencies not found, install with `poetry install --extras llms-azopenai`"

View File

@ -81,7 +81,9 @@ class DataSettings(BaseModel):
class LLMSettings(BaseModel):
mode: Literal["llamacpp", "openai", "openailike", "azopenai", "sagemaker", "mock", "ollama"]
mode: Literal[
"llamacpp", "openai", "openailike", "azopenai", "sagemaker", "mock", "ollama"
]
max_new_tokens: int = Field(
256,
description="The maximum number of token that the LLM is authorized to generate in one completion.",
@ -185,6 +187,7 @@ class OllamaSettings(BaseModel):
description="Model to use. Example: 'nomic-embed-text'.",
)
class AzureOpenAISettings(BaseModel):
api_key: str
azure_endpoint: str
@ -203,6 +206,7 @@ class AzureOpenAISettings(BaseModel):
description="OpenAI Model to use. Example: 'gpt-4'.",
)
class UISettings(BaseModel):
enabled: bool
path: str