From 0e5ca89f1d54f5606abdd57670688673ab24e907 Mon Sep 17 00:00:00 2001
From: JMARyA <jmarya@hydrar.de>
Date: Tue, 31 Dec 2024 03:18:14 +0100
Subject: [PATCH] fix ordering

---
 src/ai.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/ai.rs b/src/ai.rs
index 00ed98c..dc9a060 100644
--- 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
     }