diff options
Diffstat (limited to 'g4f/Provider/Vitalentum.py')
| -rw-r--r-- | g4f/Provider/Vitalentum.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/g4f/Provider/Vitalentum.py b/g4f/Provider/Vitalentum.py index d5265428..9125c285 100644 --- a/g4f/Provider/Vitalentum.py +++ b/g4f/Provider/Vitalentum.py @@ -18,6 +18,7 @@ class Vitalentum(AsyncGeneratorProvider): model: str, messages: list[dict[str, str]], proxy: str = None, + timeout: int = 30, **kwargs ) -> AsyncGenerator: headers = { @@ -40,7 +41,7 @@ class Vitalentum(AsyncGeneratorProvider): **kwargs } async with ClientSession( - headers=headers + headers=headers, timeout=timeout ) as session: async with session.post(cls.url + "/api/converse-edge", json=data, proxy=proxy) as response: response.raise_for_status() |