From 396493ad2be3b3914d702abecb9fec89dff700b5 Mon Sep 17 00:00:00 2001
From: lixinyue <2569494688@qq.com>
Date: Mon, 16 Mar 2026 14:28:42 +0800
Subject: [PATCH] fix/retrieve

---
 .../core/memory/agent/langgraph_graph/nodes/data_nodes.py  | 7 +++++++
 api/app/schemas/memory_agent_schema.py                     | 4 +++-
 api/app/services/draft_run_service.py                      | 2 +-
 3 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/api/app/core/memory/agent/langgraph_graph/nodes/data_nodes.py b/api/app/core/memory/agent/langgraph_graph/nodes/data_nodes.py
index ca08db76..5241ac89 100644
--- a/api/app/core/memory/agent/langgraph_graph/nodes/data_nodes.py
+++ b/api/app/core/memory/agent/langgraph_graph/nodes/data_nodes.py
@@ -1,4 +1,5 @@
 from app.core.memory.agent.utils.llm_tools import ReadState, WriteState
+from app.schemas.memory_agent_schema import AgentMemoryDataset
 
 
 def content_input_node(state: ReadState) -> ReadState:
@@ -17,6 +18,9 @@ def content_input_node(state: ReadState) -> ReadState:
 
     content = state['messages'][0].content if state.get('messages') else ''
     # Return content and maintain all state information
+    for pronoun in AgentMemoryDataset.PRONOUN:
+        content = content.replace(pronoun, AgentMemoryDataset.NAME)
+
     return {"data": content}
 
 
@@ -35,4 +39,7 @@ def content_input_write(state: WriteState) -> WriteState:
 
     content = state['messages'][0].content if state.get('messages') else ''
     # Return content and maintain all state information
+    for pronoun in AgentMemoryDataset.PRONOUN:
+        content = content.replace(pronoun, AgentMemoryDataset.NAME)
+
     return {"data": content}
diff --git a/api/app/schemas/memory_agent_schema.py b/api/app/schemas/memory_agent_schema.py
index 26a7390b..b4efe61d 100644
--- a/api/app/schemas/memory_agent_schema.py
+++ b/api/app/schemas/memory_agent_schema.py
@@ -26,5 +26,7 @@ class AgentMemory_Long_Term(ABC):
     STRATEGY_TIME = "time"
     DEFAULT_SCOPE = 6
     TIME_SCOPE=5
-
+class AgentMemoryDataset(ABC):
+    PRONOUN=['我','本人','在下','自己','咱','鄙人','吴','余']
+    NAME='用户'
 
diff --git a/api/app/services/draft_run_service.py b/api/app/services/draft_run_service.py
index b3b136a1..619a5f10 100644
--- a/api/app/services/draft_run_service.py
+++ b/api/app/services/draft_run_service.py
@@ -98,7 +98,7 @@ def create_long_term_memory_tool(
         **重要：如果用户的问题可以直接回答，不要调用此工具。只在确实需要历史信息时才使用。**
 
         Args:
-            question: 需要检索的问题（保持原问题的核心语义，使用清晰的关键词）
+            question: 需要检索的问题（保持原问题的核心语义，使用清晰的关键词，第三人称描述的偏好、行为通常指用户本人，比如（我，本人，在下，自己，咱，鄙人，吴，余）通指用户）
 
         Returns:
             检索到的历史记忆内容