Quellcode durchsuchen

fix milvus filter search (#16725)

tags/1.2.0
Jyong vor 7 Monaten
Ursprung
Commit
2174225259
Es ist kein Account mit der E-Mail-Adresse des Committers verbunden
1 geänderte Dateien mit 3 neuen und 3 gelöschten Zeilen
  1. 3
    3
      api/core/rag/datasource/vdb/milvus/milvus_vector.py

+ 3
- 3
api/core/rag/datasource/vdb/milvus/milvus_vector.py Datei anzeigen

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

Laden…
Abbrechen
Speichern