@assistant-ui/react-langchain 0.0.27 → 0.0.28

package/dist/converter.js

@@ -0,0 +1,213 @@
+import { parseDataUrl, resolveFilePartSource } from "@assistant-ui/core/internal";
+//#region src/converter.ts
+const convertLangChainContentBlock = (part) => {
+	switch (part.type) {
+		case "text":
+		case "text_delta": return {
+			type: "text",
+			text: part.text
+		};
+		case "image_url": {
+			const image = typeof part.image_url === "string" ? part.image_url : part.image_url?.url;
+			if (!image) return null;
+			return {
+				type: "image",
+				image
+			};
+		}
+		case "file": return {
+			type: "file",
+			filename: part.metadata?.filename ?? "file",
+			data: part.source_type === "url" ? part.url : part.source_type === "id" ? part.id : part.data,
+			mimeType: part.mime_type ?? "application/octet-stream",
+			...(part.source_type === "url" || part.source_type === "id") && { sourceType: part.source_type }
+		};
+		case "audio": {
+			const mimeType = part.mime_type ?? "application/octet-stream";
+			const subtype = mimeType.startsWith("audio/") ? mimeType.slice(6) : void 0;
+			return {
+				type: "file",
+				filename: subtype ? `audio.${subtype}` : "audio",
+				data: part.data,
+				mimeType
+			};
+		}
+		case "thinking": return {
+			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 ?? ""
+		};
+		case "tool_use":
+		case "input_json_delta": return null;
+		default: return;
+	}
+};
+const hasVisibleText = (text) => typeof text === "string" && text.trim() !== "";
+/**
+* Audio output arrives outside the content array: providers leave `content`
+* empty and put the spoken text in `additional_kwargs.audio.transcript`. The
+* audio bytes stay behind because no provider reports their media type, and a
+* streamed response carries raw PCM rather than a playable file.
+*/
+const withAudioTranscript = (parts, additionalKwargs) => {
+	const transcript = (additionalKwargs?.audio)?.transcript;
+	if (typeof transcript !== "string" || !hasVisibleText(transcript)) return parts;
+	if (parts.some((part) => part.type === "text" && hasVisibleText(part.text))) return parts;
+	return [...parts.filter((part) => part.type !== "text"), {
+		type: "text",
+		text: transcript
+	}];
+};
+const getCustomMetadata = (additionalKwargs) => additionalKwargs?.metadata ?? {};
+const uiMessageToDataPart = (ui) => ({
+	type: "data",
+	name: ui.name,
+	data: ui.props
+});
+/**
+* Audio media types that reach a provider's audio input through the LangChain
+* `audio` block. langchain-core derives OpenAI's `input_audio.format` by
+* splitting `mime_type` on `/`, and that format is a wav-or-mp3 enum, so
+* `audio/mpeg` passes the converter and is rejected at the provider.
+*/
+const audioBlockMimeTypes = /* @__PURE__ */ new Map([
+	["audio/mp3", "audio/mp3"],
+	["audio/mpeg", "audio/mp3"],
+	["audio/wav", "audio/wav"],
+	["audio/wave", "audio/wav"],
+	["audio/x-wav", "audio/wav"]
+]);
+const getMessageContent = (msg) => {
+	const allContent = [...msg.content, ...msg.attachments?.flatMap((a) => a.content) ?? []];
+	const hasNonText = allContent.some((part) => part.type === "file" || part.type === "image" || part.type === "audio");
+	const hasText = allContent.some((part) => part.type === "text");
+	if (hasNonText && !hasText) allContent.unshift({
+		type: "text",
+		text: " "
+	});
+	const content = allContent.flatMap((part) => {
+		const type = part.type;
+		switch (type) {
+			case "text": return {
+				type: "text",
+				text: part.text
+			};
+			case "image": return {
+				type: "image_url",
+				image_url: { url: part.image }
+			};
+			case "file": {
+				const metadata = { filename: part.filename ?? "file" };
+				if (part.sourceType === "id") return {
+					type: "file",
+					id: part.data,
+					mime_type: part.mimeType,
+					filename: metadata.filename,
+					metadata,
+					source_type: "id"
+				};
+				const source = resolveFilePartSource(part);
+				if (source.kind === "url") return {
+					type: "file",
+					url: source.url,
+					mime_type: part.mimeType,
+					filename: metadata.filename,
+					metadata,
+					source_type: "url"
+				};
+				const audioMimeType = audioBlockMimeTypes.get(source.mimeType.toLowerCase());
+				if (audioMimeType) return {
+					type: "audio",
+					data: source.data,
+					mime_type: audioMimeType,
+					source_type: "base64"
+				};
+				return {
+					type: "file",
+					data: source.data,
+					mime_type: source.mimeType,
+					filename: metadata.filename,
+					metadata,
+					source_type: "base64"
+				};
+			}
+			case "audio": return {
+				type: "audio",
+				data: parseDataUrl(part.audio.data)?.data ?? part.audio.data,
+				mime_type: `audio/${part.audio.format}`,
+				source_type: "base64"
+			};
+			case "data": return [];
+			case "tool-call": throw new Error("Tool call appends are not supported.");
+			default: throw new Error(`Unsupported append message part type: ${type}`);
+		}
+	});
+	if (content.length === 1 && content[0]?.type === "text") return content[0].text ?? "";
+	return content;
+};
+const 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 createLangChainStreamingTimingAccessors = (getType) => {
+	const findAiMessage = (messages, messageId) => messages.find((message) => getType(message) === "ai" && message.id === messageId);
+	const getTextLength = (messages, messageId) => {
+		const message = findAiMessage(messages, messageId);
+		if (!message) return 0;
+		const content = message.content;
+		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);
+		}
+		return len;
+	};
+	const getToolCallCount = (messages, messageId) => findAiMessage(messages, messageId)?.tool_calls?.length ?? 0;
+	const getAssistantMessageId = (messages) => {
+		for (let i = messages.length - 1; i >= 0; i--) {
+			const message = messages[i];
+			if (message && getType(message) === "ai" && message.id) return message.id;
+		}
+	};
+	return {
+		getAssistantMessageId,
+		getTextLength,
+		getToolCallCount
+	};
+};
+/**
+* Resolve the assistant message a `UIMessage` belongs to: the parent id comes
+* from `metadata.message_id` (Python SDK) or `metadata.id` (JS SDK).
+*/
+const getUIMessageParentId = (ui) => ui.metadata?.message_id ?? ui.metadata?.id;
+/**
+* Group the graph's accumulated `UIMessage`s by the assistant message they
+* belong to. Non-array state and entries without a parent link are dropped.
+*/
+const groupUIMessagesByParent = (value) => {
+	const map = /* @__PURE__ */ new Map();
+	if (!Array.isArray(value)) return map;
+	for (const ui of value) {
+		const parentId = getUIMessageParentId(ui);
+		if (!parentId) continue;
+		const existing = map.get(parentId);
+		if (existing) existing.push(ui);
+		else map.set(parentId, [ui]);
+	}
+	return map;
+};
+//#endregion
+export { convertLangChainContentBlock, createLangChainStreamingTimingAccessors, getCustomMetadata, getMessageContent, getUIMessageParentId, groupUIMessagesByParent, uiMessageToDataPart, withAudioTranscript };
+
+//# sourceMappingURL=converter.js.map

package/dist/converter.js.map

@@ -0,0 +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"}

package/dist/index.d.ts

@@ -1,4 +1,5 @@
-import { LangChainBaseMessage, LangChainContentBlock, LangChainToolCall, RemoveUIMessage, UIMessage, UseStreamRuntimeOptions } from "./types.js";
+import { LangChainContentBlock } from "./converter.js";
+import { LangChainBaseMessage, LangChainToolCall, RemoveUIMessage, UIMessage, UseStreamRuntimeOptions } from "./types.js";
 import { convertLangChainBaseMessage } from "./convertMessages.js";
 import { useLangChainError, useLangChainInterruptState, useLangChainInterrupts, useLangChainRespond, useLangChainRespondAll, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainStream, useLangChainSubagents, useLangChainSubgraphs, useLangChainSubmit, useLangChainToolCalls } from "./hooks.js";
 import { useStreamRuntime } from "./useStreamRuntime.js";

package/dist/streamingTiming.d.ts

@@ -1,8 +1,7 @@
 import { LangChainBaseMessage } from "./types.js";
-import { StreamingTimingAccessors } from "@assistant-ui/core/react";
 import { MessageTiming } from "@assistant-ui/core";
 //#region src/streamingTiming.d.ts
-declare const langChainStreamingTimingAccessors: StreamingTimingAccessors<LangChainBaseMessage>;
+declare const langChainStreamingTimingAccessors: import("@assistant-ui/core").StreamingTimingAccessors<LangChainBaseMessage>;
 /**
  * Tracks per-message streaming timing for LangChain messages. Delegates to
  * the shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,

package/dist/streamingTiming.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"streamingTiming.d.ts","names":[],"sources":["../src/streamingTiming.ts"],"mappings":";;;;cAmEa,mCAAmC,yBAAyB;;;;;;;cAa5D,8BACX,mBAAmB,wBACnB,uBACC,eAAe"}
+{"version":3,"file":"streamingTiming.d.ts","names":[],"sources":["../src/streamingTiming.ts"],"mappings":";;;cAQa,gEAAiC,yBAAA;;;;;;;cASjC,8BACX,mBAAmB,wBACnB,uBACC,eAAe"}

package/dist/streamingTiming.js

@@ -1,43 +1,9 @@
 "use client";
+import { createLangChainStreamingTimingAccessors } from "./converter.js";
 import { getMessageType } from "./convertMessages.js";
 import { useStreamingTiming } from "@assistant-ui/core/react";
 //#region src/streamingTiming.ts
-const findAiMessage = (messages, messageId) => messages.find((m) => getMessageType(m) === "ai" && m.id === messageId);
-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 getTextLength = (messages, messageId) => {
-	const m = findAiMessage(messages, messageId);
-	if (!m) return 0;
-	const content = m.content;
-	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);
-	}
-	return len;
-};
-const getToolCallCount = (messages, messageId) => findAiMessage(messages, messageId)?.tool_calls?.length ?? 0;
-const getAssistantMessageId = (messages) => {
-	for (let i = messages.length - 1; i >= 0; i--) {
-		const m = messages[i];
-		if (m && getMessageType(m) === "ai" && m.id) return m.id;
-	}
-};
-const langChainStreamingTimingAccessors = {
-	getAssistantMessageId,
-	getTextLength,
-	getToolCallCount
-};
+const langChainStreamingTimingAccessors = createLangChainStreamingTimingAccessors(getMessageType);
 /**
 * Tracks per-message streaming timing for LangChain messages. Delegates to
 * the shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,

package/dist/streamingTiming.js.map

@@ -1 +1 @@
-{"version":3,"file":"streamingTiming.js","names":[],"sources":["../src/streamingTiming.ts"],"sourcesContent":["\"use client\";\n\nimport type { MessageTiming } from \"@assistant-ui/core\";\nimport {\n  useStreamingTiming,\n  type StreamingTimingAccessors,\n} from \"@assistant-ui/core/react\";\nimport type { LangChainBaseMessage, LangChainContentBlock } from \"./types\";\nimport { getMessageType } from \"./convertMessages\";\n\nconst findAiMessage = (\n  messages: readonly LangChainBaseMessage[],\n  messageId: string,\n): LangChainBaseMessage | undefined =>\n  messages.find((m) => getMessageType(m) === \"ai\" && m.id === messageId);\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\nconst getTextLength = (\n  messages: readonly LangChainBaseMessage[],\n  messageId: string,\n): number => {\n  const m = findAiMessage(messages, messageId);\n  if (!m) return 0;\n  const content = m.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\nconst getToolCallCount = (\n  messages: readonly LangChainBaseMessage[],\n  messageId: string,\n): number => findAiMessage(messages, messageId)?.tool_calls?.length ?? 0;\n\nconst getAssistantMessageId = (\n  messages: readonly LangChainBaseMessage[],\n): string | undefined => {\n  for (let i = messages.length - 1; i >= 0; i--) {\n    const m = messages[i];\n    if (m && getMessageType(m) === \"ai\" && m.id) return m.id;\n  }\n  return undefined;\n};\n\nexport const langChainStreamingTimingAccessors: StreamingTimingAccessors<LangChainBaseMessage> =\n  {\n    getAssistantMessageId,\n    getTextLength,\n    getToolCallCount,\n  };\n\n/**\n * Tracks per-message streaming timing for LangChain messages. Delegates to\n * the shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,\n * adapted to the `LangChainBaseMessage` shape (`_getType() -> \"ai\"`, content\n * blocks including text/thinking/reasoning, `tool_calls`).\n */\nexport const useLangChainStreamingTiming = (\n  messages: readonly LangChainBaseMessage[],\n  isRunning: boolean,\n): Record<string, MessageTiming> =>\n  useStreamingTiming(messages, isRunning, langChainStreamingTimingAccessors);\n"],"mappings":";;;;AAUA,MAAM,iBACJ,UACA,cAEA,SAAS,MAAM,MAAM,eAAe,CAAC,MAAM,QAAQ,EAAE,OAAO,SAAS;AAEvE,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,MAAM,iBACJ,UACA,cACW;CACX,MAAM,IAAI,cAAc,UAAU,SAAS;CAC3C,IAAI,CAAC,GAAG,OAAO;CACf,MAAM,UAAU,EAAE;CAClB,IAAI,OAAO,YAAY,UAAU,OAAO,QAAQ;CAChD,IAAI,CAAC,MAAM,QAAQ,OAAO,GAAG,OAAO;CACpC,IAAI,MAAM;CACV,KAAK,MAAM,QAAQ,SACjB,QAAQ,KAAK,MAAb;EACE,KAAK;EACL,KAAK;GACH,IAAI,OAAO,KAAK,SAAS,UAAU,OAAO,KAAK,KAAK;GACpD;EACF,KAAK;GACH,IAAI,OAAO,KAAK,aAAa,UAAU,OAAO,KAAK,SAAS;GAC5D;EACF,KAAK,aACH,OAAO,oBAAoB,IAAI;CAEnC;CAEF,OAAO;AACT;AAEA,MAAM,oBACJ,UACA,cACW,cAAc,UAAU,SAAS,CAAC,EAAE,YAAY,UAAU;AAEvE,MAAM,yBACJ,aACuB;CACvB,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;EAC7C,MAAM,IAAI,SAAS;EACnB,IAAI,KAAK,eAAe,CAAC,MAAM,QAAQ,EAAE,IAAI,OAAO,EAAE;CACxD;AAEF;AAEA,MAAa,oCACX;CACE;CACA;CACA;AACF;;;;;;;AAQF,MAAa,+BACX,UACA,cAEA,mBAAmB,UAAU,WAAW,iCAAiC"}
+{"version":3,"file":"streamingTiming.js","names":[],"sources":["../src/streamingTiming.ts"],"sourcesContent":["\"use client\";\n\nimport type { MessageTiming } from \"@assistant-ui/core\";\nimport { useStreamingTiming } from \"@assistant-ui/core/react\";\nimport { createLangChainStreamingTimingAccessors } from \"./converter\";\nimport { getMessageType } from \"./convertMessages\";\nimport type { LangChainBaseMessage } from \"./types\";\n\nexport const langChainStreamingTimingAccessors =\n  createLangChainStreamingTimingAccessors<LangChainBaseMessage>(getMessageType);\n\n/**\n * Tracks per-message streaming timing for LangChain messages. Delegates to\n * the shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,\n * adapted to the `LangChainBaseMessage` shape (`_getType() -> \"ai\"`, content\n * blocks including text/thinking/reasoning, `tool_calls`).\n */\nexport const useLangChainStreamingTiming = (\n  messages: readonly LangChainBaseMessage[],\n  isRunning: boolean,\n): Record<string, MessageTiming> =>\n  useStreamingTiming(messages, isRunning, langChainStreamingTimingAccessors);\n"],"mappings":";;;;;AAQA,MAAa,oCACX,wCAA8D,cAAc;;;;;;;AAQ9E,MAAa,+BACX,UACA,cAEA,mBAAmB,UAAU,WAAW,iCAAiC"}

package/dist/types.d.ts

@@ -1,61 +1,8 @@
+import { LangChainContentBlock } from "./converter.js";
 import { AttachmentAdapter, DictationAdapter, ExternalStoreSharedOptions, FeedbackAdapter, MessageStatus, RealtimeVoiceAdapter, RemoteThreadListAdapter, SpeechSynthesisAdapter } from "@assistant-ui/core";
 import { AnyStream, AssembledToolCall, SubagentDiscoverySnapshot, SubgraphDiscoverySnapshot, UseStreamOptions } from "@langchain/react";
 import { AssistantCloud } from "assistant-cloud";
 //#region src/types.d.ts
-/** Known content block types from @langchain/core messages. */
-type LangChainContentBlock = {
-  type: "text";
-  text: string;
-} | {
-  type: "text_delta";
-  text: string;
-} | {
-  type: "image_url";
-  image_url: string | {
-    url?: string;
-  };
-} | {
-  type: "thinking";
-  thinking: string;
-} | {
-  type: "reasoning";
-  summary?: Array<{
-    type: "summary_text";
-    text?: string;
-  }>;
-  reasoning?: string;
-} | {
-  type: "file";
-  data: string;
-  mime_type: string;
-  source_type?: "base64";
-  metadata?: {
-    filename?: string;
-  };
-} | {
-  type: "file";
-  url: string;
-  mime_type?: string;
-  source_type: "url";
-  metadata?: {
-    filename?: string;
-  };
-} | {
-  type: "file";
-  id: string;
-  mime_type?: string;
-  source_type: "id";
-  metadata?: {
-    filename?: string;
-  };
-} | {
-  type: "audio";
-  data: string;
-  mime_type: string;
-  source_type: "base64";
-} | {
-  type: "tool_use" | "input_json_delta";
-};
 type LangChainToolCall = {
   id: string;
   name: string;
@@ -183,5 +130,5 @@ type LangChainRuntimeExtraOptions = ExternalStoreSharedOptions & {
 };
 type UseStreamRuntimeOptions = UseStreamOptions extends (infer O) ? O extends UseStreamOptions ? O & LangChainRuntimeExtraOptions : never : never;
 //#endregion
-export { LangChainBaseMessage, LangChainContentBlock, LangChainRuntimeExtraOptions, LangChainRuntimeExtras, LangChainToolCall, RemoveUIMessage, UIMessage, UseStreamRuntimeOptions };
+export { LangChainBaseMessage, type LangChainContentBlock, LangChainRuntimeExtraOptions, LangChainRuntimeExtras, LangChainToolCall, RemoveUIMessage, UIMessage, UseStreamRuntimeOptions };
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;;;KAoBY;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;;KAEM;EACV;EACA;EACA,MAAM;;;;;;;;;KAUI,UACV,+BACA,eAAe,0BAA0B;EAEzC;EACA;EACA,MAAM;EACN,OAAO;EACP;IACE;IACA;IACA;IACA;IACA;IACA;KACC;;;;;;;KAQO;EACV;EACA;;;;;;;;;KAUU;EACV;EACA;EACA;EACA;EACA,oBAAoB;;EAEpB,sBAAsB;;EAEtB;;EAEA,SAAS;;EAET;;KAGU;EACV;IAAa;;EACb;IAAuB;IAAa;;EACpC,oBAAoB;EACpB,WAAW,oBAAoB;EAC/B,WAAW,oBAAoB;EAC/B,QAAQ;EACR;EACA,SACE,QAAQ,4CACR,UAAU,4BACP;EACL,UACE,mBACA,UAAU,4BACP;EACL,aACE,eAAe,yBACf,UAAU,4BACP;EACL,QAAQ;EACR;;KAGU,+BAA+B;;;;;;;;;EASzC,qBAAqB;EACrB,QAAQ;EACR;IAEM,cAAc;IACd,SAAS;IACT,YAAY;IACZ,QAAQ;IACR,WAAW;;;;;;;;EASjB;;;;;EAKA;;;;;EAKA,6BAA6B;;EAE7B,gBAAgB;IAAU;;;EAE1B,WAAW,qBAAqB;;;;;;;EAOhC;;KAOU,0BAA0B,gCAA+B,KACjE,UAAU,mBACR,IAAI"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;;;KAqBY;EACV;EACA;EACA,MAAM;;;;;;;;;KAUI,UACV,+BACA,eAAe,0BAA0B;EAEzC;EACA;EACA,MAAM;EACN,OAAO;EACP;IACE;IACA;IACA;IACA;IACA;IACA;KACC;;;;;;;KAQO;EACV;EACA;;;;;;;;;KAUU;EACV;EACA;EACA;EACA;EACA,oBAAoB;;EAEpB,sBAAsB;;EAEtB;;EAEA,SAAS;;EAET;;KAGU;EACV;IAAa;;EACb;IAAuB;IAAa;;EACpC,oBAAoB;EACpB,WAAW,oBAAoB;EAC/B,WAAW,oBAAoB;EAC/B,QAAQ;EACR;EACA,SACE,QAAQ,4CACR,UAAU,4BACP;EACL,UACE,mBACA,UAAU,4BACP;EACL,aACE,eAAe,yBACf,UAAU,4BACP;EACL,QAAQ;EACR;;KAGU,+BAA+B;;;;;;;;;EASzC,qBAAqB;EACrB,QAAQ;EACR;IAEM,cAAc;IACd,SAAS;IACT,YAAY;IACZ,QAAQ;IACR,WAAW;;;;;;;;EASjB;;;;;EAKA;;;;;EAKA,6BAA6B;;EAE7B,gBAAgB;IAAU;;;EAE1B,WAAW,qBAAqB;;;;;;;EAOhC;;KAOU,0BAA0B,gCAA+B,KACjE,UAAU,mBACR,IAAI"}

package/dist/useStreamRuntime.d.ts

@@ -1,5 +1,6 @@
 /// <reference types="@assistant-ui/core/store" />
-import { UIMessage, UseStreamRuntimeOptions } from "./types.js";
+import { groupUIMessagesByParent } from "./converter.js";
+import { UseStreamRuntimeOptions } from "./types.js";
 import { AppendMessage } from "@assistant-ui/core";
 //#region src/useStreamRuntime.d.ts
 declare const runConfigToSubmitOptions: (runConfig: AppendMessage["runConfig"]) => {
@@ -7,13 +8,6 @@ declare const runConfigToSubmitOptions: (runConfig: AppendMessage["runConfig"])
     configurable: Record<string, unknown>;
   };
 } | undefined;
-/**
- * Group the graph's accumulated `UIMessage`s by the assistant message they
- * belong to. Non-array state and entries without a parent link are dropped.
- * The parent id comes from `metadata.message_id` (Python SDK) or
- * `metadata.id` (JS SDK).
- */
-declare const groupUIMessagesByParent: (value: unknown) => Map<string, UIMessage[]>;
 /**
  * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.
  * Accepts the same options as `useStream` from `@langchain/react`, plus

package/dist/useStreamRuntime.d.ts.map

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

package/dist/useStreamRuntime.js

@@ -1,9 +1,11 @@
 "use client";
-import { convertLangChainBaseMessage, getMessageContent, getMessageType } from "./convertMessages.js";
+import { getMessageContent, groupUIMessagesByParent } from "./converter.js";
+import { convertLangChainBaseMessage, getMessageType } from "./convertMessages.js";
 import { langChainExtras } from "./runtimeExtras.js";
 import { foldUIUpdates, mergeUIMessages } from "./uiMessages.js";
 import { resolveForkCheckpoint } from "./resolveForkCheckpoint.js";
 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";
@@ -12,33 +14,11 @@ import { STREAM_CONTROLLER, useChannel, useStream } from "@langchain/react";
 //#region src/useStreamRuntime.ts
 const UI_CUSTOM_CHANNELS = ["custom"];
 const runConfigToSubmitOptions = (runConfig) => runConfig?.custom ? { config: { configurable: runConfig.custom } } : void 0;
-/**
-* Group the graph's accumulated `UIMessage`s by the assistant message they
-* belong to. Non-array state and entries without a parent link are dropped.
-* The parent id comes from `metadata.message_id` (Python SDK) or
-* `metadata.id` (JS SDK).
-*/
-const groupUIMessagesByParent = (value) => {
-	const map = /* @__PURE__ */ new Map();
-	if (!Array.isArray(value)) return map;
-	for (const ui of value) {
-		const parentId = ui.metadata?.message_id ?? ui.metadata?.id;
-		if (!parentId) continue;
-		const existing = map.get(parentId);
-		if (existing) existing.push(ui);
-		else map.set(parentId, [ui]);
-	}
-	return map;
-};
-const getPendingToolCalls = (messages) => {
-	const pending = /* @__PURE__ */ new Map();
-	for (const m of messages) {
-		const type = getMessageType(m);
-		if (type === "ai") for (const tc of m.tool_calls ?? []) pending.set(tc.id, tc);
-		else if (type === "tool" && m.tool_call_id) pending.delete(m.tool_call_id);
-	}
-	return [...pending.values()];
-};
+const getPendingToolCalls = (messages) => scanPendingToolCalls(messages, (message) => {
+	const type = getMessageType(message);
+	if (type === "ai") return { toolCalls: message.tool_calls ?? [] };
+	if (type === "tool" && message.tool_call_id) return { toolCallId: message.tool_call_id };
+}, (toolCall) => toolCall.id);
 const toStagedHumanMessage = (msg, id = generateId()) => ({
 	id,
 	_getType: () => "human",
@@ -218,7 +198,7 @@ const useStreamThreadRuntime = (options) => {
 	]);
 	return useExternalStoreRuntime({
 		...pickExternalStoreSharedOptions(options),
-		isRunning: effectiveIsRunning,
+		isRunning: stream.isLoading,
 		isLoading: stream.isThreadLoading,
 		messages: threadMessages,
 		adapters,
@@ -233,13 +213,7 @@ const useStreamThreadRuntime = (options) => {
 			const stagedMessageId = stageUserMessage(msg, true).id;
 			setActiveRunConfig(msg.runConfig);
 			const content = getMessageContent(msg);
-			const cancellations = autoCancelPendingToolCalls !== false ? getPendingToolCalls(streamRef.current.messages).map((t) => ({
-				type: "tool",
-				name: t.name,
-				tool_call_id: t.id,
-				content: JSON.stringify({ cancelled: true }),
-				status: "error"
-			})) : [];
+			const cancellations = autoCancelPendingToolCalls !== false ? getPendingToolCalls(streamRef.current.messages).map(createToolCallCancellationStub) : [];
 			try {
 				const { externalId } = await aui.threadListItem.initialize();
 				await streamRef.current.submit({ [messagesKey]: [...cancellations, {
@@ -366,9 +340,10 @@ 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,
-		create,
+		create: createCloudThreadListAdapterCreateFallback(create, aui.threadListItem),
 		delete: deleteFn
 	});
 	return useRemoteThreadListRuntime({