@alexkroman1/aai-ui 0.10.2 → 0.10.4

package/dist/_components/app.d.ts

@@ -2,16 +2,16 @@ import type * as preact from "preact";
 /**
  * The default top-level UI component for an AAI voice agent.
  * Renders a {@link StartScreen} (with the AAI logo or a custom title from
- * {@link useMountConfig}) followed by a {@link ChatView} once the session starts.
+ * {@link useClientConfig}) followed by a {@link ChatView} once the session starts.
  *
- * This is the component rendered by {@link mount} when no custom component is
+ * This is the component rendered by {@link defineClient} when no custom component is
  * provided.
  *
  * @example
  * ```tsx
- * import { App, mount } from "@aai/ui";
+ * import { App, defineClient } from "@aai/ui";
  *
- * mount(App, { target: "#app", title: "My Agent" });
+ * defineClient(App, { target: "#app", title: "My Agent" });
  * ```
  *
  * @param className - Additional CSS class names applied to the root element.

package/dist/_components/app.js

@@ -1,4 +1,4 @@
-import { useMountConfig } from "../mount-context.js";
+import { useClientConfig } from "../client-context.js";
 import { ChatView } from "./chat-view.js";
 import { StartScreen } from "./start-screen.js";
 import { jsx } from "preact/jsx-runtime";
@@ -12,16 +12,16 @@ function AnsiLogo() {
 /**
 * The default top-level UI component for an AAI voice agent.
 * Renders a {@link StartScreen} (with the AAI logo or a custom title from
-* {@link useMountConfig}) followed by a {@link ChatView} once the session starts.
+* {@link useClientConfig}) followed by a {@link ChatView} once the session starts.
 *
-* This is the component rendered by {@link mount} when no custom component is
+* This is the component rendered by {@link defineClient} when no custom component is
 * provided.
 *
 * @example
 * ```tsx
-* import { App, mount } from "@aai/ui";
+* import { App, defineClient } from "@aai/ui";
 *
-* mount(App, { target: "#app", title: "My Agent" });
+* defineClient(App, { target: "#app", title: "My Agent" });
 * ```
 *
 * @param className - Additional CSS class names applied to the root element.
@@ -29,7 +29,7 @@ function AnsiLogo() {
 * @public
 */
 function App({ className }) {
-	const { title } = useMountConfig();
+	const { title } = useClientConfig();
 	return /* @__PURE__ */ jsx(StartScreen, {
 		icon: title ? void 0 : /* @__PURE__ */ jsx(AnsiLogo, {}),
 		title,

package/dist/_components/chat-view.js

@@ -1,4 +1,4 @@
-import { useMountConfig } from "../mount-context.js";
+import { useClientConfig } from "../client-context.js";
 import { useSession } from "../signals.js";
 import { Controls } from "./controls.js";
 import { ErrorBanner } from "./error-banner.js";
@@ -34,7 +34,7 @@ import { jsx, jsxs } from "preact/jsx-runtime";
 */
 function ChatView({ className }) {
 	const { session } = useSession();
-	const { title } = useMountConfig();
+	const { title } = useClientConfig();
 	return /* @__PURE__ */ jsxs("div", {
 		class: clsx("flex flex-col h-screen max-w-130 mx-auto bg-aai-bg text-aai-text font-aai text-sm", className),
 		children: [

package/dist/{mount-context.d.ts → client-context.d.ts}

@@ -3,7 +3,7 @@
  *
  * @public
  */
-export type MountTheme = {
+export type ClientTheme = {
     /** Background color. Default: `#101010`. */
     bg?: string;
     /** Primary accent color. Default: `#fab283`. */
@@ -16,18 +16,18 @@ export type MountTheme = {
     border?: string;
 };
 /**
- * Resolved mount-level configuration available to default UI components.
+ * Resolved client-level configuration available to default UI components.
  *
  * @public
  */
-export type MountConfig = {
+export type ClientConfig = {
     title?: string | undefined;
-    theme?: MountTheme | undefined;
+    theme?: ClientTheme | undefined;
 };
-export declare const MountConfigProvider: import("preact").Provider<MountConfig>;
+export declare const ClientConfigProvider: import("preact").Provider<ClientConfig>;
 /**
- * Read mount config (title, theme) from the nearest provider.
+ * Read client config (title, theme) from the nearest provider.
  *
  * @public
  */
-export declare function useMountConfig(): MountConfig;
+export declare function useClientConfig(): ClientConfig;

package/dist/client-context.js

@@ -0,0 +1,15 @@
+import { createContext } from "preact";
+import { useContext } from "preact/hooks";
+//#region client-context.ts
+const Ctx = createContext({});
+const ClientConfigProvider = Ctx.Provider;
+/**
+* Read client config (title, theme) from the nearest provider.
+*
+* @public
+*/
+function useClientConfig() {
+	return useContext(Ctx);
+}
+//#endregion
+export { ClientConfigProvider, useClientConfig };

package/dist/{mount.d.ts → define-client.d.ts}

@@ -1,13 +1,13 @@
 import type { ComponentType } from "preact";
-import { type MountTheme } from "./mount-context.ts";
-import { type VoiceSession } from "./session.ts";
+import { type ClientTheme } from "./client-context.ts";
+import { type VoiceSession, type WebSocketConstructor } from "./session.ts";
 import { type SessionSignals } from "./signals.ts";
 /**
- * Options for {@link mount}.
+ * Options for {@link defineClient}.
  *
  * @public
  */
-export type MountOptions = {
+export type ClientOptions = {
     /** CSS selector or DOM element to render into. Defaults to `"#app"`. */
     target?: string | HTMLElement;
     /** Base URL of the AAI platform server. Derived from `location.href` by default. */
@@ -15,20 +15,22 @@ export type MountOptions = {
     /** Agent title shown in the header and start screen. */
     title?: string;
     /** Theme color overrides. */
-    theme?: MountTheme;
+    theme?: ClientTheme;
     /** Called when the server sends a session ID. Store it for reconnection. */
     onSessionId?: ((sessionId: string) => void) | undefined;
     /** Session ID from a previous connection for resuming persisted state. */
     resumeSessionId?: string | undefined;
+    /** WebSocket constructor override. Passed through to VoiceSessionOptions. */
+    WebSocket?: WebSocketConstructor | undefined;
 };
 /**
- * Handle returned by {@link mount} for cleanup.
+ * Handle returned by {@link defineClient} for cleanup.
  *
  * Implements `Disposable` so it can be used with `using`.
  *
  * @public
  */
-export type MountHandle = {
+export type ClientHandle = {
     /** The underlying voice session. */
     session: VoiceSession;
     /** Reactive session controls for the mounted UI. */
@@ -39,17 +41,17 @@ export type MountHandle = {
     [Symbol.dispose](): void;
 };
 /**
- * Mount a Preact component with voice session wiring.
+ * Define and mount a client UI for a voice agent.
  *
  * Creates a {@link VoiceSession}, wraps it in
  * {@link SessionSignals}, and renders the component
  * inside a {@link SessionProvider}.
  *
  * @param Component - The Preact component to render.
- * @param options - Mount options (target element, platform URL).
- * @returns A {@link MountHandle} for cleanup.
+ * @param options - Client options (target element, platform URL, theme).
+ * @returns A {@link ClientHandle} for cleanup.
  * @throws If the target element is not found in the DOM.
  *
  * @public
  */
-export declare function mount(Component: ComponentType<any>, options?: MountOptions): MountHandle;
+export declare function defineClient(Component: ComponentType<any>, options?: ClientOptions): ClientHandle;

package/dist/{mount.js → define-client.js}

@@ -1,10 +1,10 @@
-import { MountConfigProvider } from "./mount-context.js";
+import { ClientConfigProvider } from "./client-context.js";
 import { SessionProvider, createSessionControls } from "./signals.js";
-import { t as createVoiceSession } from "./session-D0RdWlWH.js";
+import { t as createVoiceSession } from "./session-CWB7vmqz.js";
 import { render } from "preact";
 import { batch, signal } from "@preact/signals";
 import { jsx } from "preact/jsx-runtime";
-//#region mount.tsx
+//#region define-client.tsx
 function resolveContainer(target = "#app") {
 	if (typeof target !== "string") return target;
 	const el = document.querySelector(target);
@@ -12,30 +12,31 @@ function resolveContainer(target = "#app") {
 	return el;
 }
 /**
-* Mount a Preact component with voice session wiring.
+* Define and mount a client UI for a voice agent.
 *
 * Creates a {@link VoiceSession}, wraps it in
 * {@link SessionSignals}, and renders the component
 * inside a {@link SessionProvider}.
 *
 * @param Component - The Preact component to render.
-* @param options - Mount options (target element, platform URL).
-* @returns A {@link MountHandle} for cleanup.
+* @param options - Client options (target element, platform URL, theme).
+* @returns A {@link ClientHandle} for cleanup.
 * @throws If the target element is not found in the DOM.
 *
 * @public
 */
-function mount(Component, options) {
+function defineClient(Component, options) {
 	const container = resolveContainer(options?.target);
 	const session = createVoiceSession({
 		platformUrl: options?.platformUrl ?? globalThis.location.origin + globalThis.location.pathname,
 		reactiveFactory: signal,
 		batch,
 		onSessionId: options?.onSessionId,
-		resumeSessionId: options?.resumeSessionId
+		resumeSessionId: options?.resumeSessionId,
+		...options?.WebSocket ? { WebSocket: options.WebSocket } : {}
 	});
 	const signals = createSessionControls(session);
-	const mountConfig = {
+	const clientConfig = {
 		title: options?.title,
 		theme: options?.theme
 	};
@@ -48,8 +49,8 @@ function mount(Component, options) {
 		if (t.surface) el.style.setProperty("--color-aai-surface", t.surface);
 		if (t.border) el.style.setProperty("--color-aai-border", t.border);
 	}
-	render(/* @__PURE__ */ jsx(MountConfigProvider, {
-		value: mountConfig,
+	render(/* @__PURE__ */ jsx(ClientConfigProvider, {
+		value: clientConfig,
 		children: /* @__PURE__ */ jsx(SessionProvider, {
 			value: signals,
 			children: /* @__PURE__ */ jsx(Component, {})
@@ -70,4 +71,4 @@ function mount(Component, options) {
 	return handle;
 }
 //#endregion
-export { mount };
+export { defineClient };

package/dist/index.d.ts

@@ -7,9 +7,9 @@
  *
  * @example
  * ```tsx
- * import { App, mount } from "@aai/ui";
+ * import { App, defineClient } from "@aai/ui";
  *
- * mount(App, { target: "#app" });
+ * defineClient(App, { target: "#app" });
  * ```
  */
 export { App } from "./_components/app.tsx";
@@ -26,12 +26,12 @@ export { StateIndicator } from "./_components/state-indicator.tsx";
 export { ThinkingIndicator } from "./_components/thinking-indicator.tsx";
 export { ToolCallBlock } from "./_components/tool-call-block.tsx";
 export { Transcript } from "./_components/transcript.tsx";
-export type { MountHandle, MountOptions } from "./mount.tsx";
-export { mount } from "./mount.tsx";
-export type { MountConfig, MountTheme } from "./mount-context.ts";
-export { useMountConfig } from "./mount-context.ts";
+export type { ClientConfig, ClientTheme } from "./client-context.ts";
+export { useClientConfig } from "./client-context.ts";
+export type { ClientHandle, ClientOptions } from "./define-client.tsx";
+export { defineClient } from "./define-client.tsx";
 export type { VoiceSession } from "./session.ts";
 export { createVoiceSession } from "./session.ts";
 export type { SessionSignals } from "./signals.ts";
-export { createSessionControls, SessionProvider, useAutoScroll, useSession, useToolCallStart, useToolCallUpdate, useToolResult, } from "./signals.ts";
+export { createSessionControls, SessionProvider, useAutoScroll, useSession, useToolCallStart, useToolResult, } from "./signals.ts";
 export type { AgentState, ChatMessage, Reactive, SessionError, SessionErrorCode, ToolCallInfo, VoiceSessionOptions, } from "./types.ts";

package/dist/index.js

@@ -1,5 +1,5 @@
-import { useMountConfig } from "./mount-context.js";
-import { SessionProvider, createSessionControls, useAutoScroll, useSession, useToolCallStart, useToolCallUpdate, useToolResult } from "./signals.js";
+import { useClientConfig } from "./client-context.js";
+import { SessionProvider, createSessionControls, useAutoScroll, useSession, useToolCallStart, useToolResult } from "./signals.js";
 import { Button } from "./_components/button.js";
 import { Controls } from "./_components/controls.js";
 import { ErrorBanner } from "./_components/error-banner.js";
@@ -13,6 +13,6 @@ import { ChatView } from "./_components/chat-view.js";
 import { StartScreen } from "./_components/start-screen.js";
 import { App } from "./_components/app.js";
 import { SidebarLayout } from "./_components/sidebar-layout.js";
-import { t as createVoiceSession } from "./session-D0RdWlWH.js";
-import { mount } from "./mount.js";
-export { App, Button, ChatView, Controls, ErrorBanner, MessageBubble, MessageList, SessionProvider, SidebarLayout, StartScreen, StateIndicator, ThinkingIndicator, ToolCallBlock, Transcript, createSessionControls, createVoiceSession, mount, useAutoScroll, useMountConfig, useSession, useToolCallStart, useToolCallUpdate, useToolResult };
+import { t as createVoiceSession } from "./session-CWB7vmqz.js";
+import { defineClient } from "./define-client.js";
+export { App, Button, ChatView, Controls, ErrorBanner, MessageBubble, MessageList, SessionProvider, SidebarLayout, StartScreen, StateIndicator, ThinkingIndicator, ToolCallBlock, Transcript, createSessionControls, createVoiceSession, defineClient, useAutoScroll, useClientConfig, useSession, useToolCallStart, useToolResult };

package/dist/{session-D0RdWlWH.js → session-CWB7vmqz.js}

@@ -72,24 +72,9 @@ var ClientHandler = class {
 					toolName: e.toolName,
 					args: e.args,
 					status: "pending",
-					updates: [],
 					afterMessageIndex: this.#messages.value.length - 1
 				}];
 				break;
-			case "tool_call_update": {
-				const tcs = this.#toolCalls.value;
-				const idx = tcs.findIndex((tc) => tc.toolCallId === e.toolCallId);
-				if (idx !== -1) {
-					const updated = [...tcs];
-					const existing = updated[idx];
-					if (existing) updated[idx] = {
-						...existing,
-						updates: [...existing.updates, e.data]
-					};
-					this.#toolCalls.value = updated;
-				}
-				break;
-			}
 			case "tool_call_done": {
 				const tcs = this.#toolCalls.value;
 				const idx = tcs.findIndex((tc) => tc.toolCallId === e.toolCallId);
@@ -208,14 +193,7 @@ var ClientHandler = class {
 };
 //#endregion
 //#region session.ts
-/** Built-in non-reactive container (plain mutable wrapper). */
-function plainReactive(initial) {
-	return { value: initial };
-}
-/** No-op batch — just calls the function. */
-function plainBatch(fn) {
-	fn();
-}
+const WS_OPEN = 1;
 /**
 * Initialize audio capture and playback after the server sends a ready config.
 *
@@ -261,7 +239,7 @@ async function initAudioCapture(conn, msg, deps) {
 				}
 			}
 		});
-		if (conn.generation !== gen || !conn.ws || conn.ws.readyState !== WebSocket.OPEN) {
+		if (conn.generation !== gen || !conn.ws || conn.ws.readyState !== WS_OPEN) {
 			io.close();
 			return;
 		}
@@ -269,7 +247,7 @@ async function initAudioCapture(conn, msg, deps) {
 		deps.send({ type: "audio_ready" });
 		deps.state.value = "listening";
 	} catch (err) {
-		if (conn.generation !== gen || !conn.ws || conn.ws.readyState !== WebSocket.OPEN) return;
+		if (conn.generation !== gen || !conn.ws || conn.ws.readyState !== WS_OPEN) return;
 		deps.batch(() => {
 			deps.error.value = {
 				code: "audio",
@@ -299,8 +277,9 @@ function buildWsUrl(platformUrl, resume, sessionId) {
 * @public
 */
 function createVoiceSession(options) {
-	const reactive = options.reactiveFactory ?? plainReactive;
-	const batchFn = options.batch ?? plainBatch;
+	const WS = options.WebSocket ?? WebSocket;
+	const reactive = options.reactiveFactory ?? ((initial) => ({ value: initial }));
+	const batchFn = options.batch ?? ((fn) => fn());
 	const state = reactive("disconnected");
 	const messages = reactive([]);
 	const toolCalls = reactive([]);
@@ -331,10 +310,10 @@ function createVoiceSession(options) {
 		});
 	}
 	function send(msg) {
-		if (conn.ws && conn.ws.readyState === WebSocket.OPEN) conn.ws.send(JSON.stringify(msg));
+		if (conn.ws && conn.ws.readyState === WS_OPEN) conn.ws.send(JSON.stringify(msg));
 	}
 	function sendBinary(data) {
-		if (conn.ws && conn.ws.readyState === WebSocket.OPEN) conn.ws.send(data);
+		if (conn.ws && conn.ws.readyState === WS_OPEN) conn.ws.send(data);
 	}
 	const audioDeps = {
 		send,
@@ -356,8 +335,7 @@ function createVoiceSession(options) {
 		const { signal: sig } = controller;
 		if (opts?.signal) opts.signal.addEventListener("abort", () => disconnect(), { signal: sig });
 		const resumeId = !hasConnected ? options.resumeSessionId : void 0;
-		const wsUrl = buildWsUrl(options.platformUrl, hasConnected, resumeId);
-		const socket = new WebSocket(wsUrl.toString());
+		const socket = new WS(buildWsUrl(options.platformUrl, hasConnected, resumeId).toString());
 		socket.binaryType = "arraybuffer";
 		conn.ws = socket;
 		const handler = new ClientHandler({
@@ -412,7 +390,7 @@ function createVoiceSession(options) {
 	}
 	function reset() {
 		conn.voiceIO?.flush();
-		if (conn.ws && conn.ws.readyState === WebSocket.OPEN) {
+		if (conn.ws && conn.ws.readyState === WS_OPEN) {
 			send({ type: "reset" });
 			return;
 		}

package/dist/session.d.ts

@@ -1,6 +1,6 @@
 import type { AgentState, ChatMessage, Reactive, SessionError, ToolCallInfo, VoiceSessionOptions } from "./types.ts";
 export { ClientHandler } from "./client-handler.ts";
-export type { AgentState, ChatMessage, Reactive, SessionError, SessionErrorCode, ToolCallInfo, VoiceSessionOptions, } from "./types.ts";
+export type { AgentState, ChatMessage, Reactive, SessionError, SessionErrorCode, ToolCallInfo, VoiceSessionOptions, WebSocketConstructor, } from "./types.ts";
 /**
  * A reactive voice session that manages WebSocket communication,
  * audio capture/playback, and agent state transitions.

package/dist/session.js

@@ -1,2 +1,2 @@
-import { n as ClientHandler, t as createVoiceSession } from "./session-D0RdWlWH.js";
+import { n as ClientHandler, t as createVoiceSession } from "./session-CWB7vmqz.js";
 export { ClientHandler, createVoiceSession };

package/dist/signals.d.ts

@@ -110,23 +110,6 @@ export declare function useToolResult(callback: (toolName: string, result: unkno
  * @public
  */
 export declare function useToolCallStart(callback: (toolName: string, args: Record<string, unknown>, toolCall: ToolCallInfo) => void): void;
-/**
- * Hook that fires a callback for each intermediate update pushed by a tool
- * via `ctx.sendUpdate()`.
- *
- * Use this to render progressive/streaming data from a tool before it
- * finishes — e.g. showing a recipe preview card while nutrition data is
- * still loading.
- *
- * The callback fires once per update. The `data` argument is the parsed
- * JSON update (or raw string if parsing fails).
- *
- * @param callback - Called once per intermediate update with the tool name,
- *   parsed data, and full {@link ToolCallInfo}.
- *
- * @public
- */
-export declare function useToolCallUpdate(callback: (toolName: string, data: unknown, toolCall: ToolCallInfo) => void): void;
 /**
  * Auto-scroll a container to the bottom when messages, tool calls,
  * or utterances change. Returns a ref to attach to a sentinel `<div>`

package/dist/signals.js

@@ -86,13 +86,23 @@ function isNewCompletedCall(tc, seen, filterName) {
 	if (filterName && tc.toolName !== filterName) return false;
 	return true;
 }
-function useToolResult(toolNameOrCallback, maybeCallback) {
-	const filterName = typeof toolNameOrCallback === "string" ? toolNameOrCallback : void 0;
-	const callback = typeof toolNameOrCallback === "function" ? toolNameOrCallback : (_name, result, tc) => maybeCallback?.(result, tc);
-	const { session } = useSession();
+/**
+* Shared helper for hooks that need to fire a callback once per new tool call.
+*
+* Handles deduplication via a `Set<string>` ref, automatic reset when the
+* signal array is cleared, and a stable callback ref to avoid stale closures.
+*
+* @param session - The voice session whose `toolCalls` signal to watch.
+* @param shouldProcess - Predicate that decides whether a tool call is eligible.
+*   Called with the tool call and the seen-IDs set. Must NOT mutate the set.
+* @param onNew - Invoked for each eligible tool call that hasn't been seen yet.
+*/
+function useToolCallEffect(session, shouldProcess, onNew) {
 	const seenRef = useRef(/* @__PURE__ */ new Set());
-	const cbRef = useRef(callback);
-	cbRef.current = callback;
+	const cbRef = useRef(onNew);
+	cbRef.current = onNew;
+	const predicateRef = useRef(shouldProcess);
+	predicateRef.current = shouldProcess;
 	useEffect(() => effect(() => {
 		const toolCalls = session.toolCalls.value;
 		if (toolCalls.length === 0) {
@@ -100,12 +110,19 @@ function useToolResult(toolNameOrCallback, maybeCallback) {
 			return;
 		}
 		for (const tc of toolCalls) {
-			if (!isNewCompletedCall(tc, seenRef.current, filterName)) continue;
+			if (!predicateRef.current(tc, seenRef.current)) continue;
+			if (seenRef.current.has(tc.toolCallId)) continue;
 			seenRef.current.add(tc.toolCallId);
-			cbRef.current(tc.toolName, tryParseJSON(tc.result), tc);
+			cbRef.current(tc);
 		}
 	}), [session]);
 }
+function useToolResult(toolNameOrCallback, maybeCallback) {
+	const filterName = typeof toolNameOrCallback === "string" ? toolNameOrCallback : void 0;
+	const callback = typeof toolNameOrCallback === "function" ? toolNameOrCallback : (_name, result, tc) => maybeCallback?.(result, tc);
+	const { session } = useSession();
+	useToolCallEffect(session, (tc, seen) => isNewCompletedCall(tc, seen, filterName), (tc) => callback(tc.toolName, tryParseJSON(tc.result), tc));
+}
 /**
 * Hook that fires a callback when a new tool call starts (status: "pending").
 *
@@ -119,60 +136,7 @@ function useToolResult(toolNameOrCallback, maybeCallback) {
 */
 function useToolCallStart(callback) {
 	const { session } = useSession();
-	const seenRef = useRef(/* @__PURE__ */ new Set());
-	const cbRef = useRef(callback);
-	cbRef.current = callback;
-	useEffect(() => effect(() => {
-		const toolCalls = session.toolCalls.value;
-		if (toolCalls.length === 0) {
-			seenRef.current.clear();
-			return;
-		}
-		for (const tc of toolCalls) {
-			if (seenRef.current.has(tc.toolCallId)) continue;
-			seenRef.current.add(tc.toolCallId);
-			cbRef.current(tc.toolName, tc.args, tc);
-		}
-	}), [session]);
-}
-/**
-* Hook that fires a callback for each intermediate update pushed by a tool
-* via `ctx.sendUpdate()`.
-*
-* Use this to render progressive/streaming data from a tool before it
-* finishes — e.g. showing a recipe preview card while nutrition data is
-* still loading.
-*
-* The callback fires once per update. The `data` argument is the parsed
-* JSON update (or raw string if parsing fails).
-*
-* @param callback - Called once per intermediate update with the tool name,
-*   parsed data, and full {@link ToolCallInfo}.
-*
-* @public
-*/
-function useToolCallUpdate(callback) {
-	const { session } = useSession();
-	const countRef = useRef(/* @__PURE__ */ new Map());
-	const cbRef = useRef(callback);
-	cbRef.current = callback;
-	useEffect(() => effect(() => {
-		const toolCalls = session.toolCalls.value;
-		if (toolCalls.length === 0) {
-			countRef.current.clear();
-			return;
-		}
-		for (const tc of toolCalls) processNewUpdates(tc, countRef.current, cbRef.current);
-	}), [session]);
-}
-function processNewUpdates(tc, seenCounts, cb) {
-	const seen = seenCounts.get(tc.toolCallId) ?? 0;
-	if (tc.updates.length <= seen) return;
-	for (let i = seen; i < tc.updates.length; i++) {
-		const raw = tc.updates[i];
-		if (raw !== void 0) cb(tc.toolName, tryParseJSON(raw), tc);
-	}
-	seenCounts.set(tc.toolCallId, tc.updates.length);
+	useToolCallEffect(session, () => true, (tc) => callback(tc.toolName, tc.args, tc));
 }
 /**
 * Auto-scroll a container to the bottom when messages, tool calls,
@@ -194,4 +158,4 @@ function useAutoScroll() {
 	return ref;
 }
 //#endregion
-export { SessionProvider, createSessionControls, useAutoScroll, useSession, useToolCallStart, useToolCallUpdate, useToolResult };
+export { SessionProvider, createSessionControls, useAutoScroll, useSession, useToolCallStart, useToolResult };

package/dist/types.d.ts

@@ -29,8 +29,6 @@ export type ToolCallInfo = {
     args: Record<string, unknown>;
     status: "pending" | "done";
     result?: string | undefined;
-    /** Intermediate updates pushed by the tool via `ctx.sendUpdate()`. */
-    updates: string[];
     /** Index in the messages array where this tool call should appear. */
     afterMessageIndex: number;
 };
@@ -86,4 +84,18 @@ export type VoiceSessionOptions = {
      * `persistence` enabled).
      */
     resumeSessionId?: string | undefined;
+    /**
+     * WebSocket constructor override. Defaults to the native `WebSocket`.
+     * Primarily useful for testing with a mock WebSocket.
+     */
+    WebSocket?: WebSocketConstructor | undefined;
+};
+/**
+ * Minimal WebSocket constructor type accepted by {@link VoiceSessionOptions}.
+ *
+ * @public
+ */
+export type WebSocketConstructor = {
+    new (url: string | URL, protocols?: string | string[]): WebSocket;
+    readonly OPEN: number;
 };

package/dist/types.test-d.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Type-level tests for the public API surface of @alexkroman1/aai-ui.
+ *
+ * These are checked by tsc (via vitest typecheck) but never executed.
+ * A failure here means a public type contract has regressed.
+ */
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@alexkroman1/aai-ui",
-  "version": "0.10.2",
+  "version": "0.10.4",
   "type": "module",
   "files": [
     "dist",
@@ -21,7 +21,7 @@
   },
   "dependencies": {
     "clsx": "^2.1.1",
-    "@alexkroman1/aai": "0.10.2"
+    "@alexkroman1/aai": "0.10.4"
   },
   "peerDependencies": {
     "@preact/signals": "^2.8.2",
@@ -38,7 +38,8 @@
     "@testing-library/preact": "^3.2.4",
     "jsdom": "^29.0.1",
     "preact": "^10.29.0",
-    "tsdown": "^0.21.5"
+    "tsdown": "^0.21.5",
+    "vitest": "^4.1.1"
   },
   "engines": {
     "node": ">=22.6"
@@ -52,7 +53,7 @@
     "build": "tsdown && tsc -p tsconfig.build.json",
     "typecheck": "tsc --noEmit",
     "lint": "biome check .",
-    "check:api": "api-extractor run -c api-extractor.json",
+    "check:publint": "publint",
     "check:attw": "attw --pack --profile esm-only --entrypoints . ./session"
   }
 }

package/dist/_sqlite-stub.d.ts

@@ -1,3 +0,0 @@
-export declare class DatabaseSync {
-    constructor();
-}

package/dist/mount-context.js

@@ -1,15 +0,0 @@
-import { createContext } from "preact";
-import { useContext } from "preact/hooks";
-//#region mount-context.ts
-const Ctx = createContext({});
-const MountConfigProvider = Ctx.Provider;
-/**
-* Read mount config (title, theme) from the nearest provider.
-*
-* @public
-*/
-function useMountConfig() {
-	return useContext(Ctx);
-}
-//#endregion
-export { MountConfigProvider, useMountConfig };