black formatting

This commit is contained in:
rboone 2024-03-27 17:25:13 +01:00
parent e0533a40ed
commit 05ff156340
2 changed files with 3 additions and 1 deletions

View File

@ -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

View File

@ -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