Function main

cake-cli/src/main.rs:96–239

()

#[tokio::main]
async fn main() -> Result<()> {
    let cli = Cli::parse();

    // setup logging
    if std::env::var_os("RUST_LOG").is_none() {
        // set `RUST_LOG=debug` to see debug logs
        std::env::set_var("RUST_LOG", "info,tokenizers=error,actix_server=warn");
    }

    env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info"))
        .format_module_path(false)
        .format_target(false)
        .init();

    match cli.command {
        Commands::Run { model, prompt, mut args } => {
            match (model, args.cluster_key.is_some()) {
                // No model + cluster key → worker mode
                (None, true) => {
                    args.mode = Mode::Worker;
                    run_as_worker(args).await
                }
                // No model, no cluster key → error
                (None, false) => {
                    anyhow::bail!(
                        "model is required. Usage:\n  \
                         cake run <model> [prompt]           # local inference\n  \
                         cake run <model> --cluster-key K    # distributed master\n  \
                         cake run --cluster-key K            # distributed worker"
                    );
                }
                // Model specified → master mode (local or distributed)
                (Some(m), _) => {
                    args.model = m;
                    if let Some(p) = prompt {
                        args.prompt = p;
                    }
                    args.mode = Mode::Master;
                    run_as_master(args).await
                }
            }
        }
        Commands::Serve { model, address, mut args } => {
            args.model = model;
            args.api = Some(address);
            args.mode = Mode::Master;
            run_as_master(args).await
        }
        Commands::List => {
            let models = utils::models::list_models()?;
            if models.is_empty() {
                println!("No models found.");
                println!();
                println!("Download a model with:");
                println!("  cake pull <org/model-name>");
            } else {
                println!(
                    "{:<50} {:<15} {:<15} SOURCE",