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

// Custom-agent advice candidates: separate capability scale from built-ins.
// Exports: custom_candidates().
// Deps: custom registry, route predicate, capability helpers, run_model resolver.

use crate::agent::registry::load_custom_agents;
use super::super::selection_capabilities::{custom_category_score, custom_strength_bonus};
use super::super::selection_scoring::CandidateContext;
use crate::agent::run_model::{RunModelInput, resolve_run_model};
use crate::config::agent as agent_config;
use crate::config::SelectionConfig;
use crate::routing::model_catalog::AGENT_MODELS;
use crate::types::{AgentKind, DeclaredTaskProfile, TaskBudget};

use super::gate::Exclusions;
use super::{CustomAdviceCandidate, ELIGIBILITY_PENALTY, NOT_INSTALLED_PENALTY};

pub(super) fn custom_candidates(
    context: &CandidateContext<'_>,
    declared: DeclaredTaskProfile,
    selection: &SelectionConfig,
) -> Vec<CustomAdviceCandidate> {
    let floor = declared.difficulty.capability_floor();
    let mut candidates: Vec<_> = load_custom_agents().into_values()
        .filter(|config| AgentKind::parse_str(&config.id).is_none())
        .filter(|config| !agent_config::is_agent_disabled(&config.id))
        .map(|config| {
            let category_capability = custom_category_score(&config, context.profile.category);
            let strength_bonus = custom_strength_bonus(&config, context.profile.category);
            let total = category_capability + strength_bonus;
            let team_preferred = context.team.is_some_and(|team| {
                team.preferred_agents.iter().any(|item| item.eq_ignore_ascii_case(&config.id))
            });
            let input = RunModelInput::declared(
                &config.id, AgentKind::Custom, Some(&config), declared, selection,
            );
            let model = resolve_run_model(&input).model;
            let blocker = crate::agent::custom_route_blocker(&config.command);
            let mut exclusions = Exclusions::default();
            if let Some(blocker) = &blocker {
                exclusions.push(blocker.code(), blocker.reason());
            }
            let _ = exclusions.floor(
                Some(f64::from(total)), floor, declared.difficulty, context.profile.category,
            );
            exclusions.budget(custom_budget_allows(model.as_deref(), declared.budget), declared.budget);
            let eligible = exclusions.is_empty();
            let (exclusion_reason, exclusion_codes) = exclusions.into_parts();
            CustomAdviceCandidate {
                agent: config.id, installed: blocker.is_none(), eligible,
                capability_evidence: crate::routing::scores::evidence(AgentKind::Custom, model.as_deref(), context.profile.category),
                model, category_capability, strength_bonus, team_preferred,
                exclusion_reason, exclusion_codes,
            }
        })
        .collect();
    candidates.sort_by(|left, right| {
        custom_rank(right).partial_cmp(&custom_rank(left)).unwrap_or(std::cmp::Ordering::Equal)
            .then_with(|| {
                let lt = left.category_capability + left.strength_bonus;
                let rt = right.category_capability + right.strength_bonus;
                rt.cmp(&lt).then_with(|| left.agent.cmp(&right.agent))
            })
    });
    candidates
}

fn custom_rank(candidate: &CustomAdviceCandidate) -> f64 {
    let mut score = f64::from(candidate.category_capability + candidate.strength_bonus);
    if !candidate.installed { score -= NOT_INSTALLED_PENALTY; }
    if !candidate.eligible { score -= ELIGIBILITY_PENALTY; }
    score
}

fn custom_budget_allows(model: Option<&str>, budget: TaskBudget) -> bool {
    if !matches!(budget, TaskBudget::Free | TaskBudget::Cheap) { return true; }
    let Some(model) = model else { return false };
    AGENT_MODELS.iter().any(|item| item.model == model && (
        item.tier == "free" || budget == TaskBudget::Cheap && item.tier == "cheap"
    ))
}