fix multimodal embedding retrival test

2025-12-26 09:07:28 +00:00 · 2025-12-26 15:01:38 +08:00
1 changed files with 1 additions and 1 deletions
--- a/api/core/rag/datasource/retrieval_service.py
+++ b/api/core/rag/datasource/retrieval_service.py
@@ -429,7 +429,7 @@ class RetrievalService:

                for attachment in attachments:
                    segment_ids.append(attachment["segment_id"])
-                    attachment_map[attachment["segment_id"]] = attachment
+                    attachment_map[attachment["segment_id"]] = attachment["attachment_info"]
                    doc_segment_map[attachment["segment_id"]] = attachment["attachment_id"]

                child_chunk_stmt = select(ChildChunk).where(ChildChunk.index_node_id.in_(child_index_node_ids))