github/bilingual_book_maker
1
1
Fork 0
mirror of https://github.com/yihong0618/bilingual_book_maker.git synced 2025-06-05 19:15:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

替 BBM_CHATGPTAPI_SYS_MSG 與 BBM_CHATGPTAPI_USER_MSG_TEMPLATE 加入斷行能力 (#224)

Browse Source
This commit is contained in:
Will 保哥 2023-04-02 17:09:22 +08:00 committed by GitHub
parent ffb76baa21
commit cc9e816c57
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
book_maker/translator/chatgptapi_translator.py
View File

@ -50,8 +50,10 @@ class ChatGPTAPI(Base):
openai.api_key = next(self.keys) openai.api_key = next(self.keys)
def create_chat_completion(self, text): def create_chat_completion(self, text):
content = self.prompt_template.format(text=text, language=self.language) content = self.prompt_template.format(
sys_content = self.system_content or self.prompt_sys_msg text=text, language=self.language, crlf="\n"
)
sys_content = self.system_content or self.prompt_sys_msg.format(crlf="\n")
messages = [ messages = [
{"role": "system", "content": sys_content}, {"role": "system", "content": sys_content},
{"role": "user", "content": content}, {"role": "user", "content": content},