llm 1.3.8

A Rust library unifying multiple LLM backends.
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

mod ansi;
mod lab;

use ratatui::style::Color;

use super::capabilities::ColorLevel;

#[derive(Debug, Clone, Copy, PartialEq)]
pub struct Rgb {
    pub r: u8,
    pub g: u8,
    pub b: u8,
}

impl Rgb {
    pub const fn new(r: u8, g: u8, b: u8) -> Self {
        Self { r, g, b }
    }
}

#[derive(Debug, Clone)]
pub struct TerminalPalette {
    level: ColorLevel,
}

impl TerminalPalette {
    pub fn new(level: ColorLevel) -> Self {
        Self { level }
    }

    pub fn map(&self, rgb: Rgb) -> Color {
        match self.level {
            ColorLevel::TrueColor => Color::Rgb(rgb.r, rgb.g, rgb.b),
            ColorLevel::Ansi256 => Color::Indexed(ansi::nearest_index(rgb, 256)),
            ColorLevel::Ansi16 => Color::Indexed(ansi::nearest_index(rgb, 16)),
            ColorLevel::Ansi8 => Color::Indexed(ansi::nearest_index(rgb, 8)),
            ColorLevel::None => Color::Reset,
        }
    }

    pub fn blend(&self, base: Rgb, highlight: Rgb, t: f32) -> Color {
        let blended = lab::blend_lab(base, highlight, t);
        self.map(blended)
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn maps_truecolor_exact() {
        let palette = TerminalPalette::new(ColorLevel::TrueColor);
        let mapped = palette.map(Rgb::new(10, 20, 30));
        assert_eq!(mapped, Color::Rgb(10, 20, 30));
    }

    #[test]
    fn maps_ansi256_black_to_index_zero() {
        const BLACK: Rgb = Rgb::new(0, 0, 0);
        const BLACK_INDEX: u8 = 0;

        let palette = TerminalPalette::new(ColorLevel::Ansi256);
        let mapped = palette.map(BLACK);
        assert_eq!(mapped, Color::Indexed(BLACK_INDEX));
    }
}