walrus-cli 0.0.1

Command-line interface for Walrus
Documentation 
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
81
82
83
84
85
86
87
88

//! Agent management commands: list, info.

use crate::runner::gateway::GatewayRunner;
use anyhow::Result;
use clap::Subcommand;
use protocol::ServerMessage;

/// Agent management subcommands.
#[derive(Subcommand, Debug)]
pub enum AgentCommand {
    /// List registered agents.
    List,
    /// Show agent details.
    Info {
        /// Agent name.
        name: String,
    },
}

impl AgentCommand {
    /// Dispatch agent management subcommands.
    pub async fn run(&self, runner: &mut GatewayRunner) -> Result<()> {
        match self {
            Self::List => list(runner).await,
            Self::Info { name } => info(runner, name).await,
        }
    }
}

async fn list(runner: &mut GatewayRunner) -> Result<()> {
    let agents = runner.list_agents().await?;
    if agents.is_empty() {
        println!("No agents registered.");
        return Ok(());
    }
    for agent in agents {
        let desc = if agent.description.is_empty() {
            "(no description)"
        } else {
            agent.description.as_str()
        };
        println!("  {} — {}", agent.name, desc);
    }
    Ok(())
}

async fn info(runner: &mut GatewayRunner, name: &str) -> Result<()> {
    match runner.agent_info(name).await? {
        ServerMessage::AgentDetail {
            name,
            description,
            tools,
            skill_tags,
            system_prompt,
        } => {
            println!("Name:        {name}");
            println!("Description: {description}");
            let tools_str = if tools.is_empty() {
                "(none)".to_owned()
            } else {
                tools
                    .iter()
                    .map(|t| t.as_str())
                    .collect::<Vec<_>>()
                    .join(", ")
            };
            let tags_str = if skill_tags.is_empty() {
                "(none)".to_owned()
            } else {
                skill_tags
                    .iter()
                    .map(|t| t.as_str())
                    .collect::<Vec<_>>()
                    .join(", ")
            };
            println!("Tools:       {tools_str}");
            println!("Skill tags:  {tags_str}");
            if !system_prompt.is_empty() {
                println!("\nSystem prompt:\n{system_prompt}");
            }
        }
        ServerMessage::Error { code, message } => {
            anyhow::bail!("error ({code}): {message}");
        }
        other => anyhow::bail!("unexpected response: {other:?}"),
    }
    Ok(())
}