diff options
author | ostix360 <vicrobin1001@gmail.com> | 2023-10-17 09:29:12 +0200 |
---|---|---|
committer | ostix360 <vicrobin1001@gmail.com> | 2023-10-17 09:29:12 +0200 |
commit | 24f7495f24acb73997deca85d243025ba638db53 (patch) | |
tree | c8289e9fdcd1fd08d4dd51b9a86b6aa0914fadab /g4f | |
parent | ~ (diff) | |
download | gpt4free-24f7495f24acb73997deca85d243025ba638db53.tar gpt4free-24f7495f24acb73997deca85d243025ba638db53.tar.gz gpt4free-24f7495f24acb73997deca85d243025ba638db53.tar.bz2 gpt4free-24f7495f24acb73997deca85d243025ba638db53.tar.lz gpt4free-24f7495f24acb73997deca85d243025ba638db53.tar.xz gpt4free-24f7495f24acb73997deca85d243025ba638db53.tar.zst gpt4free-24f7495f24acb73997deca85d243025ba638db53.zip |
Diffstat (limited to 'g4f')
-rw-r--r-- | g4f/Provider/retry_provider.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/g4f/Provider/retry_provider.py b/g4f/Provider/retry_provider.py index 94b9b90a..72170756 100644 --- a/g4f/Provider/retry_provider.py +++ b/g4f/Provider/retry_provider.py @@ -1,5 +1,6 @@ from __future__ import annotations +import asyncio import random from typing import List, Type, Dict from ..typing import CreateResult, Messages @@ -68,7 +69,11 @@ class RetryProvider(AsyncProvider): self.exceptions: Dict[str, Exception] = {} for provider in providers: try: - return await provider.create_async(model, messages, **kwargs) + return await asyncio.wait_for(provider.create_async(model, messages, **kwargs), timeout=60) + except asyncio.TimeoutError as e: + self.exceptions[provider.__name__] = e + if logging: + print(f"{provider.__name__}: TimeoutError: {e}") except Exception as e: self.exceptions[provider.__name__] = e if logging: |