KMS/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix parent-child retrival count

Browse Source
This commit is contained in:
jyong 2025-03-07 10:27:02 +08:00
parent b33063e946
commit 1bbef1f028
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/rag/retrieval/dataset_retrieval.py
View File

@ -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,

2
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View File

@ -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,