diff --git a/README.md b/README.md index 15fd897..7f093d2 100644 --- a/README.md +++ b/README.md @@ -26,6 +26,18 @@ for token in chat_completion: ``` ### simple usage: + +providers: +```py +g4f.Providers.You +g4f.Providers.Ails +g4f.Providers.Phind + +# usage: + +response = g4f.ChatCompletion.create(..., provider=g4f.Providers.ProviderName) +``` + ```py import g4f diff --git a/g4f/Providers/Test.py b/g4f/Providers/Test.py deleted file mode 100644 index 153fa60..0000000 --- a/g4f/Providers/Test.py +++ /dev/null @@ -1,2 +0,0 @@ -def _create_completion(prompt: str, **kwargs): - return 'helloooo', prompt \ No newline at end of file diff --git a/g4f/Providers/You.py b/g4f/Providers/You.py new file mode 100644 index 0000000..a4b33d7 --- /dev/null +++ b/g4f/Providers/You.py @@ -0,0 +1,20 @@ +import os +import json +import time +import subprocess + +url = 'https://you.com' +model = 'gpt-3.5-turbo' + +def _create_completion(model: str, messages: list, **kwargs): + + path = os.path.dirname(os.path.realpath(__file__)) + config = json.dumps({ + 'messages': messages}, separators=(',', ':')) + + cmd = ['python3', f'{path}/helpers/you.py', config] + + p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) + + for line in iter(p.stdout.readline, b''): + yield line.decode('utf-8')[:-1] \ No newline at end of file diff --git a/g4f/Providers/__init__.py b/g4f/Providers/__init__.py index 34ed9b1..29bf9d7 100644 --- a/g4f/Providers/__init__.py +++ b/g4f/Providers/__init__.py @@ -1 +1 @@ -from . import Ails, Phind, Test, Provider \ No newline at end of file +from . import Ails, Phind, Provider, You \ No newline at end of file diff --git a/g4f/Providers/helpers/you.py b/g4f/Providers/helpers/you.py new file mode 100644 index 0000000..86a6fa7 --- /dev/null +++ b/g4f/Providers/helpers/you.py @@ -0,0 +1,71 @@ +import sys +import json +import urllib.parse + +from curl_cffi import requests + +config = json.loads(sys.argv[1]) +messages = config['messages'] +prompt = '' + + +def transform(messages: list) -> list: + result = [] + i = 0 + + while i < len(messages): + if messages[i]['role'] == 'user': + question = messages[i]['content'] + i += 1 + + if i < len(messages) and messages[i]['role'] == 'assistant': + answer = messages[i]['content'] + i += 1 + else: + answer = '' + + result.append({'question': question, 'answer': answer}) + + elif messages[i]['role'] == 'assistant': + result.append({'question': '', 'answer': messages[i]['content']}) + i += 1 + + elif messages[i]['role'] == 'system': + result.append({'question': messages[i]['content'], 'answer': ''}) + i += 1 + + return result + +headers = { + 'Content-Type': 'application/x-www-form-urlencoded', + 'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8', + 'Sec-Fetch-Site': 'same-origin', + 'Accept-Language': 'en-GB,en;q=0.9', + 'Sec-Fetch-Mode': 'navigate', + 'Host': 'you.com', + 'Origin': 'https://you.com', + 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.4 Safari/605.1.15', + 'Referer': 'https://you.com/api/streamingSearch?q=nice&safeSearch=Moderate&onShoppingPage=false&mkt=&responseFilter=WebPages,Translations,TimeZone,Computation,RelatedSearches&domain=youchat&queryTraceId=7a6671f8-5881-404d-8ea3-c3f8301f85ba&chat=%5B%7B%22question%22%3A%22hi%22%2C%22answer%22%3A%22Hello!%20How%20can%20I%20assist%20you%20today%3F%22%7D%5D&chatId=7a6671f8-5881-404d-8ea3-c3f8301f85ba&__cf_chl_tk=ex2bw6vn5vbLsUm8J5rDYUC0Bjzc1XZqka6vUl6765A-1684108495-0-gaNycGzNDtA', + 'Connection': 'keep-alive', + 'Sec-Fetch-Dest': 'document', + 'Priority': 'u=0, i', +} + +if messages[-1]['role'] == 'user': + prompt = messages[-1]['content'] + messages = messages[:-1] + +params = urllib.parse.urlencode({ + 'q': prompt, + 'domain': 'youchat', + 'chat': transform(messages) +}) + +def output(chunk): + if b'"youChatToken"' in chunk: + chunk_json = json.loads(chunk.decode().split('data: ')[1]) + + print(chunk_json['youChatToken'], flush=True) + +response = requests.get(f'https://you.com/api/streamingSearch?{params}', + headers=headers, content_callback=output, impersonate='safari15_5')