diff options
Diffstat (limited to 'g4f/Provider/Providers/helpers/theb.py')
| -rw-r--r-- | g4f/Provider/Providers/helpers/theb.py | 48 |
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 |