summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/GPTalk.py
diff options
context:
space:
mode:
authorHeiner Lohaus <heiner@lohaus.eu>2023-10-07 04:03:36 +0200
committerHeiner Lohaus <heiner@lohaus.eu>2023-10-07 04:03:36 +0200
commit1238d9a63839165567511cc88e49b246c453cc47 (patch)
tree39e2d809e74e2639eb4b985b3008c161173a7df4 /g4f/Provider/GPTalk.py
parent ~ | g4f `v-0.1.5.4` [new gui & interference] (diff)
downloadgpt4free-1238d9a63839165567511cc88e49b246c453cc47.tar
gpt4free-1238d9a63839165567511cc88e49b246c453cc47.tar.gz
gpt4free-1238d9a63839165567511cc88e49b246c453cc47.tar.bz2
gpt4free-1238d9a63839165567511cc88e49b246c453cc47.tar.lz
gpt4free-1238d9a63839165567511cc88e49b246c453cc47.tar.xz
gpt4free-1238d9a63839165567511cc88e49b246c453cc47.tar.zst
gpt4free-1238d9a63839165567511cc88e49b246c453cc47.zip
Diffstat (limited to '')
-rw-r--r--g4f/Provider/GPTalk.py83
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