fix ordering

2024-12-31 03:18:14 +01:00 · 2024-12-31 03:18:14 +01:00 · 0e5ca89f1d
commit 0e5ca89f1d
parent 38287e77e7
1 changed files with 6 additions and 1 deletions
--- a/src/ai.rs
+++ b/src/ai.rs
@ -263,7 +263,12 @@ impl EmbedStore {
            .flatten()
            .collect::<Vec<SearchResult>>();

-        flat.sort_by(|a, b| b.chunks.len().cmp(&a.chunks.len()));
+        flat.sort_by(|a, b| {
+            b.similarity()
+                .partial_cmp(&a.similarity())
+                .unwrap_or(std::cmp::Ordering::Equal)
+                .then(b.chunks.len().cmp(&a.chunks.len()))
+        });
        flat
    }