From c53786d229140de42ab9398acd84e608b344ca88 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Wed, 26 Feb 2025 19:59:57 +0800
Subject: [PATCH] dataset metadata update

---
 api/core/rag/datasource/retrieval_service.py | 2 ++
 1 file changed, 2 insertions(+)

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):