@assistant-ui/react-langchain 0.0.19 → 0.0.21

package/dist/convertMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessages.d.ts","names":[],"sources":["../src/convertMessages.ts"],"mappings":";;;;KAeK,oCACH,4BAA4B;EAC1B,qBAAqB,YAAY;EACjC,gBAAgB,eAAe;;cAStB,iBAAkB,SAAS;cAyE3B,8BACX,SAAS,sBACT,WAAU,sCACT,4BAA4B;cAmGlB,oBAAqB,KAAK"}
+{"version":3,"file":"convertMessages.d.ts","names":[],"sources":["../src/convertMessages.ts"],"mappings":";;;;KAgBK,oCACH,4BAA4B;EAC1B,qBAAqB,YAAY;EACjC,gBAAgB,eAAe;;cAStB,iBAAkB,SAAS;cAyE3B,8BACX,SAAS,sBACT,WAAU,sCACT,4BAA4B;cAyFlB,oBAAqB,KAAK"}

package/dist/convertMessages.js

@@ -1,4 +1,5 @@
 "use client";
+import { httpUrlPattern, parseDataUrl } from "@assistant-ui/core/internal";
 //#region src/convertMessages.ts
 const uiMessageToDataPart = (ui) => ({
 	type: "data",
@@ -116,15 +117,6 @@ const convertLangChainBaseMessage = (message, metadata = {}) => {
 		};
 	}
 };
-const parseDataUrl = (value) => {
-	const match = value.match(/^data:([^;,]+)(?:;[^;,]+)*;base64,(.+)$/);
-	if (!match) return null;
-	return {
-		mimeType: match[1],
-		data: match[2]
-	};
-};
-const httpUrlPattern = /^https?:\/\//i;
 const getMessageContent = (msg) => {
 	const allContent = [...msg.content, ...msg.attachments?.flatMap((a) => a.content) ?? []];
 	const hasNonText = allContent.some((part) => part.type === "file" || part.type === "image");

package/dist/convertMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"convertMessages.js","names":["parts","_exhaustiveCheck"],"sources":["../src/convertMessages.ts"],"sourcesContent":["\"use client\";\n\nimport type { useExternalMessageConverter } from \"@assistant-ui/core/react\";\nimport type {\n  AppendMessage,\n  DataMessagePart,\n  MessageTiming,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport type {\n  LangChainBaseMessage,\n  LangChainContentBlock,\n  UIMessage,\n} from \"./types\";\n\ntype LangChainMessageConverterMetadata =\n  useExternalMessageConverter.Metadata & {\n    uiMessagesByParent?: Map<string, UIMessage[]>;\n    messageTiming?: Record<string, MessageTiming>;\n  };\n\nconst uiMessageToDataPart = (ui: UIMessage): DataMessagePart => ({\n  type: \"data\",\n  name: ui.name,\n  data: ui.props,\n});\n\nexport const getMessageType = (message: LangChainBaseMessage): string => {\n  if (typeof message._getType === \"function\") return message._getType();\n  if (\"type\" in message)\n    return (message as Record<string, unknown>).type as string;\n  throw new Error(\"Cannot determine message type\");\n};\n\nconst contentToParts = (content: unknown) => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n\n  const parts = content as readonly LangChainContentBlock[];\n  return parts\n    .map((part) => {\n      const type = part.type;\n      switch (type) {\n        case \"text\":\n        case \"text_delta\":\n          return { type: \"text\" as const, text: part.text };\n        case \"image_url\": {\n          const image =\n            typeof part.image_url === \"string\"\n              ? part.image_url\n              : part.image_url?.url;\n          if (!image) return null;\n          return { type: \"image\" as const, image };\n        }\n        case \"file\":\n          return {\n            type: \"file\" as const,\n            filename: part.metadata?.filename ?? \"file\",\n            data:\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          };\n        case \"thinking\":\n          return { type: \"reasoning\" as const, text: part.thinking };\n        case \"reasoning\":\n          return {\n            type: \"reasoning\" as const,\n            text:\n              part.summary?.map((s) => s?.text ?? \"\").join(\"\\n\\n\\n\") ??\n              part.reasoning ??\n              \"\",\n          };\n        case \"tool_use\":\n        case \"input_json_delta\":\n          return null;\n        default:\n          return null;\n      }\n    })\n    .filter((p) => p !== null);\n};\n\nconst getCustomMetadata = (\n  additionalKwargs: Record<string, unknown> | undefined,\n): Record<string, unknown> =>\n  (additionalKwargs?.metadata as Record<string, unknown>) ?? {};\n\nconst getStringContent = (content: unknown): string => {\n  if (typeof content === \"string\") return content;\n  const parts = content as readonly LangChainContentBlock[];\n  return parts\n    .filter((c): c is { type: \"text\"; text: string } => c.type === \"text\")\n    .map((c) => c.text)\n    .join(\"\");\n};\n\nexport const convertLangChainBaseMessage = (\n  message: LangChainBaseMessage,\n  metadata: LangChainMessageConverterMetadata = {},\n): useExternalMessageConverter.Message => {\n  const type = getMessageType(message);\n\n  switch (type) {\n    case \"system\":\n      return {\n        role: \"system\",\n        id: message.id,\n        content: [{ type: \"text\", text: getStringContent(message.content) }],\n        metadata: {\n          custom: getCustomMetadata(message.additional_kwargs),\n        },\n      };\n\n    case \"human\":\n      return {\n        role: \"user\",\n        id: message.id,\n        content: contentToParts(message.content),\n        metadata: {\n          custom: getCustomMetadata(message.additional_kwargs),\n        },\n      };\n\n    case \"ai\": {\n      const toolCallParts =\n        message.tool_calls?.map((tc) => ({\n          type: \"tool-call\" as const,\n          toolCallId: tc.id,\n          toolName: tc.name,\n          args: tc.args as ReadonlyJSONObject,\n          argsText: JSON.stringify(tc.args),\n        })) ?? [];\n\n      const assistantStatus =\n        typeof message.status === \"object\" ? message.status : undefined;\n\n      const uiDataParts =\n        (message.id\n          ? metadata.uiMessagesByParent\n              ?.get(message.id)\n              ?.map(uiMessageToDataPart)\n          : undefined) ?? [];\n\n      const timing = metadata.messageTiming?.[message.id ?? \"\"];\n\n      return {\n        role: \"assistant\",\n        id: message.id,\n        content: [\n          ...contentToParts(message.content),\n          ...toolCallParts,\n          ...uiDataParts,\n        ],\n        metadata: {\n          custom: getCustomMetadata(message.additional_kwargs),\n          ...(timing && { timing }),\n        },\n        ...(assistantStatus && { status: assistantStatus }),\n      };\n    }\n\n    case \"tool\":\n      return {\n        role: \"tool\",\n        toolName: message.name ?? \"\",\n        toolCallId: message.tool_call_id ?? \"\",\n        result: message.content,\n        artifact: message.artifact,\n        isError: message.status === \"error\",\n      };\n\n    default:\n      return {\n        role: \"system\",\n        id: message.id,\n        content: [\n          {\n            type: \"text\",\n            text:\n              typeof message.content === \"string\"\n                ? message.content\n                : JSON.stringify(message.content),\n          },\n        ],\n      };\n  }\n};\n\nconst parseDataUrl = (\n  value: string,\n): { mimeType: string; data: string } | null => {\n  const match = value.match(/^data:([^;,]+)(?:;[^;,]+)*;base64,(.+)$/);\n  if (!match) return null;\n  return { mimeType: match[1]!, data: match[2]! };\n};\n\nconst httpUrlPattern = /^https?:\\/\\//i;\n\nexport const getMessageContent = (msg: AppendMessage) => {\n  const allContent = [\n    ...msg.content,\n    ...(msg.attachments?.flatMap((a) => a.content) ?? []),\n  ];\n\n  const hasNonText = allContent.some(\n    (part) => part.type === \"file\" || part.type === \"image\",\n  );\n  const hasText = allContent.some((part) => part.type === \"text\");\n  if (hasNonText && !hasText) {\n    allContent.unshift({ type: \"text\", text: \" \" });\n  }\n\n  const content = allContent.map((part) => {\n    const type = part.type;\n    switch (type) {\n      case \"text\":\n        return { type: \"text\" as const, text: part.text };\n      case \"image\":\n        return { type: \"image_url\" as const, image_url: { url: part.image } };\n      case \"file\": {\n        const metadata = { filename: part.filename ?? \"file\" };\n        if (httpUrlPattern.test(part.data)) {\n          return {\n            type: \"file\" as const,\n            url: part.data,\n            mime_type: part.mimeType,\n            metadata,\n            source_type: \"url\" as const,\n          };\n        }\n        const parsed = parseDataUrl(part.data);\n        return {\n          type: \"file\" as const,\n          data: parsed?.data ?? part.data,\n          mime_type: parsed?.mimeType ?? part.mimeType,\n          metadata,\n          source_type: \"base64\" as const,\n        };\n      }\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported.\");\n      default: {\n        const _exhaustiveCheck:\n          | \"reasoning\"\n          | \"source\"\n          | \"audio\"\n          | \"data\"\n          | \"generative-ui\" = type;\n        throw new Error(\n          `Unsupported append message part type: ${_exhaustiveCheck}`,\n        );\n      }\n    }\n  });\n\n  if (content.length === 1 && content[0]?.type === \"text\") {\n    return content[0].text ?? \"\";\n  }\n  return content;\n};\n"],"mappings":";;AAqBA,MAAM,uBAAuB,QAAoC;CAC/D,MAAM;CACN,MAAM,GAAG;CACT,MAAM,GAAG;AACX;AAEA,MAAa,kBAAkB,YAA0C;CACvE,IAAI,OAAO,QAAQ,aAAa,YAAY,OAAO,QAAQ,SAAS;CACpE,IAAI,UAAU,SACZ,OAAQ,QAAoC;CAC9C,MAAM,IAAI,MAAM,+BAA+B;AACjD;AAEA,MAAM,kBAAkB,YAAqB;CAC3C,IAAI,OAAO,YAAY,UACrB,OAAO,CAAC;EAAE,MAAM;EAAiB,MAAM;CAAQ,CAAC;CAGlD,OAAOA,QACJ,KAAK,SAAS;EAEb,QADa,KAAK,MAClB;GACE,KAAK;GACL,KAAK,cACH,OAAO;IAAE,MAAM;IAAiB,MAAM,KAAK;GAAK;GAClD,KAAK,aAAa;IAChB,MAAM,QACJ,OAAO,KAAK,cAAc,WACtB,KAAK,YACL,KAAK,WAAW;IACtB,IAAI,CAAC,OAAO,OAAO;IACnB,OAAO;KAAE,MAAM;KAAkB;IAAM;GACzC;GACA,KAAK,QACH,OAAO;IACL,MAAM;IACN,UAAU,KAAK,UAAU,YAAY;IACrC,MACE,KAAK,gBAAgB,QACjB,KAAK,MACL,KAAK,gBAAgB,OACnB,KAAK,KACL,KAAK;IACb,UAAU,KAAK,aAAa;GAC9B;GACF,KAAK,YACH,OAAO;IAAE,MAAM;IAAsB,MAAM,KAAK;GAAS;GAC3D,KAAK,aACH,OAAO;IACL,MAAM;IACN,MACE,KAAK,SAAS,KAAK,MAAM,GAAG,QAAQ,EAAE,CAAC,CAAC,KAAK,QAAQ,KACrD,KAAK,aACL;GACJ;GACF,KAAK;GACL,KAAK,oBACH,OAAO;GACT,SACE,OAAO;EACX;CACF,CAAC,CAAC,CACD,QAAQ,MAAM,MAAM,IAAI;AAC7B;AAEA,MAAM,qBACJ,qBAEC,kBAAkB,YAAwC,CAAC;AAE9D,MAAM,oBAAoB,YAA6B;CACrD,IAAI,OAAO,YAAY,UAAU,OAAO;CAExC,OAAOA,QACJ,QAAQ,MAA2C,EAAE,SAAS,MAAM,CAAC,CACrE,KAAK,MAAM,EAAE,IAAI,CAAC,CAClB,KAAK,EAAE;AACZ;AAEA,MAAa,+BACX,SACA,WAA8C,CAAC,MACP;CAGxC,QAFa,eAAe,OAEjB,GAAX;EACE,KAAK,UACH,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,CAAC;IAAE,MAAM;IAAQ,MAAM,iBAAiB,QAAQ,OAAO;GAAE,CAAC;GACnE,UAAU,EACR,QAAQ,kBAAkB,QAAQ,iBAAiB,EACrD;EACF;EAEF,KAAK,SACH,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,eAAe,QAAQ,OAAO;GACvC,UAAU,EACR,QAAQ,kBAAkB,QAAQ,iBAAiB,EACrD;EACF;EAEF,KAAK,MAAM;GACT,MAAM,gBACJ,QAAQ,YAAY,KAAK,QAAQ;IAC/B,MAAM;IACN,YAAY,GAAG;IACf,UAAU,GAAG;IACb,MAAM,GAAG;IACT,UAAU,KAAK,UAAU,GAAG,IAAI;GAClC,EAAE,KAAK,CAAC;GAEV,MAAM,kBACJ,OAAO,QAAQ,WAAW,WAAW,QAAQ,SAAS,KAAA;GAExD,MAAM,eACH,QAAQ,KACL,SAAS,oBACL,IAAI,QAAQ,EAAE,CAAC,EACf,IAAI,mBAAmB,IAC3B,KAAA,MAAc,CAAC;GAErB,MAAM,SAAS,SAAS,gBAAgB,QAAQ,MAAM;GAEtD,OAAO;IACL,MAAM;IACN,IAAI,QAAQ;IACZ,SAAS;KACP,GAAG,eAAe,QAAQ,OAAO;KACjC,GAAG;KACH,GAAG;IACL;IACA,UAAU;KACR,QAAQ,kBAAkB,QAAQ,iBAAiB;KACnD,GAAI,UAAU,EAAE,OAAO;IACzB;IACA,GAAI,mBAAmB,EAAE,QAAQ,gBAAgB;GACnD;EACF;EAEA,KAAK,QACH,OAAO;GACL,MAAM;GACN,UAAU,QAAQ,QAAQ;GAC1B,YAAY,QAAQ,gBAAgB;GACpC,QAAQ,QAAQ;GAChB,UAAU,QAAQ;GAClB,SAAS,QAAQ,WAAW;EAC9B;EAEF,SACE,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,CACP;IACE,MAAM;IACN,MACE,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,KAAK,UAAU,QAAQ,OAAO;GACtC,CACF;EACF;CACJ;AACF;AAEA,MAAM,gBACJ,UAC8C;CAC9C,MAAM,QAAQ,MAAM,MAAM,yCAAyC;CACnE,IAAI,CAAC,OAAO,OAAO;CACnB,OAAO;EAAE,UAAU,MAAM;EAAK,MAAM,MAAM;CAAI;AAChD;AAEA,MAAM,iBAAiB;AAEvB,MAAa,qBAAqB,QAAuB;CACvD,MAAM,aAAa,CACjB,GAAG,IAAI,SACP,GAAI,IAAI,aAAa,SAAS,MAAM,EAAE,OAAO,KAAK,CAAC,CACrD;CAEA,MAAM,aAAa,WAAW,MAC3B,SAAS,KAAK,SAAS,UAAU,KAAK,SAAS,OAClD;CACA,MAAM,UAAU,WAAW,MAAM,SAAS,KAAK,SAAS,MAAM;CAC9D,IAAI,cAAc,CAAC,SACjB,WAAW,QAAQ;EAAE,MAAM;EAAQ,MAAM;CAAI,CAAC;CAGhD,MAAM,UAAU,WAAW,KAAK,SAAS;EACvC,MAAM,OAAO,KAAK;EAClB,QAAQ,MAAR;GACE,KAAK,QACH,OAAO;IAAE,MAAM;IAAiB,MAAM,KAAK;GAAK;GAClD,KAAK,SACH,OAAO;IAAE,MAAM;IAAsB,WAAW,EAAE,KAAK,KAAK,MAAM;GAAE;GACtE,KAAK,QAAQ;IACX,MAAM,WAAW,EAAE,UAAU,KAAK,YAAY,OAAO;IACrD,IAAI,eAAe,KAAK,KAAK,IAAI,GAC/B,OAAO;KACL,MAAM;KACN,KAAK,KAAK;KACV,WAAW,KAAK;KAChB;KACA,aAAa;IACf;IAEF,MAAM,SAAS,aAAa,KAAK,IAAI;IACrC,OAAO;KACL,MAAM;KACN,MAAM,QAAQ,QAAQ,KAAK;KAC3B,WAAW,QAAQ,YAAY,KAAK;KACpC;KACA,aAAa;IACf;GACF;GACA,KAAK,aACH,MAAM,IAAI,MAAM,sCAAsC;GACxD,SAOE,MAAM,IAAI,MACR,yCAAyCC,MAC3C;EAEJ;CACF,CAAC;CAED,IAAI,QAAQ,WAAW,KAAK,QAAQ,EAAE,EAAE,SAAS,QAC/C,OAAO,QAAQ,EAAE,CAAC,QAAQ;CAE5B,OAAO;AACT"}
+{"version":3,"file":"convertMessages.js","names":["parts","_exhaustiveCheck"],"sources":["../src/convertMessages.ts"],"sourcesContent":["\"use client\";\n\nimport type { useExternalMessageConverter } from \"@assistant-ui/core/react\";\nimport type {\n  AppendMessage,\n  DataMessagePart,\n  MessageTiming,\n} from \"@assistant-ui/core\";\nimport { httpUrlPattern, parseDataUrl } from \"@assistant-ui/core/internal\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport type {\n  LangChainBaseMessage,\n  LangChainContentBlock,\n  UIMessage,\n} from \"./types\";\n\ntype LangChainMessageConverterMetadata =\n  useExternalMessageConverter.Metadata & {\n    uiMessagesByParent?: Map<string, UIMessage[]>;\n    messageTiming?: Record<string, MessageTiming>;\n  };\n\nconst uiMessageToDataPart = (ui: UIMessage): DataMessagePart => ({\n  type: \"data\",\n  name: ui.name,\n  data: ui.props,\n});\n\nexport const getMessageType = (message: LangChainBaseMessage): string => {\n  if (typeof message._getType === \"function\") return message._getType();\n  if (\"type\" in message)\n    return (message as Record<string, unknown>).type as string;\n  throw new Error(\"Cannot determine message type\");\n};\n\nconst contentToParts = (content: unknown) => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n\n  const parts = content as readonly LangChainContentBlock[];\n  return parts\n    .map((part) => {\n      const type = part.type;\n      switch (type) {\n        case \"text\":\n        case \"text_delta\":\n          return { type: \"text\" as const, text: part.text };\n        case \"image_url\": {\n          const image =\n            typeof part.image_url === \"string\"\n              ? part.image_url\n              : part.image_url?.url;\n          if (!image) return null;\n          return { type: \"image\" as const, image };\n        }\n        case \"file\":\n          return {\n            type: \"file\" as const,\n            filename: part.metadata?.filename ?? \"file\",\n            data:\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          };\n        case \"thinking\":\n          return { type: \"reasoning\" as const, text: part.thinking };\n        case \"reasoning\":\n          return {\n            type: \"reasoning\" as const,\n            text:\n              part.summary?.map((s) => s?.text ?? \"\").join(\"\\n\\n\\n\") ??\n              part.reasoning ??\n              \"\",\n          };\n        case \"tool_use\":\n        case \"input_json_delta\":\n          return null;\n        default:\n          return null;\n      }\n    })\n    .filter((p) => p !== null);\n};\n\nconst getCustomMetadata = (\n  additionalKwargs: Record<string, unknown> | undefined,\n): Record<string, unknown> =>\n  (additionalKwargs?.metadata as Record<string, unknown>) ?? {};\n\nconst getStringContent = (content: unknown): string => {\n  if (typeof content === \"string\") return content;\n  const parts = content as readonly LangChainContentBlock[];\n  return parts\n    .filter((c): c is { type: \"text\"; text: string } => c.type === \"text\")\n    .map((c) => c.text)\n    .join(\"\");\n};\n\nexport const convertLangChainBaseMessage = (\n  message: LangChainBaseMessage,\n  metadata: LangChainMessageConverterMetadata = {},\n): useExternalMessageConverter.Message => {\n  const type = getMessageType(message);\n\n  switch (type) {\n    case \"system\":\n      return {\n        role: \"system\",\n        id: message.id,\n        content: [{ type: \"text\", text: getStringContent(message.content) }],\n        metadata: {\n          custom: getCustomMetadata(message.additional_kwargs),\n        },\n      };\n\n    case \"human\":\n      return {\n        role: \"user\",\n        id: message.id,\n        content: contentToParts(message.content),\n        metadata: {\n          custom: getCustomMetadata(message.additional_kwargs),\n        },\n      };\n\n    case \"ai\": {\n      const toolCallParts =\n        message.tool_calls?.map((tc) => ({\n          type: \"tool-call\" as const,\n          toolCallId: tc.id,\n          toolName: tc.name,\n          args: tc.args as ReadonlyJSONObject,\n          argsText: JSON.stringify(tc.args),\n        })) ?? [];\n\n      const assistantStatus =\n        typeof message.status === \"object\" ? message.status : undefined;\n\n      const uiDataParts =\n        (message.id\n          ? metadata.uiMessagesByParent\n              ?.get(message.id)\n              ?.map(uiMessageToDataPart)\n          : undefined) ?? [];\n\n      const timing = metadata.messageTiming?.[message.id ?? \"\"];\n\n      return {\n        role: \"assistant\",\n        id: message.id,\n        content: [\n          ...contentToParts(message.content),\n          ...toolCallParts,\n          ...uiDataParts,\n        ],\n        metadata: {\n          custom: getCustomMetadata(message.additional_kwargs),\n          ...(timing && { timing }),\n        },\n        ...(assistantStatus && { status: assistantStatus }),\n      };\n    }\n\n    case \"tool\":\n      return {\n        role: \"tool\",\n        toolName: message.name ?? \"\",\n        toolCallId: message.tool_call_id ?? \"\",\n        result: message.content,\n        artifact: message.artifact,\n        isError: message.status === \"error\",\n      };\n\n    default:\n      return {\n        role: \"system\",\n        id: message.id,\n        content: [\n          {\n            type: \"text\",\n            text:\n              typeof message.content === \"string\"\n                ? message.content\n                : JSON.stringify(message.content),\n          },\n        ],\n      };\n  }\n};\n\nexport const getMessageContent = (msg: AppendMessage) => {\n  const allContent = [\n    ...msg.content,\n    ...(msg.attachments?.flatMap((a) => a.content) ?? []),\n  ];\n\n  const hasNonText = allContent.some(\n    (part) => part.type === \"file\" || part.type === \"image\",\n  );\n  const hasText = allContent.some((part) => part.type === \"text\");\n  if (hasNonText && !hasText) {\n    allContent.unshift({ type: \"text\", text: \" \" });\n  }\n\n  const content = allContent.map((part) => {\n    const type = part.type;\n    switch (type) {\n      case \"text\":\n        return { type: \"text\" as const, text: part.text };\n      case \"image\":\n        return { type: \"image_url\" as const, image_url: { url: part.image } };\n      case \"file\": {\n        const metadata = { filename: part.filename ?? \"file\" };\n        if (httpUrlPattern.test(part.data)) {\n          return {\n            type: \"file\" as const,\n            url: part.data,\n            mime_type: part.mimeType,\n            metadata,\n            source_type: \"url\" as const,\n          };\n        }\n        const parsed = parseDataUrl(part.data);\n        return {\n          type: \"file\" as const,\n          data: parsed?.data ?? part.data,\n          mime_type: parsed?.mimeType ?? part.mimeType,\n          metadata,\n          source_type: \"base64\" as const,\n        };\n      }\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported.\");\n      default: {\n        const _exhaustiveCheck:\n          | \"reasoning\"\n          | \"source\"\n          | \"audio\"\n          | \"data\"\n          | \"generative-ui\" = type;\n        throw new Error(\n          `Unsupported append message part type: ${_exhaustiveCheck}`,\n        );\n      }\n    }\n  });\n\n  if (content.length === 1 && content[0]?.type === \"text\") {\n    return content[0].text ?? \"\";\n  }\n  return content;\n};\n"],"mappings":";;;AAsBA,MAAM,uBAAuB,QAAoC;CAC/D,MAAM;CACN,MAAM,GAAG;CACT,MAAM,GAAG;AACX;AAEA,MAAa,kBAAkB,YAA0C;CACvE,IAAI,OAAO,QAAQ,aAAa,YAAY,OAAO,QAAQ,SAAS;CACpE,IAAI,UAAU,SACZ,OAAQ,QAAoC;CAC9C,MAAM,IAAI,MAAM,+BAA+B;AACjD;AAEA,MAAM,kBAAkB,YAAqB;CAC3C,IAAI,OAAO,YAAY,UACrB,OAAO,CAAC;EAAE,MAAM;EAAiB,MAAM;CAAQ,CAAC;CAGlD,OAAOA,QACJ,KAAK,SAAS;EAEb,QADa,KAAK,MAClB;GACE,KAAK;GACL,KAAK,cACH,OAAO;IAAE,MAAM;IAAiB,MAAM,KAAK;GAAK;GAClD,KAAK,aAAa;IAChB,MAAM,QACJ,OAAO,KAAK,cAAc,WACtB,KAAK,YACL,KAAK,WAAW;IACtB,IAAI,CAAC,OAAO,OAAO;IACnB,OAAO;KAAE,MAAM;KAAkB;IAAM;GACzC;GACA,KAAK,QACH,OAAO;IACL,MAAM;IACN,UAAU,KAAK,UAAU,YAAY;IACrC,MACE,KAAK,gBAAgB,QACjB,KAAK,MACL,KAAK,gBAAgB,OACnB,KAAK,KACL,KAAK;IACb,UAAU,KAAK,aAAa;GAC9B;GACF,KAAK,YACH,OAAO;IAAE,MAAM;IAAsB,MAAM,KAAK;GAAS;GAC3D,KAAK,aACH,OAAO;IACL,MAAM;IACN,MACE,KAAK,SAAS,KAAK,MAAM,GAAG,QAAQ,EAAE,CAAC,CAAC,KAAK,QAAQ,KACrD,KAAK,aACL;GACJ;GACF,KAAK;GACL,KAAK,oBACH,OAAO;GACT,SACE,OAAO;EACX;CACF,CAAC,CAAC,CACD,QAAQ,MAAM,MAAM,IAAI;AAC7B;AAEA,MAAM,qBACJ,qBAEC,kBAAkB,YAAwC,CAAC;AAE9D,MAAM,oBAAoB,YAA6B;CACrD,IAAI,OAAO,YAAY,UAAU,OAAO;CAExC,OAAOA,QACJ,QAAQ,MAA2C,EAAE,SAAS,MAAM,CAAC,CACrE,KAAK,MAAM,EAAE,IAAI,CAAC,CAClB,KAAK,EAAE;AACZ;AAEA,MAAa,+BACX,SACA,WAA8C,CAAC,MACP;CAGxC,QAFa,eAAe,OAEjB,GAAX;EACE,KAAK,UACH,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,CAAC;IAAE,MAAM;IAAQ,MAAM,iBAAiB,QAAQ,OAAO;GAAE,CAAC;GACnE,UAAU,EACR,QAAQ,kBAAkB,QAAQ,iBAAiB,EACrD;EACF;EAEF,KAAK,SACH,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,eAAe,QAAQ,OAAO;GACvC,UAAU,EACR,QAAQ,kBAAkB,QAAQ,iBAAiB,EACrD;EACF;EAEF,KAAK,MAAM;GACT,MAAM,gBACJ,QAAQ,YAAY,KAAK,QAAQ;IAC/B,MAAM;IACN,YAAY,GAAG;IACf,UAAU,GAAG;IACb,MAAM,GAAG;IACT,UAAU,KAAK,UAAU,GAAG,IAAI;GAClC,EAAE,KAAK,CAAC;GAEV,MAAM,kBACJ,OAAO,QAAQ,WAAW,WAAW,QAAQ,SAAS,KAAA;GAExD,MAAM,eACH,QAAQ,KACL,SAAS,oBACL,IAAI,QAAQ,EAAE,CAAC,EACf,IAAI,mBAAmB,IAC3B,KAAA,MAAc,CAAC;GAErB,MAAM,SAAS,SAAS,gBAAgB,QAAQ,MAAM;GAEtD,OAAO;IACL,MAAM;IACN,IAAI,QAAQ;IACZ,SAAS;KACP,GAAG,eAAe,QAAQ,OAAO;KACjC,GAAG;KACH,GAAG;IACL;IACA,UAAU;KACR,QAAQ,kBAAkB,QAAQ,iBAAiB;KACnD,GAAI,UAAU,EAAE,OAAO;IACzB;IACA,GAAI,mBAAmB,EAAE,QAAQ,gBAAgB;GACnD;EACF;EAEA,KAAK,QACH,OAAO;GACL,MAAM;GACN,UAAU,QAAQ,QAAQ;GAC1B,YAAY,QAAQ,gBAAgB;GACpC,QAAQ,QAAQ;GAChB,UAAU,QAAQ;GAClB,SAAS,QAAQ,WAAW;EAC9B;EAEF,SACE,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,CACP;IACE,MAAM;IACN,MACE,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,KAAK,UAAU,QAAQ,OAAO;GACtC,CACF;EACF;CACJ;AACF;AAEA,MAAa,qBAAqB,QAAuB;CACvD,MAAM,aAAa,CACjB,GAAG,IAAI,SACP,GAAI,IAAI,aAAa,SAAS,MAAM,EAAE,OAAO,KAAK,CAAC,CACrD;CAEA,MAAM,aAAa,WAAW,MAC3B,SAAS,KAAK,SAAS,UAAU,KAAK,SAAS,OAClD;CACA,MAAM,UAAU,WAAW,MAAM,SAAS,KAAK,SAAS,MAAM;CAC9D,IAAI,cAAc,CAAC,SACjB,WAAW,QAAQ;EAAE,MAAM;EAAQ,MAAM;CAAI,CAAC;CAGhD,MAAM,UAAU,WAAW,KAAK,SAAS;EACvC,MAAM,OAAO,KAAK;EAClB,QAAQ,MAAR;GACE,KAAK,QACH,OAAO;IAAE,MAAM;IAAiB,MAAM,KAAK;GAAK;GAClD,KAAK,SACH,OAAO;IAAE,MAAM;IAAsB,WAAW,EAAE,KAAK,KAAK,MAAM;GAAE;GACtE,KAAK,QAAQ;IACX,MAAM,WAAW,EAAE,UAAU,KAAK,YAAY,OAAO;IACrD,IAAI,eAAe,KAAK,KAAK,IAAI,GAC/B,OAAO;KACL,MAAM;KACN,KAAK,KAAK;KACV,WAAW,KAAK;KAChB;KACA,aAAa;IACf;IAEF,MAAM,SAAS,aAAa,KAAK,IAAI;IACrC,OAAO;KACL,MAAM;KACN,MAAM,QAAQ,QAAQ,KAAK;KAC3B,WAAW,QAAQ,YAAY,KAAK;KACpC;KACA,aAAa;IACf;GACF;GACA,KAAK,aACH,MAAM,IAAI,MAAM,sCAAsC;GACxD,SAOE,MAAM,IAAI,MACR,yCAAyCC,MAC3C;EAEJ;CACF,CAAC;CAED,IAAI,QAAQ,WAAW,KAAK,QAAQ,EAAE,EAAE,SAAS,QAC/C,OAAO,QAAQ,EAAE,CAAC,QAAQ;CAE5B,OAAO;AACT"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-langchain",
-  "version": "0.0.19",
+  "version": "0.0.21",
   "description": "LangChain useStream adapter for assistant-ui",
   "keywords": [
     "langchain",
@@ -29,10 +29,10 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.2.21",
-    "@assistant-ui/store": "^0.2.20",
+    "@assistant-ui/core": "^0.3.0",
+    "@assistant-ui/store": "^0.3.0",
     "assistant-cloud": "*",
-    "assistant-stream": "^0.3.26"
+    "assistant-stream": "^0.3.29"
   },
   "peerDependencies": {
     "@langchain/react": "^1.0.2",
@@ -46,14 +46,14 @@
   },
   "devDependencies": {
     "@langchain/core": "^1.2.3",
-    "@langchain/langgraph-sdk": "^1.9.25",
-    "@langchain/react": "^1.0.26",
+    "@langchain/langgraph-sdk": "^1.9.28",
+    "@langchain/react": "^1.0.29",
     "@testing-library/react": "^16.3.2",
     "@types/react": "^19.2.17",
     "jsdom": "^29.1.1",
-    "react": "^19.2.7",
+    "react": "^19.2.8",
     "vitest": "^4.1.10",
-    "@assistant-ui/x-buildutils": "0.0.18"
+    "@assistant-ui/x-buildutils": "0.0.20"
   },
   "publishConfig": {
     "access": "public",

package/src/convertMessages.ts

@@ -6,6 +6,7 @@ import type {
   DataMessagePart,
   MessageTiming,
 } from "@assistant-ui/core";
+import { httpUrlPattern, parseDataUrl } from "@assistant-ui/core/internal";
 import type { ReadonlyJSONObject } from "assistant-stream/utils";
 import type {
   LangChainBaseMessage,
@@ -190,16 +191,6 @@ export const convertLangChainBaseMessage = (
   }
 };
 
-const parseDataUrl = (
-  value: string,
-): { mimeType: string; data: string } | null => {
-  const match = value.match(/^data:([^;,]+)(?:;[^;,]+)*;base64,(.+)$/);
-  if (!match) return null;
-  return { mimeType: match[1]!, data: match[2]! };
-};
-
-const httpUrlPattern = /^https?:\/\//i;
-
 export const getMessageContent = (msg: AppendMessage) => {
   const allContent = [
     ...msg.content,

package/src/useStreamRuntime.test.tsx

@@ -103,15 +103,12 @@ const renderAui = (stream: MockStream) => {
 };
 
 const getText = (aui: ReturnType<typeof useAui>) =>
-  aui
-    .thread()
-    .getState()
-    .messages.map((m) =>
-      m.content
-        .filter((part) => part.type === "text")
-        .map((part) => part.text)
-        .join(""),
-    );
+  aui.thread.getState().messages.map((m) =>
+    m.content
+      .filter((part) => part.type === "text")
+      .map((part) => part.text)
+      .join(""),
+  );
 
 describe("useStreamRuntime staged messages", () => {
   it("stages a new user message without submitting when startRun is false", async () => {
@@ -119,7 +116,7 @@ describe("useStreamRuntime staged messages", () => {
     const { auiResult } = renderAui(stream);
 
     await act(async () => {
-      auiResult.current.thread().append({
+      auiResult.current.thread.append({
         role: "user",
         content: [{ type: "text", text: "draft" }],
         startRun: false,
@@ -141,7 +138,7 @@ describe("useStreamRuntime staged messages", () => {
     const { auiResult, rerender } = renderAui(stream);
 
     await act(async () => {
-      auiResult.current.thread().append({
+      auiResult.current.thread.append({
         role: "user",
         parentId: "u1",
         content: [{ type: "text", text: "edited" }],
@@ -171,12 +168,12 @@ describe("useStreamRuntime staged messages", () => {
     const { auiResult, rerender } = renderAui(stream);
 
     await act(async () => {
-      auiResult.current.thread().append({
+      auiResult.current.thread.append({
         role: "user",
         content: [{ type: "text", text: "first staged" }],
         startRun: false,
       });
-      auiResult.current.thread().append({
+      auiResult.current.thread.append({
         role: "user",
         content: [{ type: "text", text: "second staged" }],
         startRun: false,
@@ -191,9 +188,9 @@ describe("useStreamRuntime staged messages", () => {
       ]);
     });
 
-    const firstStagedId = auiResult.current.thread().getState().messages[1]!.id;
+    const firstStagedId = auiResult.current.thread.getState().messages[1]!.id;
     await act(async () => {
-      await auiResult.current.thread().startRun({
+      await auiResult.current.thread.startRun({
         parentId: firstStagedId,
         sourceId: null,
         runConfig: {},

package/src/runConfigToSubmitOptions.test.ts

@@ -1,24 +0,0 @@
-import { describe, expect, it } from "vitest";
-import type { AppendMessage } from "@assistant-ui/core";
-import { runConfigToSubmitOptions } from "./useStreamRuntime";
-
-type RunConfig = AppendMessage["runConfig"];
-
-describe("runConfigToSubmitOptions", () => {
-  it("returns undefined when runConfig is undefined", () => {
-    expect(runConfigToSubmitOptions(undefined)).toBeUndefined();
-  });
-
-  it("returns undefined when custom is undefined", () => {
-    expect(
-      runConfigToSubmitOptions({ custom: undefined } as unknown as RunConfig),
-    ).toBeUndefined();
-  });
-
-  it("maps custom to config.configurable", () => {
-    const runConfig = { custom: { mode: "plan" } } as RunConfig;
-    expect(runConfigToSubmitOptions(runConfig)).toEqual({
-      config: { configurable: { mode: "plan" } },
-    });
-  });
-});

package/src/streamingTiming.test.ts

@@ -1,118 +0,0 @@
-import { describe, expect, it } from "vitest";
-import type { LangChainBaseMessage } from "./types";
-import { langChainStreamingTimingAccessors } from "./streamingTiming";
-
-const ai = (
-  fields: Partial<LangChainBaseMessage> & { id: string; content: unknown },
-): LangChainBaseMessage => ({
-  _getType: () => "ai",
-  ...fields,
-});
-
-const human = (id: string, content: unknown): LangChainBaseMessage => ({
-  _getType: () => "human",
-  id,
-  content,
-});
-
-const { getAssistantMessageId, getTextLength, getToolCallCount } =
-  langChainStreamingTimingAccessors;
-
-describe("langChainStreamingTimingAccessors", () => {
-  it("resolves the last assistant message id via _getType()", () => {
-    expect(
-      getAssistantMessageId([human("u1", "hi"), ai({ id: "a1", content: "" })]),
-    ).toBe("a1");
-    expect(
-      getAssistantMessageId([
-        ai({ id: "a1", content: "" }),
-        human("u2", "hi"),
-        ai({ id: "a2", content: "" }),
-      ]),
-    ).toBe("a2");
-  });
-
-  it("returns undefined when there is no assistant message", () => {
-    expect(getAssistantMessageId([human("u1", "hi")])).toBeUndefined();
-  });
-
-  it("measures string content length", () => {
-    expect(getTextLength([ai({ id: "a1", content: "hello" })], "a1")).toBe(5);
-  });
-
-  it("sums text and thinking lengths across content blocks", () => {
-    expect(
-      getTextLength(
-        [
-          ai({
-            id: "a1",
-            content: [
-              { type: "thinking", thinking: "hmm" },
-              { type: "text", text: "answer" },
-            ],
-          }),
-        ],
-        "a1",
-      ),
-    ).toBe("hmm".length + "answer".length);
-  });
-
-  it("measures reasoning blocks (summary and reasoning fields)", () => {
-    // reasoning via summary_text entries, joined like contentToParts does.
-    expect(
-      getTextLength(
-        [
-          ai({
-            id: "a1",
-            content: [
-              {
-                type: "reasoning",
-                summary: [
-                  { type: "summary_text", text: "step one" },
-                  { type: "summary_text", text: "step two" },
-                ],
-              },
-            ],
-          }),
-        ],
-        "a1",
-      ),
-    ).toBe("step one\n\n\nstep two".length);
-
-    // reasoning via the bare `reasoning` field when no summary is present.
-    expect(
-      getTextLength(
-        [
-          ai({
-            id: "a2",
-            content: [{ type: "reasoning", reasoning: "deduced" }],
-          }),
-        ],
-        "a2",
-      ),
-    ).toBe("deduced".length);
-  });
-
-  it("counts tool_calls on the assistant message", () => {
-    expect(
-      getToolCallCount(
-        [
-          ai({
-            id: "a1",
-            content: "",
-            tool_calls: [
-              { id: "t1", name: "search", args: {} },
-              { id: "t2", name: "fetch", args: {} },
-            ],
-          }),
-        ],
-        "a1",
-      ),
-    ).toBe(2);
-  });
-
-  it("returns 0 when the message is missing or not assistant", () => {
-    expect(getTextLength([human("u1", "hi")], "a1")).toBe(0);
-    expect(getToolCallCount([human("u1", "hi")], "a1")).toBe(0);
-  });
-});

package/src/useLangChainError.test.tsx

@@ -1,48 +0,0 @@
-// @vitest-environment jsdom
-
-import { describe, expect, it, vi } from "vitest";
-import { renderHook } from "@testing-library/react";
-import {
-  createRunSelectorAgainst,
-  makeExtras,
-  type Selector,
-} from "./__tests__/langChainTestUtils";
-
-const { mockUseAuiState } = vi.hoisted(() => ({
-  mockUseAuiState: vi.fn(),
-}));
-
-vi.mock(import("@assistant-ui/store"), async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    ...actual,
-    useAuiState: ((selector: Selector) =>
-      mockUseAuiState(selector)) as typeof actual.useAuiState,
-    useAui: (() => ({})) as unknown as typeof actual.useAui,
-  };
-});
-
-import { useLangChainError } from "./hooks";
-
-const runSelectorAgainst = createRunSelectorAgainst(mockUseAuiState);
-
-describe("useLangChainError", () => {
-  it("returns undefined when extras are absent", () => {
-    runSelectorAgainst(undefined);
-    const { result } = renderHook(() => useLangChainError());
-    expect(result.current).toBeUndefined();
-  });
-
-  it("returns the error value when extras carry error", () => {
-    const error = new Error("boom");
-    runSelectorAgainst(makeExtras({ error }));
-    const { result } = renderHook(() => useLangChainError());
-    expect(result.current).toBe(error);
-  });
-
-  it("returns undefined when extras carry no error", () => {
-    runSelectorAgainst(makeExtras({ error: undefined }));
-    const { result } = renderHook(() => useLangChainError());
-    expect(result.current).toBeUndefined();
-  });
-});

package/src/useLangChainInterrupts.test.tsx

@@ -1,48 +0,0 @@
-// @vitest-environment jsdom
-
-import { describe, expect, it, vi } from "vitest";
-import { renderHook } from "@testing-library/react";
-import {
-  createRunSelectorAgainst,
-  makeExtras,
-  type Selector,
-} from "./__tests__/langChainTestUtils";
-
-const { mockUseAuiState } = vi.hoisted(() => ({
-  mockUseAuiState: vi.fn(),
-}));
-
-vi.mock(import("@assistant-ui/store"), async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    ...actual,
-    useAuiState: ((selector: Selector) =>
-      mockUseAuiState(selector)) as typeof actual.useAuiState,
-    useAui: (() => ({})) as unknown as typeof actual.useAui,
-  };
-});
-
-import { useLangChainInterrupts } from "./hooks";
-
-const runSelectorAgainst = createRunSelectorAgainst(mockUseAuiState);
-
-describe("useLangChainInterrupts", () => {
-  it("returns an empty array when extras are absent", () => {
-    runSelectorAgainst(undefined);
-    const { result } = renderHook(() => useLangChainInterrupts());
-    expect(result.current).toEqual([]);
-  });
-
-  it("returns the pending interrupts from extras", () => {
-    const interrupts = [{ id: "i1", value: { q: "ok?" } }];
-    runSelectorAgainst(makeExtras({ interrupts }));
-    const { result } = renderHook(() => useLangChainInterrupts());
-    expect(result.current).toBe(interrupts);
-  });
-
-  it("returns an empty array when extras carry no interrupts", () => {
-    runSelectorAgainst(makeExtras({ interrupts: undefined }));
-    const { result } = renderHook(() => useLangChainInterrupts());
-    expect(result.current).toEqual([]);
-  });
-});

package/src/useLangChainRespond.test.tsx

@@ -1,49 +0,0 @@
-// @vitest-environment jsdom
-
-import { describe, expect, it, vi } from "vitest";
-import { renderHook } from "@testing-library/react";
-import { makeExtras } from "./__tests__/langChainTestUtils";
-
-const { mockUseAui } = vi.hoisted(() => ({
-  mockUseAui: vi.fn(),
-}));
-
-vi.mock(import("@assistant-ui/store"), async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    ...actual,
-    useAui: (() => mockUseAui()) as unknown as typeof actual.useAui,
-  };
-});
-
-import { useLangChainRespond } from "./hooks";
-
-describe("useLangChainRespond", () => {
-  it("forwards the response and options to stream.respond", async () => {
-    const respond = vi.fn().mockResolvedValue(undefined);
-    mockUseAui.mockReturnValue({
-      thread: () => ({
-        getState: () => ({ extras: makeExtras({ respond }) }),
-      }),
-    });
-
-    const { result } = renderHook(() => useLangChainRespond());
-    await result.current({ approved: true }, { interruptId: "x" });
-
-    expect(respond).toHaveBeenCalledWith(
-      { approved: true },
-      { interruptId: "x" },
-    );
-  });
-
-  it("throws when extras are absent or unbranded", () => {
-    mockUseAui.mockReturnValue({
-      thread: () => ({ getState: () => ({ extras: { respond: vi.fn() } }) }),
-    });
-
-    const { result } = renderHook(() => useLangChainRespond());
-    expect(() => result.current({ approved: true })).toThrow(
-      "useStreamRuntime",
-    );
-  });
-});

package/src/useLangChainRespondAll.test.tsx

@@ -1,52 +0,0 @@
-// @vitest-environment jsdom
-
-import { describe, expect, it, vi } from "vitest";
-import { renderHook } from "@testing-library/react";
-import { makeExtras } from "./__tests__/langChainTestUtils";
-
-const { mockUseAui } = vi.hoisted(() => ({
-  mockUseAui: vi.fn(),
-}));
-
-vi.mock(import("@assistant-ui/store"), async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    ...actual,
-    useAui: (() => mockUseAui()) as unknown as typeof actual.useAui,
-  };
-});
-
-import { useLangChainRespondAll } from "./hooks";
-
-describe("useLangChainRespondAll", () => {
-  it("forwards the responses and options to stream.respondAll", async () => {
-    const respondAll = vi.fn().mockResolvedValue(undefined);
-    mockUseAui.mockReturnValue({
-      thread: () => ({
-        getState: () => ({ extras: makeExtras({ respondAll }) }),
-      }),
-    });
-
-    const { result } = renderHook(() => useLangChainRespondAll());
-    await result.current(
-      { a: { approved: true }, b: { approved: false } },
-      { metadata: { src: "ui" } },
-    );
-
-    expect(respondAll).toHaveBeenCalledWith(
-      { a: { approved: true }, b: { approved: false } },
-      { metadata: { src: "ui" } },
-    );
-  });
-
-  it("throws when extras are absent or unbranded", () => {
-    mockUseAui.mockReturnValue({
-      thread: () => ({ getState: () => ({ extras: { respondAll: vi.fn() } }) }),
-    });
-
-    const { result } = renderHook(() => useLangChainRespondAll());
-    expect(() => result.current({ a: { approved: true } })).toThrow(
-      "useStreamRuntime",
-    );
-  });
-});

package/src/useLangChainStream.test.tsx

@@ -1,42 +0,0 @@
-// @vitest-environment jsdom
-
-import { describe, expect, it, vi } from "vitest";
-import { renderHook } from "@testing-library/react";
-import {
-  createRunSelectorAgainst,
-  makeExtras,
-  type Selector,
-} from "./__tests__/langChainTestUtils";
-
-const { mockUseAuiState } = vi.hoisted(() => ({
-  mockUseAuiState: vi.fn(),
-}));
-
-vi.mock(import("@assistant-ui/store"), async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    ...actual,
-    useAuiState: ((selector: Selector) =>
-      mockUseAuiState(selector)) as typeof actual.useAuiState,
-    useAui: (() => ({})) as unknown as typeof actual.useAui,
-  };
-});
-
-import { useLangChainStream } from "./hooks";
-
-const runSelectorAgainst = createRunSelectorAgainst(mockUseAuiState);
-
-describe("useLangChainStream", () => {
-  it("returns undefined when extras are absent", () => {
-    runSelectorAgainst(undefined);
-    const { result } = renderHook(() => useLangChainStream());
-    expect(result.current).toBeUndefined();
-  });
-
-  it("returns the stream handle from extras", () => {
-    const stream = { submit: () => {} };
-    runSelectorAgainst(makeExtras({ stream }));
-    const { result } = renderHook(() => useLangChainStream());
-    expect(result.current).toBe(stream);
-  });
-});

package/src/useLangChainSubagents.test.tsx

@@ -1,48 +0,0 @@
-// @vitest-environment jsdom
-
-import { describe, expect, it, vi } from "vitest";
-import { renderHook } from "@testing-library/react";
-import {
-  createRunSelectorAgainst,
-  makeExtras,
-  type Selector,
-} from "./__tests__/langChainTestUtils";
-
-const { mockUseAuiState } = vi.hoisted(() => ({
-  mockUseAuiState: vi.fn(),
-}));
-
-vi.mock(import("@assistant-ui/store"), async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    ...actual,
-    useAuiState: ((selector: Selector) =>
-      mockUseAuiState(selector)) as typeof actual.useAuiState,
-    useAui: (() => ({})) as unknown as typeof actual.useAui,
-  };
-});
-
-import { useLangChainSubagents } from "./hooks";
-
-const runSelectorAgainst = createRunSelectorAgainst(mockUseAuiState);
-
-describe("useLangChainSubagents", () => {
-  it("returns the stable empty map when extras are absent", () => {
-    runSelectorAgainst(undefined);
-    const { result } = renderHook(() => useLangChainSubagents());
-    expect(result.current.size).toBe(0);
-  });
-
-  it("returns the subagents map from extras", () => {
-    const subagents = new Map([["researcher", { namespace: "researcher" }]]);
-    runSelectorAgainst(makeExtras({ subagents }));
-    const { result } = renderHook(() => useLangChainSubagents());
-    expect(result.current).toBe(subagents);
-  });
-
-  it("returns the stable empty map when extras carry no subagents", () => {
-    runSelectorAgainst(makeExtras({ subagents: undefined }));
-    const { result } = renderHook(() => useLangChainSubagents());
-    expect(result.current.size).toBe(0);
-  });
-});

package/src/useLangChainSubgraphs.test.tsx

@@ -1,48 +0,0 @@
-// @vitest-environment jsdom
-
-import { describe, expect, it, vi } from "vitest";
-import { renderHook } from "@testing-library/react";
-import {
-  createRunSelectorAgainst,
-  makeExtras,
-  type Selector,
-} from "./__tests__/langChainTestUtils";
-
-const { mockUseAuiState } = vi.hoisted(() => ({
-  mockUseAuiState: vi.fn(),
-}));
-
-vi.mock(import("@assistant-ui/store"), async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    ...actual,
-    useAuiState: ((selector: Selector) =>
-      mockUseAuiState(selector)) as typeof actual.useAuiState,
-    useAui: (() => ({})) as unknown as typeof actual.useAui,
-  };
-});
-
-import { useLangChainSubgraphs } from "./hooks";
-
-const runSelectorAgainst = createRunSelectorAgainst(mockUseAuiState);
-
-describe("useLangChainSubgraphs", () => {
-  it("returns the stable empty map when extras are absent", () => {
-    runSelectorAgainst(undefined);
-    const { result } = renderHook(() => useLangChainSubgraphs());
-    expect(result.current.size).toBe(0);
-  });
-
-  it("returns the subgraphs map from extras", () => {
-    const subgraphs = new Map([["planner", { namespace: "planner" }]]);
-    runSelectorAgainst(makeExtras({ subgraphs }));
-    const { result } = renderHook(() => useLangChainSubgraphs());
-    expect(result.current).toBe(subgraphs);
-  });
-
-  it("returns the stable empty map when extras carry no subgraphs", () => {
-    runSelectorAgainst(makeExtras({ subgraphs: undefined }));
-    const { result } = renderHook(() => useLangChainSubgraphs());
-    expect(result.current.size).toBe(0);
-  });
-});

package/src/useLangChainToolCalls.test.tsx

@@ -1,48 +0,0 @@
-// @vitest-environment jsdom
-
-import { describe, expect, it, vi } from "vitest";
-import { renderHook } from "@testing-library/react";
-import {
-  createRunSelectorAgainst,
-  makeExtras,
-  type Selector,
-} from "./__tests__/langChainTestUtils";
-
-const { mockUseAuiState } = vi.hoisted(() => ({
-  mockUseAuiState: vi.fn(),
-}));
-
-vi.mock(import("@assistant-ui/store"), async (importOriginal) => {
-  const actual = await importOriginal();
-  return {
-    ...actual,
-    useAuiState: ((selector: Selector) =>
-      mockUseAuiState(selector)) as typeof actual.useAuiState,
-    useAui: (() => ({})) as unknown as typeof actual.useAui,
-  };
-});
-
-import { useLangChainToolCalls } from "./hooks";
-
-const runSelectorAgainst = createRunSelectorAgainst(mockUseAuiState);
-
-describe("useLangChainToolCalls", () => {
-  it("returns an empty array when extras are absent", () => {
-    runSelectorAgainst(undefined);
-    const { result } = renderHook(() => useLangChainToolCalls());
-    expect(result.current).toEqual([]);
-  });
-
-  it("returns the assembled tool calls from extras", () => {
-    const toolCalls = [{ id: "a", name: "search", args: { q: "x" } }];
-    runSelectorAgainst(makeExtras({ toolCalls }));
-    const { result } = renderHook(() => useLangChainToolCalls());
-    expect(result.current).toBe(toolCalls);
-  });
-
-  it("returns an empty array when extras carry no tool calls", () => {
-    runSelectorAgainst(makeExtras({ toolCalls: undefined }));
-    const { result } = renderHook(() => useLangChainToolCalls());
-    expect(result.current).toEqual([]);
-  });
-});