@alexkroman1/aai 0.4.0 → 0.6.0

package/dist/cli.js

@@ -58,26 +58,6 @@ var init_output = __esm({
   }
 });
 
-// sdk/types.ts
-var DEFAULT_INSTRUCTIONS, DEFAULT_GREETING;
-var init_types = __esm({
-  "sdk/types.ts"() {
-    "use strict";
-    DEFAULT_INSTRUCTIONS = `You are AAI, a helpful AI assistant.
-
-Voice-First Rules:
-- Optimize for natural speech. Avoid jargon unless central to the answer. Use short, punchy sentences.
-- Never mention "search results," "sources," or "the provided text." Speak as if the knowledge is your own.
-- No visual formatting. Do not say "bullet point," "bold," or "bracketed one." If you need to list items, say "First," "Next," and "Finally."
-- Start with the most important information. No introductory filler.
-- Be concise. Keep answers to 1-3 sentences. For complex topics, provide a high-level summary.
-- Be confident. Avoid hedging phrases like "It seems that" or "I believe."
-- If you don't have enough information, say so directly rather than guessing.
-- Never use exclamation points. Keep your tone calm and conversational.`;
-    DEFAULT_GREETING = "Hey there. I'm a voice assistant. What can I help you with?";
-  }
-});
-
 // cli/_new.ts
 var new_exports = {};
 __export(new_exports, {
@@ -174,11 +154,11 @@ var init_new = __esm({
   }
 });
 
-// sdk/protocol.ts
+// dist/sdk/protocol.js
 import { z } from "zod";
 var PROTOCOL_VERSION, DEFAULT_TTS_SAMPLE_RATE, AUDIO_FORMAT, _bitsPerSample, _channels, AudioFrameSpec, KvRequestBaseSchema, HOOK_TIMEOUT_MS, SessionErrorCodeSchema, TranscriptEventSchema, ClientEventSchema, ClientMessageSchema;
 var init_protocol = __esm({
-  "sdk/protocol.ts"() {
+  "dist/sdk/protocol.js"() {
     "use strict";
     PROTOCOL_VERSION = 1;
     DEFAULT_TTS_SAMPLE_RATE = 24e3;
@@ -266,39 +246,45 @@ var init_protocol = __esm({
       z.object({ type: z.literal("reset") }),
       z.object({
         type: z.literal("history"),
-        messages: z.array(
-          z.object({
-            role: z.enum(["user", "assistant"]),
-            text: z.string().max(1e5)
-          })
-        ).max(200)
+        messages: z.array(z.object({
+          role: z.enum(["user", "assistant"]),
+          text: z.string().max(1e5)
+        })).max(200)
       })
     ]);
   }
 });
 
-// sdk/runtime.ts
+// dist/sdk/runtime.js
 var consoleLogger, noopMetrics, DEFAULT_S2S_CONFIG;
 var init_runtime = __esm({
-  "sdk/runtime.ts"() {
+  "dist/sdk/runtime.js"() {
     "use strict";
     init_protocol();
     consoleLogger = {
       info(msg, ctx) {
-        if (ctx) console.log(msg, ctx);
-        else console.log(msg);
+        if (ctx)
+          console.log(msg, ctx);
+        else
+          console.log(msg);
       },
       warn(msg, ctx) {
-        if (ctx) console.warn(msg, ctx);
-        else console.warn(msg);
+        if (ctx)
+          console.warn(msg, ctx);
+        else
+          console.warn(msg);
       },
       error(msg, ctx) {
-        if (ctx) console.error(msg, ctx);
-        else console.error(msg);
+        if (ctx)
+          console.error(msg, ctx);
+        else
+          console.error(msg);
       },
       debug(msg, ctx) {
-        if (ctx) console.debug(msg, ctx);
-        else console.debug(msg);
+        if (ctx)
+          console.debug(msg, ctx);
+        else
+          console.debug(msg);
       }
     };
     noopMetrics = {
@@ -316,7 +302,7 @@ var init_runtime = __esm({
   }
 });
 
-// sdk/_internal_types.ts
+// dist/sdk/_internal_types.js
 import { z as z2 } from "zod";
 function agentToolsToSchemas(tools) {
   return Object.entries(tools).map(([name, def]) => ({
@@ -327,13 +313,13 @@ function agentToolsToSchemas(tools) {
 }
 var EMPTY_PARAMS;
 var init_internal_types = __esm({
-  "sdk/_internal_types.ts"() {
+  "dist/sdk/_internal_types.js"() {
     "use strict";
     EMPTY_PARAMS = z2.object({});
   }
 });
 
-// sdk/builtin_tools.ts
+// dist/sdk/builtin_tools.js
 import { z as z3 } from "zod";
 function htmlToText(html) {
   return html.replace(/<script[^>]*>[\s\S]*?<\/script>/gi, "").replace(/<style[^>]*>[\s\S]*?<\/style>/gi, "").replace(/<(br|hr)\s*\/?>/gi, "\n").replace(/<\/(p|div|li|h[1-6]|tr|blockquote)>/gi, "\n").replace(/<[^>]+>/g, " ").replace(/&nbsp;/g, " ").replace(/&amp;/g, "&").replace(/&lt;/g, "<").replace(/&gt;/g, ">").replace(/&quot;/g, '"').replace(/&#0?39;/g, "'").replace(/[ \t]+/g, " ").replace(/\n /g, "\n").replace(/\n{3,}/g, "\n\n").trim();
@@ -357,10 +343,12 @@ function createWebSearch() {
         headers: { "X-Subscription-Token": apiKey },
         signal: ctx.abortSignal ?? AbortSignal.timeout(15e3)
       });
-      if (!resp.ok) return [];
+      if (!resp.ok)
+        return [];
       const raw = await resp.json();
       const data = BraveSearchResponseSchema.safeParse(raw);
-      if (!data.success) return [];
+      if (!data.success)
+        return [];
       return (data.data.web?.results ?? []).slice(0, maxResults).map((r) => ({
         title: r.title,
         url: r.url,
@@ -488,7 +476,8 @@ function getBuiltinToolDefs(names, opts) {
 function getBuiltinToolSchemas(names) {
   return names.flatMap((name) => {
     const creator = TOOL_CREATORS[name];
-    if (!creator) return [];
+    if (!creator)
+      return [];
     const def = creator();
     return [
       {
@@ -501,7 +490,7 @@ function getBuiltinToolSchemas(names) {
 }
 var EMPTY_PARAMS2, webSearchParams, BRAVE_SEARCH_URL, BraveSearchResponseSchema, MAX_PAGE_CHARS, MAX_HTML_BYTES, visitWebpageParams, fetchJsonParams, runCodeParams, vectorSearchParams, TOOL_CREATORS;
 var init_builtin_tools = __esm({
-  "sdk/builtin_tools.ts"() {
+  "dist/sdk/builtin_tools.js"() {
     "use strict";
     EMPTY_PARAMS2 = z3.object({});
     webSearchParams = z3.object({
@@ -511,13 +500,11 @@ var init_builtin_tools = __esm({
     BRAVE_SEARCH_URL = "https://api.search.brave.com/res/v1/web/search";
     BraveSearchResponseSchema = z3.object({
       web: z3.object({
-        results: z3.array(
-          z3.object({
-            title: z3.string(),
-            url: z3.string(),
-            description: z3.string()
-          })
-        )
+        results: z3.array(z3.object({
+          title: z3.string(),
+          url: z3.string(),
+          description: z3.string()
+        }))
       }).optional()
     });
     MAX_PAGE_CHARS = 1e4;
@@ -533,9 +520,7 @@ var init_builtin_tools = __esm({
       code: z3.string().describe("JavaScript code to execute. Use console.log() for output.")
     });
     vectorSearchParams = z3.object({
-      query: z3.string().describe(
-        'Short keyword query to search the knowledge base. Use specific topic terms, not full sentences. Do NOT include the company or product name since all documents are from the same source. For example, if the user asks "how much does Acme cost", search for "pricing plans rates".'
-      ),
+      query: z3.string().describe('Short keyword query to search the knowledge base. Use specific topic terms, not full sentences. Do NOT include the company or product name since all documents are from the same source. For example, if the user asks "how much does Acme cost", search for "pricing plans rates".'),
       topK: z3.number().describe("Maximum results to return (default: 5)").optional()
     });
     TOOL_CREATORS = {
@@ -549,10 +534,11 @@ var init_builtin_tools = __esm({
   }
 });
 
-// sdk/kv.ts
+// dist/sdk/kv.js
 function sortAndPaginate(entries, options) {
   entries.sort((a, b) => a.key < b.key ? -1 : a.key > b.key ? 1 : 0);
-  if (options?.reverse) entries.reverse();
+  if (options?.reverse)
+    entries.reverse();
   if (options?.limit && options.limit > 0) {
     entries.length = Math.min(entries.length, options.limit);
   }
@@ -567,7 +553,8 @@ function createMemoryKv() {
     get(key) {
       const entry = store.get(key);
       if (!entry || isExpired(entry)) {
-        if (entry) store.delete(key);
+        if (entry)
+          store.delete(key);
         return Promise.resolve(null);
       }
       return Promise.resolve(JSON.parse(entry.raw));
@@ -607,13 +594,13 @@ function createMemoryKv() {
 }
 var MAX_VALUE_SIZE;
 var init_kv = __esm({
-  "sdk/kv.ts"() {
+  "dist/sdk/kv.js"() {
     "use strict";
     MAX_VALUE_SIZE = 65536;
   }
 });
 
-// sdk/s2s.ts
+// dist/sdk/s2s.js
 function uint8ToBase64(bytes) {
   const CHUNK = 32768;
   const parts = [];
@@ -640,7 +627,8 @@ function connectS2s(opts) {
     const target = new EventTarget();
     let opened = false;
     function send(msg) {
-      if (ws.readyState !== WS_OPEN) return;
+      if (ws.readyState !== WS_OPEN)
+        return;
       const type = msg.type;
       if (type !== "input.audio") {
         log.info(`S2S >> ${JSON.stringify(msg)}`);
@@ -683,11 +671,9 @@ function connectS2s(opts) {
       }
       switch (type) {
         case "session.ready":
-          target.dispatchEvent(
-            new CustomEvent("ready", {
-              detail: { session_id: msg.session_id }
-            })
-          );
+          target.dispatchEvent(new CustomEvent("ready", {
+            detail: { session_id: msg.session_id }
+          }));
           break;
         case "session.updated":
           target.dispatchEvent(new CustomEvent("session_updated", { detail: msg }));
@@ -699,28 +685,22 @@ function connectS2s(opts) {
           target.dispatchEvent(new CustomEvent("speech_stopped"));
           break;
         case "transcript.user.delta":
-          target.dispatchEvent(
-            new CustomEvent("user_transcript_delta", {
-              detail: { text: msg.text }
-            })
-          );
+          target.dispatchEvent(new CustomEvent("user_transcript_delta", {
+            detail: { text: msg.text }
+          }));
           break;
         case "transcript.user":
-          target.dispatchEvent(
-            new CustomEvent("user_transcript", {
-              detail: {
-                item_id: msg.item_id,
-                text: msg.text
-              }
-            })
-          );
+          target.dispatchEvent(new CustomEvent("user_transcript", {
+            detail: {
+              item_id: msg.item_id,
+              text: msg.text
+            }
+          }));
           break;
         case "reply.started":
-          target.dispatchEvent(
-            new CustomEvent("reply_started", {
-              detail: { reply_id: msg.reply_id }
-            })
-          );
+          target.dispatchEvent(new CustomEvent("reply_started", {
+            detail: { reply_id: msg.reply_id }
+          }));
           break;
         case "reply.audio": {
           const audioBytes = base64ToUint8(msg.data);
@@ -728,38 +708,30 @@ function connectS2s(opts) {
           break;
         }
         case "transcript.agent":
-          target.dispatchEvent(
-            new CustomEvent("agent_transcript", {
-              detail: { text: msg.text }
-            })
-          );
+          target.dispatchEvent(new CustomEvent("agent_transcript", {
+            detail: { text: msg.text }
+          }));
           break;
         case "tool.call":
-          target.dispatchEvent(
-            new CustomEvent("tool_call", {
-              detail: {
-                call_id: msg.call_id,
-                name: msg.name,
-                args: msg.args ?? {}
-              }
-            })
-          );
+          target.dispatchEvent(new CustomEvent("tool_call", {
+            detail: {
+              call_id: msg.call_id,
+              name: msg.name,
+              args: msg.args ?? {}
+            }
+          }));
           break;
         case "reply.done":
-          target.dispatchEvent(
-            new CustomEvent("reply_done", {
-              detail: { status: msg.status ?? void 0 }
-            })
-          );
+          target.dispatchEvent(new CustomEvent("reply_done", {
+            detail: { status: msg.status ?? void 0 }
+          }));
           break;
         case "session.error": {
           const code = msg.code;
           const isExpired = code === "session_not_found" || code === "session_forbidden";
-          target.dispatchEvent(
-            new CustomEvent(isExpired ? "session_expired" : "error", {
-              detail: { code, message: msg.message }
-            })
-          );
+          target.dispatchEvent(new CustomEvent(isExpired ? "session_expired" : "error", {
+            detail: { code, message: msg.message }
+          }));
           break;
         }
       }
@@ -777,25 +749,42 @@ function connectS2s(opts) {
       if (!opened) {
         reject(errObj);
       } else {
-        target.dispatchEvent(
-          new CustomEvent("error", {
-            detail: { code: "ws_error", message: errObj.message }
-          })
-        );
+        target.dispatchEvent(new CustomEvent("error", {
+          detail: { code: "ws_error", message: errObj.message }
+        }));
       }
     });
   });
 }
 var WS_OPEN;
 var init_s2s = __esm({
-  "sdk/s2s.ts"() {
+  "dist/sdk/s2s.js"() {
     "use strict";
     init_runtime();
     WS_OPEN = 1;
   }
 });
 
-// sdk/system_prompt.ts
+// dist/sdk/types.js
+var DEFAULT_INSTRUCTIONS2;
+var init_types = __esm({
+  "dist/sdk/types.js"() {
+    "use strict";
+    DEFAULT_INSTRUCTIONS2 = `You are AAI, a helpful AI assistant.
+
+Voice-First Rules:
+- Optimize for natural speech. Avoid jargon unless central to the answer. Use short, punchy sentences.
+- Never mention "search results," "sources," or "the provided text." Speak as if the knowledge is your own.
+- No visual formatting. Do not say "bullet point," "bold," or "bracketed one." If you need to list items, say "First," "Next," and "Finally."
+- Start with the most important information. No introductory filler.
+- Be concise. Keep answers to 1-3 sentences. For complex topics, provide a high-level summary.
+- Be confident. Avoid hedging phrases like "It seems that" or "I believe."
+- If you don't have enough information, say so directly rather than guessing.
+- Never use exclamation points. Keep your tone calm and conversational.`;
+  }
+});
+
+// dist/sdk/system_prompt.js
 function buildSystemPrompt(config, opts) {
   const { hasTools } = opts;
   const agentInstructions = config.instructions ? `
@@ -809,34 +798,22 @@ ${config.instructions}` : "";
     month: "long",
     day: "numeric"
   });
-  return DEFAULT_INSTRUCTIONS + `
+  return DEFAULT_INSTRUCTIONS2 + `
 
 Today's date is ${today}.` + agentInstructions + toolPreamble + (opts?.voice ? VOICE_RULES : "");
 }
 var VOICE_RULES;
 var init_system_prompt = __esm({
-  "sdk/system_prompt.ts"() {
+  "dist/sdk/system_prompt.js"() {
     "use strict";
     init_types();
     VOICE_RULES = '\n\nCRITICAL OUTPUT RULES \u2014 you MUST follow these for EVERY response:\nYour response will be spoken aloud by a TTS system and displayed as plain text.\n- NEVER use markdown: no **, no *, no _, no #, no `, no [](), no ---\n- NEVER use bullet points (-, *, \u2022) or numbered lists (1., 2.)\n- NEVER use code blocks or inline code\n- NEVER mention tools, search, APIs, or technical failures to the user. If a tool returns no results, just answer naturally without explaining why.\n- Write exactly as you would say it out loud to a friend\n- Use short conversational sentences. To list things, say "First," "Next," "Finally,"\n- Keep responses concise \u2014 1 to 3 sentences max';
   }
 });
 
-// sdk/session.ts
+// dist/sdk/session.js
 function createS2sSession(opts) {
-  const {
-    id,
-    agent,
-    client,
-    toolSchemas,
-    apiKey,
-    s2sConfig,
-    executeTool,
-    createWebSocket,
-    hookInvoker,
-    logger: log = consoleLogger,
-    metrics = noopMetrics
-  } = opts;
+  const { id, agent, client, toolSchemas, apiKey, s2sConfig, executeTool, createWebSocket, hookInvoker, logger: log = consoleLogger, metrics = noopMetrics } = opts;
   const agentLabel = { agent };
   const agentConfig = opts.skipGreeting ? { ...opts.agentConfig, greeting: "" } : opts.agentConfig;
   const hasTools = toolSchemas.length > 0 || (agentConfig.builtinTools?.length ?? 0) > 0;
@@ -859,7 +836,8 @@ function createS2sSession(opts) {
   let s2sSessionId = null;
   let pendingTools = [];
   async function resolveTurnConfig() {
-    if (!hookInvoker) return null;
+    if (!hookInvoker)
+      return null;
     try {
       return await hookInvoker.resolveTurnConfig(id, HOOK_TIMEOUT_MS);
     } catch {
@@ -867,7 +845,8 @@ function createS2sSession(opts) {
     }
   }
   async function invokeHook(hook, ...args) {
-    if (!hookInvoker) return;
+    if (!hookInvoker)
+      return;
     try {
       switch (hook) {
         case "onConnect":
@@ -966,7 +945,8 @@ function createS2sSession(opts) {
       }
       handle.updateSession({
         system_prompt: systemPrompt,
-        tools: s2sTools
+        tools: s2sTools,
+        ...agentConfig.greeting ? { greeting: agentConfig.greeting } : {}
       });
       handle.addEventListener("ready", ((e) => {
         s2sSessionId = e.detail.session_id;
@@ -1074,10 +1054,12 @@ function createS2sSession(opts) {
       await connectAndSetup();
     },
     async stop() {
-      if (sessionAbort.signal.aborted) return;
+      if (sessionAbort.signal.aborted)
+        return;
       sessionAbort.abort();
       metrics.sessionsActive.dec(agentLabel);
-      if (turnPromise) await turnPromise;
+      if (turnPromise)
+        await turnPromise;
       s2s?.close();
       invokeHook("onDisconnect");
     },
@@ -1085,15 +1067,9 @@ function createS2sSession(opts) {
       s2s?.sendAudio(data);
     },
     onAudioReady() {
-      if (audioReady) return;
+      if (audioReady)
+        return;
       audioReady = true;
-      if (agentConfig.greeting && s2s) {
-        s2s.updateSession({
-          system_prompt: systemPrompt,
-          tools: s2sTools,
-          greeting: agentConfig.greeting
-        });
-      }
     },
     onCancel() {
       client.event({ type: "cancelled" });
@@ -1118,7 +1094,7 @@ function createS2sSession(opts) {
 }
 var _internals3;
 var init_session = __esm({
-  "sdk/session.ts"() {
+  "dist/sdk/session.js"() {
     "use strict";
     init_protocol();
     init_runtime();
@@ -1130,7 +1106,7 @@ var init_session = __esm({
   }
 });
 
-// sdk/worker_entry.ts
+// dist/sdk/worker_entry.js
 import { z as z4 } from "zod";
 async function executeToolCall(name, args, options) {
   const { tool, env, sessionId, state, kv, messages } = options;
@@ -1149,13 +1125,15 @@ async function executeToolCall(name, args, options) {
       abortSignal,
       state: state ?? {},
       get kv() {
-        if (!kv) throw new Error("KV not available");
+        if (!kv)
+          throw new Error("KV not available");
         return kv;
       },
       messages: messages ?? []
     };
     const result = await Promise.resolve(tool.execute(parsed.data, ctx));
-    if (result == null) return "null";
+    if (result == null)
+      return "null";
     return typeof result === "string" ? result : JSON.stringify(result);
   } catch (err) {
     if (err instanceof DOMException && err.name === "TimeoutError") {
@@ -1168,13 +1146,13 @@ async function executeToolCall(name, args, options) {
 }
 var TOOL_HANDLER_TIMEOUT;
 var init_worker_entry = __esm({
-  "sdk/worker_entry.ts"() {
+  "dist/sdk/worker_entry.js"() {
     "use strict";
     TOOL_HANDLER_TIMEOUT = 3e4;
   }
 });
 
-// sdk/direct_executor.ts
+// dist/sdk/direct_executor.js
 function buildAgentConfig(agent) {
   const config = {
     name: agent.name,
@@ -1182,29 +1160,22 @@ function buildAgentConfig(agent) {
     greeting: agent.greeting,
     voice: agent.voice
   };
-  if (agent.sttPrompt !== void 0) config.sttPrompt = agent.sttPrompt;
-  if (typeof agent.maxSteps !== "function") config.maxSteps = agent.maxSteps;
-  if (agent.toolChoice !== void 0) config.toolChoice = agent.toolChoice;
-  if (agent.builtinTools) config.builtinTools = [...agent.builtinTools];
-  if (agent.activeTools) config.activeTools = [...agent.activeTools];
+  if (agent.sttPrompt !== void 0)
+    config.sttPrompt = agent.sttPrompt;
+  if (typeof agent.maxSteps !== "function")
+    config.maxSteps = agent.maxSteps;
+  if (agent.toolChoice !== void 0)
+    config.toolChoice = agent.toolChoice;
+  if (agent.builtinTools)
+    config.builtinTools = [...agent.builtinTools];
+  if (agent.activeTools)
+    config.activeTools = [...agent.activeTools];
   return config;
 }
 function createDirectExecutor(opts) {
-  const {
-    agent,
-    env,
-    kv = createMemoryKv(),
-    vectorSearch,
-    createWebSocket,
-    logger = consoleLogger,
-    metrics = noopMetrics,
-    s2sConfig = DEFAULT_S2S_CONFIG
-  } = opts;
+  const { agent, env, kv = createMemoryKv(), vectorSearch, createWebSocket, logger = consoleLogger, metrics = noopMetrics, s2sConfig = DEFAULT_S2S_CONFIG } = opts;
   const agentConfig = buildAgentConfig(agent);
-  const builtinDefs = getBuiltinToolDefs(
-    agent.builtinTools ?? [],
-    vectorSearch ? { vectorSearch } : void 0
-  );
+  const builtinDefs = getBuiltinToolDefs(agent.builtinTools ?? [], vectorSearch ? { vectorSearch } : void 0);
   const allTools = {
     ...builtinDefs,
     ...agent.tools
@@ -1232,7 +1203,8 @@ function createDirectExecutor(opts) {
   }
   const executeTool = async (name, args, sessionId, messages) => {
     const tool = allTools[name];
-    if (!tool) return JSON.stringify({ error: `Unknown tool: ${name}` });
+    if (!tool)
+      return JSON.stringify({ error: `Unknown tool: ${name}` });
     return executeToolCall(name, args, {
       tool,
       env: frozenEnv,
@@ -1270,10 +1242,13 @@ function createDirectExecutor(opts) {
         const result = await agent.onBeforeStep(0, ctx);
         activeTools = result?.activeTools;
       }
-      if (maxSteps === void 0 && activeTools === void 0) return null;
+      if (maxSteps === void 0 && activeTools === void 0)
+        return null;
       const config = {};
-      if (maxSteps !== void 0) config.maxSteps = maxSteps;
-      if (activeTools !== void 0) config.activeTools = activeTools;
+      if (maxSteps !== void 0)
+        config.maxSteps = maxSteps;
+      if (activeTools !== void 0)
+        config.activeTools = activeTools;
       return config;
     }
   };
@@ -1301,7 +1276,7 @@ function createDirectExecutor(opts) {
   return { executeTool, hookInvoker, toolSchemas, createSession };
 }
 var init_direct_executor = __esm({
-  "sdk/direct_executor.ts"() {
+  "dist/sdk/direct_executor.js"() {
     "use strict";
     init_internal_types();
     init_builtin_tools();
@@ -1312,7 +1287,7 @@ var init_direct_executor = __esm({
   }
 });
 
-// sdk/ws_handler.ts
+// dist/sdk/ws_handler.js
 function isValidAudioChunk(data) {
   return data.byteLength > 0 && data.byteLength <= MAX_AUDIO_CHUNK_BYTES && data.byteLength % 2 === 0;
 }
@@ -1344,7 +1319,7 @@ function wireSessionSocket(ws, opts) {
   const sid = sessionId.slice(0, 8);
   const ctx = opts.logContext ?? {};
   let session = null;
-  ws.addEventListener("open", () => {
+  function onOpen() {
     opts.onOpen?.();
     log.info("Session connected", { ...ctx, sid });
     const client = createClientSink(ws);
@@ -1353,13 +1328,20 @@ function wireSessionSocket(ws, opts) {
     ws.send(JSON.stringify({ type: "config", ...opts.readyConfig }));
     void session.start();
     log.info("Session ready", { ...ctx, sid });
-  });
+  }
+  if (ws.readyState === 1) {
+    onOpen();
+  } else {
+    ws.addEventListener("open", onOpen);
+  }
   ws.addEventListener("message", (event) => {
-    if (!session) return;
+    if (!session)
+      return;
     const msgEvent = event;
     const { data } = msgEvent;
-    if (data instanceof ArrayBuffer) {
-      const chunk = new Uint8Array(data);
+    const isBinary = globalThis.Buffer?.isBuffer(data) || data instanceof ArrayBuffer || data instanceof Uint8Array;
+    if (isBinary) {
+      const chunk = data instanceof Uint8Array ? data : new Uint8Array(data.buffer ?? data, data.byteOffset ?? 0, data.byteLength ?? data.byteLength);
       if (!isValidAudioChunk(chunk)) {
         log.warn("Invalid audio chunk, dropping", {
           ...ctx,
@@ -1372,7 +1354,8 @@ function wireSessionSocket(ws, opts) {
       session.onAudio(chunk);
       return;
     }
-    if (typeof data !== "string") return;
+    if (typeof data !== "string")
+      return;
     let json;
     try {
       json = JSON.parse(data);
@@ -1421,7 +1404,7 @@ function wireSessionSocket(ws, opts) {
 }
 var MAX_AUDIO_CHUNK_BYTES;
 var init_ws_handler = __esm({
-  "sdk/ws_handler.ts"() {
+  "dist/sdk/ws_handler.js"() {
     "use strict";
     init_protocol();
     init_runtime();
@@ -1429,18 +1412,9 @@ var init_ws_handler = __esm({
   }
 });
 
-// sdk/winterc_server.ts
+// dist/sdk/winterc_server.js
 function createWintercServer(options) {
-  const {
-    agent,
-    env,
-    kv = createMemoryKv(),
-    vectorSearch,
-    clientHtml,
-    logger = consoleLogger,
-    metrics = noopMetrics,
-    s2sConfig = DEFAULT_S2S_CONFIG
-  } = options;
+  const { agent, env, kv = createMemoryKv(), vectorSearch, clientHtml, logger = consoleLogger, metrics = noopMetrics, s2sConfig = DEFAULT_S2S_CONFIG } = options;
   const executor = createDirectExecutor({
     agent,
     env,
@@ -1473,10 +1447,7 @@ function createWintercServer(options) {
         });
       }
       if (url.pathname === "/") {
-        return new Response(
-          `<!DOCTYPE html><html><body><h1>${agent.name}</h1><p>Agent server running.</p></body></html>`,
-          { headers: { "Content-Type": "text/html" } }
-        );
+        return new Response(`<!DOCTYPE html><html><body><h1>${agent.name}</h1><p>Agent server running.</p></body></html>`, { headers: { "Content-Type": "text/html" } });
       }
       return new Response("Not Found", { status: 404 });
     },
@@ -1502,7 +1473,7 @@ function createWintercServer(options) {
   };
 }
 var init_winterc_server = __esm({
-  "sdk/winterc_server.ts"() {
+  "dist/sdk/winterc_server.js"() {
     "use strict";
     init_direct_executor();
     init_kv();
@@ -1512,7 +1483,7 @@ var init_winterc_server = __esm({
   }
 });
 
-// sdk/server.ts
+// dist/sdk/server.js
 var server_exports = {};
 __export(server_exports, {
   createServer: () => createServer
@@ -1523,23 +1494,20 @@ async function loadWsFactory() {
     const WS = mod.default ?? mod;
     return (url, opts) => new WS(url, { headers: opts.headers });
   } catch {
-    throw new Error(
-      "WebSocket factory not provided and `ws` package not found. Install `ws` (`npm install ws`) or pass `createWebSocket` option."
-    );
+    throw new Error("WebSocket factory not provided and `ws` package not found. Install `ws` (`npm install ws`) or pass `createWebSocket` option.");
   }
 }
 function resolveEnv(env) {
   const resolved = {};
   for (const [key, value] of Object.entries(env)) {
-    if (value !== void 0) resolved[key] = value;
+    if (value !== void 0)
+      resolved[key] = value;
   }
   return resolved;
 }
 function createServer(options) {
   const { agent, kv, clientHtml, logger = consoleLogger, s2sConfig = DEFAULT_S2S_CONFIG } = options;
-  const env = resolveEnv(
-    options.env ?? (typeof process !== "undefined" ? process.env : {})
-  );
+  const env = resolveEnv(options.env ?? (typeof process !== "undefined" ? process.env : {}));
   let wsFactory = options.createWebSocket ?? null;
   async function getWsFactory() {
     if (!wsFactory) {
@@ -1555,7 +1523,7 @@ function createServer(options) {
         env,
         ...kv ? { kv } : {},
         createWebSocket: wsFactory,
-        clientHtml,
+        ...clientHtml !== void 0 ? { clientHtml } : {},
         logger,
         s2sConfig
       });
@@ -1577,7 +1545,8 @@ function createServer(options) {
           const url = new URL(req.url ?? "/", `${protocol}://${host}`);
           const headers = new Headers();
           for (const [key, val] of Object.entries(req.headers)) {
-            if (val) headers.set(key, Array.isArray(val) ? val[0] : val);
+            if (val)
+              headers.set(key, Array.isArray(val) ? val[0] : val);
           }
           const request = new Request(url, {
             method: req.method ?? "GET",
@@ -1594,23 +1563,17 @@ function createServer(options) {
       });
       try {
         const wsMod = await import("ws");
-        const WSServer = wsMod.WebSocketServer ?? wsMod.default?.Server;
+        const WSServer = wsMod.WebSocketServer;
         if (WSServer) {
           const wss = new WSServer({ noServer: true });
           nodeServer.on("upgrade", (req, socket, head) => {
-            wss.handleUpgrade(
-              req,
-              socket,
-              head,
-              (ws) => {
-                const reqUrl = new URL(
-                  req.url ?? "/",
-                  `http://localhost:${port}`
-                );
-                const resume = reqUrl.searchParams.has("resume");
-                getWinterc().handleWebSocket(ws, { skipGreeting: resume });
-              }
-            );
+            wss.handleUpgrade(req, socket, head, (ws) => {
+              const reqUrl = new URL(req.url ?? "/", `http://localhost:${port}`);
+              const resume = reqUrl.searchParams.has("resume");
+              getWinterc().handleWebSocket(ws, {
+                skipGreeting: resume
+              });
+            });
           });
         }
       } catch {
@@ -1637,7 +1600,7 @@ function createServer(options) {
   };
 }
 var init_server = __esm({
-  "sdk/server.ts"() {
+  "dist/sdk/server.js"() {
     init_runtime();
     init_winterc_server();
   }
@@ -1794,6 +1757,7 @@ async function askSelect(message, choices) {
           Select,
           {
             options: choices,
+            visibleOptionCount: choices.length,
             onChange: (value) => {
               resolve(value);
               app.unmount();
@@ -1807,7 +1771,8 @@ async function askSelect(message, choices) {
 
 // cli/_discover.ts
 function isDevMode() {
-  return path.basename(process.argv[0] ?? "") !== "aai";
+  const script = process.argv[1] ?? "";
+  return script.endsWith(".ts") || script.endsWith(".tsx");
 }
 function generateSlug() {
   return humanId({ separator: "-", capitalize: false });
@@ -2072,7 +2037,10 @@ function CommandRunner({
   const { items, log } = useStepLog();
   const [spinning, setSpinning] = useState(true);
   const [err, setErr] = useState(null);
+  const started = useRef(false);
   React.useEffect(() => {
+    if (started.current) return;
+    started.current = true;
     (async () => {
       try {
         await run(log);
@@ -2084,7 +2052,7 @@ function CommandRunner({
       setSpinning(false);
       exit();
     })();
-  }, [exit, log, onError, run]);
+  });
   return /* @__PURE__ */ jsxs2(Fragment, { children: [
     /* @__PURE__ */ jsx2(StepLog, { items }),
     err && /* @__PURE__ */ jsx2(ErrorLine, { msg: err }),
@@ -2122,12 +2090,27 @@ import { build } from "vite";
 import { viteSingleFile } from "vite-plugin-singlefile";
 
 // cli/_static_config.ts
-init_types();
 import fs2 from "node:fs/promises";
 import {
   Project,
   SyntaxKind
 } from "ts-morph";
+
+// sdk/types.ts
+var DEFAULT_INSTRUCTIONS = `You are AAI, a helpful AI assistant.
+
+Voice-First Rules:
+- Optimize for natural speech. Avoid jargon unless central to the answer. Use short, punchy sentences.
+- Never mention "search results," "sources," or "the provided text." Speak as if the knowledge is your own.
+- No visual formatting. Do not say "bullet point," "bold," or "bracketed one." If you need to list items, say "First," "Next," and "Finally."
+- Start with the most important information. No introductory filler.
+- Be concise. Keep answers to 1-3 sentences. For complex topics, provide a high-level summary.
+- Be confident. Avoid hedging phrases like "It seems that" or "I believe."
+- If you don't have enough information, say so directly rather than guessing.
+- Never use exclamation points. Keep your tone calm and conversational.`;
+var DEFAULT_GREETING = "Hey there. I'm a voice assistant. What can I help you with?";
+
+// cli/_static_config.ts
 var project = new Project({ useInMemoryFileSystem: true });
 async function extractStaticConfig(agentPath) {
   const source = await fs2.readFile(agentPath, "utf-8");
@@ -2485,6 +2468,7 @@ var DEFAULT_STYLES_CSS = `@import url("https://fonts.googleapis.com/css2?family=
 @import "tailwindcss";
 @source "./";
 @source "./components/";
+@source "./node_modules/@alexkroman1/aai/ui/";
 
 @theme {
   --color-aai-bg: #101010;
@@ -2562,7 +2546,7 @@ var INDEX_HTML = `<!DOCTYPE html>
       content="width=device-width, initial-scale=1.0, viewport-fit=cover"
     />
     <title>aai</title>
-    <link rel="icon" href="/favicon.svg" type="image/svg+xml" />
+    <link rel="icon" href="data:," />
     <link rel="stylesheet" href="../styles.css" />
   </head>
   <body>
@@ -2863,15 +2847,25 @@ init_output();
 
 // cli/_server_common.ts
 import path7 from "node:path";
+import { createServer as createViteServer } from "vite";
 init_output();
 async function loadAgentDef(cwd) {
   const agentPath = path7.resolve(cwd, "agent.ts");
-  const agentModule = await import(agentPath);
-  const agentDef = agentModule.default;
-  if (!agentDef || typeof agentDef !== "object" || !agentDef.name) {
-    throw new Error("agent.ts must export a default agent definition (from defineAgent())");
+  const vite = await createViteServer({
+    root: cwd,
+    logLevel: "silent",
+    server: { middlewareMode: true }
+  });
+  try {
+    const agentModule = await vite.ssrLoadModule(agentPath);
+    const agentDef = agentModule.default;
+    if (!agentDef || typeof agentDef !== "object" || !agentDef.name) {
+      throw new Error("agent.ts must export a default agent definition (from defineAgent())");
+    }
+    return agentDef;
+  } finally {
+    await vite.close();
   }
-  return agentDef;
 }
 async function resolveServerEnv() {
   const env = { ...process.env };
@@ -2885,13 +2879,21 @@ async function resolveServerEnv() {
   }
   return env;
 }
-async function bootServer(agentDef, html, env, port) {
+async function loadWsFromProject(cwd) {
+  const wsPath = path7.resolve(cwd, "node_modules", "ws", "index.js");
+  const mod = await import(wsPath);
+  const WS = mod.default ?? mod;
+  return (url, opts) => new WS(url, { headers: opts.headers });
+}
+async function bootServer(agentDef, html, env, port, cwd) {
   step("Start", `http://localhost:${port}`);
+  const createWebSocket = await loadWsFromProject(cwd);
   const { createServer: createServer2 } = await Promise.resolve().then(() => (init_server(), server_exports));
   const server = createServer2({
     agent: agentDef,
     clientHtml: html,
-    env
+    env,
+    createWebSocket
   });
   await server.listen(port);
   step("Ready", `http://localhost:${port}`);
@@ -2901,7 +2903,7 @@ async function bootServer(agentDef, html, env, port) {
 async function _startDevServer(cwd, port) {
   const agent = await loadAgent(cwd);
   if (!agent) {
-    throw new Error("No agent found \u2014 run `aai new` first");
+    throw new Error("No agent found \u2014 run `aai init` first");
   }
   step("Bundle", agent.slug);
   let html;
@@ -2918,7 +2920,7 @@ async function _startDevServer(cwd, port) {
   step("Load", "agent.ts");
   const agentDef = await loadAgentDef(cwd);
   const env = await resolveServerEnv();
-  await bootServer(agentDef, html, env, port);
+  await bootServer(agentDef, html, env, port, cwd);
 }
 
 // cli/dev.tsx
@@ -3332,7 +3334,7 @@ async function _startProductionServer(cwd, port) {
   step("Load", "agent");
   const agentDef = await loadAgentDef(cwd);
   const env = await resolveServerEnv();
-  await bootServer(agentDef, html, env, port);
+  await bootServer(agentDef, html, env, port, cwd);
 }
 
 // cli/start.tsx
@@ -3422,14 +3424,11 @@ async function main(args) {
       process.exit(1);
   }
 }
-var isMain = process.argv[1] === fileURLToPath3(import.meta.url);
-if (isMain) {
-  try {
-    await main(process.argv.slice(2));
-  } catch (err) {
-    error2(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
+try {
+  await main(process.argv.slice(2));
+} catch (err) {
+  error2(err instanceof Error ? err.message : String(err));
+  process.exit(1);
 }
 export {
   main