From 559397ffeec0c04ff585a7397c000943480e7dc0 Mon Sep 17 00:00:00 2001 From: "liuchangsheng@wisdomidata.com" Date: Fri, 13 Jun 2025 13:45:04 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90Dify=E3=80=91=20=E5=A2=9E=E5=8A=A0?= =?UTF-8?q?=E5=85=A8=E6=96=87=E6=A3=80=E7=B4=A2=E7=9A=84=E6=8E=A5=E5=8F=A3?= =?UTF-8?q?-=20=20=E6=9F=A5=E8=AF=A2=E6=96=87=E4=BB=B6ID?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- api/services/ext/dataset_ext_service.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/api/services/ext/dataset_ext_service.py b/api/services/ext/dataset_ext_service.py index 0d5f276a4e..3ac217e685 100644 --- a/api/services/ext/dataset_ext_service.py +++ b/api/services/ext/dataset_ext_service.py @@ -214,6 +214,7 @@ class DocumentExtService: search_data = { "title": segment.document_name, "content": segment.segment_content, + "doc_metadata": segment.metadata, "query": query_text } search_datas.append(search_data) @@ -223,7 +224,7 @@ class DocumentExtService: def get_full_search_segments(dataset_ids: list[str], query_text: str): sql = text(""" - SELECT s.id segment_id, s.document_id, s.content segment_content, d.name document_name + SELECT s.id segment_id, s.document_id, s.content segment_content, d.name document_name,d.doc_metadata FROM document_segments s left join documents d on d.id = s.document_id WHERE content ILIKE :keyword and d.dataset_id::text = ANY(:dataset_ids) @@ -236,7 +237,8 @@ class DocumentExtService: SELECT d.id AS document_id, d.name AS document_name, s.id AS segment_id, - s.content AS segment_content + s.content AS segment_content, + d.doc_metadata FROM documents d JOIN ( SELECT s1.*