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

// Dispatch and cascade identity regressions for API accounts.
// Uses a temporary config and the real resolver with an in-memory store.
use super::*;
use crate::support::paths::{self, AidHomeGuard};

#[test]
fn named_route_resolves_and_cascade_preserves_its_identity() {
    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\n",
        ),
    )
    .unwrap();
    let store = Arc::new(Store::open_memory().unwrap());
    let mut args = RunArgs {
        agent_name: "claude-api-meet".into(),
        ..Default::default()
    };
    let setup = resolve_agent_setup(&store, &mut args, None).unwrap();
    assert_eq!(setup.agent_kind, AgentKind::Custom);
    assert_eq!(setup.custom_agent_name.as_deref(), Some("claude-api-meet"));
    crate::routing::api_budget::hold("claude-api-meet", "test hold", false).unwrap();
    args.cascade = vec!["claude-api-ming".into()];
    let setup = resolve_agent_setup(&store, &mut args, None).unwrap();
    assert_eq!(setup.agent_display_name, "claude-api-ming");
    assert_eq!(setup.agent.rate_limit_name(), Some("claude-api-ming"));
    assert_eq!(args.agent_name, "claude-api-ming");
}

#[test]
fn api_container_is_rejected_before_launch_without_holding_route() {
    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 = Arc::new(Store::open_memory().unwrap());
    let mut args = RunArgs {
        agent_name: "claude-api-ming".into(),
        container: Some("image".into()),
        ..Default::default()
    };
    let error = match resolve_agent_setup(&store, &mut args, None) {
        Ok(_) => panic!("API route container must be rejected"),
        Err(error) => error.to_string(),
    };
    assert!(error.contains("route claude-api-ming does not support --container"));
    assert!(!paths::aid_dir().join("rate-limit-claude-api-ming").exists());
}

#[test]
fn ordinary_dispatch_survives_invalid_unrelated_api_route() {
    let home = tempfile::tempdir().unwrap();
    let _guard = AidHomeGuard::set(home.path());
    std::fs::write(
        paths::config_path(),
        concat!(
            "[[api_route]]\nid = 'claude-api-bad'\nagent = 'unsupported'\n",
            "keychain_service = 'test'\nmonthly_budget_usd = 200\n"
        ),
    )
    .unwrap();
    let agents = paths::aid_dir().join("agents");
    std::fs::create_dir_all(&agents).unwrap();
    std::fs::write(
        agents.join("other.toml"),
        "[agent]\nid = 'other'\ndisplay_name = 'Other'\ncommand = 'other'",
    )
    .unwrap();
    let store = Arc::new(Store::open_memory().unwrap());
    for name in ["claude", "codex", "other"] {
        let mut args = RunArgs {
            agent_name: name.into(),
            ..Default::default()
        };
        let setup = resolve_agent_setup(&store, &mut args, None).unwrap();
        assert_eq!(setup.agent_display_name, name);
    }
    let mut args = RunArgs {
        agent_name: "claude-api-bad".into(),
        ..Default::default()
    };
    let error = match resolve_agent_setup(&store, &mut args, None) {
        Ok(_) => panic!("invalid route must fail"),
        Err(error) => error.to_string(),
    };
    assert!(error.contains("only agent = \"claude\" is supported"));
}