mirror of
https://github.com/imartinez/privateGPT.git
synced 2025-08-01 23:47:54 +00:00
fix: code styling
applied "make check" changes
This commit is contained in:
parent
71e44ce738
commit
e3c0c7fbfc
@ -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`"
|
||||
|
@ -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`"
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user