feat: Add /waz slash command to generate TMP schemas via AI

Author: codeitlikemiley

app/i18n/en/warp.ftl

@@ -2426,6 +2426,8 @@ slash-cmd-fork-desc = Fork the current conversation in a new pane or a new tab
 slash-cmd-fork-hint = <optional prompt to send in forked conversation>
 slash-cmd-open-code-review-desc = Open code review
 slash-cmd-init-desc = Generate or update an AGENTS.md file
+slash-cmd-waz-desc = Generate a TMP schema for a CLI tool using AI
+slash-cmd-waz-hint = <command or tool>
 slash-cmd-open-project-rules-desc = Open the project rules file (AGENTS.md)
 slash-cmd-open-mcp-servers-desc = Open MCP servers
 slash-cmd-open-settings-file-desc = Open settings file (TOML)

app/i18n/ja/warp.ftl

@@ -2330,6 +2330,8 @@ slash-cmd-fork-desc = 現在の会話を新しいペインまたはタブにフ
 slash-cmd-fork-hint = <フォーク後の会話に送る任意のプロンプト>
 slash-cmd-open-code-review-desc = コードレビューを開く
 slash-cmd-init-desc = AGENTS.md ファイルを生成または更新
+slash-cmd-waz-desc = AIを使用してCLIツールのTMPスキーマを生成します
+slash-cmd-waz-hint = <コマンドまたはツール名>
 slash-cmd-open-project-rules-desc = プロジェクトルールファイル (AGENTS.md) を開く
 slash-cmd-open-mcp-servers-desc = MCP サーバーを開く
 slash-cmd-open-settings-file-desc = 設定ファイル (TOML) を開く

app/i18n/zh-CN/warp.ftl

@@ -2352,6 +2352,8 @@ slash-cmd-fork-desc = 在新窗格或新标签页中分叉当前对话
 slash-cmd-fork-hint = <可选：在分叉后的对话中发送的提示词>
 slash-cmd-open-code-review-desc = 打开代码评审
 slash-cmd-init-desc = 生成或更新 AGENTS.md 文件
+slash-cmd-waz-desc = 使用 AI 为 CLI 工具生成 TMP 模式 (schema)
+slash-cmd-waz-hint = <命令或工具名称>
 slash-cmd-open-project-rules-desc = 打开项目规则文件（AGENTS.md）
 slash-cmd-open-mcp-servers-desc = 打开 MCP 服务器
 slash-cmd-open-settings-file-desc = 打开设置文件（TOML）

app/src/ai/agent_providers/prompt_renderer.rs

@@ -73,6 +73,11 @@ fn build_env() -> Environment<'static> {
         include_str!("prompts/commands/init_project.j2"),
     )
     .expect("init_project command template parses");
+    env.add_template(
+        "commands/generate_schema.j2",
+        include_str!("prompts/commands/generate_schema.j2"),
+    )
+    .expect("generate_schema command template parses");
 
     // Distribute system prompt (aligned opencode) by model id substring matching
     // `packages/opencode/src/session/system.ts::provider`). The OpenRouter path looks like
@@ -217,6 +222,11 @@ struct InitProjectCommandContext {
     arguments: String,
 }
 
+#[derive(Debug, Default, Serialize)]
+struct GenerateSchemaCommandContext {
+    tool: String,
+}
+
 #[derive(Debug, Default, Serialize)]
 struct PromptContext {
     cwd: Option<String>,
@@ -383,6 +393,28 @@ pub fn render_init_project_command(arguments: Option<&str>) -> String {
     }
 }
 
+pub fn render_generate_schema_command(tool: &str) -> String {
+    let ctx = GenerateSchemaCommandContext {
+        tool: tool.trim().to_owned(),
+    };
+    let env = env();
+    let template_name = "commands/generate_schema.j2";
+    let tmpl = match env.get_template(template_name) {
+        Ok(t) => t,
+        Err(e) => {
+            log::error!("[byop prompt] failed to get template {template_name}: {e}");
+            return fallback_generate_schema_command(&ctx.tool);
+        }
+    };
+    match tmpl.render(Value::from_serialize(&ctx)) {
+        Ok(s) => s,
+        Err(e) => {
+            log::error!("[byop prompt] render {template_name} failed: {e}");
+            fallback_generate_schema_command(&ctx.tool)
+        }
+    }
+}
+
 /// Render the system message string ultimately sent to the upstream model.
 ///
 /// `ctx` usually comes from the latest `AIAgentInput::UserQuery.context` in `params.input`.
@@ -441,6 +473,13 @@ fn fallback_init_project_command(arguments: &str) -> String {
     )
 }
 
+fn fallback_generate_schema_command(tool: &str) -> String {
+    format!(
+        "Generate a Token Model Protocol (TMP) JSON schema for the CLI tool `{tool}`. \
+         Please identify its most common subcommands, options, and parameters, and design the schema."
+    )
+}
+
 /// Rendering system (only used when template loading/rendering fails, should not be triggered in normal paths).
 fn fallback_system(model_id: &str) -> String {
     format!(
@@ -465,6 +504,12 @@ mod tests {
         assert!(out.contains("## Writing rules"), "{out}");
     }
 
+    #[test]
+    fn render_generate_schema_command_uses_template() {
+        let out = render_generate_schema_command("docker");
+        assert!(out.contains("Generate a Token Model Protocol (TMP) JSON schema for the CLI tool \"docker\""), "{out}");
+    }
+
     #[test]
     fn pick_template_dispatches_by_model_family() {
         // Direct connection mode

app/src/ai/agent_providers/prompts/commands/generate_schema.j2

@@ -0,0 +1,39 @@
+Generate a Token Model Protocol (TMP) JSON schema for the CLI tool "{{ tool }}".
+
+A TMP schema defines the subcommand structure, parameters (flags and arguments), primitive token types, descriptions, and dynamic resolvers for a CLI tool.
+
+Please design a comprehensive and detailed schema file for "{{ tool }}", mapping its most common subcommands, options, and parameters.
+
+## Schema Specification and Format
+The output MUST be a valid JSON object matching the following structure:
+{
+  "meta": {
+    "tool": "{{ tool }}",
+    "description": "Short description of what the tool does."
+  },
+  "commands": [
+    {
+      "command": "{{ tool }} <subcommand>",
+      "description": "Short description of the subcommand.",
+      "group": "{{ tool }}",
+      "verified": true,
+      "tokens": [
+        {
+          "name": "parameter_name",
+          "description": "What this flag/argument does.",
+          "required": false,
+          "token_type": "Enum" | "Boolean" | "String" | "File" | "Number",
+          "flag": "--flag-name" (optional),
+          "data_source": { (optional)
+            "resolver": "git:branches" | "git:status_files" | "cargo:packages" | "npm:scripts" | etc.
+          }
+        }
+      ]
+    }
+  ]
+}
+
+## Guidelines
+1. The JSON must be valid and ready to be saved to `.waz/schemas/{{ tool }}.json`.
+2. Do not wrap the JSON in markdown code blocks or add conversational commentary; output ONLY the raw, clean JSON content.
+3. Make sure to define the correct `token_type` for each parameter. Use "Boolean" for flags that take no values (like --force or -f), "Enum" for parameters with specific values, "File" for paths/files, and "String" for generic text.

app/src/ai/blocklist/controller/slash_command.rs

@@ -31,6 +31,9 @@ pub enum SlashCommandRequest {
     InitProjectRules {
         arguments: Option<String>,
     },
+    GenerateSchema {
+        tool: String,
+    },
     Summarize {
         prompt: Option<String>,
         /// Waz BYOP local session compression: Whether this summary is automatically triggered by token-overflow.
@@ -65,6 +68,15 @@ impl SlashCommandRequest {
             });
         }
 
+        if let Some(tool) = query
+            .strip_prefix(commands::WAZ_NAME)
+            .and_then(|query| query.strip_prefix(' '))
+        {
+            return Some(Self::GenerateSchema {
+                tool: tool.trim().to_string(),
+            });
+        }
+
         // Check if query starts with /compact and route to summarize conversation
         if let Some(prompt) = query.strip_prefix(commands::COMPACT.name) {
             return Some(Self::Summarize {
@@ -222,6 +234,17 @@ impl SlashCommandRequest {
                 running_command: None,
                 intended_agent: None,
             }],
+            SlashCommandRequest::GenerateSchema { tool } => vec![AIAgentInput::UserQuery {
+                query: crate::ai::agent_providers::prompt_renderer::render_generate_schema_command(
+                    &tool,
+                ),
+                context,
+                static_query_type: None,
+                referenced_attachments: HashMap::<String, AIAgentAttachment>::new(),
+                user_query_mode: UserQueryMode::Normal,
+                running_command: None,
+                intended_agent: None,
+            }],
             SlashCommandRequest::Summarize { prompt, overflow } => {
                 vec![AIAgentInput::SummarizeConversation { prompt, overflow }]
             }
@@ -258,6 +281,7 @@ impl SlashCommandRequest {
             SlashCommandRequest::CloneRepository { .. } => EntrypointType::CloneRepository,
             SlashCommandRequest::InitProjectRules { .. } => EntrypointType::InitProjectRules,
             SlashCommandRequest::CreateNewProject { .. }
+            | SlashCommandRequest::GenerateSchema { .. }
             | SlashCommandRequest::Summarize { .. }
             | SlashCommandRequest::FetchReviewComments { .. }
             | SlashCommandRequest::InvokeSkill { .. } => EntrypointType::UserInitiated,

app/src/search/slash_command_menu/static_commands/commands.rs

@@ -146,6 +146,17 @@ pub static INIT: LazyLock<StaticCommand> = LazyLock::new(|| StaticCommand {
     argument: Some(Argument::optional()),
 });
 
+pub const WAZ_NAME: &str = "/waz";
+
+pub static WAZ: LazyLock<StaticCommand> = LazyLock::new(|| StaticCommand {
+    name: WAZ_NAME,
+    description: t_static!("slash-cmd-waz-desc"),
+    icon_path: "bundled/svg/stars-01.svg",
+    availability: Availability::AI_ENABLED,
+    auto_enter_ai_mode: true,
+    argument: Some(Argument::required().with_hint_text(t_static!("slash-cmd-waz-hint"))),
+});
+
 pub static OPEN_PROJECT_RULES: LazyLock<StaticCommand> = LazyLock::new(|| StaticCommand {
     name: "/open-project-rules",
     description: t_static!("slash-cmd-open-project-rules-desc"),
@@ -428,6 +439,7 @@ fn all_commands() -> Vec<StaticCommand> {
         ADD_PROMPT.clone(),
         ADD_RULE.clone(),
         INIT.clone(),
+        WAZ.clone(),
         OPEN_PROJECT_RULES.clone(),
         OPEN_MCP_SERVERS.clone(),
         OPEN_RULES.clone(),