honk 0.2.0

Make your computer honk like an old-school car. Honk, honk!
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

//! AI feature management, backed by the shared [Ailloy](https://crates.io/crates/ailloy) config.
//!
//! - `honk ai`          show status
//! - `honk ai test`     test the AI connection
//! - `honk ai enable`   enable AI for honk
//! - `honk ai disable`  disable AI for honk
//! - `honk ai config`   interactive AI node configuration
//!
//! Ailloy stores a shared, global config (`~/.config/ailloy/config.yaml`) so
//! every MKLab tool reuses the same providers and API keys. To actually call a
//! model from your own commands, use `ailloy::Client` (see the ailloy docs).

use anyhow::Result;

use ailloy::config::Config;
use ailloy::config_tui;

use crate::cli::AiCommands;

/// The capabilities this tool needs from Ailloy. A bare CLI just needs `chat`;
/// add more (e.g. `"image"`) as your tool grows.
const CAPABILITIES: &[&str] = &["chat"];

pub async fn run(cmd: Option<AiCommands>) -> Result<()> {
    match cmd {
        None | Some(AiCommands::Status) => config_tui::print_ai_status("honk", CAPABILITIES),
        Some(AiCommands::Test { message }) => config_tui::run_test_chat("honk", message).await,
        Some(AiCommands::Enable) => config_tui::enable_ai("honk"),
        Some(AiCommands::Disable) => config_tui::disable_ai("honk"),
        Some(AiCommands::Config) => {
            let mut config = Config::load_global()?;
            config_tui::run_interactive_config(&mut config, CAPABILITIES).await?;
            Ok(())
        }
    }
}

/// Whether AI features are active (configured via ailloy + enabled for this tool).
#[allow(dead_code)]
pub fn is_ai_active() -> bool {
    config_tui::is_ai_active("honk")
}