diff options
Diffstat (limited to 'g4f/Provider/needs_auth/OpenaiChat.py')
| -rw-r--r-- | g4f/Provider/needs_auth/OpenaiChat.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/g4f/Provider/needs_auth/OpenaiChat.py b/g4f/Provider/needs_auth/OpenaiChat.py index c57692d3..c41909e3 100644 --- a/g4f/Provider/needs_auth/OpenaiChat.py +++ b/g4f/Provider/needs_auth/OpenaiChat.py @@ -22,6 +22,7 @@ class OpenaiChat(AsyncGeneratorProvider): proxy: str = None, access_token: str = None, cookies: dict = None, + timeout: int = 30, **kwargs: dict ) -> AsyncGenerator: proxies = {"https": proxy} @@ -31,7 +32,7 @@ class OpenaiChat(AsyncGeneratorProvider): "Accept": "text/event-stream", "Authorization": f"Bearer {access_token}", } - async with StreamSession(proxies=proxies, headers=headers, impersonate="chrome107") as session: + async with StreamSession(proxies=proxies, headers=headers, impersonate="chrome107", timeout=timeout) as session: messages = [ { "id": str(uuid.uuid4()), |