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"] = {