import json import sys 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: chunk_json = json.loads(chunk.decode('utf-8')) completion_chunk = chunk_json['detail']['choices'][0]['delta']['content'] print(completion_chunk, flush=True, end = '') except Exception as e: print('[ERROR] an error occured, retrying... |', e, 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