fix/retrieve

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}

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='用户'
 

api/app/services/draft_run_service.py

@@ -98,7 +98,7 @@ def create_long_term_memory_tool(
         **重要：如果用户的问题可以直接回答，不要调用此工具。只在确实需要历史信息时才使用。**
 
         Args:
-            question: 需要检索的问题（保持原问题的核心语义，使用清晰的关键词）
+            question: 需要检索的问题（保持原问题的核心语义，使用清晰的关键词，第三人称描述的偏好、行为通常指用户本人，比如（我，本人，在下，自己，咱，鄙人，吴，余）通指用户）
 
         Returns:
             检索到的历史记忆内容