diff --git a/private_gpt/components/embedding/embedding_component.py b/private_gpt/components/embedding/embedding_component.py index 173501c..4f0ae93 100644 --- a/private_gpt/components/embedding/embedding_component.py +++ b/private_gpt/components/embedding/embedding_component.py @@ -75,7 +75,9 @@ class EmbeddingComponent: "FireworksEmbedding dependencies not found, install with `poetry install --extras embeddings-fireworks`" ) from e - api_key = settings.fireworks.embedding_api_key or settings.fireworks.api_key + api_key = ( + settings.fireworks.embedding_api_key or settings.fireworks.api_key + ) model = settings.openai.embedding_model self.embedding_model = FireworksEmbedding( api_key=api_key, diff --git a/private_gpt/settings/settings.py b/private_gpt/settings/settings.py index f94162f..dbb0d2a 100644 --- a/private_gpt/settings/settings.py +++ b/private_gpt/settings/settings.py @@ -269,6 +269,7 @@ class OpenAISettings(BaseModel): description="OpenAI embedding Model to use. Example: 'text-embedding-3-large'.", ) + class FireWorksSettings(BaseModel): api_key: str model: str = Field( @@ -285,6 +286,7 @@ class FireWorksSettings(BaseModel): description="FIREWORKS embedding Model to use. Example: 'nomic-ai/nomic-embed-text-v1.5'.", ) + class GeminiSettings(BaseModel): api_key: str model: str = Field(