diff options
Diffstat (limited to 'g4f/Provider/Yqcloud.py')
| -rw-r--r-- | g4f/Provider/Yqcloud.py | 42 |
1 files changed, 21 insertions, 21 deletions
diff --git a/g4f/Provider/Yqcloud.py b/g4f/Provider/Yqcloud.py index a3147c2d..731e4ecb 100644 --- a/g4f/Provider/Yqcloud.py +++ b/g4f/Provider/Yqcloud.py @@ -1,45 +1,45 @@ -import requests +from __future__ import annotations -from ..typing import Any, CreateResult -from .base_provider import BaseProvider +from aiohttp import ClientSession +from .base_provider import AsyncProvider, format_prompt -class Yqcloud(BaseProvider): + +class Yqcloud(AsyncProvider): url = "https://chat9.yqcloud.top/" working = True supports_gpt_35_turbo = True @staticmethod - def create_completion( + async def create_async( model: str, messages: list[dict[str, str]], - stream: bool, - **kwargs: Any, - ) -> CreateResult: - headers = _create_header() - payload = _create_payload(messages) - - url = "https://api.aichatos.cloud/api/generateStream" - response = requests.post(url=url, headers=headers, json=payload) - response.raise_for_status() - response.encoding = 'utf-8' - yield response.text + proxy: str = None, + **kwargs, + ) -> str: + async with ClientSession( + headers=_create_header() + ) as session: + payload = _create_payload(messages) + async with session.post("https://api.aichatos.cloud/api/generateStream", proxy=proxy, json=payload) as response: + response.raise_for_status() + return await response.text() def _create_header(): return { - "accept": "application/json, text/plain, */*", - "content-type": "application/json", - "origin": "https://chat9.yqcloud.top", + "accept" : "application/json, text/plain, */*", + "content-type" : "application/json", + "origin" : "https://chat9.yqcloud.top", } def _create_payload(messages: list[dict[str, str]]): - prompt = messages[-1]["content"] return { - "prompt": prompt, + "prompt": format_prompt(messages), "network": True, "system": "", "withoutContext": False, "stream": False, + "userId": "#/chat/1693025544336" } |