@assistant-ui/react-langchain 0.0.15 → 0.0.17

package/dist/convertMessages.d.ts

@@ -1,9 +1,43 @@
-import { LangChainBaseMessage } from "./types.js";
+import { LangChainBaseMessage, UIMessage } from "./types.js";
+import { AppendMessage, MessageTiming } from "@assistant-ui/core";
 import { useExternalMessageConverter } from "@assistant-ui/core/react";
 
 //#region src/convertMessages.d.ts
+type LangChainMessageConverterMetadata = useExternalMessageConverter.Metadata & {
+  uiMessagesByParent?: Map<string, UIMessage[]>;
+  messageTiming?: Record<string, MessageTiming>;
+};
 declare const getMessageType: (message: LangChainBaseMessage) => string;
-declare const convertLangChainBaseMessage: useExternalMessageConverter.Callback<LangChainBaseMessage>;
+declare const convertLangChainBaseMessage: (message: LangChainBaseMessage, metadata?: LangChainMessageConverterMetadata) => useExternalMessageConverter.Message;
+declare const getMessageContent: (msg: AppendMessage) => string | ({
+  type: "text";
+  text: string;
+  image_url?: never;
+  data?: never;
+  mime_type?: never;
+  metadata?: never;
+  source_type?: never;
+} | {
+  type: "image_url";
+  image_url: {
+    url: string;
+  };
+  text?: never;
+  data?: never;
+  mime_type?: never;
+  metadata?: never;
+  source_type?: never;
+} | {
+  type: "file";
+  data: string;
+  mime_type: string;
+  metadata: {
+    filename: string;
+  };
+  source_type: "base64";
+  text?: never;
+  image_url?: never;
+})[];
 //#endregion
-export { convertLangChainBaseMessage, getMessageType };
+export { convertLangChainBaseMessage, getMessageContent, getMessageType };
 //# sourceMappingURL=convertMessages.d.ts.map

package/dist/convertMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessages.d.ts","names":[],"sources":["../src/convertMessages.ts"],"mappings":";;;;cAMa,cAAA,GAAkB,OAA6B,EAApB,oBAAoB;AAAA,cA8D/C,2BAAA,EAA6B,2BAAA,CAA4B,QAAQ,CAC5E,oBAAA"}
+{"version":3,"file":"convertMessages.d.ts","names":[],"sources":["../src/convertMessages.ts"],"mappings":";;;;;KAeK,iCAAA,GACH,2BAAA,CAA4B,QAAA;EAC1B,kBAAA,GAAqB,GAAA,SAAY,SAAA;EACjC,aAAA,GAAgB,MAAA,SAAe,aAAA;AAAA;AAAA,cAStB,cAAA,GAAkB,OAA6B,EAApB,oBAAoB;AAAA,cAoE/C,2BAAA,GACX,OAAA,EAAS,oBAAA,EACT,QAAA,GAAU,iCAAA,KACT,2BAAA,CAA4B,OAAA;AAAA,cAyFlB,iBAAA,GAAqB,GAAA,EAAK,aAAa"}

package/dist/convertMessages.js

@@ -1,5 +1,10 @@
 "use client";
 //#region src/convertMessages.ts
+const uiMessageToDataPart = (ui) => ({
+	type: "data",
+	name: ui.name,
+	data: ui.props
+});
 const getMessageType = (message) => {
 	if (typeof message._getType === "function") return message._getType();
 	if ("type" in message) return message.type;
@@ -17,15 +22,14 @@ const contentToParts = (content) => {
 				type: "text",
 				text: part.text
 			};
-			case "image_url":
-				if (typeof part.image_url === "string") return {
-					type: "image",
-					image: part.image_url
-				};
+			case "image_url": {
+				const image = typeof part.image_url === "string" ? part.image_url : part.image_url?.url;
+				if (!image) return null;
 				return {
 					type: "image",
-					image: part.image_url.url
+					image
 				};
+			}
 			case "file": return {
 				type: "file",
 				filename: part.metadata?.filename ?? "file",
@@ -38,7 +42,7 @@ const contentToParts = (content) => {
 			};
 			case "reasoning": return {
 				type: "reasoning",
-				text: part.summary.map((s) => s.text).join("\n\n\n")
+				text: part.summary?.map((s) => s?.text ?? "").join("\n\n\n") ?? part.reasoning ?? ""
 			};
 			case "tool_use":
 			case "input_json_delta": return null;
@@ -51,7 +55,7 @@ const getStringContent = (content) => {
 	if (typeof content === "string") return content;
 	return content.filter((c) => c.type === "text").map((c) => c.text).join("");
 };
-const convertLangChainBaseMessage = (message) => {
+const convertLangChainBaseMessage = (message, metadata = {}) => {
 	switch (getMessageType(message)) {
 		case "system": return {
 			role: "system",
@@ -77,11 +81,20 @@ const convertLangChainBaseMessage = (message) => {
 				argsText: JSON.stringify(tc.args)
 			})) ?? [];
 			const assistantStatus = typeof message.status === "object" ? message.status : void 0;
+			const uiDataParts = (message.id ? metadata.uiMessagesByParent?.get(message.id)?.map(uiMessageToDataPart) : void 0) ?? [];
+			const timing = metadata.messageTiming?.[message.id ?? ""];
 			return {
 				role: "assistant",
 				id: message.id,
-				content: [...contentToParts(message.content), ...toolCallParts],
-				metadata: { custom: getCustomMetadata(message.additional_kwargs) },
+				content: [
+					...contentToParts(message.content),
+					...toolCallParts,
+					...uiDataParts
+				],
+				metadata: {
+					custom: getCustomMetadata(message.additional_kwargs),
+					...timing && { timing }
+				},
 				...assistantStatus && { status: assistantStatus }
 			};
 		}
@@ -103,7 +116,40 @@ const convertLangChainBaseMessage = (message) => {
 		};
 	}
 };
+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: throw new Error(`Unsupported append message part type: ${type}`);
+		}
+	});
+	if (content.length === 1 && content[0]?.type === "text") return content[0].text ?? "";
+	return content;
+};
 //#endregion
-export { convertLangChainBaseMessage, getMessageType };
+export { convertLangChainBaseMessage, getMessageContent, getMessageType };
 
 //# sourceMappingURL=convertMessages.js.map

package/dist/convertMessages.js.map

@@ -1 +1 @@
-{"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 \"file\":\n          return {\n            type: \"file\" as const,\n            filename: part.metadata?.filename ?? \"file\",\n            data: part.data,\n            mimeType: part.mime_type,\n          };\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,QACH,OAAO;IACL,MAAM;IACN,UAAU,KAAK,UAAU,YAAY;IACrC,MAAM,KAAK;IACX,UAAU,KAAK;GACjB;GACF,KAAK,YACH,OAAO;IAAE,MAAM;IAAsB,MAAM,KAAK;GAAS;GAC3D,KAAK,aACH,OAAO;IACL,MAAM;IACN,MAAM,KAAK,QAAQ,KAAK,MAAM,EAAE,IAAI,CAAC,CAAC,KAAK,QAAQ;GACrD;GACF,KAAK;GACL,KAAK,oBACH,OAAO;GACT,SACE,OAAO;EACX;CACF,CAAC,CAAC,CACD,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,CAAC,CACrE,KAAK,MAAM,EAAE,IAAI,CAAC,CAClB,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"}
+{"version":3,"file":"convertMessages.js","names":["parts","_exhaustiveCheck"],"sources":["../src/convertMessages.ts"],"sourcesContent":["\"use client\";\n\nimport type { useExternalMessageConverter } from \"@assistant-ui/core/react\";\nimport type {\n  AppendMessage,\n  DataMessagePart,\n  MessageTiming,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport type {\n  LangChainBaseMessage,\n  LangChainContentBlock,\n  UIMessage,\n} from \"./types\";\n\ntype LangChainMessageConverterMetadata =\n  useExternalMessageConverter.Metadata & {\n    uiMessagesByParent?: Map<string, UIMessage[]>;\n    messageTiming?: Record<string, MessageTiming>;\n  };\n\nconst uiMessageToDataPart = (ui: UIMessage): DataMessagePart => ({\n  type: \"data\",\n  name: ui.name,\n  data: ui.props,\n});\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          const image =\n            typeof part.image_url === \"string\"\n              ? part.image_url\n              : part.image_url?.url;\n          if (!image) return null;\n          return { type: \"image\" as const, image };\n        }\n        case \"file\":\n          return {\n            type: \"file\" as const,\n            filename: part.metadata?.filename ?? \"file\",\n            data: part.data,\n            mimeType: part.mime_type,\n          };\n        case \"thinking\":\n          return { type: \"reasoning\" as const, text: part.thinking };\n        case \"reasoning\":\n          return {\n            type: \"reasoning\" as const,\n            text:\n              part.summary?.map((s) => s?.text ?? \"\").join(\"\\n\\n\\n\") ??\n              part.reasoning ??\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 = (\n  message: LangChainBaseMessage,\n  metadata: LangChainMessageConverterMetadata = {},\n): useExternalMessageConverter.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      const uiDataParts =\n        (message.id\n          ? metadata.uiMessagesByParent\n              ?.get(message.id)\n              ?.map(uiMessageToDataPart)\n          : undefined) ?? [];\n\n      const timing = metadata.messageTiming?.[message.id ?? \"\"];\n\n      return {\n        role: \"assistant\",\n        id: message.id,\n        content: [\n          ...contentToParts(message.content),\n          ...toolCallParts,\n          ...uiDataParts,\n        ],\n        metadata: {\n          custom: getCustomMetadata(message.additional_kwargs),\n          ...(timing && { timing }),\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\nexport const 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"],"mappings":";;AAqBA,MAAM,uBAAuB,QAAoC;CAC/D,MAAM;CACN,MAAM,GAAG;CACT,MAAM,GAAG;AACX;AAEA,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,aAAa;IAChB,MAAM,QACJ,OAAO,KAAK,cAAc,WACtB,KAAK,YACL,KAAK,WAAW;IACtB,IAAI,CAAC,OAAO,OAAO;IACnB,OAAO;KAAE,MAAM;KAAkB;IAAM;GACzC;GACA,KAAK,QACH,OAAO;IACL,MAAM;IACN,UAAU,KAAK,UAAU,YAAY;IACrC,MAAM,KAAK;IACX,UAAU,KAAK;GACjB;GACF,KAAK,YACH,OAAO;IAAE,MAAM;IAAsB,MAAM,KAAK;GAAS;GAC3D,KAAK,aACH,OAAO;IACL,MAAM;IACN,MACE,KAAK,SAAS,KAAK,MAAM,GAAG,QAAQ,EAAE,CAAC,CAAC,KAAK,QAAQ,KACrD,KAAK,aACL;GACJ;GACF,KAAK;GACL,KAAK,oBACH,OAAO;GACT,SACE,OAAO;EACX;CACF,CAAC,CAAC,CACD,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,CAAC,CACrE,KAAK,MAAM,EAAE,IAAI,CAAC,CAClB,KAAK,EAAE;AACZ;AAEA,MAAa,+BACX,SACA,WAA8C,CAAC,MACP;CAGxC,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,MAAM,eACH,QAAQ,KACL,SAAS,oBACL,IAAI,QAAQ,EAAE,CAAC,EACf,IAAI,mBAAmB,IAC3B,KAAA,MAAc,CAAC;GAErB,MAAM,SAAS,SAAS,gBAAgB,QAAQ,MAAM;GAEtD,OAAO;IACL,MAAM;IACN,IAAI,QAAQ;IACZ,SAAS;KACP,GAAG,eAAe,QAAQ,OAAO;KACjC,GAAG;KACH,GAAG;IACL;IACA,UAAU;KACR,QAAQ,kBAAkB,QAAQ,iBAAiB;KACnD,GAAI,UAAU,EAAE,OAAO;IACzB;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;AAEA,MAAa,qBAAqB,QAAuB;CACvD,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,yCAAyCC,MAC3C;EAEJ;CACF,CAAC;CAED,IAAI,QAAQ,WAAW,KAAK,QAAQ,EAAE,EAAE,SAAS,QAC/C,OAAO,QAAQ,EAAE,CAAC,QAAQ;CAE5B,OAAO;AACT"}

package/dist/findForkCheckpointInHistory.d.ts

@@ -0,0 +1,26 @@
+import { LangChainBaseMessage } from "./types.js";
+
+//#region src/findForkCheckpointInHistory.d.ts
+type ForkCheckpointHistoryState = {
+  values?: Record<string, unknown> | undefined;
+  checkpoint?: {
+    checkpoint_id?: string | null | undefined;
+  } | undefined;
+};
+type ForkHistoryCursor = {
+  configurable: {
+    checkpoint_id: string;
+  };
+};
+type ForkCheckpointClient = {
+  threads: {
+    getHistory: (threadId: string, options?: {
+      limit?: number;
+      before?: ForkHistoryCursor;
+    }) => Promise<readonly ForkCheckpointHistoryState[]>;
+  };
+};
+declare const findForkCheckpointInHistory: (client: ForkCheckpointClient, threadId: string, messagesUpToParent: readonly LangChainBaseMessage[], messagesKey: string, pageSize?: number) => Promise<string | null>;
+//#endregion
+export { ForkCheckpointClient, findForkCheckpointInHistory };
+//# sourceMappingURL=findForkCheckpointInHistory.d.ts.map

package/dist/findForkCheckpointInHistory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"findForkCheckpointInHistory.d.ts","names":[],"sources":["../src/findForkCheckpointInHistory.ts"],"mappings":";;;KAEK,0BAAA;EACH,MAAA,GAAS,MAAM;EACf,UAAA;IAAe,aAAA;EAAA;AAAA;AAAA,KAGZ,iBAAA;EAAsB,YAAA;IAAgB,aAAa;EAAA;AAAA;AAAA,KAE5C,oBAAA;EACV,OAAA;IACE,UAAA,GACE,QAAA,UACA,OAAA;MAAY,KAAA;MAAgB,MAAA,GAAS,iBAAA;IAAA,MAClC,OAAA,UAAiB,0BAAA;EAAA;AAAA;AAAA,cAIb,2BAAA,GACX,MAAA,EAAQ,oBAAA,EACR,QAAA,UACA,kBAAA,WAA6B,oBAAA,IAC7B,WAAA,UACA,QAAA,cACC,OAAA"}

package/dist/findForkCheckpointInHistory.js

@@ -0,0 +1,34 @@
+//#region src/findForkCheckpointInHistory.ts
+const findForkCheckpointInHistory = async (client, threadId, messagesUpToParent, messagesKey, pageSize = 100) => {
+	if (!messagesUpToParent.every((m) => typeof m.id === "string")) return null;
+	const scanned = /* @__PURE__ */ new Set();
+	let before;
+	for (;;) {
+		const page = await client.threads.getHistory(threadId, before ? {
+			limit: pageSize,
+			before
+		} : { limit: pageSize });
+		let oldestCheckpointId;
+		let advanced = false;
+		for (const state of page) {
+			const checkpointId = state.checkpoint?.checkpoint_id;
+			if (typeof checkpointId === "string") {
+				oldestCheckpointId = checkpointId;
+				if (!scanned.has(checkpointId)) {
+					scanned.add(checkpointId);
+					advanced = true;
+				}
+			}
+			const stateMessages = state.values?.[messagesKey];
+			if (!Array.isArray(stateMessages) || stateMessages.length !== messagesUpToParent.length) continue;
+			if (!stateMessages.every((m) => typeof m.id === "string")) continue;
+			if (messagesUpToParent.every((m, i) => m.id === stateMessages[i]?.id) && checkpointId) return checkpointId;
+		}
+		if (page.length < pageSize || oldestCheckpointId === void 0 || !advanced) return null;
+		before = { configurable: { checkpoint_id: oldestCheckpointId } };
+	}
+};
+//#endregion
+export { findForkCheckpointInHistory };
+
+//# sourceMappingURL=findForkCheckpointInHistory.js.map

package/dist/findForkCheckpointInHistory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"findForkCheckpointInHistory.js","names":[],"sources":["../src/findForkCheckpointInHistory.ts"],"sourcesContent":["import type { LangChainBaseMessage } from \"./types\";\n\ntype ForkCheckpointHistoryState = {\n  values?: Record<string, unknown> | undefined;\n  checkpoint?: { checkpoint_id?: string | null | undefined } | undefined;\n};\n\ntype ForkHistoryCursor = { configurable: { checkpoint_id: string } };\n\nexport type ForkCheckpointClient = {\n  threads: {\n    getHistory: (\n      threadId: string,\n      options?: { limit?: number; before?: ForkHistoryCursor },\n    ) => Promise<readonly ForkCheckpointHistoryState[]>;\n  };\n};\n\nexport const findForkCheckpointInHistory = async (\n  client: ForkCheckpointClient,\n  threadId: string,\n  messagesUpToParent: readonly LangChainBaseMessage[],\n  messagesKey: string,\n  pageSize = 100,\n): Promise<string | null> => {\n  if (!messagesUpToParent.every((m) => typeof m.id === \"string\")) return null;\n\n  const scanned = new Set<string>();\n  let before: ForkHistoryCursor | undefined;\n\n  for (;;) {\n    const page = await client.threads.getHistory(\n      threadId,\n      before ? { limit: pageSize, before } : { limit: pageSize },\n    );\n\n    let oldestCheckpointId: string | undefined;\n    let advanced = false;\n    for (const state of page) {\n      const checkpointId = state.checkpoint?.checkpoint_id;\n      if (typeof checkpointId === \"string\") {\n        oldestCheckpointId = checkpointId;\n        if (!scanned.has(checkpointId)) {\n          scanned.add(checkpointId);\n          advanced = true;\n        }\n      }\n      const stateMessages = state.values?.[messagesKey] as\n        | readonly LangChainBaseMessage[]\n        | undefined;\n      if (\n        !Array.isArray(stateMessages) ||\n        stateMessages.length !== messagesUpToParent.length\n      )\n        continue;\n      if (!stateMessages.every((m) => typeof m.id === \"string\")) continue;\n      const isMatch = messagesUpToParent.every(\n        (m, i) => m.id === stateMessages[i]?.id,\n      );\n      if (isMatch && checkpointId) return checkpointId;\n    }\n\n    // `advanced` halts paging if the backend ignores `before` and repeats a page\n    if (page.length < pageSize || oldestCheckpointId === undefined || !advanced)\n      return null;\n    before = { configurable: { checkpoint_id: oldestCheckpointId } };\n  }\n};\n"],"mappings":";AAkBA,MAAa,8BAA8B,OACzC,QACA,UACA,oBACA,aACA,WAAW,QACgB;CAC3B,IAAI,CAAC,mBAAmB,OAAO,MAAM,OAAO,EAAE,OAAO,QAAQ,GAAG,OAAO;CAEvE,MAAM,0BAAU,IAAI,IAAY;CAChC,IAAI;CAEJ,SAAS;EACP,MAAM,OAAO,MAAM,OAAO,QAAQ,WAChC,UACA,SAAS;GAAE,OAAO;GAAU;EAAO,IAAI,EAAE,OAAO,SAAS,CAC3D;EAEA,IAAI;EACJ,IAAI,WAAW;EACf,KAAK,MAAM,SAAS,MAAM;GACxB,MAAM,eAAe,MAAM,YAAY;GACvC,IAAI,OAAO,iBAAiB,UAAU;IACpC,qBAAqB;IACrB,IAAI,CAAC,QAAQ,IAAI,YAAY,GAAG;KAC9B,QAAQ,IAAI,YAAY;KACxB,WAAW;IACb;GACF;GACA,MAAM,gBAAgB,MAAM,SAAS;GAGrC,IACE,CAAC,MAAM,QAAQ,aAAa,KAC5B,cAAc,WAAW,mBAAmB,QAE5C;GACF,IAAI,CAAC,cAAc,OAAO,MAAM,OAAO,EAAE,OAAO,QAAQ,GAAG;GAI3D,IAHgB,mBAAmB,OAChC,GAAG,MAAM,EAAE,OAAO,cAAc,EAAE,EAAE,EAE7B,KAAK,cAAc,OAAO;EACtC;EAGA,IAAI,KAAK,SAAS,YAAY,uBAAuB,KAAA,KAAa,CAAC,UACjE,OAAO;EACT,SAAS,EAAE,cAAc,EAAE,eAAe,mBAAmB,EAAE;CACjE;AACF"}

package/dist/hooks.d.ts

@@ -0,0 +1,88 @@
+import { LangChainBaseMessage } from "./types.js";
+import { AssembledToolCall, SubagentDiscoverySnapshot, SubgraphDiscoverySnapshot } from "@langchain/react";
+
+//#region src/hooks.d.ts
+/**
+ * Read the current LangGraph interrupt state from the runtime extras.
+ */
+declare const useLangChainInterruptState: () => {
+  value?: unknown;
+} | undefined;
+/**
+ * Read every interrupt pending at the current checkpoint, each with `id`
+ * and `value`. Defaults to an empty array. Pair with `useLangChainRespondAll`
+ * (keyed by interrupt id) to resolve several at once.
+ */
+declare const useLangChainInterrupts: () => readonly {
+  id?: string;
+  value?: unknown;
+}[];
+/** Read the last run/hydration error from the runtime extras. */
+declare const useLangChainError: () => unknown;
+/**
+ * Read the root tool calls assembled by `useStream` from the `tools`
+ * channel. Defaults to an empty array, so consumers can `.map` without
+ * a guard. Useful for rendering pending/streamed tool calls and
+ * approval UIs.
+ */
+declare const useLangChainToolCalls: () => readonly AssembledToolCall<string, unknown, unknown>[];
+/** Subagents discovered on the current run (multi-agent graphs). */
+declare const useLangChainSubagents: () => ReadonlyMap<string, SubagentDiscoverySnapshot>;
+/** Subgraphs discovered on the current run. */
+declare const useLangChainSubgraphs: () => ReadonlyMap<string, SubgraphDiscoverySnapshot>;
+/**
+ * The underlying `useStream` handle, for advanced views — pass it to v1's
+ * `useMessages(stream, target)` / `useToolCalls(stream, target)` with a
+ * subagent/subgraph from `useLangChainSubagents` / `useLangChainSubgraphs`.
+ * `undefined` outside the runtime provider.
+ */
+declare const useLangChainStream: () => import("@langchain/react").AnyStream | undefined;
+/**
+ * Returns a function to submit raw state updates to the LangGraph agent,
+ * bypassing the normal message flow. Useful for sending interrupt resume
+ * commands.
+ */
+declare const useLangChainSubmit: () => (values: Record<string, unknown> | null | undefined, options?: Record<string, unknown>) => Promise<void>;
+/**
+ * Resume a LangGraph interrupt with a response payload via
+ * `useStream().respond`. Preferred over `useLangChainSendCommand`; it
+ * carries the response cleanly and handles interrupt namespaces.
+ */
+declare const useLangChainRespond: () => (response: unknown, options?: Record<string, unknown>) => Promise<void>;
+/**
+ * Resume several LangGraph interrupts pending at the same checkpoint in
+ * one run via `useStream().respondAll`. Use when a run pauses on multiple
+ * interrupts at once; sequential `useLangChainRespond` calls can't service
+ * them (the first resume starts a run, stranding the rest).
+ */
+declare const useLangChainRespondAll: () => (responsesById: Record<string, unknown>, 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)`.
+ */
+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
+ * shape (`{ resume?, goto?, update? }`) differs from the legacy
+ * `{ resume: string }` form; to carry a payload, use the input or
+ * `stream.respond` instead.
+ */
+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
+ * stream emits new state.
+ *
+ * @example
+ * ```tsx
+ * const todos = useLangChainState<Todo[]>("todos");
+ * const files = useLangChainState<Record<string, string>>("files", {});
+ * ```
+ */
+declare function useLangChainState<T>(key: string): T | undefined;
+declare function useLangChainState<T>(key: string, defaultValue: T): T;
+//#endregion
+export { useLangChainError, useLangChainInterruptState, useLangChainInterrupts, useLangChainRespond, useLangChainRespondAll, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainStream, useLangChainSubagents, useLangChainSubgraphs, useLangChainSubmit, useLangChainToolCalls };
+//# sourceMappingURL=hooks.d.ts.map

package/dist/hooks.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hooks.d.ts","names":[],"sources":["../src/hooks.ts"],"mappings":";;;;;;AAwBA;cAAa,0BAAA;EACuC,KAAA;AAAA;;AAOpD;;;;cAAa,sBAAA;EAIV,EAAA;EAAA,KAAA;AAAA;;cAGU,iBAAA;AASb;;;;AAAkC;AAIlC;AAJA,cAAa,qBAAA,iBAAqB,iBAAA;;cAIrB,qBAAA,QAAqB,WAAA,SAAA,yBAAA;AAAA;AAAA,cAIrB,qBAAA,QAAqB,WAAA,SAAA,yBAAA;;;;AAAA;AASlC;;cAAa,kBAAA,mCAAkB,SAAA;;AAAA;AAQ/B;;;cAAa,kBAAA,SAGT,MAAA,EAAQ,MAAA,sCACR,OAAA,GAAU,MAAA,sBAAuB,OAAA;;;;;;cASxB,mBAAA,SAEH,QAAA,WAAmB,OAAA,GAAU,MAAA,sBAAuB,OAAA;;;;;AAXzB;AASrC;cAYa,sBAAA,SAGT,aAAA,EAAe,MAAA,mBACf,OAAA,GAAU,MAAA,sBAAuB,OAAA;;;;;;cASxB,gBAAA,SAGT,QAAA,WAAmB,oBAAA,IACnB,OAAA,GAAU,MAAA,sBAAuB,OAAA;;AA3ByB;AAU9D;;;;;cA+Ba,uBAAA,SAEH,OAAA,EAAS,MAAA,sBAAuB,OAAA;;;;;;;;;AA7BL;AASrC;;iBAkCgB,iBAAA,IAAqB,GAAA,WAAc,CAAC;AAAA,iBACpC,iBAAA,IAAqB,GAAA,UAAa,YAAA,EAAc,CAAA,GAAI,CAAC"}

package/dist/hooks.js

@@ -0,0 +1,99 @@
+"use client";
+import { langChainExtras } from "./runtimeExtras.js";
+import { useAui } from "@assistant-ui/store";
+//#region src/hooks.ts
+const EMPTY_TOOL_CALLS = [];
+const EMPTY_INTERRUPTS = [];
+const EMPTY_SUBAGENTS = /* @__PURE__ */ new Map();
+const EMPTY_SUBGRAPHS = /* @__PURE__ */ new Map();
+/**
+* Read the current LangGraph interrupt state from the runtime extras.
+*/
+const useLangChainInterruptState = () => langChainExtras.use((e) => e.interrupt, void 0);
+/**
+* Read every interrupt pending at the current checkpoint, each with `id`
+* and `value`. Defaults to an empty array. Pair with `useLangChainRespondAll`
+* (keyed by interrupt id) to resolve several at once.
+*/
+const useLangChainInterrupts = () => langChainExtras.use((e) => e.interrupts ?? EMPTY_INTERRUPTS, EMPTY_INTERRUPTS);
+/** Read the last run/hydration error from the runtime extras. */
+const useLangChainError = () => langChainExtras.use((e) => e.error, void 0);
+/**
+* Read the root tool calls assembled by `useStream` from the `tools`
+* channel. Defaults to an empty array, so consumers can `.map` without
+* a guard. Useful for rendering pending/streamed tool calls and
+* approval UIs.
+*/
+const useLangChainToolCalls = () => langChainExtras.use((e) => e.toolCalls ?? EMPTY_TOOL_CALLS, EMPTY_TOOL_CALLS);
+/** Subagents discovered on the current run (multi-agent graphs). */
+const useLangChainSubagents = () => langChainExtras.use((e) => e.subagents ?? EMPTY_SUBAGENTS, EMPTY_SUBAGENTS);
+/** Subgraphs discovered on the current run. */
+const useLangChainSubgraphs = () => langChainExtras.use((e) => e.subgraphs ?? EMPTY_SUBGRAPHS, EMPTY_SUBGRAPHS);
+/**
+* The underlying `useStream` handle, for advanced views — pass it to v1's
+* `useMessages(stream, target)` / `useToolCalls(stream, target)` with a
+* subagent/subgraph from `useLangChainSubagents` / `useLangChainSubgraphs`.
+* `undefined` outside the runtime provider.
+*/
+const useLangChainStream = () => langChainExtras.use((e) => e.stream, void 0);
+/**
+* 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) => langChainExtras.get(aui).submit(values, options);
+};
+/**
+* Resume a LangGraph interrupt with a response payload via
+* `useStream().respond`. Preferred over `useLangChainSendCommand`; it
+* carries the response cleanly and handles interrupt namespaces.
+*/
+const useLangChainRespond = () => {
+	const aui = useAui();
+	return (response, options) => langChainExtras.get(aui).respond(response, options);
+};
+/**
+* Resume several LangGraph interrupts pending at the same checkpoint in
+* one run via `useStream().respondAll`. Use when a run pauses on multiple
+* interrupts at once; sequential `useLangChainRespond` calls can't service
+* them (the first resume starts a run, stranding the rest).
+*/
+const useLangChainRespondAll = () => {
+	const aui = useAui();
+	return (responsesById, options) => langChainExtras.get(aui).respondAll(responsesById, 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 extras = langChainExtras.get(aui);
+		return extras.submit({ [extras.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.
+*/
+const useLangChainSendCommand = () => {
+	const submit = useLangChainSubmit();
+	return (command) => submit(null, { command });
+};
+function useLangChainState(key, defaultValue) {
+	return langChainExtras.use((e) => {
+		const value = e.values[key];
+		return value !== void 0 ? value : defaultValue;
+	}, defaultValue);
+}
+//#endregion
+export { useLangChainError, useLangChainInterruptState, useLangChainInterrupts, useLangChainRespond, useLangChainRespondAll, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainStream, useLangChainSubagents, useLangChainSubgraphs, useLangChainSubmit, useLangChainToolCalls };
+
+//# sourceMappingURL=hooks.js.map

package/dist/hooks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hooks.js","names":[],"sources":["../src/hooks.ts"],"sourcesContent":["\"use client\";\n\nimport { useAui } from \"@assistant-ui/store\";\nimport type {\n  AssembledToolCall,\n  SubagentDiscoverySnapshot,\n  SubgraphDiscoverySnapshot,\n} from \"@langchain/react\";\nimport { langChainExtras } from \"./runtimeExtras\";\nimport type { LangChainBaseMessage } from \"./types\";\n\nconst EMPTY_TOOL_CALLS: readonly AssembledToolCall[] = [];\n\nconst EMPTY_INTERRUPTS: readonly { id?: string; value?: unknown }[] = [];\n\nconst EMPTY_SUBAGENTS: ReadonlyMap<string, SubagentDiscoverySnapshot> =\n  new Map();\n\nconst EMPTY_SUBGRAPHS: ReadonlyMap<string, SubgraphDiscoverySnapshot> =\n  new Map();\n\n/**\n * Read the current LangGraph interrupt state from the runtime extras.\n */\nexport const useLangChainInterruptState = () =>\n  langChainExtras.use((e) => e.interrupt, undefined);\n\n/**\n * Read every interrupt pending at the current checkpoint, each with `id`\n * and `value`. Defaults to an empty array. Pair with `useLangChainRespondAll`\n * (keyed by interrupt id) to resolve several at once.\n */\nexport const useLangChainInterrupts = () =>\n  langChainExtras.use(\n    (e) => e.interrupts ?? EMPTY_INTERRUPTS,\n    EMPTY_INTERRUPTS,\n  );\n\n/** Read the last run/hydration error from the runtime extras. */\nexport const useLangChainError = () =>\n  langChainExtras.use((e) => e.error, undefined);\n\n/**\n * Read the root tool calls assembled by `useStream` from the `tools`\n * channel. Defaults to an empty array, so consumers can `.map` without\n * a guard. Useful for rendering pending/streamed tool calls and\n * approval UIs.\n */\nexport const useLangChainToolCalls = () =>\n  langChainExtras.use((e) => e.toolCalls ?? EMPTY_TOOL_CALLS, EMPTY_TOOL_CALLS);\n\n/** Subagents discovered on the current run (multi-agent graphs). */\nexport const useLangChainSubagents = () =>\n  langChainExtras.use((e) => e.subagents ?? EMPTY_SUBAGENTS, EMPTY_SUBAGENTS);\n\n/** Subgraphs discovered on the current run. */\nexport const useLangChainSubgraphs = () =>\n  langChainExtras.use((e) => e.subgraphs ?? EMPTY_SUBGRAPHS, EMPTY_SUBGRAPHS);\n\n/**\n * The underlying `useStream` handle, for advanced views — pass it to v1's\n * `useMessages(stream, target)` / `useToolCalls(stream, target)` with a\n * subagent/subgraph from `useLangChainSubagents` / `useLangChainSubgraphs`.\n * `undefined` outside the runtime provider.\n */\nexport const useLangChainStream = () =>\n  langChainExtras.use((e) => e.stream, undefined);\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  ) => langChainExtras.get(aui).submit(values, options);\n};\n\n/**\n * Resume a LangGraph interrupt with a response payload via\n * `useStream().respond`. Preferred over `useLangChainSendCommand`; it\n * carries the response cleanly and handles interrupt namespaces.\n */\nexport const useLangChainRespond = () => {\n  const aui = useAui();\n  return (response: unknown, options?: Record<string, unknown>) =>\n    langChainExtras.get(aui).respond(response, options);\n};\n\n/**\n * Resume several LangGraph interrupts pending at the same checkpoint in\n * one run via `useStream().respondAll`. Use when a run pauses on multiple\n * interrupts at once; sequential `useLangChainRespond` calls can't service\n * them (the first resume starts a run, stranding the rest).\n */\nexport const useLangChainRespondAll = () => {\n  const aui = useAui();\n  return (\n    responsesById: Record<string, unknown>,\n    options?: Record<string, unknown>,\n  ) => langChainExtras.get(aui).respondAll(responsesById, options);\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 extras = langChainExtras.get(aui);\n    return extras.submit({ [extras.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 langChainExtras.use((e) => {\n    const value = e.values[key] as T | undefined;\n    return value !== undefined ? value : defaultValue;\n  }, defaultValue);\n}\n"],"mappings":";;;;AAWA,MAAM,mBAAiD,CAAC;AAExD,MAAM,mBAAgE,CAAC;AAEvE,MAAM,kCACJ,IAAI,IAAI;AAEV,MAAM,kCACJ,IAAI,IAAI;;;;AAKV,MAAa,mCACX,gBAAgB,KAAK,MAAM,EAAE,WAAW,KAAA,CAAS;;;;;;AAOnD,MAAa,+BACX,gBAAgB,KACb,MAAM,EAAE,cAAc,kBACvB,gBACF;;AAGF,MAAa,0BACX,gBAAgB,KAAK,MAAM,EAAE,OAAO,KAAA,CAAS;;;;;;;AAQ/C,MAAa,8BACX,gBAAgB,KAAK,MAAM,EAAE,aAAa,kBAAkB,gBAAgB;;AAG9E,MAAa,8BACX,gBAAgB,KAAK,MAAM,EAAE,aAAa,iBAAiB,eAAe;;AAG5E,MAAa,8BACX,gBAAgB,KAAK,MAAM,EAAE,aAAa,iBAAiB,eAAe;;;;;;;AAQ5E,MAAa,2BACX,gBAAgB,KAAK,MAAM,EAAE,QAAQ,KAAA,CAAS;;;;;;AAOhD,MAAa,2BAA2B;CACtC,MAAM,MAAM,OAAO;CACnB,QACE,QACA,YACG,gBAAgB,IAAI,GAAG,CAAC,CAAC,OAAO,QAAQ,OAAO;AACtD;;;;;;AAOA,MAAa,4BAA4B;CACvC,MAAM,MAAM,OAAO;CACnB,QAAQ,UAAmB,YACzB,gBAAgB,IAAI,GAAG,CAAC,CAAC,QAAQ,UAAU,OAAO;AACtD;;;;;;;AAQA,MAAa,+BAA+B;CAC1C,MAAM,MAAM,OAAO;CACnB,QACE,eACA,YACG,gBAAgB,IAAI,GAAG,CAAC,CAAC,WAAW,eAAe,OAAO;AACjE;;;;;;AAOA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CACnB,QACE,UACA,YACG;EACH,MAAM,SAAS,gBAAgB,IAAI,GAAG;EACtC,OAAO,OAAO,OAAO,GAAG,OAAO,cAAc,SAAS,GAAG,OAAO;CAClE;AACF;;;;;;;;AASA,MAAa,gCAAgC;CAC3C,MAAM,SAAS,mBAAmB;CAClC,QAAQ,YAAqC,OAAO,MAAM,EAAE,QAAQ,CAAC;AACvE;AAeA,SAAgB,kBACd,KACA,cACe;CACf,OAAO,gBAAgB,KAAK,MAAM;EAChC,MAAM,QAAQ,EAAE,OAAO;EACvB,OAAO,UAAU,KAAA,IAAY,QAAQ;CACvC,GAAG,YAAY;AACjB"}

package/dist/index.d.ts

@@ -1,4 +1,7 @@
-import { LangChainBaseMessage, LangChainContentBlock, LangChainToolCall } from "./types.js";
+import { LangChainBaseMessage, LangChainContentBlock, LangChainToolCall, RemoveUIMessage, UIMessage, UseStreamRuntimeOptions } from "./types.js";
 import { convertLangChainBaseMessage } from "./convertMessages.js";
-import { UseStreamRuntimeOptions, useLangChainError, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useLangChainToolCalls, useStreamRuntime } from "./useStreamRuntime.js";
-export { type LangChainBaseMessage, type LangChainContentBlock, type LangChainToolCall, type UseStreamRuntimeOptions, convertLangChainBaseMessage, useLangChainError, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useLangChainToolCalls, useStreamRuntime };
+import { useLangChainError, useLangChainInterruptState, useLangChainInterrupts, useLangChainRespond, useLangChainRespondAll, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainStream, useLangChainSubagents, useLangChainSubgraphs, useLangChainSubmit, useLangChainToolCalls } from "./hooks.js";
+import { useStreamRuntime } from "./useStreamRuntime.js";
+import { useLangChainStreamingTiming } from "./streamingTiming.js";
+import { SubagentDiscoverySnapshot, SubgraphDiscoverySnapshot } from "@langchain/react";
+export { type LangChainBaseMessage, type LangChainContentBlock, type LangChainToolCall, type RemoveUIMessage, type SubagentDiscoverySnapshot, type SubgraphDiscoverySnapshot, type UIMessage, type UseStreamRuntimeOptions, convertLangChainBaseMessage, useLangChainError, useLangChainInterruptState, useLangChainInterrupts, useLangChainRespond, useLangChainRespondAll, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainStream, useLangChainStreamingTiming, useLangChainSubagents, useLangChainSubgraphs, useLangChainSubmit, useLangChainToolCalls, useStreamRuntime };

package/dist/index.js

@@ -1,3 +1,5 @@
 import { convertLangChainBaseMessage } from "./convertMessages.js";
-import { useLangChainError, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useLangChainToolCalls, useStreamRuntime } from "./useStreamRuntime.js";
-export { convertLangChainBaseMessage, useLangChainError, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useLangChainToolCalls, useStreamRuntime };
+import { useLangChainError, useLangChainInterruptState, useLangChainInterrupts, useLangChainRespond, useLangChainRespondAll, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainStream, useLangChainSubagents, useLangChainSubgraphs, useLangChainSubmit, useLangChainToolCalls } from "./hooks.js";
+import { useLangChainStreamingTiming } from "./streamingTiming.js";
+import { useStreamRuntime } from "./useStreamRuntime.js";
+export { convertLangChainBaseMessage, useLangChainError, useLangChainInterruptState, useLangChainInterrupts, useLangChainRespond, useLangChainRespondAll, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainStream, useLangChainStreamingTiming, useLangChainSubagents, useLangChainSubgraphs, useLangChainSubmit, useLangChainToolCalls, useStreamRuntime };

package/dist/resolveForkCheckpoint.d.ts

@@ -0,0 +1,17 @@
+import { LangChainBaseMessage } from "./types.js";
+import { ForkCheckpointClient } from "./findForkCheckpointInHistory.js";
+
+//#region src/resolveForkCheckpoint.d.ts
+type MessageMetadataSnapshot = ReadonlyMap<string, {
+  readonly parentCheckpointId: string | undefined;
+}> | undefined;
+/**
+ * Hydration seeds every message with the head's parent checkpoint, so only the
+ * head's recorded fork checkpoint is reliable; older turns are matched against
+ * server history by message id. A `null` `parentId` edits the first human
+ * message and forks from the thread's initial (empty-message) checkpoint.
+ */
+declare const resolveForkCheckpoint: (client: ForkCheckpointClient, threadId: string, messages: readonly LangChainBaseMessage[], parentId: string | null, sourceId: string | null | undefined, metadata: MessageMetadataSnapshot, messagesKey: string) => Promise<string | null>;
+//#endregion
+export { resolveForkCheckpoint };
+//# sourceMappingURL=resolveForkCheckpoint.d.ts.map

package/dist/resolveForkCheckpoint.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"resolveForkCheckpoint.d.ts","names":[],"sources":["../src/resolveForkCheckpoint.ts"],"mappings":";;;;KAMK,uBAAA,GACD,WAAW;EAAA,SAAoB,kBAAkB;AAAA;;;;AAAA;AASrD;;cAAa,qBAAA,GACX,MAAA,EAAQ,oBAAA,EACR,QAAA,UACA,QAAA,WAAmB,oBAAA,IACnB,QAAA,iBACA,QAAA,6BACA,QAAA,EAAU,uBAAA,EACV,WAAA,aACC,OAAA"}

package/dist/resolveForkCheckpoint.js

@@ -0,0 +1,26 @@
+import { findForkCheckpointInHistory } from "./findForkCheckpointInHistory.js";
+//#region src/resolveForkCheckpoint.ts
+/**
+* Hydration seeds every message with the head's parent checkpoint, so only the
+* head's recorded fork checkpoint is reliable; older turns are matched against
+* server history by message id. A `null` `parentId` edits the first human
+* message and forks from the thread's initial (empty-message) checkpoint.
+*/
+const resolveForkCheckpoint = async (client, threadId, messages, parentId, sourceId, metadata, messagesKey) => {
+	const lastMessage = messages[messages.length - 1];
+	let checkpointId = sourceId != null && lastMessage?.id === sourceId ? metadata?.get(sourceId)?.parentCheckpointId ?? null : null;
+	if (!checkpointId) {
+		const parentIndex = parentId == null ? -1 : messages.findIndex((m) => m.id === parentId);
+		if (parentId != null && parentIndex === -1) return null;
+		try {
+			checkpointId = await findForkCheckpointInHistory(client, threadId, messages.slice(0, parentIndex + 1), messagesKey);
+		} catch {
+			return null;
+		}
+	}
+	return checkpointId;
+};
+//#endregion
+export { resolveForkCheckpoint };
+
+//# sourceMappingURL=resolveForkCheckpoint.js.map

package/dist/resolveForkCheckpoint.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resolveForkCheckpoint.js","names":[],"sources":["../src/resolveForkCheckpoint.ts"],"sourcesContent":["import type { LangChainBaseMessage } from \"./types\";\nimport {\n  findForkCheckpointInHistory,\n  type ForkCheckpointClient,\n} from \"./findForkCheckpointInHistory\";\n\ntype MessageMetadataSnapshot =\n  | ReadonlyMap<string, { readonly parentCheckpointId: string | undefined }>\n  | undefined;\n\n/**\n * Hydration seeds every message with the head's parent checkpoint, so only the\n * head's recorded fork checkpoint is reliable; older turns are matched against\n * server history by message id. A `null` `parentId` edits the first human\n * message and forks from the thread's initial (empty-message) checkpoint.\n */\nexport const resolveForkCheckpoint = async (\n  client: ForkCheckpointClient,\n  threadId: string,\n  messages: readonly LangChainBaseMessage[],\n  parentId: string | null,\n  sourceId: string | null | undefined,\n  metadata: MessageMetadataSnapshot,\n  messagesKey: string,\n): Promise<string | null> => {\n  const lastMessage = messages[messages.length - 1];\n  let checkpointId: string | null =\n    sourceId != null && lastMessage?.id === sourceId\n      ? (metadata?.get(sourceId)?.parentCheckpointId ?? null)\n      : null;\n\n  if (!checkpointId) {\n    const parentIndex =\n      parentId == null ? -1 : messages.findIndex((m) => m.id === parentId);\n    if (parentId != null && parentIndex === -1) return null;\n    try {\n      checkpointId = await findForkCheckpointInHistory(\n        client,\n        threadId,\n        messages.slice(0, parentIndex + 1),\n        messagesKey,\n      );\n    } catch {\n      return null;\n    }\n  }\n\n  return checkpointId;\n};\n"],"mappings":";;;;;;;;AAgBA,MAAa,wBAAwB,OACnC,QACA,UACA,UACA,UACA,UACA,UACA,gBAC2B;CAC3B,MAAM,cAAc,SAAS,SAAS,SAAS;CAC/C,IAAI,eACF,YAAY,QAAQ,aAAa,OAAO,WACnC,UAAU,IAAI,QAAQ,CAAC,EAAE,sBAAsB,OAChD;CAEN,IAAI,CAAC,cAAc;EACjB,MAAM,cACJ,YAAY,OAAO,KAAK,SAAS,WAAW,MAAM,EAAE,OAAO,QAAQ;EACrE,IAAI,YAAY,QAAQ,gBAAgB,IAAI,OAAO;EACnD,IAAI;GACF,eAAe,MAAM,4BACnB,QACA,UACA,SAAS,MAAM,GAAG,cAAc,CAAC,GACjC,WACF;EACF,QAAQ;GACN,OAAO;EACT;CACF;CAEA,OAAO;AACT"}

package/dist/runtimeExtras.d.ts

@@ -0,0 +1,7 @@
+import { LangChainRuntimeExtras } from "./types.js";
+
+//#region src/runtimeExtras.d.ts
+declare const langChainExtras: import("@assistant-ui/core/internal").RuntimeExtras<LangChainRuntimeExtras>;
+//#endregion
+export { langChainExtras };
+//# sourceMappingURL=runtimeExtras.d.ts.map

package/dist/runtimeExtras.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtimeExtras.d.ts","names":[],"sources":["../src/runtimeExtras.ts"],"mappings":";;;cAGa,eAAA,wCAAe,aAAA,CAAA,sBAAA"}

package/dist/runtimeExtras.js

@@ -0,0 +1,7 @@
+import { createRuntimeExtras } from "@assistant-ui/core/internal";
+//#region src/runtimeExtras.ts
+const langChainExtras = createRuntimeExtras("useStreamRuntime");
+//#endregion
+export { langChainExtras };
+
+//# sourceMappingURL=runtimeExtras.js.map

package/dist/runtimeExtras.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtimeExtras.js","names":[],"sources":["../src/runtimeExtras.ts"],"sourcesContent":["import { createRuntimeExtras } from \"@assistant-ui/core/internal\";\nimport type { LangChainRuntimeExtras } from \"./types\";\n\nexport const langChainExtras =\n  createRuntimeExtras<LangChainRuntimeExtras>(\"useStreamRuntime\");\n"],"mappings":";;AAGA,MAAa,kBACX,oBAA4C,kBAAkB"}

package/dist/streamingTiming.d.ts

@@ -0,0 +1,16 @@
+import { LangChainBaseMessage } from "./types.js";
+import { MessageTiming } from "@assistant-ui/core";
+import { StreamingTimingAccessors } from "@assistant-ui/core/react";
+
+//#region src/streamingTiming.d.ts
+declare const langChainStreamingTimingAccessors: StreamingTimingAccessors<LangChainBaseMessage>;
+/**
+ * Tracks per-message streaming timing for LangChain messages. Delegates to
+ * the shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,
+ * adapted to the `LangChainBaseMessage` shape (`_getType() -> "ai"`, content
+ * blocks including text/thinking/reasoning, `tool_calls`).
+ */
+declare const useLangChainStreamingTiming: (messages: readonly LangChainBaseMessage[], isRunning: boolean) => Record<string, MessageTiming>;
+//#endregion
+export { langChainStreamingTimingAccessors, useLangChainStreamingTiming };
+//# sourceMappingURL=streamingTiming.d.ts.map

package/dist/streamingTiming.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamingTiming.d.ts","names":[],"sources":["../src/streamingTiming.ts"],"mappings":";;;;;cAmEa,iCAAA,EAAmC,wBAAwB,CAAC,oBAAA;;AAAzE;;;;AAA6F;cAahF,2BAAA,GACX,QAAA,WAAmB,oBAAA,IACnB,SAAA,cACC,MAAA,SAAe,aAAA"}