フォーク元 g4f/gpt4free
49 行
1.6 KiB
Python
49 行
1.6 KiB
Python
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 |