hanzo 0.1.0

Hanzo AI — Rust SDK and CLI for the Hanzo ecosystem
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

//! The `hanzo` command-line tool — a thin shell over the [`hanzo`] SDK.

use clap::{Parser, Subcommand};

/// Hanzo AI — one CLI for the whole ecosystem.
#[derive(Parser)]
#[command(name = "hanzo", version, about = "Hanzo AI — one CLI + SDK for the whole ecosystem")]
struct Cli {
    /// Engine/base URL (env: HANZO_BASE_URL)
    #[arg(long, global = true, env = "HANZO_BASE_URL")]
    base_url: Option<String>,

    #[command(subcommand)]
    command: Command,
}

#[derive(Subcommand)]
enum Command {
    /// Chat with a model (engine)
    Chat {
        #[arg(short, long, default_value = "default")]
        model: String,
        /// The prompt
        prompt: String,
    },
    /// Generate an embedding (engine)
    Embed {
        #[arg(short, long, default_value = "default")]
        model: String,
        text: String,
    },
    /// List available models (engine)
    Models,
}

fn build_client(cli: &Cli) -> hanzo::Client {
    let mut client = match &cli.base_url {
        Some(b) => hanzo::Client::new(b.clone()),
        None => hanzo::Client::from_env(),
    };
    if let Ok(key) = std::env::var("HANZO_API_KEY") {
        client = client.with_api_key(key);
    }
    client
}

fn run(cli: Cli) -> hanzo::Result<()> {
    let client = build_client(&cli);
    match cli.command {
        Command::Chat { model, prompt } => {
            let resp = client.engine().chat(&model, &[hanzo::msg("user", prompt)])?;
            println!("{}", resp.text());
        }
        Command::Embed { model, text } => {
            let resp = client.engine().embeddings(&model, vec![text])?;
            if let Some(e) = resp.data.first() {
                let preview: Vec<String> =
                    e.embedding.iter().take(6).map(|x| format!("{x:.4}")).collect();
                println!("dim={} [{}, ...]", e.embedding.len(), preview.join(", "));
            }
        }
        Command::Models => {
            for m in client.engine().models()?.data {
                println!("{}", m.id);
            }
        }
    }
    Ok(())
}

fn main() {
    let cli = Cli::parse();
    if let Err(e) = run(cli) {
        eprintln!("hanzo: error: {e}");
        std::process::exit(1);
    }
}