llm-cmd 0.11.0

A CLI tool for interacting with LLMs (OpenAI, Anthropic) via a unified interface
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

use clap::Subcommand;
use llm_core::{Config, LlmError, Paths, Result};

#[derive(Subcommand)]
pub enum AliasesCommand {
    /// Set an alias for a model
    Set {
        /// Alias name
        alias: String,
        /// Model ID the alias points to
        model: String,
    },
    /// Show what model an alias points to
    Show {
        /// Alias name
        alias: String,
    },
    /// List all aliases
    List,
    /// Remove an alias
    Remove {
        /// Alias name
        alias: String,
    },
    /// Show the path to the config file
    Path,
}

pub fn run(cmd: &AliasesCommand) -> Result<()> {
    let paths = Paths::resolve()?;
    let config_path = paths.config_file();

    match cmd {
        AliasesCommand::Set { alias, model } => {
            let mut config = Config::load(&config_path)?;
            config.set_alias(alias, model);
            config.save(&config_path)?;
        }
        AliasesCommand::Show { alias } => {
            let config = Config::load(&config_path)?;
            match config.aliases.get(alias.as_str()) {
                Some(model) => println!("{alias}: {model}"),
                None => {
                    return Err(LlmError::Config(format!("alias '{alias}' not found")));
                }
            }
        }
        AliasesCommand::List => {
            let config = Config::load(&config_path)?;
            if config.aliases.is_empty() {
                println!("No aliases set");
            } else {
                let mut entries: Vec<_> = config.aliases.iter().collect();
                entries.sort_by_key(|(k, _)| *k);
                for (alias, model) in entries {
                    println!("{alias}: {model}");
                }
            }
        }
        AliasesCommand::Remove { alias } => {
            let mut config = Config::load(&config_path)?;
            if !config.remove_alias(alias) {
                return Err(LlmError::Config(format!("alias '{alias}' not found")));
            }
            config.save(&config_path)?;
        }
        AliasesCommand::Path => {
            println!("{}", config_path.display());
        }
    }
    Ok(())
}