Merge branch 'feat/support-knowledge-metadata' into deploy/dev

This commit is contained in:
jyong
2025-03-07 10:27:17 +08:00
2 changed files with 2 additions and 2 deletions

View File

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

View File

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