diff options
| author | elvis <elvis@claros.ar> | 2026-09-06 22:05:14 -0300 |
|---|---|---|
| committer | elvis <elvis@claros.ar> | 2026-09-06 22:05:14 -0300 |
| commit | 71764752f28d31028b9c2ca486c0fc1bcea1cf95 (patch) | |
| tree | ffd463969826157819f5a3319e1895c3f51deb22 /crates/asist-tools | |
| parent | f8f98e83481e7376a235fb305a095552433f79ab (diff) | |
| download | asist-p-71764752f28d31028b9c2ca486c0fc1bcea1cf95.tar.gz asist-p-71764752f28d31028b9c2ca486c0fc1bcea1cf95.zip | |
Web search and camera vision
Dos capacidades nuevas en un crate aparte, asist-tools, registradas desde
asist-app: son el ejemplo de que el punto de extensión documentado funciona
sin tocar el orquestador.
Buscar admite tres buscadores intercambiables. Tavily devuelve una respuesta ya
redactada, que es lo que se puede leer en voz alta sin gastar otra vuelta del
modelo en resumir (0,54 s para «capital de Australia»); ddgs no necesita clave
y devuelve fragmentos que el modelo sintetiza (2,24 s). El tercero es un
backend de comando genérico: ejecuta un programa y lee JSON de su salida, así
que añadir otro buscador —o un puente a un servidor MCP— es escribir un guion.
Sobre DuckDuckGo, comprobado y no supuesto: curl contra html.duckduckgo.com
devuelve HTTP 202 con una página anti-bot y ni un resultado, y la API sin clave
api.duckduckgo.com devuelve vacío para casi todo lo que no sea una entidad de
enciclopedia. La librería ddgs —la misma que hay bajo duckduckgo-mcp— sí
funciona porque rota buscadores y cabeceras, y es la que usa
scripts/buscar-ddgs.sh.
Mirar aprovecha que el servidor ya carga el proyector multimodal: el mismo
modelo que conversa describe lo que capta la cámara. La pregunta del usuario
viaja hasta ahí, porque «¿de qué color es mi camiseta?» y «¿cuánta gente hay?»
necesitan la misma imagen y descripciones distintas. La imagen no entra en el
historial —cientos de tokens por turno para algo ya descrito— ni toca el disco.
Tres cosas más que salieron de medir, en docs/RENDIMIENTO.md:
- La resolución de captura manda en la latencia: 7,8 s a 1280x720 frente a
2,9 s a 640x480, con la misma descripción útil. 640x480 pasa a ser el valor
por defecto.
- Una herramienta ejecutándose en silencio deja el turno seis segundos mudo y
parece un cuelgue. Tool::acknowledgement pronuncia una frase antes de
ejecutar y baja el primer audio de ~8,4 s a 4,1 s.
- Con la instrucción de voz a secas en la pasada de redacción, el modelo
anunciaba lo que acababa de hacer en vez de contar lo que averiguó, ignorando
el resultado que tenía delante. general.tool_result_prompt lo corrige; ahí sí
se puede añadir estilo sin riesgo, porque la llamada ya ocurrió.
Con tres herramientas declaradas el modelo elige bien entre ellas (15 de 16
medido), pero le cuesta abstenerse y busca cosas que ya sabe. Intentar
corregirlo con instrucciones empeora los aciertos sin reducir los falsos
positivos, coherente con la fragilidad ya documentada.
Corregido además un fallo que dejaba mudo al asistente tras la primera
respuesta: el anillo de reproducción no bajaba nunca su bandera de actividad,
así que el segmentador mantenía el micrófono cerrado creyendo que seguía
hablando. El segmentador se guía ahora por la cola, que no puede desfasarse, y
hay pruebas del anillo con un mando sin dispositivo detrás.
Claude-Session: https://claude.ai/code/session_01FNxz5cSdQSscJH9H7b8uGU
Diffstat (limited to 'crates/asist-tools')
| -rw-r--r-- | crates/asist-tools/Cargo.toml | 16 | ||||
| -rw-r--r-- | crates/asist-tools/src/camera.rs | 290 | ||||
| -rw-r--r-- | crates/asist-tools/src/lib.rs | 14 | ||||
| -rw-r--r-- | crates/asist-tools/src/search.rs | 489 |
4 files changed, 809 insertions, 0 deletions
diff --git a/crates/asist-tools/Cargo.toml b/crates/asist-tools/Cargo.toml new file mode 100644 index 0000000..b0cc51a --- /dev/null +++ b/crates/asist-tools/Cargo.toml @@ -0,0 +1,16 @@ +[package] +name = "asist-tools" +version.workspace = true +edition.workspace = true +rust-version.workspace = true +license.workspace = true + +[dependencies] +asist-core.workspace = true +asist-llm.workspace = true +serde_json.workspace = true +tracing.workspace = true +# Cliente con TLS sólo para las peticiones que salen a internet. El de +# asist-core es HTTP plano a propósito —vale para localhost y permite cortar +# una descarga a media frase—, pero un buscador está detrás de HTTPS. +ureq = { version = "3", default-features = false, features = ["rustls", "json", "gzip"] } diff --git a/crates/asist-tools/src/camera.rs b/crates/asist-tools/src/camera.rs new file mode 100644 index 0000000..fe41319 --- /dev/null +++ b/crates/asist-tools/src/camera.rs @@ -0,0 +1,290 @@ +//! Mirar por la cámara. +//! +//! El servidor ya tiene cargado el proyector multimodal, así que el mismo +//! modelo que conversa puede describir una imagen. La herramienta hace tres +//! cosas: capturar un fotograma, preguntarle al modelo por él y devolver la +//! respuesta como texto. +//! +//! Que la pregunta viaje hasta la cámara importa: «¿de qué color es mi +//! camiseta?» y «¿cuánta gente hay?» necesitan la misma imagen pero +//! descripciones muy distintas, y pedir una descripción genérica para luego +//! interrogarla pierde justo el detalle que se buscaba. +//! +//! La captura no se guarda en disco salvo que se pida expresamente: un +//! asistente que deja fotogramas por ahí es un problema de privacidad, no una +//! comodidad de depuración. + +use std::path::PathBuf; +use std::process::{Command, Stdio}; +use std::sync::Arc; +use std::time::{Duration, Instant}; + +use asist_core::error::{Error, Result}; +use asist_core::http::Cancel; +use asist_core::tools::Tool; +use asist_llm::LlmClient; +use serde_json::{json, Value}; + +/// Cómo se captura el fotograma. +#[derive(Debug, Clone)] +pub struct CaptureConfig { + /// Dispositivo V4L2. + pub device: PathBuf, + pub width: u32, + pub height: u32, + /// Fotogramas que se descartan antes de quedarse con uno. + /// + /// La cámara arranca con la exposición automática sin asentar y el primer + /// fotograma suele salir quemado. Descartar unos pocos es prácticamente + /// gratis —medido, 0,45 s frente a 0,53 s— y en penumbra se nota. + pub warmup_frames: u32, + /// Plazo máximo de la captura. + pub timeout: Duration, + /// Carpeta donde dejar los fotogramas. Vacío = no se guarda ninguno. + pub save_dir: Option<PathBuf>, +} + +impl Default for CaptureConfig { + fn default() -> Self { + Self { + device: PathBuf::from("/dev/video0"), + // 640x480 es el punto de equilibrio medido: el modelo tarda 2,9 s + // y sigue describiendo bien. A 1280x720 tarda 7,8 s, y a 320x240 + // baja a 1,3 s pero deja de distinguir detalles. + width: 640, + height: 480, + warmup_frames: 5, + timeout: Duration::from_secs(15), + save_dir: None, + } + } +} + +/// Captura un fotograma en JPEG. +/// +/// Se apoya en ffmpeg en vez de hablar con V4L2 directamente: una cámara USB +/// entrega MJPEG, YUYV o lo que le parezca, y reimplementar esa negociación +/// para ahorrarse un proceso no sale a cuenta. +pub fn capture(config: &CaptureConfig) -> Result<Vec<u8>> { + let fail = |message: String| Error::Tool { + tool: "mirar_por_la_camara".into(), + message, + }; + + if !config.device.exists() { + return Err(fail(format!( + "no existe el dispositivo {}. Comprueba con «v4l2-ctl --list-devices»", + config.device.display() + ))); + } + + let started = Instant::now(); + let mut command = Command::new("ffmpeg"); + command + .args(["-hide_banner", "-loglevel", "error", "-nostdin"]) + .args(["-f", "v4l2"]) + .args([ + "-video_size", + &format!("{}x{}", config.width, config.height), + ]) + .arg("-i") + .arg(&config.device); + + if config.warmup_frames > 0 { + // Se leen N fotogramas y se conserva el último: es la forma de dejar + // que la exposición se asiente sin abrir el dispositivo dos veces. + command.args(["-vf", &format!("select=eq(n\\,{})", config.warmup_frames)]); + } + command + .args(["-frames:v", "1"]) + .args(["-f", "image2", "-c:v", "mjpeg"]) + .arg("-") + .stdin(Stdio::null()) + .stdout(Stdio::piped()) + .stderr(Stdio::piped()); + + let mut child = command.spawn().map_err(|e| { + fail(format!( + "no se pudo ejecutar ffmpeg ({e}); hace falta para leer la cámara" + )) + })?; + + let deadline = Instant::now() + config.timeout; + loop { + match child.try_wait().map_err(|e| fail(e.to_string()))? { + Some(_) => break, + None if Instant::now() >= deadline => { + let _ = child.kill(); + let _ = child.wait(); + return Err(fail(format!( + "la cámara no respondió en {} s", + config.timeout.as_secs() + ))); + } + None => std::thread::sleep(Duration::from_millis(20)), + } + } + + let output = child.wait_with_output().map_err(|e| fail(e.to_string()))?; + if !output.status.success() || output.stdout.is_empty() { + let stderr = String::from_utf8_lossy(&output.stderr); + let hint = if stderr.contains("Permission denied") { + ". Tu usuario necesita estar en el grupo «video»" + } else if stderr.contains("Device or resource busy") { + ". Otra aplicación está usando la cámara" + } else { + "" + }; + return Err(fail(format!( + "ffmpeg no capturó nada{hint}: {}", + stderr.trim().lines().next_back().unwrap_or("sin detalles") + ))); + } + + if let Some(dir) = &config.save_dir { + // Sólo si se ha pedido: por defecto la imagen no toca el disco. + if let Err(err) = std::fs::create_dir_all(dir).and_then(|()| { + let name = format!( + "frame-{}.jpg", + std::time::SystemTime::now() + .duration_since(std::time::UNIX_EPOCH) + .map(|d| d.as_secs()) + .unwrap_or(0) + ); + std::fs::write(dir.join(name), &output.stdout) + }) { + tracing::warn!(target: "camara", %err, "no se pudo guardar el fotograma"); + } + } + + tracing::info!( + target: "camara", + dispositivo = %config.device.display(), + resolucion = format!("{}x{}", config.width, config.height), + kb = output.stdout.len() / 1024, + ms = started.elapsed().as_millis(), + "fotograma capturado" + ); + Ok(output.stdout) +} + +pub struct Camera { + config: CaptureConfig, + llm: Arc<LlmClient>, +} + +impl Camera { + pub fn new(config: CaptureConfig, llm: Arc<LlmClient>) -> Self { + Self { config, llm } + } + + /// `true` si el dispositivo está presente. Sin esto no tiene sentido + /// declarar la herramienta: el modelo la llamaría y fallaría siempre. + pub fn available(config: &CaptureConfig) -> bool { + config.device.exists() + } +} + +impl Tool for Camera { + fn name(&self) -> &str { + "mirar_por_la_camara" + } + + fn description(&self) -> &str { + "Toma una foto con la cámara del equipo y responde a una pregunta sobre \ + lo que se ve. Úsala cuando te pregunten qué ves, qué hay delante, de qué \ + color es algo o cuántas cosas hay." + } + + fn parameters(&self) -> Value { + json!({ + "type": "object", + "properties": { + "pregunta": { + "type": "string", + "description": "La pregunta del usuario tal cual, sin concretarla más de lo que él dijo. Si sólo quiere saber qué hay delante, pon: ¿Qué se ve?" + } + }, + "required": ["pregunta"] + }) + } + + /// Enciende la cámara, así que se anuncia como tal. + fn is_side_effecting(&self) -> bool { + true + } + + fn acknowledgement(&self) -> Option<&str> { + Some("Voy a mirar.") + } + + fn call(&self, args: &Value) -> Result<String> { + let question = args + .get("pregunta") + .and_then(Value::as_str) + .map(str::trim) + .filter(|q| !q.is_empty()) + .unwrap_or("¿Qué se ve en esta imagen?"); + + let frame = capture(&self.config)?; + + // Se le pide al modelo el estilo hablado aquí y no en la conversación: + // esta petición va fuera del historial, así que la instrucción de voz + // del asistente no le llega. + let prompt = format!( + "{question}\n\nResponde en una o dos frases cortas en español, en texto \ + plano, describiendo sólo lo que se ve de verdad en la imagen. Si no se \ + distingue, dilo." + ); + let started = Instant::now(); + let answer = self.llm.look(&frame, &prompt, &Cancel::new())?; + tracing::info!( + target: "camara", + ms = started.elapsed().as_millis(), + "el modelo describió el fotograma" + ); + Ok(answer) + } +} + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn un_dispositivo_inexistente_se_detecta_antes_de_registrar_la_herramienta() { + let config = CaptureConfig { + device: PathBuf::from("/dev/video-que-no-existe"), + ..Default::default() + }; + assert!(!Camera::available(&config)); + } + + #[test] + fn capturar_de_un_dispositivo_inexistente_da_un_error_util() { + let config = CaptureConfig { + device: PathBuf::from("/dev/video-que-no-existe"), + ..Default::default() + }; + let err = capture(&config).unwrap_err().to_string(); + assert!(err.contains("no existe el dispositivo"), "{err}"); + assert!( + err.contains("v4l2-ctl"), + "el error debe decir cómo comprobarlo: {err}" + ); + } + + #[test] + fn por_defecto_no_se_guarda_ningun_fotograma() { + assert!( + CaptureConfig::default().save_dir.is_none(), + "guardar imágenes por defecto sería una fuga de privacidad" + ); + } + + #[test] + fn la_resolucion_por_defecto_es_la_medida_como_equilibrada() { + let config = CaptureConfig::default(); + assert_eq!((config.width, config.height), (640, 480)); + } +} diff --git a/crates/asist-tools/src/lib.rs b/crates/asist-tools/src/lib.rs new file mode 100644 index 0000000..796b3d8 --- /dev/null +++ b/crates/asist-tools/src/lib.rs @@ -0,0 +1,14 @@ +//! Herramientas que asoman el asistente al mundo: buscar en internet y mirar +//! por la cámara. +//! +//! Viven en un crate aparte de `asist-core` porque necesitan cosas que el +//! núcleo no debe arrastrar —un cliente con TLS, el modelo multimodal, el +//! dispositivo de vídeo—, y porque son el ejemplo de que el punto de extensión +//! funciona: se registran con `ToolRegistry::register` sin tocar el +//! orquestador. + +pub mod camera; +pub mod search; + +pub use camera::Camera; +pub use search::{SearchBackend, WebSearch}; diff --git a/crates/asist-tools/src/search.rs b/crates/asist-tools/src/search.rs new file mode 100644 index 0000000..b69c17d --- /dev/null +++ b/crates/asist-tools/src/search.rs @@ -0,0 +1,489 @@ +//! Búsqueda en internet. +//! +//! El resultado se va a leer en voz alta, así que lo que interesa no es una +//! lista de enlaces sino una respuesta. Por eso se prefiere un buscador que +//! sintetice —Tavily devuelve un párrafo ya redactado— y los titulares sólo +//! acompañan como respaldo cuando no hay síntesis. + +use std::path::PathBuf; +use std::process::{Command, Stdio}; +use std::time::{Duration, Instant}; + +use asist_core::error::{Error, Result}; +use asist_core::tools::Tool; +use serde_json::{json, Value}; + +/// De dónde salen los resultados. +#[derive(Debug, Clone)] +pub enum SearchBackend { + /// API de Tavily. Devuelve una respuesta ya redactada además de los + /// resultados, que es justo lo que hace falta para hablarla. + Tavily { api_key: String }, + /// Instancia de SearXNG, propia o de confianza. Sin clave, pero devuelve + /// sólo resultados: la síntesis la tiene que hacer el modelo. + SearxNG { base_url: String }, + /// Un programa externo que imprime los resultados en JSON. + /// + /// Es la vía sin clave: `scripts/buscar-ddgs.sh` consulta DuckDuckGo y + /// compañía a través de la librería `ddgs`. Vale para cualquier otra cosa + /// que escriba JSON por la salida estándar —un puente a un servidor MCP, + /// un buscador interno—, así que también es el punto de extensión del + /// apartado de búsqueda. + /// + /// En `args`, `{consulta}` y `{max}` se sustituyen antes de ejecutar. + Command { program: PathBuf, args: Vec<String> }, +} + +impl SearchBackend { + /// Backend sin clave por omisión: el guion que envuelve a ddgs. + pub fn ddgs(script: impl Into<PathBuf>) -> Self { + SearchBackend::Command { + program: script.into(), + args: vec!["{consulta}".into(), "{max}".into()], + } + } +} + +impl SearchBackend { + pub fn label(&self) -> &'static str { + match self { + SearchBackend::Tavily { .. } => "Tavily", + SearchBackend::SearxNG { .. } => "SearXNG", + SearchBackend::Command { .. } => "comando", + } + } +} + +pub struct WebSearch { + backend: SearchBackend, + max_results: usize, + timeout: Duration, +} + +impl WebSearch { + pub fn new(backend: SearchBackend, max_results: usize, timeout: Duration) -> Self { + Self { + backend, + max_results: max_results.clamp(1, 10), + timeout, + } + } + + pub fn backend(&self) -> &SearchBackend { + &self.backend + } + + fn agent(&self) -> ureq::Agent { + ureq::Agent::config_builder() + .timeout_global(Some(self.timeout)) + .build() + .into() + } + + fn fail(message: impl Into<String>) -> Error { + Error::Tool { + tool: "buscar_en_internet".into(), + message: message.into(), + } + } + + fn tavily(&self, key: &str, query: &str) -> Result<String> { + let body = json!({ + "query": query, + "max_results": self.max_results, + "search_depth": "basic", + // La respuesta redactada es la razón de usar este buscador: sin + // ella habría que gastar otra vuelta del modelo en resumir. + "include_answer": true, + }); + let mut response = self + .agent() + .post("https://api.tavily.com/search") + .header("Authorization", &format!("Bearer {key}")) + .send_json(&body) + .map_err(|e| Self::fail(describe_ureq(&e)))?; + + let parsed: Value = response + .body_mut() + .read_json() + .map_err(|e| Self::fail(format!("respuesta ilegible: {e}")))?; + + let (answer, results) = extract(&parsed, self.max_results); + compose(&answer, &results) + } + + /// Ejecuta el programa configurado y traduce lo que imprima. + fn command(&self, program: &std::path::Path, args: &[String], query: &str) -> Result<String> { + let rendered: Vec<String> = args + .iter() + .map(|arg| { + arg.replace("{consulta}", query) + .replace("{max}", &self.max_results.to_string()) + }) + .collect(); + + // Los argumentos van al `execve` tal cual: la consulta sale de lo que + // se ha oído por el micrófono, y no puede acabar interpretada por una + // shell. + let mut child = Command::new(program) + .args(&rendered) + .stdin(Stdio::null()) + .stdout(Stdio::piped()) + .stderr(Stdio::piped()) + .spawn() + .map_err(|e| Self::fail(format!("no se pudo ejecutar {}: {e}", program.display())))?; + + let deadline = Instant::now() + self.timeout; + loop { + match child.try_wait().map_err(|e| Self::fail(e.to_string()))? { + Some(_) => break, + None if Instant::now() >= deadline => { + let _ = child.kill(); + let _ = child.wait(); + return Err(Self::fail(format!( + "el buscador tardó más de {} s", + self.timeout.as_secs() + ))); + } + None => std::thread::sleep(Duration::from_millis(20)), + } + } + + let output = child + .wait_with_output() + .map_err(|e| Self::fail(e.to_string()))?; + if !output.status.success() { + let stderr = String::from_utf8_lossy(&output.stderr); + return Err(Self::fail(format!( + "el buscador falló: {}", + stderr.trim().lines().next_back().unwrap_or("sin detalles") + ))); + } + + let parsed: Value = serde_json::from_slice(&output.stdout) + .map_err(|e| Self::fail(format!("el buscador no devolvió JSON válido: {e}")))?; + let (answer, results) = extract(&parsed, self.max_results); + compose(&answer, &results) + } + + fn searxng(&self, base_url: &str, query: &str) -> Result<String> { + let url = format!( + "{}/search?q={}&format=json&language=es", + base_url.trim_end_matches('/'), + urlencode(query) + ); + let mut response = self + .agent() + .get(&url) + .call() + .map_err(|e| Self::fail(describe_ureq(&e)))?; + + let parsed: Value = response + .body_mut() + .read_json() + .map_err(|e| Self::fail(format!("respuesta ilegible: {e}")))?; + + let (answer, results) = extract(&parsed, self.max_results); + compose(&answer, &results) + } +} + +impl Tool for WebSearch { + fn name(&self) -> &str { + "buscar_en_internet" + } + + fn description(&self) -> &str { + "Busca información actual en internet y devuelve un resumen. Úsala para \ + noticias, precios, resultados, el tiempo o cualquier cosa posterior a tu \ + entrenamiento, en vez de responder de memoria." + } + + fn parameters(&self) -> Value { + json!({ + "type": "object", + "properties": { + "consulta": { + "type": "string", + "description": "Qué buscar, en lenguaje natural. Por ejemplo: tiempo en Buenos Aires mañana" + } + }, + "required": ["consulta"] + }) + } + + fn acknowledgement(&self) -> Option<&str> { + Some("Déjame que lo busque.") + } + + fn call(&self, args: &Value) -> Result<String> { + let query = args + .get("consulta") + .and_then(Value::as_str) + .map(str::trim) + .filter(|q| !q.is_empty()) + .ok_or_else(|| Self::fail("falta «consulta»"))?; + + let started = Instant::now(); + let result = match &self.backend { + SearchBackend::Tavily { api_key } => self.tavily(api_key, query), + SearchBackend::SearxNG { base_url } => self.searxng(base_url, query), + SearchBackend::Command { program, args } => self.command(program, args, query), + }; + tracing::info!( + target: "herramientas", + buscador = self.backend.label(), + consulta = query, + ms = started.elapsed().as_millis(), + ok = result.is_ok(), + "búsqueda" + ); + result + } +} + +/// Saca respuesta y resultados de un JSON sin exigir una forma concreta. +/// +/// Cada buscador nombra los campos a su manera —`href` o `url`, `body` o +/// `content`, la lista suelta o dentro de `results`— y aquí lo que interesa es +/// que un guion nuevo funcione sin tener que tocar Rust. +fn extract(parsed: &Value, max: usize) -> (String, Vec<String>) { + let answer = parsed + .get("answer") + .and_then(Value::as_str) + .unwrap_or("") + .trim() + .to_string(); + + let rows = parsed + .as_array() + .or_else(|| parsed.get("results").and_then(Value::as_array)); + + let results = rows + .map(|items| { + items + .iter() + .take(max) + .filter_map(|item| { + let pick = |keys: &[&str]| { + keys.iter() + .find_map(|k| item.get(*k).and_then(Value::as_str)) + .unwrap_or("") + .trim() + .to_string() + }; + let title = pick(&["title", "titulo", "name"]); + let snippet = pick(&["body", "content", "snippet", "description"]); + if title.is_empty() && snippet.is_empty() { + return None; + } + Some(if title.is_empty() { + clip(&snippet, 200) + } else { + format!("{title}: {}", clip(&snippet, 200)) + }) + }) + .collect() + }) + .unwrap_or_default(); + + (answer, results) +} + +/// Junta la respuesta sintetizada con los titulares. +/// +/// La síntesis va primero porque es lo que probablemente se pronuncie; los +/// titulares quedan detrás para que el modelo tenga de dónde tirar si la +/// pregunta pedía un detalle que el resumen no cubre. +fn compose(answer: &str, results: &[String]) -> Result<String> { + if answer.is_empty() && results.is_empty() { + return Err(Error::Tool { + tool: "buscar_en_internet".into(), + message: "la búsqueda no devolvió nada".into(), + }); + } + let mut out = String::new(); + if !answer.is_empty() { + out.push_str(answer); + } + if !results.is_empty() { + if !out.is_empty() { + out.push_str("\n\nFuentes:\n"); + } + for (i, result) in results.iter().enumerate() { + out.push_str(&format!("{}. {result}\n", i + 1)); + } + } + Ok(out.trim().to_string()) +} + +fn clip(text: &str, max: usize) -> String { + let flat = text.split_whitespace().collect::<Vec<_>>().join(" "); + if flat.chars().count() <= max { + return flat; + } + flat.chars().take(max).collect::<String>() + "…" +} + +/// Un error de ureq trae la cadena completa con la URL dentro, y en Tavily esa +/// URL no lleva la clave —va en la cabecera—, pero más vale no acostumbrarse: +/// se resume el error en vez de volcarlo entero. +fn describe_ureq(err: &ureq::Error) -> String { + match err { + ureq::Error::StatusCode(code) => match code { + 401 | 403 => format!("el buscador rechazó la clave (HTTP {code})"), + 429 => "se ha superado el límite de consultas del buscador".into(), + other => format!("el buscador respondió HTTP {other}"), + }, + ureq::Error::Timeout(_) => "el buscador tardó demasiado".into(), + other => format!("no se pudo consultar el buscador: {other}"), + } +} + +fn urlencode(text: &str) -> String { + let mut out = String::with_capacity(text.len()); + for byte in text.bytes() { + match byte { + b'A'..=b'Z' | b'a'..=b'z' | b'0'..=b'9' | b'-' | b'_' | b'.' | b'~' => { + out.push(byte as char) + } + b' ' => out.push('+'), + other => out.push_str(&format!("%{other:02X}")), + } + } + out +} + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn la_respuesta_sintetizada_va_delante_de_las_fuentes() { + let out = compose("Hace 17 grados.", &["Meteored: parcialmente nuboso".into()]).unwrap(); + assert!(out.starts_with("Hace 17 grados.")); + assert!(out.contains("Fuentes:")); + } + + #[test] + fn sin_sintesis_valen_los_titulares() { + let out = compose("", &["Uno: algo".into(), "Dos: otra cosa".into()]).unwrap(); + assert!(out.starts_with("1. Uno")); + assert!(out.contains("2. Dos")); + } + + #[test] + fn una_busqueda_sin_resultados_es_un_error_y_no_una_cadena_vacia() { + // Si devolviera "" el modelo se inventaría la respuesta creyendo que + // la herramienta funcionó. + assert!(compose("", &[]).is_err()); + } + + #[test] + fn los_fragmentos_largos_se_recortan() { + let largo = "palabra ".repeat(80); + assert!(clip(&largo, 100).chars().count() <= 101); + assert_eq!(clip("corto", 100), "corto"); + } + + #[test] + fn los_saltos_de_linea_de_los_fragmentos_se_aplanan() { + assert_eq!(clip("uno\n\n dos", 100), "uno dos"); + } + + #[test] + fn la_consulta_se_codifica_para_la_url() { + assert_eq!(urlencode("qué tiempo hace"), "qu%C3%A9+tiempo+hace"); + assert_eq!(urlencode("a&b=c"), "a%26b%3Dc"); + } + + #[test] + fn una_consulta_vacia_se_rechaza_sin_salir_a_la_red() { + let tool = WebSearch::new( + SearchBackend::Tavily { + api_key: "x".into(), + }, + 3, + Duration::from_secs(1), + ); + assert!(tool.call(&json!({ "consulta": " " })).is_err()); + assert!(tool.call(&json!({})).is_err()); + } + + #[test] + fn se_entiende_la_lista_suelta_que_devuelve_ddgs() { + let raw = serde_json::json!([ + { "title": "Canberra", "href": "https://x", "body": "es la capital" }, + { "title": "Sídney", "href": "https://y", "body": "no lo es" } + ]); + let (answer, results) = extract(&raw, 5); + assert!(answer.is_empty(), "ddgs no sintetiza"); + assert_eq!(results.len(), 2); + assert!(results[0].starts_with("Canberra: es la capital")); + } + + #[test] + fn se_entiende_tambien_la_forma_con_results_y_answer() { + let raw = serde_json::json!({ + "answer": "Canberra.", + "results": [{ "title": "T", "url": "https://x", "content": "C" }] + }); + let (answer, results) = extract(&raw, 5); + assert_eq!(answer, "Canberra."); + assert_eq!(results, vec!["T: C"]); + } + + #[test] + fn se_respeta_el_maximo_de_resultados() { + let raw = serde_json::json!([ + { "title": "1", "body": "a" }, { "title": "2", "body": "b" }, + { "title": "3", "body": "c" } + ]); + assert_eq!(extract(&raw, 2).1.len(), 2); + } + + #[test] + fn una_fila_sin_titulo_ni_texto_se_ignora() { + let raw = serde_json::json!([{ "href": "https://x" }, { "title": "T", "body": "C" }]); + assert_eq!(extract(&raw, 5).1, vec!["T: C"]); + } + + #[test] + fn los_marcadores_del_comando_se_sustituyen() { + let tool = WebSearch::new(SearchBackend::ddgs("/bin/echo"), 4, Duration::from_secs(5)); + let SearchBackend::Command { args, .. } = tool.backend() else { + panic!("esperaba un backend de comando"); + }; + let rendered: Vec<String> = args + .iter() + .map(|a| a.replace("{consulta}", "hola").replace("{max}", "4")) + .collect(); + assert_eq!(rendered, vec!["hola", "4"]); + } + + #[test] + fn un_comando_que_no_existe_da_un_error_util() { + let tool = WebSearch::new( + SearchBackend::ddgs("/no/existe/buscador.sh"), + 3, + Duration::from_secs(5), + ); + let err = tool + .call(&json!({ "consulta": "algo" })) + .unwrap_err() + .to_string(); + assert!(err.contains("no se pudo ejecutar"), "{err}"); + } + + #[test] + fn el_numero_de_resultados_se_mantiene_en_un_rango_sensato() { + let tool = WebSearch::new( + SearchBackend::Tavily { + api_key: "x".into(), + }, + 99, + Duration::from_secs(1), + ); + assert_eq!(tool.max_results, 10); + } +} |