@assistant-ui/react 0.15.3 → 0.15.5

package/dist/legacy-runtime/runtime-cores/assistant-transport/runManager.js

@@ -2,6 +2,9 @@ import { useLatestRef } from "./useLatestRef.js";
 import { useCallback, useEffect, useRef, useState } from "@assistant-ui/tap/react-shim";
 //#region src/legacy-runtime/runtime-cores/assistant-transport/runManager.ts
 const disposeReason = Symbol("assistant-transport-dispose");
+const reportFinishError = (error) => {
+	console.error("[assistant-ui] Assistant transport onFinish callback threw an error", error);
+};
 function useRunManager(config) {
 	const [isRunning, setIsRunning] = useState(false);
 	const stateRef = useRef({
@@ -29,7 +32,11 @@ function useRunManager(config) {
 					else await onErrorRef.current?.(error);
 				}
 			} finally {
-				if (!disposeAborted() && !stateRef.current.disposed) onFinishRef.current?.();
+				if (!disposeAborted() && !stateRef.current.disposed) try {
+					Promise.resolve(onFinishRef.current?.()).catch(reportFinishError);
+				} catch (error) {
+					reportFinishError(error);
+				}
 				if (!stateRef.current.disposed && stateRef.current.pending) startRun();
 				else {
 					setIsRunning(false);

package/dist/legacy-runtime/runtime-cores/assistant-transport/runManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"runManager.js","names":["useCallback","useEffect","useRef","useState","useLatestRef","RunManager","Readonly","isRunning","schedule","cancel","disposeReason","Symbol","useRunManager","config","onRun","signal","AbortSignal","Promise","onFinish","onCancel","onError","error","Error","setIsRunning","stateRef","pending","disposed","abortController","AbortController","onRunRef","onFinishRef","onCancelRef","onErrorRef","startRun","current","ac","disposeAborted","reason","queueMicrotask","aborted","abort"],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/runManager.ts"],"sourcesContent":["import { useCallback, useEffect, useRef, useState } from \"react\";\nimport { useLatestRef } from \"./useLatestRef\";\n\nexport type RunManager = Readonly<{\n  isRunning: boolean;\n  schedule: () => void;\n  cancel: () => void;\n}>;\n\nconst disposeReason = Symbol(\"assistant-transport-dispose\");\n\nexport function useRunManager(config: {\n  onRun: (signal: AbortSignal) => Promise<void>;\n  onFinish?: (() => void) | undefined;\n  onCancel?: (() => void) | undefined;\n  onError?: ((error: Error) => void | Promise<void>) | undefined;\n}): RunManager {\n  const [isRunning, setIsRunning] = useState(false);\n  const stateRef = useRef({\n    pending: false,\n    disposed: false,\n    abortController: null as AbortController | null,\n  });\n  const onRunRef = useLatestRef(config.onRun);\n  const onFinishRef = useLatestRef(config.onFinish);\n  const onCancelRef = useLatestRef(config.onCancel);\n  const onErrorRef = useLatestRef(config.onError);\n\n  const startRun = useCallback(() => {\n    setIsRunning(true);\n    stateRef.current.pending = false;\n    const ac = new AbortController();\n    stateRef.current.abortController = ac;\n\n    // The dispose marker rides on the abort reason so a settling run detects\n    // its own disposal even after an effect cycle re-arms the manager.\n    const disposeAborted = () => ac.signal.reason === disposeReason;\n\n    queueMicrotask(async () => {\n      try {\n        if (!disposeAborted()) {\n          await onRunRef.current(ac.signal);\n        }\n      } catch (error) {\n        if (!disposeAborted() && !stateRef.current.disposed) {\n          stateRef.current.pending = false;\n          if (ac.signal.aborted) {\n            onCancelRef.current?.();\n          } else {\n            await onErrorRef.current?.(error as Error);\n          }\n        }\n      } finally {\n        if (!disposeAborted() && !stateRef.current.disposed) {\n          onFinishRef.current?.();\n        }\n        if (!stateRef.current.disposed && stateRef.current.pending) {\n          startRun();\n        } else {\n          setIsRunning(false);\n          if (stateRef.current.abortController === ac) {\n            stateRef.current.abortController = null;\n          }\n        }\n      }\n    });\n  }, [onRunRef, onFinishRef, onErrorRef, onCancelRef]);\n\n  const schedule = useCallback(() => {\n    if (stateRef.current.disposed) return;\n    if (stateRef.current.abortController) {\n      // Coalesce multiple schedules while running into a single follow-up run.\n      stateRef.current.pending = true;\n      return;\n    }\n    startRun();\n  }, [startRun]);\n\n  // Strict-mode effects run setup / cleanup / setup on the same instance;\n  // arming on setup undoes the cleanup's dispose.\n  useEffect(() => {\n    stateRef.current.disposed = false;\n    return () => {\n      stateRef.current.disposed = true;\n      stateRef.current.pending = false;\n      stateRef.current.abortController?.abort(disposeReason);\n    };\n  }, []);\n\n  const cancel = useCallback(() => {\n    stateRef.current.pending = false;\n    stateRef.current.abortController?.abort();\n  }, []);\n\n  return {\n    isRunning,\n    schedule,\n    cancel,\n  };\n}\n"],"mappings":";;;AASA,MAAMU,gBAAgBC,OAAO,6BAA6B;AAE1D,SAAgBC,cAAcC,QAKf;CACb,MAAM,CAACN,WAAWgB,gBAAgBpB,SAAS,KAAK;CAChD,MAAMqB,WAAWtB,OAAO;EACtBuB,SAAS;EACTC,UAAU;EACVC,iBAAiB;CACnB,CAAC;CACD,MAAME,WAAWzB,aAAaS,OAAOC,KAAK;CAC1C,MAAMgB,cAAc1B,aAAaS,OAAOK,QAAQ;CAChD,MAAMa,cAAc3B,aAAaS,OAAOM,QAAQ;CAChD,MAAMa,aAAa5B,aAAaS,OAAOO,OAAO;CAE9C,MAAMa,WAAWjC,kBAAkB;EACjCuB,aAAa,IAAI;EACjBC,SAASU,QAAQT,UAAU;EAC3B,MAAMU,KAAK,IAAIP,gBAAgB;EAC/BJ,SAASU,QAAQP,kBAAkBQ;EAInC,MAAMC,uBAAuBD,GAAGpB,OAAOsB,WAAW3B;EAElD4B,eAAe,YAAY;GACzB,IAAI;IACF,IAAI,CAACF,eAAe,GAClB,MAAMP,SAASK,QAAQC,GAAGpB,MAAM;GAEpC,SAASM,OAAO;IACd,IAAI,CAACe,eAAe,KAAK,CAACZ,SAASU,QAAQR,UAAU;KACnDF,SAASU,QAAQT,UAAU;KAC3B,IAAIU,GAAGpB,OAAOwB,SACZR,YAAYG,UAAU;UAEtB,MAAMF,WAAWE,UAAUb,KAAc;IAE7C;GACF,UAAU;IACR,IAAI,CAACe,eAAe,KAAK,CAACZ,SAASU,QAAQR,UACzCI,YAAYI,UAAU;IAExB,IAAI,CAACV,SAASU,QAAQR,YAAYF,SAASU,QAAQT,SACjDQ,SAAS;SACJ;KACLV,aAAa,KAAK;KAClB,IAAIC,SAASU,QAAQP,oBAAoBQ,IACvCX,SAASU,QAAQP,kBAAkB;IAEvC;GACF;EACF,CAAC;CACH,GAAG;EAACE;EAAUC;EAAaE;EAAYD;CAAW,CAAC;CAEnD,MAAMvB,WAAWR,kBAAkB;EACjC,IAAIwB,SAASU,QAAQR,UAAU;EAC/B,IAAIF,SAASU,QAAQP,iBAAiB;GAEpCH,SAASU,QAAQT,UAAU;GAC3B;EACF;EACAQ,SAAS;CACX,GAAG,CAACA,QAAQ,CAAC;CAIbhC,gBAAgB;EACduB,SAASU,QAAQR,WAAW;EAC5B,aAAa;GACXF,SAASU,QAAQR,WAAW;GAC5BF,SAASU,QAAQT,UAAU;GAC3BD,SAASU,QAAQP,iBAAiBa,MAAM9B,aAAa;EACvD;CACF,GAAG,CAAA,CAAE;CAOL,OAAO;EACLH;EACAC;EACAC,QARaT,kBAAkB;GAC/BwB,SAASU,QAAQT,UAAU;GAC3BD,SAASU,QAAQP,iBAAiBa,MAAM;EAC1C,GAAG,CAAA,CAKD/B;CACF;AACF"}
+{"version":3,"file":"runManager.js","names":["useCallback","useEffect","useRef","useState","useLatestRef","RunManager","Readonly","isRunning","schedule","cancel","disposeReason","Symbol","reportFinishError","error","console","useRunManager","config","onRun","signal","AbortSignal","Promise","onFinish","onCancel","onError","Error","setIsRunning","stateRef","pending","disposed","abortController","AbortController","onRunRef","onFinishRef","onCancelRef","onErrorRef","startRun","current","ac","disposeAborted","reason","queueMicrotask","aborted","resolve","catch","abort"],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/runManager.ts"],"sourcesContent":["import { useCallback, useEffect, useRef, useState } from \"react\";\nimport { useLatestRef } from \"./useLatestRef\";\n\nexport type RunManager = Readonly<{\n  isRunning: boolean;\n  schedule: () => void;\n  cancel: () => void;\n}>;\n\nconst disposeReason = Symbol(\"assistant-transport-dispose\");\n\nconst reportFinishError = (error: unknown) => {\n  console.error(\n    \"[assistant-ui] Assistant transport onFinish callback threw an error\",\n    error,\n  );\n};\n\nexport function useRunManager(config: {\n  onRun: (signal: AbortSignal) => Promise<void>;\n  onFinish?: (() => void) | undefined;\n  onCancel?: (() => void) | undefined;\n  onError?: ((error: Error) => void | Promise<void>) | undefined;\n}): RunManager {\n  const [isRunning, setIsRunning] = useState(false);\n  const stateRef = useRef({\n    pending: false,\n    disposed: false,\n    abortController: null as AbortController | null,\n  });\n  const onRunRef = useLatestRef(config.onRun);\n  const onFinishRef = useLatestRef(config.onFinish);\n  const onCancelRef = useLatestRef(config.onCancel);\n  const onErrorRef = useLatestRef(config.onError);\n\n  const startRun = useCallback(() => {\n    setIsRunning(true);\n    stateRef.current.pending = false;\n    const ac = new AbortController();\n    stateRef.current.abortController = ac;\n\n    // The dispose marker rides on the abort reason so a settling run detects\n    // its own disposal even after an effect cycle re-arms the manager.\n    const disposeAborted = () => ac.signal.reason === disposeReason;\n\n    queueMicrotask(async () => {\n      try {\n        if (!disposeAborted()) {\n          await onRunRef.current(ac.signal);\n        }\n      } catch (error) {\n        if (!disposeAborted() && !stateRef.current.disposed) {\n          stateRef.current.pending = false;\n          if (ac.signal.aborted) {\n            onCancelRef.current?.();\n          } else {\n            await onErrorRef.current?.(error as Error);\n          }\n        }\n      } finally {\n        if (!disposeAborted() && !stateRef.current.disposed) {\n          try {\n            void Promise.resolve(onFinishRef.current?.()).catch(\n              reportFinishError,\n            );\n          } catch (error) {\n            reportFinishError(error);\n          }\n        }\n        if (!stateRef.current.disposed && stateRef.current.pending) {\n          startRun();\n        } else {\n          setIsRunning(false);\n          if (stateRef.current.abortController === ac) {\n            stateRef.current.abortController = null;\n          }\n        }\n      }\n    });\n  }, [onRunRef, onFinishRef, onErrorRef, onCancelRef]);\n\n  const schedule = useCallback(() => {\n    if (stateRef.current.disposed) return;\n    if (stateRef.current.abortController) {\n      // Coalesce multiple schedules while running into a single follow-up run.\n      stateRef.current.pending = true;\n      return;\n    }\n    startRun();\n  }, [startRun]);\n\n  // Strict-mode effects run setup / cleanup / setup on the same instance;\n  // arming on setup undoes the cleanup's dispose.\n  useEffect(() => {\n    stateRef.current.disposed = false;\n    return () => {\n      stateRef.current.disposed = true;\n      stateRef.current.pending = false;\n      stateRef.current.abortController?.abort(disposeReason);\n    };\n  }, []);\n\n  const cancel = useCallback(() => {\n    stateRef.current.pending = false;\n    stateRef.current.abortController?.abort();\n  }, []);\n\n  return {\n    isRunning,\n    schedule,\n    cancel,\n  };\n}\n"],"mappings":";;;AASA,MAAMU,gBAAgBC,OAAO,6BAA6B;AAE1D,MAAMC,qBAAqBC,UAAmB;CAC5CC,QAAQD,MACN,uEACAA,KACF;AACF;AAEA,SAAgBE,cAAcC,QAKf;CACb,MAAM,CAACT,WAAWkB,gBAAgBtB,SAAS,KAAK;CAChD,MAAMuB,WAAWxB,OAAO;EACtByB,SAAS;EACTC,UAAU;EACVC,iBAAiB;CACnB,CAAC;CACD,MAAME,WAAW3B,aAAaY,OAAOC,KAAK;CAC1C,MAAMe,cAAc5B,aAAaY,OAAOK,QAAQ;CAChD,MAAMY,cAAc7B,aAAaY,OAAOM,QAAQ;CAChD,MAAMY,aAAa9B,aAAaY,OAAOO,OAAO;CAE9C,MAAMY,WAAWnC,kBAAkB;EACjCyB,aAAa,IAAI;EACjBC,SAASU,QAAQT,UAAU;EAC3B,MAAMU,KAAK,IAAIP,gBAAgB;EAC/BJ,SAASU,QAAQP,kBAAkBQ;EAInC,MAAMC,uBAAuBD,GAAGnB,OAAOqB,WAAW7B;EAElD8B,eAAe,YAAY;GACzB,IAAI;IACF,IAAI,CAACF,eAAe,GAClB,MAAMP,SAASK,QAAQC,GAAGnB,MAAM;GAEpC,SAASL,OAAO;IACd,IAAI,CAACyB,eAAe,KAAK,CAACZ,SAASU,QAAQR,UAAU;KACnDF,SAASU,QAAQT,UAAU;KAC3B,IAAIU,GAAGnB,OAAOuB,SACZR,YAAYG,UAAU;UAEtB,MAAMF,WAAWE,UAAUvB,KAAc;IAE7C;GACF,UAAU;IACR,IAAI,CAACyB,eAAe,KAAK,CAACZ,SAASU,QAAQR,UACzC,IAAI;KACF,QAAac,QAAQV,YAAYI,UAAU,CAAC,CAAC,CAACO,MAC5C/B,iBACF;IACF,SAASC,OAAO;KACdD,kBAAkBC,KAAK;IACzB;IAEF,IAAI,CAACa,SAASU,QAAQR,YAAYF,SAASU,QAAQT,SACjDQ,SAAS;SACJ;KACLV,aAAa,KAAK;KAClB,IAAIC,SAASU,QAAQP,oBAAoBQ,IACvCX,SAASU,QAAQP,kBAAkB;IAEvC;GACF;EACF,CAAC;CACH,GAAG;EAACE;EAAUC;EAAaE;EAAYD;CAAW,CAAC;CAEnD,MAAMzB,WAAWR,kBAAkB;EACjC,IAAI0B,SAASU,QAAQR,UAAU;EAC/B,IAAIF,SAASU,QAAQP,iBAAiB;GAEpCH,SAASU,QAAQT,UAAU;GAC3B;EACF;EACAQ,SAAS;CACX,GAAG,CAACA,QAAQ,CAAC;CAIblC,gBAAgB;EACdyB,SAASU,QAAQR,WAAW;EAC5B,aAAa;GACXF,SAASU,QAAQR,WAAW;GAC5BF,SAASU,QAAQT,UAAU;GAC3BD,SAASU,QAAQP,iBAAiBe,MAAMlC,aAAa;EACvD;CACF,GAAG,CAAA,CAAE;CAOL,OAAO;EACLH;EACAC;EACAC,QARaT,kBAAkB;GAC/B0B,SAASU,QAAQT,UAAU;GAC3BD,SAASU,QAAQP,iBAAiBe,MAAM;EAC1C,GAAG,CAAA,CAKDnC;CACF;AACF"}

package/dist/legacy-runtime/runtime-cores/assistant-transport/types.d.ts

@@ -56,7 +56,9 @@ type HeadersValue = Record<string, string> | Headers;
 type AssistantTransportProtocol = "data-stream" | "assistant-transport";
 type SendCommandsRequestBody = {
   commands: QueuedCommand[];
-  state: unknown;
+  /** Absent on a resume with `resumeStateApi`; the server replays from its retained snapshot. */
+  state?: unknown;
+  runId?: string;
   system: string | undefined;
   tools: Record<string, unknown> | undefined;
   callSettings: LanguageModelV1CallSettings | undefined;
@@ -69,7 +71,15 @@ type AssistantTransportOptions<T> = {
   initialState: T;
   api: string;
   resumeApi?: string;
+  /** Endpoint that returns the retained initial state and run ID for a resume stream. A 204 response means no run is active and the resume is skipped. */
+  resumeStateApi?: string;
   protocol?: AssistantTransportProtocol;
+  /**
+   * When `false`, stream decoding and state reconciliation tolerate malformed
+   * input (invalid chunks are dropped with a console log) instead of throwing.
+   * Resume runs always decode leniently. Defaults to `true`.
+   */
+  strict?: boolean;
   converter: AssistantTransportStateConverter<T>;
   headers: HeadersValue | (() => Promise<HeadersValue>);
   body?: object | (() => Promise<object | undefined>);

package/dist/legacy-runtime/runtime-cores/assistant-transport/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/types.ts"],"mappings":";;;;;KAaY;WACD;WACA;;KAGC;WACD;WACA;;KAGC,kBAAkB,WAAW;KAE7B;WACD;WACA,gBAAgB;;KAGf;WACD;WACA,gBAAgB;;KAIf;WACD;WACA,SAAS,cAAc;WACvB;WACA;;KAGC;WACD;WACA;WACA;WACA,QAAQ;WACR;WACA,WAAW;WACX,wBAAwB;;KAGvB,4BACR,oBACA,uBACA;KAGQ;WACD,mBAAmB;WACnB,QAAQ;WACR;;KAGC;EACV,iBAAiB;EACjB;EACA,cAAc,eAAe;;KAGnB,iCAAiC,MAC3C,OAAO,GACP,oBAAoB,yCACjB;KAGO;EACV,QAAQ;EACR,WAAW;;KAID,gBAAgB;KAKhB,eAAe,yBAAyB;KAExC;KAEA;EACV,UAAU;EACV;EACA;EACA,OAAO;EACP,cAAc;EACd,QAAQ;EACR;EACA;GAIC;;KAGS,0BAA0B;EACpC,cAAc;EACd;EACA;EACA,WAAW;EACX,WAAW,iCAAiC;EAC5C,SAAS,sBAAsB,QAAQ;EACvC,uBAAuB;;;;;;;;;;;;;EAavB,8BACE,MAAM,4BACH,0BAA0B,QAAQ;EACvC,cAAc,UAAU;EACxB;EACA,WACE,OAAO,OACP;IACE,UAAU;IACV,cAAc,UAAU,OAAO,MAAM;eAE7B;;;;;;;EAOZ,YAAY;IACV,UAAU;IACV,cAAc,UAAU,OAAO,MAAM;IACrC,QAAQ;;EAEV;IACE;;EAEF;IACE,cAAc;IACd,UAAU"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/types.ts"],"mappings":";;;;;KAaY;WACD;WACA;;KAGC;WACD;WACA;;KAGC,kBAAkB,WAAW;KAE7B;WACD;WACA,gBAAgB;;KAGf;WACD;WACA,gBAAgB;;KAIf;WACD;WACA,SAAS,cAAc;WACvB;WACA;;KAGC;WACD;WACA;WACA;WACA,QAAQ;WACR;WACA,WAAW;WACX,wBAAwB;;KAGvB,4BACR,oBACA,uBACA;KAGQ;WACD,mBAAmB;WACnB,QAAQ;WACR;;KAGC;EACV,iBAAiB;EACjB;EACA,cAAc,eAAe;;KAGnB,iCAAiC,MAC3C,OAAO,GACP,oBAAoB,yCACjB;KAGO;EACV,QAAQ;EACR,WAAW;;KAID,gBAAgB;KAKhB,eAAe,yBAAyB;KAExC;KAEA;EACV,UAAU;;EAEV;EACA;EACA;EACA,OAAO;EACP,cAAc;EACd,QAAQ;EACR;EACA;GAIC;;KAGS,0BAA0B;EACpC,cAAc;EACd;EACA;;EAEA;EACA,WAAW;;;;;;EAMX;EACA,WAAW,iCAAiC;EAC5C,SAAS,sBAAsB,QAAQ;EACvC,uBAAuB;;;;;;;;;;;;;EAavB,8BACE,MAAM,4BACH,0BAA0B,QAAQ;EACvC,cAAc,UAAU;EACxB;EACA,WACE,OAAO,OACP;IACE,UAAU;IACV,cAAc,UAAU,OAAO,MAAM;eAE7B;;;;;;;EAOZ,YAAY;IACV,UAAU;IACV,cAAc,UAAU,OAAO,MAAM;IACrC,QAAQ;;EAEV;IACE;;EAEF;IACE,cAAc;IACd,UAAU"}

package/dist/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAssistantTransportRuntime.d.ts","names":[],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.ts"],"mappings":";;;;cA8Fa,yCAGH,SAAS;iBAOH,8BAA8B;iBAC9B,2BAA2B,GACzC,WAAW,OAAO,sBAAsB,IACvC;;;;cA6RU,+BAAgC,GAC3C,SAAS,0BAA0B,OAClC"}
+{"version":3,"file":"useAssistantTransportRuntime.d.ts","names":[],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.ts"],"mappings":";;;;cA8Ha,yCAGH,SAAS;iBAOH,8BAA8B;iBAC9B,2BAA2B,GACzC,WAAW,OAAO,sBAAsB,IACvC;;;;cAmUU,+BAAgC,GAC3C,SAAS,0BAA0B,OAClC"}

package/dist/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.js

@@ -24,6 +24,7 @@ const convertAppendMessageToCommand = (message) => {
 		type: "image",
 		image: contentPart.image
 	});
+	if (parts.length === 0) return null;
 	return {
 		type: "add-message",
 		message: {
@@ -34,6 +35,21 @@ const convertAppendMessageToCommand = (message) => {
 		sourceId: message.sourceId
 	};
 };
+const readResumeState = async (response) => {
+	if (response.status === 204) return null;
+	if (!response.ok) throw new Error(`Resume state request failed with status ${response.status}: ${await response.text()}`);
+	let value;
+	try {
+		value = await response.json();
+	} catch {
+		throw new Error("Resume state response was not valid JSON");
+	}
+	if (typeof value !== "object" || value === null || !("state" in value) || !("runId" in value) || typeof value.runId !== "string") throw new Error("Resume state response must contain state and runId");
+	return {
+		runId: value.runId,
+		state: value.state
+	};
+};
 const symbolAssistantTransportExtras = Symbol("assistant-transport-extras");
 const asAssistantTransportExtras = (extras) => {
 	if (typeof extras !== "object" || extras == null || !(symbolAssistantTransportExtras in extras)) throw new Error("This method can only be called when you are using useAssistantTransportRuntime");
@@ -64,6 +80,15 @@ const useAssistantTransportThreadRuntime = (options) => {
 	const waitForReplayRender = useReplayRenderWait();
 	const parentIdRef = useRef(void 0);
 	const commandQueue = useCommandQueue({ onQueue: () => runManager.schedule() });
+	const enqueueAppendMessage = (message) => {
+		const command = convertAppendMessageToCommand(message);
+		if (!command) {
+			console.warn("[assistant-ui] Skipped add-message command with no supported parts");
+			return;
+		}
+		parentIdRef.current = message.parentId;
+		commandQueue.enqueue(command, { schedule: message.startRun ?? message.role === "user" });
+	};
 	const threadId = useAuiState((s) => s.threadListItem.remoteId);
 	const runManager = useRunManager({
 		onRun: async (signal) => {
@@ -75,11 +100,26 @@ const useAssistantTransportThreadRuntime = (options) => {
 			const parentId = isResume ? void 0 : parentIdRef.current;
 			if (!isResume) parentIdRef.current = void 0;
 			const headers = await createRequestHeaders(options.headers);
+			let resumeState;
+			if (isResume && options.resumeStateApi) {
+				const resumeStateResponse = await fetch(options.resumeStateApi, {
+					method: "POST",
+					headers,
+					body: JSON.stringify({ threadId }),
+					signal
+				});
+				const retained = await readResumeState(resumeStateResponse);
+				if (retained === null) {
+					if (commandQueue.state.queued.length > 0) runManager.schedule();
+					return;
+				}
+				resumeState = retained;
+			}
 			const bodyValue = typeof options.body === "function" ? await options.body() : options.body;
 			const context = runtime.thread.getModelContext();
 			let requestBody = {
 				commands,
-				state: agentStateRef.current,
+				...resumeState === void 0 && { state: agentStateRef.current },
 				system: context.system,
 				tools: context.tools ? toToolsJSONSchema(context.tools) : void 0,
 				threadId,
@@ -91,6 +131,13 @@ const useAssistantTransportThreadRuntime = (options) => {
 				...bodyValue ?? {}
 			};
 			if (options.prepareSendCommandsRequest) requestBody = await options.prepareSendCommandsRequest(requestBody);
+			if (resumeState !== void 0) {
+				requestBody = {
+					...requestBody,
+					runId: resumeState.runId
+				};
+				delete requestBody["state"];
+			}
 			const response = await fetch(isResume ? options.resumeApi : options.api, {
 				method: "POST",
 				headers,
@@ -100,15 +147,22 @@ const useAssistantTransportThreadRuntime = (options) => {
 			options.onResponse?.(response);
 			if (!response.ok) throw new Error(`Status ${response.status}: ${await response.text()}`);
 			if (!response.body) throw new Error("Response body is null");
+			if (resumeState !== void 0) {
+				agentStateRef.current = resumeState.state;
+				rerender((prev) => prev + 1);
+			}
 			const body = await createReplayBoundaryStream(response, {
 				setReplaying: setIsReplaying,
 				waitForRender: waitForReplayRender
 			});
-			const decoder = (options.protocol ?? "data-stream") === "assistant-transport" ? new AssistantTransportDecoder() : new DataStreamDecoder();
+			const protocol = options.protocol ?? "data-stream";
+			const strict = isResume ? false : options.strict ?? true;
+			const decoder = protocol === "assistant-transport" ? new AssistantTransportDecoder({ strict }) : new DataStreamDecoder({ strict });
 			let err;
 			const stream = body.pipeThrough(decoder).pipeThrough(new AssistantMessageAccumulator({
 				initialMessage: unstable_createInitialMessage({ unstable_state: agentStateRef.current ?? null }),
 				throttle: isResume,
+				strict,
 				onError: (error) => {
 					err = error;
 				}
@@ -136,7 +190,7 @@ const useAssistantTransportThreadRuntime = (options) => {
 				commands: cmds,
 				updateState: (updater) => {
 					agentStateRef.current = updater(agentStateRef.current);
-					rerender((prev) => prev + 1);
+					rerender((prev_0) => prev_0 + 1);
 				}
 			});
 		},
@@ -178,21 +232,13 @@ const useAssistantTransportThreadRuntime = (options) => {
 		setToolStatuses,
 		extras: {
 			[symbolAssistantTransportExtras]: true,
-			sendCommand: (command) => {
-				commandQueue.enqueue(command);
+			sendCommand: (command_0) => {
+				commandQueue.enqueue(command_0);
 			},
 			state: agentStateRef.current
 		},
-		onNew: async (message) => {
-			parentIdRef.current = message.parentId;
-			const command_0 = convertAppendMessageToCommand(message);
-			commandQueue.enqueue(command_0, { schedule: message.startRun ?? message.role === "user" });
-		},
-		...options.capabilities?.edit && { onEdit: async (message_0) => {
-			parentIdRef.current = message_0.parentId;
-			const command_1 = convertAppendMessageToCommand(message_0);
-			commandQueue.enqueue(command_1, { schedule: message_0.startRun ?? message_0.role === "user" });
-		} },
+		onNew: async (message_0) => enqueueAppendMessage(message_0),
+		...options.capabilities?.edit && { onEdit: async (message_1) => enqueueAppendMessage(message_1) },
 		...commandQueue.state.queued.length > 0 && { onReload: async (parentId_0) => {
 			parentIdRef.current = parentId_0;
 			runManager.schedule();
@@ -206,7 +252,7 @@ const useAssistantTransportThreadRuntime = (options) => {
 			runManager.schedule();
 		},
 		onAddToolResult: async (toolOptions) => {
-			const command_2 = {
+			const command_1 = {
 				type: "add-tool-result",
 				toolCallId: toolOptions.toolCallId,
 				result: toolOptions.result,
@@ -215,11 +261,11 @@ const useAssistantTransportThreadRuntime = (options) => {
 				...toolOptions.artifact && { artifact: toolOptions.artifact },
 				...toolOptions.modelContent !== void 0 && { modelContent: toolOptions.modelContent }
 			};
-			commandQueue.enqueue(command_2);
+			commandQueue.enqueue(command_1);
 		},
 		onLoadExternalState: async (state) => {
 			agentStateRef.current = state;
-			rerender((prev_0) => prev_0 + 1);
+			rerender((prev_1) => prev_1 + 1);
 		}
 	});
 	return runtime;

package/dist/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAssistantTransportRuntime.js","names":["c","_c","AppendMessage","ReadonlyJSONObject","ReadonlyJSONValue","asAsyncIterableStream","useExternalStoreRuntime","AssistantRuntime","AddToolResultOptions","useMemo","useRef","useState","AssistantMessageAccumulator","DataStreamDecoder","AssistantTransportDecoder","unstable_createInitialMessage","createInitialMessage","toToolsJSONSchema","AssistantTransportOptions","AddMessageCommand","AddToolResultCommand","UserMessagePart","QueuedCommand","AssistantTransportCommand","SendCommandsRequestBody","useCommandQueue","createReplayBoundaryStream","useReplayRenderWait","useRunManager","useConvertedState","ToolExecutionStatus","createRequestHeaders","useRemoteThreadListRuntime","InMemoryThreadListAdapter","useAui","useAuiState","UserExternalState","convertAppendMessageToCommand","message","role","Error","parts","content","attachments","flatMap","a","contentPart","type","push","text","image","parentId","sourceId","symbolAssistantTransportExtras","Symbol","AssistantTransportExtras","sendCommand","command","state","asAssistantTransportExtras","extras","useAssistantTransportSendCommand","$","aui","t0","thread","getState","transportExtras","useAssistantTransportState","selector","T","t","s","useAssistantTransportThreadRuntime","options","agentStateRef","initialState","rerender","resumeFlagRef","isReplaying","setIsReplaying","waitForReplayRender","parentIdRef","undefined","commandQueue","onQueue","runManager","schedule","threadId","threadListItem","remoteId","onRun","signal","AbortSignal","isResume","current","commands","flush","length","headers","bodyValue","body","context","runtime","getModelContext","requestBody","Record","system","tools","callSettings","config","prepareSendCommandsRequest","response","fetch","resumeApi","api","method","JSON","stringify","onResponse","ok","status","setReplaying","waitForRender","protocol","decoder","err","stream","pipeThrough","initialMessage","unstable_state","throttle","onError","error","markedDelivered","chunk","metadata","markDelivered","prev","queued","onFinish","onCancel","cmds","inTransit","reset","updateState","updater","inTransitCmds","queuedCmds","toolStatuses","setToolStatuses","pendingCommands","converted","converter","isRunning","messages","isLoading","adapters","unstable_enableToolInvocations","enqueue","onNew","Promise","startRun","capabilities","edit","onEdit","onReload","cancel","onResume","onAddToolResult","toolOptions","toolCallId","result","toolName","isError","artifact","modelContent","onLoadExternalState","useAssistantTransportRuntime","runtimeHook","RuntimeHook","adapter","allowNesting"],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport type { AppendMessage } from \"@assistant-ui/core\";\nimport {\n  type ReadonlyJSONObject,\n  type ReadonlyJSONValue,\n  asAsyncIterableStream,\n} from \"assistant-stream/utils\";\nimport { useExternalStoreRuntime } from \"../external-store/useExternalStoreRuntime\";\nimport type { AssistantRuntime } from \"../../runtime/AssistantRuntime\";\nimport type { AddToolResultOptions } from \"@assistant-ui/core\";\nimport { useMemo, useRef, useState } from \"react\";\nimport {\n  AssistantMessageAccumulator,\n  DataStreamDecoder,\n  AssistantTransportDecoder,\n  unstable_createInitialMessage as createInitialMessage,\n  toToolsJSONSchema,\n} from \"assistant-stream\";\nimport type {\n  AssistantTransportOptions,\n  AddMessageCommand,\n  AddToolResultCommand,\n  UserMessagePart,\n  QueuedCommand,\n  AssistantTransportCommand,\n  SendCommandsRequestBody,\n} from \"./types\";\nimport { useCommandQueue } from \"./commandQueue\";\nimport {\n  createReplayBoundaryStream,\n  useReplayRenderWait,\n} from \"./replayBoundaryStream\";\nimport { useRunManager } from \"./runManager\";\nimport { useConvertedState } from \"./useConvertedState\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport { createRequestHeaders } from \"@assistant-ui/core\";\nimport { useRemoteThreadListRuntime } from \"../remote-thread-list/useRemoteThreadListRuntime\";\nimport { InMemoryThreadListAdapter } from \"@assistant-ui/core\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { UserExternalState } from \"../../../augmentations\";\n\nconst convertAppendMessageToCommand = (\n  message: AppendMessage,\n): AddMessageCommand => {\n  if (message.role !== \"user\")\n    throw new Error(\"Only user messages are supported\");\n\n  const parts: UserMessagePart[] = [];\n  const content = [\n    ...message.content,\n    ...(message.attachments?.flatMap((a) => a.content) ?? []),\n  ];\n  for (const contentPart of content) {\n    if (contentPart.type === \"text\") {\n      parts.push({ type: \"text\", text: contentPart.text });\n    } else if (contentPart.type === \"image\") {\n      parts.push({ type: \"image\", image: contentPart.image });\n    }\n  }\n\n  return {\n    type: \"add-message\",\n    message: {\n      role: \"user\",\n      parts,\n    },\n    parentId: message.parentId,\n    sourceId: message.sourceId,\n  };\n};\n\nconst symbolAssistantTransportExtras = Symbol(\"assistant-transport-extras\");\ntype AssistantTransportExtras = {\n  [symbolAssistantTransportExtras]: true;\n  sendCommand: (command: AssistantTransportCommand) => void;\n  state: UserExternalState;\n};\n\nconst asAssistantTransportExtras = (\n  extras: unknown,\n): AssistantTransportExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolAssistantTransportExtras in extras)\n  )\n    throw new Error(\n      \"This method can only be called when you are using useAssistantTransportRuntime\",\n    );\n\n  return extras as AssistantTransportExtras;\n};\n\nexport const useAssistantTransportSendCommand = () => {\n  const aui = useAui();\n\n  return (command: AssistantTransportCommand) => {\n    const extras = aui.thread.getState().extras;\n    const transportExtras = asAssistantTransportExtras(extras);\n    transportExtras.sendCommand(command);\n  };\n};\n\nexport function useAssistantTransportState(): UserExternalState;\nexport function useAssistantTransportState<T>(\n  selector: (state: UserExternalState) => T,\n): T;\nexport function useAssistantTransportState<T>(\n  selector: (state: UserExternalState) => T = (t) => t as T,\n): T | UserExternalState {\n  return useAuiState((s) =>\n    selector(asAssistantTransportExtras(s.thread.extras).state),\n  );\n}\n\nconst useAssistantTransportThreadRuntime = <T>(\n  options: AssistantTransportOptions<T>,\n): AssistantRuntime => {\n  const agentStateRef = useRef(options.initialState);\n  const [, rerender] = useState(0);\n  const resumeFlagRef = useRef(false);\n  const [isReplaying, setIsReplaying] = useState(false);\n  const waitForReplayRender = useReplayRenderWait();\n  const parentIdRef = useRef<string | null | undefined>(undefined);\n  const commandQueue = useCommandQueue({\n    onQueue: () => runManager.schedule(),\n  });\n\n  const threadId = useAuiState((s) => s.threadListItem.remoteId);\n\n  const runManager = useRunManager({\n    onRun: async (signal: AbortSignal) => {\n      const isResume = resumeFlagRef.current;\n      resumeFlagRef.current = false;\n      setIsReplaying(false);\n      const commands: QueuedCommand[] = isResume ? [] : commandQueue.flush();\n      if (commands.length === 0 && !isResume) return;\n\n      // The flushed batch consumes the parentId; read it alongside the flush\n      // (before any awaits) so a mid-run append keeps its own value. Resume\n      // runs send no commands, so they neither send nor consume it.\n      const parentId = isResume ? undefined : parentIdRef.current;\n      if (!isResume) parentIdRef.current = undefined;\n\n      const headers = await createRequestHeaders(options.headers);\n      const bodyValue =\n        typeof options.body === \"function\"\n          ? await options.body()\n          : options.body;\n      const context = runtime.thread.getModelContext();\n\n      let requestBody: Record<string, unknown> = {\n        commands,\n        state: agentStateRef.current,\n        system: context.system,\n        tools: context.tools ? toToolsJSONSchema(context.tools) : undefined,\n        threadId,\n        ...(parentId !== undefined && {\n          parentId,\n        }),\n        // nested (new format, aligned with AssistantChatTransport)\n        callSettings: context.callSettings,\n        config: context.config,\n        // @deprecated spread at top level — use nested `callSettings`/`config` instead. Will be removed in a future version.\n        ...context.callSettings,\n        ...context.config,\n        ...(bodyValue ?? {}),\n      };\n\n      if (options.prepareSendCommandsRequest) {\n        requestBody = await options.prepareSendCommandsRequest(\n          requestBody as SendCommandsRequestBody,\n        );\n      }\n\n      const response = await fetch(\n        isResume ? options.resumeApi! : options.api,\n        {\n          method: \"POST\",\n          headers,\n          body: JSON.stringify(requestBody),\n          signal,\n        },\n      );\n\n      options.onResponse?.(response);\n\n      if (!response.ok) {\n        throw new Error(`Status ${response.status}: ${await response.text()}`);\n      }\n\n      if (!response.body) {\n        throw new Error(\"Response body is null\");\n      }\n\n      const body = await createReplayBoundaryStream(response, {\n        setReplaying: setIsReplaying,\n        waitForRender: waitForReplayRender,\n      });\n\n      // Select decoder based on protocol option\n      const protocol = options.protocol ?? \"data-stream\";\n      const decoder =\n        protocol === \"assistant-transport\"\n          ? new AssistantTransportDecoder()\n          : new DataStreamDecoder();\n\n      let err: string | undefined;\n      const stream = body.pipeThrough(decoder).pipeThrough(\n        new AssistantMessageAccumulator({\n          initialMessage: createInitialMessage({\n            unstable_state:\n              (agentStateRef.current as ReadonlyJSONValue) ?? null,\n          }),\n          throttle: isResume,\n          onError: (error) => {\n            err = error;\n          },\n        }),\n      );\n\n      let markedDelivered = false;\n\n      for await (const chunk of asAsyncIterableStream(stream)) {\n        if (chunk.metadata.unstable_state === agentStateRef.current) continue;\n\n        if (!markedDelivered) {\n          commandQueue.markDelivered();\n          markedDelivered = true;\n        }\n\n        agentStateRef.current = chunk.metadata.unstable_state as T;\n        rerender((prev) => prev + 1);\n      }\n\n      if (err) {\n        throw new Error(err);\n      }\n\n      // A successful run confirms delivery even when no state-changing\n      // chunk was observed.\n      if (!markedDelivered) {\n        commandQueue.markDelivered();\n      }\n\n      // commands that coalesced into this resume run must not starve\n      if (isResume && commandQueue.state.queued.length > 0) {\n        runManager.schedule();\n      }\n    },\n    onFinish: options.onFinish,\n    onCancel: () => {\n      setIsReplaying(false);\n      const cmds = [\n        ...commandQueue.state.inTransit,\n        ...commandQueue.state.queued,\n      ];\n\n      commandQueue.reset();\n\n      options.onCancel?.({\n        commands: cmds,\n        updateState: (updater) => {\n          agentStateRef.current = updater(agentStateRef.current);\n          rerender((prev) => prev + 1);\n        },\n      });\n    },\n    onError: async (error) => {\n      setIsReplaying(false);\n      const inTransitCmds = [...commandQueue.state.inTransit];\n      const queuedCmds = [...commandQueue.state.queued];\n\n      commandQueue.reset();\n\n      try {\n        await options.onError?.(error as Error, {\n          commands: inTransitCmds,\n          updateState: (updater) => {\n            agentStateRef.current = updater(agentStateRef.current);\n            rerender((prev) => prev + 1);\n          },\n        });\n      } finally {\n        options.onCancel?.({\n          commands: queuedCmds,\n          updateState: (updater) => {\n            agentStateRef.current = updater(agentStateRef.current);\n            rerender((prev) => prev + 1);\n          },\n          error: error as Error,\n        });\n      }\n    },\n  });\n\n  // Tool execution status state\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n\n  // Reactive conversion of agent state + connection metadata → UI state\n  const pendingCommands = useMemo(\n    () => [...commandQueue.state.inTransit, ...commandQueue.state.queued],\n    [commandQueue.state],\n  );\n  const converted = useConvertedState(\n    options.converter,\n    agentStateRef.current,\n    pendingCommands,\n    runManager.isRunning,\n    toolStatuses,\n  );\n\n  // Create runtime\n  const runtime = useExternalStoreRuntime({\n    messages: converted.messages,\n    state: converted.state,\n    isRunning: converted.isRunning,\n    isLoading: isReplaying,\n    adapters: options.adapters,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    extras: {\n      [symbolAssistantTransportExtras]: true,\n      sendCommand: (command: AssistantTransportCommand) => {\n        commandQueue.enqueue(command);\n      },\n      state: agentStateRef.current as UserExternalState,\n    } satisfies AssistantTransportExtras,\n    onNew: async (message: AppendMessage): Promise<void> => {\n      parentIdRef.current = message.parentId;\n      const command = convertAppendMessageToCommand(message);\n      commandQueue.enqueue(command, {\n        schedule: message.startRun ?? message.role === \"user\",\n      });\n    },\n    ...(options.capabilities?.edit && {\n      onEdit: async (message: AppendMessage): Promise<void> => {\n        parentIdRef.current = message.parentId;\n        const command = convertAppendMessageToCommand(message);\n        commandQueue.enqueue(command, {\n          schedule: message.startRun ?? message.role === \"user\",\n        });\n      },\n    }),\n    ...(commandQueue.state.queued.length > 0 && {\n      onReload: async (parentId: string | null) => {\n        parentIdRef.current = parentId;\n        runManager.schedule();\n      },\n    }),\n    onCancel: async () => {\n      runManager.cancel();\n    },\n    onResume: async () => {\n      if (!options.resumeApi)\n        throw new Error(\"Must pass resumeApi to options to resume runs\");\n\n      resumeFlagRef.current = true;\n      runManager.schedule();\n    },\n    onAddToolResult: async (\n      toolOptions: AddToolResultOptions,\n    ): Promise<void> => {\n      const command: AddToolResultCommand = {\n        type: \"add-tool-result\",\n        toolCallId: toolOptions.toolCallId,\n        result: toolOptions.result as ReadonlyJSONObject,\n        toolName: toolOptions.toolName,\n        isError: toolOptions.isError,\n        ...(toolOptions.artifact && { artifact: toolOptions.artifact }),\n        ...(toolOptions.modelContent !== undefined && {\n          modelContent: toolOptions.modelContent,\n        }),\n      };\n\n      commandQueue.enqueue(command);\n    },\n    onLoadExternalState: async (state) => {\n      agentStateRef.current = state as T;\n      rerender((prev) => prev + 1);\n    },\n  });\n\n  return runtime;\n};\n\n/**\n * @alpha This is an experimental API that is subject to change.\n */\nexport const useAssistantTransportRuntime = <T>(\n  options: AssistantTransportOptions<T>,\n): AssistantRuntime => {\n  const runtime = useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useAssistantTransportThreadRuntime(options);\n    },\n    adapter: new InMemoryThreadListAdapter(),\n    allowNesting: true,\n  });\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;AA0CA,MAAMqC,iCACJC,YACsB;CACtB,IAAIA,QAAQC,SAAS,QACnB,MAAM,IAAIC,MAAM,kCAAkC;CAEpD,MAAMC,QAA2B,CAAA;CACjC,MAAMC,UAAU,CACd,GAAGJ,QAAQI,SACX,GAAIJ,QAAQK,aAAaC,SAASC,MAAMA,EAAEH,OAAO,KAAK,CAAA,CAAG;CAE3D,KAAK,MAAMI,eAAeJ,SACxB,IAAII,YAAYC,SAAS,QACvBN,MAAMO,KAAK;EAAED,MAAM;EAAQE,MAAMH,YAAYG;CAAK,CAAC;MAC9C,IAAIH,YAAYC,SAAS,SAC9BN,MAAMO,KAAK;EAAED,MAAM;EAASG,OAAOJ,YAAYI;CAAM,CAAC;CAI1D,OAAO;EACLH,MAAM;EACNT,SAAS;GACPC,MAAM;GACNE;EACF;EACAU,UAAUb,QAAQa;EAClBC,UAAUd,QAAQc;CACpB;AACF;AAEA,MAAMC,iCAAiCC,OAAO,4BAA4B;AAO1E,MAAMK,8BACJC,WAC6B;CAC7B,IACE,OAAOA,WAAW,YAClBA,UAAU,QACV,EAAEP,kCAAkCO,SAEpC,MAAM,IAAIpB,MACR,gFACF;CAEF,OAAOoB;AACT;AAEA,MAAaC,yCAAmC;CAAA,MAAAC,IAAA7D,EAAA,CAAA;CAC9C,MAAA8D,MAAY7B,OAAO;CAAE,IAAA8B;CAAA,IAAAF,EAAA,OAAAC,IAAAE,QAAA;EAEdD,MAAAP,YAAA;GACL,MAAAG,SAAeG,IAAGE,OAAOC,SAAU,CAAC,CAAAN;GAEpCO,2BADmDP,MACnDO,CAAe,CAAAX,YAAaC,OAAO;EAAC;EACrCK,EAAA,KAAAC,IAAAE;EAAAH,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAA,OAJME;AAIN;AAOH,SAAgBI,2BACdC,YAA6CE,MAAMA,GAC5B;CACvB,OAAOpC,aAAaqC,MAClBH,SAASV,2BAA2Ba,EAAEP,OAAOL,MAAM,CAAC,CAACF,KAAK,CAC5D;AACF;AAEA,MAAMe,sCACJC,YACqB;CACrB,MAAMC,gBAAgBjE,OAAOgE,QAAQE,YAAY;CACjD,MAAM,GAAGC,YAAYlE,SAAS,CAAC;CAC/B,MAAMmE,gBAAgBpE,OAAO,KAAK;CAClC,MAAM,CAACqE,aAAaC,kBAAkBrE,SAAS,KAAK;CACpD,MAAMsE,sBAAsBtD,oBAAoB;CAChD,MAAMuD,cAAcxE,OAAkCyE,KAAAA,CAAS;CAC/D,MAAMC,eAAe3D,gBAAgB,EACnC4D,eAAeC,WAAWC,SAAS,EACrC,CAAC;CAED,MAAMC,WAAWrD,aAAaqC,MAAMA,EAAEiB,eAAeC,QAAQ;CAE7D,MAAMJ,aAAa1D,cAAc;EAC/B+D,OAAO,OAAOC,WAAwB;GACpC,MAAME,WAAWhB,cAAciB;GAC/BjB,cAAciB,UAAU;GACxBf,eAAe,KAAK;GACpB,MAAMgB,WAA4BF,WAAW,CAAA,IAAKV,aAAaa,MAAM;GACrE,IAAID,SAASE,WAAW,KAAK,CAACJ,UAAU;GAKxC,MAAM3C,WAAW2C,WAAWX,KAAAA,IAAYD,YAAYa;GACpD,IAAI,CAACD,UAAUZ,YAAYa,UAAUZ,KAAAA;GAErC,MAAMgB,UAAU,MAAMpE,qBAAqB2C,QAAQyB,OAAO;GAC1D,MAAMC,YACJ,OAAO1B,QAAQ2B,SAAS,aACpB,MAAM3B,QAAQ2B,KAAK,IACnB3B,QAAQ2B;GACd,MAAMC,UAAUC,QAAQtC,OAAOuC,gBAAgB;GAE/C,IAAIC,cAAuC;IACzCT;IACAtC,OAAOiB,cAAcoB;IACrBY,QAAQL,QAAQK;IAChBC,OAAON,QAAQM,QAAQ3F,kBAAkBqF,QAAQM,KAAK,IAAIzB,KAAAA;IAC1DK;IACA,GAAIrC,aAAagC,KAAAA,KAAa,EAC5BhC,SACF;IAEA0D,cAAcP,QAAQO;IACtBC,QAAQR,QAAQQ;IAEhB,GAAGR,QAAQO;IACX,GAAGP,QAAQQ;IACX,GAAIV,aAAa,CAAC;GACpB;GAEA,IAAI1B,QAAQqC,4BACVN,cAAc,MAAM/B,QAAQqC,2BAC1BN,WACF;GAGF,MAAMO,WAAW,MAAMC,MACrBnB,WAAWpB,QAAQwC,YAAaxC,QAAQyC,KACxC;IACEC,QAAQ;IACRjB;IACAE,MAAMgB,KAAKC,UAAUb,WAAW;IAChCb;GACF,CACF;GAEAlB,QAAQ6C,aAAaP,QAAQ;GAE7B,IAAI,CAACA,SAASQ,IACZ,MAAM,IAAIhF,MAAM,UAAUwE,SAASS,OAAM,IAAK,MAAMT,SAAS/D,KAAK,GAAG;GAGvE,IAAI,CAAC+D,SAASX,MACZ,MAAM,IAAI7D,MAAM,uBAAuB;GAGzC,MAAM6D,OAAO,MAAM3E,2BAA2BsF,UAAU;IACtDU,cAAc1C;IACd2C,eAAe1C;GACjB,CAAC;GAID,MAAM4C,WADWnD,QAAQkD,YAAY,mBAEtB,wBACT,IAAI9G,0BAA0B,IAC9B,IAAID,kBAAkB;GAE5B,IAAIiH;GACJ,MAAMC,SAAS1B,KAAK2B,YAAYH,OAAO,CAAC,CAACG,YACvC,IAAIpH,4BAA4B;IAC9BqH,gBAAgBjH,8BAAqB,EACnCkH,gBACGvD,cAAcoB,WAAiC,KACpD,CAAC;IACDoC,UAAUrC;IACVsC,UAAUC,UAAU;KAClBP,MAAMO;IACR;GACF,CAAC,CACH;GAEA,IAAIC,kBAAkB;GAEtB,WAAW,MAAMC,SAASlI,sBAAsB0H,MAAM,GAAG;IACvD,IAAIQ,MAAMC,SAASN,mBAAmBvD,cAAcoB,SAAS;IAE7D,IAAI,CAACuC,iBAAiB;KACpBlD,aAAaqD,cAAc;KAC3BH,kBAAkB;IACpB;IAEA3D,cAAcoB,UAAUwC,MAAMC,SAASN;IACvCrD,UAAU6D,SAASA,OAAO,CAAC;GAC7B;GAEA,IAAIZ,KACF,MAAM,IAAItF,MAAMsF,GAAG;GAKrB,IAAI,CAACQ,iBACHlD,aAAaqD,cAAc;GAI7B,IAAI3C,YAAYV,aAAa1B,MAAMiF,OAAOzC,SAAS,GACjDZ,WAAWC,SAAS;EAExB;EACAqD,UAAUlE,QAAQkE;EAClBC,gBAAgB;GACd7D,eAAe,KAAK;GACpB,MAAM8D,OAAO,CACX,GAAG1D,aAAa1B,MAAMqF,WACtB,GAAG3D,aAAa1B,MAAMiF,MAAM;GAG9BvD,aAAa4D,MAAM;GAEnBtE,QAAQmE,WAAW;IACjB7C,UAAU8C;IACVG,cAAcC,YAAY;KACxBvE,cAAcoB,UAAUmD,QAAQvE,cAAcoB,OAAO;KACrDlB,UAAU6D,SAASA,OAAO,CAAC;IAC7B;GACF,CAAC;EACH;EACAN,SAAS,OAAOC,YAAU;GACxBrD,eAAe,KAAK;GACpB,MAAMmE,gBAAgB,CAAC,GAAG/D,aAAa1B,MAAMqF,SAAS;GACtD,MAAMK,aAAa,CAAC,GAAGhE,aAAa1B,MAAMiF,MAAM;GAEhDvD,aAAa4D,MAAM;GAEnB,IAAI;IACF,MAAMtE,QAAQ0D,UAAUC,SAAgB;KACtCrC,UAAUmD;KACVF,cAAcC,YAAY;MACxBvE,cAAcoB,UAAUmD,QAAQvE,cAAcoB,OAAO;MACrDlB,UAAU6D,SAASA,OAAO,CAAC;KAC7B;IACF,CAAC;GACH,UAAU;IACRhE,QAAQmE,WAAW;KACjB7C,UAAUoD;KACVH,cAAcC,YAAY;MACxBvE,cAAcoB,UAAUmD,QAAQvE,cAAcoB,OAAO;MACrDlB,UAAU6D,SAASA,OAAO,CAAC;KAC7B;KACAL,OAAOA;IACT,CAAC;GACH;EACF;CACF,CAAC;CAGD,MAAM,CAACgB,cAAcC,mBAAmB3I,SAEtC,CAAC,CAAC;CAGJ,MAAM4I,kBAAkB9I,cAChB,CAAC,GAAG2E,aAAa1B,MAAMqF,WAAW,GAAG3D,aAAa1B,MAAMiF,MAAM,GACpE,CAACvD,aAAa1B,KAAK,CACrB;CACA,MAAM8F,YAAY3H,kBAChB6C,QAAQ+E,WACR9E,cAAcoB,SACdwD,iBACAjE,WAAWoE,WACXL,YACF;CAGA,MAAM9C,UAAUjG,wBAAwB;EACtCqJ,UAAUH,UAAUG;EACpBjG,OAAO8F,UAAU9F;EACjBgG,WAAWF,UAAUE;EACrBE,WAAW7E;EACX8E,UAAUnF,QAAQmF;EAClBC,gCAAgC;EAChCR;EACA1F,QAAQ;IACLP,iCAAiC;GAClCG,cAAcC,YAAuC;IACnD2B,aAAa2E,QAAQtG,OAAO;GAC9B;GACAC,OAAOiB,cAAcoB;EACvB;EACAiE,OAAO,OAAO1H,YAA0C;GACtD4C,YAAYa,UAAUzD,QAAQa;GAC9B,MAAMM,YAAUpB,8BAA8BC,OAAO;GACrD8C,aAAa2E,QAAQtG,WAAS,EAC5B8B,UAAUjD,QAAQ4H,YAAY5H,QAAQC,SAAS,OACjD,CAAC;EACH;EACA,GAAImC,QAAQyF,cAAcC,QAAQ,EAChCC,QAAQ,OAAO/H,cAA0C;GACvD4C,YAAYa,UAAUzD,UAAQa;GAC9B,MAAMM,YAAUpB,8BAA8BC,SAAO;GACrD8C,aAAa2E,QAAQtG,WAAS,EAC5B8B,UAAUjD,UAAQ4H,YAAY5H,UAAQC,SAAS,OACjD,CAAC;EACH,EACF;EACA,GAAI6C,aAAa1B,MAAMiF,OAAOzC,SAAS,KAAK,EAC1CoE,UAAU,OAAOnH,eAA4B;GAC3C+B,YAAYa,UAAU5C;GACtBmC,WAAWC,SAAS;EACtB,EACF;EACAsD,UAAU,YAAY;GACpBvD,WAAWiF,OAAO;EACpB;EACAC,UAAU,YAAY;GACpB,IAAI,CAAC9F,QAAQwC,WACX,MAAM,IAAI1E,MAAM,+CAA+C;GAEjEsC,cAAciB,UAAU;GACxBT,WAAWC,SAAS;EACtB;EACAkF,iBAAiB,OACfC,gBACkB;GAClB,MAAMjH,YAAgC;IACpCV,MAAM;IACN4H,YAAYD,YAAYC;IACxBC,QAAQF,YAAYE;IACpBC,UAAUH,YAAYG;IACtBC,SAASJ,YAAYI;IACrB,GAAIJ,YAAYK,YAAY,EAAEA,UAAUL,YAAYK,SAAS;IAC7D,GAAIL,YAAYM,iBAAiB7F,KAAAA,KAAa,EAC5C6F,cAAcN,YAAYM,aAC5B;GACF;GAEA5F,aAAa2E,QAAQtG,SAAO;EAC9B;EACAwH,qBAAqB,OAAOvH,UAAU;GACpCiB,cAAcoB,UAAUrC;GACxBmB,UAAU6D,WAASA,SAAO,CAAC;EAC7B;CACF,CAAC;CAED,OAAOnC;AACT;;;;AAKA,MAAa2E,gCACXxG,YACqB;CAQrB,OAPgB1C,2BAA2B;EACzCmJ,aAAa,SAASC,cAAc;GAClC,OAAO3G,mCAAmCC,OAAO;EACnD;EACA2G,SAAS,IAAIpJ,0BAA0B;EACvCqJ,cAAc;CAChB,CACO/E;AACT"}
+{"version":3,"file":"useAssistantTransportRuntime.js","names":["c","_c","AppendMessage","ReadonlyJSONObject","ReadonlyJSONValue","asAsyncIterableStream","useExternalStoreRuntime","AssistantRuntime","AddToolResultOptions","useMemo","useRef","useState","AssistantMessageAccumulator","DataStreamDecoder","AssistantTransportDecoder","unstable_createInitialMessage","createInitialMessage","toToolsJSONSchema","AssistantTransportOptions","AddMessageCommand","AddToolResultCommand","UserMessagePart","QueuedCommand","AssistantTransportCommand","SendCommandsRequestBody","useCommandQueue","createReplayBoundaryStream","useReplayRenderWait","useRunManager","useConvertedState","ToolExecutionStatus","createRequestHeaders","useRemoteThreadListRuntime","InMemoryThreadListAdapter","useAui","useAuiState","UserExternalState","convertAppendMessageToCommand","message","role","Error","parts","content","attachments","flatMap","a","contentPart","type","push","text","image","length","parentId","sourceId","readResumeState","response","Response","Promise","runId","state","T","status","ok","value","json","symbolAssistantTransportExtras","Symbol","AssistantTransportExtras","sendCommand","command","asAssistantTransportExtras","extras","useAssistantTransportSendCommand","$","aui","t0","thread","getState","transportExtras","useAssistantTransportState","selector","t","s","useAssistantTransportThreadRuntime","options","agentStateRef","initialState","rerender","resumeFlagRef","isReplaying","setIsReplaying","waitForReplayRender","parentIdRef","undefined","commandQueue","onQueue","runManager","schedule","enqueueAppendMessage","console","warn","current","enqueue","startRun","threadId","threadListItem","remoteId","onRun","signal","AbortSignal","isResume","commands","flush","headers","resumeState","resumeStateApi","resumeStateResponse","fetch","method","body","JSON","stringify","retained","queued","bodyValue","context","runtime","getModelContext","requestBody","Record","system","tools","callSettings","config","prepareSendCommandsRequest","resumeApi","api","onResponse","prev","setReplaying","waitForRender","protocol","strict","decoder","err","stream","pipeThrough","initialMessage","unstable_state","throttle","onError","error","markedDelivered","chunk","metadata","markDelivered","onFinish","onCancel","cmds","inTransit","reset","updateState","updater","inTransitCmds","queuedCmds","toolStatuses","setToolStatuses","pendingCommands","converted","converter","isRunning","messages","isLoading","adapters","unstable_enableToolInvocations","onNew","capabilities","edit","onEdit","onReload","cancel","onResume","onAddToolResult","toolOptions","toolCallId","result","toolName","isError","artifact","modelContent","onLoadExternalState","useAssistantTransportRuntime","runtimeHook","RuntimeHook","adapter","allowNesting"],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport type { AppendMessage } from \"@assistant-ui/core\";\nimport {\n  type ReadonlyJSONObject,\n  type ReadonlyJSONValue,\n  asAsyncIterableStream,\n} from \"assistant-stream/utils\";\nimport { useExternalStoreRuntime } from \"../external-store/useExternalStoreRuntime\";\nimport type { AssistantRuntime } from \"../../runtime/AssistantRuntime\";\nimport type { AddToolResultOptions } from \"@assistant-ui/core\";\nimport { useMemo, useRef, useState } from \"react\";\nimport {\n  AssistantMessageAccumulator,\n  DataStreamDecoder,\n  AssistantTransportDecoder,\n  unstable_createInitialMessage as createInitialMessage,\n  toToolsJSONSchema,\n} from \"assistant-stream\";\nimport type {\n  AssistantTransportOptions,\n  AddMessageCommand,\n  AddToolResultCommand,\n  UserMessagePart,\n  QueuedCommand,\n  AssistantTransportCommand,\n  SendCommandsRequestBody,\n} from \"./types\";\nimport { useCommandQueue } from \"./commandQueue\";\nimport {\n  createReplayBoundaryStream,\n  useReplayRenderWait,\n} from \"./replayBoundaryStream\";\nimport { useRunManager } from \"./runManager\";\nimport { useConvertedState } from \"./useConvertedState\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport { createRequestHeaders } from \"@assistant-ui/core\";\nimport { useRemoteThreadListRuntime } from \"../remote-thread-list/useRemoteThreadListRuntime\";\nimport { InMemoryThreadListAdapter } from \"@assistant-ui/core\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { UserExternalState } from \"../../../augmentations\";\n\nconst convertAppendMessageToCommand = (\n  message: AppendMessage,\n): AddMessageCommand | null => {\n  if (message.role !== \"user\")\n    throw new Error(\"Only user messages are supported\");\n\n  const parts: UserMessagePart[] = [];\n  const content = [\n    ...message.content,\n    ...(message.attachments?.flatMap((a) => a.content) ?? []),\n  ];\n  for (const contentPart of content) {\n    if (contentPart.type === \"text\") {\n      parts.push({ type: \"text\", text: contentPart.text });\n    } else if (contentPart.type === \"image\") {\n      parts.push({ type: \"image\", image: contentPart.image });\n    }\n  }\n\n  if (parts.length === 0) return null;\n\n  return {\n    type: \"add-message\",\n    message: {\n      role: \"user\",\n      parts,\n    },\n    parentId: message.parentId,\n    sourceId: message.sourceId,\n  };\n};\n\nconst readResumeState = async <T>(\n  response: Response,\n): Promise<{ runId: string; state: T } | null> => {\n  if (response.status === 204) return null;\n  if (!response.ok) {\n    throw new Error(\n      `Resume state request failed with status ${response.status}: ${await response.text()}`,\n    );\n  }\n\n  let value: unknown;\n  try {\n    value = await response.json();\n  } catch {\n    throw new Error(\"Resume state response was not valid JSON\");\n  }\n\n  if (\n    typeof value !== \"object\" ||\n    value === null ||\n    !(\"state\" in value) ||\n    !(\"runId\" in value) ||\n    typeof value.runId !== \"string\"\n  ) {\n    throw new Error(\"Resume state response must contain state and runId\");\n  }\n\n  return { runId: value.runId, state: value.state as T };\n};\n\nconst symbolAssistantTransportExtras = Symbol(\"assistant-transport-extras\");\ntype AssistantTransportExtras = {\n  [symbolAssistantTransportExtras]: true;\n  sendCommand: (command: AssistantTransportCommand) => void;\n  state: UserExternalState;\n};\n\nconst asAssistantTransportExtras = (\n  extras: unknown,\n): AssistantTransportExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolAssistantTransportExtras in extras)\n  )\n    throw new Error(\n      \"This method can only be called when you are using useAssistantTransportRuntime\",\n    );\n\n  return extras as AssistantTransportExtras;\n};\n\nexport const useAssistantTransportSendCommand = () => {\n  const aui = useAui();\n\n  return (command: AssistantTransportCommand) => {\n    const extras = aui.thread.getState().extras;\n    const transportExtras = asAssistantTransportExtras(extras);\n    transportExtras.sendCommand(command);\n  };\n};\n\nexport function useAssistantTransportState(): UserExternalState;\nexport function useAssistantTransportState<T>(\n  selector: (state: UserExternalState) => T,\n): T;\nexport function useAssistantTransportState<T>(\n  selector: (state: UserExternalState) => T = (t) => t as T,\n): T | UserExternalState {\n  return useAuiState((s) =>\n    selector(asAssistantTransportExtras(s.thread.extras).state),\n  );\n}\n\nconst useAssistantTransportThreadRuntime = <T>(\n  options: AssistantTransportOptions<T>,\n): AssistantRuntime => {\n  const agentStateRef = useRef(options.initialState);\n  const [, rerender] = useState(0);\n  const resumeFlagRef = useRef(false);\n  const [isReplaying, setIsReplaying] = useState(false);\n  const waitForReplayRender = useReplayRenderWait();\n  const parentIdRef = useRef<string | null | undefined>(undefined);\n  const commandQueue = useCommandQueue({\n    onQueue: () => runManager.schedule(),\n  });\n\n  const enqueueAppendMessage = (message: AppendMessage) => {\n    const command = convertAppendMessageToCommand(message);\n    if (!command) {\n      console.warn(\n        \"[assistant-ui] Skipped add-message command with no supported parts\",\n      );\n      return;\n    }\n    parentIdRef.current = message.parentId;\n    commandQueue.enqueue(command, {\n      schedule: message.startRun ?? message.role === \"user\",\n    });\n  };\n\n  const threadId = useAuiState((s) => s.threadListItem.remoteId);\n\n  const runManager = useRunManager({\n    onRun: async (signal: AbortSignal) => {\n      const isResume = resumeFlagRef.current;\n      resumeFlagRef.current = false;\n      setIsReplaying(false);\n      const commands: QueuedCommand[] = isResume ? [] : commandQueue.flush();\n      if (commands.length === 0 && !isResume) return;\n\n      // The flushed batch consumes the parentId; read it alongside the flush\n      // (before any awaits) so a mid-run append keeps its own value. Resume\n      // runs send no commands, so they neither send nor consume it.\n      const parentId = isResume ? undefined : parentIdRef.current;\n      if (!isResume) parentIdRef.current = undefined;\n\n      const headers = await createRequestHeaders(options.headers);\n      let resumeState: { runId: string; state: T } | undefined;\n      if (isResume && options.resumeStateApi) {\n        const resumeStateResponse = await fetch(options.resumeStateApi, {\n          method: \"POST\",\n          headers,\n          body: JSON.stringify({ threadId }),\n          signal,\n        });\n        const retained = await readResumeState<T>(resumeStateResponse);\n        if (retained === null) {\n          if (commandQueue.state.queued.length > 0) {\n            runManager.schedule();\n          }\n          return;\n        }\n        resumeState = retained;\n      }\n\n      const bodyValue =\n        typeof options.body === \"function\"\n          ? await options.body()\n          : options.body;\n      const context = runtime.thread.getModelContext();\n\n      let requestBody: Record<string, unknown> = {\n        commands,\n        ...(resumeState === undefined && { state: agentStateRef.current }),\n        system: context.system,\n        tools: context.tools ? toToolsJSONSchema(context.tools) : undefined,\n        threadId,\n        ...(parentId !== undefined && {\n          parentId,\n        }),\n        // nested (new format, aligned with AssistantChatTransport)\n        callSettings: context.callSettings,\n        config: context.config,\n        // @deprecated spread at top level — use nested `callSettings`/`config` instead. Will be removed in a future version.\n        ...context.callSettings,\n        ...context.config,\n        ...(bodyValue ?? {}),\n      };\n\n      if (options.prepareSendCommandsRequest) {\n        requestBody = await options.prepareSendCommandsRequest(\n          requestBody as SendCommandsRequestBody,\n        );\n      }\n\n      if (resumeState !== undefined) {\n        // The server replays a resume from the snapshot it retained for this\n        // runId. Body overrides and prepare hooks can neither substitute a\n        // state nor drop the ID the server validates against.\n        requestBody = { ...requestBody, runId: resumeState.runId };\n        delete requestBody[\"state\"];\n      }\n\n      const response = await fetch(\n        isResume ? options.resumeApi! : options.api,\n        {\n          method: \"POST\",\n          headers,\n          body: JSON.stringify(requestBody),\n          signal,\n        },\n      );\n\n      options.onResponse?.(response);\n\n      if (!response.ok) {\n        throw new Error(`Status ${response.status}: ${await response.text()}`);\n      }\n\n      if (!response.body) {\n        throw new Error(\"Response body is null\");\n      }\n\n      if (resumeState !== undefined) {\n        agentStateRef.current = resumeState.state;\n        rerender((prev) => prev + 1);\n      }\n\n      const body = await createReplayBoundaryStream(response, {\n        setReplaying: setIsReplaying,\n        waitForRender: waitForReplayRender,\n      });\n\n      // Select decoder based on protocol option\n      const protocol = options.protocol ?? \"data-stream\";\n      // Resume replays a best-effort buffer; always reconcile leniently.\n      const strict = isResume ? false : (options.strict ?? true);\n      const decoder =\n        protocol === \"assistant-transport\"\n          ? new AssistantTransportDecoder({ strict })\n          : new DataStreamDecoder({ strict });\n\n      let err: string | undefined;\n      const stream = body.pipeThrough(decoder).pipeThrough(\n        new AssistantMessageAccumulator({\n          initialMessage: createInitialMessage({\n            unstable_state:\n              (agentStateRef.current as ReadonlyJSONValue) ?? null,\n          }),\n          throttle: isResume,\n          strict,\n          onError: (error) => {\n            err = error;\n          },\n        }),\n      );\n\n      let markedDelivered = false;\n\n      for await (const chunk of asAsyncIterableStream(stream)) {\n        if (chunk.metadata.unstable_state === agentStateRef.current) continue;\n\n        if (!markedDelivered) {\n          commandQueue.markDelivered();\n          markedDelivered = true;\n        }\n\n        agentStateRef.current = chunk.metadata.unstable_state as T;\n        rerender((prev) => prev + 1);\n      }\n\n      if (err) {\n        throw new Error(err);\n      }\n\n      // A successful run confirms delivery even when no state-changing\n      // chunk was observed.\n      if (!markedDelivered) {\n        commandQueue.markDelivered();\n      }\n\n      // commands that coalesced into this resume run must not starve\n      if (isResume && commandQueue.state.queued.length > 0) {\n        runManager.schedule();\n      }\n    },\n    onFinish: options.onFinish,\n    onCancel: () => {\n      setIsReplaying(false);\n      const cmds = [\n        ...commandQueue.state.inTransit,\n        ...commandQueue.state.queued,\n      ];\n\n      commandQueue.reset();\n\n      options.onCancel?.({\n        commands: cmds,\n        updateState: (updater) => {\n          agentStateRef.current = updater(agentStateRef.current);\n          rerender((prev) => prev + 1);\n        },\n      });\n    },\n    onError: async (error) => {\n      setIsReplaying(false);\n      const inTransitCmds = [...commandQueue.state.inTransit];\n      const queuedCmds = [...commandQueue.state.queued];\n\n      commandQueue.reset();\n\n      try {\n        await options.onError?.(error as Error, {\n          commands: inTransitCmds,\n          updateState: (updater) => {\n            agentStateRef.current = updater(agentStateRef.current);\n            rerender((prev) => prev + 1);\n          },\n        });\n      } finally {\n        options.onCancel?.({\n          commands: queuedCmds,\n          updateState: (updater) => {\n            agentStateRef.current = updater(agentStateRef.current);\n            rerender((prev) => prev + 1);\n          },\n          error: error as Error,\n        });\n      }\n    },\n  });\n\n  // Tool execution status state\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n\n  // Reactive conversion of agent state + connection metadata → UI state\n  const pendingCommands = useMemo(\n    () => [...commandQueue.state.inTransit, ...commandQueue.state.queued],\n    [commandQueue.state],\n  );\n  const converted = useConvertedState(\n    options.converter,\n    agentStateRef.current,\n    pendingCommands,\n    runManager.isRunning,\n    toolStatuses,\n  );\n\n  // Create runtime\n  const runtime = useExternalStoreRuntime({\n    messages: converted.messages,\n    state: converted.state,\n    isRunning: converted.isRunning,\n    isLoading: isReplaying,\n    adapters: options.adapters,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    extras: {\n      [symbolAssistantTransportExtras]: true,\n      sendCommand: (command: AssistantTransportCommand) => {\n        commandQueue.enqueue(command);\n      },\n      state: agentStateRef.current as UserExternalState,\n    } satisfies AssistantTransportExtras,\n    onNew: async (message: AppendMessage): Promise<void> =>\n      enqueueAppendMessage(message),\n    ...(options.capabilities?.edit && {\n      onEdit: async (message: AppendMessage): Promise<void> =>\n        enqueueAppendMessage(message),\n    }),\n    ...(commandQueue.state.queued.length > 0 && {\n      onReload: async (parentId: string | null) => {\n        parentIdRef.current = parentId;\n        runManager.schedule();\n      },\n    }),\n    onCancel: async () => {\n      runManager.cancel();\n    },\n    onResume: async () => {\n      if (!options.resumeApi)\n        throw new Error(\"Must pass resumeApi to options to resume runs\");\n\n      resumeFlagRef.current = true;\n      runManager.schedule();\n    },\n    onAddToolResult: async (\n      toolOptions: AddToolResultOptions,\n    ): Promise<void> => {\n      const command: AddToolResultCommand = {\n        type: \"add-tool-result\",\n        toolCallId: toolOptions.toolCallId,\n        result: toolOptions.result as ReadonlyJSONObject,\n        toolName: toolOptions.toolName,\n        isError: toolOptions.isError,\n        ...(toolOptions.artifact && { artifact: toolOptions.artifact }),\n        ...(toolOptions.modelContent !== undefined && {\n          modelContent: toolOptions.modelContent,\n        }),\n      };\n\n      commandQueue.enqueue(command);\n    },\n    onLoadExternalState: async (state) => {\n      agentStateRef.current = state as T;\n      rerender((prev) => prev + 1);\n    },\n  });\n\n  return runtime;\n};\n\n/**\n * @alpha This is an experimental API that is subject to change.\n */\nexport const useAssistantTransportRuntime = <T>(\n  options: AssistantTransportOptions<T>,\n): AssistantRuntime => {\n  const runtime = useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useAssistantTransportThreadRuntime(options);\n    },\n    adapter: new InMemoryThreadListAdapter(),\n    allowNesting: true,\n  });\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;AA0CA,MAAMqC,iCACJC,YAC6B;CAC7B,IAAIA,QAAQC,SAAS,QACnB,MAAM,IAAIC,MAAM,kCAAkC;CAEpD,MAAMC,QAA2B,CAAA;CACjC,MAAMC,UAAU,CACd,GAAGJ,QAAQI,SACX,GAAIJ,QAAQK,aAAaC,SAASC,MAAMA,EAAEH,OAAO,KAAK,CAAA,CAAG;CAE3D,KAAK,MAAMI,eAAeJ,SACxB,IAAII,YAAYC,SAAS,QACvBN,MAAMO,KAAK;EAAED,MAAM;EAAQE,MAAMH,YAAYG;CAAK,CAAC;MAC9C,IAAIH,YAAYC,SAAS,SAC9BN,MAAMO,KAAK;EAAED,MAAM;EAASG,OAAOJ,YAAYI;CAAM,CAAC;CAI1D,IAAIT,MAAMU,WAAW,GAAG,OAAO;CAE/B,OAAO;EACLJ,MAAM;EACNT,SAAS;GACPC,MAAM;GACNE;EACF;EACAW,UAAUd,QAAQc;EAClBC,UAAUf,QAAQe;CACpB;AACF;AAEA,MAAMC,kBAAkB,OACtBC,aACgD;CAChD,IAAIA,SAASM,WAAW,KAAK,OAAO;CACpC,IAAI,CAACN,SAASO,IACZ,MAAM,IAAItB,MACR,2CAA2Ce,SAASM,OAAM,IAAK,MAAMN,SAASN,KAAK,GACrF;CAGF,IAAIc;CACJ,IAAI;EACFA,QAAQ,MAAMR,SAASS,KAAK;CAC9B,QAAQ;EACN,MAAM,IAAIxB,MAAM,0CAA0C;CAC5D;CAEA,IACE,OAAOuB,UAAU,YACjBA,UAAU,QACV,EAAE,WAAWA,UACb,EAAE,WAAWA,UACb,OAAOA,MAAML,UAAU,UAEvB,MAAM,IAAIlB,MAAM,oDAAoD;CAGtE,OAAO;EAAEkB,OAAOK,MAAML;EAAOC,OAAOI,MAAMJ;CAAW;AACvD;AAEA,MAAMM,iCAAiCC,OAAO,4BAA4B;AAO1E,MAAMI,8BACJC,WAC6B;CAC7B,IACE,OAAOA,WAAW,YAClBA,UAAU,QACV,EAAEN,kCAAkCM,SAEpC,MAAM,IAAI/B,MACR,gFACF;CAEF,OAAO+B;AACT;AAEA,MAAaC,yCAAmC;CAAA,MAAAC,IAAAxE,EAAA,CAAA;CAC9C,MAAAyE,MAAYxC,OAAO;CAAE,IAAAyC;CAAA,IAAAF,EAAA,OAAAC,IAAAE,QAAA;EAEdD,MAAAN,YAAA;GACL,MAAAE,SAAeG,IAAGE,OAAOC,SAAU,CAAC,CAAAN;GAEpCO,2BADmDP,MACnDO,CAAe,CAAAV,YAAaC,OAAO;EAAC;EACrCI,EAAA,KAAAC,IAAAE;EAAAH,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAA,OAJME;AAIN;AAOH,SAAgBI,2BACdC,YAA6CC,MAAMA,GAC5B;CACvB,OAAO9C,aAAa+C,MAClBF,SAASV,2BAA2BY,EAAEN,OAAOL,MAAM,CAAC,CAACZ,KAAK,CAC5D;AACF;AAEA,MAAMwB,sCACJC,YACqB;CACrB,MAAMC,gBAAgB3E,OAAO0E,QAAQE,YAAY;CACjD,MAAM,GAAGC,YAAY5E,SAAS,CAAC;CAC/B,MAAM6E,gBAAgB9E,OAAO,KAAK;CAClC,MAAM,CAAC+E,aAAaC,kBAAkB/E,SAAS,KAAK;CACpD,MAAMgF,sBAAsBhE,oBAAoB;CAChD,MAAMiE,cAAclF,OAAkCmF,KAAAA,CAAS;CAC/D,MAAMC,eAAerE,gBAAgB,EACnCsE,eAAeC,WAAWC,SAAS,EACrC,CAAC;CAED,MAAMC,wBAAwB5D,YAA2B;EACvD,MAAM+B,UAAUhC,8BAA8BC,OAAO;EACrD,IAAI,CAAC+B,SAAS;GACZ8B,QAAQC,KACN,oEACF;GACA;EACF;EACAR,YAAYS,UAAU/D,QAAQc;EAC9B0C,aAAaQ,QAAQjC,SAAS,EAC5B4B,UAAU3D,QAAQiE,YAAYjE,QAAQC,SAAS,OACjD,CAAC;CACH;CAEA,MAAMiE,WAAWrE,aAAa+C,MAAMA,EAAEuB,eAAeC,QAAQ;CAE7D,MAAMV,aAAapE,cAAc;EAC/B+E,OAAO,OAAOC,WAAwB;GACpC,MAAME,WAAWtB,cAAca;GAC/Bb,cAAca,UAAU;GACxBX,eAAe,KAAK;GACpB,MAAMqB,WAA4BD,WAAW,CAAA,IAAKhB,aAAakB,MAAM;GACrE,IAAID,SAAS5D,WAAW,KAAK,CAAC2D,UAAU;GAKxC,MAAM1D,WAAW0D,WAAWjB,KAAAA,IAAYD,YAAYS;GACpD,IAAI,CAACS,UAAUlB,YAAYS,UAAUR,KAAAA;GAErC,MAAMoB,UAAU,MAAMlF,qBAAqBqD,QAAQ6B,OAAO;GAC1D,IAAIC;GACJ,IAAIJ,YAAY1B,QAAQ+B,gBAAgB;IACtC,MAAMC,sBAAsB,MAAMC,MAAMjC,QAAQ+B,gBAAgB;KAC9DG,QAAQ;KACRL;KACAM,MAAMC,KAAKC,UAAU,EAAEjB,SAAS,CAAC;KACjCI;IACF,CAAC;IACD,MAAMc,WAAW,MAAMpE,gBAAmB8D,mBAAmB;IAC7D,IAAIM,aAAa,MAAM;KACrB,IAAI5B,aAAanC,MAAMgE,OAAOxE,SAAS,GACrC6C,WAAWC,SAAS;KAEtB;IACF;IACAiB,cAAcQ;GAChB;GAEA,MAAME,YACJ,OAAOxC,QAAQmC,SAAS,aACpB,MAAMnC,QAAQmC,KAAK,IACnBnC,QAAQmC;GACd,MAAMM,UAAUC,QAAQlD,OAAOmD,gBAAgB;GAE/C,IAAIC,cAAuC;IACzCjB;IACA,GAAIG,gBAAgBrB,KAAAA,KAAa,EAAElC,OAAO0B,cAAcgB,QAAQ;IAChE6B,QAAQL,QAAQK;IAChBC,OAAON,QAAQM,QAAQlH,kBAAkB4G,QAAQM,KAAK,IAAItC,KAAAA;IAC1DW;IACA,GAAIpD,aAAayC,KAAAA,KAAa,EAC5BzC,SACF;IAEAgF,cAAcP,QAAQO;IACtBC,QAAQR,QAAQQ;IAEhB,GAAGR,QAAQO;IACX,GAAGP,QAAQQ;IACX,GAAIT,aAAa,CAAC;GACpB;GAEA,IAAIxC,QAAQkD,4BACVN,cAAc,MAAM5C,QAAQkD,2BAC1BN,WACF;GAGF,IAAId,gBAAgBrB,KAAAA,GAAW;IAI7BmC,cAAc;KAAE,GAAGA;KAAatE,OAAOwD,YAAYxD;IAAM;IACzD,OAAOsE,YAAY;GACrB;GAEA,MAAMzE,WAAW,MAAM8D,MACrBP,WAAW1B,QAAQmD,YAAanD,QAAQoD,KACxC;IACElB,QAAQ;IACRL;IACAM,MAAMC,KAAKC,UAAUO,WAAW;IAChCpB;GACF,CACF;GAEAxB,QAAQqD,aAAalF,QAAQ;GAE7B,IAAI,CAACA,SAASO,IACZ,MAAM,IAAItB,MAAM,UAAUe,SAASM,OAAM,IAAK,MAAMN,SAASN,KAAK,GAAG;GAGvE,IAAI,CAACM,SAASgE,MACZ,MAAM,IAAI/E,MAAM,uBAAuB;GAGzC,IAAI0E,gBAAgBrB,KAAAA,GAAW;IAC7BR,cAAcgB,UAAUa,YAAYvD;IACpC4B,UAAUmD,SAASA,OAAO,CAAC;GAC7B;GAEA,MAAMnB,OAAO,MAAM7F,2BAA2B6B,UAAU;IACtDoF,cAAcjD;IACdkD,eAAejD;GACjB,CAAC;GAGD,MAAMkD,WAAWzD,QAAQyD,YAAY;GAErC,MAAMC,SAAShC,WAAW,QAAS1B,QAAQ0D,UAAU;GACrD,MAAMC,UACJF,aAAa,wBACT,IAAI/H,0BAA0B,EAAEgI,OAAO,CAAC,IACxC,IAAIjI,kBAAkB,EAAEiI,OAAO,CAAC;GAEtC,IAAIE;GACJ,MAAMC,SAAS1B,KAAK2B,YAAYH,OAAO,CAAC,CAACG,YACvC,IAAItI,4BAA4B;IAC9BuI,gBAAgBnI,8BAAqB,EACnCoI,gBACG/D,cAAcgB,WAAiC,KACpD,CAAC;IACDgD,UAAUvC;IACVgC;IACAQ,UAAUC,UAAU;KAClBP,MAAMO;IACR;GACF,CAAC,CACH;GAEA,IAAIC,kBAAkB;GAEtB,WAAW,MAAMC,SAASpJ,sBAAsB4I,MAAM,GAAG;IACvD,IAAIQ,MAAMC,SAASN,mBAAmB/D,cAAcgB,SAAS;IAE7D,IAAI,CAACmD,iBAAiB;KACpB1D,aAAa6D,cAAc;KAC3BH,kBAAkB;IACpB;IAEAnE,cAAcgB,UAAUoD,MAAMC,SAASN;IACvC7D,UAAUmD,SAASA,OAAO,CAAC;GAC7B;GAEA,IAAIM,KACF,MAAM,IAAIxG,MAAMwG,GAAG;GAKrB,IAAI,CAACQ,iBACH1D,aAAa6D,cAAc;GAI7B,IAAI7C,YAAYhB,aAAanC,MAAMgE,OAAOxE,SAAS,GACjD6C,WAAWC,SAAS;EAExB;EACA2D,UAAUxE,QAAQwE;EAClBC,gBAAgB;GACdnE,eAAe,KAAK;GACpB,MAAMoE,OAAO,CACX,GAAGhE,aAAanC,MAAMoG,WACtB,GAAGjE,aAAanC,MAAMgE,MAAM;GAG9B7B,aAAakE,MAAM;GAEnB5E,QAAQyE,WAAW;IACjB9C,UAAU+C;IACVG,cAAcC,YAAY;KACxB7E,cAAcgB,UAAU6D,QAAQ7E,cAAcgB,OAAO;KACrDd,UAAUmD,WAASA,SAAO,CAAC;IAC7B;GACF,CAAC;EACH;EACAY,SAAS,OAAOC,YAAU;GACxB7D,eAAe,KAAK;GACpB,MAAMyE,gBAAgB,CAAC,GAAGrE,aAAanC,MAAMoG,SAAS;GACtD,MAAMK,aAAa,CAAC,GAAGtE,aAAanC,MAAMgE,MAAM;GAEhD7B,aAAakE,MAAM;GAEnB,IAAI;IACF,MAAM5E,QAAQkE,UAAUC,SAAgB;KACtCxC,UAAUoD;KACVF,cAAcC,YAAY;MACxB7E,cAAcgB,UAAU6D,QAAQ7E,cAAcgB,OAAO;MACrDd,UAAUmD,SAASA,OAAO,CAAC;KAC7B;IACF,CAAC;GACH,UAAU;IACRtD,QAAQyE,WAAW;KACjB9C,UAAUqD;KACVH,cAAcC,YAAY;MACxB7E,cAAcgB,UAAU6D,QAAQ7E,cAAcgB,OAAO;MACrDd,UAAUmD,SAASA,OAAO,CAAC;KAC7B;KACAa,OAAOA;IACT,CAAC;GACH;EACF;CACF,CAAC;CAGD,MAAM,CAACc,cAAcC,mBAAmB3J,SAEtC,CAAC,CAAC;CAGJ,MAAM4J,kBAAkB9J,cAChB,CAAC,GAAGqF,aAAanC,MAAMoG,WAAW,GAAGjE,aAAanC,MAAMgE,MAAM,GACpE,CAAC7B,aAAanC,KAAK,CACrB;CACA,MAAM6G,YAAY3I,kBAChBuD,QAAQqF,WACRpF,cAAcgB,SACdkE,iBACAvE,WAAW0E,WACXL,YACF;CAGA,MAAMvC,UAAUxH,wBAAwB;EACtCqK,UAAUH,UAAUG;EACpBhH,OAAO6G,UAAU7G;EACjB+G,WAAWF,UAAUE;EACrBE,WAAWnF;EACXoF,UAAUzF,QAAQyF;EAClBC,gCAAgC;EAChCR;EACA/F,QAAQ;IACLN,iCAAiC;GAClCG,cAAcC,cAAuC;IACnDyB,aAAaQ,QAAQjC,SAAO;GAC9B;GACAV,OAAO0B,cAAcgB;EACvB;EACA0E,OAAO,OAAOzI,cACZ4D,qBAAqB5D,SAAO;EAC9B,GAAI8C,QAAQ4F,cAAcC,QAAQ,EAChCC,QAAQ,OAAO5I,cACb4D,qBAAqB5D,SAAO,EAChC;EACA,GAAIwD,aAAanC,MAAMgE,OAAOxE,SAAS,KAAK,EAC1CgI,UAAU,OAAO/H,eAA4B;GAC3CwC,YAAYS,UAAUjD;GACtB4C,WAAWC,SAAS;EACtB,EACF;EACA4D,UAAU,YAAY;GACpB7D,WAAWoF,OAAO;EACpB;EACAC,UAAU,YAAY;GACpB,IAAI,CAACjG,QAAQmD,WACX,MAAM,IAAI/F,MAAM,+CAA+C;GAEjEgD,cAAca,UAAU;GACxBL,WAAWC,SAAS;EACtB;EACAqF,iBAAiB,OACfC,gBACkB;GAClB,MAAMlH,YAAgC;IACpCtB,MAAM;IACNyI,YAAYD,YAAYC;IACxBC,QAAQF,YAAYE;IACpBC,UAAUH,YAAYG;IACtBC,SAASJ,YAAYI;IACrB,GAAIJ,YAAYK,YAAY,EAAEA,UAAUL,YAAYK,SAAS;IAC7D,GAAIL,YAAYM,iBAAiBhG,KAAAA,KAAa,EAC5CgG,cAAcN,YAAYM,aAC5B;GACF;GAEA/F,aAAaQ,QAAQjC,SAAO;EAC9B;EACAyH,qBAAqB,OAAOnI,UAAU;GACpC0B,cAAcgB,UAAU1C;GACxB4B,UAAUmD,WAASA,SAAO,CAAC;EAC7B;CACF,CAAC;CAED,OAAOZ;AACT;;;;AAKA,MAAaiE,gCACX3G,YACqB;CAQrB,OAPgBpD,2BAA2B;EACzCgK,aAAa,SAASC,cAAc;GAClC,OAAO9G,mCAAmCC,OAAO;EACnD;EACA8G,SAAS,IAAIjK,0BAA0B;EACvCkK,cAAc;CAChB,CACOrE;AACT"}

package/dist/primitives/actionBar/ActionBarExportMarkdown.js

@@ -32,7 +32,7 @@ const useActionBarExportMarkdown = (t0) => {
 			a.href = url;
 			a.download = filename ?? `message-${Date.now()}.md`;
 			a.click();
-			URL.revokeObjectURL(url);
+			setTimeout(() => URL.revokeObjectURL(url), 4e4);
 		};
 		$[2] = aui.message;
 		$[3] = filename;

package/dist/primitives/actionBar/ActionBarExportMarkdown.js.map

@@ -1 +1 @@
-{"version":3,"file":"ActionBarExportMarkdown.js","names":["c","_c","forwardRef","useCallback","ActionButtonProps","composeEventHandlers","Primitive","useAuiState","useAui","useActionBarExportMarkdown","t0","$","t1","undefined","filename","onExport","aui","hasExportableContent","_temp2","t2","message","content","getCopyText","blob","Blob","type","url","URL","createObjectURL","a","document","createElement","href","download","Date","now","click","revokeObjectURL","callback","ActionBarPrimitiveExportMarkdown","Element","HTMLButtonElement","Props","forwardedRef","disabled","onClick","props","t3","t4","t5","displayName","_temp","text","length","s","role","status","parts","some"],"sources":["../../../src/primitives/actionBar/ActionBarExportMarkdown.tsx"],"sourcesContent":["\"use client\";\n\nimport { forwardRef, useCallback } from \"react\";\nimport type { ActionButtonProps } from \"../../utils/createActionButton\";\nimport { composeEventHandlers } from \"@radix-ui/primitive\";\nimport { Primitive } from \"../../utils/Primitive\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\n\nconst useActionBarExportMarkdown = ({\n  filename,\n  onExport,\n}: {\n  filename?: string | undefined;\n  onExport?: ((content: string) => void | Promise<void>) | undefined;\n} = {}) => {\n  const aui = useAui();\n  const hasExportableContent = useAuiState((s) => {\n    return (\n      (s.message.role !== \"assistant\" ||\n        s.message.status?.type !== \"running\") &&\n      s.message.parts.some((c) => c.type === \"text\" && c.text.length > 0)\n    );\n  });\n\n  const callback = useCallback(async () => {\n    const content = aui.message.getCopyText();\n    if (!content) return;\n\n    if (onExport) {\n      await onExport(content);\n      return;\n    }\n\n    const blob = new Blob([content], { type: \"text/markdown\" });\n    const url = URL.createObjectURL(blob);\n    const a = document.createElement(\"a\");\n    a.href = url;\n    a.download = filename ?? `message-${Date.now()}.md`;\n    a.click();\n    URL.revokeObjectURL(url);\n  }, [aui, filename, onExport]);\n\n  if (!hasExportableContent) return null;\n  return callback;\n};\n\nexport namespace ActionBarPrimitiveExportMarkdown {\n  export type Element = HTMLButtonElement;\n  export type Props = ActionButtonProps<typeof useActionBarExportMarkdown>;\n}\n\nexport const ActionBarPrimitiveExportMarkdown = forwardRef<\n  ActionBarPrimitiveExportMarkdown.Element,\n  ActionBarPrimitiveExportMarkdown.Props\n>(({ filename, onExport, onClick, disabled, ...props }, forwardedRef) => {\n  const callback = useActionBarExportMarkdown({ filename, onExport });\n  return (\n    <Primitive.button\n      type=\"button\"\n      {...props}\n      ref={forwardedRef}\n      disabled={disabled || !callback}\n      onClick={composeEventHandlers(onClick, () => {\n        callback?.();\n      })}\n    />\n  );\n});\n\nActionBarPrimitiveExportMarkdown.displayName =\n  \"ActionBarPrimitive.ExportMarkdown\";\n"],"mappings":";;;;;;;;AAQA,MAAMS,8BAA6BC,OAAA;CAAA,MAAAC,IAAAV,EAAA,CAAA;CAAA,IAAAW;CAAA,IAAAD,EAAA,OAAAD,IAAA;EAACE,KAAAF,OAAAG,KAAAA,IAAA,CAM/B,IAN+BH;EAM9BC,EAAA,KAAAD;EAAAC,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAN8B,MAAA,EAAAG,UAAAC,aAAAH;CAOlC,MAAAI,MAAYR,OAAO;CACnB,MAAAS,uBAA6BV,YAAYW,MAMxC;CAAE,IAAAC;CAAA,IAAAR,EAAA,OAAAK,IAAAI,WAAAT,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EAE0BI,KAAA,YAAA;GAC3B,MAAAE,UAAgBL,IAAGI,QAAQE,YAAa;GACxC,IAAI,CAACD,SAAO;GAEZ,IAAIN,UAAQ;IACV,MAAMA,SAASM,OAAO;IAAC;GAAA;GAIzB,MAAAE,OAAa,IAAIC,KAAK,CAACH,OAAO,GAAG,EAAAI,MAAQ,gBAAgB,CAAC;GAC1D,MAAAC,MAAYC,IAAGC,gBAAiBL,IAAI;GACpC,MAAAM,IAAUC,SAAQC,cAAe,GAAG;GACpCF,EAACG,OAAQN;GACTG,EAACI,WAAYnB,YAAA,WAAuBoB,KAAIC,IAAK,EAAC;GAC9CN,EAACO,MAAO;GACRT,IAAGU,gBAAiBX,GAAG;EAAC;EACzBf,EAAA,KAAAK,IAAAI;EAAAT,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAhBD,MAAA2B,WAAiBnB;CAkBjB,IAAI,CAACF,sBAAoB,OAAS;CAAK,OAChCqB;AAAQ;AAQjB,MAAaC,mCAAmCrC,YAG9CQ,IAAAiC,iBAAA;CAAA,MAAAhC,IAAAV,EAAA,EAAA;CAAA,IAAA2C;CAAA,IAAA9B;CAAA,IAAA+B;CAAA,IAAA9B;CAAA,IAAA+B;CAAA,IAAAnC,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAI,UAAAC,UAAA8B,SAAAD,aAAAE,SAAApC;EAAmDC,EAAA,KAAAD;EAAAC,EAAA,KAAAiC;EAAAjC,EAAA,KAAAG;EAAAH,EAAA,KAAAkC;EAAAlC,EAAA,KAAAI;EAAAJ,EAAA,KAAAmC;CAAA,OAAA;EAAAF,WAAAjC,EAAA;EAAAG,WAAAH,EAAA;EAAAkC,UAAAlC,EAAA;EAAAI,WAAAJ,EAAA;EAAAmC,QAAAnC,EAAA;CAAA;CAAA,IAAAC;CAAA,IAAAD,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EACRH,KAAA;GAAAE;GAAAC;EAAqB;EAACJ,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAAlE,MAAA2B,WAAiB7B,2BAA2BG,EAAsB;CAMpD,MAAAO,KAAAyB,YAAA,CAAaN;CAAQ,IAAAS;CAAA,IAAApC,EAAA,OAAA2B,UAAA;EACQS,WAAA;GACrCT,WAAW;EAAC;EACb3B,EAAA,KAAA2B;EAAA3B,EAAA,MAAAoC;CAAA,OAAAA,KAAApC,EAAA;CAAA,IAAAqC;CAAA,IAAArC,EAAA,QAAAkC,WAAAlC,EAAA,QAAAoC,IAAA;EAFQC,KAAA3C,qBAAqBwC,SAASE,EAEtC;EAACpC,EAAA,MAAAkC;EAAAlC,EAAA,MAAAoC;EAAApC,EAAA,MAAAqC;CAAA,OAAAA,KAAArC,EAAA;CAAA,IAAAsC;CAAA,IAAAtC,EAAA,QAAAgC,gBAAAhC,EAAA,QAAAmC,SAAAnC,EAAA,QAAAQ,MAAAR,EAAA,QAAAqC,IAAA;EAPJC,KAAA,oBAAA,UAAA,QAAA;GACO,MAAA;GAAQ,GACTH;GACCH,KAAAA;GACK,UAAAxB;GACD,SAAA6B;EAEP,CAAA;EACFrC,EAAA,MAAAgC;EAAAhC,EAAA,MAAAmC;EAAAnC,EAAA,MAAAQ;EAAAR,EAAA,MAAAqC;EAAArC,EAAA,MAAAsC;CAAA,OAAAA,KAAAtC,EAAA;CAAA,OARFsC;AAQE,CAEL;AAEDV,iCAAiCW,cAC/B;AA9DiC,SAAAC,MAAAnD,GAAA;CAAA,OAYDA,EAACyB,SAAU,UAAUzB,EAACoD,KAAKC,SAAU;AAAC;AAZrC,SAAAnC,OAAAoC,GAAA;CAAA,QAU5BA,EAAClC,QAAQmC,SAAU,eAClBD,EAAClC,QAAQoC,QAAa/B,SAAK,cAC7B6B,EAAClC,QAAQqC,MAAMC,KAAMP,KAA6C;AAAC"}
+{"version":3,"file":"ActionBarExportMarkdown.js","names":["c","_c","forwardRef","useCallback","ActionButtonProps","composeEventHandlers","Primitive","useAuiState","useAui","useActionBarExportMarkdown","t0","$","t1","undefined","filename","onExport","aui","hasExportableContent","_temp2","t2","message","content","getCopyText","blob","Blob","type","url","URL","createObjectURL","a","document","createElement","href","download","Date","now","click","setTimeout","revokeObjectURL","callback","ActionBarPrimitiveExportMarkdown","Element","HTMLButtonElement","Props","forwardedRef","disabled","onClick","props","t3","t4","t5","displayName","_temp","text","length","s","role","status","parts","some"],"sources":["../../../src/primitives/actionBar/ActionBarExportMarkdown.tsx"],"sourcesContent":["\"use client\";\n\nimport { forwardRef, useCallback } from \"react\";\nimport type { ActionButtonProps } from \"../../utils/createActionButton\";\nimport { composeEventHandlers } from \"@radix-ui/primitive\";\nimport { Primitive } from \"../../utils/Primitive\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\n\nconst useActionBarExportMarkdown = ({\n  filename,\n  onExport,\n}: {\n  filename?: string | undefined;\n  onExport?: ((content: string) => void | Promise<void>) | undefined;\n} = {}) => {\n  const aui = useAui();\n  const hasExportableContent = useAuiState((s) => {\n    return (\n      (s.message.role !== \"assistant\" ||\n        s.message.status?.type !== \"running\") &&\n      s.message.parts.some((c) => c.type === \"text\" && c.text.length > 0)\n    );\n  });\n\n  const callback = useCallback(async () => {\n    const content = aui.message.getCopyText();\n    if (!content) return;\n\n    if (onExport) {\n      await onExport(content);\n      return;\n    }\n\n    const blob = new Blob([content], { type: \"text/markdown\" });\n    const url = URL.createObjectURL(blob);\n    const a = document.createElement(\"a\");\n    a.href = url;\n    a.download = filename ?? `message-${Date.now()}.md`;\n    a.click();\n    setTimeout(() => URL.revokeObjectURL(url), 40_000);\n  }, [aui, filename, onExport]);\n\n  if (!hasExportableContent) return null;\n  return callback;\n};\n\nexport namespace ActionBarPrimitiveExportMarkdown {\n  export type Element = HTMLButtonElement;\n  export type Props = ActionButtonProps<typeof useActionBarExportMarkdown>;\n}\n\nexport const ActionBarPrimitiveExportMarkdown = forwardRef<\n  ActionBarPrimitiveExportMarkdown.Element,\n  ActionBarPrimitiveExportMarkdown.Props\n>(({ filename, onExport, onClick, disabled, ...props }, forwardedRef) => {\n  const callback = useActionBarExportMarkdown({ filename, onExport });\n  return (\n    <Primitive.button\n      type=\"button\"\n      {...props}\n      ref={forwardedRef}\n      disabled={disabled || !callback}\n      onClick={composeEventHandlers(onClick, () => {\n        callback?.();\n      })}\n    />\n  );\n});\n\nActionBarPrimitiveExportMarkdown.displayName =\n  \"ActionBarPrimitive.ExportMarkdown\";\n"],"mappings":";;;;;;;;AAQA,MAAMS,8BAA6BC,OAAA;CAAA,MAAAC,IAAAV,EAAA,CAAA;CAAA,IAAAW;CAAA,IAAAD,EAAA,OAAAD,IAAA;EAACE,KAAAF,OAAAG,KAAAA,IAAA,CAM/B,IAN+BH;EAM9BC,EAAA,KAAAD;EAAAC,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAN8B,MAAA,EAAAG,UAAAC,aAAAH;CAOlC,MAAAI,MAAYR,OAAO;CACnB,MAAAS,uBAA6BV,YAAYW,MAMxC;CAAE,IAAAC;CAAA,IAAAR,EAAA,OAAAK,IAAAI,WAAAT,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EAE0BI,KAAA,YAAA;GAC3B,MAAAE,UAAgBL,IAAGI,QAAQE,YAAa;GACxC,IAAI,CAACD,SAAO;GAEZ,IAAIN,UAAQ;IACV,MAAMA,SAASM,OAAO;IAAC;GAAA;GAIzB,MAAAE,OAAa,IAAIC,KAAK,CAACH,OAAO,GAAG,EAAAI,MAAQ,gBAAgB,CAAC;GAC1D,MAAAC,MAAYC,IAAGC,gBAAiBL,IAAI;GACpC,MAAAM,IAAUC,SAAQC,cAAe,GAAG;GACpCF,EAACG,OAAQN;GACTG,EAACI,WAAYnB,YAAA,WAAuBoB,KAAIC,IAAK,EAAC;GAC9CN,EAACO,MAAO;GACRC,iBAAiBV,IAAGW,gBAAiBZ,GAAG,GAAG,GAAM;EAAC;EACnDf,EAAA,KAAAK,IAAAI;EAAAT,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAhBD,MAAA4B,WAAiBpB;CAkBjB,IAAI,CAACF,sBAAoB,OAAS;CAAK,OAChCsB;AAAQ;AAQjB,MAAaC,mCAAmCtC,YAG9CQ,IAAAkC,iBAAA;CAAA,MAAAjC,IAAAV,EAAA,EAAA;CAAA,IAAA4C;CAAA,IAAA/B;CAAA,IAAAgC;CAAA,IAAA/B;CAAA,IAAAgC;CAAA,IAAApC,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAI,UAAAC,UAAA+B,SAAAD,aAAAE,SAAArC;EAAmDC,EAAA,KAAAD;EAAAC,EAAA,KAAAkC;EAAAlC,EAAA,KAAAG;EAAAH,EAAA,KAAAmC;EAAAnC,EAAA,KAAAI;EAAAJ,EAAA,KAAAoC;CAAA,OAAA;EAAAF,WAAAlC,EAAA;EAAAG,WAAAH,EAAA;EAAAmC,UAAAnC,EAAA;EAAAI,WAAAJ,EAAA;EAAAoC,QAAApC,EAAA;CAAA;CAAA,IAAAC;CAAA,IAAAD,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EACRH,KAAA;GAAAE;GAAAC;EAAqB;EAACJ,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAAlE,MAAA4B,WAAiB9B,2BAA2BG,EAAsB;CAMpD,MAAAO,KAAA0B,YAAA,CAAaN;CAAQ,IAAAS;CAAA,IAAArC,EAAA,OAAA4B,UAAA;EACQS,WAAA;GACrCT,WAAW;EAAC;EACb5B,EAAA,KAAA4B;EAAA5B,EAAA,MAAAqC;CAAA,OAAAA,KAAArC,EAAA;CAAA,IAAAsC;CAAA,IAAAtC,EAAA,QAAAmC,WAAAnC,EAAA,QAAAqC,IAAA;EAFQC,KAAA5C,qBAAqByC,SAASE,EAEtC;EAACrC,EAAA,MAAAmC;EAAAnC,EAAA,MAAAqC;EAAArC,EAAA,MAAAsC;CAAA,OAAAA,KAAAtC,EAAA;CAAA,IAAAuC;CAAA,IAAAvC,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAoC,SAAApC,EAAA,QAAAQ,MAAAR,EAAA,QAAAsC,IAAA;EAPJC,KAAA,oBAAA,UAAA,QAAA;GACO,MAAA;GAAQ,GACTH;GACCH,KAAAA;GACK,UAAAzB;GACD,SAAA8B;EAEP,CAAA;EACFtC,EAAA,MAAAiC;EAAAjC,EAAA,MAAAoC;EAAApC,EAAA,MAAAQ;EAAAR,EAAA,MAAAsC;EAAAtC,EAAA,MAAAuC;CAAA,OAAAA,KAAAvC,EAAA;CAAA,OARFuC;AAQE,CAEL;AAEDV,iCAAiCW,cAC/B;AA9DiC,SAAAC,MAAApD,GAAA;CAAA,OAYDA,EAACyB,SAAU,UAAUzB,EAACqD,KAAKC,SAAU;AAAC;AAZrC,SAAApC,OAAAqC,GAAA;CAAA,QAU5BA,EAACnC,QAAQoC,SAAU,eAClBD,EAACnC,QAAQqC,QAAahC,SAAK,cAC7B8B,EAACnC,QAAQsC,MAAMC,KAAMP,KAA6C;AAAC"}

package/dist/primitives/attachment/AttachmentThumb.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AttachmentThumb.d.ts","names":[],"sources":["../../../src/primitives/attachment/AttachmentThumb.tsx"],"mappings":";;;KAUK,oBAAoB,gCAAgC,UAAU;kBAElD;OACH,UAAU,oBAAoB,UAAU;OACxC,QAAQ;;cAGT,0CAAwB,0BAAA,KAAA,qBAAA,gBAAA,kCAAA,eAAA;;;;oBAanC,cAAA,0CAAA,cAAA"}
+{"version":3,"file":"AttachmentThumb.d.ts","names":[],"sources":["../../../src/primitives/attachment/AttachmentThumb.tsx"],"mappings":";;;KAUK,oBAAoB,gCAAgC,UAAU;kBAElD;OACH,UAAU,oBAAoB,UAAU;OACxC,QAAQ;;cAGT,0CAAwB,0BAAA,KAAA,qBAAA,gBAAA,kCAAA,eAAA;;;;oBAiBnC,cAAA,0CAAA,cAAA"}

package/dist/primitives/attachment/AttachmentThumb.js

@@ -3,29 +3,32 @@ import { Primitive } from "../../utils/Primitive.js";
 import { useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { forwardRef } from "@assistant-ui/tap/react-shim";
-import { jsxs } from "react/jsx-runtime";
+import { jsx } from "react/jsx-runtime";
 //#region src/primitives/attachment/AttachmentThumb.tsx
 const AttachmentPrimitiveThumb = forwardRef((props, ref) => {
 	const $ = c(4);
-	const ext = useAuiState(_temp);
-	let t0;
-	if ($[0] !== ext || $[1] !== props || $[2] !== ref) {
-		t0 = /* @__PURE__ */ jsxs(Primitive.div, {
+	const label = useAuiState(_temp);
+	const t0 = props.children ?? label;
+	let t1;
+	if ($[0] !== props || $[1] !== ref || $[2] !== t0) {
+		t1 = /* @__PURE__ */ jsx(Primitive.div, {
 			...props,
 			ref,
-			children: [".", ext]
+			children: t0
 		});
-		$[0] = ext;
-		$[1] = props;
-		$[2] = ref;
-		$[3] = t0;
-	} else t0 = $[3];
-	return t0;
+		$[0] = props;
+		$[1] = ref;
+		$[2] = t0;
+		$[3] = t1;
+	} else t1 = $[3];
+	return t1;
 });
 AttachmentPrimitiveThumb.displayName = "AttachmentPrimitive.unstable_Thumb";
 function _temp(s) {
-	const parts = s.attachment.name.split(".");
-	return parts.length > 1 ? parts.pop() : "";
+	const name = s.attachment.name;
+	const dot = name.lastIndexOf(".");
+	if (dot > 0 && dot < name.length - 1) return `.${name.slice(dot + 1)}`;
+	return s.attachment.type;
 }
 //#endregion
 export { AttachmentPrimitiveThumb };

package/dist/primitives/attachment/AttachmentThumb.js.map

@@ -1 +1 @@
-{"version":3,"file":"AttachmentThumb.js","names":["c","_c","ComponentPropsWithoutRef","forwardRef","ComponentRef","useAuiState","Primitive","PrimitiveDivProps","div","AttachmentPrimitiveThumb","Element","Props","props","ref","$","ext","_temp","t0","displayName","s","parts","attachment","name","split","length","pop"],"sources":["../../../src/primitives/attachment/AttachmentThumb.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  type ComponentPropsWithoutRef,\n  forwardRef,\n  type ComponentRef,\n} from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { Primitive } from \"../../utils/Primitive\";\n\ntype PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;\n\nexport namespace AttachmentPrimitiveThumb {\n  export type Element = ComponentRef<typeof Primitive.div>;\n  export type Props = PrimitiveDivProps;\n}\n\nexport const AttachmentPrimitiveThumb = forwardRef<\n  AttachmentPrimitiveThumb.Element,\n  AttachmentPrimitiveThumb.Props\n>((props, ref) => {\n  const ext = useAuiState((s) => {\n    const parts = s.attachment.name.split(\".\");\n    return parts.length > 1 ? parts.pop()! : \"\";\n  });\n  return (\n    <Primitive.div {...props} ref={ref}>\n      .{ext}\n    </Primitive.div>\n  );\n});\n\nAttachmentPrimitiveThumb.displayName = \"AttachmentPrimitive.unstable_Thumb\";\n"],"mappings":";;;;;;;AAiBA,MAAaS,2BAA2BN,YAGtCS,OAAAC,QAAA;CAAA,MAAAC,IAAAb,EAAA,CAAA;CACA,MAAAc,MAAYV,YAAYW,KAGvB;CAAE,IAAAC;CAAA,IAAAH,EAAA,OAAAC,OAAAD,EAAA,OAAAF,SAAAE,EAAA,OAAAD,KAAA;EAEDI,KAAA,qBAAA,UAAA,KAAA;GAAA,GAAmBL;GAAYC;GAA/B,UAAA,CAAoC,KAChCE,GACJ;;EAAgBD,EAAA,KAAAC;EAAAD,EAAA,KAAAF;EAAAE,EAAA,KAAAD;EAAAC,EAAA,KAAAG;CAAA,OAAAA,KAAAH,EAAA;CAAA,OAFhBG;AAEgB,CAEnB;AAEDR,yBAAyBS,cAAc;AAZrC,SAAAF,MAAAG,GAAA;CAEE,MAAAC,QAAcD,EAACE,WAAWC,KAAKC,MAAO,GAAG;CAAE,OACpCH,MAAKI,SAAU,IAAIJ,MAAKK,IAAW,IAAnC;AAAoC"}
+{"version":3,"file":"AttachmentThumb.js","names":["c","_c","ComponentPropsWithoutRef","forwardRef","ComponentRef","useAuiState","Primitive","PrimitiveDivProps","div","AttachmentPrimitiveThumb","Element","Props","props","ref","$","label","_temp","t0","children","t1","displayName","s","name","attachment","dot","lastIndexOf","length","slice","type"],"sources":["../../../src/primitives/attachment/AttachmentThumb.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  type ComponentPropsWithoutRef,\n  forwardRef,\n  type ComponentRef,\n} from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { Primitive } from \"../../utils/Primitive\";\n\ntype PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;\n\nexport namespace AttachmentPrimitiveThumb {\n  export type Element = ComponentRef<typeof Primitive.div>;\n  export type Props = PrimitiveDivProps;\n}\n\nexport const AttachmentPrimitiveThumb = forwardRef<\n  AttachmentPrimitiveThumb.Element,\n  AttachmentPrimitiveThumb.Props\n>((props, ref) => {\n  const label = useAuiState((s) => {\n    const name = s.attachment.name;\n    const dot = name.lastIndexOf(\".\");\n    if (dot > 0 && dot < name.length - 1) {\n      return `.${name.slice(dot + 1)}`;\n    }\n    return s.attachment.type;\n  });\n  return (\n    <Primitive.div {...props} ref={ref}>\n      {props.children ?? label}\n    </Primitive.div>\n  );\n});\n\nAttachmentPrimitiveThumb.displayName = \"AttachmentPrimitive.unstable_Thumb\";\n"],"mappings":";;;;;;;AAiBA,MAAaS,2BAA2BN,YAGtCS,OAAAC,QAAA;CAAA,MAAAC,IAAAb,EAAA,CAAA;CACA,MAAAc,QAAcV,YAAYW,KAOzB;CAGI,MAAAC,KAAAL,MAAKM,YAALH;CAAuB,IAAAI;CAAA,IAAAL,EAAA,OAAAF,SAAAE,EAAA,OAAAD,OAAAC,EAAA,OAAAG,IAAA;EAD1BE,KAAA,oBAAA,UAAA,KAAA;GAAA,GAAmBP;GAAYC;GAC5BI,UAAAA;EACH,CAAA;EAAgBH,EAAA,KAAAF;EAAAE,EAAA,KAAAD;EAAAC,EAAA,KAAAG;EAAAH,EAAA,KAAAK;CAAA,OAAAA,KAAAL,EAAA;CAAA,OAFhBK;AAEgB,CAEnB;AAEDV,yBAAyBW,cAAc;AAhBrC,SAAAJ,MAAAK,GAAA;CAEE,MAAAC,OAAaD,EAACE,WAAWD;CACzB,MAAAE,MAAYF,KAAIG,YAAa,GAAG;CAChC,IAAID,MAAM,KAAKA,MAAMF,KAAII,SAAU,GAAC,OAC3B,IAAIJ,KAAIK,MAAOH,MAAM,CAAC;CAC9B,OACMH,EAACE,WAAWK;AAAK"}

package/dist/primitives/composer/ComposerInput.js

@@ -63,6 +63,7 @@ const ComposerPrimitiveInput = forwardRef(({ autoFocus = false, asChild, render,
 	const compositionRef = useRef(false);
 	useEscapeKeydown((e) => {
 		if (!textareaRef.current?.contains(e.target)) return;
+		if (e.isComposing) return;
 		if (pluginRegistry) {
 			for (const plugin of pluginRegistry.getPlugins()) if (plugin.handleKeyDown(e)) return;
 		}