diff options
Diffstat (limited to 'g4f/Provider/Yqcloud.py')
| -rw-r--r-- | g4f/Provider/Yqcloud.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/g4f/Provider/Yqcloud.py b/g4f/Provider/Yqcloud.py index ac93315c..1b22de9e 100644 --- a/g4f/Provider/Yqcloud.py +++ b/g4f/Provider/Yqcloud.py @@ -16,10 +16,11 @@ class Yqcloud(AsyncGeneratorProvider): model: str, messages: list[dict[str, str]], proxy: str = None, + timeout: int = 30, **kwargs, ) -> AsyncGenerator: async with ClientSession( - headers=_create_header() + headers=_create_header(), timeout=timeout ) as session: payload = _create_payload(messages) async with session.post("https://api.aichatos.cloud/api/generateStream", proxy=proxy, json=payload) as response: |