build fix

tembo-io · Oct 20, 2024 · c28b5d3 · c28b5d3
1 parent d29cfbb
commit c28b5d3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/extension/src/api.rs b/extension/src/api.rs
@@ -105,7 +105,7 @@ fn init_rag(
         columns,
         unique_record_id,
         None,
-        index_dist_type,
+        index_dist_type.into(),
         &transformer_model,
         table_method.into(),
         schedule,

diff --git a/extension/src/types.rs b/extension/src/types.rs
@@ -1,5 +1,7 @@
 use pgrx::*;
-use vectorize_core::types::{IndexDist as CoreIndexDist, TableMethod as CoreTableMethod};
+use vectorize_core::types::{
+    IndexDist as CoreIndexDist, SimilarityAlg as CoreSimilarityAlg, TableMethod as CoreTableMethod,
+};
 
 use serde::{Deserialize, Serialize};
 pub const VECTORIZE_SCHEMA: &str = "vectorize";