kowalski-core 1.3.0

Kowalski Core Module: A Rust-based agent for interacting with Ollama models
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

//! Penguin avatar ids for UI display (maps to `ui/src/assets/pinguins/<id>.png`).

/// Infer a stable avatar id from pipeline step `kind` and `name` (kebab-case step id).
pub fn infer_penguin_avatar(kind: &str, name: &str) -> String {
    let n = name.to_lowercase().replace('-', "_");
    let k = kind.to_lowercase();

    for (pat, avatar) in NAME_AVATAR_PATTERNS {
        if n.contains(pat) {
            return (*avatar).to_string();
        }
    }

    match k.as_str() {
        "ingest" => "ingest",
        "deliver" | "final" => "deliver",
        "ask" => "ask",
        "lint" => "lint",
        "compile" => "compile",
        "investigate" => "investigate",
        "structure" => "structure",
        "scaffold" => "scaffold",
        "process" | "step" => "process",
        _ => "default",
    }
    .into()
}

/// Fill missing `avatar` on every penguin in the draft.
pub fn assign_penguin_avatars(draft: &mut crate::rookery::types::RookeryDraft) {
    for p in &mut draft.penguins {
        if p.avatar.as_deref().unwrap_or("").trim().is_empty() {
            p.avatar = Some(infer_penguin_avatar(&p.kind, &p.name));
        }
    }
}

const NAME_AVATAR_PATTERNS: &[(&str, &str)] = &[
    ("mock_builder", "mock_builder"),
    ("mock", "mock_builder"),
    ("todo_generator", "todo_generator"),
    ("todo_list", "todo_generator"),
    ("todo", "todo_generator"),
    ("structure", "structure"),
    ("investigate", "investigate"),
    ("scaffold", "scaffold"),
    ("research", "researcher"),
    ("explorer", "explorer"),
    ("security", "security"),
    ("advisor", "advisor"),
    ("translator", "translator"),
    ("coordinator", "coordinator"),
    ("director", "director"),
    ("compile", "compile"),
    ("thinker", "thinker"),
    ("ingest", "ingest"),
    ("deliver", "deliver"),
    ("lint", "lint"),
    ("ask", "ask"),
];

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

    #[test]
    fn infers_kind_defaults() {
        assert_eq!(infer_penguin_avatar("ingest", "collect"), "ingest");
        assert_eq!(infer_penguin_avatar("deliver", "ship"), "deliver");
        assert_eq!(infer_penguin_avatar("ask", "qa"), "ask");
        assert_eq!(infer_penguin_avatar("lint", "check"), "lint");
    }

    #[test]
    fn infers_name_patterns() {
        assert_eq!(infer_penguin_avatar("step", "mock-builder"), "mock_builder");
        assert_eq!(
            infer_penguin_avatar("step", "todo-generator"),
            "todo_generator"
        );
        assert_eq!(infer_penguin_avatar("step", "investigate"), "investigate");
    }

    #[test]
    fn unknown_falls_back_to_default() {
        assert_eq!(infer_penguin_avatar("custom", "xyzzy"), "default");
    }
}