v2

1 files changed, 48 insertions, 0 deletions

diff --git a/g4f/Provider/Providers/helpers/theb.py b/g4f/Provider/Providers/helpers/theb.py
new file mode 100644
index 00000000..71cfd23f
--- /dev/null
+++ b/g4f/Provider/Providers/helpers/theb.py
@@ -0,0 +1,48 @@
+import json
+import sys
+from re import findall
+from curl_cffi import requests
+
+config = json.loads(sys.argv[1])
+prompt = config['messages'][-1]['content']
+
+headers = {
+    'authority': 'chatbot.theb.ai',
+    'accept': 'application/json, text/plain, */*',
+    'accept-language': 'en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3',
+    'content-type': 'application/json',
+    'origin': 'https://chatbot.theb.ai',
+    'referer': 'https://chatbot.theb.ai/',
+    'sec-ch-ua': '"Google Chrome";v="113", "Chromium";v="113", "Not-A.Brand";v="24"',
+    'sec-ch-ua-mobile': '?0',
+    'sec-ch-ua-platform': '"macOS"',
+    'sec-fetch-dest': 'empty',
+    'sec-fetch-mode': 'cors',
+    'sec-fetch-site': 'same-origin',
+    'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/113.0.0.0 Safari/537.36',
+}
+
+json_data = {
+    'prompt': prompt,
+    'options': {}
+}
+
+def format(chunk):
+    try:
+        completion_chunk = findall(r'content":"(.*)"},"fin', chunk.decode())[0]
+        print(completion_chunk, flush=True, end='')
+
+    except Exception as e:
+        print(f'[ERROR] an error occured, retrying... | [[{chunk.decode()}]]', flush=True)
+        return
+
+while True:
+    try: 
+        response = requests.post('https://chatbot.theb.ai/api/chat-process', 
+                            headers=headers, json=json_data, content_callback=format, impersonate='chrome110')
+        
+        exit(0)
+    
+    except Exception as e:
+        print('[ERROR] an error occured, retrying... |', e, flush=True)
+        continue
\ No newline at end of file