github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-08-07 02:23:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: code styling

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
private_gpt/components/embedding/embedding_component.py
View File

@ -74,7 +74,9 @@ class EmbeddingComponent:
) )
case "azopenai": case "azopenai":
try: 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: except ImportError as e:
raise ImportError( raise ImportError(
"Azure OpenAI dependencies not found, install with `poetry install --extras embeddings-azopenai`" "Azure OpenAI dependencies not found, install with `poetry install --extras embeddings-azopenai`"

4
private_gpt/components/llm/llm_component.py
View File

@ -113,7 +113,9 @@ class LLMComponent:
) )
case "azopenai": case "azopenai":
try: 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: except ImportError as e:
raise ImportError( raise ImportError(
"Azure OpenAI dependencies not found, install with `poetry install --extras llms-azopenai`" "Azure OpenAI dependencies not found, install with `poetry install --extras llms-azopenai`"

6
private_gpt/settings/settings.py
View File

@ -81,7 +81,9 @@ class DataSettings(BaseModel):
class LLMSettings(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( max_new_tokens: int = Field(
256, 256,
description="The maximum number of token that the LLM is authorized to generate in one completion.", 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'.", description="Model to use. Example: 'nomic-embed-text'.",
) )
class AzureOpenAISettings(BaseModel): class AzureOpenAISettings(BaseModel):
api_key: str api_key: str
azure_endpoint: str azure_endpoint: str
@ -203,6 +206,7 @@ class AzureOpenAISettings(BaseModel):
description="OpenAI Model to use. Example: 'gpt-4'.", description="OpenAI Model to use. Example: 'gpt-4'.",
) )
class UISettings(BaseModel): class UISettings(BaseModel):
enabled: bool enabled: bool
path: str path: str