granite-cli 0.2.0

CLI for discovering, configuring, and launching AI workflows powered by IBM Granite models.
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

// Local
use crate::utils::hardware::detect_hardware;
use anyhow::Result;

pub struct HardwareCommands;

/*-- public --*/

impl HardwareCommands {
    /// Key-value fields for the hardware detail panel.
    /// Shared by the CLI command and the TUI.
    pub(crate) fn hardware_fields() -> Vec<(&'static str, String)> {
        let profile = detect_hardware();
        vec![
            ("CPU Cores", profile.cpu_cores.to_string()),
            ("CPU Architecture", profile.cpu_arch.clone()),
            ("RAM", format!("{:.2} GB", profile.ram_gb)),
            (
                "GPU Vendor",
                profile
                    .gpu_vendor
                    .clone()
                    .unwrap_or_else(|| "None".to_string()),
            ),
            (
                "VRAM",
                profile
                    .vram_gb
                    .map(|v| format!("{v:.2} GB"))
                    .unwrap_or_else(|| "None".to_string()),
            ),
        ]
    }

    pub fn show(ctx: &crate::AppContext) -> Result<()> {
        ctx.ui.detail("Hardware Profile", &Self::hardware_fields());
        Ok(())
    }
}

/*-- tests --*/

#[cfg(test)]
mod tests {
    use super::*;
    use crate::utils::ui::base::tests::CaptureUi;
    use std::sync::Arc;

    fn ctx() -> crate::AppContext {
        crate::AppContext {
            config: crate::config::Config::default(),
            ui: Arc::new(CaptureUi::default()),
        }
    }

    #[test]
    fn hardware_show_renders_detail() {
        let ctx = ctx();
        HardwareCommands::show(&ctx).unwrap();
        let details = (&*ctx.ui as &dyn std::any::Any)
            .downcast_ref::<CaptureUi>()
            .unwrap()
            .details
            .borrow();
        assert_eq!(details.len(), 1);
    }

    #[test]
    fn hardware_detail_has_cpu_and_ram_fields() {
        let fields = HardwareCommands::hardware_fields();
        assert!(fields.iter().any(|(k, _)| *k == "CPU Cores"));
        assert!(fields.iter().any(|(k, _)| *k == "RAM"));
    }
}