From 819476f612a8ba49122b345994c913c6bad80203 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=81=8D=E5=85=AE=E6=83=9A=E5=85=AE?=
 <101191390+HIllya51@users.noreply.github.com>
Date: Sat, 17 Aug 2024 18:31:01 +0800
Subject: [PATCH] Update sakura.py

---
 .../LunaTranslator/translator/sakura.py       | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/LunaTranslator/LunaTranslator/translator/sakura.py b/LunaTranslator/LunaTranslator/translator/sakura.py
index bc242fd2..df16c0d6 100644
--- a/LunaTranslator/LunaTranslator/translator/sakura.py
+++ b/LunaTranslator/LunaTranslator/translator/sakura.py
@@ -54,11 +54,15 @@ class TS(basetrans):
             src = gpt["src"]
             if self.needzhconv:
                 dst = zhconv.convert(gpt["dst"], "zh-hans")
-                info = zhconv.convert(gpt["info"], "zh-hans") if "info" in gpt.keys() else None
+                info = (
+                    zhconv.convert(gpt["info"], "zh-hans")
+                    if "info" in gpt.keys()
+                    else None
+                )
             else:
                 dst = gpt["dst"]
                 info = gpt["info"] if "info" in gpt.keys() else None
-            
+
             if info:
                 single = f"{src}->{dst} #{info}"
             else:
@@ -118,6 +122,7 @@ class TS(basetrans):
             )
             # print(content)
             messages.append({"role": "user", "content": content})
+        print(messages)
         return messages
 
     def send_request(self, query, is_test=False, **kwargs):
@@ -293,7 +298,9 @@ class TS(basetrans):
             # output_text = output["choices"][0]["message"]["content"]
 
             if bool(self.config["流式输出"]) == True:
-                output = self.send_request_stream(query, history_zh=history_prompt)
+                output = self.send_request_stream(
+                    query, history_zh=history_prompt, gpt_dict=gpt_dict
+                )
                 completion_tokens = 0
                 output_text = ""
                 for o in output:
@@ -305,7 +312,9 @@ class TS(basetrans):
                     else:
                         finish_reason = o["choices"][0]["finish_reason"]
             else:
-                output = self.send_request(query, history_zh=history_prompt)
+                output = self.send_request(
+                    query, history_zh=history_prompt, gpt_dict=gpt_dict
+                )
                 for o in output:
                     completion_tokens = o["usage"]["completion_tokens"]
                     output_text = o["choices"][0]["message"]["content"]
@@ -324,6 +333,7 @@ class TS(basetrans):
                             query,
                             history_zh=history_prompt,
                             frequency_penalty=frequency_penalty,
+                            gpt_dict=gpt_dict,
                         )
                         completion_tokens = 0
                         output_text = ""
@@ -341,6 +351,7 @@ class TS(basetrans):
                             query,
                             history_zh=history_prompt,
                             frequency_penalty=frequency_penalty,
+                            gpt_dict=gpt_dict,
                         )
                         yield "\0"
                         for o in output: