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) _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: def check_connection(client: Client) -> bool:
try: try:
client.list() client.list()

2
private_gpt/utils/retry.py
View file

@ -28,4 +28,4 @@ def retry(
jitter=jitter, jitter=jitter,
logger=logger, logger=logger,
) )
return wrapped # type: ignore return wrapped # type: ignore