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

fix metadata

Browse Source
This commit is contained in:
jyong 2025-03-10 15:06:53 +08:00
parent 6b8c84eff3
commit d30e16c4c3
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -773,11 +773,11 @@ class DatasetRetrieval:
metadata_filtering_conditions: Optional[MetadataFilteringCondition], metadata_filtering_conditions: Optional[MetadataFilteringCondition],
inputs: dict, inputs: dict,
) -> Optional[dict[str, list[str]]]: ) -> Optional[dict[str, list[str]]]:
document_query = db.session.query(Document).filter( document_query = db.session.query(DatasetDocument).filter(
Document.dataset_id.in_(dataset_ids), DatasetDocument.dataset_id.in_(dataset_ids),
Document.indexing_status == "completed", DatasetDocument.indexing_status == "completed",
Document.enabled == True, DatasetDocument.enabled == True,
Document.archived == False, DatasetDocument.archived == False,
) )
if metadata_filtering_mode == "disabled": if metadata_filtering_mode == "disabled":
return None return None
@ -821,7 +821,7 @@ class DatasetRetrieval:
) -> list[dict[str, Any]]: ) -> list[dict[str, Any]]:
# get all metadata field # get all metadata field
metadata_fields = db.session.query(DatasetMetadata).filter(DatasetMetadata.dataset_id.in_(dataset_ids)).all() metadata_fields = db.session.query(DatasetMetadata).filter(DatasetMetadata.dataset_id.in_(dataset_ids)).all()
all_metadata_fields = [metadata_field.field_name for metadata_field in metadata_fields] all_metadata_fields = [metadata_field.name for metadata_field in metadata_fields]
# get metadata model config # get metadata model config
if metadata_model_config is None: if metadata_model_config is None:
raise ValueError("metadata_model_config is required") raise ValueError("metadata_model_config is required")