aa-cli 0.0.1-alpha.9

aasm — command-line tool for Agent Assembly
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
118
119
120
121
122
123
124

//! Rendering utilities for topology subcommands.

use std::collections::{BTreeMap, HashMap};

use serde::{Deserialize, Serialize};

use crate::output::OutputFormat;

pub mod json;
pub mod table;
pub mod tree;

/// Overview of the entire agent topology across all teams.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct TopologyOverview {
    pub team_count: usize,
    pub root_agent_count: usize,
    pub total_agent_count: usize,
    pub teams: Vec<TeamSummary>,
    pub standalone_root_agents: Vec<AgentNode>,
}

/// High-level statistics for a single team.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct TeamSummary {
    pub team_id: String,
    pub agent_count: usize,
    pub root_agent_count: usize,
}

/// Minimal agent representation used in list and tree responses.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AgentNode {
    pub id: String,
    pub name: String,
    pub depth: u32,
    pub status: String,
    pub team_id: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub governance_level: Option<String>,
}

/// Recursive tree node representing an agent and all its descendants.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AgentTree {
    pub id: String,
    pub name: String,
    pub depth: u32,
    pub status: String,
    pub team_id: Option<String>,
    pub delegation_reason: Option<String>,
    pub spawned_by_tool: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub governance_level: Option<String>,
    pub children: Vec<AgentTree>,
}

/// All agents belonging to a single team.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct TeamTopology {
    pub team_id: String,
    pub agent_count: usize,
    pub members: Vec<AgentNode>,
}

/// An agent's complete ancestry chain ordered root-first.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AgentLineage {
    pub agent_id: String,
    pub ancestor_count: usize,
    pub ancestors: Vec<LineageStep>,
}

/// One step in an agent's ancestry chain.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct LineageStep {
    pub id: String,
    pub name: String,
    pub depth: u32,
    pub delegation_reason: Option<String>,
    pub team_id: Option<String>,
}

/// Aggregate topology statistics across all registered agents.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct TopologyStats {
    pub total_agents: usize,
    pub root_agent_count: usize,
    pub max_depth: u32,
    pub active_count: usize,
    pub suspended_count: usize,
    pub deregistered_count: usize,
    pub team_count: usize,
    pub team_sizes: HashMap<String, usize>,
    pub depth_histogram: BTreeMap<String, u32>,
    pub team_size_histogram: BTreeMap<String, u32>,
    pub spawn_count_histogram: BTreeMap<String, u32>,
    pub orphan_count: usize,
    pub avg_children_per_parent: f64,
}

/// Union of all topology API response shapes for rendering.
pub enum TopologyPayload<'a> {
    Overview(&'a TopologyOverview),
    Tree(&'a AgentTree),
    Team(&'a TeamTopology),
    Lineage(&'a AgentLineage),
    Stats(&'a TopologyStats),
}

/// Render a topology payload in the requested output format.
pub fn render(payload: TopologyPayload<'_>, output: OutputFormat) {
    match output {
        OutputFormat::Json => json::render_json(&payload),
        OutputFormat::Yaml => json::render_yaml(&payload),
        OutputFormat::Table => match payload {
            TopologyPayload::Overview(v) => table::render_overview_table(v),
            TopologyPayload::Tree(v) => tree::render_agent_tree(v, "", true),
            TopologyPayload::Team(v) => table::render_team_table(v),
            TopologyPayload::Lineage(v) => table::render_lineage_table(v),
            TopologyPayload::Stats(v) => table::render_stats_table(v),
        },
    }
}