@assistant-ui/react-langchain 0.0.28 → 0.0.30

package/README.md

@@ -17,7 +17,7 @@ npm install @assistant-ui/react @assistant-ui/react-langchain @langchain/react
 ```tsx
 import { useStreamRuntime } from "@assistant-ui/react-langchain";
 import { AssistantRuntimeProvider } from "@assistant-ui/react";
-import { Thread } from "@/components/assistant-ui/thread";
+import { Thread } from "@/components/assistant-ui/elements/thread.aui";
 
 export function App() {
   const runtime = useStreamRuntime({

package/dist/convertMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessages.d.ts","names":[],"sources":["../src/convertMessages.ts"],"mappings":";;;;;KAiBK,oCACH,4BAA4B;EAC1B,qBAAqB,YAAY;EACjC,gBAAgB,eAAe;;cAGtB,iBAAkB,SAAS;cA0B3B,8BACX,SAAS,sBACT,WAAU,sCACT,4BAA4B"}
+{"version":3,"file":"convertMessages.d.ts","names":[],"sources":["../src/convertMessages.ts"],"mappings":";;;;;KAiBK,oCACH,4BAA4B;EAC1B,qBAAqB,YAAY;EACjC,gBAAgB,eAAe;;cAetB,iBAAkB,SAAS;cAuC3B,8BACX,SAAS,sBACT,WAAU,sCACT,4BAA4B"}

package/dist/convertMessages.js

@@ -1,21 +1,35 @@
 "use client";
 import { convertLangChainContentBlock, getCustomMetadata, getMessageContent, uiMessageToDataPart, withAudioTranscript } from "./converter.js";
 //#region src/convertMessages.ts
+const warnedMalformedMessages = /* @__PURE__ */ new Set();
+const warnOnceInDevelopment = (message) => {
+	if (typeof process === "undefined" || process?.env?.NODE_ENV !== "development") return;
+	if (warnedMalformedMessages.has(message)) return;
+	warnedMalformedMessages.add(message);
+	console.warn(message);
+};
 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");
+	warnOnceInDevelopment("Cannot determine message type; rendering the message as system text");
+	return "unknown";
+};
+const contentBlocks = (content) => {
+	if (content == null) return [];
+	if (Array.isArray(content)) return content.filter((block) => typeof block === "object" && block !== null);
+	warnOnceInDevelopment(`Ignoring message content that is neither a string nor an array: ${typeof content}`);
+	return [];
 };
 const contentToParts = (content) => {
 	if (typeof content === "string") return [{
 		type: "text",
 		text: content
 	}];
-	return content.map(convertLangChainContentBlock).filter((part) => part !== null && part !== void 0);
+	return contentBlocks(content).map(convertLangChainContentBlock).filter((part) => part !== null && part !== void 0);
 };
 const getStringContent = (content) => {
 	if (typeof content === "string") return content;
-	return content.filter((c) => c.type === "text").map((c) => c.text).join("");
+	return contentBlocks(content).filter((c) => c.type === "text").map((c) => c.text).join("");
 };
 const convertLangChainBaseMessage = (message, metadata = {}) => {
 	switch (getMessageType(message)) {
@@ -73,7 +87,7 @@ const convertLangChainBaseMessage = (message, metadata = {}) => {
 			id: message.id,
 			content: [{
 				type: "text",
-				text: typeof message.content === "string" ? message.content : JSON.stringify(message.content)
+				text: typeof message.content === "string" ? message.content : JSON.stringify(message.content) ?? ""
 			}]
 		};
 	}

package/dist/convertMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessages.js","names":["parts"],"sources":["../src/convertMessages.ts"],"sourcesContent":["\"use client\";\n\nimport type { MessageTiming } from \"@assistant-ui/core\";\nimport type { useExternalMessageConverter } from \"@assistant-ui/core/react\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport {\n  convertLangChainContentBlock,\n  getCustomMetadata,\n  uiMessageToDataPart,\n  withAudioTranscript,\n} from \"./converter\";\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\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(convertLangChainContentBlock)\n    .filter((part) => part !== null && part !== undefined);\n};\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          ...withAudioTranscript(\n            contentToParts(message.content),\n            message.additional_kwargs,\n          ),\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        // `joinExternalMessages` only checks the name against the tool call\n        // when it is non-null, so an empty name manufactures a mismatch.\n        toolName: message.name || undefined,\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 { getMessageContent } from \"./converter\";\n"],"mappings":";;;AAuBA,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,IAAI,4BAA4B,CAAC,CACjC,QAAQ,SAAS,SAAS,QAAQ,SAAS,KAAA,CAAS;AACzD;AAEA,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,oBACD,eAAe,QAAQ,OAAO,GAC9B,QAAQ,iBACV;KACA,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;GAGN,UAAU,QAAQ,QAAQ,KAAA;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":[],"sources":["../src/convertMessages.ts"],"sourcesContent":["\"use client\";\n\nimport type { MessageTiming } from \"@assistant-ui/core\";\nimport type { useExternalMessageConverter } from \"@assistant-ui/core/react\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport {\n  convertLangChainContentBlock,\n  getCustomMetadata,\n  uiMessageToDataPart,\n  withAudioTranscript,\n} from \"./converter\";\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 warnedMalformedMessages = new Set<string>();\nconst warnOnceInDevelopment = (message: string) => {\n  if (\n    typeof process === \"undefined\" ||\n    process?.env?.NODE_ENV !== \"development\"\n  )\n    return;\n  if (warnedMalformedMessages.has(message)) return;\n  warnedMalformedMessages.add(message);\n  console.warn(message);\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  warnOnceInDevelopment(\n    \"Cannot determine message type; rendering the message as system text\",\n  );\n  return \"unknown\";\n};\n\nconst contentBlocks = (content: unknown): readonly LangChainContentBlock[] => {\n  if (content == null) return [];\n  if (Array.isArray(content))\n    return content.filter(\n      (block) => typeof block === \"object\" && block !== null,\n    );\n  warnOnceInDevelopment(\n    `Ignoring message content that is neither a string nor an array: ${typeof content}`,\n  );\n  return [];\n};\n\nconst contentToParts = (content: unknown) => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n\n  return contentBlocks(content)\n    .map(convertLangChainContentBlock)\n    .filter((part) => part !== null && part !== undefined);\n};\n\nconst getStringContent = (content: unknown): string => {\n  if (typeof content === \"string\") return content;\n  return contentBlocks(content)\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          ...withAudioTranscript(\n            contentToParts(message.content),\n            message.additional_kwargs,\n          ),\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        // `joinExternalMessages` only checks the name against the tool call\n        // when it is non-null, so an empty name manufactures a mismatch.\n        toolName: message.name || undefined,\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 { getMessageContent } from \"./converter\";\n"],"mappings":";;;AAuBA,MAAM,0CAA0B,IAAI,IAAY;AAChD,MAAM,yBAAyB,YAAoB;CACjD,IACE,OAAO,YAAY,eACnB,SAAS,KAAK,aAAa,eAE3B;CACF,IAAI,wBAAwB,IAAI,OAAO,GAAG;CAC1C,wBAAwB,IAAI,OAAO;CACnC,QAAQ,KAAK,OAAO;AACtB;AAEA,MAAa,kBAAkB,YAA0C;CACvE,IAAI,OAAO,QAAQ,aAAa,YAAY,OAAO,QAAQ,SAAS;CACpE,IAAI,UAAU,SACZ,OAAQ,QAAoC;CAC9C,sBACE,qEACF;CACA,OAAO;AACT;AAEA,MAAM,iBAAiB,YAAuD;CAC5E,IAAI,WAAW,MAAM,OAAO,CAAC;CAC7B,IAAI,MAAM,QAAQ,OAAO,GACvB,OAAO,QAAQ,QACZ,UAAU,OAAO,UAAU,YAAY,UAAU,IACpD;CACF,sBACE,mEAAmE,OAAO,SAC5E;CACA,OAAO,CAAC;AACV;AAEA,MAAM,kBAAkB,YAAqB;CAC3C,IAAI,OAAO,YAAY,UACrB,OAAO,CAAC;EAAE,MAAM;EAAiB,MAAM;CAAQ,CAAC;CAElD,OAAO,cAAc,OAAO,CAAC,CAC1B,IAAI,4BAA4B,CAAC,CACjC,QAAQ,SAAS,SAAS,QAAQ,SAAS,KAAA,CAAS;AACzD;AAEA,MAAM,oBAAoB,YAA6B;CACrD,IAAI,OAAO,YAAY,UAAU,OAAO;CACxC,OAAO,cAAc,OAAO,CAAC,CAC1B,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,oBACD,eAAe,QAAQ,OAAO,GAC9B,QAAQ,iBACV;KACA,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;GAGN,UAAU,QAAQ,QAAQ,KAAA;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,UACP,KAAK,UAAU,QAAQ,OAAO,KAAK;GAC5C,CACF;EACF;CACJ;AACF"}

package/dist/converter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"converter.d.ts","names":[],"sources":["../src/converter.ts"],"mappings":";;;;KAaY;EACN;EAAc;;EACd;EAAoB;;EACpB;EAAmB;IAAsB;;;EACzC;EAAkB;;EAElB;EACA,UAAU;IAAQ;IAAsB;;EACxC;;EAGA;EACA;EACA;EACA;EACA;IAAa;;;EAGb;EACA;EACA;EACA;EACA;IAAa;;;EAGb;EACA;EACA;EACA;EACA;IAAa;;;EAGb;EACA;EACA;EACA;;EAEA;;KAED,uBACD,uCACA;cAES,+BACX,MAAM,0BACL;;;;;;;cAoEU,sBAAuB;EAAY;EAAc;GAC5D,gBAAgB,KAChB,kBAAkB,kDACP;EAAM;EAAc;;cAapB,oBACX,kBAAkB,wCACjB;cAGU,sBACX;EAAqB;EAAc,OAAO;GAE1C,IAAI,eACH;cAoBU,oBAAqB,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAsG1B,0CACX;EACE;EACA;EACA;GAGF,UAAU,SAAS,wBAClB,yBAAyB;;;;;cA8Df,uBAAwB;EACnC;IAAa;IAAqB;;;;;;;cAOvB,0BACX;EACE;IAAa;IAAqB;;GAGpC,mBACC,YAAY"}
+{"version":3,"file":"converter.d.ts","names":[],"sources":["../src/converter.ts"],"mappings":";;;;KAaY;EACN;EAAc;;EACd;EAAoB;;EACpB;EAAmB;IAAsB;;;EACzC;EAAkB;;EAElB;EACA,UAAU;IAAQ;IAAsB;;EACxC;;EAGA;EACA;EACA;EACA;EACA;IAAa;;;EAGb;EACA;EACA;EACA;EACA;IAAa;;;EAGb;EACA;EACA;EACA;EACA;IAAa;;;EAGb;EACA;EACA;EACA;;EAEA;;KAED,uBACD,uCACA;cAES,+BACX,MAAM,0BACL;;;;;;;cAqEU,sBAAuB;EAAY;EAAc;GAC5D,gBAAgB,KAChB,kBAAkB,kDACP;EAAM;EAAc;;cAapB,oBACX,kBAAkB,wCACjB;cAGU,sBACX;EAAqB;EAAc,OAAO;GAE1C,IAAI,eACH;cAoBU,oBAAqB,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAuG1B,0CACX;EACE;EACA;EACA;GAGF,UAAU,SAAS,wBAClB,yBAAyB;;;;;cA+Df,uBAAwB;EACnC;IAAa;IAAqB;;;;;;;cAOvB,0BACX;EACE;IAAa;IAAqB;;GAGpC,mBACC,YAAY"}

package/dist/converter.js

@@ -5,7 +5,7 @@ const convertLangChainContentBlock = (part) => {
 		case "text":
 		case "text_delta": return {
 			type: "text",
-			text: part.text
+			text: typeof part.text === "string" ? part.text : ""
 		};
 		case "image_url": {
 			const image = typeof part.image_url === "string" ? part.image_url : part.image_url?.url;
@@ -32,14 +32,17 @@ const convertLangChainContentBlock = (part) => {
 				mimeType
 			};
 		}
-		case "thinking": return {
+		case "thinking": return hasVisibleText(part.thinking) ? {
 			type: "reasoning",
 			text: part.thinking
-		};
-		case "reasoning": return {
-			type: "reasoning",
-			text: part.summary && part.summary.length > 0 ? part.summary.map((s) => s?.text ?? "").join("\n\n\n") : part.reasoning ?? ""
-		};
+		} : null;
+		case "reasoning": {
+			const text = getReasoningText(part);
+			return text ? {
+				type: "reasoning",
+				text
+			} : null;
+		}
 		case "tool_use":
 		case "input_json_delta": return null;
 		default: return;
@@ -148,9 +151,11 @@ const getMessageContent = (msg) => {
 	if (content.length === 1 && content[0]?.type === "text") return content[0].text ?? "";
 	return content;
 };
-const reasoningTextLength = (part) => {
-	if (part.summary && part.summary.length > 0) return part.summary.map((s) => s?.text ?? "").join("\n\n\n").length;
-	return part.reasoning?.length ?? 0;
+const getReasoningText = (part) => {
+	const summary = part.summary?.map((s) => s?.text ?? "").join("\n\n\n") ?? "";
+	if (hasVisibleText(summary)) return summary;
+	const reasoning = part.reasoning ?? "";
+	return hasVisibleText(reasoning) ? reasoning : "";
 };
 const createLangChainStreamingTimingAccessors = (getType) => {
 	const findAiMessage = (messages, messageId) => messages.find((message) => getType(message) === "ai" && message.id === messageId);
@@ -161,15 +166,18 @@ const createLangChainStreamingTimingAccessors = (getType) => {
 		if (typeof content === "string") return content.length;
 		if (!Array.isArray(content)) return 0;
 		let len = 0;
-		for (const part of content) switch (part.type) {
-			case "text":
-			case "text_delta":
-				if (typeof part.text === "string") len += part.text.length;
-				break;
-			case "thinking":
-				if (typeof part.thinking === "string") len += part.thinking.length;
-				break;
-			case "reasoning": len += reasoningTextLength(part);
+		for (const part of content) {
+			if (typeof part !== "object" || part === null) continue;
+			switch (part.type) {
+				case "text":
+				case "text_delta":
+					if (typeof part.text === "string") len += part.text.length;
+					break;
+				case "thinking":
+					if (hasVisibleText(part.thinking)) len += part.thinking.length;
+					break;
+				case "reasoning": len += getReasoningText(part).length;
+			}
 		}
 		return len;
 	};

package/dist/converter.js.map

@@ -1 +1 @@
-{"version":3,"file":"converter.js","names":["_exhaustiveCheck"],"sources":["../src/converter.ts"],"sourcesContent":["import type {\n  AppendMessage,\n  DataMessagePart,\n  ThreadAssistantMessage,\n  ThreadUserMessage,\n} from \"@assistant-ui/core\";\nimport {\n  parseDataUrl,\n  resolveFilePartSource,\n} from \"@assistant-ui/core/internal\";\nimport type { StreamingTimingAccessors } from \"@assistant-ui/core/react\";\n\n/** Known content block types from @langchain/core messages. */\nexport type LangChainContentBlock =\n  | { type: \"text\"; text: string }\n  | { type: \"text_delta\"; text: string }\n  | { type: \"image_url\"; image_url: string | { url?: string } }\n  | { type: \"thinking\"; thinking: string }\n  | {\n      type: \"reasoning\";\n      summary?: Array<{ type: \"summary_text\"; text?: string }>;\n      reasoning?: string;\n    }\n  | {\n      type: \"file\";\n      data: string;\n      mime_type: string;\n      source_type?: \"base64\";\n      metadata?: { filename?: string };\n    }\n  | {\n      type: \"file\";\n      url: string;\n      mime_type?: string;\n      source_type: \"url\";\n      metadata?: { filename?: string };\n    }\n  | {\n      type: \"file\";\n      id: string;\n      mime_type?: string;\n      source_type: \"id\";\n      metadata?: { filename?: string };\n    }\n  | {\n      type: \"audio\";\n      data: string;\n      mime_type: string;\n      source_type: \"base64\";\n    }\n  | { type: \"tool_use\" | \"input_json_delta\" };\n\ntype ConvertedContentPart =\n  | ThreadUserMessage[\"content\"][number]\n  | ThreadAssistantMessage[\"content\"][number];\n\nexport const convertLangChainContentBlock = (\n  part: LangChainContentBlock,\n): ConvertedContentPart | null | undefined => {\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:\n          part.source_type === \"url\"\n            ? part.url\n            : part.source_type === \"id\"\n              ? part.id\n              : part.data,\n        mimeType: part.mime_type ?? \"application/octet-stream\",\n        ...((part.source_type === \"url\" || part.source_type === \"id\") && {\n          sourceType: part.source_type,\n        }),\n      };\n    case \"audio\": {\n      const mimeType = part.mime_type ?? \"application/octet-stream\";\n      const subtype = mimeType.startsWith(\"audio/\")\n        ? mimeType.slice(\"audio/\".length)\n        : undefined;\n      return {\n        type: \"file\" as const,\n        filename: subtype ? `audio.${subtype}` : \"audio\",\n        data: part.data,\n        mimeType,\n      };\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 && part.summary.length > 0\n            ? part.summary.map((s) => s?.text ?? \"\").join(\"\\n\\n\\n\")\n            : (part.reasoning ?? \"\"),\n      };\n    case \"tool_use\":\n    case \"input_json_delta\":\n      return null;\n    default:\n      return undefined;\n  }\n};\n\nconst hasVisibleText = (text: unknown): boolean =>\n  typeof text === \"string\" && text.trim() !== \"\";\n\n/**\n * Audio output arrives outside the content array: providers leave `content`\n * empty and put the spoken text in `additional_kwargs.audio.transcript`. The\n * audio bytes stay behind because no provider reports their media type, and a\n * streamed response carries raw PCM rather than a playable file.\n */\nexport const withAudioTranscript = <T extends { type: string; text?: unknown }>(\n  parts: readonly T[],\n  additionalKwargs: Record<string, unknown> | undefined,\n): readonly (T | { type: \"text\"; text: string })[] => {\n  const audio = additionalKwargs?.audio as { transcript?: unknown } | undefined;\n  const transcript = audio?.transcript;\n  if (typeof transcript !== \"string\" || !hasVisibleText(transcript))\n    return parts;\n  if (parts.some((part) => part.type === \"text\" && hasVisibleText(part.text)))\n    return parts;\n  return [\n    ...parts.filter((part) => part.type !== \"text\"),\n    { type: \"text\" as const, text: transcript },\n  ];\n};\n\nexport const getCustomMetadata = (\n  additionalKwargs: Record<string, unknown> | undefined,\n): Record<string, unknown> =>\n  (additionalKwargs?.metadata as Record<string, unknown>) ?? {};\n\nexport const uiMessageToDataPart = <\n  TUIMessage extends { name: string; props: Record<string, unknown> },\n>(\n  ui: TUIMessage,\n): DataMessagePart => ({\n  type: \"data\",\n  name: ui.name,\n  data: ui.props,\n});\n\n/**\n * Audio media types that reach a provider's audio input through the LangChain\n * `audio` block. langchain-core derives OpenAI's `input_audio.format` by\n * splitting `mime_type` on `/`, and that format is a wav-or-mp3 enum, so\n * `audio/mpeg` passes the converter and is rejected at the provider.\n */\nconst audioBlockMimeTypes = new Map<string, \"audio/mp3\" | \"audio/wav\">([\n  [\"audio/mp3\", \"audio/mp3\"],\n  [\"audio/mpeg\", \"audio/mp3\"],\n  [\"audio/wav\", \"audio/wav\"],\n  [\"audio/wave\", \"audio/wav\"],\n  [\"audio/x-wav\", \"audio/wav\"],\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) =>\n      part.type === \"file\" || part.type === \"image\" || part.type === \"audio\",\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.flatMap((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        const metadata = { filename: part.filename ?? \"file\" };\n        if (part.sourceType === \"id\") {\n          return {\n            type: \"file\" as const,\n            id: part.data,\n            mime_type: part.mimeType,\n            filename: metadata.filename,\n            metadata,\n            source_type: \"id\" as const,\n          };\n        }\n        const source = resolveFilePartSource(part);\n        if (source.kind === \"url\") {\n          return {\n            type: \"file\" as const,\n            url: source.url,\n            mime_type: part.mimeType,\n            filename: metadata.filename,\n            metadata,\n            source_type: \"url\" as const,\n          };\n        }\n        const audioMimeType = audioBlockMimeTypes.get(\n          source.mimeType.toLowerCase(),\n        );\n        if (audioMimeType) {\n          return {\n            type: \"audio\" as const,\n            data: source.data,\n            mime_type: audioMimeType,\n            source_type: \"base64\" as const,\n          };\n        }\n        return {\n          type: \"file\" as const,\n          data: source.data,\n          mime_type: source.mimeType,\n          filename: metadata.filename,\n          metadata,\n          source_type: \"base64\" as const,\n        };\n      }\n      case \"audio\": {\n        const parsed = parseDataUrl(part.audio.data);\n        return {\n          type: \"audio\" as const,\n          data: parsed?.data ?? part.audio.data,\n          mime_type: `audio/${part.audio.format}`,\n          source_type: \"base64\" as const,\n        };\n      }\n      case \"data\":\n        return [];\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported.\");\n      default: {\n        const _exhaustiveCheck: \"reasoning\" | \"source\" | \"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\nconst reasoningTextLength = (part: {\n  readonly summary?: ReadonlyArray<{ readonly text?: string }>;\n  readonly reasoning?: string;\n}): number => {\n  if (part.summary && part.summary.length > 0)\n    return part.summary.map((s) => s?.text ?? \"\").join(\"\\n\\n\\n\").length;\n  return part.reasoning?.length ?? 0;\n};\n\nexport const createLangChainStreamingTimingAccessors = <\n  TMessage extends {\n    id?: string | undefined;\n    content?: unknown;\n    tool_calls?: readonly unknown[] | undefined;\n  },\n>(\n  getType: (message: TMessage) => string,\n): StreamingTimingAccessors<TMessage> => {\n  const findAiMessage = (\n    messages: readonly TMessage[],\n    messageId: string,\n  ): TMessage | undefined =>\n    messages.find(\n      (message) => getType(message) === \"ai\" && message.id === messageId,\n    );\n\n  const getTextLength = (\n    messages: readonly TMessage[],\n    messageId: string,\n  ): number => {\n    const message = findAiMessage(messages, messageId);\n    if (!message) return 0;\n    const content = message.content;\n    if (typeof content === \"string\") return content.length;\n    if (!Array.isArray(content)) return 0;\n    let len = 0;\n    for (const part of content as readonly LangChainContentBlock[]) {\n      switch (part.type) {\n        case \"text\":\n        case \"text_delta\":\n          if (typeof part.text === \"string\") len += part.text.length;\n          break;\n        case \"thinking\":\n          if (typeof part.thinking === \"string\") len += part.thinking.length;\n          break;\n        case \"reasoning\":\n          len += reasoningTextLength(part);\n          break;\n      }\n    }\n    return len;\n  };\n\n  const getToolCallCount = (\n    messages: readonly TMessage[],\n    messageId: string,\n  ): number => findAiMessage(messages, messageId)?.tool_calls?.length ?? 0;\n\n  const getAssistantMessageId = (\n    messages: readonly TMessage[],\n  ): string | undefined => {\n    for (let i = messages.length - 1; i >= 0; i--) {\n      const message = messages[i];\n      if (message && getType(message) === \"ai\" && message.id) return message.id;\n    }\n    return undefined;\n  };\n\n  return {\n    getAssistantMessageId,\n    getTextLength,\n    getToolCallCount,\n  };\n};\n\n/**\n * Resolve the assistant message a `UIMessage` belongs to: the parent id comes\n * from `metadata.message_id` (Python SDK) or `metadata.id` (JS SDK).\n */\nexport const getUIMessageParentId = (ui: {\n  metadata?: { message_id?: string; id?: string } | undefined;\n}): string | undefined => ui.metadata?.message_id ?? ui.metadata?.id;\n\n/**\n * Group the graph's accumulated `UIMessage`s by the assistant message they\n * belong to. Non-array state and entries without a parent link are dropped.\n */\nexport const groupUIMessagesByParent = <\n  T extends {\n    metadata?: { message_id?: string; id?: string } | undefined;\n  },\n>(\n  value: unknown,\n): Map<string, T[]> => {\n  const map = new Map<string, T[]>();\n  if (!Array.isArray(value)) return map;\n  for (const ui of value as T[]) {\n    const parentId = getUIMessageParentId(ui);\n    if (!parentId) continue;\n    const existing = map.get(parentId);\n    if (existing) {\n      existing.push(ui);\n    } else {\n      map.set(parentId, [ui]);\n    }\n  }\n  return map;\n};\n"],"mappings":";;AAwDA,MAAa,gCACX,SAC4C;CAE5C,QADa,KAAK,MAClB;EACE,KAAK;EACL,KAAK,cACH,OAAO;GAAE,MAAM;GAAiB,MAAM,KAAK;EAAK;EAClD,KAAK,aAAa;GAChB,MAAM,QACJ,OAAO,KAAK,cAAc,WACtB,KAAK,YACL,KAAK,WAAW;GACtB,IAAI,CAAC,OAAO,OAAO;GACnB,OAAO;IAAE,MAAM;IAAkB;GAAM;EACzC;EACA,KAAK,QACH,OAAO;GACL,MAAM;GACN,UAAU,KAAK,UAAU,YAAY;GACrC,MACE,KAAK,gBAAgB,QACjB,KAAK,MACL,KAAK,gBAAgB,OACnB,KAAK,KACL,KAAK;GACb,UAAU,KAAK,aAAa;GAC5B,IAAK,KAAK,gBAAgB,SAAS,KAAK,gBAAgB,SAAS,EAC/D,YAAY,KAAK,YACnB;EACF;EACF,KAAK,SAAS;GACZ,MAAM,WAAW,KAAK,aAAa;GACnC,MAAM,UAAU,SAAS,WAAW,QAAQ,IACxC,SAAS,MAAM,CAAe,IAC9B,KAAA;GACJ,OAAO;IACL,MAAM;IACN,UAAU,UAAU,SAAS,YAAY;IACzC,MAAM,KAAK;IACX;GACF;EACF;EACA,KAAK,YACH,OAAO;GAAE,MAAM;GAAsB,MAAM,KAAK;EAAS;EAC3D,KAAK,aACH,OAAO;GACL,MAAM;GACN,MACE,KAAK,WAAW,KAAK,QAAQ,SAAS,IAClC,KAAK,QAAQ,KAAK,MAAM,GAAG,QAAQ,EAAE,CAAC,CAAC,KAAK,QAAQ,IACnD,KAAK,aAAa;EAC3B;EACF,KAAK;EACL,KAAK,oBACH,OAAO;EACT,SACE;CACJ;AACF;AAEA,MAAM,kBAAkB,SACtB,OAAO,SAAS,YAAY,KAAK,KAAK,MAAM;;;;;;;AAQ9C,MAAa,uBACX,OACA,qBACoD;CAEpD,MAAM,cADQ,kBAAkB,MAAA,EACN;CAC1B,IAAI,OAAO,eAAe,YAAY,CAAC,eAAe,UAAU,GAC9D,OAAO;CACT,IAAI,MAAM,MAAM,SAAS,KAAK,SAAS,UAAU,eAAe,KAAK,IAAI,CAAC,GACxE,OAAO;CACT,OAAO,CACL,GAAG,MAAM,QAAQ,SAAS,KAAK,SAAS,MAAM,GAC9C;EAAE,MAAM;EAAiB,MAAM;CAAW,CAC5C;AACF;AAEA,MAAa,qBACX,qBAEC,kBAAkB,YAAwC,CAAC;AAE9D,MAAa,uBAGX,QACqB;CACrB,MAAM;CACN,MAAM,GAAG;CACT,MAAM,GAAG;AACX;;;;;;;AAQA,MAAM,sCAAsB,IAAI,IAAuC;CACrE,CAAC,aAAa,WAAW;CACzB,CAAC,cAAc,WAAW;CAC1B,CAAC,aAAa,WAAW;CACzB,CAAC,cAAc,WAAW;CAC1B,CAAC,eAAe,WAAW;AAC7B,CAAC;AAED,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,SACC,KAAK,SAAS,UAAU,KAAK,SAAS,WAAW,KAAK,SAAS,OACnE;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,SAAS,SAAS;EAC3C,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,QAAQ;IACX,MAAM,WAAW,EAAE,UAAU,KAAK,YAAY,OAAO;IACrD,IAAI,KAAK,eAAe,MACtB,OAAO;KACL,MAAM;KACN,IAAI,KAAK;KACT,WAAW,KAAK;KAChB,UAAU,SAAS;KACnB;KACA,aAAa;IACf;IAEF,MAAM,SAAS,sBAAsB,IAAI;IACzC,IAAI,OAAO,SAAS,OAClB,OAAO;KACL,MAAM;KACN,KAAK,OAAO;KACZ,WAAW,KAAK;KAChB,UAAU,SAAS;KACnB;KACA,aAAa;IACf;IAEF,MAAM,gBAAgB,oBAAoB,IACxC,OAAO,SAAS,YAAY,CAC9B;IACA,IAAI,eACF,OAAO;KACL,MAAM;KACN,MAAM,OAAO;KACb,WAAW;KACX,aAAa;IACf;IAEF,OAAO;KACL,MAAM;KACN,MAAM,OAAO;KACb,WAAW,OAAO;KAClB,UAAU,SAAS;KACnB;KACA,aAAa;IACf;GACF;GACA,KAAK,SAEH,OAAO;IACL,MAAM;IACN,MAHa,aAAa,KAAK,MAAM,IAG1B,CAAC,EAAE,QAAQ,KAAK,MAAM;IACjC,WAAW,SAAS,KAAK,MAAM;IAC/B,aAAa;GACf;GAEF,KAAK,QACH,OAAO,CAAC;GACV,KAAK,aACH,MAAM,IAAI,MAAM,sCAAsC;GACxD,SAEE,MAAM,IAAI,MACR,yCAAyCA,MAC3C;EAEJ;CACF,CAAC;CAED,IAAI,QAAQ,WAAW,KAAK,QAAQ,EAAE,EAAE,SAAS,QAC/C,OAAO,QAAQ,EAAE,CAAC,QAAQ;CAE5B,OAAO;AACT;AAEA,MAAM,uBAAuB,SAGf;CACZ,IAAI,KAAK,WAAW,KAAK,QAAQ,SAAS,GACxC,OAAO,KAAK,QAAQ,KAAK,MAAM,GAAG,QAAQ,EAAE,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC;CAC/D,OAAO,KAAK,WAAW,UAAU;AACnC;AAEA,MAAa,2CAOX,YACuC;CACvC,MAAM,iBACJ,UACA,cAEA,SAAS,MACN,YAAY,QAAQ,OAAO,MAAM,QAAQ,QAAQ,OAAO,SAC3D;CAEF,MAAM,iBACJ,UACA,cACW;EACX,MAAM,UAAU,cAAc,UAAU,SAAS;EACjD,IAAI,CAAC,SAAS,OAAO;EACrB,MAAM,UAAU,QAAQ;EACxB,IAAI,OAAO,YAAY,UAAU,OAAO,QAAQ;EAChD,IAAI,CAAC,MAAM,QAAQ,OAAO,GAAG,OAAO;EACpC,IAAI,MAAM;EACV,KAAK,MAAM,QAAQ,SACjB,QAAQ,KAAK,MAAb;GACE,KAAK;GACL,KAAK;IACH,IAAI,OAAO,KAAK,SAAS,UAAU,OAAO,KAAK,KAAK;IACpD;GACF,KAAK;IACH,IAAI,OAAO,KAAK,aAAa,UAAU,OAAO,KAAK,SAAS;IAC5D;GACF,KAAK,aACH,OAAO,oBAAoB,IAAI;EAEnC;EAEF,OAAO;CACT;CAEA,MAAM,oBACJ,UACA,cACW,cAAc,UAAU,SAAS,CAAC,EAAE,YAAY,UAAU;CAEvE,MAAM,yBACJ,aACuB;EACvB,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;GAC7C,MAAM,UAAU,SAAS;GACzB,IAAI,WAAW,QAAQ,OAAO,MAAM,QAAQ,QAAQ,IAAI,OAAO,QAAQ;EACzE;CAEF;CAEA,OAAO;EACL;EACA;EACA;CACF;AACF;;;;;AAMA,MAAa,wBAAwB,OAEX,GAAG,UAAU,cAAc,GAAG,UAAU;;;;;AAMlE,MAAa,2BAKX,UACqB;CACrB,MAAM,sBAAM,IAAI,IAAiB;CACjC,IAAI,CAAC,MAAM,QAAQ,KAAK,GAAG,OAAO;CAClC,KAAK,MAAM,MAAM,OAAc;EAC7B,MAAM,WAAW,qBAAqB,EAAE;EACxC,IAAI,CAAC,UAAU;EACf,MAAM,WAAW,IAAI,IAAI,QAAQ;EACjC,IAAI,UACF,SAAS,KAAK,EAAE;OAEhB,IAAI,IAAI,UAAU,CAAC,EAAE,CAAC;CAE1B;CACA,OAAO;AACT"}
+{"version":3,"file":"converter.js","names":["_exhaustiveCheck"],"sources":["../src/converter.ts"],"sourcesContent":["import type {\n  AppendMessage,\n  DataMessagePart,\n  ThreadAssistantMessage,\n  ThreadUserMessage,\n} from \"@assistant-ui/core\";\nimport {\n  parseDataUrl,\n  resolveFilePartSource,\n} from \"@assistant-ui/core/internal\";\nimport type { StreamingTimingAccessors } from \"@assistant-ui/core/react\";\n\n/** Known content block types from @langchain/core messages. */\nexport type LangChainContentBlock =\n  | { type: \"text\"; text: string }\n  | { type: \"text_delta\"; text: string }\n  | { type: \"image_url\"; image_url: string | { url?: string } }\n  | { type: \"thinking\"; thinking: string }\n  | {\n      type: \"reasoning\";\n      summary?: Array<{ type: \"summary_text\"; text?: string }>;\n      reasoning?: string;\n    }\n  | {\n      type: \"file\";\n      data: string;\n      mime_type: string;\n      source_type?: \"base64\";\n      metadata?: { filename?: string };\n    }\n  | {\n      type: \"file\";\n      url: string;\n      mime_type?: string;\n      source_type: \"url\";\n      metadata?: { filename?: string };\n    }\n  | {\n      type: \"file\";\n      id: string;\n      mime_type?: string;\n      source_type: \"id\";\n      metadata?: { filename?: string };\n    }\n  | {\n      type: \"audio\";\n      data: string;\n      mime_type: string;\n      source_type: \"base64\";\n    }\n  | { type: \"tool_use\" | \"input_json_delta\" };\n\ntype ConvertedContentPart =\n  | ThreadUserMessage[\"content\"][number]\n  | ThreadAssistantMessage[\"content\"][number];\n\nexport const convertLangChainContentBlock = (\n  part: LangChainContentBlock,\n): ConvertedContentPart | null | undefined => {\n  const type = part.type;\n  switch (type) {\n    case \"text\":\n    case \"text_delta\":\n      return {\n        type: \"text\" as const,\n        text: typeof part.text === \"string\" ? part.text : \"\",\n      };\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:\n          part.source_type === \"url\"\n            ? part.url\n            : part.source_type === \"id\"\n              ? part.id\n              : part.data,\n        mimeType: part.mime_type ?? \"application/octet-stream\",\n        ...((part.source_type === \"url\" || part.source_type === \"id\") && {\n          sourceType: part.source_type,\n        }),\n      };\n    case \"audio\": {\n      const mimeType = part.mime_type ?? \"application/octet-stream\";\n      const subtype = mimeType.startsWith(\"audio/\")\n        ? mimeType.slice(\"audio/\".length)\n        : undefined;\n      return {\n        type: \"file\" as const,\n        filename: subtype ? `audio.${subtype}` : \"audio\",\n        data: part.data,\n        mimeType,\n      };\n    }\n    case \"thinking\":\n      return hasVisibleText(part.thinking)\n        ? { type: \"reasoning\" as const, text: part.thinking }\n        : null;\n    case \"reasoning\": {\n      const text = getReasoningText(part);\n      return text ? { type: \"reasoning\" as const, text } : null;\n    }\n    case \"tool_use\":\n    case \"input_json_delta\":\n      return null;\n    default:\n      return undefined;\n  }\n};\n\nconst hasVisibleText = (text: unknown): boolean =>\n  typeof text === \"string\" && text.trim() !== \"\";\n\n/**\n * Audio output arrives outside the content array: providers leave `content`\n * empty and put the spoken text in `additional_kwargs.audio.transcript`. The\n * audio bytes stay behind because no provider reports their media type, and a\n * streamed response carries raw PCM rather than a playable file.\n */\nexport const withAudioTranscript = <T extends { type: string; text?: unknown }>(\n  parts: readonly T[],\n  additionalKwargs: Record<string, unknown> | undefined,\n): readonly (T | { type: \"text\"; text: string })[] => {\n  const audio = additionalKwargs?.audio as { transcript?: unknown } | undefined;\n  const transcript = audio?.transcript;\n  if (typeof transcript !== \"string\" || !hasVisibleText(transcript))\n    return parts;\n  if (parts.some((part) => part.type === \"text\" && hasVisibleText(part.text)))\n    return parts;\n  return [\n    ...parts.filter((part) => part.type !== \"text\"),\n    { type: \"text\" as const, text: transcript },\n  ];\n};\n\nexport const getCustomMetadata = (\n  additionalKwargs: Record<string, unknown> | undefined,\n): Record<string, unknown> =>\n  (additionalKwargs?.metadata as Record<string, unknown>) ?? {};\n\nexport const uiMessageToDataPart = <\n  TUIMessage extends { name: string; props: Record<string, unknown> },\n>(\n  ui: TUIMessage,\n): DataMessagePart => ({\n  type: \"data\",\n  name: ui.name,\n  data: ui.props,\n});\n\n/**\n * Audio media types that reach a provider's audio input through the LangChain\n * `audio` block. langchain-core derives OpenAI's `input_audio.format` by\n * splitting `mime_type` on `/`, and that format is a wav-or-mp3 enum, so\n * `audio/mpeg` passes the converter and is rejected at the provider.\n */\nconst audioBlockMimeTypes = new Map<string, \"audio/mp3\" | \"audio/wav\">([\n  [\"audio/mp3\", \"audio/mp3\"],\n  [\"audio/mpeg\", \"audio/mp3\"],\n  [\"audio/wav\", \"audio/wav\"],\n  [\"audio/wave\", \"audio/wav\"],\n  [\"audio/x-wav\", \"audio/wav\"],\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) =>\n      part.type === \"file\" || part.type === \"image\" || part.type === \"audio\",\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.flatMap((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        const metadata = { filename: part.filename ?? \"file\" };\n        if (part.sourceType === \"id\") {\n          return {\n            type: \"file\" as const,\n            id: part.data,\n            mime_type: part.mimeType,\n            filename: metadata.filename,\n            metadata,\n            source_type: \"id\" as const,\n          };\n        }\n        const source = resolveFilePartSource(part);\n        if (source.kind === \"url\") {\n          return {\n            type: \"file\" as const,\n            url: source.url,\n            mime_type: part.mimeType,\n            filename: metadata.filename,\n            metadata,\n            source_type: \"url\" as const,\n          };\n        }\n        const audioMimeType = audioBlockMimeTypes.get(\n          source.mimeType.toLowerCase(),\n        );\n        if (audioMimeType) {\n          return {\n            type: \"audio\" as const,\n            data: source.data,\n            mime_type: audioMimeType,\n            source_type: \"base64\" as const,\n          };\n        }\n        return {\n          type: \"file\" as const,\n          data: source.data,\n          mime_type: source.mimeType,\n          filename: metadata.filename,\n          metadata,\n          source_type: \"base64\" as const,\n        };\n      }\n      case \"audio\": {\n        const parsed = parseDataUrl(part.audio.data);\n        return {\n          type: \"audio\" as const,\n          data: parsed?.data ?? part.audio.data,\n          mime_type: `audio/${part.audio.format}`,\n          source_type: \"base64\" as const,\n        };\n      }\n      case \"data\":\n        return [];\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported.\");\n      default: {\n        const _exhaustiveCheck: \"reasoning\" | \"source\" | \"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\nconst getReasoningText = (part: {\n  readonly summary?: ReadonlyArray<{ readonly text?: string }>;\n  readonly reasoning?: string;\n}): string => {\n  const summary = part.summary?.map((s) => s?.text ?? \"\").join(\"\\n\\n\\n\") ?? \"\";\n  if (hasVisibleText(summary)) return summary;\n  const reasoning = part.reasoning ?? \"\";\n  return hasVisibleText(reasoning) ? reasoning : \"\";\n};\n\nexport const createLangChainStreamingTimingAccessors = <\n  TMessage extends {\n    id?: string | undefined;\n    content?: unknown;\n    tool_calls?: readonly unknown[] | undefined;\n  },\n>(\n  getType: (message: TMessage) => string,\n): StreamingTimingAccessors<TMessage> => {\n  const findAiMessage = (\n    messages: readonly TMessage[],\n    messageId: string,\n  ): TMessage | undefined =>\n    messages.find(\n      (message) => getType(message) === \"ai\" && message.id === messageId,\n    );\n\n  const getTextLength = (\n    messages: readonly TMessage[],\n    messageId: string,\n  ): number => {\n    const message = findAiMessage(messages, messageId);\n    if (!message) return 0;\n    const content = message.content;\n    if (typeof content === \"string\") return content.length;\n    if (!Array.isArray(content)) return 0;\n    let len = 0;\n    for (const part of content as readonly LangChainContentBlock[]) {\n      if (typeof part !== \"object\" || part === null) continue;\n      switch (part.type) {\n        case \"text\":\n        case \"text_delta\":\n          if (typeof part.text === \"string\") len += part.text.length;\n          break;\n        case \"thinking\":\n          if (hasVisibleText(part.thinking)) len += part.thinking.length;\n          break;\n        case \"reasoning\":\n          len += getReasoningText(part).length;\n          break;\n      }\n    }\n    return len;\n  };\n\n  const getToolCallCount = (\n    messages: readonly TMessage[],\n    messageId: string,\n  ): number => findAiMessage(messages, messageId)?.tool_calls?.length ?? 0;\n\n  const getAssistantMessageId = (\n    messages: readonly TMessage[],\n  ): string | undefined => {\n    for (let i = messages.length - 1; i >= 0; i--) {\n      const message = messages[i];\n      if (message && getType(message) === \"ai\" && message.id) return message.id;\n    }\n    return undefined;\n  };\n\n  return {\n    getAssistantMessageId,\n    getTextLength,\n    getToolCallCount,\n  };\n};\n\n/**\n * Resolve the assistant message a `UIMessage` belongs to: the parent id comes\n * from `metadata.message_id` (Python SDK) or `metadata.id` (JS SDK).\n */\nexport const getUIMessageParentId = (ui: {\n  metadata?: { message_id?: string; id?: string } | undefined;\n}): string | undefined => ui.metadata?.message_id ?? ui.metadata?.id;\n\n/**\n * Group the graph's accumulated `UIMessage`s by the assistant message they\n * belong to. Non-array state and entries without a parent link are dropped.\n */\nexport const groupUIMessagesByParent = <\n  T extends {\n    metadata?: { message_id?: string; id?: string } | undefined;\n  },\n>(\n  value: unknown,\n): Map<string, T[]> => {\n  const map = new Map<string, T[]>();\n  if (!Array.isArray(value)) return map;\n  for (const ui of value as T[]) {\n    const parentId = getUIMessageParentId(ui);\n    if (!parentId) continue;\n    const existing = map.get(parentId);\n    if (existing) {\n      existing.push(ui);\n    } else {\n      map.set(parentId, [ui]);\n    }\n  }\n  return map;\n};\n"],"mappings":";;AAwDA,MAAa,gCACX,SAC4C;CAE5C,QADa,KAAK,MAClB;EACE,KAAK;EACL,KAAK,cACH,OAAO;GACL,MAAM;GACN,MAAM,OAAO,KAAK,SAAS,WAAW,KAAK,OAAO;EACpD;EACF,KAAK,aAAa;GAChB,MAAM,QACJ,OAAO,KAAK,cAAc,WACtB,KAAK,YACL,KAAK,WAAW;GACtB,IAAI,CAAC,OAAO,OAAO;GACnB,OAAO;IAAE,MAAM;IAAkB;GAAM;EACzC;EACA,KAAK,QACH,OAAO;GACL,MAAM;GACN,UAAU,KAAK,UAAU,YAAY;GACrC,MACE,KAAK,gBAAgB,QACjB,KAAK,MACL,KAAK,gBAAgB,OACnB,KAAK,KACL,KAAK;GACb,UAAU,KAAK,aAAa;GAC5B,IAAK,KAAK,gBAAgB,SAAS,KAAK,gBAAgB,SAAS,EAC/D,YAAY,KAAK,YACnB;EACF;EACF,KAAK,SAAS;GACZ,MAAM,WAAW,KAAK,aAAa;GACnC,MAAM,UAAU,SAAS,WAAW,QAAQ,IACxC,SAAS,MAAM,CAAe,IAC9B,KAAA;GACJ,OAAO;IACL,MAAM;IACN,UAAU,UAAU,SAAS,YAAY;IACzC,MAAM,KAAK;IACX;GACF;EACF;EACA,KAAK,YACH,OAAO,eAAe,KAAK,QAAQ,IAC/B;GAAE,MAAM;GAAsB,MAAM,KAAK;EAAS,IAClD;EACN,KAAK,aAAa;GAChB,MAAM,OAAO,iBAAiB,IAAI;GAClC,OAAO,OAAO;IAAE,MAAM;IAAsB;GAAK,IAAI;EACvD;EACA,KAAK;EACL,KAAK,oBACH,OAAO;EACT,SACE;CACJ;AACF;AAEA,MAAM,kBAAkB,SACtB,OAAO,SAAS,YAAY,KAAK,KAAK,MAAM;;;;;;;AAQ9C,MAAa,uBACX,OACA,qBACoD;CAEpD,MAAM,cADQ,kBAAkB,MAAA,EACN;CAC1B,IAAI,OAAO,eAAe,YAAY,CAAC,eAAe,UAAU,GAC9D,OAAO;CACT,IAAI,MAAM,MAAM,SAAS,KAAK,SAAS,UAAU,eAAe,KAAK,IAAI,CAAC,GACxE,OAAO;CACT,OAAO,CACL,GAAG,MAAM,QAAQ,SAAS,KAAK,SAAS,MAAM,GAC9C;EAAE,MAAM;EAAiB,MAAM;CAAW,CAC5C;AACF;AAEA,MAAa,qBACX,qBAEC,kBAAkB,YAAwC,CAAC;AAE9D,MAAa,uBAGX,QACqB;CACrB,MAAM;CACN,MAAM,GAAG;CACT,MAAM,GAAG;AACX;;;;;;;AAQA,MAAM,sCAAsB,IAAI,IAAuC;CACrE,CAAC,aAAa,WAAW;CACzB,CAAC,cAAc,WAAW;CAC1B,CAAC,aAAa,WAAW;CACzB,CAAC,cAAc,WAAW;CAC1B,CAAC,eAAe,WAAW;AAC7B,CAAC;AAED,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,SACC,KAAK,SAAS,UAAU,KAAK,SAAS,WAAW,KAAK,SAAS,OACnE;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,SAAS,SAAS;EAC3C,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,QAAQ;IACX,MAAM,WAAW,EAAE,UAAU,KAAK,YAAY,OAAO;IACrD,IAAI,KAAK,eAAe,MACtB,OAAO;KACL,MAAM;KACN,IAAI,KAAK;KACT,WAAW,KAAK;KAChB,UAAU,SAAS;KACnB;KACA,aAAa;IACf;IAEF,MAAM,SAAS,sBAAsB,IAAI;IACzC,IAAI,OAAO,SAAS,OAClB,OAAO;KACL,MAAM;KACN,KAAK,OAAO;KACZ,WAAW,KAAK;KAChB,UAAU,SAAS;KACnB;KACA,aAAa;IACf;IAEF,MAAM,gBAAgB,oBAAoB,IACxC,OAAO,SAAS,YAAY,CAC9B;IACA,IAAI,eACF,OAAO;KACL,MAAM;KACN,MAAM,OAAO;KACb,WAAW;KACX,aAAa;IACf;IAEF,OAAO;KACL,MAAM;KACN,MAAM,OAAO;KACb,WAAW,OAAO;KAClB,UAAU,SAAS;KACnB;KACA,aAAa;IACf;GACF;GACA,KAAK,SAEH,OAAO;IACL,MAAM;IACN,MAHa,aAAa,KAAK,MAAM,IAG1B,CAAC,EAAE,QAAQ,KAAK,MAAM;IACjC,WAAW,SAAS,KAAK,MAAM;IAC/B,aAAa;GACf;GAEF,KAAK,QACH,OAAO,CAAC;GACV,KAAK,aACH,MAAM,IAAI,MAAM,sCAAsC;GACxD,SAEE,MAAM,IAAI,MACR,yCAAyCA,MAC3C;EAEJ;CACF,CAAC;CAED,IAAI,QAAQ,WAAW,KAAK,QAAQ,EAAE,EAAE,SAAS,QAC/C,OAAO,QAAQ,EAAE,CAAC,QAAQ;CAE5B,OAAO;AACT;AAEA,MAAM,oBAAoB,SAGZ;CACZ,MAAM,UAAU,KAAK,SAAS,KAAK,MAAM,GAAG,QAAQ,EAAE,CAAC,CAAC,KAAK,QAAQ,KAAK;CAC1E,IAAI,eAAe,OAAO,GAAG,OAAO;CACpC,MAAM,YAAY,KAAK,aAAa;CACpC,OAAO,eAAe,SAAS,IAAI,YAAY;AACjD;AAEA,MAAa,2CAOX,YACuC;CACvC,MAAM,iBACJ,UACA,cAEA,SAAS,MACN,YAAY,QAAQ,OAAO,MAAM,QAAQ,QAAQ,OAAO,SAC3D;CAEF,MAAM,iBACJ,UACA,cACW;EACX,MAAM,UAAU,cAAc,UAAU,SAAS;EACjD,IAAI,CAAC,SAAS,OAAO;EACrB,MAAM,UAAU,QAAQ;EACxB,IAAI,OAAO,YAAY,UAAU,OAAO,QAAQ;EAChD,IAAI,CAAC,MAAM,QAAQ,OAAO,GAAG,OAAO;EACpC,IAAI,MAAM;EACV,KAAK,MAAM,QAAQ,SAA6C;GAC9D,IAAI,OAAO,SAAS,YAAY,SAAS,MAAM;GAC/C,QAAQ,KAAK,MAAb;IACE,KAAK;IACL,KAAK;KACH,IAAI,OAAO,KAAK,SAAS,UAAU,OAAO,KAAK,KAAK;KACpD;IACF,KAAK;KACH,IAAI,eAAe,KAAK,QAAQ,GAAG,OAAO,KAAK,SAAS;KACxD;IACF,KAAK,aACH,OAAO,iBAAiB,IAAI,CAAC,CAAC;GAElC;EACF;EACA,OAAO;CACT;CAEA,MAAM,oBACJ,UACA,cACW,cAAc,UAAU,SAAS,CAAC,EAAE,YAAY,UAAU;CAEvE,MAAM,yBACJ,aACuB;EACvB,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;GAC7C,MAAM,UAAU,SAAS;GACzB,IAAI,WAAW,QAAQ,OAAO,MAAM,QAAQ,QAAQ,IAAI,OAAO,QAAQ;EACzE;CAEF;CAEA,OAAO;EACL;EACA;EACA;CACF;AACF;;;;;AAMA,MAAa,wBAAwB,OAEX,GAAG,UAAU,cAAc,GAAG,UAAU;;;;;AAMlE,MAAa,2BAKX,UACqB;CACrB,MAAM,sBAAM,IAAI,IAAiB;CACjC,IAAI,CAAC,MAAM,QAAQ,KAAK,GAAG,OAAO;CAClC,KAAK,MAAM,MAAM,OAAc;EAC7B,MAAM,WAAW,qBAAqB,EAAE;EACxC,IAAI,CAAC,UAAU;EACf,MAAM,WAAW,IAAI,IAAI,QAAQ;EACjC,IAAI,UACF,SAAS,KAAK,EAAE;OAEhB,IAAI,IAAI,UAAU,CAAC,EAAE,CAAC;CAE1B;CACA,OAAO;AACT"}

package/dist/useStreamRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamRuntime.d.ts","names":[],"sources":["../src/useStreamRuntime.ts"],"mappings":";;;;cA6Ca,2BACX,WAAW;;kBAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;cA+iBA,mBAAoB,YAAY,yDAAuB"}
+{"version":3,"file":"useStreamRuntime.d.ts","names":[],"sources":["../src/useStreamRuntime.ts"],"mappings":";;;;cAoDa,2BACX,WAAW;;kBAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;cAwjBA,mBAAoB,YAAY,yDAAuB"}

package/dist/useStreamRuntime.js

@@ -8,7 +8,7 @@ import { useLangChainStreamingTiming } from "./streamingTiming.js";
 import { createCloudThreadListAdapterCreateFallback, createToolCallCancellationStub, scanPendingToolCalls } from "@assistant-ui/core/internal";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import { useCloudThreadListAdapter, useExternalMessageConverter, useExternalStoreRuntime, useRemoteThreadListRuntime } from "@assistant-ui/core/react";
-import { useCallback, useEffect, useMemo, useRef, useState } from "react";
+import { useCallback, useEffect, useInsertionEffect, useMemo, useRef, useState } from "react";
 import { generateId, getExternalStoreMessages, pickExternalStoreSharedOptions } from "@assistant-ui/core";
 import { STREAM_CONTROLLER, useChannel, useStream } from "@langchain/react";
 //#region src/useStreamRuntime.ts
@@ -69,7 +69,9 @@ const useStreamThreadRuntime = (options) => {
 		isRunning: effectiveIsRunning
 	});
 	const streamRef = useRef(stream);
-	streamRef.current = stream;
+	useInsertionEffect(() => {
+		streamRef.current = stream;
+	}, [stream]);
 	const activeRunConfigRef = useRef(void 0);
 	const runConfigByMessageIdRef = useRef(/* @__PURE__ */ new Map());
 	const activeThreadIdRef = useRef(externalId);
@@ -106,9 +108,13 @@ const useStreamThreadRuntime = (options) => {
 		}
 	}, [stream.messages]);
 	const visibleMessagesRef = useRef(visibleMessages);
-	visibleMessagesRef.current = visibleMessages;
+	useInsertionEffect(() => {
+		visibleMessagesRef.current = visibleMessages;
+	}, [visibleMessages]);
 	const threadMessagesRef = useRef(threadMessages);
-	threadMessagesRef.current = threadMessages;
+	useInsertionEffect(() => {
+		threadMessagesRef.current = threadMessages;
+	}, [threadMessages]);
 	const stagedMessagesRef = useRef(/* @__PURE__ */ new Map());
 	const stagedBaseMessagesRef = useRef(null);
 	useEffect(() => {
@@ -338,8 +344,6 @@ const useStreamThreadRuntime = (options) => {
 */
 const useStreamRuntime = (rawOptions) => {
 	const { cloud, unstable_threadListAdapter, create, delete: deleteFn, onThreadIdChange, ...options } = rawOptions;
-	const optionsRef = useRef(options);
-	optionsRef.current = options;
 	const aui = useAui();
 	const cloudAdapter = useCloudThreadListAdapter({
 		cloud,
@@ -348,7 +352,7 @@ const useStreamRuntime = (rawOptions) => {
 	});
 	return useRemoteThreadListRuntime({
 		runtimeHook: function RuntimeHook() {
-			return useStreamThreadRuntime(optionsRef.current);
+			return useStreamThreadRuntime(options);
 		},
 		adapter: unstable_threadListAdapter ?? cloudAdapter,
 		allowNesting: true,

package/dist/useStreamRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamRuntime.js","names":[],"sources":["../src/useStreamRuntime.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/store\" />\n\"use client\";\n\nimport { useCallback, useEffect, useMemo, useRef, useState } from \"react\";\nimport type { AppendMessage, ToolExecutionStatus } from \"@assistant-ui/core\";\nimport {\n  generateId,\n  getExternalStoreMessages,\n  pickExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport type { ThreadMessage } from \"@assistant-ui/core\";\nimport {\n  createCloudThreadListAdapterCreateFallback,\n  createToolCallCancellationStub,\n  scanPendingToolCalls,\n} from \"@assistant-ui/core/internal\";\nimport {\n  useCloudThreadListAdapter,\n  useExternalStoreRuntime,\n  useExternalMessageConverter,\n  useRemoteThreadListRuntime,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport { STREAM_CONTROLLER, useChannel, useStream } from \"@langchain/react\";\nimport type { Channel } from \"@langchain/react\";\nimport type {\n  LangChainBaseMessage,\n  LangChainToolCall,\n  UIMessage,\n  UseStreamRuntimeOptions,\n} from \"./types\";\nimport { groupUIMessagesByParent } from \"./converter\";\nexport { groupUIMessagesByParent } from \"./converter\";\nimport {\n  convertLangChainBaseMessage,\n  getMessageContent,\n  getMessageType,\n} from \"./convertMessages\";\nimport { foldUIUpdates, mergeUIMessages } from \"./uiMessages\";\nimport { langChainExtras } from \"./runtimeExtras\";\nimport { resolveForkCheckpoint } from \"./resolveForkCheckpoint\";\nimport { useLangChainStreamingTiming } from \"./streamingTiming\";\n\nconst UI_CUSTOM_CHANNELS: readonly Channel[] = [\"custom\"];\n\nexport const runConfigToSubmitOptions = (\n  runConfig: AppendMessage[\"runConfig\"],\n) =>\n  runConfig?.custom\n    ? { config: { configurable: runConfig.custom } }\n    : undefined;\n\ntype NormalizedRunConfigOptions = NonNullable<\n  ReturnType<typeof runConfigToSubmitOptions>\n>;\n\nconst getPendingToolCalls = (\n  messages: readonly LangChainBaseMessage[],\n): LangChainToolCall[] =>\n  scanPendingToolCalls(\n    messages,\n    (message) => {\n      const type = getMessageType(message);\n      if (type === \"ai\") return { toolCalls: message.tool_calls ?? [] };\n      if (type === \"tool\" && message.tool_call_id) {\n        return { toolCallId: message.tool_call_id };\n      }\n      return undefined;\n    },\n    (toolCall) => toolCall.id,\n  );\n\nconst toStagedHumanMessage = (\n  msg: AppendMessage,\n  id = generateId(),\n): LangChainBaseMessage & { id: string } => ({\n  id,\n  _getType: () => \"human\",\n  content: getMessageContent(msg),\n});\n\nconst humanContentText = (content: LangChainBaseMessage[\"content\"]) => {\n  if (typeof content === \"string\") return content;\n  if (!Array.isArray(content)) return \"\";\n  return content\n    .filter(\n      (part): part is { type: \"text\"; text: string } =>\n        typeof part === \"object\" &&\n        part !== null &&\n        part.type === \"text\" &&\n        typeof part.text === \"string\",\n    )\n    .map((part) => part.text)\n    .join(\"\");\n};\n\nconst hasSameMessageContent = (\n  a: LangChainBaseMessage,\n  b: LangChainBaseMessage,\n) => humanContentText(a.content) === humanContentText(b.content);\n\nconst truncateLangChainBaseMessages = (\n  threadMessages: readonly ThreadMessage[],\n  parentId: string | null,\n): LangChainBaseMessage[] => {\n  if (parentId === null) return [];\n  const parentIndex = threadMessages.findIndex((m) => m.id === parentId);\n  if (parentIndex === -1) return [];\n  const truncated: LangChainBaseMessage[] = [];\n  for (let i = 0; i <= parentIndex && i < threadMessages.length; i++) {\n    truncated.push(\n      ...getExternalStoreMessages<LangChainBaseMessage>(threadMessages[i]!),\n    );\n  }\n  return truncated;\n};\n\ntype DistributiveOmit<T, K extends keyof any> = T extends unknown\n  ? Omit<T, K>\n  : never;\n\nconst useStreamThreadRuntime = (\n  options: DistributiveOmit<\n    UseStreamRuntimeOptions,\n    \"cloud\" | \"unstable_threadListAdapter\" | \"create\" | \"delete\"\n  >,\n) => {\n  const { adapters, autoCancelPendingToolCalls, unstable_allowCancellation } =\n    options;\n  const aui = useAui();\n  const messagesKey = options.messagesKey ?? \"messages\";\n  const uiStateKey = options.uiStateKey ?? \"ui\";\n\n  const externalId = useAuiState((s) => s.threadListItem.externalId) as\n    | string\n    | null;\n  // Object.assign preserves the discriminated transport union; object spread\n  // collapses its arms and no longer satisfies UseStreamOptions.\n  const stream = useStream(\n    Object.assign({}, options, { threadId: externalId }),\n  );\n  const [stagedMessages, setStagedMessages] = useState<\n    LangChainBaseMessage[] | null\n  >(null);\n\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const effectiveIsRunning = stream.isLoading || hasExecutingTools;\n\n  const uiStateValue = stream.values[uiStateKey];\n\n  const customEvents = useChannel(stream, UI_CUSTOM_CHANNELS);\n  const liveUiMessages = useMemo(\n    () => foldUIUpdates(customEvents),\n    [customEvents],\n  );\n\n  const mergedUiMessages = useMemo(\n    () => mergeUIMessages(liveUiMessages, uiStateValue),\n    [liveUiMessages, uiStateValue],\n  );\n\n  const visibleMessages =\n    stagedMessages ?? (stream.messages as LangChainBaseMessage[]);\n\n  const messageTiming = useLangChainStreamingTiming(\n    visibleMessages,\n    effectiveIsRunning,\n  );\n\n  const convertWithUI = useMemo<\n    useExternalMessageConverter.Callback<LangChainBaseMessage>\n  >(() => {\n    const uiMessagesByParent =\n      groupUIMessagesByParent<UIMessage>(mergedUiMessages);\n    return (message, metadata) =>\n      convertLangChainBaseMessage(message, {\n        ...metadata,\n        uiMessagesByParent,\n        messageTiming,\n      });\n  }, [mergedUiMessages, messageTiming]);\n\n  const threadMessages = useExternalMessageConverter({\n    callback: convertWithUI,\n    messages: visibleMessages,\n    isRunning: effectiveIsRunning,\n  });\n\n  const streamRef = useRef(stream);\n  streamRef.current = stream;\n\n  const activeRunConfigRef = useRef<\n    NormalizedRunConfigOptions[\"config\"] | undefined\n  >(undefined);\n  const runConfigByMessageIdRef = useRef(\n    new Map<string, NormalizedRunConfigOptions[\"config\"] | undefined>(),\n  );\n  const activeThreadIdRef = useRef(externalId);\n  const setActiveRunConfig = useCallback(\n    (runConfig: AppendMessage[\"runConfig\"]) => {\n      activeRunConfigRef.current = runConfigToSubmitOptions(runConfig)?.config;\n    },\n    [],\n  );\n  const withActiveRunConfig = useCallback(\n    (submitOptions?: Record<string, unknown>) => {\n      if (submitOptions && \"config\" in submitOptions) return submitOptions;\n      if (activeRunConfigRef.current === undefined) return submitOptions;\n      return { ...submitOptions, config: activeRunConfigRef.current };\n    },\n    [],\n  );\n\n  useEffect(() => {\n    if (\n      activeThreadIdRef.current !== null &&\n      activeThreadIdRef.current !== externalId\n    ) {\n      activeRunConfigRef.current = undefined;\n      runConfigByMessageIdRef.current.clear();\n    }\n    activeThreadIdRef.current = externalId;\n  }, [externalId]);\n\n  useEffect(() => {\n    const messages = stream.messages as readonly LangChainBaseMessage[];\n    const owned = runConfigByMessageIdRef.current;\n    for (let i = messages.length - 1; i >= 0; i--) {\n      const message = messages.at(i);\n      if (\n        !message?.id ||\n        getMessageType(message) !== \"ai\" ||\n        !message.tool_calls?.length\n      ) {\n        continue;\n      }\n      if (owned.has(message.id)) return;\n      break;\n    }\n    for (const message of messages) {\n      if (\n        !message.id ||\n        getMessageType(message) !== \"ai\" ||\n        !message.tool_calls?.length ||\n        owned.has(message.id)\n      ) {\n        continue;\n      }\n      owned.set(message.id, activeRunConfigRef.current);\n    }\n  }, [stream.messages]);\n\n  const visibleMessagesRef = useRef(visibleMessages);\n  visibleMessagesRef.current = visibleMessages;\n\n  const threadMessagesRef = useRef(threadMessages);\n  threadMessagesRef.current = threadMessages;\n\n  const stagedMessagesRef = useRef(\n    new Map<\n      string,\n      {\n        message: LangChainBaseMessage & { id: string };\n        runConfig: AppendMessage[\"runConfig\"];\n        reconcileOnEcho: boolean;\n        baseMessageCount: number;\n      }\n    >(),\n  );\n  const stagedBaseMessagesRef = useRef<LangChainBaseMessage[] | null>(null);\n  useEffect(() => {\n    if (stagedMessagesRef.current.size === 0) return;\n\n    // Staged edits must keep their truncated base while stream updates arrive before promotion.\n    const baseMessages =\n      stagedBaseMessagesRef.current ??\n      (stream.messages as LangChainBaseMessage[]);\n    const remainingStagedMessages: LangChainBaseMessage[] = [];\n    const matchedBaseMessageIndexes = new Set<number>();\n    const visibleStagedIds = new Set(\n      visibleMessagesRef.current.flatMap((m) => (m.id ? [m.id] : [])),\n    );\n    for (const [id, staged] of stagedMessagesRef.current) {\n      if (!visibleStagedIds.has(id)) continue;\n      const echoed = baseMessages.some((message, index) => {\n        if (matchedBaseMessageIndexes.has(index)) return false;\n        if (message.id === id) {\n          matchedBaseMessageIndexes.add(index);\n          return true;\n        }\n        if (\n          !staged.reconcileOnEcho ||\n          index < staged.baseMessageCount ||\n          getMessageType(message) !== \"human\" ||\n          !hasSameMessageContent(message, staged.message)\n        ) {\n          return false;\n        }\n        matchedBaseMessageIndexes.add(index);\n        return true;\n      });\n      if (echoed) stagedMessagesRef.current.delete(id);\n      else remainingStagedMessages.push(staged.message);\n    }\n\n    if (remainingStagedMessages.length === 0) {\n      stagedBaseMessagesRef.current = null;\n      visibleMessagesRef.current = baseMessages;\n      setStagedMessages(null);\n      return;\n    }\n\n    const nextMessages = [...baseMessages, ...remainingStagedMessages];\n    visibleMessagesRef.current = nextMessages;\n    setStagedMessages(nextMessages);\n  }, [stream.messages]);\n\n  const getStagedRun = (parentId: string | null) => {\n    if (!parentId || !stagedMessagesRef.current.has(parentId)) return null;\n\n    const staged: LangChainBaseMessage[] = [];\n    for (const message of visibleMessagesRef.current) {\n      if (message.id && stagedMessagesRef.current.has(message.id)) {\n        staged.push(stagedMessagesRef.current.get(message.id)!.message);\n      }\n      if (message.id === parentId) break;\n    }\n\n    return {\n      messages: staged,\n      runConfig: stagedMessagesRef.current.get(parentId)!.runConfig,\n    };\n  };\n\n  const stageUserMessage = (msg: AppendMessage, reconcileOnEcho = false) => {\n    const stagedMessage = toStagedHumanMessage(msg);\n    stagedMessagesRef.current.set(stagedMessage.id, {\n      message: stagedMessage,\n      runConfig: msg.runConfig,\n      reconcileOnEcho,\n      baseMessageCount: streamRef.current.messages.length,\n    });\n    const nextMessages = [...visibleMessagesRef.current, stagedMessage];\n    visibleMessagesRef.current = nextMessages;\n    setStagedMessages(nextMessages);\n    return stagedMessage;\n  };\n\n  const removeStagedMessage = (id: string) => {\n    if (!stagedMessagesRef.current.delete(id)) return;\n    const nextMessages = visibleMessagesRef.current.filter(\n      (message) => message.id !== id,\n    );\n    visibleMessagesRef.current = nextMessages;\n    if (stagedMessagesRef.current.size === 0) {\n      stagedBaseMessagesRef.current = null;\n      setStagedMessages(null);\n    } else {\n      setStagedMessages(nextMessages);\n    }\n  };\n\n  const extras = useMemo(\n    () =>\n      langChainExtras.provide({\n        interrupt: stream.interrupt,\n        interrupts: stream.interrupts,\n        toolCalls: stream.toolCalls,\n        subagents: stream.subagents,\n        subgraphs: stream.subgraphs,\n        stream,\n        error: stream.error,\n        submit: (values, submitOptions) => {\n          const isResume = values == null || submitOptions?.command != null;\n          return stream.submit(\n            values,\n            isResume ? withActiveRunConfig(submitOptions) : submitOptions,\n          );\n        },\n        respond: (response, respondOptions) =>\n          stream.respond(response, withActiveRunConfig(respondOptions)),\n        respondAll: (responsesById, respondOptions) =>\n          stream.respondAll(responsesById, withActiveRunConfig(respondOptions)),\n        values: stream.values,\n        messagesKey,\n      }),\n    [stream, messagesKey, withActiveRunConfig],\n  );\n\n  const runtime = useExternalStoreRuntime({\n    ...pickExternalStoreSharedOptions(options),\n    isRunning: stream.isLoading,\n    isLoading: stream.isThreadLoading,\n    messages: threadMessages,\n    adapters,\n    extras,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    onNew: async (msg) => {\n      if (!(msg.startRun ?? msg.role === \"user\")) {\n        stageUserMessage(msg);\n        return;\n      }\n\n      const stagedMessage = stageUserMessage(msg, true);\n      const stagedMessageId = stagedMessage.id;\n      setActiveRunConfig(msg.runConfig);\n      const content = getMessageContent(msg);\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(\n              streamRef.current.messages as readonly LangChainBaseMessage[],\n            ).map(createToolCallCancellationStub)\n          : [];\n      // A null threadId is not a no-op for the SDK: it rebinds the controller\n      // away from its self-created thread and forces a fresh one, so the\n      // submit waits for initialization to produce an identity; core no\n      // longer holds appends on that barrier.\n      try {\n        const { externalId } = await aui.threadListItem.initialize();\n        await streamRef.current.submit(\n          {\n            [messagesKey]: [\n              ...cancellations,\n              {\n                id: stagedMessageId,\n                type: \"human\",\n                content,\n              },\n            ],\n          },\n          {\n            ...runConfigToSubmitOptions(msg.runConfig),\n            ...(externalId != null ? { threadId: externalId } : {}),\n          },\n        );\n      } catch (error) {\n        removeStagedMessage(stagedMessageId);\n        throw error;\n      }\n    },\n    onAddToolResult: async ({\n      messageId,\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      artifact,\n    }) => {\n      const runConfig = runConfigByMessageIdRef.current.has(messageId)\n        ? runConfigByMessageIdRef.current.get(messageId)\n        : activeRunConfigRef.current;\n      await stream.submit(\n        {\n          [messagesKey]: [\n            {\n              type: \"tool\",\n              name: toolName,\n              tool_call_id: toolCallId,\n              content: JSON.stringify(result),\n              ...(artifact !== undefined && { artifact }),\n              status: isError ? \"error\" : \"success\",\n            },\n          ],\n        },\n        runConfig === undefined ? undefined : { config: runConfig },\n      );\n    },\n    onReload: async (parentId, config) => {\n      const stagedRun = getStagedRun(parentId);\n      if (stagedRun) {\n        const promotedIds = new Set<string>();\n        for (const message of stagedRun.messages) {\n          if (!message.id) continue;\n          promotedIds.add(message.id);\n          stagedMessagesRef.current.delete(message.id);\n        }\n        stagedBaseMessagesRef.current = null;\n        if (stagedMessagesRef.current.size > 0) {\n          const nextMessages = visibleMessagesRef.current.filter(\n            (message) => !message.id || !promotedIds.has(message.id),\n          );\n          visibleMessagesRef.current = nextMessages;\n          setStagedMessages(nextMessages);\n        } else {\n          setStagedMessages(null);\n        }\n        const runConfig = config.runConfig ?? stagedRun.runConfig;\n        setActiveRunConfig(runConfig);\n        await stream.submit(\n          {\n            [messagesKey]: stagedRun.messages.map((message) => ({\n              id: message.id,\n              type: \"human\",\n              content: message.content,\n            })),\n          },\n          runConfigToSubmitOptions(runConfig),\n        );\n        return;\n      }\n\n      const threadId = externalId;\n      if (!threadId || parentId == null) return;\n      const s = streamRef.current;\n      const checkpointId = await resolveForkCheckpoint(\n        s.client,\n        threadId,\n        s.messages as readonly LangChainBaseMessage[],\n        parentId,\n        config.sourceId,\n        s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(),\n        messagesKey,\n      );\n      if (!checkpointId) return;\n      setActiveRunConfig(config.runConfig);\n      await s.submit(null, {\n        forkFrom: checkpointId,\n        ...runConfigToSubmitOptions(config.runConfig),\n      });\n    },\n    onEdit: async (message) => {\n      if (!(message.startRun ?? message.role === \"user\")) {\n        const truncated = truncateLangChainBaseMessages(\n          threadMessagesRef.current,\n          message.parentId,\n        );\n        const stagedMessage = toStagedHumanMessage(message);\n        stagedMessagesRef.current.set(stagedMessage.id, {\n          message: stagedMessage,\n          runConfig: message.runConfig,\n          reconcileOnEcho: false,\n          baseMessageCount: 0,\n        });\n        stagedBaseMessagesRef.current = truncated;\n        const nextMessages = [...truncated, stagedMessage];\n        visibleMessagesRef.current = nextMessages;\n        setStagedMessages(nextMessages);\n        return;\n      }\n\n      const threadId = externalId;\n      if (!threadId) return;\n      const s = streamRef.current;\n      const checkpointId = await resolveForkCheckpoint(\n        s.client,\n        threadId,\n        s.messages as readonly LangChainBaseMessage[],\n        message.parentId,\n        message.sourceId,\n        s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(),\n        messagesKey,\n      );\n      if (!checkpointId) return;\n      const content = getMessageContent(message);\n      setActiveRunConfig(message.runConfig);\n      await s.submit(\n        { [messagesKey]: [{ type: \"human\", content }] },\n        {\n          forkFrom: checkpointId,\n          ...runConfigToSubmitOptions(message.runConfig),\n        },\n      );\n    },\n    onCancel:\n      unstable_allowCancellation !== false\n        ? async () => {\n            activeRunConfigRef.current = undefined;\n            await stream.stop();\n          }\n        : undefined,\n  });\n\n  return runtime;\n};\n\n/**\n * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.\n * Accepts the same options as `useStream` from `@langchain/react`, plus\n * `cloud` and `adapters`.\n *\n * @example\n * ```tsx\n * import { useStreamRuntime } from \"@assistant-ui/react-langchain\";\n * import { AssistantRuntimeProvider, Thread } from \"@assistant-ui/react\";\n *\n * function App() {\n *   const runtime = useStreamRuntime({\n *     assistantId: \"agent\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   return (\n *     <AssistantRuntimeProvider runtime={runtime}>\n *       <Thread />\n *     </AssistantRuntimeProvider>\n *   );\n * }\n * ```\n */\nexport const useStreamRuntime = (rawOptions: UseStreamRuntimeOptions) => {\n  const {\n    cloud,\n    unstable_threadListAdapter,\n    create,\n    delete: deleteFn,\n    onThreadIdChange,\n    ...options\n  } = rawOptions;\n\n  const optionsRef = useRef(options);\n  optionsRef.current = options;\n\n  const aui = useAui();\n  const cloudAdapter = useCloudThreadListAdapter({\n    cloud,\n    create: createCloudThreadListAdapterCreateFallback(\n      create,\n      aui.threadListItem,\n    ),\n    delete: deleteFn,\n  });\n  const adapter = unstable_threadListAdapter ?? cloudAdapter;\n\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useStreamThreadRuntime(optionsRef.current);\n    },\n    adapter,\n    allowNesting: true,\n    onThreadIdChange,\n  });\n};\n"],"mappings":";;;;;;;;;;;;;;AA2CA,MAAM,qBAAyC,CAAC,QAAQ;AAExD,MAAa,4BACX,cAEA,WAAW,SACP,EAAE,QAAQ,EAAE,cAAc,UAAU,OAAO,EAAE,IAC7C,KAAA;AAMN,MAAM,uBACJ,aAEA,qBACE,WACC,YAAY;CACX,MAAM,OAAO,eAAe,OAAO;CACnC,IAAI,SAAS,MAAM,OAAO,EAAE,WAAW,QAAQ,cAAc,CAAC,EAAE;CAChE,IAAI,SAAS,UAAU,QAAQ,cAC7B,OAAO,EAAE,YAAY,QAAQ,aAAa;AAG9C,IACC,aAAa,SAAS,EACzB;AAEF,MAAM,wBACJ,KACA,KAAK,WAAW,OAC2B;CAC3C;CACA,gBAAgB;CAChB,SAAS,kBAAkB,GAAG;AAChC;AAEA,MAAM,oBAAoB,YAA6C;CACrE,IAAI,OAAO,YAAY,UAAU,OAAO;CACxC,IAAI,CAAC,MAAM,QAAQ,OAAO,GAAG,OAAO;CACpC,OAAO,QACJ,QACE,SACC,OAAO,SAAS,YAChB,SAAS,QACT,KAAK,SAAS,UACd,OAAO,KAAK,SAAS,QACzB,CAAC,CACA,KAAK,SAAS,KAAK,IAAI,CAAC,CACxB,KAAK,EAAE;AACZ;AAEA,MAAM,yBACJ,GACA,MACG,iBAAiB,EAAE,OAAO,MAAM,iBAAiB,EAAE,OAAO;AAE/D,MAAM,iCACJ,gBACA,aAC2B;CAC3B,IAAI,aAAa,MAAM,OAAO,CAAC;CAC/B,MAAM,cAAc,eAAe,WAAW,MAAM,EAAE,OAAO,QAAQ;CACrE,IAAI,gBAAgB,IAAI,OAAO,CAAC;CAChC,MAAM,YAAoC,CAAC;CAC3C,KAAK,IAAI,IAAI,GAAG,KAAK,eAAe,IAAI,eAAe,QAAQ,KAC7D,UAAU,KACR,GAAG,yBAA+C,eAAe,EAAG,CACtE;CAEF,OAAO;AACT;AAMA,MAAM,0BACJ,YAIG;CACH,MAAM,EAAE,UAAU,4BAA4B,+BAC5C;CACF,MAAM,MAAM,OAAO;CACnB,MAAM,cAAc,QAAQ,eAAe;CAC3C,MAAM,aAAa,QAAQ,cAAc;CAEzC,MAAM,aAAa,aAAa,MAAM,EAAE,eAAe,UAAU;CAKjE,MAAM,SAAS,UACb,OAAO,OAAO,CAAC,GAAG,SAAS,EAAE,UAAU,WAAW,CAAC,CACrD;CACA,MAAM,CAAC,gBAAgB,qBAAqB,SAE1C,IAAI;CAEN,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,CAAC,CAAC,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,qBAAqB,OAAO,aAAa;CAE/C,MAAM,eAAe,OAAO,OAAO;CAEnC,MAAM,eAAe,WAAW,QAAQ,kBAAkB;CAC1D,MAAM,iBAAiB,cACf,cAAc,YAAY,GAChC,CAAC,YAAY,CACf;CAEA,MAAM,mBAAmB,cACjB,gBAAgB,gBAAgB,YAAY,GAClD,CAAC,gBAAgB,YAAY,CAC/B;CAEA,MAAM,kBACJ,kBAAmB,OAAO;CAE5B,MAAM,gBAAgB,4BACpB,iBACA,kBACF;CAEA,MAAM,gBAAgB,cAEd;EACN,MAAM,qBACJ,wBAAmC,gBAAgB;EACrD,QAAQ,SAAS,aACf,4BAA4B,SAAS;GACnC,GAAG;GACH;GACA;EACF,CAAC;CACL,GAAG,CAAC,kBAAkB,aAAa,CAAC;CAEpC,MAAM,iBAAiB,4BAA4B;EACjD,UAAU;EACV,UAAU;EACV,WAAW;CACb,CAAC;CAED,MAAM,YAAY,OAAO,MAAM;CAC/B,UAAU,UAAU;CAEpB,MAAM,qBAAqB,OAEzB,KAAA,CAAS;CACX,MAAM,0BAA0B,uBAC9B,IAAI,IAA8D,CACpE;CACA,MAAM,oBAAoB,OAAO,UAAU;CAC3C,MAAM,qBAAqB,aACxB,cAA0C;EACzC,mBAAmB,UAAU,yBAAyB,SAAS,CAAC,EAAE;CACpE,GACA,CAAC,CACH;CACA,MAAM,sBAAsB,aACzB,kBAA4C;EAC3C,IAAI,iBAAiB,YAAY,eAAe,OAAO;EACvD,IAAI,mBAAmB,YAAY,KAAA,GAAW,OAAO;EACrD,OAAO;GAAE,GAAG;GAAe,QAAQ,mBAAmB;EAAQ;CAChE,GACA,CAAC,CACH;CAEA,gBAAgB;EACd,IACE,kBAAkB,YAAY,QAC9B,kBAAkB,YAAY,YAC9B;GACA,mBAAmB,UAAU,KAAA;GAC7B,wBAAwB,QAAQ,MAAM;EACxC;EACA,kBAAkB,UAAU;CAC9B,GAAG,CAAC,UAAU,CAAC;CAEf,gBAAgB;EACd,MAAM,WAAW,OAAO;EACxB,MAAM,QAAQ,wBAAwB;EACtC,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;GAC7C,MAAM,UAAU,SAAS,GAAG,CAAC;GAC7B,IACE,CAAC,SAAS,MACV,eAAe,OAAO,MAAM,QAC5B,CAAC,QAAQ,YAAY,QAErB;GAEF,IAAI,MAAM,IAAI,QAAQ,EAAE,GAAG;GAC3B;EACF;EACA,KAAK,MAAM,WAAW,UAAU;GAC9B,IACE,CAAC,QAAQ,MACT,eAAe,OAAO,MAAM,QAC5B,CAAC,QAAQ,YAAY,UACrB,MAAM,IAAI,QAAQ,EAAE,GAEpB;GAEF,MAAM,IAAI,QAAQ,IAAI,mBAAmB,OAAO;EAClD;CACF,GAAG,CAAC,OAAO,QAAQ,CAAC;CAEpB,MAAM,qBAAqB,OAAO,eAAe;CACjD,mBAAmB,UAAU;CAE7B,MAAM,oBAAoB,OAAO,cAAc;CAC/C,kBAAkB,UAAU;CAE5B,MAAM,oBAAoB,uBACxB,IAAI,IAQF,CACJ;CACA,MAAM,wBAAwB,OAAsC,IAAI;CACxE,gBAAgB;EACd,IAAI,kBAAkB,QAAQ,SAAS,GAAG;EAG1C,MAAM,eACJ,sBAAsB,WACrB,OAAO;EACV,MAAM,0BAAkD,CAAC;EACzD,MAAM,4CAA4B,IAAI,IAAY;EAClD,MAAM,mBAAmB,IAAI,IAC3B,mBAAmB,QAAQ,SAAS,MAAO,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,CAAE,CAChE;EACA,KAAK,MAAM,CAAC,IAAI,WAAW,kBAAkB,SAAS;GACpD,IAAI,CAAC,iBAAiB,IAAI,EAAE,GAAG;GAkB/B,IAjBe,aAAa,MAAM,SAAS,UAAU;IACnD,IAAI,0BAA0B,IAAI,KAAK,GAAG,OAAO;IACjD,IAAI,QAAQ,OAAO,IAAI;KACrB,0BAA0B,IAAI,KAAK;KACnC,OAAO;IACT;IACA,IACE,CAAC,OAAO,mBACR,QAAQ,OAAO,oBACf,eAAe,OAAO,MAAM,WAC5B,CAAC,sBAAsB,SAAS,OAAO,OAAO,GAE9C,OAAO;IAET,0BAA0B,IAAI,KAAK;IACnC,OAAO;GACT,CACS,GAAG,kBAAkB,QAAQ,OAAO,EAAE;QAC1C,wBAAwB,KAAK,OAAO,OAAO;EAClD;EAEA,IAAI,wBAAwB,WAAW,GAAG;GACxC,sBAAsB,UAAU;GAChC,mBAAmB,UAAU;GAC7B,kBAAkB,IAAI;GACtB;EACF;EAEA,MAAM,eAAe,CAAC,GAAG,cAAc,GAAG,uBAAuB;EACjE,mBAAmB,UAAU;EAC7B,kBAAkB,YAAY;CAChC,GAAG,CAAC,OAAO,QAAQ,CAAC;CAEpB,MAAM,gBAAgB,aAA4B;EAChD,IAAI,CAAC,YAAY,CAAC,kBAAkB,QAAQ,IAAI,QAAQ,GAAG,OAAO;EAElE,MAAM,SAAiC,CAAC;EACxC,KAAK,MAAM,WAAW,mBAAmB,SAAS;GAChD,IAAI,QAAQ,MAAM,kBAAkB,QAAQ,IAAI,QAAQ,EAAE,GACxD,OAAO,KAAK,kBAAkB,QAAQ,IAAI,QAAQ,EAAE,CAAC,CAAE,OAAO;GAEhE,IAAI,QAAQ,OAAO,UAAU;EAC/B;EAEA,OAAO;GACL,UAAU;GACV,WAAW,kBAAkB,QAAQ,IAAI,QAAQ,CAAC,CAAE;EACtD;CACF;CAEA,MAAM,oBAAoB,KAAoB,kBAAkB,UAAU;EACxE,MAAM,gBAAgB,qBAAqB,GAAG;EAC9C,kBAAkB,QAAQ,IAAI,cAAc,IAAI;GAC9C,SAAS;GACT,WAAW,IAAI;GACf;GACA,kBAAkB,UAAU,QAAQ,SAAS;EAC/C,CAAC;EACD,MAAM,eAAe,CAAC,GAAG,mBAAmB,SAAS,aAAa;EAClE,mBAAmB,UAAU;EAC7B,kBAAkB,YAAY;EAC9B,OAAO;CACT;CAEA,MAAM,uBAAuB,OAAe;EAC1C,IAAI,CAAC,kBAAkB,QAAQ,OAAO,EAAE,GAAG;EAC3C,MAAM,eAAe,mBAAmB,QAAQ,QAC7C,YAAY,QAAQ,OAAO,EAC9B;EACA,mBAAmB,UAAU;EAC7B,IAAI,kBAAkB,QAAQ,SAAS,GAAG;GACxC,sBAAsB,UAAU;GAChC,kBAAkB,IAAI;EACxB,OACE,kBAAkB,YAAY;CAElC;CAEA,MAAM,SAAS,cAEX,gBAAgB,QAAQ;EACtB,WAAW,OAAO;EAClB,YAAY,OAAO;EACnB,WAAW,OAAO;EAClB,WAAW,OAAO;EAClB,WAAW,OAAO;EAClB;EACA,OAAO,OAAO;EACd,SAAS,QAAQ,kBAAkB;GACjC,MAAM,WAAW,UAAU,QAAQ,eAAe,WAAW;GAC7D,OAAO,OAAO,OACZ,QACA,WAAW,oBAAoB,aAAa,IAAI,aAClD;EACF;EACA,UAAU,UAAU,mBAClB,OAAO,QAAQ,UAAU,oBAAoB,cAAc,CAAC;EAC9D,aAAa,eAAe,mBAC1B,OAAO,WAAW,eAAe,oBAAoB,cAAc,CAAC;EACtE,QAAQ,OAAO;EACf;CACF,CAAC,GACH;EAAC;EAAQ;EAAa;CAAmB,CAC3C;CA0LA,OAxLgB,wBAAwB;EACtC,GAAG,+BAA+B,OAAO;EACzC,WAAW,OAAO;EAClB,WAAW,OAAO;EAClB,UAAU;EACV;EACA;EACA,gCAAgC;EAChC;EACA,OAAO,OAAO,QAAQ;GACpB,IAAI,EAAE,IAAI,YAAY,IAAI,SAAS,SAAS;IAC1C,iBAAiB,GAAG;IACpB;GACF;GAGA,MAAM,kBADgB,iBAAiB,KAAK,IACR,CAAC,CAAC;GACtC,mBAAmB,IAAI,SAAS;GAChC,MAAM,UAAU,kBAAkB,GAAG;GACrC,MAAM,gBACJ,+BAA+B,QAC3B,oBACE,UAAU,QAAQ,QACpB,CAAC,CAAC,IAAI,8BAA8B,IACpC,CAAC;GAKP,IAAI;IACF,MAAM,EAAE,eAAe,MAAM,IAAI,eAAe,WAAW;IAC3D,MAAM,UAAU,QAAQ,OACtB,GACG,cAAc,CACb,GAAG,eACH;KACE,IAAI;KACJ,MAAM;KACN;IACF,CACF,EACF,GACA;KACE,GAAG,yBAAyB,IAAI,SAAS;KACzC,GAAI,cAAc,OAAO,EAAE,UAAU,WAAW,IAAI,CAAC;IACvD,CACF;GACF,SAAS,OAAO;IACd,oBAAoB,eAAe;IACnC,MAAM;GACR;EACF;EACA,iBAAiB,OAAO,EACtB,WACA,YACA,UACA,QACA,SACA,eACI;GACJ,MAAM,YAAY,wBAAwB,QAAQ,IAAI,SAAS,IAC3D,wBAAwB,QAAQ,IAAI,SAAS,IAC7C,mBAAmB;GACvB,MAAM,OAAO,OACX,GACG,cAAc,CACb;IACE,MAAM;IACN,MAAM;IACN,cAAc;IACd,SAAS,KAAK,UAAU,MAAM;IAC9B,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;IACzC,QAAQ,UAAU,UAAU;GAC9B,CACF,EACF,GACA,cAAc,KAAA,IAAY,KAAA,IAAY,EAAE,QAAQ,UAAU,CAC5D;EACF;EACA,UAAU,OAAO,UAAU,WAAW;GACpC,MAAM,YAAY,aAAa,QAAQ;GACvC,IAAI,WAAW;IACb,MAAM,8BAAc,IAAI,IAAY;IACpC,KAAK,MAAM,WAAW,UAAU,UAAU;KACxC,IAAI,CAAC,QAAQ,IAAI;KACjB,YAAY,IAAI,QAAQ,EAAE;KAC1B,kBAAkB,QAAQ,OAAO,QAAQ,EAAE;IAC7C;IACA,sBAAsB,UAAU;IAChC,IAAI,kBAAkB,QAAQ,OAAO,GAAG;KACtC,MAAM,eAAe,mBAAmB,QAAQ,QAC7C,YAAY,CAAC,QAAQ,MAAM,CAAC,YAAY,IAAI,QAAQ,EAAE,CACzD;KACA,mBAAmB,UAAU;KAC7B,kBAAkB,YAAY;IAChC,OACE,kBAAkB,IAAI;IAExB,MAAM,YAAY,OAAO,aAAa,UAAU;IAChD,mBAAmB,SAAS;IAC5B,MAAM,OAAO,OACX,GACG,cAAc,UAAU,SAAS,KAAK,aAAa;KAClD,IAAI,QAAQ;KACZ,MAAM;KACN,SAAS,QAAQ;IACnB,EAAE,EACJ,GACA,yBAAyB,SAAS,CACpC;IACA;GACF;GAEA,MAAM,WAAW;GACjB,IAAI,CAAC,YAAY,YAAY,MAAM;GACnC,MAAM,IAAI,UAAU;GACpB,MAAM,eAAe,MAAM,sBACzB,EAAE,QACF,UACA,EAAE,UACF,UACA,OAAO,UACP,EAAE,kBAAkB,EAAE,sBAAsB,cAAc,GAC1D,WACF;GACA,IAAI,CAAC,cAAc;GACnB,mBAAmB,OAAO,SAAS;GACnC,MAAM,EAAE,OAAO,MAAM;IACnB,UAAU;IACV,GAAG,yBAAyB,OAAO,SAAS;GAC9C,CAAC;EACH;EACA,QAAQ,OAAO,YAAY;GACzB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,MAAM,YAAY,8BAChB,kBAAkB,SAClB,QAAQ,QACV;IACA,MAAM,gBAAgB,qBAAqB,OAAO;IAClD,kBAAkB,QAAQ,IAAI,cAAc,IAAI;KAC9C,SAAS;KACT,WAAW,QAAQ;KACnB,iBAAiB;KACjB,kBAAkB;IACpB,CAAC;IACD,sBAAsB,UAAU;IAChC,MAAM,eAAe,CAAC,GAAG,WAAW,aAAa;IACjD,mBAAmB,UAAU;IAC7B,kBAAkB,YAAY;IAC9B;GACF;GAEA,MAAM,WAAW;GACjB,IAAI,CAAC,UAAU;GACf,MAAM,IAAI,UAAU;GACpB,MAAM,eAAe,MAAM,sBACzB,EAAE,QACF,UACA,EAAE,UACF,QAAQ,UACR,QAAQ,UACR,EAAE,kBAAkB,EAAE,sBAAsB,cAAc,GAC1D,WACF;GACA,IAAI,CAAC,cAAc;GACnB,MAAM,UAAU,kBAAkB,OAAO;GACzC,mBAAmB,QAAQ,SAAS;GACpC,MAAM,EAAE,OACN,GAAG,cAAc,CAAC;IAAE,MAAM;IAAS;GAAQ,CAAC,EAAE,GAC9C;IACE,UAAU;IACV,GAAG,yBAAyB,QAAQ,SAAS;GAC/C,CACF;EACF;EACA,UACE,+BAA+B,QAC3B,YAAY;GACV,mBAAmB,UAAU,KAAA;GAC7B,MAAM,OAAO,KAAK;EACpB,IACA,KAAA;CACR,CAEa;AACf;;;;;;;;;;;;;;;;;;;;;;;;;AA0BA,MAAa,oBAAoB,eAAwC;CACvE,MAAM,EACJ,OACA,4BACA,QACA,QAAQ,UACR,kBACA,GAAG,YACD;CAEJ,MAAM,aAAa,OAAO,OAAO;CACjC,WAAW,UAAU;CAErB,MAAM,MAAM,OAAO;CACnB,MAAM,eAAe,0BAA0B;EAC7C;EACA,QAAQ,2CACN,QACA,IAAI,cACN;EACA,QAAQ;CACV,CAAC;CAGD,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAClC,OAAO,uBAAuB,WAAW,OAAO;EAClD;EACA,SANc,8BAA8B;EAO5C,cAAc;EACd;CACF,CAAC;AACH"}
+{"version":3,"file":"useStreamRuntime.js","names":[],"sources":["../src/useStreamRuntime.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/store\" />\n\"use client\";\n\nimport {\n  useCallback,\n  useEffect,\n  useInsertionEffect,\n  useMemo,\n  useRef,\n  useState,\n} from \"react\";\nimport type { AppendMessage, ToolExecutionStatus } from \"@assistant-ui/core\";\nimport {\n  generateId,\n  getExternalStoreMessages,\n  pickExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport type { ThreadMessage } from \"@assistant-ui/core\";\nimport {\n  createCloudThreadListAdapterCreateFallback,\n  createToolCallCancellationStub,\n  scanPendingToolCalls,\n} from \"@assistant-ui/core/internal\";\nimport {\n  useCloudThreadListAdapter,\n  useExternalStoreRuntime,\n  useExternalMessageConverter,\n  useRemoteThreadListRuntime,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport { STREAM_CONTROLLER, useChannel, useStream } from \"@langchain/react\";\nimport type { Channel } from \"@langchain/react\";\nimport type {\n  LangChainBaseMessage,\n  LangChainToolCall,\n  UIMessage,\n  UseStreamRuntimeOptions,\n} from \"./types\";\nimport { groupUIMessagesByParent } from \"./converter\";\nexport { groupUIMessagesByParent } from \"./converter\";\nimport {\n  convertLangChainBaseMessage,\n  getMessageContent,\n  getMessageType,\n} from \"./convertMessages\";\nimport { foldUIUpdates, mergeUIMessages } from \"./uiMessages\";\nimport { langChainExtras } from \"./runtimeExtras\";\nimport { resolveForkCheckpoint } from \"./resolveForkCheckpoint\";\nimport { useLangChainStreamingTiming } from \"./streamingTiming\";\n\nconst UI_CUSTOM_CHANNELS: readonly Channel[] = [\"custom\"];\n\nexport const runConfigToSubmitOptions = (\n  runConfig: AppendMessage[\"runConfig\"],\n) =>\n  runConfig?.custom\n    ? { config: { configurable: runConfig.custom } }\n    : undefined;\n\ntype NormalizedRunConfigOptions = NonNullable<\n  ReturnType<typeof runConfigToSubmitOptions>\n>;\n\nconst getPendingToolCalls = (\n  messages: readonly LangChainBaseMessage[],\n): LangChainToolCall[] =>\n  scanPendingToolCalls(\n    messages,\n    (message) => {\n      const type = getMessageType(message);\n      if (type === \"ai\") return { toolCalls: message.tool_calls ?? [] };\n      if (type === \"tool\" && message.tool_call_id) {\n        return { toolCallId: message.tool_call_id };\n      }\n      return undefined;\n    },\n    (toolCall) => toolCall.id,\n  );\n\nconst toStagedHumanMessage = (\n  msg: AppendMessage,\n  id = generateId(),\n): LangChainBaseMessage & { id: string } => ({\n  id,\n  _getType: () => \"human\",\n  content: getMessageContent(msg),\n});\n\nconst humanContentText = (content: LangChainBaseMessage[\"content\"]) => {\n  if (typeof content === \"string\") return content;\n  if (!Array.isArray(content)) return \"\";\n  return content\n    .filter(\n      (part): part is { type: \"text\"; text: string } =>\n        typeof part === \"object\" &&\n        part !== null &&\n        part.type === \"text\" &&\n        typeof part.text === \"string\",\n    )\n    .map((part) => part.text)\n    .join(\"\");\n};\n\nconst hasSameMessageContent = (\n  a: LangChainBaseMessage,\n  b: LangChainBaseMessage,\n) => humanContentText(a.content) === humanContentText(b.content);\n\nconst truncateLangChainBaseMessages = (\n  threadMessages: readonly ThreadMessage[],\n  parentId: string | null,\n): LangChainBaseMessage[] => {\n  if (parentId === null) return [];\n  const parentIndex = threadMessages.findIndex((m) => m.id === parentId);\n  if (parentIndex === -1) return [];\n  const truncated: LangChainBaseMessage[] = [];\n  for (let i = 0; i <= parentIndex && i < threadMessages.length; i++) {\n    truncated.push(\n      ...getExternalStoreMessages<LangChainBaseMessage>(threadMessages[i]!),\n    );\n  }\n  return truncated;\n};\n\ntype DistributiveOmit<T, K extends keyof any> = T extends unknown\n  ? Omit<T, K>\n  : never;\n\nconst useStreamThreadRuntime = (\n  options: DistributiveOmit<\n    UseStreamRuntimeOptions,\n    \"cloud\" | \"unstable_threadListAdapter\" | \"create\" | \"delete\"\n  >,\n) => {\n  const { adapters, autoCancelPendingToolCalls, unstable_allowCancellation } =\n    options;\n  const aui = useAui();\n  const messagesKey = options.messagesKey ?? \"messages\";\n  const uiStateKey = options.uiStateKey ?? \"ui\";\n\n  const externalId = useAuiState((s) => s.threadListItem.externalId) as\n    | string\n    | null;\n  // Object.assign preserves the discriminated transport union; object spread\n  // collapses its arms and no longer satisfies UseStreamOptions.\n  const stream = useStream(\n    Object.assign({}, options, { threadId: externalId }),\n  );\n  const [stagedMessages, setStagedMessages] = useState<\n    LangChainBaseMessage[] | null\n  >(null);\n\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const effectiveIsRunning = stream.isLoading || hasExecutingTools;\n\n  const uiStateValue = stream.values[uiStateKey];\n\n  const customEvents = useChannel(stream, UI_CUSTOM_CHANNELS);\n  const liveUiMessages = useMemo(\n    () => foldUIUpdates(customEvents),\n    [customEvents],\n  );\n\n  const mergedUiMessages = useMemo(\n    () => mergeUIMessages(liveUiMessages, uiStateValue),\n    [liveUiMessages, uiStateValue],\n  );\n\n  const visibleMessages =\n    stagedMessages ?? (stream.messages as LangChainBaseMessage[]);\n\n  const messageTiming = useLangChainStreamingTiming(\n    visibleMessages,\n    effectiveIsRunning,\n  );\n\n  const convertWithUI = useMemo<\n    useExternalMessageConverter.Callback<LangChainBaseMessage>\n  >(() => {\n    const uiMessagesByParent =\n      groupUIMessagesByParent<UIMessage>(mergedUiMessages);\n    return (message, metadata) =>\n      convertLangChainBaseMessage(message, {\n        ...metadata,\n        uiMessagesByParent,\n        messageTiming,\n      });\n  }, [mergedUiMessages, messageTiming]);\n\n  const threadMessages = useExternalMessageConverter({\n    callback: convertWithUI,\n    messages: visibleMessages,\n    isRunning: effectiveIsRunning,\n  });\n\n  const streamRef = useRef(stream);\n  useInsertionEffect(() => {\n    streamRef.current = stream;\n  }, [stream]);\n\n  const activeRunConfigRef = useRef<\n    NormalizedRunConfigOptions[\"config\"] | undefined\n  >(undefined);\n  const runConfigByMessageIdRef = useRef(\n    new Map<string, NormalizedRunConfigOptions[\"config\"] | undefined>(),\n  );\n  const activeThreadIdRef = useRef(externalId);\n  const setActiveRunConfig = useCallback(\n    (runConfig: AppendMessage[\"runConfig\"]) => {\n      activeRunConfigRef.current = runConfigToSubmitOptions(runConfig)?.config;\n    },\n    [],\n  );\n  const withActiveRunConfig = useCallback(\n    (submitOptions?: Record<string, unknown>) => {\n      if (submitOptions && \"config\" in submitOptions) return submitOptions;\n      if (activeRunConfigRef.current === undefined) return submitOptions;\n      return { ...submitOptions, config: activeRunConfigRef.current };\n    },\n    [],\n  );\n\n  useEffect(() => {\n    if (\n      activeThreadIdRef.current !== null &&\n      activeThreadIdRef.current !== externalId\n    ) {\n      activeRunConfigRef.current = undefined;\n      runConfigByMessageIdRef.current.clear();\n    }\n    activeThreadIdRef.current = externalId;\n  }, [externalId]);\n\n  useEffect(() => {\n    const messages = stream.messages as readonly LangChainBaseMessage[];\n    const owned = runConfigByMessageIdRef.current;\n    for (let i = messages.length - 1; i >= 0; i--) {\n      const message = messages.at(i);\n      if (\n        !message?.id ||\n        getMessageType(message) !== \"ai\" ||\n        !message.tool_calls?.length\n      ) {\n        continue;\n      }\n      if (owned.has(message.id)) return;\n      break;\n    }\n    for (const message of messages) {\n      if (\n        !message.id ||\n        getMessageType(message) !== \"ai\" ||\n        !message.tool_calls?.length ||\n        owned.has(message.id)\n      ) {\n        continue;\n      }\n      owned.set(message.id, activeRunConfigRef.current);\n    }\n  }, [stream.messages]);\n\n  const visibleMessagesRef = useRef(visibleMessages);\n  useInsertionEffect(() => {\n    visibleMessagesRef.current = visibleMessages;\n  }, [visibleMessages]);\n\n  const threadMessagesRef = useRef(threadMessages);\n  useInsertionEffect(() => {\n    threadMessagesRef.current = threadMessages;\n  }, [threadMessages]);\n\n  const stagedMessagesRef = useRef(\n    new Map<\n      string,\n      {\n        message: LangChainBaseMessage & { id: string };\n        runConfig: AppendMessage[\"runConfig\"];\n        reconcileOnEcho: boolean;\n        baseMessageCount: number;\n      }\n    >(),\n  );\n  const stagedBaseMessagesRef = useRef<LangChainBaseMessage[] | null>(null);\n  useEffect(() => {\n    if (stagedMessagesRef.current.size === 0) return;\n\n    // Staged edits must keep their truncated base while stream updates arrive before promotion.\n    const baseMessages =\n      stagedBaseMessagesRef.current ??\n      (stream.messages as LangChainBaseMessage[]);\n    const remainingStagedMessages: LangChainBaseMessage[] = [];\n    const matchedBaseMessageIndexes = new Set<number>();\n    const visibleStagedIds = new Set(\n      visibleMessagesRef.current.flatMap((m) => (m.id ? [m.id] : [])),\n    );\n    for (const [id, staged] of stagedMessagesRef.current) {\n      if (!visibleStagedIds.has(id)) continue;\n      const echoed = baseMessages.some((message, index) => {\n        if (matchedBaseMessageIndexes.has(index)) return false;\n        if (message.id === id) {\n          matchedBaseMessageIndexes.add(index);\n          return true;\n        }\n        if (\n          !staged.reconcileOnEcho ||\n          index < staged.baseMessageCount ||\n          getMessageType(message) !== \"human\" ||\n          !hasSameMessageContent(message, staged.message)\n        ) {\n          return false;\n        }\n        matchedBaseMessageIndexes.add(index);\n        return true;\n      });\n      if (echoed) stagedMessagesRef.current.delete(id);\n      else remainingStagedMessages.push(staged.message);\n    }\n\n    if (remainingStagedMessages.length === 0) {\n      stagedBaseMessagesRef.current = null;\n      visibleMessagesRef.current = baseMessages;\n      // Reconciling against the upstream stream mutates the staged refs above,\n      // which cannot happen during render.\n      // eslint-disable-next-line react-hooks/set-state-in-effect\n      setStagedMessages(null);\n      return;\n    }\n\n    const nextMessages = [...baseMessages, ...remainingStagedMessages];\n    visibleMessagesRef.current = nextMessages;\n    setStagedMessages(nextMessages);\n  }, [stream.messages]);\n\n  const getStagedRun = (parentId: string | null) => {\n    if (!parentId || !stagedMessagesRef.current.has(parentId)) return null;\n\n    const staged: LangChainBaseMessage[] = [];\n    for (const message of visibleMessagesRef.current) {\n      if (message.id && stagedMessagesRef.current.has(message.id)) {\n        staged.push(stagedMessagesRef.current.get(message.id)!.message);\n      }\n      if (message.id === parentId) break;\n    }\n\n    return {\n      messages: staged,\n      runConfig: stagedMessagesRef.current.get(parentId)!.runConfig,\n    };\n  };\n\n  const stageUserMessage = (msg: AppendMessage, reconcileOnEcho = false) => {\n    const stagedMessage = toStagedHumanMessage(msg);\n    stagedMessagesRef.current.set(stagedMessage.id, {\n      message: stagedMessage,\n      runConfig: msg.runConfig,\n      reconcileOnEcho,\n      baseMessageCount: streamRef.current.messages.length,\n    });\n    const nextMessages = [...visibleMessagesRef.current, stagedMessage];\n    visibleMessagesRef.current = nextMessages;\n    setStagedMessages(nextMessages);\n    return stagedMessage;\n  };\n\n  const removeStagedMessage = (id: string) => {\n    if (!stagedMessagesRef.current.delete(id)) return;\n    const nextMessages = visibleMessagesRef.current.filter(\n      (message) => message.id !== id,\n    );\n    visibleMessagesRef.current = nextMessages;\n    if (stagedMessagesRef.current.size === 0) {\n      stagedBaseMessagesRef.current = null;\n      setStagedMessages(null);\n    } else {\n      setStagedMessages(nextMessages);\n    }\n  };\n\n  const extras = useMemo(\n    () =>\n      langChainExtras.provide({\n        interrupt: stream.interrupt,\n        interrupts: stream.interrupts,\n        toolCalls: stream.toolCalls,\n        subagents: stream.subagents,\n        subgraphs: stream.subgraphs,\n        stream,\n        error: stream.error,\n        submit: (values, submitOptions) => {\n          const isResume = values == null || submitOptions?.command != null;\n          return stream.submit(\n            values,\n            isResume ? withActiveRunConfig(submitOptions) : submitOptions,\n          );\n        },\n        respond: (response, respondOptions) =>\n          stream.respond(response, withActiveRunConfig(respondOptions)),\n        respondAll: (responsesById, respondOptions) =>\n          stream.respondAll(responsesById, withActiveRunConfig(respondOptions)),\n        values: stream.values,\n        messagesKey,\n      }),\n    [stream, messagesKey, withActiveRunConfig],\n  );\n\n  const runtime = useExternalStoreRuntime({\n    ...pickExternalStoreSharedOptions(options),\n    isRunning: stream.isLoading,\n    isLoading: stream.isThreadLoading,\n    messages: threadMessages,\n    adapters,\n    extras,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    onNew: async (msg) => {\n      if (!(msg.startRun ?? msg.role === \"user\")) {\n        stageUserMessage(msg);\n        return;\n      }\n\n      const stagedMessage = stageUserMessage(msg, true);\n      const stagedMessageId = stagedMessage.id;\n      setActiveRunConfig(msg.runConfig);\n      const content = getMessageContent(msg);\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(\n              streamRef.current.messages as readonly LangChainBaseMessage[],\n            ).map(createToolCallCancellationStub)\n          : [];\n      // A null threadId is not a no-op for the SDK: it rebinds the controller\n      // away from its self-created thread and forces a fresh one, so the\n      // submit waits for initialization to produce an identity; core no\n      // longer holds appends on that barrier.\n      try {\n        const { externalId } = await aui.threadListItem.initialize();\n        await streamRef.current.submit(\n          {\n            [messagesKey]: [\n              ...cancellations,\n              {\n                id: stagedMessageId,\n                type: \"human\",\n                content,\n              },\n            ],\n          },\n          {\n            ...runConfigToSubmitOptions(msg.runConfig),\n            ...(externalId != null ? { threadId: externalId } : {}),\n          },\n        );\n      } catch (error) {\n        removeStagedMessage(stagedMessageId);\n        throw error;\n      }\n    },\n    onAddToolResult: async ({\n      messageId,\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      artifact,\n    }) => {\n      const runConfig = runConfigByMessageIdRef.current.has(messageId)\n        ? runConfigByMessageIdRef.current.get(messageId)\n        : activeRunConfigRef.current;\n      await stream.submit(\n        {\n          [messagesKey]: [\n            {\n              type: \"tool\",\n              name: toolName,\n              tool_call_id: toolCallId,\n              content: JSON.stringify(result),\n              ...(artifact !== undefined && { artifact }),\n              status: isError ? \"error\" : \"success\",\n            },\n          ],\n        },\n        runConfig === undefined ? undefined : { config: runConfig },\n      );\n    },\n    onReload: async (parentId, config) => {\n      const stagedRun = getStagedRun(parentId);\n      if (stagedRun) {\n        const promotedIds = new Set<string>();\n        for (const message of stagedRun.messages) {\n          if (!message.id) continue;\n          promotedIds.add(message.id);\n          stagedMessagesRef.current.delete(message.id);\n        }\n        stagedBaseMessagesRef.current = null;\n        if (stagedMessagesRef.current.size > 0) {\n          const nextMessages = visibleMessagesRef.current.filter(\n            (message) => !message.id || !promotedIds.has(message.id),\n          );\n          visibleMessagesRef.current = nextMessages;\n          setStagedMessages(nextMessages);\n        } else {\n          setStagedMessages(null);\n        }\n        const runConfig = config.runConfig ?? stagedRun.runConfig;\n        setActiveRunConfig(runConfig);\n        await stream.submit(\n          {\n            [messagesKey]: stagedRun.messages.map((message) => ({\n              id: message.id,\n              type: \"human\",\n              content: message.content,\n            })),\n          },\n          runConfigToSubmitOptions(runConfig),\n        );\n        return;\n      }\n\n      const threadId = externalId;\n      if (!threadId || parentId == null) return;\n      const s = streamRef.current;\n      const checkpointId = await resolveForkCheckpoint(\n        s.client,\n        threadId,\n        s.messages as readonly LangChainBaseMessage[],\n        parentId,\n        config.sourceId,\n        s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(),\n        messagesKey,\n      );\n      if (!checkpointId) return;\n      setActiveRunConfig(config.runConfig);\n      await s.submit(null, {\n        forkFrom: checkpointId,\n        ...runConfigToSubmitOptions(config.runConfig),\n      });\n    },\n    onEdit: async (message) => {\n      if (!(message.startRun ?? message.role === \"user\")) {\n        const truncated = truncateLangChainBaseMessages(\n          threadMessagesRef.current,\n          message.parentId,\n        );\n        const stagedMessage = toStagedHumanMessage(message);\n        stagedMessagesRef.current.set(stagedMessage.id, {\n          message: stagedMessage,\n          runConfig: message.runConfig,\n          reconcileOnEcho: false,\n          baseMessageCount: 0,\n        });\n        stagedBaseMessagesRef.current = truncated;\n        const nextMessages = [...truncated, stagedMessage];\n        visibleMessagesRef.current = nextMessages;\n        setStagedMessages(nextMessages);\n        return;\n      }\n\n      const threadId = externalId;\n      if (!threadId) return;\n      const s = streamRef.current;\n      const checkpointId = await resolveForkCheckpoint(\n        s.client,\n        threadId,\n        s.messages as readonly LangChainBaseMessage[],\n        message.parentId,\n        message.sourceId,\n        s[STREAM_CONTROLLER]?.messageMetadataStore?.getSnapshot?.(),\n        messagesKey,\n      );\n      if (!checkpointId) return;\n      const content = getMessageContent(message);\n      setActiveRunConfig(message.runConfig);\n      await s.submit(\n        { [messagesKey]: [{ type: \"human\", content }] },\n        {\n          forkFrom: checkpointId,\n          ...runConfigToSubmitOptions(message.runConfig),\n        },\n      );\n    },\n    onCancel:\n      unstable_allowCancellation !== false\n        ? async () => {\n            activeRunConfigRef.current = undefined;\n            await stream.stop();\n          }\n        : undefined,\n  });\n\n  return runtime;\n};\n\n/**\n * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.\n * Accepts the same options as `useStream` from `@langchain/react`, plus\n * `cloud` and `adapters`.\n *\n * @example\n * ```tsx\n * import { useStreamRuntime } from \"@assistant-ui/react-langchain\";\n * import { AssistantRuntimeProvider, Thread } from \"@assistant-ui/react\";\n *\n * function App() {\n *   const runtime = useStreamRuntime({\n *     assistantId: \"agent\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   return (\n *     <AssistantRuntimeProvider runtime={runtime}>\n *       <Thread />\n *     </AssistantRuntimeProvider>\n *   );\n * }\n * ```\n */\nexport const useStreamRuntime = (rawOptions: UseStreamRuntimeOptions) => {\n  const {\n    cloud,\n    unstable_threadListAdapter,\n    create,\n    delete: deleteFn,\n    onThreadIdChange,\n    ...options\n  } = rawOptions;\n\n  const aui = useAui();\n  const cloudAdapter = useCloudThreadListAdapter({\n    cloud,\n    create: createCloudThreadListAdapterCreateFallback(\n      create,\n      aui.threadListItem,\n    ),\n    delete: deleteFn,\n  });\n  const adapter = unstable_threadListAdapter ?? cloudAdapter;\n\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useStreamThreadRuntime(options);\n    },\n    adapter,\n    allowNesting: true,\n    onThreadIdChange,\n  });\n};\n"],"mappings":";;;;;;;;;;;;;;AAkDA,MAAM,qBAAyC,CAAC,QAAQ;AAExD,MAAa,4BACX,cAEA,WAAW,SACP,EAAE,QAAQ,EAAE,cAAc,UAAU,OAAO,EAAE,IAC7C,KAAA;AAMN,MAAM,uBACJ,aAEA,qBACE,WACC,YAAY;CACX,MAAM,OAAO,eAAe,OAAO;CACnC,IAAI,SAAS,MAAM,OAAO,EAAE,WAAW,QAAQ,cAAc,CAAC,EAAE;CAChE,IAAI,SAAS,UAAU,QAAQ,cAC7B,OAAO,EAAE,YAAY,QAAQ,aAAa;AAG9C,IACC,aAAa,SAAS,EACzB;AAEF,MAAM,wBACJ,KACA,KAAK,WAAW,OAC2B;CAC3C;CACA,gBAAgB;CAChB,SAAS,kBAAkB,GAAG;AAChC;AAEA,MAAM,oBAAoB,YAA6C;CACrE,IAAI,OAAO,YAAY,UAAU,OAAO;CACxC,IAAI,CAAC,MAAM,QAAQ,OAAO,GAAG,OAAO;CACpC,OAAO,QACJ,QACE,SACC,OAAO,SAAS,YAChB,SAAS,QACT,KAAK,SAAS,UACd,OAAO,KAAK,SAAS,QACzB,CAAC,CACA,KAAK,SAAS,KAAK,IAAI,CAAC,CACxB,KAAK,EAAE;AACZ;AAEA,MAAM,yBACJ,GACA,MACG,iBAAiB,EAAE,OAAO,MAAM,iBAAiB,EAAE,OAAO;AAE/D,MAAM,iCACJ,gBACA,aAC2B;CAC3B,IAAI,aAAa,MAAM,OAAO,CAAC;CAC/B,MAAM,cAAc,eAAe,WAAW,MAAM,EAAE,OAAO,QAAQ;CACrE,IAAI,gBAAgB,IAAI,OAAO,CAAC;CAChC,MAAM,YAAoC,CAAC;CAC3C,KAAK,IAAI,IAAI,GAAG,KAAK,eAAe,IAAI,eAAe,QAAQ,KAC7D,UAAU,KACR,GAAG,yBAA+C,eAAe,EAAG,CACtE;CAEF,OAAO;AACT;AAMA,MAAM,0BACJ,YAIG;CACH,MAAM,EAAE,UAAU,4BAA4B,+BAC5C;CACF,MAAM,MAAM,OAAO;CACnB,MAAM,cAAc,QAAQ,eAAe;CAC3C,MAAM,aAAa,QAAQ,cAAc;CAEzC,MAAM,aAAa,aAAa,MAAM,EAAE,eAAe,UAAU;CAKjE,MAAM,SAAS,UACb,OAAO,OAAO,CAAC,GAAG,SAAS,EAAE,UAAU,WAAW,CAAC,CACrD;CACA,MAAM,CAAC,gBAAgB,qBAAqB,SAE1C,IAAI;CAEN,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,CAAC,CAAC,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,qBAAqB,OAAO,aAAa;CAE/C,MAAM,eAAe,OAAO,OAAO;CAEnC,MAAM,eAAe,WAAW,QAAQ,kBAAkB;CAC1D,MAAM,iBAAiB,cACf,cAAc,YAAY,GAChC,CAAC,YAAY,CACf;CAEA,MAAM,mBAAmB,cACjB,gBAAgB,gBAAgB,YAAY,GAClD,CAAC,gBAAgB,YAAY,CAC/B;CAEA,MAAM,kBACJ,kBAAmB,OAAO;CAE5B,MAAM,gBAAgB,4BACpB,iBACA,kBACF;CAEA,MAAM,gBAAgB,cAEd;EACN,MAAM,qBACJ,wBAAmC,gBAAgB;EACrD,QAAQ,SAAS,aACf,4BAA4B,SAAS;GACnC,GAAG;GACH;GACA;EACF,CAAC;CACL,GAAG,CAAC,kBAAkB,aAAa,CAAC;CAEpC,MAAM,iBAAiB,4BAA4B;EACjD,UAAU;EACV,UAAU;EACV,WAAW;CACb,CAAC;CAED,MAAM,YAAY,OAAO,MAAM;CAC/B,yBAAyB;EACvB,UAAU,UAAU;CACtB,GAAG,CAAC,MAAM,CAAC;CAEX,MAAM,qBAAqB,OAEzB,KAAA,CAAS;CACX,MAAM,0BAA0B,uBAC9B,IAAI,IAA8D,CACpE;CACA,MAAM,oBAAoB,OAAO,UAAU;CAC3C,MAAM,qBAAqB,aACxB,cAA0C;EACzC,mBAAmB,UAAU,yBAAyB,SAAS,CAAC,EAAE;CACpE,GACA,CAAC,CACH;CACA,MAAM,sBAAsB,aACzB,kBAA4C;EAC3C,IAAI,iBAAiB,YAAY,eAAe,OAAO;EACvD,IAAI,mBAAmB,YAAY,KAAA,GAAW,OAAO;EACrD,OAAO;GAAE,GAAG;GAAe,QAAQ,mBAAmB;EAAQ;CAChE,GACA,CAAC,CACH;CAEA,gBAAgB;EACd,IACE,kBAAkB,YAAY,QAC9B,kBAAkB,YAAY,YAC9B;GACA,mBAAmB,UAAU,KAAA;GAC7B,wBAAwB,QAAQ,MAAM;EACxC;EACA,kBAAkB,UAAU;CAC9B,GAAG,CAAC,UAAU,CAAC;CAEf,gBAAgB;EACd,MAAM,WAAW,OAAO;EACxB,MAAM,QAAQ,wBAAwB;EACtC,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;GAC7C,MAAM,UAAU,SAAS,GAAG,CAAC;GAC7B,IACE,CAAC,SAAS,MACV,eAAe,OAAO,MAAM,QAC5B,CAAC,QAAQ,YAAY,QAErB;GAEF,IAAI,MAAM,IAAI,QAAQ,EAAE,GAAG;GAC3B;EACF;EACA,KAAK,MAAM,WAAW,UAAU;GAC9B,IACE,CAAC,QAAQ,MACT,eAAe,OAAO,MAAM,QAC5B,CAAC,QAAQ,YAAY,UACrB,MAAM,IAAI,QAAQ,EAAE,GAEpB;GAEF,MAAM,IAAI,QAAQ,IAAI,mBAAmB,OAAO;EAClD;CACF,GAAG,CAAC,OAAO,QAAQ,CAAC;CAEpB,MAAM,qBAAqB,OAAO,eAAe;CACjD,yBAAyB;EACvB,mBAAmB,UAAU;CAC/B,GAAG,CAAC,eAAe,CAAC;CAEpB,MAAM,oBAAoB,OAAO,cAAc;CAC/C,yBAAyB;EACvB,kBAAkB,UAAU;CAC9B,GAAG,CAAC,cAAc,CAAC;CAEnB,MAAM,oBAAoB,uBACxB,IAAI,IAQF,CACJ;CACA,MAAM,wBAAwB,OAAsC,IAAI;CACxE,gBAAgB;EACd,IAAI,kBAAkB,QAAQ,SAAS,GAAG;EAG1C,MAAM,eACJ,sBAAsB,WACrB,OAAO;EACV,MAAM,0BAAkD,CAAC;EACzD,MAAM,4CAA4B,IAAI,IAAY;EAClD,MAAM,mBAAmB,IAAI,IAC3B,mBAAmB,QAAQ,SAAS,MAAO,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,CAAE,CAChE;EACA,KAAK,MAAM,CAAC,IAAI,WAAW,kBAAkB,SAAS;GACpD,IAAI,CAAC,iBAAiB,IAAI,EAAE,GAAG;GAkB/B,IAjBe,aAAa,MAAM,SAAS,UAAU;IACnD,IAAI,0BAA0B,IAAI,KAAK,GAAG,OAAO;IACjD,IAAI,QAAQ,OAAO,IAAI;KACrB,0BAA0B,IAAI,KAAK;KACnC,OAAO;IACT;IACA,IACE,CAAC,OAAO,mBACR,QAAQ,OAAO,oBACf,eAAe,OAAO,MAAM,WAC5B,CAAC,sBAAsB,SAAS,OAAO,OAAO,GAE9C,OAAO;IAET,0BAA0B,IAAI,KAAK;IACnC,OAAO;GACT,CACS,GAAG,kBAAkB,QAAQ,OAAO,EAAE;QAC1C,wBAAwB,KAAK,OAAO,OAAO;EAClD;EAEA,IAAI,wBAAwB,WAAW,GAAG;GACxC,sBAAsB,UAAU;GAChC,mBAAmB,UAAU;GAI7B,kBAAkB,IAAI;GACtB;EACF;EAEA,MAAM,eAAe,CAAC,GAAG,cAAc,GAAG,uBAAuB;EACjE,mBAAmB,UAAU;EAC7B,kBAAkB,YAAY;CAChC,GAAG,CAAC,OAAO,QAAQ,CAAC;CAEpB,MAAM,gBAAgB,aAA4B;EAChD,IAAI,CAAC,YAAY,CAAC,kBAAkB,QAAQ,IAAI,QAAQ,GAAG,OAAO;EAElE,MAAM,SAAiC,CAAC;EACxC,KAAK,MAAM,WAAW,mBAAmB,SAAS;GAChD,IAAI,QAAQ,MAAM,kBAAkB,QAAQ,IAAI,QAAQ,EAAE,GACxD,OAAO,KAAK,kBAAkB,QAAQ,IAAI,QAAQ,EAAE,CAAC,CAAE,OAAO;GAEhE,IAAI,QAAQ,OAAO,UAAU;EAC/B;EAEA,OAAO;GACL,UAAU;GACV,WAAW,kBAAkB,QAAQ,IAAI,QAAQ,CAAC,CAAE;EACtD;CACF;CAEA,MAAM,oBAAoB,KAAoB,kBAAkB,UAAU;EACxE,MAAM,gBAAgB,qBAAqB,GAAG;EAC9C,kBAAkB,QAAQ,IAAI,cAAc,IAAI;GAC9C,SAAS;GACT,WAAW,IAAI;GACf;GACA,kBAAkB,UAAU,QAAQ,SAAS;EAC/C,CAAC;EACD,MAAM,eAAe,CAAC,GAAG,mBAAmB,SAAS,aAAa;EAClE,mBAAmB,UAAU;EAC7B,kBAAkB,YAAY;EAC9B,OAAO;CACT;CAEA,MAAM,uBAAuB,OAAe;EAC1C,IAAI,CAAC,kBAAkB,QAAQ,OAAO,EAAE,GAAG;EAC3C,MAAM,eAAe,mBAAmB,QAAQ,QAC7C,YAAY,QAAQ,OAAO,EAC9B;EACA,mBAAmB,UAAU;EAC7B,IAAI,kBAAkB,QAAQ,SAAS,GAAG;GACxC,sBAAsB,UAAU;GAChC,kBAAkB,IAAI;EACxB,OACE,kBAAkB,YAAY;CAElC;CAEA,MAAM,SAAS,cAEX,gBAAgB,QAAQ;EACtB,WAAW,OAAO;EAClB,YAAY,OAAO;EACnB,WAAW,OAAO;EAClB,WAAW,OAAO;EAClB,WAAW,OAAO;EAClB;EACA,OAAO,OAAO;EACd,SAAS,QAAQ,kBAAkB;GACjC,MAAM,WAAW,UAAU,QAAQ,eAAe,WAAW;GAC7D,OAAO,OAAO,OACZ,QACA,WAAW,oBAAoB,aAAa,IAAI,aAClD;EACF;EACA,UAAU,UAAU,mBAClB,OAAO,QAAQ,UAAU,oBAAoB,cAAc,CAAC;EAC9D,aAAa,eAAe,mBAC1B,OAAO,WAAW,eAAe,oBAAoB,cAAc,CAAC;EACtE,QAAQ,OAAO;EACf;CACF,CAAC,GACH;EAAC;EAAQ;EAAa;CAAmB,CAC3C;CA0LA,OAxLgB,wBAAwB;EACtC,GAAG,+BAA+B,OAAO;EACzC,WAAW,OAAO;EAClB,WAAW,OAAO;EAClB,UAAU;EACV;EACA;EACA,gCAAgC;EAChC;EACA,OAAO,OAAO,QAAQ;GACpB,IAAI,EAAE,IAAI,YAAY,IAAI,SAAS,SAAS;IAC1C,iBAAiB,GAAG;IACpB;GACF;GAGA,MAAM,kBADgB,iBAAiB,KAAK,IACR,CAAC,CAAC;GACtC,mBAAmB,IAAI,SAAS;GAChC,MAAM,UAAU,kBAAkB,GAAG;GACrC,MAAM,gBACJ,+BAA+B,QAC3B,oBACE,UAAU,QAAQ,QACpB,CAAC,CAAC,IAAI,8BAA8B,IACpC,CAAC;GAKP,IAAI;IACF,MAAM,EAAE,eAAe,MAAM,IAAI,eAAe,WAAW;IAC3D,MAAM,UAAU,QAAQ,OACtB,GACG,cAAc,CACb,GAAG,eACH;KACE,IAAI;KACJ,MAAM;KACN;IACF,CACF,EACF,GACA;KACE,GAAG,yBAAyB,IAAI,SAAS;KACzC,GAAI,cAAc,OAAO,EAAE,UAAU,WAAW,IAAI,CAAC;IACvD,CACF;GACF,SAAS,OAAO;IACd,oBAAoB,eAAe;IACnC,MAAM;GACR;EACF;EACA,iBAAiB,OAAO,EACtB,WACA,YACA,UACA,QACA,SACA,eACI;GACJ,MAAM,YAAY,wBAAwB,QAAQ,IAAI,SAAS,IAC3D,wBAAwB,QAAQ,IAAI,SAAS,IAC7C,mBAAmB;GACvB,MAAM,OAAO,OACX,GACG,cAAc,CACb;IACE,MAAM;IACN,MAAM;IACN,cAAc;IACd,SAAS,KAAK,UAAU,MAAM;IAC9B,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;IACzC,QAAQ,UAAU,UAAU;GAC9B,CACF,EACF,GACA,cAAc,KAAA,IAAY,KAAA,IAAY,EAAE,QAAQ,UAAU,CAC5D;EACF;EACA,UAAU,OAAO,UAAU,WAAW;GACpC,MAAM,YAAY,aAAa,QAAQ;GACvC,IAAI,WAAW;IACb,MAAM,8BAAc,IAAI,IAAY;IACpC,KAAK,MAAM,WAAW,UAAU,UAAU;KACxC,IAAI,CAAC,QAAQ,IAAI;KACjB,YAAY,IAAI,QAAQ,EAAE;KAC1B,kBAAkB,QAAQ,OAAO,QAAQ,EAAE;IAC7C;IACA,sBAAsB,UAAU;IAChC,IAAI,kBAAkB,QAAQ,OAAO,GAAG;KACtC,MAAM,eAAe,mBAAmB,QAAQ,QAC7C,YAAY,CAAC,QAAQ,MAAM,CAAC,YAAY,IAAI,QAAQ,EAAE,CACzD;KACA,mBAAmB,UAAU;KAC7B,kBAAkB,YAAY;IAChC,OACE,kBAAkB,IAAI;IAExB,MAAM,YAAY,OAAO,aAAa,UAAU;IAChD,mBAAmB,SAAS;IAC5B,MAAM,OAAO,OACX,GACG,cAAc,UAAU,SAAS,KAAK,aAAa;KAClD,IAAI,QAAQ;KACZ,MAAM;KACN,SAAS,QAAQ;IACnB,EAAE,EACJ,GACA,yBAAyB,SAAS,CACpC;IACA;GACF;GAEA,MAAM,WAAW;GACjB,IAAI,CAAC,YAAY,YAAY,MAAM;GACnC,MAAM,IAAI,UAAU;GACpB,MAAM,eAAe,MAAM,sBACzB,EAAE,QACF,UACA,EAAE,UACF,UACA,OAAO,UACP,EAAE,kBAAkB,EAAE,sBAAsB,cAAc,GAC1D,WACF;GACA,IAAI,CAAC,cAAc;GACnB,mBAAmB,OAAO,SAAS;GACnC,MAAM,EAAE,OAAO,MAAM;IACnB,UAAU;IACV,GAAG,yBAAyB,OAAO,SAAS;GAC9C,CAAC;EACH;EACA,QAAQ,OAAO,YAAY;GACzB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,MAAM,YAAY,8BAChB,kBAAkB,SAClB,QAAQ,QACV;IACA,MAAM,gBAAgB,qBAAqB,OAAO;IAClD,kBAAkB,QAAQ,IAAI,cAAc,IAAI;KAC9C,SAAS;KACT,WAAW,QAAQ;KACnB,iBAAiB;KACjB,kBAAkB;IACpB,CAAC;IACD,sBAAsB,UAAU;IAChC,MAAM,eAAe,CAAC,GAAG,WAAW,aAAa;IACjD,mBAAmB,UAAU;IAC7B,kBAAkB,YAAY;IAC9B;GACF;GAEA,MAAM,WAAW;GACjB,IAAI,CAAC,UAAU;GACf,MAAM,IAAI,UAAU;GACpB,MAAM,eAAe,MAAM,sBACzB,EAAE,QACF,UACA,EAAE,UACF,QAAQ,UACR,QAAQ,UACR,EAAE,kBAAkB,EAAE,sBAAsB,cAAc,GAC1D,WACF;GACA,IAAI,CAAC,cAAc;GACnB,MAAM,UAAU,kBAAkB,OAAO;GACzC,mBAAmB,QAAQ,SAAS;GACpC,MAAM,EAAE,OACN,GAAG,cAAc,CAAC;IAAE,MAAM;IAAS;GAAQ,CAAC,EAAE,GAC9C;IACE,UAAU;IACV,GAAG,yBAAyB,QAAQ,SAAS;GAC/C,CACF;EACF;EACA,UACE,+BAA+B,QAC3B,YAAY;GACV,mBAAmB,UAAU,KAAA;GAC7B,MAAM,OAAO,KAAK;EACpB,IACA,KAAA;CACR,CAEa;AACf;;;;;;;;;;;;;;;;;;;;;;;;;AA0BA,MAAa,oBAAoB,eAAwC;CACvE,MAAM,EACJ,OACA,4BACA,QACA,QAAQ,UACR,kBACA,GAAG,YACD;CAEJ,MAAM,MAAM,OAAO;CACnB,MAAM,eAAe,0BAA0B;EAC7C;EACA,QAAQ,2CACN,QACA,IAAI,cACN;EACA,QAAQ;CACV,CAAC;CAGD,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAClC,OAAO,uBAAuB,OAAO;EACvC;EACA,SANc,8BAA8B;EAO5C,cAAc;EACd;CACF,CAAC;AACH"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-langchain",
-  "version": "0.0.28",
+  "version": "0.0.30",
   "description": "LangChain useStream adapter for assistant-ui",
   "keywords": [
     "langchain",
@@ -40,10 +40,10 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.3.16",
-    "@assistant-ui/store": "^0.3.11",
+    "@assistant-ui/core": "^0.3.18",
+    "@assistant-ui/store": "^0.3.13",
     "assistant-cloud": "*",
-    "assistant-stream": "^0.3.40"
+    "assistant-stream": "^0.3.42"
   },
   "peerDependencies": {
     "@langchain/react": "^1.0.2",
@@ -59,11 +59,11 @@
     }
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.24",
+    "@assistant-ui/x-buildutils": "0.0.26",
     "@langchain/core": "^1.2.9",
-    "@langchain/langgraph-sdk": "^1.9.31",
-    "@langchain/react": "^1.0.32",
-    "@testing-library/react": "^16.3.2",
+    "@langchain/langgraph-sdk": "^1.10.2",
+    "@langchain/react": "^1.0.35",
+    "@testing-library/react": "^16.3.3",
     "@types/react": "^19.2.18",
     "jsdom": "^30.0.1",
     "react": "^19.2.8",

package/src/convertMessages.test.ts

@@ -1,10 +1,12 @@
-import { describe, expect, it } from "vitest";
+import { describe, expect, it, vi } from "vitest";
 import type { AppendMessage } from "@assistant-ui/core";
 import { convertExternalMessages } from "@assistant-ui/core/react";
 import {
   convertLangChainBaseMessage,
   getMessageContent,
+  getMessageType,
 } from "./convertMessages";
+import { createLangChainStreamingTimingAccessors } from "./converter";
 import type { LangChainBaseMessage, UIMessage } from "./types";
 
 const humanMessage = (content: unknown): LangChainBaseMessage => ({
@@ -754,13 +756,38 @@ describe("convertLangChainBaseMessage reasoning content parts", () => {
     ]);
   });
 
-  it("does not throw when a reasoning block omits both summary and reasoning", () => {
+  it("omits reasoning parts that contain only provider metadata", () => {
     const result = convertLangChainBaseMessage(
-      aiMessage([{ type: "reasoning" }]),
+      aiMessage([
+        { type: "reasoning", signature: "signed", index: 0 },
+        { type: "reasoning", summary: [{ type: "summary_text" }], index: 1 },
+        { type: "text", text: "Answer." },
+      ]),
       {},
     );
 
-    expect(contentOf(result)).toEqual([{ type: "reasoning", text: "" }]);
+    expect(contentOf(result)).toEqual([{ type: "text", text: "Answer." }]);
+  });
+
+  it("renders the summary, falling back to the reasoning string when it is blank", () => {
+    const convert = (summary: string) =>
+      contentOf(
+        convertLangChainBaseMessage(
+          aiMessage([
+            {
+              type: "reasoning",
+              reasoning: "Check net",
+              summary: [{ type: "summary_text", text: summary }],
+            },
+          ]),
+          {},
+        ),
+      );
+
+    expect(convert("temperature")).toEqual([
+      { type: "reasoning", text: "temperature" },
+    ]);
+    expect(convert("  ")).toEqual([{ type: "reasoning", text: "Check net" }]);
   });
 
   it("tolerates null entries inside the summary array", () => {
@@ -1047,3 +1074,162 @@ describe("convertLangChainBaseMessage tool messages", () => {
     ).toThrow(/does not match existing tool call/);
   });
 });
+
+describe("convertLangChainBaseMessage malformed messages", () => {
+  it("reports an unknown type for a message without _getType or type", () => {
+    const message = { id: "msg-3", content: "hello" } as LangChainBaseMessage;
+
+    expect(getMessageType(message)).toBe("unknown");
+    expect(convertLangChainBaseMessage(message, {})).toEqual({
+      role: "system",
+      id: "msg-3",
+      content: [{ type: "text", text: "hello" }],
+    });
+  });
+
+  it("converts a human message with null content to empty content", () => {
+    expect(
+      contentOf(convertLangChainBaseMessage(humanMessage(null), {})),
+    ).toEqual([]);
+  });
+
+  it("keeps tool calls when an ai message carries object content", () => {
+    const result = convertLangChainBaseMessage(
+      {
+        ...aiMessage({ text: "not an array" }),
+        tool_calls: [{ id: "call-1", name: "lookup", args: { q: "x" } }],
+      },
+      {},
+    );
+
+    expect(contentOf(result)).toEqual([
+      {
+        type: "tool-call",
+        toolCallId: "call-1",
+        toolName: "lookup",
+        args: { q: "x" },
+        argsText: '{"q":"x"}',
+      },
+    ]);
+  });
+
+  it("converts a system message with null content to empty text", () => {
+    const result = convertLangChainBaseMessage(
+      { _getType: () => "system", id: "msg-4", content: null },
+      {},
+    );
+
+    expect(contentOf(result)).toEqual([{ type: "text", text: "" }]);
+  });
+
+  it("renders a message without a type or content as empty system text", () => {
+    const result = convertLangChainBaseMessage(
+      { id: "msg-6" } as LangChainBaseMessage,
+      {},
+    );
+
+    expect(contentOf(result)).toEqual([{ type: "text", text: "" }]);
+  });
+
+  it("skips null entries inside a content array", () => {
+    const result = convertLangChainBaseMessage(
+      humanMessage([null, { type: "text", text: "kept" }, undefined]),
+      {},
+    );
+
+    expect(contentOf(result)).toEqual([{ type: "text", text: "kept" }]);
+  });
+
+  it("coalesces a text block without a text field to empty text", () => {
+    const result = convertLangChainBaseMessage(
+      humanMessage([{ type: "text" }]),
+      {},
+    );
+
+    expect(contentOf(result)).toEqual([{ type: "text", text: "" }]);
+  });
+
+  it("skips null entries when measuring streamed text length", () => {
+    const { getTextLength } = createLangChainStreamingTimingAccessors<{
+      id?: string | undefined;
+      content?: unknown;
+      _getType: () => string;
+    }>((message) => message._getType());
+
+    expect(
+      getTextLength(
+        [
+          {
+            _getType: () => "ai",
+            id: "ai-1",
+            content: [null, { type: "text", text: "abc" }, undefined],
+          },
+        ],
+        "ai-1",
+      ),
+    ).toBe(3);
+  });
+
+  it("renders a textless block as empty system text", () => {
+    const result = convertLangChainBaseMessage(
+      {
+        _getType: () => "system",
+        id: "msg-7",
+        content: [{ type: "text" }, { type: "text", text: "kept" }],
+      },
+      {},
+    );
+
+    expect(contentOf(result)).toEqual([{ type: "text", text: "kept" }]);
+  });
+
+  it("skips null entries when collecting system text", () => {
+    const result = convertLangChainBaseMessage(
+      {
+        _getType: () => "system",
+        id: "msg-7",
+        content: [null, { type: "text", text: "kept" }],
+      },
+      {},
+    );
+
+    expect(contentOf(result)).toEqual([{ type: "text", text: "kept" }]);
+  });
+
+  it("converts a system message with object content to empty text", () => {
+    const result = convertLangChainBaseMessage(
+      { _getType: () => "system", id: "msg-5", content: { text: "x" } },
+      {},
+    );
+
+    expect(contentOf(result)).toEqual([{ type: "text", text: "" }]);
+  });
+
+  it("stays silent about non-array content outside development", () => {
+    vi.stubEnv("NODE_ENV", "test");
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    try {
+      convertLangChainBaseMessage(humanMessage(true), {});
+      expect(warn).not.toHaveBeenCalled();
+    } finally {
+      warn.mockRestore();
+      vi.unstubAllEnvs();
+    }
+  });
+
+  it("warns once in development about non-array content", () => {
+    vi.stubEnv("NODE_ENV", "development");
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    try {
+      convertLangChainBaseMessage(humanMessage(42), {});
+      convertLangChainBaseMessage(humanMessage(42), {});
+      expect(warn).toHaveBeenCalledTimes(1);
+      expect(warn).toHaveBeenCalledWith(
+        "Ignoring message content that is neither a string nor an array: number",
+      );
+    } finally {
+      warn.mockRestore();
+      vi.unstubAllEnvs();
+    }
+  });
+});

package/src/convertMessages.ts

@@ -21,27 +21,52 @@ type LangChainMessageConverterMetadata =
     messageTiming?: Record<string, MessageTiming>;
   };
 
+const warnedMalformedMessages = new Set<string>();
+const warnOnceInDevelopment = (message: string) => {
+  if (
+    typeof process === "undefined" ||
+    process?.env?.NODE_ENV !== "development"
+  )
+    return;
+  if (warnedMalformedMessages.has(message)) return;
+  warnedMalformedMessages.add(message);
+  console.warn(message);
+};
+
 export const getMessageType = (message: LangChainBaseMessage): string => {
   if (typeof message._getType === "function") return message._getType();
   if ("type" in message)
     return (message as Record<string, unknown>).type as string;
-  throw new Error("Cannot determine message type");
+  warnOnceInDevelopment(
+    "Cannot determine message type; rendering the message as system text",
+  );
+  return "unknown";
+};
+
+const contentBlocks = (content: unknown): readonly LangChainContentBlock[] => {
+  if (content == null) return [];
+  if (Array.isArray(content))
+    return content.filter(
+      (block) => typeof block === "object" && block !== null,
+    );
+  warnOnceInDevelopment(
+    `Ignoring message content that is neither a string nor an array: ${typeof content}`,
+  );
+  return [];
 };
 
 const contentToParts = (content: unknown) => {
   if (typeof content === "string")
     return [{ type: "text" as const, text: content }];
 
-  const parts = content as readonly LangChainContentBlock[];
-  return parts
+  return contentBlocks(content)
     .map(convertLangChainContentBlock)
     .filter((part) => part !== null && part !== undefined);
 };
 
 const getStringContent = (content: unknown): string => {
   if (typeof content === "string") return content;
-  const parts = content as readonly LangChainContentBlock[];
-  return parts
+  return contentBlocks(content)
     .filter((c): c is { type: "text"; text: string } => c.type === "text")
     .map((c) => c.text)
     .join("");
@@ -137,7 +162,7 @@ export const convertLangChainBaseMessage = (
             text:
               typeof message.content === "string"
                 ? message.content
-                : JSON.stringify(message.content),
+                : (JSON.stringify(message.content) ?? ""),
           },
         ],
       };

package/src/converter.ts

@@ -61,7 +61,10 @@ export const convertLangChainContentBlock = (
   switch (type) {
     case "text":
     case "text_delta":
-      return { type: "text" as const, text: part.text };
+      return {
+        type: "text" as const,
+        text: typeof part.text === "string" ? part.text : "",
+      };
     case "image_url": {
       const image =
         typeof part.image_url === "string"
@@ -98,15 +101,13 @@ export const convertLangChainContentBlock = (
       };
     }
     case "thinking":
-      return { type: "reasoning" as const, text: part.thinking };
-    case "reasoning":
-      return {
-        type: "reasoning" as const,
-        text:
-          part.summary && part.summary.length > 0
-            ? part.summary.map((s) => s?.text ?? "").join("\n\n\n")
-            : (part.reasoning ?? ""),
-      };
+      return hasVisibleText(part.thinking)
+        ? { type: "reasoning" as const, text: part.thinking }
+        : null;
+    case "reasoning": {
+      const text = getReasoningText(part);
+      return text ? { type: "reasoning" as const, text } : null;
+    }
     case "tool_use":
     case "input_json_delta":
       return null;
@@ -262,13 +263,14 @@ export const getMessageContent = (msg: AppendMessage) => {
   return content;
 };
 
-const reasoningTextLength = (part: {
+const getReasoningText = (part: {
   readonly summary?: ReadonlyArray<{ readonly text?: string }>;
   readonly reasoning?: string;
-}): number => {
-  if (part.summary && part.summary.length > 0)
-    return part.summary.map((s) => s?.text ?? "").join("\n\n\n").length;
-  return part.reasoning?.length ?? 0;
+}): string => {
+  const summary = part.summary?.map((s) => s?.text ?? "").join("\n\n\n") ?? "";
+  if (hasVisibleText(summary)) return summary;
+  const reasoning = part.reasoning ?? "";
+  return hasVisibleText(reasoning) ? reasoning : "";
 };
 
 export const createLangChainStreamingTimingAccessors = <
@@ -299,16 +301,17 @@ export const createLangChainStreamingTimingAccessors = <
     if (!Array.isArray(content)) return 0;
     let len = 0;
     for (const part of content as readonly LangChainContentBlock[]) {
+      if (typeof part !== "object" || part === null) continue;
       switch (part.type) {
         case "text":
         case "text_delta":
           if (typeof part.text === "string") len += part.text.length;
           break;
         case "thinking":
-          if (typeof part.thinking === "string") len += part.thinking.length;
+          if (hasVisibleText(part.thinking)) len += part.thinking.length;
           break;
         case "reasoning":
-          len += reasoningTextLength(part);
+          len += getReasoningText(part).length;
           break;
       }
     }

package/src/streamingTiming.test.tsx

@@ -34,4 +34,54 @@ describe("useLangChainStreamingTiming", () => {
       Math.ceil("deduced".length / 4),
     );
   });
+
+  it("counts a thinking block only when the converter renders it", () => {
+    const tokenCount = (thinking: string) => {
+      const messages: LangChainBaseMessage[] = [
+        {
+          id: "msg-1",
+          _getType: () => "ai",
+          content: [{ type: "thinking", thinking }],
+        },
+      ];
+      const { result, rerender } = renderHook(
+        ({ msgs, running }) => useLangChainStreamingTiming(msgs, running),
+        { initialProps: { msgs: messages, running: true } },
+      );
+      act(() => {
+        rerender({ msgs: messages, running: false });
+      });
+      return result.current["msg-1"]?.tokenCount;
+    };
+
+    expect(tokenCount("deduced")).toBe(Math.ceil("deduced".length / 4));
+    expect(tokenCount("   ")).toBeUndefined();
+  });
+
+  it("counts the summary text the converter renders, not the reasoning it shadows", () => {
+    const messages: LangChainBaseMessage[] = [
+      {
+        id: "msg-1",
+        _getType: () => "ai",
+        content: [
+          {
+            type: "reasoning",
+            reasoning: "partial thinking",
+            summary: [{ type: "summary_text", text: "first summary" }],
+          },
+        ],
+      },
+    ];
+    const { result, rerender } = renderHook(
+      ({ running }) => useLangChainStreamingTiming(messages, running),
+      { initialProps: { running: true } },
+    );
+    act(() => {
+      rerender({ running: false });
+    });
+
+    expect(result.current["msg-1"]?.tokenCount).toBe(
+      Math.ceil("first summary".length / 4),
+    );
+  });
 });

package/src/useStreamRuntime.test.tsx

@@ -10,7 +10,7 @@ import type {
 } from "@assistant-ui/core";
 import { useAui } from "@assistant-ui/store";
 import type { LangChainBaseMessage } from "./types";
-import type { ReactNode } from "react";
+import { startTransition, Suspense, type ReactNode } from "react";
 import {
   useLangChainRespond,
   useLangChainRespondAll,
@@ -898,3 +898,76 @@ describe("useStreamRuntime staged messages", () => {
     });
   });
 });
+
+describe("useStreamRuntime committed refs", () => {
+  it("submits through the committed stream after an abandoned render", async () => {
+    const streamA = createMockStream();
+    const streamB = createMockStream();
+    const adapter = makeThreadListAdapter();
+    mockUseStream.mockImplementation((options: { apiUrl: string }) =>
+      options.apiUrl === "/api/b" ? streamB : streamA,
+    );
+    const host = renderHook(() =>
+      useStreamRuntime({
+        apiUrl: "/api/a",
+        unstable_threadListAdapter: adapter,
+      } as never),
+    );
+
+    const pending = new Promise<never>(() => {});
+    let blocked = false;
+    const interruptedRender = vi.fn();
+    const Blocker = () => {
+      if (blocked) {
+        interruptedRender();
+        throw pending;
+      }
+      return null;
+    };
+
+    const capture: { runtime: AssistantRuntime | null } = { runtime: null };
+    const Nested = ({ apiUrl }: { apiUrl: string }) => {
+      capture.runtime = useStreamRuntime({
+        apiUrl,
+        unstable_threadListAdapter: adapter,
+      } as never);
+      return null;
+    };
+    const Tree = ({ apiUrl }: { apiUrl: string }) => (
+      <AssistantRuntimeProvider runtime={host.result.current}>
+        <Suspense fallback={null}>
+          <Nested apiUrl={apiUrl} />
+          <Blocker />
+        </Suspense>
+      </AssistantRuntimeProvider>
+    );
+
+    const view = render(<Tree apiUrl="/api/a" />);
+    expect(capture.runtime).not.toBeNull();
+
+    act(() => {
+      blocked = true;
+      startTransition(() => view.rerender(<Tree apiUrl="/api/b" />));
+    });
+    expect(interruptedRender).toHaveBeenCalled();
+
+    await act(async () => {
+      await capture.runtime!.thread.append("hello");
+    });
+
+    expect(streamA.submit).toHaveBeenCalledOnce();
+    expect(streamB.submit).not.toHaveBeenCalled();
+
+    await act(async () => {
+      blocked = false;
+      view.rerender(<Tree apiUrl="/api/b" />);
+    });
+    await act(async () => {
+      await capture.runtime!.thread.append("second");
+    });
+
+    expect(streamB.submit).toHaveBeenCalledOnce();
+    view.unmount();
+    host.unmount();
+  });
+});

package/src/useStreamRuntime.ts

@@ -1,7 +1,14 @@
 /// <reference types="@assistant-ui/core/store" />
 "use client";
 
-import { useCallback, useEffect, useMemo, useRef, useState } from "react";
+import {
+  useCallback,
+  useEffect,
+  useInsertionEffect,
+  useMemo,
+  useRef,
+  useState,
+} from "react";
 import type { AppendMessage, ToolExecutionStatus } from "@assistant-ui/core";
 import {
   generateId,
@@ -192,7 +199,9 @@ const useStreamThreadRuntime = (
   });
 
   const streamRef = useRef(stream);
-  streamRef.current = stream;
+  useInsertionEffect(() => {
+    streamRef.current = stream;
+  }, [stream]);
 
   const activeRunConfigRef = useRef<
     NormalizedRunConfigOptions["config"] | undefined
@@ -256,10 +265,14 @@ const useStreamThreadRuntime = (
   }, [stream.messages]);
 
   const visibleMessagesRef = useRef(visibleMessages);
-  visibleMessagesRef.current = visibleMessages;
+  useInsertionEffect(() => {
+    visibleMessagesRef.current = visibleMessages;
+  }, [visibleMessages]);
 
   const threadMessagesRef = useRef(threadMessages);
-  threadMessagesRef.current = threadMessages;
+  useInsertionEffect(() => {
+    threadMessagesRef.current = threadMessages;
+  }, [threadMessages]);
 
   const stagedMessagesRef = useRef(
     new Map<
@@ -311,6 +324,9 @@ const useStreamThreadRuntime = (
     if (remainingStagedMessages.length === 0) {
       stagedBaseMessagesRef.current = null;
       visibleMessagesRef.current = baseMessages;
+      // Reconciling against the upstream stream mutates the staged refs above,
+      // which cannot happen during render.
+      // eslint-disable-next-line react-hooks/set-state-in-effect
       setStagedMessages(null);
       return;
     }
@@ -613,9 +629,6 @@ export const useStreamRuntime = (rawOptions: UseStreamRuntimeOptions) => {
     ...options
   } = rawOptions;
 
-  const optionsRef = useRef(options);
-  optionsRef.current = options;
-
   const aui = useAui();
   const cloudAdapter = useCloudThreadListAdapter({
     cloud,
@@ -629,7 +642,7 @@ export const useStreamRuntime = (rawOptions: UseStreamRuntimeOptions) => {
 
   return useRemoteThreadListRuntime({
     runtimeHook: function RuntimeHook() {
-      return useStreamThreadRuntime(optionsRef.current);
+      return useStreamThreadRuntime(options);
     },
     adapter,
     allowNesting: true,