ai-dispatch 9.7.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

// E2E coverage for `aid run auto`.
// Verifies the stderr selection message and dispatch through a fake agent CLI.
// Deps: compiled `aid` binary, tempfile, and a POSIX shell.

use std::path::Path;

#[cfg(unix)]
use std::os::unix::fs::PermissionsExt;
use tempfile::TempDir;

mod common;
use common::aid_cmd_in;

/// Verifies auto mode selects a research-capable agent and reports why.
#[test]
fn auto_run_selects_research_agent_and_reports_reason() {
    let aid_home = TempDir::new().unwrap();
    let bin_dir = TempDir::new().unwrap();
    write_script(
        bin_dir.path(),
        "gemini",
        "#!/bin/sh\nprintf '%s' '{\"response\":\"ok\",\"usageMetadata\":{\"totalTokenCount\":7}}'\n",
    );
    write_script(bin_dir.path(), "agy", "#!/bin/sh\nprintf '%s' 'ok'\n");

    let path = format!(
        "{}:{}",
        bin_dir.path().display(),
        std::env::var("PATH").unwrap_or_default()
    );
    let output = aid_cmd_in(aid_home.path())
        .env("PATH", path)
        .args(["run", "auto", "Explain the retry flow?"])
        .output()
        .unwrap();

    assert!(
        output.status.success(),
        "run failed: {}",
        String::from_utf8_lossy(&output.stderr)
    );
    let stderr = String::from_utf8_lossy(&output.stderr);
    let auto_selected = stderr.contains("[aid] Auto-selected: gemini")
        || stderr.contains("[aid] Auto-selected: agy");
    assert!(auto_selected, "expected gemini or agy to be auto-selected, got:\n{}", stderr);
    assert!(stderr.contains("research task"));
}

fn write_script(dir: &Path, name: &str, contents: &str) {
    let path = dir.join(name);
    std::fs::write(&path, contents).unwrap();
    #[cfg(unix)]
    {
        let permissions = std::fs::Permissions::from_mode(0o755);
        std::fs::set_permissions(&path, permissions).unwrap();
    }
}