diff options
Diffstat (limited to '')
| -rw-r--r-- | g4f/Provider/GPTalk.py | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/g4f/Provider/GPTalk.py b/g4f/Provider/GPTalk.py new file mode 100644 index 00000000..c85399c1 --- /dev/null +++ b/g4f/Provider/GPTalk.py @@ -0,0 +1,83 @@ +from __future__ import annotations + +import secrets, time, json +from aiohttp import ClientSession +from typing import AsyncGenerator + +from .base_provider import AsyncGeneratorProvider +from .helper import format_prompt + + +class GPTalk(AsyncGeneratorProvider): + url = "https://gptalk.net" + supports_gpt_35_turbo = True + working = True + _auth = None + + @classmethod + async def create_async_generator( + cls, + model: str, + messages: list[dict[str, str]], + **kwargs + ) -> AsyncGenerator: + if not model: + model = "gpt-3.5-turbo" + timestamp = int(time.time()) + headers = { + 'authority': 'gptalk.net', + 'accept': '*/*', + 'accept-language': 'de-DE,de;q=0.9,en-DE;q=0.8,en;q=0.7,en-US;q=0.6,nl;q=0.5,zh-CN;q=0.4,zh-TW;q=0.3,zh;q=0.2', + 'content-type': 'application/json', + 'origin': 'https://gptalk.net', + 'sec-ch-ua': '"Google Chrome";v="117", "Not;A=Brand";v="8", "Chromium";v="117"', + 'sec-ch-ua-mobile': '?0', + 'sec-ch-ua-platform': '"Linux"', + 'sec-fetch-dest': 'empty', + 'sec-fetch-mode': 'cors', + 'sec-fetch-site': 'same-origin', + 'user-agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/117.0.0.0 Safari/537.36', + 'x-auth-appid': '2229', + 'x-auth-openid': '', + 'x-auth-platform': '', + 'x-auth-timestamp': f"{timestamp}", + } + async with ClientSession(headers=headers) as session: + if not cls._auth or cls._auth["expires_at"] < timestamp: + data = { + "fingerprint": secrets.token_hex(16).zfill(32), + "platform": "fingerprint" + } + async with session.post(cls.url + "/api/chatgpt/user/login", json=data) as response: + response.raise_for_status() + cls._auth = (await response.json())["data"] + data = { + "content": format_prompt(messages), + "accept": "stream", + "from": 1, + "model": model, + "is_mobile": 0, + "user_agent": headers["user-agent"], + "is_open_ctx": 0, + "prompt": "", + "roid": 111, + "temperature": 0, + "ctx_msg_count": 3, + "created_at": timestamp + } + headers = { + 'authorization': f'Bearer {cls._auth["token"]}', + } + async with session.post(cls.url + "/api/chatgpt/chatapi/text", json=data, headers=headers) as response: + response.raise_for_status() + token = (await response.json())["data"]["token"] + last_message = "" + async with session.get(cls.url + "/api/chatgpt/chatapi/stream", params={"token": token}) as response: + response.raise_for_status() + async for line in response.content: + if line.startswith(b"data: "): + if line.startswith(b"data: [DONE]"): + break + message = json.loads(line[6:-1])["content"] + yield message[len(last_message):] + last_message = message
\ No newline at end of file |