diff options
Diffstat (limited to 'g4f/Provider/Aichat.py')
| -rw-r--r-- | g4f/Provider/Aichat.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/g4f/Provider/Aichat.py b/g4f/Provider/Aichat.py index 8edd17e2..41496765 100644 --- a/g4f/Provider/Aichat.py +++ b/g4f/Provider/Aichat.py @@ -15,6 +15,7 @@ class Aichat(AsyncProvider): model: str, messages: list[dict[str, str]], proxy: str = None, + timeout: int = 30, **kwargs ) -> str: headers = { @@ -33,7 +34,7 @@ class Aichat(AsyncProvider): "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/113.0.0.0 Safari/537.36", } async with ClientSession( - headers=headers + headers=headers, timeout=timeout ) as session: json_data = { "message": format_prompt(messages), |