From b2037302e984a214014e32b00b2453a69adb8ef7 Mon Sep 17 00:00:00 2001 From: Andrew Tsegaye Date: Thu, 27 Apr 2023 09:17:07 -0700 Subject: well updated --- unfinished/bard/__init__.py | 129 ++++++++++++++++++-------------------------- 1 file changed, 53 insertions(+), 76 deletions(-) (limited to 'unfinished/bard/__init__.py') diff --git a/unfinished/bard/__init__.py b/unfinished/bard/__init__.py index 2b26997f..ef8980d7 100644 --- a/unfinished/bard/__init__.py +++ b/unfinished/bard/__init__.py @@ -1,19 +1,19 @@ -from requests import Session -from re import search -from random import randint -from json import dumps, loads -from random import randint +from requests import Session +from re import search +from random import randint +from json import dumps, loads from urllib.parse import urlencode -from dotenv import load_dotenv; load_dotenv() -from os import getenv +from dotenv import load_dotenv +from os import getenv from bard.typings import BardResponse +load_dotenv() token = getenv('1psid') proxy = getenv('proxy') temperatures = { - 0 : "Generate text strictly following known patterns, with no creativity.", + 0: "Generate text strictly following known patterns, with no creativity.", 0.1: "Produce text adhering closely to established patterns, allowing minimal creativity.", 0.2: "Create text with modest deviations from familiar patterns, injecting a slight creative touch.", 0.3: "Craft text with a mild level of creativity, deviating somewhat from common patterns.", @@ -23,93 +23,70 @@ temperatures = { 0.7: "Produce text favoring creativity over typical patterns for more original results.", 0.8: "Create text heavily focused on creativity, with limited concern for familiar patterns.", 0.9: "Craft text with a strong emphasis on unique and inventive ideas, largely ignoring established patterns.", - 1 : "Generate text with maximum creativity, disregarding any constraints of known patterns or structures." + 1: "Generate text with maximum creativity, disregarding any constraints of known patterns or structures." } + class Completion: - # def __init__(self, _token, proxy: str or None = None) -> None: - # self.client = Session() - # self.client.proxies = { - # 'http': f'http://{proxy}', - # 'https': f'http://{proxy}' } if proxy else None - - # self.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={_token}' - # } - - # self.snlm0e = self.__init_client() - # self.conversation_id = '' - # self.response_id = '' - # self.choice_id = '' - # self.reqid = randint(1111, 9999) - def create( - prompt : str = 'hello world', - temperature : int = None, - conversation_id : str = '', - response_id : str = '', - choice_id : str = '') -> BardResponse: - + prompt: str = 'hello world', + temperature: int = None, + conversation_id: str = '', + response_id: str = '', + choice_id: str = '') -> BardResponse: + if temperature: prompt = f'''settings: follow these settings for your response: [temperature: {temperature} - {temperatures[temperature]}] | prompt : {prompt}''' - - client = Session() + + client = Session() client.proxies = { 'http': f'http://{proxy}', - 'https': f'http://{proxy}' } if proxy else None + '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={token}' + '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={token}' } - snlm0e = search(r'SNlM0e\":\"(.*?)\"', client.get('https://bard.google.com/').text).group(1) - + snlm0e = search(r'SNlM0e\":\"(.*?)\"', + client.get('https://bard.google.com/').text).group(1) + params = urlencode({ - 'bl' : 'boq_assistant-bard-web-server_20230326.21_p0', - '_reqid' : randint(1111, 9999), - 'rt' : 'c', + 'bl': 'boq_assistant-bard-web-server_20230326.21_p0', + '_reqid': randint(1111, 9999), + 'rt': 'c', }) - response = client.post(f'https://bard.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate?{params}', - data = { - 'at': snlm0e, - 'f.req': dumps([None, dumps([ - [prompt], - None, - [conversation_id, response_id, choice_id], - ]) - ]) - } - ) - + response = client.post(f'https://bard.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate?{params}', + data={ + 'at': snlm0e, + 'f.req': dumps([None, dumps([ + [prompt], + None, + [conversation_id, response_id, choice_id], + ])]) + } + ) + chat_data = loads(response.content.splitlines()[3])[0][2] - if not chat_data: print('error, retrying'); Completion.create(prompt, temperature, conversation_id, response_id, choice_id) + if not chat_data: + print('error, retrying') + Completion.create(prompt, temperature, + conversation_id, response_id, choice_id) json_chat_data = loads(chat_data) results = { - 'content' : json_chat_data[0][0], - 'conversation_id' : json_chat_data[1][0], - 'response_id' : json_chat_data[1][1], - 'factualityQueries' : json_chat_data[3], - 'textQuery' : json_chat_data[2][0] if json_chat_data[2] is not None else '', - 'choices' : [{'id': i[0], 'content': i[1]} for i in json_chat_data[4]], + 'content': json_chat_data[0][0], + 'conversation_id': json_chat_data[1][0], + 'response_id': json_chat_data[1][1], + 'factualityQueries': json_chat_data[3], + 'textQuery': json_chat_data[2][0] if json_chat_data[2] is not None else '', + 'choices': [{'id': i[0], 'content': i[1]} for i in json_chat_data[4]], } - - # self.conversation_id = results['conversation_id'] - # self.response_id = results['response_id'] - # self.choice_id = results['choices'][0]['id'] - # self.reqid += 100000 return BardResponse(results) -- cgit v1.2.3