aboutsummaryrefslogtreecommitdiff
path: root/src/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs4
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)
}