summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Vitalentum.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/Provider/Vitalentum.py')
-rw-r--r--g4f/Provider/Vitalentum.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/g4f/Provider/Vitalentum.py b/g4f/Provider/Vitalentum.py
index 9125c285..ccaaeb00 100644
--- a/g4f/Provider/Vitalentum.py
+++ b/g4f/Provider/Vitalentum.py
@@ -1,7 +1,7 @@
from __future__ import annotations
import json
-from aiohttp import ClientSession
+from aiohttp import ClientSession, ClientTimeout
from .base_provider import AsyncGeneratorProvider
from ..typing import AsyncGenerator
@@ -41,7 +41,7 @@ class Vitalentum(AsyncGeneratorProvider):
**kwargs
}
async with ClientSession(
- headers=headers, timeout=timeout
+ headers=headers, timeout=ClientTimeout(timeout)
) as session:
async with session.post(cls.url + "/api/converse-edge", json=data, proxy=proxy) as response:
response.raise_for_status()