routerctl 0.3.3

Command-line control interface for syntrop-routerd
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

use crate::cli::TestArgs;
use crate::client::RouterctlClient;
use anyhow::Result;
use colored::Colorize;

pub async fn run_benchmark(
    client: &RouterctlClient,
    args: TestArgs,
    json_output: bool,
) -> Result<()> {
    if !json_output {
        println!(
            "Benchmarking route for model '{}' (tier: {:?})...",
            args.model.bold(),
            args.tier
        );
    }

    match client.test_benchmark(&args.model, &args.prompt, args.tier.as_deref()).await {
        Ok(bench) => {
            if json_output {
                let obj = serde_json::json!({
                    "model": bench.model,
                    "ttft_ms": bench.ttft_ms,
                    "total_latency_ms": bench.total_latency_ms,
                    "chunks": bench.chunks,
                    "response": bench.content,
                    "success": true
                });
                println!("{}", serde_json::to_string_pretty(&obj)?);
            } else {
                println!(
                    "{} TTFT: {:.1} ms | Total Latency: {:.1} ms (chunks: {})",
                    "Success:".bold().green(),
                    bench.ttft_ms,
                    bench.total_latency_ms,
                    bench.chunks
                );
                if !bench.content.is_empty() {
                    let preview = if bench.content.len() > 120 {
                        format!("{}...", &bench.content[..120])
                    } else {
                        bench.content
                    };
                    println!("  Response preview: \"{}\"", preview.dimmed());
                }
            }
        }
        Err(e) => {
            if json_output {
                let obj = serde_json::json!({
                    "model": args.model,
                    "success": false,
                    "error": e.to_string()
                });
                println!("{}", serde_json::to_string_pretty(&obj)?);
            } else {
                println!("{} {}", "Benchmark failed:".bold().red(), e);
            }
        }
    }

    Ok(())
}