From 902293ab5278900c903887acfe0a1bd97c19fb5c Mon Sep 17 00:00:00 2001
From: Kevin Zimmerman <4733573+kczimm@users.noreply.github.com>
Date: Mon, 7 Aug 2023 17:29:59 -0500
Subject: [PATCH] disable embed when use_as_lib feature enabled

---
 pgml-extension/src/api.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pgml-extension/src/api.rs b/pgml-extension/src/api.rs
index 0a6418e26..a556a33c3 100644
--- 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,