diff --git a/agent.py b/agent.py
index c48fb17..eaacf3a 100644
--- a/agent.py
+++ b/agent.py
@@ -726,7 +726,7 @@ class ProductionIncidentCommander(A2AAgent[CommanderConfig, NoAuth]):
         ConsumerSetupField.secret("GITEA_TOKEN", label="Gitea token", description="Optional read-only repository token.", required=False),
     )
 
-    @a2a.tool(description="Diagnose a production incident, collect bounded evidence, rank hypotheses, and write diagnosis artifacts", timeout_seconds=900, cost_class="incident-diagnostic", grant_mode="read_write_overlay", grant_write_prefixes=("outputs/incidents/",), grant_allow_patterns=("**",), grant_deny_patterns=("**/.env", "**/secrets/**", "**/*kubeconfig*"))
+    @a2a.tool(description="Diagnose a production incident, collect bounded evidence, rank hypotheses, and write diagnosis artifacts", input_schema=DIAGNOSE_INPUT_SCHEMA, timeout_seconds=900, cost_class="incident-diagnostic", grant_mode="read_write_overlay", grant_write_prefixes=("outputs/incidents/",), grant_allow_patterns=("**",), grant_deny_patterns=("**/.env", "**/secrets/**", "**/*kubeconfig*"))
     async def diagnose_incident(
         self,
         ctx: RunContext[NoAuth],