ai-dispatch 11.3.0

Multi-AI CLI team orchestrator
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

// API account quota display is distinct from the Claude subscription.
// Uses an empty temporary database, without reading any real keychain item.
use super::*;
use crate::support::paths::{self, AidHomeGuard};

#[test]
fn api_quota_rows_show_plan_state_and_monthly_dollars() {
    let home = tempfile::tempdir().unwrap();
    let _guard = AidHomeGuard::set(home.path());
    std::fs::write(
        paths::config_path(),
        concat!(
            "[[api_route]]\nid = 'claude-api-meet'\nagent = 'claude'\n",
            "keychain_service = 'test'\nmonthly_budget_usd = 200\n",
        ),
    )
    .unwrap();
    let line = api_quota_line("claude-api-meet").unwrap().unwrap();
    assert!(line.contains("claude-api-meet"));
    assert!(line.contains("IN PLAN"));
    assert!(line.contains("$0.00 / $200 this month"));
    assert!(api_quota_line("claude").unwrap().is_none());
    show_quota().unwrap();
    list_agents().unwrap();
}

#[test]
fn api_quota_displays_exhausted_and_bill_overage() {
    let home = tempfile::tempdir().unwrap();
    let _guard = AidHomeGuard::set(home.path());
    let config = concat!(
        "[[api_route]]\nid = 'claude-api-meet'\nagent = 'claude'\n",
        "keychain_service = 'test'\nmonthly_budget_usd = 200\nover_budget = 'stop'\n",
    );
    std::fs::write(paths::config_path(), config).unwrap();
    let store = crate::store::Store::open(&paths::db_path()).unwrap();
    store
        .insert_waiting_task(
            "t-spend", "claude", "test", None, None, None, None, None, None, false, false,
        )
        .unwrap();
    store
        .db()
        .execute(
            "UPDATE tasks SET agent = 'custom', custom_agent_name = 'claude-api-meet',
        cost_usd = 200, status = 'done', completed_at = strftime('%Y-%m-%dT%H:%M:%fZ', 'now') WHERE id = 't-spend'",
            [],
        )
        .unwrap();
    let line = api_quota_line("claude-api-meet").unwrap().unwrap();
    assert!(line.contains("EXHAUSTED"));
    assert!(line.contains("$200.00 / $200 this month"));
    std::fs::write(paths::config_path(), config.replace("'stop'", "'bill'")).unwrap();
    assert!(
        api_quota_line("claude-api-meet")
            .unwrap()
            .unwrap()
            .contains("OVERAGE")
    );
}

#[test]
fn quota_displays_unknown_completed_task_prices() {
    let home = tempfile::tempdir().unwrap();
    let _guard = AidHomeGuard::set(home.path());
    std::fs::write(
        paths::config_path(),
        concat!(
            "[[api_route]]\nid = 'claude-api-ming'\nagent = 'claude'\n",
            "keychain_service = 'test'\nmonthly_budget_usd = 200\n"
        ),
    )
    .unwrap();
    let store = crate::store::Store::open(&paths::db_path()).unwrap();
    store
        .insert_waiting_task(
            "t-unpriced",
            "claude-api-ming",
            "test",
            None,
            None,
            None,
            None,
            None,
            None,
            false,
            false,
        )
        .unwrap();
    store
        .db()
        .execute(
            "UPDATE tasks SET agent = 'custom', custom_agent_name = 'claude-api-ming',
        status = 'failed', completed_at = strftime('%Y-%m-%dT%H:%M:%fZ', 'now'),
        observed_model = 'claude-opus-99-9', tokens = 100 WHERE id = 't-unpriced'",
            [],
        )
        .unwrap();
    let line = api_quota_line("claude-api-ming").unwrap().unwrap();
    assert!(line.contains("$0.00 / $200 this month (1 tasks unpriced)"));
}