From 1bbef1f028d155a7d52004478289f4fe31c5143b Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Fri, 7 Mar 2025 10:27:02 +0800
Subject: [PATCH] fix parent-child retrival count

---
 api/core/rag/retrieval/dataset_retrieval.py                     | 2 +-
 .../nodes/knowledge_retrieval/knowledge_retrieval_node.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/core/rag/retrieval/dataset_retrieval.py b/api/core/rag/retrieval/dataset_retrieval.py
index 7332993c27..1a443614ca 100644
--- a/api/core/rag/retrieval/dataset_retrieval.py
+++ b/api/core/rag/retrieval/dataset_retrieval.py
@@ -773,7 +773,7 @@ class DatasetRetrieval:
         metadata_filtering_conditions: MetadataFilteringCondition,
         inputs: dict,
     ) -> dict[str, list[str]]:
-        document_query = db.session.query(Document.id).filter(
+        document_query = db.session.query(Document).filter(
             Document.dataset_id.in_(dataset_ids),
             Document.indexing_status == "completed",
             Document.enabled == True,
diff --git a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
index 3657c073d1..56dc4bc665 100644
--- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
+++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
@@ -284,7 +284,7 @@ class KnowledgeRetrievalNode(LLMNode):
     def _get_metadata_filter_condition(
         self, dataset_ids: list, query: str, node_data: KnowledgeRetrievalNodeData
     ) -> Optional[dict[str, list[str]]]:
-        document_query = db.session.query(Document.id).filter(
+        document_query = db.session.query(Document).filter(
             Document.dataset_id.in_(dataset_ids),
             Document.indexing_status == "completed",
             Document.enabled == True,