diff --git a/book_maker/translator/chatgptapi_translator.py b/book_maker/translator/chatgptapi_translator.py index dfdfdbc..0dbd6c4 100644 --- a/book_maker/translator/chatgptapi_translator.py +++ b/book_maker/translator/chatgptapi_translator.py @@ -1,6 +1,7 @@ import time import openai +from os import environ from .base_translator import Base @@ -20,10 +21,14 @@ class ChatGPTAPI(Base): completion = openai.ChatCompletion.create( model="gpt-3.5-turbo", messages=[ + { + "role": "system", + "content": environ.get("OPENAI_API_SYS_MSG"), + }, { "role": "user", "content": f"Please help me to translate,`{text}` to {self.language}, please return only translated content not include the origin text", - } + }, ], ) t_text = (