github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-08-21 08:54:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

black formatting

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
private_gpt/components/llm/custom/ollama.py
View File

@ -11,7 +11,8 @@ class CustomOllama(Ollama):
) )
def __init__(self, *args, **kwargs) -> None: 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) super().__init__(*args, **kwargs)
self.keep_alive = keep_alive self.keep_alive = keep_alive

1
private_gpt/settings/settings.py
View File

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