diff --git a/api/core/rag/datasource/retrieval_service.py b/api/core/rag/datasource/retrieval_service.py index bcfcd83ca0..532a7e8464 100644 --- a/api/core/rag/datasource/retrieval_service.py +++ b/api/core/rag/datasource/retrieval_service.py @@ -66,6 +66,7 @@ class RetrievalService: top_k=top_k, all_documents=all_documents, exceptions=exceptions, + document_ids_filter=document_ids_filter, ) ) if RetrievalMethod.is_support_semantic_search(retrieval_method): @@ -81,6 +82,7 @@ class RetrievalService: all_documents=all_documents, retrieval_method=retrieval_method, exceptions=exceptions, + document_ids_filter=document_ids_filter, ) ) if RetrievalMethod.is_support_fulltext_search(retrieval_method):