diff options
Diffstat (limited to 'g4f/Provider/ChatForAi.py')
| -rw-r--r-- | g4f/Provider/ChatForAi.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/g4f/Provider/ChatForAi.py b/g4f/Provider/ChatForAi.py index 779799cf..47ac46f0 100644 --- a/g4f/Provider/ChatForAi.py +++ b/g4f/Provider/ChatForAi.py @@ -17,9 +17,10 @@ class ChatForAi(AsyncGeneratorProvider): cls, model: str, messages: list[dict[str, str]], + timeout: int = 30, **kwargs ) -> AsyncGenerator: - async with StreamSession(impersonate="chrome107") as session: + async with StreamSession(impersonate="chrome107", timeout=timeout) as session: conversation_id = f"id_{int(time.time())}" prompt = messages[-1]["content"] timestamp = int(time.time()) |