Jyong
7 月之前
+ 3
- 3
api/core/rag/datasource/vdb/milvus/milvus_vector.py
查看文件
| 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) | |||||
| 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( | results = self._client.search( | ||||
| collection_name=self._collection_name, | collection_name=self._collection_name, | ||||
| data=[query_vector], | data=[query_vector], | ||||
| 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, |
Loading…