From e8ab616a5dabe2ae8d77772466f92410ebee1048 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Mon, 9 Dec 2024 18:19:38 +0100
Subject: fix bug with angular distance and floating point error; added cosine
 distance

---
 src/main.rs | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index 3055768..1cc380f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -26,6 +26,7 @@ enum Embedder {
     Color,
     ContentEuclidean,
     ContentAngularDistance,
+    ContentCosine,
     ContentManhatten,
 }
 
@@ -230,6 +231,9 @@ fn main() -> Result<()> {
         Embedder::ContentAngularDistance => {
             process_embedder(ContentEmbedder::<AngularDistance>::new(&cfg), &args, &cfg)
         }
+        Embedder::ContentCosine => {
+            process_embedder(ContentEmbedder::<CosineDistance>::new(&cfg), &args, &cfg)
+        }
         Embedder::ContentEuclidean => {
             process_embedder(ContentEmbedder::<EuclidianDistance>::new(&cfg), &args, &cfg)
         }
-- 
cgit v1.2.3-70-g09d2