modify: fix CI error

yihong0618 · Aug 14, 2024 · 15be109 · 15be109
1 parent dafcdbf
commit 15be109
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/book_maker/translator/chatgptapi_translator.py b/book_maker/translator/chatgptapi_translator.py
@@ -102,7 +102,12 @@ def create_chat_completion(self, text):
         ]
         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 @@ def get_translation(self, text):
             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)