From f2494cacdc018ee169683abcf2cd216b07ee668d Mon Sep 17 00:00:00 2001 From: George Date: Fri, 11 Oct 2024 20:43:54 +0200 Subject: [PATCH] fix: fix default vector distance matrix (#810) --- qdrant_client/local/local_collection.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/qdrant_client/local/local_collection.py b/qdrant_client/local/local_collection.py index bef5bbee..b0dcba22 100644 --- a/qdrant_client/local/local_collection.py +++ b/qdrant_client/local/local_collection.py @@ -1548,7 +1548,12 @@ class LocalCollection: for sampled_id_index, sampled in enumerate(samples): ids_to_includes = [x for (i, x) in enumerate(ids) if i != sampled_id_index] sampling_filter = _include_ids_in_filter(query_filter, ids_to_includes) - search_vector = sampled.vector[search_in_vector_name] + sampled_vector = sampled.vector + search_vector = ( + sampled_vector[search_in_vector_name] + if isinstance(sampled_vector, dict) + else sampled_vector + ) samples_scores = self.search( query_vector=(search_in_vector_name, search_vector), query_filter=sampling_filter,