mirror of
https://github.com/hanzoai/kai.git
synced 2026-10-17 11:18:06 +00:00
zooqueenandHanzo Dev
45e8e6c18f
The curl scripts piped any response into jq and exited 0, so a refused key read as success; they now run under set -euo pipefail with --fail-with-body. noul.sh sent "spent 2,000" because the shell expanded $1 inside "$12,000". Go returns errors instead of panicking and gets a go.mod so `go run .` and the CI toolchain find it. Rust reports a missing key as an error rather than a panic and uses rustls, so it builds without system OpenSSL. Both say in one line why they use plain HTTP: go-sdk v8.5.623 refuses a choice question with named criteria, and hanzo-client 8.5.156 has no /v1/decisions. Co-authored-by: Hanzo Dev <[email protected]>
58 lines
2.2 KiB
Rust
58 lines
2.2 KiB
Rust
// reqwest rather than the hanzo-client crate: its newest release, 8.5.156, has no /v1/decisions.
|
|
use serde_json::{json, Value};
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let api_key = std::env::var("HANZO_API_KEY").map_err(|_| "HANZO_API_KEY environment variable required")?;
|
|
let model = std::env::var("KAI_MODEL").unwrap_or_else(|_| "kai".to_string()); // "kai" or "typesafe/jev-1.13"
|
|
let base_url = std::env::var("HANZO_BASE_URL").unwrap_or_else(|_| "https://api.hanzo.ai".to_string());
|
|
let url = format!("{}/v1/decisions", base_url.trim_end_matches('/'));
|
|
|
|
let client = reqwest::Client::new();
|
|
|
|
let res = client
|
|
.post(&url)
|
|
.bearer_auth(api_key)
|
|
.json(&json!({
|
|
"model": model,
|
|
"state": "Kubernetes pod evicted: OOMKilled",
|
|
"questions": {
|
|
"triage": {
|
|
"type": "choice",
|
|
"instructions": "Identify next step",
|
|
"criteria": {
|
|
"increase_limits": "raise memory requests and limits",
|
|
"restart": "restart pod on clean node",
|
|
"profile_memory": "attach memory profiler to inspect leak"
|
|
}
|
|
}
|
|
}
|
|
}))
|
|
.send()
|
|
.await?;
|
|
|
|
let status = res.status();
|
|
if !status.is_success() {
|
|
let err_text = res.text().await?;
|
|
eprintln!("HTTP {} Error: {}", status, err_text);
|
|
std::process::exit(1);
|
|
}
|
|
|
|
let decision: Value = res.json().await?;
|
|
|
|
println!("Model: {}", decision["model"].as_str().unwrap_or("unknown"));
|
|
println!("Decision ID: {}", decision["id"].as_str().unwrap_or(""));
|
|
println!("Latency: {}ms", decision["latency_ms"]);
|
|
|
|
let ans = &decision["answers"]["triage"];
|
|
let choice = ans["choice"].as_str().ok_or_else(|| format!("no choice answer for \"triage\" in {}", decision))?;
|
|
println!("Triage Choice: {}", choice);
|
|
println!("Confidence: {:.4}", ans["confidence"].as_f64().unwrap_or_default());
|
|
println!("Probabilities:");
|
|
for (opt, prob) in ans["probabilities"].as_object().into_iter().flatten() {
|
|
println!(" - {}: {:.2}%", opt, prob.as_f64().unwrap_or_default() * 100.0);
|
|
}
|
|
|
|
Ok(())
|
|
}
|