SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 07:40:12 +01:00
Code Issues Projects Releases Packages Activity

fix: mypy

Browse source
This commit is contained in:
Javier Martinez 2024-09-11 16:24:53 +02:00
parent d67d0c0f5e
commit 6177913544
No known key found for this signature in database
2 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
private_gpt/utils/ollama.py
View file

@ -21,7 +21,13 @@ _MAX_RETRIES = 5
_JITTER = (3.0, 10.0)
@retry(is_async=False, exceptions=(ConnectError, ResponseError), tries=_MAX_RETRIES, jitter=_JITTER, logger=logger)
@retry(
is_async=False,
exceptions=(ConnectError, ResponseError),
tries=_MAX_RETRIES,
jitter=_JITTER,
logger=logger,
)
def check_connection(client: Client) -> bool:
try:
client.list()