youngkingdom/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-05-03 00:37:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: record user_id in memory (#13585)

Browse Source 
### What problem does this PR solve?

Get user_id from canvas and record it.

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
Lynn
2026-03-13 15:38:35 +08:00
committed by GitHub
parent 5c955a31cc
commit 02070bab2a
5 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/db/joint_services/memory_message_service.py
View File

@ -66,7 +66,7 @@ async def save_to_memory(memory_id: str, message_dict: dict):
"message_type": MemoryType.RAW.name.lower(),
"source_id": 0,
"memory_id": memory_id,
"user_id": "",
"user_id": message_dict.get("user_id", ""),
"agent_id": message_dict["agent_id"],
"session_id": message_dict["session_id"],
"content": f"User Input: {message_dict.get('user_input')}\nAgent Response: {message_dict.get('agent_response')}",
@ -79,7 +79,7 @@ async def save_to_memory(memory_id: str, message_dict: dict):
"message_type": content["message_type"],
"source_id": raw_message_id,
"memory_id": memory_id,
"user_id": "",
"user_id": message_dict.get("user_id", ""),
"agent_id": message_dict["agent_id"],
"session_id": message_dict["session_id"],
"content": content["content"],
@ -121,7 +121,7 @@ async def save_extracted_to_memory_only(memory_id: str, message_dict, source_mes
"message_type": content["message_type"],
"source_id": source_message_id,
"memory_id": memory_id,
"user_id": "",
"user_id": message_dict.get("user_id", ""),
"agent_id": message_dict["agent_id"],
"session_id": message_dict["session_id"],
"content": content["content"],
@ -227,6 +227,7 @@ def query_message(filter_dict: dict, params: dict):
"memory_id": List[str],
"agent_id": optional
"session_id": optional
"user_id": optional
}
:param params: {
"query": question str,
@ -374,7 +375,7 @@ async def queue_save_to_memory_task(memory_ids: list[str], message_dict: dict):
"message_type": MemoryType.RAW.name.lower(),
"source_id": 0,
"memory_id": memory_id,
"user_id": "",
"user_id": message_dict.get("user_id", ""),
"agent_id": message_dict["agent_id"],
"session_id": message_dict["session_id"],
"content": f"User Input: {message_dict.get('user_input')}\nAgent Response: {message_dict.get('agent_response')}",