SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 17:05:41 +01:00
Code Issues Projects Releases Packages Activity

fixed test black error

Browse source
This commit is contained in:
Somashekar B R 2024-09-24 23:37:03 +05:30
parent 9c3590e555
commit cecec30ca8
2 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -75,7 +75,9 @@ class EmbeddingComponent:
"FireworksEmbedding dependencies not found, install with `poetry install --extras embeddings-fireworks`" "FireworksEmbedding dependencies not found, install with `poetry install --extras embeddings-fireworks`"
) from e ) 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 model = settings.openai.embedding_model
self.embedding_model = FireworksEmbedding( self.embedding_model = FireworksEmbedding(
api_key=api_key, api_key=api_key,

2
private_gpt/settings/settings.py
View file

@ -269,6 +269,7 @@ class OpenAISettings(BaseModel):
description="OpenAI embedding Model to use. Example: 'text-embedding-3-large'.", description="OpenAI embedding Model to use. Example: 'text-embedding-3-large'.",
) )
class FireWorksSettings(BaseModel): class FireWorksSettings(BaseModel):
api_key: str api_key: str
model: str = Field( model: str = Field(
@ -285,6 +286,7 @@ class FireWorksSettings(BaseModel):
description="FIREWORKS embedding Model to use. Example: 'nomic-ai/nomic-embed-text-v1.5'.", description="FIREWORKS embedding Model to use. Example: 'nomic-ai/nomic-embed-text-v1.5'.",
) )
class GeminiSettings(BaseModel): class GeminiSettings(BaseModel):
api_key: str api_key: str
model: str = Field( model: str = Field(