github/bilingual_book_maker
1
1
Fork 0
mirror of https://github.com/yihong0618/bilingual_book_maker.git synced 2025-07-21 17:40:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

reformat with black

Browse Source
This commit is contained in:
cce 2024-12-06 23:48:41 -05:00
parent fc2d663773
commit acf01251f3
No known key found for this signature in database
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
book_maker/translator/claude_translator.py
View File

@ -43,10 +43,13 @@ class Claude(Base):
def create_messages(self, text, intermediate_messages=None):
"""Create messages for the current translation request"""
current_msg = {"role": "user", "content": self.prompt_template.format(
current_msg = {
"role": "user",
"content": self.prompt_template.format(
text=text,
language=self.language,
)}
),
}
messages = []
if intermediate_messages:
@ -67,12 +70,9 @@ class Claude(Base):
"content": self.prompt_template.format(
text="\n\n".join(self.context_list),
language=self.language,
)
),
},
{
"role": "assistant",
"content": "\n\n".join(self.context_translated_list)
}
{"role": "assistant", "content": "\n\n".join(self.context_translated_list)},
]
def save_context(self, text, t_text):