perft:记忆 (#231)

1:优化异常处理

Co-authored-by: 李彬 <01780391@yto.net.cn>
This commit is contained in:
yeyuezhishou
2025-03-07 18:44:08 +08:00
committed by GitHub
co-authored by 李彬
parent dd5fecdf5a
commit dc4b8a5002
@@ -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: