summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/needs_auth
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--g4f/Provider/needs_auth/OpenaiChat.py49
1 files changed, 46 insertions, 3 deletions
diff --git a/g4f/Provider/needs_auth/OpenaiChat.py b/g4f/Provider/needs_auth/OpenaiChat.py
index 14619958..eaccb7f1 100644
--- a/g4f/Provider/needs_auth/OpenaiChat.py
+++ b/g4f/Provider/needs_auth/OpenaiChat.py
@@ -1,11 +1,13 @@
from __future__ import annotations
-import uuid, json, time
+import uuid, json, time, os
+import tempfile, subprocess, shutil
from ..base_provider import AsyncGeneratorProvider
from ..helper import get_browser, get_cookies, format_prompt, get_event_loop
from ...typing import AsyncResult, Messages
from ...requests import StreamSession
+from ... import debug
class OpenaiChat(AsyncGeneratorProvider):
url = "https://chat.openai.com"
@@ -47,6 +49,7 @@ class OpenaiChat(AsyncGeneratorProvider):
]
data = {
"action": "next",
+ "arkose_token": await get_arkose_token(proxy),
"messages": messages,
"conversation_id": None,
"parent_message_id": str(uuid.uuid4()),
@@ -65,7 +68,11 @@ class OpenaiChat(AsyncGeneratorProvider):
line = json.loads(line)
except:
continue
- if "message" not in line or "message_type" not in line["message"]["metadata"]:
+ if "message" not in line:
+ continue
+ if "error" in line and line["error"]:
+ raise RuntimeError(line["error"])
+ if "message_type" not in line["message"]["metadata"]:
continue
if line["message"]["metadata"]["message_type"] == "next":
new_message = line["message"]["content"]["parts"][0]
@@ -134,4 +141,40 @@ class OpenaiChat(AsyncGeneratorProvider):
("cookies", "dict[str, str]")
]
param = ", ".join([": ".join(p) for p in params])
- return f"g4f.provider.{cls.__name__} supports: ({param})" \ No newline at end of file
+ return f"g4f.provider.{cls.__name__} supports: ({param})"
+
+async def get_arkose_token(proxy: str = None) -> str:
+ node = shutil.which("node")
+ if not node:
+ if debug.logging:
+ print('OpenaiChat: "node" not found')
+ return
+ dir = os.path.dirname(os.path.dirname(__file__))
+ include = f'{dir}/npm/node_modules/funcaptcha'
+ config = {
+ "pkey": "3D86FBBA-9D22-402A-B512-3420086BA6CC",
+ "surl": "https://tcr9i.chat.openai.com",
+ "data": {},
+ "headers": {
+ "User-Agent": 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36'
+ },
+ "site": "https://chat.openai.com",
+ "proxy": proxy
+ }
+ source = """
+fun = require({include})
+config = {config}
+fun.getToken(config).then(token => {
+ console.log(token.token)
+})
+"""
+ source = source.replace('{include}', json.dumps(include))
+ source = source.replace('{config}', json.dumps(config))
+ tmp = tempfile.NamedTemporaryFile(delete=False)
+ tmp.write(source.encode())
+ tmp.close()
+ try:
+ p = subprocess.Popen([node, tmp.name], stdout=subprocess.PIPE)
+ return p.stdout.read().decode()
+ finally:
+ os.unlink(tmp.name) \ No newline at end of file