Skip to content

Commit cecec30

Browse files
fixed test black error
1 parent 9c3590e commit cecec30

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

private_gpt/components/embedding/embedding_component.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,9 @@ def __init__(self, settings: Settings) -> None:
7575
"FireworksEmbedding dependencies not found, install with `poetry install --extras embeddings-fireworks`"
7676
) from e
7777

78-
api_key = settings.fireworks.embedding_api_key or settings.fireworks.api_key
78+
api_key = (
79+
settings.fireworks.embedding_api_key or settings.fireworks.api_key
80+
)
7981
model = settings.openai.embedding_model
8082
self.embedding_model = FireworksEmbedding(
8183
api_key=api_key,

private_gpt/settings/settings.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -269,6 +269,7 @@ class OpenAISettings(BaseModel):
269269
description="OpenAI embedding Model to use. Example: 'text-embedding-3-large'.",
270270
)
271271

272+
272273
class FireWorksSettings(BaseModel):
273274
api_key: str
274275
model: str = Field(
@@ -285,6 +286,7 @@ class FireWorksSettings(BaseModel):
285286
description="FIREWORKS embedding Model to use. Example: 'nomic-ai/nomic-embed-text-v1.5'.",
286287
)
287288

289+
288290
class GeminiSettings(BaseModel):
289291
api_key: str
290292
model: str = Field(

0 commit comments

Comments
 (0)