github/bilingual_book_maker
1
1
Fork 0
mirror of https://github.com/yihong0618/bilingual_book_maker.git synced 2025-06-06 11:35:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

modify: fix CI error

Browse Source
This commit is contained in:
mkXultra 2024-08-14 10:48:29 +09:00
parent dafcdbfd60
commit 15be109c71
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
book_maker/translator/chatgptapi_translator.py
View File

@ -102,7 +102,12 @@ class ChatGPTAPI(Base):
] ]
if self.context_flag: if self.context_flag:
messages.append({"role": "user", "content": "\n".join(self.context_list)}) messages.append({"role": "user", "content": "\n".join(self.context_list)})
messages.append({"role": "assistant", "content": "\n".join(self.context_translated_list)}) messages.append(
{
"role": "assistant",
"content": "\n".join(self.context_translated_list),
}
)
messages.append({"role": "user", "content": content}) messages.append({"role": "user", "content": content})
completion = self.openai_client.chat.completions.create( completion = self.openai_client.chat.completions.create(
@ -129,6 +134,7 @@ class ChatGPTAPI(Base):
self.save_context(text, t_text) self.save_context(text, t_text)
return t_text return t_text
def save_context(self, text, t_text): def save_context(self, text, t_text):
if self.context_paragraph_limit > 0: if self.context_paragraph_limit > 0:
self.context_list.append(text) self.context_list.append(text)