rust by default (#1213)

montanalow · web-flow · commit a02592d4f2df · 2023-12-04T20:57:40.000-08:00
diff --git a/pgml-dashboard/Cargo.toml b/pgml-dashboard/Cargo.toml
@@ -29,7 +29,7 @@ log = "0.4"
 markdown = "1.0.0-alpha.14"
 num-traits = "0.2"
 once_cell = "1.18"
-pgml = { version = "0.10.0", path = "../pgml-sdks/pgml/" }
+pgml = { path = "../pgml-sdks/pgml/" }
 pgml-components = { path = "../packages/pgml-components" }
 pgvector = { version = "0.2.2", features = [ "sqlx", "postgres" ] }
 rand = "0.8"
diff --git a/pgml-sdks/pgml/src/cli.rs b/pgml-sdks/pgml/src/cli.rs
@@ -36,6 +36,19 @@ struct Javascript {
     subcommand: Subcommands,
 }
 
+/// PostgresML CLI is Rust by default
+#[cfg(all(not(feature = "python"), not(feature = "javascript")))]
+#[derive(Parser, Debug, Clone)]
+#[command(author, version, about, long_about = None, name = "pgml", bin_name = "pgml")]
+struct Rust {
+    /// TODO comment on the necessity of this argument.
+    #[arg(name = "pgmlcli")]
+    pgmlcli: Option<String>,
+
+    #[command(subcommand)]
+    subcommand: Subcommands,
+}
+
 #[derive(Subcommand, Debug, Clone)]
 enum Subcommands {
     /// Connect your PostgresML database to another PostgreSQL database.
@@ -160,6 +173,13 @@ async fn cli_internal() -> anyhow::Result<()> {
         args.subcommand
     };
 
+    // Rust by default
+    #[cfg(all(not(feature = "python"), not(feature = "javascript")))]
+    let subcommand = {
+        let args = Rust::parse();
+        args.subcommand
+    };
+
     match subcommand {
         Subcommands::Connect {
             name,
