From 10a1e8faa6ad53ffa8c0ff7becbe25972504071c Mon Sep 17 00:00:00 2001
From: Rock Chin <1010553892@qq.com>
Date: Wed, 5 Apr 2023 09:56:27 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=9B=9E=E5=A4=8D=E5=86=85=E5=AE=B9?=
 =?UTF-8?q?=E4=B8=8D=E5=AE=8C=E6=95=B4=E9=97=AE=E9=A2=98=20(#208)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/openai/session.py | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/pkg/openai/session.py b/pkg/openai/session.py
index 6575de6e..52cb5ea9 100644
--- a/pkg/openai/session.py
+++ b/pkg/openai/session.py
@@ -229,13 +229,7 @@ class Session:
 
         # 成功获取，处理回复
         res_test = message
-        res_ans = res_test
-
-        # 去除开头可能的提示
-        res_ans_spt = res_test.split("\n\n")
-        if len(res_ans_spt) > 1:
-            del (res_ans_spt[0])
-            res_ans = '\n\n'.join(res_ans_spt)
+        res_ans = res_test.strip()
 
         # 将此次对话的双方内容加入到prompt中
         self.prompt.append({'role': 'user', 'content': text})