@assistant-ui/react-langchain 0.0.32 → 0.0.34

package/dist/useSubagentTranscripts.js.map

@@ -1 +1 @@
-{"version":3,"file":"useSubagentTranscripts.js","names":[],"sources":["../src/useSubagentTranscripts.ts"],"sourcesContent":["\"use client\";\n\nimport { useEffect, useRef, useSyncExternalStore } from \"react\";\nimport {\n  stepStreamingTiming,\n  type MessageTiming,\n  type StreamingTimingState,\n  type ThreadMessage,\n  type ToolCallTiming,\n} from \"@assistant-ui/core\";\nimport {\n  convertExternalMessages,\n  createExternalMessageConversionCache,\n  type ExternalMessageConversionCache,\n  type useExternalMessageConverter,\n} from \"@assistant-ui/core/react\";\nimport { STREAM_CONTROLLER, type AnyStream } from \"@langchain/react\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\nimport {\n  channelProjection,\n  messagesProjection,\n  type Event,\n} from \"@langchain/langgraph-sdk/stream\";\nimport type { SubagentDiscoverySnapshot } from \"@langchain/react\";\nimport {\n  attachSubagentTranscripts,\n  type AttachMemo,\n  createAttachMemo,\n  type SubagentTranscript,\n} from \"./attachSubagentTranscripts\";\nimport { convertLangChainBaseMessage } from \"./convertMessages\";\nimport { groupUIMessagesByParent } from \"./converter\";\nimport { langChainStreamingTimingAccessors } from \"./streamingTiming\";\nimport type { LangChainBaseMessage, UIMessage } from \"./types\";\nimport {\n  createUIFoldMemo,\n  foldUIUpdates,\n  mergeUIMessages,\n  UI_CUSTOM_CHANNELS,\n  type UIFoldMemo,\n} from \"./uiMessages\";\n\nexport const MAX_SUBAGENT_DEPTH = 16;\n\nconst ROOT_UI_CHANNEL_DEPTH = 1;\n\nconst TRANSCRIPT_METADATA = {};\n\nconst NO_UI_MESSAGES: readonly UIMessage[] = [];\n\nconst NO_MESSAGE_TIMING: Record<string, MessageTiming> = {};\n\ntype ProjectionStore<T> = {\n  getSnapshot(): T;\n  subscribe(listener: () => void): () => void;\n};\n\ntype ProjectionResource = {\n  snapshot: SubagentDiscoverySnapshot;\n  namespace: readonly string[];\n  store: ProjectionStore<BaseMessage[]>;\n  uiStore: ProjectionStore<readonly Event[]> | undefined;\n  dispose: () => void;\n  storeSnapshot: BaseMessage[] | undefined;\n  status: SubagentDiscoverySnapshot[\"status\"] | undefined;\n  uiFoldMemo: UIFoldMemo;\n  localUiMessages: readonly UIMessage[];\n  rootUiMessagesByParent: Map<string, UIMessage[]>;\n  uiMessagesByParent: Map<string, UIMessage[]>;\n  timingState: StreamingTimingState | null;\n  messageTiming: Record<string, MessageTiming>;\n  convertedMessageTiming: Record<string, MessageTiming> | undefined;\n  convert: useExternalMessageConverter.Callback<LangChainBaseMessage>;\n  uiMessages: readonly UIMessage[];\n  converted: readonly ThreadMessage[] | undefined;\n  childTranscripts: ReadonlyMap<string, SubagentTranscript> | undefined;\n  transcript: readonly ThreadMessage[] | undefined;\n  timing: ToolCallTiming | undefined;\n  entry: SubagentTranscript | undefined;\n  memo: AttachMemo;\n  cache: ExternalMessageConversionCache;\n};\n\ntype MeasuredTiming = {\n  timing: ToolCallTiming | undefined;\n  timingState: StreamingTimingState | null;\n  messageTiming: Record<string, MessageTiming>;\n};\n\ntype NamespaceRequest = {\n  id: string;\n  attempts: number;\n  pending: boolean;\n  retryQueued: boolean;\n  status: SubagentDiscoverySnapshot[\"status\"];\n};\n\ntype SubagentTranscriptSource = {\n  resources: Map<string, ProjectionResource>;\n  namespaceRequests: Map<string, NamespaceRequest>;\n  snapshot: ReadonlyMap<string, SubagentTranscript>;\n  listeners: Set<() => void>;\n  controller: AnyStream[typeof STREAM_CONTROLLER] | undefined;\n  uiMessagesByParent: Map<string, UIMessage[]>;\n  subscribe(listener: () => void): () => void;\n  getSnapshot(): ReadonlyMap<string, SubagentTranscript>;\n  reconcile(\n    controller: AnyStream[typeof STREAM_CONTROLLER],\n    subagents: AnyStream[\"subagents\"],\n    uiMessagesByParent: Map<string, UIMessage[]>,\n  ): void;\n  dispose(): void;\n};\n\nconst sameNamespace = (a: readonly string[], b: readonly string[]) =>\n  a.length === b.length && a.every((segment, index) => segment === b[index]);\n\nconst needsNamespaceResolution = (snapshot: SubagentDiscoverySnapshot) =>\n  snapshot.namespace.length === 1 &&\n  snapshot.namespace[0] === `tools:${snapshot.id}`;\n\nconst requestSubagentNamespace = (\n  source: SubagentTranscriptSource,\n  controller: AnyStream[typeof STREAM_CONTROLLER],\n  request: NamespaceRequest,\n) => {\n  request.attempts += 1;\n  request.pending = true;\n  request.retryQueued = false;\n  void controller\n    .resolveSubagentNamespace(request.id)\n    .catch(() => {})\n    .finally(() => {\n      if (\n        source.controller !== controller ||\n        source.namespaceRequests.get(request.id) !== request\n      )\n        return;\n\n      request.pending = false;\n      if (\n        !request.retryQueued &&\n        request.attempts === 1 &&\n        request.status !== \"running\"\n      ) {\n        request.retryQueued = true;\n      }\n      if (!request.retryQueued || request.attempts >= 2) {\n        request.retryQueued = false;\n        return;\n      }\n      requestSubagentNamespace(source, controller, request);\n    });\n};\n\nconst sameTranscriptEntries = (\n  a: ReadonlyMap<string, SubagentTranscript> | undefined,\n  b: ReadonlyMap<string, SubagentTranscript>,\n) =>\n  a?.size === b.size &&\n  [...b].every(([id, transcript]) => a.get(id) === transcript);\n\nconst collectUIMessages = (\n  messages: readonly BaseMessage[],\n  uiMessagesByParent: Map<string, UIMessage[]>,\n) => {\n  const collected: UIMessage[] = [];\n  if (uiMessagesByParent.size === 0) return collected;\n  for (const message of messages) {\n    const uiMessages = message.id && uiMessagesByParent.get(message.id);\n    if (uiMessages) collected.push(...uiMessages);\n  }\n  return collected;\n};\n\nconst sameUIMessages = (a: readonly UIMessage[], b: readonly UIMessage[]) =>\n  a.length === b.length && a.every((ui, index) => ui === b[index]);\n\nconst foldLocalUIMessages = (resource: ProjectionResource) => {\n  const events = resource.uiStore?.getSnapshot();\n  if (events === undefined) return NO_UI_MESSAGES;\n  return foldUIUpdates(events, resource.uiFoldMemo);\n};\n\nconst mergeLocalUIMessages = (\n  uiMessagesByParent: Map<string, UIMessage[]>,\n  localUiMessages: readonly UIMessage[],\n) => {\n  const merged = new Map(uiMessagesByParent);\n  for (const [parentId, messages] of groupUIMessagesByParent<UIMessage>(\n    localUiMessages,\n  )) {\n    merged.set(parentId, mergeUIMessages(messages, merged.get(parentId)));\n  }\n  return merged;\n};\n\nconst createConverter =\n  (\n    uiMessagesByParent: Map<string, UIMessage[]>,\n    messageTiming: Record<string, MessageTiming>,\n  ): useExternalMessageConverter.Callback<LangChainBaseMessage> =>\n  (message, metadata) =>\n    convertLangChainBaseMessage(message, {\n      ...metadata,\n      uiMessagesByParent,\n      messageTiming,\n    });\n\nconst updateTiming = (resource: ProjectionResource) => {\n  if (resource.timing === undefined) return;\n  const startedAt = resource.snapshot.startedAt.getTime();\n  const completedAt = resource.snapshot.completedAt?.getTime();\n  if (\n    resource.timing.startedAt === startedAt &&\n    resource.timing.completedAt === completedAt\n  )\n    return;\n  resource.timing = {\n    startedAt,\n    ...(completedAt !== undefined && { completedAt }),\n  };\n};\n\nconst createSubagentTranscriptSource = (): SubagentTranscriptSource => {\n  const uiMessagesByParent = new Map<string, UIMessage[]>();\n  const source: SubagentTranscriptSource = {\n    resources: new Map(),\n    namespaceRequests: new Map(),\n    snapshot: new Map(),\n    listeners: new Set(),\n    controller: undefined,\n    uiMessagesByParent,\n    subscribe(listener) {\n      source.listeners.add(listener);\n      return () => source.listeners.delete(listener);\n    },\n    getSnapshot() {\n      return source.snapshot;\n    },\n    reconcile(controller, subagents, uiMessagesByParent) {\n      source.uiMessagesByParent = uiMessagesByParent;\n      // A resource is rebuilt from scratch when its namespace resolves, which\n      // routinely happens after the subagent finished. Its timing was measured\n      // while the client watched the task run and cannot be measured again.\n      const rebound = new Map<string, MeasuredTiming>();\n\n      if (source.controller !== controller) {\n        source.dispose();\n        source.controller = controller;\n      }\n\n      for (const id of source.namespaceRequests.keys()) {\n        if (!subagents.has(id)) source.namespaceRequests.delete(id);\n      }\n\n      for (const [id, resource] of source.resources) {\n        const snapshot = subagents.get(id);\n        if (\n          !snapshot ||\n          snapshot.depth > MAX_SUBAGENT_DEPTH ||\n          !sameNamespace(resource.namespace, snapshot.namespace)\n        ) {\n          if (snapshot && snapshot.depth <= MAX_SUBAGENT_DEPTH)\n            rebound.set(id, {\n              timing: resource.timing,\n              timingState: resource.timingState,\n              messageTiming: resource.messageTiming,\n            });\n          resource.dispose();\n          source.resources.delete(id);\n          continue;\n        }\n        resource.snapshot = snapshot;\n      }\n\n      for (const snapshot of subagents.values()) {\n        if (snapshot.depth > MAX_SUBAGENT_DEPTH) continue;\n        if (!needsNamespaceResolution(snapshot)) {\n          source.namespaceRequests.delete(snapshot.id);\n        } else {\n          const request = source.namespaceRequests.get(snapshot.id);\n          if (!request) {\n            const nextRequest: NamespaceRequest = {\n              id: snapshot.id,\n              attempts: 0,\n              pending: false,\n              retryQueued: false,\n              status: snapshot.status,\n            };\n            source.namespaceRequests.set(snapshot.id, nextRequest);\n            requestSubagentNamespace(source, controller, nextRequest);\n          } else if (request.status !== snapshot.status) {\n            request.status = snapshot.status;\n            if (request.pending) {\n              request.retryQueued = request.attempts < 2;\n            } else if (request.attempts < 2) {\n              requestSubagentNamespace(source, controller, request);\n            }\n          }\n        }\n        if (source.resources.has(snapshot.id)) continue;\n        const acquired = controller.registry.acquire(\n          messagesProjection(snapshot.namespace),\n        );\n        const acquiredUI =\n          snapshot.namespace.length > ROOT_UI_CHANNEL_DEPTH\n            ? controller.registry.acquire(\n                channelProjection(UI_CUSTOM_CHANNELS, snapshot.namespace),\n              )\n            : undefined;\n        const resource: ProjectionResource = {\n          snapshot,\n          namespace: snapshot.namespace,\n          store: acquired.store,\n          uiStore: acquiredUI?.store,\n          dispose: () => {},\n          storeSnapshot: undefined,\n          status: undefined,\n          uiFoldMemo: createUIFoldMemo(),\n          localUiMessages: NO_UI_MESSAGES,\n          rootUiMessagesByParent: source.uiMessagesByParent,\n          uiMessagesByParent: source.uiMessagesByParent,\n          timingState: rebound.get(snapshot.id)?.timingState ?? null,\n          messageTiming:\n            rebound.get(snapshot.id)?.messageTiming ?? NO_MESSAGE_TIMING,\n          convertedMessageTiming: undefined,\n          convert: createConverter(\n            source.uiMessagesByParent,\n            NO_MESSAGE_TIMING,\n          ),\n          uiMessages: [],\n          converted: undefined,\n          childTranscripts: undefined,\n          transcript: undefined,\n          timing: rebound.has(snapshot.id)\n            ? rebound.get(snapshot.id)!.timing\n            : snapshot.status === \"running\"\n              ? { startedAt: snapshot.startedAt.getTime() }\n              : undefined,\n          entry: undefined,\n          memo: createAttachMemo(),\n          cache: createExternalMessageConversionCache(),\n        };\n        const unsubscribe = resource.store.subscribe(() => rebuild());\n        const unsubscribeUI = resource.uiStore?.subscribe(() => rebuild());\n        resource.dispose = () => {\n          unsubscribe();\n          unsubscribeUI?.();\n          acquired.release();\n          acquiredUI?.release();\n        };\n        source.resources.set(snapshot.id, resource);\n      }\n\n      rebuild();\n    },\n    dispose() {\n      for (const resource of source.resources.values()) resource.dispose();\n      source.resources.clear();\n      source.namespaceRequests.clear();\n      source.snapshot = new Map();\n      for (const listener of source.listeners) listener();\n    },\n  };\n\n  const rebuild = () => {\n    const { uiMessagesByParent } = source;\n    const resources = [...source.resources.values()];\n    const childrenByParent = new Map<string, ProjectionResource[]>();\n\n    for (const resource of resources) {\n      const parentId = resource.snapshot.parentId;\n      if (parentId == null) continue;\n      const children = childrenByParent.get(parentId);\n      if (children) children.push(resource);\n      else childrenByParent.set(parentId, [resource]);\n    }\n\n    const transcripts = new Map<string, SubagentTranscript>();\n    let changed = source.snapshot.size !== resources.length;\n    const built = new Set<string>();\n\n    const build = (resource: ProjectionResource, depth: number) => {\n      if (built.has(resource.snapshot.id)) return;\n      built.add(resource.snapshot.id);\n      const children =\n        depth < MAX_SUBAGENT_DEPTH\n          ? (childrenByParent.get(resource.snapshot.id) ?? [])\n          : [];\n      for (const child of children) build(child, depth + 1);\n      const childTranscripts = new Map(\n        children.flatMap((child) =>\n          child.entry ? [[child.snapshot.id, child.entry] as const] : [],\n        ),\n      );\n      const storeSnapshot = resource.store.getSnapshot();\n      const status = resource.snapshot.status;\n      const stepped = stepStreamingTiming(\n        resource.timingState,\n        storeSnapshot as LangChainBaseMessage[],\n        status === \"running\",\n        langChainStreamingTimingAccessors,\n        undefined,\n      );\n      resource.timingState = stepped.state;\n      if (Object.keys(stepped.timings).length > 0) {\n        resource.messageTiming = {\n          ...resource.messageTiming,\n          ...stepped.timings,\n        };\n      }\n      const localUiMessages = foldLocalUIMessages(resource);\n      if (\n        resource.localUiMessages !== localUiMessages ||\n        resource.rootUiMessagesByParent !== uiMessagesByParent ||\n        resource.convertedMessageTiming !== resource.messageTiming\n      ) {\n        resource.localUiMessages = localUiMessages;\n        resource.rootUiMessagesByParent = uiMessagesByParent;\n        resource.uiMessagesByParent =\n          localUiMessages.length === 0\n            ? uiMessagesByParent\n            : mergeLocalUIMessages(uiMessagesByParent, localUiMessages);\n        resource.convert = createConverter(\n          resource.uiMessagesByParent,\n          resource.messageTiming,\n        );\n      }\n      const uiMessages = collectUIMessages(\n        storeSnapshot,\n        resource.uiMessagesByParent,\n      );\n\n      const conversionChanged =\n        resource.converted === undefined ||\n        resource.storeSnapshot !== storeSnapshot ||\n        resource.status !== status ||\n        resource.convertedMessageTiming !== resource.messageTiming ||\n        !sameUIMessages(resource.uiMessages, uiMessages);\n      if (conversionChanged) {\n        resource.converted = convertExternalMessages(\n          storeSnapshot as LangChainBaseMessage[],\n          resource.convert,\n          status === \"running\",\n          TRANSCRIPT_METADATA,\n          resource.cache,\n        );\n        resource.storeSnapshot = storeSnapshot;\n        resource.status = status;\n        resource.uiMessages = uiMessages;\n        resource.convertedMessageTiming = resource.messageTiming;\n      }\n\n      if (\n        resource.transcript === undefined ||\n        conversionChanged ||\n        !sameTranscriptEntries(resource.childTranscripts, childTranscripts)\n      ) {\n        const transcript = attachSubagentTranscripts(\n          resource.converted!,\n          childTranscripts,\n          resource.memo,\n        );\n        changed ||= resource.transcript !== transcript;\n        resource.transcript = transcript;\n        resource.childTranscripts = childTranscripts;\n      }\n\n      updateTiming(resource);\n      if (\n        resource.entry === undefined ||\n        resource.entry.messages !== resource.transcript ||\n        resource.entry.timing !== resource.timing\n      ) {\n        resource.entry = {\n          messages: resource.transcript!,\n          ...(resource.timing && { timing: resource.timing }),\n        };\n        changed = true;\n      }\n\n      if (!source.snapshot.has(resource.snapshot.id)) changed = true;\n      transcripts.set(resource.snapshot.id, resource.entry);\n    };\n\n    for (const resource of resources) {\n      if (\n        resource.snapshot.parentId == null ||\n        !source.resources.has(resource.snapshot.parentId)\n      )\n        build(resource, 1);\n    }\n    for (const resource of resources) build(resource, MAX_SUBAGENT_DEPTH);\n\n    if (!changed) return;\n    source.snapshot = transcripts;\n    for (const listener of source.listeners) listener();\n  };\n\n  return source;\n};\n\nexport const useSubagentTranscripts = (\n  stream: AnyStream,\n  uiMessagesByParent: Map<string, UIMessage[]>,\n): ReadonlyMap<string, SubagentTranscript> => {\n  const sourceRef = useRef<SubagentTranscriptSource | undefined>(undefined);\n  if (!sourceRef.current) {\n    sourceRef.current = createSubagentTranscriptSource();\n  }\n  const source = sourceRef.current;\n  const controller = stream[STREAM_CONTROLLER];\n\n  useEffect(() => {\n    source.reconcile(controller, stream.subagents, uiMessagesByParent);\n  }, [controller, source, stream.subagents, uiMessagesByParent]);\n\n  useEffect(() => () => source.dispose(), [source]);\n\n  return useSyncExternalStore(\n    source.subscribe,\n    source.getSnapshot,\n    source.getSnapshot,\n  );\n};\n"],"mappings":";;;;;;;;;;;;AA0CA,MAAa,qBAAqB;AAElC,MAAM,wBAAwB;AAE9B,MAAM,sBAAsB,CAAC;AAE7B,MAAM,iBAAuC,CAAC;AAE9C,MAAM,oBAAmD,CAAC;AAgE1D,MAAM,iBAAiB,GAAsB,MAC3C,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,SAAS,UAAU,YAAY,EAAE,MAAM;AAE3E,MAAM,4BAA4B,aAChC,SAAS,UAAU,WAAW,KAC9B,SAAS,UAAU,OAAO,SAAS,SAAS;AAE9C,MAAM,4BACJ,QACA,YACA,YACG;CACH,QAAQ,YAAY;CACpB,QAAQ,UAAU;CAClB,QAAQ,cAAc;CACtB,WACG,yBAAyB,QAAQ,EAAE,CAAC,CACpC,YAAY,CAAC,CAAC,CAAC,CACf,cAAc;EACb,IACE,OAAO,eAAe,cACtB,OAAO,kBAAkB,IAAI,QAAQ,EAAE,MAAM,SAE7C;EAEF,QAAQ,UAAU;EAClB,IACE,CAAC,QAAQ,eACT,QAAQ,aAAa,KACrB,QAAQ,WAAW,WAEnB,QAAQ,cAAc;EAExB,IAAI,CAAC,QAAQ,eAAe,QAAQ,YAAY,GAAG;GACjD,QAAQ,cAAc;GACtB;EACF;EACA,yBAAyB,QAAQ,YAAY,OAAO;CACtD,CAAC;AACL;AAEA,MAAM,yBACJ,GACA,MAEA,GAAG,SAAS,EAAE,QACd,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,gBAAgB,EAAE,IAAI,EAAE,MAAM,UAAU;AAE7D,MAAM,qBACJ,UACA,uBACG;CACH,MAAM,YAAyB,CAAC;CAChC,IAAI,mBAAmB,SAAS,GAAG,OAAO;CAC1C,KAAK,MAAM,WAAW,UAAU;EAC9B,MAAM,aAAa,QAAQ,MAAM,mBAAmB,IAAI,QAAQ,EAAE;EAClE,IAAI,YAAY,UAAU,KAAK,GAAG,UAAU;CAC9C;CACA,OAAO;AACT;AAEA,MAAM,kBAAkB,GAAyB,MAC/C,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,IAAI,UAAU,OAAO,EAAE,MAAM;AAEjE,MAAM,uBAAuB,aAAiC;CAC5D,MAAM,SAAS,SAAS,SAAS,YAAY;CAC7C,IAAI,WAAW,KAAA,GAAW,OAAO;CACjC,OAAO,cAAc,QAAQ,SAAS,UAAU;AAClD;AAEA,MAAM,wBACJ,oBACA,oBACG;CACH,MAAM,SAAS,IAAI,IAAI,kBAAkB;CACzC,KAAK,MAAM,CAAC,UAAU,aAAa,wBACjC,eACF,GACE,OAAO,IAAI,UAAU,gBAAgB,UAAU,OAAO,IAAI,QAAQ,CAAC,CAAC;CAEtE,OAAO;AACT;AAEA,MAAM,mBAEF,oBACA,mBAED,SAAS,aACR,4BAA4B,SAAS;CACnC,GAAG;CACH;CACA;AACF,CAAC;AAEL,MAAM,gBAAgB,aAAiC;CACrD,IAAI,SAAS,WAAW,KAAA,GAAW;CACnC,MAAM,YAAY,SAAS,SAAS,UAAU,QAAQ;CACtD,MAAM,cAAc,SAAS,SAAS,aAAa,QAAQ;CAC3D,IACE,SAAS,OAAO,cAAc,aAC9B,SAAS,OAAO,gBAAgB,aAEhC;CACF,SAAS,SAAS;EAChB;EACA,GAAI,gBAAgB,KAAA,KAAa,EAAE,YAAY;CACjD;AACF;AAEA,MAAM,uCAAiE;CAErE,MAAM,SAAmC;EACvC,2BAAW,IAAI,IAAI;EACnB,mCAAmB,IAAI,IAAI;EAC3B,0BAAU,IAAI,IAAI;EAClB,2BAAW,IAAI,IAAI;EACnB,YAAY,KAAA;EACZ,oCAAA,IAP6B,IAOZ;EACjB,UAAU,UAAU;GAClB,OAAO,UAAU,IAAI,QAAQ;GAC7B,aAAa,OAAO,UAAU,OAAO,QAAQ;EAC/C;EACA,cAAc;GACZ,OAAO,OAAO;EAChB;EACA,UAAU,YAAY,WAAW,oBAAoB;GACnD,OAAO,qBAAqB;GAI5B,MAAM,0BAAU,IAAI,IAA4B;GAEhD,IAAI,OAAO,eAAe,YAAY;IACpC,OAAO,QAAQ;IACf,OAAO,aAAa;GACtB;GAEA,KAAK,MAAM,MAAM,OAAO,kBAAkB,KAAK,GAC7C,IAAI,CAAC,UAAU,IAAI,EAAE,GAAG,OAAO,kBAAkB,OAAO,EAAE;GAG5D,KAAK,MAAM,CAAC,IAAI,aAAa,OAAO,WAAW;IAC7C,MAAM,WAAW,UAAU,IAAI,EAAE;IACjC,IACE,CAAC,YACD,SAAS,QAAA,MACT,CAAC,cAAc,SAAS,WAAW,SAAS,SAAS,GACrD;KACA,IAAI,YAAY,SAAS,SAAA,IACvB,QAAQ,IAAI,IAAI;MACd,QAAQ,SAAS;MACjB,aAAa,SAAS;MACtB,eAAe,SAAS;KAC1B,CAAC;KACH,SAAS,QAAQ;KACjB,OAAO,UAAU,OAAO,EAAE;KAC1B;IACF;IACA,SAAS,WAAW;GACtB;GAEA,KAAK,MAAM,YAAY,UAAU,OAAO,GAAG;IACzC,IAAI,SAAS,QAAA,IAA4B;IACzC,IAAI,CAAC,yBAAyB,QAAQ,GACpC,OAAO,kBAAkB,OAAO,SAAS,EAAE;SACtC;KACL,MAAM,UAAU,OAAO,kBAAkB,IAAI,SAAS,EAAE;KACxD,IAAI,CAAC,SAAS;MACZ,MAAM,cAAgC;OACpC,IAAI,SAAS;OACb,UAAU;OACV,SAAS;OACT,aAAa;OACb,QAAQ,SAAS;MACnB;MACA,OAAO,kBAAkB,IAAI,SAAS,IAAI,WAAW;MACrD,yBAAyB,QAAQ,YAAY,WAAW;KAC1D,OAAO,IAAI,QAAQ,WAAW,SAAS,QAAQ;MAC7C,QAAQ,SAAS,SAAS;MAC1B,IAAI,QAAQ,SACV,QAAQ,cAAc,QAAQ,WAAW;WACpC,IAAI,QAAQ,WAAW,GAC5B,yBAAyB,QAAQ,YAAY,OAAO;KAExD;IACF;IACA,IAAI,OAAO,UAAU,IAAI,SAAS,EAAE,GAAG;IACvC,MAAM,WAAW,WAAW,SAAS,QACnC,mBAAmB,SAAS,SAAS,CACvC;IACA,MAAM,aACJ,SAAS,UAAU,SAAS,wBACxB,WAAW,SAAS,QAClB,kBAAkB,oBAAoB,SAAS,SAAS,CAC1D,IACA,KAAA;IACN,MAAM,WAA+B;KACnC;KACA,WAAW,SAAS;KACpB,OAAO,SAAS;KAChB,SAAS,YAAY;KACrB,eAAe,CAAC;KAChB,eAAe,KAAA;KACf,QAAQ,KAAA;KACR,YAAY,iBAAiB;KAC7B,iBAAiB;KACjB,wBAAwB,OAAO;KAC/B,oBAAoB,OAAO;KAC3B,aAAa,QAAQ,IAAI,SAAS,EAAE,CAAC,EAAE,eAAe;KACtD,eACE,QAAQ,IAAI,SAAS,EAAE,CAAC,EAAE,iBAAiB;KAC7C,wBAAwB,KAAA;KACxB,SAAS,gBACP,OAAO,oBACP,iBACF;KACA,YAAY,CAAC;KACb,WAAW,KAAA;KACX,kBAAkB,KAAA;KAClB,YAAY,KAAA;KACZ,QAAQ,QAAQ,IAAI,SAAS,EAAE,IAC3B,QAAQ,IAAI,SAAS,EAAE,CAAC,CAAE,SAC1B,SAAS,WAAW,YAClB,EAAE,WAAW,SAAS,UAAU,QAAQ,EAAE,IAC1C,KAAA;KACN,OAAO,KAAA;KACP,MAAM,iBAAiB;KACvB,OAAO,qCAAqC;IAC9C;IACA,MAAM,cAAc,SAAS,MAAM,gBAAgB,QAAQ,CAAC;IAC5D,MAAM,gBAAgB,SAAS,SAAS,gBAAgB,QAAQ,CAAC;IACjE,SAAS,gBAAgB;KACvB,YAAY;KACZ,gBAAgB;KAChB,SAAS,QAAQ;KACjB,YAAY,QAAQ;IACtB;IACA,OAAO,UAAU,IAAI,SAAS,IAAI,QAAQ;GAC5C;GAEA,QAAQ;EACV;EACA,UAAU;GACR,KAAK,MAAM,YAAY,OAAO,UAAU,OAAO,GAAG,SAAS,QAAQ;GACnE,OAAO,UAAU,MAAM;GACvB,OAAO,kBAAkB,MAAM;GAC/B,OAAO,2BAAW,IAAI,IAAI;GAC1B,KAAK,MAAM,YAAY,OAAO,WAAW,SAAS;EACpD;CACF;CAEA,MAAM,gBAAgB;EACpB,MAAM,EAAE,uBAAuB;EAC/B,MAAM,YAAY,CAAC,GAAG,OAAO,UAAU,OAAO,CAAC;EAC/C,MAAM,mCAAmB,IAAI,IAAkC;EAE/D,KAAK,MAAM,YAAY,WAAW;GAChC,MAAM,WAAW,SAAS,SAAS;GACnC,IAAI,YAAY,MAAM;GACtB,MAAM,WAAW,iBAAiB,IAAI,QAAQ;GAC9C,IAAI,UAAU,SAAS,KAAK,QAAQ;QAC/B,iBAAiB,IAAI,UAAU,CAAC,QAAQ,CAAC;EAChD;EAEA,MAAM,8BAAc,IAAI,IAAgC;EACxD,IAAI,UAAU,OAAO,SAAS,SAAS,UAAU;EACjD,MAAM,wBAAQ,IAAI,IAAY;EAE9B,MAAM,SAAS,UAA8B,UAAkB;GAC7D,IAAI,MAAM,IAAI,SAAS,SAAS,EAAE,GAAG;GACrC,MAAM,IAAI,SAAS,SAAS,EAAE;GAC9B,MAAM,WACJ,QAAA,KACK,iBAAiB,IAAI,SAAS,SAAS,EAAE,KAAK,CAAC,IAChD,CAAC;GACP,KAAK,MAAM,SAAS,UAAU,MAAM,OAAO,QAAQ,CAAC;GACpD,MAAM,mBAAmB,IAAI,IAC3B,SAAS,SAAS,UAChB,MAAM,QAAQ,CAAC,CAAC,MAAM,SAAS,IAAI,MAAM,KAAK,CAAU,IAAI,CAAC,CAC/D,CACF;GACA,MAAM,gBAAgB,SAAS,MAAM,YAAY;GACjD,MAAM,SAAS,SAAS,SAAS;GACjC,MAAM,UAAU,oBACd,SAAS,aACT,eACA,WAAW,WACX,mCACA,KAAA,CACF;GACA,SAAS,cAAc,QAAQ;GAC/B,IAAI,OAAO,KAAK,QAAQ,OAAO,CAAC,CAAC,SAAS,GACxC,SAAS,gBAAgB;IACvB,GAAG,SAAS;IACZ,GAAG,QAAQ;GACb;GAEF,MAAM,kBAAkB,oBAAoB,QAAQ;GACpD,IACE,SAAS,oBAAoB,mBAC7B,SAAS,2BAA2B,sBACpC,SAAS,2BAA2B,SAAS,eAC7C;IACA,SAAS,kBAAkB;IAC3B,SAAS,yBAAyB;IAClC,SAAS,qBACP,gBAAgB,WAAW,IACvB,qBACA,qBAAqB,oBAAoB,eAAe;IAC9D,SAAS,UAAU,gBACjB,SAAS,oBACT,SAAS,aACX;GACF;GACA,MAAM,aAAa,kBACjB,eACA,SAAS,kBACX;GAEA,MAAM,oBACJ,SAAS,cAAc,KAAA,KACvB,SAAS,kBAAkB,iBAC3B,SAAS,WAAW,UACpB,SAAS,2BAA2B,SAAS,iBAC7C,CAAC,eAAe,SAAS,YAAY,UAAU;GACjD,IAAI,mBAAmB;IACrB,SAAS,YAAY,wBACnB,eACA,SAAS,SACT,WAAW,WACX,qBACA,SAAS,KACX;IACA,SAAS,gBAAgB;IACzB,SAAS,SAAS;IAClB,SAAS,aAAa;IACtB,SAAS,yBAAyB,SAAS;GAC7C;GAEA,IACE,SAAS,eAAe,KAAA,KACxB,qBACA,CAAC,sBAAsB,SAAS,kBAAkB,gBAAgB,GAClE;IACA,MAAM,aAAa,0BACjB,SAAS,WACT,kBACA,SAAS,IACX;IACA,YAAY,SAAS,eAAe;IACpC,SAAS,aAAa;IACtB,SAAS,mBAAmB;GAC9B;GAEA,aAAa,QAAQ;GACrB,IACE,SAAS,UAAU,KAAA,KACnB,SAAS,MAAM,aAAa,SAAS,cACrC,SAAS,MAAM,WAAW,SAAS,QACnC;IACA,SAAS,QAAQ;KACf,UAAU,SAAS;KACnB,GAAI,SAAS,UAAU,EAAE,QAAQ,SAAS,OAAO;IACnD;IACA,UAAU;GACZ;GAEA,IAAI,CAAC,OAAO,SAAS,IAAI,SAAS,SAAS,EAAE,GAAG,UAAU;GAC1D,YAAY,IAAI,SAAS,SAAS,IAAI,SAAS,KAAK;EACtD;EAEA,KAAK,MAAM,YAAY,WACrB,IACE,SAAS,SAAS,YAAY,QAC9B,CAAC,OAAO,UAAU,IAAI,SAAS,SAAS,QAAQ,GAEhD,MAAM,UAAU,CAAC;EAErB,KAAK,MAAM,YAAY,WAAW,MAAM,UAAA,EAA4B;EAEpE,IAAI,CAAC,SAAS;EACd,OAAO,WAAW;EAClB,KAAK,MAAM,YAAY,OAAO,WAAW,SAAS;CACpD;CAEA,OAAO;AACT;AAEA,MAAa,0BACX,QACA,uBAC4C;CAC5C,MAAM,YAAY,OAA6C,KAAA,CAAS;CACxE,IAAI,CAAC,UAAU,SACb,UAAU,UAAU,+BAA+B;CAErD,MAAM,SAAS,UAAU;CACzB,MAAM,aAAa,OAAO;CAE1B,gBAAgB;EACd,OAAO,UAAU,YAAY,OAAO,WAAW,kBAAkB;CACnE,GAAG;EAAC;EAAY;EAAQ,OAAO;EAAW;CAAkB,CAAC;CAE7D,sBAAsB,OAAO,QAAQ,GAAG,CAAC,MAAM,CAAC;CAEhD,OAAO,qBACL,OAAO,WACP,OAAO,aACP,OAAO,WACT;AACF"}
+{"version":3,"file":"useSubagentTranscripts.js","names":[],"sources":["../src/useSubagentTranscripts.ts"],"sourcesContent":["\"use client\";\n\nimport { useEffect, useRef, useSyncExternalStore } from \"react\";\nimport {\n  stepStreamingTiming,\n  type MessageTiming,\n  type StreamingTimingState,\n  type ThreadMessage,\n  type ToolCallTiming,\n} from \"@assistant-ui/core\";\nimport {\n  convertExternalMessages,\n  createExternalMessageConversionCache,\n  type ExternalMessageConversionCache,\n  type useExternalMessageConverter,\n} from \"@assistant-ui/core/react\";\nimport { useReplaySafeEffect } from \"@assistant-ui/store/internal\";\nimport { STREAM_CONTROLLER, type AnyStream } from \"@langchain/react\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\nimport { channelProjection, type Event } from \"@langchain/langgraph-sdk/stream\";\nimport type { SubagentDiscoverySnapshot } from \"@langchain/react\";\nimport {\n  attachSubagentTranscripts,\n  type AttachMemo,\n  createAttachMemo,\n  type SubagentTranscript,\n} from \"./attachSubagentTranscripts\";\nimport { convertLangChainBaseMessage } from \"./convertMessages\";\nimport { groupUIMessagesByParent } from \"./converter\";\nimport { langChainStreamingTimingAccessors } from \"./streamingTiming\";\nimport { subagentMessagesProjection } from \"./subagentMessagesProjection\";\nimport type { LangChainBaseMessage, UIMessage } from \"./types\";\nimport {\n  createUIFoldMemo,\n  foldUIUpdates,\n  mergeUIMessages,\n  UI_CUSTOM_CHANNELS,\n  type UIFoldMemo,\n} from \"./uiMessages\";\n\nexport const MAX_SUBAGENT_DEPTH = 16;\n\nconst ROOT_UI_CHANNEL_DEPTH = 1;\n\nconst TRANSCRIPT_METADATA = {};\n\nconst NO_UI_MESSAGES: readonly UIMessage[] = [];\n\nconst NO_MESSAGE_TIMING: Record<string, MessageTiming> = {};\n\ntype ProjectionStore<T> = {\n  getSnapshot(): T;\n  subscribe(listener: () => void): () => void;\n};\n\ntype ProjectionResource = {\n  snapshot: SubagentDiscoverySnapshot;\n  namespace: readonly string[];\n  store: ProjectionStore<BaseMessage[]>;\n  uiStore: ProjectionStore<readonly Event[]> | undefined;\n  dispose: () => void;\n  storeSnapshot: BaseMessage[] | undefined;\n  status: SubagentDiscoverySnapshot[\"status\"] | undefined;\n  uiFoldMemo: UIFoldMemo;\n  localUiMessages: readonly UIMessage[];\n  rootUiMessagesByParent: Map<string, UIMessage[]>;\n  uiMessagesByParent: Map<string, UIMessage[]>;\n  timingState: StreamingTimingState | null;\n  messageTiming: Record<string, MessageTiming>;\n  convertedMessageTiming: Record<string, MessageTiming> | undefined;\n  convert: useExternalMessageConverter.Callback<LangChainBaseMessage>;\n  uiMessages: readonly UIMessage[];\n  converted: readonly ThreadMessage[] | undefined;\n  childTranscripts: ReadonlyMap<string, SubagentTranscript> | undefined;\n  transcript: readonly ThreadMessage[] | undefined;\n  timing: ToolCallTiming | undefined;\n  entry: SubagentTranscript | undefined;\n  memo: AttachMemo;\n  cache: ExternalMessageConversionCache;\n};\n\ntype MeasuredTiming = {\n  timing: ToolCallTiming | undefined;\n  timingState: StreamingTimingState | null;\n  messageTiming: Record<string, MessageTiming>;\n};\n\ntype NamespaceRequest = {\n  id: string;\n  attempts: number;\n  pending: boolean;\n  retryQueued: boolean;\n  status: SubagentDiscoverySnapshot[\"status\"];\n};\n\ntype SubagentTranscriptSource = {\n  resources: Map<string, ProjectionResource>;\n  namespaceRequests: Map<string, NamespaceRequest>;\n  snapshot: ReadonlyMap<string, SubagentTranscript>;\n  listeners: Set<() => void>;\n  controller: AnyStream[typeof STREAM_CONTROLLER] | undefined;\n  uiMessagesByParent: Map<string, UIMessage[]>;\n  subscribe(listener: () => void): () => void;\n  getSnapshot(): ReadonlyMap<string, SubagentTranscript>;\n  reconcile(\n    controller: AnyStream[typeof STREAM_CONTROLLER],\n    subagents: AnyStream[\"subagents\"],\n    uiMessagesByParent: Map<string, UIMessage[]>,\n  ): void;\n  dispose(): void;\n};\n\nconst sameNamespace = (a: readonly string[], b: readonly string[]) =>\n  a.length === b.length && a.every((segment, index) => segment === b[index]);\n\nconst needsNamespaceResolution = (snapshot: SubagentDiscoverySnapshot) =>\n  snapshot.namespace.length === 1 &&\n  snapshot.namespace[0] === `tools:${snapshot.id}`;\n\nconst requestSubagentNamespace = (\n  source: SubagentTranscriptSource,\n  controller: AnyStream[typeof STREAM_CONTROLLER],\n  request: NamespaceRequest,\n) => {\n  request.attempts += 1;\n  request.pending = true;\n  request.retryQueued = false;\n  void controller\n    .resolveSubagentNamespace(request.id)\n    .catch(() => {})\n    .finally(() => {\n      if (\n        source.controller !== controller ||\n        source.namespaceRequests.get(request.id) !== request\n      )\n        return;\n\n      request.pending = false;\n      if (\n        !request.retryQueued &&\n        request.attempts === 1 &&\n        request.status !== \"running\"\n      ) {\n        request.retryQueued = true;\n      }\n      if (!request.retryQueued || request.attempts >= 2) {\n        request.retryQueued = false;\n        return;\n      }\n      requestSubagentNamespace(source, controller, request);\n    });\n};\n\nconst sameTranscriptEntries = (\n  a: ReadonlyMap<string, SubagentTranscript> | undefined,\n  b: ReadonlyMap<string, SubagentTranscript>,\n) =>\n  a?.size === b.size &&\n  [...b].every(([id, transcript]) => a.get(id) === transcript);\n\nconst collectUIMessages = (\n  messages: readonly BaseMessage[],\n  uiMessagesByParent: Map<string, UIMessage[]>,\n) => {\n  const collected: UIMessage[] = [];\n  if (uiMessagesByParent.size === 0) return collected;\n  for (const message of messages) {\n    const uiMessages = message.id && uiMessagesByParent.get(message.id);\n    if (uiMessages) collected.push(...uiMessages);\n  }\n  return collected;\n};\n\nconst sameUIMessages = (a: readonly UIMessage[], b: readonly UIMessage[]) =>\n  a.length === b.length && a.every((ui, index) => ui === b[index]);\n\nconst foldLocalUIMessages = (resource: ProjectionResource) => {\n  const events = resource.uiStore?.getSnapshot();\n  if (events === undefined) return NO_UI_MESSAGES;\n  return foldUIUpdates(events, resource.uiFoldMemo);\n};\n\nconst mergeLocalUIMessages = (\n  uiMessagesByParent: Map<string, UIMessage[]>,\n  localUiMessages: readonly UIMessage[],\n) => {\n  const merged = new Map(uiMessagesByParent);\n  for (const [parentId, messages] of groupUIMessagesByParent<UIMessage>(\n    localUiMessages,\n  )) {\n    merged.set(parentId, mergeUIMessages(messages, merged.get(parentId)));\n  }\n  return merged;\n};\n\nconst createConverter =\n  (\n    uiMessagesByParent: Map<string, UIMessage[]>,\n    messageTiming: Record<string, MessageTiming>,\n  ): useExternalMessageConverter.Callback<LangChainBaseMessage> =>\n  (message, metadata) =>\n    convertLangChainBaseMessage(message, {\n      ...metadata,\n      uiMessagesByParent,\n      messageTiming,\n    });\n\nconst updateTiming = (resource: ProjectionResource) => {\n  if (resource.timing === undefined) return;\n  const startedAt = resource.snapshot.startedAt.getTime();\n  const completedAt = resource.snapshot.completedAt?.getTime();\n  if (\n    resource.timing.startedAt === startedAt &&\n    resource.timing.completedAt === completedAt\n  )\n    return;\n  resource.timing = {\n    startedAt,\n    ...(completedAt !== undefined && { completedAt }),\n  };\n};\n\nconst createSubagentTranscriptSource = (): SubagentTranscriptSource => {\n  const uiMessagesByParent = new Map<string, UIMessage[]>();\n  const source: SubagentTranscriptSource = {\n    resources: new Map(),\n    namespaceRequests: new Map(),\n    snapshot: new Map(),\n    listeners: new Set(),\n    controller: undefined,\n    uiMessagesByParent,\n    subscribe(listener) {\n      source.listeners.add(listener);\n      return () => source.listeners.delete(listener);\n    },\n    getSnapshot() {\n      return source.snapshot;\n    },\n    reconcile(controller, subagents, uiMessagesByParent) {\n      source.uiMessagesByParent = uiMessagesByParent;\n      // A resource is rebuilt from scratch when its namespace resolves, which\n      // routinely happens after the subagent finished. Its timing was measured\n      // while the client watched the task run and cannot be measured again.\n      const rebound = new Map<string, MeasuredTiming>();\n\n      if (source.controller !== controller) {\n        source.dispose();\n        source.controller = controller;\n      }\n\n      for (const id of source.namespaceRequests.keys()) {\n        if (!subagents.has(id)) source.namespaceRequests.delete(id);\n      }\n\n      for (const [id, resource] of source.resources) {\n        const snapshot = subagents.get(id);\n        if (\n          !snapshot ||\n          snapshot.depth > MAX_SUBAGENT_DEPTH ||\n          !sameNamespace(resource.namespace, snapshot.namespace)\n        ) {\n          if (snapshot && snapshot.depth <= MAX_SUBAGENT_DEPTH)\n            rebound.set(id, {\n              timing: resource.timing,\n              timingState: resource.timingState,\n              messageTiming: resource.messageTiming,\n            });\n          resource.dispose();\n          source.resources.delete(id);\n          continue;\n        }\n        resource.snapshot = snapshot;\n      }\n\n      for (const snapshot of subagents.values()) {\n        if (snapshot.depth > MAX_SUBAGENT_DEPTH) continue;\n        if (!needsNamespaceResolution(snapshot)) {\n          source.namespaceRequests.delete(snapshot.id);\n        } else {\n          const request = source.namespaceRequests.get(snapshot.id);\n          if (!request) {\n            const nextRequest: NamespaceRequest = {\n              id: snapshot.id,\n              attempts: 0,\n              pending: false,\n              retryQueued: false,\n              status: snapshot.status,\n            };\n            source.namespaceRequests.set(snapshot.id, nextRequest);\n            requestSubagentNamespace(source, controller, nextRequest);\n          } else if (request.status !== snapshot.status) {\n            request.status = snapshot.status;\n            if (request.pending) {\n              request.retryQueued = request.attempts < 2;\n            } else if (request.attempts < 2) {\n              requestSubagentNamespace(source, controller, request);\n            }\n          }\n        }\n        if (source.resources.has(snapshot.id)) continue;\n        const acquired = controller.registry.acquire(\n          subagentMessagesProjection(snapshot.namespace),\n        );\n        const acquiredUI =\n          snapshot.namespace.length > ROOT_UI_CHANNEL_DEPTH\n            ? controller.registry.acquire(\n                channelProjection(UI_CUSTOM_CHANNELS, snapshot.namespace),\n              )\n            : undefined;\n        const resource: ProjectionResource = {\n          snapshot,\n          namespace: snapshot.namespace,\n          store: acquired.store,\n          uiStore: acquiredUI?.store,\n          dispose: () => {},\n          storeSnapshot: undefined,\n          status: undefined,\n          uiFoldMemo: createUIFoldMemo(),\n          localUiMessages: NO_UI_MESSAGES,\n          rootUiMessagesByParent: source.uiMessagesByParent,\n          uiMessagesByParent: source.uiMessagesByParent,\n          timingState: rebound.get(snapshot.id)?.timingState ?? null,\n          messageTiming:\n            rebound.get(snapshot.id)?.messageTiming ?? NO_MESSAGE_TIMING,\n          convertedMessageTiming: undefined,\n          convert: createConverter(\n            source.uiMessagesByParent,\n            NO_MESSAGE_TIMING,\n          ),\n          uiMessages: [],\n          converted: undefined,\n          childTranscripts: undefined,\n          transcript: undefined,\n          timing: rebound.has(snapshot.id)\n            ? rebound.get(snapshot.id)!.timing\n            : snapshot.status === \"running\"\n              ? { startedAt: snapshot.startedAt.getTime() }\n              : undefined,\n          entry: undefined,\n          memo: createAttachMemo(),\n          cache: createExternalMessageConversionCache(),\n        };\n        const unsubscribe = resource.store.subscribe(() => rebuild());\n        const unsubscribeUI = resource.uiStore?.subscribe(() => rebuild());\n        resource.dispose = () => {\n          unsubscribe();\n          unsubscribeUI?.();\n          acquired.release();\n          acquiredUI?.release();\n        };\n        source.resources.set(snapshot.id, resource);\n      }\n\n      rebuild();\n    },\n    dispose() {\n      for (const resource of source.resources.values()) resource.dispose();\n      source.resources.clear();\n      source.namespaceRequests.clear();\n      source.snapshot = new Map();\n      for (const listener of source.listeners) listener();\n    },\n  };\n\n  const rebuild = () => {\n    const { uiMessagesByParent } = source;\n    const resources = [...source.resources.values()];\n    const childrenByParent = new Map<string, ProjectionResource[]>();\n\n    for (const resource of resources) {\n      const parentId = resource.snapshot.parentId;\n      if (parentId == null) continue;\n      const children = childrenByParent.get(parentId);\n      if (children) children.push(resource);\n      else childrenByParent.set(parentId, [resource]);\n    }\n\n    const transcripts = new Map<string, SubagentTranscript>();\n    let changed = source.snapshot.size !== resources.length;\n    const built = new Set<string>();\n\n    const build = (resource: ProjectionResource, depth: number) => {\n      if (built.has(resource.snapshot.id)) return;\n      built.add(resource.snapshot.id);\n      const children =\n        depth < MAX_SUBAGENT_DEPTH\n          ? (childrenByParent.get(resource.snapshot.id) ?? [])\n          : [];\n      for (const child of children) build(child, depth + 1);\n      const childTranscripts = new Map(\n        children.flatMap((child) =>\n          child.entry ? [[child.snapshot.id, child.entry] as const] : [],\n        ),\n      );\n      const storeSnapshot = resource.store.getSnapshot();\n      const status = resource.snapshot.status;\n      const stepped = stepStreamingTiming(\n        resource.timingState,\n        storeSnapshot as LangChainBaseMessage[],\n        status === \"running\",\n        langChainStreamingTimingAccessors,\n        undefined,\n      );\n      resource.timingState = stepped.state;\n      if (Object.keys(stepped.timings).length > 0) {\n        resource.messageTiming = {\n          ...resource.messageTiming,\n          ...stepped.timings,\n        };\n      }\n      const localUiMessages = foldLocalUIMessages(resource);\n      if (\n        resource.localUiMessages !== localUiMessages ||\n        resource.rootUiMessagesByParent !== uiMessagesByParent ||\n        resource.convertedMessageTiming !== resource.messageTiming\n      ) {\n        resource.localUiMessages = localUiMessages;\n        resource.rootUiMessagesByParent = uiMessagesByParent;\n        resource.uiMessagesByParent =\n          localUiMessages.length === 0\n            ? uiMessagesByParent\n            : mergeLocalUIMessages(uiMessagesByParent, localUiMessages);\n        resource.convert = createConverter(\n          resource.uiMessagesByParent,\n          resource.messageTiming,\n        );\n      }\n      const uiMessages = collectUIMessages(\n        storeSnapshot,\n        resource.uiMessagesByParent,\n      );\n\n      const conversionChanged =\n        resource.converted === undefined ||\n        resource.storeSnapshot !== storeSnapshot ||\n        resource.status !== status ||\n        resource.convertedMessageTiming !== resource.messageTiming ||\n        !sameUIMessages(resource.uiMessages, uiMessages);\n      if (conversionChanged) {\n        resource.converted = convertExternalMessages(\n          storeSnapshot as LangChainBaseMessage[],\n          resource.convert,\n          status === \"running\",\n          TRANSCRIPT_METADATA,\n          resource.cache,\n        );\n        resource.storeSnapshot = storeSnapshot;\n        resource.status = status;\n        resource.uiMessages = uiMessages;\n        resource.convertedMessageTiming = resource.messageTiming;\n      }\n\n      if (\n        resource.transcript === undefined ||\n        conversionChanged ||\n        !sameTranscriptEntries(resource.childTranscripts, childTranscripts)\n      ) {\n        const transcript = attachSubagentTranscripts(\n          resource.converted!,\n          childTranscripts,\n          resource.memo,\n        );\n        changed ||= resource.transcript !== transcript;\n        resource.transcript = transcript;\n        resource.childTranscripts = childTranscripts;\n      }\n\n      updateTiming(resource);\n      if (\n        resource.entry === undefined ||\n        resource.entry.messages !== resource.transcript ||\n        resource.entry.timing !== resource.timing\n      ) {\n        resource.entry = {\n          messages: resource.transcript!,\n          ...(resource.timing && { timing: resource.timing }),\n        };\n        changed = true;\n      }\n\n      if (!source.snapshot.has(resource.snapshot.id)) changed = true;\n      transcripts.set(resource.snapshot.id, resource.entry);\n    };\n\n    for (const resource of resources) {\n      if (\n        resource.snapshot.parentId == null ||\n        !source.resources.has(resource.snapshot.parentId)\n      )\n        build(resource, 1);\n    }\n    for (const resource of resources) build(resource, MAX_SUBAGENT_DEPTH);\n\n    if (!changed) return;\n    source.snapshot = transcripts;\n    for (const listener of source.listeners) listener();\n  };\n\n  return source;\n};\n\nexport const useSubagentTranscripts = (\n  stream: AnyStream,\n  uiMessagesByParent: Map<string, UIMessage[]>,\n): ReadonlyMap<string, SubagentTranscript> => {\n  const sourceRef = useRef<SubagentTranscriptSource | undefined>(undefined);\n  if (!sourceRef.current) {\n    sourceRef.current = createSubagentTranscriptSource();\n  }\n  const source = sourceRef.current;\n  const controller = stream[STREAM_CONTROLLER];\n\n  useEffect(() => {\n    source.reconcile(controller, stream.subagents, uiMessagesByParent);\n  }, [controller, source, stream.subagents, uiMessagesByParent]);\n\n  useReplaySafeEffect(() => () => source.dispose(), [source]);\n\n  return useSyncExternalStore(\n    source.subscribe,\n    source.getSnapshot,\n    source.getSnapshot,\n  );\n};\n"],"mappings":";;;;;;;;;;;;;;AAwCA,MAAa,qBAAqB;AAElC,MAAM,wBAAwB;AAE9B,MAAM,sBAAsB,CAAC;AAE7B,MAAM,iBAAuC,CAAC;AAE9C,MAAM,oBAAmD,CAAC;AAgE1D,MAAM,iBAAiB,GAAsB,MAC3C,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,SAAS,UAAU,YAAY,EAAE,MAAM;AAE3E,MAAM,4BAA4B,aAChC,SAAS,UAAU,WAAW,KAC9B,SAAS,UAAU,OAAO,SAAS,SAAS;AAE9C,MAAM,4BACJ,QACA,YACA,YACG;CACH,QAAQ,YAAY;CACpB,QAAQ,UAAU;CAClB,QAAQ,cAAc;CACtB,WACG,yBAAyB,QAAQ,EAAE,CAAC,CACpC,YAAY,CAAC,CAAC,CAAC,CACf,cAAc;EACb,IACE,OAAO,eAAe,cACtB,OAAO,kBAAkB,IAAI,QAAQ,EAAE,MAAM,SAE7C;EAEF,QAAQ,UAAU;EAClB,IACE,CAAC,QAAQ,eACT,QAAQ,aAAa,KACrB,QAAQ,WAAW,WAEnB,QAAQ,cAAc;EAExB,IAAI,CAAC,QAAQ,eAAe,QAAQ,YAAY,GAAG;GACjD,QAAQ,cAAc;GACtB;EACF;EACA,yBAAyB,QAAQ,YAAY,OAAO;CACtD,CAAC;AACL;AAEA,MAAM,yBACJ,GACA,MAEA,GAAG,SAAS,EAAE,QACd,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,gBAAgB,EAAE,IAAI,EAAE,MAAM,UAAU;AAE7D,MAAM,qBACJ,UACA,uBACG;CACH,MAAM,YAAyB,CAAC;CAChC,IAAI,mBAAmB,SAAS,GAAG,OAAO;CAC1C,KAAK,MAAM,WAAW,UAAU;EAC9B,MAAM,aAAa,QAAQ,MAAM,mBAAmB,IAAI,QAAQ,EAAE;EAClE,IAAI,YAAY,UAAU,KAAK,GAAG,UAAU;CAC9C;CACA,OAAO;AACT;AAEA,MAAM,kBAAkB,GAAyB,MAC/C,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,IAAI,UAAU,OAAO,EAAE,MAAM;AAEjE,MAAM,uBAAuB,aAAiC;CAC5D,MAAM,SAAS,SAAS,SAAS,YAAY;CAC7C,IAAI,WAAW,KAAA,GAAW,OAAO;CACjC,OAAO,cAAc,QAAQ,SAAS,UAAU;AAClD;AAEA,MAAM,wBACJ,oBACA,oBACG;CACH,MAAM,SAAS,IAAI,IAAI,kBAAkB;CACzC,KAAK,MAAM,CAAC,UAAU,aAAa,wBACjC,eACF,GACE,OAAO,IAAI,UAAU,gBAAgB,UAAU,OAAO,IAAI,QAAQ,CAAC,CAAC;CAEtE,OAAO;AACT;AAEA,MAAM,mBAEF,oBACA,mBAED,SAAS,aACR,4BAA4B,SAAS;CACnC,GAAG;CACH;CACA;AACF,CAAC;AAEL,MAAM,gBAAgB,aAAiC;CACrD,IAAI,SAAS,WAAW,KAAA,GAAW;CACnC,MAAM,YAAY,SAAS,SAAS,UAAU,QAAQ;CACtD,MAAM,cAAc,SAAS,SAAS,aAAa,QAAQ;CAC3D,IACE,SAAS,OAAO,cAAc,aAC9B,SAAS,OAAO,gBAAgB,aAEhC;CACF,SAAS,SAAS;EAChB;EACA,GAAI,gBAAgB,KAAA,KAAa,EAAE,YAAY;CACjD;AACF;AAEA,MAAM,uCAAiE;CAErE,MAAM,SAAmC;EACvC,2BAAW,IAAI,IAAI;EACnB,mCAAmB,IAAI,IAAI;EAC3B,0BAAU,IAAI,IAAI;EAClB,2BAAW,IAAI,IAAI;EACnB,YAAY,KAAA;EACZ,oCAAA,IAP6B,IAOZ;EACjB,UAAU,UAAU;GAClB,OAAO,UAAU,IAAI,QAAQ;GAC7B,aAAa,OAAO,UAAU,OAAO,QAAQ;EAC/C;EACA,cAAc;GACZ,OAAO,OAAO;EAChB;EACA,UAAU,YAAY,WAAW,oBAAoB;GACnD,OAAO,qBAAqB;GAI5B,MAAM,0BAAU,IAAI,IAA4B;GAEhD,IAAI,OAAO,eAAe,YAAY;IACpC,OAAO,QAAQ;IACf,OAAO,aAAa;GACtB;GAEA,KAAK,MAAM,MAAM,OAAO,kBAAkB,KAAK,GAC7C,IAAI,CAAC,UAAU,IAAI,EAAE,GAAG,OAAO,kBAAkB,OAAO,EAAE;GAG5D,KAAK,MAAM,CAAC,IAAI,aAAa,OAAO,WAAW;IAC7C,MAAM,WAAW,UAAU,IAAI,EAAE;IACjC,IACE,CAAC,YACD,SAAS,QAAA,MACT,CAAC,cAAc,SAAS,WAAW,SAAS,SAAS,GACrD;KACA,IAAI,YAAY,SAAS,SAAA,IACvB,QAAQ,IAAI,IAAI;MACd,QAAQ,SAAS;MACjB,aAAa,SAAS;MACtB,eAAe,SAAS;KAC1B,CAAC;KACH,SAAS,QAAQ;KACjB,OAAO,UAAU,OAAO,EAAE;KAC1B;IACF;IACA,SAAS,WAAW;GACtB;GAEA,KAAK,MAAM,YAAY,UAAU,OAAO,GAAG;IACzC,IAAI,SAAS,QAAA,IAA4B;IACzC,IAAI,CAAC,yBAAyB,QAAQ,GACpC,OAAO,kBAAkB,OAAO,SAAS,EAAE;SACtC;KACL,MAAM,UAAU,OAAO,kBAAkB,IAAI,SAAS,EAAE;KACxD,IAAI,CAAC,SAAS;MACZ,MAAM,cAAgC;OACpC,IAAI,SAAS;OACb,UAAU;OACV,SAAS;OACT,aAAa;OACb,QAAQ,SAAS;MACnB;MACA,OAAO,kBAAkB,IAAI,SAAS,IAAI,WAAW;MACrD,yBAAyB,QAAQ,YAAY,WAAW;KAC1D,OAAO,IAAI,QAAQ,WAAW,SAAS,QAAQ;MAC7C,QAAQ,SAAS,SAAS;MAC1B,IAAI,QAAQ,SACV,QAAQ,cAAc,QAAQ,WAAW;WACpC,IAAI,QAAQ,WAAW,GAC5B,yBAAyB,QAAQ,YAAY,OAAO;KAExD;IACF;IACA,IAAI,OAAO,UAAU,IAAI,SAAS,EAAE,GAAG;IACvC,MAAM,WAAW,WAAW,SAAS,QACnC,2BAA2B,SAAS,SAAS,CAC/C;IACA,MAAM,aACJ,SAAS,UAAU,SAAS,wBACxB,WAAW,SAAS,QAClB,kBAAkB,oBAAoB,SAAS,SAAS,CAC1D,IACA,KAAA;IACN,MAAM,WAA+B;KACnC;KACA,WAAW,SAAS;KACpB,OAAO,SAAS;KAChB,SAAS,YAAY;KACrB,eAAe,CAAC;KAChB,eAAe,KAAA;KACf,QAAQ,KAAA;KACR,YAAY,iBAAiB;KAC7B,iBAAiB;KACjB,wBAAwB,OAAO;KAC/B,oBAAoB,OAAO;KAC3B,aAAa,QAAQ,IAAI,SAAS,EAAE,CAAC,EAAE,eAAe;KACtD,eACE,QAAQ,IAAI,SAAS,EAAE,CAAC,EAAE,iBAAiB;KAC7C,wBAAwB,KAAA;KACxB,SAAS,gBACP,OAAO,oBACP,iBACF;KACA,YAAY,CAAC;KACb,WAAW,KAAA;KACX,kBAAkB,KAAA;KAClB,YAAY,KAAA;KACZ,QAAQ,QAAQ,IAAI,SAAS,EAAE,IAC3B,QAAQ,IAAI,SAAS,EAAE,CAAC,CAAE,SAC1B,SAAS,WAAW,YAClB,EAAE,WAAW,SAAS,UAAU,QAAQ,EAAE,IAC1C,KAAA;KACN,OAAO,KAAA;KACP,MAAM,iBAAiB;KACvB,OAAO,qCAAqC;IAC9C;IACA,MAAM,cAAc,SAAS,MAAM,gBAAgB,QAAQ,CAAC;IAC5D,MAAM,gBAAgB,SAAS,SAAS,gBAAgB,QAAQ,CAAC;IACjE,SAAS,gBAAgB;KACvB,YAAY;KACZ,gBAAgB;KAChB,SAAS,QAAQ;KACjB,YAAY,QAAQ;IACtB;IACA,OAAO,UAAU,IAAI,SAAS,IAAI,QAAQ;GAC5C;GAEA,QAAQ;EACV;EACA,UAAU;GACR,KAAK,MAAM,YAAY,OAAO,UAAU,OAAO,GAAG,SAAS,QAAQ;GACnE,OAAO,UAAU,MAAM;GACvB,OAAO,kBAAkB,MAAM;GAC/B,OAAO,2BAAW,IAAI,IAAI;GAC1B,KAAK,MAAM,YAAY,OAAO,WAAW,SAAS;EACpD;CACF;CAEA,MAAM,gBAAgB;EACpB,MAAM,EAAE,uBAAuB;EAC/B,MAAM,YAAY,CAAC,GAAG,OAAO,UAAU,OAAO,CAAC;EAC/C,MAAM,mCAAmB,IAAI,IAAkC;EAE/D,KAAK,MAAM,YAAY,WAAW;GAChC,MAAM,WAAW,SAAS,SAAS;GACnC,IAAI,YAAY,MAAM;GACtB,MAAM,WAAW,iBAAiB,IAAI,QAAQ;GAC9C,IAAI,UAAU,SAAS,KAAK,QAAQ;QAC/B,iBAAiB,IAAI,UAAU,CAAC,QAAQ,CAAC;EAChD;EAEA,MAAM,8BAAc,IAAI,IAAgC;EACxD,IAAI,UAAU,OAAO,SAAS,SAAS,UAAU;EACjD,MAAM,wBAAQ,IAAI,IAAY;EAE9B,MAAM,SAAS,UAA8B,UAAkB;GAC7D,IAAI,MAAM,IAAI,SAAS,SAAS,EAAE,GAAG;GACrC,MAAM,IAAI,SAAS,SAAS,EAAE;GAC9B,MAAM,WACJ,QAAA,KACK,iBAAiB,IAAI,SAAS,SAAS,EAAE,KAAK,CAAC,IAChD,CAAC;GACP,KAAK,MAAM,SAAS,UAAU,MAAM,OAAO,QAAQ,CAAC;GACpD,MAAM,mBAAmB,IAAI,IAC3B,SAAS,SAAS,UAChB,MAAM,QAAQ,CAAC,CAAC,MAAM,SAAS,IAAI,MAAM,KAAK,CAAU,IAAI,CAAC,CAC/D,CACF;GACA,MAAM,gBAAgB,SAAS,MAAM,YAAY;GACjD,MAAM,SAAS,SAAS,SAAS;GACjC,MAAM,UAAU,oBACd,SAAS,aACT,eACA,WAAW,WACX,mCACA,KAAA,CACF;GACA,SAAS,cAAc,QAAQ;GAC/B,IAAI,OAAO,KAAK,QAAQ,OAAO,CAAC,CAAC,SAAS,GACxC,SAAS,gBAAgB;IACvB,GAAG,SAAS;IACZ,GAAG,QAAQ;GACb;GAEF,MAAM,kBAAkB,oBAAoB,QAAQ;GACpD,IACE,SAAS,oBAAoB,mBAC7B,SAAS,2BAA2B,sBACpC,SAAS,2BAA2B,SAAS,eAC7C;IACA,SAAS,kBAAkB;IAC3B,SAAS,yBAAyB;IAClC,SAAS,qBACP,gBAAgB,WAAW,IACvB,qBACA,qBAAqB,oBAAoB,eAAe;IAC9D,SAAS,UAAU,gBACjB,SAAS,oBACT,SAAS,aACX;GACF;GACA,MAAM,aAAa,kBACjB,eACA,SAAS,kBACX;GAEA,MAAM,oBACJ,SAAS,cAAc,KAAA,KACvB,SAAS,kBAAkB,iBAC3B,SAAS,WAAW,UACpB,SAAS,2BAA2B,SAAS,iBAC7C,CAAC,eAAe,SAAS,YAAY,UAAU;GACjD,IAAI,mBAAmB;IACrB,SAAS,YAAY,wBACnB,eACA,SAAS,SACT,WAAW,WACX,qBACA,SAAS,KACX;IACA,SAAS,gBAAgB;IACzB,SAAS,SAAS;IAClB,SAAS,aAAa;IACtB,SAAS,yBAAyB,SAAS;GAC7C;GAEA,IACE,SAAS,eAAe,KAAA,KACxB,qBACA,CAAC,sBAAsB,SAAS,kBAAkB,gBAAgB,GAClE;IACA,MAAM,aAAa,0BACjB,SAAS,WACT,kBACA,SAAS,IACX;IACA,YAAY,SAAS,eAAe;IACpC,SAAS,aAAa;IACtB,SAAS,mBAAmB;GAC9B;GAEA,aAAa,QAAQ;GACrB,IACE,SAAS,UAAU,KAAA,KACnB,SAAS,MAAM,aAAa,SAAS,cACrC,SAAS,MAAM,WAAW,SAAS,QACnC;IACA,SAAS,QAAQ;KACf,UAAU,SAAS;KACnB,GAAI,SAAS,UAAU,EAAE,QAAQ,SAAS,OAAO;IACnD;IACA,UAAU;GACZ;GAEA,IAAI,CAAC,OAAO,SAAS,IAAI,SAAS,SAAS,EAAE,GAAG,UAAU;GAC1D,YAAY,IAAI,SAAS,SAAS,IAAI,SAAS,KAAK;EACtD;EAEA,KAAK,MAAM,YAAY,WACrB,IACE,SAAS,SAAS,YAAY,QAC9B,CAAC,OAAO,UAAU,IAAI,SAAS,SAAS,QAAQ,GAEhD,MAAM,UAAU,CAAC;EAErB,KAAK,MAAM,YAAY,WAAW,MAAM,UAAA,EAA4B;EAEpE,IAAI,CAAC,SAAS;EACd,OAAO,WAAW;EAClB,KAAK,MAAM,YAAY,OAAO,WAAW,SAAS;CACpD;CAEA,OAAO;AACT;AAEA,MAAa,0BACX,QACA,uBAC4C;CAC5C,MAAM,YAAY,OAA6C,KAAA,CAAS;CACxE,IAAI,CAAC,UAAU,SACb,UAAU,UAAU,+BAA+B;CAErD,MAAM,SAAS,UAAU;CACzB,MAAM,aAAa,OAAO;CAE1B,gBAAgB;EACd,OAAO,UAAU,YAAY,OAAO,WAAW,kBAAkB;CACnE,GAAG;EAAC;EAAY;EAAQ,OAAO;EAAW;CAAkB,CAAC;CAE7D,gCAAgC,OAAO,QAAQ,GAAG,CAAC,MAAM,CAAC;CAE1D,OAAO,qBACL,OAAO,WACP,OAAO,aACP,OAAO,WACT;AACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-langchain",
-  "version": "0.0.32",
+  "version": "0.0.34",
   "description": "LangChain useStream adapter for assistant-ui",
   "keywords": [
     "langchain",
@@ -40,18 +40,22 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.3.20",
-    "@assistant-ui/store": "^0.3.14",
+    "@assistant-ui/core": "^0.3.23",
+    "@assistant-ui/store": "^0.3.17",
     "assistant-cloud": "*",
-    "assistant-stream": "^0.3.44"
+    "assistant-stream": "^0.3.47"
   },
   "peerDependencies": {
+    "@langchain/core": "^1.0.0",
     "@langchain/langgraph-sdk": "^1.9.20",
     "@langchain/react": "^1.0.20",
     "@types/react": "*",
     "react": "^18 || ^19"
   },
   "peerDependenciesMeta": {
+    "@langchain/core": {
+      "optional": true
+    },
     "@langchain/langgraph-sdk": {
       "optional": true
     },
@@ -63,15 +67,16 @@
     }
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.28",
-    "@langchain/core": "^1.2.11",
-    "@langchain/langgraph-sdk": "^1.11.0",
-    "@langchain/react": "^1.1.0",
+    "@assistant-ui/x-buildutils": "0.0.30",
+    "@assistant-ui/x-react18": "0.0.0",
+    "@langchain/core": "^1.2.13",
+    "@langchain/langgraph-sdk": "^1.12.0",
+    "@langchain/react": "^1.2.0",
     "@testing-library/react": "^16.3.3",
     "@types/react": "^19.3.0",
     "jsdom": "^30.0.1",
     "react": "^19.3.0",
-    "vitest": "^5.0.1"
+    "vitest": "^5.0.3"
   },
   "publishConfig": {
     "access": "public",
@@ -90,6 +95,8 @@
     "build": "aui-build",
     "test": "vitest run",
     "typecheck": "tsc --noEmit",
+    "test:peer-react18": "vitest run --config vitest.peer-react18.config.ts",
+    "test:types:peer-react18": "aui-react18-types",
     "test:watch": "vitest"
   }
 }

package/src/convertMessages.test.ts

@@ -27,6 +27,27 @@ const contentOf = (result: ReturnType<typeof convertLangChainBaseMessage>) => {
   return result.content;
 };
 
+describe("convertLangChainBaseMessage modality", () => {
+  it("lifts voice modality onto human and ai messages and ignores unknown values", () => {
+    for (const message of [humanMessage("Question"), aiMessage("Answer")]) {
+      const spoken = convertLangChainBaseMessage({
+        ...message,
+        additional_kwargs: { modality: "voice" },
+      });
+      const unknown = convertLangChainBaseMessage({
+        ...message,
+        additional_kwargs: { modality: "video" },
+      });
+
+      expect(spoken).toHaveProperty("metadata", {
+        custom: {},
+        modality: "voice",
+      });
+      expect(unknown).toHaveProperty("metadata", { custom: {} });
+    }
+  });
+});
+
 describe("convertLangChainBaseMessage file content parts", () => {
   it("converts a base64 file block", () => {
     const result = convertLangChainBaseMessage(
@@ -1113,6 +1134,118 @@ describe("convertLangChainBaseMessage malformed messages", () => {
     ]);
   });
 
+  it("normalizes missing, string, and array tool-call args to an object", () => {
+    for (const args of [undefined, "not-json", ["x"], null]) {
+      const result = convertLangChainBaseMessage(
+        {
+          ...aiMessage([]),
+          tool_calls: [{ id: "call-1", name: "lookup", args }],
+        } as unknown as LangChainBaseMessage,
+        {},
+      );
+      expect(contentOf(result)).toEqual([
+        {
+          type: "tool-call",
+          toolCallId: "call-1",
+          toolName: "lookup",
+          args: {},
+          argsText: "{}",
+        },
+      ]);
+    }
+  });
+
+  it("normalizes unsafe object tool-call args to an empty object", () => {
+    const cyclic: Record<string, unknown> = {};
+    cyclic.self = cyclic;
+    class CustomArgs {
+      query = "x";
+    }
+
+    for (const args of [new Date(0), new Map(), new CustomArgs(), cyclic]) {
+      const result = convertLangChainBaseMessage(
+        {
+          ...aiMessage([]),
+          tool_calls: [{ id: "call-1", name: "lookup", args }],
+        } as unknown as LangChainBaseMessage,
+        {},
+      );
+      expect(contentOf(result)).toEqual([
+        {
+          type: "tool-call",
+          toolCallId: "call-1",
+          toolName: "lookup",
+          args: {},
+          argsText: "{}",
+        },
+      ]);
+    }
+  });
+
+  it("skips a null tool call entry and keeps the rest", () => {
+    const result = convertLangChainBaseMessage(
+      {
+        ...aiMessage([]),
+        tool_calls: [null, { id: "call-1", name: "lookup", args: {} }],
+      } as unknown as LangChainBaseMessage,
+      {},
+    );
+
+    expect(contentOf(result)).toEqual([
+      {
+        type: "tool-call",
+        toolCallId: "call-1",
+        toolName: "lookup",
+        args: {},
+        argsText: "{}",
+      },
+    ]);
+  });
+
+  it("skips a tool call without a name so its named result does not throw", () => {
+    const messages = convertExternalMessages(
+      [
+        {
+          ...aiMessage([]),
+          tool_calls: [{ id: "call-1", args: {} }],
+        } as unknown as LangChainBaseMessage,
+        {
+          _getType: () => "tool",
+          id: "msg-tool",
+          name: "search",
+          tool_call_id: "call-1",
+          content: "3 results",
+        } as LangChainBaseMessage,
+      ],
+      (message) => convertLangChainBaseMessage(message, {}),
+      false,
+      {},
+    );
+
+    expect(messages).toHaveLength(1);
+    expect(messages[0]!.content).toEqual([]);
+  });
+
+  it("keeps a tool call whose name is empty", () => {
+    const result = convertLangChainBaseMessage(
+      {
+        ...aiMessage([]),
+        tool_calls: [{ id: "call-1", name: "", args: {} }],
+      },
+      {},
+    );
+
+    expect(contentOf(result)).toEqual([
+      {
+        type: "tool-call",
+        toolCallId: "call-1",
+        toolName: "",
+        args: {},
+        argsText: "{}",
+      },
+    ]);
+  });
+
   it("converts a system message with null content to empty text", () => {
     const result = convertLangChainBaseMessage(
       { _getType: () => "system", id: "msg-4", content: null },
@@ -1217,6 +1350,26 @@ describe("convertLangChainBaseMessage malformed messages", () => {
     }
   });
 
+  it("warns once in development about a skipped tool call without a name", () => {
+    vi.stubEnv("NODE_ENV", "development");
+    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+    try {
+      const message = {
+        ...aiMessage([]),
+        tool_calls: [null, { id: "call-1", args: {} }],
+      } as unknown as LangChainBaseMessage;
+      convertLangChainBaseMessage(message, {});
+      convertLangChainBaseMessage(message, {});
+      expect(warn).toHaveBeenCalledTimes(1);
+      expect(warn).toHaveBeenCalledWith(
+        "Skipping a tool call without a name; its result is not shown either",
+      );
+    } 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(() => {});

package/src/convertMessages.ts

@@ -6,6 +6,7 @@ import type { ReadonlyJSONObject } from "assistant-stream/utils";
 import {
   convertLangChainContentBlock,
   getCustomMetadata,
+  getMessageModality,
   uiMessageToDataPart,
   withAudioTranscript,
 } from "./converter";
@@ -55,6 +56,35 @@ const contentBlocks = (content: unknown): readonly LangChainContentBlock[] => {
   return [];
 };
 
+const normalizeToolCallArgs = (args: unknown): ReadonlyJSONObject => {
+  if (typeof args !== "object" || args === null || Array.isArray(args)) {
+    return {};
+  }
+
+  try {
+    const prototype = Object.getPrototypeOf(args);
+    return prototype === Object.prototype || prototype === null
+      ? (args as ReadonlyJSONObject)
+      : {};
+  } catch {
+    return {};
+  }
+};
+
+const toolCallArgs = (
+  value: unknown,
+): { args: ReadonlyJSONObject; argsText: string } => {
+  const args = normalizeToolCallArgs(value);
+  try {
+    const argsText = JSON.stringify(args);
+    return typeof argsText === "string"
+      ? { args, argsText }
+      : { args: {}, argsText: "{}" };
+  } catch {
+    return { args: {}, argsText: "{}" };
+  }
+};
+
 const contentToParts = (content: unknown) => {
   if (typeof content === "string")
     return [{ type: "text" as const, text: content }];
@@ -89,25 +119,39 @@ export const convertLangChainBaseMessage = (
         },
       };
 
-    case "human":
+    case "human": {
+      const modality = getMessageModality(message.additional_kwargs);
       return {
         role: "user",
         id: message.id,
         content: contentToParts(message.content),
         metadata: {
           custom: getCustomMetadata(message.additional_kwargs),
+          ...(modality && { modality }),
         },
       };
+    }
 
     case "ai": {
       const toolCallParts =
-        message.tool_calls?.map((tc) => ({
-          type: "tool-call" as const,
-          toolCallId: tc.id,
-          toolName: tc.name,
-          args: tc.args as ReadonlyJSONObject,
-          argsText: JSON.stringify(tc.args),
-        })) ?? [];
+        message.tool_calls
+          ?.filter((tc) => {
+            if (typeof tc?.name === "string") return true;
+            warnOnceInDevelopment(
+              "Skipping a tool call without a name; its result is not shown either",
+            );
+            return false;
+          })
+          .map((tc) => {
+            const { args, argsText } = toolCallArgs(tc.args);
+            return {
+              type: "tool-call" as const,
+              toolCallId: tc.id,
+              toolName: tc.name,
+              args,
+              argsText,
+            };
+          }) ?? [];
 
       const assistantStatus =
         typeof message.status === "object" ? message.status : undefined;
@@ -120,6 +164,7 @@ export const convertLangChainBaseMessage = (
           : undefined) ?? [];
 
       const timing = metadata.messageTiming?.[message.id ?? ""];
+      const modality = getMessageModality(message.additional_kwargs);
 
       return {
         role: "assistant",
@@ -135,6 +180,7 @@ export const convertLangChainBaseMessage = (
         metadata: {
           custom: getCustomMetadata(message.additional_kwargs),
           ...(timing && { timing }),
+          ...(modality && { modality }),
         },
         ...(assistantStatus && { status: assistantStatus }),
       };

package/src/converter.ts

@@ -1,6 +1,7 @@
 import type {
   AppendMessage,
   DataMessagePart,
+  MessageModality,
   ThreadAssistantMessage,
   ThreadUserMessage,
 } from "@assistant-ui/core";
@@ -146,6 +147,11 @@ export const getCustomMetadata = (
 ): Record<string, unknown> =>
   (additionalKwargs?.metadata as Record<string, unknown>) ?? {};
 
+export const getMessageModality = (
+  additionalKwargs: Record<string, unknown> | undefined,
+): MessageModality | undefined =>
+  additionalKwargs?.modality === "voice" ? "voice" : undefined;
+
 export const uiMessageToDataPart = <
   TUIMessage extends { name: string; props: Record<string, unknown> },
 >(

package/src/subagentMessagesProjection.test.ts

@@ -0,0 +1,175 @@
+import { describe, expect, it, vi } from "vitest";
+import type { BaseMessage } from "@langchain/core/messages";
+import {
+  matchesSubscription,
+  SubscriptionHandle,
+} from "@langchain/langgraph-sdk/client";
+import {
+  type Event,
+  messagesProjection,
+  type ProjectionSpec,
+  StreamStore,
+} from "@langchain/langgraph-sdk/stream";
+import { subagentMessagesProjection } from "./subagentMessagesProjection";
+
+const PARENT = ["tools:parent"];
+const CHILD = ["tools:parent", "tools:child"];
+
+const human = (id: string, content: string) => ({
+  id,
+  type: "human",
+  content,
+});
+const ai = (id: string, content: string, tool_calls: unknown[] = []) => ({
+  id,
+  type: "ai",
+  content,
+  tool_calls,
+});
+const tool = (id: string, content: string, tool_call_id: string) => ({
+  id,
+  type: "tool",
+  content,
+  tool_call_id,
+});
+
+const values = (namespace: string[], messages: unknown[]) =>
+  ({
+    method: "values",
+    params: { namespace, data: { messages } },
+  }) as unknown as Event;
+
+const messagesEvent = (namespace: string[], data: Record<string, unknown>) =>
+  ({
+    method: "messages",
+    params: { namespace, node: "model", data },
+  }) as unknown as Event;
+
+/**
+ * Feeds the projection through the per-subscription matcher the SDK client
+ * applies before it hands an event to a subscription.
+ */
+const openProjection = (spec: ProjectionSpec<BaseMessage[]>) => {
+  let handle: SubscriptionHandle<Event> | undefined;
+  const unsubscribe = vi.fn(async () => {});
+  const subscribe = vi.fn(
+    async (params: SubscriptionHandle<Event>["params"]) => {
+      handle = new SubscriptionHandle<Event>(
+        "subscription",
+        params,
+        unsubscribe,
+      );
+      return handle;
+    },
+  );
+  const store = new StreamStore<BaseMessage[]>(spec.initial);
+  const runtime = spec.open({
+    thread: { subscribe } as never,
+    store,
+    rootBus: {
+      channels: [
+        "values",
+        "checkpoints",
+        "lifecycle",
+        "input",
+        "messages",
+        "tools",
+      ],
+      subscribe: () => () => {},
+    },
+  });
+  const push = (event: Event) => {
+    if (matchesSubscription(event, handle!.params)) handle!.push(event);
+  };
+  const ids = () => store.getSnapshot().map((message) => message.id);
+  return { subscribe, unsubscribe, runtime, push, ids };
+};
+
+const parentTurn = [
+  human("parent-human", "research"),
+  ai("parent-ai", "", [
+    { id: "call-child", name: "task", args: { subagent_type: "worker" } },
+  ]),
+];
+
+const childRun = [
+  values(CHILD, [human("child-human", "sub task")]),
+  messagesEvent(CHILD, { event: "message-start", id: "child-ai", role: "ai" }),
+  messagesEvent(CHILD, {
+    event: "content-block-start",
+    index: 0,
+    content: { type: "text", text: "hello from the child" },
+  }),
+  messagesEvent(CHILD, { event: "message-finish" }),
+  values(CHILD, [
+    human("child-human", "sub task"),
+    ai("child-ai", "hello from the child"),
+  ]),
+];
+
+describe("subagentMessagesProjection", () => {
+  it("keeps the SDK projection's identity and pins its subscription to depth 0", async () => {
+    const spec = subagentMessagesProjection(PARENT);
+    expect(spec.namespace).toEqual(PARENT);
+    expect(spec.initial).toEqual([]);
+    expect(spec.key).not.toBe(messagesProjection(PARENT).key);
+
+    const { subscribe, runtime } = openProjection(spec);
+    await vi.waitFor(() => expect(subscribe).toHaveBeenCalledOnce());
+    expect(subscribe).toHaveBeenCalledWith({
+      channels: ["messages", "values"],
+      namespaces: [PARENT],
+      depth: 0,
+    });
+    await runtime.dispose();
+  });
+
+  it("ignores a nested subagent's events while the parent's tool call runs", async () => {
+    const { subscribe, runtime, push, ids } = openProjection(
+      subagentMessagesProjection(PARENT),
+    );
+    await vi.waitFor(() => expect(subscribe).toHaveBeenCalledOnce());
+
+    push(values(PARENT, parentTurn));
+    await vi.waitFor(() =>
+      expect(ids()).toEqual(["parent-human", "parent-ai"]),
+    );
+
+    for (const event of childRun) push(event);
+    push(
+      messagesEvent(PARENT, {
+        event: "message-start",
+        id: "parent-sentinel",
+        role: "ai",
+      }),
+    );
+    push(messagesEvent(PARENT, { event: "message-finish" }));
+    await vi.waitFor(() => expect(ids()).toContain("parent-sentinel"));
+    expect(ids()).toEqual(["parent-human", "parent-ai", "parent-sentinel"]);
+
+    push(
+      values(PARENT, [
+        ...parentTurn,
+        tool("parent-tool", "done", "call-child"),
+      ]),
+    );
+    await vi.waitFor(() =>
+      expect(ids()).toEqual([
+        "parent-human",
+        "parent-ai",
+        "parent-tool",
+        "parent-sentinel",
+      ]),
+    );
+    await runtime.dispose();
+  });
+
+  it("unsubscribes the underlying subscription on dispose", async () => {
+    const { subscribe, unsubscribe, runtime } = openProjection(
+      subagentMessagesProjection(PARENT),
+    );
+    await vi.waitFor(() => expect(subscribe).toHaveBeenCalledOnce());
+    await runtime.dispose();
+    await vi.waitFor(() => expect(unsubscribe).toHaveBeenCalledOnce());
+  });
+});