Skip to content

Commit 52593c1

Browse files
committed
fix: fix default vector distance matrix (#810)
1 parent 4a2c0ab commit 52593c1

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

qdrant_client/local/local_collection.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1548,7 +1548,12 @@ def _search_distance_matrix(
15481548
for sampled_id_index, sampled in enumerate(samples):
15491549
ids_to_includes = [x for (i, x) in enumerate(ids) if i != sampled_id_index]
15501550
sampling_filter = _include_ids_in_filter(query_filter, ids_to_includes)
1551-
search_vector = sampled.vector[search_in_vector_name]
1551+
sampled_vector = sampled.vector
1552+
search_vector = (
1553+
sampled_vector[search_in_vector_name]
1554+
if isinstance(sampled_vector, dict)
1555+
else sampled_vector
1556+
)
15521557
samples_scores = self.search(
15531558
query_vector=(search_in_vector_name, search_vector),
15541559
query_filter=sampling_filter,

0 commit comments

Comments
 (0)