| @@ -328,16 +328,13 @@ class TidbOnQdrantVector(BaseVector): | |||
| ) | |||
| document_ids_filter = kwargs.get("document_ids_filter") | |||
| if document_ids_filter: | |||
| should_conditions = [] | |||
| for document_id_filter in document_ids_filter: | |||
| should_conditions.append( | |||
| if filter.must: | |||
| filter.must.append( | |||
| models.FieldCondition( | |||
| key="metadata.document_id", | |||
| match=models.MatchValue(value=document_id_filter), | |||
| match=models.MatchAny(any=document_ids_filter), | |||
| ) | |||
| ) | |||
| if should_conditions: | |||
| filter.should = should_conditions # type: ignore | |||
| results = self._client.search( | |||
| collection_name=self._collection_name, | |||
| query_vector=query_vector, | |||
| @@ -382,16 +379,13 @@ class TidbOnQdrantVector(BaseVector): | |||
| ) | |||
| document_ids_filter = kwargs.get("document_ids_filter") | |||
| if document_ids_filter: | |||
| should_conditions = [] | |||
| for document_id_filter in document_ids_filter: | |||
| should_conditions.append( | |||
| if scroll_filter.must: | |||
| scroll_filter.must.append( | |||
| models.FieldCondition( | |||
| key="metadata.document_id", | |||
| match=models.MatchValue(value=document_id_filter), | |||
| match=models.MatchAny(any=document_ids_filter), | |||
| ) | |||
| ) | |||
| if should_conditions: | |||
| scroll_filter.should = should_conditions # type: ignore | |||
| response = self._client.scroll( | |||
| collection_name=self._collection_name, | |||
| scroll_filter=scroll_filter, | |||