perf: 强化ltm异常处理

Soulter · Jan 27, 2025 · d2f694c · d2f694c
1 parent 7a885b0
commit d2f694c
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/packages/astrbot/main.py b/packages/astrbot/main.py
@@ -423,7 +423,10 @@ async def gewe_logout(self, event: AstrMessageEvent):
     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 @@ async def decorate_llm_req(self, event: AstrMessageEvent, req: ProviderRequest):
                 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):