From faa7e9f8f464bceba44d41cbeacc4c8bb48d98fd Mon Sep 17 00:00:00 2001
From: Arnaud Gourlay <arnaud.gourlay@gmail.com>
Date: Fri, 6 Sep 2024 14:53:20 +0200
Subject: [PATCH] Fix distance matrix min samples found (#5036)

---
 lib/collection/src/collection/distance_matrix.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/lib/collection/src/collection/distance_matrix.rs b/lib/collection/src/collection/distance_matrix.rs
index 4ee76e2148..b07b023ca0 100644
--- a/lib/collection/src/collection/distance_matrix.rs
+++ b/lib/collection/src/collection/distance_matrix.rs
@@ -204,6 +204,11 @@ impl Collection {
             sampled_points.extend(filtered);
         }
 
+        // if we have less than 2 points, we can't build a matrix
+        if sampled_points.len() < 2 {
+            return Ok(CollectionSearchMatrixResponse::default());
+        }
+
         sampled_points.truncate(sample_size);
         // sort by id for a deterministic order
         sampled_points.sort_unstable_by(|(id1, _), (id2, _)| id1.cmp(id2));