From dc4b8a5002eec3365bc0c179c4ad091e24522024 Mon Sep 17 00:00:00 2001
From: yeyuezhishou <626732147@qq.com>
Date: Fri, 7 Mar 2025 18:44:08 +0800
Subject: [PATCH] =?UTF-8?q?perft:=E8=AE=B0=E5=BF=86=20(#231)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

1：优化异常处理

Co-authored-by: 李彬 <01780391@yto.net.cn>
---
 .../core/providers/memory/mem0ai/mem0ai.py             | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/main/xiaozhi-server/core/providers/memory/mem0ai/mem0ai.py b/main/xiaozhi-server/core/providers/memory/mem0ai/mem0ai.py
index a904a02d..9a82efe5 100644
--- a/main/xiaozhi-server/core/providers/memory/mem0ai/mem0ai.py
+++ b/main/xiaozhi-server/core/providers/memory/mem0ai/mem0ai.py
@@ -1,3 +1,5 @@
+import traceback
+
 from ..base import MemoryProviderBase, logger
 from mem0 import MemoryClient
 from core.utils.util import check_model_key
@@ -15,7 +17,13 @@ class MemoryProvider(MemoryProviderBase):
             return
         else:
             self.use_mem0 = True
-        self.client = MemoryClient(api_key=self.api_key)
+        try:
+            self.client = MemoryClient(api_key=self.api_key)
+            logger.bind(tag=TAG).info("成功连接到 Mem0ai 服务")
+        except Exception as e:
+            logger.bind(tag=TAG).error(f"连接到 Mem0ai 服务时发生错误: {str(e)}")
+            logger.bind(tag=TAG).error(f"详细错误: {traceback.format_exc()}")
+            self.use_mem0 = False
 
     async def save_memory(self, msgs):
         if not self.use_mem0: