File size: 3,528 Bytes
633bcd8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
from tls_client import Session
from re import findall
from json import loads, dumps
from uuid import uuid4
class Completion:
def create(
prompt : str,
page : int = 1,
count : int = 10,
safeSearch : str = "Moderate",
onShoppingpage : bool = False,
mkt : str = "",
responseFilter : str = "WebPages,Translations,TimeZone,Computation,RelatedSearches",
domain : str = "youchat",
queryTraceId : str = None,
chat : list = [],
includelinks : bool = False,
detailed : bool = False,
debug : bool = False ) -> dict:
client = Session(client_identifier="chrome_108")
client.headers = {
"authority" : "you.com",
"accept" : "text/event-stream",
"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",
"cache-control" : "no-cache",
"referer" : "https://you.com/search?q=who+are+you&tbm=youchat",
"sec-ch-ua" : '"Not_A Brand";v="99", "Google Chrome";v="109", "Chromium";v="109"',
"sec-ch-ua-mobile" : "?0",
"sec-ch-ua-platform": '"Windows"',
"sec-fetch-dest" : "empty",
"sec-fetch-mode" : "cors",
"sec-fetch-site" : "same-origin",
'cookie' : f'safesearch_guest=Moderate; uuid_guest={str(uuid4())}',
"user-agent" : "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36",
}
response = client.get(f"https://you.com/api/streamingSearch", params = {
"q" : prompt,
"page" : page,
"count" : count,
"safeSearch" : safeSearch,
"onShoppingPage" : onShoppingpage,
"mkt" : mkt,
"responseFilter" : responseFilter,
"domain" : domain,
"queryTraceId" : str(uuid4()) if queryTraceId is None else queryTraceId,
"chat" : str(chat), # {"question":"","answer":" '"}
}
)
if debug:
print('\n\n------------------\n\n')
print(response.text)
print('\n\n------------------\n\n')
youChatSerpResults = findall(r'youChatSerpResults\ndata: (.*)\n\nevent', response.text)[0]
thirdPartySearchResults = findall(r"thirdPartySearchResults\ndata: (.*)\n\nevent", response.text)[0]
slots = findall(r"slots\ndata: (.*)\n\nevent", response.text)[0]
text = response.text.split('}]}\n\nevent: youChatToken\ndata: {"youChatToken": "')[-1]
text = text.replace('"}\n\nevent: youChatToken\ndata: {"youChatToken": "', '')
text = text.replace('event: done\ndata: I\'m Mr. Meeseeks. Look at me.\n\n', '')
extra = {
'youChatSerpResults' : loads(youChatSerpResults),
'slots' : loads(slots)
}
return {
'response': text,
'links' : loads(thirdPartySearchResults)['search']["third_party_search_results"] if includelinks else None,
'extra' : extra if detailed else None,
} |