From 05ff1563407f2b48347aaeeef627b464815b19f6 Mon Sep 17 00:00:00 2001 From: rboone Date: Wed, 27 Mar 2024 17:25:13 +0100 Subject: [PATCH] black formatting --- private_gpt/components/llm/custom/ollama.py | 3 ++- private_gpt/settings/settings.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/private_gpt/components/llm/custom/ollama.py b/private_gpt/components/llm/custom/ollama.py index af4d3702..80447e6e 100644 --- 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 index 8f036910..b6db64e2 100644 --- 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