use scv_core::{ToolError, ToolFailure, ToolOutput};
use scv_protocol::JobStatus;
use serde_json::Value;
use crate::delegate::{
adapters,
agent::{Accepts, Offered},
output::{self, AgentReply},
};
const UNAVAILABLE: &[&str] = &[
"not found on path",
"no such file or directory",
"the agent exited",
"acp server exited",
"not logged in",
"not signed in",
"not authenticated",
"unauthorized",
"authentication",
"missing_credential",
"no api key",
"auth_required",
"forbidden",
"401",
"403",
"404",
"429",
"502",
"503",
"529",
"rate limit",
"rate_limit",
"quota",
"overloaded",
"service unavailable",
"model not found",
"no such model",
"does not exist",
"insufficient",
"billing",
"connection refused",
"connection reset",
];
pub(crate) fn reports_unavailable(text: &str) -> bool {
let lower = text.to_ascii_lowercase();
UNAVAILABLE.iter().any(|needle| lower.contains(needle))
}
pub fn model_effort_phrase(model: Option<&str>, effort: Option<&str>) -> Option<String> {
match (model, effort) {
(Some(model), Some(effort)) => Some(format!("model {model} and effort {effort}")),
(Some(model), None) => Some(format!("model {model}")),
(None, Some(effort)) => Some(format!("effort {effort}")),
(None, None) => None,
}
}
fn choice_note(use_for: Option<&str>, model: Option<&str>, effort: Option<&str>) -> Option<String> {
let defaults = model_effort_phrase(model, effort);
match (use_for, defaults) {
(Some(use_for), Some(defaults)) => Some(format!(
" The user's note on when to use it: {use_for}. For that work, pass {defaults}; \
omit model and effort for other work so the agent uses its own default."
)),
(Some(use_for), None) => Some(format!(" The user's note on when to use it: {use_for}")),
(None, Some(defaults)) => Some(format!(
" Pass {defaults} unless the user asks for another; omit them to use the agent's \
own default."
)),
(None, None) => None,
}
}
fn takes(accepts: Accepts, model_hint: &str) -> String {
let mut taken = Vec::new();
if accepts.model {
let hint = model_hint.trim().trim_end_matches('.');
taken.push(if hint.is_empty() {
"model".to_owned()
} else {
format!("model ({hint})")
});
}
if accepts.effort {
taken.push("effort".to_owned());
}
if accepts.session {
taken.push("session".to_owned());
}
match taken.as_slice() {
[] => "Takes no model, effort, or session.".to_owned(),
[one] => format!("Takes {one}."),
[first, second] => format!("Takes {first} and {second}."),
[rest @ .., last] => format!("Takes {}, and {last}.", rest.join(", ")),
}
}
pub(crate) fn entry(agent: &Offered) -> String {
let mut line = match adapters::adapter(&agent.name) {
Some(descriptor) => format!(
"{} ({}): {}.",
agent.name, descriptor.product, descriptor.offers
),
None => format!("{}.", agent.name),
};
line.push(' ');
line.push_str(&takes(agent.accepts, &agent.model_hint));
if let Some(note) = choice_note(
agent.use_for.as_deref(),
agent.model.as_deref(),
agent.effort.as_deref(),
) {
line.push_str(¬e);
}
line
}
fn fallback(others: &[&str]) -> Option<String> {
(!others.is_empty()).then(|| {
format!(
"This agent could not run: it is missing, signed out, or its provider returned an \
error. Other agents are available: {}. Call agent again with one of them.",
others.join(", ")
)
})
}
fn unavailable_result(reply: &AgentReply) -> bool {
reply.status == Some(JobStatus::Failed)
&& reply.error.as_deref().is_some_and(reports_unavailable)
}
pub(crate) fn settle(
result: Result<ToolOutput, ToolError>,
others: &[&str],
) -> Result<ToolOutput, ToolError> {
match result {
Ok(mut result) if result.is_error() => {
if let Ok(value) = serde_json::from_str::<Value>(&result.content)
&& let Some(reply) = AgentReply::read(&value)
&& let Value::Object(mut content) = value
{
if unavailable_result(&reply) {
result.failure = Some(ToolFailure::Unavailable);
if let Some(fallback) = fallback(others) {
content.insert("fallback".into(), fallback.into());
result.content = Value::Object(content).to_string();
}
} else if reply.status == Some(JobStatus::Declined) && others.contains(&"grok") {
content.insert("note".into(), output::DECLINED_NOTE_TRY_GROK.into());
result.content = Value::Object(content).to_string();
}
}
Ok(result)
}
Err(error) if error.kind == ToolFailure::Unavailable => Err(match fallback(others) {
Some(fallback) => ToolError::unavailable(format!("{} {fallback}", error.message)),
None => error,
}),
other => other,
}
}
pub fn product(agent: &str) -> &str {
adapters::adapter(agent).map_or(agent, |descriptor| descriptor.product)
}
#[cfg(test)]
mod tests;