From a1f60a7024ef36a295dde59200ed9b08b480f832 Mon Sep 17 00:00:00 2001
From: zhuchao <zhuchaowe@163.com>
Date: Mon, 2 Mar 2026 12:39:05 +0000
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=20get=5Fall=5Fmemorie?=
 =?UTF-8?q?s=20=E8=B0=83=E7=94=A8=E6=97=B6=20config=20=E4=B8=BA=20None=20?=
 =?UTF-8?q?=E5=AF=BC=E8=87=B4=E7=9A=84=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

问题：当调用 get_all_memories API 时，没有传递 config 参数，
导致 _create_mem0_instance 中的 config 为 None，
调用 config.get_custom_fact_extraction_prompt() 时抛出 AttributeError。

修复：添加 config 的空值检查，只在 config 存在时才添加 custom_fact_extraction_prompt。

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 agent/mem0_manager.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/agent/mem0_manager.py b/agent/mem0_manager.py
index 0c9cef1..93f30c9 100644
--- a/agent/mem0_manager.py
+++ b/agent/mem0_manager.py
@@ -301,7 +301,6 @@ class Mem0Manager:
         # 设置一个假的 base_url，这样 HuggingFaceEmbedding 就不会加载 SentenceTransformer
         
         config_dict = {
-            "custom_fact_extraction_prompt": config.get_custom_fact_extraction_prompt(),
             "vector_store": {
                 "provider": "pgvector",
                 "config": {
@@ -320,6 +319,10 @@ class Mem0Manager:
             }
         }
 
+        # 添加自定义记忆提取提示词（如果提供了 config）
+        if config is not None:
+            config_dict["custom_fact_extraction_prompt"] = config.get_custom_fact_extraction_prompt()
+
         # 添加 LangChain LLM 配置（如果提供了）
         if config and config.llm_instance is not None:
             config_dict["llm"] = {