routerctl 0.3.8

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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use crate::cli::AskArgs;
use crate::client::RouterctlClient;
use anyhow::Result;

pub async fn run_ask(
    client: &RouterctlClient,
    args: AskArgs,
    json_output: bool,
) -> Result<()> {
    let prompt = join_prompt(&args.prompt);
    match client
        .chat_completion(&args.model, &prompt, args.max_tokens)
        .await
    {
        Ok(text) => {
            if json_output {
                let obj = serde_json::json!({
                    "model": args.model,
                    "prompt": prompt,
                    "response": text,
                    "success": true
                });
                println!("{}", serde_json::to_string_pretty(&obj)?);
            } else {
                println!("{}", text);
            }
        }
        Err(e) => {
            if json_output {
                let obj = serde_json::json!({
                    "model": args.model,
                    "prompt": prompt,
                    "success": false,
                    "error": e.to_string()
                });
                println!("{}", serde_json::to_string_pretty(&obj)?);
            } else {
                eprintln!("ask failed: {}", e);
                std::process::exit(1);
            }
        }
    }

    Ok(())
}

/// Question words become one prompt: `ask say hello` asks "say hello".
fn join_prompt(words: &[String]) -> String {
    words.join(" ")
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::cli::{Cli, Commands};
    use clap::Parser;

    #[test]
    fn prompt_words_join_with_spaces() {
        assert_eq!(join_prompt(&[]), "");
        assert_eq!(
            join_prompt(&["say".to_string(), "hello".to_string()]),
            "say hello"
        );
    }

    #[test]
    fn ask_parses_model_and_words() {
        let cli = Cli::try_parse_from(["routerctl", "ask", "say", "hello"]).unwrap();
        let Commands::Ask(args) = cli.command else {
            panic!("want Ask");
        };
        assert_eq!(args.model, "router:auto");
        assert_eq!(args.max_tokens, 256);
        assert_eq!(join_prompt(&args.prompt), "say hello");
        let cli = Cli::try_parse_from(["routerctl", "ask", "-m", "router:fast", "hi"]).unwrap();
        let Commands::Ask(args) = cli.command else {
            panic!("want Ask");
        };
        assert_eq!(args.model, "router:fast");
    }
}