From 2234e9db0ebf3e3a28ee62d55ee03e6429c80680 Mon Sep 17 00:00:00 2001
From: LINSTCL <linstcl@qq.com>
Date: Thu, 2 Mar 2023 23:25:42 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=AF=B9=E8=AF=9D=E6=8B=BC?=
 =?UTF-8?q?=E6=8E=A5=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/openai/modelmgr.py | 4 ++--
 pkg/openai/session.py  | 5 +----
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/pkg/openai/modelmgr.py b/pkg/openai/modelmgr.py
index f8263ad7..66b3c080 100644
--- a/pkg/openai/modelmgr.py
+++ b/pkg/openai/modelmgr.py
@@ -94,10 +94,10 @@ class CompletionModel(ModelRequest):
     def msg_handle(self, msgs):
         prompt = ''
         for msg in msgs:
-            if msg['role'] == '':
+            if msg['role'] == 'assistant':
                 prompt = prompt + "{}\n".format(msg['content'])
             else:
-                prompt = prompt + "{}:{}\n".format(msg['role'] if msg['role']!='system' else '你的回答要遵守此规则', msg['content'])
+                prompt = prompt + "{}:{}\n".format(msg['role'] , msg['content'])
         return prompt
     
     def get_text(self):
diff --git a/pkg/openai/session.py b/pkg/openai/session.py
index 28e7611a..81c8cbe4 100644
--- a/pkg/openai/session.py
+++ b/pkg/openai/session.py
@@ -210,10 +210,7 @@ class Session:
             del (res_ans_spt[0])
             res_ans = '\n\n'.join(res_ans_spt)
 
-        if config.completion_api_params['model'] in pkg.openai.modelmgr.CHAT_COMPLETION_MODELS:
-            self.prompt.append({'role':'assistant', 'content':res_ans})
-        elif config.completion_api_params['model'] in pkg.openai.modelmgr.COMPLETION_MODELS:
-            self.prompt.append({'role':'', 'content':res_ans})
+        self.prompt.append({'role':'assistant', 'content':res_ans})
 
         if self.just_switched_to_exist_session:
             self.just_switched_to_exist_session = False