@assistant-ui/react-langchain 0.0.6 → 0.0.8

package/dist/useStreamRuntime.js

@@ -1,216 +1,238 @@
 "use client";
+import { convertLangChainBaseMessage, getMessageType } from "./convertMessages.js";
 import { useMemo, useRef, useState } from "react";
-import { useCloudThreadListAdapter, useExternalStoreRuntime, useExternalMessageConverter, useRemoteThreadListRuntime, useToolInvocations, } from "@assistant-ui/core/react";
+import { useCloudThreadListAdapter, useExternalMessageConverter, useExternalStoreRuntime, useRemoteThreadListRuntime, useToolInvocations } from "@assistant-ui/core/react";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import { useStream } from "@langchain/react";
-import { convertLangChainBaseMessage } from "./convertMessages.js";
+//#region src/useStreamRuntime.ts
 const symbolLangChainRuntimeExtras = Symbol("langchain-runtime-extras");
 const asLangChainRuntimeExtras = (extras) => {
-    if (typeof extras !== "object" ||
-        extras == null ||
-        !(symbolLangChainRuntimeExtras in extras))
-        throw new Error("This method can only be called when you are using useStreamRuntime");
-    return extras;
+	if (typeof extras !== "object" || extras == null || !(symbolLangChainRuntimeExtras in extras)) throw new Error("This method can only be called when you are using useStreamRuntime");
+	return extras;
+};
+const getPendingToolCalls = (messages) => {
+	const pending = /* @__PURE__ */ new Map();
+	for (const m of messages) {
+		const type = getMessageType(m);
+		if (type === "ai") for (const tc of m.tool_calls ?? []) pending.set(tc.id, tc);
+		else if (type === "tool" && m.tool_call_id) pending.delete(m.tool_call_id);
+	}
+	return [...pending.values()];
 };
 const getMessageContent = (msg) => {
-    const allContent = [
-        ...msg.content,
-        ...(msg.attachments?.flatMap((a) => a.content) ?? []),
-    ];
-    const hasNonText = allContent.some((part) => part.type === "file" || part.type === "image");
-    const hasText = allContent.some((part) => part.type === "text");
-    if (hasNonText && !hasText) {
-        allContent.unshift({ type: "text", text: " " });
-    }
-    const content = allContent.map((part) => {
-        const type = part.type;
-        switch (type) {
-            case "text":
-                return { type: "text", text: part.text };
-            case "image":
-                return { type: "image_url", image_url: { url: part.image } };
-            case "file":
-                return {
-                    type: "file",
-                    data: part.data,
-                    mime_type: part.mimeType,
-                    metadata: { filename: part.filename ?? "file" },
-                    source_type: "base64",
-                };
-            case "tool-call":
-                throw new Error("Tool call appends are not supported.");
-            default: {
-                const _exhaustiveCheck = type;
-                throw new Error(`Unsupported append message part type: ${_exhaustiveCheck}`);
-            }
-        }
-    });
-    if (content.length === 1 && content[0]?.type === "text") {
-        return content[0].text ?? "";
-    }
-    return content;
+	const allContent = [...msg.content, ...msg.attachments?.flatMap((a) => a.content) ?? []];
+	const hasNonText = allContent.some((part) => part.type === "file" || part.type === "image");
+	const hasText = allContent.some((part) => part.type === "text");
+	if (hasNonText && !hasText) allContent.unshift({
+		type: "text",
+		text: " "
+	});
+	const content = allContent.map((part) => {
+		const type = part.type;
+		switch (type) {
+			case "text": return {
+				type: "text",
+				text: part.text
+			};
+			case "image": return {
+				type: "image_url",
+				image_url: { url: part.image }
+			};
+			case "file": return {
+				type: "file",
+				data: part.data,
+				mime_type: part.mimeType,
+				metadata: { filename: part.filename ?? "file" },
+				source_type: "base64"
+			};
+			case "tool-call": throw new Error("Tool call appends are not supported.");
+			default: throw new Error(`Unsupported append message part type: ${type}`);
+		}
+	});
+	if (content.length === 1 && content[0]?.type === "text") return content[0].text ?? "";
+	return content;
+};
+const useStreamThreadRuntime = (options) => {
+	const { adapters, autoCancelPendingToolCalls, unstable_allowCancellation } = options;
+	const messagesKey = options.messagesKey ?? "messages";
+	options.threadId = useAuiState((s) => s.threadListItem.externalId);
+	const stream = useStream(options);
+	const [toolStatuses, setToolStatuses] = useState({});
+	const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
+	const effectiveIsRunning = stream.isLoading || hasExecutingTools;
+	const threadMessages = useExternalMessageConverter({
+		callback: convertLangChainBaseMessage,
+		messages: stream.messages,
+		isRunning: effectiveIsRunning
+	});
+	const [runtimeRef] = useState(() => ({ get current() {
+		return runtime;
+	} }));
+	const streamRef = useRef(stream);
+	streamRef.current = stream;
+	const toolInvocations = useToolInvocations({
+		state: {
+			messages: threadMessages,
+			isRunning: effectiveIsRunning
+		},
+		getTools: () => runtimeRef.current.thread.getModelContext().tools,
+		onResult: (command) => {
+			if (command.type === "add-tool-result") streamRef.current.submit({ [messagesKey]: [{
+				type: "tool",
+				name: command.toolName,
+				tool_call_id: command.toolCallId,
+				content: JSON.stringify(command.result),
+				status: command.isError ? "error" : "success"
+			}] }, {});
+		},
+		setToolStatuses
+	});
+	const runtime = useExternalStoreRuntime({
+		isRunning: effectiveIsRunning,
+		messages: threadMessages,
+		adapters,
+		extras: useMemo(() => ({
+			[symbolLangChainRuntimeExtras]: true,
+			interrupt: stream.interrupt,
+			interrupts: stream.interrupts,
+			submit: stream.submit,
+			values: stream.values,
+			messagesKey
+		}), [
+			stream.interrupt,
+			stream.interrupts,
+			stream.submit,
+			stream.values,
+			messagesKey
+		]),
+		onNew: async (msg) => {
+			await toolInvocations.abort();
+			const content = getMessageContent(msg);
+			const cancellations = autoCancelPendingToolCalls !== false ? getPendingToolCalls(streamRef.current.messages).map((t) => ({
+				type: "tool",
+				name: t.name,
+				tool_call_id: t.id,
+				content: JSON.stringify({ cancelled: true }),
+				status: "error"
+			})) : [];
+			await stream.submit({ [messagesKey]: [...cancellations, {
+				type: "human",
+				content
+			}] });
+		},
+		onAddToolResult: async ({ toolCallId, toolName, result, isError, artifact }) => {
+			await stream.submit({ [messagesKey]: [{
+				type: "tool",
+				name: toolName,
+				tool_call_id: toolCallId,
+				content: JSON.stringify(result),
+				...artifact !== void 0 && { artifact },
+				status: isError ? "error" : "success"
+			}] });
+		},
+		onCancel: unstable_allowCancellation !== false ? async () => {
+			await stream.stop();
+			await toolInvocations.abort();
+		} : void 0
+	});
+	return runtime;
 };
-const useStreamThreadRuntime = ({ adapters, messagesKey = "messages", ...streamOptions }) => {
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const externalId = useAuiState((s) => s.threadListItem.externalId);
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const stream = useStream({
-        ...streamOptions,
-        threadId: externalId,
-    });
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const [toolStatuses, setToolStatuses] = useState({});
-    const hasExecutingTools = Object.values(toolStatuses).some((s) => s?.type === "executing");
-    const effectiveIsRunning = stream.isLoading || hasExecutingTools;
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const threadMessages = useExternalMessageConverter({
-        callback: convertLangChainBaseMessage,
-        messages: stream.messages,
-        isRunning: effectiveIsRunning,
-    });
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const [runtimeRef] = useState(() => ({
-        get current() {
-            return runtime;
-        },
-    }));
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const streamRef = useRef(stream);
-    streamRef.current = stream;
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const toolInvocations = useToolInvocations({
-        state: {
-            messages: threadMessages,
-            isRunning: effectiveIsRunning,
-        },
-        getTools: () => runtimeRef.current.thread.getModelContext().tools,
-        onResult: (command) => {
-            if (command.type === "add-tool-result") {
-                void streamRef.current.submit({
-                    [messagesKey]: [
-                        {
-                            type: "tool",
-                            name: command.toolName,
-                            tool_call_id: command.toolCallId,
-                            content: JSON.stringify(command.result),
-                            status: command.isError ? "error" : "success",
-                        },
-                    ],
-                }, {});
-            }
-        },
-        setToolStatuses,
-    });
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const extras = useMemo(() => ({
-        [symbolLangChainRuntimeExtras]: true,
-        interrupt: stream.interrupt,
-        interrupts: stream.interrupts,
-        submit: stream.submit,
-        values: stream.values,
-    }), [stream.interrupt, stream.interrupts, stream.submit, stream.values]);
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const runtime = useExternalStoreRuntime({
-        isRunning: effectiveIsRunning,
-        messages: threadMessages,
-        adapters,
-        extras,
-        onNew: async (msg) => {
-            await toolInvocations.abort();
-            const content = getMessageContent(msg);
-            await stream.submit({
-                [messagesKey]: [{ type: "human", content }],
-            });
-        },
-        onAddToolResult: async ({ 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",
-                    },
-                ],
-            });
-        },
-        onCancel: async () => {
-            await stream.stop();
-            await toolInvocations.abort();
-        },
-    });
-    return runtime;
+/**
+* Creates an assistant-ui runtime backed by LangChain's `useStream` hook.
+* Accepts the same options as `useStream` from `@langchain/react`, plus
+* `cloud` and `adapters`.
+*
+* @example
+* ```tsx
+* import { useStreamRuntime } from "@assistant-ui/react-langchain";
+* import { AssistantRuntimeProvider, Thread } from "@assistant-ui/react";
+*
+* function App() {
+*   const runtime = useStreamRuntime({
+*     assistantId: "agent",
+*     apiUrl: "http://localhost:2024",
+*   });
+*
+*   return (
+*     <AssistantRuntimeProvider runtime={runtime}>
+*       <Thread />
+*     </AssistantRuntimeProvider>
+*   );
+* }
+* ```
+*/
+const useStreamRuntime = (rawOptions) => {
+	const { cloud, unstable_threadListAdapter, create, delete: deleteFn, ...options } = rawOptions;
+	const optionsRef = useRef(options);
+	optionsRef.current = options;
+	const cloudAdapter = useCloudThreadListAdapter({
+		cloud,
+		create,
+		delete: deleteFn
+	});
+	return useRemoteThreadListRuntime({
+		runtimeHook: function RuntimeHook() {
+			return useStreamThreadRuntime(optionsRef.current);
+		},
+		adapter: unstable_threadListAdapter ?? cloudAdapter,
+		allowNesting: true
+	});
+};
+/**
+* Read the current LangGraph interrupt state from the runtime extras.
+*/
+const useLangChainInterruptState = () => {
+	return useAuiState((s) => {
+		const extras = s.thread.extras;
+		if (!extras) return void 0;
+		return asLangChainRuntimeExtras(extras).interrupt;
+	});
 };
 /**
- * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.
- * Accepts the same options as `useStream` from `@langchain/react`, plus
- * `cloud` and `adapters`.
- *
- * @example
- * ```tsx
- * import { useStreamRuntime } from "@assistant-ui/react-langchain";
- * import { AssistantRuntimeProvider, Thread } from "@assistant-ui/react";
- *
- * function App() {
- *   const runtime = useStreamRuntime({
- *     assistantId: "agent",
- *     apiUrl: "http://localhost:2024",
- *   });
- *
- *   return (
- *     <AssistantRuntimeProvider runtime={runtime}>
- *       <Thread />
- *     </AssistantRuntimeProvider>
- *   );
- * }
- * ```
- */
-export const useStreamRuntime = ({ cloud, ...options }) => {
-    const optionsRef = useRef(options);
-    optionsRef.current = options;
-    const cloudAdapter = useCloudThreadListAdapter({ cloud });
-    return useRemoteThreadListRuntime({
-        runtimeHook: function RuntimeHook() {
-            // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-            return useStreamThreadRuntime(optionsRef.current);
-        },
-        adapter: cloudAdapter,
-        allowNesting: true,
-    });
+* Returns a function to submit raw state updates to the LangGraph agent,
+* bypassing the normal message flow. Useful for sending interrupt resume
+* commands.
+*/
+const useLangChainSubmit = () => {
+	const aui = useAui();
+	return (values, options) => {
+		const extras = aui.thread().getState().extras;
+		const { submit } = asLangChainRuntimeExtras(extras);
+		return submit(values, options);
+	};
 };
 /**
- * Read the current LangGraph interrupt state from the runtime extras.
- */
-export const useLangChainInterruptState = () => {
-    return useAuiState((s) => {
-        const extras = s.thread.extras;
-        if (!extras)
-            return undefined;
-        return asLangChainRuntimeExtras(extras).interrupt;
-    });
+* Submit a list of LangChain-shaped messages on the current thread.
+* Parity helper for migrating from `useLangGraphSend`. Routes to
+* `useStream().submit({ [messagesKey]: messages }, options)`.
+*/
+const useLangChainSend = () => {
+	const aui = useAui();
+	return (messages, options) => {
+		const { submit, messagesKey } = asLangChainRuntimeExtras(aui.thread().getState().extras);
+		return submit({ [messagesKey]: messages }, options);
+	};
 };
 /**
- * Returns a function to submit raw state updates to the LangGraph agent,
- * bypassing the normal message flow. Useful for sending interrupt resume
- * commands.
- */
-export const useLangChainSubmit = () => {
-    const aui = useAui();
-    return (values, options) => {
-        const extras = aui.thread().getState().extras;
-        const { submit } = asLangChainRuntimeExtras(extras);
-        return submit(values, options);
-    };
+* Submit a `useStream` command (e.g. interrupt resume). Parity helper
+* for migrating from `useLangGraphSendCommand`. Note that v1's command
+* shape (`{ resume?, goto?, update? }`) differs from the legacy
+* `{ resume: string }` form — to carry a payload, use the input or
+* `stream.respond` instead.
+*/
+const useLangChainSendCommand = () => {
+	const submit = useLangChainSubmit();
+	return (command) => submit(null, { command });
 };
-export function useLangChainState(key, defaultValue) {
-    return useAuiState((s) => {
-        const extras = s.thread.extras;
-        if (!extras)
-            return defaultValue;
-        const value = asLangChainRuntimeExtras(extras).values[key];
-        return value !== undefined ? value : defaultValue;
-    });
+function useLangChainState(key, defaultValue) {
+	return useAuiState((s) => {
+		const extras = s.thread.extras;
+		if (!extras) return defaultValue;
+		const value = asLangChainRuntimeExtras(extras).values[key];
+		return value !== void 0 ? value : defaultValue;
+	});
 }
+//#endregion
+export { useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useStreamRuntime };
+
 //# sourceMappingURL=useStreamRuntime.js.map

package/dist/useStreamRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamRuntime.js","sourceRoot":"","sources":["../src/useStreamRuntime.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAOlD,OAAO,EACL,yBAAyB,EACzB,uBAAuB,EACvB,2BAA2B,EAC3B,0BAA0B,EAC1B,kBAAkB,GAEnB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAE1D,OAAO,EAAE,SAAS,EAAyB,MAAM,kBAAkB,CAAC;AAEpE,OAAO,EAAE,2BAA2B,EAAE,6BAA0B;AAEhE,MAAM,4BAA4B,GAAG,MAAM,CAAC,0BAA0B,CAAC,CAAC;AAaxE,MAAM,wBAAwB,GAAG,CAAC,MAAe,EAA0B,EAAE;IAC3E,IACE,OAAO,MAAM,KAAK,QAAQ;QAC1B,MAAM,IAAI,IAAI;QACd,CAAC,CAAC,4BAA4B,IAAI,MAAM,CAAC;QAEzC,MAAM,IAAI,KAAK,CACb,oEAAoE,CACrE,CAAC;IACJ,OAAO,MAAgC,CAAC;AAC1C,CAAC,CAAC;AAeF,MAAM,iBAAiB,GAAG,CAAC,GAAkB,EAAE,EAAE;IAC/C,MAAM,UAAU,GAAG;QACjB,GAAG,GAAG,CAAC,OAAO;QACd,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;KACtD,CAAC;IAEF,MAAM,UAAU,GAAG,UAAU,CAAC,IAAI,CAChC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,CACxD,CAAC;IACF,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;IAChE,IAAI,UAAU,IAAI,CAAC,OAAO,EAAE,CAAC;QAC3B,UAAU,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACtC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACvB,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,MAAM;gBACT,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACpD,KAAK,OAAO;gBACV,OAAO,EAAE,IAAI,EAAE,WAAoB,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC;YACxE,KAAK,MAAM;gBACT,OAAO;oBACL,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,SAAS,EAAE,IAAI,CAAC,QAAQ;oBACxB,QAAQ,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,MAAM,EAAE;oBAC/C,WAAW,EAAE,QAAiB;iBAC/B,CAAC;YACJ,KAAK,WAAW;gBACd,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAC1D,OAAO,CAAC,CAAC,CAAC;gBACR,MAAM,gBAAgB,GACpB,IAAI,CAAC;gBACP,MAAM,IAAI,KAAK,CACb,yCAAyC,gBAAgB,EAAE,CAC5D,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;QACxD,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;IAC/B,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,sBAAsB,GAAG,CAAC,EAC9B,QAAQ,EACR,WAAW,GAAG,UAAU,EACxB,GAAG,aAAa,EACuB,EAAE,EAAE;IAC3C,6FAA6F;IAC7F,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,UAAU,CAEzD,CAAC;IAET,6FAA6F;IAC7F,MAAM,MAAM,GAAG,SAAS,CAAC;QACvB,GAAG,aAAa;QAChB,QAAQ,EAAE,UAAU;KACrB,CAAC,CAAC;IAEH,6FAA6F;IAC7F,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAE9C,EAAE,CAAC,CAAC;IACN,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CACxD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,KAAK,WAAW,CAC/B,CAAC;IACF,MAAM,kBAAkB,GAAG,MAAM,CAAC,SAAS,IAAI,iBAAiB,CAAC;IAEjE,6FAA6F;IAC7F,MAAM,cAAc,GAAG,2BAA2B,CAAC;QACjD,QAAQ,EAAE,2BAA2B;QACrC,QAAQ,EAAE,MAAM,CAAC,QAAkC;QACnD,SAAS,EAAE,kBAAkB;KAC9B,CAAC,CAAC;IAEH,6FAA6F;IAC7F,MAAM,CAAC,UAAU,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC;QACnC,IAAI,OAAO;YACT,OAAO,OAAO,CAAC;QACjB,CAAC;KACF,CAAC,CAAC,CAAC;IAEJ,6FAA6F;IAC7F,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;IACjC,SAAS,CAAC,OAAO,GAAG,MAAM,CAAC;IAE3B,6FAA6F;IAC7F,MAAM,eAAe,GAAG,kBAAkB,CAAC;QACzC,KAAK,EAAE;YACL,QAAQ,EAAE,cAAc;YACxB,SAAS,EAAE,kBAAkB;SAC9B;QACD,QAAQ,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,KAAK;QACjE,QAAQ,EAAE,CAAC,OAAO,EAAE,EAAE;YACpB,IAAI,OAAO,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;gBACvC,KAAK,SAAS,CAAC,OAAO,CAAC,MAAM,CAC3B;oBACE,CAAC,WAAW,CAAC,EAAE;wBACb;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,OAAO,CAAC,QAAQ;4BACtB,YAAY,EAAE,OAAO,CAAC,UAAU;4BAChC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC;4BACvC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;yBAC9C;qBACF;iBACF,EACD,EAAE,CACH,CAAC;YACJ,CAAC;QACH,CAAC;QACD,eAAe;KAChB,CAAC,CAAC;IAEH,6FAA6F;IAC7F,MAAM,MAAM,GAAG,OAAO,CACpB,GAA2B,EAAE,CAAC,CAAC;QAC7B,CAAC,4BAA4B,CAAC,EAAE,IAAI;QACpC,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC,EACF,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CACpE,CAAC;IAEF,6FAA6F;IAC7F,MAAM,OAAO,GAAG,uBAAuB,CAAC;QACtC,SAAS,EAAE,kBAAkB;QAC7B,QAAQ,EAAE,cAAc;QACxB,QAAQ;QACR,MAAM;QACN,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACnB,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;YAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC;YACvC,MAAM,MAAM,CAAC,MAAM,CAAC;gBAClB,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;aAC5C,CAAC,CAAC;QACL,CAAC;QACD,eAAe,EAAE,KAAK,EAAE,EACtB,UAAU,EACV,QAAQ,EACR,MAAM,EACN,OAAO,EACP,QAAQ,GACT,EAAE,EAAE;YACH,MAAM,MAAM,CAAC,MAAM,CAAC;gBAClB,CAAC,WAAW,CAAC,EAAE;oBACb;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,QAAQ;wBACd,YAAY,EAAE,UAAU;wBACxB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;wBAC/B,GAAG,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,CAAC;wBAC3C,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;qBACtC;iBACF;aACF,CAAC,CAAC;QACL,CAAC;QACD,QAAQ,EAAE,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YACpB,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;QAChC,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,EAC/B,KAAK,EACL,GAAG,OAAO,EACc,EAAE,EAAE;IAC5B,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;IACnC,UAAU,CAAC,OAAO,GAAG,OAAO,CAAC;IAE7B,MAAM,YAAY,GAAG,yBAAyB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1D,OAAO,0BAA0B,CAAC;QAChC,WAAW,EAAE,SAAS,WAAW;YAC/B,6FAA6F;YAC7F,OAAO,sBAAsB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,EAAE,YAAY;QACrB,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;AACL,CAAC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,GAAG,EAAE;IAC7C,OAAO,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;QAC/B,IAAI,CAAC,MAAM;YAAE,OAAO,SAAS,CAAC;QAC9B,OAAO,wBAAwB,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,GAAG,EAAE;IACrC,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IACrB,OAAO,CACL,MAAkD,EAClD,OAAiC,EACjC,EAAE;QACF,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC;QAC9C,MAAM,EAAE,MAAM,EAAE,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;QACpD,OAAO,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,CAAC,CAAC;AACJ,CAAC,CAAC;AAeF,MAAM,UAAU,iBAAiB,CAC/B,GAAW,EACX,YAAgB;IAEhB,OAAO,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;QAC/B,IAAI,CAAC,MAAM;YAAE,OAAO,YAAY,CAAC;QACjC,MAAM,KAAK,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,GAAG,CAAkB,CAAC;QAC5E,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC"}
+{"version":3,"file":"useStreamRuntime.js","names":["_exhaustiveCheck"],"sources":["../src/useStreamRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useMemo, useRef, useState } from \"react\";\nimport type {\n  AppendMessage,\n  AttachmentAdapter,\n  FeedbackAdapter,\n  RemoteThreadListAdapter,\n  SpeechSynthesisAdapter,\n} from \"@assistant-ui/core\";\nimport {\n  useCloudThreadListAdapter,\n  useExternalStoreRuntime,\n  useExternalMessageConverter,\n  useRemoteThreadListRuntime,\n  useToolInvocations,\n  type ToolExecutionStatus,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport { useStream, type UseStreamOptions } from \"@langchain/react\";\nimport type { LangChainBaseMessage, LangChainToolCall } from \"./types\";\nimport { convertLangChainBaseMessage, getMessageType } from \"./convertMessages\";\n\nconst symbolLangChainRuntimeExtras = Symbol(\"langchain-runtime-extras\");\n\ntype LangChainRuntimeExtras = {\n  [symbolLangChainRuntimeExtras]: true;\n  interrupt: { value?: unknown } | undefined;\n  interrupts: readonly { value?: unknown }[];\n  submit: (\n    values: Record<string, unknown> | null | undefined,\n    options?: Record<string, unknown>,\n  ) => Promise<void>;\n  values: Record<string, unknown>;\n  messagesKey: string;\n};\n\nconst asLangChainRuntimeExtras = (extras: unknown): LangChainRuntimeExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolLangChainRuntimeExtras in extras)\n  )\n    throw new Error(\n      \"This method can only be called when you are using useStreamRuntime\",\n    );\n  return extras as LangChainRuntimeExtras;\n};\n\ntype LangChainRuntimeExtraOptions = {\n  cloud?: AssistantCloud | undefined;\n  adapters?:\n    | {\n        attachments?: AttachmentAdapter | undefined;\n        speech?: SpeechSynthesisAdapter | undefined;\n        feedback?: FeedbackAdapter | undefined;\n      }\n    | undefined;\n  /**\n   * When the user sends a new message while previous tool calls are\n   * still pending, automatically submit `tool` messages that cancel\n   * them so the agent's tool-call accounting stays consistent.\n   * Defaults to `true`.\n   */\n  autoCancelPendingToolCalls?: boolean | undefined;\n  /**\n   * Routes the Cancel button's click to `useStream().stop()`. On by\n   * default. Pass `false` to disable the Cancel button.\n   */\n  unstable_allowCancellation?: boolean | undefined;\n  /**\n   * Custom `RemoteThreadListAdapter`. When provided, replaces the\n   * cloud-backed thread list adapter.\n   */\n  unstable_threadListAdapter?: RemoteThreadListAdapter | undefined;\n  /** Custom thread-creation hook, forwarded to the cloud adapter. */\n  create?: (() => Promise<{ externalId: string | undefined }>) | undefined;\n  /** Custom thread-deletion hook, forwarded to the cloud adapter. */\n  delete?: ((threadId: string) => Promise<void>) | undefined;\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\n// Distribute the intersection through the union arms of `UseStreamOptions`\n// (`AgentServerOptions | CustomAdapterOptions`). Writing `UseStreamOptions & X`\n// directly collapses arm tracking, so `Omit<…, \"cloud\">` and the like would\n// produce a flattened structural type that no longer matches either arm.\nexport type UseStreamRuntimeOptions = UseStreamOptions extends infer O\n  ? O extends UseStreamOptions\n    ? O & LangChainRuntimeExtraOptions\n    : never\n  : never;\n\nconst getMessageContent = (msg: AppendMessage) => {\n  const allContent = [\n    ...msg.content,\n    ...(msg.attachments?.flatMap((a) => a.content) ?? []),\n  ];\n\n  const hasNonText = allContent.some(\n    (part) => part.type === \"file\" || part.type === \"image\",\n  );\n  const hasText = allContent.some((part) => part.type === \"text\");\n  if (hasNonText && !hasText) {\n    allContent.unshift({ type: \"text\", text: \" \" });\n  }\n\n  const content = allContent.map((part) => {\n    const type = part.type;\n    switch (type) {\n      case \"text\":\n        return { type: \"text\" as const, text: part.text };\n      case \"image\":\n        return { type: \"image_url\" as const, image_url: { url: part.image } };\n      case \"file\":\n        return {\n          type: \"file\" as const,\n          data: part.data,\n          mime_type: part.mimeType,\n          metadata: { filename: part.filename ?? \"file\" },\n          source_type: \"base64\" as const,\n        };\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported.\");\n      default: {\n        const _exhaustiveCheck:\n          | \"reasoning\"\n          | \"source\"\n          | \"audio\"\n          | \"data\"\n          | \"generative-ui\" = type;\n        throw new Error(\n          `Unsupported append message part type: ${_exhaustiveCheck}`,\n        );\n      }\n    }\n  });\n\n  if (content.length === 1 && content[0]?.type === \"text\") {\n    return content[0].text ?? \"\";\n  }\n  return content;\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\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const externalId = useAuiState((s) => s.threadListItem.externalId) as\n    | string\n    | null;\n  // Mutate in place rather than `{ ...options, threadId }`: spreading\n  // `UseStreamOptions` (a discriminated union on `transport`) into an object\n  // literal merges both arms' transport types, breaking arm assignment.\n  options.threadId = externalId;\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const stream = useStream(options);\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\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  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const threadMessages = useExternalMessageConverter({\n    callback: convertLangChainBaseMessage,\n    messages: stream.messages as LangChainBaseMessage[],\n    isRunning: effectiveIsRunning,\n  });\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const [runtimeRef] = useState(() => ({\n    get current() {\n      return runtime;\n    },\n  }));\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const streamRef = useRef(stream);\n  streamRef.current = stream;\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const toolInvocations = useToolInvocations({\n    state: {\n      messages: threadMessages,\n      isRunning: effectiveIsRunning,\n    },\n    getTools: () => runtimeRef.current.thread.getModelContext().tools,\n    onResult: (command) => {\n      if (command.type === \"add-tool-result\") {\n        void streamRef.current.submit(\n          {\n            [messagesKey]: [\n              {\n                type: \"tool\",\n                name: command.toolName,\n                tool_call_id: command.toolCallId,\n                content: JSON.stringify(command.result),\n                status: command.isError ? \"error\" : \"success\",\n              },\n            ],\n          },\n          {},\n        );\n      }\n    },\n    setToolStatuses,\n  });\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const extras = useMemo(\n    (): LangChainRuntimeExtras => ({\n      [symbolLangChainRuntimeExtras]: true,\n      interrupt: stream.interrupt,\n      interrupts: stream.interrupts,\n      submit: stream.submit,\n      values: stream.values,\n      messagesKey,\n    }),\n    [\n      stream.interrupt,\n      stream.interrupts,\n      stream.submit,\n      stream.values,\n      messagesKey,\n    ],\n  );\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const runtime = useExternalStoreRuntime({\n    isRunning: effectiveIsRunning,\n    messages: threadMessages,\n    adapters,\n    extras,\n    onNew: async (msg) => {\n      await toolInvocations.abort();\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      });\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    onCancel:\n      unstable_allowCancellation !== false\n        ? async () => {\n            await stream.stop();\n            await toolInvocations.abort();\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    ...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      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n      return useStreamThreadRuntime(optionsRef.current);\n    },\n    adapter,\n    allowNesting: true,\n  });\n};\n\n/**\n * Read the current LangGraph interrupt state from the runtime extras.\n */\nexport const useLangChainInterruptState = () => {\n  return useAuiState((s) => {\n    const extras = s.thread.extras;\n    if (!extras) return undefined;\n    return asLangChainRuntimeExtras(extras).interrupt;\n  });\n};\n\n/**\n * Returns a function to submit raw state updates to the LangGraph agent,\n * bypassing the normal message flow. Useful for sending interrupt resume\n * commands.\n */\nexport const useLangChainSubmit = () => {\n  const aui = useAui();\n  return (\n    values: Record<string, unknown> | null | undefined,\n    options?: Record<string, unknown>,\n  ) => {\n    const extras = aui.thread().getState().extras;\n    const { submit } = asLangChainRuntimeExtras(extras);\n    return submit(values, options);\n  };\n};\n\n/**\n * Submit a list of LangChain-shaped messages on the current thread.\n * Parity helper for migrating from `useLangGraphSend`. Routes to\n * `useStream().submit({ [messagesKey]: messages }, options)`.\n */\nexport const useLangChainSend = () => {\n  const aui = useAui();\n  return (\n    messages: readonly LangChainBaseMessage[],\n    options?: Record<string, unknown>,\n  ) => {\n    const { submit, messagesKey } = asLangChainRuntimeExtras(\n      aui.thread().getState().extras,\n    );\n    return submit({ [messagesKey]: messages }, options);\n  };\n};\n\n/**\n * Submit a `useStream` command (e.g. interrupt resume). Parity helper\n * for migrating from `useLangGraphSendCommand`. Note that v1's command\n * shape (`{ resume?, goto?, update? }`) differs from the legacy\n * `{ resume: string }` form — to carry a payload, use the input or\n * `stream.respond` instead.\n */\nexport const useLangChainSendCommand = () => {\n  const submit = useLangChainSubmit();\n  return (command: Record<string, unknown>) => submit(null, { command });\n};\n\n/**\n * Read a custom LangGraph state key from the current thread. Mirrors\n * `useStream().values[key]` from `@langchain/react` and updates when the\n * stream emits new state.\n *\n * @example\n * ```tsx\n * const todos = useLangChainState<Todo[]>(\"todos\");\n * const files = useLangChainState<Record<string, string>>(\"files\", {});\n * ```\n */\nexport function useLangChainState<T>(key: string): T | undefined;\nexport function useLangChainState<T>(key: string, defaultValue: T): T;\nexport function useLangChainState<T>(\n  key: string,\n  defaultValue?: T,\n): T | undefined {\n  return useAuiState((s) => {\n    const extras = s.thread.extras;\n    if (!extras) return defaultValue;\n    const value = asLangChainRuntimeExtras(extras).values[key] as T | undefined;\n    return value !== undefined ? value : defaultValue;\n  });\n}\n"],"mappings":";;;;;;;AAwBA,MAAM,+BAA+B,OAAO,0BAA0B;AActE,MAAM,4BAA4B,WAA4C;CAC5E,IACE,OAAO,WAAW,YAClB,UAAU,QACV,EAAE,gCAAgC,SAElC,MAAM,IAAI,MACR,oEACF;CACF,OAAO;AACT;AAkCA,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;AAYA,MAAM,qBAAqB,QAAuB;CAChD,MAAM,aAAa,CACjB,GAAG,IAAI,SACP,GAAI,IAAI,aAAa,SAAS,MAAM,EAAE,OAAO,KAAK,CAAC,CACrD;CAEA,MAAM,aAAa,WAAW,MAC3B,SAAS,KAAK,SAAS,UAAU,KAAK,SAAS,OAClD;CACA,MAAM,UAAU,WAAW,MAAM,SAAS,KAAK,SAAS,MAAM;CAC9D,IAAI,cAAc,CAAC,SACjB,WAAW,QAAQ;EAAE,MAAM;EAAQ,MAAM;CAAI,CAAC;CAGhD,MAAM,UAAU,WAAW,KAAK,SAAS;EACvC,MAAM,OAAO,KAAK;EAClB,QAAQ,MAAR;GACE,KAAK,QACH,OAAO;IAAE,MAAM;IAAiB,MAAM,KAAK;GAAK;GAClD,KAAK,SACH,OAAO;IAAE,MAAM;IAAsB,WAAW,EAAE,KAAK,KAAK,MAAM;GAAE;GACtE,KAAK,QACH,OAAO;IACL,MAAM;IACN,MAAM,KAAK;IACX,WAAW,KAAK;IAChB,UAAU,EAAE,UAAU,KAAK,YAAY,OAAO;IAC9C,aAAa;GACf;GACF,KAAK,aACH,MAAM,IAAI,MAAM,sCAAsC;GACxD,SAOE,MAAM,IAAI,MACR,yCAAyCA,MAC3C;EAEJ;CACF,CAAC;CAED,IAAI,QAAQ,WAAW,KAAK,QAAQ,IAAI,SAAS,QAC/C,OAAO,QAAQ,GAAG,QAAQ;CAE5B,OAAO;AACT;AAMA,MAAM,0BACJ,YAIG;CACH,MAAM,EAAE,UAAU,4BAA4B,+BAC5C;CACF,MAAM,cAAc,QAAQ,eAAe;CAS3C,QAAQ,WANW,aAAa,MAAM,EAAE,eAAe,UAM3B;CAG5B,MAAM,SAAS,UAAU,OAAO;CAGhC,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,EAAE,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,qBAAqB,OAAO,aAAa;CAG/C,MAAM,iBAAiB,4BAA4B;EACjD,UAAU;EACV,UAAU,OAAO;EACjB,WAAW;CACb,CAAC;CAGD,MAAM,CAAC,cAAc,gBAAgB,EACnC,IAAI,UAAU;EACZ,OAAO;CACT,EACF,EAAE;CAGF,MAAM,YAAY,OAAO,MAAM;CAC/B,UAAU,UAAU;CAGpB,MAAM,kBAAkB,mBAAmB;EACzC,OAAO;GACL,UAAU;GACV,WAAW;EACb;EACA,gBAAgB,WAAW,QAAQ,OAAO,gBAAgB,EAAE;EAC5D,WAAW,YAAY;GACrB,IAAI,QAAQ,SAAS,mBACnB,UAAe,QAAQ,OACrB,GACG,cAAc,CACb;IACE,MAAM;IACN,MAAM,QAAQ;IACd,cAAc,QAAQ;IACtB,SAAS,KAAK,UAAU,QAAQ,MAAM;IACtC,QAAQ,QAAQ,UAAU,UAAU;GACtC,CACF,EACF,GACA,CAAC,CACH;EAEJ;EACA;CACF,CAAC;CAsBD,MAAM,UAAU,wBAAwB;EACtC,WAAW;EACX,UAAU;EACV;EACA,QAvBa,eACkB;IAC5B,+BAA+B;GAChC,WAAW,OAAO;GAClB,YAAY,OAAO;GACnB,QAAQ,OAAO;GACf,QAAQ,OAAO;GACf;EACF,IACA;GACE,OAAO;GACP,OAAO;GACP,OAAO;GACP,OAAO;GACP;EACF,CAQK;EACL,OAAO,OAAO,QAAQ;GACpB,MAAM,gBAAgB,MAAM;GAC5B,MAAM,UAAU,kBAAkB,GAAG;GACrC,MAAM,gBACJ,+BAA+B,QAC3B,oBACE,UAAU,QAAQ,QACpB,EAAE,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,OAAO,GACjB,cAAc,CAAC,GAAG,eAAe;IAAE,MAAM;IAAS;GAAQ,CAAC,EAC9D,CAAC;EACH;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,UACE,+BAA+B,QAC3B,YAAY;GACV,MAAM,OAAO,KAAK;GAClB,MAAM,gBAAgB,MAAM;EAC9B,IACA,KAAA;CACR,CAAC;CAED,OAAO;AACT;;;;;;;;;;;;;;;;;;;;;;;;;AA0BA,MAAa,oBAAoB,eAAwC;CACvE,MAAM,EACJ,OACA,4BACA,QACA,QAAQ,UACR,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;GAElC,OAAO,uBAAuB,WAAW,OAAO;EAClD;EACA,SAPc,8BAA8B;EAQ5C,cAAc;CAChB,CAAC;AACH;;;;AAKA,MAAa,mCAAmC;CAC9C,OAAO,aAAa,MAAM;EACxB,MAAM,SAAS,EAAE,OAAO;EACxB,IAAI,CAAC,QAAQ,OAAO,KAAA;EACpB,OAAO,yBAAyB,MAAM,EAAE;CAC1C,CAAC;AACH;;;;;;AAOA,MAAa,2BAA2B;CACtC,MAAM,MAAM,OAAO;CACnB,QACE,QACA,YACG;EACH,MAAM,SAAS,IAAI,OAAO,EAAE,SAAS,EAAE;EACvC,MAAM,EAAE,WAAW,yBAAyB,MAAM;EAClD,OAAO,OAAO,QAAQ,OAAO;CAC/B;AACF;;;;;;AAOA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CACnB,QACE,UACA,YACG;EACH,MAAM,EAAE,QAAQ,gBAAgB,yBAC9B,IAAI,OAAO,EAAE,SAAS,EAAE,MAC1B;EACA,OAAO,OAAO,GAAG,cAAc,SAAS,GAAG,OAAO;CACpD;AACF;;;;;;;;AASA,MAAa,gCAAgC;CAC3C,MAAM,SAAS,mBAAmB;CAClC,QAAQ,YAAqC,OAAO,MAAM,EAAE,QAAQ,CAAC;AACvE;AAeA,SAAgB,kBACd,KACA,cACe;CACf,OAAO,aAAa,MAAM;EACxB,MAAM,SAAS,EAAE,OAAO;EACxB,IAAI,CAAC,QAAQ,OAAO;EACpB,MAAM,QAAQ,yBAAyB,MAAM,EAAE,OAAO;EACtD,OAAO,UAAU,KAAA,IAAY,QAAQ;CACvC,CAAC;AACH"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-langchain",
-  "version": "0.0.6",
+  "version": "0.0.8",
   "description": "LangChain useStream adapter for assistant-ui",
   "keywords": [
     "langchain",
@@ -29,13 +29,13 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.2.0",
-    "@assistant-ui/store": "^0.2.10",
+    "@assistant-ui/core": "^0.2.5",
+    "@assistant-ui/store": "^0.2.12",
     "assistant-cloud": "*",
-    "assistant-stream": "^0.3.13"
+    "assistant-stream": "^0.3.16"
   },
   "peerDependencies": {
-    "@langchain/react": "^0.3.3",
+    "@langchain/react": "^1.0.2",
     "@types/react": "*",
     "react": "^18 || ^19"
   },
@@ -45,15 +45,15 @@
     }
   },
   "devDependencies": {
-    "@langchain/core": "^1.1.44",
-    "@langchain/langgraph-sdk": "^1.9.1",
-    "@langchain/react": "^0.3.3",
+    "@langchain/core": "^1.1.47",
+    "@langchain/langgraph-sdk": "^1.9.4",
+    "@langchain/react": "^1.0.4",
     "@testing-library/react": "^16.3.2",
-    "@types/react": "^19.2.14",
+    "@types/react": "^19.2.15",
     "jsdom": "^29.1.1",
-    "react": "^19.2.5",
-    "vitest": "^4.1.5",
-    "@assistant-ui/x-buildutils": "0.0.7"
+    "react": "^19.2.6",
+    "vitest": "^4.1.7",
+    "@assistant-ui/x-buildutils": "0.0.9"
   },
   "publishConfig": {
     "access": "public",

package/src/convertMessages.ts

@@ -4,7 +4,7 @@ import type { useExternalMessageConverter } from "@assistant-ui/core/react";
 import type { ReadonlyJSONObject } from "assistant-stream/utils";
 import type { LangChainBaseMessage, LangChainContentBlock } from "./types";
 
-const getMessageType = (message: LangChainBaseMessage): string => {
+export const getMessageType = (message: LangChainBaseMessage): string => {
   if (typeof message._getType === "function") return message._getType();
   if ("type" in message)
     return (message as Record<string, unknown>).type as string;

package/src/index.ts

@@ -1,6 +1,8 @@
 export {
   useStreamRuntime,
   useLangChainInterruptState,
+  useLangChainSend,
+  useLangChainSendCommand,
   useLangChainState,
   useLangChainSubmit,
 } from "./useStreamRuntime";