From c0b3e0c2d5f72e32818cbc4863d010bb54e46161 Mon Sep 17 00:00:00 2001
From: hleft <89069008+hleft@users.noreply.github.com>
Date: Sat, 11 Mar 2023 20:31:16 +0800
Subject: [PATCH] support system meesage in environment variable (#144)

* support config tags to translate

* support system meesage in envirment
---
 book_maker/translator/chatgptapi_translator.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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 = (