zeph_core/agent/

agent_access_impl.rs

// SPDX-FileCopyrightText: 2026 Andrei G <bug-ops>
// SPDX-License-Identifier: MIT OR Apache-2.0

//! Implementation of [`zeph_commands::traits::agent::AgentAccess`] for [`Agent<C>`].
//!
//! Each method in `AgentAccess` returns a formatted `String` result (without sending to the
//! channel directly), so that `CommandContext::sink` does not conflict with this borrow.
//! The one exception is methods for subsystems that are already channel-free (memory, graph).
//!
//! [`Agent<C>`]: super::Agent

use std::fmt::Write as _;
use std::future::Future;
use std::pin::Pin;

use zeph_commands::CommandError;
use zeph_commands::traits::agent::AgentAccess;
use zeph_memory::{GraphExtractionConfig, MessageId, extract_and_store};

use super::{Agent, error::AgentError};
use crate::channel::Channel;

impl<C: Channel + Send + 'static> AgentAccess for Agent<C> {
    // ----- /memory -----

    fn memory_tiers<'a>(
        &'a mut self,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            let Some(memory) = self.memory_state.persistence.memory.clone() else {
                return Ok("Memory not configured.".to_owned());
            };
            match memory.sqlite().count_messages_by_tier().await {
                Ok((episodic, semantic)) => {
                    let mut out = String::new();
                    let _ = writeln!(out, "Memory tiers:");
                    let _ = writeln!(out, "  Working:  (current context window — virtual)");
                    let _ = writeln!(out, "  Episodic: {episodic} messages");
                    let _ = writeln!(out, "  Semantic: {semantic} facts");
                    Ok(out.trim_end().to_owned())
                }
                Err(e) => Ok(format!("Failed to query tier stats: {e}")),
            }
        })
    }

    fn memory_promote<'a>(
        &'a mut self,
        ids_str: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            let Some(memory) = self.memory_state.persistence.memory.clone() else {
                return Ok("Memory not configured.".to_owned());
            };
            let ids: Vec<MessageId> = ids_str
                .split_whitespace()
                .filter_map(|s| s.parse::<i64>().ok().map(MessageId))
                .collect();
            if ids.is_empty() {
                return Ok(
                    "Usage: /memory promote <id> [id...]\nExample: /memory promote 42 43 44"
                        .to_owned(),
                );
            }
            match memory.sqlite().manual_promote(&ids).await {
                Ok(count) => Ok(format!("Promoted {count} message(s) to semantic tier.")),
                Err(e) => Ok(format!("Promotion failed: {e}")),
            }
        })
    }

    // ----- /graph -----

    fn graph_stats<'a>(
        &'a mut self,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            let Some(memory) = self.memory_state.persistence.memory.as_ref() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };
            let Some(store) = memory.graph_store.as_ref() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };

            let (entities, edges, communities, distribution) = tokio::join!(
                store.entity_count(),
                store.active_edge_count(),
                store.community_count(),
                store.edge_type_distribution()
            );
            let mut msg = format!(
                "Graph memory: {} entities, {} edges, {} communities",
                entities.unwrap_or(0),
                edges.unwrap_or(0),
                communities.unwrap_or(0)
            );
            if let Ok(dist) = distribution
                && !dist.is_empty()
            {
                let dist_str: Vec<String> = dist.iter().map(|(t, c)| format!("{t}={c}")).collect();
                write!(msg, "\nEdge types: {}", dist_str.join(", ")).unwrap_or(());
            }
            Ok(msg)
        })
    }

    fn graph_entities<'a>(
        &'a mut self,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            let Some(memory) = self.memory_state.persistence.memory.as_ref() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };
            let Some(store) = memory.graph_store.as_ref() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };

            let entities = store
                .all_entities()
                .await
                .map_err(|e| CommandError::new(e.to_string()))?;
            if entities.is_empty() {
                return Ok("No entities found.".to_owned());
            }

            let total = entities.len();
            let display: Vec<String> = entities
                .iter()
                .take(50)
                .map(|e| {
                    format!(
                        "  {:<40}  {:<15}  {}",
                        e.name,
                        e.entity_type.as_str(),
                        e.last_seen_at.split('T').next().unwrap_or(&e.last_seen_at)
                    )
                })
                .collect();
            let mut msg = format!(
                "Entities ({total} total):\n  {:<40}  {:<15}  {}\n{}",
                "NAME",
                "TYPE",
                "LAST SEEN",
                display.join("\n")
            );
            if total > 50 {
                write!(msg, "\n  ...and {} more", total - 50).unwrap_or(());
            }
            Ok(msg)
        })
    }

    fn graph_facts<'a>(
        &'a mut self,
        name: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            let Some(memory) = self.memory_state.persistence.memory.as_ref() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };
            let Some(store) = memory.graph_store.as_ref() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };

            let matches = store
                .find_entity_by_name(name)
                .await
                .map_err(|e| CommandError::new(e.to_string()))?;
            if matches.is_empty() {
                return Ok(format!("No entity found matching '{name}'."));
            }

            let entity = &matches[0];
            let edges = store
                .edges_for_entity(entity.id)
                .await
                .map_err(|e| CommandError::new(e.to_string()))?;
            if edges.is_empty() {
                return Ok(format!("Entity '{}' has no known facts.", entity.name));
            }

            let mut entity_names: std::collections::HashMap<i64, String> =
                std::collections::HashMap::new();
            entity_names.insert(entity.id, entity.name.clone());
            for edge in &edges {
                let other_id = if edge.source_entity_id == entity.id {
                    edge.target_entity_id
                } else {
                    edge.source_entity_id
                };
                entity_names.entry(other_id).or_default();
            }
            for (&id, name_val) in &mut entity_names {
                if name_val.is_empty() {
                    if let Ok(Some(other)) = store.find_entity_by_id(id).await {
                        *name_val = other.name;
                    } else {
                        *name_val = format!("#{id}");
                    }
                }
            }

            let lines: Vec<String> = edges
                .iter()
                .map(|e| {
                    let src = entity_names
                        .get(&e.source_entity_id)
                        .cloned()
                        .unwrap_or_else(|| format!("#{}", e.source_entity_id));
                    let tgt = entity_names
                        .get(&e.target_entity_id)
                        .cloned()
                        .unwrap_or_else(|| format!("#{}", e.target_entity_id));
                    format!(
                        "  {} --[{}/{}]--> {}: {} (confidence: {:.2})",
                        src, e.relation, e.edge_type, tgt, e.fact, e.confidence
                    )
                })
                .collect();
            Ok(format!(
                "Facts for '{}':\n{}",
                entity.name,
                lines.join("\n")
            ))
        })
    }

    fn graph_history<'a>(
        &'a mut self,
        name: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            let Some(memory) = self.memory_state.persistence.memory.as_ref() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };
            let Some(store) = memory.graph_store.as_ref() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };

            let matches = store
                .find_entity_by_name(name)
                .await
                .map_err(|e| CommandError::new(e.to_string()))?;
            if matches.is_empty() {
                return Ok(format!("No entity found matching '{name}'."));
            }

            let entity = &matches[0];
            let edges = store
                .edge_history_for_entity(entity.id, 50)
                .await
                .map_err(|e| CommandError::new(e.to_string()))?;
            if edges.is_empty() {
                return Ok(format!("Entity '{}' has no edge history.", entity.name));
            }

            let mut entity_names: std::collections::HashMap<i64, String> =
                std::collections::HashMap::new();
            entity_names.insert(entity.id, entity.name.clone());
            for edge in &edges {
                for &id in &[edge.source_entity_id, edge.target_entity_id] {
                    entity_names.entry(id).or_default();
                }
            }
            for (&id, name_val) in &mut entity_names {
                if name_val.is_empty() {
                    if let Ok(Some(other)) = store.find_entity_by_id(id).await {
                        *name_val = other.name;
                    } else {
                        *name_val = format!("#{id}");
                    }
                }
            }

            let n = edges.len();
            let lines: Vec<String> = edges
                .iter()
                .map(|e| {
                    let status = if e.valid_to.is_some() {
                        let date = e
                            .valid_to
                            .as_deref()
                            .and_then(|s| s.split('T').next().or_else(|| s.split(' ').next()))
                            .unwrap_or("?");
                        format!("[expired {date}]")
                    } else {
                        "[active]".to_string()
                    };
                    let src = entity_names
                        .get(&e.source_entity_id)
                        .cloned()
                        .unwrap_or_else(|| format!("#{}", e.source_entity_id));
                    let tgt = entity_names
                        .get(&e.target_entity_id)
                        .cloned()
                        .unwrap_or_else(|| format!("#{}", e.target_entity_id));
                    format!(
                        "  {status} {} --[{}/{}]--> {}: {} (confidence: {:.2})",
                        src, e.relation, e.edge_type, tgt, e.fact, e.confidence
                    )
                })
                .collect();
            Ok(format!(
                "Edge history for '{}' ({n} edges):\n{}",
                entity.name,
                lines.join("\n")
            ))
        })
    }

    fn graph_communities<'a>(
        &'a mut self,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            let Some(memory) = self.memory_state.persistence.memory.as_ref() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };
            let Some(store) = memory.graph_store.as_ref() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };

            let communities = store
                .all_communities()
                .await
                .map_err(|e| CommandError::new(e.to_string()))?;
            if communities.is_empty() {
                return Ok("No communities detected yet. Run graph backfill first.".to_owned());
            }

            let lines: Vec<String> = communities
                .iter()
                .map(|c| format!("  [{}]: {}", c.name, c.summary))
                .collect();
            Ok(format!(
                "Communities ({}):\n{}",
                communities.len(),
                lines.join("\n")
            ))
        })
    }

    fn graph_backfill<'a>(
        &'a mut self,
        limit: Option<usize>,
        progress_cb: &'a mut (dyn FnMut(String) + Send),
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            let Some(memory) = self.memory_state.persistence.memory.clone() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };
            let Some(store) = memory.graph_store.clone() else {
                return Ok("Graph memory is not enabled.".to_owned());
            };

            let total = store.unprocessed_message_count().await.unwrap_or(0);
            let cap = limit.unwrap_or(usize::MAX);

            progress_cb(format!(
                "Starting graph backfill... ({total} unprocessed messages)"
            ));

            let batch_size = 50usize;
            let mut processed = 0usize;
            let mut total_entities = 0usize;
            let mut total_edges = 0usize;

            let graph_cfg = self.memory_state.extraction.graph_config.clone();
            let provider = self.provider.clone();

            loop {
                let remaining_cap = cap.saturating_sub(processed);
                if remaining_cap == 0 {
                    break;
                }
                let batch_limit = batch_size.min(remaining_cap);
                let messages = store
                    .unprocessed_messages_for_backfill(batch_limit)
                    .await
                    .map_err(|e| CommandError::new(e.to_string()))?;
                if messages.is_empty() {
                    break;
                }

                let ids: Vec<zeph_memory::types::MessageId> =
                    messages.iter().map(|(id, _)| *id).collect();

                for (_id, content) in &messages {
                    if content.trim().is_empty() {
                        continue;
                    }
                    let extraction_cfg = GraphExtractionConfig {
                        max_entities: graph_cfg.max_entities_per_message,
                        max_edges: graph_cfg.max_edges_per_message,
                        extraction_timeout_secs: graph_cfg.extraction_timeout_secs,
                        community_refresh_interval: 0,
                        expired_edge_retention_days: graph_cfg.expired_edge_retention_days,
                        max_entities_cap: graph_cfg.max_entities,
                        community_summary_max_prompt_bytes: graph_cfg
                            .community_summary_max_prompt_bytes,
                        community_summary_concurrency: graph_cfg.community_summary_concurrency,
                        lpa_edge_chunk_size: graph_cfg.lpa_edge_chunk_size,
                        note_linking: zeph_memory::NoteLinkingConfig::default(),
                        link_weight_decay_lambda: graph_cfg.link_weight_decay_lambda,
                        link_weight_decay_interval_secs: graph_cfg.link_weight_decay_interval_secs,
                        belief_revision_enabled: graph_cfg.belief_revision.enabled,
                        belief_revision_similarity_threshold: graph_cfg
                            .belief_revision
                            .similarity_threshold,
                        conversation_id: None,
                    };
                    let pool = store.pool().clone();
                    match extract_and_store(
                        content.clone(),
                        vec![],
                        provider.clone(),
                        pool,
                        extraction_cfg,
                        None,
                        None,
                    )
                    .await
                    {
                        Ok(result) => {
                            total_entities += result.stats.entities_upserted;
                            total_edges += result.stats.edges_inserted;
                        }
                        Err(e) => {
                            tracing::warn!("backfill extraction error: {e:#}");
                        }
                    }
                }

                store
                    .mark_messages_graph_processed(&ids)
                    .await
                    .map_err(|e| CommandError::new(e.to_string()))?;
                processed += messages.len();

                progress_cb(format!(
                    "Backfill progress: {processed} messages processed, \
                     {total_entities} entities, {total_edges} edges"
                ));
            }

            Ok(format!(
                "Backfill complete: {total_entities} entities, {total_edges} edges \
                 extracted from {processed} messages"
            ))
        })
    }

    // ----- /guidelines -----

    fn guidelines<'a>(
        &'a mut self,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            const MAX_DISPLAY_CHARS: usize = 4096;

            let Some(memory) = &self.memory_state.persistence.memory else {
                return Ok("No memory backend initialised.".to_owned());
            };

            let cid = self.memory_state.persistence.conversation_id;
            let sqlite = memory.sqlite();

            let (version, text) = sqlite
                .load_compression_guidelines(cid)
                .await
                .map_err(|e: zeph_memory::MemoryError| CommandError::new(e.to_string()))?;

            if version == 0 || text.is_empty() {
                return Ok("No compression guidelines generated yet.".to_owned());
            }

            let (_, created_at) = sqlite
                .load_compression_guidelines_meta(cid)
                .await
                .unwrap_or((0, String::new()));

            let (body, truncated) = if text.len() > MAX_DISPLAY_CHARS {
                let end = text.floor_char_boundary(MAX_DISPLAY_CHARS);
                (&text[..end], true)
            } else {
                (text.as_str(), false)
            };

            let mut output =
                format!("Compression Guidelines (v{version}, updated {created_at}):\n\n{body}");
            if truncated {
                output.push_str("\n\n[truncated]");
            }
            Ok(output)
        })
    }

    // ----- /model, /provider -----

    fn handle_model<'a>(
        &'a mut self,
        arg: &'a str,
    ) -> Pin<Box<dyn Future<Output = String> + Send + 'a>> {
        Box::pin(async move {
            let input = if arg.is_empty() {
                "/model".to_owned()
            } else {
                format!("/model {arg}")
            };
            self.handle_model_command_as_string(&input).await
        })
    }

    fn handle_provider<'a>(
        &'a mut self,
        arg: &'a str,
    ) -> Pin<Box<dyn Future<Output = String> + Send + 'a>> {
        Box::pin(async move { self.handle_provider_command_as_string(arg) })
    }

    // ----- /policy -----

    fn handle_policy<'a>(
        &'a mut self,
        args: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move { Ok(self.handle_policy_command_as_string(args)) })
    }

    // ----- /scheduler -----

    #[cfg(feature = "scheduler")]
    fn list_scheduled_tasks<'a>(
        &'a mut self,
    ) -> Pin<Box<dyn Future<Output = Result<Option<String>, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            let result = self
                .handle_scheduler_list_as_string()
                .await
                .map_err(|e| CommandError::new(e.to_string()))?;
            Ok(Some(result))
        })
    }

    #[cfg(not(feature = "scheduler"))]
    fn list_scheduled_tasks<'a>(
        &'a mut self,
    ) -> Pin<Box<dyn Future<Output = Result<Option<String>, CommandError>> + Send + 'a>> {
        Box::pin(async move { Ok(None) })
    }

    // ----- /lsp -----

    fn lsp_status<'a>(
        &'a mut self,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            self.handle_lsp_status_as_string()
                .await
                .map_err(|e| CommandError::new(e.to_string()))
        })
    }

    // ----- /compact -----

    fn compact_context<'a>(
        &'a mut self,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(self.compact_context_command())
    }

    // ----- /new -----

    fn reset_conversation<'a>(
        &'a mut self,
        keep_plan: bool,
        no_digest: bool,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            match self.reset_conversation(keep_plan, no_digest).await {
                Ok((old_id, new_id)) => {
                    let old = old_id.map_or_else(|| "none".to_string(), |id| id.0.to_string());
                    let new = new_id.map_or_else(|| "none".to_string(), |id| id.0.to_string());
                    let keep_note = if keep_plan { " (plan preserved)" } else { "" };
                    Ok(format!(
                        "New conversation started. Previous: {old} → Current: {new}{keep_note}"
                    ))
                }
                Err(e) => Ok(format!("Failed to start new conversation: {e}")),
            }
        })
    }

    // ----- /cache-stats -----

    fn cache_stats(&self) -> String {
        self.tool_orchestrator.cache_stats()
    }

    // ----- /status -----

    fn session_status<'a>(
        &'a mut self,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move { Ok(self.handle_status_as_string()) })
    }

    // ----- /guardrail -----

    fn guardrail_status(&self) -> String {
        self.format_guardrail_status()
    }

    // ----- /focus -----

    fn focus_status(&self) -> String {
        self.format_focus_status()
    }

    // ----- /sidequest -----

    fn sidequest_status(&self) -> String {
        self.format_sidequest_status()
    }

    // ----- /image -----

    fn load_image<'a>(
        &'a mut self,
        path: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move { Ok(self.handle_image_as_string(path)) })
    }

    // ----- /mcp -----

    fn handle_mcp<'a>(
        &'a mut self,
        args: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        // Extract all owned data before the async block so no &mut self reference is
        // held across an .await point, satisfying the `for<'a>` Send bound.
        let args_owned = args.to_owned();
        let parts: Vec<String> = args_owned.split_whitespace().map(str::to_owned).collect();
        let sub = parts.first().cloned().unwrap_or_default();

        match sub.as_str() {
            "list" => {
                // Read-only: clone all data before async.
                let manager = self.mcp.manager.clone();
                let tools_snapshot: Vec<(String, String)> = self
                    .mcp
                    .tools
                    .iter()
                    .map(|t| (t.server_id.clone(), t.name.clone()))
                    .collect();
                Box::pin(async move {
                    use std::fmt::Write;
                    let Some(manager) = manager else {
                        return Ok("MCP is not enabled.".to_owned());
                    };
                    let server_ids = manager.list_servers().await;
                    if server_ids.is_empty() {
                        return Ok("No MCP servers connected.".to_owned());
                    }
                    let mut output = String::from("Connected MCP servers:\n");
                    let mut total = 0usize;
                    for id in &server_ids {
                        let count = tools_snapshot.iter().filter(|(sid, _)| sid == id).count();
                        total += count;
                        let _ = writeln!(output, "- {id} ({count} tools)");
                    }
                    let _ = write!(output, "Total: {total} tool(s)");
                    Ok(output)
                })
            }
            "tools" => {
                // Read-only: collect tool info before async.
                let server_id = parts.get(1).cloned();
                let owned_tools: Vec<(String, String)> = if let Some(ref sid) = server_id {
                    self.mcp
                        .tools
                        .iter()
                        .filter(|t| &t.server_id == sid)
                        .map(|t| (t.name.clone(), t.description.clone()))
                        .collect()
                } else {
                    Vec::new()
                };
                Box::pin(async move {
                    use std::fmt::Write;
                    let Some(server_id) = server_id else {
                        return Ok("Usage: /mcp tools <server_id>".to_owned());
                    };
                    if owned_tools.is_empty() {
                        return Ok(format!("No tools found for server '{server_id}'."));
                    }
                    let mut output =
                        format!("Tools for '{server_id}' ({} total):\n", owned_tools.len());
                    for (name, desc) in &owned_tools {
                        if desc.is_empty() {
                            let _ = writeln!(output, "- {name}");
                        } else {
                            let _ = writeln!(output, "- {name} — {desc}");
                        }
                    }
                    Ok(output)
                })
            }
            // add/remove require mutating self after async I/O.
            // handle_mcp_command is structured so the only .await crossing a &mut self
            // boundary goes through a cloned Arc<McpManager> — no &self fields are held
            // across that .await.  The subsequent state-change methods (rebuild_semantic_index,
            // sync_mcp_registry) are also async fn(&mut self), but they only hold owned locals
            // across their own .await points (cloned tools Vec, cloned Arcs).
            _ => Box::pin(async move {
                self.handle_mcp_command(&args_owned)
                    .await
                    .map_err(|e| CommandError::new(e.to_string()))
            }),
        }
    }

    // ----- /skill -----

    fn handle_skill<'a>(
        &'a mut self,
        args: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        let args_owned = args.to_owned();
        Box::pin(async move {
            self.handle_skill_command_as_string(&args_owned)
                .await
                .map_err(|e| CommandError::new(e.to_string()))
        })
    }

    // ----- /skills -----

    fn handle_skills<'a>(
        &'a mut self,
        args: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        let args_owned = args.to_owned();
        Box::pin(async move {
            self.handle_skills_as_string(&args_owned)
                .await
                .map_err(|e| CommandError::new(e.to_string()))
        })
    }

    // ----- /feedback -----

    fn handle_feedback_command<'a>(
        &'a mut self,
        args: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        let args_owned = args.to_owned();
        Box::pin(async move {
            self.handle_feedback_as_string(&args_owned)
                .await
                .map_err(|e| CommandError::new(e.to_string()))
        })
    }

    // ----- /plan -----

    #[cfg(feature = "scheduler")]
    fn handle_plan<'a>(
        &'a mut self,
        input: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            self.dispatch_plan_command_as_string(input)
                .await
                .map_err(|e| CommandError::new(e.to_string()))
        })
    }

    #[cfg(not(feature = "scheduler"))]
    fn handle_plan<'a>(
        &'a mut self,
        _input: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move { Ok(String::new()) })
    }

    // ----- /experiment -----

    fn handle_experiment<'a>(
        &'a mut self,
        input: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<String, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            self.handle_experiment_command_as_string(input)
                .await
                .map_err(|e| CommandError::new(e.to_string()))
        })
    }

    // ----- /agent, @mention -----

    fn handle_agent_dispatch<'a>(
        &'a mut self,
        input: &'a str,
    ) -> Pin<Box<dyn Future<Output = Result<Option<String>, CommandError>> + Send + 'a>> {
        Box::pin(async move {
            match self.dispatch_agent_command(input).await {
                Some(Err(e)) => Err(CommandError::new(e.to_string())),
                Some(Ok(())) | None => Ok(None),
            }
        })
    }
}

/// Convert `AgentError` to `CommandError` for the trait boundary.
impl From<AgentError> for CommandError {
    fn from(e: AgentError) -> Self {
        Self(e.to_string())
    }
}