feat: add support for tool calling to beeai compiler

Signed-off-by: Nick Mitchell <[email protected]>

Author: starpit

pdl-live-react/src-tauri/src/cli/setup.rs

@@ -3,7 +3,7 @@ use ::std::path::Path;
 use serde_json::Value;
 use urlencoding::encode;
 
-use tauri_plugin_cli::CliExt;
+use tauri_plugin_cli::{ArgData, CliExt};
 
 use crate::cli::run;
 use crate::compile;
@@ -18,7 +18,9 @@ pub fn cli(app: &mut tauri::App) -> Result<(), Box<dyn ::std::error::Error>> {
     // `subcommand` is `Option<Box<SubcommandMatches>>` where `SubcommandMatches` is a struct with { name, matches }.
     let Some(subcommand_matches) = app.cli().matches()?.subcommand else {
         if let Some(help) = app.cli().matches()?.args.get("help") {
-            return Err(Box::from(help.value.as_str().or(Some("Internal Error")).unwrap()));
+            return Err(Box::from(
+                help.value.as_str().or(Some("Internal Error")).unwrap(),
+            ));
         } else {
             return Err(Box::from("Internal Error"));
         }
@@ -32,19 +34,29 @@ pub fn cli(app: &mut tauri::App) -> Result<(), Box<dyn ::std::error::Error>> {
 
             match compile_subcommand_matches.name.as_str() {
                 "beeai" => {
-                    let Some(source) = compile_subcommand_matches.matches.args.get("source") else {
-                        return Err(Box::from("Missing source file"));
+                    // TODO this probably fails if the source is a number??
+                    let Some(ArgData {
+                        value: Value::String(source_file_path),
+                        ..
+                    }) = compile_subcommand_matches.matches.args.get("source")
+                    else {
+                        return Err(Box::from("Invalid source file"));
                     };
-                    let Value::String(source_file_path) = &source.value else {
-                        return Err(Box::from("Invalid source file argument"));
+                    let Some(ArgData {
+                        value: Value::String(output_file_path),
+                        ..
+                    }) = compile_subcommand_matches.matches.args.get("output")
+                    else {
+                        return Err(Box::from("Invalid output argument"));
                     };
-                    let Some(output) = compile_subcommand_matches.matches.args.get("output") else {
-                        return Err(Box::from("Missing output argument"));
+                    let Some(ArgData {
+                        value: Value::Bool(debug),
+                        ..
+                    }) = compile_subcommand_matches.matches.args.get("debug")
+                    else {
+                        return Err(Box::from("Invalid debug argument"));
                     };
-                    let Value::String(output_file_path) = &output.value else {
-                        return Err(Box::from("Invalid output file argument"));
-                    };
-                    return compile::beeai::compile(source_file_path, output_file_path);
+                    return compile::beeai::compile(source_file_path, output_file_path, debug);
                 }
                 _ => {}
             }