From 15be109c71d6b5b3c6feb0b1cf798338e8ce871f Mon Sep 17 00:00:00 2001 From: mkXultra Date: Wed, 14 Aug 2024 10:48:29 +0900 Subject: [PATCH] modify: fix CI error --- book_maker/translator/chatgptapi_translator.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/book_maker/translator/chatgptapi_translator.py b/book_maker/translator/chatgptapi_translator.py index 5326719..94ecfae 100644 --- a/book_maker/translator/chatgptapi_translator.py +++ b/book_maker/translator/chatgptapi_translator.py @@ -102,7 +102,12 @@ class ChatGPTAPI(Base): ] if self.context_flag: 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}) completion = self.openai_client.chat.completions.create( @@ -129,6 +134,7 @@ class ChatGPTAPI(Base): self.save_context(text, t_text) return t_text + def save_context(self, text, t_text): if self.context_paragraph_limit > 0: self.context_list.append(text)