Public-Mirror/LunaTranslator
1
0
Fork 0
mirror of https://github.com/HIllya51/LunaTranslator.git synced 2024-12-29 16:44:13 +08:00
Code Issues Projects Releases Wiki Activity

Update sakura.py

Browse Source
This commit is contained in:
恍兮惚兮 2024-08-17 18:31:01 +08:00
parent a9190ec194
commit 819476f612
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
LunaTranslator/LunaTranslator/translator/sakura.py
View File

@ -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: