diff options
Diffstat (limited to 'g4f/Provider/You.py')
| -rw-r--r-- | g4f/Provider/You.py | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/g4f/Provider/You.py b/g4f/Provider/You.py index a8de7dec..7ea699a2 100644 --- a/g4f/Provider/You.py +++ b/g4f/Provider/You.py @@ -1,5 +1,6 @@ import re import urllib.parse +import json from curl_cffi import requests @@ -28,8 +29,12 @@ class You(BaseProvider): impersonate="chrome107", ) response.raise_for_status() - yield _parse_output(response.text).encode().decode("unicode_escape") - + + start = 'data: {"youChatToken": ' + for line in response.content.splitlines(): + line = line.decode('utf-8') + if line.startswith(start): + yield json.loads(line[len(start): -1]) def _create_url_param(messages: list[dict[str, str]], history: list[dict[str, str]]): prompt = "" @@ -53,10 +58,4 @@ def _create_header(): return { "accept": "text/event-stream", "referer": "https://you.com/search?fromSearchBar=true&tbm=youchat", - } - - -def _parse_output(output: str) -> str: - regex = r"^data:\s{\"youChatToken\": \"(.*)\"}$" - tokens = [token for token in re.findall(regex, output, re.MULTILINE)] - return "".join(tokens) + }
\ No newline at end of file |