diff options
Diffstat (limited to 'g4f/Provider/Aibn.py')
-rw-r--r-- | g4f/Provider/Aibn.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/g4f/Provider/Aibn.py b/g4f/Provider/Aibn.py index df56189b..3399d613 100644 --- a/g4f/Provider/Aibn.py +++ b/g4f/Provider/Aibn.py @@ -18,9 +18,10 @@ class Aibn(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: timestamp = int(time.time()) data = { "messages": messages, |