|
@@ -97,6 +97,7 @@ class RetrievalService:
|
|
|
all_documents=all_documents,
|
|
|
retrieval_method=retrieval_method,
|
|
|
exceptions=exceptions,
|
|
|
+ document_ids_filter=document_ids_filter,
|
|
|
)
|
|
|
)
|
|
|
concurrent.futures.wait(futures, timeout=30, return_when=concurrent.futures.ALL_COMPLETED)
|
|
@@ -222,6 +223,7 @@ class RetrievalService:
|
|
|
all_documents: list,
|
|
|
retrieval_method: str,
|
|
|
exceptions: list,
|
|
|
+ document_ids_filter: Optional[list[str]] = None,
|
|
|
):
|
|
|
with flask_app.app_context():
|
|
|
try:
|
|
@@ -231,7 +233,9 @@ class RetrievalService:
|
|
|
|
|
|
vector_processor = Vector(dataset=dataset)
|
|
|
|
|
|
- documents = vector_processor.search_by_full_text(cls.escape_query_for_search(query), top_k=top_k)
|
|
|
+ documents = vector_processor.search_by_full_text(
|
|
|
+ cls.escape_query_for_search(query), top_k=top_k, document_ids_filter=document_ids_filter
|
|
|
+ )
|
|
|
if documents:
|
|
|
if (
|
|
|
reranking_model
|