Browse Source

fix milvus filter search (#16725)

Jyong 2 months ago
parent
commit
2174225259
1 changed files with 3 additions and 3 deletions
  1. 3 3
      api/core/rag/datasource/vdb/milvus/milvus_vector.py

+ 3 - 3
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,