disable embed when use_as_lib feature enabled (#897)

kczimm · web-flow · commit d7fb281e787a · 2023-08-07T17:30:34.000-05:00
diff --git a/pgml-extension/src/api.rs b/pgml-extension/src/api.rs
@@ -566,6 +566,7 @@ fn load_dataset(
     TableIterator::new(vec![(name, rows)].into_iter())
 }
 
+#[cfg(all(feature = "python", not(feature = "use_as_lib")))]
 #[pg_extern(immutable, parallel_safe, name = "embed")]
 pub fn embed(transformer: &str, text: &str, kwargs: default!(JsonB, "'{}'")) -> Vec<f32> {
     embed_batch(transformer, Vec::from([text]), kwargs)
@@ -574,6 +575,7 @@ pub fn embed(transformer: &str, text: &str, kwargs: default!(JsonB, "'{}'")) ->
         .to_vec()
 }
 
+#[cfg(all(feature = "python", not(feature = "use_as_lib")))]
 #[pg_extern(immutable, parallel_safe, name = "embed")]
 pub fn embed_batch(
     transformer: &str,

Original file line number	Diff line number	Diff line change
`@@ -566,6 +566,7 @@ fn load_dataset(`
`566`	`566`	`TableIterator::new(vec![(name, rows)].into_iter())`
`567`	`567`	`}`
`568`	`568`
	`569`	`+#[cfg(all(feature = "python", not(feature = "use_as_lib")))]`
`569`	`570`	`#[pg_extern(immutable, parallel_safe, name = "embed")]`
`570`	`571`	`pub fn embed(transformer: &str, text: &str, kwargs: default!(JsonB, "'{}'")) -> Vec<f32> {`
`571`	`572`	`embed_batch(transformer, Vec::from([text]), kwargs)`
`@@ -574,6 +575,7 @@ pub fn embed(transformer: &str, text: &str, kwargs: default!(JsonB, "'{}'")) ->`
`574`	`575`	`.to_vec()`
`575`	`576`	`}`
`576`	`577`
	`578`	`+#[cfg(all(feature = "python", not(feature = "use_as_lib")))]`
`577`	`579`	`#[pg_extern(immutable, parallel_safe, name = "embed")]`
`578`	`580`	`pub fn embed_batch(`
`579`	`581`	`transformer: &str,`