black formatting

zylon-ai · Mar 27, 2024 · 05ff156 · 05ff156
1 parent e0533a4
commit 05ff156
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/private_gpt/components/llm/custom/ollama.py b/private_gpt/components/llm/custom/ollama.py
@@ -11,7 +11,8 @@ class CustomOllama(Ollama):
     )
 
     def __init__(self, *args, **kwargs) -> None:
-        keep_alive = kwargs.pop('keep_alive', '5m')  # fetch keep_alive from kwargs or use 5m if not found.
+        # fetch keep_alive from kwargs or use 5m if not found.
+        keep_alive = kwargs.pop("keep_alive", "5m")
         super().__init__(*args, **kwargs)
         self.keep_alive = keep_alive
 

diff --git a/private_gpt/settings/settings.py b/private_gpt/settings/settings.py
@@ -254,6 +254,7 @@ class OllamaSettings(BaseModel):
         description="Time elapsed until ollama times out the request. Default is 120s. Format is float. ",
     )
 
+
 class AzureOpenAISettings(BaseModel):
     api_key: str
     azure_endpoint: str