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

//! Data models for the `aasm cost` command.

use serde::{Deserialize, Serialize};

/// Per-agent cost entry from the API.
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct AgentCostEntry {
    pub agent_id: String,
    pub daily_spend_usd: String,
    pub monthly_spend_usd: Option<String>,
    pub date: String,
}

/// API response from `GET /api/v1/costs`.
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct CostResponse {
    pub daily_spend_usd: String,
    pub monthly_spend_usd: Option<String>,
    pub date: String,
    #[serde(default)]
    pub daily_limit_usd: Option<String>,
    #[serde(default)]
    pub monthly_limit_usd: Option<String>,
    #[serde(default)]
    pub per_agent: Vec<AgentCostEntry>,
}

/// Display model for cost summary output.
#[derive(Debug, Clone, Serialize)]
pub struct CostSummaryDisplay {
    pub daily_spend_usd: String,
    pub monthly_spend_usd: Option<String>,
    pub date: String,
    pub daily_limit_usd: Option<String>,
    pub monthly_limit_usd: Option<String>,
    pub per_agent: Vec<AgentCostEntry>,
}

/// Display model for cost forecast output.
#[derive(Debug, Clone, Serialize)]
pub struct CostForecastDisplay {
    pub date: String,
    pub day_of_month: u32,
    pub days_in_month: u32,
    pub current_daily_spend: String,
    pub projected_monthly_spend: String,
    pub monthly_limit_usd: Option<String>,
    pub projected_utilization_pct: Option<String>,
}

impl From<CostResponse> for CostSummaryDisplay {
    fn from(resp: CostResponse) -> Self {
        Self {
            daily_spend_usd: resp.daily_spend_usd,
            monthly_spend_usd: resp.monthly_spend_usd,
            date: resp.date,
            daily_limit_usd: resp.daily_limit_usd,
            monthly_limit_usd: resp.monthly_limit_usd,
            per_agent: resp.per_agent,
        }
    }
}

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

    #[test]
    fn cost_response_deserializes_minimal() {
        let json = r#"{"daily_spend_usd":"0.00","date":"2026-04-30"}"#;
        let resp: CostResponse = serde_json::from_str(json).unwrap();
        assert_eq!(resp.daily_spend_usd, "0.00");
        assert!(resp.monthly_spend_usd.is_none());
        assert!(resp.daily_limit_usd.is_none());
        assert!(resp.per_agent.is_empty());
    }

    #[test]
    fn cost_response_deserializes_full() {
        let json = r#"{
            "daily_spend_usd": "8.10",
            "monthly_spend_usd": "142.50",
            "date": "2026-04-30",
            "daily_limit_usd": "50.00",
            "monthly_limit_usd": "500.00",
            "per_agent": [{
                "agent_id": "abc123",
                "daily_spend_usd": "4.00",
                "monthly_spend_usd": "80.00",
                "date": "2026-04-30"
            }]
        }"#;
        let resp: CostResponse = serde_json::from_str(json).unwrap();
        assert_eq!(resp.daily_limit_usd.as_deref(), Some("50.00"));
        assert_eq!(resp.per_agent.len(), 1);
        assert_eq!(resp.per_agent[0].agent_id, "abc123");
    }
}