mirror of
https://github.com/imartinez/privateGPT.git
synced 2025-08-17 23:16:55 +00:00
black formatting
This commit is contained in:
parent
e0533a40ed
commit
05ff156340
@ -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
|
||||
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user