diff --git a/api/core/rag/datasource/vdb/milvus/milvus_vector.py b/api/core/rag/datasource/vdb/milvus/milvus_vector.py index a1180a650..7a3319f4a 100644 --- a/api/core/rag/datasource/vdb/milvus/milvus_vector.py +++ b/api/core/rag/datasource/vdb/milvus/milvus_vector.py @@ -231,8 +231,8 @@ class MilvusVector(BaseVector): document_ids_filter = kwargs.get("document_ids_filter") filter = "" if document_ids_filter: - document_ids = ", ".join(f"'{id}'" for id in document_ids_filter) - filter = f'metadata["document_id"] in ({document_ids})' + document_ids = ", ".join(f'"{id}"' for id in document_ids_filter) + filter = f'metadata["document_id"] in [{document_ids}]' results = self._client.search( collection_name=self._collection_name, data=[query_vector], @@ -259,7 +259,7 @@ class MilvusVector(BaseVector): filter = "" if document_ids_filter: document_ids = ", ".join(f"'{id}'" for id in document_ids_filter) - filter = f'metadata["document_id"] in ({document_ids})' + filter = f'metadata["document_id"] in [{document_ids}]' results = self._client.search( collection_name=self._collection_name,