fixed test black error

zylon-ai · Sep 24, 2024 · cecec30 · cecec30
1 parent 9c3590e
commit cecec30
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/private_gpt/components/embedding/embedding_component.py b/private_gpt/components/embedding/embedding_component.py
@@ -75,7 +75,9 @@ def __init__(self, settings: Settings) -> None:
                         "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
@@ -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(