|
@@ -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,
|