From a6a285174fa53161ef3ebba78ac259785287e39d Mon Sep 17 00:00:00 2001
From: Montana Low <montana.low@gmail.com>
Date: Mon, 4 Dec 2023 22:13:00 -0800
Subject: [PATCH] use CLI

---
 pgml-sdks/pgml/src/cli.rs | 5 +++++
 pgml-sdks/pgml/src/lib.rs | 1 +
 2 files changed, 6 insertions(+)

diff --git a/pgml-sdks/pgml/src/cli.rs b/pgml-sdks/pgml/src/cli.rs
index c586b86f6..709e5c1ab 100644
--- a/pgml-sdks/pgml/src/cli.rs
+++ b/pgml-sdks/pgml/src/cli.rs
@@ -160,6 +160,11 @@ pub fn cli(
     Ok(promise)
 }
 
+#[cfg(all(not(feature = "python"), not(feature = "javascript")))]
+pub async fn cli() -> anyhow::Result<()> {
+    cli_internal().await
+}
+
 async fn cli_internal() -> anyhow::Result<()> {
     #[cfg(feature = "python")]
     let subcommand = {
diff --git a/pgml-sdks/pgml/src/lib.rs b/pgml-sdks/pgml/src/lib.rs
index 0c4cd6e81..4fb0b7caf 100644
--- a/pgml-sdks/pgml/src/lib.rs
+++ b/pgml-sdks/pgml/src/lib.rs
@@ -34,6 +34,7 @@ mod utils;
 
 // Re-export
 pub use builtins::Builtins;
+pub use cli::cli;
 pub use collection::Collection;
 pub use model::Model;
 pub use open_source_ai::OpenSourceAI;