@assistant-ui/react-ai-sdk 1.3.30 → 1.3.32

package/dist/toolOutputConversion.d.ts

@@ -0,0 +1,34 @@
+import { ToolModelContentPart } from "./assistant-stream/dist/core/tool/tool-types.js";
+import { JSONValue } from "ai";
+
+//#region src/toolOutputConversion.d.ts
+declare const toAISDKContent: (parts: readonly ToolModelContentPart[]) => {
+  type: "content";
+  value: ({
+    type: "text";
+    text: string;
+    data?: never;
+    mediaType?: never;
+  } | {
+    type: "image-data";
+    data: string;
+    mediaType: string;
+    text?: never;
+  } | {
+    filename?: string;
+    type: "file-data";
+    data: string;
+    mediaType: string;
+    text?: never;
+  })[];
+};
+declare const toAISDKDefaultOutput: (output: unknown) => {
+  type: "text";
+  value: string;
+} | {
+  type: "json";
+  value: JSONValue;
+};
+//#endregion
+export { toAISDKContent, toAISDKDefaultOutput };
+//# sourceMappingURL=toolOutputConversion.d.ts.map

package/dist/toolOutputConversion.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolOutputConversion.d.ts","names":[],"sources":["../src/toolOutputConversion.ts"],"mappings":";;;;cAGa,cAAA,GAAkB,KAAA,WAAgB,oBAAoB;;;;;;;;;;;;;;;;;;;;cAsBtD,oBAAA,GAAwB,MAAA;;;;;SAGqB,SAAS;AAAA"}

package/dist/toolOutputConversion.js

@@ -0,0 +1,31 @@
+//#region src/toolOutputConversion.ts
+const toAISDKContent = (parts) => ({
+	type: "content",
+	value: parts.map((part) => {
+		if (part.type === "text") return {
+			type: "text",
+			text: part.text
+		};
+		return part.mediaType.startsWith("image/") ? {
+			type: "image-data",
+			data: part.data,
+			mediaType: part.mediaType
+		} : {
+			type: "file-data",
+			data: part.data,
+			mediaType: part.mediaType,
+			...part.filename !== void 0 && { filename: part.filename }
+		};
+	})
+});
+const toAISDKDefaultOutput = (output) => typeof output === "string" ? {
+	type: "text",
+	value: output
+} : {
+	type: "json",
+	value: output ?? null
+};
+//#endregion
+export { toAISDKContent, toAISDKDefaultOutput };
+
+//# sourceMappingURL=toolOutputConversion.js.map

package/dist/toolOutputConversion.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolOutputConversion.js","names":[],"sources":["../src/toolOutputConversion.ts"],"sourcesContent":["import type { JSONValue } from \"ai\";\nimport type { ToolModelContentPart } from \"assistant-stream\";\n\nexport const toAISDKContent = (parts: readonly ToolModelContentPart[]) => ({\n  type: \"content\" as const,\n  value: parts.map((part) => {\n    if (part.type === \"text\") {\n      return { type: \"text\" as const, text: part.text };\n    }\n    const isImage = part.mediaType.startsWith(\"image/\");\n    return isImage\n      ? {\n          type: \"image-data\" as const,\n          data: part.data,\n          mediaType: part.mediaType,\n        }\n      : {\n          type: \"file-data\" as const,\n          data: part.data,\n          mediaType: part.mediaType,\n          ...(part.filename !== undefined && { filename: part.filename }),\n        };\n  }),\n});\n\nexport const toAISDKDefaultOutput = (output: unknown) =>\n  typeof output === \"string\"\n    ? { type: \"text\" as const, value: output }\n    : { type: \"json\" as const, value: (output ?? null) as JSONValue };\n"],"mappings":";AAGA,MAAa,kBAAkB,WAA4C;CACzE,MAAM;CACN,OAAO,MAAM,KAAK,SAAS;EACzB,IAAI,KAAK,SAAS,QAChB,OAAO;GAAE,MAAM;GAAiB,MAAM,KAAK;EAAK;EAGlD,OADgB,KAAK,UAAU,WAAW,QAC7B,IACT;GACE,MAAM;GACN,MAAM,KAAK;GACX,WAAW,KAAK;EAClB,IACA;GACE,MAAM;GACN,MAAM,KAAK;GACX,WAAW,KAAK;GAChB,GAAI,KAAK,aAAa,KAAA,KAAa,EAAE,UAAU,KAAK,SAAS;EAC/D;CACN,CAAC;AACH;AAEA,MAAa,wBAAwB,WACnC,OAAO,WAAW,WACd;CAAE,MAAM;CAAiB,OAAO;AAAO,IACvC;CAAE,MAAM;CAAiB,OAAQ,UAAU;AAAmB"}

package/dist/ui/resumable.d.ts

@@ -1,4 +1,4 @@
-import { RESUMABLE_STREAM_ID_HEADER } from "../packages/assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
+import { RESUMABLE_STREAM_ID_HEADER } from "../assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
 //#region src/ui/resumable.d.ts
 type ResumableClientStorage = {
   getStreamId(): string | null;

package/dist/ui/resumable.js

@@ -1,5 +1,5 @@
 "use client";
-import { RESUMABLE_STREAM_ID_HEADER } from "../packages/assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
+import { RESUMABLE_STREAM_ID_HEADER } from "../assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
 //#region src/ui/resumable.ts
 const DEFAULT_STORAGE_KEY = "aui-resumable-stream-id";
 /** `sessionStorage`-backed storage for the pending resumable stream id. */

package/dist/ui/use-chat/AssistantChatTransport.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantChatTransport.d.ts","names":[],"sources":["../../../src/ui/use-chat/AssistantChatTransport.ts"],"mappings":";;;;;KAeK,2BAAA,GAA8B,IAAI,CAAC,qBAAA;AAAA,KAM5B,iCAAA,oBAAqD,SAAA,IAC/D,4BAAA,CAA6B,UAAA;EAC3B,SAAA,GAAY,6BAAA;AAAA;AAAA,cAGH,sBAAA,oBACQ,SAAA,UACX,oBAAA,CAAqB,UAAA;EAAA,QACrB,OAAA;EAAA,QACA,iBAAA;EAAA,iBAGS,SAAA;cAEL,WAAA,GAAc,iCAAA,CAAkC,UAAA;EAkD5D,UAAA,CAAW,OAAA,EAAS,gBAAA;EAIpB,mBAAA,CAAA,GAAuB,6BAAA;EAIvB,+BAAA,CACE,MAAA,QAAc,2BAAA;AAAA"}
+{"version":3,"file":"AssistantChatTransport.d.ts","names":[],"sources":["../../../src/ui/use-chat/AssistantChatTransport.ts"],"mappings":";;;;;KAeK,2BAAA,GAA8B,IAAI,CAAC,qBAAA;AAAA,KAM5B,iCAAA,oBAAqD,SAAA,IAC/D,4BAAA,CAA6B,UAAA;EAC3B,SAAA,GAAY,6BAAA;AAAA;AAAA,cAGH,sBAAA,oBACQ,SAAA,UACX,oBAAA,CAAqB,UAAA;EAAA,QACrB,OAAA;EAAA,QACA,iBAAA;EAAA,iBAGS,SAAA;cAEL,WAAA,GAAc,iCAAA,CAAkC,UAAA;EAkD5D,UAAA,CAAW,OAAA,EAAS,gBAAA;EAIpB,mBAAA,IAAuB,6BAAA;EAIvB,+BAAA,CACE,MAAA,QAAc,2BAAA;AAAA"}

package/dist/ui/use-chat/AssistantChatTransport.js

@@ -1,5 +1,5 @@
-import { toToolsJSONSchema } from "../../packages/assistant-stream/dist/core/tool/schema-utils.js";
-import { RESUMABLE_STREAM_ID_HEADER } from "../../packages/assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
+import { toToolsJSONSchema } from "../../assistant-stream/dist/core/tool/schema-utils.js";
+import { RESUMABLE_STREAM_ID_HEADER } from "../../assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
 import { DefaultChatTransport } from "ai";
 //#region src/ui/use-chat/AssistantChatTransport.ts
 const FINISH_MARKER = "\"type\":\"finish\"";

package/dist/ui/use-chat/useAISDKRuntime.d.ts

@@ -1,9 +1,9 @@
-import { AppendMessage, AssistantRuntime, ExternalStoreAdapter, ThreadHistoryAdapter, ThreadSuggestion } from "@assistant-ui/core";
+import { AppendMessage, AssistantRuntime, ExternalStoreAdapter, ExternalStoreSharedOptions, ThreadHistoryAdapter } from "@assistant-ui/core";
 import { CreateUIMessage, UIMessage, useChat } from "@ai-sdk/react";
 
 //#region src/ui/use-chat/useAISDKRuntime.d.ts
 type CustomToCreateMessageFunction = <UI_MESSAGE extends UIMessage = UIMessage>(message: AppendMessage) => CreateUIMessage<UI_MESSAGE>;
-type AISDKRuntimeAdapter = {
+type AISDKRuntimeAdapter = ExternalStoreSharedOptions & {
   adapters?: (NonNullable<ExternalStoreAdapter["adapters"]> & {
     history?: ThreadHistoryAdapter | undefined;
   }) | undefined;
@@ -25,21 +25,8 @@ type AISDKRuntimeAdapter = {
    * (for example, an SSE reconnect endpoint keyed by turn id).
    */
   onResume?: ExternalStoreAdapter["onResume"];
-  /**
-   * Follow up suggestions to surface on the thread. Use this to drive
-   * dynamic suggestions from application state, tool results, or backend
-   * responses; flows into `thread.suggestions` and is rendered by
-   * components that read it (such as the shadcn `ThreadFollowupSuggestions`).
-   */
-  suggestions?: readonly ThreadSuggestion[] | undefined;
 };
-declare const useAISDKRuntime: <UI_MESSAGE extends UIMessage = UIMessage>(chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>, {
-  adapters,
-  toCreateMessage: customToCreateMessage,
-  cancelPendingToolCallsOnSend,
-  onResume,
-  suggestions
-}?: AISDKRuntimeAdapter) => AssistantRuntime;
+declare const useAISDKRuntime: <UI_MESSAGE extends UIMessage = UIMessage>(chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>, adapter?: AISDKRuntimeAdapter) => AssistantRuntime;
 //#endregion
 export { AISDKRuntimeAdapter, CustomToCreateMessageFunction, useAISDKRuntime };
 //# sourceMappingURL=useAISDKRuntime.d.ts.map

package/dist/ui/use-chat/useAISDKRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAISDKRuntime.d.ts","names":[],"sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"mappings":";;;;KAyCY,6BAAA,uBACS,SAAA,GAAY,SAAA,EAE/B,OAAA,EAAS,aAAA,KACN,eAAA,CAAgB,UAAA;AAAA,KAYT,mBAAA;EACV,QAAA,IACK,WAAA,CAAY,oBAAA;IACX,OAAA,GAAU,oBAAA;EAAA;EAGhB,eAAA,GAAkB,6BAAA;EArBa;;;;;;;;EA8B/B,4BAAA;EA5BS;;;;;AACoB;AAY/B;EAuBE,QAAA,GAAW,oBAAA;;;;;;;EAOX,WAAA,YAAuB,gBAAA;AAAA;AAAA,cAGZ,eAAA,sBAAsC,SAAA,GAAY,SAAA,EAC7D,WAAA,EAAa,UAAA,QAAkB,OAAA,CAAQ,UAAA;EACvC,QAAA;EAAA,eAAA,EAAA,qBAAA;EAAA,4BAAA;EAAA,QAAA;EAAA;AAAA,IAMG,mBAAA,KAAwB,gBAAA"}
+{"version":3,"file":"useAISDKRuntime.d.ts","names":[],"sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"mappings":";;;;KA4CY,6BAAA,uBACS,SAAA,GAAY,SAAA,EAE/B,OAAA,EAAS,aAAA,KACN,eAAA,CAAgB,UAAA;AAAA,KAYT,mBAAA,GAAsB,0BAAA;EAChC,QAAA,IACK,WAAA,CAAY,oBAAA;IACX,OAAA,GAAU,oBAAA;EAAA;EAGhB,eAAA,GAAkB,6BAAA;EArBa;;;;;;;;EA8B/B,4BAAA;EA5BS;;;;;AACoB;AAY/B;EAuBE,QAAA,GAAW,oBAAA;AAAA;AAAA,cAGA,eAAA,sBAAsC,SAAA,GAAY,SAAA,EAC7D,WAAA,EAAa,UAAA,QAAkB,OAAA,CAAQ,UAAA,IACvC,OAAA,GAAS,mBAAA,KAAwB,gBAAA"}

package/dist/ui/use-chat/useAISDKRuntime.js

@@ -11,14 +11,15 @@ import { useStreamingTiming } from "./useStreamingTiming.js";
 import { generateId, isToolUIPart } from "ai";
 import { useMemo, useRef, useState } from "react";
 import { useExternalStoreRuntime, useRuntimeAdapters } from "@assistant-ui/core/react";
-import { getExternalStoreMessages } from "@assistant-ui/core";
+import { getExternalStoreMessages, pickExternalStoreSharedOptions } from "@assistant-ui/core";
 //#region src/ui/use-chat/useAISDKRuntime.ts
 const toUIMessage = (createMessage, fallbackRole) => ({
 	...createMessage,
 	id: createMessage.id ?? generateId(),
 	role: createMessage.role ?? fallbackRole
 });
-const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, onResume, suggestions } = {}) => {
+const useAISDKRuntime = (chatHelpers, adapter = {}) => {
+	const { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, onResume } = adapter;
 	const contextAdapters = useRuntimeAdapters();
 	const [toolStatuses, setToolStatuses] = useState({});
 	const toolArgsKeyOrderCacheRef = useRef(/* @__PURE__ */ new Map());
@@ -28,6 +29,8 @@ const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: customToCreat
 	const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
 	const isRunning = chatHelpers.status === "submitted" || chatHelpers.status === "streaming" || hasExecutingTools;
 	const messageTiming = useStreamingTiming(chatHelpers.messages, isRunning);
+	const lastMessage = chatHelpers.messages.at(-1);
+	const optimisticMessageId = isRunning && lastMessage?.role === "assistant" ? lastMessage.id : void 0;
 	const messages = AISDKMessageConverter.useThreadMessages({
 		isRunning,
 		messages: chatHelpers.messages,
@@ -37,10 +40,12 @@ const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: customToCreat
 			toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,
 			toolLastInputCache: toolLastInputCacheRef.current,
 			mcpAppMetadataCache: mcpAppMetadataCacheRef.current,
+			...optimisticMessageId && { optimisticMessageId },
 			...chatHelpers.error && { error: chatHelpers.error.message }
 		}), [
 			toolStatuses,
 			messageTiming,
+			optimisticMessageId,
 			chatHelpers.error
 		])
 	});
@@ -60,8 +65,9 @@ const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: customToCreat
 				if (!isToolUIPart(part)) return part;
 				if (part.state === "output-available" || part.state === "output-error") return part;
 				hasChanges = true;
+				const { approval: _approval, ...rest } = part;
 				return {
-					...part,
+					...rest,
 					state: "output-error",
 					errorText: "User cancelled tool call by sending a new message."
 				};
@@ -160,8 +166,8 @@ const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: customToCreat
 				options: { metadata: lastRunConfigRef.current }
 			});
 		},
+		...pickExternalStoreSharedOptions(adapter),
 		...onResume && { onResume },
-		...suggestions && { suggestions },
 		adapters: {
 			attachments: vercelAttachmentAdapter,
 			...contextAdapters,

package/dist/ui/use-chat/useAISDKRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAISDKRuntime.js","names":[],"sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useMemo, useRef, useState } from \"react\";\nimport type { UIMessage, useChat, CreateUIMessage } from \"@ai-sdk/react\";\nimport { isToolUIPart, generateId } from \"ai\";\nimport {\n  useExternalStoreRuntime,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport type {\n  ExternalStoreAdapter,\n  ThreadHistoryAdapter,\n  AssistantRuntime,\n  ThreadMessage,\n  ThreadSuggestion,\n  MessageFormatAdapter,\n  MessageFormatItem,\n  MessageFormatRepository,\n  AppendMessage,\n  RunConfig,\n  McpAppMetadata,\n} from \"@assistant-ui/core\";\nimport { getExternalStoreMessages } from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../getVercelAIMessages\";\nimport { AISDKMessageConverter } from \"../utils/convertMessage\";\nimport { wrapModelContentEnvelope } from \"../../modelContentEnvelope\";\nimport {\n  type AISDKStorageFormat,\n  aiSDKV6FormatAdapter,\n} from \"../adapters/aiSDKFormatAdapter\";\nimport {\n  useExternalHistory,\n  toExportedMessageRepository,\n} from \"./useExternalHistory\";\nimport { useStreamingTiming } from \"./useStreamingTiming\";\n\nexport type CustomToCreateMessageFunction = <\n  UI_MESSAGE extends UIMessage = UIMessage,\n>(\n  message: AppendMessage,\n) => CreateUIMessage<UI_MESSAGE>;\n\nconst toUIMessage = <UI_MESSAGE extends UIMessage>(\n  createMessage: CreateUIMessage<UI_MESSAGE>,\n  fallbackRole: UI_MESSAGE[\"role\"],\n): UI_MESSAGE =>\n  ({\n    ...createMessage,\n    id: createMessage.id ?? generateId(),\n    role: createMessage.role ?? fallbackRole,\n  }) as UI_MESSAGE;\n\nexport type AISDKRuntimeAdapter = {\n  adapters?:\n    | (NonNullable<ExternalStoreAdapter[\"adapters\"]> & {\n        history?: ThreadHistoryAdapter | undefined;\n      })\n    | undefined;\n  toCreateMessage?: CustomToCreateMessageFunction;\n  /**\n   * Whether to automatically cancel pending interactive tool calls when the user sends a new message.\n   *\n   * When enabled (default), the pending tool calls will be marked as failed with an error message\n   * indicating the user cancelled the tool call by sending a new message.\n   *\n   * @default true\n   */\n  cancelPendingToolCallsOnSend?: boolean | undefined;\n  /**\n   * Called when `runtime.thread.resumeRun(config)` is invoked.\n   *\n   * When omitted, `resumeRun` throws `\"Runtime does not support resuming runs.\"`.\n   * Provide this to bridge resume invocations into a custom replay channel\n   * (for example, an SSE reconnect endpoint keyed by turn id).\n   */\n  onResume?: ExternalStoreAdapter[\"onResume\"];\n  /**\n   * Follow up suggestions to surface on the thread. Use this to drive\n   * dynamic suggestions from application state, tool results, or backend\n   * responses; flows into `thread.suggestions` and is rendered by\n   * components that read it (such as the shadcn `ThreadFollowupSuggestions`).\n   */\n  suggestions?: readonly ThreadSuggestion[] | undefined;\n};\n\nexport const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>,\n  {\n    adapters,\n    toCreateMessage: customToCreateMessage,\n    cancelPendingToolCallsOnSend = true,\n    onResume,\n    suggestions,\n  }: AISDKRuntimeAdapter = {},\n) => {\n  const contextAdapters = useRuntimeAdapters();\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const toolArgsKeyOrderCacheRef = useRef<Map<string, Map<string, string[]>>>(\n    new Map(),\n  );\n  const toolLastInputCacheRef = useRef<Map<string, ReadonlyJSONObject>>(\n    new Map(),\n  );\n  const mcpAppMetadataCacheRef = useRef<Map<string, McpAppMetadata>>(new Map());\n  const lastRunConfigRef = useRef<RunConfig | undefined>(undefined);\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const isRunning =\n    chatHelpers.status === \"submitted\" ||\n    chatHelpers.status === \"streaming\" ||\n    hasExecutingTools;\n\n  const messageTiming = useStreamingTiming(chatHelpers.messages, isRunning);\n\n  const messages = AISDKMessageConverter.useThreadMessages({\n    isRunning,\n    messages: chatHelpers.messages,\n    metadata: useMemo(\n      () => ({\n        toolStatuses,\n        messageTiming,\n        toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,\n        toolLastInputCache: toolLastInputCacheRef.current,\n        mcpAppMetadataCache: mcpAppMetadataCacheRef.current,\n        ...(chatHelpers.error && { error: chatHelpers.error.message }),\n      }),\n      [toolStatuses, messageTiming, chatHelpers.error],\n    ),\n  });\n\n  const [runtimeRef] = useState(() => ({\n    get current(): AssistantRuntime {\n      return runtime;\n    },\n  }));\n\n  const isLoading = useExternalHistory(\n    runtimeRef,\n    adapters?.history ?? contextAdapters?.history,\n    AISDKMessageConverter.toThreadMessages as (\n      messages: UI_MESSAGE[],\n    ) => ThreadMessage[],\n    aiSDKV6FormatAdapter as MessageFormatAdapter<\n      UI_MESSAGE,\n      AISDKStorageFormat\n    >,\n    (messages) => {\n      chatHelpers.setMessages(messages);\n    },\n  );\n\n  const completePendingToolCalls = async () => {\n    if (!cancelPendingToolCallsOnSend) return;\n\n    // The runtime auto-aborts in-flight tool invocations when a new run\n    // is dispatched (append() / startRun()). All we need to do here is\n    // mark any tool without a result as cancelled in the UI message list.\n\n    // Mark any tool without a result as cancelled (uses setMessages to avoid triggering sendAutomaticallyWhen)\n    chatHelpers.setMessages((messages) => {\n      const lastMessage = messages.at(-1);\n      if (lastMessage?.role !== \"assistant\") return messages;\n\n      let hasChanges = false;\n      const parts = lastMessage.parts?.map((part) => {\n        if (!isToolUIPart(part)) return part;\n        if (part.state === \"output-available\" || part.state === \"output-error\")\n          return part;\n\n        hasChanges = true;\n        return {\n          ...part,\n          state: \"output-error\" as const,\n          errorText: \"User cancelled tool call by sending a new message.\",\n        };\n      });\n\n      if (!hasChanges) return messages;\n      return [...messages.slice(0, -1), { ...lastMessage, parts }];\n    });\n  };\n\n  const runtime = useExternalStoreRuntime({\n    isRunning,\n    messages,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onImport: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onExportExternalState: (): MessageFormatRepository<UI_MESSAGE> => {\n      const exported = runtimeRef.current.thread.export();\n\n      const expandedMessages: MessageFormatItem<UI_MESSAGE>[] = [];\n      const lastInnerIdMap = new Map<string, string>();\n\n      for (const item of exported.messages) {\n        const innerMessages = getExternalStoreMessages<UI_MESSAGE>(\n          item.message,\n        );\n        let parentId =\n          item.parentId != null\n            ? (lastInnerIdMap.get(item.parentId) ?? item.parentId)\n            : null;\n        for (const innerMessage of innerMessages) {\n          expandedMessages.push({ parentId, message: innerMessage });\n          parentId = aiSDKV6FormatAdapter.getId(innerMessage as UIMessage);\n        }\n        if (innerMessages.length > 0) {\n          lastInnerIdMap.set(\n            item.message.id,\n            aiSDKV6FormatAdapter.getId(\n              innerMessages[innerMessages.length - 1]! as UIMessage,\n            ),\n          );\n        }\n      }\n\n      const result: MessageFormatRepository<UI_MESSAGE> = {\n        messages: expandedMessages,\n      };\n\n      if (exported.headId != null) {\n        result.headId = lastInnerIdMap.get(exported.headId) ?? exported.headId;\n      }\n\n      return result;\n    },\n    onLoadExternalState: (repo: MessageFormatRepository<UI_MESSAGE>) => {\n      // Convert MessageFormatRepository to ExportedMessageRepository\n      const exportedRepo = toExportedMessageRepository(\n        AISDKMessageConverter.toThreadMessages,\n        repo,\n      );\n\n      // Import into the thread's MessageRepository\n      runtimeRef.current.thread.import(exportedRepo);\n    },\n    onCancel: async () => {\n      chatHelpers.stop();\n    },\n    onNew: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...current,\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      await completePendingToolCalls();\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onEdit: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...sliceMessagesUntil(current, message.parentId),\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      chatHelpers.setMessages((current) =>\n        sliceMessagesUntil(current, message.parentId),\n      );\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onReload: async (parentId: string | null, config) => {\n      lastRunConfigRef.current = config.runConfig;\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.regenerate({ metadata: config.runConfig });\n    },\n    onAddToolResult: ({\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      modelContent,\n    }) => {\n      const options = { metadata: lastRunConfigRef.current };\n      if (isError) {\n        chatHelpers.addToolOutput({\n          state: \"output-error\",\n          tool: toolName ?? toolCallId,\n          toolCallId,\n          errorText:\n            typeof result === \"string\" ? result : JSON.stringify(result),\n          options,\n        });\n      } else {\n        const output =\n          modelContent !== undefined\n            ? wrapModelContentEnvelope(result, modelContent)\n            : result;\n        chatHelpers.addToolResult({\n          tool: toolName,\n          toolCallId,\n          output,\n          options,\n        });\n      }\n    },\n    onRespondToToolApproval: ({ approvalId, approved, reason }) => {\n      void chatHelpers.addToolApprovalResponse({\n        id: approvalId,\n        approved,\n        ...(reason != null && { reason }),\n        options: { metadata: lastRunConfigRef.current },\n      });\n    },\n    ...(onResume && { onResume }),\n    ...(suggestions && { suggestions }),\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n      ...contextAdapters,\n      ...adapters,\n    },\n    isLoading,\n  });\n\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;;AA+CA,MAAM,eACJ,eACA,kBAEC;CACC,GAAG;CACH,IAAI,cAAc,MAAM,WAAW;CACnC,MAAM,cAAc,QAAQ;AAC9B;AAmCF,MAAa,mBACX,aACA,EACE,UACA,iBAAiB,uBACjB,+BAA+B,MAC/B,UACA,gBACuB,CAAC,MACvB;CACH,MAAM,kBAAkB,mBAAmB;CAC3C,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,2BAA2B,uBAC/B,IAAI,IAAI,CACV;CACA,MAAM,wBAAwB,uBAC5B,IAAI,IAAI,CACV;CACA,MAAM,yBAAyB,uBAAoC,IAAI,IAAI,CAAC;CAC5E,MAAM,mBAAmB,OAA8B,KAAA,CAAS;CAEhE,MAAM,oBAAoB,OAAO,OAAO,YAAY,EAAE,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,YACJ,YAAY,WAAW,eACvB,YAAY,WAAW,eACvB;CAEF,MAAM,gBAAgB,mBAAmB,YAAY,UAAU,SAAS;CAExE,MAAM,WAAW,sBAAsB,kBAAkB;EACvD;EACA,UAAU,YAAY;EACtB,UAAU,eACD;GACL;GACA;GACA,uBAAuB,yBAAyB;GAChD,oBAAoB,sBAAsB;GAC1C,qBAAqB,uBAAuB;GAC5C,GAAI,YAAY,SAAS,EAAE,OAAO,YAAY,MAAM,QAAQ;EAC9D,IACA;GAAC;GAAc;GAAe,YAAY;EAAK,CACjD;CACF,CAAC;CAED,MAAM,CAAC,cAAc,gBAAgB,EACnC,IAAI,UAA4B;EAC9B,OAAO;CACT,EACF,EAAE;CAEF,MAAM,YAAY,mBAChB,YACA,UAAU,WAAW,iBAAiB,SACtC,sBAAsB,kBAGtB,uBAIC,aAAa;EACZ,YAAY,YAAY,QAAQ;CAClC,CACF;CAEA,MAAM,2BAA2B,YAAY;EAC3C,IAAI,CAAC,8BAA8B;EAOnC,YAAY,aAAa,aAAa;GACpC,MAAM,cAAc,SAAS,GAAG,EAAE;GAClC,IAAI,aAAa,SAAS,aAAa,OAAO;GAE9C,IAAI,aAAa;GACjB,MAAM,QAAQ,YAAY,OAAO,KAAK,SAAS;IAC7C,IAAI,CAAC,aAAa,IAAI,GAAG,OAAO;IAChC,IAAI,KAAK,UAAU,sBAAsB,KAAK,UAAU,gBACtD,OAAO;IAET,aAAa;IACb,OAAO;KACL,GAAG;KACH,OAAO;KACP,WAAW;IACb;GACF,CAAC;GAED,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,CAAC,GAAG,SAAS,MAAM,GAAG,EAAE,GAAG;IAAE,GAAG;IAAa;GAAM,CAAC;EAC7D,CAAC;CACH;CAEA,MAAM,UAAU,wBAAwB;EACtC;EACA;EACA,gCAAgC;EAChC;EACA,cAAc,aACZ,YAAY,YACV,SACG,IAAI,mBAA+B,EACnC,OAAO,OAAO,EACd,KAAK,CACV;EACF,WAAW,aACT,YAAY,YACV,SACG,IAAI,mBAA+B,EACnC,OAAO,OAAO,EACd,KAAK,CACV;EACF,6BAAkE;GAChE,MAAM,WAAW,WAAW,QAAQ,OAAO,OAAO;GAElD,MAAM,mBAAoD,CAAC;GAC3D,MAAM,iCAAiB,IAAI,IAAoB;GAE/C,KAAK,MAAM,QAAQ,SAAS,UAAU;IACpC,MAAM,gBAAgB,yBACpB,KAAK,OACP;IACA,IAAI,WACF,KAAK,YAAY,OACZ,eAAe,IAAI,KAAK,QAAQ,KAAK,KAAK,WAC3C;IACN,KAAK,MAAM,gBAAgB,eAAe;KACxC,iBAAiB,KAAK;MAAE;MAAU,SAAS;KAAa,CAAC;KACzD,WAAW,qBAAqB,MAAM,YAAyB;IACjE;IACA,IAAI,cAAc,SAAS,GACzB,eAAe,IACb,KAAK,QAAQ,IACb,qBAAqB,MACnB,cAAc,cAAc,SAAS,EACvC,CACF;GAEJ;GAEA,MAAM,SAA8C,EAClD,UAAU,iBACZ;GAEA,IAAI,SAAS,UAAU,MACrB,OAAO,SAAS,eAAe,IAAI,SAAS,MAAM,KAAK,SAAS;GAGlE,OAAO;EACT;EACA,sBAAsB,SAA8C;GAElE,MAAM,eAAe,4BACnB,sBAAsB,kBACtB,IACF;GAGA,WAAW,QAAQ,OAAO,OAAO,YAAY;EAC/C;EACA,UAAU,YAAY;GACpB,YAAY,KAAK;EACnB;EACA,OAAO,OAAO,YAAY;GACxB,MAAM,iBACJ,yBAAyB,iBACb,OAAO;GAErB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,YAAY,aAAa,YAAY,CACnC,GAAG,SACH,YAAwB,eAAe,QAAQ,IAAI,CACrD,CAAC;IACD;GACF;GAEA,iBAAiB,UAAU,QAAQ;GACnC,MAAM,yBAAyB;GAC/B,MAAM,YAAY,YAAY,eAAe,EAC3C,UAAU,QAAQ,UACpB,CAAC;EACH;EACA,QAAQ,OAAO,YAAY;GACzB,MAAM,iBACJ,yBAAyB,iBACb,OAAO;GAErB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,YAAY,aAAa,YAAY,CACnC,GAAG,mBAAmB,SAAS,QAAQ,QAAQ,GAC/C,YAAwB,eAAe,QAAQ,IAAI,CACrD,CAAC;IACD;GACF;GAEA,iBAAiB,UAAU,QAAQ;GACnC,YAAY,aAAa,YACvB,mBAAmB,SAAS,QAAQ,QAAQ,CAC9C;GACA,MAAM,YAAY,YAAY,eAAe,EAC3C,UAAU,QAAQ,UACpB,CAAC;EACH;EACA,UAAU,OAAO,UAAyB,WAAW;GACnD,iBAAiB,UAAU,OAAO;GAClC,MAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;GACrE,YAAY,YAAY,WAAW;GAEnC,MAAM,YAAY,WAAW,EAAE,UAAU,OAAO,UAAU,CAAC;EAC7D;EACA,kBAAkB,EAChB,YACA,UACA,QACA,SACA,mBACI;GACJ,MAAM,UAAU,EAAE,UAAU,iBAAiB,QAAQ;GACrD,IAAI,SACF,YAAY,cAAc;IACxB,OAAO;IACP,MAAM,YAAY;IAClB;IACA,WACE,OAAO,WAAW,WAAW,SAAS,KAAK,UAAU,MAAM;IAC7D;GACF,CAAC;QACI;IACL,MAAM,SACJ,iBAAiB,KAAA,IACb,yBAAyB,QAAQ,YAAY,IAC7C;IACN,YAAY,cAAc;KACxB,MAAM;KACN;KACA;KACA;IACF,CAAC;GACH;EACF;EACA,0BAA0B,EAAE,YAAY,UAAU,aAAa;GAC7D,YAAiB,wBAAwB;IACvC,IAAI;IACJ;IACA,GAAI,UAAU,QAAQ,EAAE,OAAO;IAC/B,SAAS,EAAE,UAAU,iBAAiB,QAAQ;GAChD,CAAC;EACH;EACA,GAAI,YAAY,EAAE,SAAS;EAC3B,GAAI,eAAe,EAAE,YAAY;EACjC,UAAU;GACR,aAAa;GACb,GAAG;GACH,GAAG;EACL;EACA;CACF,CAAC;CAED,OAAO;AACT"}
+{"version":3,"file":"useAISDKRuntime.js","names":[],"sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useMemo, useRef, useState } from \"react\";\nimport type { UIMessage, useChat, CreateUIMessage } from \"@ai-sdk/react\";\nimport { isToolUIPart, generateId } from \"ai\";\nimport {\n  useExternalStoreRuntime,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport type {\n  ExternalStoreAdapter,\n  ExternalStoreSharedOptions,\n  ThreadHistoryAdapter,\n  AssistantRuntime,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatItem,\n  MessageFormatRepository,\n  AppendMessage,\n  RunConfig,\n  McpAppMetadata,\n} from \"@assistant-ui/core\";\nimport {\n  getExternalStoreMessages,\n  pickExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../getVercelAIMessages\";\nimport { AISDKMessageConverter } from \"../utils/convertMessage\";\nimport { wrapModelContentEnvelope } from \"../../modelContentEnvelope\";\nimport {\n  type AISDKStorageFormat,\n  aiSDKV6FormatAdapter,\n} from \"../adapters/aiSDKFormatAdapter\";\nimport {\n  useExternalHistory,\n  toExportedMessageRepository,\n} from \"./useExternalHistory\";\nimport { useStreamingTiming } from \"./useStreamingTiming\";\n\nexport type CustomToCreateMessageFunction = <\n  UI_MESSAGE extends UIMessage = UIMessage,\n>(\n  message: AppendMessage,\n) => CreateUIMessage<UI_MESSAGE>;\n\nconst toUIMessage = <UI_MESSAGE extends UIMessage>(\n  createMessage: CreateUIMessage<UI_MESSAGE>,\n  fallbackRole: UI_MESSAGE[\"role\"],\n): UI_MESSAGE =>\n  ({\n    ...createMessage,\n    id: createMessage.id ?? generateId(),\n    role: createMessage.role ?? fallbackRole,\n  }) as UI_MESSAGE;\n\nexport type AISDKRuntimeAdapter = ExternalStoreSharedOptions & {\n  adapters?:\n    | (NonNullable<ExternalStoreAdapter[\"adapters\"]> & {\n        history?: ThreadHistoryAdapter | undefined;\n      })\n    | undefined;\n  toCreateMessage?: CustomToCreateMessageFunction;\n  /**\n   * Whether to automatically cancel pending interactive tool calls when the user sends a new message.\n   *\n   * When enabled (default), the pending tool calls will be marked as failed with an error message\n   * indicating the user cancelled the tool call by sending a new message.\n   *\n   * @default true\n   */\n  cancelPendingToolCallsOnSend?: boolean | undefined;\n  /**\n   * Called when `runtime.thread.resumeRun(config)` is invoked.\n   *\n   * When omitted, `resumeRun` throws `\"Runtime does not support resuming runs.\"`.\n   * Provide this to bridge resume invocations into a custom replay channel\n   * (for example, an SSE reconnect endpoint keyed by turn id).\n   */\n  onResume?: ExternalStoreAdapter[\"onResume\"];\n};\n\nexport const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>,\n  adapter: AISDKRuntimeAdapter = {},\n) => {\n  const {\n    adapters,\n    toCreateMessage: customToCreateMessage,\n    cancelPendingToolCallsOnSend = true,\n    onResume,\n  } = adapter;\n  const contextAdapters = useRuntimeAdapters();\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const toolArgsKeyOrderCacheRef = useRef<Map<string, Map<string, string[]>>>(\n    new Map(),\n  );\n  const toolLastInputCacheRef = useRef<Map<string, ReadonlyJSONObject>>(\n    new Map(),\n  );\n  const mcpAppMetadataCacheRef = useRef<Map<string, McpAppMetadata>>(new Map());\n  const lastRunConfigRef = useRef<RunConfig | undefined>(undefined);\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const isRunning =\n    chatHelpers.status === \"submitted\" ||\n    chatHelpers.status === \"streaming\" ||\n    hasExecutingTools;\n\n  const messageTiming = useStreamingTiming(chatHelpers.messages, isRunning);\n\n  // Flag the streaming message optimistic: its id can be swapped for a server\n  // id mid-run, and the repository then drops the orphaned pre-swap id (#4037).\n  const lastMessage = chatHelpers.messages.at(-1);\n  const optimisticMessageId =\n    isRunning && lastMessage?.role === \"assistant\" ? lastMessage.id : undefined;\n\n  const messages = AISDKMessageConverter.useThreadMessages({\n    isRunning,\n    messages: chatHelpers.messages,\n    metadata: useMemo(\n      () => ({\n        toolStatuses,\n        messageTiming,\n        toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,\n        toolLastInputCache: toolLastInputCacheRef.current,\n        mcpAppMetadataCache: mcpAppMetadataCacheRef.current,\n        ...(optimisticMessageId && { optimisticMessageId }),\n        ...(chatHelpers.error && { error: chatHelpers.error.message }),\n      }),\n      [toolStatuses, messageTiming, optimisticMessageId, chatHelpers.error],\n    ),\n  });\n\n  const [runtimeRef] = useState(() => ({\n    get current(): AssistantRuntime {\n      return runtime;\n    },\n  }));\n\n  const isLoading = useExternalHistory(\n    runtimeRef,\n    adapters?.history ?? contextAdapters?.history,\n    AISDKMessageConverter.toThreadMessages as (\n      messages: UI_MESSAGE[],\n    ) => ThreadMessage[],\n    aiSDKV6FormatAdapter as MessageFormatAdapter<\n      UI_MESSAGE,\n      AISDKStorageFormat\n    >,\n    (messages) => {\n      chatHelpers.setMessages(messages);\n    },\n  );\n\n  const completePendingToolCalls = async () => {\n    if (!cancelPendingToolCallsOnSend) return;\n\n    // The runtime auto-aborts in-flight tool invocations when a new run\n    // is dispatched (append() / startRun()). All we need to do here is\n    // mark any tool without a result as cancelled in the UI message list.\n\n    // Mark any tool without a result as cancelled (uses setMessages to avoid triggering sendAutomaticallyWhen)\n    chatHelpers.setMessages((messages) => {\n      const lastMessage = messages.at(-1);\n      if (lastMessage?.role !== \"assistant\") return messages;\n\n      let hasChanges = false;\n      const parts = lastMessage.parts?.map((part) => {\n        if (!isToolUIPart(part)) return part;\n        if (part.state === \"output-available\" || part.state === \"output-error\")\n          return part;\n\n        hasChanges = true;\n        const { approval: _approval, ...rest } = part;\n        return {\n          ...rest,\n          state: \"output-error\" as const,\n          errorText: \"User cancelled tool call by sending a new message.\",\n        };\n      });\n\n      if (!hasChanges) return messages;\n      return [...messages.slice(0, -1), { ...lastMessage, parts }];\n    });\n  };\n\n  const runtime = useExternalStoreRuntime({\n    isRunning,\n    messages,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onImport: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onExportExternalState: (): MessageFormatRepository<UI_MESSAGE> => {\n      const exported = runtimeRef.current.thread.export();\n\n      const expandedMessages: MessageFormatItem<UI_MESSAGE>[] = [];\n      const lastInnerIdMap = new Map<string, string>();\n\n      for (const item of exported.messages) {\n        const innerMessages = getExternalStoreMessages<UI_MESSAGE>(\n          item.message,\n        );\n        let parentId =\n          item.parentId != null\n            ? (lastInnerIdMap.get(item.parentId) ?? item.parentId)\n            : null;\n        for (const innerMessage of innerMessages) {\n          expandedMessages.push({ parentId, message: innerMessage });\n          parentId = aiSDKV6FormatAdapter.getId(innerMessage as UIMessage);\n        }\n        if (innerMessages.length > 0) {\n          lastInnerIdMap.set(\n            item.message.id,\n            aiSDKV6FormatAdapter.getId(\n              innerMessages[innerMessages.length - 1]! as UIMessage,\n            ),\n          );\n        }\n      }\n\n      const result: MessageFormatRepository<UI_MESSAGE> = {\n        messages: expandedMessages,\n      };\n\n      if (exported.headId != null) {\n        result.headId = lastInnerIdMap.get(exported.headId) ?? exported.headId;\n      }\n\n      return result;\n    },\n    onLoadExternalState: (repo: MessageFormatRepository<UI_MESSAGE>) => {\n      // Convert MessageFormatRepository to ExportedMessageRepository\n      const exportedRepo = toExportedMessageRepository(\n        AISDKMessageConverter.toThreadMessages,\n        repo,\n      );\n\n      // Import into the thread's MessageRepository\n      runtimeRef.current.thread.import(exportedRepo);\n    },\n    onCancel: async () => {\n      chatHelpers.stop();\n    },\n    onNew: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...current,\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      await completePendingToolCalls();\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onEdit: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...sliceMessagesUntil(current, message.parentId),\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      chatHelpers.setMessages((current) =>\n        sliceMessagesUntil(current, message.parentId),\n      );\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onReload: async (parentId: string | null, config) => {\n      lastRunConfigRef.current = config.runConfig;\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.regenerate({ metadata: config.runConfig });\n    },\n    onAddToolResult: ({\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      modelContent,\n    }) => {\n      const options = { metadata: lastRunConfigRef.current };\n      if (isError) {\n        chatHelpers.addToolOutput({\n          state: \"output-error\",\n          tool: toolName ?? toolCallId,\n          toolCallId,\n          errorText:\n            typeof result === \"string\" ? result : JSON.stringify(result),\n          options,\n        });\n      } else {\n        const output =\n          modelContent !== undefined\n            ? wrapModelContentEnvelope(result, modelContent)\n            : result;\n        chatHelpers.addToolResult({\n          tool: toolName,\n          toolCallId,\n          output,\n          options,\n        });\n      }\n    },\n    onRespondToToolApproval: ({ approvalId, approved, reason }) => {\n      void chatHelpers.addToolApprovalResponse({\n        id: approvalId,\n        approved,\n        ...(reason != null && { reason }),\n        options: { metadata: lastRunConfigRef.current },\n      });\n    },\n    ...pickExternalStoreSharedOptions(adapter),\n    ...(onResume && { onResume }),\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n      ...contextAdapters,\n      ...adapters,\n    },\n    isLoading,\n  });\n\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;;AAkDA,MAAM,eACJ,eACA,kBAEC;CACC,GAAG;CACH,IAAI,cAAc,MAAM,WAAW;CACnC,MAAM,cAAc,QAAQ;AAC9B;AA4BF,MAAa,mBACX,aACA,UAA+B,CAAC,MAC7B;CACH,MAAM,EACJ,UACA,iBAAiB,uBACjB,+BAA+B,MAC/B,aACE;CACJ,MAAM,kBAAkB,mBAAmB;CAC3C,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,2BAA2B,uBAC/B,IAAI,IAAI,CACV;CACA,MAAM,wBAAwB,uBAC5B,IAAI,IAAI,CACV;CACA,MAAM,yBAAyB,uBAAoC,IAAI,IAAI,CAAC;CAC5E,MAAM,mBAAmB,OAA8B,KAAA,CAAS;CAEhE,MAAM,oBAAoB,OAAO,OAAO,YAAY,EAAE,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,YACJ,YAAY,WAAW,eACvB,YAAY,WAAW,eACvB;CAEF,MAAM,gBAAgB,mBAAmB,YAAY,UAAU,SAAS;CAIxE,MAAM,cAAc,YAAY,SAAS,GAAG,EAAE;CAC9C,MAAM,sBACJ,aAAa,aAAa,SAAS,cAAc,YAAY,KAAK,KAAA;CAEpE,MAAM,WAAW,sBAAsB,kBAAkB;EACvD;EACA,UAAU,YAAY;EACtB,UAAU,eACD;GACL;GACA;GACA,uBAAuB,yBAAyB;GAChD,oBAAoB,sBAAsB;GAC1C,qBAAqB,uBAAuB;GAC5C,GAAI,uBAAuB,EAAE,oBAAoB;GACjD,GAAI,YAAY,SAAS,EAAE,OAAO,YAAY,MAAM,QAAQ;EAC9D,IACA;GAAC;GAAc;GAAe;GAAqB,YAAY;EAAK,CACtE;CACF,CAAC;CAED,MAAM,CAAC,cAAc,gBAAgB,EACnC,IAAI,UAA4B;EAC9B,OAAO;CACT,EACF,EAAE;CAEF,MAAM,YAAY,mBAChB,YACA,UAAU,WAAW,iBAAiB,SACtC,sBAAsB,kBAGtB,uBAIC,aAAa;EACZ,YAAY,YAAY,QAAQ;CAClC,CACF;CAEA,MAAM,2BAA2B,YAAY;EAC3C,IAAI,CAAC,8BAA8B;EAOnC,YAAY,aAAa,aAAa;GACpC,MAAM,cAAc,SAAS,GAAG,EAAE;GAClC,IAAI,aAAa,SAAS,aAAa,OAAO;GAE9C,IAAI,aAAa;GACjB,MAAM,QAAQ,YAAY,OAAO,KAAK,SAAS;IAC7C,IAAI,CAAC,aAAa,IAAI,GAAG,OAAO;IAChC,IAAI,KAAK,UAAU,sBAAsB,KAAK,UAAU,gBACtD,OAAO;IAET,aAAa;IACb,MAAM,EAAE,UAAU,WAAW,GAAG,SAAS;IACzC,OAAO;KACL,GAAG;KACH,OAAO;KACP,WAAW;IACb;GACF,CAAC;GAED,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,CAAC,GAAG,SAAS,MAAM,GAAG,EAAE,GAAG;IAAE,GAAG;IAAa;GAAM,CAAC;EAC7D,CAAC;CACH;CAEA,MAAM,UAAU,wBAAwB;EACtC;EACA;EACA,gCAAgC;EAChC;EACA,cAAc,aACZ,YAAY,YACV,SACG,IAAI,mBAA+B,EACnC,OAAO,OAAO,EACd,KAAK,CACV;EACF,WAAW,aACT,YAAY,YACV,SACG,IAAI,mBAA+B,EACnC,OAAO,OAAO,EACd,KAAK,CACV;EACF,6BAAkE;GAChE,MAAM,WAAW,WAAW,QAAQ,OAAO,OAAO;GAElD,MAAM,mBAAoD,CAAC;GAC3D,MAAM,iCAAiB,IAAI,IAAoB;GAE/C,KAAK,MAAM,QAAQ,SAAS,UAAU;IACpC,MAAM,gBAAgB,yBACpB,KAAK,OACP;IACA,IAAI,WACF,KAAK,YAAY,OACZ,eAAe,IAAI,KAAK,QAAQ,KAAK,KAAK,WAC3C;IACN,KAAK,MAAM,gBAAgB,eAAe;KACxC,iBAAiB,KAAK;MAAE;MAAU,SAAS;KAAa,CAAC;KACzD,WAAW,qBAAqB,MAAM,YAAyB;IACjE;IACA,IAAI,cAAc,SAAS,GACzB,eAAe,IACb,KAAK,QAAQ,IACb,qBAAqB,MACnB,cAAc,cAAc,SAAS,EACvC,CACF;GAEJ;GAEA,MAAM,SAA8C,EAClD,UAAU,iBACZ;GAEA,IAAI,SAAS,UAAU,MACrB,OAAO,SAAS,eAAe,IAAI,SAAS,MAAM,KAAK,SAAS;GAGlE,OAAO;EACT;EACA,sBAAsB,SAA8C;GAElE,MAAM,eAAe,4BACnB,sBAAsB,kBACtB,IACF;GAGA,WAAW,QAAQ,OAAO,OAAO,YAAY;EAC/C;EACA,UAAU,YAAY;GACpB,YAAY,KAAK;EACnB;EACA,OAAO,OAAO,YAAY;GACxB,MAAM,iBACJ,yBAAyB,iBACb,OAAO;GAErB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,YAAY,aAAa,YAAY,CACnC,GAAG,SACH,YAAwB,eAAe,QAAQ,IAAI,CACrD,CAAC;IACD;GACF;GAEA,iBAAiB,UAAU,QAAQ;GACnC,MAAM,yBAAyB;GAC/B,MAAM,YAAY,YAAY,eAAe,EAC3C,UAAU,QAAQ,UACpB,CAAC;EACH;EACA,QAAQ,OAAO,YAAY;GACzB,MAAM,iBACJ,yBAAyB,iBACb,OAAO;GAErB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,YAAY,aAAa,YAAY,CACnC,GAAG,mBAAmB,SAAS,QAAQ,QAAQ,GAC/C,YAAwB,eAAe,QAAQ,IAAI,CACrD,CAAC;IACD;GACF;GAEA,iBAAiB,UAAU,QAAQ;GACnC,YAAY,aAAa,YACvB,mBAAmB,SAAS,QAAQ,QAAQ,CAC9C;GACA,MAAM,YAAY,YAAY,eAAe,EAC3C,UAAU,QAAQ,UACpB,CAAC;EACH;EACA,UAAU,OAAO,UAAyB,WAAW;GACnD,iBAAiB,UAAU,OAAO;GAClC,MAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;GACrE,YAAY,YAAY,WAAW;GAEnC,MAAM,YAAY,WAAW,EAAE,UAAU,OAAO,UAAU,CAAC;EAC7D;EACA,kBAAkB,EAChB,YACA,UACA,QACA,SACA,mBACI;GACJ,MAAM,UAAU,EAAE,UAAU,iBAAiB,QAAQ;GACrD,IAAI,SACF,YAAY,cAAc;IACxB,OAAO;IACP,MAAM,YAAY;IAClB;IACA,WACE,OAAO,WAAW,WAAW,SAAS,KAAK,UAAU,MAAM;IAC7D;GACF,CAAC;QACI;IACL,MAAM,SACJ,iBAAiB,KAAA,IACb,yBAAyB,QAAQ,YAAY,IAC7C;IACN,YAAY,cAAc;KACxB,MAAM;KACN;KACA;KACA;IACF,CAAC;GACH;EACF;EACA,0BAA0B,EAAE,YAAY,UAAU,aAAa;GAC7D,YAAiB,wBAAwB;IACvC,IAAI;IACJ;IACA,GAAI,UAAU,QAAQ,EAAE,OAAO;IAC/B,SAAS,EAAE,UAAU,iBAAiB,QAAQ;GAChD,CAAC;EACH;EACA,GAAG,+BAA+B,OAAO;EACzC,GAAI,YAAY,EAAE,SAAS;EAC3B,UAAU;GACR,aAAa;GACb,GAAG;GACH,GAAG;EACL;EACA;CACF,CAAC;CAED,OAAO;AACT"}

package/dist/ui/use-chat/useChatRuntime.d.ts

@@ -1,16 +1,15 @@
 import { AISDKRuntimeAdapter, CustomToCreateMessageFunction } from "./useAISDKRuntime.js";
 import { ChatInit } from "ai";
-import { AssistantRuntime } from "@assistant-ui/core";
+import { AssistantRuntime, ExternalStoreSharedOptions } from "@assistant-ui/core";
 import { UIMessage as UIMessage$1 } from "@ai-sdk/react";
 import { AssistantCloud } from "assistant-cloud";
 
 //#region src/ui/use-chat/useChatRuntime.d.ts
-type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage$1 = UIMessage$1> = ChatInit<UI_MESSAGE> & {
+type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage$1 = UIMessage$1> = ChatInit<UI_MESSAGE> & ExternalStoreSharedOptions & {
   cloud?: AssistantCloud | undefined;
   adapters?: AISDKRuntimeAdapter["adapters"] | undefined;
   toCreateMessage?: CustomToCreateMessageFunction;
   onResume?: AISDKRuntimeAdapter["onResume"];
-  suggestions?: AISDKRuntimeAdapter["suggestions"];
 };
 declare const useChatRuntime: <UI_MESSAGE extends UIMessage$1 = UIMessage$1>({
   cloud,

package/dist/ui/use-chat/useChatRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useChatRuntime.d.ts","names":[],"sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"mappings":";;;;;;;KAoBY,qBAAA,oBAAyC,WAAA,GAAY,WAAA,IAC/D,QAAA,CAAS,UAAA;EACP,KAAA,GAAQ,cAAA;EACR,QAAA,GAAW,mBAAA;EACX,eAAA,GAAkB,6BAAA;EAClB,QAAA,GAAW,mBAAA;EACX,WAAA,GAAc,mBAAA;AAAA;AAAA,cA0GL,cAAA,sBAAqC,WAAA,GAAY,WAAA;EAAW,KAAA;EAAA,GAAA;AAAA,IAGtE,qBAAA,CAAsB,UAAA,MAAmB,gBAAA"}
+{"version":3,"file":"useChatRuntime.d.ts","names":[],"sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"mappings":";;;;;;;KAwBY,qBAAA,oBAAyC,WAAA,GAAY,WAAA,IAC/D,QAAA,CAAS,UAAA,IACP,0BAAA;EACE,KAAA,GAAQ,cAAA;EACR,QAAA,GAAW,mBAAA;EACX,eAAA,GAAkB,6BAAA;EAClB,QAAA,GAAW,mBAAA;AAAA;AAAA,cAwGJ,cAAA,sBAAqC,WAAA,GAAY,WAAA;EAAW,KAAA;EAAA,GAAA;AAAA,IAGtE,qBAAA,CAAsB,UAAA,MAAmB,gBAAA"}

package/dist/ui/use-chat/useChatRuntime.js

@@ -3,6 +3,7 @@ import { useAISDKRuntime } from "./useAISDKRuntime.js";
 import { AssistantChatTransport } from "./AssistantChatTransport.js";
 import { useEffect, useMemo, useRef } from "react";
 import { useCloudThreadListAdapter, useRemoteThreadListRuntime } from "@assistant-ui/core/react";
+import { pickExternalStoreSharedOptions } from "@assistant-ui/core";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import { useChat } from "@ai-sdk/react";
 //#region src/ui/use-chat/useChatRuntime.ts
@@ -23,7 +24,7 @@ const getResumableAdapter = (transport) => {
 	return candidate.call(transport);
 };
 const useChatThreadRuntime = (options) => {
-	const { adapters, transport: transportOptions, toCreateMessage, onResume, suggestions, ...chatOptions } = options ?? {};
+	const { adapters, transport: transportOptions, toCreateMessage, isDisabled: _isDisabled, isSendDisabled: _isSendDisabled, unstable_capabilities: _unstable_capabilities, suggestions: _suggestions, onResume, ...chatOptions } = options ?? {};
 	const transport = useDynamicChatTransport(transportOptions ?? new AssistantChatTransport());
 	const id = useAuiState((s) => s.threadListItem.id);
 	const aui = useAui();
@@ -34,9 +35,9 @@ const useChatThreadRuntime = (options) => {
 	});
 	const runtime = useAISDKRuntime(chat, {
 		adapters,
+		...pickExternalStoreSharedOptions(options ?? {}),
 		...toCreateMessage && { toCreateMessage },
-		...onResume && { onResume },
-		...suggestions && { suggestions }
+		...onResume && { onResume }
 	});
 	if (transport instanceof AssistantChatTransport) {
 		transport.setRuntime(runtime);

package/dist/ui/use-chat/useChatRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useChatRuntime.js","names":[],"sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useChat, type UIMessage } from \"@ai-sdk/react\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport type { AssistantRuntime } from \"@assistant-ui/core\";\nimport {\n  useCloudThreadListAdapter,\n  useRemoteThreadListRuntime,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport {\n  useAISDKRuntime,\n  type AISDKRuntimeAdapter,\n  type CustomToCreateMessageFunction,\n} from \"./useAISDKRuntime\";\nimport type { ChatInit, ChatTransport } from \"ai\";\nimport { AssistantChatTransport } from \"./AssistantChatTransport\";\nimport type { AssistantChatResumableOptions } from \"../resumable\";\nimport { useEffect, useMemo, useRef } from \"react\";\n\nexport type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  ChatInit<UI_MESSAGE> & {\n    cloud?: AssistantCloud | undefined;\n    adapters?: AISDKRuntimeAdapter[\"adapters\"] | undefined;\n    toCreateMessage?: CustomToCreateMessageFunction;\n    onResume?: AISDKRuntimeAdapter[\"onResume\"];\n    suggestions?: AISDKRuntimeAdapter[\"suggestions\"];\n  };\n\nconst useDynamicChatTransport = <UI_MESSAGE extends UIMessage = UIMessage>(\n  transport: ChatTransport<UI_MESSAGE>,\n): ChatTransport<UI_MESSAGE> => {\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const transportRef = useRef<ChatTransport<UI_MESSAGE>>(transport);\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  useEffect(() => {\n    transportRef.current = transport;\n  });\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\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 useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options?: UseChatRuntimeOptions<UI_MESSAGE>,\n): AssistantRuntime => {\n  const {\n    adapters,\n    transport: transportOptions,\n    toCreateMessage,\n    onResume,\n    suggestions,\n    ...chatOptions\n  } = options ?? {};\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const transport = useDynamicChatTransport(\n    transportOptions ?? new AssistantChatTransport(),\n  );\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const id = useAuiState((s) => s.threadListItem.id);\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const aui = useAui();\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const chat = useChat({\n    ...chatOptions,\n    id,\n    transport,\n  });\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const runtime = useAISDKRuntime(chat, {\n    adapters,\n    ...(toCreateMessage && { toCreateMessage }),\n    ...(onResume && { onResume }),\n    ...(suggestions && { suggestions }),\n  });\n\n  if (transport instanceof AssistantChatTransport) {\n    transport.setRuntime(runtime);\n    transport.__internal_setGetThreadListItem(() =>\n      aui.threadListItem.source ? aui.threadListItem() : undefined,\n    );\n  }\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const resumeFiredRef = useRef(false);\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  useEffect(() => {\n    if (resumeFiredRef.current) return;\n    const adapter = getResumableAdapter(transport);\n    if (!adapter) return;\n    const pending = adapter.storage.getStreamId();\n    if (!pending) return;\n    resumeFiredRef.current = true;\n    chat.resumeStream().catch((err: unknown) => {\n      console.warn(\n        \"[assistant-ui] resumable: resume failed; clearing stored stream id\",\n        err,\n      );\n      adapter.storage.clear();\n    });\n  }, [transport, chat]);\n\n  return runtime;\n};\n\nexport const useChatRuntime = <UI_MESSAGE extends UIMessage = UIMessage>({\n  cloud,\n  ...options\n}: UseChatRuntimeOptions<UI_MESSAGE> = {}): AssistantRuntime => {\n  const cloudAdapter = useCloudThreadListAdapter({ cloud });\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n      return useChatThreadRuntime(options);\n    },\n    adapter: cloudAdapter,\n    allowNesting: true,\n  });\n};\n"],"mappings":";;;;;;;;AA6BA,MAAM,2BACJ,cAC8B;CAE9B,MAAM,eAAe,OAAkC,SAAS;CAEhE,gBAAgB;EACd,aAAa,UAAU;CACzB,CAAC;CAeD,OAbyB,cAErB,IAAI,MAAM,aAAa,SAAS,EAC9B,IAAI,GAAG,MAAM;EACX,MAAM,MACJ,aAAa,QAAQ;EACvB,OAAO,OAAO,QAAQ,aAClB,IAAI,KAAK,aAAa,OAAO,IAC7B;CACN,EACF,CAAC,GACH,CAAC,CAEmB;AACxB;AAEA,MAAM,uBACJ,cAC8C;CAC9C,IAAI,qBAAqB,wBACvB,OAAO,UAAU,oBAAoB;CAEvC,MAAM,YAAa,UAChB;CACH,IAAI,OAAO,cAAc,YAAY,OAAO,KAAA;CAC5C,OAAO,UAAU,KAAK,SAAS;AACjC;AAEA,MAAM,wBACJ,YACqB;CACrB,MAAM,EACJ,UACA,WAAW,kBACX,iBACA,UACA,aACA,GAAG,gBACD,WAAW,CAAC;CAGhB,MAAM,YAAY,wBAChB,oBAAoB,IAAI,uBAAuB,CACjD;CAGA,MAAM,KAAK,aAAa,MAAM,EAAE,eAAe,EAAE;CAEjD,MAAM,MAAM,OAAO;CAEnB,MAAM,OAAO,QAAQ;EACnB,GAAG;EACH;EACA;CACF,CAAC;CAGD,MAAM,UAAU,gBAAgB,MAAM;EACpC;EACA,GAAI,mBAAmB,EAAE,gBAAgB;EACzC,GAAI,YAAY,EAAE,SAAS;EAC3B,GAAI,eAAe,EAAE,YAAY;CACnC,CAAC;CAED,IAAI,qBAAqB,wBAAwB;EAC/C,UAAU,WAAW,OAAO;EAC5B,UAAU,sCACR,IAAI,eAAe,SAAS,IAAI,eAAe,IAAI,KAAA,CACrD;CACF;CAGA,MAAM,iBAAiB,OAAO,KAAK;CAEnC,gBAAgB;EACd,IAAI,eAAe,SAAS;EAC5B,MAAM,UAAU,oBAAoB,SAAS;EAC7C,IAAI,CAAC,SAAS;EAEd,IAAI,CADY,QAAQ,QAAQ,YACrB,GAAG;EACd,eAAe,UAAU;EACzB,KAAK,aAAa,EAAE,OAAO,QAAiB;GAC1C,QAAQ,KACN,sEACA,GACF;GACA,QAAQ,QAAQ,MAAM;EACxB,CAAC;CACH,GAAG,CAAC,WAAW,IAAI,CAAC;CAEpB,OAAO;AACT;AAEA,MAAa,kBAA4D,EACvE,OACA,GAAG,YACkC,CAAC,MAAwB;CAE9D,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAElC,OAAO,qBAAqB,OAAO;EACrC;EACA,SANmB,0BAA0B,EAAE,MAAM,CAMjC;EACpB,cAAc;CAChB,CAAC;AACH"}
+{"version":3,"file":"useChatRuntime.js","names":[],"sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useChat, type UIMessage } from \"@ai-sdk/react\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport {\n  pickExternalStoreSharedOptions,\n  type AssistantRuntime,\n  type ExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport {\n  useCloudThreadListAdapter,\n  useRemoteThreadListRuntime,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport {\n  useAISDKRuntime,\n  type AISDKRuntimeAdapter,\n  type CustomToCreateMessageFunction,\n} from \"./useAISDKRuntime\";\nimport type { ChatInit, ChatTransport } from \"ai\";\nimport { AssistantChatTransport } from \"./AssistantChatTransport\";\nimport type { AssistantChatResumableOptions } from \"../resumable\";\nimport { useEffect, useMemo, useRef } from \"react\";\n\nexport type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  ChatInit<UI_MESSAGE> &\n    ExternalStoreSharedOptions & {\n      cloud?: AssistantCloud | undefined;\n      adapters?: AISDKRuntimeAdapter[\"adapters\"] | undefined;\n      toCreateMessage?: CustomToCreateMessageFunction;\n      onResume?: AISDKRuntimeAdapter[\"onResume\"];\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 useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options?: UseChatRuntimeOptions<UI_MESSAGE>,\n): AssistantRuntime => {\n  const {\n    adapters,\n    transport: transportOptions,\n    toCreateMessage,\n    isDisabled: _isDisabled,\n    isSendDisabled: _isSendDisabled,\n    unstable_capabilities: _unstable_capabilities,\n    suggestions: _suggestions,\n    onResume,\n    ...chatOptions\n  } = options ?? {};\n  // peel guard: any shared key left in `chatOptions` collapses this to `never`\n  true satisfies keyof typeof chatOptions &\n    keyof ExternalStoreSharedOptions extends never\n    ? true\n    : never;\n\n  const transport = useDynamicChatTransport(\n    transportOptions ?? new AssistantChatTransport(),\n  );\n\n  const id = useAuiState((s) => s.threadListItem.id);\n  const aui = useAui();\n  const chat = useChat({\n    ...chatOptions,\n    id,\n    transport,\n  });\n\n  const runtime = useAISDKRuntime(chat, {\n    adapters,\n    ...pickExternalStoreSharedOptions(options ?? {}),\n    ...(toCreateMessage && { toCreateMessage }),\n    ...(onResume && { onResume }),\n  });\n\n  if (transport instanceof AssistantChatTransport) {\n    transport.setRuntime(runtime);\n    transport.__internal_setGetThreadListItem(() =>\n      aui.threadListItem.source ? aui.threadListItem() : undefined,\n    );\n  }\n\n  const resumeFiredRef = useRef(false);\n  useEffect(() => {\n    if (resumeFiredRef.current) return;\n    const adapter = getResumableAdapter(transport);\n    if (!adapter) return;\n    const pending = adapter.storage.getStreamId();\n    if (!pending) return;\n    resumeFiredRef.current = true;\n    chat.resumeStream().catch((err: unknown) => {\n      console.warn(\n        \"[assistant-ui] resumable: resume failed; clearing stored stream id\",\n        err,\n      );\n      adapter.storage.clear();\n    });\n  }, [transport, chat]);\n\n  return runtime;\n};\n\nexport const useChatRuntime = <UI_MESSAGE extends UIMessage = UIMessage>({\n  cloud,\n  ...options\n}: UseChatRuntimeOptions<UI_MESSAGE> = {}): AssistantRuntime => {\n  const cloudAdapter = useCloudThreadListAdapter({ cloud });\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useChatThreadRuntime(options);\n    },\n    adapter: cloudAdapter,\n    allowNesting: true,\n  });\n};\n"],"mappings":";;;;;;;;;AAiCA,MAAM,2BACJ,cAC8B;CAC9B,MAAM,eAAe,OAAkC,SAAS;CAChE,gBAAgB;EACd,aAAa,UAAU;CACzB,CAAC;CAcD,OAbyB,cAErB,IAAI,MAAM,aAAa,SAAS,EAC9B,IAAI,GAAG,MAAM;EACX,MAAM,MACJ,aAAa,QAAQ;EACvB,OAAO,OAAO,QAAQ,aAClB,IAAI,KAAK,aAAa,OAAO,IAC7B;CACN,EACF,CAAC,GACH,CAAC,CAEmB;AACxB;AAEA,MAAM,uBACJ,cAC8C;CAC9C,IAAI,qBAAqB,wBACvB,OAAO,UAAU,oBAAoB;CAEvC,MAAM,YAAa,UAChB;CACH,IAAI,OAAO,cAAc,YAAY,OAAO,KAAA;CAC5C,OAAO,UAAU,KAAK,SAAS;AACjC;AAEA,MAAM,wBACJ,YACqB;CACrB,MAAM,EACJ,UACA,WAAW,kBACX,iBACA,YAAY,aACZ,gBAAgB,iBAChB,uBAAuB,wBACvB,aAAa,cACb,UACA,GAAG,gBACD,WAAW,CAAC;CAOhB,MAAM,YAAY,wBAChB,oBAAoB,IAAI,uBAAuB,CACjD;CAEA,MAAM,KAAK,aAAa,MAAM,EAAE,eAAe,EAAE;CACjD,MAAM,MAAM,OAAO;CACnB,MAAM,OAAO,QAAQ;EACnB,GAAG;EACH;EACA;CACF,CAAC;CAED,MAAM,UAAU,gBAAgB,MAAM;EACpC;EACA,GAAG,+BAA+B,WAAW,CAAC,CAAC;EAC/C,GAAI,mBAAmB,EAAE,gBAAgB;EACzC,GAAI,YAAY,EAAE,SAAS;CAC7B,CAAC;CAED,IAAI,qBAAqB,wBAAwB;EAC/C,UAAU,WAAW,OAAO;EAC5B,UAAU,sCACR,IAAI,eAAe,SAAS,IAAI,eAAe,IAAI,KAAA,CACrD;CACF;CAEA,MAAM,iBAAiB,OAAO,KAAK;CACnC,gBAAgB;EACd,IAAI,eAAe,SAAS;EAC5B,MAAM,UAAU,oBAAoB,SAAS;EAC7C,IAAI,CAAC,SAAS;EAEd,IAAI,CADY,QAAQ,QAAQ,YACrB,GAAG;EACd,eAAe,UAAU;EACzB,KAAK,aAAa,EAAE,OAAO,QAAiB;GAC1C,QAAQ,KACN,sEACA,GACF;GACA,QAAQ,QAAQ,MAAM;EACxB,CAAC;CACH,GAAG,CAAC,WAAW,IAAI,CAAC;CAEpB,OAAO;AACT;AAEA,MAAa,kBAA4D,EACvE,OACA,GAAG,YACkC,CAAC,MAAwB;CAE9D,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAClC,OAAO,qBAAqB,OAAO;EACrC;EACA,SALmB,0BAA0B,EAAE,MAAM,CAKjC;EACpB,cAAc;CAChB,CAAC;AACH"}

package/dist/ui/use-chat/useExternalHistory.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useExternalHistory.d.ts","names":[],"sources":["../../../src/ui/use-chat/useExternalHistory.ts"],"mappings":";;;;cAsBa,2BAAA,aACX,gBAAA,GAAmB,QAAA,EAAU,QAAA,OAAe,aAAA,IAC5C,QAAA,EAAU,uBAAA,CAAwB,QAAA,MACjC,yBAAA;AAAA,cAaU,kBAAA,aACX,UAAA,EAAY,SAAA,CAAU,gBAAA,GACtB,cAAA,EAAgB,oBAAA,cAChB,gBAAA,GAAmB,QAAA,EAAU,QAAA,OAAe,aAAA,IAC5C,oBAAA,EAAsB,oBAAA,CAAqB,QAAA,QAC3C,aAAA,GAAgB,QAAA,EAAU,QAAA"}
+{"version":3,"file":"useExternalHistory.d.ts","names":[],"sources":["../../../src/ui/use-chat/useExternalHistory.ts"],"mappings":";;;;cAsBa,2BAAA,aACX,gBAAA,GAAmB,QAAA,EAAU,QAAA,OAAe,aAAA,IAC5C,QAAA,EAAU,uBAAA,CAAwB,QAAA,MACjC,yBAAA;AAAA,cAsBU,kBAAA,aACX,UAAA,EAAY,SAAA,CAAU,gBAAA,GACtB,cAAA,EAAgB,oBAAA,cAChB,gBAAA,GAAmB,QAAA,EAAU,QAAA,OAAe,aAAA,IAC5C,oBAAA,EAAsB,oBAAA,CAAqB,QAAA,QAC3C,aAAA,GAAgB,QAAA,EAAU,QAAA"}

package/dist/ui/use-chat/useExternalHistory.js

@@ -5,15 +5,23 @@ import { MessageRepository } from "@assistant-ui/core/internal";
 import { useAui } from "@assistant-ui/store";
 //#region src/ui/use-chat/useExternalHistory.ts
 const toExportedMessageRepository = (toThreadMessages, messages) => {
+	const survivingIds = /* @__PURE__ */ new Set();
+	const survivors = messages.messages.flatMap((m) => {
+		const message = toThreadMessages([m.message])[0];
+		if (!message) {
+			console.warn("Skipping a stored message that could not be loaded.");
+			return [];
+		}
+		if (m.parentId && !survivingIds.has(m.parentId)) return [];
+		survivingIds.add(message.id);
+		return [{
+			...m,
+			message
+		}];
+	});
 	return {
-		headId: messages.headId,
-		messages: messages.messages.map((m) => {
-			const message = toThreadMessages([m.message])[0];
-			return {
-				...m,
-				message
-			};
-		})
+		headId: messages.headId && survivingIds.has(messages.headId) ? messages.headId : null,
+		messages: survivors
 	};
 };
 const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storageFormatAdapter, onSetMessages) => {

package/dist/ui/use-chat/useExternalHistory.js.map

@@ -1 +1 @@
-{"version":3,"file":"useExternalHistory.js","names":[],"sources":["../../../src/ui/use-chat/useExternalHistory.ts"],"sourcesContent":["\"use client\";\n\nimport type {\n  AssistantRuntime,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatRepository,\n  ExportedMessageRepository,\n} from \"@assistant-ui/core\";\nimport { getExternalStoreMessages } from \"@assistant-ui/core\";\nimport { MessageRepository } from \"@assistant-ui/core/internal\";\nimport { useAui } from \"@assistant-ui/store\";\nimport {\n  useRef,\n  useEffect,\n  useState,\n  type RefObject,\n  useCallback,\n  useMemo,\n} from \"react\";\n\nexport const toExportedMessageRepository = <TMessage>(\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  messages: MessageFormatRepository<TMessage>,\n): ExportedMessageRepository => {\n  return {\n    headId: messages.headId!,\n    messages: messages.messages.map((m) => {\n      const message = toThreadMessages([m.message])[0]!;\n      return {\n        ...m,\n        message,\n      };\n    }),\n  };\n};\n\nexport const useExternalHistory = <TMessage>(\n  runtimeRef: RefObject<AssistantRuntime>,\n  historyAdapter: ThreadHistoryAdapter | undefined,\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  storageFormatAdapter: MessageFormatAdapter<TMessage, any>,\n  onSetMessages: (messages: TMessage[]) => void,\n) => {\n  const loadedRef = useRef(false);\n\n  const aui = useAui();\n  const optionalThreadListItem = useCallback(\n    () => (aui.threadListItem.source ? aui.threadListItem() : null),\n    [aui],\n  );\n\n  const [isLoading, setIsLoading] = useState(false);\n\n  const historyIds = useRef(new Set<string>());\n\n  const onSetMessagesRef = useRef(onSetMessages);\n  useEffect(() => {\n    onSetMessagesRef.current = onSetMessages;\n  });\n\n  const formatAdapter = useMemo(() => {\n    if (!historyAdapter) return undefined;\n    if (!historyAdapter.withFormat) {\n      throw new Error(\n        \"useAISDKRuntime: ThreadHistoryAdapter is missing the required `withFormat` method.\",\n      );\n    }\n    return historyAdapter.withFormat<TMessage, any>(storageFormatAdapter);\n  }, [historyAdapter, storageFormatAdapter]);\n\n  useEffect(() => {\n    if (!formatAdapter || loadedRef.current) return;\n\n    const loadHistory = async () => {\n      setIsLoading(true);\n      try {\n        const repo = await formatAdapter.load();\n        if (repo && repo.messages.length > 0) {\n          const converted = toExportedMessageRepository(toThreadMessages, repo);\n          runtimeRef.current.thread.import(converted);\n\n          const tempRepo = new MessageRepository();\n          tempRepo.import(converted);\n          const messages = tempRepo.getMessages();\n\n          onSetMessagesRef.current(\n            messages.flatMap(getExternalStoreMessages<TMessage>),\n          );\n\n          historyIds.current = new Set(\n            converted.messages.map((m) => m.message.id),\n          );\n        }\n      } catch (error) {\n        console.error(\"Failed to load message history:\", error);\n      } finally {\n        setIsLoading(false);\n      }\n    };\n\n    loadedRef.current = true;\n\n    if (!optionalThreadListItem()?.getState().remoteId) {\n      setIsLoading(false);\n      return;\n    }\n\n    loadHistory();\n  }, [formatAdapter, toThreadMessages, runtimeRef, optionalThreadListItem]);\n\n  const runStartRef = useRef<number | null>(null);\n  const persistTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);\n  const stepBoundariesRef = useRef<number[]>([]);\n  const wasRunningRef = useRef(false);\n  const toolCallCountRef = useRef(0);\n\n  useEffect(() => {\n    if (!formatAdapter) return;\n\n    const unsubscribe = runtimeRef.current.thread.subscribe(() => {\n      const { isRunning } = runtimeRef.current.thread.getState();\n      const wasRunning = wasRunningRef.current;\n      wasRunningRef.current = isRunning;\n\n      // Track step boundaries by content changes (more reliable than isRunning)\n      if (runStartRef.current != null) {\n        const lastMsg = runtimeRef.current.thread.getState().messages.at(-1);\n        if (lastMsg?.role === \"assistant\") {\n          const currentToolCallCount = lastMsg.content.filter(\n            (p) => p.type === \"tool-call\",\n          ).length;\n          while (toolCallCountRef.current < currentToolCallCount) {\n            stepBoundariesRef.current.push(Date.now() - runStartRef.current);\n            toolCallCountRef.current++;\n          }\n        }\n      }\n\n      if (isRunning) {\n        if (runStartRef.current == null) {\n          runStartRef.current = Date.now();\n          stepBoundariesRef.current = [];\n          toolCallCountRef.current = 0;\n        }\n        // Cancel any pending persist — isRunning went back to true\n        if (persistTimerRef.current) {\n          clearTimeout(persistTimerRef.current);\n          persistTimerRef.current = null;\n        }\n        return;\n      }\n\n      // Only act on the true→false transition\n      if (!wasRunning) return;\n\n      // Record step boundary offset (synchronous for accuracy)\n      if (runStartRef.current != null) {\n        stepBoundariesRef.current.push(Date.now() - runStartRef.current);\n      }\n\n      // Debounce: wait one macrotask so agentic step flickers are absorbed\n      if (persistTimerRef.current) clearTimeout(persistTimerRef.current);\n      persistTimerRef.current = setTimeout(async () => {\n        persistTimerRef.current = null;\n\n        // Re-read latest state — may have changed since the timeout was scheduled\n        const latest = runtimeRef.current.thread.getState();\n        if (latest.isRunning) return; // was just a flicker\n\n        // Derive durationMs from the last boundary (covers all steps)\n        const boundaries = stepBoundariesRef.current;\n        const durationMs =\n          boundaries.length > 0 ? boundaries.at(-1) : undefined;\n\n        // Fallback: if only 1 boundary but message has multiple steps, distribute evenly\n        if (boundaries.length === 1 && durationMs != null) {\n          const lastAssistant = latest.messages.findLast(\n            (m) => m.role === \"assistant\",\n          );\n          if (lastAssistant) {\n            const tcCount = lastAssistant.content.filter(\n              (p) => p.type === \"tool-call\",\n            ).length;\n            if (tcCount > 0) {\n              const totalSteps = tcCount + 1;\n              const stepDur = durationMs / totalSteps;\n              boundaries.length = 0;\n              for (let i = 0; i < totalSteps; i++) {\n                boundaries.push(Math.round((i + 1) * stepDur));\n              }\n            }\n          }\n        }\n\n        // Build per-step timestamps when there are multiple steps\n        const stepTimestamps =\n          boundaries.length > 1\n            ? boundaries.map((endMs, i) => ({\n                start_ms: i === 0 ? 0 : boundaries[i - 1]!,\n                end_ms: endMs,\n              }))\n            : undefined;\n\n        runStartRef.current = null;\n        stepBoundariesRef.current = [];\n\n        const telemetryOptions = {\n          ...(durationMs != null ? { durationMs } : undefined),\n          ...(stepTimestamps != null ? { stepTimestamps } : undefined),\n        };\n\n        const { messages } = latest;\n        let lastInnerMessageId: string | null = null;\n\n        const getLastInnerId = (msgs: TMessage[]): string | null =>\n          msgs.length > 0 ? storageFormatAdapter.getId(msgs.at(-1)!) : null;\n\n        const toBatchItems = (msgs: TMessage[]) =>\n          msgs.map((msg, idx) => ({\n            parentId:\n              idx === 0\n                ? lastInnerMessageId\n                : storageFormatAdapter.getId(msgs[idx - 1]!),\n            message: msg,\n          }));\n\n        for (const message of messages) {\n          const innerMessages = getExternalStoreMessages<TMessage>(message);\n\n          const isReady =\n            message.status === undefined ||\n            message.status.type === \"complete\" ||\n            message.status.type === \"incomplete\";\n\n          if (!isReady) {\n            lastInnerMessageId =\n              getLastInnerId(innerMessages) ?? lastInnerMessageId;\n            continue;\n          }\n\n          if (historyIds.current.has(message.id)) {\n            if (durationMs !== undefined) {\n              let parentId = lastInnerMessageId;\n              for (const innerMessage of innerMessages) {\n                try {\n                  await formatAdapter.update?.(\n                    { parentId, message: innerMessage },\n                    storageFormatAdapter.getId(innerMessage),\n                  );\n                } catch {\n                  // ignore update failures to avoid breaking the message processing loop\n                }\n                parentId = storageFormatAdapter.getId(innerMessage);\n              }\n            }\n            lastInnerMessageId =\n              getLastInnerId(innerMessages) ?? lastInnerMessageId;\n            continue;\n          }\n          historyIds.current.add(message.id);\n\n          const batchItems = toBatchItems(innerMessages);\n          for (const item of batchItems) {\n            await formatAdapter.append(item);\n          }\n\n          lastInnerMessageId =\n            getLastInnerId(innerMessages) ?? lastInnerMessageId;\n\n          formatAdapter.reportTelemetry?.(batchItems, telemetryOptions);\n        }\n      }, 0);\n    });\n\n    return () => {\n      unsubscribe();\n      if (persistTimerRef.current) {\n        clearTimeout(persistTimerRef.current);\n        persistTimerRef.current = null;\n      }\n    };\n  }, [formatAdapter, storageFormatAdapter, runtimeRef]);\n\n  return isLoading;\n};\n"],"mappings":";;;;;;AAsBA,MAAa,+BACX,kBACA,aAC8B;CAC9B,OAAO;EACL,QAAQ,SAAS;EACjB,UAAU,SAAS,SAAS,KAAK,MAAM;GACrC,MAAM,UAAU,iBAAiB,CAAC,EAAE,OAAO,CAAC,EAAE;GAC9C,OAAO;IACL,GAAG;IACH;GACF;EACF,CAAC;CACH;AACF;AAEA,MAAa,sBACX,YACA,gBACA,kBACA,sBACA,kBACG;CACH,MAAM,YAAY,OAAO,KAAK;CAE9B,MAAM,MAAM,OAAO;CACnB,MAAM,yBAAyB,kBACtB,IAAI,eAAe,SAAS,IAAI,eAAe,IAAI,MAC1D,CAAC,GAAG,CACN;CAEA,MAAM,CAAC,WAAW,gBAAgB,SAAS,KAAK;CAEhD,MAAM,aAAa,uBAAO,IAAI,IAAY,CAAC;CAE3C,MAAM,mBAAmB,OAAO,aAAa;CAC7C,gBAAgB;EACd,iBAAiB,UAAU;CAC7B,CAAC;CAED,MAAM,gBAAgB,cAAc;EAClC,IAAI,CAAC,gBAAgB,OAAO,KAAA;EAC5B,IAAI,CAAC,eAAe,YAClB,MAAM,IAAI,MACR,oFACF;EAEF,OAAO,eAAe,WAA0B,oBAAoB;CACtE,GAAG,CAAC,gBAAgB,oBAAoB,CAAC;CAEzC,gBAAgB;EACd,IAAI,CAAC,iBAAiB,UAAU,SAAS;EAEzC,MAAM,cAAc,YAAY;GAC9B,aAAa,IAAI;GACjB,IAAI;IACF,MAAM,OAAO,MAAM,cAAc,KAAK;IACtC,IAAI,QAAQ,KAAK,SAAS,SAAS,GAAG;KACpC,MAAM,YAAY,4BAA4B,kBAAkB,IAAI;KACpE,WAAW,QAAQ,OAAO,OAAO,SAAS;KAE1C,MAAM,WAAW,IAAI,kBAAkB;KACvC,SAAS,OAAO,SAAS;KACzB,MAAM,WAAW,SAAS,YAAY;KAEtC,iBAAiB,QACf,SAAS,QAAQ,wBAAkC,CACrD;KAEA,WAAW,UAAU,IAAI,IACvB,UAAU,SAAS,KAAK,MAAM,EAAE,QAAQ,EAAE,CAC5C;IACF;GACF,SAAS,OAAO;IACd,QAAQ,MAAM,mCAAmC,KAAK;GACxD,UAAU;IACR,aAAa,KAAK;GACpB;EACF;EAEA,UAAU,UAAU;EAEpB,IAAI,CAAC,uBAAuB,GAAG,SAAS,EAAE,UAAU;GAClD,aAAa,KAAK;GAClB;EACF;EAEA,YAAY;CACd,GAAG;EAAC;EAAe;EAAkB;EAAY;CAAsB,CAAC;CAExE,MAAM,cAAc,OAAsB,IAAI;CAC9C,MAAM,kBAAkB,OAA6C,IAAI;CACzE,MAAM,oBAAoB,OAAiB,CAAC,CAAC;CAC7C,MAAM,gBAAgB,OAAO,KAAK;CAClC,MAAM,mBAAmB,OAAO,CAAC;CAEjC,gBAAgB;EACd,IAAI,CAAC,eAAe;EAEpB,MAAM,cAAc,WAAW,QAAQ,OAAO,gBAAgB;GAC5D,MAAM,EAAE,cAAc,WAAW,QAAQ,OAAO,SAAS;GACzD,MAAM,aAAa,cAAc;GACjC,cAAc,UAAU;GAGxB,IAAI,YAAY,WAAW,MAAM;IAC/B,MAAM,UAAU,WAAW,QAAQ,OAAO,SAAS,EAAE,SAAS,GAAG,EAAE;IACnE,IAAI,SAAS,SAAS,aAAa;KACjC,MAAM,uBAAuB,QAAQ,QAAQ,QAC1C,MAAM,EAAE,SAAS,WACpB,EAAE;KACF,OAAO,iBAAiB,UAAU,sBAAsB;MACtD,kBAAkB,QAAQ,KAAK,KAAK,IAAI,IAAI,YAAY,OAAO;MAC/D,iBAAiB;KACnB;IACF;GACF;GAEA,IAAI,WAAW;IACb,IAAI,YAAY,WAAW,MAAM;KAC/B,YAAY,UAAU,KAAK,IAAI;KAC/B,kBAAkB,UAAU,CAAC;KAC7B,iBAAiB,UAAU;IAC7B;IAEA,IAAI,gBAAgB,SAAS;KAC3B,aAAa,gBAAgB,OAAO;KACpC,gBAAgB,UAAU;IAC5B;IACA;GACF;GAGA,IAAI,CAAC,YAAY;GAGjB,IAAI,YAAY,WAAW,MACzB,kBAAkB,QAAQ,KAAK,KAAK,IAAI,IAAI,YAAY,OAAO;GAIjE,IAAI,gBAAgB,SAAS,aAAa,gBAAgB,OAAO;GACjE,gBAAgB,UAAU,WAAW,YAAY;IAC/C,gBAAgB,UAAU;IAG1B,MAAM,SAAS,WAAW,QAAQ,OAAO,SAAS;IAClD,IAAI,OAAO,WAAW;IAGtB,MAAM,aAAa,kBAAkB;IACrC,MAAM,aACJ,WAAW,SAAS,IAAI,WAAW,GAAG,EAAE,IAAI,KAAA;IAG9C,IAAI,WAAW,WAAW,KAAK,cAAc,MAAM;KACjD,MAAM,gBAAgB,OAAO,SAAS,UACnC,MAAM,EAAE,SAAS,WACpB;KACA,IAAI,eAAe;MACjB,MAAM,UAAU,cAAc,QAAQ,QACnC,MAAM,EAAE,SAAS,WACpB,EAAE;MACF,IAAI,UAAU,GAAG;OACf,MAAM,aAAa,UAAU;OAC7B,MAAM,UAAU,aAAa;OAC7B,WAAW,SAAS;OACpB,KAAK,IAAI,IAAI,GAAG,IAAI,YAAY,KAC9B,WAAW,KAAK,KAAK,OAAO,IAAI,KAAK,OAAO,CAAC;MAEjD;KACF;IACF;IAGA,MAAM,iBACJ,WAAW,SAAS,IAChB,WAAW,KAAK,OAAO,OAAO;KAC5B,UAAU,MAAM,IAAI,IAAI,WAAW,IAAI;KACvC,QAAQ;IACV,EAAE,IACF,KAAA;IAEN,YAAY,UAAU;IACtB,kBAAkB,UAAU,CAAC;IAE7B,MAAM,mBAAmB;KACvB,GAAI,cAAc,OAAO,EAAE,WAAW,IAAI,KAAA;KAC1C,GAAI,kBAAkB,OAAO,EAAE,eAAe,IAAI,KAAA;IACpD;IAEA,MAAM,EAAE,aAAa;IACrB,IAAI,qBAAoC;IAExC,MAAM,kBAAkB,SACtB,KAAK,SAAS,IAAI,qBAAqB,MAAM,KAAK,GAAG,EAAE,CAAE,IAAI;IAE/D,MAAM,gBAAgB,SACpB,KAAK,KAAK,KAAK,SAAS;KACtB,UACE,QAAQ,IACJ,qBACA,qBAAqB,MAAM,KAAK,MAAM,EAAG;KAC/C,SAAS;IACX,EAAE;IAEJ,KAAK,MAAM,WAAW,UAAU;KAC9B,MAAM,gBAAgB,yBAAmC,OAAO;KAOhE,IAAI,EAJF,QAAQ,WAAW,KAAA,KACnB,QAAQ,OAAO,SAAS,cACxB,QAAQ,OAAO,SAAS,eAEZ;MACZ,qBACE,eAAe,aAAa,KAAK;MACnC;KACF;KAEA,IAAI,WAAW,QAAQ,IAAI,QAAQ,EAAE,GAAG;MACtC,IAAI,eAAe,KAAA,GAAW;OAC5B,IAAI,WAAW;OACf,KAAK,MAAM,gBAAgB,eAAe;QACxC,IAAI;SACF,MAAM,cAAc,SAClB;UAAE;UAAU,SAAS;SAAa,GAClC,qBAAqB,MAAM,YAAY,CACzC;QACF,QAAQ,CAER;QACA,WAAW,qBAAqB,MAAM,YAAY;OACpD;MACF;MACA,qBACE,eAAe,aAAa,KAAK;MACnC;KACF;KACA,WAAW,QAAQ,IAAI,QAAQ,EAAE;KAEjC,MAAM,aAAa,aAAa,aAAa;KAC7C,KAAK,MAAM,QAAQ,YACjB,MAAM,cAAc,OAAO,IAAI;KAGjC,qBACE,eAAe,aAAa,KAAK;KAEnC,cAAc,kBAAkB,YAAY,gBAAgB;IAC9D;GACF,GAAG,CAAC;EACN,CAAC;EAED,aAAa;GACX,YAAY;GACZ,IAAI,gBAAgB,SAAS;IAC3B,aAAa,gBAAgB,OAAO;IACpC,gBAAgB,UAAU;GAC5B;EACF;CACF,GAAG;EAAC;EAAe;EAAsB;CAAU,CAAC;CAEpD,OAAO;AACT"}
+{"version":3,"file":"useExternalHistory.js","names":[],"sources":["../../../src/ui/use-chat/useExternalHistory.ts"],"sourcesContent":["\"use client\";\n\nimport type {\n  AssistantRuntime,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatRepository,\n  ExportedMessageRepository,\n} from \"@assistant-ui/core\";\nimport { getExternalStoreMessages } from \"@assistant-ui/core\";\nimport { MessageRepository } from \"@assistant-ui/core/internal\";\nimport { useAui } from \"@assistant-ui/store\";\nimport {\n  useRef,\n  useEffect,\n  useState,\n  type RefObject,\n  useCallback,\n  useMemo,\n} from \"react\";\n\nexport const toExportedMessageRepository = <TMessage>(\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  messages: MessageFormatRepository<TMessage>,\n): ExportedMessageRepository => {\n  const survivingIds = new Set<string>();\n  const survivors = messages.messages.flatMap((m) => {\n    const message = toThreadMessages([m.message])[0];\n    if (!message) {\n      console.warn(\"Skipping a stored message that could not be loaded.\");\n      return [];\n    }\n    if (m.parentId && !survivingIds.has(m.parentId)) return [];\n    survivingIds.add(message.id);\n    return [{ ...m, message }];\n  });\n\n  return {\n    headId:\n      messages.headId && survivingIds.has(messages.headId)\n        ? messages.headId\n        : null,\n    messages: survivors,\n  };\n};\n\nexport const useExternalHistory = <TMessage>(\n  runtimeRef: RefObject<AssistantRuntime>,\n  historyAdapter: ThreadHistoryAdapter | undefined,\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  storageFormatAdapter: MessageFormatAdapter<TMessage, any>,\n  onSetMessages: (messages: TMessage[]) => void,\n) => {\n  const loadedRef = useRef(false);\n\n  const aui = useAui();\n  const optionalThreadListItem = useCallback(\n    () => (aui.threadListItem.source ? aui.threadListItem() : null),\n    [aui],\n  );\n\n  const [isLoading, setIsLoading] = useState(false);\n\n  const historyIds = useRef(new Set<string>());\n\n  const onSetMessagesRef = useRef(onSetMessages);\n  useEffect(() => {\n    onSetMessagesRef.current = onSetMessages;\n  });\n\n  const formatAdapter = useMemo(() => {\n    if (!historyAdapter) return undefined;\n    if (!historyAdapter.withFormat) {\n      throw new Error(\n        \"useAISDKRuntime: ThreadHistoryAdapter is missing the required `withFormat` method.\",\n      );\n    }\n    return historyAdapter.withFormat<TMessage, any>(storageFormatAdapter);\n  }, [historyAdapter, storageFormatAdapter]);\n\n  useEffect(() => {\n    if (!formatAdapter || loadedRef.current) return;\n\n    const loadHistory = async () => {\n      setIsLoading(true);\n      try {\n        const repo = await formatAdapter.load();\n        if (repo && repo.messages.length > 0) {\n          const converted = toExportedMessageRepository(toThreadMessages, repo);\n          runtimeRef.current.thread.import(converted);\n\n          const tempRepo = new MessageRepository();\n          tempRepo.import(converted);\n          const messages = tempRepo.getMessages();\n\n          onSetMessagesRef.current(\n            messages.flatMap(getExternalStoreMessages<TMessage>),\n          );\n\n          historyIds.current = new Set(\n            converted.messages.map((m) => m.message.id),\n          );\n        }\n      } catch (error) {\n        console.error(\"Failed to load message history:\", error);\n      } finally {\n        setIsLoading(false);\n      }\n    };\n\n    loadedRef.current = true;\n\n    if (!optionalThreadListItem()?.getState().remoteId) {\n      setIsLoading(false);\n      return;\n    }\n\n    loadHistory();\n  }, [formatAdapter, toThreadMessages, runtimeRef, optionalThreadListItem]);\n\n  const runStartRef = useRef<number | null>(null);\n  const persistTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);\n  const stepBoundariesRef = useRef<number[]>([]);\n  const wasRunningRef = useRef(false);\n  const toolCallCountRef = useRef(0);\n\n  useEffect(() => {\n    if (!formatAdapter) return;\n\n    const unsubscribe = runtimeRef.current.thread.subscribe(() => {\n      const { isRunning } = runtimeRef.current.thread.getState();\n      const wasRunning = wasRunningRef.current;\n      wasRunningRef.current = isRunning;\n\n      // Track step boundaries by content changes (more reliable than isRunning)\n      if (runStartRef.current != null) {\n        const lastMsg = runtimeRef.current.thread.getState().messages.at(-1);\n        if (lastMsg?.role === \"assistant\") {\n          const currentToolCallCount = lastMsg.content.filter(\n            (p) => p.type === \"tool-call\",\n          ).length;\n          while (toolCallCountRef.current < currentToolCallCount) {\n            stepBoundariesRef.current.push(Date.now() - runStartRef.current);\n            toolCallCountRef.current++;\n          }\n        }\n      }\n\n      if (isRunning) {\n        if (runStartRef.current == null) {\n          runStartRef.current = Date.now();\n          stepBoundariesRef.current = [];\n          toolCallCountRef.current = 0;\n        }\n        // Cancel any pending persist — isRunning went back to true\n        if (persistTimerRef.current) {\n          clearTimeout(persistTimerRef.current);\n          persistTimerRef.current = null;\n        }\n        return;\n      }\n\n      // Only act on the true→false transition\n      if (!wasRunning) return;\n\n      // Record step boundary offset (synchronous for accuracy)\n      if (runStartRef.current != null) {\n        stepBoundariesRef.current.push(Date.now() - runStartRef.current);\n      }\n\n      // Debounce: wait one macrotask so agentic step flickers are absorbed\n      if (persistTimerRef.current) clearTimeout(persistTimerRef.current);\n      persistTimerRef.current = setTimeout(async () => {\n        persistTimerRef.current = null;\n\n        // Re-read latest state — may have changed since the timeout was scheduled\n        const latest = runtimeRef.current.thread.getState();\n        if (latest.isRunning) return; // was just a flicker\n\n        // Derive durationMs from the last boundary (covers all steps)\n        const boundaries = stepBoundariesRef.current;\n        const durationMs =\n          boundaries.length > 0 ? boundaries.at(-1) : undefined;\n\n        // Fallback: if only 1 boundary but message has multiple steps, distribute evenly\n        if (boundaries.length === 1 && durationMs != null) {\n          const lastAssistant = latest.messages.findLast(\n            (m) => m.role === \"assistant\",\n          );\n          if (lastAssistant) {\n            const tcCount = lastAssistant.content.filter(\n              (p) => p.type === \"tool-call\",\n            ).length;\n            if (tcCount > 0) {\n              const totalSteps = tcCount + 1;\n              const stepDur = durationMs / totalSteps;\n              boundaries.length = 0;\n              for (let i = 0; i < totalSteps; i++) {\n                boundaries.push(Math.round((i + 1) * stepDur));\n              }\n            }\n          }\n        }\n\n        // Build per-step timestamps when there are multiple steps\n        const stepTimestamps =\n          boundaries.length > 1\n            ? boundaries.map((endMs, i) => ({\n                start_ms: i === 0 ? 0 : boundaries[i - 1]!,\n                end_ms: endMs,\n              }))\n            : undefined;\n\n        runStartRef.current = null;\n        stepBoundariesRef.current = [];\n\n        const telemetryOptions = {\n          ...(durationMs != null ? { durationMs } : undefined),\n          ...(stepTimestamps != null ? { stepTimestamps } : undefined),\n        };\n\n        const { messages } = latest;\n        let lastInnerMessageId: string | null = null;\n\n        const getLastInnerId = (msgs: TMessage[]): string | null =>\n          msgs.length > 0 ? storageFormatAdapter.getId(msgs.at(-1)!) : null;\n\n        const toBatchItems = (msgs: TMessage[]) =>\n          msgs.map((msg, idx) => ({\n            parentId:\n              idx === 0\n                ? lastInnerMessageId\n                : storageFormatAdapter.getId(msgs[idx - 1]!),\n            message: msg,\n          }));\n\n        for (const message of messages) {\n          const innerMessages = getExternalStoreMessages<TMessage>(message);\n\n          const isReady =\n            message.status === undefined ||\n            message.status.type === \"complete\" ||\n            message.status.type === \"incomplete\";\n\n          if (!isReady) {\n            lastInnerMessageId =\n              getLastInnerId(innerMessages) ?? lastInnerMessageId;\n            continue;\n          }\n\n          if (historyIds.current.has(message.id)) {\n            if (durationMs !== undefined) {\n              let parentId = lastInnerMessageId;\n              for (const innerMessage of innerMessages) {\n                try {\n                  await formatAdapter.update?.(\n                    { parentId, message: innerMessage },\n                    storageFormatAdapter.getId(innerMessage),\n                  );\n                } catch {\n                  // ignore update failures to avoid breaking the message processing loop\n                }\n                parentId = storageFormatAdapter.getId(innerMessage);\n              }\n            }\n            lastInnerMessageId =\n              getLastInnerId(innerMessages) ?? lastInnerMessageId;\n            continue;\n          }\n          historyIds.current.add(message.id);\n\n          const batchItems = toBatchItems(innerMessages);\n          for (const item of batchItems) {\n            await formatAdapter.append(item);\n          }\n\n          lastInnerMessageId =\n            getLastInnerId(innerMessages) ?? lastInnerMessageId;\n\n          formatAdapter.reportTelemetry?.(batchItems, telemetryOptions);\n        }\n      }, 0);\n    });\n\n    return () => {\n      unsubscribe();\n      if (persistTimerRef.current) {\n        clearTimeout(persistTimerRef.current);\n        persistTimerRef.current = null;\n      }\n    };\n  }, [formatAdapter, storageFormatAdapter, runtimeRef]);\n\n  return isLoading;\n};\n"],"mappings":";;;;;;AAsBA,MAAa,+BACX,kBACA,aAC8B;CAC9B,MAAM,+BAAe,IAAI,IAAY;CACrC,MAAM,YAAY,SAAS,SAAS,SAAS,MAAM;EACjD,MAAM,UAAU,iBAAiB,CAAC,EAAE,OAAO,CAAC,EAAE;EAC9C,IAAI,CAAC,SAAS;GACZ,QAAQ,KAAK,qDAAqD;GAClE,OAAO,CAAC;EACV;EACA,IAAI,EAAE,YAAY,CAAC,aAAa,IAAI,EAAE,QAAQ,GAAG,OAAO,CAAC;EACzD,aAAa,IAAI,QAAQ,EAAE;EAC3B,OAAO,CAAC;GAAE,GAAG;GAAG;EAAQ,CAAC;CAC3B,CAAC;CAED,OAAO;EACL,QACE,SAAS,UAAU,aAAa,IAAI,SAAS,MAAM,IAC/C,SAAS,SACT;EACN,UAAU;CACZ;AACF;AAEA,MAAa,sBACX,YACA,gBACA,kBACA,sBACA,kBACG;CACH,MAAM,YAAY,OAAO,KAAK;CAE9B,MAAM,MAAM,OAAO;CACnB,MAAM,yBAAyB,kBACtB,IAAI,eAAe,SAAS,IAAI,eAAe,IAAI,MAC1D,CAAC,GAAG,CACN;CAEA,MAAM,CAAC,WAAW,gBAAgB,SAAS,KAAK;CAEhD,MAAM,aAAa,uBAAO,IAAI,IAAY,CAAC;CAE3C,MAAM,mBAAmB,OAAO,aAAa;CAC7C,gBAAgB;EACd,iBAAiB,UAAU;CAC7B,CAAC;CAED,MAAM,gBAAgB,cAAc;EAClC,IAAI,CAAC,gBAAgB,OAAO,KAAA;EAC5B,IAAI,CAAC,eAAe,YAClB,MAAM,IAAI,MACR,oFACF;EAEF,OAAO,eAAe,WAA0B,oBAAoB;CACtE,GAAG,CAAC,gBAAgB,oBAAoB,CAAC;CAEzC,gBAAgB;EACd,IAAI,CAAC,iBAAiB,UAAU,SAAS;EAEzC,MAAM,cAAc,YAAY;GAC9B,aAAa,IAAI;GACjB,IAAI;IACF,MAAM,OAAO,MAAM,cAAc,KAAK;IACtC,IAAI,QAAQ,KAAK,SAAS,SAAS,GAAG;KACpC,MAAM,YAAY,4BAA4B,kBAAkB,IAAI;KACpE,WAAW,QAAQ,OAAO,OAAO,SAAS;KAE1C,MAAM,WAAW,IAAI,kBAAkB;KACvC,SAAS,OAAO,SAAS;KACzB,MAAM,WAAW,SAAS,YAAY;KAEtC,iBAAiB,QACf,SAAS,QAAQ,wBAAkC,CACrD;KAEA,WAAW,UAAU,IAAI,IACvB,UAAU,SAAS,KAAK,MAAM,EAAE,QAAQ,EAAE,CAC5C;IACF;GACF,SAAS,OAAO;IACd,QAAQ,MAAM,mCAAmC,KAAK;GACxD,UAAU;IACR,aAAa,KAAK;GACpB;EACF;EAEA,UAAU,UAAU;EAEpB,IAAI,CAAC,uBAAuB,GAAG,SAAS,EAAE,UAAU;GAClD,aAAa,KAAK;GAClB;EACF;EAEA,YAAY;CACd,GAAG;EAAC;EAAe;EAAkB;EAAY;CAAsB,CAAC;CAExE,MAAM,cAAc,OAAsB,IAAI;CAC9C,MAAM,kBAAkB,OAA6C,IAAI;CACzE,MAAM,oBAAoB,OAAiB,CAAC,CAAC;CAC7C,MAAM,gBAAgB,OAAO,KAAK;CAClC,MAAM,mBAAmB,OAAO,CAAC;CAEjC,gBAAgB;EACd,IAAI,CAAC,eAAe;EAEpB,MAAM,cAAc,WAAW,QAAQ,OAAO,gBAAgB;GAC5D,MAAM,EAAE,cAAc,WAAW,QAAQ,OAAO,SAAS;GACzD,MAAM,aAAa,cAAc;GACjC,cAAc,UAAU;GAGxB,IAAI,YAAY,WAAW,MAAM;IAC/B,MAAM,UAAU,WAAW,QAAQ,OAAO,SAAS,EAAE,SAAS,GAAG,EAAE;IACnE,IAAI,SAAS,SAAS,aAAa;KACjC,MAAM,uBAAuB,QAAQ,QAAQ,QAC1C,MAAM,EAAE,SAAS,WACpB,EAAE;KACF,OAAO,iBAAiB,UAAU,sBAAsB;MACtD,kBAAkB,QAAQ,KAAK,KAAK,IAAI,IAAI,YAAY,OAAO;MAC/D,iBAAiB;KACnB;IACF;GACF;GAEA,IAAI,WAAW;IACb,IAAI,YAAY,WAAW,MAAM;KAC/B,YAAY,UAAU,KAAK,IAAI;KAC/B,kBAAkB,UAAU,CAAC;KAC7B,iBAAiB,UAAU;IAC7B;IAEA,IAAI,gBAAgB,SAAS;KAC3B,aAAa,gBAAgB,OAAO;KACpC,gBAAgB,UAAU;IAC5B;IACA;GACF;GAGA,IAAI,CAAC,YAAY;GAGjB,IAAI,YAAY,WAAW,MACzB,kBAAkB,QAAQ,KAAK,KAAK,IAAI,IAAI,YAAY,OAAO;GAIjE,IAAI,gBAAgB,SAAS,aAAa,gBAAgB,OAAO;GACjE,gBAAgB,UAAU,WAAW,YAAY;IAC/C,gBAAgB,UAAU;IAG1B,MAAM,SAAS,WAAW,QAAQ,OAAO,SAAS;IAClD,IAAI,OAAO,WAAW;IAGtB,MAAM,aAAa,kBAAkB;IACrC,MAAM,aACJ,WAAW,SAAS,IAAI,WAAW,GAAG,EAAE,IAAI,KAAA;IAG9C,IAAI,WAAW,WAAW,KAAK,cAAc,MAAM;KACjD,MAAM,gBAAgB,OAAO,SAAS,UACnC,MAAM,EAAE,SAAS,WACpB;KACA,IAAI,eAAe;MACjB,MAAM,UAAU,cAAc,QAAQ,QACnC,MAAM,EAAE,SAAS,WACpB,EAAE;MACF,IAAI,UAAU,GAAG;OACf,MAAM,aAAa,UAAU;OAC7B,MAAM,UAAU,aAAa;OAC7B,WAAW,SAAS;OACpB,KAAK,IAAI,IAAI,GAAG,IAAI,YAAY,KAC9B,WAAW,KAAK,KAAK,OAAO,IAAI,KAAK,OAAO,CAAC;MAEjD;KACF;IACF;IAGA,MAAM,iBACJ,WAAW,SAAS,IAChB,WAAW,KAAK,OAAO,OAAO;KAC5B,UAAU,MAAM,IAAI,IAAI,WAAW,IAAI;KACvC,QAAQ;IACV,EAAE,IACF,KAAA;IAEN,YAAY,UAAU;IACtB,kBAAkB,UAAU,CAAC;IAE7B,MAAM,mBAAmB;KACvB,GAAI,cAAc,OAAO,EAAE,WAAW,IAAI,KAAA;KAC1C,GAAI,kBAAkB,OAAO,EAAE,eAAe,IAAI,KAAA;IACpD;IAEA,MAAM,EAAE,aAAa;IACrB,IAAI,qBAAoC;IAExC,MAAM,kBAAkB,SACtB,KAAK,SAAS,IAAI,qBAAqB,MAAM,KAAK,GAAG,EAAE,CAAE,IAAI;IAE/D,MAAM,gBAAgB,SACpB,KAAK,KAAK,KAAK,SAAS;KACtB,UACE,QAAQ,IACJ,qBACA,qBAAqB,MAAM,KAAK,MAAM,EAAG;KAC/C,SAAS;IACX,EAAE;IAEJ,KAAK,MAAM,WAAW,UAAU;KAC9B,MAAM,gBAAgB,yBAAmC,OAAO;KAOhE,IAAI,EAJF,QAAQ,WAAW,KAAA,KACnB,QAAQ,OAAO,SAAS,cACxB,QAAQ,OAAO,SAAS,eAEZ;MACZ,qBACE,eAAe,aAAa,KAAK;MACnC;KACF;KAEA,IAAI,WAAW,QAAQ,IAAI,QAAQ,EAAE,GAAG;MACtC,IAAI,eAAe,KAAA,GAAW;OAC5B,IAAI,WAAW;OACf,KAAK,MAAM,gBAAgB,eAAe;QACxC,IAAI;SACF,MAAM,cAAc,SAClB;UAAE;UAAU,SAAS;SAAa,GAClC,qBAAqB,MAAM,YAAY,CACzC;QACF,QAAQ,CAER;QACA,WAAW,qBAAqB,MAAM,YAAY;OACpD;MACF;MACA,qBACE,eAAe,aAAa,KAAK;MACnC;KACF;KACA,WAAW,QAAQ,IAAI,QAAQ,EAAE;KAEjC,MAAM,aAAa,aAAa,aAAa;KAC7C,KAAK,MAAM,QAAQ,YACjB,MAAM,cAAc,OAAO,IAAI;KAGjC,qBACE,eAAe,aAAa,KAAK;KAEnC,cAAc,kBAAkB,YAAY,gBAAgB;IAC9D;GACF,GAAG,CAAC;EACN,CAAC;EAED,aAAa;GACX,YAAY;GACZ,IAAI,gBAAgB,SAAS;IAC3B,aAAa,gBAAgB,OAAO;IACpC,gBAAgB,UAAU;GAC5B;EACF;CACF,GAAG;EAAC;EAAe;EAAsB;CAAU,CAAC;CAEpD,OAAO;AACT"}

package/dist/ui/utils/convertMessage.d.ts

@@ -1,4 +1,4 @@
-import { ReadonlyJSONObject } from "../../packages/assistant-stream/dist/utils/json/json-value.js";
+import { ReadonlyJSONObject } from "../../assistant-stream/dist/utils/json/json-value.js";
 import { UIMessage } from "ai";
 import { useExternalMessageConverter } from "@assistant-ui/core/react";
 import { McpAppMetadata } from "@assistant-ui/core";
@@ -7,7 +7,8 @@ import { McpAppMetadata } from "@assistant-ui/core";
 type AISDKMessageConverterMetadata = useExternalMessageConverter.Metadata & {
   toolArgsKeyOrderCache?: Map<string, Map<string, string[]>>;
   toolLastInputCache?: Map<string, ReadonlyJSONObject>;
-  mcpAppMetadataCache?: Map<string, McpAppMetadata>;
+  mcpAppMetadataCache?: Map<string, McpAppMetadata>; /** Id of the currently-streaming message, flagged optimistic (#4037). */
+  optimisticMessageId?: string | undefined;
 };
 declare const AISDKMessageConverter: {
   useThreadMessages: ({

package/dist/ui/utils/convertMessage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessage.d.ts","names":[],"sources":["../../../src/ui/utils/convertMessage.ts"],"mappings":";;;;;;KAqBY,6BAAA,GACV,2BAAA,CAA4B,QAAA;EAC1B,qBAAA,GAAwB,GAAA,SAAY,GAAA;EACpC,kBAAA,GAAqB,GAAA,SAAY,kBAAA;EACjC,mBAAA,GAAsB,GAAA,SAAY,cAAA;AAAA;AAAA,cAyUzB,qBAAA;;;;;;;;;;;;;mJA3UyC,aAAA;kJAEf,aAAA"}
+{"version":3,"file":"convertMessage.d.ts","names":[],"sources":["../../../src/ui/utils/convertMessage.ts"],"mappings":";;;;;;KAqBY,6BAAA,GACV,2BAAA,CAA4B,QAAA;EAC1B,qBAAA,GAAwB,GAAA,SAAY,GAAA;EACpC,kBAAA,GAAqB,GAAA,SAAY,kBAAA;EACjC,mBAAA,GAAsB,GAAA,SAAY,cAAA,GAJG;EAMrC,mBAAA;AAAA;AAAA,cAyUS,qBAAA;;;;;;;;;;;;;mJA7UyC,aAAA;kJAEf,aAAA"}

package/dist/ui/utils/convertMessage.js

@@ -201,6 +201,7 @@ const AISDKMessageConverter = createMessageConverter((message, metadata) => {
 		case "system":
 		case "assistant": {
 			const timing = metadata.messageTiming?.[message.id];
+			const isOptimistic = message.role === "assistant" && message.id === metadata.optimisticMessageId;
 			return {
 				role: message.role,
 				id: message.id,
@@ -208,7 +209,8 @@ const AISDKMessageConverter = createMessageConverter((message, metadata) => {
 				content,
 				metadata: {
 					...message.metadata,
-					...timing && { timing }
+					...timing && { timing },
+					...isOptimistic && { isOptimistic: true }
 				}
 			};
 		}