フォーク元 g4f/gpt4free
76 行
2.5 KiB
Python
76 行
2.5 KiB
Python
|
# implement proxy argument
|
||
|
|
||
|
import os, requests, json, browser_cookie3, re, random
|
||
|
from ...typing import sha256, Dict, get_type_hints
|
||
|
|
||
|
url = 'https://bard.google.com'
|
||
|
model = ['Palm2']
|
||
|
supports_stream = False
|
||
|
|
||
|
def _create_completion(model: str, messages: list, stream: bool, **kwargs):
|
||
|
psid = {cookie.name: cookie.value for cookie in browser_cookie3.chrome(
|
||
|
domain_name='.google.com')}['__Secure-1PSID']
|
||
|
|
||
|
formatted = '\n'.join([
|
||
|
'%s: %s' % (message['role'], message['content']) for message in messages
|
||
|
])
|
||
|
prompt = f'{formatted}\nAssistant:'
|
||
|
|
||
|
proxy = None
|
||
|
|
||
|
if proxy == None:
|
||
|
raise Exception('Proxy is required for Bard (set in g4f/Provider/Providers/Bard.py line 18)')
|
||
|
|
||
|
snlm0e = False
|
||
|
conversation_id = None
|
||
|
response_id = None
|
||
|
choice_id = None
|
||
|
|
||
|
client = requests.Session()
|
||
|
client.proxies = {
|
||
|
'http': f'https://{proxy}',
|
||
|
'https': f'https://{proxy}'} if proxy else None
|
||
|
|
||
|
client.headers = {
|
||
|
'authority': 'bard.google.com',
|
||
|
'content-type': 'application/x-www-form-urlencoded;charset=UTF-8',
|
||
|
'origin': 'https://bard.google.com',
|
||
|
'referer': 'https://bard.google.com/',
|
||
|
'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36',
|
||
|
'x-same-domain': '1',
|
||
|
'cookie': f'__Secure-1PSID={psid}'
|
||
|
}
|
||
|
|
||
|
snlm0e = re.search(r'SNlM0e\":\"(.*?)\"',
|
||
|
client.get('https://bard.google.com/').text).group(1) if not snlm0e else snlm0e
|
||
|
|
||
|
params = {
|
||
|
'bl': 'boq_assistant-bard-web-server_20230326.21_p0',
|
||
|
'_reqid': random.randint(1111, 9999),
|
||
|
'rt': 'c'
|
||
|
}
|
||
|
|
||
|
data = {
|
||
|
'at': snlm0e,
|
||
|
'f.req': json.dumps([None, json.dumps([[prompt], None, [conversation_id, response_id, choice_id]])])}
|
||
|
|
||
|
intents = '.'.join([
|
||
|
'assistant',
|
||
|
'lamda',
|
||
|
'BardFrontendService'
|
||
|
])
|
||
|
|
||
|
response = client.post(f'https://bard.google.com/_/BardChatUi/data/{intents}/StreamGenerate',
|
||
|
data=data, params=params)
|
||
|
|
||
|
chat_data = json.loads(response.content.splitlines()[3])[0][2]
|
||
|
if chat_data:
|
||
|
json_chat_data = json.loads(chat_data)
|
||
|
|
||
|
yield json_chat_data[0][0]
|
||
|
|
||
|
else:
|
||
|
yield 'error'
|
||
|
|
||
|
params = f'g4f.Providers.{os.path.basename(__file__)[:-3]} supports: ' + \
|
||
|
'(%s)' % ', '.join([f"{name}: {get_type_hints(_create_completion)[name].__name__}" for name in _create_completion.__code__.co_varnames[:_create_completion.__code__.co_argcount]])
|