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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

// API advice displays route billing and independent pace/hold penalties.
// Tests the full advice engine against persisted account costs.
use super::*;
use crate::support::paths::{self, AidHomeGuard};

#[test]
fn named_api_candidates_have_independent_billing_pace_and_holds() {
    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",
            "[[api_route]]\nid = 'claude-api-meet'\nagent = 'claude'\n",
            "keychain_service = 'test'\nmonthly_budget_usd = 200\nover_budget = 'bill'\n",
        ),
    )
    .unwrap();
    let store = Store::open_memory().unwrap();
    store
        .insert_waiting_task(
            "t-spend",
            "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',
        cost_usd = 200, status = 'done', completed_at = strftime('%Y-%m-%dT%H:%M:%fZ', 'now') WHERE id = 't-spend'",
            [],
        )
        .unwrap();
    crate::routing::api_budget::hold("claude-api-meet", "test human hold", false).unwrap();
    let report = advise(
        "test",
        DeclaredTaskProfile {
            difficulty: Default::default(),
            budget: Default::default(),
            urgency: Default::default(),
            rigor: Default::default(),
        },
        None,
        None,
        Some(&store),
        0,
        None,
    );
    let ming = report
        .candidates
        .iter()
        .find(|c| c.agent == "claude-api-ming")
        .unwrap();
    let meet = report
        .candidates
        .iter()
        .find(|c| c.agent == "claude-api-meet")
        .unwrap();
    assert_eq!(
        ming.billing.as_ref().unwrap().state,
        crate::routing::quota::billing::BillingState::Exhausted
    );
    assert!(ming.exclusion_codes.iter().any(|code| code == "billing"));
    assert!(ming.breakdown.pace == -2.0 || ming.breakdown.pace == 0.0);
    assert_eq!(
        meet.billing.as_ref().unwrap().state,
        crate::routing::quota::billing::BillingState::InPlan
    );
    assert_eq!(meet.quota.status, "held");
    assert_eq!(ming.quota.status, "unknown");
    assert_eq!(
        meet.breakdown.pace,
        if (crate::routing::api_budget::month_bounds(chrono::Utc::now())
            .unwrap()
            .1
            - chrono::Utc::now())
            < chrono::Duration::days(7)
        {
            1.0
        } else {
            0.0
        }
    );
    assert!(
        !report
            .custom_candidates
            .iter()
            .any(|c| c.agent.starts_with("claude-api-"))
    );
}

#[test]
fn history_scores_do_not_share_other_accounts_or_claude() {
    let store = Store::open_memory().unwrap();
    for index in 0..5 {
        let id = format!("t-api-history-{index}");
        store
            .insert_waiting_task(
                &id, "claude", "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 = 'done' WHERE id = ?1",
                [&id],
            )
            .unwrap();
    }
    assert_eq!(history_bonus(Some(&store), "claude-api-ming"), 4.0);
    assert_eq!(history_bonus(Some(&store), "claude-api-meet"), 0.0);
    assert_eq!(history_bonus(Some(&store), "claude"), 0.0);
}