@assistant-ui/ai-sdk 0.0.7 → 0.0.8

package/dist/runtime/useResourceCleanup.js

@@ -2,12 +2,12 @@ import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { useEffect, useRef } from "@assistant-ui/tap/react-shim";
 import { useAssistantClientDestroySignal } from "@assistant-ui/store/internal";
 //#region src/runtime/useResourceCleanup.ts
-const useResourceCleanup = (enabled, cleanup) => {
-	const $ = c(7);
+const useResourceCleanup = (enabled, cleanup, hostDestroySignal) => {
+	const $ = c(8);
 	const destroySignal = useAssistantClientDestroySignal();
 	const cleanupRef = useRef(cleanup);
 	const enabledRef = useRef(enabled);
-	const registeredSignalRef = useRef(void 0);
+	const registrationRef = useRef(null);
 	let t0;
 	if ($[0] !== cleanup || $[1] !== enabled) {
 		t0 = () => {
@@ -21,23 +21,47 @@ const useResourceCleanup = (enabled, cleanup) => {
 	useEffect(t0);
 	let t1;
 	let t2;
-	if ($[3] !== destroySignal || $[4] !== enabled) {
+	if ($[3] !== destroySignal || $[4] !== enabled || $[5] !== hostDestroySignal) {
 		t1 = () => {
-			if (!enabled || !destroySignal) return;
-			if (registeredSignalRef.current === destroySignal) return;
-			registeredSignalRef.current = destroySignal;
-			destroySignal.addEventListener("abort", () => {
+			if (!enabled || !destroySignal && !hostDestroySignal) return;
+			const current = registrationRef.current;
+			if (current !== null && current.destroySignal === destroySignal && current.hostDestroySignal === hostDestroySignal) return;
+			current?.controller.abort();
+			const registration = new AbortController();
+			registrationRef.current = {
+				destroySignal,
+				hostDestroySignal,
+				controller: registration
+			};
+			const run = () => {
+				if (registration.signal.aborted) return;
+				registration.abort();
 				if (enabledRef.current) cleanupRef.current();
-			}, { once: true });
+			};
+			if (destroySignal?.aborted || hostDestroySignal?.aborted) {
+				run();
+				return;
+			}
+			const options = {
+				once: true,
+				signal: registration.signal
+			};
+			destroySignal?.addEventListener("abort", run, options);
+			hostDestroySignal?.addEventListener("abort", run, options);
 		};
-		t2 = [destroySignal, enabled];
+		t2 = [
+			destroySignal,
+			enabled,
+			hostDestroySignal
+		];
 		$[3] = destroySignal;
 		$[4] = enabled;
-		$[5] = t1;
-		$[6] = t2;
+		$[5] = hostDestroySignal;
+		$[6] = t1;
+		$[7] = t2;
 	} else {
-		t1 = $[5];
-		t2 = $[6];
+		t1 = $[6];
+		t2 = $[7];
 	}
 	useEffect(t1, t2);
 };

package/dist/runtime/useResourceCleanup.js.map

@@ -1 +1 @@
-{"version":3,"file":"useResourceCleanup.js","names":["useEffect","useRef","useAssistantClientDestroySignal","useResourceCleanup","enabled","cleanup","$","_c","destroySignal","cleanupRef","enabledRef","registeredSignalRef","undefined","t0","current","t1","t2","addEventListener","once"],"sources":["../../src/runtime/useResourceCleanup.ts"],"sourcesContent":["import { useEffect, useRef } from \"react\";\nimport { useAssistantClientDestroySignal } from \"@assistant-ui/store/internal\";\n\nexport const useResourceCleanup = (enabled: boolean, cleanup: () => void) => {\n  const destroySignal = useAssistantClientDestroySignal();\n  const cleanupRef = useRef(cleanup);\n  const enabledRef = useRef(enabled);\n  const registeredSignalRef = useRef<AbortSignal | undefined>(undefined);\n\n  useEffect(() => {\n    cleanupRef.current = cleanup;\n    enabledRef.current = enabled;\n  });\n\n  useEffect(() => {\n    if (!enabled || !destroySignal) return undefined;\n    if (registeredSignalRef.current === destroySignal) return undefined;\n\n    registeredSignalRef.current = destroySignal;\n    destroySignal.addEventListener(\n      \"abort\",\n      () => {\n        if (enabledRef.current) cleanupRef.current();\n      },\n      { once: true },\n    );\n\n    // The listener must survive standalone soft unmounts so a later permanent\n    // client destroy still cleans up the retained resource state.\n    return undefined;\n  }, [destroySignal, enabled]);\n};\n"],"mappings":";;;;AAGA,MAAaG,sBAAqBC,SAAAC,YAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAChC,MAAAC,gBAAsBN,gCAAgC;CACtD,MAAAO,aAAmBR,OAAOI,OAAO;CACjC,MAAAK,aAAmBT,OAAOG,OAAO;CACjC,MAAAO,sBAA4BV,OAAgCW,KAAAA,CAAS;CAAE,IAAAC;CAAA,IAAAP,EAAA,OAAAD,WAAAC,EAAA,OAAAF,SAAA;EAE7DS,WAAA;GACRJ,WAAUK,UAAWT;GACrBK,WAAUI,UAAWV;EAAH;EACnBE,EAAA,KAAAD;EAAAC,EAAA,KAAAF;EAAAE,EAAA,KAAAO;CAAA,OAAAA,KAAAP,EAAA;CAHDN,UAAUa,EAGT;CAAC,IAAAE;CAAA,IAAAC;CAAA,IAAAV,EAAA,OAAAE,iBAAAF,EAAA,OAAAF,SAAA;EAEQW,WAAA;GACR,IAAI,CAACX,WAAD,CAAaI,eAAa;GAC9B,IAAIG,oBAAmBG,YAAaN,eAAa;GAEjDG,oBAAmBG,UAAWN;GAC9BA,cAAaS,iBACX,eACA;IACE,IAAIP,WAAUI,SAAUL,WAAUK,QAAS;GAAE,GAE/C,EAAAI,MAAQ,KAAK,CACf;EAAC;EAKAF,KAAA,CAACR,eAAeJ,OAAO;EAACE,EAAA,KAAAE;EAAAF,EAAA,KAAAF;EAAAE,EAAA,KAAAS;EAAAT,EAAA,KAAAU;CAAA,OAAA;EAAAD,KAAAT,EAAA;EAAAU,KAAAV,EAAA;CAAA;CAhB3BN,UAAUe,IAgBPC,EAAwB;AAAC"}
+{"version":3,"file":"useResourceCleanup.js","names":["useEffect","useRef","useAssistantClientDestroySignal","useResourceCleanup","enabled","cleanup","hostDestroySignal","$","_c","destroySignal","cleanupRef","enabledRef","registrationRef","t0","current","t1","t2","controller","abort","registration","AbortController","run","signal","aborted","options","once","addEventListener"],"sources":["../../src/runtime/useResourceCleanup.ts"],"sourcesContent":["import { useEffect, useRef } from \"react\";\nimport { useAssistantClientDestroySignal } from \"@assistant-ui/store/internal\";\n\nexport const useResourceCleanup = (\n  enabled: boolean,\n  cleanup: () => void,\n  hostDestroySignal?: AbortSignal,\n) => {\n  const destroySignal = useAssistantClientDestroySignal();\n  const cleanupRef = useRef(cleanup);\n  const enabledRef = useRef(enabled);\n  const registrationRef = useRef<{\n    destroySignal: AbortSignal | undefined;\n    hostDestroySignal: AbortSignal | undefined;\n    controller: AbortController;\n  } | null>(null);\n\n  useEffect(() => {\n    cleanupRef.current = cleanup;\n    enabledRef.current = enabled;\n  });\n\n  useEffect(() => {\n    if (!enabled || (!destroySignal && !hostDestroySignal)) return undefined;\n    const current = registrationRef.current;\n    if (\n      current !== null &&\n      current.destroySignal === destroySignal &&\n      current.hostDestroySignal === hostDestroySignal\n    ) {\n      return undefined;\n    }\n    current?.controller.abort();\n\n    const registration = new AbortController();\n    registrationRef.current = {\n      destroySignal,\n      hostDestroySignal,\n      controller: registration,\n    };\n    const run = () => {\n      if (registration.signal.aborted) return;\n      registration.abort();\n      if (enabledRef.current) cleanupRef.current();\n    };\n    if (destroySignal?.aborted || hostDestroySignal?.aborted) {\n      run();\n      return undefined;\n    }\n    const options = { once: true, signal: registration.signal };\n    destroySignal?.addEventListener(\"abort\", run, options);\n    hostDestroySignal?.addEventListener(\"abort\", run, options);\n\n    // The listeners must survive standalone soft unmounts so a later\n    // permanent destroy still cleans up the retained resource state.\n    return undefined;\n  }, [destroySignal, enabled, hostDestroySignal]);\n};\n"],"mappings":";;;;AAGA,MAAaG,sBAAqBC,SAAAC,SAAAC,sBAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAKhC,MAAAC,gBAAsBP,gCAAgC;CACtD,MAAAQ,aAAmBT,OAAOI,OAAO;CACjC,MAAAM,aAAmBV,OAAOG,OAAO;CACjC,MAAAQ,kBAAwBX,OAId,IAAI;CAAE,IAAAY;CAAA,IAAAN,EAAA,OAAAF,WAAAE,EAAA,OAAAH,SAAA;EAENS,WAAA;GACRH,WAAUI,UAAWT;GACrBM,WAAUG,UAAWV;EAAH;EACnBG,EAAA,KAAAF;EAAAE,EAAA,KAAAH;EAAAG,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAHDP,UAAUa,EAGT;CAAC,IAAAE;CAAA,IAAAC;CAAA,IAAAT,EAAA,OAAAE,iBAAAF,EAAA,OAAAH,WAAAG,EAAA,OAAAD,mBAAA;EAEQS,WAAA;GACR,IAAI,CAACX,WAAY,CAACK,iBAAD,CAAmBH,mBAAkB;GACtD,MAAAQ,UAAgBF,gBAAeE;GAC/B,IACEA,YAAY,QACZA,QAAOL,kBAAmBA,iBAC1BK,QAAOR,sBAAuBA,mBAAiB;GAIjDQ,SAAOG,WAAkBC,MAAC;GAE1B,MAAAC,eAAqB,IAAIC,gBAAgB;GACzCR,gBAAeE,UAAW;IAAAL;IAAAH;IAAAW,YAGZE;GACd;GACA,MAAAE,YAAY;IACV,IAAIF,aAAYG,OAAOC,SAAQ;IAC/BJ,aAAYD,MAAO;IACnB,IAAIP,WAAUG,SAAUJ,WAAUI,QAAS;GAAE;GAE/C,IAAIL,eAAac,WAAajB,mBAAiBiB,SAAS;IACtDF,IAAI;IAAC;GAAA;GAGP,MAAAG,UAAgB;IAAAC,MAAQ;IAAIH,QAAUH,aAAYG;GAAQ;GAC1Db,eAAaiB,iBAAmB,SAASL,KAAKG,OAAO;GACrDlB,mBAAiBoB,iBAAmB,SAASL,KAAKG,OAAO;EAAC;EAKzDR,KAAA;GAACP;GAAeL;GAASE;EAAiB;EAACC,EAAA,KAAAE;EAAAF,EAAA,KAAAH;EAAAG,EAAA,KAAAD;EAAAC,EAAA,KAAAQ;EAAAR,EAAA,KAAAS;CAAA,OAAA;EAAAD,KAAAR,EAAA;EAAAS,KAAAT,EAAA;CAAA;CAlC9CP,UAAUe,IAkCPC,EAA2C;AAAC"}

package/dist/runtime/useStreamingTiming.d.ts

@@ -1,7 +1,6 @@
-import { StreamingTimingAccessors } from "@assistant-ui/core/react";
-import { MessageTiming } from "@assistant-ui/core";
-import { UIMessage } from "@ai-sdk/react";
-//#region src/runtime/useStreamingTiming.d.ts
+import type { UIMessage } from "@ai-sdk/react";
+import type { MessageTiming } from "@assistant-ui/core";
+import { type StreamingTimingAccessors } from "@assistant-ui/core/react";
 export declare const aiSdkStreamingTimingAccessors: StreamingTimingAccessors<UIMessage>;
 /**
  * Tracks streaming timing for AI SDK messages client-side. Delegates to the
@@ -10,5 +9,4 @@ export declare const aiSdkStreamingTimingAccessors: StreamingTimingAccessors<UIM
  * and stored per message id.
  */
 export declare const useStreamingTiming: (messages: UIMessage[], isRunning: boolean) => Record<string, MessageTiming>;
-//#endregion
 //# sourceMappingURL=useStreamingTiming.d.ts.map

package/dist/runtime/useStreamingTiming.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamingTiming.d.ts","names":[],"sources":["../../src/runtime/useStreamingTiming.ts"],"mappings":";;;;qBA8Ca,+BAA+B,yBAAyB;;;;;;;qBAaxD,qBACX,UAAU,aACV,uBACC,eAAe"}
+{"version":3,"file":"useStreamingTiming.d.ts","sourceRoot":"","sources":["../../src/runtime/useStreamingTiming.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE/C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,EAEL,KAAK,wBAAwB,EAC9B,MAAM,0BAA0B,CAAC;AAsClC,eAAO,MAAM,6BAA6B,EAAE,wBAAwB,CAAC,SAAS,CAK3E,CAAC;AAEJ;;;;;GAKG;AACH,eAAO,MAAM,kBAAkB,GAC7B,UAAU,SAAS,EAAE,EACrB,WAAW,OAAO,KACjB,MAAM,CAAC,MAAM,EAAE,aAAa,CAK5B,CAAC"}

package/dist/runtime/useStreamingTiming.js

@@ -1,6 +1,6 @@
 "use client";
-import { useStreamingTiming as useStreamingTiming$1 } from "@assistant-ui/core/react";
 import { isToolUIPart } from "ai";
+import { useStreamingTiming as useStreamingTiming$1 } from "@assistant-ui/core/react";
 //#region src/runtime/useStreamingTiming.ts
 const findAssistant = (messages, messageId) => messages.find((m) => m.role === "assistant" && m.id === messageId);
 const getTextLength = (messages, messageId) => {

package/dist/tools/frontendTools.d.ts

@@ -1,32 +1,30 @@
-import { ToolSet } from "ai";
-import { ToolJSONSchema } from "assistant-stream";
-//#region src/tools/frontendTools.d.ts
+import { type ToolSet } from "ai";
+import type { ToolJSONSchema } from "assistant-stream";
 /** Frontend tool definitions uploaded by AssistantChatTransport. */
 export type FrontendTools = Record<string, ToolJSONSchema>;
 export declare const defaultToModelOutput: ({ output }: {
-  output: unknown;
+    output: unknown;
 }) => {
-  type: "content";
-  value: ({
-    type: "text";
-    text: string;
-  } | {
-    filename?: string;
-    type: "file";
-    data: {
-      type: "data";
-      data: string;
-    };
-    mediaType: string;
-    text?: never;
-  })[];
+    type: "content";
+    value: ({
+        type: "text";
+        text: string;
+    } | {
+        filename?: string;
+        type: "file";
+        data: {
+            type: "data";
+            data: string;
+        };
+        mediaType: string;
+        text?: never;
+    })[];
 } | {
-  type: "text";
-  value: string;
+    type: "text";
+    value: string;
 } | {
-  type: "json";
-  value: import("ai").JSONValue;
+    type: "json";
+    value: import("ai").JSONValue;
 };
 export declare const frontendTools: (tools: FrontendTools) => ToolSet;
-//#endregion
 //# sourceMappingURL=frontendTools.d.ts.map

package/dist/tools/frontendTools.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"frontendTools.d.ts","names":[],"sources":["../../src/tools/frontendTools.ts"],"mappings":";;;;YASY,gBAAgB,eAAe;qBAE9B,yBAAwB;EAAc;;;;;;;;;;;;;;;;;;;;;;;qBAuDtC,gBAAiB,OAAO,kBAAgB"}
+{"version":3,"file":"frontendTools.d.ts","sourceRoot":"","sources":["../../src/tools/frontendTools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,KAAK,OAAO,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAOvD,oEAAoE;AACpE,MAAM,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;AAE3D,eAAO,MAAM,oBAAoB,GAAI,YAAY;IAAE,MAAM,EAAE,OAAO,CAAA;CAAE;;;;;;;;;;;;;;;;;;;;;CAMnE,CAAC;AAiDF,eAAO,MAAM,aAAa,GAAI,OAAO,aAAa,KAAG,OAgBpD,CAAC"}

package/dist/tools/generativeTools.d.ts

@@ -1,34 +1,33 @@
-import { FrontendTools } from "./frontendTools.js";
-import { Toolkit } from "@assistant-ui/core/react";
-import { ToolSet } from "ai";
-//#region src/tools/generativeTools.d.ts
+import { type ToolSet } from "ai";
+import type { Toolkit } from "@assistant-ui/core/react";
+import { type FrontendTools } from "./frontendTools.js";
 /**
  * @deprecated Options for the deprecated {@link generativeTools}. Use
  * {@link AISDKToolkit} with {@link AISDKToolkitOptions} /
  * {@link AISDKToolkitToolsOptions} instead.
  */
 export interface GenerativeToolsOptions {
-  /**
-   * The server build of a generative toolkit (schema + server `execute`). Typed
-   * as the canonical {@link Toolkit} so callers don't need to cast; the server
-   * build carries `execute`, recovered internally as {@link ToolkitDefinition}.
-   */
-  toolkit: Toolkit;
-  /**
-   * Tools uploaded by the frontend (the request body's `tools`). Merged in
-   * alongside the `toolkit`; a server `execute` from `toolkit` takes precedence
-   * over an uploaded entry of the same name.
-   */
-  frontendTools?: FrontendTools;
+    /**
+     * The server build of a generative toolkit (schema + server `execute`). Typed
+     * as the canonical {@link Toolkit} so callers don't need to cast; the server
+     * build carries `execute`, recovered internally as {@link ToolkitDefinition}.
+     */
+    toolkit: Toolkit;
+    /**
+     * Tools uploaded by the frontend (the request body's `tools`). Merged in
+     * alongside the `toolkit`; a server `execute` from `toolkit` takes precedence
+     * over an uploaded entry of the same name.
+     */
+    frontendTools?: FrontendTools;
 }
 export type AISDKToolkitOptions = {
-  toolkit: Toolkit;
+    toolkit: Toolkit;
 };
 export type AISDKToolkitToolsOptions = {
-  /**
-   * Tools uploaded by the frontend request body.
-   */
-  frontend?: FrontendTools;
+    /**
+     * Tools uploaded by the frontend request body.
+     */
+    frontend?: FrontendTools;
 };
 /**
  * Builds an AI SDK `ToolSet` for server-side use with `streamText` /
@@ -63,10 +62,9 @@ export type AISDKToolkitToolsOptions = {
  */
 export declare const generativeTools: (options: GenerativeToolsOptions) => ToolSet;
 export declare class AISDKToolkit {
-  #private;
-  constructor(options: AISDKToolkitOptions);
-  tools(options?: AISDKToolkitToolsOptions): Promise<ToolSet>;
-  close(): Promise<void>;
+    #private;
+    constructor(options: AISDKToolkitOptions);
+    tools(options?: AISDKToolkitToolsOptions): Promise<ToolSet>;
+    close(): Promise<void>;
 }
-//#endregion
 //# sourceMappingURL=generativeTools.d.ts.map

package/dist/tools/generativeTools.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generativeTools.d.ts","names":[],"sources":["../../src/tools/generativeTools.ts"],"mappings":";;;;;;;;;iBAsFiB;;;;;;EAMf,SAAS;;;;;;EAMT,gBAAgB;;YAGN;EACV,SAAS;;YAGC;;;;EAIV,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qBAkCA,kBAAmB,SAAS,2BAAyB;qBAYrD;;EAIC,YAAA,SAAS;EAIf,MAAM,UAAS,2BAAgC,QAAQ;EAsBvD,SAAS"}
+{"version":3,"file":"generativeTools.d.ts","sourceRoot":"","sources":["../../src/tools/generativeTools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,KAAK,OAAO,EAAE,MAAM,IAAI,CAAC;AAU9C,OAAO,KAAK,EAEV,OAAO,EAER,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAiB,KAAK,aAAa,EAAE,2BAAwB;AAkEpE;;;;GAIG;AACH,MAAM,WAAW,sBAAsB;IACrC;;;;OAIG;IACH,OAAO,EAAE,OAAO,CAAC;IACjB;;;;OAIG;IACH,aAAa,CAAC,EAAE,aAAa,CAAC;CAC/B;AAED,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,EAAE,OAAO,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC;;OAEG;IACH,QAAQ,CAAC,EAAE,aAAa,CAAC;CAC1B,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,eAAO,MAAM,eAAe,GAAI,SAAS,sBAAsB,KAAG,OAUjE,CAAC;AAEF,qBAAa,YAAY;;gBAIX,OAAO,EAAE,mBAAmB;IAIlC,KAAK,CAAC,OAAO,GAAE,wBAA6B,GAAG,OAAO,CAAC,OAAO,CAAC;IAsB/D,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAmH7B"}

package/dist/tools/mcp-stdio.node.d.ts

@@ -1,2 +1,2 @@
-import { Experimental_StdioMCPTransport } from "@ai-sdk/mcp/mcp-stdio";
-export { Experimental_StdioMCPTransport };
+export { Experimental_StdioMCPTransport } from "@ai-sdk/mcp/mcp-stdio";
+//# sourceMappingURL=mcp-stdio.node.d.ts.map

package/dist/tools/mcp-stdio.node.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-stdio.node.d.ts","sourceRoot":"","sources":["../../src/tools/mcp-stdio.node.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,8BAA8B,EAAE,MAAM,uBAAuB,CAAC"}

package/dist/tools/mcp-stdio.unsupported.d.ts

@@ -1,5 +1,3 @@
-import { Experimental_StdioMCPTransport as Experimental_StdioMCPTransport$1 } from "@ai-sdk/mcp/mcp-stdio";
-//#region src/tools/mcp-stdio.unsupported.d.ts
-export declare const Experimental_StdioMCPTransport: typeof Experimental_StdioMCPTransport$1;
-//#endregion
+import type { Experimental_StdioMCPTransport as NodeStdioMCPTransport } from "@ai-sdk/mcp/mcp-stdio";
+export declare const Experimental_StdioMCPTransport: typeof NodeStdioMCPTransport;
 //# sourceMappingURL=mcp-stdio.unsupported.d.ts.map

package/dist/tools/mcp-stdio.unsupported.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mcp-stdio.unsupported.d.ts","names":[],"sources":["../../src/tools/mcp-stdio.unsupported.ts"],"mappings":";;qBAUa,uCACuC"}
+{"version":3,"file":"mcp-stdio.unsupported.d.ts","sourceRoot":"","sources":["../../src/tools/mcp-stdio.unsupported.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,8BAA8B,IAAI,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAUrG,eAAO,MAAM,8BAA8B,EACE,OAAO,qBAAqB,CAAC"}

package/dist/transport/AssistantChatTransport.d.ts

@@ -1,22 +1,20 @@
-import { AssistantChatResumableOptions } from "./resumable.js";
-import { DefaultChatTransport, HttpChatTransportInitOptions, UIMessage } from "ai";
-import { AssistantRuntime, ThreadListItemRuntime } from "@assistant-ui/core";
-//#region src/transport/AssistantChatTransport.d.ts
+import type { AssistantRuntime, ThreadListItemRuntime } from "@assistant-ui/core";
+import { DefaultChatTransport, type HttpChatTransportInitOptions, type UIMessage } from "ai";
+import { type AssistantChatResumableOptions } from "./resumable.js";
 export type InitializableThreadListItem = Pick<ThreadListItemRuntime, "initialize">;
 export type AssistantChatTransportInitOptions<UI_MESSAGE extends UIMessage> = HttpChatTransportInitOptions<UI_MESSAGE> & {
-  resumable?: AssistantChatResumableOptions;
+    resumable?: AssistantChatResumableOptions;
 };
 export declare class AssistantChatTransport<UI_MESSAGE extends UIMessage> extends DefaultChatTransport<UI_MESSAGE> {
-  private runtime;
-  private getThreadListItem;
-  private readonly resumable;
-  private readonly __internal_initOptions;
-  /** Constructs an unwired copy with the same init options. */
-  __internal_clone(): AssistantChatTransport<UI_MESSAGE>;
-  constructor(initOptions?: AssistantChatTransportInitOptions<UI_MESSAGE>);
-  setRuntime(runtime: AssistantRuntime): void;
-  getResumableAdapter(): AssistantChatResumableOptions | undefined;
-  __internal_setGetThreadListItem(getter: () => InitializableThreadListItem | undefined): void;
+    private runtime;
+    private getThreadListItem;
+    private readonly resumable;
+    private readonly __internal_initOptions;
+    /** Constructs an unwired copy with the same init options. */
+    __internal_clone(): AssistantChatTransport<UI_MESSAGE>;
+    constructor(initOptions?: AssistantChatTransportInitOptions<UI_MESSAGE>);
+    setRuntime(runtime: AssistantRuntime): void;
+    getResumableAdapter(): AssistantChatResumableOptions | undefined;
+    __internal_setGetThreadListItem(getter: () => InitializableThreadListItem | undefined): void;
 }
-//#endregion
 //# sourceMappingURL=AssistantChatTransport.d.ts.map

package/dist/transport/AssistantChatTransport.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantChatTransport.d.ts","names":[],"sources":["../../src/transport/AssistantChatTransport.ts"],"mappings":";;;;YAeY,8BAA8B,KACxC;YAaU,kCAAkC,mBAAmB,aAC/D,6BAA6B;EAC3B,YAAY;;qBAGH,uBACX,mBAAmB,mBACX,qBAAqB;UACrB;UACA;mBAGS;mBAEA;;EAKV,oBAAoB,uBAAuB;EAOtC,YAAA,cAAc,kCAAkC;EA0D5D,WAAW,SAAS;EAIpB,uBAAuB;EAIvB,gCACE,cAAc"}
+{"version":3,"file":"AssistantChatTransport.d.ts","sourceRoot":"","sources":["../../src/transport/AssistantChatTransport.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,gBAAgB,EAChB,qBAAqB,EACtB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,oBAAoB,EACpB,KAAK,4BAA4B,EACjC,KAAK,SAAS,EACf,MAAM,IAAI,CAAC;AAEZ,OAAO,EAEL,KAAK,6BAA6B,EACnC,uBAAoB;AAErB,MAAM,MAAM,2BAA2B,GAAG,IAAI,CAC5C,qBAAqB,EACrB,YAAY,CACb,CAAC;AAWF,MAAM,MAAM,iCAAiC,CAAC,UAAU,SAAS,SAAS,IACxE,4BAA4B,CAAC,UAAU,CAAC,GAAG;IACzC,SAAS,CAAC,EAAE,6BAA6B,CAAC;CAC3C,CAAC;AAEJ,qBAAa,sBAAsB,CACjC,UAAU,SAAS,SAAS,CAC5B,SAAQ,oBAAoB,CAAC,UAAU,CAAC;IACxC,OAAO,CAAC,OAAO,CAA+B;IAC9C,OAAO,CAAC,iBAAiB,CAEX;IACd,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA4C;IAEtE,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAEzB;IAEd,6DAA6D;IACtD,gBAAgB,IAAI,sBAAsB,CAAC,UAAU,CAAC;gBAOjD,WAAW,CAAC,EAAE,iCAAiC,CAAC,UAAU,CAAC;IA0DvE,UAAU,CAAC,OAAO,EAAE,gBAAgB;IAIpC,mBAAmB,IAAI,6BAA6B,GAAG,SAAS;IAIhE,+BAA+B,CAC7B,MAAM,EAAE,MAAM,2BAA2B,GAAG,SAAS;CAIxD"}

package/dist/transport/resumable.d.ts

@@ -1,37 +1,35 @@
-//#region src/transport/resumable.d.ts
 /** Response header used by the [Resumable Streams](/docs/guides/resumable-streams) server and client wiring. */
 export declare const RESUMABLE_STREAM_ID_HEADER = "x-resumable-stream-id";
 export type ResumableClientStorage = {
-  getStreamId(threadId?: string): string | null;
-  setStreamId(id: string, threadId?: string): void;
-  clear(threadId?: string): void;
-  /** Subscribes to stream id changes so automatic resume can react after mount. */
-  subscribe?(listener: () => void, threadId?: string): () => void;
+    getStreamId(threadId?: string): string | null;
+    setStreamId(id: string, threadId?: string): void;
+    clear(threadId?: string): void;
+    /** Subscribes to stream id changes so automatic resume can react after mount. */
+    subscribe?(listener: () => void, threadId?: string): () => void;
 };
 /** `sessionStorage`-backed storage for the pending resumable stream id. See the [Resumable Streams](/docs/guides/resumable-streams) guide for end-to-end wiring. */
 export declare function createResumableSessionStorage(options?: {
-  /**
-   * Storage key for the pending stream id. A static string namespaces per route
-   * or chat surface. A getter is read lazily on every access, so the key can be
-   * derived from the active thread's identity; while the getter returns
-   * `undefined`, reads report no pending stream and writes are dropped, so a
-   * thread whose identity is not known yet never touches another thread's key.
-   *
-   * Under a remote thread list with more than one thread, scope the key per
-   * thread and create one storage instance per thread runtime rather than a
-   * single shared one. A shared key is written and cleared by whichever thread
-   * acts last, so one conversation's stream can resume inside another.
-   */
-  key?: string | (() => string | undefined);
+    /**
+     * Storage key for the pending stream id. A static string namespaces per route
+     * or chat surface. A getter is read lazily on every access, so the key can be
+     * derived from the active thread's identity; while the getter returns
+     * `undefined`, reads report no pending stream and writes are dropped, so a
+     * thread whose identity is not known yet never touches another thread's key.
+     *
+     * Under a remote thread list with more than one thread, scope the key per
+     * thread and create one storage instance per thread runtime rather than a
+     * single shared one. A shared key is written and cleared by whichever thread
+     * acts last, so one conversation's stream can resume inside another.
+     */
+    key?: string | (() => string | undefined);
 }): ResumableClientStorage;
 export type AssistantChatResumableOptions = {
-  storage: ResumableClientStorage;
-  resumeApi: string | ((streamId: string) => string);
-  /**
-   * Defaults to scanning for the AI SDK UIMessageStream `finish` marker.
-   * Cancellation never invokes this callback, only natural completion does.
-   */
-  isFinishEvent?: (chunk: Uint8Array, accumulator: string) => boolean;
+    storage: ResumableClientStorage;
+    resumeApi: string | ((streamId: string) => string);
+    /**
+     * Defaults to scanning for the AI SDK UIMessageStream `finish` marker.
+     * Cancellation never invokes this callback, only natural completion does.
+     */
+    isFinishEvent?: (chunk: Uint8Array, accumulator: string) => boolean;
 };
-//#endregion
 //# sourceMappingURL=resumable.d.ts.map

package/dist/transport/resumable.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"resumable.d.ts","names":[],"sources":["../../src/transport/resumable.ts"],"mappings":";;qBAKa;YAID;EACV,YAAY;EACZ,YAAY,YAAY;EACxB,MAAM;;EAEN,WAAW,sBAAsB;;;wBAkBnB,8BAA8B;;;;;;;;;;;;;EAa5C;IACE;YA0GQ;EACV,SAAS;EACT,sBAAsB;;;;;EAKtB,iBAAiB,OAAO,YAAY"}
+{"version":3,"file":"resumable.d.ts","sourceRoot":"","sources":["../../src/transport/resumable.ts"],"names":[],"mappings":"AAIA,gHAAgH;AAChH,eAAO,MAAM,0BAA0B,0BAAmC,CAAC;AAI3E,MAAM,MAAM,sBAAsB,GAAG;IACnC,WAAW,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;IAC9C,WAAW,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACjD,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,iFAAiF;IACjF,SAAS,CAAC,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,IAAI,CAAC;CACjE,CAAC;AAgBF,oKAAoK;AACpK,wBAAgB,6BAA6B,CAAC,OAAO,CAAC,EAAE;IACtD;;;;;;;;;;;OAWG;IACH,GAAG,CAAC,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC;CAC3C,GAAG,sBAAsB,CAwGzB;AAED,MAAM,MAAM,6BAA6B,GAAG;IAC1C,OAAO,EAAE,sBAAsB,CAAC;IAChC,SAAS,EAAE,MAAM,GAAG,CAAC,CAAC,QAAQ,EAAE,MAAM,KAAK,MAAM,CAAC,CAAC;IACnD;;;OAGG;IACH,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,KAAK,OAAO,CAAC;CACrE,CAAC"}

package/dist/usage.d.ts

@@ -1,15 +1,14 @@
-/// <reference types="@assistant-ui/core/react" />
-//#region src/usage.d.ts
+/// <reference types="@assistant-ui/core/react" preserve="true" />
 export type ThreadTokenUsage = {
-  totalTokens?: number;
-  inputTokens?: number;
-  outputTokens?: number;
-  reasoningTokens?: number;
-  cachedInputTokens?: number;
+    totalTokens?: number;
+    inputTokens?: number;
+    outputTokens?: number;
+    reasoningTokens?: number;
+    cachedInputTokens?: number;
 };
 export interface TokenUsageExtractableMessage {
-  role?: string;
-  metadata?: unknown;
+    role?: string;
+    metadata?: unknown;
 }
 export declare function getThreadMessageTokenUsage(message: TokenUsageExtractableMessage | undefined): ThreadTokenUsage | undefined;
 /**
@@ -21,5 +20,4 @@ export declare function getThreadMessageTokenUsage(message: TokenUsageExtractabl
  * looks.
  */
 export declare function useThreadTokenUsage(): ThreadTokenUsage | undefined;
-//#endregion
 //# sourceMappingURL=usage.d.ts.map

package/dist/usage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"usage.d.ts","names":[],"sources":["../src/usage.ts"],"mappings":";YAIY;EACV;EACA;EACA;EACA;EACA;;iBAGe;EACf;EACA;;wBAuHc,2BACd,SAAS,2CACR;;;;;;;;;wBAsCa,uBAAuB"}
+{"version":3,"file":"usage.d.ts","sourceRoot":"","sources":["../src/usage.ts"],"names":[],"mappings":";AAIA,MAAM,MAAM,gBAAgB,GAAG;IAC7B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,WAAW,4BAA4B;IAC3C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAsHD,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,4BAA4B,GAAG,SAAS,GAChD,gBAAgB,GAAG,SAAS,CAa9B;AAiBD;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,IAAI,gBAAgB,GAAG,SAAS,CAGlE"}

package/dist/usage.js.map

@@ -1 +1 @@
-{"version":3,"file":"usage.js","names":["useMemo","useAuiState","ThreadTokenUsage","totalTokens","inputTokens","outputTokens","reasoningTokens","cachedInputTokens","TokenUsageExtractableMessage","role","metadata","UsageRecord","Record","USAGE_KEYS","const","asRecord","value","Array","isArray","undefined","asPositiveTokenCount","Number","isFinite","computeTotalTokens","usage","normalizeUsage","record","result","hasFields","key","count","outputTokenDetails","inputTokenDetails","cacheReadTokens","withComputedTotal","usageFromSteps","steps","sums","present","stepsWithUsage","stepsWithComputableTotal","step","stepTotal","getThreadMessageTokenUsage","message","topLevelUsage","customUsage","custom","findLatestMessageWithUsage","messages","idx","length","useThreadTokenUsage","$","_c","msg","_temp","t0","s","thread"],"sources":["../src/usage.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/react\" />\nimport { useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\n\nexport type ThreadTokenUsage = {\n  totalTokens?: number;\n  inputTokens?: number;\n  outputTokens?: number;\n  reasoningTokens?: number;\n  cachedInputTokens?: number;\n};\n\nexport interface TokenUsageExtractableMessage {\n  role?: string;\n  metadata?: unknown;\n}\n\ntype UsageRecord = Record<string, unknown>;\n\nconst USAGE_KEYS = [\n  \"inputTokens\",\n  \"outputTokens\",\n  \"reasoningTokens\",\n  \"cachedInputTokens\",\n  \"totalTokens\",\n] as const satisfies (keyof ThreadTokenUsage)[];\n\nfunction asRecord(value: unknown): UsageRecord | undefined {\n  if (!value || typeof value !== \"object\" || Array.isArray(value))\n    return undefined;\n  return value as UsageRecord;\n}\n\nfunction asPositiveTokenCount(value: unknown): number | undefined {\n  if (typeof value !== \"number\" || !Number.isFinite(value) || value < 0) {\n    return undefined;\n  }\n  return value;\n}\n\nfunction computeTotalTokens(usage: ThreadTokenUsage): number | undefined {\n  if (usage.totalTokens !== undefined) return usage.totalTokens;\n  if (usage.inputTokens !== undefined && usage.outputTokens !== undefined) {\n    return usage.inputTokens + usage.outputTokens;\n  }\n  return undefined;\n}\n\nfunction normalizeUsage(value: unknown): ThreadTokenUsage | undefined {\n  const record = asRecord(value);\n  if (!record) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  let hasFields = false;\n  for (const key of USAGE_KEYS) {\n    const count = asPositiveTokenCount(record[key]);\n    if (count !== undefined) {\n      result[key] = count;\n      hasFields = true;\n    }\n  }\n  // AI SDK v7 moved these under token detail objects; v6 kept them top-level.\n  if (result.reasoningTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.outputTokenDetails)?.reasoningTokens,\n    );\n    if (count !== undefined) {\n      result.reasoningTokens = count;\n      hasFields = true;\n    }\n  }\n  if (result.cachedInputTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.inputTokenDetails)?.cacheReadTokens,\n    );\n    if (count !== undefined) {\n      result.cachedInputTokens = count;\n      hasFields = true;\n    }\n  }\n  return hasFields ? result : undefined;\n}\n\nfunction withComputedTotal(\n  usage: ThreadTokenUsage,\n): ThreadTokenUsage | undefined {\n  const totalTokens = computeTotalTokens(usage);\n  return { ...usage, ...(totalTokens !== undefined && { totalTokens }) };\n}\n\nfunction usageFromSteps(value: unknown): ThreadTokenUsage | undefined {\n  const steps = Array.isArray(value) ? value : [];\n\n  const sums: Record<string, number> = {};\n  const present: Record<string, boolean> = {};\n  let stepsWithUsage = 0;\n  let stepsWithComputableTotal = 0;\n\n  for (const step of steps) {\n    const usage = normalizeUsage(asRecord(step)?.usage);\n    if (!usage) continue;\n    stepsWithUsage++;\n\n    const stepTotal = computeTotalTokens(usage);\n    if (stepTotal !== undefined) {\n      sums.totalTokens = (sums.totalTokens ?? 0) + stepTotal;\n      stepsWithComputableTotal++;\n    }\n\n    for (const key of USAGE_KEYS) {\n      if (key === \"totalTokens\") continue;\n      if (usage[key] !== undefined) {\n        sums[key] = (sums[key] ?? 0) + usage[key];\n        present[key] = true;\n      }\n    }\n  }\n\n  if (stepsWithUsage === 0) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  if (stepsWithComputableTotal === stepsWithUsage) {\n    result.totalTokens = sums.totalTokens!;\n  }\n  for (const key of USAGE_KEYS) {\n    if (key === \"totalTokens\") continue;\n    if (present[key]) {\n      result[key] = sums[key]!;\n    }\n  }\n  return result;\n}\n\nexport function getThreadMessageTokenUsage(\n  message: TokenUsageExtractableMessage | undefined,\n): ThreadTokenUsage | undefined {\n  if (!message || message.role !== \"assistant\") return undefined;\n\n  const metadata = asRecord(message.metadata);\n  if (!metadata) return undefined;\n\n  const topLevelUsage = normalizeUsage(metadata.usage);\n  if (topLevelUsage) return withComputedTotal(topLevelUsage);\n\n  const customUsage = normalizeUsage(asRecord(metadata.custom)?.usage);\n  if (customUsage) return withComputedTotal(customUsage);\n\n  return usageFromSteps(metadata.steps);\n}\n\nfunction findLatestMessageWithUsage(\n  messages: readonly TokenUsageExtractableMessage[] | undefined,\n): TokenUsageExtractableMessage | undefined {\n  if (!messages) return undefined;\n\n  for (let idx = messages.length - 1; idx >= 0; idx -= 1) {\n    const message = messages[idx];\n    if (getThreadMessageTokenUsage(message)) {\n      return message;\n    }\n  }\n\n  return undefined;\n}\n\n/**\n * Reads token usage from the newest assistant message that reports any.\n *\n * A route attaches usage through the AI SDK's `messageMetadata` option. Because\n * a thread message carries a fixed metadata shape, the converter moves every\n * other key the route returns into `metadata.custom`, which is where this hook\n * looks.\n */\nexport function useThreadTokenUsage(): ThreadTokenUsage | undefined {\n  const msg = useAuiState((s) => findLatestMessageWithUsage(s.thread.messages));\n  return useMemo(() => getThreadMessageTokenUsage(msg), [msg]);\n}\n"],"mappings":";;;;AAmBA,MAAMa,aAAa;CACjB;CACA;CACA;CACA;CACA;AAAa;AAGf,SAASE,SAASC,OAAyC;CACzD,IAAI,CAACA,SAAS,OAAOA,UAAU,YAAYC,MAAMC,QAAQF,KAAK,GAC5D,OAAOG,KAAAA;CACT,OAAOH;AACT;AAEA,SAASI,qBAAqBJ,OAAoC;CAChE,IAAI,OAAOA,UAAU,YAAY,CAACK,OAAOC,SAASN,KAAK,KAAKA,QAAQ,GAClE;CAEF,OAAOA;AACT;AAEA,SAASO,mBAAmBC,OAA6C;CACvE,IAAIA,MAAMrB,gBAAgBgB,KAAAA,GAAW,OAAOK,MAAMrB;CAClD,IAAIqB,MAAMpB,gBAAgBe,KAAAA,KAAaK,MAAMnB,iBAAiBc,KAAAA,GAC5D,OAAOK,MAAMpB,cAAcoB,MAAMnB;AAGrC;AAEA,SAASoB,eAAeT,OAA8C;CACpE,MAAMU,SAASX,SAASC,KAAK;CAC7B,IAAI,CAACU,QAAQ,OAAOP,KAAAA;CAEpB,MAAMQ,SAA2B,CAAC;CAClC,IAAIC,YAAY;CAChB,KAAK,MAAMC,OAAOhB,YAAY;EAC5B,MAAMiB,QAAQV,qBAAqBM,OAAOG,IAAI;EAC9C,IAAIC,UAAUX,KAAAA,GAAW;GACvBQ,OAAOE,OAAOC;GACdF,YAAY;EACd;CACF;CAEA,IAAID,OAAOrB,oBAAoBa,KAAAA,GAAW;EACxC,MAAMW,QAAQV,qBACZL,SAASW,OAAOK,kBAAkB,CAAC,EAAEzB,eACvC;EACA,IAAIwB,UAAUX,KAAAA,GAAW;GACvBQ,OAAOrB,kBAAkBwB;GACzBF,YAAY;EACd;CACF;CACA,IAAID,OAAOpB,sBAAsBY,KAAAA,GAAW;EAC1C,MAAMW,QAAQV,qBACZL,SAASW,OAAOM,iBAAiB,CAAC,EAAEC,eACtC;EACA,IAAIH,UAAUX,KAAAA,GAAW;GACvBQ,OAAOpB,oBAAoBuB;GAC3BF,YAAY;EACd;CACF;CACA,OAAOA,YAAYD,SAASR,KAAAA;AAC9B;AAEA,SAASe,kBACPV,OAC8B;CAC9B,MAAMrB,cAAcoB,mBAAmBC,KAAK;CAC5C,OAAO;EAAE,GAAGA;EAAO,GAAIrB,gBAAgBgB,KAAAA,KAAa,EAAEhB,YAAY;CAAG;AACvE;AAEA,SAASgC,eAAenB,OAA8C;CACpE,MAAMoB,QAAQnB,MAAMC,QAAQF,KAAK,IAAIA,QAAQ,CAAA;CAE7C,MAAMqB,OAA+B,CAAC;CACtC,MAAMC,UAAmC,CAAC;CAC1C,IAAIC,iBAAiB;CACrB,IAAIC,2BAA2B;CAE/B,KAAK,MAAMC,QAAQL,OAAO;EACxB,MAAMZ,QAAQC,eAAeV,SAAS0B,IAAI,CAAC,EAAEjB,KAAK;EAClD,IAAI,CAACA,OAAO;EACZe;EAEA,MAAMG,YAAYnB,mBAAmBC,KAAK;EAC1C,IAAIkB,cAAcvB,KAAAA,GAAW;GAC3BkB,KAAKlC,eAAekC,KAAKlC,eAAe,KAAKuC;GAC7CF;EACF;EAEA,KAAK,MAAMX,OAAOhB,YAAY;GAC5B,IAAIgB,QAAQ,eAAe;GAC3B,IAAIL,MAAMK,SAASV,KAAAA,GAAW;IAC5BkB,KAAKR,QAAQQ,KAAKR,QAAQ,KAAKL,MAAMK;IACrCS,QAAQT,OAAO;GACjB;EACF;CACF;CAEA,IAAIU,mBAAmB,GAAG,OAAOpB,KAAAA;CAEjC,MAAMQ,SAA2B,CAAC;CAClC,IAAIa,6BAA6BD,gBAC/BZ,OAAOxB,cAAckC,KAAKlC;CAE5B,KAAK,MAAM0B,OAAOhB,YAAY;EAC5B,IAAIgB,QAAQ,eAAe;EAC3B,IAAIS,QAAQT,MACVF,OAAOE,OAAOQ,KAAKR;CAEvB;CACA,OAAOF;AACT;AAEA,SAAgBgB,2BACdC,SAC8B;CAC9B,IAAI,CAACA,WAAWA,QAAQnC,SAAS,aAAa,OAAOU,KAAAA;CAErD,MAAMT,WAAWK,SAAS6B,QAAQlC,QAAQ;CAC1C,IAAI,CAACA,UAAU,OAAOS,KAAAA;CAEtB,MAAM0B,gBAAgBpB,eAAef,SAASc,KAAK;CACnD,IAAIqB,eAAe,OAAOX,kBAAkBW,aAAa;CAEzD,MAAMC,cAAcrB,eAAeV,SAASL,SAASqC,MAAM,CAAC,EAAEvB,KAAK;CACnE,IAAIsB,aAAa,OAAOZ,kBAAkBY,WAAW;CAErD,OAAOX,eAAezB,SAAS0B,KAAK;AACtC;AAEA,SAASY,2BACPC,UAC0C;CAC1C,IAAI,CAACA,UAAU,OAAO9B,KAAAA;CAEtB,KAAK,IAAI+B,MAAMD,SAASE,SAAS,GAAGD,OAAO,GAAGA,OAAO,GAAG;EACtD,MAAMN,UAAUK,SAASC;EACzB,IAAIP,2BAA2BC,OAAO,GACpC,OAAOA;CAEX;AAGF;;;;;;;;;AAUA,SAAOQ,sBAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CACL,MAAAC,MAAYtD,YAAYuD,KAAoD;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAAE,KAAA;EACzDE,KAAAd,2BAA2BY,GAAG;EAACF,EAAA,KAAAE;EAAAF,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,OAA/BI;AAA+B;AAF/C,SAAAD,MAAAE,GAAA;CAAA,OAC0BV,2BAA2BU,EAACC,OAAOV,QAAS;AAAC"}
+{"version":3,"file":"usage.js","names":["useMemo","useAuiState","ThreadTokenUsage","totalTokens","inputTokens","outputTokens","reasoningTokens","cachedInputTokens","TokenUsageExtractableMessage","role","metadata","UsageRecord","Record","USAGE_KEYS","const","asRecord","value","Array","isArray","undefined","asPositiveTokenCount","Number","isFinite","computeTotalTokens","usage","normalizeUsage","record","result","hasFields","key","count","outputTokenDetails","inputTokenDetails","cacheReadTokens","withComputedTotal","usageFromSteps","steps","sums","present","stepsWithUsage","stepsWithComputableTotal","step","stepTotal","getThreadMessageTokenUsage","message","topLevelUsage","customUsage","custom","findLatestMessageWithUsage","messages","idx","length","useThreadTokenUsage","$","_c","msg","_temp","t0","s","thread"],"sources":["../src/usage.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/react\" preserve=\"true\" />\nimport { useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\n\nexport type ThreadTokenUsage = {\n  totalTokens?: number;\n  inputTokens?: number;\n  outputTokens?: number;\n  reasoningTokens?: number;\n  cachedInputTokens?: number;\n};\n\nexport interface TokenUsageExtractableMessage {\n  role?: string;\n  metadata?: unknown;\n}\n\ntype UsageRecord = Record<string, unknown>;\n\nconst USAGE_KEYS = [\n  \"inputTokens\",\n  \"outputTokens\",\n  \"reasoningTokens\",\n  \"cachedInputTokens\",\n  \"totalTokens\",\n] as const satisfies (keyof ThreadTokenUsage)[];\n\nfunction asRecord(value: unknown): UsageRecord | undefined {\n  if (!value || typeof value !== \"object\" || Array.isArray(value))\n    return undefined;\n  return value as UsageRecord;\n}\n\nfunction asPositiveTokenCount(value: unknown): number | undefined {\n  if (typeof value !== \"number\" || !Number.isFinite(value) || value < 0) {\n    return undefined;\n  }\n  return value;\n}\n\nfunction computeTotalTokens(usage: ThreadTokenUsage): number | undefined {\n  if (usage.totalTokens !== undefined) return usage.totalTokens;\n  if (usage.inputTokens !== undefined && usage.outputTokens !== undefined) {\n    return usage.inputTokens + usage.outputTokens;\n  }\n  return undefined;\n}\n\nfunction normalizeUsage(value: unknown): ThreadTokenUsage | undefined {\n  const record = asRecord(value);\n  if (!record) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  let hasFields = false;\n  for (const key of USAGE_KEYS) {\n    const count = asPositiveTokenCount(record[key]);\n    if (count !== undefined) {\n      result[key] = count;\n      hasFields = true;\n    }\n  }\n  // AI SDK v7 moved these under token detail objects; v6 kept them top-level.\n  if (result.reasoningTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.outputTokenDetails)?.reasoningTokens,\n    );\n    if (count !== undefined) {\n      result.reasoningTokens = count;\n      hasFields = true;\n    }\n  }\n  if (result.cachedInputTokens === undefined) {\n    const count = asPositiveTokenCount(\n      asRecord(record.inputTokenDetails)?.cacheReadTokens,\n    );\n    if (count !== undefined) {\n      result.cachedInputTokens = count;\n      hasFields = true;\n    }\n  }\n  return hasFields ? result : undefined;\n}\n\nfunction withComputedTotal(\n  usage: ThreadTokenUsage,\n): ThreadTokenUsage | undefined {\n  const totalTokens = computeTotalTokens(usage);\n  return { ...usage, ...(totalTokens !== undefined && { totalTokens }) };\n}\n\nfunction usageFromSteps(value: unknown): ThreadTokenUsage | undefined {\n  const steps = Array.isArray(value) ? value : [];\n\n  const sums: Record<string, number> = {};\n  const present: Record<string, boolean> = {};\n  let stepsWithUsage = 0;\n  let stepsWithComputableTotal = 0;\n\n  for (const step of steps) {\n    const usage = normalizeUsage(asRecord(step)?.usage);\n    if (!usage) continue;\n    stepsWithUsage++;\n\n    const stepTotal = computeTotalTokens(usage);\n    if (stepTotal !== undefined) {\n      sums.totalTokens = (sums.totalTokens ?? 0) + stepTotal;\n      stepsWithComputableTotal++;\n    }\n\n    for (const key of USAGE_KEYS) {\n      if (key === \"totalTokens\") continue;\n      if (usage[key] !== undefined) {\n        sums[key] = (sums[key] ?? 0) + usage[key];\n        present[key] = true;\n      }\n    }\n  }\n\n  if (stepsWithUsage === 0) return undefined;\n\n  const result: ThreadTokenUsage = {};\n  if (stepsWithComputableTotal === stepsWithUsage) {\n    result.totalTokens = sums.totalTokens!;\n  }\n  for (const key of USAGE_KEYS) {\n    if (key === \"totalTokens\") continue;\n    if (present[key]) {\n      result[key] = sums[key]!;\n    }\n  }\n  return result;\n}\n\nexport function getThreadMessageTokenUsage(\n  message: TokenUsageExtractableMessage | undefined,\n): ThreadTokenUsage | undefined {\n  if (!message || message.role !== \"assistant\") return undefined;\n\n  const metadata = asRecord(message.metadata);\n  if (!metadata) return undefined;\n\n  const topLevelUsage = normalizeUsage(metadata.usage);\n  if (topLevelUsage) return withComputedTotal(topLevelUsage);\n\n  const customUsage = normalizeUsage(asRecord(metadata.custom)?.usage);\n  if (customUsage) return withComputedTotal(customUsage);\n\n  return usageFromSteps(metadata.steps);\n}\n\nfunction findLatestMessageWithUsage(\n  messages: readonly TokenUsageExtractableMessage[] | undefined,\n): TokenUsageExtractableMessage | undefined {\n  if (!messages) return undefined;\n\n  for (let idx = messages.length - 1; idx >= 0; idx -= 1) {\n    const message = messages[idx];\n    if (getThreadMessageTokenUsage(message)) {\n      return message;\n    }\n  }\n\n  return undefined;\n}\n\n/**\n * Reads token usage from the newest assistant message that reports any.\n *\n * A route attaches usage through the AI SDK's `messageMetadata` option. Because\n * a thread message carries a fixed metadata shape, the converter moves every\n * other key the route returns into `metadata.custom`, which is where this hook\n * looks.\n */\nexport function useThreadTokenUsage(): ThreadTokenUsage | undefined {\n  const msg = useAuiState((s) => findLatestMessageWithUsage(s.thread.messages));\n  return useMemo(() => getThreadMessageTokenUsage(msg), [msg]);\n}\n"],"mappings":";;;;AAmBA,MAAMa,aAAa;CACjB;CACA;CACA;CACA;CACA;AAAa;AAGf,SAASE,SAASC,OAAyC;CACzD,IAAI,CAACA,SAAS,OAAOA,UAAU,YAAYC,MAAMC,QAAQF,KAAK,GAC5D,OAAOG,KAAAA;CACT,OAAOH;AACT;AAEA,SAASI,qBAAqBJ,OAAoC;CAChE,IAAI,OAAOA,UAAU,YAAY,CAACK,OAAOC,SAASN,KAAK,KAAKA,QAAQ,GAClE;CAEF,OAAOA;AACT;AAEA,SAASO,mBAAmBC,OAA6C;CACvE,IAAIA,MAAMrB,gBAAgBgB,KAAAA,GAAW,OAAOK,MAAMrB;CAClD,IAAIqB,MAAMpB,gBAAgBe,KAAAA,KAAaK,MAAMnB,iBAAiBc,KAAAA,GAC5D,OAAOK,MAAMpB,cAAcoB,MAAMnB;AAGrC;AAEA,SAASoB,eAAeT,OAA8C;CACpE,MAAMU,SAASX,SAASC,KAAK;CAC7B,IAAI,CAACU,QAAQ,OAAOP,KAAAA;CAEpB,MAAMQ,SAA2B,CAAC;CAClC,IAAIC,YAAY;CAChB,KAAK,MAAMC,OAAOhB,YAAY;EAC5B,MAAMiB,QAAQV,qBAAqBM,OAAOG,IAAI;EAC9C,IAAIC,UAAUX,KAAAA,GAAW;GACvBQ,OAAOE,OAAOC;GACdF,YAAY;EACd;CACF;CAEA,IAAID,OAAOrB,oBAAoBa,KAAAA,GAAW;EACxC,MAAMW,QAAQV,qBACZL,SAASW,OAAOK,kBAAkB,CAAC,EAAEzB,eACvC;EACA,IAAIwB,UAAUX,KAAAA,GAAW;GACvBQ,OAAOrB,kBAAkBwB;GACzBF,YAAY;EACd;CACF;CACA,IAAID,OAAOpB,sBAAsBY,KAAAA,GAAW;EAC1C,MAAMW,QAAQV,qBACZL,SAASW,OAAOM,iBAAiB,CAAC,EAAEC,eACtC;EACA,IAAIH,UAAUX,KAAAA,GAAW;GACvBQ,OAAOpB,oBAAoBuB;GAC3BF,YAAY;EACd;CACF;CACA,OAAOA,YAAYD,SAASR,KAAAA;AAC9B;AAEA,SAASe,kBACPV,OAC8B;CAC9B,MAAMrB,cAAcoB,mBAAmBC,KAAK;CAC5C,OAAO;EAAE,GAAGA;EAAO,GAAIrB,gBAAgBgB,KAAAA,KAAa,EAAEhB,YAAY;CAAG;AACvE;AAEA,SAASgC,eAAenB,OAA8C;CACpE,MAAMoB,QAAQnB,MAAMC,QAAQF,KAAK,IAAIA,QAAQ,CAAA;CAE7C,MAAMqB,OAA+B,CAAC;CACtC,MAAMC,UAAmC,CAAC;CAC1C,IAAIC,iBAAiB;CACrB,IAAIC,2BAA2B;CAE/B,KAAK,MAAMC,QAAQL,OAAO;EACxB,MAAMZ,QAAQC,eAAeV,SAAS0B,IAAI,CAAC,EAAEjB,KAAK;EAClD,IAAI,CAACA,OAAO;EACZe;EAEA,MAAMG,YAAYnB,mBAAmBC,KAAK;EAC1C,IAAIkB,cAAcvB,KAAAA,GAAW;GAC3BkB,KAAKlC,eAAekC,KAAKlC,eAAe,KAAKuC;GAC7CF;EACF;EAEA,KAAK,MAAMX,OAAOhB,YAAY;GAC5B,IAAIgB,QAAQ,eAAe;GAC3B,IAAIL,MAAMK,SAASV,KAAAA,GAAW;IAC5BkB,KAAKR,QAAQQ,KAAKR,QAAQ,KAAKL,MAAMK;IACrCS,QAAQT,OAAO;GACjB;EACF;CACF;CAEA,IAAIU,mBAAmB,GAAG,OAAOpB,KAAAA;CAEjC,MAAMQ,SAA2B,CAAC;CAClC,IAAIa,6BAA6BD,gBAC/BZ,OAAOxB,cAAckC,KAAKlC;CAE5B,KAAK,MAAM0B,OAAOhB,YAAY;EAC5B,IAAIgB,QAAQ,eAAe;EAC3B,IAAIS,QAAQT,MACVF,OAAOE,OAAOQ,KAAKR;CAEvB;CACA,OAAOF;AACT;AAEA,SAAgBgB,2BACdC,SAC8B;CAC9B,IAAI,CAACA,WAAWA,QAAQnC,SAAS,aAAa,OAAOU,KAAAA;CAErD,MAAMT,WAAWK,SAAS6B,QAAQlC,QAAQ;CAC1C,IAAI,CAACA,UAAU,OAAOS,KAAAA;CAEtB,MAAM0B,gBAAgBpB,eAAef,SAASc,KAAK;CACnD,IAAIqB,eAAe,OAAOX,kBAAkBW,aAAa;CAEzD,MAAMC,cAAcrB,eAAeV,SAASL,SAASqC,MAAM,CAAC,EAAEvB,KAAK;CACnE,IAAIsB,aAAa,OAAOZ,kBAAkBY,WAAW;CAErD,OAAOX,eAAezB,SAAS0B,KAAK;AACtC;AAEA,SAASY,2BACPC,UAC0C;CAC1C,IAAI,CAACA,UAAU,OAAO9B,KAAAA;CAEtB,KAAK,IAAI+B,MAAMD,SAASE,SAAS,GAAGD,OAAO,GAAGA,OAAO,GAAG;EACtD,MAAMN,UAAUK,SAASC;EACzB,IAAIP,2BAA2BC,OAAO,GACpC,OAAOA;CAEX;AAGF;;;;;;;;;AAUA,SAAOQ,sBAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CACL,MAAAC,MAAYtD,YAAYuD,KAAoD;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAAE,KAAA;EACzDE,KAAAd,2BAA2BY,GAAG;EAACF,EAAA,KAAAE;EAAAF,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,OAA/BI;AAA+B;AAF/C,SAAAD,MAAAE,GAAA;CAAA,OAC0BV,2BAA2BU,EAACC,OAAOV,QAAS;AAAC"}

package/dist/utils/getVercelAIMessages.d.ts

@@ -1,6 +1,4 @@
-import { UIMessage } from "ai";
-import { ThreadMessage } from "@assistant-ui/core";
-//#region src/utils/getVercelAIMessages.d.ts
+import { type ThreadMessage } from "@assistant-ui/core";
+import type { UIMessage } from "ai";
 export declare const getVercelAIMessages: <UI_MESSAGE extends UIMessage = UIMessage>(message: ThreadMessage) => UI_MESSAGE[];
-//#endregion
 //# sourceMappingURL=getVercelAIMessages.d.ts.map

package/dist/utils/getVercelAIMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"getVercelAIMessages.d.ts","names":[],"sources":["../../src/utils/getVercelAIMessages.ts"],"mappings":";;;qBAMa,sBAAuB,mBAAmB,YAAY,WACjE,SAAS,kBAEmC"}
+{"version":3,"file":"getVercelAIMessages.d.ts","sourceRoot":"","sources":["../../src/utils/getVercelAIMessages.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,aAAa,EACnB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAEpC,eAAO,MAAM,mBAAmB,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EAC1E,SAAS,aAAa,KAEsB,UAAU,EACvD,CAAC"}

package/dist/utils/sliceMessagesUntil.d.ts

@@ -1,5 +1,3 @@
-import { UIMessage } from "ai";
-//#region src/utils/sliceMessagesUntil.d.ts
+import type { UIMessage } from "ai";
 export declare const sliceMessagesUntil: <UI_MESSAGE extends UIMessage = UIMessage>(messages: UI_MESSAGE[], messageId: string | null) => UI_MESSAGE[];
-//#endregion
 //# sourceMappingURL=sliceMessagesUntil.d.ts.map

package/dist/utils/sliceMessagesUntil.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sliceMessagesUntil.d.ts","names":[],"sources":["../../src/utils/sliceMessagesUntil.ts"],"mappings":";;qBAEa,qBAAsB,mBAAmB,YAAY,WAChE,UAAU,cACV,6BAAwB"}
+{"version":3,"file":"sliceMessagesUntil.d.ts","sourceRoot":"","sources":["../../src/utils/sliceMessagesUntil.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAEpC,eAAO,MAAM,kBAAkB,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EACzE,UAAU,UAAU,EAAE,EACtB,WAAW,MAAM,GAAG,IAAI,iBAezB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/ai-sdk",
-  "version": "0.0.7",
+  "version": "0.0.8",
   "description": "Vercel AI SDK adapter for assistant-ui",
   "keywords": [
     "ai-sdk",
@@ -48,15 +48,15 @@
   "dependencies": {
     "@ai-sdk/mcp": "^2.0.49",
     "@ai-sdk/react": "^4.0.104",
-    "@assistant-ui/core": "^0.3.20",
-    "@assistant-ui/store": "^0.3.14",
-    "@assistant-ui/tap": "^0.9.18",
-    "ai": "^7.0.101",
-    "assistant-cloud": "^0.2.2",
-    "assistant-stream": "^0.3.44"
+    "@assistant-ui/core": "^0.3.21",
+    "@assistant-ui/store": "^0.3.15",
+    "@assistant-ui/tap": "^0.9.19",
+    "assistant-cloud": "^0.2.3",
+    "assistant-stream": "^0.3.45"
   },
   "peerDependencies": {
     "@types/react": "*",
+    "ai": "^7.0.101",
     "react": "^18 || ^19"
   },
   "peerDependenciesMeta": {
@@ -68,11 +68,12 @@
     }
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.28",
+    "@assistant-ui/x-buildutils": "0.0.29",
     "@testing-library/react": "^16.3.3",
     "@types/json-schema": "^7.0.15",
     "@types/react": "^19.3.0",
     "@types/react-dom": "^19.3.0",
+    "ai": "^7.0.101",
     "jsdom": "^30.0.1",
     "react": "^19.3.0",
     "vitest": "^5.0.1"

package/src/adapters/vercelAttachmentAdapter.test.ts

@@ -26,4 +26,27 @@ describe("vercelAttachmentAdapter", () => {
     if (part?.type !== "file") throw new Error("expected file content part");
     expect(part.data).toBe("data:text/plain;base64,aGVsbG8=");
   });
+
+  it("aborts a file send via the shared core encoder", async () => {
+    globalThis.FileReader = undefined as unknown as typeof FileReader;
+
+    const file = new File(["hello"], "a.txt", { type: "text/plain" });
+    const controller = new AbortController();
+    controller.abort();
+
+    await expect(
+      vercelAttachmentAdapter.send(
+        {
+          id: "1",
+          type: "file",
+          name: file.name,
+          file,
+          contentType: file.type,
+          content: [],
+          status: { type: "requires-action", reason: "composer-send" },
+        },
+        { signal: controller.signal },
+      ),
+    ).rejects.toBe(controller.signal.reason);
+  });
 });

package/src/adapters/vercelAttachmentAdapter.ts

@@ -15,7 +15,7 @@ export const vercelAttachmentAdapter: AttachmentAdapter = {
       status: { type: "requires-action", reason: "composer-send" },
     };
   },
-  async send(attachment) {
+  async send(attachment, options) {
     // noop
     return {
       ...attachment,
@@ -25,7 +25,7 @@ export const vercelAttachmentAdapter: AttachmentAdapter = {
           type: "file",
           mimeType: attachment.contentType ?? "",
           filename: attachment.name,
-          data: await getFileDataURL(attachment.file),
+          data: await getFileDataURL(attachment.file, options),
         },
       ],
     };