feat(uni): add for cosine simliarity

unit-mesh · Nov 14, 2023 · 31c8df7 · 31c8df7
1 parent 4bb5198
commit 31c8df7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/inference_core/src/inference.udl b/inference_core/src/inference.udl
@@ -1,6 +1,7 @@
 namespace inference {
     [Throws=SemanticError]
     Semantic init_semantic(sequence<u8> model, sequence<u8> tokenizer_data);
+    Similarity get_cosine_similarity();
 };
 
 [Custom]

diff --git a/inference_core/src/lib.rs b/inference_core/src/lib.rs
@@ -3,6 +3,7 @@ pub mod embedding;
 pub mod similarity;
 pub mod store;
 
+use std::sync::Arc;
 // make the following types available in the generated bindings
 pub use document::Document;
 pub use document::Metadata;
@@ -19,5 +20,8 @@ pub use similarity::RelevanceScore;
 pub use store::EmbeddingStore;
 pub use store::InMemoryEmbeddingStore;
 
+pub fn get_cosine_similarity() -> Arc<dyn Similarity> {
+    Arc::new(CosineSimilarity {})
+}
 
 uniffi::include_scaffolding!("inference");