@assistant-ui/react-langchain 0.0.7 → 0.0.10

package/dist/convertMessages.d.ts

@@ -1,5 +1,9 @@
-import type { useExternalMessageConverter } from "@assistant-ui/core/react";
-import type { LangChainBaseMessage } from "./types.js";
-export declare const getMessageType: (message: LangChainBaseMessage) => string;
-export declare const convertLangChainBaseMessage: useExternalMessageConverter.Callback<LangChainBaseMessage>;
+import { LangChainBaseMessage } from "./types.js";
+import { useExternalMessageConverter } from "@assistant-ui/core/react";
+
+//#region src/convertMessages.d.ts
+declare const getMessageType: (message: LangChainBaseMessage) => string;
+declare const convertLangChainBaseMessage: useExternalMessageConverter.Callback<LangChainBaseMessage>;
+//#endregion
+export { convertLangChainBaseMessage, getMessageType };
 //# sourceMappingURL=convertMessages.d.ts.map

package/dist/convertMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessages.d.ts","sourceRoot":"","sources":["../src/convertMessages.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,0BAA0B,CAAC;AAE5E,OAAO,KAAK,EAAE,oBAAoB,EAAyB,mBAAgB;AAE3E,eAAO,MAAM,cAAc,GAAI,SAAS,oBAAoB,KAAG,MAK9D,CAAC;AAkDF,eAAO,MAAM,2BAA2B,EAAE,2BAA2B,CAAC,QAAQ,CAC5E,oBAAoB,CA0ErB,CAAC"}
+{"version":3,"file":"convertMessages.d.ts","names":[],"sources":["../src/convertMessages.ts"],"mappings":";;;;cAMa,cAAA,GAAkB,OAA6B,EAApB,oBAAoB;AAAA,cAuD/C,2BAAA,EAA6B,2BAAA,CAA4B,QAAQ,CAC5E,oBAAA"}

package/dist/convertMessages.js

@@ -1,115 +1,103 @@
 "use client";
-export const getMessageType = (message) => {
-    if (typeof message._getType === "function")
-        return message._getType();
-    if ("type" in message)
-        return message.type;
-    throw new Error("Cannot determine message type");
+//#region src/convertMessages.ts
+const getMessageType = (message) => {
+	if (typeof message._getType === "function") return message._getType();
+	if ("type" in message) return message.type;
+	throw new Error("Cannot determine message type");
 };
 const contentToParts = (content) => {
-    if (typeof content === "string")
-        return [{ type: "text", text: content }];
-    const parts = content;
-    return parts
-        .map((part) => {
-        const type = part.type;
-        switch (type) {
-            case "text":
-            case "text_delta":
-                return { type: "text", text: part.text };
-            case "image_url":
-                if (typeof part.image_url === "string") {
-                    return { type: "image", image: part.image_url };
-                }
-                return { type: "image", image: part.image_url.url };
-            case "thinking":
-                return { type: "reasoning", text: part.thinking };
-            case "reasoning":
-                return {
-                    type: "reasoning",
-                    text: part.summary.map((s) => s.text).join("\n\n\n"),
-                };
-            case "tool_use":
-            case "input_json_delta":
-                return null;
-            default:
-                return null;
-        }
-    })
-        .filter((p) => p !== null);
+	if (typeof content === "string") return [{
+		type: "text",
+		text: content
+	}];
+	return content.map((part) => {
+		switch (part.type) {
+			case "text":
+			case "text_delta": return {
+				type: "text",
+				text: part.text
+			};
+			case "image_url":
+				if (typeof part.image_url === "string") return {
+					type: "image",
+					image: part.image_url
+				};
+				return {
+					type: "image",
+					image: part.image_url.url
+				};
+			case "thinking": return {
+				type: "reasoning",
+				text: part.thinking
+			};
+			case "reasoning": return {
+				type: "reasoning",
+				text: part.summary.map((s) => s.text).join("\n\n\n")
+			};
+			case "tool_use":
+			case "input_json_delta": return null;
+			default: return null;
+		}
+	}).filter((p) => p !== null);
 };
 const getCustomMetadata = (additionalKwargs) => additionalKwargs?.metadata ?? {};
 const getStringContent = (content) => {
-    if (typeof content === "string")
-        return content;
-    const parts = content;
-    return parts
-        .filter((c) => c.type === "text")
-        .map((c) => c.text)
-        .join("");
+	if (typeof content === "string") return content;
+	return content.filter((c) => c.type === "text").map((c) => c.text).join("");
 };
-export const convertLangChainBaseMessage = (message) => {
-    const type = getMessageType(message);
-    switch (type) {
-        case "system":
-            return {
-                role: "system",
-                id: message.id,
-                content: [{ type: "text", text: getStringContent(message.content) }],
-                metadata: {
-                    custom: getCustomMetadata(message.additional_kwargs),
-                },
-            };
-        case "human":
-            return {
-                role: "user",
-                id: message.id,
-                content: contentToParts(message.content),
-                metadata: {
-                    custom: getCustomMetadata(message.additional_kwargs),
-                },
-            };
-        case "ai": {
-            const toolCallParts = message.tool_calls?.map((tc) => ({
-                type: "tool-call",
-                toolCallId: tc.id,
-                toolName: tc.name,
-                args: tc.args,
-                argsText: JSON.stringify(tc.args),
-            })) ?? [];
-            const assistantStatus = typeof message.status === "object" ? message.status : undefined;
-            return {
-                role: "assistant",
-                id: message.id,
-                content: [...contentToParts(message.content), ...toolCallParts],
-                metadata: {
-                    custom: getCustomMetadata(message.additional_kwargs),
-                },
-                ...(assistantStatus && { status: assistantStatus }),
-            };
-        }
-        case "tool":
-            return {
-                role: "tool",
-                toolName: message.name ?? "",
-                toolCallId: message.tool_call_id ?? "",
-                result: message.content,
-                artifact: message.artifact,
-                isError: message.status === "error",
-            };
-        default:
-            return {
-                role: "system",
-                id: message.id,
-                content: [
-                    {
-                        type: "text",
-                        text: typeof message.content === "string"
-                            ? message.content
-                            : JSON.stringify(message.content),
-                    },
-                ],
-            };
-    }
+const convertLangChainBaseMessage = (message) => {
+	switch (getMessageType(message)) {
+		case "system": return {
+			role: "system",
+			id: message.id,
+			content: [{
+				type: "text",
+				text: getStringContent(message.content)
+			}],
+			metadata: { custom: getCustomMetadata(message.additional_kwargs) }
+		};
+		case "human": return {
+			role: "user",
+			id: message.id,
+			content: contentToParts(message.content),
+			metadata: { custom: getCustomMetadata(message.additional_kwargs) }
+		};
+		case "ai": {
+			const toolCallParts = message.tool_calls?.map((tc) => ({
+				type: "tool-call",
+				toolCallId: tc.id,
+				toolName: tc.name,
+				args: tc.args,
+				argsText: JSON.stringify(tc.args)
+			})) ?? [];
+			const assistantStatus = typeof message.status === "object" ? message.status : void 0;
+			return {
+				role: "assistant",
+				id: message.id,
+				content: [...contentToParts(message.content), ...toolCallParts],
+				metadata: { custom: getCustomMetadata(message.additional_kwargs) },
+				...assistantStatus && { status: assistantStatus }
+			};
+		}
+		case "tool": return {
+			role: "tool",
+			toolName: message.name ?? "",
+			toolCallId: message.tool_call_id ?? "",
+			result: message.content,
+			artifact: message.artifact,
+			isError: message.status === "error"
+		};
+		default: return {
+			role: "system",
+			id: message.id,
+			content: [{
+				type: "text",
+				text: typeof message.content === "string" ? message.content : JSON.stringify(message.content)
+			}]
+		};
+	}
 };
+//#endregion
+export { convertLangChainBaseMessage, getMessageType };
+
 //# sourceMappingURL=convertMessages.js.map

package/dist/convertMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessages.js","sourceRoot":"","sources":["../src/convertMessages.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAMb,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,OAA6B,EAAU,EAAE;IACtE,IAAI,OAAO,OAAO,CAAC,QAAQ,KAAK,UAAU;QAAE,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;IACtE,IAAI,MAAM,IAAI,OAAO;QACnB,OAAQ,OAAmC,CAAC,IAAc,CAAC;IAC7D,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;AACnD,CAAC,CAAC;AAEF,MAAM,cAAc,GAAG,CAAC,OAAgB,EAAE,EAAE;IAC1C,IAAI,OAAO,OAAO,KAAK,QAAQ;QAC7B,OAAO,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAEpD,MAAM,KAAK,GAAG,OAA2C,CAAC;IAC1D,OAAO,KAAK;SACT,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACvB,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,MAAM,CAAC;YACZ,KAAK,YAAY;gBACf,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACpD,KAAK,WAAW;gBACd,IAAI,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBACvC,OAAO,EAAE,IAAI,EAAE,OAAgB,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC3D,CAAC;gBACD,OAAO,EAAE,IAAI,EAAE,OAAgB,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC;YAC/D,KAAK,UAAU;gBACb,OAAO,EAAE,IAAI,EAAE,WAAoB,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC7D,KAAK,WAAW;gBACd,OAAO;oBACL,IAAI,EAAE,WAAoB;oBAC1B,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC;iBACrD,CAAC;YACJ,KAAK,UAAU,CAAC;YAChB,KAAK,kBAAkB;gBACrB,OAAO,IAAI,CAAC;YACd;gBACE,OAAO,IAAI,CAAC;QAChB,CAAC;IACH,CAAC,CAAC;SACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;AAC/B,CAAC,CAAC;AAEF,MAAM,iBAAiB,GAAG,CACxB,gBAAqD,EAC5B,EAAE,CAC1B,gBAAgB,EAAE,QAAoC,IAAI,EAAE,CAAC;AAEhE,MAAM,gBAAgB,GAAG,CAAC,OAAgB,EAAU,EAAE;IACpD,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAC;IAChD,MAAM,KAAK,GAAG,OAA2C,CAAC;IAC1D,OAAO,KAAK;SACT,MAAM,CAAC,CAAC,CAAC,EAAuC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;SACrE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SAClB,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAEpC,CAAC,OAAO,EAAE,EAAE;IACd,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IAErC,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,QAAQ;YACX,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBACpE,QAAQ,EAAE;oBACR,MAAM,EAAE,iBAAiB,CAAC,OAAO,CAAC,iBAAiB,CAAC;iBACrD;aACF,CAAC;QAEJ,KAAK,OAAO;YACV,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC;gBACxC,QAAQ,EAAE;oBACR,MAAM,EAAE,iBAAiB,CAAC,OAAO,CAAC,iBAAiB,CAAC;iBACrD;aACF,CAAC;QAEJ,KAAK,IAAI,CAAC,CAAC,CAAC;YACV,MAAM,aAAa,GACjB,OAAO,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC/B,IAAI,EAAE,WAAoB;gBAC1B,UAAU,EAAE,EAAE,CAAC,EAAE;gBACjB,QAAQ,EAAE,EAAE,CAAC,IAAI;gBACjB,IAAI,EAAE,EAAE,CAAC,IAA0B;gBACnC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,CAAC;aAClC,CAAC,CAAC,IAAI,EAAE,CAAC;YAEZ,MAAM,eAAe,GACnB,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;YAElE,OAAO;gBACL,IAAI,EAAE,WAAW;gBACjB,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,OAAO,EAAE,CAAC,GAAG,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,GAAG,aAAa,CAAC;gBAC/D,QAAQ,EAAE;oBACR,MAAM,EAAE,iBAAiB,CAAC,OAAO,CAAC,iBAAiB,CAAC;iBACrD;gBACD,GAAG,CAAC,eAAe,IAAI,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC;aACpD,CAAC;QACJ,CAAC;QAED,KAAK,MAAM;YACT,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;gBAC5B,UAAU,EAAE,OAAO,CAAC,YAAY,IAAI,EAAE;gBACtC,MAAM,EAAE,OAAO,CAAC,OAAO;gBACvB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,OAAO,EAAE,OAAO,CAAC,MAAM,KAAK,OAAO;aACpC,CAAC;QAEJ;YACE,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EACF,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;4BACjC,CAAC,CAAC,OAAO,CAAC,OAAO;4BACjB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;qBACtC;iBACF;aACF,CAAC;IACN,CAAC;AACH,CAAC,CAAC"}
+{"version":3,"file":"convertMessages.js","names":["parts"],"sources":["../src/convertMessages.ts"],"sourcesContent":["\"use client\";\n\nimport type { useExternalMessageConverter } from \"@assistant-ui/core/react\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport type { LangChainBaseMessage, LangChainContentBlock } from \"./types\";\n\nexport const getMessageType = (message: LangChainBaseMessage): string => {\n  if (typeof message._getType === \"function\") return message._getType();\n  if (\"type\" in message)\n    return (message as Record<string, unknown>).type as string;\n  throw new Error(\"Cannot determine message type\");\n};\n\nconst contentToParts = (content: unknown) => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n\n  const parts = content as readonly LangChainContentBlock[];\n  return parts\n    .map((part) => {\n      const type = part.type;\n      switch (type) {\n        case \"text\":\n        case \"text_delta\":\n          return { type: \"text\" as const, text: part.text };\n        case \"image_url\":\n          if (typeof part.image_url === \"string\") {\n            return { type: \"image\" as const, image: part.image_url };\n          }\n          return { type: \"image\" as const, image: part.image_url.url };\n        case \"thinking\":\n          return { type: \"reasoning\" as const, text: part.thinking };\n        case \"reasoning\":\n          return {\n            type: \"reasoning\" as const,\n            text: part.summary.map((s) => s.text).join(\"\\n\\n\\n\"),\n          };\n        case \"tool_use\":\n        case \"input_json_delta\":\n          return null;\n        default:\n          return null;\n      }\n    })\n    .filter((p) => p !== null);\n};\n\nconst getCustomMetadata = (\n  additionalKwargs: Record<string, unknown> | undefined,\n): Record<string, unknown> =>\n  (additionalKwargs?.metadata as Record<string, unknown>) ?? {};\n\nconst getStringContent = (content: unknown): string => {\n  if (typeof content === \"string\") return content;\n  const parts = content as readonly LangChainContentBlock[];\n  return parts\n    .filter((c): c is { type: \"text\"; text: string } => c.type === \"text\")\n    .map((c) => c.text)\n    .join(\"\");\n};\n\nexport const convertLangChainBaseMessage: useExternalMessageConverter.Callback<\n  LangChainBaseMessage\n> = (message) => {\n  const type = getMessageType(message);\n\n  switch (type) {\n    case \"system\":\n      return {\n        role: \"system\",\n        id: message.id,\n        content: [{ type: \"text\", text: getStringContent(message.content) }],\n        metadata: {\n          custom: getCustomMetadata(message.additional_kwargs),\n        },\n      };\n\n    case \"human\":\n      return {\n        role: \"user\",\n        id: message.id,\n        content: contentToParts(message.content),\n        metadata: {\n          custom: getCustomMetadata(message.additional_kwargs),\n        },\n      };\n\n    case \"ai\": {\n      const toolCallParts =\n        message.tool_calls?.map((tc) => ({\n          type: \"tool-call\" as const,\n          toolCallId: tc.id,\n          toolName: tc.name,\n          args: tc.args as ReadonlyJSONObject,\n          argsText: JSON.stringify(tc.args),\n        })) ?? [];\n\n      const assistantStatus =\n        typeof message.status === \"object\" ? message.status : undefined;\n\n      return {\n        role: \"assistant\",\n        id: message.id,\n        content: [...contentToParts(message.content), ...toolCallParts],\n        metadata: {\n          custom: getCustomMetadata(message.additional_kwargs),\n        },\n        ...(assistantStatus && { status: assistantStatus }),\n      };\n    }\n\n    case \"tool\":\n      return {\n        role: \"tool\",\n        toolName: message.name ?? \"\",\n        toolCallId: message.tool_call_id ?? \"\",\n        result: message.content,\n        artifact: message.artifact,\n        isError: message.status === \"error\",\n      };\n\n    default:\n      return {\n        role: \"system\",\n        id: message.id,\n        content: [\n          {\n            type: \"text\",\n            text:\n              typeof message.content === \"string\"\n                ? message.content\n                : JSON.stringify(message.content),\n          },\n        ],\n      };\n  }\n};\n"],"mappings":";;AAMA,MAAa,kBAAkB,YAA0C;CACvE,IAAI,OAAO,QAAQ,aAAa,YAAY,OAAO,QAAQ,SAAS;CACpE,IAAI,UAAU,SACZ,OAAQ,QAAoC;CAC9C,MAAM,IAAI,MAAM,+BAA+B;AACjD;AAEA,MAAM,kBAAkB,YAAqB;CAC3C,IAAI,OAAO,YAAY,UACrB,OAAO,CAAC;EAAE,MAAM;EAAiB,MAAM;CAAQ,CAAC;CAGlD,OAAOA,QACJ,KAAK,SAAS;EAEb,QADa,KAAK,MAClB;GACE,KAAK;GACL,KAAK,cACH,OAAO;IAAE,MAAM;IAAiB,MAAM,KAAK;GAAK;GAClD,KAAK;IACH,IAAI,OAAO,KAAK,cAAc,UAC5B,OAAO;KAAE,MAAM;KAAkB,OAAO,KAAK;IAAU;IAEzD,OAAO;KAAE,MAAM;KAAkB,OAAO,KAAK,UAAU;IAAI;GAC7D,KAAK,YACH,OAAO;IAAE,MAAM;IAAsB,MAAM,KAAK;GAAS;GAC3D,KAAK,aACH,OAAO;IACL,MAAM;IACN,MAAM,KAAK,QAAQ,KAAK,MAAM,EAAE,IAAI,EAAE,KAAK,QAAQ;GACrD;GACF,KAAK;GACL,KAAK,oBACH,OAAO;GACT,SACE,OAAO;EACX;CACF,CAAC,EACA,QAAQ,MAAM,MAAM,IAAI;AAC7B;AAEA,MAAM,qBACJ,qBAEC,kBAAkB,YAAwC,CAAC;AAE9D,MAAM,oBAAoB,YAA6B;CACrD,IAAI,OAAO,YAAY,UAAU,OAAO;CAExC,OAAOA,QACJ,QAAQ,MAA2C,EAAE,SAAS,MAAM,EACpE,KAAK,MAAM,EAAE,IAAI,EACjB,KAAK,EAAE;AACZ;AAEA,MAAa,+BAER,YAAY;CAGf,QAFa,eAAe,OAEjB,GAAX;EACE,KAAK,UACH,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,CAAC;IAAE,MAAM;IAAQ,MAAM,iBAAiB,QAAQ,OAAO;GAAE,CAAC;GACnE,UAAU,EACR,QAAQ,kBAAkB,QAAQ,iBAAiB,EACrD;EACF;EAEF,KAAK,SACH,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,eAAe,QAAQ,OAAO;GACvC,UAAU,EACR,QAAQ,kBAAkB,QAAQ,iBAAiB,EACrD;EACF;EAEF,KAAK,MAAM;GACT,MAAM,gBACJ,QAAQ,YAAY,KAAK,QAAQ;IAC/B,MAAM;IACN,YAAY,GAAG;IACf,UAAU,GAAG;IACb,MAAM,GAAG;IACT,UAAU,KAAK,UAAU,GAAG,IAAI;GAClC,EAAE,KAAK,CAAC;GAEV,MAAM,kBACJ,OAAO,QAAQ,WAAW,WAAW,QAAQ,SAAS,KAAA;GAExD,OAAO;IACL,MAAM;IACN,IAAI,QAAQ;IACZ,SAAS,CAAC,GAAG,eAAe,QAAQ,OAAO,GAAG,GAAG,aAAa;IAC9D,UAAU,EACR,QAAQ,kBAAkB,QAAQ,iBAAiB,EACrD;IACA,GAAI,mBAAmB,EAAE,QAAQ,gBAAgB;GACnD;EACF;EAEA,KAAK,QACH,OAAO;GACL,MAAM;GACN,UAAU,QAAQ,QAAQ;GAC1B,YAAY,QAAQ,gBAAgB;GACpC,QAAQ,QAAQ;GAChB,UAAU,QAAQ;GAClB,SAAS,QAAQ,WAAW;EAC9B;EAEF,SACE,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,CACP;IACE,MAAM;IACN,MACE,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,KAAK,UAAU,QAAQ,OAAO;GACtC,CACF;EACF;CACJ;AACF"}

package/dist/index.d.ts

@@ -1,5 +1,4 @@
-export { useStreamRuntime, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, } from "./useStreamRuntime.js";
-export type { UseStreamRuntimeOptions } from "./useStreamRuntime.js";
-export { convertLangChainBaseMessage } from "./convertMessages.js";
-export type { LangChainBaseMessage, LangChainContentBlock, LangChainToolCall, } from "./types.js";
-//# sourceMappingURL=index.d.ts.map
+import { LangChainBaseMessage, LangChainContentBlock, LangChainToolCall } from "./types.js";
+import { convertLangChainBaseMessage } from "./convertMessages.js";
+import { UseStreamRuntimeOptions, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useStreamRuntime } from "./useStreamRuntime.js";
+export { type LangChainBaseMessage, type LangChainContentBlock, type LangChainToolCall, type UseStreamRuntimeOptions, convertLangChainBaseMessage, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useStreamRuntime };

package/dist/index.js

@@ -1,3 +1,3 @@
-export { useStreamRuntime, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, } from "./useStreamRuntime.js";
-export { convertLangChainBaseMessage } from "./convertMessages.js";
-//# sourceMappingURL=index.js.map
+import { convertLangChainBaseMessage } from "./convertMessages.js";
+import { useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useStreamRuntime } from "./useStreamRuntime.js";
+export { convertLangChainBaseMessage, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useStreamRuntime };

package/dist/types.d.ts

@@ -1,32 +1,34 @@
-import type { MessageStatus } from "@assistant-ui/core";
+import { MessageStatus } from "@assistant-ui/core";
+
+//#region src/types.d.ts
 /** Known content block types from @langchain/core messages. */
-export type LangChainContentBlock = {
-    type: "text";
-    text: string;
+type LangChainContentBlock = {
+  type: "text";
+  text: string;
 } | {
-    type: "text_delta";
-    text: string;
+  type: "text_delta";
+  text: string;
 } | {
-    type: "image_url";
-    image_url: string | {
-        url: string;
-    };
+  type: "image_url";
+  image_url: string | {
+    url: string;
+  };
 } | {
-    type: "thinking";
-    thinking: string;
+  type: "thinking";
+  thinking: string;
 } | {
-    type: "reasoning";
-    summary: Array<{
-        type: "summary_text";
-        text: string;
-    }>;
+  type: "reasoning";
+  summary: Array<{
+    type: "summary_text";
+    text: string;
+  }>;
 } | {
-    type: "tool_use" | "input_json_delta";
+  type: "tool_use" | "input_json_delta";
 };
-export type LangChainToolCall = {
-    id: string;
-    name: string;
-    args: Record<string, unknown>;
+type LangChainToolCall = {
+  id: string;
+  name: string;
+  args: Record<string, unknown>;
 };
 /**
  * Minimal duck-typed interface for BaseMessage class instances returned by
@@ -35,19 +37,17 @@ export type LangChainToolCall = {
  * Uses `unknown` for `content` to remain compatible with
  * `@langchain/core`'s `MessageContent` union.
  */
-export type LangChainBaseMessage = {
-    _getType: () => string;
-    content: unknown;
-    id?: string | undefined;
-    name?: string | undefined;
-    additional_kwargs?: Record<string, unknown> | undefined;
-    /** Present on AIMessage */
-    tool_calls?: readonly LangChainToolCall[] | undefined;
-    /** Present on ToolMessage */
-    tool_call_id?: string | undefined;
-    /** Completion status (AIMessage) or outcome status (ToolMessage) */
-    status?: MessageStatus | "success" | "error" | undefined;
-    /** Present on ToolMessage */
-    artifact?: unknown;
+type LangChainBaseMessage = {
+  _getType: () => string;
+  content: unknown;
+  id?: string | undefined;
+  name?: string | undefined;
+  additional_kwargs?: Record<string, unknown> | undefined; /** Present on AIMessage */
+  tool_calls?: readonly LangChainToolCall[] | undefined; /** Present on ToolMessage */
+  tool_call_id?: string | undefined; /** Completion status (AIMessage) or outcome status (ToolMessage) */
+  status?: MessageStatus | "success" | "error" | undefined; /** Present on ToolMessage */
+  artifact?: unknown;
 };
+//#endregion
+export { LangChainBaseMessage, LangChainContentBlock, LangChainToolCall };
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAExD,+DAA+D;AAC/D,MAAM,MAAM,qBAAqB,GAC7B;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAC9B;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACpC;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,SAAS,EAAE,MAAM,GAAG;QAAE,GAAG,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GAC1D;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,GACtC;IACE,IAAI,EAAE,WAAW,CAAC;IAClB,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,cAAc,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACxD,GACD;IAAE,IAAI,EAAE,UAAU,GAAG,kBAAkB,CAAA;CAAE,CAAC;AAE9C,MAAM,MAAM,iBAAiB,GAAG;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,EAAE,MAAM,MAAM,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;IACxD,2BAA2B;IAC3B,UAAU,CAAC,EAAE,SAAS,iBAAiB,EAAE,GAAG,SAAS,CAAC;IACtD,6BAA6B;IAC7B,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,oEAAoE;IACpE,MAAM,CAAC,EAAE,aAAa,GAAG,SAAS,GAAG,OAAO,GAAG,SAAS,CAAC;IACzD,6BAA6B;IAC7B,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB,CAAC"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;;KAGY,qBAAA;EACN,IAAA;EAAc,IAAA;AAAA;EACd,IAAA;EAAoB,IAAA;AAAA;EACpB,IAAA;EAAmB,SAAA;IAAsB,GAAA;EAAA;AAAA;EACzC,IAAA;EAAkB,QAAA;AAAA;EAElB,IAAA;EACA,OAAA,EAAS,KAAK;IAAG,IAAA;IAAsB,IAAA;EAAA;AAAA;EAEvC,IAAA;AAAA;AAAA,KAEM,iBAAA;EACV,EAAA;EACA,IAAA;EACA,IAAA,EAAM,MAAM;AAAA;;;;AAAA;AAUd;;;KAAY,oBAAA;EACV,QAAA;EACA,OAAA;EACA,EAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA,+BAHpB;EAKA,UAAA,YAAsB,iBAAA,gBAHtB;EAKA,YAAA,uBAJoB;EAMpB,MAAA,GAAS,aAAA,oCAJa;EAMtB,QAAA;AAAA"}

package/dist/types.js

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=types.js.map

package/dist/useStreamRuntime.d.ts

@@ -1,39 +1,40 @@
-import type { AttachmentAdapter, FeedbackAdapter, RemoteThreadListAdapter, SpeechSynthesisAdapter } from "@assistant-ui/core";
-import type { AssistantCloud } from "assistant-cloud";
-import { type UseStreamOptions } from "@langchain/react";
-import type { LangChainBaseMessage } from "./types.js";
+/// <reference types="@assistant-ui/core/store" />
+import { LangChainBaseMessage } from "./types.js";
+import { UseStreamOptions } from "@langchain/react";
+import { AttachmentAdapter, FeedbackAdapter, RemoteThreadListAdapter, SpeechSynthesisAdapter } from "@assistant-ui/core";
+import { AssistantCloud } from "assistant-cloud";
+
+//#region src/useStreamRuntime.d.ts
 type LangChainRuntimeExtraOptions = {
-    cloud?: AssistantCloud | undefined;
-    adapters?: {
-        attachments?: AttachmentAdapter | undefined;
-        speech?: SpeechSynthesisAdapter | undefined;
-        feedback?: FeedbackAdapter | undefined;
-    } | undefined;
-    /**
-     * When the user sends a new message while previous tool calls are
-     * still pending, automatically submit `tool` messages that cancel
-     * them so the agent's tool-call accounting stays consistent.
-     * Defaults to `true`.
-     */
-    autoCancelPendingToolCalls?: boolean | undefined;
-    /**
-     * Routes the Cancel button's click to `useStream().stop()`. On by
-     * default. Pass `false` to disable the Cancel button.
-     */
-    unstable_allowCancellation?: boolean | undefined;
-    /**
-     * Custom `RemoteThreadListAdapter`. When provided, replaces the
-     * cloud-backed thread list adapter.
-     */
-    unstable_threadListAdapter?: RemoteThreadListAdapter | undefined;
-    /** Custom thread-creation hook, forwarded to the cloud adapter. */
-    create?: (() => Promise<{
-        externalId: string | undefined;
-    }>) | undefined;
-    /** Custom thread-deletion hook, forwarded to the cloud adapter. */
-    delete?: ((threadId: string) => Promise<void>) | undefined;
+  cloud?: AssistantCloud | undefined;
+  adapters?: {
+    attachments?: AttachmentAdapter | undefined;
+    speech?: SpeechSynthesisAdapter | undefined;
+    feedback?: FeedbackAdapter | undefined;
+  } | undefined;
+  /**
+   * When the user sends a new message while previous tool calls are
+   * still pending, automatically submit `tool` messages that cancel
+   * them so the agent's tool-call accounting stays consistent.
+   * Defaults to `true`.
+   */
+  autoCancelPendingToolCalls?: boolean | undefined;
+  /**
+   * Routes the Cancel button's click to `useStream().stop()`. On by
+   * default. Pass `false` to disable the Cancel button.
+   */
+  unstable_allowCancellation?: boolean | undefined;
+  /**
+   * Custom `RemoteThreadListAdapter`. When provided, replaces the
+   * cloud-backed thread list adapter.
+   */
+  unstable_threadListAdapter?: RemoteThreadListAdapter | undefined; /** Custom thread-creation hook, forwarded to the cloud adapter. */
+  create?: (() => Promise<{
+    externalId: string | undefined;
+  }>) | undefined; /** Custom thread-deletion hook, forwarded to the cloud adapter. */
+  delete?: ((threadId: string) => Promise<void>) | undefined;
 };
-export type UseStreamRuntimeOptions = UseStreamOptions extends infer O ? O extends UseStreamOptions ? O & LangChainRuntimeExtraOptions : never : never;
+type UseStreamRuntimeOptions = UseStreamOptions extends infer O ? O extends UseStreamOptions ? O & LangChainRuntimeExtraOptions : never : never;
 /**
  * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.
  * Accepts the same options as `useStream` from `@langchain/react`, plus
@@ -58,25 +59,25 @@ export type UseStreamRuntimeOptions = UseStreamOptions extends infer O ? O exten
  * }
  * ```
  */
-export declare const useStreamRuntime: (rawOptions: UseStreamRuntimeOptions) => import("@assistant-ui/core").AssistantRuntime;
+declare const useStreamRuntime: (rawOptions: UseStreamRuntimeOptions) => import("@assistant-ui/core").AssistantRuntime;
 /**
  * Read the current LangGraph interrupt state from the runtime extras.
  */
-export declare const useLangChainInterruptState: () => {
-    value?: unknown;
+declare const useLangChainInterruptState: () => {
+  value?: unknown;
 } | undefined;
 /**
  * Returns a function to submit raw state updates to the LangGraph agent,
  * bypassing the normal message flow. Useful for sending interrupt resume
  * commands.
  */
-export declare const useLangChainSubmit: () => (values: Record<string, unknown> | null | undefined, options?: Record<string, unknown>) => Promise<void>;
+declare const useLangChainSubmit: () => (values: Record<string, unknown> | null | undefined, options?: Record<string, unknown>) => Promise<void>;
 /**
  * Submit a list of LangChain-shaped messages on the current thread.
  * Parity helper for migrating from `useLangGraphSend`. Routes to
  * `useStream().submit({ [messagesKey]: messages }, options)`.
  */
-export declare const useLangChainSend: () => (messages: readonly LangChainBaseMessage[], options?: Record<string, unknown>) => Promise<void>;
+declare const useLangChainSend: () => (messages: readonly LangChainBaseMessage[], options?: Record<string, unknown>) => Promise<void>;
 /**
  * Submit a `useStream` command (e.g. interrupt resume). Parity helper
  * for migrating from `useLangGraphSendCommand`. Note that v1's command
@@ -84,7 +85,7 @@ export declare const useLangChainSend: () => (messages: readonly LangChainBaseMe
  * `{ resume: string }` form — to carry a payload, use the input or
  * `stream.respond` instead.
  */
-export declare const useLangChainSendCommand: () => (command: Record<string, unknown>) => Promise<void>;
+declare const useLangChainSendCommand: () => (command: Record<string, unknown>) => Promise<void>;
 /**
  * Read a custom LangGraph state key from the current thread. Mirrors
  * `useStream().values[key]` from `@langchain/react` and updates when the
@@ -96,7 +97,8 @@ export declare const useLangChainSendCommand: () => (command: Record<string, unk
  * const files = useLangChainState<Record<string, string>>("files", {});
  * ```
  */
-export declare function useLangChainState<T>(key: string): T | undefined;
-export declare function useLangChainState<T>(key: string, defaultValue: T): T;
-export {};
+declare function useLangChainState<T>(key: string): T | undefined;
+declare function useLangChainState<T>(key: string, defaultValue: T): T;
+//#endregion
+export { UseStreamRuntimeOptions, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useStreamRuntime };
 //# sourceMappingURL=useStreamRuntime.d.ts.map

package/dist/useStreamRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamRuntime.d.ts","sourceRoot":"","sources":["../src/useStreamRuntime.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAEV,iBAAiB,EACjB,eAAe,EACf,uBAAuB,EACvB,sBAAsB,EACvB,MAAM,oBAAoB,CAAC;AAU5B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAa,KAAK,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpE,OAAO,KAAK,EAAE,oBAAoB,EAAqB,mBAAgB;AA6BvE,KAAK,4BAA4B,GAAG;IAClC,KAAK,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACnC,QAAQ,CAAC,EACL;QACE,WAAW,CAAC,EAAE,iBAAiB,GAAG,SAAS,CAAC;QAC5C,MAAM,CAAC,EAAE,sBAAsB,GAAG,SAAS,CAAC;QAC5C,QAAQ,CAAC,EAAE,eAAe,GAAG,SAAS,CAAC;KACxC,GACD,SAAS,CAAC;IACd;;;;;OAKG;IACH,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD;;;OAGG;IACH,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD;;;OAGG;IACH,0BAA0B,CAAC,EAAE,uBAAuB,GAAG,SAAS,CAAC;IACjE,mEAAmE;IACnE,MAAM,CAAC,EAAE,CAAC,MAAM,OAAO,CAAC;QAAE,UAAU,EAAE,MAAM,GAAG,SAAS,CAAA;KAAE,CAAC,CAAC,GAAG,SAAS,CAAC;IACzE,mEAAmE;IACnE,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,CAAC;CAC5D,CAAC;AAqBF,MAAM,MAAM,uBAAuB,GAAG,gBAAgB,SAAS,MAAM,CAAC,GAClE,CAAC,SAAS,gBAAgB,GACxB,CAAC,GAAG,4BAA4B,GAChC,KAAK,GACP,KAAK,CAAC;AA8MV;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,eAAO,MAAM,gBAAgB,GAAI,YAAY,uBAAuB,kDA2BnE,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,0BAA0B;YAnVhB,OAAO;aAyV7B,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,kBAAkB,SAG3B,QAAQ,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,GAAG,SAAS,EAClD,UAAU,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,kBAMpC,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,gBAAgB,SAGzB,UAAU,SAAS,oBAAoB,EAAE,EACzC,UAAU,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,kBAOpC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,uBAAuB,SAE1B,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,kBACzC,CAAC;AAEF;;;;;;;;;;GAUG;AACH,wBAAgB,iBAAiB,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,GAAG,CAAC,GAAG,SAAS,CAAC;AACjE,wBAAgB,iBAAiB,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,GAAG,CAAC,CAAC"}
+{"version":3,"file":"useStreamRuntime.d.ts","names":[],"sources":["../src/useStreamRuntime.ts"],"mappings":";;;;;;KAkDK,4BAAA;EACH,KAAA,GAAQ,cAAA;EACR,QAAA;IAEM,WAAA,GAAc,iBAAA;IACd,MAAA,GAAS,sBAAA;IACT,QAAA,GAAW,eAAA;EAAA;EADF;;;;;;EAUf,0BAAA;EAdA;;;;EAmBA,0BAAA;EAfM;;;;EAoBN,0BAAA,GAA6B,uBAAA,cAL7B;EAOA,MAAA,UAAgB,OAAA;IAAU,UAAA;EAAA,iBAAV;EAEhB,MAAA,KAAW,QAAA,aAAqB,OAAA;AAAA;AAAA,KAsBtB,uBAAA,GAA0B,gBAAA,mBAClC,CAAA,SAAU,gBAAA,GACR,CAAA,GAAI,4BAAA;;;AAxB+B;AAsBzC;;;;;;;;;;;;;;;;AAEsC;AAyMtC;;;;cAAa,gBAAA,GAAoB,UAAA,EAAY,uBAAuB,kCAAA,gBAAA;;;;cAgCvD,0BAAA;EAMZ,KAAA;AAAA;;;;AAOD;;cAAa,kBAAA,SAGT,MAAA,EAAQ,MAAA,sCACR,OAAA,GAAU,MAAA,sBAAuB,OAAA;;;;;;cAaxB,gBAAA,SAGT,QAAA,WAAmB,oBAAA,IACnB,OAAA,GAAU,MAAA,sBAAuB,OAAA;;;;;;AAjBA;AAarC;cAoBa,uBAAA,SAEH,OAAA,EAAS,MAAA,sBAAuB,OAAA;;;;;;;;;;;;iBAc1B,iBAAA,GAAA,CAAqB,GAAA,WAAc,CAAC;AAAA,iBACpC,iBAAA,GAAA,CAAqB,GAAA,UAAa,YAAA,EAAc,CAAA,GAAI,CAAC"}

package/dist/useStreamRuntime.js

@@ -1,280 +1,217 @@
 "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 } from "@assistant-ui/core/react";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import { useStream } from "@langchain/react";
-import { convertLangChainBaseMessage, getMessageType } 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 = 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 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";
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const externalId = useAuiState((s) => s.threadListItem.externalId);
-    // Mutate in place rather than `{ ...options, threadId }`: spreading
-    // `UseStreamOptions` (a discriminated union on `transport`) into an object
-    // literal merges both arms' transport types, breaking arm assignment.
-    options.threadId = externalId;
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const stream = useStream(options);
-    // 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,
-        messagesKey,
-    }), [
-        stream.interrupt,
-        stream.interrupts,
-        stream.submit,
-        stream.values,
-        messagesKey,
-    ]);
-    // 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);
-            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 !== undefined && { artifact }),
-                        status: isError ? "error" : "success",
-                    },
-                ],
-            });
-        },
-        onCancel: unstable_allowCancellation !== false
-            ? async () => {
-                await stream.stop();
-                await toolInvocations.abort();
-            }
-            : undefined,
-    });
-    return runtime;
+	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 streamRef = useRef(stream);
+	streamRef.current = stream;
+	return 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
+		]),
+		unstable_enableToolInvocations: true,
+		setToolStatuses,
+		onNew: async (msg) => {
+			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();
+		} : void 0
+	});
 };
 /**
- * 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 = (rawOptions) => {
-    const { cloud, unstable_threadListAdapter, create, delete: deleteFn, ...options } = rawOptions;
-    const optionsRef = useRef(options);
-    optionsRef.current = options;
-    const cloudAdapter = useCloudThreadListAdapter({
-        cloud,
-        create,
-        delete: deleteFn,
-    });
-    const adapter = unstable_threadListAdapter ?? cloudAdapter;
-    return useRemoteThreadListRuntime({
-        runtimeHook: function RuntimeHook() {
-            // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-            return useStreamThreadRuntime(optionsRef.current);
-        },
-        adapter,
-        allowNesting: true,
-    });
+* 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.
- */
-export const useLangChainInterruptState = () => {
-    return useAuiState((s) => {
-        const extras = s.thread.extras;
-        if (!extras)
-            return undefined;
-        return asLangChainRuntimeExtras(extras).interrupt;
-    });
+* 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;
+	});
 };
 /**
- * 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);
-    };
+* 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);
+	};
 };
 /**
- * Submit a list of LangChain-shaped messages on the current thread.
- * Parity helper for migrating from `useLangGraphSend`. Routes to
- * `useStream().submit({ [messagesKey]: messages }, options)`.
- */
-export const useLangChainSend = () => {
-    const aui = useAui();
-    return (messages, options) => {
-        const { submit, messagesKey } = asLangChainRuntimeExtras(aui.thread().getState().extras);
-        return submit({ [messagesKey]: messages }, options);
-    };
+* 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);
+	};
 };
 /**
- * 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.
- */
-export const useLangChainSendCommand = () => {
-    const submit = useLangChainSubmit();
-    return (command) => submit(null, { command });
+* 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;AAQlD,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,cAAc,EAAE,6BAA0B;AAEhF,MAAM,4BAA4B,GAAG,MAAM,CAAC,0BAA0B,CAAC,CAAC;AAcxE,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;AAkCF,MAAM,mBAAmB,GAAG,CAC1B,QAAyC,EACpB,EAAE;IACvB,MAAM,OAAO,GAAG,IAAI,GAAG,EAA6B,CAAC;IACrD,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;QAC/B,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAClB,KAAK,MAAM,EAAE,IAAI,CAAC,CAAC,UAAU,IAAI,EAAE;gBAAE,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9D,CAAC;aAAM,IAAI,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,YAAY,EAAE,CAAC;YAC7C,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IACD,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;AAC/B,CAAC,CAAC;AAYF,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;AAMF,MAAM,sBAAsB,GAAG,CAC7B,OAGC,EACD,EAAE;IACF,MAAM,EAAE,QAAQ,EAAE,0BAA0B,EAAE,0BAA0B,EAAE,GACxE,OAAO,CAAC;IACV,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,UAAU,CAAC;IAEtD,6FAA6F;IAC7F,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,UAAU,CAEzD,CAAC;IACT,oEAAoE;IACpE,2EAA2E;IAC3E,sEAAsE;IACtE,OAAO,CAAC,QAAQ,GAAG,UAAU,CAAC;IAE9B,6FAA6F;IAC7F,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC;IAElC,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;QACrB,WAAW;KACZ,CAAC,EACF;QACE,MAAM,CAAC,SAAS;QAChB,MAAM,CAAC,UAAU;QACjB,MAAM,CAAC,MAAM;QACb,MAAM,CAAC,MAAM;QACb,WAAW;KACZ,CACF,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,aAAa,GACjB,0BAA0B,KAAK,KAAK;gBAClC,CAAC,CAAC,mBAAmB,CACjB,SAAS,CAAC,OAAO,CAAC,QAA2C,CAC9D,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACZ,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,YAAY,EAAE,CAAC,CAAC,EAAE;oBAClB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;oBAC5C,MAAM,EAAE,OAAgB;iBACzB,CAAC,CAAC;gBACL,CAAC,CAAC,EAAE,CAAC;YACT,MAAM,MAAM,CAAC,MAAM,CAAC;gBAClB,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;aAC9D,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,EACN,0BAA0B,KAAK,KAAK;YAClC,CAAC,CAAC,KAAK,IAAI,EAAE;gBACT,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBACpB,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;YAChC,CAAC;YACH,CAAC,CAAC,SAAS;KAChB,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,UAAmC,EAAE,EAAE;IACtE,MAAM,EACJ,KAAK,EACL,0BAA0B,EAC1B,MAAM,EACN,MAAM,EAAE,QAAQ,EAChB,GAAG,OAAO,EACX,GAAG,UAAU,CAAC;IAEf,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;IACnC,UAAU,CAAC,OAAO,GAAG,OAAO,CAAC;IAE7B,MAAM,YAAY,GAAG,yBAAyB,CAAC;QAC7C,KAAK;QACL,MAAM;QACN,MAAM,EAAE,QAAQ;KACjB,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,0BAA0B,IAAI,YAAY,CAAC;IAE3D,OAAO,0BAA0B,CAAC;QAChC,WAAW,EAAE,SAAS,WAAW;YAC/B,6FAA6F;YAC7F,OAAO,sBAAsB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC;QACD,OAAO;QACP,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;;;;GAIG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,GAAG,EAAE;IACnC,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IACrB,OAAO,CACL,QAAyC,EACzC,OAAiC,EACjC,EAAE;QACF,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,wBAAwB,CACtD,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,CAC/B,CAAC;QACF,OAAO,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,GAAG,EAAE;IAC1C,MAAM,MAAM,GAAG,kBAAkB,EAAE,CAAC;IACpC,OAAO,CAAC,OAAgC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;AACzE,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":["/// <reference types=\"@assistant-ui/core/store\" />\n\"use client\";\n\nimport { useMemo, useRef, useState } from \"react\";\nimport type {\n  AppendMessage,\n  AttachmentAdapter,\n  FeedbackAdapter,\n  RemoteThreadListAdapter,\n  SpeechSynthesisAdapter,\n  ToolExecutionStatus,\n} 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 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 streamRef = useRef(stream);\n  streamRef.current = stream;\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    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    onNew: async (msg) => {\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          }\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,YAAY,OAAO,MAAM;CAC/B,UAAU,UAAU;CA2EpB,OArDgB,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,gCAAgC;EAChC;EACA,OAAO,OAAO,QAAQ;GACpB,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;EACpB,IACA,KAAA;CACR,CAEa;AACf;;;;;;;;;;;;;;;;;;;;;;;;;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.7",
+  "version": "0.0.10",
   "description": "LangChain useStream adapter for assistant-ui",
   "keywords": [
     "langchain",
@@ -29,10 +29,10 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.2.3",
-    "@assistant-ui/store": "^0.2.11",
+    "@assistant-ui/core": "^0.2.7",
+    "@assistant-ui/store": "^0.2.12",
     "assistant-cloud": "*",
-    "assistant-stream": "^0.3.15"
+    "assistant-stream": "^0.3.16"
   },
   "peerDependencies": {
     "@langchain/react": "^1.0.2",
@@ -45,15 +45,15 @@
     }
   },
   "devDependencies": {
-    "@langchain/core": "^1.1.44",
-    "@langchain/langgraph-sdk": "^1.9.2",
-    "@langchain/react": "^1.0.2",
+    "@langchain/core": "^1.1.48",
+    "@langchain/langgraph-sdk": "^1.9.6",
+    "@langchain/react": "^1.0.6",
     "@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.8"
+    "react": "^19.2.6",
+    "vitest": "^4.1.7",
+    "@assistant-ui/x-buildutils": "0.0.9"
   },
   "publishConfig": {
     "access": "public",

package/src/useStreamRuntime.ts

@@ -1,3 +1,4 @@
+/// <reference types="@assistant-ui/core/store" />
 "use client";
 
 import { useMemo, useRef, useState } from "react";
@@ -7,14 +8,13 @@ import type {
   FeedbackAdapter,
   RemoteThreadListAdapter,
   SpeechSynthesisAdapter,
+  ToolExecutionStatus,
 } from "@assistant-ui/core";
 import {
   useCloudThreadListAdapter,
   useExternalStoreRuntime,
   useExternalMessageConverter,
   useRemoteThreadListRuntime,
-  useToolInvocations,
-  type ToolExecutionStatus,
 } from "@assistant-ui/core/react";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import type { AssistantCloud } from "assistant-cloud";
@@ -137,8 +137,12 @@ const getMessageContent = (msg: AppendMessage) => {
       case "tool-call":
         throw new Error("Tool call appends are not supported.");
       default: {
-        const _exhaustiveCheck: "reasoning" | "source" | "audio" | "data" =
-          type;
+        const _exhaustiveCheck:
+          | "reasoning"
+          | "source"
+          | "audio"
+          | "data"
+          | "generative-ui" = type;
         throw new Error(
           `Unsupported append message part type: ${_exhaustiveCheck}`,
         );
@@ -194,45 +198,10 @@ const useStreamThreadRuntime = (
     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(
     (): LangChainRuntimeExtras => ({
@@ -258,8 +227,9 @@ const useStreamThreadRuntime = (
     messages: threadMessages,
     adapters,
     extras,
+    unstable_enableToolInvocations: true,
+    setToolStatuses,
     onNew: async (msg) => {
-      await toolInvocations.abort();
       const content = getMessageContent(msg);
       const cancellations =
         autoCancelPendingToolCalls !== false
@@ -301,7 +271,6 @@ const useStreamThreadRuntime = (
       unstable_allowCancellation !== false
         ? async () => {
             await stream.stop();
-            await toolInvocations.abort();
           }
         : undefined,
   });

package/dist/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gBAAgB,EAChB,0BAA0B,EAC1B,gBAAgB,EAChB,uBAAuB,EACvB,iBAAiB,EACjB,kBAAkB,GACnB,8BAA2B;AAC5B,YAAY,EAAE,uBAAuB,EAAE,8BAA2B;AAElE,OAAO,EAAE,2BAA2B,EAAE,6BAA0B;AAEhE,YAAY,EACV,oBAAoB,EACpB,qBAAqB,EACrB,iBAAiB,GAClB,mBAAgB"}

package/dist/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gBAAgB,EAChB,0BAA0B,EAC1B,gBAAgB,EAChB,uBAAuB,EACvB,iBAAiB,EACjB,kBAAkB,GACnB,8BAA2B;AAG5B,OAAO,EAAE,2BAA2B,EAAE,6BAA0B"}

package/dist/types.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}