summaryrefslogtreecommitdiffstats
path: root/you
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--you/__init__.py77
1 files changed, 77 insertions, 0 deletions
diff --git a/you/__init__.py b/you/__init__.py
new file mode 100644
index 00000000..dc440b79
--- /dev/null
+++ b/you/__init__.py
@@ -0,0 +1,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,
+ } \ No newline at end of file