Skip to content

Commit 1be0d26

Browse files
authored
fix metadata filter not affect in keyword-search and fulltext-search (#16644)
1 parent c167a1f commit 1be0d26

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

api/core/rag/datasource/retrieval_service.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,7 @@ def retrieve(
9797
all_documents=all_documents,
9898
retrieval_method=retrieval_method,
9999
exceptions=exceptions,
100+
document_ids_filter=document_ids_filter,
100101
)
101102
)
102103
concurrent.futures.wait(futures, timeout=30, return_when=concurrent.futures.ALL_COMPLETED)
@@ -222,6 +223,7 @@ def full_text_index_search(
222223
all_documents: list,
223224
retrieval_method: str,
224225
exceptions: list,
226+
document_ids_filter: Optional[list[str]] = None,
225227
):
226228
with flask_app.app_context():
227229
try:
@@ -231,7 +233,9 @@ def full_text_index_search(
231233

232234
vector_processor = Vector(dataset=dataset)
233235

234-
documents = vector_processor.search_by_full_text(cls.escape_query_for_search(query), top_k=top_k)
236+
documents = vector_processor.search_by_full_text(
237+
cls.escape_query_for_search(query), top_k=top_k, document_ids_filter=document_ids_filter
238+
)
235239
if documents:
236240
if (
237241
reranking_model

api/core/rag/retrieval/dataset_retrieval.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -610,7 +610,11 @@ def _retriever(
610610
if dataset.indexing_technique == "economy":
611611
# use keyword table query
612612
documents = RetrievalService.retrieve(
613-
retrieval_method="keyword_search", dataset_id=dataset.id, query=query, top_k=top_k
613+
retrieval_method="keyword_search",
614+
dataset_id=dataset.id,
615+
query=query,
616+
top_k=top_k,
617+
document_ids_filter=document_ids_filter,
614618
)
615619
if documents:
616620
all_documents.extend(documents)

0 commit comments

Comments
 (0)