From 269e561497dce1cb730de128b89beb65fb531bc4 Mon Sep 17 00:00:00 2001 From: RockChinQ <1010553892@qq.com> Date: Tue, 14 May 2024 22:41:39 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20messages=20=E5=AD=98=E5=9B=9E=20convers?= =?UTF-8?q?ation=20=E5=BA=94=E8=AF=A5=E4=BB=85=E5=9C=A8=E6=88=90=E5=8A=9F?= =?UTF-8?q?=E6=89=A7=E8=A1=8C=E6=9C=AC=E6=AC=A1=E8=AF=B7=E6=B1=82=E6=97=B6?= =?UTF-8?q?=E6=89=A7=E8=A1=8C=20(#769)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/pipeline/process/handlers/chat.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pkg/pipeline/process/handlers/chat.py b/pkg/pipeline/process/handlers/chat.py index 2f0616f3..26f73b61 100644 --- a/pkg/pipeline/process/handlers/chat.py +++ b/pkg/pipeline/process/handlers/chat.py @@ -89,6 +89,9 @@ class ChatMessageHandler(handler.MessageHandler): result_type=entities.ResultType.CONTINUE, new_query=query ) + + query.session.using_conversation.messages.append(query.user_message) + query.session.using_conversation.messages.extend(query.resp_messages) except Exception as e: self.ap.logger.error(f'对话({query.query_id})请求失败: {str(e)}') @@ -101,8 +104,6 @@ class ChatMessageHandler(handler.MessageHandler): debug_notice=traceback.format_exc() ) finally: - query.session.using_conversation.messages.append(query.user_message) - query.session.using_conversation.messages.extend(query.resp_messages) await self.ap.ctr_mgr.usage.post_query_record( session_type=query.session.launcher_type.value,