batuta 0.7.3

Sovereign AI orchestration: autonomous agents, ML serving, code analysis, and transpilation pipelines
Documentation 
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

//! Pacha Helper Functions
//!
//! Utility functions for the Pacha model registry CLI including
//! model resolution, caching, and display formatting.

use crate::ansi_colors::Colorize;

// ============================================================================
// Display Helpers
// ============================================================================

/// Print a styled command header
pub fn print_command_header(title: &str) {
    println!("{}", title.bright_cyan().bold());
    println!("{}", "=".repeat(60).dimmed());
    println!();
}

/// Print a success message with checkmark
pub fn print_success(message: impl std::fmt::Display) {
    println!("{} {}", "ok".bright_green().bold(), message);
}

// ============================================================================
// Model Resolution
// ============================================================================

/// Resolve a model reference to full URI
pub fn resolve_model_ref(model: &str, quant: Option<&str>) -> anyhow::Result<String> {
    // Check for known aliases
    let aliases = [
        ("llama3", "hf://meta-llama/Meta-Llama-3-8B-Instruct-GGUF"),
        ("llama3:8b", "hf://meta-llama/Meta-Llama-3-8B-Instruct-GGUF"),
        ("llama3:70b", "hf://meta-llama/Meta-Llama-3-70B-Instruct-GGUF"),
        ("mistral", "hf://mistralai/Mistral-7B-Instruct-v0.2-GGUF"),
        ("mixtral", "hf://mistralai/Mixtral-8x7B-Instruct-v0.1-GGUF"),
        ("phi3", "hf://microsoft/Phi-3-mini-4k-instruct-gguf"),
    ];

    for (alias, target) in &aliases {
        if model == *alias {
            let mut result = (*target).to_string();
            if let Some(q) = quant {
                result = format!("{}:{}", result, q.to_uppercase());
            }
            return Ok(result);
        }
    }

    // If already a full URI, return as-is
    if model.contains("://") {
        return Ok(model.to_string());
    }

    // Otherwise, assume pacha:// scheme
    Ok(format!("pacha://{}", model))
}

// ============================================================================
// Cache Operations
// ============================================================================

/// Check if a model is cached (simulation)
pub fn is_cached(model: &str) -> bool {
    // GH-47: model parameter not yet used — always returns false (simulation stub)
    let _ = model;
    false
}

/// Get cached models (simulation)
pub fn get_cached_models() -> Vec<(String, u64, String)> {
    vec![
        ("llama3:8b-q4_k_m".to_string(), 4_690_000_000, "2 hours ago".to_string()),
        ("mistral:7b-q4_k_m".to_string(), 4_110_000_000, "1 day ago".to_string()),
        ("phi3:mini-q4_k_m".to_string(), 2_390_000_000, "3 days ago".to_string()),
    ]
}

// ============================================================================
// Formatting
// ============================================================================

/// Format size in human-readable form
pub fn format_size(bytes: u64) -> String {
    const KB: u64 = 1024;
    const MB: u64 = KB * 1024;
    const GB: u64 = MB * 1024;

    if bytes >= GB {
        format!("{:.2} GB", bytes as f64 / GB as f64)
    } else if bytes >= MB {
        format!("{:.1} MB", bytes as f64 / MB as f64)
    } else if bytes >= KB {
        format!("{:.0} KB", bytes as f64 / KB as f64)
    } else {
        format!("{} B", bytes)
    }
}

/// Create a progress bar string
pub fn create_progress_bar(percent: f64, width: usize) -> String {
    let filled = (percent / 100.0 * width as f64) as usize;
    let empty = width.saturating_sub(filled);

    format!(
        "{}[{}{}]{}",
        "\x1b[36m", // Cyan
        "=".repeat(filled),
        " ".repeat(empty),
        "\x1b[0m" // Reset
    )
}

/// Truncate string with ellipsis (delegates to batuta-common).
pub fn truncate_str(s: &str, max_len: usize) -> String {
    batuta_common::display::truncate_str(s, max_len)
}