From 1238d9a63839165567511cc88e49b246c453cc47 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <heiner@lohaus.eu>
Date: Sat, 7 Oct 2023 04:03:36 +0200
Subject: Add GPTalk and GptForLove Provider

---
 g4f/Provider/GPTalk.py | 83 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)
 create mode 100644 g4f/Provider/GPTalk.py

(limited to 'g4f/Provider/GPTalk.py')

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
-- 
cgit v1.2.3