@assistant-ui/ai-sdk 0.0.11 → 0.0.12

package/dist/runtime/useChatThread.js

@@ -2,37 +2,14 @@
 import { useAISDKRuntime } from "./useAISDKRuntime.js";
 import { AssistantChatTransport } from "../transport/AssistantChatTransport.js";
 import { useResourceCleanup } from "./useResourceCleanup.js";
+import { getResumableAdapter } from "./getResumableAdapter.js";
+import { DynamicChatTransport, getResumedStreamIds } from "./DynamicChatTransport.js";
+import { useDynamicChatTransport } from "./useDynamicChatTransport.js";
 import { pickExternalStoreSharedOptions } from "@assistant-ui/core";
-import { useCallback, useEffect, useMemo, useRef, useState, useSyncExternalStore } from "@assistant-ui/tap/react-shim";
+import { useCallback, useEffect, useEffectEvent, useInsertionEffect, useMemo, useRef, useState, useSyncExternalStore } from "@assistant-ui/tap/react-shim";
 import { Chat, useChat } from "@ai-sdk/react";
 //#region src/runtime/useChatThread.ts
-const useDynamicChatTransport = (transport) => {
-	const transportRef = useRef(transport);
-	useEffect(() => {
-		transportRef.current = transport;
-	});
-	return useMemo(() => new Proxy(transportRef.current, { get(_, prop) {
-		const res = transportRef.current[prop];
-		return typeof res === "function" ? res.bind(transportRef.current) : res;
-	} }), []);
-};
-const getResumableAdapter = (transport) => {
-	if (transport instanceof AssistantChatTransport) return transport.getResumableAdapter();
-	const candidate = transport.getResumableAdapter;
-	if (typeof candidate !== "function") return void 0;
-	return candidate.call(transport);
-};
 const getNoPendingStreamId = () => null;
-const resumedStreamIdsByStorage = /* @__PURE__ */ new WeakMap();
-const getResumedStreamIds = (storage) => {
-	if (!storage) return /* @__PURE__ */ new Set();
-	let resumedStreamIds = resumedStreamIdsByStorage.get(storage);
-	if (!resumedStreamIds) {
-		resumedStreamIds = /* @__PURE__ */ new Set();
-		resumedStreamIdsByStorage.set(storage, resumedStreamIds);
-	}
-	return resumedStreamIds;
-};
 /**
 * Splits the combined options into the assistant-ui side and the `ChatInit`
 * remainder the AI SDK consumes, so external `Chat` construction forwards the
@@ -86,8 +63,27 @@ const createChat = (init, callbacksRef) => {
 const useChatThread = (options, env) => {
 	const { adapters, transport: transportOptions, throttle, toCreateMessage, onResume, onResumeToolCall, onRespondToToolApproval, onResumeError, joinStrategy, messageRepository, unstable_onBranchChange, chatInit: chatOptions } = splitChatThreadOptions(options);
 	const { id, isMainThread, getThreadListItem, stopOnClientDestroy = true, hostDestroySignal, chat: externalChat, messageRepositoryInstance } = env;
-	const sourceTransport = useMemo(() => transportOptions === void 0 ? new AssistantChatTransport() : externalChat === void 0 && transportOptions instanceof AssistantChatTransport ? transportOptions.__internal_clone() : transportOptions, [transportOptions, externalChat]);
-	const transport = useDynamicChatTransport(sourceTransport);
+	const defaultTransport = useMemo(() => new AssistantChatTransport(), []);
+	const configuredTransport = transportOptions ?? defaultTransport;
+	const transport = useDynamicChatTransport(configuredTransport, externalChat === void 0);
+	const transportContextOwner = useMemo(() => ({}), []);
+	const getThreadListItemRef = useRef(getThreadListItem);
+	useInsertionEffect(() => {
+		getThreadListItemRef.current = getThreadListItem;
+	}, [getThreadListItem]);
+	const getCurrentThreadListItem = useCallback(() => getThreadListItemRef.current(), []);
+	const resumableStorage = useMemo(() => transport instanceof DynamicChatTransport ? transport.getCurrentResumableStorage() : getResumableAdapter(transport)?.storage, [transport]);
+	const transportBindingRef = useRef(null);
+	let initialTransportBinding = null;
+	const chatTransport = useMemo(() => transport instanceof DynamicChatTransport ? transport.createThreadProxy(transportContextOwner, () => {
+		const binding = transportBindingRef.current ?? initialTransportBinding;
+		if (!binding) throw new Error("Chat transport used before runtime setup");
+		return binding;
+	}) : transport, [
+		initialTransportBinding,
+		transport,
+		transportContextOwner
+	]);
 	const latestChatOptionsRef = useRef(chatOptions);
 	useEffect(() => {
 		latestChatOptionsRef.current = chatOptions;
@@ -95,7 +91,7 @@ const useChatThread = (options, env) => {
 	const [ownedChat] = useState(() => externalChat ?? createChat({
 		...chatOptions,
 		id,
-		transport
+		transport: chatTransport
 	}, latestChatOptionsRef));
 	const chat = useChat({
 		chat: externalChat ?? ownedChat,
@@ -123,14 +119,36 @@ const useChatThread = (options, env) => {
 		unstable_hostApprovalOwner: externalChat ?? ownedChat,
 		...unstable_onBranchChange && { unstable_onBranchChange }
 	});
-	if (sourceTransport instanceof AssistantChatTransport) {
-		sourceTransport.setRuntime(runtime);
-		sourceTransport.__internal_setGetThreadListItem(getThreadListItem);
+	initialTransportBinding = {
+		runtime,
+		getThreadListItem: getCurrentThreadListItem
+	};
+	useInsertionEffect(() => {
+		transportBindingRef.current = {
+			runtime,
+			getThreadListItem: getCurrentThreadListItem
+		};
+	}, [runtime, getCurrentThreadListItem]);
+	const registerTransportContext = useEffectEvent((dynamicTransport, chatId) => {
+		dynamicTransport.setThreadContext(chatId, transportContextOwner, runtime, getCurrentThreadListItem);
+		return () => dynamicTransport.unregisterThread(chatId, transportContextOwner);
+	});
+	useInsertionEffect(() => {
+		if (!(transport instanceof DynamicChatTransport)) return void 0;
+		return registerTransportContext(transport, id);
+	}, [
+		id,
+		runtime,
+		transport,
+		transportContextOwner
+	]);
+	if (!(transport instanceof DynamicChatTransport) && configuredTransport instanceof AssistantChatTransport) {
+		configuredTransport.setRuntime(runtime);
+		configuredTransport.__internal_setGetThreadListItem(getCurrentThreadListItem);
 	}
 	const subscribeToRuntime = useCallback((callback) => runtime.thread.subscribe(callback), [runtime]);
 	const getHistoryLoadingSnapshot = useCallback(() => runtime.thread.getState().isLoading, [runtime]);
 	const isLoadingHistory = useSyncExternalStore(subscribeToRuntime, getHistoryLoadingSnapshot, getHistoryLoadingSnapshot);
-	const resumableStorage = useMemo(() => getResumableAdapter(sourceTransport)?.storage, [sourceTransport]);
 	const subscribeToResumableStorage = useCallback((callback_0) => isMainThread ? resumableStorage?.subscribe?.(callback_0, id) ?? (() => {}) : () => {}, [
 		id,
 		isMainThread,
@@ -143,12 +161,13 @@ const useChatThread = (options, env) => {
 	]);
 	const pendingStreamId = useSyncExternalStore(subscribeToResumableStorage, getPendingStreamId, getNoPendingStreamId);
 	const isChatRunning = chat.status === "submitted" || chat.status === "streaming";
-	const resumedStreamIds = useMemo(() => getResumedStreamIds(resumableStorage), [resumableStorage]);
+	const staticResumedStreamIds = useMemo(() => getResumedStreamIds(resumableStorage), [resumableStorage]);
 	const onResumeErrorRef = useRef(onResumeError);
 	useEffect(() => {
 		onResumeErrorRef.current = onResumeError;
 	});
 	useEffect(() => {
+		const resumedStreamIds = transport instanceof DynamicChatTransport ? transport.getResumedStreamIds() : staticResumedStreamIds;
 		if (!pendingStreamId || resumedStreamIds.has(pendingStreamId)) return;
 		if (isChatRunning) {
 			resumedStreamIds.add(pendingStreamId);
@@ -181,7 +200,8 @@ const useChatThread = (options, env) => {
 		pendingStreamId,
 		ownedChat,
 		resumableStorage,
-		resumedStreamIds
+		staticResumedStreamIds,
+		transport
 	]);
 	return runtime;
 };

package/dist/runtime/useChatThread.js.map

@@ -1 +1 @@
-{"version":3,"file":"useChatThread.js","names":["Chat","useChat","UIMessage","MessageRepository","pickExternalStoreSharedOptions","AssistantRuntime","ExternalStoreSharedOptions","useAISDKRuntime","AISDKRuntimeAdapter","CustomToCreateMessageFunction","ChatInit","ChatTransport","AssistantChatTransport","InitializableThreadListItem","AssistantChatResumableOptions","ResumableClientStorage","useCallback","useEffect","useMemo","useRef","useState","useSyncExternalStore","useResourceCleanup","ChatThreadOptions","UI_MESSAGE","throttle","adapters","toCreateMessage","onResume","onResumeToolCall","onRespondToToolApproval","onResumeError","error","joinStrategy","messageRepository","unstable_onBranchChange","ChatThreadEnvironment","id","isMainThread","getThreadListItem","stopOnClientDestroy","hostDestroySignal","AbortSignal","chat","messageRepositoryInstance","useDynamicChatTransport","transport","transportRef","current","dynamicTransport","Proxy","get","_","prop","res","bind","getResumableAdapter","candidate","undefined","call","getNoPendingStreamId","resumedStreamIdsByStorage","WeakMap","Set","getResumedStreamIds","storage","resumedStreamIds","set","splitChatThreadOptions","options","isDisabled","_isDisabled","isSendDisabled","_isSendDisabled","unstable_capabilities","_unstable_capabilities","suggestions","_suggestions","chatInit","ChatCallbacks","Pick","requestsByChat","createChat","init","callbacksRef","sendMessages","Symbol","reconnectToStream","onToolCall","arg","onData","onFinish","onError","sendAutomaticallyWhen","useChatThread","env","transportOptions","chatOptions","externalChat","sourceTransport","__internal_clone","latestChatOptionsRef","ownedChat","stop","catch","runtime","threadList","threadId","unstable_messageRepositoryInstance","unstable_hostApprovalOwner","setRuntime","__internal_setGetThreadListItem","subscribeToRuntime","callback","thread","subscribe","getHistoryLoadingSnapshot","getState","isLoading","isLoadingHistory","resumableStorage","subscribeToResumableStorage","getPendingStreamId","getStreamId","pendingStreamId","isChatRunning","status","onResumeErrorRef","has","add","activeChat","clearError","pending","resumeStream","request","then","err","console","warn","callbackError","clear"],"sources":["../../src/runtime/useChatThread.ts"],"sourcesContent":["\"use client\";\n\nimport { Chat, useChat, type UIMessage } from \"@ai-sdk/react\";\nimport type { MessageRepository } from \"@assistant-ui/core/internal\";\nimport {\n  pickExternalStoreSharedOptions,\n  type AssistantRuntime,\n  type ExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport {\n  useAISDKRuntime,\n  type AISDKRuntimeAdapter,\n  type CustomToCreateMessageFunction,\n} from \"./useAISDKRuntime\";\nimport type { ChatInit, ChatTransport } from \"ai\";\nimport {\n  AssistantChatTransport,\n  type InitializableThreadListItem,\n} from \"../transport/AssistantChatTransport\";\nimport type {\n  AssistantChatResumableOptions,\n  ResumableClientStorage,\n} from \"../transport/resumable\";\nimport {\n  useCallback,\n  useEffect,\n  useMemo,\n  useRef,\n  useState,\n  useSyncExternalStore,\n} from \"react\";\nimport { useResourceCleanup } from \"./useResourceCleanup\";\n\nexport type ChatThreadOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  ChatInit<UI_MESSAGE> &\n    ExternalStoreSharedOptions & {\n      throttle?: number | undefined;\n      adapters?: AISDKRuntimeAdapter[\"adapters\"] | undefined;\n      toCreateMessage?: CustomToCreateMessageFunction;\n      onResume?: AISDKRuntimeAdapter[\"onResume\"];\n      onResumeToolCall?: AISDKRuntimeAdapter[\"onResumeToolCall\"];\n      onRespondToToolApproval?: AISDKRuntimeAdapter[\"onRespondToToolApproval\"];\n      /**\n       * Called when an automatic resumable stream reconnect fails. Use this to\n       * surface a toast, report telemetry, or mark the thread as needing a\n       * retry. The failed stream id is cleared after the callback unless a\n       * newer id has replaced it.\n       */\n      onResumeError?: ((error: unknown) => void) | undefined;\n      joinStrategy?: AISDKRuntimeAdapter[\"joinStrategy\"];\n      messageRepository?: AISDKRuntimeAdapter<UI_MESSAGE>[\"messageRepository\"];\n      unstable_onBranchChange?: AISDKRuntimeAdapter[\"unstable_onBranchChange\"];\n    };\n\nexport type ChatThreadEnvironment<UI_MESSAGE extends UIMessage = UIMessage> = {\n  id: string;\n  isMainThread: boolean;\n  getThreadListItem: () => InitializableThreadListItem | undefined;\n  stopOnClientDestroy?: boolean;\n  /**\n   * Aborts when the React component hosting the runtime is deleted. A nested\n   * runtime resolves the destroy signal of the provider above it, which\n   * outlives the nested component, so this stops the chat on its own unmount.\n   */\n  hostDestroySignal?: AbortSignal | undefined;\n  /**\n   * An externally owned chat instance. State lives on the instance, so it\n   * survives the hosting resource unmounting; construction options are read\n   * from the instance.\n   */\n  chat?: Chat<UI_MESSAGE> | undefined;\n  /**\n   * An externally owned per-thread message repository. Hosts that route\n   * multiple threads through one mounting pass a distinct instance per\n   * thread so histories and branches stay isolated.\n   */\n  messageRepositoryInstance?: MessageRepository | undefined;\n};\n\nconst useDynamicChatTransport = <UI_MESSAGE extends UIMessage = UIMessage>(\n  transport: ChatTransport<UI_MESSAGE>,\n): ChatTransport<UI_MESSAGE> => {\n  const transportRef = useRef<ChatTransport<UI_MESSAGE>>(transport);\n  useEffect(() => {\n    transportRef.current = transport;\n  });\n  const dynamicTransport = useMemo(\n    () =>\n      new Proxy(transportRef.current, {\n        get(_, prop) {\n          const res =\n            transportRef.current[prop as keyof ChatTransport<UI_MESSAGE>];\n          return typeof res === \"function\"\n            ? res.bind(transportRef.current)\n            : res;\n        },\n      }),\n    [],\n  );\n  return dynamicTransport;\n};\n\nconst getResumableAdapter = <UI_MESSAGE extends UIMessage>(\n  transport: ChatTransport<UI_MESSAGE>,\n): AssistantChatResumableOptions | undefined => {\n  if (transport instanceof AssistantChatTransport) {\n    return transport.getResumableAdapter();\n  }\n  const candidate = (transport as { getResumableAdapter?: () => unknown })\n    .getResumableAdapter;\n  if (typeof candidate !== \"function\") return undefined;\n  return candidate.call(transport) as AssistantChatResumableOptions | undefined;\n};\n\nconst getNoPendingStreamId = () => null;\n\nconst resumedStreamIdsByStorage = new WeakMap<\n  ResumableClientStorage,\n  Set<string>\n>();\n\nconst getResumedStreamIds = (storage: ResumableClientStorage | undefined) => {\n  if (!storage) return new Set<string>();\n  let resumedStreamIds = resumedStreamIdsByStorage.get(storage);\n  if (!resumedStreamIds) {\n    resumedStreamIds = new Set();\n    resumedStreamIdsByStorage.set(storage, resumedStreamIds);\n  }\n  return resumedStreamIds;\n};\n\n/**\n * Splits the combined options into the assistant-ui side and the `ChatInit`\n * remainder the AI SDK consumes, so external `Chat` construction forwards the\n * same fields `useChat` would.\n */\nexport const splitChatThreadOptions = <UI_MESSAGE extends UIMessage>(\n  options: ChatThreadOptions<UI_MESSAGE> | undefined,\n) => {\n  const {\n    adapters,\n    transport,\n    throttle,\n    toCreateMessage,\n    isDisabled: _isDisabled,\n    isSendDisabled: _isSendDisabled,\n    unstable_capabilities: _unstable_capabilities,\n    suggestions: _suggestions,\n    onResume,\n    onResumeToolCall,\n    onRespondToToolApproval,\n    onResumeError,\n    joinStrategy,\n    messageRepository,\n    unstable_onBranchChange,\n    ...chatInit\n  } = options ?? {};\n  // peel guard: any shared key left in `chatInit` collapses this to `never`\n  true satisfies keyof typeof chatInit &\n    keyof ExternalStoreSharedOptions extends never\n    ? true\n    : never;\n  return {\n    adapters,\n    transport,\n    throttle,\n    toCreateMessage,\n    onResume,\n    onResumeToolCall,\n    onRespondToToolApproval,\n    onResumeError,\n    joinStrategy,\n    messageRepository,\n    unstable_onBranchChange,\n    chatInit,\n  };\n};\n\ntype ChatCallbacks<UI_MESSAGE extends UIMessage> = Pick<\n  ChatInit<UI_MESSAGE>,\n  \"onToolCall\" | \"onData\" | \"onFinish\" | \"onError\" | \"sendAutomaticallyWhen\"\n>;\n\nconst requestsByChat = new WeakMap<object, symbol>();\n\n/**\n * Constructs a `Chat` whose callbacks read the latest options through\n * `callbacksRef`, the forwarding `useChat` applies only to a chat it\n * constructs itself.\n */\nexport const createChat = <UI_MESSAGE extends UIMessage>(\n  init: ChatInit<UI_MESSAGE>,\n  callbacksRef: { readonly current: ChatCallbacks<UI_MESSAGE> | undefined },\n): Chat<UI_MESSAGE> => {\n  const transport = init.transport;\n  const chat = new Chat<UI_MESSAGE>({\n    ...init,\n    ...(transport && {\n      transport: {\n        sendMessages: (options) => {\n          requestsByChat.set(chat, Symbol());\n          return transport.sendMessages(options);\n        },\n        reconnectToStream: (options) => {\n          requestsByChat.set(chat, Symbol());\n          return transport.reconnectToStream(options);\n        },\n      },\n    }),\n    onToolCall: (arg) => callbacksRef.current?.onToolCall?.(arg),\n    onData: (arg) => callbacksRef.current?.onData?.(arg),\n    onFinish: (arg) => callbacksRef.current?.onFinish?.(arg),\n    onError: (arg) => callbacksRef.current?.onError?.(arg),\n    sendAutomaticallyWhen: (arg) =>\n      callbacksRef.current?.sendAutomaticallyWhen?.(arg) ?? false,\n  });\n  return chat;\n};\n\nexport const useChatThread = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options: ChatThreadOptions<UI_MESSAGE> | undefined,\n  env: ChatThreadEnvironment<UI_MESSAGE>,\n): AssistantRuntime => {\n  const {\n    adapters,\n    transport: transportOptions,\n    throttle,\n    toCreateMessage,\n    onResume,\n    onResumeToolCall,\n    onRespondToToolApproval,\n    onResumeError,\n    joinStrategy,\n    messageRepository,\n    unstable_onBranchChange,\n    chatInit: chatOptions,\n  } = splitChatThreadOptions(options);\n\n  const {\n    id,\n    isMainThread,\n    getThreadListItem,\n    stopOnClientDestroy = true,\n    hostDestroySignal,\n    chat: externalChat,\n    messageRepositoryInstance,\n  } = env;\n\n  // Wiring below is per thread and mutated on the instance, so a transport\n  // shared across simultaneously mounted threads is last-writer-wins. A\n  // caller-owned chat is already bound to its own clone, so cloning again\n  // here would wire a copy the chat never sends through.\n  const sourceTransport = useMemo(\n    () =>\n      transportOptions === undefined\n        ? new AssistantChatTransport()\n        : externalChat === undefined &&\n            transportOptions instanceof AssistantChatTransport\n          ? transportOptions.__internal_clone()\n          : transportOptions,\n    [transportOptions, externalChat],\n  );\n  const transport = useDynamicChatTransport(sourceTransport);\n\n  const latestChatOptionsRef = useRef(chatOptions);\n  useEffect(() => {\n    latestChatOptionsRef.current = chatOptions;\n  });\n  // `useChat` stops a chat it constructs whenever it unmounts, and a\n  // resource's soft unmount runs that cleanup, so the thread owns its chat.\n  const [ownedChat] = useState(\n    () =>\n      externalChat ??\n      createChat({ ...chatOptions, id, transport }, latestChatOptionsRef),\n  );\n\n  const chat = useChat({\n    chat: externalChat ?? ownedChat,\n    ...(throttle !== undefined && { throttle }),\n  });\n\n  useResourceCleanup(\n    stopOnClientDestroy,\n    () => {\n      void chat.stop().catch(() => {});\n    },\n    hostDestroySignal,\n  );\n\n  const runtime = useAISDKRuntime(chat, {\n    adapters: {\n      ...adapters,\n      threadList: { threadId: id, ...adapters?.threadList },\n    },\n    ...pickExternalStoreSharedOptions(options ?? {}),\n    ...(toCreateMessage && { toCreateMessage }),\n    ...(onResume && { onResume }),\n    ...(onResumeToolCall && { onResumeToolCall }),\n    ...(onRespondToToolApproval && { onRespondToToolApproval }),\n    ...(joinStrategy && { joinStrategy }),\n    ...(messageRepository && { messageRepository }),\n    ...(messageRepositoryInstance && {\n      unstable_messageRepositoryInstance: messageRepositoryInstance,\n    }),\n    // The chat outlives this runtime when a host mounts only the visible\n    // thread, so a host approval answer is kept with it. This is the Chat\n    // instance, not the useChat helpers, which are re-minted every render and\n    // would be a dead WeakMap key by the next one.\n    unstable_hostApprovalOwner: externalChat ?? ownedChat,\n    ...(unstable_onBranchChange && { unstable_onBranchChange }),\n  });\n\n  // Wire in render, not an effect: a send from a descendant's mount effect\n  // runs before this hook's effect would (effects fire child-first), and must\n  // see a wired transport. The clone is per thread, so a discarded render's\n  // wiring is discarded with it and the committed render re-wires the same\n  // instance.\n  if (sourceTransport instanceof AssistantChatTransport) {\n    sourceTransport.setRuntime(runtime);\n    sourceTransport.__internal_setGetThreadListItem(getThreadListItem);\n  }\n\n  const subscribeToRuntime = useCallback(\n    (callback: () => void) => runtime.thread.subscribe(callback),\n    [runtime],\n  );\n  const getHistoryLoadingSnapshot = useCallback(\n    () => runtime.thread.getState().isLoading,\n    [runtime],\n  );\n  const isLoadingHistory = useSyncExternalStore(\n    subscribeToRuntime,\n    getHistoryLoadingSnapshot,\n    getHistoryLoadingSnapshot,\n  );\n\n  const resumableStorage = useMemo(\n    () => getResumableAdapter(sourceTransport)?.storage,\n    [sourceTransport],\n  );\n  const subscribeToResumableStorage = useCallback(\n    (callback: () => void) =>\n      isMainThread\n        ? (resumableStorage?.subscribe?.(callback, id) ?? (() => {}))\n        : () => {},\n    [id, isMainThread, resumableStorage],\n  );\n  const getPendingStreamId = useCallback(\n    () => (isMainThread ? (resumableStorage?.getStreamId(id) ?? null) : null),\n    [id, isMainThread, resumableStorage],\n  );\n  const pendingStreamId = useSyncExternalStore(\n    subscribeToResumableStorage,\n    getPendingStreamId,\n    getNoPendingStreamId,\n  );\n  const isChatRunning =\n    chat.status === \"submitted\" || chat.status === \"streaming\";\n\n  const resumedStreamIds = useMemo(\n    () => getResumedStreamIds(resumableStorage),\n    [resumableStorage],\n  );\n  const onResumeErrorRef = useRef(onResumeError);\n  useEffect(() => {\n    onResumeErrorRef.current = onResumeError;\n  });\n  useEffect(() => {\n    if (!pendingStreamId || resumedStreamIds.has(pendingStreamId)) {\n      return;\n    }\n    if (isChatRunning) {\n      resumedStreamIds.add(pendingStreamId);\n      return;\n    }\n    if (isLoadingHistory) return;\n    resumedStreamIds.add(pendingStreamId);\n    const activeChat = externalChat ?? ownedChat;\n    activeChat.clearError();\n    const pending = chat.resumeStream();\n    const request = requestsByChat.get(activeChat);\n    pending\n      .then(() => {\n        // Chat.error is shared with sends and resumes that can start before\n        // this promise settles, including inside the caller's onFinish.\n        if (requestsByChat.get(activeChat) === request && activeChat.error) {\n          throw activeChat.error;\n        }\n      })\n      .catch((err: unknown) => {\n        console.warn(\"[assistant-ui] resumable: resume failed\", err);\n        try {\n          onResumeErrorRef.current?.(err);\n        } catch (callbackError) {\n          console.error(\n            \"[assistant-ui] resumable: onResumeError callback failed\",\n            callbackError,\n          );\n        } finally {\n          if (resumableStorage?.getStreamId(id) === pendingStreamId) {\n            resumableStorage.clear(id);\n          }\n        }\n      });\n  }, [\n    chat,\n    externalChat,\n    id,\n    isChatRunning,\n    isLoadingHistory,\n    pendingStreamId,\n    ownedChat,\n    resumableStorage,\n    resumedStreamIds,\n  ]);\n\n  return runtime;\n};\n"],"mappings":";;;;;;;;AA+EA,MAAM6C,2BACJC,cAC8B;CAC9B,MAAMC,eAAe5B,OAAkC2B,SAAS;CAChE7B,gBAAgB;EACd8B,aAAaC,UAAUF;CACzB,CAAC;CAcD,OAbyB5B,cAErB,IAAIgC,MAAMH,aAAaC,SAAS,EAC9BG,IAAIC,GAAGC,MAAM;EACX,MAAMC,MACJP,aAAaC,QAAQK;EACvB,OAAO,OAAOC,QAAQ,aAClBA,IAAIC,KAAKR,aAAaC,OAAO,IAC7BM;CACN,EACF,CAAC,GACH,CAAA,CAEKL;AACT;AAEA,MAAMO,uBACJV,cAC8C;CAC9C,IAAIA,qBAAqBlC,wBACvB,OAAOkC,UAAUU,oBAAoB;CAEvC,MAAMC,YAAaX,UAChBU;CACH,IAAI,OAAOC,cAAc,YAAY,OAAOC,KAAAA;CAC5C,OAAOD,UAAUE,KAAKb,SAAS;AACjC;AAEA,MAAMc,6BAA6B;AAEnC,MAAMC,4CAA4B,IAAIC,QAGpC;AAEF,MAAME,uBAAuBC,YAAgD;CAC3E,IAAI,CAACA,SAAS,uBAAO,IAAIF,IAAY;CACrC,IAAIG,mBAAmBL,0BAA0BV,IAAIc,OAAO;CAC5D,IAAI,CAACC,kBAAkB;EACrBA,mCAAmB,IAAIH,IAAI;EAC3BF,0BAA0BM,IAAIF,SAASC,gBAAgB;CACzD;CACA,OAAOA;AACT;;;;;;AAOA,MAAaE,0BACXC,YACG;CACH,MAAM,EACJ3C,UACAoB,WACArB,UACAE,iBACA2C,YAAYC,aACZC,gBAAgBC,iBAChBC,uBAAuBC,wBACvBC,aAAaC,cACbjD,UACAC,kBACAC,yBACAC,eACAE,cACAC,mBACAC,yBACA,GAAG2C,aACDT,WAAW,CAAC;CAMhB,OAAO;EACL3C;EACAoB;EACArB;EACAE;EACAC;EACAC;EACAC;EACAC;EACAE;EACAC;EACAC;EACA2C;CACF;AACF;AAOA,MAAMG,iCAAiB,IAAInB,QAAwB;;;;;;AAOnD,MAAaoB,cACXC,MACAC,iBACqB;CACrB,MAAMtC,YAAYqC,KAAKrC;CACvB,MAAMH,OAAO,IAAI3C,KAAiB;EAChC,GAAGmF;EACH,GAAIrC,aAAa,EACfA,WAAW;GACTuC,eAAehB,YAAY;IACzBY,eAAed,IAAIxB,MAAM2C,OAAO,CAAC;IACjC,OAAOxC,UAAUuC,aAAahB,OAAO;GACvC;GACAkB,oBAAoBlB,YAAY;IAC9BY,eAAed,IAAIxB,MAAM2C,OAAO,CAAC;IACjC,OAAOxC,UAAUyC,kBAAkBlB,OAAO;GAC5C;EACF,EACF;EACAmB,aAAaC,QAAQL,aAAapC,SAASwC,aAAaC,GAAG;EAC3DC,SAASD,QAAQL,aAAapC,SAAS0C,SAASD,GAAG;EACnDE,WAAWF,QAAQL,aAAapC,SAAS2C,WAAWF,GAAG;EACvDG,UAAUH,QAAQL,aAAapC,SAAS4C,UAAUH,GAAG;EACrDI,wBAAwBJ,QACtBL,aAAapC,SAAS6C,wBAAwBJ,GAAG,KAAK;CAC1D,CAAC;CACD,OAAO9C;AACT;AAEA,MAAamD,iBACXzB,SACA0B,QACqB;CACrB,MAAM,EACJrE,UACAoB,WAAWkD,kBACXvE,UACAE,iBACAC,UACAC,kBACAC,yBACAC,eACAE,cACAC,mBACAC,yBACA2C,UAAUmB,gBACR7B,uBAAuBC,OAAO;CAElC,MAAM,EACJhC,IACAC,cACAC,mBACAC,sBAAsB,MACtBC,mBACAE,MAAMuD,cACNtD,8BACEmD;CAMJ,MAAMI,kBAAkBjF,cAEpB8E,qBAAqBtC,KAAAA,IACjB,IAAI9C,uBAAuB,IAC3BsF,iBAAiBxC,KAAAA,KACfsC,4BAA4BpF,yBAC5BoF,iBAAiBI,iBAAiB,IAClCJ,kBACR,CAACA,kBAAkBE,YAAY,CACjC;CACA,MAAMpD,YAAYD,wBAAwBsD,eAAe;CAEzD,MAAME,uBAAuBlF,OAAO8E,WAAW;CAC/ChF,gBAAgB;EACdoF,qBAAqBrD,UAAUiD;CACjC,CAAC;CAGD,MAAM,CAACK,aAAalF,eAEhB8E,gBACAhB,WAAW;EAAE,GAAGe;EAAa5D;EAAIS;CAAU,GAAGuD,oBAAoB,CACtE;CAEA,MAAM1D,OAAO1C,QAAQ;EACnB0C,MAAMuD,gBAAgBI;EACtB,GAAI7E,aAAaiC,KAAAA,KAAa,EAAEjC,SAAS;CAC3C,CAAC;CAEDH,mBACEkB,2BACM;EACJ,KAAU+D,KAAK,CAAC,CAACC,YAAY,CAAC,CAAC;CACjC,GACA/D,iBACF;CAEA,MAAMgE,UAAUlG,gBAAgBoC,MAAM;EACpCjB,UAAU;GACR,GAAGA;GACHgF,YAAY;IAAEC,UAAUtE;IAAI,GAAGX,UAAUgF;GAAW;EACtD;EACA,GAAGtG,+BAA+BiE,WAAW,CAAC,CAAC;EAC/C,GAAI1C,mBAAmB,EAAEA,gBAAgB;EACzC,GAAIC,YAAY,EAAEA,SAAS;EAC3B,GAAIC,oBAAoB,EAAEA,iBAAiB;EAC3C,GAAIC,2BAA2B,EAAEA,wBAAwB;EACzD,GAAIG,gBAAgB,EAAEA,aAAa;EACnC,GAAIC,qBAAqB,EAAEA,kBAAkB;EAC7C,GAAIU,6BAA6B,EAC/BgE,oCAAoChE,0BACtC;EAKAiE,4BAA4BX,gBAAgBI;EAC5C,GAAInE,2BAA2B,EAAEA,wBAAwB;CAC3D,CAAC;CAOD,IAAIgE,2BAA2BvF,wBAAwB;EACrDuF,gBAAgBW,WAAWL,OAAO;EAClCN,gBAAgBY,gCAAgCxE,iBAAiB;CACnE;CAEA,MAAMyE,qBAAqBhG,aACxBiG,aAAyBR,QAAQS,OAAOC,UAAUF,QAAQ,GAC3D,CAACR,OAAO,CACV;CACA,MAAMW,4BAA4BpG,kBAC1ByF,QAAQS,OAAOG,SAAS,CAAC,CAACC,WAChC,CAACb,OAAO,CACV;CACA,MAAMc,mBAAmBlG,qBACvB2F,oBACAI,2BACAA,yBACF;CAEA,MAAMI,mBAAmBtG,cACjBsC,oBAAoB2C,eAAe,CAAC,EAAElC,SAC5C,CAACkC,eAAe,CAClB;CACA,MAAMsB,8BAA8BzG,aACjCiG,eACC3E,eACKkF,kBAAkBL,YAAYF,YAAU5E,EAAE,YAAY,CAAC,WAClD,CAAC,GACb;EAACA;EAAIC;EAAckF;CAAgB,CACrC;CACA,MAAME,qBAAqB1G,kBAClBsB,eAAgBkF,kBAAkBG,YAAYtF,EAAE,KAAK,OAAQ,MACpE;EAACA;EAAIC;EAAckF;CAAgB,CACrC;CACA,MAAMI,kBAAkBvG,qBACtBoG,6BACAC,oBACA9D,oBACF;CACA,MAAMiE,gBACJlF,KAAKmF,WAAW,eAAenF,KAAKmF,WAAW;CAEjD,MAAM5D,mBAAmBhD,cACjB8C,oBAAoBwD,gBAAgB,GAC1C,CAACA,gBAAgB,CACnB;CACA,MAAMO,mBAAmB5G,OAAOY,aAAa;CAC7Cd,gBAAgB;EACd8G,iBAAiB/E,UAAUjB;CAC7B,CAAC;CACDd,gBAAgB;EACd,IAAI,CAAC2G,mBAAmB1D,iBAAiB8D,IAAIJ,eAAe,GAC1D;EAEF,IAAIC,eAAe;GACjB3D,iBAAiB+D,IAAIL,eAAe;GACpC;EACF;EACA,IAAIL,kBAAkB;EACtBrD,iBAAiB+D,IAAIL,eAAe;EACpC,MAAMM,aAAahC,gBAAgBI;EACnC4B,WAAWC,WAAW;EACtB,MAAMC,UAAUzF,KAAK0F,aAAa;EAClC,MAAMC,UAAUrD,eAAe9B,IAAI+E,UAAU;EAC7CE,QACGG,WAAW;GAGV,IAAItD,eAAe9B,IAAI+E,UAAU,MAAMI,WAAWJ,WAAWlG,OAC3D,MAAMkG,WAAWlG;EAErB,CAAC,CAAC,CACDwE,OAAOgC,QAAiB;GACvBC,QAAQC,KAAK,2CAA2CF,GAAG;GAC3D,IAAI;IACFT,iBAAiB/E,UAAUwF,GAAG;GAChC,SAASG,eAAe;IACtBF,QAAQzG,MACN,2DACA2G,aACF;GACF,UAAU;IACR,IAAInB,kBAAkBG,YAAYtF,EAAE,MAAMuF,iBACxCJ,iBAAiBoB,MAAMvG,EAAE;GAE7B;EACF,CAAC;CACL,GAAG;EACDM;EACAuD;EACA7D;EACAwF;EACAN;EACAK;EACAtB;EACAkB;EACAtD;CAAgB,CACjB;CAED,OAAOuC;AACT"}
+{"version":3,"file":"useChatThread.js","names":["Chat","useChat","UIMessage","MessageRepository","pickExternalStoreSharedOptions","AssistantRuntime","ExternalStoreSharedOptions","useAISDKRuntime","AISDKRuntimeAdapter","CustomToCreateMessageFunction","ChatInit","AssistantChatTransport","InitializableThreadListItem","useCallback","useEffect","useEffectEvent","useInsertionEffect","useMemo","useRef","useState","useSyncExternalStore","useResourceCleanup","DynamicChatTransport","getResumedStreamIds","getResumableAdapter","useDynamicChatTransport","ChatThreadOptions","UI_MESSAGE","throttle","adapters","toCreateMessage","onResume","onResumeToolCall","onRespondToToolApproval","onResumeError","error","joinStrategy","messageRepository","unstable_onBranchChange","ChatThreadEnvironment","id","isMainThread","getThreadListItem","stopOnClientDestroy","hostDestroySignal","AbortSignal","chat","messageRepositoryInstance","ChatThreadTransportBinding","runtime","getNoPendingStreamId","splitChatThreadOptions","options","transport","isDisabled","_isDisabled","isSendDisabled","_isSendDisabled","unstable_capabilities","_unstable_capabilities","suggestions","_suggestions","chatInit","ChatCallbacks","Pick","requestsByChat","WeakMap","createChat","init","callbacksRef","current","sendMessages","set","Symbol","reconnectToStream","onToolCall","arg","onData","onFinish","onError","sendAutomaticallyWhen","useChatThread","env","transportOptions","chatOptions","externalChat","defaultTransport","configuredTransport","undefined","transportContextOwner","getThreadListItemRef","getCurrentThreadListItem","resumableStorage","getCurrentResumableStorage","storage","transportBindingRef","initialTransportBinding","chatTransport","createThreadProxy","binding","Error","latestChatOptionsRef","ownedChat","stop","catch","threadList","threadId","unstable_messageRepositoryInstance","unstable_hostApprovalOwner","registerTransportContext","dynamicTransport","chatId","setThreadContext","unregisterThread","setRuntime","__internal_setGetThreadListItem","subscribeToRuntime","callback","thread","subscribe","getHistoryLoadingSnapshot","getState","isLoading","isLoadingHistory","subscribeToResumableStorage","getPendingStreamId","getStreamId","pendingStreamId","isChatRunning","status","staticResumedStreamIds","onResumeErrorRef","resumedStreamIds","has","add","activeChat","clearError","pending","resumeStream","request","get","then","err","console","warn","callbackError","clear"],"sources":["../../src/runtime/useChatThread.ts"],"sourcesContent":["\"use client\";\n\nimport { Chat, useChat, type UIMessage } from \"@ai-sdk/react\";\nimport type { MessageRepository } from \"@assistant-ui/core/internal\";\nimport {\n  pickExternalStoreSharedOptions,\n  type AssistantRuntime,\n  type ExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport {\n  useAISDKRuntime,\n  type AISDKRuntimeAdapter,\n  type CustomToCreateMessageFunction,\n} from \"./useAISDKRuntime\";\nimport type { ChatInit } from \"ai\";\nimport {\n  AssistantChatTransport,\n  type InitializableThreadListItem,\n} from \"../transport/AssistantChatTransport\";\nimport {\n  useCallback,\n  useEffect,\n  useEffectEvent,\n  useInsertionEffect,\n  useMemo,\n  useRef,\n  useState,\n  useSyncExternalStore,\n} from \"react\";\nimport { useResourceCleanup } from \"./useResourceCleanup\";\nimport {\n  DynamicChatTransport,\n  getResumedStreamIds,\n} from \"./DynamicChatTransport\";\nimport { getResumableAdapter } from \"./getResumableAdapter\";\nimport { useDynamicChatTransport } from \"./useDynamicChatTransport\";\n\nexport type ChatThreadOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  ChatInit<UI_MESSAGE> &\n    ExternalStoreSharedOptions & {\n      throttle?: number | undefined;\n      adapters?: AISDKRuntimeAdapter[\"adapters\"] | undefined;\n      toCreateMessage?: CustomToCreateMessageFunction;\n      onResume?: AISDKRuntimeAdapter[\"onResume\"];\n      onResumeToolCall?: AISDKRuntimeAdapter[\"onResumeToolCall\"];\n      onRespondToToolApproval?: AISDKRuntimeAdapter[\"onRespondToToolApproval\"];\n      /**\n       * Called when an automatic resumable stream reconnect fails. Use this to\n       * surface a toast, report telemetry, or mark the thread as needing a\n       * retry. The failed stream id is cleared after the callback unless a\n       * newer id has replaced it.\n       */\n      onResumeError?: ((error: unknown) => void) | undefined;\n      joinStrategy?: AISDKRuntimeAdapter[\"joinStrategy\"];\n      messageRepository?: AISDKRuntimeAdapter<UI_MESSAGE>[\"messageRepository\"];\n      /** @deprecated Experimental since 2026-06-23. Not scheduled for removal; the API may change in any release. */\n      unstable_onBranchChange?: AISDKRuntimeAdapter[\"unstable_onBranchChange\"];\n    };\n\nexport type ChatThreadEnvironment<UI_MESSAGE extends UIMessage = UIMessage> = {\n  id: string;\n  isMainThread: boolean;\n  getThreadListItem: () => InitializableThreadListItem | undefined;\n  stopOnClientDestroy?: boolean;\n  /**\n   * Aborts when the React component hosting the runtime is deleted. A nested\n   * runtime resolves the destroy signal of the provider above it, which\n   * outlives the nested component, so this stops the chat on its own unmount.\n   */\n  hostDestroySignal?: AbortSignal | undefined;\n  /**\n   * An externally owned chat instance. State lives on the instance, so it\n   * survives the hosting resource unmounting; construction options are read\n   * from the instance.\n   */\n  chat?: Chat<UI_MESSAGE> | undefined;\n  /**\n   * An externally owned per-thread message repository. Hosts that route\n   * multiple threads through one mounting pass a distinct instance per\n   * thread so histories and branches stay isolated.\n   */\n  messageRepositoryInstance?: MessageRepository | undefined;\n};\n\ntype ChatThreadTransportBinding = {\n  runtime: AssistantRuntime;\n  getThreadListItem: () => InitializableThreadListItem | undefined;\n};\n\nconst getNoPendingStreamId = () => null;\n\n/**\n * Splits the combined options into the assistant-ui side and the `ChatInit`\n * remainder the AI SDK consumes, so external `Chat` construction forwards the\n * same fields `useChat` would.\n */\nexport const splitChatThreadOptions = <UI_MESSAGE extends UIMessage>(\n  options: ChatThreadOptions<UI_MESSAGE> | undefined,\n) => {\n  const {\n    adapters,\n    transport,\n    throttle,\n    toCreateMessage,\n    isDisabled: _isDisabled,\n    isSendDisabled: _isSendDisabled,\n    unstable_capabilities: _unstable_capabilities,\n    suggestions: _suggestions,\n    onResume,\n    onResumeToolCall,\n    onRespondToToolApproval,\n    onResumeError,\n    joinStrategy,\n    messageRepository,\n    unstable_onBranchChange,\n    ...chatInit\n  } = options ?? {};\n  // peel guard: any shared key left in `chatInit` collapses this to `never`\n  true satisfies keyof typeof chatInit &\n    keyof ExternalStoreSharedOptions extends never\n    ? true\n    : never;\n  return {\n    adapters,\n    transport,\n    throttle,\n    toCreateMessage,\n    onResume,\n    onResumeToolCall,\n    onRespondToToolApproval,\n    onResumeError,\n    joinStrategy,\n    messageRepository,\n    unstable_onBranchChange,\n    chatInit,\n  };\n};\n\ntype ChatCallbacks<UI_MESSAGE extends UIMessage> = Pick<\n  ChatInit<UI_MESSAGE>,\n  \"onToolCall\" | \"onData\" | \"onFinish\" | \"onError\" | \"sendAutomaticallyWhen\"\n>;\n\nconst requestsByChat = new WeakMap<object, symbol>();\n\n/**\n * Constructs a `Chat` whose callbacks read the latest options through\n * `callbacksRef`, the forwarding `useChat` applies only to a chat it\n * constructs itself.\n */\nexport const createChat = <UI_MESSAGE extends UIMessage>(\n  init: ChatInit<UI_MESSAGE>,\n  callbacksRef: { readonly current: ChatCallbacks<UI_MESSAGE> | undefined },\n): Chat<UI_MESSAGE> => {\n  const transport = init.transport;\n  const chat = new Chat<UI_MESSAGE>({\n    ...init,\n    ...(transport && {\n      transport: {\n        sendMessages: (options) => {\n          requestsByChat.set(chat, Symbol());\n          return transport.sendMessages(options);\n        },\n        reconnectToStream: (options) => {\n          requestsByChat.set(chat, Symbol());\n          return transport.reconnectToStream(options);\n        },\n      },\n    }),\n    onToolCall: (arg) => callbacksRef.current?.onToolCall?.(arg),\n    onData: (arg) => callbacksRef.current?.onData?.(arg),\n    onFinish: (arg) => callbacksRef.current?.onFinish?.(arg),\n    onError: (arg) => callbacksRef.current?.onError?.(arg),\n    sendAutomaticallyWhen: (arg) =>\n      callbacksRef.current?.sendAutomaticallyWhen?.(arg) ?? false,\n  });\n  return chat;\n};\n\nexport const useChatThread = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options: ChatThreadOptions<UI_MESSAGE> | undefined,\n  env: ChatThreadEnvironment<UI_MESSAGE>,\n): AssistantRuntime => {\n  const {\n    adapters,\n    transport: transportOptions,\n    throttle,\n    toCreateMessage,\n    onResume,\n    onResumeToolCall,\n    onRespondToToolApproval,\n    onResumeError,\n    joinStrategy,\n    messageRepository,\n    unstable_onBranchChange,\n    chatInit: chatOptions,\n  } = splitChatThreadOptions(options);\n\n  const {\n    id,\n    isMainThread,\n    getThreadListItem,\n    stopOnClientDestroy = true,\n    hostDestroySignal,\n    chat: externalChat,\n    messageRepositoryInstance,\n  } = env;\n\n  const defaultTransport = useMemo(() => new AssistantChatTransport(), []);\n  const configuredTransport = transportOptions ?? defaultTransport;\n  const transport = useDynamicChatTransport(\n    configuredTransport,\n    externalChat === undefined,\n  );\n  const transportContextOwner = useMemo(() => ({}), []);\n  const getThreadListItemRef = useRef(getThreadListItem);\n  useInsertionEffect(() => {\n    getThreadListItemRef.current = getThreadListItem;\n  }, [getThreadListItem]);\n  const getCurrentThreadListItem = useCallback(\n    () => getThreadListItemRef.current(),\n    [],\n  );\n  const resumableStorage = useMemo(\n    () =>\n      transport instanceof DynamicChatTransport\n        ? transport.getCurrentResumableStorage()\n        : getResumableAdapter(transport)?.storage,\n    [transport],\n  );\n\n  const transportBindingRef = useRef<ChatThreadTransportBinding | null>(null);\n  // This is null when useMemo runs, so the proxy pins its mount render's binding. The fallback\n  // serves only pre-commit sends; after insertion effects, transportBindingRef is authoritative.\n  let initialTransportBinding: ChatThreadTransportBinding | null = null;\n  const chatTransport = useMemo(\n    () =>\n      transport instanceof DynamicChatTransport\n        ? transport.createThreadProxy(transportContextOwner, () => {\n            const binding =\n              transportBindingRef.current ?? initialTransportBinding;\n            if (!binding) {\n              throw new Error(\"Chat transport used before runtime setup\");\n            }\n            return binding;\n          })\n        : transport,\n    [initialTransportBinding, transport, transportContextOwner],\n  );\n\n  const latestChatOptionsRef = useRef(chatOptions);\n  useEffect(() => {\n    latestChatOptionsRef.current = chatOptions;\n  });\n  // `useChat` stops a chat it constructs whenever it unmounts, and a\n  // resource's soft unmount runs that cleanup, so the thread owns its chat.\n  const [ownedChat] = useState(\n    () =>\n      externalChat ??\n      createChat(\n        { ...chatOptions, id, transport: chatTransport },\n        latestChatOptionsRef,\n      ),\n  );\n\n  const chat = useChat({\n    chat: externalChat ?? ownedChat,\n    ...(throttle !== undefined && { throttle }),\n  });\n\n  useResourceCleanup(\n    stopOnClientDestroy,\n    () => {\n      void chat.stop().catch(() => {});\n    },\n    hostDestroySignal,\n  );\n\n  const runtime = useAISDKRuntime(chat, {\n    adapters: {\n      ...adapters,\n      threadList: { threadId: id, ...adapters?.threadList },\n    },\n    ...pickExternalStoreSharedOptions(options ?? {}),\n    ...(toCreateMessage && { toCreateMessage }),\n    ...(onResume && { onResume }),\n    ...(onResumeToolCall && { onResumeToolCall }),\n    ...(onRespondToToolApproval && { onRespondToToolApproval }),\n    ...(joinStrategy && { joinStrategy }),\n    ...(messageRepository && { messageRepository }),\n    ...(messageRepositoryInstance && {\n      unstable_messageRepositoryInstance: messageRepositoryInstance,\n    }),\n    // The chat outlives this runtime when a host mounts only the visible\n    // thread, so a host approval answer is kept with it. This is the Chat\n    // instance, not the useChat helpers, which are re-minted every render and\n    // would be a dead WeakMap key by the next one.\n    unstable_hostApprovalOwner: externalChat ?? ownedChat,\n    ...(unstable_onBranchChange && { unstable_onBranchChange }),\n  });\n  initialTransportBinding = {\n    runtime,\n    getThreadListItem: getCurrentThreadListItem,\n  };\n  useInsertionEffect(() => {\n    transportBindingRef.current = {\n      runtime,\n      getThreadListItem: getCurrentThreadListItem,\n    };\n  }, [runtime, getCurrentThreadListItem]);\n\n  const registerTransportContext = useEffectEvent(\n    (dynamicTransport: DynamicChatTransport<UI_MESSAGE>, chatId: string) => {\n      dynamicTransport.setThreadContext(\n        chatId,\n        transportContextOwner,\n        runtime,\n        getCurrentThreadListItem,\n      );\n      return () =>\n        dynamicTransport.unregisterThread(chatId, transportContextOwner);\n    },\n  );\n  useInsertionEffect(() => {\n    if (!(transport instanceof DynamicChatTransport)) return undefined;\n    return registerTransportContext(transport, id);\n  }, [id, runtime, transport, transportContextOwner]);\n\n  if (\n    !(transport instanceof DynamicChatTransport) &&\n    configuredTransport instanceof AssistantChatTransport\n  ) {\n    configuredTransport.setRuntime(runtime);\n    configuredTransport.__internal_setGetThreadListItem(\n      getCurrentThreadListItem,\n    );\n  }\n\n  const subscribeToRuntime = useCallback(\n    (callback: () => void) => runtime.thread.subscribe(callback),\n    [runtime],\n  );\n  const getHistoryLoadingSnapshot = useCallback(\n    () => runtime.thread.getState().isLoading,\n    [runtime],\n  );\n  const isLoadingHistory = useSyncExternalStore(\n    subscribeToRuntime,\n    getHistoryLoadingSnapshot,\n    getHistoryLoadingSnapshot,\n  );\n\n  const subscribeToResumableStorage = useCallback(\n    (callback: () => void) =>\n      isMainThread\n        ? (resumableStorage?.subscribe?.(callback, id) ?? (() => {}))\n        : () => {},\n    [id, isMainThread, resumableStorage],\n  );\n  const getPendingStreamId = useCallback(\n    () => (isMainThread ? (resumableStorage?.getStreamId(id) ?? null) : null),\n    [id, isMainThread, resumableStorage],\n  );\n  const pendingStreamId = useSyncExternalStore(\n    subscribeToResumableStorage,\n    getPendingStreamId,\n    getNoPendingStreamId,\n  );\n  const isChatRunning =\n    chat.status === \"submitted\" || chat.status === \"streaming\";\n\n  const staticResumedStreamIds = useMemo(\n    () => getResumedStreamIds(resumableStorage),\n    [resumableStorage],\n  );\n  const onResumeErrorRef = useRef(onResumeError);\n  useEffect(() => {\n    onResumeErrorRef.current = onResumeError;\n  });\n  useEffect(() => {\n    const resumedStreamIds =\n      transport instanceof DynamicChatTransport\n        ? transport.getResumedStreamIds()\n        : staticResumedStreamIds;\n    if (!pendingStreamId || resumedStreamIds.has(pendingStreamId)) {\n      return;\n    }\n    if (isChatRunning) {\n      resumedStreamIds.add(pendingStreamId);\n      return;\n    }\n    if (isLoadingHistory) return;\n    resumedStreamIds.add(pendingStreamId);\n    const activeChat = externalChat ?? ownedChat;\n    activeChat.clearError();\n    const pending = chat.resumeStream();\n    const request = requestsByChat.get(activeChat);\n    pending\n      .then(() => {\n        // Chat.error is shared with sends and resumes that can start before\n        // this promise settles, including inside the caller's onFinish.\n        if (requestsByChat.get(activeChat) === request && activeChat.error) {\n          throw activeChat.error;\n        }\n      })\n      .catch((err: unknown) => {\n        console.warn(\"[assistant-ui] resumable: resume failed\", err);\n        try {\n          onResumeErrorRef.current?.(err);\n        } catch (callbackError) {\n          console.error(\n            \"[assistant-ui] resumable: onResumeError callback failed\",\n            callbackError,\n          );\n        } finally {\n          if (resumableStorage?.getStreamId(id) === pendingStreamId) {\n            resumableStorage.clear(id);\n          }\n        }\n      });\n  }, [\n    chat,\n    externalChat,\n    id,\n    isChatRunning,\n    isLoadingHistory,\n    pendingStreamId,\n    ownedChat,\n    resumableStorage,\n    staticResumedStreamIds,\n    transport,\n  ]);\n\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;AAyFA,MAAMkD,6BAA6B;;;;;;AAOnC,MAAaC,0BACXC,YACG;CACH,MAAM,EACJvB,UACAwB,WACAzB,UACAE,iBACAwB,YAAYC,aACZC,gBAAgBC,iBAChBC,uBAAuBC,wBACvBC,aAAaC,cACb9B,UACAC,kBACAC,yBACAC,eACAE,cACAC,mBACAC,yBACA,GAAGwB,aACDV,WAAW,CAAC;CAMhB,OAAO;EACLvB;EACAwB;EACAzB;EACAE;EACAC;EACAC;EACAC;EACAC;EACAE;EACAC;EACAC;EACAwB;CACF;AACF;AAOA,MAAMG,iCAAiB,IAAIC,QAAwB;;;;;;AAOnD,MAAaC,cACXC,MACAC,iBACqB;CACrB,MAAMhB,YAAYe,KAAKf;CACvB,MAAMP,OAAO,IAAI9C,KAAiB;EAChC,GAAGoE;EACH,GAAIf,aAAa,EACfA,WAAW;GACTkB,eAAenB,YAAY;IACzBa,eAAeO,IAAI1B,MAAM2B,OAAO,CAAC;IACjC,OAAOpB,UAAUkB,aAAanB,OAAO;GACvC;GACAsB,oBAAoBtB,YAAY;IAC9Ba,eAAeO,IAAI1B,MAAM2B,OAAO,CAAC;IACjC,OAAOpB,UAAUqB,kBAAkBtB,OAAO;GAC5C;EACF,EACF;EACAuB,aAAaC,QAAQP,aAAaC,SAASK,aAAaC,GAAG;EAC3DC,SAASD,QAAQP,aAAaC,SAASO,SAASD,GAAG;EACnDE,WAAWF,QAAQP,aAAaC,SAASQ,WAAWF,GAAG;EACvDG,UAAUH,QAAQP,aAAaC,SAASS,UAAUH,GAAG;EACrDI,wBAAwBJ,QACtBP,aAAaC,SAASU,wBAAwBJ,GAAG,KAAK;CAC1D,CAAC;CACD,OAAO9B;AACT;AAEA,MAAamC,iBACX7B,SACA8B,QACqB;CACrB,MAAM,EACJrD,UACAwB,WAAW8B,kBACXvD,UACAE,iBACAC,UACAC,kBACAC,yBACAC,eACAE,cACAC,mBACAC,yBACAwB,UAAUsB,gBACRjC,uBAAuBC,OAAO;CAElC,MAAM,EACJZ,IACAC,cACAC,mBACAC,sBAAsB,MACtBC,mBACAE,MAAMuC,cACNtC,8BACEmC;CAEJ,MAAMI,mBAAmBrE,cAAc,IAAIN,uBAAuB,GAAG,CAAA,CAAE;CACvE,MAAM4E,sBAAsBJ,oBAAoBG;CAChD,MAAMjC,YAAY5B,wBAChB8D,qBACAF,iBAAiBG,KAAAA,CACnB;CACA,MAAMC,wBAAwBxE,eAAe,CAAC,IAAI,CAAA,CAAE;CACpD,MAAMyE,uBAAuBxE,OAAOwB,iBAAiB;CACrD1B,yBAAyB;EACvB0E,qBAAqBpB,UAAU5B;CACjC,GAAG,CAACA,iBAAiB,CAAC;CACtB,MAAMiD,2BAA2B9E,kBACzB6E,qBAAqBpB,QAAQ,GACnC,CAAA,CACF;CACA,MAAMsB,mBAAmB3E,cAErBoC,qBAAqB/B,uBACjB+B,UAAUwC,2BAA2B,IACrCrE,oBAAoB6B,SAAS,CAAC,EAAEyC,SACtC,CAACzC,SAAS,CACZ;CAEA,MAAM0C,sBAAsB7E,OAA0C,IAAI;CAG1E,IAAI8E,0BAA6D;CACjE,MAAMC,gBAAgBhF,cAElBoC,qBAAqB/B,uBACjB+B,UAAU6C,kBAAkBT,6BAA6B;EACvD,MAAMU,UACJJ,oBAAoBzB,WAAW0B;EACjC,IAAI,CAACG,SACH,MAAM,IAAIC,MAAM,0CAA0C;EAE5D,OAAOD;CACT,CAAC,IACD9C,WACN;EAAC2C;EAAyB3C;EAAWoC;CAAqB,CAC5D;CAEA,MAAMY,uBAAuBnF,OAAOkE,WAAW;CAC/CtE,gBAAgB;EACduF,qBAAqB/B,UAAUc;CACjC,CAAC;CAGD,MAAM,CAACkB,aAAanF,eAEhBkE,gBACAlB,WACE;EAAE,GAAGiB;EAAa5C;EAAIa,WAAW4C;CAAc,GAC/CI,oBACF,CACJ;CAEA,MAAMvD,OAAO7C,QAAQ;EACnB6C,MAAMuC,gBAAgBiB;EACtB,GAAI1E,aAAa4D,KAAAA,KAAa,EAAE5D,SAAS;CAC3C,CAAC;CAEDP,mBACEsB,2BACM;EACJ,KAAU4D,KAAK,CAAC,CAACC,YAAY,CAAC,CAAC;CACjC,GACA5D,iBACF;CAEA,MAAMK,UAAU1C,gBAAgBuC,MAAM;EACpCjB,UAAU;GACR,GAAGA;GACH4E,YAAY;IAAEC,UAAUlE;IAAI,GAAGX,UAAU4E;GAAW;EACtD;EACA,GAAGrG,+BAA+BgD,WAAW,CAAC,CAAC;EAC/C,GAAItB,mBAAmB,EAAEA,gBAAgB;EACzC,GAAIC,YAAY,EAAEA,SAAS;EAC3B,GAAIC,oBAAoB,EAAEA,iBAAiB;EAC3C,GAAIC,2BAA2B,EAAEA,wBAAwB;EACzD,GAAIG,gBAAgB,EAAEA,aAAa;EACnC,GAAIC,qBAAqB,EAAEA,kBAAkB;EAC7C,GAAIU,6BAA6B,EAC/B4D,oCAAoC5D,0BACtC;EAKA6D,4BAA4BvB,gBAAgBiB;EAC5C,GAAIhE,2BAA2B,EAAEA,wBAAwB;CAC3D,CAAC;CACD0D,0BAA0B;EACxB/C;EACAP,mBAAmBiD;CACrB;CACA3E,yBAAyB;EACvB+E,oBAAoBzB,UAAU;GAC5BrB;GACAP,mBAAmBiD;EACrB;CACF,GAAG,CAAC1C,SAAS0C,wBAAwB,CAAC;CAEtC,MAAMkB,2BAA2B9F,gBAC9B+F,kBAAoDC,WAAmB;EACtED,iBAAiBE,iBACfD,QACAtB,uBACAxC,SACA0C,wBACF;EACA,aACEmB,iBAAiBG,iBAAiBF,QAAQtB,qBAAqB;CACnE,CACF;CACAzE,yBAAyB;EACvB,IAAI,EAAEqC,qBAAqB/B,uBAAuB,OAAOkE,KAAAA;EACzD,OAAOqB,yBAAyBxD,WAAWb,EAAE;CAC/C,GAAG;EAACA;EAAIS;EAASI;EAAWoC;CAAqB,CAAC;CAElD,IACE,EAAEpC,qBAAqB/B,yBACvBiE,+BAA+B5E,wBAC/B;EACA4E,oBAAoB2B,WAAWjE,OAAO;EACtCsC,oBAAoB4B,gCAClBxB,wBACF;CACF;CAEA,MAAMyB,qBAAqBvG,aACxBwG,aAAyBpE,QAAQqE,OAAOC,UAAUF,QAAQ,GAC3D,CAACpE,OAAO,CACV;CACA,MAAMuE,4BAA4B3G,kBAC1BoC,QAAQqE,OAAOG,SAAS,CAAC,CAACC,WAChC,CAACzE,OAAO,CACV;CACA,MAAM0E,mBAAmBvG,qBACvBgG,oBACAI,2BACAA,yBACF;CAEA,MAAMI,8BAA8B/G,aACjCwG,eACC5E,eACKmD,kBAAkB2B,YAAYF,YAAU7E,EAAE,YAAY,CAAC,WAClD,CAAC,GACb;EAACA;EAAIC;EAAcmD;CAAgB,CACrC;CACA,MAAMiC,qBAAqBhH,kBAClB4B,eAAgBmD,kBAAkBkC,YAAYtF,EAAE,KAAK,OAAQ,MACpE;EAACA;EAAIC;EAAcmD;CAAgB,CACrC;CACA,MAAMmC,kBAAkB3G,qBACtBwG,6BACAC,oBACA3E,oBACF;CACA,MAAM8E,gBACJlF,KAAKmF,WAAW,eAAenF,KAAKmF,WAAW;CAEjD,MAAMC,yBAAyBjH,cACvBM,oBAAoBqE,gBAAgB,GAC1C,CAACA,gBAAgB,CACnB;CACA,MAAMuC,mBAAmBjH,OAAOgB,aAAa;CAC7CpB,gBAAgB;EACdqH,iBAAiB7D,UAAUpC;CAC7B,CAAC;CACDpB,gBAAgB;EACd,MAAMsH,mBACJ/E,qBAAqB/B,uBACjB+B,UAAU9B,oBAAoB,IAC9B2G;EACN,IAAI,CAACH,mBAAmBK,iBAAiBC,IAAIN,eAAe,GAC1D;EAEF,IAAIC,eAAe;GACjBI,iBAAiBE,IAAIP,eAAe;GACpC;EACF;EACA,IAAIJ,kBAAkB;EACtBS,iBAAiBE,IAAIP,eAAe;EACpC,MAAMQ,aAAalD,gBAAgBiB;EACnCiC,WAAWC,WAAW;EACtB,MAAMC,UAAU3F,KAAK4F,aAAa;EAClC,MAAMC,UAAU1E,eAAe2E,IAAIL,UAAU;EAC7CE,QACGI,WAAW;GAGV,IAAI5E,eAAe2E,IAAIL,UAAU,MAAMI,WAAWJ,WAAWpG,OAC3D,MAAMoG,WAAWpG;EAErB,CAAC,CAAC,CACDqE,OAAOsC,QAAiB;GACvBC,QAAQC,KAAK,2CAA2CF,GAAG;GAC3D,IAAI;IACFX,iBAAiB7D,UAAUwE,GAAG;GAChC,SAASG,eAAe;IACtBF,QAAQ5G,MACN,2DACA8G,aACF;GACF,UAAU;IACR,IAAIrD,kBAAkBkC,YAAYtF,EAAE,MAAMuF,iBACxCnC,iBAAiBsD,MAAM1G,EAAE;GAE7B;EACF,CAAC;CACL,GAAG;EACDM;EACAuC;EACA7C;EACAwF;EACAL;EACAI;EACAzB;EACAV;EACAsC;EACA7E;CAAS,CACV;CAED,OAAOJ;AACT"}

package/dist/runtime/useDynamicChatTransport.d.ts

@@ -0,0 +1,4 @@
+import type { UIMessage } from "@ai-sdk/react";
+import type { ChatTransport } from "ai";
+export declare const useDynamicChatTransport: <UI_MESSAGE extends UIMessage>(transport: ChatTransport<UI_MESSAGE>, enabled?: boolean) => ChatTransport<UI_MESSAGE>;
+//# sourceMappingURL=useDynamicChatTransport.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"useDynamicChatTransport.d.ts","sourceRoot":"","sources":["../../src/runtime/useDynamicChatTransport.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAIxC,eAAO,MAAM,uBAAuB,GAAI,UAAU,SAAS,SAAS,EAClE,WAAW,aAAa,CAAC,UAAU,CAAC,EACpC,iBAAc,KACb,aAAa,CAAC,UAAU,CAiB1B,CAAC"}

package/dist/runtime/useDynamicChatTransport.js

@@ -0,0 +1,64 @@
+import { DynamicChatTransport } from "./DynamicChatTransport.js";
+import { useEffect, useInsertionEffect, useState } from "@assistant-ui/tap/react-shim";
+import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+//#region src/runtime/useDynamicChatTransport.ts
+const useDynamicChatTransport = (transport, t0) => {
+	const $ = c(12);
+	const enabled = t0 === void 0 ? true : t0;
+	let t1;
+	if ($[0] !== transport) {
+		t1 = () => transport instanceof DynamicChatTransport ? transport : new DynamicChatTransport(transport);
+		$[0] = transport;
+		$[1] = t1;
+	} else t1 = $[1];
+	const [dynamicTransport] = useState(t1);
+	let t2;
+	let t3;
+	if ($[2] !== dynamicTransport || $[3] !== enabled || $[4] !== transport) {
+		t2 = () => {
+			if (!enabled || dynamicTransport === transport) return;
+			dynamicTransport.setTransport(transport);
+		};
+		t3 = [
+			dynamicTransport,
+			enabled,
+			transport
+		];
+		$[2] = dynamicTransport;
+		$[3] = enabled;
+		$[4] = transport;
+		$[5] = t2;
+		$[6] = t3;
+	} else {
+		t2 = $[5];
+		t3 = $[6];
+	}
+	useInsertionEffect(t2, t3);
+	let t4;
+	let t5;
+	if ($[7] !== dynamicTransport || $[8] !== enabled || $[9] !== transport) {
+		t4 = () => {
+			if (!enabled || dynamicTransport === transport) return;
+			dynamicTransport.flushTransportChange();
+		};
+		t5 = [
+			dynamicTransport,
+			enabled,
+			transport
+		];
+		$[7] = dynamicTransport;
+		$[8] = enabled;
+		$[9] = transport;
+		$[10] = t4;
+		$[11] = t5;
+	} else {
+		t4 = $[10];
+		t5 = $[11];
+	}
+	useEffect(t4, t5);
+	return enabled ? dynamicTransport : transport;
+};
+//#endregion
+export { useDynamicChatTransport };
+
+//# sourceMappingURL=useDynamicChatTransport.js.map

package/dist/runtime/useDynamicChatTransport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useDynamicChatTransport.js","names":["UIMessage","ChatTransport","useEffect","useInsertionEffect","useState","DynamicChatTransport","useDynamicChatTransport","transport","t0","$","_c","enabled","undefined","t1","dynamicTransport","t2","t3","setTransport","t4","t5","flushTransportChange","_c"],"sources":["../../src/runtime/useDynamicChatTransport.ts"],"sourcesContent":["import type { UIMessage } from \"@ai-sdk/react\";\nimport type { ChatTransport } from \"ai\";\nimport { useEffect, useInsertionEffect, useState } from \"react\";\nimport { DynamicChatTransport } from \"./DynamicChatTransport\";\n\nexport const useDynamicChatTransport = <UI_MESSAGE extends UIMessage>(\n  transport: ChatTransport<UI_MESSAGE>,\n  enabled = true,\n): ChatTransport<UI_MESSAGE> => {\n  const [dynamicTransport] = useState(() =>\n    transport instanceof DynamicChatTransport\n      ? transport\n      : new DynamicChatTransport(transport),\n  );\n\n  useInsertionEffect(() => {\n    if (!enabled || dynamicTransport === transport) return;\n    dynamicTransport.setTransport(transport);\n  }, [dynamicTransport, enabled, transport]);\n  useEffect(() => {\n    if (!enabled || dynamicTransport === transport) return;\n    dynamicTransport.flushTransportChange();\n  }, [dynamicTransport, enabled, transport]);\n\n  return enabled ? dynamicTransport : transport;\n};\n"],"mappings":";;;;AAKA,MAAaM,2BAA0BC,WAAAC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAErC,MAAAC,UAAAH,OAAAI,KAAAA,IAAA,OAAAJ;CAAc,IAAAK;CAAA,IAAAJ,EAAA,OAAAF,WAAA;EAEsBM,WAClCN,qBAAqBF,uBAArBE,YAAA,IAEQF,qBAAqBE,SAAS;EAACE,EAAA,KAAAF;EAAAE,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAHzC,MAAA,CAAAK,oBAA2BV,SAASS,EAIpC;CAAE,IAAAE;CAAA,IAAAC;CAAA,IAAAP,EAAA,OAAAK,oBAAAL,EAAA,OAAAE,WAAAF,EAAA,OAAAF,WAAA;EAEiBQ,WAAA;GACjB,IAAI,CAACJ,WAAWG,qBAAqBP,WAAS;GAC9CO,iBAAgBG,aAAcV,SAAS;EAAC;EACvCS,KAAA;GAACF;GAAkBH;GAASJ;EAAS;EAACE,EAAA,KAAAK;EAAAL,EAAA,KAAAE;EAAAF,EAAA,KAAAF;EAAAE,EAAA,KAAAM;EAAAN,EAAA,KAAAO;CAAA,OAAA;EAAAD,KAAAN,EAAA;EAAAO,KAAAP,EAAA;CAAA;CAHzCN,mBAAmBY,IAGhBC,EAAsC;CAAC,IAAAE;CAAA,IAAAC;CAAA,IAAAV,EAAA,OAAAK,oBAAAL,EAAA,OAAAE,WAAAF,EAAA,OAAAF,WAAA;EAChCW,WAAA;GACR,IAAI,CAACP,WAAWG,qBAAqBP,WAAS;GAC9CO,iBAAgBM,qBAAsB;EAAC;EACtCD,KAAA;GAACL;GAAkBH;GAASJ;EAAS;EAACE,EAAA,KAAAK;EAAAL,EAAA,KAAAE;EAAAF,EAAA,KAAAF;EAAAE,EAAA,MAAAS;EAAAT,EAAA,MAAAU;CAAA,OAAA;EAAAD,KAAAT,EAAA;EAAAU,KAAAV,EAAA;CAAA;CAHzCP,UAAUgB,IAGPC,EAAsC;CAAC,OAEnCR,UAAAG,mBAAAP;AAAsC"}

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

@@ -1 +1 @@
-{"version":3,"file":"useExternalHistory.d.ts","sourceRoot":"","sources":["../../src/runtime/useExternalHistory.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,gBAAgB,EAChB,oBAAoB,EACpB,aAAa,EACb,oBAAoB,EAEpB,uBAAuB,EACvB,yBAAyB,EACzB,4BAA4B,EAC5B,2BAA2B,EAC5B,MAAM,oBAAoB,CAAC;AAO5B,OAAO,EAIL,KAAK,SAAS,EAGf,MAAM,OAAO,CAAC;AAGf,eAAO,MAAM,2BAA2B,GAAI,QAAQ,EAClD,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,aAAa,EAAE,EAC3D,UAAU,uBAAuB,CAAC,QAAQ,CAAC,KAC1C,yBAoBF,CAAC;AAsOF,eAAO,MAAM,kBAAkB,GAAI,QAAQ,EACzC,YAAY,SAAS,CAAC,gBAAgB,CAAC,EACvC,gBAAgB,oBAAoB,GAAG,SAAS,EAChD,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,aAAa,EAAE,EAC3D,sBAAsB,oBAAoB,CAAC,QAAQ,EAAE,GAAG,CAAC,EACzD,eAAe,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,IAAI,EAC7C,gBAAgB,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,EACpC,0BAA0B,MAAM,IAAI,EACpC,mBAAmB,GAAG,CAAC,MAAM,EAAE,2BAA2B,CAAC,EAC3D,6BAA6B,MAAM,IAAI,EACvC,wBAAwB,GAAG,CAAC,MAAM,EAAE,4BAA4B,CAAC,EACjE,kCAAkC,MAAM,IAAI;;+BA+dxB,MAAM;yCA3UZ,MAAM;8CAAN,MAAM;CAyXrB,CAAC"}
+{"version":3,"file":"useExternalHistory.d.ts","sourceRoot":"","sources":["../../src/runtime/useExternalHistory.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,gBAAgB,EAChB,oBAAoB,EACpB,aAAa,EACb,oBAAoB,EAEpB,uBAAuB,EACvB,yBAAyB,EACzB,4BAA4B,EAC5B,2BAA2B,EAC5B,MAAM,oBAAoB,CAAC;AAI5B,OAAO,EAIL,KAAK,SAAS,EAGf,MAAM,OAAO,CAAC;AAUf,eAAO,MAAM,2BAA2B,GAAI,QAAQ,EAClD,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,aAAa,EAAE,EAC3D,UAAU,uBAAuB,CAAC,QAAQ,CAAC,KAC1C,yBAoBF,CAAC;AA2BF,eAAO,MAAM,kBAAkB,GAAI,QAAQ,EACzC,YAAY,SAAS,CAAC,gBAAgB,CAAC,EACvC,gBAAgB,oBAAoB,GAAG,SAAS,EAChD,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,aAAa,EAAE,EAC3D,sBAAsB,oBAAoB,CAAC,QAAQ,EAAE,GAAG,CAAC,EACzD,eAAe,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,IAAI,EAC7C,gBAAgB,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,EACpC,0BAA0B,MAAM,IAAI,EACpC,mBAAmB,GAAG,CAAC,MAAM,EAAE,2BAA2B,CAAC,EAC3D,6BAA6B,MAAM,IAAI,EACvC,wBAAwB,GAAG,CAAC,MAAM,EAAE,4BAA4B,CAAC,EACjE,kCAAkC,MAAM,IAAI;;+BAkexB,MAAM;yCA3UZ,MAAM;8CAAN,MAAM;CAyXrB,CAAC"}

package/dist/runtime/useExternalHistory.js

@@ -1,6 +1,6 @@
 "use client";
-import { normalizeToolApprovalAnswers } from "../converters/toolApprovalAnswers.js";
-import { MessageRepository, readToolInteractionLog } from "@assistant-ui/core/internal";
+import { addToolData, collectToolApprovalResponses, collectToolArtifacts, collectToolInteractions, restoreToolData } from "./toolHistoryCodec.js";
+import { MessageRepository } from "@assistant-ui/core/internal";
 import { getExternalStoreMessages } from "@assistant-ui/core";
 import { useCallback, useEffect, useMemo, useRef, useState } from "@assistant-ui/tap/react-shim";
 import { useAui } from "@assistant-ui/store";
@@ -28,107 +28,6 @@ const toExportedMessageRepository = (toThreadMessages, messages) => {
 const hasUnansweredApproval = (message) => message.content.some((part) => part.type === "tool-call" && part.approval != null && part.approval.approved === void 0 && part.approval.resolution === void 0);
 const isAwaitingToolApproval = (message) => message.status?.type === "requires-action" && (message.status.reason === "tool-calls" || message.status.reason === "interrupt" && hasUnansweredApproval(message));
 const isTerminalMessage = (message) => message.status === void 0 || message.status.type === "complete" || message.status.type === "incomplete";
-const TOOL_ARTIFACTS_METADATA_KEY = "__aui_toolArtifacts";
-const TOOL_INTERACTIONS_METADATA_KEY = "__aui_toolInteractions";
-const TOOL_APPROVAL_RESPONSES_METADATA_KEY = "__aui_toolApprovalResponses";
-const isRecord = (value) => value !== null && typeof value === "object" && !Array.isArray(value);
-const collectToolArtifacts = (message, toolArtifacts) => {
-	if (!toolArtifacts) return void 0;
-	const entries = message.content.flatMap((part) => {
-		if (part.type !== "tool-call") return [];
-		const artifact = toolArtifacts.get(part.toolCallId);
-		return artifact === void 0 ? [] : [[part.toolCallId, artifact]];
-	});
-	return entries.length > 0 ? Object.fromEntries(entries) : void 0;
-};
-const collectToolInteractions = (message, toolInteractions) => {
-	if (!toolInteractions) return void 0;
-	const entries = message.content.flatMap((part) => {
-		if (part.type !== "tool-call") return [];
-		const interactions = toolInteractions.get(part.toolCallId);
-		return interactions === void 0 ? [] : [[part.toolCallId, interactions]];
-	});
-	return entries.length > 0 ? Object.fromEntries(entries) : void 0;
-};
-const collectToolApprovalResponses = (message, toolApprovalResponses) => {
-	if (!toolApprovalResponses) return void 0;
-	const entries = message.content.flatMap((part) => {
-		if (part.type !== "tool-call" || !part.approval) return [];
-		const response = toolApprovalResponses.get(part.approval.id);
-		if (!response) return [];
-		return [[part.approval.id, {
-			approved: response.approved,
-			...response.optionId != null && { optionId: response.optionId },
-			...response.text != null && { text: response.text },
-			...response.answers != null && { answers: response.answers },
-			...response.reason != null && { reason: response.reason }
-		}]];
-	});
-	return entries.length > 0 ? Object.fromEntries(entries) : void 0;
-};
-const addToolData = (message, toolArtifacts, toolInteractions, toolApprovalResponses) => {
-	if (!toolArtifacts && !toolInteractions && !toolApprovalResponses || !isRecord(message) || !Array.isArray(message.parts)) return message;
-	const toolCallIds = message.parts.flatMap((part) => {
-		if (!isRecord(part) || typeof part.toolCallId !== "string") return [];
-		return [part.toolCallId];
-	});
-	const artifacts = toolArtifacts ? Object.fromEntries(toolCallIds.flatMap((toolCallId) => Object.hasOwn(toolArtifacts, toolCallId) ? [[toolCallId, toolArtifacts[toolCallId]]] : [])) : void 0;
-	const interactions = toolInteractions ? Object.fromEntries(toolCallIds.flatMap((toolCallId) => Object.hasOwn(toolInteractions, toolCallId) ? [[toolCallId, toolInteractions[toolCallId]]] : [])) : void 0;
-	const approvalIds = message.parts.flatMap((part) => {
-		if (!isRecord(part) || !isRecord(part.approval)) return [];
-		const approvalId = part.approval.id;
-		return typeof approvalId === "string" ? [approvalId] : [];
-	});
-	const approvalResponses = toolApprovalResponses ? Object.fromEntries(approvalIds.flatMap((approvalId) => Object.hasOwn(toolApprovalResponses, approvalId) ? [[approvalId, toolApprovalResponses[approvalId]]] : [])) : void 0;
-	const hasArtifacts = !!artifacts && Object.keys(artifacts).length > 0;
-	const hasInteractions = !!interactions && Object.keys(interactions).length > 0;
-	const hasApprovalResponses = !!approvalResponses && Object.keys(approvalResponses).length > 0;
-	if (!hasArtifacts && !hasInteractions && !hasApprovalResponses) return message;
-	const metadata = isRecord(message.metadata) ? message.metadata : {};
-	return {
-		...message,
-		metadata: {
-			...metadata,
-			...hasArtifacts && { [TOOL_ARTIFACTS_METADATA_KEY]: artifacts },
-			...hasInteractions && { [TOOL_INTERACTIONS_METADATA_KEY]: interactions },
-			...hasApprovalResponses && { [TOOL_APPROVAL_RESPONSES_METADATA_KEY]: approvalResponses }
-		}
-	};
-};
-const restoreToolData = (message, toolArtifacts, toolInteractions, toolApprovalResponses) => {
-	if (!isRecord(message) || !isRecord(message.metadata)) return message;
-	const metadata = message.metadata;
-	const hasArtifacts = Object.hasOwn(metadata, TOOL_ARTIFACTS_METADATA_KEY);
-	const hasInteractions = Object.hasOwn(metadata, TOOL_INTERACTIONS_METADATA_KEY);
-	const hasApprovalResponses = Object.hasOwn(metadata, TOOL_APPROVAL_RESPONSES_METADATA_KEY);
-	if (!hasArtifacts && !hasInteractions && !hasApprovalResponses) return message;
-	const artifacts = metadata[TOOL_ARTIFACTS_METADATA_KEY];
-	if (toolArtifacts && isRecord(artifacts)) for (const [toolCallId, artifact] of Object.entries(artifacts)) toolArtifacts.set(toolCallId, artifact);
-	const interactions = metadata[TOOL_INTERACTIONS_METADATA_KEY];
-	if (toolInteractions && isRecord(interactions)) for (const [toolCallId, value] of Object.entries(interactions)) {
-		const log = readToolInteractionLog(value);
-		if (log) toolInteractions.set(toolCallId, log);
-	}
-	const approvalResponses = metadata[TOOL_APPROVAL_RESPONSES_METADATA_KEY];
-	if (toolApprovalResponses && isRecord(approvalResponses)) for (const [approvalId, value] of Object.entries(approvalResponses)) {
-		if (!isRecord(value) || typeof value.approved !== "boolean") continue;
-		const answers = normalizeToolApprovalAnswers(value.answers);
-		toolApprovalResponses.set(approvalId, {
-			approvalId,
-			approved: value.approved,
-			...typeof value.optionId === "string" && { optionId: value.optionId },
-			...typeof value.text === "string" && { text: value.text },
-			...answers && Object.keys(answers).length > 0 && { answers },
-			...typeof value.reason === "string" && { reason: value.reason }
-		});
-	}
-	const { [TOOL_ARTIFACTS_METADATA_KEY]: _, [TOOL_INTERACTIONS_METADATA_KEY]: __, [TOOL_APPROVAL_RESPONSES_METADATA_KEY]: ___, ...restMetadata } = metadata;
-	const { metadata: _metadata, ...restMessage } = message;
-	return Object.keys(restMetadata).length === 0 ? restMessage : {
-		...restMessage,
-		metadata: restMetadata
-	};
-};
 const encodeContent = (storageFormatAdapter, item) => JSON.stringify(storageFormatAdapter.encode(item));
 const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storageFormatAdapter, onSetMessages, toolArtifacts, onToolArtifactsRestored, toolInteractions, onToolInteractionsRestored, toolApprovalResponses, onToolApprovalResponsesRestored) => {
 	const loadedRef = useRef(false);
@@ -154,6 +53,8 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 		const loadHistory = async () => {
 			try {
 				const repo = await formatAdapter.load();
+				const threadState = runtimeRef.current.thread.getState();
+				if (threadState.isRunning || threadState.messages.length > 0) return;
 				toolArtifacts?.clear();
 				toolInteractions?.clear();
 				toolApprovalResponses?.clear();
@@ -198,13 +99,14 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 				if (optionalThreadListItem()?.getState().remoteId) setItemEpoch((n) => n + 1);
 			});
 		}
-		const threadState = runtimeRef.current.thread.getState();
-		if (threadState.isRunning || threadState.messages.length > 0) {
+		const threadState_0 = runtimeRef.current.thread.getState();
+		if (threadState_0.isRunning || threadState_0.messages.length > 0) {
 			loadedRef.current = true;
 			setHasLoaded(true);
 			return;
 		}
 		loadedRef.current = true;
+		setHasLoaded(false);
 		loadHistory();
 	}, [
 		formatAdapter,
@@ -277,12 +179,12 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 		const idleScheduledInnerIds = /* @__PURE__ */ new Set();
 		const unpersistedInnerIds = (message_0) => isTerminalMessage(message_0) ? getExternalStoreMessages(message_0).map((innerMessage_1) => storageFormatAdapter.getId(innerMessage_1)).filter((innerId_0) => !persistedInnerMessages.current.has(innerId_0)) : [];
 		const unsubscribe = runtimeRef.current.thread.subscribe(() => {
-			const threadState_0 = runtimeRef.current.thread.getState();
-			const { isRunning } = threadState_0;
+			const threadState_1 = runtimeRef.current.thread.getState();
+			const { isRunning } = threadState_1;
 			const wasRunning = wasRunningRef.current;
 			wasRunningRef.current = isRunning;
 			if (runStartRef.current != null) {
-				const lastMsg = threadState_0.messages.at(-1);
+				const lastMsg = threadState_1.messages.at(-1);
 				if (lastMsg?.role === "assistant") {
 					const currentToolCallCount = lastMsg.content.filter((p) => p.type === "tool-call").length;
 					while (toolCallCountRef.current < currentToolCallCount) {
@@ -307,7 +209,7 @@ const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storag
 			if (wasRunning) {
 				if (runStartRef.current != null) stepBoundariesRef.current.push(Date.now() - runStartRef.current);
 			} else {
-				const pending = threadState_0.messages.flatMap(unpersistedInnerIds);
+				const pending = threadState_1.messages.flatMap(unpersistedInnerIds);
 				if (pending.every((innerId_1) => idleScheduledInnerIds.has(innerId_1))) return;
 				for (const innerId_2 of pending) idleScheduledInnerIds.add(innerId_2);
 			}