zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-08 00:18:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: display embd_name in list memory api response (#17932)

Browse Source
This commit is contained in:
Lynn
2026-08-06 15:43:07 +08:00
committed by GitHub
parent 405275935f
commit 457830f312
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/apps/services/memory_api_service.py
View File

@@ -20,6 +20,7 @@ from api.db.services.user_service import UserTenantService
from api.db.services.canvas_service import UserCanvasService
from api.db.services.task_service import TaskService
from api.db.joint_services.memory_message_service import get_memory_size_cache, judge_system_prompt_is_default, queue_save_to_memory_task, query_message
from api.db.joint_services.tenant_model_service import get_composite_model_name_by_ids
from api.utils.memory_utils import format_ret_data_from_memory, get_memory_type_human
from api.constants import MEMORY_NAME_LIMIT, MEMORY_SIZE_LIMIT
from memory.services.messages import MessageService
@@ -269,7 +270,8 @@ async def list_memory(filter_params: dict, keywords: str, page: int = 1, page_si
filter_dict["memory_type"] = memory_types
memory_list, count = MemoryService.get_by_filter(filter_dict, keywords, page, page_size)
[memory.update({"memory_type": get_memory_type_human(memory["memory_type"])}) for memory in memory_list]
embd_name_map = get_composite_model_name_by_ids([memory["embd_id"] for memory in memory_list])
[memory.update({"memory_type": get_memory_type_human(memory["memory_type"]), "embd_name": embd_name_map.get(memory["embd_id"], "")}) for memory in memory_list]
memory_list.sort(key=lambda m: m["create_time"], reverse=True)
return {"memory_list": memory_list, "total_count": count}