diff options
Diffstat (limited to '')
| -rw-r--r-- | g4f/Provider/AiChatOnline.py | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/g4f/Provider/AiChatOnline.py b/g4f/Provider/AiChatOnline.py new file mode 100644 index 00000000..dc774fe0 --- /dev/null +++ b/g4f/Provider/AiChatOnline.py @@ -0,0 +1,59 @@ +from __future__ import annotations + +import json +from aiohttp import ClientSession + +from ..typing import AsyncResult, Messages +from .base_provider import AsyncGeneratorProvider +from .helper import get_random_string + +class AiChatOnline(AsyncGeneratorProvider): + url = "https://aichatonline.org" + working = True + supports_gpt_35_turbo = True + supports_message_history = False + + @classmethod + async def create_async_generator( + cls, + model: str, + messages: Messages, + proxy: str = None, + **kwargs + ) -> AsyncResult: + headers = { + "User-Agent": "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/119.0", + "Accept": "text/event-stream", + "Accept-Language": "de,en-US;q=0.7,en;q=0.3", + "Accept-Encoding": "gzip, deflate, br", + "Referer": f"{cls.url}/chatgpt/chat/", + "Content-Type": "application/json", + "Origin": cls.url, + "Alt-Used": "aichatonline.org", + "Connection": "keep-alive", + "Sec-Fetch-Dest": "empty", + "Sec-Fetch-Mode": "cors", + "Sec-Fetch-Site": "same-origin", + "TE": "trailers" + } + async with ClientSession(headers=headers) as session: + data = { + "botId": "default", + "customId": None, + "session": get_random_string(16), + "chatId": get_random_string(), + "contextId": 7, + "messages": messages, + "newMessage": messages[-1]["content"], + "newImageId": None, + "stream": True + } + async with session.post(f"{cls.url}/chatgpt/wp-json/mwai-ui/v1/chats/submit", json=data, proxy=proxy) as response: + response.raise_for_status() + async for chunk in response.content: + if chunk.startswith(b"data: "): + data = json.loads(chunk[6:]) + if data["type"] == "live": + yield data["data"] + elif data["type"] == "end": + break
\ No newline at end of file |