From 6f2d7d96d028486d1aae18b155b541f58210c630 Mon Sep 17 00:00:00 2001 From: RockChinQ <1010553892@qq.com> Date: Thu, 1 Feb 2024 16:43:44 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=E5=AE=8C=E5=96=84=E5=8E=86=E5=8F=B2?= =?UTF-8?q?=E6=B6=88=E6=81=AF=E5=A4=84=E7=90=86=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/pipeline/process/handlers/chat.py | 28 +++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/pkg/pipeline/process/handlers/chat.py b/pkg/pipeline/process/handlers/chat.py index bcacf939..b57de98f 100644 --- a/pkg/pipeline/process/handlers/chat.py +++ b/pkg/pipeline/process/handlers/chat.py @@ -62,27 +62,27 @@ class ChatMessageHandler(handler.MessageHandler): query.user_message ) - query.session.using_conversation.messages.append( - query.user_message - ) - text_length = 0 start_time = time.time() - async for result in query.use_model.requester.request(query): - query.resp_messages.append(result) + try: - # 消息同步到会话 - query.session.using_conversation.messages.append(result) + async for result in query.use_model.requester.request(query): + query.resp_messages.append(result) - if result.content is not None: - text_length += len(result.content) + if result.content is not None: + text_length += len(result.content) - yield entities.StageProcessResult( - result_type=entities.ResultType.CONTINUE, - new_query=query - ) + yield entities.StageProcessResult( + result_type=entities.ResultType.CONTINUE, + new_query=query + ) + finally: + query.session.using_conversation.messages.append(query.user_message) + query.session.using_conversation.messages.extend(query.resp_messages) + + print(query.session.using_conversation.messages) await self.ap.ctr_mgr.usage.post_query_record( session_type=query.session.launcher_type.value,