@alexkroman1/aai 0.7.12 → 0.8.0

package/templates/dispatch-center/agent.ts

@@ -1,4 +1,4 @@
-import { defineAgent } from "@alexkroman1/aai";
+import { defineAgent, tool } from "@alexkroman1/aai";
 import { z } from "zod";
 import type { HookContext, ToolContext } from "@alexkroman1/aai";
 
@@ -619,7 +619,7 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
   },
 
   tools: {
-    incident_create: {
+    incident_create: tool({
       description: "Create a new incident from an incoming emergency call.",
       parameters: z.object({
         location: z.string().describe("Address or location description"),
@@ -643,17 +643,10 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           callerPhone,
           estimatedCasualties,
           hazards,
-        }: {
-          location: string;
-          description: string;
-          callerName?: string;
-          callerPhone?: string;
-          estimatedCasualties?: number;
-          hazards?: string[];
         },
         ctx,
       ) => {
-        const state = ctx.state;
+        const state = ctx.state as DispatchState;
         state.incidentCounter++;
         const id = `INC-${String(state.incidentCounter).padStart(4, "0")}`;
 
@@ -722,9 +715,9 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
             : `${id} created. Triage score ${triageScore}. ${recommended.length} resource(s) recommended.`,
         };
       },
-    },
+    }),
 
-    incident_triage: {
+    incident_triage: tool({
       description:
         "Triage an incident — confirm or override severity, type, hazards, and casualty count.",
       parameters: z.object({
@@ -756,17 +749,10 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           additionalHazards,
           casualtyUpdate,
           notes,
-        }: {
-          incidentId: string;
-          severity?: "critical" | "urgent" | "moderate" | "minor";
-          type?: "medical" | "fire" | "hazmat" | "traffic" | "crime" | "natural_disaster" | "utility" | "other";
-          additionalHazards?: string[];
-          casualtyUpdate?: number;
-          notes?: string;
         },
         ctx,
       ) => {
-        const state = ctx.state;
+        const state = ctx.state as DispatchState;
         const inc = state.incidents[incidentId];
         if (!inc) return { error: `Incident ${incidentId} not found` };
 
@@ -821,9 +807,9 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           systemAlertLevel: state.alertLevel,
         };
       },
-    },
+    }),
 
-    incident_update_status: {
+    incident_update_status: tool({
       description:
         "Update an incident's status (en_route, on_scene, resolved, escalated).",
       parameters: z.object({
@@ -837,15 +823,10 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
         }).describe("Updated casualty numbers").optional(),
       }),
       execute: async (
-        { incidentId, status, notes, casualtyUpdate }: {
-          incidentId: string;
-          status: "en_route" | "on_scene" | "resolved" | "escalated";
-          notes?: string;
-          casualtyUpdate?: { confirmed?: number; treated?: number };
-        },
+        { incidentId, status, notes, casualtyUpdate },
         ctx,
       ) => {
-        const state = ctx.state;
+        const state = ctx.state as DispatchState;
         const inc = state.incidents[incidentId];
         if (!inc) return { error: `Incident ${incidentId} not found` };
 
@@ -905,9 +886,9 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           systemAlertLevel: state.alertLevel,
         };
       },
-    },
+    }),
 
-    incident_escalate: {
+    incident_escalate: tool({
       description:
         "Escalate an incident when it exceeds current capacity or severity increases.",
       parameters: z.object({
@@ -921,15 +902,10 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
         ).optional(),
       }),
       execute: async (
-        { incidentId, reason, requestMutualAid, newSeverity }: {
-          incidentId: string;
-          reason: string;
-          requestMutualAid?: boolean;
-          newSeverity?: "critical" | "urgent";
-        },
+        { incidentId, reason, requestMutualAid, newSeverity },
         ctx,
       ) => {
-        const state = ctx.state;
+        const state = ctx.state as DispatchState;
         const inc = state.incidents[incidentId];
         if (!inc) return { error: `Incident ${incidentId} not found` };
 
@@ -1004,16 +980,16 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
             `ESCALATION CONFIRMED — ${incidentId} now Level ${inc.escalationLevel}. ${additionalResources.length} additional resource(s) available for dispatch.`,
         };
       },
-    },
+    }),
 
-    incident_get: {
+    incident_get: tool({
       description:
         "Get full details on a specific incident including timeline and assigned resources.",
       parameters: z.object({
         incidentId: z.string().describe("The incident ID"),
       }),
-      execute: ({ incidentId }: { incidentId: string }, ctx) => {
-        const state = ctx.state;
+      execute: ({ incidentId }, ctx) => {
+        const state = ctx.state as DispatchState;
         const inc = state.incidents[incidentId];
         if (!inc) return { error: `Incident ${incidentId} not found` };
 
@@ -1044,9 +1020,9 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
             .map((p) => p.name),
         };
       },
-    },
+    }),
 
-    incident_add_note: {
+    incident_add_note: tool({
       description: "Add a situational update note to an incident.",
       parameters: z.object({
         incidentId: z.string().describe("The incident ID"),
@@ -1055,8 +1031,8 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           "Who reported this — unit callsign or caller",
         ).optional(),
       }),
-      execute: async ({ incidentId, note, source }: { incidentId: string; note: string; source?: string }, ctx) => {
-        const state = ctx.state;
+      execute: async ({ incidentId, note, source }, ctx) => {
+        const state = ctx.state as DispatchState;
         const inc = state.incidents[incidentId];
         if (!inc) return { error: `Incident ${incidentId} not found` };
 
@@ -1072,9 +1048,9 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           totalNotes: inc.notes.length,
         };
       },
-    },
+    }),
 
-    resources_dispatch: {
+    resources_dispatch: tool({
       description:
         "Dispatch units to an incident. Can auto-dispatch recommended resources or manually specify callsigns.",
       parameters: z.object({
@@ -1090,15 +1066,10 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
         ).optional(),
       }),
       execute: async (
-        { incidentId, callsigns, autoDispatch, priority }: {
-          incidentId: string;
-          callsigns?: string[];
-          autoDispatch?: boolean;
-          priority?: "routine" | "priority" | "emergency";
-        },
+        { incidentId, callsigns, autoDispatch, priority },
         ctx,
       ) => {
-        const state = ctx.state;
+        const state = ctx.state as DispatchState;
         const inc = state.incidents[incidentId];
         if (!inc) return { error: `Incident ${incidentId} not found` };
 
@@ -1180,9 +1151,9 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
             : undefined,
         };
       },
-    },
+    }),
 
-    resources_get_available: {
+    resources_get_available: tool({
       description: "List available resources, optionally filtered by type.",
       parameters: z.object({
         type: z.enum([
@@ -1197,8 +1168,8 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           "all",
         ]).describe("Filter by resource type, or 'all'").optional(),
       }),
-      execute: ({ type }: { type?: "ambulance" | "fire_engine" | "police" | "hazmat_team" | "helicopter" | "k9_unit" | "swat" | "ems_supervisor" | "all" }, ctx) => {
-        const state = ctx.state;
+      execute: ({ type }, ctx) => {
+        const state = ctx.state as DispatchState;
         let resources = state.resources;
         if (type && type !== "all") {
           resources = resources.filter((r) => r.type === type);
@@ -1222,9 +1193,9 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           },
         };
       },
-    },
+    }),
 
-    resources_update_status: {
+    resources_update_status: tool({
       description: "Update a resource unit's status when it radios in.",
       parameters: z.object({
         callsign: z.string().describe("The resource callsign"),
@@ -1237,8 +1208,8 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
         ]).describe("New status"),
         notes: z.string().describe("Status notes").optional(),
       }),
-      execute: async ({ callsign, status, notes }: { callsign: string; status: "available" | "dispatched" | "en_route" | "on_scene" | "returning"; notes?: string }, ctx) => {
-        const state = ctx.state;
+      execute: async ({ callsign, status, notes }, ctx) => {
+        const state = ctx.state as DispatchState;
         const resource = state.resources.find((r) =>
           r.callsign.toLowerCase() === callsign.toLowerCase()
         );
@@ -1279,13 +1250,13 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           systemAlertLevel: state.alertLevel,
         };
       },
-    },
+    }),
 
     ops_dashboard: {
       description:
         "Get the full operational dashboard: alert level, resource utilization, active incidents, and available resources.",
       execute: (_args, ctx) => {
-        const state = ctx.state;
+        const state = ctx.state as DispatchState;
 
         const activeIncidents = Object.values(state.incidents)
           .filter((i) => i.status !== "resolved")
@@ -1342,7 +1313,7 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
       },
     },
 
-    ops_protocols: {
+    ops_protocols: tool({
       description:
         "Look up step-by-step response protocols for a given incident type and severity.",
       parameters: z.object({
@@ -1359,7 +1330,7 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
         severity: z.enum(["critical", "urgent", "moderate", "minor"])
           .describe("Severity level"),
       }),
-      execute: ({ incidentType, severity }: { incidentType: IncidentType; severity: Severity }) => {
+      execute: ({ incidentType, severity }) => {
         const protocols = getApplicableProtocols(
           incidentType,
           severity,
@@ -1379,9 +1350,9 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           })),
         };
       },
-    },
+    }),
 
-    ops_run_scenario: {
+    ops_run_scenario: tool({
       description:
         "Run a training scenario that creates simulated incidents for dispatch practice.",
       parameters: z.object({
@@ -1393,8 +1364,8 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
           "highway_pileup",
         ]).describe("Scenario type to simulate"),
       }),
-      execute: async ({ scenario }: { scenario: string }, ctx) => {
-        const state = ctx.state;
+      execute: async ({ scenario }, ctx) => {
+        const state = ctx.state as DispatchState;
         const scenarios: Record<
           string,
           { incidents: Partial<Incident>[]; narrative: string }
@@ -1559,6 +1530,6 @@ Radio style: "Medic-1, respond priority one to 400 Oak Street, report of cardiac
             `SCENARIO ACTIVE: ${s.narrative}. ${created.length} incidents created. Awaiting dispatch orders.`,
         };
       },
-    },
+    }),
   },
 });

package/templates/embedded-assets/agent.ts

@@ -1,4 +1,4 @@
-import { defineAgent } from "@alexkroman1/aai";
+import { defineAgent, tool } from "@alexkroman1/aai";
 import { z } from "zod";
 import knowledge from "./knowledge.json" with { type: "json" };
 
@@ -22,15 +22,14 @@ Rules:
 - Always be helpful and polite`,
   greeting:
     "Hi! I'm your FAQ assistant. Ask me anything about the AAI agent framework and I'll look it up in my knowledge base.",
-  voice: "156fb8d2-335b-4950-9cb3-a2d33befec77", // Helpful Woman
   tools: {
-    search_knowledge: {
+    search_knowledge: tool({
       description:
         "Search the embedded FAQ knowledge base for an answer matching the user's question.",
       parameters: z.object({
         query: z.string().describe("The user's question to search for"),
       }),
-      execute: ({ query }: { query: string }) => {
+      execute: ({ query }) => {
         const q = query.toLowerCase();
         const match = faqs.find((f) =>
           f.question.toLowerCase().includes(q) ||
@@ -39,7 +38,7 @@ Rules:
         );
         return match ?? { result: "No matching FAQ found." };
       },
-    },
+    }),
     list_topics: {
       description:
         "List all available topics in the embedded FAQ knowledge base.",

package/templates/health-assistant/agent.ts

@@ -1,4 +1,4 @@
-import { defineAgent } from "@alexkroman1/aai";
+import { defineAgent, tool } from "@alexkroman1/aai";
 import { z } from "zod";
 
 function first(field: unknown): string | undefined {
@@ -136,7 +136,7 @@ Use run_code for health calculations:
     "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.",
   builtinTools: ["web_search", "run_code"],
   tools: {
-    medication_lookup: {
+    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({
@@ -144,9 +144,9 @@ Use run_code for health calculations:
           "Medication name (generic or brand, e.g. 'ibuprofen' or 'Advil')",
         ),
       }),
-      execute: ({ name }: { name: string }) => lookupDrug(name),
-    },
-    check_drug_interaction: {
+      execute: ({ name }) => lookupDrug(name),
+    }),
+    check_drug_interaction: tool({
       description:
         "Check for known interactions between two or more medications. Resolves drug names via RxNorm and returns interaction details with severity levels.",
       parameters: z.object({
@@ -154,7 +154,7 @@ Use run_code for health calculations:
           "Comma-separated medication names (e.g. 'ibuprofen, warfarin')",
         ),
       }),
-      execute: ({ drugs }: { drugs: string }) => checkInteractions(drugs),
-    },
+      execute: ({ drugs }) => checkInteractions(drugs),
+    }),
   },
 });

package/templates/infocom-adventure/agent.ts

@@ -1,6 +1,6 @@
-import { defineAgent } from "@alexkroman1/aai";
-import { z } from "zod";
+import { defineAgent, tool } from "@alexkroman1/aai";
 import type { ToolContext } from "@alexkroman1/aai";
+import { z } from "zod";
 
 type GameState = {
   inventory: string[];
@@ -89,76 +89,76 @@ ATMOSPHERE:
         };
       },
     },
-    game_state_move: {
+    game_state_move: tool({
       description:
         "Move the player to a new room and increment the move counter.",
       parameters: z.object({
         value: z.string().describe("Room name to move to"),
       }),
-      execute: ({ value }: { value: string }, ctx) => {
+      execute: ({ value }, ctx) => {
         const g = s(ctx);
         g.currentRoom = value;
         g.moves++;
         return { currentRoom: g.currentRoom, moves: g.moves };
       },
-    },
-    game_state_take: {
+    }),
+    game_state_take: tool({
       description: "Add an item to the player's inventory.",
       parameters: z.object({
         value: z.string().describe("Item name to take"),
       }),
-      execute: ({ value }: { value: string }, ctx) => {
+      execute: ({ value }, ctx) => {
         const g = s(ctx);
         if (!g.inventory.includes(value)) g.inventory.push(value);
         return { inventory: g.inventory };
       },
-    },
-    game_state_drop: {
+    }),
+    game_state_drop: tool({
       description: "Remove an item from the player's inventory.",
       parameters: z.object({
         value: z.string().describe("Item name to drop"),
       }),
-      execute: ({ value }: { value: string }, ctx) => {
+      execute: ({ value }, ctx) => {
         const g = s(ctx);
         g.inventory = g.inventory.filter((i) => i !== value);
         return { inventory: g.inventory };
       },
-    },
-    game_state_score: {
+    }),
+    game_state_score: tool({
       description: "Add points to the player's score.",
       parameters: z.object({
         value: z.number().describe("Points to add"),
       }),
-      execute: ({ value }: { value: number }, ctx) => {
+      execute: ({ value }, ctx) => {
         const g = s(ctx);
         g.score += value;
         return { score: g.score };
       },
-    },
-    game_state_flag: {
+    }),
+    game_state_flag: tool({
       description:
         "Set a game flag to true, used for tracking puzzle and event state.",
       parameters: z.object({
         value: z.string().describe("Flag name to set"),
       }),
-      execute: ({ value }: { value: string }, ctx) => {
+      execute: ({ value }, ctx) => {
         const g = s(ctx);
         g.flags[value] = true;
         return { flags: g.flags };
       },
-    },
-    game_state_history: {
+    }),
+    game_state_history: tool({
       description:
         "Log a player command to the history and increment the move counter.",
       parameters: z.object({
         value: z.string().describe("Command text to log"),
       }),
-      execute: ({ value }: { value: string }, ctx) => {
+      execute: ({ value }, ctx) => {
         const g = s(ctx);
         g.history.push(value);
         g.moves++;
         return { moves: g.moves, recentHistory: g.history.slice(-5) };
       },
-    },
+    }),
   },
 });

package/templates/memory-agent/agent.ts

@@ -1,5 +1,4 @@
 import { defineAgent } from "@alexkroman1/aai";
-import { z } from "zod";
 
 export default defineAgent({
   name: "Memory Agent",
@@ -17,58 +16,5 @@ Keep responses concise and conversational. Never say "I saved that to my \
 database" — just confirm naturally, like "Got it, I'll remember that."`,
   greeting:
     "Hey there. I'm an assistant with a long-term memory. Tell me things you want me to remember, and I'll recall them in future conversations.",
-  builtinTools: ["web_search"],
-  tools: {
-    save_memory: {
-      description:
-        "Save a piece of information to persistent memory. Use a descriptive key like 'user:name' or 'project:status'.",
-      parameters: z.object({
-        key: z.string().describe(
-          "A descriptive key for this memory (e.g. 'user:name', 'preference:color')",
-        ),
-        value: z.string().describe("The information to remember"),
-      }),
-      execute: async (
-        { key, value }: { key: string; value: string },
-        ctx,
-      ) => {
-        await ctx.kv.set(key, value);
-        return { saved: key };
-      },
-    },
-    recall_memory: {
-      description: "Retrieve a previously saved memory by its key.",
-      parameters: z.object({
-        key: z.string().describe("The key to look up"),
-      }),
-      execute: async ({ key }: { key: string }, ctx) => {
-        const value = await ctx.kv.get(key);
-        if (value === null) return { found: false, key };
-        return { found: true, key, value };
-      },
-    },
-    list_memories: {
-      description:
-        "List all saved memory keys, optionally filtered by a prefix (e.g. 'user:').",
-      parameters: z.object({
-        prefix: z.string().describe(
-          "Prefix to filter keys (e.g. 'user:'). Use empty string for all.",
-        ).optional(),
-      }),
-      execute: async ({ prefix }: { prefix?: string }, ctx) => {
-        const entries = await ctx.kv.list(prefix ?? "");
-        return { count: entries.length, keys: entries.map((e) => e.key) };
-      },
-    },
-    forget_memory: {
-      description: "Delete a previously saved memory by its key.",
-      parameters: z.object({
-        key: z.string().describe("The key to delete"),
-      }),
-      execute: async ({ key }: { key: string }, ctx) => {
-        await ctx.kv.delete(key);
-        return { deleted: key };
-      },
-    },
-  },
+  builtinTools: ["web_search", "memory"],
 });

package/templates/night-owl/agent.ts

@@ -1,4 +1,4 @@
-import { defineAgent } from "@alexkroman1/aai";
+import { defineAgent, tool } from "@alexkroman1/aai";
 import { z } from "zod";
 
 const PICKS: Record<string, Record<string, string[]>> = {
@@ -79,20 +79,20 @@ Use run_code for sleep calculations:
     "Hey there, night owl. Try asking me for a cozy movie recommendation, or tell me what time you need to wake up and I'll calculate the best time to fall asleep.",
   builtinTools: ["run_code"],
   tools: {
-    recommend: {
+    recommend: tool({
       description:
         "Get recommendations for movies, music, or books based on mood.",
       parameters: z.object({
         category: z.enum(["movie", "music", "book"]),
         mood: z.enum(["chill", "intense", "cozy", "spooky", "funny"]),
       }),
-      execute: ({ category, mood }: { category: string; mood: string }) => {
+      execute: ({ category, mood }) => {
         return {
           category,
           mood,
           picks: PICKS[category]?.[mood] ?? [],
         };
       },
-    },
+    }),
   },
 });

package/templates/night-owl/client.tsx

@@ -1,29 +1,12 @@
 import "@alexkroman1/aai/ui/styles.css";
-import { ChatView, mount, useSession } from "@alexkroman1/aai/ui";
+import { ChatView, StartScreen, mount } from "@alexkroman1/aai/ui";
 
 function NightOwl() {
-  const { started, start } = useSession();
-
-  if (!started.value) {
-    return (
-      <div class="flex items-center justify-center h-screen bg-aai-bg font-aai">
-        <div class="flex flex-col items-center gap-6 bg-aai-surface border border-aai-border rounded-lg px-12 py-10">
-          <div class="text-5xl">&#x1F989;</div>
-          <h1 class="text-xl font-semibold text-aai-text m-0">Night Owl</h1>
-          <p class="text-sm text-aai-text-muted m-0">your evening companion</p>
-          <button
-            type="button"
-            class="mt-2 px-8 py-3 rounded-aai text-sm font-medium cursor-pointer bg-aai-primary text-white border-none"
-            onClick={start}
-          >
-            Start Conversation
-          </button>
-        </div>
-      </div>
-    );
-  }
-
-  return <ChatView />;
+  return (
+    <StartScreen icon={<span>&#x1F989;</span>} title="Night Owl" subtitle="your evening companion" buttonText="Start Conversation">
+      <ChatView />
+    </StartScreen>
+  );
 }
 
 mount(NightOwl);