@assistant-ui/react-langchain 0.0.25 → 0.0.27

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;;;;;;;;;;;;;;;;;;;;;;;;;cAyjBF,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
@@ -44,6 +44,12 @@ const toStagedHumanMessage = (msg, id = generateId()) => ({
 	_getType: () => "human",
 	content: getMessageContent(msg)
 });
+const humanContentText = (content) => {
+	if (typeof content === "string") return content;
+	if (!Array.isArray(content)) return "";
+	return content.filter((part) => typeof part === "object" && part !== null && part.type === "text" && typeof part.text === "string").map((part) => part.text).join("");
+};
+const hasSameMessageContent = (a, b) => humanContentText(a.content) === humanContentText(b.content);
 const truncateLangChainBaseMessages = (threadMessages, parentId) => {
 	if (parentId === null) return [];
 	const parentIndex = threadMessages.findIndex((m) => m.id === parentId);
@@ -54,6 +60,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 +90,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);
@@ -92,16 +134,22 @@ const useStreamThreadRuntime = (options) => {
 	useEffect(() => {
 		if (stagedMessagesRef.current.size === 0) return;
 		const baseMessages = stagedBaseMessagesRef.current ?? stream.messages;
-		const baseMessageIds = new Set(baseMessages.flatMap((message) => message.id ? [message.id] : []));
 		const remainingStagedMessages = [];
-		const seenStagedIds = /* @__PURE__ */ new Set();
-		for (const message of visibleMessagesRef.current) {
-			if (!message.id || seenStagedIds.has(message.id)) continue;
-			if (baseMessageIds.has(message.id)) continue;
-			const staged = stagedMessagesRef.current.get(message.id);
-			if (!staged) continue;
-			remainingStagedMessages.push(staged.message);
-			seenStagedIds.add(message.id);
+		const matchedBaseMessageIndexes = /* @__PURE__ */ new Set();
+		const visibleStagedIds = new Set(visibleMessagesRef.current.flatMap((m) => m.id ? [m.id] : []));
+		for (const [id, staged] of stagedMessagesRef.current) {
+			if (!visibleStagedIds.has(id)) continue;
+			if (baseMessages.some((message, index) => {
+				if (matchedBaseMessageIndexes.has(index)) return false;
+				if (message.id === id) {
+					matchedBaseMessageIndexes.add(index);
+					return true;
+				}
+				if (!staged.reconcileOnEcho || index < staged.baseMessageCount || getMessageType(message) !== "human" || !hasSameMessageContent(message, staged.message)) return false;
+				matchedBaseMessageIndexes.add(index);
+				return true;
+			})) stagedMessagesRef.current.delete(id);
+			else remainingStagedMessages.push(staged.message);
 		}
 		if (remainingStagedMessages.length === 0) {
 			stagedBaseMessagesRef.current = null;
@@ -125,15 +173,27 @@ const useStreamThreadRuntime = (options) => {
 			runConfig: stagedMessagesRef.current.get(parentId).runConfig
 		};
 	};
-	const stageUserMessage = (msg) => {
+	const stageUserMessage = (msg, reconcileOnEcho = false) => {
 		const stagedMessage = toStagedHumanMessage(msg);
 		stagedMessagesRef.current.set(stagedMessage.id, {
 			message: stagedMessage,
-			runConfig: msg.runConfig
+			runConfig: msg.runConfig,
+			reconcileOnEcho,
+			baseMessageCount: streamRef.current.messages.length
 		});
 		const nextMessages = [...visibleMessagesRef.current, stagedMessage];
 		visibleMessagesRef.current = nextMessages;
 		setStagedMessages(nextMessages);
+		return stagedMessage;
+	};
+	const removeStagedMessage = (id) => {
+		if (!stagedMessagesRef.current.delete(id)) return;
+		const nextMessages = visibleMessagesRef.current.filter((message) => message.id !== id);
+		visibleMessagesRef.current = nextMessages;
+		if (stagedMessagesRef.current.size === 0) {
+			stagedBaseMessagesRef.current = null;
+			setStagedMessages(null);
+		} else setStagedMessages(nextMessages);
 	};
 	const extras = useMemo(() => langChainExtras.provide({
 		interrupt: stream.interrupt,
@@ -143,12 +203,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 +230,8 @@ const useStreamThreadRuntime = (options) => {
 				stageUserMessage(msg);
 				return;
 			}
+			const stagedMessageId = stageUserMessage(msg, true).id;
+			setActiveRunConfig(msg.runConfig);
 			const content = getMessageContent(msg);
 			const cancellations = autoCancelPendingToolCalls !== false ? getPendingToolCalls(streamRef.current.messages).map((t) => ({
 				type: "tool",
@@ -171,12 +240,23 @@ const useStreamThreadRuntime = (options) => {
 				content: JSON.stringify({ cancelled: true }),
 				status: "error"
 			})) : [];
-			await stream.submit({ [messagesKey]: [...cancellations, {
-				type: "human",
-				content
-			}] }, runConfigToSubmitOptions(msg.runConfig));
+			try {
+				const { externalId } = await aui.threadListItem.initialize();
+				await streamRef.current.submit({ [messagesKey]: [...cancellations, {
+					id: stagedMessageId,
+					type: "human",
+					content
+				}] }, {
+					...runConfigToSubmitOptions(msg.runConfig),
+					...externalId != null ? { threadId: externalId } : {}
+				});
+			} catch (error) {
+				removeStagedMessage(stagedMessageId);
+				throw error;
+			}
 		},
-		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 +264,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 +281,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 +295,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)
@@ -224,7 +307,9 @@ const useStreamThreadRuntime = (options) => {
 				const stagedMessage = toStagedHumanMessage(message);
 				stagedMessagesRef.current.set(stagedMessage.id, {
 					message: stagedMessage,
-					runConfig: message.runConfig
+					runConfig: message.runConfig,
+					reconcileOnEcho: false,
+					baseMessageCount: 0
 				});
 				stagedBaseMessagesRef.current = truncated;
 				const nextMessages = [...truncated, stagedMessage];
@@ -238,6 +323,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 +333,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 humanContentText = (content: LangChainBaseMessage[\"content\"]) => {\n  if (typeof content === \"string\") return content;\n  if (!Array.isArray(content)) return \"\";\n  return content\n    .filter(\n      (part): part is { type: \"text\"; text: string } =>\n        typeof part === \"object\" &&\n        part !== null &&\n        part.type === \"text\" &&\n        typeof part.text === \"string\",\n    )\n    .map((part) => part.text)\n    .join(\"\");\n};\n\nconst hasSameMessageContent = (\n  a: LangChainBaseMessage,\n  b: LangChainBaseMessage,\n) => humanContentText(a.content) === humanContentText(b.content);\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        reconcileOnEcho: boolean;\n        baseMessageCount: number;\n      }\n    >(),\n  );\n  const stagedBaseMessagesRef = useRef<LangChainBaseMessage[] | null>(null);\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 remainingStagedMessages: LangChainBaseMessage[] = [];\n    const matchedBaseMessageIndexes = new Set<number>();\n    const visibleStagedIds = new Set(\n      visibleMessagesRef.current.flatMap((m) => (m.id ? [m.id] : [])),\n    );\n    for (const [id, staged] of stagedMessagesRef.current) {\n      if (!visibleStagedIds.has(id)) continue;\n      const echoed = baseMessages.some((message, index) => {\n        if (matchedBaseMessageIndexes.has(index)) return false;\n        if (message.id === id) {\n          matchedBaseMessageIndexes.add(index);\n          return true;\n        }\n        if (\n          !staged.reconcileOnEcho ||\n          index < staged.baseMessageCount ||\n          getMessageType(message) !== \"human\" ||\n          !hasSameMessageContent(message, staged.message)\n        ) {\n          return false;\n        }\n        matchedBaseMessageIndexes.add(index);\n        return true;\n      });\n      if (echoed) stagedMessagesRef.current.delete(id);\n      else remainingStagedMessages.push(staged.message);\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, reconcileOnEcho = false) => {\n    const stagedMessage = toStagedHumanMessage(msg);\n    stagedMessagesRef.current.set(stagedMessage.id, {\n      message: stagedMessage,\n      runConfig: msg.runConfig,\n      reconcileOnEcho,\n      baseMessageCount: streamRef.current.messages.length,\n    });\n    const nextMessages = [...visibleMessagesRef.current, stagedMessage];\n    visibleMessagesRef.current = nextMessages;\n    setStagedMessages(nextMessages);\n    return stagedMessage;\n  };\n\n  const removeStagedMessage = (id: string) => {\n    if (!stagedMessagesRef.current.delete(id)) return;\n    const nextMessages = visibleMessagesRef.current.filter(\n      (message) => message.id !== id,\n    );\n    visibleMessagesRef.current = nextMessages;\n    if (stagedMessagesRef.current.size === 0) {\n      stagedBaseMessagesRef.current = null;\n      setStagedMessages(null);\n    } else {\n      setStagedMessages(nextMessages);\n    }\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      const stagedMessage = stageUserMessage(msg, true);\n      const stagedMessageId = stagedMessage.id;\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      try {\n        const { externalId } = await aui.threadListItem.initialize();\n        await streamRef.current.submit(\n          {\n            [messagesKey]: [\n              ...cancellations,\n              {\n                id: stagedMessageId,\n                type: \"human\",\n                content,\n              },\n            ],\n          },\n          {\n            ...runConfigToSubmitOptions(msg.runConfig),\n            ...(externalId != null ? { threadId: externalId } : {}),\n          },\n        );\n      } catch (error) {\n        removeStagedMessage(stagedMessageId);\n        throw error;\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          reconcileOnEcho: false,\n          baseMessageCount: 0,\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,oBAAoB,YAA6C;CACrE,IAAI,OAAO,YAAY,UAAU,OAAO;CACxC,IAAI,CAAC,MAAM,QAAQ,OAAO,GAAG,OAAO;CACpC,OAAO,QACJ,QACE,SACC,OAAO,SAAS,YAChB,SAAS,QACT,KAAK,SAAS,UACd,OAAO,KAAK,SAAS,QACzB,CAAC,CACA,KAAK,SAAS,KAAK,IAAI,CAAC,CACxB,KAAK,EAAE;AACZ;AAEA,MAAM,yBACJ,GACA,MACG,iBAAiB,EAAE,OAAO,MAAM,iBAAiB,EAAE,OAAO;AAE/D,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,IAQF,CACJ;CACA,MAAM,wBAAwB,OAAsC,IAAI;CACxE,gBAAgB;EACd,IAAI,kBAAkB,QAAQ,SAAS,GAAG;EAG1C,MAAM,eACJ,sBAAsB,WACrB,OAAO;EACV,MAAM,0BAAkD,CAAC;EACzD,MAAM,4CAA4B,IAAI,IAAY;EAClD,MAAM,mBAAmB,IAAI,IAC3B,mBAAmB,QAAQ,SAAS,MAAO,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,CAAE,CAChE;EACA,KAAK,MAAM,CAAC,IAAI,WAAW,kBAAkB,SAAS;GACpD,IAAI,CAAC,iBAAiB,IAAI,EAAE,GAAG;GAkB/B,IAjBe,aAAa,MAAM,SAAS,UAAU;IACnD,IAAI,0BAA0B,IAAI,KAAK,GAAG,OAAO;IACjD,IAAI,QAAQ,OAAO,IAAI;KACrB,0BAA0B,IAAI,KAAK;KACnC,OAAO;IACT;IACA,IACE,CAAC,OAAO,mBACR,QAAQ,OAAO,oBACf,eAAe,OAAO,MAAM,WAC5B,CAAC,sBAAsB,SAAS,OAAO,OAAO,GAE9C,OAAO;IAET,0BAA0B,IAAI,KAAK;IACnC,OAAO;GACT,CACS,GAAG,kBAAkB,QAAQ,OAAO,EAAE;QAC1C,wBAAwB,KAAK,OAAO,OAAO;EAClD;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,KAAoB,kBAAkB,UAAU;EACxE,MAAM,gBAAgB,qBAAqB,GAAG;EAC9C,kBAAkB,QAAQ,IAAI,cAAc,IAAI;GAC9C,SAAS;GACT,WAAW,IAAI;GACf;GACA,kBAAkB,UAAU,QAAQ,SAAS;EAC/C,CAAC;EACD,MAAM,eAAe,CAAC,GAAG,mBAAmB,SAAS,aAAa;EAClE,mBAAmB,UAAU;EAC7B,kBAAkB,YAAY;EAC9B,OAAO;CACT;CAEA,MAAM,uBAAuB,OAAe;EAC1C,IAAI,CAAC,kBAAkB,QAAQ,OAAO,EAAE,GAAG;EAC3C,MAAM,eAAe,mBAAmB,QAAQ,QAC7C,YAAY,QAAQ,OAAO,EAC9B;EACA,mBAAmB,UAAU;EAC7B,IAAI,kBAAkB,QAAQ,SAAS,GAAG;GACxC,sBAAsB,UAAU;GAChC,kBAAkB,IAAI;EACxB,OACE,kBAAkB,YAAY;CAElC;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;CAgMA,OA9LgB,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;GAGA,MAAM,kBADgB,iBAAiB,KAAK,IACR,CAAC,CAAC;GACtC,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,IAAI;IACF,MAAM,EAAE,eAAe,MAAM,IAAI,eAAe,WAAW;IAC3D,MAAM,UAAU,QAAQ,OACtB,GACG,cAAc,CACb,GAAG,eACH;KACE,IAAI;KACJ,MAAM;KACN;IACF,CACF,EACF,GACA;KACE,GAAG,yBAAyB,IAAI,SAAS;KACzC,GAAI,cAAc,OAAO,EAAE,UAAU,WAAW,IAAI,CAAC;IACvD,CACF;GACF,SAAS,OAAO;IACd,oBAAoB,eAAe;IACnC,MAAM;GACR;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;KACnB,iBAAiB;KACjB,kBAAkB;IACpB,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.27",
   "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.15",
+    "@assistant-ui/store": "^0.3.10",
     "assistant-cloud": "*",
-    "assistant-stream": "^0.3.36"
+    "assistant-stream": "^0.3.39"
   },
   "peerDependencies": {
     "@langchain/react": "^1.0.2",
@@ -45,15 +45,15 @@
     }
   },
   "devDependencies": {
-    "@langchain/core": "^1.2.5",
-    "@langchain/langgraph-sdk": "^1.9.28",
-    "@langchain/react": "^1.0.29",
+    "@langchain/core": "^1.2.8",
+    "@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",