estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix milvus filter search (#16725)

Browse Source
This commit is contained in:
Jyong
2025-03-25 16:22:43 +08:00
committed by GitHub
parent 6a0f1dad7f
commit 2174225259
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/rag/datasource/vdb/milvus/milvus_vector.py
View File

@@ -231,8 +231,8 @@ class MilvusVector(BaseVector):
document_ids_filter = kwargs.get("document_ids_filter") document_ids_filter = kwargs.get("document_ids_filter")
filter = "" filter = ""
if document_ids_filter: if document_ids_filter:
document_ids = ", ".join(f"'{id}'" for id in 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( results = self._client.search(
collection_name=self._collection_name, collection_name=self._collection_name,
data=[query_vector], data=[query_vector],
@@ -259,7 +259,7 @@ class MilvusVector(BaseVector):
filter = "" filter = ""
if document_ids_filter: if document_ids_filter:
document_ids = ", ".join(f"'{id}'" for id in 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( results = self._client.search(
collection_name=self._collection_name, collection_name=self._collection_name,