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

// Declared-profile validation and agent resolution for `aid run`.
// Exports: validate_task_profile(), resolve_run_agent().
// Deps: selection advice, routing hints, config/team/store, task-profile types.

use anyhow::Result;

use crate::agent;
use crate::agent::classifier::TaskCategory;
use crate::cmd_dispatch::recommend_hint;
use crate::store;
use crate::team;
use crate::types::{
    DeclaredTaskProfile, TaskBudget, TaskDifficulty, TaskEgress, TaskRigor, TaskUrgency,
};

pub(super) fn validate_task_profile(
    difficulty: Option<TaskDifficulty>,
    budget: Option<TaskBudget>,
    urgency: Option<TaskUrgency>,
    rigor: Option<TaskRigor>,
) -> Result<()> {
    let missing = [
        difficulty.is_none().then_some("difficulty"),
        budget.is_none().then_some("budget"),
        urgency.is_none().then_some("urgency"),
        rigor.is_none().then_some("rigor"),
    ].into_iter().flatten().collect::<Vec<_>>();
    if missing.is_empty() { return Ok(()) }
    let project = crate::project::detect_project();
    if project.as_ref().is_some_and(|item| item.require_task_profile) {
        anyhow::bail!("Task profile is required; missing --{}", missing.join(", --"));
    }
    aid_warn!(
        "[aid] Warning: task profile incomplete (missing --{}); undeclared values will be stored as null",
        missing.join(", --")
    );
    Ok(())
}

#[allow(clippy::too_many_arguments)]
pub(super) fn resolve_run_agent(
    store: &store::Store, prompt: &str,
    difficulty: Option<TaskDifficulty>, declared_budget: Option<TaskBudget>,
    urgency: Option<TaskUrgency>, rigor: Option<TaskRigor>, egress: TaskEgress,
    kind: Option<TaskCategory>, no_hint: bool, team_flag: &Option<String>, agent_name: String,
) -> Result<String> {
    if agent::selection::is_removed_auto_agent(&agent_name) {
        anyhow::bail!("{}", agent::selection::AUTO_AGENT_REMOVED_MSG);
    }
    let declared = DeclaredTaskProfile {
        difficulty: difficulty.unwrap_or_default(), budget: declared_budget.unwrap_or_default(),
        urgency: urgency.unwrap_or_default(), rigor: rigor.unwrap_or_default(),
    };
    let team_config = team_flag.as_deref().and_then(team::resolve_team);
    if let Some(hint) = recommend_hint::recommendation_hint(
        &agent_name, prompt, no_hint, declared, kind, store, team_config.as_ref(),
    ) {
        aid_hint!("{hint}");
    }
    explicit_agent(agent_name, egress)
}

/// Egress gates only: the model is resolved at dispatch by `resolve_run_model`.
fn explicit_agent(agent_name: String, egress: TaskEgress) -> Result<String> {
    if egress.requires_local() {
        agent::egress::require_local_egress(&agent_name)?;
    }
    if egress.requires_private_network() {
        agent::egress::require_private_network_egress(&agent_name)?;
    }
    Ok(agent_name)
}

#[cfg(test)]
#[path = "run_profile_tests.rs"]
mod tests;