fix bug with angular distance and floating point error; added cosine distance

1 files changed, 4 insertions, 0 deletions

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)
         }