From d34ad7343947c5867b5d3c191f796cdc208477df Mon Sep 17 00:00:00 2001
From: lixinyue <2569494688@qq.com>
Date: Wed, 28 Jan 2026 10:56:41 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BA=94=E7=94=A8=E5=B1=82memory=5Fcontent->me?=
 =?UTF-8?q?mory=5Fconfig?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/app/schemas/app_schema.py         |  2 +-
 api/app/services/draft_run_service.py | 12 ++++++------
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/api/app/schemas/app_schema.py b/api/app/schemas/app_schema.py
index 09410091..b350f17c 100644
--- a/api/app/schemas/app_schema.py
+++ b/api/app/schemas/app_schema.py
@@ -51,7 +51,7 @@ class ToolOldConfig(BaseModel):
 class MemoryConfig(BaseModel):
     """记忆配置"""
     enabled: bool = Field(default=True, description="是否启用对话历史记忆")
-    memory_content: Optional[str] = Field(default=None, description="选择记忆的内容类型")
+    memory_config: Optional[str] = Field(default=None, description="选择记忆的内容类型")
     max_history: int = Field(default=10, ge=0, le=100, description="最大保留的历史对话轮数")
 
 
diff --git a/api/app/services/draft_run_service.py b/api/app/services/draft_run_service.py
index 9766eec0..0d1f51a4 100644
--- a/api/app/services/draft_run_service.py
+++ b/api/app/services/draft_run_service.py
@@ -56,7 +56,7 @@ def create_long_term_memory_tool(memory_config: Dict[str, Any], end_user_id: str
         长期记忆工具
     """
     # search_switch = memory_config.get("search_switch", "2")
-    config_id= memory_config.get("memory_content",None)
+    config_id= memory_config.get("memory_content") or memory_config.get("memory_config",None)
     logger.info(f"创建长期记忆工具，配置: end_user_id={end_user_id}, config_id={config_id}, storage_type={storage_type}")
     @tool(args_schema=LongTermMemoryInput)
     def long_term_memory(question: str) -> str:
@@ -106,9 +106,9 @@ def create_long_term_memory_tool(memory_config: Dict[str, Any], end_user_id: str
                     "app.core.memory.agent.read_message",
                     args=[end_user_id, question, [], "1", config_id, storage_type, user_rag_memory_id]
                 )
-                # result = task_service.get_task_memory_read_result(task.id)
-                # status = result.get("status")
-                # logger.info(f"读取任务状态：{status}")
+                result = task_service.get_task_memory_read_result(task.id)
+                status = result.get("status")
+                logger.info(f"读取任务状态：{status}")
 
             finally:
                 db.close()
@@ -418,7 +418,7 @@ class DraftRunService:
             )
 
             memory_config_= agent_config.memory
-            config_id = memory_config_.get("memory_content")
+            config_id = memory_config_.get("memory_content") or memory_config_.get("memory_config",None)
 
             # 7. 调用 Agent
             result = await agent.chat(
@@ -644,7 +644,7 @@ class DraftRunService:
             })
 
             memory_config_ = agent_config.memory
-            config_id = memory_config_.get("memory_content")
+            config_id = memory_config_.get("memory_content") or memory_config_.get("memory_config",None)
 
             # 9. 流式调用 Agent
             full_content = ""