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;