@assistant-ui/ai-sdk 0.0.7 → 0.0.9

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;AAYF,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/AssistantChatTransport.js

@@ -5,6 +5,7 @@ import { toToolsJSONSchema } from "assistant-stream";
 const FINISH_MARKER = "\"type\":\"finish\"";
 const FINISH_BUFFER_LIMIT = 4096;
 const RESUMABLE_THREAD_ID_HEADER = "x-assistant-ui-resumable-thread-id";
+const RESUMABLE_RECONNECT_ID_HEADER = "x-assistant-ui-resumable-reconnect-id";
 const NULL_BODY_STATUSES = /* @__PURE__ */ new Set([
 	101,
 	204,
@@ -81,13 +82,18 @@ function wrapFetchWithResumable(resumable, userFetch) {
 	return async (input, init) => {
 		const headers = new Headers(init?.headers);
 		const threadId = headers.get(RESUMABLE_THREAD_ID_HEADER) ?? void 0;
+		const reconnectingStreamId = headers.get(RESUMABLE_RECONNECT_ID_HEADER);
+		const checkpointAtRequest = reconnectingStreamId ?? resumable.storage.getStreamId(threadId);
 		headers.delete(RESUMABLE_THREAD_ID_HEADER);
+		headers.delete(RESUMABLE_RECONNECT_ID_HEADER);
 		const res = await baseFetch(input, {
 			...init,
 			headers
 		});
 		const id = res.headers.get(RESUMABLE_STREAM_ID_HEADER);
-		if (id) resumable.storage.setStreamId(id, threadId);
+		const ownsCheckpoint = !reconnectingStreamId || resumable.storage.getStreamId(threadId) === reconnectingStreamId;
+		if (id && ownsCheckpoint) resumable.storage.setStreamId(id, threadId);
+		if ((res.status === 204 || res.status === 404) && reconnectingStreamId && resumable.storage.getStreamId(threadId) === reconnectingStreamId) resumable.storage.clear(threadId);
 		if (!res.body || NULL_BODY_STATUSES.has(res.status)) return res;
 		const detectFinish = resumable.isFinishEvent ?? defaultIsFinishEvent;
 		const decoder = new TextDecoder();
@@ -96,7 +102,7 @@ function wrapFetchWithResumable(resumable, userFetch) {
 			controller.enqueue(chunk);
 			accumulator += decoder.decode(chunk, { stream: true });
 			if (detectFinish(chunk, accumulator)) {
-				if (!id || resumable.storage.getStreamId(threadId) === id) resumable.storage.clear(threadId);
+				if (ownsCheckpoint && resumable.storage.getStreamId(threadId) === (id ?? checkpointAtRequest)) resumable.storage.clear(threadId);
 				accumulator = "";
 			} else if (accumulator.length > FINISH_BUFFER_LIMIT) accumulator = accumulator.slice(-1024);
 		} });
@@ -121,6 +127,7 @@ function wrapPrepareReconnect(resumable, userPrepareReconnect) {
 		});
 		const headers = new Headers(userPrepared?.headers ?? options.headers);
 		headers.set(RESUMABLE_THREAD_ID_HEADER, options.id);
+		headers.set(RESUMABLE_RECONNECT_ID_HEADER, streamId);
 		return {
 			...userPrepared,
 			headers,

package/dist/transport/AssistantChatTransport.js.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantChatTransport.js","names":["AssistantRuntime","ThreadListItemRuntime","DefaultChatTransport","HttpChatTransportInitOptions","UIMessage","toToolsJSONSchema","RESUMABLE_STREAM_ID_HEADER","AssistantChatResumableOptions","InitializableThreadListItem","Pick","FINISH_MARKER","FINISH_BUFFER_LIMIT","FINISH_BUFFER_TAIL","RESUMABLE_THREAD_ID_HEADER","NULL_BODY_STATUSES","Set","AssistantChatTransportInitOptions","UI_MESSAGE","resumable","AssistantChatTransport","runtime","getThreadListItem","__internal_initOptions","__internal_clone","Constructor","constructor","initOptions","rest","userFetch","fetch","userPrepareReconnect","prepareReconnectToStreamRequest","wrapFetchWithResumable","wrapPrepareReconnect","prepareSendMessagesRequest","options","threadId","id","context","thread","getModelContext","threadListItem","threads","mainItem","initialize","remoteId","optionsEx","body","callSettings","system","config","tools","preparedRequest","headers","Headers","undefined","set","messages","trigger","messageId","metadata","requestMetadata","setRuntime","getResumableAdapter","__internal_setGetThreadListItem","getter","NonNullable","baseFetch","globalThis","input","init","RequestInfo","URL","bind","get","delete","res","storage","setStreamId","has","status","detectFinish","isFinishEvent","defaultIsFinishEvent","decoder","TextDecoder","accumulator","tap","TransformStream","Uint8Array","transform","chunk","controller","enqueue","decode","stream","getStreamId","clear","length","slice","Response","pipeThrough","statusText","_chunk","includes","streamId","Error","api","resumeApi","userPrepared"],"sources":["../../src/transport/AssistantChatTransport.ts"],"sourcesContent":["import type {\n  AssistantRuntime,\n  ThreadListItemRuntime,\n} from \"@assistant-ui/core\";\nimport {\n  DefaultChatTransport,\n  type HttpChatTransportInitOptions,\n  type UIMessage,\n} from \"ai\";\nimport { toToolsJSONSchema } from \"assistant-stream\";\nimport {\n  RESUMABLE_STREAM_ID_HEADER,\n  type AssistantChatResumableOptions,\n} from \"./resumable\";\n\nexport type InitializableThreadListItem = Pick<\n  ThreadListItemRuntime,\n  \"initialize\"\n>;\n\nconst FINISH_MARKER = '\"type\":\"finish\"';\nconst FINISH_BUFFER_LIMIT = 4096;\nconst FINISH_BUFFER_TAIL = 1024;\nconst RESUMABLE_THREAD_ID_HEADER = \"x-assistant-ui-resumable-thread-id\";\n\n// 101/204/205/304 are null-body statuses per the fetch spec: `new Response(body, { status })`\n// throws for them, and WebKit returns a non-null empty body, so the body check alone does not guard it.\nconst NULL_BODY_STATUSES = new Set([101, 204, 205, 304]);\n\nexport type AssistantChatTransportInitOptions<UI_MESSAGE extends UIMessage> =\n  HttpChatTransportInitOptions<UI_MESSAGE> & {\n    resumable?: AssistantChatResumableOptions;\n  };\n\nexport class AssistantChatTransport<\n  UI_MESSAGE extends UIMessage,\n> extends DefaultChatTransport<UI_MESSAGE> {\n  private runtime: AssistantRuntime | undefined;\n  private getThreadListItem:\n    | (() => InitializableThreadListItem | undefined)\n    | undefined;\n  private readonly resumable: AssistantChatResumableOptions | undefined;\n\n  private readonly __internal_initOptions:\n    | AssistantChatTransportInitOptions<UI_MESSAGE>\n    | undefined;\n\n  /** Constructs an unwired copy with the same init options. */\n  public __internal_clone(): AssistantChatTransport<UI_MESSAGE> {\n    const Constructor = this.constructor as new (\n      initOptions?: AssistantChatTransportInitOptions<UI_MESSAGE>,\n    ) => AssistantChatTransport<UI_MESSAGE>;\n    return new Constructor(this.__internal_initOptions);\n  }\n\n  constructor(initOptions?: AssistantChatTransportInitOptions<UI_MESSAGE>) {\n    const { resumable, ...rest } = initOptions ?? {};\n    const userFetch = rest.fetch;\n    const userPrepareReconnect = rest.prepareReconnectToStreamRequest;\n\n    super({\n      ...rest,\n      ...(resumable && {\n        fetch: wrapFetchWithResumable(resumable, userFetch),\n        prepareReconnectToStreamRequest: wrapPrepareReconnect(\n          resumable,\n          userPrepareReconnect,\n        ),\n      }),\n      prepareSendMessagesRequest: async (options) => {\n        const threadId = options.id;\n        const context = this.runtime?.thread.getModelContext();\n        const threadListItem =\n          this.getThreadListItem?.() ?? this.runtime?.threads.mainItem;\n        const id = (await threadListItem?.initialize())?.remoteId ?? options.id;\n\n        const optionsEx = {\n          ...options,\n          id,\n          body: {\n            callSettings: context?.callSettings,\n            system: context?.system,\n            config: context?.config,\n            tools: toToolsJSONSchema(context?.tools ?? {}),\n            ...options?.body,\n          },\n        };\n        const preparedRequest =\n          await rest.prepareSendMessagesRequest?.(optionsEx);\n        const headers = resumable\n          ? new Headers(preparedRequest?.headers ?? options.headers)\n          : undefined;\n        headers?.set(RESUMABLE_THREAD_ID_HEADER, threadId);\n\n        return {\n          ...preparedRequest,\n          ...(headers && { headers }),\n          body: preparedRequest?.body ?? {\n            ...optionsEx.body,\n            id,\n            messages: options.messages,\n            trigger: options.trigger,\n            messageId: options.messageId,\n            metadata: options.requestMetadata,\n          },\n        };\n      },\n    });\n\n    this.resumable = resumable;\n    this.__internal_initOptions = initOptions;\n  }\n\n  setRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  getResumableAdapter(): AssistantChatResumableOptions | undefined {\n    return this.resumable;\n  }\n\n  __internal_setGetThreadListItem(\n    getter: () => InitializableThreadListItem | undefined,\n  ) {\n    this.getThreadListItem = getter;\n  }\n}\n\nfunction wrapFetchWithResumable(\n  resumable: AssistantChatResumableOptions,\n  userFetch: HttpChatTransportInitOptions<UIMessage>[\"fetch\"],\n): NonNullable<HttpChatTransportInitOptions<UIMessage>[\"fetch\"]> {\n  const baseFetch: typeof globalThis.fetch = userFetch\n    ? (input, init) => userFetch(input as RequestInfo | URL, init)\n    : globalThis.fetch.bind(globalThis);\n\n  return async (input, init) => {\n    const headers = new Headers(init?.headers);\n    const threadId = headers.get(RESUMABLE_THREAD_ID_HEADER) ?? undefined;\n    headers.delete(RESUMABLE_THREAD_ID_HEADER);\n    const res = await baseFetch(input, { ...init, headers });\n    const id = res.headers.get(RESUMABLE_STREAM_ID_HEADER);\n    if (id) resumable.storage.setStreamId(id, threadId);\n    if (!res.body || NULL_BODY_STATUSES.has(res.status)) return res;\n\n    const detectFinish = resumable.isFinishEvent ?? defaultIsFinishEvent;\n    // a single decoder is required so multi-byte sequences split across\n    // chunks buffer via stream: true rather than getting dropped.\n    const decoder = new TextDecoder();\n    let accumulator = \"\";\n    const tap = new TransformStream<Uint8Array, Uint8Array>({\n      transform(chunk, controller) {\n        controller.enqueue(chunk);\n        accumulator += decoder.decode(chunk, { stream: true });\n        if (detectFinish(chunk, accumulator)) {\n          if (!id || resumable.storage.getStreamId(threadId) === id) {\n            resumable.storage.clear(threadId);\n          }\n          accumulator = \"\";\n        } else if (accumulator.length > FINISH_BUFFER_LIMIT) {\n          accumulator = accumulator.slice(-FINISH_BUFFER_TAIL);\n        }\n      },\n    });\n\n    return new Response(res.body.pipeThrough(tap), {\n      status: res.status,\n      statusText: res.statusText,\n      headers: res.headers,\n    });\n  };\n}\n\nfunction defaultIsFinishEvent(_chunk: Uint8Array, accumulator: string) {\n  return accumulator.includes(FINISH_MARKER);\n}\n\nfunction wrapPrepareReconnect(\n  resumable: AssistantChatResumableOptions,\n  userPrepareReconnect: HttpChatTransportInitOptions<UIMessage>[\"prepareReconnectToStreamRequest\"],\n): NonNullable<\n  HttpChatTransportInitOptions<UIMessage>[\"prepareReconnectToStreamRequest\"]\n> {\n  return async (options) => {\n    const streamId = resumable.storage.getStreamId(options.id);\n    if (!streamId) {\n      throw new Error(\n        \"AssistantChatTransport: no resumable stream id available; nothing to resume\",\n      );\n    }\n    const api =\n      typeof resumable.resumeApi === \"function\"\n        ? resumable.resumeApi(streamId)\n        : resumable.resumeApi;\n    const userPrepared = await userPrepareReconnect?.({ ...options, api });\n    const headers = new Headers(userPrepared?.headers ?? options.headers);\n    headers.set(RESUMABLE_THREAD_ID_HEADER, options.id);\n    return {\n      ...userPrepared,\n      headers,\n      api: userPrepared?.api ?? api,\n    };\n  };\n}\n"],"mappings":";;;;AAoBA,MAAMU,gBAAgB;AACtB,MAAMC,sBAAsB;AAE5B,MAAME,6BAA6B;AAInC,MAAMC,qCAAqB,IAAIC,IAAI;CAAC;CAAK;CAAK;CAAK;AAAG,CAAC;AAOvD,IAAaI,yBAAb,cAEUjB,qBAAiC;CACzC;CACA;CAGA;CAEA;;CAKA,mBAA8D;EAC5D,MAAMsB,cAAc,KAAKC;EAGzB,OAAO,IAAID,YAAY,KAAKF,sBAAsB;CACpD;CAEAG,YAAYC,aAA6D;EACvE,MAAM,EAAER,WAAW,GAAGS,SAASD,eAAe,CAAC;EAC/C,MAAME,YAAYD,KAAKE;EACvB,MAAMC,uBAAuBH,KAAKI;EAElC,MAAM;GACJ,GAAGJ;GACH,GAAIT,aAAa;IACfW,OAAOG,uBAAuBd,WAAWU,SAAS;IAClDG,iCAAiCE,qBAC/Bf,WACAY,oBACF;GACF;GACAI,4BAA4B,OAAOC,YAAY;IAC7C,MAAMC,WAAWD,QAAQE;IACzB,MAAMC,UAAU,KAAKlB,SAASmB,OAAOC,gBAAgB;IAGrD,MAAMH,MAAM,OADV,KAAKhB,oBAAoB,KAAK,KAAKD,SAASsB,QAAQC,SAAAA,EACpBC,WAAW,EAAA,EAAIC,YAAYV,QAAQE;IAErE,MAAMS,YAAY;KAChB,GAAGX;KACHE;KACAU,MAAM;MACJC,cAAcV,SAASU;MACvBC,QAAQX,SAASW;MACjBC,QAAQZ,SAASY;MACjBC,OAAO9C,kBAAkBiC,SAASa,SAAS,CAAC,CAAC;MAC7C,GAAGhB,SAASY;KACd;IACF;IACA,MAAMK,kBACJ,MAAMzB,KAAKO,6BAA6BY,SAAS;IACnD,MAAMO,UAAUnC,YACZ,IAAIoC,QAAQF,iBAAiBC,WAAWlB,QAAQkB,OAAO,IACvDE,KAAAA;IACJF,SAASG,IAAI3C,4BAA4BuB,QAAQ;IAEjD,OAAO;KACL,GAAGgB;KACH,GAAIC,WAAW,EAAEA,QAAQ;KACzBN,MAAMK,iBAAiBL,QAAQ;MAC7B,GAAGD,UAAUC;MACbV;MACAoB,UAAUtB,QAAQsB;MAClBC,SAASvB,QAAQuB;MACjBC,WAAWxB,QAAQwB;MACnBC,UAAUzB,QAAQ0B;KACpB;IACF;GACF;EACF,CAAC;EAED,KAAK3C,YAAYA;EACjB,KAAKI,yBAAyBI;CAChC;CAEAoC,WAAW1C,SAA2B;EACpC,KAAKA,UAAUA;CACjB;CAEA2C,sBAAiE;EAC/D,OAAO,KAAK7C;CACd;CAEA8C,gCACEC,QACA;EACA,KAAK5C,oBAAoB4C;CAC3B;AACF;AAEA,SAASjC,uBACPd,WACAU,WAC+D;CAC/D,MAAMuC,YAAqCvC,aACtCyC,OAAOC,SAAS1C,UAAUyC,OAA4BC,IAAI,IAC3DF,WAAWvC,MAAM4C,KAAKL,UAAU;CAEpC,OAAO,OAAOC,OAAOC,SAAS;EAC5B,MAAMjB,UAAU,IAAIC,QAAQgB,MAAMjB,OAAO;EACzC,MAAMjB,WAAWiB,QAAQqB,IAAI7D,0BAA0B,KAAK0C,KAAAA;EAC5DF,QAAQsB,OAAO9D,0BAA0B;EACzC,MAAM+D,MAAM,MAAMT,UAAUE,OAAO;GAAE,GAAGC;GAAMjB;EAAQ,CAAC;EACvD,MAAMhB,KAAKuC,IAAIvB,QAAQqB,IAAIpE,0BAA0B;EACrD,IAAI+B,IAAInB,UAAU2D,QAAQC,YAAYzC,IAAID,QAAQ;EAClD,IAAI,CAACwC,IAAI7B,QAAQjC,mBAAmBiE,IAAIH,IAAII,MAAM,GAAG,OAAOJ;EAE5D,MAAMK,eAAe/D,UAAUgE,iBAAiBC;EAGhD,MAAMC,UAAU,IAAIC,YAAY;EAChC,IAAIC,cAAc;EAClB,MAAMC,MAAM,IAAIC,gBAAwC,EACtDE,UAAUC,OAAOC,YAAY;GAC3BA,WAAWC,QAAQF,KAAK;GACxBL,eAAeF,QAAQU,OAAOH,OAAO,EAAEI,QAAQ,KAAK,CAAC;GACrD,IAAId,aAAaU,OAAOL,WAAW,GAAG;IACpC,IAAI,CAACjD,MAAMnB,UAAU2D,QAAQmB,YAAY5D,QAAQ,MAAMC,IACrDnB,UAAU2D,QAAQoB,MAAM7D,QAAQ;IAElCkD,cAAc;GAChB,OAAO,IAAIA,YAAYY,SAASvF,qBAC9B2E,cAAcA,YAAYa,MAAM,KAAmB;EAEvD,EACF,CAAC;EAED,OAAO,IAAIC,SAASxB,IAAI7B,KAAKsD,YAAYd,GAAG,GAAG;GAC7CP,QAAQJ,IAAII;GACZsB,YAAY1B,IAAI0B;GAChBjD,SAASuB,IAAIvB;EACf,CAAC;CACH;AACF;AAEA,SAAS8B,qBAAqBoB,QAAoBjB,aAAqB;CACrE,OAAOA,YAAYkB,SAAS9F,aAAa;AAC3C;AAEA,SAASuB,qBACPf,WACAY,sBAGA;CACA,OAAO,OAAOK,YAAY;EACxB,MAAMsE,WAAWvF,UAAU2D,QAAQmB,YAAY7D,QAAQE,EAAE;EACzD,IAAI,CAACoE,UACH,MAAM,IAAIC,MACR,6EACF;EAEF,MAAMC,MACJ,OAAOzF,UAAU0F,cAAc,aAC3B1F,UAAU0F,UAAUH,QAAQ,IAC5BvF,UAAU0F;EAChB,MAAMC,eAAe,MAAM/E,uBAAuB;GAAE,GAAGK;GAASwE;EAAI,CAAC;EACrE,MAAMtD,UAAU,IAAIC,QAAQuD,cAAcxD,WAAWlB,QAAQkB,OAAO;EACpEA,QAAQG,IAAI3C,4BAA4BsB,QAAQE,EAAE;EAClD,OAAO;GACL,GAAGwE;GACHxD;GACAsD,KAAKE,cAAcF,OAAOA;EAC5B;CACF;AACF"}
+{"version":3,"file":"AssistantChatTransport.js","names":["AssistantRuntime","ThreadListItemRuntime","DefaultChatTransport","HttpChatTransportInitOptions","UIMessage","toToolsJSONSchema","RESUMABLE_STREAM_ID_HEADER","AssistantChatResumableOptions","InitializableThreadListItem","Pick","FINISH_MARKER","FINISH_BUFFER_LIMIT","FINISH_BUFFER_TAIL","RESUMABLE_THREAD_ID_HEADER","RESUMABLE_RECONNECT_ID_HEADER","NULL_BODY_STATUSES","Set","AssistantChatTransportInitOptions","UI_MESSAGE","resumable","AssistantChatTransport","runtime","getThreadListItem","__internal_initOptions","__internal_clone","Constructor","constructor","initOptions","rest","userFetch","fetch","userPrepareReconnect","prepareReconnectToStreamRequest","wrapFetchWithResumable","wrapPrepareReconnect","prepareSendMessagesRequest","options","threadId","id","context","thread","getModelContext","threadListItem","threads","mainItem","initialize","remoteId","optionsEx","body","callSettings","system","config","tools","preparedRequest","headers","Headers","undefined","set","messages","trigger","messageId","metadata","requestMetadata","setRuntime","getResumableAdapter","__internal_setGetThreadListItem","getter","NonNullable","baseFetch","globalThis","input","init","RequestInfo","URL","bind","get","reconnectingStreamId","checkpointAtRequest","storage","getStreamId","delete","res","ownsCheckpoint","setStreamId","status","clear","has","detectFinish","isFinishEvent","defaultIsFinishEvent","decoder","TextDecoder","accumulator","tap","TransformStream","Uint8Array","transform","chunk","controller","enqueue","decode","stream","length","slice","Response","pipeThrough","statusText","_chunk","includes","streamId","Error","api","resumeApi","userPrepared"],"sources":["../../src/transport/AssistantChatTransport.ts"],"sourcesContent":["import type {\n  AssistantRuntime,\n  ThreadListItemRuntime,\n} from \"@assistant-ui/core\";\nimport {\n  DefaultChatTransport,\n  type HttpChatTransportInitOptions,\n  type UIMessage,\n} from \"ai\";\nimport { toToolsJSONSchema } from \"assistant-stream\";\nimport {\n  RESUMABLE_STREAM_ID_HEADER,\n  type AssistantChatResumableOptions,\n} from \"./resumable\";\n\nexport type InitializableThreadListItem = Pick<\n  ThreadListItemRuntime,\n  \"initialize\"\n>;\n\nconst FINISH_MARKER = '\"type\":\"finish\"';\nconst FINISH_BUFFER_LIMIT = 4096;\nconst FINISH_BUFFER_TAIL = 1024;\nconst RESUMABLE_THREAD_ID_HEADER = \"x-assistant-ui-resumable-thread-id\";\nconst RESUMABLE_RECONNECT_ID_HEADER = \"x-assistant-ui-resumable-reconnect-id\";\n\n// 101/204/205/304 are null-body statuses per the fetch spec: `new Response(body, { status })`\n// throws for them, and WebKit returns a non-null empty body, so the body check alone does not guard it.\nconst NULL_BODY_STATUSES = new Set([101, 204, 205, 304]);\n\nexport type AssistantChatTransportInitOptions<UI_MESSAGE extends UIMessage> =\n  HttpChatTransportInitOptions<UI_MESSAGE> & {\n    resumable?: AssistantChatResumableOptions;\n  };\n\nexport class AssistantChatTransport<\n  UI_MESSAGE extends UIMessage,\n> extends DefaultChatTransport<UI_MESSAGE> {\n  private runtime: AssistantRuntime | undefined;\n  private getThreadListItem:\n    | (() => InitializableThreadListItem | undefined)\n    | undefined;\n  private readonly resumable: AssistantChatResumableOptions | undefined;\n\n  private readonly __internal_initOptions:\n    | AssistantChatTransportInitOptions<UI_MESSAGE>\n    | undefined;\n\n  /** Constructs an unwired copy with the same init options. */\n  public __internal_clone(): AssistantChatTransport<UI_MESSAGE> {\n    const Constructor = this.constructor as new (\n      initOptions?: AssistantChatTransportInitOptions<UI_MESSAGE>,\n    ) => AssistantChatTransport<UI_MESSAGE>;\n    return new Constructor(this.__internal_initOptions);\n  }\n\n  constructor(initOptions?: AssistantChatTransportInitOptions<UI_MESSAGE>) {\n    const { resumable, ...rest } = initOptions ?? {};\n    const userFetch = rest.fetch;\n    const userPrepareReconnect = rest.prepareReconnectToStreamRequest;\n\n    super({\n      ...rest,\n      ...(resumable && {\n        fetch: wrapFetchWithResumable(resumable, userFetch),\n        prepareReconnectToStreamRequest: wrapPrepareReconnect(\n          resumable,\n          userPrepareReconnect,\n        ),\n      }),\n      prepareSendMessagesRequest: async (options) => {\n        const threadId = options.id;\n        const context = this.runtime?.thread.getModelContext();\n        const threadListItem =\n          this.getThreadListItem?.() ?? this.runtime?.threads.mainItem;\n        const id = (await threadListItem?.initialize())?.remoteId ?? options.id;\n\n        const optionsEx = {\n          ...options,\n          id,\n          body: {\n            callSettings: context?.callSettings,\n            system: context?.system,\n            config: context?.config,\n            tools: toToolsJSONSchema(context?.tools ?? {}),\n            ...options?.body,\n          },\n        };\n        const preparedRequest =\n          await rest.prepareSendMessagesRequest?.(optionsEx);\n        const headers = resumable\n          ? new Headers(preparedRequest?.headers ?? options.headers)\n          : undefined;\n        headers?.set(RESUMABLE_THREAD_ID_HEADER, threadId);\n\n        return {\n          ...preparedRequest,\n          ...(headers && { headers }),\n          body: preparedRequest?.body ?? {\n            ...optionsEx.body,\n            id,\n            messages: options.messages,\n            trigger: options.trigger,\n            messageId: options.messageId,\n            metadata: options.requestMetadata,\n          },\n        };\n      },\n    });\n\n    this.resumable = resumable;\n    this.__internal_initOptions = initOptions;\n  }\n\n  setRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  getResumableAdapter(): AssistantChatResumableOptions | undefined {\n    return this.resumable;\n  }\n\n  __internal_setGetThreadListItem(\n    getter: () => InitializableThreadListItem | undefined,\n  ) {\n    this.getThreadListItem = getter;\n  }\n}\n\nfunction wrapFetchWithResumable(\n  resumable: AssistantChatResumableOptions,\n  userFetch: HttpChatTransportInitOptions<UIMessage>[\"fetch\"],\n): NonNullable<HttpChatTransportInitOptions<UIMessage>[\"fetch\"]> {\n  const baseFetch: typeof globalThis.fetch = userFetch\n    ? (input, init) => userFetch(input as RequestInfo | URL, init)\n    : globalThis.fetch.bind(globalThis);\n\n  return async (input, init) => {\n    const headers = new Headers(init?.headers);\n    const threadId = headers.get(RESUMABLE_THREAD_ID_HEADER) ?? undefined;\n    const reconnectingStreamId = headers.get(RESUMABLE_RECONNECT_ID_HEADER);\n    const checkpointAtRequest =\n      reconnectingStreamId ?? resumable.storage.getStreamId(threadId);\n    headers.delete(RESUMABLE_THREAD_ID_HEADER);\n    headers.delete(RESUMABLE_RECONNECT_ID_HEADER);\n    const res = await baseFetch(input, { ...init, headers });\n    const id = res.headers.get(RESUMABLE_STREAM_ID_HEADER);\n    const ownsCheckpoint =\n      !reconnectingStreamId ||\n      resumable.storage.getStreamId(threadId) === reconnectingStreamId;\n    if (id && ownsCheckpoint) resumable.storage.setStreamId(id, threadId);\n    if (\n      (res.status === 204 || res.status === 404) &&\n      reconnectingStreamId &&\n      resumable.storage.getStreamId(threadId) === reconnectingStreamId\n    ) {\n      resumable.storage.clear(threadId);\n    }\n    if (!res.body || NULL_BODY_STATUSES.has(res.status)) return res;\n\n    const detectFinish = resumable.isFinishEvent ?? defaultIsFinishEvent;\n    // a single decoder is required so multi-byte sequences split across\n    // chunks buffer via stream: true rather than getting dropped.\n    const decoder = new TextDecoder();\n    let accumulator = \"\";\n    const tap = new TransformStream<Uint8Array, Uint8Array>({\n      transform(chunk, controller) {\n        controller.enqueue(chunk);\n        accumulator += decoder.decode(chunk, { stream: true });\n        if (detectFinish(chunk, accumulator)) {\n          if (\n            ownsCheckpoint &&\n            resumable.storage.getStreamId(threadId) ===\n              (id ?? checkpointAtRequest)\n          ) {\n            resumable.storage.clear(threadId);\n          }\n          accumulator = \"\";\n        } else if (accumulator.length > FINISH_BUFFER_LIMIT) {\n          accumulator = accumulator.slice(-FINISH_BUFFER_TAIL);\n        }\n      },\n    });\n\n    return new Response(res.body.pipeThrough(tap), {\n      status: res.status,\n      statusText: res.statusText,\n      headers: res.headers,\n    });\n  };\n}\n\nfunction defaultIsFinishEvent(_chunk: Uint8Array, accumulator: string) {\n  return accumulator.includes(FINISH_MARKER);\n}\n\nfunction wrapPrepareReconnect(\n  resumable: AssistantChatResumableOptions,\n  userPrepareReconnect: HttpChatTransportInitOptions<UIMessage>[\"prepareReconnectToStreamRequest\"],\n): NonNullable<\n  HttpChatTransportInitOptions<UIMessage>[\"prepareReconnectToStreamRequest\"]\n> {\n  return async (options) => {\n    const streamId = resumable.storage.getStreamId(options.id);\n    if (!streamId) {\n      throw new Error(\n        \"AssistantChatTransport: no resumable stream id available; nothing to resume\",\n      );\n    }\n    const api =\n      typeof resumable.resumeApi === \"function\"\n        ? resumable.resumeApi(streamId)\n        : resumable.resumeApi;\n    const userPrepared = await userPrepareReconnect?.({ ...options, api });\n    const headers = new Headers(userPrepared?.headers ?? options.headers);\n    headers.set(RESUMABLE_THREAD_ID_HEADER, options.id);\n    headers.set(RESUMABLE_RECONNECT_ID_HEADER, streamId);\n    return {\n      ...userPrepared,\n      headers,\n      api: userPrepared?.api ?? api,\n    };\n  };\n}\n"],"mappings":";;;;AAoBA,MAAMU,gBAAgB;AACtB,MAAMC,sBAAsB;AAE5B,MAAME,6BAA6B;AACnC,MAAMC,gCAAgC;AAItC,MAAMC,qCAAqB,IAAIC,IAAI;CAAC;CAAK;CAAK;CAAK;AAAG,CAAC;AAOvD,IAAaI,yBAAb,cAEUlB,qBAAiC;CACzC;CACA;CAGA;CAEA;;CAKA,mBAA8D;EAC5D,MAAMuB,cAAc,KAAKC;EAGzB,OAAO,IAAID,YAAY,KAAKF,sBAAsB;CACpD;CAEAG,YAAYC,aAA6D;EACvE,MAAM,EAAER,WAAW,GAAGS,SAASD,eAAe,CAAC;EAC/C,MAAME,YAAYD,KAAKE;EACvB,MAAMC,uBAAuBH,KAAKI;EAElC,MAAM;GACJ,GAAGJ;GACH,GAAIT,aAAa;IACfW,OAAOG,uBAAuBd,WAAWU,SAAS;IAClDG,iCAAiCE,qBAC/Bf,WACAY,oBACF;GACF;GACAI,4BAA4B,OAAOC,YAAY;IAC7C,MAAMC,WAAWD,QAAQE;IACzB,MAAMC,UAAU,KAAKlB,SAASmB,OAAOC,gBAAgB;IAGrD,MAAMH,MAAM,OADV,KAAKhB,oBAAoB,KAAK,KAAKD,SAASsB,QAAQC,SAAAA,EACpBC,WAAW,EAAA,EAAIC,YAAYV,QAAQE;IAErE,MAAMS,YAAY;KAChB,GAAGX;KACHE;KACAU,MAAM;MACJC,cAAcV,SAASU;MACvBC,QAAQX,SAASW;MACjBC,QAAQZ,SAASY;MACjBC,OAAO/C,kBAAkBkC,SAASa,SAAS,CAAC,CAAC;MAC7C,GAAGhB,SAASY;KACd;IACF;IACA,MAAMK,kBACJ,MAAMzB,KAAKO,6BAA6BY,SAAS;IACnD,MAAMO,UAAUnC,YACZ,IAAIoC,QAAQF,iBAAiBC,WAAWlB,QAAQkB,OAAO,IACvDE,KAAAA;IACJF,SAASG,IAAI5C,4BAA4BwB,QAAQ;IAEjD,OAAO;KACL,GAAGgB;KACH,GAAIC,WAAW,EAAEA,QAAQ;KACzBN,MAAMK,iBAAiBL,QAAQ;MAC7B,GAAGD,UAAUC;MACbV;MACAoB,UAAUtB,QAAQsB;MAClBC,SAASvB,QAAQuB;MACjBC,WAAWxB,QAAQwB;MACnBC,UAAUzB,QAAQ0B;KACpB;IACF;GACF;EACF,CAAC;EAED,KAAK3C,YAAYA;EACjB,KAAKI,yBAAyBI;CAChC;CAEAoC,WAAW1C,SAA2B;EACpC,KAAKA,UAAUA;CACjB;CAEA2C,sBAAiE;EAC/D,OAAO,KAAK7C;CACd;CAEA8C,gCACEC,QACA;EACA,KAAK5C,oBAAoB4C;CAC3B;AACF;AAEA,SAASjC,uBACPd,WACAU,WAC+D;CAC/D,MAAMuC,YAAqCvC,aACtCyC,OAAOC,SAAS1C,UAAUyC,OAA4BC,IAAI,IAC3DF,WAAWvC,MAAM4C,KAAKL,UAAU;CAEpC,OAAO,OAAOC,OAAOC,SAAS;EAC5B,MAAMjB,UAAU,IAAIC,QAAQgB,MAAMjB,OAAO;EACzC,MAAMjB,WAAWiB,QAAQqB,IAAI9D,0BAA0B,KAAK2C,KAAAA;EAC5D,MAAMoB,uBAAuBtB,QAAQqB,IAAI7D,6BAA6B;EACtE,MAAM+D,sBACJD,wBAAwBzD,UAAU2D,QAAQC,YAAY1C,QAAQ;EAChEiB,QAAQ0B,OAAOnE,0BAA0B;EACzCyC,QAAQ0B,OAAOlE,6BAA6B;EAC5C,MAAMmE,MAAM,MAAMb,UAAUE,OAAO;GAAE,GAAGC;GAAMjB;EAAQ,CAAC;EACvD,MAAMhB,KAAK2C,IAAI3B,QAAQqB,IAAIrE,0BAA0B;EACrD,MAAM4E,iBACJ,CAACN,wBACDzD,UAAU2D,QAAQC,YAAY1C,QAAQ,MAAMuC;EAC9C,IAAItC,MAAM4C,gBAAgB/D,UAAU2D,QAAQK,YAAY7C,IAAID,QAAQ;EACpE,KACG4C,IAAIG,WAAW,OAAOH,IAAIG,WAAW,QACtCR,wBACAzD,UAAU2D,QAAQC,YAAY1C,QAAQ,MAAMuC,sBAE5CzD,UAAU2D,QAAQO,MAAMhD,QAAQ;EAElC,IAAI,CAAC4C,IAAIjC,QAAQjC,mBAAmBuE,IAAIL,IAAIG,MAAM,GAAG,OAAOH;EAE5D,MAAMM,eAAepE,UAAUqE,iBAAiBC;EAGhD,MAAMC,UAAU,IAAIC,YAAY;EAChC,IAAIC,cAAc;EAClB,MAAMC,MAAM,IAAIC,gBAAwC,EACtDE,UAAUC,OAAOC,YAAY;GAC3BA,WAAWC,QAAQF,KAAK;GACxBL,eAAeF,QAAQU,OAAOH,OAAO,EAAEI,QAAQ,KAAK,CAAC;GACrD,IAAId,aAAaU,OAAOL,WAAW,GAAG;IACpC,IACEV,kBACA/D,UAAU2D,QAAQC,YAAY1C,QAAQ,OACnCC,MAAMuC,sBAET1D,UAAU2D,QAAQO,MAAMhD,QAAQ;IAElCuD,cAAc;GAChB,OAAO,IAAIA,YAAYU,SAAS3F,qBAC9BiF,cAAcA,YAAYW,MAAM,KAAmB;EAEvD,EACF,CAAC;EAED,OAAO,IAAIC,SAASvB,IAAIjC,KAAKyD,YAAYZ,GAAG,GAAG;GAC7CT,QAAQH,IAAIG;GACZsB,YAAYzB,IAAIyB;GAChBpD,SAAS2B,IAAI3B;EACf,CAAC;CACH;AACF;AAEA,SAASmC,qBAAqBkB,QAAoBf,aAAqB;CACrE,OAAOA,YAAYgB,SAASlG,aAAa;AAC3C;AAEA,SAASwB,qBACPf,WACAY,sBAGA;CACA,OAAO,OAAOK,YAAY;EACxB,MAAMyE,WAAW1F,UAAU2D,QAAQC,YAAY3C,QAAQE,EAAE;EACzD,IAAI,CAACuE,UACH,MAAM,IAAIC,MACR,6EACF;EAEF,MAAMC,MACJ,OAAO5F,UAAU6F,cAAc,aAC3B7F,UAAU6F,UAAUH,QAAQ,IAC5B1F,UAAU6F;EAChB,MAAMC,eAAe,MAAMlF,uBAAuB;GAAE,GAAGK;GAAS2E;EAAI,CAAC;EACrE,MAAMzD,UAAU,IAAIC,QAAQ0D,cAAc3D,WAAWlB,QAAQkB,OAAO;EACpEA,QAAQG,IAAI5C,4BAA4BuB,QAAQE,EAAE;EAClDgB,QAAQG,IAAI3C,+BAA+B+F,QAAQ;EACnD,OAAO;GACL,GAAGI;GACH3D;GACAyD,KAAKE,cAAcF,OAAOA;EAC5B;CACF;AACF"}

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"}