@assistant-ui/react-langchain 0.0.25 → 0.0.26

package/dist/useStreamRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamRuntime.d.ts","names":[],"sources":["../src/useStreamRuntime.ts"],"mappings":";;;cAsCa,2BACX,WAAW;;kBAAA;;;;;;;;;cAYA,0BACX,mBACC,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;cAwZF,mBAAoB,YAAY,yDAAuB"}
+{"version":3,"file":"useStreamRuntime.d.ts","names":[],"sources":["../src/useStreamRuntime.ts"],"mappings":";;;cAsCa,2BACX,WAAW;;kBAAA;;;;;;;;;cAgBA,0BACX,mBACC,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;cAmfF,mBAAoB,YAAY,yDAAuB"}

package/dist/useStreamRuntime.js

@@ -4,9 +4,9 @@ import { langChainExtras } from "./runtimeExtras.js";
 import { foldUIUpdates, mergeUIMessages } from "./uiMessages.js";
 import { resolveForkCheckpoint } from "./resolveForkCheckpoint.js";
 import { useLangChainStreamingTiming } from "./streamingTiming.js";
-import { useAuiState } from "@assistant-ui/store";
+import { useAui, useAuiState } from "@assistant-ui/store";
 import { useCloudThreadListAdapter, useExternalMessageConverter, useExternalStoreRuntime, useRemoteThreadListRuntime } from "@assistant-ui/core/react";
-import { useEffect, useMemo, useRef, useState } from "react";
+import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { generateId, getExternalStoreMessages, pickExternalStoreSharedOptions } from "@assistant-ui/core";
 import { STREAM_CONTROLLER, useChannel, useStream } from "@langchain/react";
 //#region src/useStreamRuntime.ts
@@ -54,6 +54,7 @@ const truncateLangChainBaseMessages = (threadMessages, parentId) => {
 };
 const useStreamThreadRuntime = (options) => {
 	const { adapters, autoCancelPendingToolCalls, unstable_allowCancellation } = options;
+	const aui = useAui();
 	const messagesKey = options.messagesKey ?? "messages";
 	const uiStateKey = options.uiStateKey ?? "ui";
 	const externalId = useAuiState((s) => s.threadListItem.externalId);
@@ -83,6 +84,41 @@ const useStreamThreadRuntime = (options) => {
 	});
 	const streamRef = useRef(stream);
 	streamRef.current = stream;
+	const activeRunConfigRef = useRef(void 0);
+	const runConfigByMessageIdRef = useRef(/* @__PURE__ */ new Map());
+	const activeThreadIdRef = useRef(externalId);
+	const setActiveRunConfig = useCallback((runConfig) => {
+		activeRunConfigRef.current = runConfigToSubmitOptions(runConfig)?.config;
+	}, []);
+	const withActiveRunConfig = useCallback((submitOptions) => {
+		if (submitOptions && "config" in submitOptions) return submitOptions;
+		if (activeRunConfigRef.current === void 0) return submitOptions;
+		return {
+			...submitOptions,
+			config: activeRunConfigRef.current
+		};
+	}, []);
+	useEffect(() => {
+		if (activeThreadIdRef.current !== null && activeThreadIdRef.current !== externalId) {
+			activeRunConfigRef.current = void 0;
+			runConfigByMessageIdRef.current.clear();
+		}
+		activeThreadIdRef.current = externalId;
+	}, [externalId]);
+	useEffect(() => {
+		const messages = stream.messages;
+		const owned = runConfigByMessageIdRef.current;
+		for (let i = messages.length - 1; i >= 0; i--) {
+			const message = messages.at(i);
+			if (!message?.id || getMessageType(message) !== "ai" || !message.tool_calls?.length) continue;
+			if (owned.has(message.id)) return;
+			break;
+		}
+		for (const message of messages) {
+			if (!message.id || getMessageType(message) !== "ai" || !message.tool_calls?.length || owned.has(message.id)) continue;
+			owned.set(message.id, activeRunConfigRef.current);
+		}
+	}, [stream.messages]);
 	const visibleMessagesRef = useRef(visibleMessages);
 	visibleMessagesRef.current = visibleMessages;
 	const threadMessagesRef = useRef(threadMessages);
@@ -143,12 +179,19 @@ const useStreamThreadRuntime = (options) => {
 		subgraphs: stream.subgraphs,
 		stream,
 		error: stream.error,
-		submit: stream.submit,
-		respond: stream.respond,
-		respondAll: stream.respondAll,
+		submit: (values, submitOptions) => {
+			const isResume = values == null || submitOptions?.command != null;
+			return stream.submit(values, isResume ? withActiveRunConfig(submitOptions) : submitOptions);
+		},
+		respond: (response, respondOptions) => stream.respond(response, withActiveRunConfig(respondOptions)),
+		respondAll: (responsesById, respondOptions) => stream.respondAll(responsesById, withActiveRunConfig(respondOptions)),
 		values: stream.values,
 		messagesKey
-	}), [stream, messagesKey]);
+	}), [
+		stream,
+		messagesKey,
+		withActiveRunConfig
+	]);
 	return useExternalStoreRuntime({
 		...pickExternalStoreSharedOptions(options),
 		isRunning: effectiveIsRunning,
@@ -163,6 +206,7 @@ const useStreamThreadRuntime = (options) => {
 				stageUserMessage(msg);
 				return;
 			}
+			setActiveRunConfig(msg.runConfig);
 			const content = getMessageContent(msg);
 			const cancellations = autoCancelPendingToolCalls !== false ? getPendingToolCalls(streamRef.current.messages).map((t) => ({
 				type: "tool",
@@ -171,12 +215,17 @@ const useStreamThreadRuntime = (options) => {
 				content: JSON.stringify({ cancelled: true }),
 				status: "error"
 			})) : [];
-			await stream.submit({ [messagesKey]: [...cancellations, {
+			const { externalId } = await aui.threadListItem.initialize();
+			await streamRef.current.submit({ [messagesKey]: [...cancellations, {
 				type: "human",
 				content
-			}] }, runConfigToSubmitOptions(msg.runConfig));
+			}] }, {
+				...runConfigToSubmitOptions(msg.runConfig),
+				...externalId != null ? { threadId: externalId } : {}
+			});
 		},
-		onAddToolResult: async ({ toolCallId, toolName, result, isError, artifact }) => {
+		onAddToolResult: async ({ messageId, toolCallId, toolName, result, isError, artifact }) => {
+			const runConfig = runConfigByMessageIdRef.current.has(messageId) ? runConfigByMessageIdRef.current.get(messageId) : activeRunConfigRef.current;
 			await stream.submit({ [messagesKey]: [{
 				type: "tool",
 				name: toolName,
@@ -184,7 +233,7 @@ const useStreamThreadRuntime = (options) => {
 				content: JSON.stringify(result),
 				...artifact !== void 0 && { artifact },
 				status: isError ? "error" : "success"
-			}] });
+			}] }, runConfig === void 0 ? void 0 : { config: runConfig });
 		},
 		onReload: async (parentId, config) => {
 			const stagedRun = getStagedRun(parentId);
@@ -201,11 +250,13 @@ const useStreamThreadRuntime = (options) => {
 					visibleMessagesRef.current = nextMessages;
 					setStagedMessages(nextMessages);
 				} else setStagedMessages(null);
+				const runConfig = config.runConfig ?? stagedRun.runConfig;
+				setActiveRunConfig(runConfig);
 				await stream.submit({ [messagesKey]: stagedRun.messages.map((message) => ({
 					id: message.id,
 					type: "human",
 					content: message.content
-				})) }, runConfigToSubmitOptions(config.runConfig ?? stagedRun.runConfig));
+				})) }, runConfigToSubmitOptions(runConfig));
 				return;
 			}
 			const threadId = externalId;
@@ -213,6 +264,7 @@ const useStreamThreadRuntime = (options) => {
 			const s = streamRef.current;
 			const checkpointId = await resolveForkCheckpoint(s.client, threadId, s.messages, parentId, config.sourceId, s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(), messagesKey);
 			if (!checkpointId) return;
+			setActiveRunConfig(config.runConfig);
 			await s.submit(null, {
 				forkFrom: checkpointId,
 				...runConfigToSubmitOptions(config.runConfig)
@@ -238,6 +290,7 @@ const useStreamThreadRuntime = (options) => {
 			const checkpointId = await resolveForkCheckpoint(s.client, threadId, s.messages, message.parentId, message.sourceId, s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(), messagesKey);
 			if (!checkpointId) return;
 			const content = getMessageContent(message);
+			setActiveRunConfig(message.runConfig);
 			await s.submit({ [messagesKey]: [{
 				type: "human",
 				content
@@ -247,6 +300,7 @@ const useStreamThreadRuntime = (options) => {
 			});
 		},
 		onCancel: unstable_allowCancellation !== false ? async () => {
+			activeRunConfigRef.current = void 0;
 			await stream.stop();
 		} : void 0
 	});

package/dist/useStreamRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamRuntime.js","names":[],"sources":["../src/useStreamRuntime.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/store\" />\n\"use client\";\n\nimport { useEffect, useMemo, useRef, useState } from \"react\";\nimport type { AppendMessage, ToolExecutionStatus } from \"@assistant-ui/core\";\nimport {\n  generateId,\n  getExternalStoreMessages,\n  pickExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport type { ThreadMessage } from \"@assistant-ui/core\";\nimport {\n  useCloudThreadListAdapter,\n  useExternalStoreRuntime,\n  useExternalMessageConverter,\n  useRemoteThreadListRuntime,\n} from \"@assistant-ui/core/react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { STREAM_CONTROLLER, useChannel, useStream } from \"@langchain/react\";\nimport type { Channel } from \"@langchain/react\";\nimport type {\n  LangChainBaseMessage,\n  LangChainToolCall,\n  UIMessage,\n  UseStreamRuntimeOptions,\n} from \"./types\";\nimport {\n  convertLangChainBaseMessage,\n  getMessageContent,\n  getMessageType,\n} from \"./convertMessages\";\nimport { foldUIUpdates, mergeUIMessages } from \"./uiMessages\";\nimport { langChainExtras } from \"./runtimeExtras\";\nimport { resolveForkCheckpoint } from \"./resolveForkCheckpoint\";\nimport { useLangChainStreamingTiming } from \"./streamingTiming\";\n\nconst UI_CUSTOM_CHANNELS: readonly Channel[] = [\"custom\"];\n\nexport const runConfigToSubmitOptions = (\n  runConfig: AppendMessage[\"runConfig\"],\n) =>\n  runConfig?.custom\n    ? { config: { configurable: runConfig.custom } }\n    : undefined;\n\n/**\n * Group the graph's accumulated `UIMessage`s by the assistant message they\n * belong to. Non-array state and entries without a parent link are dropped.\n * The parent id comes from `metadata.message_id` (Python SDK) or\n * `metadata.id` (JS SDK).\n */\nexport const groupUIMessagesByParent = (\n  value: unknown,\n): Map<string, UIMessage[]> => {\n  const map = new Map<string, UIMessage[]>();\n  if (!Array.isArray(value)) return map;\n  for (const ui of value as UIMessage[]) {\n    const parentId = ui.metadata?.message_id ?? ui.metadata?.id;\n    if (!parentId) continue;\n    const existing = map.get(parentId);\n    if (existing) {\n      existing.push(ui);\n    } else {\n      map.set(parentId, [ui]);\n    }\n  }\n  return map;\n};\n\nconst getPendingToolCalls = (\n  messages: readonly LangChainBaseMessage[],\n): LangChainToolCall[] => {\n  const pending = new Map<string, LangChainToolCall>();\n  for (const m of messages) {\n    const type = getMessageType(m);\n    if (type === \"ai\") {\n      for (const tc of m.tool_calls ?? []) pending.set(tc.id, tc);\n    } else if (type === \"tool\" && m.tool_call_id) {\n      pending.delete(m.tool_call_id);\n    }\n  }\n  return [...pending.values()];\n};\n\nconst toStagedHumanMessage = (\n  msg: AppendMessage,\n  id = generateId(),\n): LangChainBaseMessage & { id: string } => ({\n  id,\n  _getType: () => \"human\",\n  content: getMessageContent(msg),\n});\n\nconst truncateLangChainBaseMessages = (\n  threadMessages: readonly ThreadMessage[],\n  parentId: string | null,\n): LangChainBaseMessage[] => {\n  if (parentId === null) return [];\n  const parentIndex = threadMessages.findIndex((m) => m.id === parentId);\n  if (parentIndex === -1) return [];\n  const truncated: LangChainBaseMessage[] = [];\n  for (let i = 0; i <= parentIndex && i < threadMessages.length; i++) {\n    truncated.push(\n      ...getExternalStoreMessages<LangChainBaseMessage>(threadMessages[i]!),\n    );\n  }\n  return truncated;\n};\n\ntype DistributiveOmit<T, K extends keyof any> = T extends unknown\n  ? Omit<T, K>\n  : never;\n\nconst useStreamThreadRuntime = (\n  options: DistributiveOmit<\n    UseStreamRuntimeOptions,\n    \"cloud\" | \"unstable_threadListAdapter\" | \"create\" | \"delete\"\n  >,\n) => {\n  const { adapters, autoCancelPendingToolCalls, unstable_allowCancellation } =\n    options;\n  const messagesKey = options.messagesKey ?? \"messages\";\n  const uiStateKey = options.uiStateKey ?? \"ui\";\n\n  const externalId = useAuiState((s) => s.threadListItem.externalId) as\n    | string\n    | null;\n  // Object.assign preserves the discriminated transport union; object spread\n  // collapses its arms and no longer satisfies UseStreamOptions.\n  const stream = useStream(\n    Object.assign({}, options, { threadId: externalId }),\n  );\n  const [stagedMessages, setStagedMessages] = useState<\n    LangChainBaseMessage[] | null\n  >(null);\n\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const effectiveIsRunning = stream.isLoading || hasExecutingTools;\n\n  const uiStateValue = stream.values[uiStateKey];\n\n  const customEvents = useChannel(stream, UI_CUSTOM_CHANNELS);\n  const liveUiMessages = useMemo(\n    () => foldUIUpdates(customEvents),\n    [customEvents],\n  );\n\n  const mergedUiMessages = useMemo(\n    () => mergeUIMessages(liveUiMessages, uiStateValue),\n    [liveUiMessages, uiStateValue],\n  );\n\n  const visibleMessages =\n    stagedMessages ?? (stream.messages as LangChainBaseMessage[]);\n\n  const messageTiming = useLangChainStreamingTiming(\n    visibleMessages,\n    effectiveIsRunning,\n  );\n\n  const convertWithUI = useMemo<\n    useExternalMessageConverter.Callback<LangChainBaseMessage>\n  >(() => {\n    const uiMessagesByParent = groupUIMessagesByParent(mergedUiMessages);\n    return (message, metadata) =>\n      convertLangChainBaseMessage(message, {\n        ...metadata,\n        uiMessagesByParent,\n        messageTiming,\n      });\n  }, [mergedUiMessages, messageTiming]);\n\n  const threadMessages = useExternalMessageConverter({\n    callback: convertWithUI,\n    messages: visibleMessages,\n    isRunning: effectiveIsRunning,\n  });\n\n  const streamRef = useRef(stream);\n  streamRef.current = stream;\n\n  const visibleMessagesRef = useRef(visibleMessages);\n  visibleMessagesRef.current = visibleMessages;\n\n  const threadMessagesRef = useRef(threadMessages);\n  threadMessagesRef.current = threadMessages;\n\n  const stagedMessagesRef = useRef(\n    new Map<\n      string,\n      {\n        message: LangChainBaseMessage & { id: string };\n        runConfig: AppendMessage[\"runConfig\"];\n      }\n    >(),\n  );\n  const stagedBaseMessagesRef = useRef<LangChainBaseMessage[] | null>(null);\n\n  useEffect(() => {\n    if (stagedMessagesRef.current.size === 0) return;\n\n    // Staged edits must keep their truncated base while stream updates arrive before promotion.\n    const baseMessages =\n      stagedBaseMessagesRef.current ??\n      (stream.messages as LangChainBaseMessage[]);\n    const baseMessageIds = new Set(\n      baseMessages.flatMap((message) => (message.id ? [message.id] : [])),\n    );\n    const remainingStagedMessages: LangChainBaseMessage[] = [];\n    const seenStagedIds = new Set<string>();\n    for (const message of visibleMessagesRef.current) {\n      if (!message.id || seenStagedIds.has(message.id)) continue;\n      if (baseMessageIds.has(message.id)) continue;\n      const staged = stagedMessagesRef.current.get(message.id);\n      if (!staged) continue;\n      remainingStagedMessages.push(staged.message);\n      seenStagedIds.add(message.id);\n    }\n\n    if (remainingStagedMessages.length === 0) {\n      stagedBaseMessagesRef.current = null;\n      visibleMessagesRef.current = baseMessages;\n      setStagedMessages(null);\n      return;\n    }\n\n    const nextMessages = [...baseMessages, ...remainingStagedMessages];\n    visibleMessagesRef.current = nextMessages;\n    setStagedMessages(nextMessages);\n  }, [stream.messages]);\n\n  const getStagedRun = (parentId: string | null) => {\n    if (!parentId || !stagedMessagesRef.current.has(parentId)) return null;\n\n    const staged: LangChainBaseMessage[] = [];\n    for (const message of visibleMessagesRef.current) {\n      if (message.id && stagedMessagesRef.current.has(message.id)) {\n        staged.push(stagedMessagesRef.current.get(message.id)!.message);\n      }\n      if (message.id === parentId) break;\n    }\n\n    return {\n      messages: staged,\n      runConfig: stagedMessagesRef.current.get(parentId)!.runConfig,\n    };\n  };\n\n  const stageUserMessage = (msg: AppendMessage) => {\n    const stagedMessage = toStagedHumanMessage(msg);\n    stagedMessagesRef.current.set(stagedMessage.id, {\n      message: stagedMessage,\n      runConfig: msg.runConfig,\n    });\n    const nextMessages = [...visibleMessagesRef.current, stagedMessage];\n    visibleMessagesRef.current = nextMessages;\n    setStagedMessages(nextMessages);\n  };\n\n  const extras = useMemo(\n    () =>\n      langChainExtras.provide({\n        interrupt: stream.interrupt,\n        interrupts: stream.interrupts,\n        toolCalls: stream.toolCalls,\n        subagents: stream.subagents,\n        subgraphs: stream.subgraphs,\n        stream,\n        error: stream.error,\n        submit: stream.submit,\n        respond: stream.respond,\n        respondAll: stream.respondAll,\n        values: stream.values,\n        messagesKey,\n      }),\n    [stream, messagesKey],\n  );\n\n  const runtime = useExternalStoreRuntime({\n    ...pickExternalStoreSharedOptions(options),\n    isRunning: effectiveIsRunning,\n    isLoading: stream.isThreadLoading,\n    messages: threadMessages,\n    adapters,\n    extras,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    onNew: async (msg) => {\n      if (!(msg.startRun ?? msg.role === \"user\")) {\n        stageUserMessage(msg);\n        return;\n      }\n\n      const content = getMessageContent(msg);\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(\n              streamRef.current.messages as readonly LangChainBaseMessage[],\n            ).map((t) => ({\n              type: \"tool\" as const,\n              name: t.name,\n              tool_call_id: t.id,\n              content: JSON.stringify({ cancelled: true }),\n              status: \"error\" as const,\n            }))\n          : [];\n      await stream.submit(\n        { [messagesKey]: [...cancellations, { type: \"human\", content }] },\n        runConfigToSubmitOptions(msg.runConfig),\n      );\n    },\n    onAddToolResult: async ({\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      artifact,\n    }) => {\n      await stream.submit({\n        [messagesKey]: [\n          {\n            type: \"tool\",\n            name: toolName,\n            tool_call_id: toolCallId,\n            content: JSON.stringify(result),\n            ...(artifact !== undefined && { artifact }),\n            status: isError ? \"error\" : \"success\",\n          },\n        ],\n      });\n    },\n    onReload: async (parentId, config) => {\n      const stagedRun = getStagedRun(parentId);\n      if (stagedRun) {\n        const promotedIds = new Set<string>();\n        for (const message of stagedRun.messages) {\n          if (!message.id) continue;\n          promotedIds.add(message.id);\n          stagedMessagesRef.current.delete(message.id);\n        }\n        stagedBaseMessagesRef.current = null;\n        if (stagedMessagesRef.current.size > 0) {\n          const nextMessages = visibleMessagesRef.current.filter(\n            (message) => !message.id || !promotedIds.has(message.id),\n          );\n          visibleMessagesRef.current = nextMessages;\n          setStagedMessages(nextMessages);\n        } else {\n          setStagedMessages(null);\n        }\n        await stream.submit(\n          {\n            [messagesKey]: stagedRun.messages.map((message) => ({\n              id: message.id,\n              type: \"human\",\n              content: message.content,\n            })),\n          },\n          runConfigToSubmitOptions(config.runConfig ?? stagedRun.runConfig),\n        );\n        return;\n      }\n\n      const threadId = externalId;\n      if (!threadId || parentId == null) return;\n      const s = streamRef.current;\n      const checkpointId = await resolveForkCheckpoint(\n        s.client,\n        threadId,\n        s.messages as readonly LangChainBaseMessage[],\n        parentId,\n        config.sourceId,\n        s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(),\n        messagesKey,\n      );\n      if (!checkpointId) return;\n      await s.submit(null, {\n        forkFrom: checkpointId,\n        ...runConfigToSubmitOptions(config.runConfig),\n      });\n    },\n    onEdit: async (message) => {\n      if (!(message.startRun ?? message.role === \"user\")) {\n        const truncated = truncateLangChainBaseMessages(\n          threadMessagesRef.current,\n          message.parentId,\n        );\n        const stagedMessage = toStagedHumanMessage(message);\n        stagedMessagesRef.current.set(stagedMessage.id, {\n          message: stagedMessage,\n          runConfig: message.runConfig,\n        });\n        stagedBaseMessagesRef.current = truncated;\n        const nextMessages = [...truncated, stagedMessage];\n        visibleMessagesRef.current = nextMessages;\n        setStagedMessages(nextMessages);\n        return;\n      }\n\n      const threadId = externalId;\n      if (!threadId) return;\n      const s = streamRef.current;\n      const checkpointId = await resolveForkCheckpoint(\n        s.client,\n        threadId,\n        s.messages as readonly LangChainBaseMessage[],\n        message.parentId,\n        message.sourceId,\n        s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(),\n        messagesKey,\n      );\n      if (!checkpointId) return;\n      const content = getMessageContent(message);\n      await s.submit(\n        { [messagesKey]: [{ type: \"human\", content }] },\n        {\n          forkFrom: checkpointId,\n          ...runConfigToSubmitOptions(message.runConfig),\n        },\n      );\n    },\n    onCancel:\n      unstable_allowCancellation !== false\n        ? async () => {\n            await stream.stop();\n          }\n        : undefined,\n  });\n\n  return runtime;\n};\n\n/**\n * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.\n * Accepts the same options as `useStream` from `@langchain/react`, plus\n * `cloud` and `adapters`.\n *\n * @example\n * ```tsx\n * import { useStreamRuntime } from \"@assistant-ui/react-langchain\";\n * import { AssistantRuntimeProvider, Thread } from \"@assistant-ui/react\";\n *\n * function App() {\n *   const runtime = useStreamRuntime({\n *     assistantId: \"agent\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   return (\n *     <AssistantRuntimeProvider runtime={runtime}>\n *       <Thread />\n *     </AssistantRuntimeProvider>\n *   );\n * }\n * ```\n */\nexport const useStreamRuntime = (rawOptions: UseStreamRuntimeOptions) => {\n  const {\n    cloud,\n    unstable_threadListAdapter,\n    create,\n    delete: deleteFn,\n    onThreadIdChange,\n    ...options\n  } = rawOptions;\n\n  const optionsRef = useRef(options);\n  optionsRef.current = options;\n\n  const cloudAdapter = useCloudThreadListAdapter({\n    cloud,\n    create,\n    delete: deleteFn,\n  });\n  const adapter = unstable_threadListAdapter ?? cloudAdapter;\n\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useStreamThreadRuntime(optionsRef.current);\n    },\n    adapter,\n    allowNesting: true,\n    onThreadIdChange,\n  });\n};\n"],"mappings":";;;;;;;;;;;;AAoCA,MAAM,qBAAyC,CAAC,QAAQ;AAExD,MAAa,4BACX,cAEA,WAAW,SACP,EAAE,QAAQ,EAAE,cAAc,UAAU,OAAO,EAAE,IAC7C,KAAA;;;;;;;AAQN,MAAa,2BACX,UAC6B;CAC7B,MAAM,sBAAM,IAAI,IAAyB;CACzC,IAAI,CAAC,MAAM,QAAQ,KAAK,GAAG,OAAO;CAClC,KAAK,MAAM,MAAM,OAAsB;EACrC,MAAM,WAAW,GAAG,UAAU,cAAc,GAAG,UAAU;EACzD,IAAI,CAAC,UAAU;EACf,MAAM,WAAW,IAAI,IAAI,QAAQ;EACjC,IAAI,UACF,SAAS,KAAK,EAAE;OAEhB,IAAI,IAAI,UAAU,CAAC,EAAE,CAAC;CAE1B;CACA,OAAO;AACT;AAEA,MAAM,uBACJ,aACwB;CACxB,MAAM,0BAAU,IAAI,IAA+B;CACnD,KAAK,MAAM,KAAK,UAAU;EACxB,MAAM,OAAO,eAAe,CAAC;EAC7B,IAAI,SAAS,MACX,KAAK,MAAM,MAAM,EAAE,cAAc,CAAC,GAAG,QAAQ,IAAI,GAAG,IAAI,EAAE;OACrD,IAAI,SAAS,UAAU,EAAE,cAC9B,QAAQ,OAAO,EAAE,YAAY;CAEjC;CACA,OAAO,CAAC,GAAG,QAAQ,OAAO,CAAC;AAC7B;AAEA,MAAM,wBACJ,KACA,KAAK,WAAW,OAC2B;CAC3C;CACA,gBAAgB;CAChB,SAAS,kBAAkB,GAAG;AAChC;AAEA,MAAM,iCACJ,gBACA,aAC2B;CAC3B,IAAI,aAAa,MAAM,OAAO,CAAC;CAC/B,MAAM,cAAc,eAAe,WAAW,MAAM,EAAE,OAAO,QAAQ;CACrE,IAAI,gBAAgB,IAAI,OAAO,CAAC;CAChC,MAAM,YAAoC,CAAC;CAC3C,KAAK,IAAI,IAAI,GAAG,KAAK,eAAe,IAAI,eAAe,QAAQ,KAC7D,UAAU,KACR,GAAG,yBAA+C,eAAe,EAAG,CACtE;CAEF,OAAO;AACT;AAMA,MAAM,0BACJ,YAIG;CACH,MAAM,EAAE,UAAU,4BAA4B,+BAC5C;CACF,MAAM,cAAc,QAAQ,eAAe;CAC3C,MAAM,aAAa,QAAQ,cAAc;CAEzC,MAAM,aAAa,aAAa,MAAM,EAAE,eAAe,UAAU;CAKjE,MAAM,SAAS,UACb,OAAO,OAAO,CAAC,GAAG,SAAS,EAAE,UAAU,WAAW,CAAC,CACrD;CACA,MAAM,CAAC,gBAAgB,qBAAqB,SAE1C,IAAI;CAEN,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,CAAC,CAAC,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,qBAAqB,OAAO,aAAa;CAE/C,MAAM,eAAe,OAAO,OAAO;CAEnC,MAAM,eAAe,WAAW,QAAQ,kBAAkB;CAC1D,MAAM,iBAAiB,cACf,cAAc,YAAY,GAChC,CAAC,YAAY,CACf;CAEA,MAAM,mBAAmB,cACjB,gBAAgB,gBAAgB,YAAY,GAClD,CAAC,gBAAgB,YAAY,CAC/B;CAEA,MAAM,kBACJ,kBAAmB,OAAO;CAE5B,MAAM,gBAAgB,4BACpB,iBACA,kBACF;CAEA,MAAM,gBAAgB,cAEd;EACN,MAAM,qBAAqB,wBAAwB,gBAAgB;EACnE,QAAQ,SAAS,aACf,4BAA4B,SAAS;GACnC,GAAG;GACH;GACA;EACF,CAAC;CACL,GAAG,CAAC,kBAAkB,aAAa,CAAC;CAEpC,MAAM,iBAAiB,4BAA4B;EACjD,UAAU;EACV,UAAU;EACV,WAAW;CACb,CAAC;CAED,MAAM,YAAY,OAAO,MAAM;CAC/B,UAAU,UAAU;CAEpB,MAAM,qBAAqB,OAAO,eAAe;CACjD,mBAAmB,UAAU;CAE7B,MAAM,oBAAoB,OAAO,cAAc;CAC/C,kBAAkB,UAAU;CAE5B,MAAM,oBAAoB,uBACxB,IAAI,IAMF,CACJ;CACA,MAAM,wBAAwB,OAAsC,IAAI;CAExE,gBAAgB;EACd,IAAI,kBAAkB,QAAQ,SAAS,GAAG;EAG1C,MAAM,eACJ,sBAAsB,WACrB,OAAO;EACV,MAAM,iBAAiB,IAAI,IACzB,aAAa,SAAS,YAAa,QAAQ,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAE,CACpE;EACA,MAAM,0BAAkD,CAAC;EACzD,MAAM,gCAAgB,IAAI,IAAY;EACtC,KAAK,MAAM,WAAW,mBAAmB,SAAS;GAChD,IAAI,CAAC,QAAQ,MAAM,cAAc,IAAI,QAAQ,EAAE,GAAG;GAClD,IAAI,eAAe,IAAI,QAAQ,EAAE,GAAG;GACpC,MAAM,SAAS,kBAAkB,QAAQ,IAAI,QAAQ,EAAE;GACvD,IAAI,CAAC,QAAQ;GACb,wBAAwB,KAAK,OAAO,OAAO;GAC3C,cAAc,IAAI,QAAQ,EAAE;EAC9B;EAEA,IAAI,wBAAwB,WAAW,GAAG;GACxC,sBAAsB,UAAU;GAChC,mBAAmB,UAAU;GAC7B,kBAAkB,IAAI;GACtB;EACF;EAEA,MAAM,eAAe,CAAC,GAAG,cAAc,GAAG,uBAAuB;EACjE,mBAAmB,UAAU;EAC7B,kBAAkB,YAAY;CAChC,GAAG,CAAC,OAAO,QAAQ,CAAC;CAEpB,MAAM,gBAAgB,aAA4B;EAChD,IAAI,CAAC,YAAY,CAAC,kBAAkB,QAAQ,IAAI,QAAQ,GAAG,OAAO;EAElE,MAAM,SAAiC,CAAC;EACxC,KAAK,MAAM,WAAW,mBAAmB,SAAS;GAChD,IAAI,QAAQ,MAAM,kBAAkB,QAAQ,IAAI,QAAQ,EAAE,GACxD,OAAO,KAAK,kBAAkB,QAAQ,IAAI,QAAQ,EAAE,CAAC,CAAE,OAAO;GAEhE,IAAI,QAAQ,OAAO,UAAU;EAC/B;EAEA,OAAO;GACL,UAAU;GACV,WAAW,kBAAkB,QAAQ,IAAI,QAAQ,CAAC,CAAE;EACtD;CACF;CAEA,MAAM,oBAAoB,QAAuB;EAC/C,MAAM,gBAAgB,qBAAqB,GAAG;EAC9C,kBAAkB,QAAQ,IAAI,cAAc,IAAI;GAC9C,SAAS;GACT,WAAW,IAAI;EACjB,CAAC;EACD,MAAM,eAAe,CAAC,GAAG,mBAAmB,SAAS,aAAa;EAClE,mBAAmB,UAAU;EAC7B,kBAAkB,YAAY;CAChC;CAEA,MAAM,SAAS,cAEX,gBAAgB,QAAQ;EACtB,WAAW,OAAO;EAClB,YAAY,OAAO;EACnB,WAAW,OAAO;EAClB,WAAW,OAAO;EAClB,WAAW,OAAO;EAClB;EACA,OAAO,OAAO;EACd,QAAQ,OAAO;EACf,SAAS,OAAO;EAChB,YAAY,OAAO;EACnB,QAAQ,OAAO;EACf;CACF,CAAC,GACH,CAAC,QAAQ,WAAW,CACtB;CAyJA,OAvJgB,wBAAwB;EACtC,GAAG,+BAA+B,OAAO;EACzC,WAAW;EACX,WAAW,OAAO;EAClB,UAAU;EACV;EACA;EACA,gCAAgC;EAChC;EACA,OAAO,OAAO,QAAQ;GACpB,IAAI,EAAE,IAAI,YAAY,IAAI,SAAS,SAAS;IAC1C,iBAAiB,GAAG;IACpB;GACF;GAEA,MAAM,UAAU,kBAAkB,GAAG;GACrC,MAAM,gBACJ,+BAA+B,QAC3B,oBACE,UAAU,QAAQ,QACpB,CAAC,CAAC,KAAK,OAAO;IACZ,MAAM;IACN,MAAM,EAAE;IACR,cAAc,EAAE;IAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;IAC3C,QAAQ;GACV,EAAE,IACF,CAAC;GACP,MAAM,OAAO,OACX,GAAG,cAAc,CAAC,GAAG,eAAe;IAAE,MAAM;IAAS;GAAQ,CAAC,EAAE,GAChE,yBAAyB,IAAI,SAAS,CACxC;EACF;EACA,iBAAiB,OAAO,EACtB,YACA,UACA,QACA,SACA,eACI;GACJ,MAAM,OAAO,OAAO,GACjB,cAAc,CACb;IACE,MAAM;IACN,MAAM;IACN,cAAc;IACd,SAAS,KAAK,UAAU,MAAM;IAC9B,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;IACzC,QAAQ,UAAU,UAAU;GAC9B,CACF,EACF,CAAC;EACH;EACA,UAAU,OAAO,UAAU,WAAW;GACpC,MAAM,YAAY,aAAa,QAAQ;GACvC,IAAI,WAAW;IACb,MAAM,8BAAc,IAAI,IAAY;IACpC,KAAK,MAAM,WAAW,UAAU,UAAU;KACxC,IAAI,CAAC,QAAQ,IAAI;KACjB,YAAY,IAAI,QAAQ,EAAE;KAC1B,kBAAkB,QAAQ,OAAO,QAAQ,EAAE;IAC7C;IACA,sBAAsB,UAAU;IAChC,IAAI,kBAAkB,QAAQ,OAAO,GAAG;KACtC,MAAM,eAAe,mBAAmB,QAAQ,QAC7C,YAAY,CAAC,QAAQ,MAAM,CAAC,YAAY,IAAI,QAAQ,EAAE,CACzD;KACA,mBAAmB,UAAU;KAC7B,kBAAkB,YAAY;IAChC,OACE,kBAAkB,IAAI;IAExB,MAAM,OAAO,OACX,GACG,cAAc,UAAU,SAAS,KAAK,aAAa;KAClD,IAAI,QAAQ;KACZ,MAAM;KACN,SAAS,QAAQ;IACnB,EAAE,EACJ,GACA,yBAAyB,OAAO,aAAa,UAAU,SAAS,CAClE;IACA;GACF;GAEA,MAAM,WAAW;GACjB,IAAI,CAAC,YAAY,YAAY,MAAM;GACnC,MAAM,IAAI,UAAU;GACpB,MAAM,eAAe,MAAM,sBACzB,EAAE,QACF,UACA,EAAE,UACF,UACA,OAAO,UACP,EAAE,kBAAkB,EAAE,sBAAsB,cAAc,GAC1D,WACF;GACA,IAAI,CAAC,cAAc;GACnB,MAAM,EAAE,OAAO,MAAM;IACnB,UAAU;IACV,GAAG,yBAAyB,OAAO,SAAS;GAC9C,CAAC;EACH;EACA,QAAQ,OAAO,YAAY;GACzB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,MAAM,YAAY,8BAChB,kBAAkB,SAClB,QAAQ,QACV;IACA,MAAM,gBAAgB,qBAAqB,OAAO;IAClD,kBAAkB,QAAQ,IAAI,cAAc,IAAI;KAC9C,SAAS;KACT,WAAW,QAAQ;IACrB,CAAC;IACD,sBAAsB,UAAU;IAChC,MAAM,eAAe,CAAC,GAAG,WAAW,aAAa;IACjD,mBAAmB,UAAU;IAC7B,kBAAkB,YAAY;IAC9B;GACF;GAEA,MAAM,WAAW;GACjB,IAAI,CAAC,UAAU;GACf,MAAM,IAAI,UAAU;GACpB,MAAM,eAAe,MAAM,sBACzB,EAAE,QACF,UACA,EAAE,UACF,QAAQ,UACR,QAAQ,UACR,EAAE,kBAAkB,EAAE,sBAAsB,cAAc,GAC1D,WACF;GACA,IAAI,CAAC,cAAc;GACnB,MAAM,UAAU,kBAAkB,OAAO;GACzC,MAAM,EAAE,OACN,GAAG,cAAc,CAAC;IAAE,MAAM;IAAS;GAAQ,CAAC,EAAE,GAC9C;IACE,UAAU;IACV,GAAG,yBAAyB,QAAQ,SAAS;GAC/C,CACF;EACF;EACA,UACE,+BAA+B,QAC3B,YAAY;GACV,MAAM,OAAO,KAAK;EACpB,IACA,KAAA;CACR,CAEa;AACf;;;;;;;;;;;;;;;;;;;;;;;;;AA0BA,MAAa,oBAAoB,eAAwC;CACvE,MAAM,EACJ,OACA,4BACA,QACA,QAAQ,UACR,kBACA,GAAG,YACD;CAEJ,MAAM,aAAa,OAAO,OAAO;CACjC,WAAW,UAAU;CAErB,MAAM,eAAe,0BAA0B;EAC7C;EACA;EACA,QAAQ;CACV,CAAC;CAGD,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAClC,OAAO,uBAAuB,WAAW,OAAO;EAClD;EACA,SANc,8BAA8B;EAO5C,cAAc;EACd;CACF,CAAC;AACH"}
+{"version":3,"file":"useStreamRuntime.js","names":[],"sources":["../src/useStreamRuntime.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/store\" />\n\"use client\";\n\nimport { useCallback, useEffect, useMemo, useRef, useState } from \"react\";\nimport type { AppendMessage, ToolExecutionStatus } from \"@assistant-ui/core\";\nimport {\n  generateId,\n  getExternalStoreMessages,\n  pickExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport type { ThreadMessage } from \"@assistant-ui/core\";\nimport {\n  useCloudThreadListAdapter,\n  useExternalStoreRuntime,\n  useExternalMessageConverter,\n  useRemoteThreadListRuntime,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport { STREAM_CONTROLLER, useChannel, useStream } from \"@langchain/react\";\nimport type { Channel } from \"@langchain/react\";\nimport type {\n  LangChainBaseMessage,\n  LangChainToolCall,\n  UIMessage,\n  UseStreamRuntimeOptions,\n} from \"./types\";\nimport {\n  convertLangChainBaseMessage,\n  getMessageContent,\n  getMessageType,\n} from \"./convertMessages\";\nimport { foldUIUpdates, mergeUIMessages } from \"./uiMessages\";\nimport { langChainExtras } from \"./runtimeExtras\";\nimport { resolveForkCheckpoint } from \"./resolveForkCheckpoint\";\nimport { useLangChainStreamingTiming } from \"./streamingTiming\";\n\nconst UI_CUSTOM_CHANNELS: readonly Channel[] = [\"custom\"];\n\nexport const runConfigToSubmitOptions = (\n  runConfig: AppendMessage[\"runConfig\"],\n) =>\n  runConfig?.custom\n    ? { config: { configurable: runConfig.custom } }\n    : undefined;\n\ntype NormalizedRunConfigOptions = NonNullable<\n  ReturnType<typeof runConfigToSubmitOptions>\n>;\n\n/**\n * Group the graph's accumulated `UIMessage`s by the assistant message they\n * belong to. Non-array state and entries without a parent link are dropped.\n * The parent id comes from `metadata.message_id` (Python SDK) or\n * `metadata.id` (JS SDK).\n */\nexport const groupUIMessagesByParent = (\n  value: unknown,\n): Map<string, UIMessage[]> => {\n  const map = new Map<string, UIMessage[]>();\n  if (!Array.isArray(value)) return map;\n  for (const ui of value as UIMessage[]) {\n    const parentId = ui.metadata?.message_id ?? ui.metadata?.id;\n    if (!parentId) continue;\n    const existing = map.get(parentId);\n    if (existing) {\n      existing.push(ui);\n    } else {\n      map.set(parentId, [ui]);\n    }\n  }\n  return map;\n};\n\nconst getPendingToolCalls = (\n  messages: readonly LangChainBaseMessage[],\n): LangChainToolCall[] => {\n  const pending = new Map<string, LangChainToolCall>();\n  for (const m of messages) {\n    const type = getMessageType(m);\n    if (type === \"ai\") {\n      for (const tc of m.tool_calls ?? []) pending.set(tc.id, tc);\n    } else if (type === \"tool\" && m.tool_call_id) {\n      pending.delete(m.tool_call_id);\n    }\n  }\n  return [...pending.values()];\n};\n\nconst toStagedHumanMessage = (\n  msg: AppendMessage,\n  id = generateId(),\n): LangChainBaseMessage & { id: string } => ({\n  id,\n  _getType: () => \"human\",\n  content: getMessageContent(msg),\n});\n\nconst truncateLangChainBaseMessages = (\n  threadMessages: readonly ThreadMessage[],\n  parentId: string | null,\n): LangChainBaseMessage[] => {\n  if (parentId === null) return [];\n  const parentIndex = threadMessages.findIndex((m) => m.id === parentId);\n  if (parentIndex === -1) return [];\n  const truncated: LangChainBaseMessage[] = [];\n  for (let i = 0; i <= parentIndex && i < threadMessages.length; i++) {\n    truncated.push(\n      ...getExternalStoreMessages<LangChainBaseMessage>(threadMessages[i]!),\n    );\n  }\n  return truncated;\n};\n\ntype DistributiveOmit<T, K extends keyof any> = T extends unknown\n  ? Omit<T, K>\n  : never;\n\nconst useStreamThreadRuntime = (\n  options: DistributiveOmit<\n    UseStreamRuntimeOptions,\n    \"cloud\" | \"unstable_threadListAdapter\" | \"create\" | \"delete\"\n  >,\n) => {\n  const { adapters, autoCancelPendingToolCalls, unstable_allowCancellation } =\n    options;\n  const aui = useAui();\n  const messagesKey = options.messagesKey ?? \"messages\";\n  const uiStateKey = options.uiStateKey ?? \"ui\";\n\n  const externalId = useAuiState((s) => s.threadListItem.externalId) as\n    | string\n    | null;\n  // Object.assign preserves the discriminated transport union; object spread\n  // collapses its arms and no longer satisfies UseStreamOptions.\n  const stream = useStream(\n    Object.assign({}, options, { threadId: externalId }),\n  );\n  const [stagedMessages, setStagedMessages] = useState<\n    LangChainBaseMessage[] | null\n  >(null);\n\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const effectiveIsRunning = stream.isLoading || hasExecutingTools;\n\n  const uiStateValue = stream.values[uiStateKey];\n\n  const customEvents = useChannel(stream, UI_CUSTOM_CHANNELS);\n  const liveUiMessages = useMemo(\n    () => foldUIUpdates(customEvents),\n    [customEvents],\n  );\n\n  const mergedUiMessages = useMemo(\n    () => mergeUIMessages(liveUiMessages, uiStateValue),\n    [liveUiMessages, uiStateValue],\n  );\n\n  const visibleMessages =\n    stagedMessages ?? (stream.messages as LangChainBaseMessage[]);\n\n  const messageTiming = useLangChainStreamingTiming(\n    visibleMessages,\n    effectiveIsRunning,\n  );\n\n  const convertWithUI = useMemo<\n    useExternalMessageConverter.Callback<LangChainBaseMessage>\n  >(() => {\n    const uiMessagesByParent = groupUIMessagesByParent(mergedUiMessages);\n    return (message, metadata) =>\n      convertLangChainBaseMessage(message, {\n        ...metadata,\n        uiMessagesByParent,\n        messageTiming,\n      });\n  }, [mergedUiMessages, messageTiming]);\n\n  const threadMessages = useExternalMessageConverter({\n    callback: convertWithUI,\n    messages: visibleMessages,\n    isRunning: effectiveIsRunning,\n  });\n\n  const streamRef = useRef(stream);\n  streamRef.current = stream;\n\n  const activeRunConfigRef = useRef<\n    NormalizedRunConfigOptions[\"config\"] | undefined\n  >(undefined);\n  const runConfigByMessageIdRef = useRef(\n    new Map<string, NormalizedRunConfigOptions[\"config\"] | undefined>(),\n  );\n  const activeThreadIdRef = useRef(externalId);\n  const setActiveRunConfig = useCallback(\n    (runConfig: AppendMessage[\"runConfig\"]) => {\n      activeRunConfigRef.current = runConfigToSubmitOptions(runConfig)?.config;\n    },\n    [],\n  );\n  const withActiveRunConfig = useCallback(\n    (submitOptions?: Record<string, unknown>) => {\n      if (submitOptions && \"config\" in submitOptions) return submitOptions;\n      if (activeRunConfigRef.current === undefined) return submitOptions;\n      return { ...submitOptions, config: activeRunConfigRef.current };\n    },\n    [],\n  );\n\n  useEffect(() => {\n    if (\n      activeThreadIdRef.current !== null &&\n      activeThreadIdRef.current !== externalId\n    ) {\n      activeRunConfigRef.current = undefined;\n      runConfigByMessageIdRef.current.clear();\n    }\n    activeThreadIdRef.current = externalId;\n  }, [externalId]);\n\n  useEffect(() => {\n    const messages = stream.messages as readonly LangChainBaseMessage[];\n    const owned = runConfigByMessageIdRef.current;\n    for (let i = messages.length - 1; i >= 0; i--) {\n      const message = messages.at(i);\n      if (\n        !message?.id ||\n        getMessageType(message) !== \"ai\" ||\n        !message.tool_calls?.length\n      ) {\n        continue;\n      }\n      if (owned.has(message.id)) return;\n      break;\n    }\n    for (const message of messages) {\n      if (\n        !message.id ||\n        getMessageType(message) !== \"ai\" ||\n        !message.tool_calls?.length ||\n        owned.has(message.id)\n      ) {\n        continue;\n      }\n      owned.set(message.id, activeRunConfigRef.current);\n    }\n  }, [stream.messages]);\n\n  const visibleMessagesRef = useRef(visibleMessages);\n  visibleMessagesRef.current = visibleMessages;\n\n  const threadMessagesRef = useRef(threadMessages);\n  threadMessagesRef.current = threadMessages;\n\n  const stagedMessagesRef = useRef(\n    new Map<\n      string,\n      {\n        message: LangChainBaseMessage & { id: string };\n        runConfig: AppendMessage[\"runConfig\"];\n      }\n    >(),\n  );\n  const stagedBaseMessagesRef = useRef<LangChainBaseMessage[] | null>(null);\n\n  useEffect(() => {\n    if (stagedMessagesRef.current.size === 0) return;\n\n    // Staged edits must keep their truncated base while stream updates arrive before promotion.\n    const baseMessages =\n      stagedBaseMessagesRef.current ??\n      (stream.messages as LangChainBaseMessage[]);\n    const baseMessageIds = new Set(\n      baseMessages.flatMap((message) => (message.id ? [message.id] : [])),\n    );\n    const remainingStagedMessages: LangChainBaseMessage[] = [];\n    const seenStagedIds = new Set<string>();\n    for (const message of visibleMessagesRef.current) {\n      if (!message.id || seenStagedIds.has(message.id)) continue;\n      if (baseMessageIds.has(message.id)) continue;\n      const staged = stagedMessagesRef.current.get(message.id);\n      if (!staged) continue;\n      remainingStagedMessages.push(staged.message);\n      seenStagedIds.add(message.id);\n    }\n\n    if (remainingStagedMessages.length === 0) {\n      stagedBaseMessagesRef.current = null;\n      visibleMessagesRef.current = baseMessages;\n      setStagedMessages(null);\n      return;\n    }\n\n    const nextMessages = [...baseMessages, ...remainingStagedMessages];\n    visibleMessagesRef.current = nextMessages;\n    setStagedMessages(nextMessages);\n  }, [stream.messages]);\n\n  const getStagedRun = (parentId: string | null) => {\n    if (!parentId || !stagedMessagesRef.current.has(parentId)) return null;\n\n    const staged: LangChainBaseMessage[] = [];\n    for (const message of visibleMessagesRef.current) {\n      if (message.id && stagedMessagesRef.current.has(message.id)) {\n        staged.push(stagedMessagesRef.current.get(message.id)!.message);\n      }\n      if (message.id === parentId) break;\n    }\n\n    return {\n      messages: staged,\n      runConfig: stagedMessagesRef.current.get(parentId)!.runConfig,\n    };\n  };\n\n  const stageUserMessage = (msg: AppendMessage) => {\n    const stagedMessage = toStagedHumanMessage(msg);\n    stagedMessagesRef.current.set(stagedMessage.id, {\n      message: stagedMessage,\n      runConfig: msg.runConfig,\n    });\n    const nextMessages = [...visibleMessagesRef.current, stagedMessage];\n    visibleMessagesRef.current = nextMessages;\n    setStagedMessages(nextMessages);\n  };\n\n  const extras = useMemo(\n    () =>\n      langChainExtras.provide({\n        interrupt: stream.interrupt,\n        interrupts: stream.interrupts,\n        toolCalls: stream.toolCalls,\n        subagents: stream.subagents,\n        subgraphs: stream.subgraphs,\n        stream,\n        error: stream.error,\n        submit: (values, submitOptions) => {\n          const isResume = values == null || submitOptions?.command != null;\n          return stream.submit(\n            values,\n            isResume ? withActiveRunConfig(submitOptions) : submitOptions,\n          );\n        },\n        respond: (response, respondOptions) =>\n          stream.respond(response, withActiveRunConfig(respondOptions)),\n        respondAll: (responsesById, respondOptions) =>\n          stream.respondAll(responsesById, withActiveRunConfig(respondOptions)),\n        values: stream.values,\n        messagesKey,\n      }),\n    [stream, messagesKey, withActiveRunConfig],\n  );\n\n  const runtime = useExternalStoreRuntime({\n    ...pickExternalStoreSharedOptions(options),\n    isRunning: effectiveIsRunning,\n    isLoading: stream.isThreadLoading,\n    messages: threadMessages,\n    adapters,\n    extras,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    onNew: async (msg) => {\n      if (!(msg.startRun ?? msg.role === \"user\")) {\n        stageUserMessage(msg);\n        return;\n      }\n\n      setActiveRunConfig(msg.runConfig);\n      const content = getMessageContent(msg);\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(\n              streamRef.current.messages as readonly LangChainBaseMessage[],\n            ).map((t) => ({\n              type: \"tool\" as const,\n              name: t.name,\n              tool_call_id: t.id,\n              content: JSON.stringify({ cancelled: true }),\n              status: \"error\" as const,\n            }))\n          : [];\n      // A null threadId is not a no-op for the SDK: it rebinds the controller\n      // away from its self-created thread and forces a fresh one, so the\n      // submit waits for initialization to produce an identity; core no\n      // longer holds appends on that barrier.\n      const { externalId } = await aui.threadListItem.initialize();\n      await streamRef.current.submit(\n        { [messagesKey]: [...cancellations, { type: \"human\", content }] },\n        {\n          ...runConfigToSubmitOptions(msg.runConfig),\n          ...(externalId != null ? { threadId: externalId } : {}),\n        },\n      );\n    },\n    onAddToolResult: async ({\n      messageId,\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      artifact,\n    }) => {\n      const runConfig = runConfigByMessageIdRef.current.has(messageId)\n        ? runConfigByMessageIdRef.current.get(messageId)\n        : activeRunConfigRef.current;\n      await stream.submit(\n        {\n          [messagesKey]: [\n            {\n              type: \"tool\",\n              name: toolName,\n              tool_call_id: toolCallId,\n              content: JSON.stringify(result),\n              ...(artifact !== undefined && { artifact }),\n              status: isError ? \"error\" : \"success\",\n            },\n          ],\n        },\n        runConfig === undefined ? undefined : { config: runConfig },\n      );\n    },\n    onReload: async (parentId, config) => {\n      const stagedRun = getStagedRun(parentId);\n      if (stagedRun) {\n        const promotedIds = new Set<string>();\n        for (const message of stagedRun.messages) {\n          if (!message.id) continue;\n          promotedIds.add(message.id);\n          stagedMessagesRef.current.delete(message.id);\n        }\n        stagedBaseMessagesRef.current = null;\n        if (stagedMessagesRef.current.size > 0) {\n          const nextMessages = visibleMessagesRef.current.filter(\n            (message) => !message.id || !promotedIds.has(message.id),\n          );\n          visibleMessagesRef.current = nextMessages;\n          setStagedMessages(nextMessages);\n        } else {\n          setStagedMessages(null);\n        }\n        const runConfig = config.runConfig ?? stagedRun.runConfig;\n        setActiveRunConfig(runConfig);\n        await stream.submit(\n          {\n            [messagesKey]: stagedRun.messages.map((message) => ({\n              id: message.id,\n              type: \"human\",\n              content: message.content,\n            })),\n          },\n          runConfigToSubmitOptions(runConfig),\n        );\n        return;\n      }\n\n      const threadId = externalId;\n      if (!threadId || parentId == null) return;\n      const s = streamRef.current;\n      const checkpointId = await resolveForkCheckpoint(\n        s.client,\n        threadId,\n        s.messages as readonly LangChainBaseMessage[],\n        parentId,\n        config.sourceId,\n        s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(),\n        messagesKey,\n      );\n      if (!checkpointId) return;\n      setActiveRunConfig(config.runConfig);\n      await s.submit(null, {\n        forkFrom: checkpointId,\n        ...runConfigToSubmitOptions(config.runConfig),\n      });\n    },\n    onEdit: async (message) => {\n      if (!(message.startRun ?? message.role === \"user\")) {\n        const truncated = truncateLangChainBaseMessages(\n          threadMessagesRef.current,\n          message.parentId,\n        );\n        const stagedMessage = toStagedHumanMessage(message);\n        stagedMessagesRef.current.set(stagedMessage.id, {\n          message: stagedMessage,\n          runConfig: message.runConfig,\n        });\n        stagedBaseMessagesRef.current = truncated;\n        const nextMessages = [...truncated, stagedMessage];\n        visibleMessagesRef.current = nextMessages;\n        setStagedMessages(nextMessages);\n        return;\n      }\n\n      const threadId = externalId;\n      if (!threadId) return;\n      const s = streamRef.current;\n      const checkpointId = await resolveForkCheckpoint(\n        s.client,\n        threadId,\n        s.messages as readonly LangChainBaseMessage[],\n        message.parentId,\n        message.sourceId,\n        s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(),\n        messagesKey,\n      );\n      if (!checkpointId) return;\n      const content = getMessageContent(message);\n      setActiveRunConfig(message.runConfig);\n      await s.submit(\n        { [messagesKey]: [{ type: \"human\", content }] },\n        {\n          forkFrom: checkpointId,\n          ...runConfigToSubmitOptions(message.runConfig),\n        },\n      );\n    },\n    onCancel:\n      unstable_allowCancellation !== false\n        ? async () => {\n            activeRunConfigRef.current = undefined;\n            await stream.stop();\n          }\n        : undefined,\n  });\n\n  return runtime;\n};\n\n/**\n * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.\n * Accepts the same options as `useStream` from `@langchain/react`, plus\n * `cloud` and `adapters`.\n *\n * @example\n * ```tsx\n * import { useStreamRuntime } from \"@assistant-ui/react-langchain\";\n * import { AssistantRuntimeProvider, Thread } from \"@assistant-ui/react\";\n *\n * function App() {\n *   const runtime = useStreamRuntime({\n *     assistantId: \"agent\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   return (\n *     <AssistantRuntimeProvider runtime={runtime}>\n *       <Thread />\n *     </AssistantRuntimeProvider>\n *   );\n * }\n * ```\n */\nexport const useStreamRuntime = (rawOptions: UseStreamRuntimeOptions) => {\n  const {\n    cloud,\n    unstable_threadListAdapter,\n    create,\n    delete: deleteFn,\n    onThreadIdChange,\n    ...options\n  } = rawOptions;\n\n  const optionsRef = useRef(options);\n  optionsRef.current = options;\n\n  const cloudAdapter = useCloudThreadListAdapter({\n    cloud,\n    create,\n    delete: deleteFn,\n  });\n  const adapter = unstable_threadListAdapter ?? cloudAdapter;\n\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useStreamThreadRuntime(optionsRef.current);\n    },\n    adapter,\n    allowNesting: true,\n    onThreadIdChange,\n  });\n};\n"],"mappings":";;;;;;;;;;;;AAoCA,MAAM,qBAAyC,CAAC,QAAQ;AAExD,MAAa,4BACX,cAEA,WAAW,SACP,EAAE,QAAQ,EAAE,cAAc,UAAU,OAAO,EAAE,IAC7C,KAAA;;;;;;;AAYN,MAAa,2BACX,UAC6B;CAC7B,MAAM,sBAAM,IAAI,IAAyB;CACzC,IAAI,CAAC,MAAM,QAAQ,KAAK,GAAG,OAAO;CAClC,KAAK,MAAM,MAAM,OAAsB;EACrC,MAAM,WAAW,GAAG,UAAU,cAAc,GAAG,UAAU;EACzD,IAAI,CAAC,UAAU;EACf,MAAM,WAAW,IAAI,IAAI,QAAQ;EACjC,IAAI,UACF,SAAS,KAAK,EAAE;OAEhB,IAAI,IAAI,UAAU,CAAC,EAAE,CAAC;CAE1B;CACA,OAAO;AACT;AAEA,MAAM,uBACJ,aACwB;CACxB,MAAM,0BAAU,IAAI,IAA+B;CACnD,KAAK,MAAM,KAAK,UAAU;EACxB,MAAM,OAAO,eAAe,CAAC;EAC7B,IAAI,SAAS,MACX,KAAK,MAAM,MAAM,EAAE,cAAc,CAAC,GAAG,QAAQ,IAAI,GAAG,IAAI,EAAE;OACrD,IAAI,SAAS,UAAU,EAAE,cAC9B,QAAQ,OAAO,EAAE,YAAY;CAEjC;CACA,OAAO,CAAC,GAAG,QAAQ,OAAO,CAAC;AAC7B;AAEA,MAAM,wBACJ,KACA,KAAK,WAAW,OAC2B;CAC3C;CACA,gBAAgB;CAChB,SAAS,kBAAkB,GAAG;AAChC;AAEA,MAAM,iCACJ,gBACA,aAC2B;CAC3B,IAAI,aAAa,MAAM,OAAO,CAAC;CAC/B,MAAM,cAAc,eAAe,WAAW,MAAM,EAAE,OAAO,QAAQ;CACrE,IAAI,gBAAgB,IAAI,OAAO,CAAC;CAChC,MAAM,YAAoC,CAAC;CAC3C,KAAK,IAAI,IAAI,GAAG,KAAK,eAAe,IAAI,eAAe,QAAQ,KAC7D,UAAU,KACR,GAAG,yBAA+C,eAAe,EAAG,CACtE;CAEF,OAAO;AACT;AAMA,MAAM,0BACJ,YAIG;CACH,MAAM,EAAE,UAAU,4BAA4B,+BAC5C;CACF,MAAM,MAAM,OAAO;CACnB,MAAM,cAAc,QAAQ,eAAe;CAC3C,MAAM,aAAa,QAAQ,cAAc;CAEzC,MAAM,aAAa,aAAa,MAAM,EAAE,eAAe,UAAU;CAKjE,MAAM,SAAS,UACb,OAAO,OAAO,CAAC,GAAG,SAAS,EAAE,UAAU,WAAW,CAAC,CACrD;CACA,MAAM,CAAC,gBAAgB,qBAAqB,SAE1C,IAAI;CAEN,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,CAAC,CAAC,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,qBAAqB,OAAO,aAAa;CAE/C,MAAM,eAAe,OAAO,OAAO;CAEnC,MAAM,eAAe,WAAW,QAAQ,kBAAkB;CAC1D,MAAM,iBAAiB,cACf,cAAc,YAAY,GAChC,CAAC,YAAY,CACf;CAEA,MAAM,mBAAmB,cACjB,gBAAgB,gBAAgB,YAAY,GAClD,CAAC,gBAAgB,YAAY,CAC/B;CAEA,MAAM,kBACJ,kBAAmB,OAAO;CAE5B,MAAM,gBAAgB,4BACpB,iBACA,kBACF;CAEA,MAAM,gBAAgB,cAEd;EACN,MAAM,qBAAqB,wBAAwB,gBAAgB;EACnE,QAAQ,SAAS,aACf,4BAA4B,SAAS;GACnC,GAAG;GACH;GACA;EACF,CAAC;CACL,GAAG,CAAC,kBAAkB,aAAa,CAAC;CAEpC,MAAM,iBAAiB,4BAA4B;EACjD,UAAU;EACV,UAAU;EACV,WAAW;CACb,CAAC;CAED,MAAM,YAAY,OAAO,MAAM;CAC/B,UAAU,UAAU;CAEpB,MAAM,qBAAqB,OAEzB,KAAA,CAAS;CACX,MAAM,0BAA0B,uBAC9B,IAAI,IAA8D,CACpE;CACA,MAAM,oBAAoB,OAAO,UAAU;CAC3C,MAAM,qBAAqB,aACxB,cAA0C;EACzC,mBAAmB,UAAU,yBAAyB,SAAS,CAAC,EAAE;CACpE,GACA,CAAC,CACH;CACA,MAAM,sBAAsB,aACzB,kBAA4C;EAC3C,IAAI,iBAAiB,YAAY,eAAe,OAAO;EACvD,IAAI,mBAAmB,YAAY,KAAA,GAAW,OAAO;EACrD,OAAO;GAAE,GAAG;GAAe,QAAQ,mBAAmB;EAAQ;CAChE,GACA,CAAC,CACH;CAEA,gBAAgB;EACd,IACE,kBAAkB,YAAY,QAC9B,kBAAkB,YAAY,YAC9B;GACA,mBAAmB,UAAU,KAAA;GAC7B,wBAAwB,QAAQ,MAAM;EACxC;EACA,kBAAkB,UAAU;CAC9B,GAAG,CAAC,UAAU,CAAC;CAEf,gBAAgB;EACd,MAAM,WAAW,OAAO;EACxB,MAAM,QAAQ,wBAAwB;EACtC,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;GAC7C,MAAM,UAAU,SAAS,GAAG,CAAC;GAC7B,IACE,CAAC,SAAS,MACV,eAAe,OAAO,MAAM,QAC5B,CAAC,QAAQ,YAAY,QAErB;GAEF,IAAI,MAAM,IAAI,QAAQ,EAAE,GAAG;GAC3B;EACF;EACA,KAAK,MAAM,WAAW,UAAU;GAC9B,IACE,CAAC,QAAQ,MACT,eAAe,OAAO,MAAM,QAC5B,CAAC,QAAQ,YAAY,UACrB,MAAM,IAAI,QAAQ,EAAE,GAEpB;GAEF,MAAM,IAAI,QAAQ,IAAI,mBAAmB,OAAO;EAClD;CACF,GAAG,CAAC,OAAO,QAAQ,CAAC;CAEpB,MAAM,qBAAqB,OAAO,eAAe;CACjD,mBAAmB,UAAU;CAE7B,MAAM,oBAAoB,OAAO,cAAc;CAC/C,kBAAkB,UAAU;CAE5B,MAAM,oBAAoB,uBACxB,IAAI,IAMF,CACJ;CACA,MAAM,wBAAwB,OAAsC,IAAI;CAExE,gBAAgB;EACd,IAAI,kBAAkB,QAAQ,SAAS,GAAG;EAG1C,MAAM,eACJ,sBAAsB,WACrB,OAAO;EACV,MAAM,iBAAiB,IAAI,IACzB,aAAa,SAAS,YAAa,QAAQ,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAE,CACpE;EACA,MAAM,0BAAkD,CAAC;EACzD,MAAM,gCAAgB,IAAI,IAAY;EACtC,KAAK,MAAM,WAAW,mBAAmB,SAAS;GAChD,IAAI,CAAC,QAAQ,MAAM,cAAc,IAAI,QAAQ,EAAE,GAAG;GAClD,IAAI,eAAe,IAAI,QAAQ,EAAE,GAAG;GACpC,MAAM,SAAS,kBAAkB,QAAQ,IAAI,QAAQ,EAAE;GACvD,IAAI,CAAC,QAAQ;GACb,wBAAwB,KAAK,OAAO,OAAO;GAC3C,cAAc,IAAI,QAAQ,EAAE;EAC9B;EAEA,IAAI,wBAAwB,WAAW,GAAG;GACxC,sBAAsB,UAAU;GAChC,mBAAmB,UAAU;GAC7B,kBAAkB,IAAI;GACtB;EACF;EAEA,MAAM,eAAe,CAAC,GAAG,cAAc,GAAG,uBAAuB;EACjE,mBAAmB,UAAU;EAC7B,kBAAkB,YAAY;CAChC,GAAG,CAAC,OAAO,QAAQ,CAAC;CAEpB,MAAM,gBAAgB,aAA4B;EAChD,IAAI,CAAC,YAAY,CAAC,kBAAkB,QAAQ,IAAI,QAAQ,GAAG,OAAO;EAElE,MAAM,SAAiC,CAAC;EACxC,KAAK,MAAM,WAAW,mBAAmB,SAAS;GAChD,IAAI,QAAQ,MAAM,kBAAkB,QAAQ,IAAI,QAAQ,EAAE,GACxD,OAAO,KAAK,kBAAkB,QAAQ,IAAI,QAAQ,EAAE,CAAC,CAAE,OAAO;GAEhE,IAAI,QAAQ,OAAO,UAAU;EAC/B;EAEA,OAAO;GACL,UAAU;GACV,WAAW,kBAAkB,QAAQ,IAAI,QAAQ,CAAC,CAAE;EACtD;CACF;CAEA,MAAM,oBAAoB,QAAuB;EAC/C,MAAM,gBAAgB,qBAAqB,GAAG;EAC9C,kBAAkB,QAAQ,IAAI,cAAc,IAAI;GAC9C,SAAS;GACT,WAAW,IAAI;EACjB,CAAC;EACD,MAAM,eAAe,CAAC,GAAG,mBAAmB,SAAS,aAAa;EAClE,mBAAmB,UAAU;EAC7B,kBAAkB,YAAY;CAChC;CAEA,MAAM,SAAS,cAEX,gBAAgB,QAAQ;EACtB,WAAW,OAAO;EAClB,YAAY,OAAO;EACnB,WAAW,OAAO;EAClB,WAAW,OAAO;EAClB,WAAW,OAAO;EAClB;EACA,OAAO,OAAO;EACd,SAAS,QAAQ,kBAAkB;GACjC,MAAM,WAAW,UAAU,QAAQ,eAAe,WAAW;GAC7D,OAAO,OAAO,OACZ,QACA,WAAW,oBAAoB,aAAa,IAAI,aAClD;EACF;EACA,UAAU,UAAU,mBAClB,OAAO,QAAQ,UAAU,oBAAoB,cAAc,CAAC;EAC9D,aAAa,eAAe,mBAC1B,OAAO,WAAW,eAAe,oBAAoB,cAAc,CAAC;EACtE,QAAQ,OAAO;EACf;CACF,CAAC,GACH;EAAC;EAAQ;EAAa;CAAmB,CAC3C;CA8KA,OA5KgB,wBAAwB;EACtC,GAAG,+BAA+B,OAAO;EACzC,WAAW;EACX,WAAW,OAAO;EAClB,UAAU;EACV;EACA;EACA,gCAAgC;EAChC;EACA,OAAO,OAAO,QAAQ;GACpB,IAAI,EAAE,IAAI,YAAY,IAAI,SAAS,SAAS;IAC1C,iBAAiB,GAAG;IACpB;GACF;GAEA,mBAAmB,IAAI,SAAS;GAChC,MAAM,UAAU,kBAAkB,GAAG;GACrC,MAAM,gBACJ,+BAA+B,QAC3B,oBACE,UAAU,QAAQ,QACpB,CAAC,CAAC,KAAK,OAAO;IACZ,MAAM;IACN,MAAM,EAAE;IACR,cAAc,EAAE;IAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;IAC3C,QAAQ;GACV,EAAE,IACF,CAAC;GAKP,MAAM,EAAE,eAAe,MAAM,IAAI,eAAe,WAAW;GAC3D,MAAM,UAAU,QAAQ,OACtB,GAAG,cAAc,CAAC,GAAG,eAAe;IAAE,MAAM;IAAS;GAAQ,CAAC,EAAE,GAChE;IACE,GAAG,yBAAyB,IAAI,SAAS;IACzC,GAAI,cAAc,OAAO,EAAE,UAAU,WAAW,IAAI,CAAC;GACvD,CACF;EACF;EACA,iBAAiB,OAAO,EACtB,WACA,YACA,UACA,QACA,SACA,eACI;GACJ,MAAM,YAAY,wBAAwB,QAAQ,IAAI,SAAS,IAC3D,wBAAwB,QAAQ,IAAI,SAAS,IAC7C,mBAAmB;GACvB,MAAM,OAAO,OACX,GACG,cAAc,CACb;IACE,MAAM;IACN,MAAM;IACN,cAAc;IACd,SAAS,KAAK,UAAU,MAAM;IAC9B,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;IACzC,QAAQ,UAAU,UAAU;GAC9B,CACF,EACF,GACA,cAAc,KAAA,IAAY,KAAA,IAAY,EAAE,QAAQ,UAAU,CAC5D;EACF;EACA,UAAU,OAAO,UAAU,WAAW;GACpC,MAAM,YAAY,aAAa,QAAQ;GACvC,IAAI,WAAW;IACb,MAAM,8BAAc,IAAI,IAAY;IACpC,KAAK,MAAM,WAAW,UAAU,UAAU;KACxC,IAAI,CAAC,QAAQ,IAAI;KACjB,YAAY,IAAI,QAAQ,EAAE;KAC1B,kBAAkB,QAAQ,OAAO,QAAQ,EAAE;IAC7C;IACA,sBAAsB,UAAU;IAChC,IAAI,kBAAkB,QAAQ,OAAO,GAAG;KACtC,MAAM,eAAe,mBAAmB,QAAQ,QAC7C,YAAY,CAAC,QAAQ,MAAM,CAAC,YAAY,IAAI,QAAQ,EAAE,CACzD;KACA,mBAAmB,UAAU;KAC7B,kBAAkB,YAAY;IAChC,OACE,kBAAkB,IAAI;IAExB,MAAM,YAAY,OAAO,aAAa,UAAU;IAChD,mBAAmB,SAAS;IAC5B,MAAM,OAAO,OACX,GACG,cAAc,UAAU,SAAS,KAAK,aAAa;KAClD,IAAI,QAAQ;KACZ,MAAM;KACN,SAAS,QAAQ;IACnB,EAAE,EACJ,GACA,yBAAyB,SAAS,CACpC;IACA;GACF;GAEA,MAAM,WAAW;GACjB,IAAI,CAAC,YAAY,YAAY,MAAM;GACnC,MAAM,IAAI,UAAU;GACpB,MAAM,eAAe,MAAM,sBACzB,EAAE,QACF,UACA,EAAE,UACF,UACA,OAAO,UACP,EAAE,kBAAkB,EAAE,sBAAsB,cAAc,GAC1D,WACF;GACA,IAAI,CAAC,cAAc;GACnB,mBAAmB,OAAO,SAAS;GACnC,MAAM,EAAE,OAAO,MAAM;IACnB,UAAU;IACV,GAAG,yBAAyB,OAAO,SAAS;GAC9C,CAAC;EACH;EACA,QAAQ,OAAO,YAAY;GACzB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,MAAM,YAAY,8BAChB,kBAAkB,SAClB,QAAQ,QACV;IACA,MAAM,gBAAgB,qBAAqB,OAAO;IAClD,kBAAkB,QAAQ,IAAI,cAAc,IAAI;KAC9C,SAAS;KACT,WAAW,QAAQ;IACrB,CAAC;IACD,sBAAsB,UAAU;IAChC,MAAM,eAAe,CAAC,GAAG,WAAW,aAAa;IACjD,mBAAmB,UAAU;IAC7B,kBAAkB,YAAY;IAC9B;GACF;GAEA,MAAM,WAAW;GACjB,IAAI,CAAC,UAAU;GACf,MAAM,IAAI,UAAU;GACpB,MAAM,eAAe,MAAM,sBACzB,EAAE,QACF,UACA,EAAE,UACF,QAAQ,UACR,QAAQ,UACR,EAAE,kBAAkB,EAAE,sBAAsB,cAAc,GAC1D,WACF;GACA,IAAI,CAAC,cAAc;GACnB,MAAM,UAAU,kBAAkB,OAAO;GACzC,mBAAmB,QAAQ,SAAS;GACpC,MAAM,EAAE,OACN,GAAG,cAAc,CAAC;IAAE,MAAM;IAAS;GAAQ,CAAC,EAAE,GAC9C;IACE,UAAU;IACV,GAAG,yBAAyB,QAAQ,SAAS;GAC/C,CACF;EACF;EACA,UACE,+BAA+B,QAC3B,YAAY;GACV,mBAAmB,UAAU,KAAA;GAC7B,MAAM,OAAO,KAAK;EACpB,IACA,KAAA;CACR,CAEa;AACf;;;;;;;;;;;;;;;;;;;;;;;;;AA0BA,MAAa,oBAAoB,eAAwC;CACvE,MAAM,EACJ,OACA,4BACA,QACA,QAAQ,UACR,kBACA,GAAG,YACD;CAEJ,MAAM,aAAa,OAAO,OAAO;CACjC,WAAW,UAAU;CAErB,MAAM,eAAe,0BAA0B;EAC7C;EACA;EACA,QAAQ;CACV,CAAC;CAGD,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAClC,OAAO,uBAAuB,WAAW,OAAO;EAClD;EACA,SANc,8BAA8B;EAO5C,cAAc;EACd;CACF,CAAC;AACH"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-langchain",
-  "version": "0.0.25",
+  "version": "0.0.26",
   "description": "LangChain useStream adapter for assistant-ui",
   "keywords": [
     "langchain",
@@ -29,10 +29,10 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.3.10",
-    "@assistant-ui/store": "^0.3.8",
+    "@assistant-ui/core": "^0.3.14",
+    "@assistant-ui/store": "^0.3.10",
     "assistant-cloud": "*",
-    "assistant-stream": "^0.3.36"
+    "assistant-stream": "^0.3.38"
   },
   "peerDependencies": {
     "@langchain/react": "^1.0.2",
@@ -46,14 +46,14 @@
   },
   "devDependencies": {
     "@langchain/core": "^1.2.5",
-    "@langchain/langgraph-sdk": "^1.9.28",
-    "@langchain/react": "^1.0.29",
+    "@langchain/langgraph-sdk": "^1.9.29",
+    "@langchain/react": "^1.0.30",
     "@testing-library/react": "^16.3.2",
     "@types/react": "^19.2.18",
     "jsdom": "^30.0.1",
     "react": "^19.2.8",
     "vitest": "^4.1.10",
-    "@assistant-ui/x-buildutils": "0.0.22"
+    "@assistant-ui/x-buildutils": "0.0.23"
   },
   "publishConfig": {
     "access": "public",

package/src/useStreamRuntime.test.tsx

@@ -10,6 +10,13 @@ import type {
 import { useAui } from "@assistant-ui/store";
 import type { LangChainBaseMessage } from "./types";
 import type { ReactNode } from "react";
+import {
+  useLangChainRespond,
+  useLangChainRespondAll,
+  useLangChainSend,
+  useLangChainSendCommand,
+  useLangChainSubmit,
+} from "./hooks";
 
 const { mockUseChannel, mockUseStream, streamController } = vi.hoisted(() => ({
   mockUseChannel: vi.fn(() => []),
@@ -147,6 +154,14 @@ const makeThreadListAdapter = (): RemoteThreadListAdapter => ({
   })),
 });
 
+const deferred = <T,>() => {
+  let resolve!: (value: T) => void;
+  const promise = new Promise<T>((res) => {
+    resolve = res;
+  });
+  return { promise, resolve };
+};
+
 describe("useStreamRuntime thread options", () => {
   it("keeps stream options isolated between mounted threads", async () => {
     mockUseStream.mockReturnValue(createMockStream());
@@ -186,6 +201,409 @@ describe("useStreamRuntime thread options", () => {
 
     view.unmount();
   });
+
+  it("submits with the initialized thread id when initialization resolves before rerender", async () => {
+    const stream = createMockStream();
+    mockUseStream.mockReturnValue(stream);
+    const initialization = deferred<{
+      remoteId: string;
+      externalId: string;
+    }>();
+    const threadListAdapter = makeThreadListAdapter();
+    threadListAdapter.list = vi.fn(async () => ({ threads: [] }));
+    threadListAdapter.initialize = vi.fn(() => initialization.promise);
+
+    const capture: { runtime: AssistantRuntime | null } = { runtime: null };
+    const TestRuntime = () => {
+      const runtime = useStreamRuntime({
+        apiUrl: "/api",
+        unstable_threadListAdapter: threadListAdapter,
+      } as never);
+      capture.runtime = runtime;
+      return <AssistantRuntimeProvider runtime={runtime} />;
+    };
+
+    const view = render(<TestRuntime />);
+    await waitFor(() => expect(capture.runtime).not.toBeNull());
+
+    await act(async () => {
+      capture.runtime!.thread.append({
+        role: "user",
+        content: [{ type: "text", text: "hello" }],
+      });
+      await Promise.resolve();
+    });
+
+    expect(stream.submit).not.toHaveBeenCalled();
+
+    await act(async () => {
+      initialization.resolve({ remoteId: "thread-b", externalId: "thread-b" });
+    });
+
+    await waitFor(() =>
+      expect(stream.submit).toHaveBeenCalledWith(
+        { messages: [{ type: "human", content: "hello" }] },
+        { threadId: "thread-b" },
+      ),
+    );
+    view.unmount();
+  });
+
+  it("omits the threadId override when initialization yields no external id", async () => {
+    const stream = createMockStream();
+    mockUseStream.mockReturnValue(stream);
+    const capture: { runtime: AssistantRuntime | null } = { runtime: null };
+    const TestRuntime = () => {
+      const runtime = useStreamRuntime({ apiUrl: "/api" } as never);
+      capture.runtime = runtime;
+      return <AssistantRuntimeProvider runtime={runtime} />;
+    };
+    const view = render(<TestRuntime />);
+    await waitFor(() => expect(capture.runtime).not.toBeNull());
+
+    await act(async () => {
+      await capture.runtime!.thread.append({
+        role: "user",
+        content: [{ type: "text", text: "one" }],
+      });
+    });
+    await act(async () => {
+      await capture.runtime!.thread.append({
+        role: "user",
+        content: [{ type: "text", text: "two" }],
+      });
+    });
+
+    expect(stream.submit).toHaveBeenCalledTimes(2);
+    for (const call of stream.submit.mock.calls) {
+      expect(call[1]).not.toHaveProperty("threadId");
+    }
+    view.unmount();
+  });
+});
+
+describe("useStreamRuntime run configuration", () => {
+  it("preserves custom configuration for automatic tool-result resumes", async () => {
+    const stream = createMockStream();
+    const { auiResult, rerender } = renderAui(stream);
+
+    await act(async () => {
+      await auiResult.current.thread.append({
+        role: "user",
+        content: [{ type: "text", text: "hello" }],
+        runConfig: { custom: { model_name: "gpt-5.4-nano" } },
+      });
+    });
+
+    stream.messages = [
+      {
+        id: "assistant-1",
+        _getType: () => "ai",
+        content: "",
+        tool_calls: [{ id: "tool-1", name: "lookup", args: {} }],
+      },
+    ];
+    rerender();
+
+    await waitFor(() => {
+      expect(auiResult.current.thread.getState().messages).toContainEqual(
+        expect.objectContaining({
+          id: "assistant-1",
+          content: expect.arrayContaining([
+            expect.objectContaining({
+              type: "tool-call",
+              toolCallId: "tool-1",
+            }),
+          ]),
+        }),
+      );
+    });
+
+    act(() => {
+      auiResult.current.thread
+        .message({ id: "assistant-1" })
+        .part({ toolCallId: "tool-1" })
+        .addToolResult({ answer: 42 });
+    });
+    await waitFor(() => expect(stream.submit).toHaveBeenCalledTimes(2));
+
+    const config = { config: { configurable: { model_name: "gpt-5.4-nano" } } };
+    expect(stream.submit).toHaveBeenNthCalledWith(
+      1,
+      {
+        messages: [{ type: "human", content: "hello" }],
+      },
+      config,
+    );
+    expect(stream.submit).toHaveBeenNthCalledWith(
+      2,
+      {
+        messages: [
+          {
+            type: "tool",
+            name: "lookup",
+            tool_call_id: "tool-1",
+            content: JSON.stringify({ answer: 42 }),
+            status: "success",
+          },
+        ],
+      },
+      config,
+    );
+  });
+
+  it("inherits custom configuration through exposed resume helpers", async () => {
+    const stream = createMockStream();
+    mockUseStream.mockReturnValue(stream);
+    const capture: {
+      runtime: AssistantRuntime | null;
+      respond?: ReturnType<typeof useLangChainRespond>;
+      respondAll?: ReturnType<typeof useLangChainRespondAll>;
+      sendCommand?: ReturnType<typeof useLangChainSendCommand>;
+    } = { runtime: null };
+
+    const Capture = () => {
+      capture.respond = useLangChainRespond();
+      capture.respondAll = useLangChainRespondAll();
+      capture.sendCommand = useLangChainSendCommand();
+      return null;
+    };
+    Capture.displayName = "Capture";
+
+    const TestRuntime = () => {
+      const runtime = useStreamRuntime({ apiUrl: "/api" } as never);
+      capture.runtime = runtime;
+      return (
+        <AssistantRuntimeProvider runtime={runtime}>
+          <Capture />
+        </AssistantRuntimeProvider>
+      );
+    };
+    TestRuntime.displayName = "TestRuntime";
+
+    const view = render(<TestRuntime />);
+    await waitFor(() => expect(capture.respond).toBeDefined());
+
+    await act(async () => {
+      await capture.runtime!.thread.append({
+        role: "user",
+        content: [{ type: "text", text: "hello" }],
+        runConfig: { custom: { model_name: "gpt-5.4-nano" } },
+      });
+    });
+
+    const config = { config: { configurable: { model_name: "gpt-5.4-nano" } } };
+    await act(async () => {
+      await capture.respond!({ approved: true });
+      await capture.respondAll!({ "interrupt-1": { approved: true } });
+      await capture.sendCommand!({ resume: "continue" });
+    });
+
+    expect(stream.respond).toHaveBeenCalledWith({ approved: true }, config);
+    expect(stream.respondAll).toHaveBeenCalledWith(
+      { "interrupt-1": { approved: true } },
+      config,
+    );
+    expect(stream.submit).toHaveBeenLastCalledWith(null, {
+      command: { resume: "continue" },
+      ...config,
+    });
+    view.unmount();
+  });
+
+  it("keeps a delayed tool result on the run that produced it", async () => {
+    const stream = createMockStream();
+    const { auiResult, rerender } = renderAui(stream);
+
+    await act(async () => {
+      await auiResult.current.thread.append({
+        role: "user",
+        content: [{ type: "text", text: "first" }],
+        runConfig: { custom: { model_name: "model-a" } },
+      });
+    });
+
+    stream.messages = [
+      {
+        id: "assistant-1",
+        _getType: () => "ai",
+        content: "",
+        tool_calls: [{ id: "tool-1", name: "lookup", args: {} }],
+      },
+    ];
+    rerender();
+    await waitFor(() => {
+      expect(auiResult.current.thread.getState().messages).toContainEqual(
+        expect.objectContaining({ id: "assistant-1" }),
+      );
+    });
+
+    await act(async () => {
+      await auiResult.current.thread.append({
+        role: "user",
+        content: [{ type: "text", text: "second" }],
+        runConfig: { custom: { model_name: "model-b" } },
+      });
+    });
+
+    act(() => {
+      auiResult.current.thread
+        .message({ id: "assistant-1" })
+        .part({ toolCallId: "tool-1" })
+        .addToolResult({ answer: 42 });
+    });
+    await waitFor(() => expect(stream.submit).toHaveBeenCalledTimes(3));
+
+    expect(stream.submit).toHaveBeenLastCalledWith(
+      {
+        messages: [
+          {
+            type: "tool",
+            name: "lookup",
+            tool_call_id: "tool-1",
+            content: JSON.stringify({ answer: 42 }),
+            status: "success",
+          },
+        ],
+      },
+      { config: { configurable: { model_name: "model-a" } } },
+    );
+  });
+
+  it("does not let a caller-supplied resume config replace the recorded configurable", async () => {
+    const stream = createMockStream();
+    mockUseStream.mockReturnValue(stream);
+    const capture: {
+      runtime: AssistantRuntime | null;
+      aui?: ReturnType<typeof useAui>;
+      submit?: ReturnType<typeof useLangChainSubmit>;
+    } = { runtime: null };
+
+    const Capture = () => {
+      capture.aui = useAui();
+      capture.submit = useLangChainSubmit();
+      return null;
+    };
+    Capture.displayName = "Capture";
+
+    const TestRuntime = () => {
+      const runtime = useStreamRuntime({ apiUrl: "/api" } as never);
+      capture.runtime = runtime;
+      return (
+        <AssistantRuntimeProvider runtime={runtime}>
+          <Capture />
+        </AssistantRuntimeProvider>
+      );
+    };
+    TestRuntime.displayName = "TestRuntime";
+
+    const view = render(<TestRuntime />);
+    await waitFor(() => expect(capture.submit).toBeDefined());
+
+    await act(async () => {
+      await capture.runtime!.thread.append({
+        role: "user",
+        content: [{ type: "text", text: "hello" }],
+        runConfig: { custom: { model_name: "gpt-5.4-nano" } },
+      });
+    });
+
+    await act(async () => {
+      await capture.submit!(null, { command: { resume: "continue" } });
+      await capture.submit!(null, {
+        command: { resume: "continue" },
+        config: { recursion_limit: 5 },
+      });
+    });
+
+    stream.messages = [
+      {
+        id: "assistant-1",
+        _getType: () => "ai",
+        content: "",
+        tool_calls: [{ id: "tool-1", name: "lookup", args: {} }],
+      },
+    ];
+    view.rerender(<TestRuntime />);
+    await waitFor(() => {
+      expect(capture.aui!.thread.getState().messages).toContainEqual(
+        expect.objectContaining({ id: "assistant-1" }),
+      );
+    });
+
+    act(() => {
+      capture
+        .aui!.thread.message({ id: "assistant-1" })
+        .part({ toolCallId: "tool-1" })
+        .addToolResult({ answer: 42 });
+    });
+    await waitFor(() => expect(stream.submit).toHaveBeenCalledTimes(4));
+
+    expect(stream.submit).toHaveBeenNthCalledWith(2, null, {
+      command: { resume: "continue" },
+      config: { configurable: { model_name: "gpt-5.4-nano" } },
+    });
+    expect(stream.submit).toHaveBeenNthCalledWith(3, null, {
+      command: { resume: "continue" },
+      config: { recursion_limit: 5 },
+    });
+    expect(stream.submit).toHaveBeenLastCalledWith(
+      expect.objectContaining({
+        messages: expect.arrayContaining([
+          expect.objectContaining({ tool_call_id: "tool-1" }),
+        ]),
+      }),
+      { config: { configurable: { model_name: "gpt-5.4-nano" } } },
+    );
+    view.unmount();
+  });
+
+  it("does not inject the recorded config into a raw new-run submit", async () => {
+    const stream = createMockStream();
+    mockUseStream.mockReturnValue(stream);
+    const capture: {
+      runtime: AssistantRuntime | null;
+      send?: ReturnType<typeof useLangChainSend>;
+    } = { runtime: null };
+
+    const Capture = () => {
+      capture.send = useLangChainSend();
+      return null;
+    };
+    Capture.displayName = "Capture";
+
+    const TestRuntime = () => {
+      const runtime = useStreamRuntime({ apiUrl: "/api" } as never);
+      capture.runtime = runtime;
+      return (
+        <AssistantRuntimeProvider runtime={runtime}>
+          <Capture />
+        </AssistantRuntimeProvider>
+      );
+    };
+    TestRuntime.displayName = "TestRuntime";
+
+    const view = render(<TestRuntime />);
+    await waitFor(() => expect(capture.send).toBeDefined());
+
+    await act(async () => {
+      await capture.runtime!.thread.append({
+        role: "user",
+        content: [{ type: "text", text: "hello" }],
+        runConfig: { custom: { model_name: "gpt-5.4-nano" } },
+      });
+    });
+
+    await act(async () => {
+      await capture.send!([{ type: "human", content: "next" }]);
+    });
+
+    expect(stream.submit).toHaveBeenLastCalledWith(
+      { messages: [{ type: "human", content: "next" }] },
+      undefined,
+    );
+    view.unmount();
+  });
 });
 
 describe("useStreamRuntime staged messages", () => {

package/src/useStreamRuntime.ts

@@ -1,7 +1,7 @@
 /// <reference types="@assistant-ui/core/store" />
 "use client";
 
-import { useEffect, useMemo, useRef, useState } from "react";
+import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import type { AppendMessage, ToolExecutionStatus } from "@assistant-ui/core";
 import {
   generateId,
@@ -15,7 +15,7 @@ import {
   useExternalMessageConverter,
   useRemoteThreadListRuntime,
 } from "@assistant-ui/core/react";
-import { useAuiState } from "@assistant-ui/store";
+import { useAui, useAuiState } from "@assistant-ui/store";
 import { STREAM_CONTROLLER, useChannel, useStream } from "@langchain/react";
 import type { Channel } from "@langchain/react";
 import type {
@@ -43,6 +43,10 @@ export const runConfigToSubmitOptions = (
     ? { config: { configurable: runConfig.custom } }
     : undefined;
 
+type NormalizedRunConfigOptions = NonNullable<
+  ReturnType<typeof runConfigToSubmitOptions>
+>;
+
 /**
  * Group the graph's accumulated `UIMessage`s by the assistant message they
  * belong to. Non-array state and entries without a parent link are dropped.
@@ -119,6 +123,7 @@ const useStreamThreadRuntime = (
 ) => {
   const { adapters, autoCancelPendingToolCalls, unstable_allowCancellation } =
     options;
+  const aui = useAui();
   const messagesKey = options.messagesKey ?? "messages";
   const uiStateKey = options.uiStateKey ?? "ui";
 
@@ -184,6 +189,67 @@ const useStreamThreadRuntime = (
   const streamRef = useRef(stream);
   streamRef.current = stream;
 
+  const activeRunConfigRef = useRef<
+    NormalizedRunConfigOptions["config"] | undefined
+  >(undefined);
+  const runConfigByMessageIdRef = useRef(
+    new Map<string, NormalizedRunConfigOptions["config"] | undefined>(),
+  );
+  const activeThreadIdRef = useRef(externalId);
+  const setActiveRunConfig = useCallback(
+    (runConfig: AppendMessage["runConfig"]) => {
+      activeRunConfigRef.current = runConfigToSubmitOptions(runConfig)?.config;
+    },
+    [],
+  );
+  const withActiveRunConfig = useCallback(
+    (submitOptions?: Record<string, unknown>) => {
+      if (submitOptions && "config" in submitOptions) return submitOptions;
+      if (activeRunConfigRef.current === undefined) return submitOptions;
+      return { ...submitOptions, config: activeRunConfigRef.current };
+    },
+    [],
+  );
+
+  useEffect(() => {
+    if (
+      activeThreadIdRef.current !== null &&
+      activeThreadIdRef.current !== externalId
+    ) {
+      activeRunConfigRef.current = undefined;
+      runConfigByMessageIdRef.current.clear();
+    }
+    activeThreadIdRef.current = externalId;
+  }, [externalId]);
+
+  useEffect(() => {
+    const messages = stream.messages as readonly LangChainBaseMessage[];
+    const owned = runConfigByMessageIdRef.current;
+    for (let i = messages.length - 1; i >= 0; i--) {
+      const message = messages.at(i);
+      if (
+        !message?.id ||
+        getMessageType(message) !== "ai" ||
+        !message.tool_calls?.length
+      ) {
+        continue;
+      }
+      if (owned.has(message.id)) return;
+      break;
+    }
+    for (const message of messages) {
+      if (
+        !message.id ||
+        getMessageType(message) !== "ai" ||
+        !message.tool_calls?.length ||
+        owned.has(message.id)
+      ) {
+        continue;
+      }
+      owned.set(message.id, activeRunConfigRef.current);
+    }
+  }, [stream.messages]);
+
   const visibleMessagesRef = useRef(visibleMessages);
   visibleMessagesRef.current = visibleMessages;
 
@@ -272,13 +338,21 @@ const useStreamThreadRuntime = (
         subgraphs: stream.subgraphs,
         stream,
         error: stream.error,
-        submit: stream.submit,
-        respond: stream.respond,
-        respondAll: stream.respondAll,
+        submit: (values, submitOptions) => {
+          const isResume = values == null || submitOptions?.command != null;
+          return stream.submit(
+            values,
+            isResume ? withActiveRunConfig(submitOptions) : submitOptions,
+          );
+        },
+        respond: (response, respondOptions) =>
+          stream.respond(response, withActiveRunConfig(respondOptions)),
+        respondAll: (responsesById, respondOptions) =>
+          stream.respondAll(responsesById, withActiveRunConfig(respondOptions)),
         values: stream.values,
         messagesKey,
       }),
-    [stream, messagesKey],
+    [stream, messagesKey, withActiveRunConfig],
   );
 
   const runtime = useExternalStoreRuntime({
@@ -296,6 +370,7 @@ const useStreamThreadRuntime = (
         return;
       }
 
+      setActiveRunConfig(msg.runConfig);
       const content = getMessageContent(msg);
       const cancellations =
         autoCancelPendingToolCalls !== false
@@ -309,30 +384,45 @@ const useStreamThreadRuntime = (
               status: "error" as const,
             }))
           : [];
-      await stream.submit(
+      // A null threadId is not a no-op for the SDK: it rebinds the controller
+      // away from its self-created thread and forces a fresh one, so the
+      // submit waits for initialization to produce an identity; core no
+      // longer holds appends on that barrier.
+      const { externalId } = await aui.threadListItem.initialize();
+      await streamRef.current.submit(
         { [messagesKey]: [...cancellations, { type: "human", content }] },
-        runConfigToSubmitOptions(msg.runConfig),
+        {
+          ...runConfigToSubmitOptions(msg.runConfig),
+          ...(externalId != null ? { threadId: externalId } : {}),
+        },
       );
     },
     onAddToolResult: async ({
+      messageId,
       toolCallId,
       toolName,
       result,
       isError,
       artifact,
     }) => {
-      await stream.submit({
-        [messagesKey]: [
-          {
-            type: "tool",
-            name: toolName,
-            tool_call_id: toolCallId,
-            content: JSON.stringify(result),
-            ...(artifact !== undefined && { artifact }),
-            status: isError ? "error" : "success",
-          },
-        ],
-      });
+      const runConfig = runConfigByMessageIdRef.current.has(messageId)
+        ? runConfigByMessageIdRef.current.get(messageId)
+        : activeRunConfigRef.current;
+      await stream.submit(
+        {
+          [messagesKey]: [
+            {
+              type: "tool",
+              name: toolName,
+              tool_call_id: toolCallId,
+              content: JSON.stringify(result),
+              ...(artifact !== undefined && { artifact }),
+              status: isError ? "error" : "success",
+            },
+          ],
+        },
+        runConfig === undefined ? undefined : { config: runConfig },
+      );
     },
     onReload: async (parentId, config) => {
       const stagedRun = getStagedRun(parentId);
@@ -353,6 +443,8 @@ const useStreamThreadRuntime = (
         } else {
           setStagedMessages(null);
         }
+        const runConfig = config.runConfig ?? stagedRun.runConfig;
+        setActiveRunConfig(runConfig);
         await stream.submit(
           {
             [messagesKey]: stagedRun.messages.map((message) => ({
@@ -361,7 +453,7 @@ const useStreamThreadRuntime = (
               content: message.content,
             })),
           },
-          runConfigToSubmitOptions(config.runConfig ?? stagedRun.runConfig),
+          runConfigToSubmitOptions(runConfig),
         );
         return;
       }
@@ -379,6 +471,7 @@ const useStreamThreadRuntime = (
         messagesKey,
       );
       if (!checkpointId) return;
+      setActiveRunConfig(config.runConfig);
       await s.submit(null, {
         forkFrom: checkpointId,
         ...runConfigToSubmitOptions(config.runConfig),
@@ -416,6 +509,7 @@ const useStreamThreadRuntime = (
       );
       if (!checkpointId) return;
       const content = getMessageContent(message);
+      setActiveRunConfig(message.runConfig);
       await s.submit(
         { [messagesKey]: [{ type: "human", content }] },
         {
@@ -427,6 +521,7 @@ const useStreamThreadRuntime = (
     onCancel:
       unstable_allowCancellation !== false
         ? async () => {
+            activeRunConfigRef.current = undefined;
             await stream.stop();
           }
         : undefined,