diff options
author | t.me/xtekky <98614666+xtekky@users.noreply.github.com> | 2023-06-24 03:47:00 +0200 |
---|---|---|
committer | t.me/xtekky <98614666+xtekky@users.noreply.github.com> | 2023-06-24 03:47:00 +0200 |
commit | 5db58fd87f230fbe5bae599bb4b120ab42cad3be (patch) | |
tree | 770be13bca77c5d04dfe3265f378431df788706f /g4f/Provider/Providers/Bard.py | |
parent | Merge pull request #664 from LopeKinz/main (diff) | |
download | gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar.gz gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar.bz2 gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar.lz gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar.xz gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar.zst gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.zip |
Diffstat (limited to '')
-rw-r--r-- | g4f/Provider/Providers/Bard.py | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/g4f/Provider/Providers/Bard.py b/g4f/Provider/Providers/Bard.py new file mode 100644 index 00000000..4c37c4b7 --- /dev/null +++ b/g4f/Provider/Providers/Bard.py @@ -0,0 +1,74 @@ +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 +needs_auth = True + +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 = kwargs.get('proxy', False) + if proxy == False: + print('warning!, you did not give a proxy, a lot of countries are banned from Google Bard, so it may not work') + + snlm0e = None + conversation_id = None + response_id = None + choice_id = None + + client = requests.Session() + client.proxies = { + 'http': f'http://{proxy}', + 'https': f'http://{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]])
\ No newline at end of file |