@alexkroman1/aai-cli 5.3.0 → 5.5.0

package/dist/scaffold/package.json

@@ -11,15 +11,15 @@
     "deploy": "aai deploy"
   },
   "dependencies": {
-    "@alexkroman1/aai": "^5.3.0",
-    "@alexkroman1/aai-ui": "^5.3.0",
+    "@alexkroman1/aai": "^5.5.0",
+    "@alexkroman1/aai-ui": "^5.5.0",
     "react": "^19.2.8",
     "react-dom": "^19.2.8",
     "tailwindcss": "^4.0.0",
     "zod": "^4.4.3"
   },
   "devDependencies": {
-    "@alexkroman1/aai-cli": "^5.3.0",
+    "@alexkroman1/aai-cli": "^5.5.0",
     "@tailwindcss/vite": "^4.3.3",
     "@types/node": "^26.1.1",
     "@types/react": "^19.2.17",

package/dist/{secret-DOva9OGk.mjs → secret-CGAIAbUx.mjs}

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 import { a as unwrapCancel, n as log, s as fail, u as ok } from "./_ui-8kOEB-JH.mjs";
-import { t as slugRequest } from "./_slug-api-19R5kZ0U.mjs";
+import { t as slugRequest } from "./_slug-api-DaqQJHk8.mjs";
 import * as p from "@clack/prompts";
 import { text } from "node:stream/consumers";
 //#region secret.ts

package/dist/{storage-vM6HjDZZ.mjs → storage-CnhOayhm.mjs}

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 import { n as log, s as fail, u as ok } from "./_ui-8kOEB-JH.mjs";
-import { t as slugRequest } from "./_slug-api-19R5kZ0U.mjs";
+import { t as slugRequest } from "./_slug-api-DaqQJHk8.mjs";
 import * as p from "@clack/prompts";
 //#region storage.ts
 async function storageRequest(cwd, init, server) {

package/dist/templates/code-interpreter/agent.ts

@@ -1,9 +1,8 @@
-import { agent, assemblyAIPipeline } from "@alexkroman1/aai";
+import { agent } from "@alexkroman1/aai";
 import systemPrompt from "./system-prompt.md?raw";
 
 export default agent({
   name: "Coda",
-  ...assemblyAIPipeline(),
   systemPrompt,
   greeting:
     "Hey, I'm Coda. I solve problems by writing and running code. Try asking me something like, what's the 50th fibonacci number, or what day of the week was January 1st 2000.",

package/dist/templates/dispatch-center/agent.test.ts

@@ -90,17 +90,17 @@ describe("dispatch-center template", () => {
 
   test("negative casualty counts are rejected by the parameter schemas", () => {
     expect(
-      incidentTriage.parameters?.safeParse({ incidentId: "INC-0001", casualtyUpdate: -5 }).success,
+      incidentTriage.inputSchema?.safeParse({ incidentId: "INC-0001", casualtyUpdate: -5 }).success,
     ).toBe(false);
     expect(
-      incidentCreate.parameters?.safeParse({
+      incidentCreate.inputSchema?.safeParse({
         location: "1 First St",
         description: "fire",
         estimatedCasualties: -1,
       }).success,
     ).toBe(false);
     expect(
-      incidentUpdateStatus.parameters?.safeParse({
+      incidentUpdateStatus.inputSchema?.safeParse({
         incidentId: "INC-0001",
         status: "on_scene",
         casualtyUpdate: { confirmed: -2 },

package/dist/templates/dispatch-center/agent.ts

@@ -1,5 +1,5 @@
-import { agent, assemblyAIPipeline } from "@alexkroman1/aai";
-import { assemblyAI } from "@alexkroman1/aai/stt";
+import { agent } from "@alexkroman1/aai";
+import { assemblyAIStt } from "@alexkroman1/aai/stt";
 import { dashboardView } from "./shared.ts";
 import systemPrompt from "./system-prompt.md?raw";
 import { incidentAddNote } from "./tools/incident_add_note.ts";
@@ -17,14 +17,14 @@ import { resourcesUpdateStatus } from "./tools/resources_update_status.ts";
 
 export default agent({
   name: "Dispatch Command Center",
-  ...assemblyAIPipeline(),
   // One projection replaces eleven `ctx.send("incidents", ...)` calls, and
   // is the single place that decides caller PII stays server-side.
   syncState: dashboardView,
-  // Overriding the STT stage: a dispatcher reads addresses and unit numbers
-  // in bursts with pauses inside one message, so end-of-turn silence has to
-  // be longer than the default or "unit twelve … respond to" splits in two.
-  stt: assemblyAI({ minTurnSilenceMs: 2200 }),
+  // Only the STT stage is declared (LLM and TTS stay on the AssemblyAI
+  // defaults): a dispatcher reads addresses and unit numbers in bursts with
+  // pauses inside one message, so end-of-turn silence has to be longer than
+  // the default or "unit twelve … respond to" splits in two.
+  stt: assemblyAIStt({ minTurnSilenceMs: 2200 }),
   systemPrompt,
   greeting:
     "Dispatch Command Center online. Restoring operational state. I'm ready to take incoming calls, manage active incidents, or run dispatch operations. Say 'dashboard' for a full status report. What do we have.",

package/dist/templates/dispatch-center/client.tsx

@@ -300,13 +300,18 @@ function App() {
                   >
                     {session.running ? "Pause" : "Resume"}
                   </button>
+                  {/* end() hangs up and flips `started` back, so the UI
+                      returns to "Start Dispatch" and the next start is a
+                      brand-new shift (fresh incident board, greeting
+                      included). reset() would keep the call live — the
+                      buttons never toggle back. */}
                   <button
                     type="button"
                     className="px-4 py-2 border-none rounded-md font-mono text-xs font-semibold uppercase tracking-wider cursor-pointer text-white"
                     style={{ background: "#dc2626" }}
-                    onClick={() => session.reset()}
+                    onClick={() => session.end()}
                   >
-                    Reset
+                    End Shift
                   </button>
                 </>
               )}

package/dist/templates/dispatch-center/tools/incident_add_note.ts

@@ -4,7 +4,7 @@ import { findIncident, logEvent, updateState } from "../shared.ts";
 
 export const incidentAddNote = tool({
   description: "Add a situational update note to an incident's timeline.",
-  parameters: z.object({
+  inputSchema: z.object({
     incidentId: z.string().max(20).describe("The incident ID"),
     note: z.string().max(1000).describe("The note to add"),
     source: z.string().max(100).describe("Who reported this — unit callsign or caller").optional(),

package/dist/templates/dispatch-center/tools/incident_create.ts

@@ -13,7 +13,7 @@ import {
 
 export const incidentCreate = tool({
   description: "Create a new incident from an incoming emergency call.",
-  parameters: z.object({
+  inputSchema: z.object({
     location: z.string().max(300).describe("Address or location description"),
     description: z.string().max(2000).describe("Nature of the emergency as described by caller"),
     callerName: z.string().max(120).describe("Caller's name").optional(),

package/dist/templates/dispatch-center/tools/incident_escalate.ts

@@ -11,7 +11,7 @@ import {
 
 export const incidentEscalate = tool({
   description: "Escalate an incident when it exceeds current capacity or severity increases.",
-  parameters: z.object({
+  inputSchema: z.object({
     incidentId: z.string().max(20).describe("The incident ID"),
     reason: z.string().max(1000).describe("Reason for escalation"),
     requestMutualAid: z

package/dist/templates/dispatch-center/tools/incident_get.ts

@@ -4,7 +4,7 @@ import { findIncident, getApplicableProtocols, getState, incidentAgeMinutes } fr
 
 export const incidentGet = tool({
   description: "Get full details on a specific incident including timeline and assigned resources.",
-  parameters: z.object({
+  inputSchema: z.object({
     incidentId: z.string().max(20).describe("The incident ID"),
   }),
   async execute(args, ctx) {

package/dist/templates/dispatch-center/tools/incident_triage.ts

@@ -15,7 +15,7 @@ import {
 export const incidentTriage = tool({
   description:
     "Triage an incident — confirm or override severity, type, hazards, and casualty count.",
-  parameters: z.object({
+  inputSchema: z.object({
     incidentId: z.string().max(20).describe("The incident ID"),
     severity: z.enum(SEVERITIES).describe("Confirmed severity after triage").optional(),
     type: z.enum(INCIDENT_TYPES).describe("Confirmed incident type").optional(),

package/dist/templates/dispatch-center/tools/incident_update_status.ts

@@ -4,7 +4,7 @@ import { assertNotResolved, findIncident, logEvent, updateState } from "../share
 
 export const incidentUpdateStatus = tool({
   description: "Update an incident's status (en_route, on_scene, resolved, escalated).",
-  parameters: z.object({
+  inputSchema: z.object({
     incidentId: z.string().max(20).describe("The incident ID"),
     status: z.enum(["en_route", "on_scene", "resolved", "escalated"]).describe("New status"),
     notes: z.string().max(1000).describe("Status update notes").optional(),

package/dist/templates/dispatch-center/tools/ops_protocols.ts

@@ -4,7 +4,7 @@ import { getApplicableProtocols, INCIDENT_TYPES, SEVERITIES } from "../shared.ts
 
 export const opsProtocols = tool({
   description: "Look up step-by-step response protocols for a given incident type and severity.",
-  parameters: z.object({
+  inputSchema: z.object({
     incidentType: z.enum(INCIDENT_TYPES).describe("Type of incident"),
     severity: z.enum(SEVERITIES).describe("Severity level"),
   }),

package/dist/templates/dispatch-center/tools/ops_run_scenario.ts

@@ -119,7 +119,7 @@ const SCENARIO_NAMES = Object.keys(scenarios) as [ScenarioName, ...ScenarioName[
 
 export const opsRunScenario = tool({
   description: "Run a training scenario that creates simulated incidents for dispatch practice.",
-  parameters: z.object({
+  inputSchema: z.object({
     scenario: z.enum(SCENARIO_NAMES).describe("Scenario type to simulate"),
   }),
   async execute(args, ctx) {

package/dist/templates/dispatch-center/tools/resources_dispatch.ts

@@ -12,7 +12,7 @@ import {
 export const resourcesDispatch = tool({
   description:
     "Dispatch units to an incident. Can auto-dispatch recommended resources or manually specify callsigns.",
-  parameters: z.object({
+  inputSchema: z.object({
     incidentId: z.string().max(20).describe("The incident ID"),
     callsigns: z
       .array(z.string().max(50))

package/dist/templates/dispatch-center/tools/resources_get_available.ts

@@ -4,7 +4,7 @@ import { getState, RESOURCE_TYPES } from "../shared.ts";
 
 export const resourcesGetAvailable = tool({
   description: "List available resources, optionally filtered by type.",
-  parameters: z.object({
+  inputSchema: z.object({
     type: z
       .enum([...RESOURCE_TYPES, "all"])
       .describe("Filter by resource type, or 'all'")

package/dist/templates/dispatch-center/tools/resources_update_status.ts

@@ -4,7 +4,7 @@ import { logEvent, RESOURCE_STATUSES, updateState } from "../shared.ts";
 
 export const resourcesUpdateStatus = tool({
   description: "Update a resource unit's status when it radios in.",
-  parameters: z.object({
+  inputSchema: z.object({
     callsign: z.string().max(50).describe("The resource callsign"),
     status: z.enum(RESOURCE_STATUSES).describe("New status"),
     notes: z.string().max(1000).describe("Status notes").optional(),

package/dist/templates/embedded-assets/agent.ts

@@ -1,4 +1,4 @@
-import { agent, assemblyAIPipeline, tool } from "@alexkroman1/aai";
+import { agent, tool } from "@alexkroman1/aai";
 import { z } from "zod";
 import knowledge from "./knowledge.json" with { type: "json" };
 
@@ -14,7 +14,6 @@ const searchable = faqs.map((entry) => ({
 
 export default agent({
   name: "FAQ Bot",
-  ...assemblyAIPipeline(),
   systemPrompt:
     "You are a friendly FAQ assistant. Answer questions using ONLY the information from your embedded knowledge base. If the user asks something not covered by your knowledge base, say you don't have that information and suggest they check the official documentation.\n\nRules:\n- Keep answers concise and conversational — this is a voice agent\n- Quote the knowledge base accurately, do not embellish\n- If a question is ambiguous, ask the user to clarify\n- Use 'search_knowledge' to find answers to specific questions\n- Use 'list_topics' to see all available FAQ topics\n- Always be helpful and polite",
   greeting:
@@ -31,7 +30,7 @@ export default agent({
     search_knowledge: tool({
       description:
         "Search the embedded FAQ knowledge base for an answer matching the user's question.",
-      parameters: z.object({
+      inputSchema: z.object({
         query: z.string().describe("The user's question to search for"),
       }),
       async execute(args) {

package/dist/templates/health-assistant/agent.ts

@@ -1,4 +1,4 @@
-import { agent, assemblyAIPipeline, tool } from "@alexkroman1/aai";
+import { agent, tool } from "@alexkroman1/aai";
 import { z } from "zod";
 import systemPrompt from "./system-prompt.md?raw";
 
@@ -76,7 +76,6 @@ function excerptAround(text: string, alias: string): string {
 
 export default agent({
   name: "Dr. Sage",
-  ...assemblyAIPipeline(),
   systemPrompt,
   greeting:
     "Hey, I'm Dr. Sage. Try asking me something like, what are the side effects of ibuprofen, can I take aspirin and warfarin together, or calculate my BMI. Just remember, I'm not a real doctor, so always check with your healthcare provider.",
@@ -86,7 +85,7 @@ export default agent({
     check_drug_interaction: tool({
       description:
         "Check for interactions between two or more medications using FDA drug label data. Looks up each drug's official label and reports where one drug's Drug Interactions section mentions another. Absence of a mention does not guarantee safety.",
-      parameters: z.object({
+      inputSchema: z.object({
         drugs: z
           .array(z.string().min(1))
           .min(2)
@@ -142,7 +141,7 @@ export default agent({
     medication_lookup: tool({
       description:
         "Look up detailed information about a single medication, including purpose, warnings, dosage, side effects, and manufacturer. Works with both generic and brand names.",
-      parameters: z.object({
+      inputSchema: z.object({
         name: z
           .string()
           .describe("Medication name (generic or brand, e.g. 'ibuprofen' or 'Advil')"),

package/dist/templates/infocom-adventure/agent.ts

@@ -1,15 +1,13 @@
-import { agent, assemblyAIPipeline, tool } from "@alexkroman1/aai";
-import { assemblyAI as assemblyAITts } from "@alexkroman1/aai/tts";
+import { agent, tool } from "@alexkroman1/aai";
 import { z } from "zod";
 import { getGameState, resetGameState } from "./shared.ts";
 import systemPrompt from "./system-prompt.md?raw";
 
 export default agent({
   name: "Cavern Adventure",
-  ...assemblyAIPipeline(),
-  // Overriding one stage of the preset: a narrator wants a narrative voice,
-  // and the other two stages stay as the preset set them.
-  tts: assemblyAITts({ voice: "paul" }),
+  // A narrator wants a narrative voice; everything else stays on the
+  // default all-AssemblyAI pipeline.
+  voice: "paul",
   systemPrompt,
   // The opening scene here must agree with DEFAULT_GAME_STATE.currentRoom
   // (shared.ts) and the world map in system-prompt.md.
@@ -19,7 +17,7 @@ export default agent({
   tools: {
     game_state_drop: tool({
       description: "Remove an item from the player's inventory.",
-      parameters: z.object({
+      inputSchema: z.object({
         value: z.string().describe("Item name to drop"),
       }),
       async execute(args, ctx) {
@@ -31,7 +29,7 @@ export default agent({
 
     game_state_flag: tool({
       description: "Set a game flag to true, used for tracking puzzle and event state.",
-      parameters: z.object({
+      inputSchema: z.object({
         value: z.string().describe("Flag name to set"),
       }),
       async execute(args, ctx) {
@@ -59,7 +57,7 @@ export default agent({
 
     game_state_history: tool({
       description: "Log a player command to the history and increment the move counter.",
-      parameters: z.object({
+      inputSchema: z.object({
         value: z.string().describe("Command text to log"),
       }),
       async execute(args, ctx) {
@@ -72,7 +70,7 @@ export default agent({
 
     game_state_move: tool({
       description: "Move the player to a new room and increment the move counter.",
-      parameters: z.object({
+      inputSchema: z.object({
         value: z.string().describe("Room name to move to"),
       }),
       async execute(args, ctx) {
@@ -94,7 +92,7 @@ export default agent({
 
     game_state_score: tool({
       description: "Add points to the player's score.",
-      parameters: z.object({
+      inputSchema: z.object({
         value: z.number().describe("Points to add"),
       }),
       async execute(args, ctx) {
@@ -106,7 +104,7 @@ export default agent({
 
     game_state_take: tool({
       description: "Add an item to the player's inventory.",
-      parameters: z.object({
+      inputSchema: z.object({
         value: z.string().describe("Item name to take"),
       }),
       async execute(args, ctx) {

package/dist/templates/infocom-adventure/client.tsx

@@ -241,15 +241,15 @@ function InfocomAdventure() {
               >
                 {session.running ? "[P]ause" : "[R]esume"}
               </button>
-              {/* Full reload = new WebSocket = fresh sessionId, so the
-                  session-scoped game state starts over too.
-                  (session.reset() keeps the same sessionId and would
-                  resume the old game.) */}
+              {/* end() hangs up and drops the sessionId, so the
+                  session-scoped game state starts over and the title
+                  screen returns. (session.reset() keeps the same
+                  sessionId and would resume the old game.) */}
               <button
                 type="button"
                 className="px-4 py-1 bg-transparent cursor-pointer uppercase tracking-wider font-mono text-[11px]"
                 style={{ color: GREEN_DIM, border: `1px solid ${GREEN_DARK}` }}
-                onClick={() => window.location.reload()}
+                onClick={() => session.end()}
               >
                 [N]ew Game
               </button>

package/dist/templates/math-buddy/agent.ts

@@ -1,13 +1,14 @@
-import { agent, assemblyAIPipeline } from "@alexkroman1/aai";
-import { assemblyAI as assemblyAILlm } from "@alexkroman1/aai/llm";
+import { agent } from "@alexkroman1/aai";
+import { assemblyAILlm } from "@alexkroman1/aai/llm";
 import systemPrompt from "./system-prompt.md?raw";
 
 export default agent({
   name: "Math Buddy",
-  ...assemblyAIPipeline(),
-  // Overriding the LLM stage: this tutor delegates every calculation to
-  // run_code, so it needs turn-taking speed more than reasoning depth —
-  // and Flash-Lite is both quicker and cheaper than the preset's default.
+  // Only the LLM stage is declared: this tutor delegates every calculation
+  // to run_code, so it needs turn-taking speed more than reasoning depth —
+  // and Flash-Lite is both quicker and cheaper than the default model. STT
+  // and TTS stay on the AssemblyAI defaults. (The string shorthand
+  // `llm: "gemini-2.5-flash-lite"` desugars to this same descriptor.)
   llm: assemblyAILlm({ model: "gemini-2.5-flash-lite" }),
   systemPrompt,
   greeting:

package/dist/templates/night-owl/agent.ts

@@ -1,4 +1,4 @@
-import { agent, assemblyAIPipeline, tool } from "@alexkroman1/aai";
+import { agent, tool } from "@alexkroman1/aai";
 import { z } from "zod";
 import { CATEGORIES, type Category, MOODS, type Mood } from "./shared.ts";
 
@@ -48,7 +48,6 @@ const PICKS: Record<Category, Record<Mood, string[]>> = {
 
 export default agent({
   name: "Night Owl",
-  ...assemblyAIPipeline(),
   systemPrompt:
     "You are Night Owl, a cozy evening companion. You help people wind down, recommend entertainment, and share interesting facts about the night sky. Keep your tone warm and relaxed. Use short, conversational responses.\n\nUse run_code for sleep calculations:\n- Each sleep cycle is 90 minutes, plus 15 minutes to fall asleep\n- Bedtime = wake_time - (cycles * 90 + 15) minutes\n- If result is negative, add 1440 (24 hours in minutes)\n- Format as HH:MM",
   greeting:
@@ -58,7 +57,7 @@ export default agent({
   tools: {
     recommend: tool({
       description: "Get recommendations for movies, music, or books based on mood.",
-      parameters: z.object({
+      inputSchema: z.object({
         category: z.enum(CATEGORIES),
         mood: z.enum(MOODS),
       }),

package/dist/templates/personal-finance/agent.ts

@@ -1,9 +1,8 @@
-import { agent, assemblyAIPipeline } from "@alexkroman1/aai";
+import { agent } from "@alexkroman1/aai";
 import systemPrompt from "./system-prompt.md?raw";
 
 export default agent({
   name: "Penny",
-  ...assemblyAIPipeline(),
   systemPrompt,
   greeting:
     "Hey, I'm Penny, your personal finance helper. Try asking me something like, what's 100 dollars in euros, what's the price of bitcoin, or help me split a 120 dollar bill four ways with 20 percent tip.",

package/dist/templates/pipeline-simple/agent.test.ts

@@ -4,29 +4,29 @@ import agentDef from "./agent.ts";
 
 describe("pipeline-simple template", () => {
   test("config passes manifest validation", () => {
-    // Same conversion `aai build`/`aai deploy` run — catches a partial
-    // provider triple or invalid tuning, not just descriptor presence.
+    // Same conversion `aai build`/`aai deploy` run — catches an invalid
+    // provider combination or tuning, not just descriptor presence.
     expect(() => toAgentConfig(agentDef)).not.toThrow();
   });
 
-  test("exports an agent with all three pipeline providers", () => {
-    // Presence of stt + llm + tts is what flips the runtime into
-    // pipeline mode (see parseManifest). Smoke-test each is wired up.
+  test("declares only the LLM stage on the def", () => {
+    // The template's point: swap one stage, leave the rest unset. The
+    // unset stages are filled with the AssemblyAI defaults at parse time.
     expect(agentDef.name).toBe("pipeline-simple");
-    expect(agentDef.stt).toBeDefined();
     expect(agentDef.llm).toBeDefined();
-    expect(agentDef.tts).toBeDefined();
-  });
-
-  test("STT descriptor is AssemblyAI", () => {
-    expect(agentDef.stt?.kind).toBe("assemblyai");
+    expect(agentDef.stt).toBeUndefined();
+    expect(agentDef.tts).toBeUndefined();
   });
 
   test("LLM descriptor is Anthropic", () => {
     expect(agentDef.llm?.kind).toBe("anthropic");
   });
 
-  test("TTS descriptor is AssemblyAI", () => {
-    expect(agentDef.tts?.kind).toBe("assemblyai");
+  test("unset stages fill to AssemblyAI in the deployable config", () => {
+    const config = toAgentConfig(agentDef);
+    expect(config.mode).toBe("pipeline");
+    expect(config.stt?.kind).toBe("assemblyai");
+    expect(config.tts?.kind).toBe("assemblyai");
+    expect(config.llm?.kind).toBe("anthropic");
   });
 });

package/dist/templates/pipeline-simple/agent.ts

@@ -1,13 +1,12 @@
-import { agent, assemblyAIPipeline } from "@alexkroman1/aai";
+import { agent } from "@alexkroman1/aai";
 import { anthropic } from "@alexkroman1/aai/llm";
 
 export default agent({
   name: "pipeline-simple",
   greeting: "Hi! I'm running in pipeline mode. What can I help with?",
-  // The preset fills all three stages with AssemblyAI, billed to the one key
-  // a published agent is guaranteed to have. Spread it, then override the one
-  // stage you want elsewhere — here the LLM. Anything not overridden stays
-  // AssemblyAI, so swapping a stage never means restating the other two.
-  ...assemblyAIPipeline(),
+  // Declare only the stage you want elsewhere — here the LLM. Every stage
+  // left unset (STT and TTS here) runs on the AssemblyAI default, billed to
+  // the one key a published agent is guaranteed to have, so swapping a stage
+  // never means restating the other two.
   llm: anthropic({ model: "claude-haiku-4-5" }),
 });

package/dist/templates/pizza-ordering/agent.test.ts

@@ -76,17 +76,19 @@ describe("pricing (shared.ts)", () => {
     expect(calculateTotal([a, b])).toBeCloseTo(8.99 * 4, 5);
   });
 
-  test("add_pizza schema defaults quantity to 1 and rejects non-positive quantities", () => {
-    const schema = getTool("add_pizza").parameters;
-    if (!schema) throw new Error("add_pizza has no parameters schema");
-    const parsed = schema.parse({ size: "small", crust: "thin", toppings: [] });
-    expect(parsed.quantity).toBe(1);
-    expect(() =>
-      schema.parse({ size: "small", crust: "thin", toppings: [], quantity: 0 }),
-    ).toThrow();
-    expect(() =>
-      schema.parse({ size: "small", crust: "thin", toppings: [], quantity: 1.5 }),
-    ).toThrow();
+  test("add_pizza schema defaults quantity to 1 and rejects non-positive quantities", async () => {
+    const schema = getTool("add_pizza").inputSchema;
+    if (!schema) throw new Error("add_pizza has no input schema");
+    // Validate through the Standard Schema contract — the vendor-neutral
+    // interface every inputSchema carries.
+    const validate = (value: unknown) => schema["~standard"].validate(value);
+    const ok = await validate({ size: "small", crust: "thin", toppings: [] });
+    if (ok.issues) throw new Error("expected valid input");
+    expect((ok.value as { quantity: number }).quantity).toBe(1);
+    const zero = await validate({ size: "small", crust: "thin", toppings: [], quantity: 0 });
+    expect(zero.issues).toBeDefined();
+    const frac = await validate({ size: "small", crust: "thin", toppings: [], quantity: 1.5 });
+    expect(frac.issues).toBeDefined();
   });
 });
 

package/dist/templates/pizza-ordering/agent.ts

@@ -1,4 +1,4 @@
-import { agent, assemblyAIPipeline, tool } from "@alexkroman1/aai";
+import { agent, tool } from "@alexkroman1/aai";
 import { z } from "zod";
 import {
   CRUSTS,
@@ -21,7 +21,6 @@ const crusts = z.enum(CRUSTS);
 
 export default agent({
   name: "Pizza Palace",
-  ...assemblyAIPipeline(),
   // The cart, pushed to the client after every tool call. Replaces a
   // `ctx.send("order", ...)` in each of the five order tools, and the
   // event-diffing the client had to do to rebuild the cart from them.
@@ -35,7 +34,7 @@ export default agent({
   tools: {
     add_pizza: tool({
       description: "Add a pizza to the order. Use when the customer has decided on a pizza.",
-      parameters: z.object({
+      inputSchema: z.object({
         size: sizes,
         crust: crusts,
         toppings: z
@@ -93,7 +92,7 @@ export default agent({
 
     remove_pizza: tool({
       description: "Remove a pizza from the order by its ID.",
-      parameters: z.object({
+      inputSchema: z.object({
         pizza_id: z.number().describe("The pizza ID to remove"),
       }),
       async execute(args, ctx) {
@@ -113,7 +112,7 @@ export default agent({
 
     set_customer_name: tool({
       description: "Set the customer name for the order.",
-      parameters: z.object({
+      inputSchema: z.object({
         name: z.string(),
       }),
       async execute(args, ctx) {
@@ -124,7 +123,7 @@ export default agent({
 
     update_pizza: tool({
       description: "Update an existing pizza in the order. Only provided fields are changed.",
-      parameters: z.object({
+      inputSchema: z.object({
         pizza_id: z.number(),
         size: sizes.optional(),
         crust: crusts.optional(),

package/dist/templates/retail/address.ts

@@ -0,0 +1,29 @@
+import { z } from "zod";
+import type { Address } from "./shared.ts";
+
+/** Spread into both address tools' schemas. Two hand-copied zod shapes is how
+ *  the order address and the profile address drift apart. */
+export const AddressFields = {
+  address1: z.string().max(200).describe("First line, e.g. '123 Main St'"),
+  address2: z.string().max(200).describe("Second line, e.g. 'Apt 1' — pass '' if there is none"),
+  city: z.string().max(100).describe("City, e.g. 'San Francisco'"),
+  state: z.string().max(60).describe("State, e.g. 'CA'"),
+  country: z.string().max(60).describe("Country, e.g. 'USA'"),
+  zip: z.string().max(20).describe("Zip code, e.g. '12345'"),
+};
+
+export function toAddress(args: Address): Address {
+  return {
+    address1: args.address1,
+    address2: args.address2,
+    city: args.city,
+    state: args.state,
+    country: args.country,
+    zip: args.zip,
+  };
+}
+
+export function formatAddress(address: Address): string {
+  const lines = [address.address1, address.address2].filter(Boolean).join(", ");
+  return `${lines}, ${address.city} ${address.state} ${address.zip}, ${address.country}`;
+}