atm-core 0.2.3

Core domain types for ATM - Claude Code agent management
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

//! Agent type identification.

use serde::{Deserialize, Serialize};
use std::fmt;

/// Type of Claude Code agent.
///
/// Claude Code spawns different agent types for different purposes:
/// - Main agent for general tasks
/// - Specialized subagents for exploration, planning, code review
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[serde(rename_all = "snake_case")]
#[derive(Default)]
pub enum AgentType {
    /// General-purpose main agent
    #[default]
    GeneralPurpose,

    /// Task/explore subagent for file exploration
    Explore,

    /// Planning subagent for task breakdown
    Plan,

    /// Code review subagent
    CodeReviewer,

    /// File search/analysis subagent
    FileSearch,

    /// Custom or unknown agent type
    Custom(String),
}

impl AgentType {
    /// Returns a short identifier for display.
    pub fn short_name(&self) -> &str {
        match self {
            Self::GeneralPurpose => "main",
            Self::Explore => "explore",
            Self::Plan => "plan",
            Self::CodeReviewer => "review",
            Self::FileSearch => "search",
            Self::Custom(name) => name.as_str(),
        }
    }

    /// Returns a descriptive label for the agent type.
    pub fn label(&self) -> &str {
        match self {
            Self::GeneralPurpose => "General Purpose",
            Self::Explore => "Explorer",
            Self::Plan => "Planner",
            Self::CodeReviewer => "Code Reviewer",
            Self::FileSearch => "File Search",
            Self::Custom(_) => "Custom",
        }
    }

    /// Parses an agent type from a subagent_type string.
    pub fn from_subagent_type(s: &str) -> Self {
        match s.to_lowercase().as_str() {
            "general-purpose" | "general_purpose" => Self::GeneralPurpose,
            "explore" | "explorer" => Self::Explore,
            "plan" | "planner" => Self::Plan,
            "code-reviewer" | "code_reviewer" | "codereview" => Self::CodeReviewer,
            "file-search" | "file_search" | "filesearch" => Self::FileSearch,
            _ => Self::Custom(s.to_string()),
        }
    }
}

impl fmt::Display for AgentType {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.label())
    }
}

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

    #[test]
    fn test_agent_type_parsing() {
        assert_eq!(
            AgentType::from_subagent_type("general-purpose"),
            AgentType::GeneralPurpose
        );
        assert_eq!(AgentType::from_subagent_type("explore"), AgentType::Explore);
        assert_eq!(
            AgentType::from_subagent_type("custom-agent"),
            AgentType::Custom("custom-agent".to_string())
        );
    }

    #[test]
    fn test_agent_type_short_name() {
        assert_eq!(AgentType::GeneralPurpose.short_name(), "main");
        assert_eq!(AgentType::Explore.short_name(), "explore");
    }
}