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