use zeph_llm::provider::LlmProvider;
use super::Agent;
use super::error;
pub(crate) fn format_overlay_section(plugins_dir: &std::path::Path) -> String {
let mut cfg = zeph_config::Config::default();
match zeph_plugins::apply_plugin_config_overlays(&mut cfg, plugins_dir) {
Err(e) => format!("overlay resolution failed: {e}"),
Ok(overlay) => {
if overlay.source_plugins.is_empty() && overlay.skipped_plugins.is_empty() {
return "No plugin overlay active.".to_owned();
}
let mut out = String::from("Active plugin overlay:\n");
if overlay.source_plugins.is_empty() {
out.push_str(" Source plugins: (none)\n");
} else {
out.push_str(" Source plugins: ");
out.push_str(&overlay.source_plugins.join(", "));
out.push('\n');
}
if overlay.skipped_plugins.is_empty() {
out.push_str(" Skipped plugins: (none)\n");
} else {
out.push_str(" Skipped plugins:\n");
for reason in &overlay.skipped_plugins {
out.push_str(" - ");
out.push_str(reason);
out.push('\n');
}
}
out.push_str(
" Note: overlay values shown against default config — run with --config for live intersection.",
);
out
}
}
}
impl<C: crate::channel::Channel> Agent<C> {
#[allow(clippy::unused_self)]
pub(super) fn handle_builtin_command(&self, _trimmed: &str) -> Option<bool> {
None
}
pub(super) async fn dispatch_slash_command(
&mut self,
trimmed: &str,
) -> Option<Result<(), error::AgentError>> {
if trimmed.starts_with('@') {
return self.dispatch_agent_command(trimmed).await;
}
if trimmed.eq_ignore_ascii_case("/subagent")
|| trimmed.to_ascii_lowercase().starts_with("/subagent ")
{
let args = trimmed.get("/subagent".len()..).unwrap_or("").trim();
return Some(self.handle_subagent_slash(args).await);
}
None
}
async fn handle_subagent_slash(&mut self, args: &str) -> Result<(), error::AgentError> {
let msg: String = if args.is_empty() {
"Usage: /subagent <subcommand>\n\nSubcommands:\n spawn <command> Spawn an ACP sub-agent process".to_owned()
} else {
let (subcmd, rest) = args.split_once(' ').unwrap_or((args, ""));
match subcmd {
"spawn" => {
let cmd = rest.trim();
if cmd.is_empty() {
"Usage: /subagent spawn <command>\n\nExample: /subagent spawn zeph --acp"
.to_owned()
} else if let Some(spawn_fn) = self.runtime.acp_subagent_spawn_fn.clone() {
let cmd = cmd.to_owned();
match spawn_fn(cmd).await {
Ok(output) => output,
Err(e) => format!("Sub-agent error: {e}"),
}
} else {
"ACP sub-agent spawning is not available in this mode.\n\
Use `zeph acp run-agent --command <CMD> --prompt <TEXT>` for one-shot sessions."
.to_owned()
}
}
other => format!("Unknown /subagent subcommand: '{other}'. Available: spawn"),
}
};
let _ = self.channel.send(&msg).await;
let _ = self.channel.flush_chunks().await;
Ok(())
}
pub(super) async fn dispatch_agent_command(
&mut self,
trimmed: &str,
) -> Option<Result<(), error::AgentError>> {
let known: Vec<String> = self
.orchestration
.subagent_manager
.as_ref()
.map(|m| m.definitions().iter().map(|d| d.name.clone()).collect())
.unwrap_or_default();
match zeph_subagent::AgentCommand::parse(trimmed, &known) {
Ok(cmd) => {
if let Some(msg) = self.handle_agent_command(cmd).await
&& let Err(e) = self.channel.send(&msg).await
{
return Some(Err(e.into()));
}
let _ = self.channel.flush_chunks().await;
Some(Ok(()))
}
Err(e) if trimmed.starts_with('@') => {
tracing::debug!("@mention not matched as agent: {e}");
None
}
Err(e) => {
if let Err(send_err) = self.channel.send(&e.to_string()).await {
return Some(Err(send_err.into()));
}
let _ = self.channel.flush_chunks().await;
Some(Ok(()))
}
}
}
#[allow(clippy::too_many_lines)]
pub(super) fn handle_status_as_string(&mut self) -> String {
use std::fmt::Write;
use zeph_llm::provider::Role;
let uptime = self.lifecycle.start_time.elapsed().as_secs();
let msg_count = self
.msg
.messages
.iter()
.filter(|m| m.role == Role::User)
.count();
let (
api_calls,
prompt_tokens,
completion_tokens,
cost_cents,
mcp_servers,
orch_plans,
orch_tasks,
orch_completed,
orch_failed,
orch_skipped,
provider_breakdown,
) = if let Some(ref tx) = self.metrics.metrics_tx {
let m = tx.borrow();
(
m.api_calls,
m.prompt_tokens,
m.completion_tokens,
m.cost_spent_cents,
m.mcp_server_count,
m.orchestration.plans_total,
m.orchestration.tasks_total,
m.orchestration.tasks_completed,
m.orchestration.tasks_failed,
m.orchestration.tasks_skipped,
m.provider_cost_breakdown.clone(),
)
} else {
(0, 0, 0, 0.0, 0, 0, 0, 0, 0, 0, vec![])
};
let skill_count = self.skill_state.registry.read().all_meta().len();
let mut out = String::from("Session status:\n\n");
let _ = writeln!(out, "Provider: {}", self.provider.name());
let _ = writeln!(out, "Model: {}", self.runtime.model_name);
let _ = writeln!(out, "Uptime: {uptime}s");
let _ = writeln!(out, "Turns: {msg_count}");
let _ = writeln!(out, "API calls: {api_calls}");
let _ = writeln!(
out,
"Tokens: {prompt_tokens} prompt / {completion_tokens} completion"
);
let _ = writeln!(out, "Skills: {skill_count}");
let _ = writeln!(out, "MCP: {mcp_servers} server(s)");
if let Some(ref tf) = self.tool_state.tool_schema_filter {
let _ = writeln!(
out,
"Filter: enabled (top_k={}, always_on={}, {} embeddings)",
tf.top_k(),
tf.always_on_count(),
tf.embedding_count(),
);
}
if let Some(ref adv) = self.runtime.adversarial_policy_info {
let provider_display = if adv.provider.is_empty() {
"default"
} else {
adv.provider.as_str()
};
let _ = writeln!(
out,
"Adv gate: enabled (provider={}, policies={}, fail_open={})",
provider_display, adv.policy_count, adv.fail_open
);
}
if cost_cents > 0.0 {
let _ = writeln!(out, "Cost: ${:.4}", cost_cents / 100.0);
if !provider_breakdown.is_empty() {
let _ = writeln!(
out,
" {:<16} {:>8} {:>8} {:>8}",
"Provider", "Requests", "Tokens", "Cost"
);
for (name, usage) in &provider_breakdown {
let total_tokens = usage.input_tokens + usage.output_tokens;
let _ = writeln!(
out,
" {:<16} {:>8} {:>8} {:>8}",
name,
usage.request_count,
total_tokens,
format!("${:.4}", usage.cost_cents / 100.0),
);
}
}
}
if orch_plans > 0 {
let _ = writeln!(out);
let _ = writeln!(out, "Orchestration:");
let _ = writeln!(out, " Plans: {orch_plans}");
let _ = writeln!(out, " Tasks: {orch_completed}/{orch_tasks} completed");
if orch_failed > 0 {
let _ = writeln!(out, " Failed: {orch_failed}");
}
if orch_skipped > 0 {
let _ = writeln!(out, " Skipped: {orch_skipped}");
}
}
{
use crate::config::PruningStrategy;
if matches!(
self.context_manager.compression.pruning_strategy,
PruningStrategy::Subgoal | PruningStrategy::SubgoalMig
) {
let _ = writeln!(out);
let _ = writeln!(
out,
"Pruning: {}",
match self.context_manager.compression.pruning_strategy {
PruningStrategy::SubgoalMig => "subgoal_mig",
_ => "subgoal",
}
);
let subgoal_count = self.compression.subgoal_registry.subgoals.len();
let _ = writeln!(out, "Subgoals: {subgoal_count} tracked");
if let Some(active) = self.compression.subgoal_registry.active_subgoal() {
let _ = writeln!(out, "Active: \"{}\"", active.description);
} else {
let _ = writeln!(out, "Active: (none yet)");
}
}
}
let gc = &self.memory_state.extraction.graph_config;
if gc.enabled {
let _ = writeln!(out);
if gc.spreading_activation.enabled {
let _ = writeln!(
out,
"Graph recall: spreading activation (lambda={:.2}, hops={})",
gc.spreading_activation.decay_lambda, gc.spreading_activation.max_hops,
);
} else {
let _ = writeln!(out, "Graph recall: BFS (hops={})", gc.max_hops);
}
}
out.trim_end().to_owned()
}
pub(super) fn format_guardrail_status(&self) -> String {
use std::fmt::Write;
let mut out = String::new();
if let Some(ref guardrail) = self.security.guardrail {
let stats = guardrail.stats();
let _ = writeln!(out, "Guardrail: enabled");
let _ = writeln!(out, "Action: {:?}", guardrail.action());
let _ = writeln!(out, "Fail strategy: {:?}", guardrail.fail_strategy());
let _ = writeln!(out, "Timeout: {}ms", guardrail.timeout_ms());
let _ = writeln!(
out,
"Tool scan: {}",
if guardrail.scan_tool_output() {
"enabled"
} else {
"disabled"
}
);
let _ = writeln!(out, "\nStats:");
let _ = writeln!(out, " Total checks: {}", stats.total_checks);
let _ = writeln!(out, " Flagged: {}", stats.flagged_count);
let _ = writeln!(out, " Errors: {}", stats.error_count);
let _ = writeln!(out, " Avg latency: {}ms", stats.avg_latency_ms());
} else {
out.push_str("Guardrail: disabled\n");
out.push_str(
"Enable with: --guardrail flag or [security.guardrail] enabled = true in config",
);
}
out.trim_end().to_owned()
}
pub(super) fn format_focus_status(&self) -> String {
use std::fmt::Write;
let mut out = String::from("Focus Agent status\n\n");
let _ = writeln!(out, "Enabled: {}", self.focus.config.enabled);
let _ = writeln!(out, "Active session: {}", self.focus.is_active());
if let Some(ref scope) = self.focus.active_scope {
let _ = writeln!(out, "Active scope: {scope}");
}
let _ = writeln!(
out,
"Knowledge blocks: {}",
self.focus.knowledge_blocks.len()
);
let _ = writeln!(out, "Turns since focus: {}", self.focus.turns_since_focus);
out.trim_end().to_owned()
}
pub(super) fn format_sidequest_status(&self) -> String {
use std::fmt::Write;
let mut out = String::from("SideQuest status\n\n");
let _ = writeln!(out, "Enabled: {}", self.sidequest.config.enabled);
let _ = writeln!(
out,
"Interval turns: {}",
self.sidequest.config.interval_turns
);
let _ = writeln!(out, "Turn counter: {}", self.sidequest.turn_counter);
let _ = writeln!(out, "Passes run: {}", self.sidequest.passes_run);
let _ = writeln!(
out,
"Total evicted: {} tool outputs",
self.sidequest.total_evicted
);
out.trim_end().to_owned()
}
pub(super) fn handle_image_as_string(&mut self, path: &str) -> String {
use std::path::Component;
use zeph_llm::provider::{ImageData, MessagePart};
let p = std::path::Path::new(path);
if p.is_absolute()
|| path.starts_with('/')
|| p.components().any(|c| c == Component::ParentDir)
{
return "Invalid image path: path traversal not allowed".to_owned();
}
let data = match std::fs::read(path) {
Ok(d) => d,
Err(e) => return format!("Cannot read image {path}: {e}"),
};
if data.len() > super::message_queue::MAX_IMAGE_BYTES {
return format!(
"Image {path} exceeds size limit ({} MB), skipping",
super::message_queue::MAX_IMAGE_BYTES / 1024 / 1024
);
}
let mime_type = super::message_queue::detect_image_mime(Some(path)).to_string();
self.msg
.pending_image_parts
.push(MessagePart::Image(Box::new(ImageData { data, mime_type })));
format!("Image loaded: {path}. Send your message.")
}
pub(super) fn run_plugin_command(
args: &str,
managed_dir: Option<std::path::PathBuf>,
mcp_allowed: Vec<String>,
base_shell_allowed: Vec<String>,
) -> String {
let plugins_dir = zeph_plugins::PluginManager::default_plugins_dir();
let (subcmd, rest) = args.trim().split_once(' ').unwrap_or((args.trim(), ""));
if subcmd == "overlay" || (matches!(subcmd, "" | "list") && rest.trim() == "--overlay") {
return format_overlay_section(&plugins_dir);
}
let managed_dir = managed_dir
.unwrap_or_else(|| zeph_config::defaults::default_vault_dir().join("skills"));
let mgr = zeph_plugins::PluginManager::new(
plugins_dir,
managed_dir,
mcp_allowed,
base_shell_allowed,
);
match subcmd {
"" | "list" => match mgr.list_installed() {
Ok(plugins) if plugins.is_empty() => "No plugins installed.".to_owned(),
Ok(plugins) => plugins
.iter()
.map(|p| format!("{} v{} — {}", p.name, p.version, p.description))
.collect::<Vec<_>>()
.join("\n"),
Err(e) => format!("plugin list failed: {e}"),
},
"add" => {
use std::fmt::Write as _;
if rest.is_empty() {
return "Usage: /plugins add <source>".to_owned();
}
match mgr.add(rest.trim()) {
Ok(r) => {
let mut out = format!("Installed plugin \"{}\"", r.name);
if !r.installed_skills.is_empty() {
let _ = write!(out, "\n Skills: {}", r.installed_skills.join(", "));
}
if !r.mcp_server_ids.is_empty() {
let _ = write!(
out,
"\n MCP servers (restart required): {}",
r.mcp_server_ids.join(", ")
);
}
for w in &r.warnings {
let _ = write!(out, "\n warning: {w}");
}
out
}
Err(e) => format!("plugin add failed: {e}"),
}
}
"remove" => {
use std::fmt::Write as _;
if rest.is_empty() {
return "Usage: /plugins remove <name>".to_owned();
}
match mgr.remove(rest.trim()) {
Ok(r) => {
let mut out = format!("Removed plugin \"{}\"", rest.trim());
if !r.removed_skills.is_empty() {
let _ =
write!(out, "\n Removed skills: {}", r.removed_skills.join(", "));
}
out
}
Err(e) => format!("plugin remove failed: {e}"),
}
}
other => {
format!(
"Unknown /plugins subcommand: '{other}'. Available: list, list --overlay, overlay, add, remove"
)
}
}
}
pub(super) async fn handle_skills_as_string(
&mut self,
subcommand: &str,
) -> Result<String, error::AgentError> {
match subcommand {
"" => self.handle_skills_command_as_string().await,
"confusability" => self.handle_skills_confusability_as_string().await,
other => Ok(format!(
"Unknown /skills subcommand: '{other}'. Available: confusability"
)),
}
}
async fn handle_skills_command_as_string(&mut self) -> Result<String, error::AgentError> {
use std::collections::BTreeMap;
use std::fmt::Write;
let all_meta: Vec<zeph_skills::loader::SkillMeta> = self
.skill_state
.registry
.read()
.all_meta()
.into_iter()
.cloned()
.collect();
let memory = self.memory_state.persistence.memory.clone();
let mut trust_map: std::collections::HashMap<String, String> =
std::collections::HashMap::new();
for meta in &all_meta {
if let Some(ref memory) = memory {
let info = memory
.sqlite()
.load_skill_trust(&meta.name)
.await
.ok()
.flatten()
.map_or_else(String::new, |r| format!(" [{}]", r.trust_level));
trust_map.insert(meta.name.clone(), info);
}
}
let mut output = String::from("Available skills:\n\n");
let has_categories = all_meta.iter().any(|m| m.category.is_some());
if has_categories {
let mut by_category: BTreeMap<&str, Vec<&zeph_skills::loader::SkillMeta>> =
BTreeMap::new();
for meta in &all_meta {
let cat = meta.category.as_deref().unwrap_or("other");
by_category.entry(cat).or_default().push(meta);
}
for (cat, skills) in &by_category {
let _ = writeln!(output, "[{cat}]");
for meta in skills {
let trust_info = trust_map.get(&meta.name).map_or("", String::as_str);
let _ = writeln!(output, "- {} — {}{trust_info}", meta.name, meta.description);
}
output.push('\n');
}
} else {
for meta in &all_meta {
let trust_info = trust_map.get(&meta.name).map_or("", String::as_str);
let _ = writeln!(output, "- {} — {}{trust_info}", meta.name, meta.description);
}
}
if let Some(ref memory) = memory {
match memory.sqlite().load_skill_usage().await {
Ok(usage) if !usage.is_empty() => {
output.push_str("\nUsage statistics:\n\n");
for row in &usage {
let _ = writeln!(
output,
"- {}: {} invocations (last: {})",
row.skill_name, row.invocation_count, row.last_used_at,
);
}
}
Ok(_) => {}
Err(e) => tracing::warn!("failed to load skill usage: {e:#}"),
}
}
Ok(output)
}
pub(crate) fn start_user_loop(&mut self, prompt: String, interval_secs: u64) {
use std::time::Duration;
use tokio::time::{Instant, MissedTickBehavior};
let period = Duration::from_secs(interval_secs);
let mut interval = tokio::time::interval_at(Instant::now() + period, period);
interval.set_missed_tick_behavior(MissedTickBehavior::Skip);
let cancel_tx = tokio_util::sync::CancellationToken::new();
self.lifecycle.user_loop = Some(crate::agent::state::LoopState {
prompt,
iteration: 0,
interval,
cancel_tx,
});
}
pub(crate) fn stop_user_loop(&mut self) -> String {
if let Some(ls) = self.lifecycle.user_loop.take() {
let iters = ls.iteration;
ls.cancel_tx.cancel();
format!("Loop stopped after {iters} iteration(s).")
} else {
"No active loop.".to_owned()
}
}
async fn handle_skills_confusability_as_string(&mut self) -> Result<String, error::AgentError> {
let threshold = self.skill_state.confusability_threshold;
if threshold <= 0.0 {
return Ok("Confusability monitoring is disabled. \
Set [skills] confusability_threshold in config (e.g. 0.85) to enable."
.to_owned());
}
let Some(matcher) = &self.skill_state.matcher else {
return Ok(
"Skill matcher not available (no embedding provider configured).".to_owned(),
);
};
let all_meta: Vec<zeph_skills::loader::SkillMeta> = self
.skill_state
.registry
.read()
.all_meta()
.into_iter()
.cloned()
.collect();
let refs: Vec<&zeph_skills::loader::SkillMeta> = all_meta.iter().collect();
let report = matcher.confusability_report(&refs, threshold).await;
Ok(report.to_string())
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn format_overlay_section_empty_dir() {
let tmp = tempfile::tempdir().unwrap();
let out = format_overlay_section(tmp.path());
assert_eq!(out, "No plugin overlay active.");
}
#[test]
fn format_overlay_section_with_source_plugin() {
let tmp = tempfile::tempdir().unwrap();
let plugin_dir = tmp.path().join("myplugin");
std::fs::create_dir_all(&plugin_dir).unwrap();
let manifest = r#"
[plugin]
name = "myplugin"
version = "0.1.0"
description = "test"
[config.tools.shell]
blocked_commands = ["curl"]
"#;
std::fs::write(plugin_dir.join(".plugin.toml"), manifest).unwrap();
let out = format_overlay_section(tmp.path());
assert!(out.contains("Active plugin overlay:"));
assert!(out.contains("myplugin"));
assert!(out.contains("Source plugins:"));
assert!(out.contains("Note:"));
}
#[test]
fn run_plugin_command_overlay_subcommand() {
let tmp = tempfile::tempdir().unwrap();
let out = format_overlay_section(tmp.path());
assert_eq!(out, "No plugin overlay active.");
}
#[test]
fn format_overlay_section_skipped_plugin_shows_reason() {
let tmp = tempfile::tempdir().unwrap();
let plugin_dir = tmp.path().join("badplugin");
std::fs::create_dir_all(&plugin_dir).unwrap();
std::fs::write(
plugin_dir.join(".plugin.toml"),
b"not valid toml at all {{{{",
)
.unwrap();
let out = format_overlay_section(tmp.path());
assert!(out.contains("No plugin overlay active.") || out.contains("badplugin"));
}
}