From 7222d9b7cea4db173258c1189f5069b544606eeb Mon Sep 17 00:00:00 2001
From: zengrr <47846202+zeng-rr@users.noreply.github.com>
Date: Wed, 23 Aug 2023 15:26:00 +0800
Subject: "You" unicode decode and fix prompt just is last message

---
 g4f/Provider/You.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

(limited to 'g4f/Provider')

diff --git a/g4f/Provider/You.py b/g4f/Provider/You.py
index 0d8114a8..a8de7dec 100644
--- a/g4f/Provider/You.py
+++ b/g4f/Provider/You.py
@@ -19,7 +19,7 @@ class You(BaseProvider):
         stream: bool,
         **kwargs: Any,
     ) -> CreateResult:
-        url_param = _create_url_param(messages)
+        url_param = _create_url_param(messages, kwargs.get("history", []))
         headers = _create_header()
         url = f"https://you.com/api/streamingSearch?{url_param}"
         response = requests.get(
@@ -28,12 +28,15 @@ class You(BaseProvider):
             impersonate="chrome107",
         )
         response.raise_for_status()
-        yield _parse_output(response.text)
+        yield _parse_output(response.text).encode().decode("unicode_escape")
 
 
-def _create_url_param(messages: list[dict[str, str]]):
-    prompt = messages.pop()["content"]
-    chat = _convert_chat(messages)
+def _create_url_param(messages: list[dict[str, str]], history: list[dict[str, str]]):
+    prompt = ""
+    for message in messages:
+        prompt += "%s: %s\n" % (message["role"], message["content"])
+    prompt += "assistant:"
+    chat = _convert_chat(history)
     param = {"q": prompt, "domain": "youchat", "chat": chat}
     return urllib.parse.urlencode(param)
 
-- 
cgit v1.2.3