@assistant-ui/core 0.3.13 → 0.3.15

package/dist/react/runtimes/assistant-transport/useAssistantTransportRuntime.js

@@ -0,0 +1,296 @@
+"use client";
+import { InMemoryThreadListAdapter } from "../../../runtimes/remote-thread-list/adapter/in-memory.js";
+import { createRequestHeaders } from "../../../runtimes/assistant-transport/utils.js";
+import { useExternalStoreRuntime } from "../useExternalStoreRuntime.js";
+import { useRemoteThreadListRuntime } from "../useRemoteThreadListRuntime.js";
+import { useCommandQueue } from "./commandQueue.js";
+import { createReplayBoundaryStream, useReplayRenderWait } from "./replayBoundaryStream.js";
+import { useRunManager } from "./runManager.js";
+import { useConvertedState } from "./useConvertedState.js";
+import { AssistantMessageAccumulator, AssistantTransportDecoder, DataStreamDecoder, toToolsJSONSchema, unstable_createInitialMessage } from "assistant-stream";
+import { asAsyncIterableStream } from "assistant-stream/utils";
+import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useMemo, useRef, useState } from "@assistant-ui/tap/react-shim";
+import { useAui, useAuiState } from "@assistant-ui/store";
+//#region src/react/runtimes/assistant-transport/useAssistantTransportRuntime.ts
+const convertAppendMessageToCommand = (message) => {
+	if (message.role !== "user") throw new Error("Only user messages are supported");
+	const parts = [];
+	const content = [...message.content, ...message.attachments?.flatMap((a) => a.content) ?? []];
+	for (const contentPart of content) if (contentPart.type === "text") parts.push({
+		type: "text",
+		text: contentPart.text
+	});
+	else if (contentPart.type === "image") parts.push({
+		type: "image",
+		image: contentPart.image
+	});
+	if (parts.length === 0) return null;
+	return {
+		type: "add-message",
+		message: {
+			role: "user",
+			parts
+		},
+		parentId: message.parentId,
+		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");
+	return extras;
+};
+const useAssistantTransportSendCommand = () => {
+	const $ = c(2);
+	const aui = useAui();
+	let t0;
+	if ($[0] !== aui.thread) {
+		t0 = (command) => {
+			const extras = aui.thread.getState().extras;
+			asAssistantTransportExtras(extras).sendCommand(command);
+		};
+		$[0] = aui.thread;
+		$[1] = t0;
+	} else t0 = $[1];
+	return t0;
+};
+function useAssistantTransportState(selector = (t) => t) {
+	return useAuiState((s) => selector(asAssistantTransportExtras(s.thread.extras).state));
+}
+const useAssistantTransportThreadRuntime = (options) => {
+	const agentStateRef = useRef(options.initialState);
+	const [, rerender] = useState(0);
+	const resumeFlagRef = useRef(false);
+	const [isReplaying, setIsReplaying] = useState(false);
+	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) => {
+			const isResume = resumeFlagRef.current;
+			resumeFlagRef.current = false;
+			setIsReplaying(false);
+			const commands = isResume ? [] : commandQueue.flush();
+			if (commands.length === 0 && !isResume) return;
+			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,
+				...resumeState === void 0 && { state: agentStateRef.current },
+				system: context.system,
+				tools: context.tools ? toToolsJSONSchema(context.tools) : void 0,
+				threadId,
+				...parentId !== void 0 && { parentId },
+				callSettings: context.callSettings,
+				config: context.config,
+				...context.callSettings,
+				...context.config,
+				...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,
+				body: JSON.stringify(requestBody),
+				signal
+			});
+			try {
+				await options.onResponse?.(response);
+			} catch (error) {
+				response.body?.cancel().catch(() => {});
+				throw error;
+			}
+			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 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_0) => {
+					err = error_0;
+				}
+			}));
+			let markedDelivered = false;
+			for await (const chunk of asAsyncIterableStream(stream)) {
+				if (chunk.metadata.unstable_state === agentStateRef.current) continue;
+				if (!markedDelivered) {
+					commandQueue.markDelivered();
+					markedDelivered = true;
+				}
+				agentStateRef.current = chunk.metadata.unstable_state;
+				rerender((prev) => prev + 1);
+			}
+			if (err) throw new Error(err);
+			if (!markedDelivered) commandQueue.markDelivered();
+			if (isResume && commandQueue.state.queued.length > 0) runManager.schedule();
+		},
+		onFinish: options.onFinish,
+		onCancel: () => {
+			setIsReplaying(false);
+			const cmds = [...commandQueue.state.inTransit, ...commandQueue.state.queued];
+			commandQueue.reset();
+			options.onCancel?.({
+				commands: cmds,
+				updateState: (updater) => {
+					agentStateRef.current = updater(agentStateRef.current);
+					rerender((prev_0) => prev_0 + 1);
+				}
+			});
+		},
+		onError: async (error_1) => {
+			resumeFlagRef.current = false;
+			setIsReplaying(false);
+			const inTransitCmds = [...commandQueue.state.inTransit];
+			const queuedCmds = [...commandQueue.state.queued];
+			commandQueue.reset();
+			try {
+				await options.onError?.(error_1, {
+					commands: inTransitCmds,
+					updateState: (updater) => {
+						agentStateRef.current = updater(agentStateRef.current);
+						rerender((prev) => prev + 1);
+					}
+				});
+			} finally {
+				options.onCancel?.({
+					commands: queuedCmds,
+					updateState: (updater) => {
+						agentStateRef.current = updater(agentStateRef.current);
+						rerender((prev) => prev + 1);
+					},
+					error: error_1
+				});
+			}
+		}
+	});
+	const [toolStatuses, setToolStatuses] = useState({});
+	const pendingCommands = useMemo(() => [...commandQueue.state.inTransit, ...commandQueue.state.queued], [commandQueue.state]);
+	const converted = useConvertedState(options.converter, agentStateRef.current, pendingCommands, runManager.isRunning, toolStatuses);
+	const runtime = useExternalStoreRuntime({
+		messages: converted.messages,
+		state: converted.state,
+		isRunning: converted.isRunning,
+		isLoading: isReplaying,
+		adapters: options.adapters,
+		unstable_enableToolInvocations: true,
+		setToolStatuses,
+		extras: {
+			[symbolAssistantTransportExtras]: true,
+			sendCommand: (command_0) => {
+				commandQueue.enqueue(command_0);
+			},
+			state: agentStateRef.current
+		},
+		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();
+		} },
+		onCancel: async () => {
+			resumeFlagRef.current = false;
+			runManager.cancel();
+		},
+		onResume: async () => {
+			if (!options.resumeApi) throw new Error("Must pass resumeApi to options to resume runs");
+			resumeFlagRef.current = true;
+			runManager.schedule();
+		},
+		onAddToolResult: async (toolOptions) => {
+			const command_1 = {
+				type: "add-tool-result",
+				toolCallId: toolOptions.toolCallId,
+				result: toolOptions.result,
+				toolName: toolOptions.toolName,
+				isError: toolOptions.isError,
+				...toolOptions.artifact !== void 0 && { artifact: toolOptions.artifact },
+				...toolOptions.modelContent !== void 0 && { modelContent: toolOptions.modelContent }
+			};
+			commandQueue.enqueue(command_1);
+		},
+		onLoadExternalState: async (state) => {
+			agentStateRef.current = state;
+			rerender((prev_1) => prev_1 + 1);
+		}
+	});
+	return runtime;
+};
+/**
+* @alpha This is an experimental API that is subject to change.
+*/
+const useAssistantTransportRuntime = (options) => {
+	return useRemoteThreadListRuntime({
+		runtimeHook: function RuntimeHook() {
+			return useAssistantTransportThreadRuntime(options);
+		},
+		adapter: new InMemoryThreadListAdapter(),
+		allowNesting: true
+	});
+};
+//#endregion
+export { useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState };
+
+//# sourceMappingURL=useAssistantTransportRuntime.js.map

package/dist/react/runtimes/assistant-transport/useAssistantTransportRuntime.js.map

@@ -0,0 +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","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","error","cancel","catch","prev","setReplaying","waitForRender","protocol","strict","decoder","err","stream","pipeThrough","initialMessage","unstable_state","throttle","onError","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","onResume","onAddToolResult","toolOptions","toolCallId","result","toolName","isError","artifact","modelContent","onLoadExternalState","useAssistantTransportRuntime","runtimeHook","RuntimeHook","adapter","allowNesting"],"sources":["../../../../src/react/runtimes/assistant-transport/useAssistantTransportRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport type { AppendMessage } from \"../../../types/message\";\nimport {\n  type ReadonlyJSONObject,\n  type ReadonlyJSONValue,\n  asAsyncIterableStream,\n} from \"assistant-stream/utils\";\nimport { useExternalStoreRuntime } from \"../useExternalStoreRuntime\";\nimport type { AssistantRuntime } from \"../../../runtime/api/assistant-runtime\";\nimport type { AddToolResultOptions } from \"../../../runtime/interfaces/thread-runtime-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 \"../../../runtimes/tool-invocations/ToolInvocationTracker\";\nimport { createRequestHeaders } from \"../../../runtimes/assistant-transport/utils\";\nimport { useRemoteThreadListRuntime } from \"../useRemoteThreadListRuntime\";\nimport { InMemoryThreadListAdapter } from \"../../../runtimes/remote-thread-list/adapter/in-memory\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { UserExternalState } from \"../../../types/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      try {\n        await options.onResponse?.(response);\n      } catch (error) {\n        void response.body?.cancel().catch(() => {});\n        throw error;\n      }\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      resumeFlagRef.current = false;\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      resumeFlagRef.current = false;\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 !== undefined && {\n          artifact: toolOptions.artifact,\n        }),\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;GAEA,IAAI;IACF,MAAMxB,QAAQqD,aAAalF,QAAQ;GACrC,SAASmF,OAAO;IACd,SAAcnB,MAAMoB,OAAO,CAAC,CAACC,YAAY,CAAC,CAAC;IAC3C,MAAMF;GACR;GAEA,IAAI,CAACnF,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,UAAUsD,SAASA,OAAO,CAAC;GAC7B;GAEA,MAAMtB,OAAO,MAAM7F,2BAA2B6B,UAAU;IACtDuF,cAAcpD;IACdqD,eAAepD;GACjB,CAAC;GAGD,MAAMqD,WAAW5D,QAAQ4D,YAAY;GAErC,MAAMC,SAASnC,WAAW,QAAS1B,QAAQ6D,UAAU;GACrD,MAAMC,UACJF,aAAa,wBACT,IAAIlI,0BAA0B,EAAEmI,OAAO,CAAC,IACxC,IAAIpI,kBAAkB,EAAEoI,OAAO,CAAC;GAEtC,IAAIE;GACJ,MAAMC,SAAS7B,KAAK8B,YAAYH,OAAO,CAAC,CAACG,YACvC,IAAIzI,4BAA4B;IAC9B0I,gBAAgBtI,8BAAqB,EACnCuI,gBACGlE,cAAcgB,WAAiC,KACpD,CAAC;IACDmD,UAAU1C;IACVmC;IACAQ,UAAUf,YAAU;KAClBS,MAAMT;IACR;GACF,CAAC,CACH;GAEA,IAAIgB,kBAAkB;GAEtB,WAAW,MAAMC,SAAStJ,sBAAsB+I,MAAM,GAAG;IACvD,IAAIO,MAAMC,SAASL,mBAAmBlE,cAAcgB,SAAS;IAE7D,IAAI,CAACqD,iBAAiB;KACpB5D,aAAa+D,cAAc;KAC3BH,kBAAkB;IACpB;IAEArE,cAAcgB,UAAUsD,MAAMC,SAASL;IACvChE,UAAUsD,SAASA,OAAO,CAAC;GAC7B;GAEA,IAAIM,KACF,MAAM,IAAI3G,MAAM2G,GAAG;GAKrB,IAAI,CAACO,iBACH5D,aAAa+D,cAAc;GAI7B,IAAI/C,YAAYhB,aAAanC,MAAMgE,OAAOxE,SAAS,GACjD6C,WAAWC,SAAS;EAExB;EACA6D,UAAU1E,QAAQ0E;EAClBC,gBAAgB;GACdrE,eAAe,KAAK;GACpB,MAAMsE,OAAO,CACX,GAAGlE,aAAanC,MAAMsG,WACtB,GAAGnE,aAAanC,MAAMgE,MAAM;GAG9B7B,aAAaoE,MAAM;GAEnB9E,QAAQ2E,WAAW;IACjBhD,UAAUiD;IACVG,cAAcC,YAAY;KACxB/E,cAAcgB,UAAU+D,QAAQ/E,cAAcgB,OAAO;KACrDd,UAAUsD,WAASA,SAAO,CAAC;IAC7B;GACF,CAAC;EACH;EACAY,SAAS,OAAOf,YAAU;GACxBlD,cAAca,UAAU;GACxBX,eAAe,KAAK;GACpB,MAAM2E,gBAAgB,CAAC,GAAGvE,aAAanC,MAAMsG,SAAS;GACtD,MAAMK,aAAa,CAAC,GAAGxE,aAAanC,MAAMgE,MAAM;GAEhD7B,aAAaoE,MAAM;GAEnB,IAAI;IACF,MAAM9E,QAAQqE,UAAUf,SAAgB;KACtC3B,UAAUsD;KACVF,cAAcC,YAAY;MACxB/E,cAAcgB,UAAU+D,QAAQ/E,cAAcgB,OAAO;MACrDd,UAAUsD,SAASA,OAAO,CAAC;KAC7B;IACF,CAAC;GACH,UAAU;IACRzD,QAAQ2E,WAAW;KACjBhD,UAAUuD;KACVH,cAAcC,YAAY;MACxB/E,cAAcgB,UAAU+D,QAAQ/E,cAAcgB,OAAO;MACrDd,UAAUsD,SAASA,OAAO,CAAC;KAC7B;KACAH,OAAOA;IACT,CAAC;GACH;EACF;CACF,CAAC;CAGD,MAAM,CAAC6B,cAAcC,mBAAmB7J,SAEtC,CAAC,CAAC;CAGJ,MAAM8J,kBAAkBhK,cAChB,CAAC,GAAGqF,aAAanC,MAAMsG,WAAW,GAAGnE,aAAanC,MAAMgE,MAAM,GACpE,CAAC7B,aAAanC,KAAK,CACrB;CACA,MAAM+G,YAAY7I,kBAChBuD,QAAQuF,WACRtF,cAAcgB,SACdoE,iBACAzE,WAAW4E,WACXL,YACF;CAGA,MAAMzC,UAAUxH,wBAAwB;EACtCuK,UAAUH,UAAUG;EACpBlH,OAAO+G,UAAU/G;EACjBiH,WAAWF,UAAUE;EACrBE,WAAWrF;EACXsF,UAAU3F,QAAQ2F;EAClBC,gCAAgC;EAChCR;EACAjG,QAAQ;IACLN,iCAAiC;GAClCG,cAAcC,cAAuC;IACnDyB,aAAaQ,QAAQjC,SAAO;GAC9B;GACAV,OAAO0B,cAAcgB;EACvB;EACA4E,OAAO,OAAO3I,cACZ4D,qBAAqB5D,SAAO;EAC9B,GAAI8C,QAAQ8F,cAAcC,QAAQ,EAChCC,QAAQ,OAAO9I,cACb4D,qBAAqB5D,SAAO,EAChC;EACA,GAAIwD,aAAanC,MAAMgE,OAAOxE,SAAS,KAAK,EAC1CkI,UAAU,OAAOjI,eAA4B;GAC3CwC,YAAYS,UAAUjD;GACtB4C,WAAWC,SAAS;EACtB,EACF;EACA8D,UAAU,YAAY;GACpBvE,cAAca,UAAU;GACxBL,WAAW2C,OAAO;EACpB;EACA2C,UAAU,YAAY;GACpB,IAAI,CAAClG,QAAQmD,WACX,MAAM,IAAI/F,MAAM,+CAA+C;GAEjEgD,cAAca,UAAU;GACxBL,WAAWC,SAAS;EACtB;EACAsF,iBAAiB,OACfC,gBACkB;GAClB,MAAMnH,YAAgC;IACpCtB,MAAM;IACN0I,YAAYD,YAAYC;IACxBC,QAAQF,YAAYE;IACpBC,UAAUH,YAAYG;IACtBC,SAASJ,YAAYI;IACrB,GAAIJ,YAAYK,aAAahG,KAAAA,KAAa,EACxCgG,UAAUL,YAAYK,SACxB;IACA,GAAIL,YAAYM,iBAAiBjG,KAAAA,KAAa,EAC5CiG,cAAcN,YAAYM,aAC5B;GACF;GAEAhG,aAAaQ,QAAQjC,SAAO;EAC9B;EACA0H,qBAAqB,OAAOpI,UAAU;GACpC0B,cAAcgB,UAAU1C;GACxB4B,UAAUsD,WAASA,SAAO,CAAC;EAC7B;CACF,CAAC;CAED,OAAOf;AACT;;;;AAKA,MAAakE,gCACX5G,YACqB;CAQrB,OAPgBpD,2BAA2B;EACzCiK,aAAa,SAASC,cAAc;GAClC,OAAO/G,mCAAmCC,OAAO;EACnD;EACA+G,SAAS,IAAIlK,0BAA0B;EACvCmK,cAAc;CAChB,CACOtE;AACT"}

package/dist/react/runtimes/assistant-transport/useConvertedState.d.ts

@@ -0,0 +1,7 @@
+import { ToolExecutionStatus } from "../../../runtimes/tool-invocations/ToolInvocationTracker.js";
+import { AssistantTransportCommand, AssistantTransportState, AssistantTransportStateConverter } from "./types.js";
+//#region src/react/runtimes/assistant-transport/useConvertedState.d.ts
+declare function useConvertedState<T>(converter: AssistantTransportStateConverter<T>, agentState: T, pendingCommands: AssistantTransportCommand[], isSending: boolean, toolStatuses: Record<string, ToolExecutionStatus>): AssistantTransportState;
+//#endregion
+export { useConvertedState };
+//# sourceMappingURL=useConvertedState.d.ts.map

package/dist/react/runtimes/assistant-transport/useConvertedState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useConvertedState.d.ts","names":[],"sources":["../../../../src/react/runtimes/assistant-transport/useConvertedState.ts"],"mappings":";;;iBAQgB,kBAAkB,GAChC,WAAW,iCAAiC,IAC5C,YAAY,GACZ,iBAAiB,6BACjB,oBACA,cAAc,eAAe,uBAC5B"}

package/dist/react/runtimes/assistant-transport/useConvertedState.js

@@ -0,0 +1,25 @@
+import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import "@assistant-ui/tap/react-shim";
+//#region src/react/runtimes/assistant-transport/useConvertedState.ts
+function useConvertedState(converter, agentState, pendingCommands, isSending, toolStatuses) {
+	const $ = c(6);
+	let t0;
+	if ($[0] !== agentState || $[1] !== converter || $[2] !== isSending || $[3] !== pendingCommands || $[4] !== toolStatuses) {
+		t0 = converter(agentState, {
+			pendingCommands,
+			isSending,
+			toolStatuses
+		});
+		$[0] = agentState;
+		$[1] = converter;
+		$[2] = isSending;
+		$[3] = pendingCommands;
+		$[4] = toolStatuses;
+		$[5] = t0;
+	} else t0 = $[5];
+	return t0;
+}
+//#endregion
+export { useConvertedState };
+
+//# sourceMappingURL=useConvertedState.js.map

package/dist/react/runtimes/assistant-transport/useConvertedState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useConvertedState.js","names":["useMemo","AssistantTransportCommand","AssistantTransportState","AssistantTransportStateConverter","ToolExecutionStatus","useConvertedState","converter","agentState","pendingCommands","isSending","toolStatuses","$","_c","t0"],"sources":["../../../../src/react/runtimes/assistant-transport/useConvertedState.ts"],"sourcesContent":["import { useMemo } from \"react\";\nimport type {\n  AssistantTransportCommand,\n  AssistantTransportState,\n  AssistantTransportStateConverter,\n} from \"./types\";\nimport type { ToolExecutionStatus } from \"../../../runtimes/tool-invocations/ToolInvocationTracker\";\n\nexport function useConvertedState<T>(\n  converter: AssistantTransportStateConverter<T>,\n  agentState: T,\n  pendingCommands: AssistantTransportCommand[],\n  isSending: boolean,\n  toolStatuses: Record<string, ToolExecutionStatus>,\n): AssistantTransportState {\n  return useMemo(\n    () => converter(agentState, { pendingCommands, isSending, toolStatuses }),\n    [converter, agentState, pendingCommands, isSending, toolStatuses],\n  );\n}\n"],"mappings":";;;AAQA,SAAOK,kBAAAC,WAAAC,YAAAC,iBAAAC,WAAAC,cAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAA,IAAAC;CAAA,IAAAF,EAAA,OAAAJ,cAAAI,EAAA,OAAAL,aAAAK,EAAA,OAAAF,aAAAE,EAAA,OAAAH,mBAAAG,EAAA,OAAAD,cAAA;EAQGG,KAAAP,UAAUC,YAAY;GAAAC;GAAAC;GAAAC;EAA2C,CAAC;EAACC,EAAA,KAAAJ;EAAAI,EAAA,KAAAL;EAAAK,EAAA,KAAAF;EAAAE,EAAA,KAAAH;EAAAG,EAAA,KAAAD;EAAAC,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAA,OAAnEE;AAAmE"}

package/dist/react/runtimes/assistant-transport/useLatestRef.d.ts

@@ -0,0 +1,7 @@
+//#region src/react/runtimes/assistant-transport/useLatestRef.d.ts
+declare function useLatestRef<T>(value: T): {
+  current: T;
+};
+//#endregion
+export { useLatestRef };
+//# sourceMappingURL=useLatestRef.d.ts.map

package/dist/react/runtimes/assistant-transport/useLatestRef.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useLatestRef.d.ts","names":[],"sources":["../../../../src/react/runtimes/assistant-transport/useLatestRef.ts"],"mappings":";iBAEgB,aAAa,GAAG,OAAO;EAKrB,SAAS"}

package/dist/react/runtimes/assistant-transport/useLatestRef.js

@@ -0,0 +1,27 @@
+import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useEffect, useRef } from "@assistant-ui/tap/react-shim";
+//#region src/react/runtimes/assistant-transport/useLatestRef.ts
+function useLatestRef(value) {
+	const $ = c(3);
+	const ref = useRef(value);
+	let t0;
+	let t1;
+	if ($[0] !== value) {
+		t0 = () => {
+			ref.current = value;
+		};
+		t1 = [value];
+		$[0] = value;
+		$[1] = t0;
+		$[2] = t1;
+	} else {
+		t0 = $[1];
+		t1 = $[2];
+	}
+	useEffect(t0, t1);
+	return ref;
+}
+//#endregion
+export { useLatestRef };
+
+//# sourceMappingURL=useLatestRef.js.map

package/dist/react/runtimes/assistant-transport/useLatestRef.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useLatestRef.js","names":["useEffect","useRef","useLatestRef","value","$","_c","ref","t0","t1","current","T"],"sources":["../../../../src/react/runtimes/assistant-transport/useLatestRef.ts"],"sourcesContent":["import { useEffect, useRef } from \"react\";\n\nexport function useLatestRef<T>(value: T) {\n  const ref = useRef(value);\n  useEffect(() => {\n    ref.current = value;\n  }, [value]);\n  return ref as { current: T };\n}\n"],"mappings":";;;AAEA,SAAOE,aAAAC,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CACL,MAAAC,MAAYL,OAAOE,KAAK;CAAE,IAAAI;CAAA,IAAAC;CAAA,IAAAJ,EAAA,OAAAD,OAAA;EAChBI,WAAA;GACRD,IAAGG,UAAWN;EAAH;EACVK,KAAA,CAACL,KAAK;EAACC,EAAA,KAAAD;EAAAC,EAAA,KAAAG;EAAAH,EAAA,KAAAI;CAAA,OAAA;EAAAD,KAAAH,EAAA;EAAAI,KAAAJ,EAAA;CAAA;CAFVJ,UAAUO,IAEPC,EAAO;CAAC,OACJF;AAAqB"}

package/dist/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantCloudThreadHistoryAdapter.d.ts","names":[],"sources":["../../../../src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.ts"],"mappings":";;;;KAgPK;EACH;EACA;EACA;EACA;EACA;;KAoDG;EACH;EACA;EACA;EACA;EACA,aAAa;EACb;EACA;;KAmBG;EACH;EACA,YAAY;EACZ;EACA;EACA;EACA;EACA;EACA;EACA,WAAW;EACX,QAAQ;EACR;;iBAiCc,aAAa,GAAG,SAAS,IAAI;iBA+b7B,sCACd,UAAU,UAAU,kBACnB"}
+{"version":3,"file":"AssistantCloudThreadHistoryAdapter.d.ts","names":[],"sources":["../../../../src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.ts"],"mappings":";;;;KA8RK;EACH;EACA;EACA;EACA;EACA;;KAoDG;EACH;EACA;EACA;EACA;EACA,aAAa;EACb;EACA;;KAmBG;EACH;EACA,YAAY;EACZ;EACA;EACA;EACA;EACA;EACA;EACA,WAAW;EACX,QAAQ;EACR;;iBAiCc,aAAa,GAAG,SAAS,IAAI;iBA+b7B,sCACd,UAAU,UAAU,kBACnB"}

package/dist/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.js

@@ -14,35 +14,61 @@ var AssistantCloudThreadHistoryAdapter = class {
 	get aui() {
 		return this.getAui();
 	}
-	get _persistence() {
-		const key = getClientId(this.aui.threadListItem);
+	getPersistence(threadListItem = this.aui.threadListItem) {
+		const key = getClientId(threadListItem);
 		if (!globalPersistence.has(key)) globalPersistence.set(key, new CloudMessagePersistence(() => this.cloudRef.current));
 		return globalPersistence.get(key);
 	}
+	get _persistence() {
+		return this.getPersistence();
+	}
+	tryGetKeyedThreadListItem() {
+		const live = this.aui.threadListItem;
+		if (!live.source) return void 0;
+		const id = live.getState().id;
+		if (id === void 0) return void 0;
+		return this.aui.threads.item({ id });
+	}
 	withFormat(formatAdapter) {
 		const adapter = this;
-		const formatted = createFormattedPersistence(this._persistence, formatAdapter);
+		let threadListItem;
+		const pinCurrent = () => {
+			const next = adapter.tryGetKeyedThreadListItem();
+			if (next) threadListItem = next;
+			return threadListItem;
+		};
+		const resolvePinned = () => threadListItem ?? pinCurrent();
+		const getFormatted = () => createFormattedPersistence(adapter._persistence, formatAdapter);
+		const getTargetFormatted = (item) => createFormattedPersistence(adapter.getPersistence(item), formatAdapter);
 		return {
+			pin() {
+				pinCurrent();
+			},
 			async append(item) {
-				const { remoteId } = await adapter.aui.threadListItem.initialize();
-				await formatted.append(remoteId, item);
+				const pinned = resolvePinned();
+				if (!pinned) throw new Error("Cannot persist cloud history without a thread list item.");
+				const remoteId = pinned.getState().remoteId ?? (await pinned.initialize()).remoteId;
+				await getTargetFormatted(pinned).append(remoteId, item);
 			},
 			async update(item, localMessageId) {
-				const remoteId = adapter.aui.threadListItem.getState().remoteId;
-				if (!remoteId) return;
-				await formatted.update?.(remoteId, item, localMessageId);
+				const pinned = resolvePinned();
+				const remoteId = pinned?.getState().remoteId;
+				if (!remoteId || !pinned) return;
+				await getTargetFormatted(pinned).update?.(remoteId, item, localMessageId);
 			},
 			async delete() {
 				throw new Error("Assistant Cloud does not support deleting thread messages yet.");
 			},
 			reportTelemetry(items, options) {
 				const encodedRunMessages = items.map((item) => formatAdapter.encode(item));
-				adapter._reportRunTelemetry(formatAdapter.format, encodedRunMessages, options);
+				adapter._reportRunTelemetry(formatAdapter.format, encodedRunMessages, options, resolvePinned());
 			},
 			async load() {
-				const remoteId = adapter.aui.threadListItem.getState().remoteId;
+				pinCurrent();
+				const live = adapter.aui.threadListItem;
+				const remoteId = live.source ? live.getState().remoteId : void 0;
 				if (!remoteId) return { messages: [] };
-				return formatted.load(remoteId);
+				return getFormatted().load(remoteId);
 			}
 		};
 	}
@@ -69,9 +95,9 @@ var AssistantCloudThreadHistoryAdapter = class {
 		if (!remoteId) return { messages: [] };
 		return { messages: (await this._persistence.load(remoteId, "aui/v0")).filter((m) => m.format === "aui/v0").map(auiV0Decode).reverse() };
 	}
-	_reportRunTelemetry(format, runMessages, options) {
+	_reportRunTelemetry(format, runMessages, options, threadListItem) {
 		if (!this.cloudRef.current.telemetry.enabled) return;
-		const remoteId = this.aui.threadListItem.getState().remoteId;
+		const remoteId = (threadListItem ?? this.aui.threadListItem).getState().remoteId;
 		if (!remoteId) return;
 		const extracted = extractRunTelemetry(format, runMessages);
 		if (!extracted) return;