ai-agent 0.13.4

Idiomatic agent sdk inspired by the claude code source leak
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

use std::collections::HashSet;

const ACTION_KEYWORDS: &[&str] = &[
    "create",
    "add",
    "implement",
    "write",
    "make",
    "build",
    "generate",
    "set up",
    "initialize",
    "configure",
];

const MODIFICATION_KEYWORDS: &[&str] = &[
    "fix", "bug", "issue", "problem", "error", "broken", "repair", "patch", "update", "change",
    "modify", "refactor",
];

const ANALYSIS_KEYWORDS: &[&str] = &[
    "analyze", "review", "check", "examine", "inspect", "audit", "find", "search", "debug",
    "explain",
];

pub fn get_action_keywords() -> HashSet<&'static str> {
    ACTION_KEYWORDS.iter().cloned().collect()
}

pub fn get_modification_keywords() -> HashSet<&'static str> {
    MODIFICATION_KEYWORDS.iter().cloned().collect()
}

pub fn get_analysis_keywords() -> HashSet<&'static str> {
    ANALYSIS_KEYWORDS.iter().cloned().collect()
}

pub fn is_action_prompt(prompt: &str) -> bool {
    let prompt_lower = prompt.to_lowercase();
    ACTION_KEYWORDS.iter().any(|kw| prompt_lower.contains(kw))
}

pub fn is_modification_prompt(prompt: &str) -> bool {
    let prompt_lower = prompt.to_lowercase();
    MODIFICATION_KEYWORDS
        .iter()
        .any(|kw| prompt_lower.contains(kw))
}

pub fn is_analysis_prompt(prompt: &str) -> bool {
    let prompt_lower = prompt.to_lowercase();
    ANALYSIS_KEYWORDS.iter().any(|kw| prompt_lower.contains(kw))
}

pub fn classify_prompt(prompt: &str) -> &str {
    if is_modification_prompt(prompt) {
        "modification"
    } else if is_action_prompt(prompt) {
        "action"
    } else if is_analysis_prompt(prompt) {
        "analysis"
    } else {
        "unknown"
    }
}

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

    #[test]
    fn test_action_keyword() {
        assert!(is_action_prompt("Create a new file"));
        assert!(is_action_prompt("Implement feature"));
    }

    #[test]
    fn test_modification_keyword() {
        assert!(is_modification_prompt("Fix the bug"));
        assert!(is_modification_prompt("Update the code"));
    }

    #[test]
    fn test_classification() {
        assert_eq!(classify_prompt("Fix this issue"), "modification");
        assert_eq!(classify_prompt("Create new file"), "action");
        assert_eq!(classify_prompt("Analyze code"), "analysis");
    }
}