From 3589a5e5be068a158b472bd28603ab7a8cb3bcc7 Mon Sep 17 00:00:00 2001 From: Soulter <905617992@qq.com> Date: Mon, 27 Jan 2025 21:47:35 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=E5=BC=BA=E5=8C=96ltm=E5=BC=82=E5=B8=B8?= =?UTF-8?q?=E5=A4=84=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- packages/astrbot/main.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/packages/astrbot/main.py b/packages/astrbot/main.py index d3f80d06d..b52a6e824 100644 --- a/packages/astrbot/main.py +++ b/packages/astrbot/main.py @@ -423,7 +423,10 @@ UID: {user_id} 此 ID 可用于设置管理员。/op 授权管理员, /deo async def on_message(self, event: AstrMessageEvent): '''长期记忆''' if self.ltm: - await self.ltm.handle_message(event) + try: + await self.ltm.handle_message(event) + except BaseException as e: + logger.error(e) @filter.on_llm_request() @@ -452,14 +455,20 @@ UID: {user_id} 此 ID 可用于设置管理员。/op 授权管理员, /deo req.contexts[:0] = begin_dialogs if self.ltm: - await self.ltm.on_req_llm(event, req) + try: + await self.ltm.on_req_llm(event, req) + except BaseException as e: + logger.error(f"ltm: {e}") @filter.after_message_sent() async def after_llm_req(self, event: AstrMessageEvent): '''在 LLM 请求后记录对话''' if self.ltm: - await self.ltm.after_req_llm(event) + try: + await self.ltm.after_req_llm(event) + except BaseException as e: + logger.error(f"ltm: {e}") # @filter.command_group("kdb") # def kdb(self):