@assistant-ui/core 0.3.18 → 0.3.20

package/dist/react/primitives/message/MessageGroupedParts.js.map

@@ -1 +1 @@
-{"version":3,"file":"MessageGroupedParts.js","names":["Fragment","FC","ReactNode","useMemo","useAuiState","useShallowSelector","PartState","MessagePartStatus","ToolCallMessagePartStatus","getGroupStatus","buildGroupTree","GROUPBY_MEMO_KEY","GroupByContext","GroupNode","MessagePartChildren","EnrichedPartState","MessagePrimitiveGroupedParts","GroupPart","type","TKey","status","indices","IndicatorPart","IndicatorMode","RenderInfo","part","children","Props","groupBy","context","indicator","info","shouldShowIndicator","mode","parts","isRunning","length","last","undefined","PartChildrenSentinel","Error","renderNode","node","render","idKey","index","groupPart","key","nodeKey","map","child","s","message","toolUIs","tools","memoKey","memoDep","tree","toolCallId","displayName"],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"sourcesContent":["\"use client\";\n\nimport { Fragment, type FC, type ReactNode, useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { useShallowSelector } from \"@assistant-ui/store/internal\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport type {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"../../../types/message\";\nimport { getGroupStatus } from \"../../../utils/getGroupStatus\";\nimport {\n  buildGroupTree,\n  GROUPBY_MEMO_KEY,\n  type GroupByContext,\n  type GroupNode,\n} from \"../../utils/groupParts\";\nimport { MessagePartChildren, type EnrichedPartState } from \"./MessageParts\";\n\nexport namespace MessagePrimitiveGroupedParts {\n  /**\n   * A coalesced group of adjacent parts. Surfaced through the same\n   * `{ part }` channel as a leaf {@link EnrichedPartState} so consumers\n   * dispatch on a single `switch (part.type)`. `type` is the group key\n   * (always `\"group-…\"`); `status` is running when any contained part runs,\n   * otherwise it mirrors the last contained part.\n   */\n  export type GroupPart<TKey extends `group-${string}` = `group-${string}`> = {\n    readonly type: TKey;\n    readonly status: MessagePartStatus | ToolCallMessagePartStatus;\n    readonly indices: readonly number[];\n  };\n\n  /**\n   * Synthetic trailing slot for a streaming/loading affordance (a\n   * \"thinking…\" dot, etc.). Surfaced through the same `{ part }` channel\n   * as groups and leaf parts so a single `switch (part.type)` renders it\n   * via `case \"indicator\"`.\n   *\n   * It is only ever emitted while the message is running, so its presence\n   * alone means \"render your loading UI here\" — there's no `status` to\n   * branch on.\n   */\n  export type IndicatorPart = {\n    readonly type: \"indicator\";\n  };\n\n  /**\n   * When to emit the synthetic {@link IndicatorPart}. It is **only** emitted\n   * while the message is running (streaming); the mode further restricts\n   * which running states qualify:\n   * - `\"never\"` — never.\n   * - `\"empty\"` — only when the message has no parts yet.\n   * - `\"no-text\"` (default) — when the message has no parts yet or the last\n   *   part isn't `text`/`reasoning` (e.g. it ended on a tool call, so the\n   *   assistant likely isn't done).\n   * - `\"always\"` — whenever the message is running, regardless of parts.\n   */\n  export type IndicatorMode = \"never\" | \"empty\" | \"no-text\" | \"always\";\n\n  export type RenderInfo<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Either a coalesced group ({@link GroupPart}, identified by a\n     * `group-…` `type`), a single enriched leaf part, or the synthetic\n     * {@link IndicatorPart} (`type: \"indicator\"`). Use one switch over\n     * `part.type` to handle all three.\n     */\n    readonly part: GroupPart<TKey> | EnrichedPartState | IndicatorPart;\n    /**\n     * For group nodes: the recursively-rendered subtree (subgroups +\n     * leaf parts). For leaf parts: a sentinel that throws when rendered\n     * — accidental fall-through (`default: return children;`) errors\n     * loudly instead of silently rendering nothing.\n     */\n    readonly children: ReactNode;\n  };\n\n  export type Props<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Maps each part to a group-key path. Adjacent parts that share a\n     * prefix coalesce into the same group. Return `[]` (or `null`) to\n     * leave a part ungrouped.\n     *\n     * Group keys must start with `\"group-\"` so the renderer's\n     * `switch (part.type)` can tell groups apart from real part types.\n     *\n     * **Prefer {@link groupPartByType}** for the common case of mapping by\n     * `part.type` — it ships a stable memo fingerprint so the tree\n     * survives unrelated re-renders. Use an inline function only when\n     * the helper isn't expressive enough (e.g. branching on\n     * `part.toolName` or part metadata).\n     *\n     * The second argument is a {@link GroupByContext} carrying the tool-UI\n     * registry, for grouping that depends on it (e.g. standalone tool calls).\n     *\n     * @example\n     * ```tsx\n     * import { groupPartByType } from \"@assistant-ui/react\";\n     *\n     * <MessagePrimitive.GroupedParts\n     *   groupBy={groupPartByType({\n     *     reasoning: [\"group-thought\", \"group-reasoning\"],\n     *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n     *   })}\n     * >\n     * ```\n     */\n    readonly groupBy: (\n      part: PartState,\n      context: GroupByContext,\n    ) => readonly TKey[] | null;\n\n    /**\n     * Controls emission of the synthetic {@link IndicatorPart} — a\n     * trailing `{ part: { type: \"indicator\", status } }` render call you\n     * handle with `case \"indicator\"` to show loading/status UI.\n     *\n     * @default \"no-text\"\n     * @see IndicatorMode\n     */\n    readonly indicator?: IndicatorMode;\n\n    /**\n     * Render function called once per group node, once per leaf part, and\n     * (when the `indicator` condition is met) once for the trailing\n     * {@link IndicatorPart}. Switch on `part.type`: `\"group-…\"` cases wrap\n     * `children`; real part types (`\"text\"`, `\"tool-call\"`, …) render the\n     * part directly; `\"indicator\"` renders status/loading UI.\n     *\n     * Leaf parts receive the same {@link EnrichedPartState} that\n     * `<MessagePrimitive.Parts>` would produce (`toolUI`, `addResult`,\n     * `resume`, `respondToApproval`, `dataRendererUI`).\n     */\n    readonly children: (info: RenderInfo<TKey>) => ReactNode;\n  };\n}\n\nconst shouldShowIndicator = (\n  mode: MessagePrimitiveGroupedParts.IndicatorMode,\n  parts: readonly PartState[],\n  isRunning: boolean,\n): boolean => {\n  // The indicator is a streaming affordance — never show it on a settled\n  // message, whatever the mode.\n  if (!isRunning) return false;\n\n  switch (mode) {\n    case \"never\":\n      return false;\n    case \"always\":\n      return true;\n    case \"empty\":\n      return parts.length === 0;\n    case \"no-text\": {\n      const last = parts[parts.length - 1];\n      return (\n        last === undefined ||\n        (last.type !== \"text\" && last.type !== \"reasoning\")\n      );\n    }\n  }\n};\n\n/**\n * `children` placeholder passed for leaf-part renders. Leaf parts have no\n * inner subtree; rendering this sentinel signals the consumer wrote\n * `default: return children;` and accidentally fell through for a part —\n * surface the bug loudly instead of silently rendering nothing.\n */\nconst PartChildrenSentinel: FC = () => {\n  throw new Error(\n    \"MessagePrimitive.GroupedParts: rendered `children` under a leaf \" +\n      \"part. `children` is only meaningful for `group-…` cases — add a \" +\n      \"matching case for the part type or return `null` to skip it.\",\n  );\n};\n\nconst renderNode = <TKey extends `group-${string}`>(\n  node: GroupNode,\n  parts: readonly PartState[],\n  render: (info: MessagePrimitiveGroupedParts.RenderInfo<TKey>) => ReactNode,\n): ReactNode => {\n  if (node.type === \"part\") {\n    // Key by part identity when available, else absolute part index — never\n    // the structural nodeKey, which leaves zombie fiber subscriptions when\n    // parts reshape (#4051).\n    return (\n      <MessagePartChildren\n        key={node.idKey ? `part-${node.idKey}` : `part-${node.index}`}\n        index={node.index}\n      >\n        {({ part }) => render({ part, children: <PartChildrenSentinel /> })}\n      </MessagePartChildren>\n    );\n  }\n\n  const status = getGroupStatus(parts, node.indices);\n  const groupPart: MessagePrimitiveGroupedParts.GroupPart<TKey> = {\n    type: node.key as TKey,\n    status,\n    indices: node.indices,\n  };\n\n  return (\n    <Fragment key={node.idKey ?? node.nodeKey}>\n      {render({\n        part: groupPart,\n        children: (\n          <>{node.children.map((child) => renderNode(child, parts, render))}</>\n        ),\n      })}\n    </Fragment>\n  );\n};\n\n/**\n * Groups adjacent message parts into a tree of coalesced runs and\n * renders each node — group or part — through a single `children`\n * function.\n *\n * The render function receives `{ part, children }` where `part.type`\n * is either a `\"group-…\"` literal (for a group, `children` is the\n * recursively-rendered subtree) or a real part type (`\"text\"`,\n * `\"tool-call\"`, …) for a leaf (`children` is a sentinel that throws\n * if rendered — use `part.type` to distinguish).\n *\n * @example\n * ```tsx\n * <MessagePrimitive.GroupedParts\n *   groupBy={groupPartByType({\n *     reasoning: [\"group-thought\", \"group-reasoning\"],\n *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n *   })}\n * >\n *   {({ part, children }) => {\n *     switch (part.type) {\n *       case \"group-thought\":   return <Thought>{children}</Thought>;\n *       case \"group-reasoning\": return <Reasoning>{children}</Reasoning>;\n *       case \"group-tool\":      return <ToolStack>{children}</ToolStack>;\n *       case \"text\":            return <MarkdownText />;\n *       case \"tool-call\":       return part.toolUI ?? <ToolFallback {...part} />;\n *       case \"indicator\":       return <LoadingDots />;\n *       default:                return null;\n *     }\n *   }}\n * </MessagePrimitive.GroupedParts>\n * ```\n */\nexport const MessagePrimitiveGroupedParts = <TKey extends `group-${string}`>({\n  groupBy,\n  indicator = \"no-text\",\n  children,\n}: MessagePrimitiveGroupedParts.Props<TKey>): ReactNode => {\n  const parts = useAuiState(useShallowSelector((s) => s.message.parts));\n  // Handed to `groupBy` as its `context` argument (see GroupByContext).\n  const toolUIs = useAuiState((s) => s.tools.toolUIs);\n  // Subscribe to a boolean, not the status object: the tree only needs to\n  // re-render when running-ness flips, and `\"never\"` opts out entirely.\n  const isRunning = useAuiState((s) =>\n    indicator === \"never\" ? false : s.message.status?.type === \"running\",\n  );\n\n  // Helpers like `groupPartByType` tag the function with `GROUPBY_MEMO_KEY`\n  // (a stable string fingerprint of the helper config). When present,\n  // memo on `[parts, memoKey]` so the tree survives unrelated renders.\n  // For inline `groupBy`, fall back to recomputing each render — O(n)\n  // and cheap.\n  const memoKey = (groupBy as { [GROUPBY_MEMO_KEY]?: string })[\n    GROUPBY_MEMO_KEY\n  ];\n  const memoDep = memoKey ?? groupBy;\n  const tree = useMemo(() => {\n    const context: GroupByContext = { toolUIs };\n    return buildGroupTree(\n      parts.map((part) => groupBy(part, context) ?? []),\n      parts.map((part) =>\n        part.type === \"tool-call\" ? part.toolCallId : undefined,\n      ),\n    );\n    // oxlint-disable-next-line react/exhaustive-deps -- groupBy is captured via memoDep (either its identity or the helper's memoKey fingerprint); listing it directly would defeat the helper-tagged memo path\n  }, [parts, memoDep, toolUIs]);\n\n  return (\n    <>\n      {tree.map((node) => renderNode(node, parts, children))}\n      {shouldShowIndicator(indicator, parts, isRunning) &&\n        children({\n          part: { type: \"indicator\" },\n          children: <PartChildrenSentinel />,\n        })}\n    </>\n  );\n};\n\nMessagePrimitiveGroupedParts.displayName = \"MessagePrimitive.GroupedParts\";\n"],"mappings":";;;;;;;;;AAyIA,MAAMgC,uBACJC,MACAC,OACAC,cACY;CAGZ,IAAI,CAACA,WAAW,OAAO;CAEvB,QAAQF,MAAR;EACE,KAAK,SACH,OAAO;EACT,KAAK,UACH,OAAO;EACT,KAAK,SACH,OAAOC,MAAME,WAAW;EAC1B,KAAK,WAAW;GACd,MAAMC,OAAOH,MAAMA,MAAME,SAAS;GAClC,OACEC,SAASC,KAAAA,KACRD,KAAKnB,SAAS,UAAUmB,KAAKnB,SAAS;EAE3C;CACF;AACF;;;;;;;AAQA,MAAMqB,6BAAiC;CACrC,MAAM,IAAIC,MACR,8LAGF;AACF;AAEA,MAAMC,cACJC,MACAR,OACAS,WACc;CACd,IAAID,KAAKxB,SAAS,QAIhB,OACE,oBAAC,qBAAD;EAEE,OAAOwB,KAAKG;EAEV,WAAA,EAAEpB,WAAWkB,OAAO;GAAElB;GAAMC,UAAU,oBAAC,sBAAD,CAAqB,CAAA;EAAI,CAAC;CAC/C,GAJdgB,KAAKE,QAAQ,QAAQF,KAAKE,UAAU,QAAQF,KAAKG,OAInC;CAIzB,MAAMzB,SAASX,eAAeyB,OAAOQ,KAAKrB,OAAO;CACjD,MAAMyB,YAA0D;EAC9D5B,MAAMwB,KAAKK;EACX3B;EACAC,SAASqB,KAAKrB;CAChB;CAEA,OACE,oBAAC,UAAD,EAAA,UACGsB,OAAO;EACNlB,MAAMqB;EACNpB,UACE,oBAAA,YAAA,EAAA,UAAGgB,KAAKhB,SAASuB,KAAKC,UAAUT,WAAWS,OAAOhB,OAAOS,MAAM,CAAC,EAAE,CAAA;CAEtE,CAAC,EACO,GAPKD,KAAKE,SAASF,KAAKM,OAOxB;AAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmCA,MAAahC,gCAAgE,EAC3EY,SACAE,YAAY,WACZJ,eACyD;CACzD,MAAMQ,QAAQ9B,YAAYC,oBAAoB8C,MAAMA,EAAEC,QAAQlB,KAAK,CAAC;CAEpE,MAAMmB,UAAUjD,aAAa+C,QAAMA,IAAEG,MAAMD,OAAO;CAGlD,MAAMlB,YAAY/B,aAAa+C,QAC7BrB,cAAc,UAAU,QAAQqB,IAAEC,QAAQhC,QAAQF,SAAS,SAC7D;CAUA,MAAMsC,UAHW5B,QACfjB,qBAEyBiB;CAC3B,MAAM6B,OAAOtD,cAAc;EACzB,MAAM0B,UAA0B,EAAEwB,QAAQ;EAC1C,OAAO3C,eACLwB,MAAMe,KAAKxB,SAASG,QAAQH,MAAMI,OAAO,KAAK,CAAA,CAAE,GAChDK,MAAMe,KAAKxB,WACTA,OAAKP,SAAS,cAAcO,OAAKiC,aAAapB,KAAAA,CAChD,CACF;CAEF,GAAG;EAACJ;EAAOsB;EAASH;CAAO,CAAC;CAE5B,OACE,qBAAA,YAAA,EAAA,UAAA,CACGI,KAAKR,KAAKP,SAASD,WAAWC,MAAMR,OAAOR,QAAQ,CAAC,GACpDM,oBAAoBF,WAAWI,OAAOC,SAAS,KAC9CT,SAAS;EACPD,MAAM,EAAEP,MAAM,YAAY;EAC1BQ,UAAU,oBAAC,sBAAD,CAAqB,CAAA;CACjC,CAAC,CACL,EAAA,CAAA;AAEJ;AAEAV,6BAA6B2C,cAAc"}
+{"version":3,"file":"MessageGroupedParts.js","names":["Fragment","FC","ReactNode","useMemo","useAuiState","useShallowSelector","PartState","MessagePartStatus","ToolCallMessagePartStatus","getGroupSummary","buildGroupTree","GROUPBY_MEMO_KEY","GroupByContext","GroupNode","MessagePartChildren","EnrichedPartState","MessagePrimitiveGroupedParts","GroupCounts","running","complete","incomplete","requiresAction","GroupPart","type","TKey","status","counts","indices","IndicatorPart","IndicatorMode","RenderInfo","part","children","Props","groupBy","context","indicator","info","shouldShowIndicator","mode","parts","isRunning","length","last","undefined","PartChildrenSentinel","Error","renderNode","node","render","idKey","index","groupPart","key","nodeKey","map","child","s","message","toolUIs","tools","memoKey","memoDep","tree","toolCallId","displayName"],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"sourcesContent":["\"use client\";\n\nimport { Fragment, type FC, type ReactNode, useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { useShallowSelector } from \"@assistant-ui/store/internal\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport type {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"../../../types/message\";\nimport { getGroupSummary } from \"../../../utils/getGroupStatus\";\nimport {\n  buildGroupTree,\n  GROUPBY_MEMO_KEY,\n  type GroupByContext,\n  type GroupNode,\n} from \"../../utils/groupParts\";\nimport { MessagePartChildren, type EnrichedPartState } from \"./MessageParts\";\n\nexport namespace MessagePrimitiveGroupedParts {\n  /** Per status tallies over the group's `indices`; they sum to `indices.length`. */\n  export type GroupCounts = {\n    readonly running: number;\n    readonly complete: number;\n    readonly incomplete: number;\n    readonly requiresAction: number;\n  };\n\n  /**\n   * A coalesced group of adjacent parts. Surfaced through the same\n   * `{ part }` channel as a leaf {@link EnrichedPartState} so consumers\n   * dispatch on a single `switch (part.type)`. `type` is the group key\n   * (always `\"group-…\"`); `status` is running when any contained part runs,\n   * otherwise it mirrors the last contained part.\n   */\n  export type GroupPart<TKey extends `group-${string}` = `group-${string}`> = {\n    readonly type: TKey;\n    readonly status: MessagePartStatus | ToolCallMessagePartStatus;\n    /** Per status tallies over `indices`. */\n    readonly counts: GroupCounts;\n    readonly indices: readonly number[];\n  };\n\n  /**\n   * Synthetic trailing slot for a streaming/loading affordance (a\n   * \"thinking…\" dot, etc.). Surfaced through the same `{ part }` channel\n   * as groups and leaf parts so a single `switch (part.type)` renders it\n   * via `case \"indicator\"`.\n   *\n   * It is only ever emitted while the message is running, so its presence\n   * alone means \"render your loading UI here\" — there's no `status` to\n   * branch on.\n   */\n  export type IndicatorPart = {\n    readonly type: \"indicator\";\n  };\n\n  /**\n   * When to emit the synthetic {@link IndicatorPart}. It is **only** emitted\n   * while the message is running (streaming); the mode further restricts\n   * which running states qualify:\n   * - `\"never\"` — never.\n   * - `\"empty\"` — only when the message has no parts yet.\n   * - `\"no-text\"` (default) — when the message has no parts yet or the last\n   *   part isn't `text`/`reasoning` (e.g. it ended on a tool call, so the\n   *   assistant likely isn't done).\n   * - `\"always\"` — whenever the message is running, regardless of parts.\n   */\n  export type IndicatorMode = \"never\" | \"empty\" | \"no-text\" | \"always\";\n\n  export type RenderInfo<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Either a coalesced group ({@link GroupPart}, identified by a\n     * `group-…` `type`), a single enriched leaf part, or the synthetic\n     * {@link IndicatorPart} (`type: \"indicator\"`). Use one switch over\n     * `part.type` to handle all three.\n     */\n    readonly part: GroupPart<TKey> | EnrichedPartState | IndicatorPart;\n    /**\n     * For group nodes: the recursively-rendered subtree (subgroups +\n     * leaf parts). For leaf parts: a sentinel that throws when rendered\n     * — accidental fall-through (`default: return children;`) errors\n     * loudly instead of silently rendering nothing.\n     */\n    readonly children: ReactNode;\n  };\n\n  export type Props<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Maps each part to a group-key path. Adjacent parts that share a\n     * prefix coalesce into the same group. Return `[]` (or `null`) to\n     * leave a part ungrouped.\n     *\n     * Group keys must start with `\"group-\"` so the renderer's\n     * `switch (part.type)` can tell groups apart from real part types.\n     *\n     * **Prefer {@link groupPartByType}** for the common case of mapping by\n     * `part.type` or tool name — it ships a stable memo fingerprint so the tree\n     * survives unrelated re-renders. Use an inline function only when\n     * the helper isn't expressive enough (e.g. branching on\n     * `part.parentId` or part metadata).\n     *\n     * The second argument is a {@link GroupByContext} carrying the tool-UI\n     * registry, for grouping that depends on it (e.g. standalone tool calls).\n     *\n     * @example\n     * ```tsx\n     * import { groupPartByType } from \"@assistant-ui/react\";\n     *\n     * <MessagePrimitive.GroupedParts\n     *   groupBy={groupPartByType({\n     *     reasoning: [\"group-thought\", \"group-reasoning\"],\n     *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n     *   })}\n     * >\n     * ```\n     */\n    readonly groupBy: (\n      part: PartState,\n      context: GroupByContext,\n    ) => readonly TKey[] | null;\n\n    /**\n     * Controls emission of the synthetic {@link IndicatorPart} — a\n     * trailing `{ part: { type: \"indicator\", status } }` render call you\n     * handle with `case \"indicator\"` to show loading/status UI.\n     *\n     * @default \"no-text\"\n     * @see IndicatorMode\n     */\n    readonly indicator?: IndicatorMode;\n\n    /**\n     * Render function called once per group node, once per leaf part, and\n     * (when the `indicator` condition is met) once for the trailing\n     * {@link IndicatorPart}. Switch on `part.type`: `\"group-…\"` cases wrap\n     * `children`; real part types (`\"text\"`, `\"tool-call\"`, …) render the\n     * part directly; `\"indicator\"` renders status/loading UI.\n     *\n     * Leaf parts receive the same {@link EnrichedPartState} that\n     * `<MessagePrimitive.Parts>` would produce (`toolUI`, `addResult`,\n     * `resume`, `respondToApproval`, `dataRendererUI`).\n     */\n    readonly children: (info: RenderInfo<TKey>) => ReactNode;\n  };\n}\n\nconst shouldShowIndicator = (\n  mode: MessagePrimitiveGroupedParts.IndicatorMode,\n  parts: readonly PartState[],\n  isRunning: boolean,\n): boolean => {\n  // The indicator is a streaming affordance — never show it on a settled\n  // message, whatever the mode.\n  if (!isRunning) return false;\n\n  switch (mode) {\n    case \"never\":\n      return false;\n    case \"always\":\n      return true;\n    case \"empty\":\n      return parts.length === 0;\n    case \"no-text\": {\n      const last = parts[parts.length - 1];\n      return (\n        last === undefined ||\n        (last.type !== \"text\" && last.type !== \"reasoning\")\n      );\n    }\n  }\n};\n\n/**\n * `children` placeholder passed for leaf-part renders. Leaf parts have no\n * inner subtree; rendering this sentinel signals the consumer wrote\n * `default: return children;` and accidentally fell through for a part —\n * surface the bug loudly instead of silently rendering nothing.\n */\nconst PartChildrenSentinel: FC = () => {\n  throw new Error(\n    \"MessagePrimitive.GroupedParts: rendered `children` under a leaf \" +\n      \"part. `children` is only meaningful for `group-…` cases — add a \" +\n      \"matching case for the part type or return `null` to skip it.\",\n  );\n};\n\nconst renderNode = <TKey extends `group-${string}`>(\n  node: GroupNode,\n  parts: readonly PartState[],\n  render: (info: MessagePrimitiveGroupedParts.RenderInfo<TKey>) => ReactNode,\n): ReactNode => {\n  if (node.type === \"part\") {\n    // Key by part identity when available, else absolute part index — never\n    // the structural nodeKey, which leaves zombie fiber subscriptions when\n    // parts reshape (#4051).\n    return (\n      <MessagePartChildren\n        key={node.idKey ? `part-${node.idKey}` : `part-${node.index}`}\n        index={node.index}\n      >\n        {({ part }) => render({ part, children: <PartChildrenSentinel /> })}\n      </MessagePartChildren>\n    );\n  }\n\n  const { status, counts } = getGroupSummary(parts, node.indices);\n  const groupPart: MessagePrimitiveGroupedParts.GroupPart<TKey> = {\n    type: node.key as TKey,\n    status,\n    counts,\n    indices: node.indices,\n  };\n\n  return (\n    <Fragment key={node.idKey ?? node.nodeKey}>\n      {render({\n        part: groupPart,\n        children: (\n          <>{node.children.map((child) => renderNode(child, parts, render))}</>\n        ),\n      })}\n    </Fragment>\n  );\n};\n\n/**\n * Groups adjacent message parts into a tree of coalesced runs and\n * renders each node — group or part — through a single `children`\n * function.\n *\n * The render function receives `{ part, children }` where `part.type`\n * is either a `\"group-…\"` literal (for a group, `children` is the\n * recursively-rendered subtree) or a real part type (`\"text\"`,\n * `\"tool-call\"`, …) for a leaf (`children` is a sentinel that throws\n * if rendered — use `part.type` to distinguish).\n *\n * @example\n * ```tsx\n * <MessagePrimitive.GroupedParts\n *   groupBy={groupPartByType({\n *     reasoning: [\"group-thought\", \"group-reasoning\"],\n *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n *   })}\n * >\n *   {({ part, children }) => {\n *     switch (part.type) {\n *       case \"group-thought\":   return <Thought>{children}</Thought>;\n *       case \"group-reasoning\": return <Reasoning>{children}</Reasoning>;\n *       case \"group-tool\":      return <ToolStack>{children}</ToolStack>;\n *       case \"text\":            return <MarkdownText />;\n *       case \"tool-call\":       return part.toolUI ?? <ToolFallback {...part} />;\n *       case \"indicator\":       return <LoadingDots />;\n *       default:                return null;\n *     }\n *   }}\n * </MessagePrimitive.GroupedParts>\n * ```\n */\nexport const MessagePrimitiveGroupedParts = <TKey extends `group-${string}`>({\n  groupBy,\n  indicator = \"no-text\",\n  children,\n}: MessagePrimitiveGroupedParts.Props<TKey>): ReactNode => {\n  const parts = useAuiState(useShallowSelector((s) => s.message.parts));\n  // Handed to `groupBy` as its `context` argument (see GroupByContext).\n  const toolUIs = useAuiState((s) => s.tools.toolUIs);\n  // Subscribe to a boolean, not the status object: the tree only needs to\n  // re-render when running-ness flips, and `\"never\"` opts out entirely.\n  const isRunning = useAuiState((s) =>\n    indicator === \"never\" ? false : s.message.status?.type === \"running\",\n  );\n\n  // Helpers like `groupPartByType` tag the function with `GROUPBY_MEMO_KEY`\n  // (a stable string fingerprint of the helper config). When present,\n  // memo on `[parts, memoKey]` so the tree survives unrelated renders.\n  // For inline `groupBy`, fall back to recomputing each render — O(n)\n  // and cheap.\n  const memoKey = (groupBy as { [GROUPBY_MEMO_KEY]?: string })[\n    GROUPBY_MEMO_KEY\n  ];\n  const memoDep = memoKey ?? groupBy;\n  const tree = useMemo(() => {\n    const context: GroupByContext = { toolUIs };\n    return buildGroupTree(\n      parts.map((part) => groupBy(part, context) ?? []),\n      parts.map((part) =>\n        part.type === \"tool-call\" ? part.toolCallId : undefined,\n      ),\n    );\n    // oxlint-disable-next-line react/exhaustive-deps -- groupBy is captured via memoDep (either its identity or the helper's memoKey fingerprint); listing it directly would defeat the helper-tagged memo path\n  }, [parts, memoDep, toolUIs]);\n\n  return (\n    <>\n      {tree.map((node) => renderNode(node, parts, children))}\n      {shouldShowIndicator(indicator, parts, isRunning) &&\n        children({\n          part: { type: \"indicator\" },\n          children: <PartChildrenSentinel />,\n        })}\n    </>\n  );\n};\n\nMessagePrimitiveGroupedParts.displayName = \"MessagePrimitive.GroupedParts\";\n"],"mappings":";;;;;;;;;AAmJA,MAAMsC,uBACJC,MACAC,OACAC,cACY;CAGZ,IAAI,CAACA,WAAW,OAAO;CAEvB,QAAQF,MAAR;EACE,KAAK,SACH,OAAO;EACT,KAAK,UACH,OAAO;EACT,KAAK,SACH,OAAOC,MAAME,WAAW;EAC1B,KAAK,WAAW;GACd,MAAMC,OAAOH,MAAMA,MAAME,SAAS;GAClC,OACEC,SAASC,KAAAA,KACRD,KAAKpB,SAAS,UAAUoB,KAAKpB,SAAS;EAE3C;CACF;AACF;;;;;;;AAQA,MAAMsB,6BAAiC;CACrC,MAAM,IAAIC,MACR,8LAGF;AACF;AAEA,MAAMC,cACJC,MACAR,OACAS,WACc;CACd,IAAID,KAAKzB,SAAS,QAIhB,OACE,oBAAC,qBAAD;EAEE,OAAOyB,KAAKG;EAEV,WAAA,EAAEpB,WAAWkB,OAAO;GAAElB;GAAMC,UAAU,oBAAC,sBAAD,CAAqB,CAAA;EAAI,CAAC;CAC/C,GAJdgB,KAAKE,QAAQ,QAAQF,KAAKE,UAAU,QAAQF,KAAKG,OAInC;CAIzB,MAAM,EAAE1B,QAAQC,WAAWjB,gBAAgB+B,OAAOQ,KAAKrB,OAAO;CAC9D,MAAMyB,YAA0D;EAC9D7B,MAAMyB,KAAKK;EACX5B;EACAC;EACAC,SAASqB,KAAKrB;CAChB;CAEA,OACE,oBAAC,UAAD,EAAA,UACGsB,OAAO;EACNlB,MAAMqB;EACNpB,UACE,oBAAA,YAAA,EAAA,UAAGgB,KAAKhB,SAASuB,KAAKC,UAAUT,WAAWS,OAAOhB,OAAOS,MAAM,CAAC,EAAE,CAAA;CAEtE,CAAC,EACO,GAPKD,KAAKE,SAASF,KAAKM,OAOxB;AAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmCA,MAAatC,gCAAgE,EAC3EkB,SACAE,YAAY,WACZJ,eACyD;CACzD,MAAMQ,QAAQpC,YAAYC,oBAAoBoD,MAAMA,EAAEC,QAAQlB,KAAK,CAAC;CAEpE,MAAMmB,UAAUvD,aAAaqD,QAAMA,IAAEG,MAAMD,OAAO;CAGlD,MAAMlB,YAAYrC,aAAaqD,QAC7BrB,cAAc,UAAU,QAAQqB,IAAEC,QAAQjC,QAAQF,SAAS,SAC7D;CAUA,MAAMuC,UAHW5B,QACfvB,qBAEyBuB;CAC3B,MAAM6B,OAAO5D,cAAc;EACzB,MAAMgC,UAA0B,EAAEwB,QAAQ;EAC1C,OAAOjD,eACL8B,MAAMe,KAAKxB,SAASG,QAAQH,MAAMI,OAAO,KAAK,CAAA,CAAE,GAChDK,MAAMe,KAAKxB,WACTA,OAAKR,SAAS,cAAcQ,OAAKiC,aAAapB,KAAAA,CAChD,CACF;CAEF,GAAG;EAACJ;EAAOsB;EAASH;CAAO,CAAC;CAE5B,OACE,qBAAA,YAAA,EAAA,UAAA,CACGI,KAAKR,KAAKP,SAASD,WAAWC,MAAMR,OAAOR,QAAQ,CAAC,GACpDM,oBAAoBF,WAAWI,OAAOC,SAAS,KAC9CT,SAAS;EACPD,MAAM,EAAER,MAAM,YAAY;EAC1BS,UAAU,oBAAC,sBAAD,CAAqB,CAAA;CACjC,CAAC,CACL,EAAA,CAAA;AAEJ;AAEAhB,6BAA6BiD,cAAc"}

package/dist/react/primitives/message/MessageParts.d.ts

@@ -28,8 +28,8 @@ type MessagePartRange = {
  * `partIds[i]` optionally carries a stable identity for part `i`; group
  * ranges derive an `idKey` from their first part's id (first claim wins).
  */
-declare const groupMessageParts: (messageTypes: readonly string[], useChainOfThought: boolean, partIds?: readonly (string | undefined)[]) => MessagePartRange[];
-declare namespace MessagePrimitiveParts {
+export declare const groupMessageParts: (messageTypes: readonly string[], useChainOfThought: boolean, partIds?: readonly (string | undefined)[]) => MessagePartRange[];
+export declare namespace MessagePrimitiveParts {
   type DataConfig = {
     /** Map data event names to specific components */
     by_name?: Record<string, DataMessagePartComponent | undefined> | undefined;
@@ -172,7 +172,7 @@ declare namespace MessagePrimitiveParts {
  * Platform-agnostic no-op default components.
  * Each platform (web, RN) wraps MessagePrimitiveParts with its own defaults.
  */
-declare const defaultComponents: {
+export declare const defaultComponents: {
   Text: () => null;
   Reasoning: () => null;
   Source: () => null;
@@ -185,8 +185,8 @@ declare const defaultComponents: {
 type MessagePartComponentProps = {
   components: MessagePrimitiveParts.Props["components"];
 };
-declare const MessagePartComponent: FC<MessagePartComponentProps>;
-declare namespace MessagePrimitivePartByIndex {
+export declare const MessagePartComponent: FC<MessagePartComponentProps>;
+export declare namespace MessagePrimitivePartByIndex {
   type Props = {
     index: number;
     components: MessagePrimitiveParts.Props["components"];
@@ -195,7 +195,7 @@ declare namespace MessagePrimitivePartByIndex {
 /**
  * Renders a single message part at the specified index.
  */
-declare const MessagePrimitivePartByIndex: FC<MessagePrimitivePartByIndex.Props>;
+export declare const MessagePrimitivePartByIndex: FC<MessagePrimitivePartByIndex.Props>;
 /**
  * Enriched part state passed to children render functions.
  *
@@ -208,7 +208,7 @@ declare const MessagePrimitivePartByIndex: FC<MessagePrimitivePartByIndex.Props>
  * loading indicator can render. Differentiate this from a real empty text
  * via `part.status?.type === "running" && part.text === ""`.
  */
-type EnrichedPartState = (Extract<PartState, {
+export type EnrichedPartState = (Extract<PartState, {
   type: "tool-call";
 }> & {
   /** The registered tool UI element, or null if none registered. */
@@ -242,14 +242,14 @@ type MessagePartChildrenProps = {
     part: EnrichedPartState;
   }) => ReactNode;
 };
-declare const MessagePartChildren: FC<MessagePartChildrenProps>;
+export declare const MessagePartChildren: FC<MessagePartChildrenProps>;
 /**
  * Renders the parts of a message with support for multiple content types.
  *
  * This is the platform-agnostic base. Each platform wraps this with its own
  * default components (web uses `<p>`, `<span>`; RN would use `<Text>`, etc.).
  */
-declare const MessagePrimitiveParts: FC<MessagePrimitiveParts.Props>;
+export declare const MessagePrimitiveParts: FC<MessagePrimitiveParts.Props>;
 //#endregion
-export { EnrichedPartState, MessagePartChildren, MessagePartComponent, MessagePrimitivePartByIndex, MessagePrimitiveParts, type PartState, defaultComponents, groupMessageParts };
+export type { PartState };
 //# sourceMappingURL=MessageParts.d.ts.map

package/dist/react/primitives/message/MessageParts.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageParts.d.ts","names":[],"sources":["../../../../src/react/primitives/message/MessageParts.tsx"],"mappings":";;;;KA4CK;EACC;EAAgB;;EAEhB;EACA;EACA;EACA;;EAGA;EACA;EACA;EACA;;EAGA;EACA;EACA;EACA;;;;;;;;;cAiDO,oBACX,iCACA,4BACA,8CACC;kBAkFc;OACV;;IAEH,UAAU,eAAe;;IAEzB,WAAW;;OAGR;;IAEH,QAAQ;;IAER,OAAO;;IAEP,SAAS;;IAET,QAAQ;;IAER,OAAO;;;;;;;IAOP,iBAAiB;;IAEjB,OAAO;;IAEP,QAAQ;;;;;;;;;IASR;;MAGM,YAAY;;MAEZ,WACI;QAAgB;QAAmB;;;;OAM1C;;IAGC,UACI,eAAe;;IAGnB,WAAW,cAAc;;;IAIzB,UAAU,cAAc;;;;;;;;OASzB,qBAAqB;;IAExB,YAAY;;IAEZ,QAAQ;;;;;;;;;;IAWR,YAAY,cACV;MAAoB;MAAoB;;;;;;;;;;;IAY1C,iBAAiB;IAEjB;;;;;;;;;OAUG,2BAA2B;;;;;;IAM9B,gBAAgB;IAEhB;IACA;IACA;IACA;;cAGU;;;;;;;;IASN,aAAa,qBAAqB;;;;;;;;IAQlC;IACA;;;IAIA,WAAW;MAAS,MAAM;UAAwB;IAClD;IACA;;;;;;;;cA4CK;;;;;;;EAOe,cAAA,YAAA,sBAAiB;EACZ,mBAAA,YAAA,sBAAiB;;KAG7C;EACH,YAAY,sBAAsB;;cAGvB,sBAAsB,GAAG;kBAkGrB;OACH;IACV;IACA,YAAY,sBAAsB;;;;;;cAOzB,6BAA6B,GAAG,4BAA4B;;;;;;;;;;;;;KA+L7D,qBACP,QAAQ;EAAa;;;WAEX,QAAQ;;EAEjB,WAAW;;EAEX,QAAQ;;EAER,mBAAmB;MAEpB,QAAQ;EAAa;;;WAEX,gBAAgB;KAE3B,QAAQ;EAAa;;EAAwB;;;;;;;;;KAgB5C;EACH;EACA,WAAW;IAAS,MAAM;QAAwB;;cAgDvC,qBAAqB,GAAG;;;;;;;cA8CxB,uBAAuB,GAAG,sBAAsB"}
+{"version":3,"file":"MessageParts.d.ts","names":[],"sources":["../../../../src/react/primitives/message/MessageParts.tsx"],"mappings":";;;;KA4CK;EACC;EAAgB;;EAEhB;EACA;EACA;EACA;;EAGA;EACA;EACA;EACA;;EAGA;EACA;EACA;EACA;;;;;;;;;qBAiDO,oBACX,iCACA,4BACA,8CACC;yBAkFc;OACV;;IAEH,UAAU,eAAe;;IAEzB,WAAW;;OAGR;;IAEH,QAAQ;;IAER,OAAO;;IAEP,SAAS;;IAET,QAAQ;;IAER,OAAO;;;;;;;IAOP,iBAAiB;;IAEjB,OAAO;;IAEP,QAAQ;;;;;;;;;IASR;;MAGM,YAAY;;MAEZ,WACI;QAAgB;QAAmB;;;;OAM1C;;IAGC,UACI,eAAe;;IAGnB,WAAW,cAAc;;;IAIzB,UAAU,cAAc;;;;;;;;OASzB,qBAAqB;;IAExB,YAAY;;IAEZ,QAAQ;;;;;;;;;;IAWR,YAAY,cACV;MAAoB;MAAoB;;;;;;;;;;;IAY1C,iBAAiB;IAEjB;;;;;;;;;OAUG,2BAA2B;;;;;;IAM9B,gBAAgB;IAEhB;IACA;IACA;IACA;;cAGU;;;;;;;;IASN,aAAa,qBAAqB;;;;;;;;IAQlC;IACA;;;IAIA,WAAW;MAAS,MAAM;UAAwB;IAClD;IACA;;;;;;;;qBA4CK;;;;;;;EAOe,cAAA,YAAA,sBAAiB;EACZ,mBAAA,YAAA,sBAAiB;;KAG7C;EACH,YAAY,sBAAsB;;qBAGvB,sBAAsB,GAAG;yBAkGrB;OACH;IACV;IACA,YAAY,sBAAsB;;;;;;qBAOzB,6BAA6B,GAAG,4BAA4B;;;;;;;;;;;;;YA+L7D,qBACP,QAAQ;EAAa;;;WAEX,QAAQ;;EAEjB,WAAW;;EAEX,QAAQ;;EAER,mBAAmB;MAEpB,QAAQ;EAAa;;;WAEX,gBAAgB;KAE3B,QAAQ;EAAa;;EAAwB;;;;;;;;;KAgB5C;EACH;EACA,WAAW;IAAS,MAAM;QAAwB;;qBAgDvC,qBAAqB,GAAG;;;;;;;qBA8CxB,uBAAuB,GAAG,sBAAsB"}

package/dist/react/primitives/message/MessageQuote.d.ts

@@ -1,13 +1,12 @@
 import { QuoteInfo } from "../../../types/quote.js";
 import { ReactNode } from "react";
 //#region src/react/primitives/message/MessageQuote.d.ts
-declare namespace MessagePrimitiveQuote {
+export declare namespace MessagePrimitiveQuote {
   type Props = {
     /** Render function called when a quote is present. Receives quote info. */
     children: (value: QuoteInfo) => ReactNode;
   };
 }
-declare const MessagePrimitiveQuote: import("react").NamedExoticComponent<MessagePrimitiveQuote.Props>;
+export declare const MessagePrimitiveQuote: import("react").NamedExoticComponent<MessagePrimitiveQuote.Props>;
 //#endregion
-export { MessagePrimitiveQuote };
 //# sourceMappingURL=MessageQuote.d.ts.map

package/dist/react/primitives/message/MessageQuote.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageQuote.d.ts","names":[],"sources":["../../../../src/react/primitives/message/MessageQuote.tsx"],"mappings":";;;kBAKiB;OACH;;IAEV,WAAW,OAAO,cAAc;;;cA0BvB,uCAAqB,qBAAA,sBAAA"}
+{"version":3,"file":"MessageQuote.d.ts","names":[],"sources":["../../../../src/react/primitives/message/MessageQuote.tsx"],"mappings":";;;yBAKiB;OACH;;IAEV,WAAW,OAAO,cAAc;;;qBA0BvB,uCAAqB,qBAAA,sBAAA"}

package/dist/react/primitives/messagePart/MessagePartInProgress.d.ts

@@ -1,9 +1,8 @@
 import { FC, PropsWithChildren } from "react";
 //#region src/react/primitives/messagePart/MessagePartInProgress.d.ts
-declare namespace MessagePartPrimitiveInProgress {
+export declare namespace MessagePartPrimitiveInProgress {
   type Props = PropsWithChildren;
 }
-declare const MessagePartPrimitiveInProgress: FC<MessagePartPrimitiveInProgress.Props>;
+export declare const MessagePartPrimitiveInProgress: FC<MessagePartPrimitiveInProgress.Props>;
 //#endregion
-export { MessagePartPrimitiveInProgress };
 //# sourceMappingURL=MessagePartInProgress.d.ts.map

package/dist/react/primitives/messagePart/MessagePartInProgress.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessagePartInProgress.d.ts","names":[],"sources":["../../../../src/react/primitives/messagePart/MessagePartInProgress.ts"],"mappings":";;kBAGiB;OACH,QAAQ;;cAGT,gCAAgC,GAC3C,+BAA+B"}
+{"version":3,"file":"MessagePartInProgress.d.ts","names":[],"sources":["../../../../src/react/primitives/messagePart/MessagePartInProgress.ts"],"mappings":";;yBAGiB;OACH,QAAQ;;qBAGT,gCAAgC,GAC3C,+BAA+B"}

package/dist/react/primitives/part/PartMessages.d.ts

@@ -2,7 +2,7 @@ import { ThreadMessage } from "../../../types/message.js";
 import { ThreadPrimitiveMessages } from "../thread/ThreadMessages.js";
 import { FC, ReactNode } from "react";
 //#region src/react/primitives/part/PartMessages.d.ts
-declare namespace PartPrimitiveMessages {
+export declare namespace PartPrimitiveMessages {
   type Props = {
     components: NonNullable<ThreadPrimitiveMessages.Props["components"]>;
     children?: never;
@@ -20,6 +20,7 @@ declare namespace PartPrimitiveMessages {
  * This primitive reads `messages` from the current tool call part in the PartScope
  * and renders them using a readonly thread context. All existing message and part
  * primitives work inside, and parent tool UI registrations are inherited.
+ * The nested thread reports `isRunning` while its trailing assistant message is running.
  *
  * @example
  * ```tsx
@@ -38,8 +39,7 @@ declare namespace PartPrimitiveMessages {
  * });
  * ```
  */
-declare const PartPrimitiveMessagesImpl: FC<PartPrimitiveMessages.Props>;
-declare const PartPrimitiveMessages: import("react").NamedExoticComponent<PartPrimitiveMessages.Props>;
+export declare const PartPrimitiveMessagesImpl: FC<PartPrimitiveMessages.Props>;
+export declare const PartPrimitiveMessages: import("react").NamedExoticComponent<PartPrimitiveMessages.Props>;
 //#endregion
-export { PartPrimitiveMessages, PartPrimitiveMessagesImpl };
 //# sourceMappingURL=PartMessages.d.ts.map

package/dist/react/primitives/part/PartMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"PartMessages.d.ts","names":[],"sources":["../../../../src/react/primitives/part/PartMessages.tsx"],"mappings":";;;;kBASiB;OACH;IAEN,YAAY,YAAY,wBAAwB;IAChD;;;IAIA,WAAW;MAAS,SAAS;UAAoB;IACjD;;;;;;;;;;;;;;;;;;;;;;;;;;;cAsCK,2BAA2B,GAAG,sBAAsB;cAyBpD,uCAAqB,qBAAA,sBAAA"}
+{"version":3,"file":"PartMessages.d.ts","names":[],"sources":["../../../../src/react/primitives/part/PartMessages.tsx"],"mappings":";;;;yBASiB;OACH;IAEN,YAAY,YAAY,wBAAwB;IAChD;;;IAIA,WAAW;MAAS,SAAS;UAAoB;IACjD;;;;;;;;;;;;;;;;;;;;;;;;;;;;qBAuCK,2BAA2B,GAAG,sBAAsB;qBAyBpD,uCAAqB,qBAAA,sBAAA"}

package/dist/react/primitives/part/PartMessages.js

@@ -14,6 +14,7 @@ const usePartMessages = () => {
 * This primitive reads `messages` from the current tool call part in the PartScope
 * and renders them using a readonly thread context. All existing message and part
 * primitives work inside, and parent tool UI registrations are inherited.
+* The nested thread reports `isRunning` while its trailing assistant message is running.
 *
 * @example
 * ```tsx

package/dist/react/primitives/part/PartMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"PartMessages.js","names":["FC","ReactNode","memo","useAuiState","ThreadMessage","ReadonlyThreadProvider","ThreadPrimitiveMessages","ThreadPrimitiveMessagesImpl","PartPrimitiveMessages","Props","components","NonNullable","children","value","message","usePartMessages","_temp","PartPrimitiveMessagesImpl","t0","$","_c","messages","length","t1","t2","displayName","s","part","type","undefined"],"sources":["../../../../src/react/primitives/part/PartMessages.tsx"],"sourcesContent":["import { type FC, type ReactNode, memo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport type { ThreadMessage } from \"../../../types/message\";\nimport { ReadonlyThreadProvider } from \"../../providers/ReadonlyThreadProvider\";\nimport {\n  type ThreadPrimitiveMessages,\n  ThreadPrimitiveMessagesImpl,\n} from \"../thread/ThreadMessages\";\n\nexport namespace PartPrimitiveMessages {\n  export type Props =\n    | {\n        components: NonNullable<ThreadPrimitiveMessages.Props[\"components\"]>;\n        children?: never;\n      }\n    | {\n        /** Render function called for each message. Receives the message. */\n        children: (value: { message: ThreadMessage }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst usePartMessages = (): readonly ThreadMessage[] | undefined => {\n  return useAuiState((s) => {\n    const part = s.part;\n    if (part.type !== \"tool-call\") return undefined;\n    return \"messages\" in part\n      ? (part.messages as readonly ThreadMessage[] | undefined)\n      : undefined;\n  });\n};\n\n/**\n * Renders the nested messages of a tool call part (e.g. sub-agent conversation).\n *\n * This primitive reads `messages` from the current tool call part in the PartScope\n * and renders them using a readonly thread context. All existing message and part\n * primitives work inside, and parent tool UI registrations are inherited.\n *\n * @example\n * ```tsx\n * const toolkit = defineToolkit({\n *   invoke_sub_agent: {\n *     type: \"backend\",\n *     render: () => (\n *       <PartPrimitive.Messages>\n *         {({ message }) => {\n *           if (message.role === \"user\") return <MyUserMessage />;\n *           return <MyAssistantMessage />;\n *         }}\n *       </PartPrimitive.Messages>\n *     ),\n *   },\n * });\n * ```\n */\nexport const PartPrimitiveMessagesImpl: FC<PartPrimitiveMessages.Props> = ({\n  components,\n  children,\n}) => {\n  const messages = usePartMessages();\n\n  if (!messages?.length) return null;\n\n  if (components) {\n    return (\n      <ReadonlyThreadProvider messages={messages}>\n        <ThreadPrimitiveMessagesImpl components={components} />\n      </ReadonlyThreadProvider>\n    );\n  }\n\n  return (\n    <ReadonlyThreadProvider messages={messages}>\n      <ThreadPrimitiveMessagesImpl>{children}</ThreadPrimitiveMessagesImpl>\n    </ReadonlyThreadProvider>\n  );\n};\n\nPartPrimitiveMessagesImpl.displayName = \"PartPrimitive.Messages\";\n\nexport const PartPrimitiveMessages = memo(PartPrimitiveMessagesImpl);\n"],"mappings":";;;;;;;AAsBA,MAAMe,wBAAkB;CAAA,OACfZ,YAAYa,KAMlB;AAAC;;;;;;;;;;;;;;;;;;;;;;;;;AA2BJ,MAAaC,6BAA6DC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAV,YAAAE,aAAAM;CAIzE,MAAAG,WAAiBN,gBAAgB;CAEjC,IAAI,CAACM,UAAQC,QAAQ,OAAS;CAE9B,IAAIZ,YAAU;EAAA,IAAAa;EAAA,IAAAJ,EAAA,OAAAT,YAAA;GAGRa,KAAA,oBAAC,6BAAD,EAAyCb,WAAU,CAAA;GAAIS,EAAA,KAAAT;GAAAS,EAAA,KAAAI;EAAA,OAAAA,KAAAJ,EAAA;EAAA,IAAAK;EAAA,IAAAL,EAAA,OAAAE,YAAAF,EAAA,OAAAI,IAAA;GADzDC,KAAA,oBAAC,wBAAD;IAAkCH;IAChCE,UAAAA;GADqB,CAAA;GAEEJ,EAAA,KAAAE;GAAAF,EAAA,KAAAI;GAAAJ,EAAA,KAAAK;EAAA,OAAAA,KAAAL,EAAA;EAAA,OAFzBK;CAEyB;CAE5B,IAAAD;CAAA,IAAAJ,EAAA,OAAAP,UAAA;EAIGW,KAAA,oBAAC,6BAAD,EAA8BX,SAAF,CAAA;EAAyCO,EAAA,KAAAP;EAAAO,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,IAAAK;CAAA,IAAAL,EAAA,OAAAE,YAAAF,EAAA,OAAAI,IAAA;EADvEC,KAAA,oBAAC,wBAAD;GAAkCH;GAChCE,UAAAA;EADqB,CAAA;EAEEJ,EAAA,KAAAE;EAAAF,EAAA,KAAAI;EAAAJ,EAAA,KAAAK;CAAA,OAAAA,KAAAL,EAAA;CAAA,OAFzBK;AAEyB;AAI7BP,0BAA0BQ,cAAc;AAExC,MAAajB,wBAAwBN,KAAKe,yBAAyB;AA3D3C,SAAAD,MAAAU,GAAA;CAEpB,MAAAC,OAAaD,EAACC;CACd,IAAIA,KAAIC,SAAU,aAAW;CAAmB,OACzC,cAAcD,OAChBA,KAAIN,WADFQ,KAAAA;AAEM"}
+{"version":3,"file":"PartMessages.js","names":["FC","ReactNode","memo","useAuiState","ThreadMessage","ReadonlyThreadProvider","ThreadPrimitiveMessages","ThreadPrimitiveMessagesImpl","PartPrimitiveMessages","Props","components","NonNullable","children","value","message","usePartMessages","_temp","PartPrimitiveMessagesImpl","t0","$","_c","messages","length","t1","t2","displayName","s","part","type","undefined"],"sources":["../../../../src/react/primitives/part/PartMessages.tsx"],"sourcesContent":["import { type FC, type ReactNode, memo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport type { ThreadMessage } from \"../../../types/message\";\nimport { ReadonlyThreadProvider } from \"../../providers/ReadonlyThreadProvider\";\nimport {\n  type ThreadPrimitiveMessages,\n  ThreadPrimitiveMessagesImpl,\n} from \"../thread/ThreadMessages\";\n\nexport namespace PartPrimitiveMessages {\n  export type Props =\n    | {\n        components: NonNullable<ThreadPrimitiveMessages.Props[\"components\"]>;\n        children?: never;\n      }\n    | {\n        /** Render function called for each message. Receives the message. */\n        children: (value: { message: ThreadMessage }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst usePartMessages = (): readonly ThreadMessage[] | undefined => {\n  return useAuiState((s) => {\n    const part = s.part;\n    if (part.type !== \"tool-call\") return undefined;\n    return \"messages\" in part\n      ? (part.messages as readonly ThreadMessage[] | undefined)\n      : undefined;\n  });\n};\n\n/**\n * Renders the nested messages of a tool call part (e.g. sub-agent conversation).\n *\n * This primitive reads `messages` from the current tool call part in the PartScope\n * and renders them using a readonly thread context. All existing message and part\n * primitives work inside, and parent tool UI registrations are inherited.\n * The nested thread reports `isRunning` while its trailing assistant message is running.\n *\n * @example\n * ```tsx\n * const toolkit = defineToolkit({\n *   invoke_sub_agent: {\n *     type: \"backend\",\n *     render: () => (\n *       <PartPrimitive.Messages>\n *         {({ message }) => {\n *           if (message.role === \"user\") return <MyUserMessage />;\n *           return <MyAssistantMessage />;\n *         }}\n *       </PartPrimitive.Messages>\n *     ),\n *   },\n * });\n * ```\n */\nexport const PartPrimitiveMessagesImpl: FC<PartPrimitiveMessages.Props> = ({\n  components,\n  children,\n}) => {\n  const messages = usePartMessages();\n\n  if (!messages?.length) return null;\n\n  if (components) {\n    return (\n      <ReadonlyThreadProvider messages={messages}>\n        <ThreadPrimitiveMessagesImpl components={components} />\n      </ReadonlyThreadProvider>\n    );\n  }\n\n  return (\n    <ReadonlyThreadProvider messages={messages}>\n      <ThreadPrimitiveMessagesImpl>{children}</ThreadPrimitiveMessagesImpl>\n    </ReadonlyThreadProvider>\n  );\n};\n\nPartPrimitiveMessagesImpl.displayName = \"PartPrimitive.Messages\";\n\nexport const PartPrimitiveMessages = memo(PartPrimitiveMessagesImpl);\n"],"mappings":";;;;;;;AAsBA,MAAMe,wBAAkB;CAAA,OACfZ,YAAYa,KAMlB;AAAC;;;;;;;;;;;;;;;;;;;;;;;;;;AA4BJ,MAAaC,6BAA6DC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAV,YAAAE,aAAAM;CAIzE,MAAAG,WAAiBN,gBAAgB;CAEjC,IAAI,CAACM,UAAQC,QAAQ,OAAS;CAE9B,IAAIZ,YAAU;EAAA,IAAAa;EAAA,IAAAJ,EAAA,OAAAT,YAAA;GAGRa,KAAA,oBAAC,6BAAD,EAAyCb,WAAU,CAAA;GAAIS,EAAA,KAAAT;GAAAS,EAAA,KAAAI;EAAA,OAAAA,KAAAJ,EAAA;EAAA,IAAAK;EAAA,IAAAL,EAAA,OAAAE,YAAAF,EAAA,OAAAI,IAAA;GADzDC,KAAA,oBAAC,wBAAD;IAAkCH;IAChCE,UAAAA;GADqB,CAAA;GAEEJ,EAAA,KAAAE;GAAAF,EAAA,KAAAI;GAAAJ,EAAA,KAAAK;EAAA,OAAAA,KAAAL,EAAA;EAAA,OAFzBK;CAEyB;CAE5B,IAAAD;CAAA,IAAAJ,EAAA,OAAAP,UAAA;EAIGW,KAAA,oBAAC,6BAAD,EAA8BX,SAAF,CAAA;EAAyCO,EAAA,KAAAP;EAAAO,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,IAAAK;CAAA,IAAAL,EAAA,OAAAE,YAAAF,EAAA,OAAAI,IAAA;EADvEC,KAAA,oBAAC,wBAAD;GAAkCH;GAChCE,UAAAA;EADqB,CAAA;EAEEJ,EAAA,KAAAE;EAAAF,EAAA,KAAAI;EAAAJ,EAAA,KAAAK;CAAA,OAAAA,KAAAL,EAAA;CAAA,OAFzBK;AAEyB;AAI7BP,0BAA0BQ,cAAc;AAExC,MAAajB,wBAAwBN,KAAKe,yBAAyB;AA5D3C,SAAAD,MAAAU,GAAA;CAEpB,MAAAC,OAAaD,EAACC;CACd,IAAIA,KAAIC,SAAU,aAAW;CAAmB,OACzC,cAAcD,OAChBA,KAAIN,WADFQ,KAAAA;AAEM"}

package/dist/react/primitives/thread/ThreadMessages.d.ts

@@ -37,7 +37,7 @@ type MessagesComponentConfig = {
   /** Component used to render system messages */
   SystemMessage?: ComponentType | undefined;
 };
-declare namespace ThreadPrimitiveMessages {
+export declare namespace ThreadPrimitiveMessages {
   type Props = {
     /** @deprecated Use the children render function instead. */
     components: MessagesComponentConfig;
@@ -50,7 +50,7 @@ declare namespace ThreadPrimitiveMessages {
     components?: never;
   };
 }
-declare namespace ThreadPrimitiveMessageByIndex {
+export declare namespace ThreadPrimitiveMessageByIndex {
   type Props = {
     index: number;
     components: MessagesComponentConfig;
@@ -59,8 +59,8 @@ declare namespace ThreadPrimitiveMessageByIndex {
 /**
  * Renders a single message at the specified index in the current thread.
  */
-declare const ThreadPrimitiveMessageByIndex: FC<ThreadPrimitiveMessageByIndex.Props>;
-declare namespace ThreadPrimitiveUnstable_MessageById {
+export declare const ThreadPrimitiveMessageByIndex: FC<ThreadPrimitiveMessageByIndex.Props>;
+export declare namespace ThreadPrimitiveUnstable_MessageById {
   type Props = {
     messageId: string;
     components: MessagesComponentConfig;
@@ -89,7 +89,7 @@ declare namespace ThreadPrimitiveUnstable_MessageById {
  * ));
  * ```
  */
-declare const ThreadPrimitiveUnstable_MessageById: FC<ThreadPrimitiveUnstable_MessageById.Props>;
+export declare const ThreadPrimitiveUnstable_MessageById: FC<ThreadPrimitiveUnstable_MessageById.Props>;
 /**
  * Renders all messages in the current thread.
  *
@@ -103,8 +103,7 @@ declare const ThreadPrimitiveUnstable_MessageById: FC<ThreadPrimitiveUnstable_Me
  * </ThreadPrimitive.Messages>
  * ```
  */
-declare const ThreadPrimitiveMessagesImpl: FC<ThreadPrimitiveMessages.Props>;
-declare const ThreadPrimitiveMessages: import("react").NamedExoticComponent<ThreadPrimitiveMessages.Props>;
+export declare const ThreadPrimitiveMessagesImpl: FC<ThreadPrimitiveMessages.Props>;
+export declare const ThreadPrimitiveMessages: import("react").NamedExoticComponent<ThreadPrimitiveMessages.Props>;
 //#endregion
-export { ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveUnstable_MessageById };
 //# sourceMappingURL=ThreadMessages.d.ts.map

package/dist/react/primitives/thread/ThreadMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ThreadMessages.d.ts","names":[],"sources":["../../../../src/react/primitives/thread/ThreadMessages.tsx"],"mappings":";;;;KAYK;;EAGC,SAAS;;EAET,eAAe;;EAEf,mBAAmB;;EAEnB,wBAAwB;;EAExB,qBAAqB;;EAErB,cAAc;;EAEd,mBAAmB;;EAEnB,gBAAgB;;;EAIhB,UAAU;;EAEV,eAAe;;EAEf,mBAAmB;;EAEnB,wBAAwB;;EAExB,qBAAqB;;EAErB,aAAa;;EAEb,kBAAkB;;EAElB,gBAAgB;;kBAGL;OACH;;IAGN,YAAY;IACZ;;;IAIA,WAAW;MAAS,SAAS;UAAmB;IAChD;;;kBAsGS;OACH;IACV;IACA,YAAY;;;;;;cAOH,+BAA+B,GAAG,8BAA8B;kBAgB5D;OACH;IACV;IACA,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;cA2BH,qCAAqC,GAAG,oCAAoC;;;;;;;;;;;;;;cA4D5E,6BAA6B,GAAG,wBAAwB;cAkBxD,yCAAuB,qBAAA,wBAAA"}
+{"version":3,"file":"ThreadMessages.d.ts","names":[],"sources":["../../../../src/react/primitives/thread/ThreadMessages.tsx"],"mappings":";;;;KAaK;;EAGC,SAAS;;EAET,eAAe;;EAEf,mBAAmB;;EAEnB,wBAAwB;;EAExB,qBAAqB;;EAErB,cAAc;;EAEd,mBAAmB;;EAEnB,gBAAgB;;;EAIhB,UAAU;;EAEV,eAAe;;EAEf,mBAAmB;;EAEnB,wBAAwB;;EAExB,qBAAqB;;EAErB,aAAa;;EAEb,kBAAkB;;EAElB,gBAAgB;;yBAGL;OACH;;IAGN,YAAY;IACZ;;;IAIA,WAAW;MAAS,SAAS;UAAmB;IAChD;;;yBAsGS;OACH;IACV;IACA,YAAY;;;;;;qBAOH,+BAA+B,GAAG,8BAA8B;yBAgB5D;OACH;IACV;IACA,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;qBA2BH,qCAAqC,GAAG,oCAAoC;;;;;;;;;;;;;;qBA8D5E,6BAA6B,GAAG,wBAAwB;qBAkBxD,yCAAuB,qBAAA,wBAAA"}

package/dist/react/primitives/thread/ThreadMessages.js

@@ -3,6 +3,7 @@ import { MessageByIdProvider } from "../../providers/MessageByIdProvider.js";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { memo, useMemo } from "@assistant-ui/tap/react-shim";
 import { RenderChildrenWithAccessor, useAuiState } from "@assistant-ui/store";
+import { useShallowSelector } from "@assistant-ui/store/internal";
 import { jsx } from "react/jsx-runtime";
 //#region src/react/primitives/thread/ThreadMessages.tsx
 const isComponentsSame = (prev, next) => {
@@ -129,10 +130,10 @@ const ThreadPrimitiveUnstable_MessageById = memo((t0) => {
 }, (prev, next) => prev.messageId === next.messageId && isComponentsSame(prev.components, next.components));
 ThreadPrimitiveUnstable_MessageById.displayName = "ThreadPrimitive.Unstable_MessageById";
 const ThreadPrimitiveMessagesInner = ({ children }) => {
-	const messagesLength = useAuiState((s) => s.thread.messages.length);
+	const messageIds = useAuiState(useShallowSelector((s) => s.thread.messages.map((message) => message.id)));
 	return useMemo(() => {
-		if (messagesLength === 0) return null;
-		return Array.from({ length: messagesLength }, (_, index) => /* @__PURE__ */ jsx(MessageByIndexProvider, {
+		if (messageIds.length === 0) return null;
+		return messageIds.map((messageId, index) => /* @__PURE__ */ jsx(MessageByIndexProvider, {
 			index,
 			children: /* @__PURE__ */ jsx(RenderChildrenWithAccessor, {
 				getItemState: (aui) => aui.thread.message({ index }).getState(),
@@ -140,8 +141,8 @@ const ThreadPrimitiveMessagesInner = ({ children }) => {
 					return getItem();
 				} })
 			})
-		}, index));
-	}, [messagesLength, children]);
+		}, messageId));
+	}, [messageIds, children]);
 };
 /**
 * Renders all messages in the current thread.

package/dist/react/primitives/thread/ThreadMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"ThreadMessages.js","names":["ComponentType","FC","ReactNode","memo","useMemo","RenderChildrenWithAccessor","useAuiState","MessageByIndexProvider","MessageByIdProvider","MessageState","MessagesComponentConfig","Message","EditComposer","UserEditComposer","AssistantEditComposer","SystemEditComposer","UserMessage","AssistantMessage","SystemMessage","ThreadPrimitiveMessages","Props","components","children","value","message","isComponentsSame","prev","next","DEFAULT_SYSTEM_MESSAGE","messageIdSetCache","WeakMap","ReadonlySet","hasMessageId","messages","messageId","ids","get","Set","map","m","id","set","has","getComponent","role","isEditing","_exhaustiveCheck","Error","ThreadMessageComponentProps","ThreadMessageComponent","t0","$","_c","_temp","_temp2","t1","Component","t2","ThreadPrimitiveMessageByIndex","index","displayName","ThreadPrimitiveUnstable_MessageById","s","thread","exists","t3","ThreadPrimitiveMessagesInner","messagesLength","length","Array","from","_","aui","getState","getItem","ThreadPrimitiveMessagesImpl","s_0","composer"],"sources":["../../../../src/react/primitives/thread/ThreadMessages.tsx"],"sourcesContent":["import {\n  type ComponentType,\n  type FC,\n  type ReactNode,\n  memo,\n  useMemo,\n} from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport { MessageByIndexProvider } from \"../../providers/MessageByIndexProvider\";\nimport { MessageByIdProvider } from \"../../providers/MessageByIdProvider\";\nimport type { MessageState } from \"../../../store\";\n\ntype MessagesComponentConfig =\n  | {\n      /** Component used to render all message types */\n      Message: ComponentType;\n      /** Component used when editing any message type */\n      EditComposer?: ComponentType | undefined;\n      /** Component used when editing user messages specifically */\n      UserEditComposer?: ComponentType | undefined;\n      /** Component used when editing assistant messages specifically */\n      AssistantEditComposer?: ComponentType | undefined;\n      /** Component used when editing system messages specifically */\n      SystemEditComposer?: ComponentType | undefined;\n      /** Component used to render user messages specifically */\n      UserMessage?: ComponentType | undefined;\n      /** Component used to render assistant messages specifically */\n      AssistantMessage?: ComponentType | undefined;\n      /** Component used to render system messages specifically */\n      SystemMessage?: ComponentType | undefined;\n    }\n  | {\n      /** Component used to render all message types (fallback) */\n      Message?: ComponentType | undefined;\n      /** Component used when editing any message type */\n      EditComposer?: ComponentType | undefined;\n      /** Component used when editing user messages specifically */\n      UserEditComposer?: ComponentType | undefined;\n      /** Component used when editing assistant messages specifically */\n      AssistantEditComposer?: ComponentType | undefined;\n      /** Component used when editing system messages specifically */\n      SystemEditComposer?: ComponentType | undefined;\n      /** Component used to render user messages */\n      UserMessage: ComponentType;\n      /** Component used to render assistant messages */\n      AssistantMessage: ComponentType;\n      /** Component used to render system messages */\n      SystemMessage?: ComponentType | undefined;\n    };\n\nexport namespace ThreadPrimitiveMessages {\n  export type Props =\n    | {\n        /** @deprecated Use the children render function instead. */\n        components: MessagesComponentConfig;\n        children?: never;\n      }\n    | {\n        /** Render function called for each message. Receives the message. */\n        children: (value: { message: MessageState }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst isComponentsSame = (\n  prev: MessagesComponentConfig,\n  next: MessagesComponentConfig,\n) => {\n  return (\n    prev.Message === next.Message &&\n    prev.EditComposer === next.EditComposer &&\n    prev.UserEditComposer === next.UserEditComposer &&\n    prev.AssistantEditComposer === next.AssistantEditComposer &&\n    prev.SystemEditComposer === next.SystemEditComposer &&\n    prev.UserMessage === next.UserMessage &&\n    prev.AssistantMessage === next.AssistantMessage &&\n    prev.SystemMessage === next.SystemMessage\n  );\n};\n\nconst DEFAULT_SYSTEM_MESSAGE = () => null;\n\nconst messageIdSetCache = new WeakMap<\n  readonly MessageState[],\n  ReadonlySet<string>\n>();\n\nconst hasMessageId = (messages: readonly MessageState[], messageId: string) => {\n  let ids = messageIdSetCache.get(messages);\n  if (!ids) {\n    ids = new Set(messages.map((m) => m.id));\n    messageIdSetCache.set(messages, ids);\n  }\n  return ids.has(messageId);\n};\n\nconst getComponent = (\n  components: MessagesComponentConfig,\n  role: MessageState[\"role\"],\n  isEditing: boolean,\n) => {\n  switch (role) {\n    case \"user\":\n      if (isEditing) {\n        return (\n          components.UserEditComposer ??\n          components.EditComposer ??\n          components.UserMessage ??\n          (components.Message as ComponentType)\n        );\n      } else {\n        return components.UserMessage ?? (components.Message as ComponentType);\n      }\n    case \"assistant\":\n      if (isEditing) {\n        return (\n          components.AssistantEditComposer ??\n          components.EditComposer ??\n          components.AssistantMessage ??\n          (components.Message as ComponentType)\n        );\n      } else {\n        return (\n          components.AssistantMessage ?? (components.Message as ComponentType)\n        );\n      }\n    case \"system\":\n      if (isEditing) {\n        return (\n          components.SystemEditComposer ??\n          components.EditComposer ??\n          components.SystemMessage ??\n          (components.Message as ComponentType) ??\n          DEFAULT_SYSTEM_MESSAGE\n        );\n      } else {\n        return (\n          components.SystemMessage ??\n          (components.Message as ComponentType) ??\n          DEFAULT_SYSTEM_MESSAGE\n        );\n      }\n    default: {\n      const _exhaustiveCheck: never = role;\n      throw new Error(`Unknown message role: ${_exhaustiveCheck}`);\n    }\n  }\n};\n\ntype ThreadMessageComponentProps = {\n  components: MessagesComponentConfig;\n};\n\nconst ThreadMessageComponent: FC<ThreadMessageComponentProps> = ({\n  components,\n}) => {\n  const role = useAuiState((s) => s.message.role);\n  const isEditing = useAuiState((s) => s.message.composer.isEditing);\n  const Component = getComponent(components, role, isEditing);\n\n  return <Component />;\n};\nexport namespace ThreadPrimitiveMessageByIndex {\n  export type Props = {\n    index: number;\n    components: MessagesComponentConfig;\n  };\n}\n\n/**\n * Renders a single message at the specified index in the current thread.\n */\nexport const ThreadPrimitiveMessageByIndex: FC<ThreadPrimitiveMessageByIndex.Props> =\n  memo(\n    ({ index, components }) => {\n      return (\n        <MessageByIndexProvider index={index}>\n          <ThreadMessageComponent components={components} />\n        </MessageByIndexProvider>\n      );\n    },\n    (prev, next) =>\n      prev.index === next.index &&\n      isComponentsSame(prev.components, next.components),\n  );\n\nThreadPrimitiveMessageByIndex.displayName = \"ThreadPrimitive.MessageByIndex\";\n\nexport namespace ThreadPrimitiveUnstable_MessageById {\n  export type Props = {\n    messageId: string;\n    components: MessagesComponentConfig;\n  };\n}\n\n/**\n * Renders the message with the given id in the current thread.\n *\n * Unlike {@link ThreadPrimitiveMessageByIndex}, this keys off the message id,\n * so it stays attached to the same message across reordering and windowing -\n * the shape needed to drive a virtualized or custom message list together with\n * `unstable_useThreadMessageIds`. A missing or removed id renders `null` rather\n * than throwing.\n *\n * @deprecated Unstable / Experimental - may change in any release.\n *\n * @example\n * ```tsx\n * const messageIds = unstable_useThreadMessageIds();\n * return messageIds.map((messageId) => (\n *   <ThreadPrimitive.Unstable_MessageById\n *     key={messageId}\n *     messageId={messageId}\n *     components={MESSAGE_COMPONENTS}\n *   />\n * ));\n * ```\n */\nexport const ThreadPrimitiveUnstable_MessageById: FC<ThreadPrimitiveUnstable_MessageById.Props> =\n  memo(\n    ({ messageId, components }) => {\n      const exists = useAuiState((s) =>\n        hasMessageId(s.thread.messages, messageId),\n      );\n      if (!exists) return null;\n\n      return (\n        <MessageByIdProvider id={messageId}>\n          <ThreadMessageComponent components={components} />\n        </MessageByIdProvider>\n      );\n    },\n    (prev, next) =>\n      prev.messageId === next.messageId &&\n      isComponentsSame(prev.components, next.components),\n  );\n\nThreadPrimitiveUnstable_MessageById.displayName =\n  \"ThreadPrimitive.Unstable_MessageById\";\n\nconst ThreadPrimitiveMessagesInner: FC<{\n  children: (value: { message: MessageState }) => ReactNode;\n}> = ({ children }) => {\n  const messagesLength = useAuiState((s) => s.thread.messages.length);\n\n  return useMemo(() => {\n    if (messagesLength === 0) return null;\n    return Array.from({ length: messagesLength }, (_, index) => (\n      <MessageByIndexProvider key={index} index={index}>\n        <RenderChildrenWithAccessor\n          getItemState={(aui) => aui.thread.message({ index }).getState()}\n        >\n          {(getItem) =>\n            children({\n              get message() {\n                return getItem();\n              },\n            })\n          }\n        </RenderChildrenWithAccessor>\n      </MessageByIndexProvider>\n    ));\n  }, [messagesLength, children]);\n};\n\n/**\n * Renders all messages in the current thread.\n *\n * @example\n * ```tsx\n * <ThreadPrimitive.Messages>\n *   {({ message }) => {\n *     if (message.role === \"user\") return <MyUserMessage />;\n *     return <MyAssistantMessage />;\n *   }}\n * </ThreadPrimitive.Messages>\n * ```\n */\nexport const ThreadPrimitiveMessagesImpl: FC<ThreadPrimitiveMessages.Props> = ({\n  components,\n  children,\n}) => {\n  if (components) {\n    return (\n      <ThreadPrimitiveMessagesInner>\n        {() => <ThreadMessageComponent components={components} />}\n      </ThreadPrimitiveMessagesInner>\n    );\n  }\n  return (\n    <ThreadPrimitiveMessagesInner>{children}</ThreadPrimitiveMessagesInner>\n  );\n};\n\nThreadPrimitiveMessagesImpl.displayName = \"ThreadPrimitive.Messages\";\n\nexport const ThreadPrimitiveMessages = memo(\n  ThreadPrimitiveMessagesImpl,\n  (prev, next) => {\n    if (prev.children || next.children) {\n      return prev.children === next.children;\n    }\n    return isComponentsSame(prev.components!, next.components!);\n  },\n);\n"],"mappings":";;;;;;;AAgEA,MAAMyB,oBACJC,MACAC,SACG;CACH,OACED,KAAKf,YAAYgB,KAAKhB,WACtBe,KAAKd,iBAAiBe,KAAKf,gBAC3Bc,KAAKb,qBAAqBc,KAAKd,oBAC/Ba,KAAKZ,0BAA0Ba,KAAKb,yBACpCY,KAAKX,uBAAuBY,KAAKZ,sBACjCW,KAAKV,gBAAgBW,KAAKX,eAC1BU,KAAKT,qBAAqBU,KAAKV,oBAC/BS,KAAKR,kBAAkBS,KAAKT;AAEhC;AAEA,MAAMU,+BAA+B;AAErC,MAAMC,oCAAoB,IAAIC,QAG5B;AAEF,MAAME,gBAAgBC,UAAmCC,cAAsB;CAC7E,IAAIC,MAAMN,kBAAkBO,IAAIH,QAAQ;CACxC,IAAI,CAACE,KAAK;EACRA,MAAM,IAAIE,IAAIJ,SAASK,KAAKC,MAAMA,EAAEC,EAAE,CAAC;EACvCX,kBAAkBY,IAAIR,UAAUE,GAAG;CACrC;CACA,OAAOA,IAAIO,IAAIR,SAAS;AAC1B;AAEA,MAAMS,gBACJtB,YACAuB,MACAC,cACG;CACH,QAAQD,MAAR;EACE,KAAK,QACH,IAAIC,WACF,OACExB,WAAWR,oBACXQ,WAAWT,gBACXS,WAAWL,eACVK,WAAWV;OAGd,OAAOU,WAAWL,eAAgBK,WAAWV;EAEjD,KAAK,aACH,IAAIkC,WACF,OACExB,WAAWP,yBACXO,WAAWT,gBACXS,WAAWJ,oBACVI,WAAWV;OAGd,OACEU,WAAWJ,oBAAqBI,WAAWV;EAGjD,KAAK,UACH,IAAIkC,WACF,OACExB,WAAWN,sBACXM,WAAWT,gBACXS,WAAWH,iBACVG,WAAWV,WACZiB;OAGF,OACEP,WAAWH,iBACVG,WAAWV,WACZiB;EAGN,SAEE,MAAM,IAAImB,MAAM,yBAAyBD,MAAkB;CAE/D;AACF;AAMA,MAAMG,0BAA0DC,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAA/B,eAAA6B;CAG/D,MAAAN,OAAatC,YAAY+C,KAAqB;CAC9C,MAAAR,YAAkBvC,YAAYgD,MAAmC;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAA9B,cAAA8B,EAAA,OAAAN,aAAAM,EAAA,OAAAP,MAAA;EACjDW,KAAAZ,aAAatB,YAAYuB,MAAMC,SAAS;EAACM,EAAA,KAAA9B;EAAA8B,EAAA,KAAAN;EAAAM,EAAA,KAAAP;EAAAO,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAA3D,MAAAK,YAAkBD;CAA0C,IAAAE;CAAA,IAAAN,EAAA,OAAAK,WAAA;EAErDC,KAAA,oBAAC,WAAD,CAAU,CAAA;EAAGN,EAAA,KAAAK;EAAAL,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OAAbM;AAAa;;;;AAYtB,MAAaC,gCACXvD,MACE+C,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAO,OAAAtC,eAAA6B;CAAqB,IAAAK;CAAA,IAAAJ,EAAA,OAAA9B,YAAA;EAGhBkC,KAAA,oBAAC,wBAAD,EAAoClC,WAAU,CAAA;EAAI8B,EAAA,KAAA9B;EAAA8B,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,IAAAM;CAAA,IAAAN,EAAA,OAAAQ,SAAAR,EAAA,OAAAI,IAAA;EADpDE,KAAA,oBAAC,wBAAD;GAA+BE;GAC7BJ,UAAAA;EADqB,CAAA;EAEEJ,EAAA,KAAAQ;EAAAR,EAAA,KAAAI;EAAAJ,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OAFzBM;AAEyB,IAG5B/B,MAAMC,SACLD,KAAKiC,UAAUhC,KAAKgC,SACpBlC,iBAAiBC,KAAKL,YAAYM,KAAKN,UAAU,CACrD;AAEFqC,8BAA8BE,cAAc;;;;;;;;;;;;;;;;;;;;;;;;AAgC5C,MAAaC,sCACX1D,MACE+C,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAlB,WAAAb,eAAA6B;CAAyB,IAAAK;CAAA,IAAAJ,EAAA,OAAAjB,WAAA;EACGqB,MAAAO,MACzB9B,aAAa8B,EAACC,OAAO9B,UAAWC,SAAS;EAACiB,EAAA,KAAAjB;EAAAiB,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAE5C,IAAI,CAHW7C,YAAYiD,EAGtBS,GAAM,OAAS;CAAK,IAAAP;CAAA,IAAAN,EAAA,OAAA9B,YAAA;EAIrBoC,KAAA,oBAAC,wBAAD,EAAoCpC,WAAU,CAAA;EAAI8B,EAAA,KAAA9B;EAAA8B,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,IAAAc;CAAA,IAAAd,EAAA,OAAAjB,aAAAiB,EAAA,OAAAM,IAAA;EADpDQ,KAAA,oBAAC,qBAAD;GAAyB/B,IAAAA;GACvBuB,UAAAA;EADkB,CAAA;EAEEN,EAAA,KAAAjB;EAAAiB,EAAA,KAAAM;EAAAN,EAAA,KAAAc;CAAA,OAAAA,KAAAd,EAAA;CAAA,OAFtBc;AAEsB,IAGzBvC,MAAMC,SACLD,KAAKQ,cAAcP,KAAKO,aACxBT,iBAAiBC,KAAKL,YAAYM,KAAKN,UAAU,CACrD;AAEFwC,oCAAoCD,cAClC;AAEF,MAAMM,gCAEA,EAAE5C,eAAe;CACrB,MAAM6C,iBAAiB7D,aAAawD,MAAMA,EAAEC,OAAO9B,SAASmC,MAAM;CAElE,OAAOhE,cAAc;EACnB,IAAI+D,mBAAmB,GAAG,OAAO;EACjC,OAAOE,MAAMC,KAAK,EAAEF,QAAQD,eAAe,IAAII,GAAGZ,UAChD,oBAAC,wBAAD;GAA2CA;GACzC,UAAA,oBAAC,4BAAD;IACE,eAAea,QAAQA,IAAIT,OAAOvC,QAAQ,EAAEmC,MAAM,CAAC,CAAC,CAACc,SAAS;IAE5DC,WAAAA,YACApD,SAAS,EACP,IAAIE,UAAU;KACZ,OAAOkD,QAAQ;IACjB,EACF,CAAC;GAEuB,CAAA;EACN,GAZKf,KAYL,CACzB;CACH,GAAG,CAACQ,gBAAgB7C,QAAQ,CAAC;AAC/B;;;;;;;;;;;;;;AAeA,MAAaqD,+BAAiEzB,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAA/B,YAAAC,aAAA4B;CAI7E,IAAI7B,YAAU;EAAA,IAAAkC;EAAA,IAAAJ,EAAA,OAAA9B,YAAA;GAEVkC,KAAA,oBAAC,8BAAD,EAAA,gBACS,oBAAC,wBAAD,EAAoClC,WAAU,CAAA,EAD1B,CAAA;GAEE8B,EAAA,KAAA9B;GAAA8B,EAAA,KAAAI;EAAA,OAAAA,KAAAJ,EAAA;EAAA,OAF/BI;CAE+B;CAElC,IAAAA;CAAA,IAAAJ,EAAA,OAAA7B,UAAA;EAECiC,KAAA,oBAAC,8BAAD,EAA+BjC,SAAF,CAAA;EAA0C6B,EAAA,KAAA7B;EAAA6B,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,OAAvEI;AAAuE;AAI3EoB,4BAA4Bf,cAAc;AAE1C,MAAazC,0BAA0BhB,KACrCwE,8BACCjD,MAAMC,SAAS;CACd,IAAID,KAAKJ,YAAYK,KAAKL,UACxB,OAAOI,KAAKJ,aAAaK,KAAKL;CAEhC,OAAOG,iBAAiBC,KAAKL,YAAaM,KAAKN,UAAW;AAC5D,CACF;AAvJgE,SAAAgC,MAAAS,GAAA;CAAA,OAG9BA,EAACtC,QAAQoB;AAAK;AAHgB,SAAAU,OAAAsB,KAAA;CAAA,OAIzBd,IAACtC,QAAQqD,SAAShC;AAAU"}
+{"version":3,"file":"ThreadMessages.js","names":["ComponentType","FC","ReactNode","memo","useMemo","RenderChildrenWithAccessor","useAuiState","useShallowSelector","MessageByIndexProvider","MessageByIdProvider","MessageState","MessagesComponentConfig","Message","EditComposer","UserEditComposer","AssistantEditComposer","SystemEditComposer","UserMessage","AssistantMessage","SystemMessage","ThreadPrimitiveMessages","Props","components","children","value","message","isComponentsSame","prev","next","DEFAULT_SYSTEM_MESSAGE","messageIdSetCache","WeakMap","ReadonlySet","hasMessageId","messages","messageId","ids","get","Set","map","m","id","set","has","getComponent","role","isEditing","_exhaustiveCheck","Error","ThreadMessageComponentProps","ThreadMessageComponent","t0","$","_c","_temp","_temp2","t1","Component","t2","ThreadPrimitiveMessageByIndex","index","displayName","ThreadPrimitiveUnstable_MessageById","s","thread","exists","t3","ThreadPrimitiveMessagesInner","messageIds","length","aui","getState","getItem","ThreadPrimitiveMessagesImpl","s_0","composer"],"sources":["../../../../src/react/primitives/thread/ThreadMessages.tsx"],"sourcesContent":["import {\n  type ComponentType,\n  type FC,\n  type ReactNode,\n  memo,\n  useMemo,\n} from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport { useShallowSelector } from \"@assistant-ui/store/internal\";\nimport { MessageByIndexProvider } from \"../../providers/MessageByIndexProvider\";\nimport { MessageByIdProvider } from \"../../providers/MessageByIdProvider\";\nimport type { MessageState } from \"../../../store\";\n\ntype MessagesComponentConfig =\n  | {\n      /** Component used to render all message types */\n      Message: ComponentType;\n      /** Component used when editing any message type */\n      EditComposer?: ComponentType | undefined;\n      /** Component used when editing user messages specifically */\n      UserEditComposer?: ComponentType | undefined;\n      /** Component used when editing assistant messages specifically */\n      AssistantEditComposer?: ComponentType | undefined;\n      /** Component used when editing system messages specifically */\n      SystemEditComposer?: ComponentType | undefined;\n      /** Component used to render user messages specifically */\n      UserMessage?: ComponentType | undefined;\n      /** Component used to render assistant messages specifically */\n      AssistantMessage?: ComponentType | undefined;\n      /** Component used to render system messages specifically */\n      SystemMessage?: ComponentType | undefined;\n    }\n  | {\n      /** Component used to render all message types (fallback) */\n      Message?: ComponentType | undefined;\n      /** Component used when editing any message type */\n      EditComposer?: ComponentType | undefined;\n      /** Component used when editing user messages specifically */\n      UserEditComposer?: ComponentType | undefined;\n      /** Component used when editing assistant messages specifically */\n      AssistantEditComposer?: ComponentType | undefined;\n      /** Component used when editing system messages specifically */\n      SystemEditComposer?: ComponentType | undefined;\n      /** Component used to render user messages */\n      UserMessage: ComponentType;\n      /** Component used to render assistant messages */\n      AssistantMessage: ComponentType;\n      /** Component used to render system messages */\n      SystemMessage?: ComponentType | undefined;\n    };\n\nexport namespace ThreadPrimitiveMessages {\n  export type Props =\n    | {\n        /** @deprecated Use the children render function instead. */\n        components: MessagesComponentConfig;\n        children?: never;\n      }\n    | {\n        /** Render function called for each message. Receives the message. */\n        children: (value: { message: MessageState }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst isComponentsSame = (\n  prev: MessagesComponentConfig,\n  next: MessagesComponentConfig,\n) => {\n  return (\n    prev.Message === next.Message &&\n    prev.EditComposer === next.EditComposer &&\n    prev.UserEditComposer === next.UserEditComposer &&\n    prev.AssistantEditComposer === next.AssistantEditComposer &&\n    prev.SystemEditComposer === next.SystemEditComposer &&\n    prev.UserMessage === next.UserMessage &&\n    prev.AssistantMessage === next.AssistantMessage &&\n    prev.SystemMessage === next.SystemMessage\n  );\n};\n\nconst DEFAULT_SYSTEM_MESSAGE = () => null;\n\nconst messageIdSetCache = new WeakMap<\n  readonly MessageState[],\n  ReadonlySet<string>\n>();\n\nconst hasMessageId = (messages: readonly MessageState[], messageId: string) => {\n  let ids = messageIdSetCache.get(messages);\n  if (!ids) {\n    ids = new Set(messages.map((m) => m.id));\n    messageIdSetCache.set(messages, ids);\n  }\n  return ids.has(messageId);\n};\n\nconst getComponent = (\n  components: MessagesComponentConfig,\n  role: MessageState[\"role\"],\n  isEditing: boolean,\n) => {\n  switch (role) {\n    case \"user\":\n      if (isEditing) {\n        return (\n          components.UserEditComposer ??\n          components.EditComposer ??\n          components.UserMessage ??\n          (components.Message as ComponentType)\n        );\n      } else {\n        return components.UserMessage ?? (components.Message as ComponentType);\n      }\n    case \"assistant\":\n      if (isEditing) {\n        return (\n          components.AssistantEditComposer ??\n          components.EditComposer ??\n          components.AssistantMessage ??\n          (components.Message as ComponentType)\n        );\n      } else {\n        return (\n          components.AssistantMessage ?? (components.Message as ComponentType)\n        );\n      }\n    case \"system\":\n      if (isEditing) {\n        return (\n          components.SystemEditComposer ??\n          components.EditComposer ??\n          components.SystemMessage ??\n          (components.Message as ComponentType) ??\n          DEFAULT_SYSTEM_MESSAGE\n        );\n      } else {\n        return (\n          components.SystemMessage ??\n          (components.Message as ComponentType) ??\n          DEFAULT_SYSTEM_MESSAGE\n        );\n      }\n    default: {\n      const _exhaustiveCheck: never = role;\n      throw new Error(`Unknown message role: ${_exhaustiveCheck}`);\n    }\n  }\n};\n\ntype ThreadMessageComponentProps = {\n  components: MessagesComponentConfig;\n};\n\nconst ThreadMessageComponent: FC<ThreadMessageComponentProps> = ({\n  components,\n}) => {\n  const role = useAuiState((s) => s.message.role);\n  const isEditing = useAuiState((s) => s.message.composer.isEditing);\n  const Component = getComponent(components, role, isEditing);\n\n  return <Component />;\n};\nexport namespace ThreadPrimitiveMessageByIndex {\n  export type Props = {\n    index: number;\n    components: MessagesComponentConfig;\n  };\n}\n\n/**\n * Renders a single message at the specified index in the current thread.\n */\nexport const ThreadPrimitiveMessageByIndex: FC<ThreadPrimitiveMessageByIndex.Props> =\n  memo(\n    ({ index, components }) => {\n      return (\n        <MessageByIndexProvider index={index}>\n          <ThreadMessageComponent components={components} />\n        </MessageByIndexProvider>\n      );\n    },\n    (prev, next) =>\n      prev.index === next.index &&\n      isComponentsSame(prev.components, next.components),\n  );\n\nThreadPrimitiveMessageByIndex.displayName = \"ThreadPrimitive.MessageByIndex\";\n\nexport namespace ThreadPrimitiveUnstable_MessageById {\n  export type Props = {\n    messageId: string;\n    components: MessagesComponentConfig;\n  };\n}\n\n/**\n * Renders the message with the given id in the current thread.\n *\n * Unlike {@link ThreadPrimitiveMessageByIndex}, this keys off the message id,\n * so it stays attached to the same message across reordering and windowing -\n * the shape needed to drive a virtualized or custom message list together with\n * `unstable_useThreadMessageIds`. A missing or removed id renders `null` rather\n * than throwing.\n *\n * @deprecated Unstable / Experimental - may change in any release.\n *\n * @example\n * ```tsx\n * const messageIds = unstable_useThreadMessageIds();\n * return messageIds.map((messageId) => (\n *   <ThreadPrimitive.Unstable_MessageById\n *     key={messageId}\n *     messageId={messageId}\n *     components={MESSAGE_COMPONENTS}\n *   />\n * ));\n * ```\n */\nexport const ThreadPrimitiveUnstable_MessageById: FC<ThreadPrimitiveUnstable_MessageById.Props> =\n  memo(\n    ({ messageId, components }) => {\n      const exists = useAuiState((s) =>\n        hasMessageId(s.thread.messages, messageId),\n      );\n      if (!exists) return null;\n\n      return (\n        <MessageByIdProvider id={messageId}>\n          <ThreadMessageComponent components={components} />\n        </MessageByIdProvider>\n      );\n    },\n    (prev, next) =>\n      prev.messageId === next.messageId &&\n      isComponentsSame(prev.components, next.components),\n  );\n\nThreadPrimitiveUnstable_MessageById.displayName =\n  \"ThreadPrimitive.Unstable_MessageById\";\n\nconst ThreadPrimitiveMessagesInner: FC<{\n  children: (value: { message: MessageState }) => ReactNode;\n}> = ({ children }) => {\n  const messageIds = useAuiState(\n    useShallowSelector((s) => s.thread.messages.map((message) => message.id)),\n  );\n\n  return useMemo(() => {\n    if (messageIds.length === 0) return null;\n    return messageIds.map((messageId, index) => (\n      <MessageByIndexProvider key={messageId} index={index}>\n        <RenderChildrenWithAccessor\n          getItemState={(aui) => aui.thread.message({ index }).getState()}\n        >\n          {(getItem) =>\n            children({\n              get message() {\n                return getItem();\n              },\n            })\n          }\n        </RenderChildrenWithAccessor>\n      </MessageByIndexProvider>\n    ));\n  }, [messageIds, children]);\n};\n\n/**\n * Renders all messages in the current thread.\n *\n * @example\n * ```tsx\n * <ThreadPrimitive.Messages>\n *   {({ message }) => {\n *     if (message.role === \"user\") return <MyUserMessage />;\n *     return <MyAssistantMessage />;\n *   }}\n * </ThreadPrimitive.Messages>\n * ```\n */\nexport const ThreadPrimitiveMessagesImpl: FC<ThreadPrimitiveMessages.Props> = ({\n  components,\n  children,\n}) => {\n  if (components) {\n    return (\n      <ThreadPrimitiveMessagesInner>\n        {() => <ThreadMessageComponent components={components} />}\n      </ThreadPrimitiveMessagesInner>\n    );\n  }\n  return (\n    <ThreadPrimitiveMessagesInner>{children}</ThreadPrimitiveMessagesInner>\n  );\n};\n\nThreadPrimitiveMessagesImpl.displayName = \"ThreadPrimitive.Messages\";\n\nexport const ThreadPrimitiveMessages = memo(\n  ThreadPrimitiveMessagesImpl,\n  (prev, next) => {\n    if (prev.children || next.children) {\n      return prev.children === next.children;\n    }\n    return isComponentsSame(prev.components!, next.components!);\n  },\n);\n"],"mappings":";;;;;;;;AAiEA,MAAM0B,oBACJC,MACAC,SACG;CACH,OACED,KAAKf,YAAYgB,KAAKhB,WACtBe,KAAKd,iBAAiBe,KAAKf,gBAC3Bc,KAAKb,qBAAqBc,KAAKd,oBAC/Ba,KAAKZ,0BAA0Ba,KAAKb,yBACpCY,KAAKX,uBAAuBY,KAAKZ,sBACjCW,KAAKV,gBAAgBW,KAAKX,eAC1BU,KAAKT,qBAAqBU,KAAKV,oBAC/BS,KAAKR,kBAAkBS,KAAKT;AAEhC;AAEA,MAAMU,+BAA+B;AAErC,MAAMC,oCAAoB,IAAIC,QAG5B;AAEF,MAAME,gBAAgBC,UAAmCC,cAAsB;CAC7E,IAAIC,MAAMN,kBAAkBO,IAAIH,QAAQ;CACxC,IAAI,CAACE,KAAK;EACRA,MAAM,IAAIE,IAAIJ,SAASK,KAAKC,MAAMA,EAAEC,EAAE,CAAC;EACvCX,kBAAkBY,IAAIR,UAAUE,GAAG;CACrC;CACA,OAAOA,IAAIO,IAAIR,SAAS;AAC1B;AAEA,MAAMS,gBACJtB,YACAuB,MACAC,cACG;CACH,QAAQD,MAAR;EACE,KAAK,QACH,IAAIC,WACF,OACExB,WAAWR,oBACXQ,WAAWT,gBACXS,WAAWL,eACVK,WAAWV;OAGd,OAAOU,WAAWL,eAAgBK,WAAWV;EAEjD,KAAK,aACH,IAAIkC,WACF,OACExB,WAAWP,yBACXO,WAAWT,gBACXS,WAAWJ,oBACVI,WAAWV;OAGd,OACEU,WAAWJ,oBAAqBI,WAAWV;EAGjD,KAAK,UACH,IAAIkC,WACF,OACExB,WAAWN,sBACXM,WAAWT,gBACXS,WAAWH,iBACVG,WAAWV,WACZiB;OAGF,OACEP,WAAWH,iBACVG,WAAWV,WACZiB;EAGN,SAEE,MAAM,IAAImB,MAAM,yBAAyBD,MAAkB;CAE/D;AACF;AAMA,MAAMG,0BAA0DC,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAA/B,eAAA6B;CAG/D,MAAAN,OAAavC,YAAYgD,KAAqB;CAC9C,MAAAR,YAAkBxC,YAAYiD,MAAmC;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAA9B,cAAA8B,EAAA,OAAAN,aAAAM,EAAA,OAAAP,MAAA;EACjDW,KAAAZ,aAAatB,YAAYuB,MAAMC,SAAS;EAACM,EAAA,KAAA9B;EAAA8B,EAAA,KAAAN;EAAAM,EAAA,KAAAP;EAAAO,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAA3D,MAAAK,YAAkBD;CAA0C,IAAAE;CAAA,IAAAN,EAAA,OAAAK,WAAA;EAErDC,KAAA,oBAAC,WAAD,CAAU,CAAA;EAAGN,EAAA,KAAAK;EAAAL,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OAAbM;AAAa;;;;AAYtB,MAAaC,gCACXxD,MACEgD,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAO,OAAAtC,eAAA6B;CAAqB,IAAAK;CAAA,IAAAJ,EAAA,OAAA9B,YAAA;EAGhBkC,KAAA,oBAAC,wBAAD,EAAoClC,WAAU,CAAA;EAAI8B,EAAA,KAAA9B;EAAA8B,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,IAAAM;CAAA,IAAAN,EAAA,OAAAQ,SAAAR,EAAA,OAAAI,IAAA;EADpDE,KAAA,oBAAC,wBAAD;GAA+BE;GAC7BJ,UAAAA;EADqB,CAAA;EAEEJ,EAAA,KAAAQ;EAAAR,EAAA,KAAAI;EAAAJ,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OAFzBM;AAEyB,IAG5B/B,MAAMC,SACLD,KAAKiC,UAAUhC,KAAKgC,SACpBlC,iBAAiBC,KAAKL,YAAYM,KAAKN,UAAU,CACrD;AAEFqC,8BAA8BE,cAAc;;;;;;;;;;;;;;;;;;;;;;;;AAgC5C,MAAaC,sCACX3D,MACEgD,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAAlB,WAAAb,eAAA6B;CAAyB,IAAAK;CAAA,IAAAJ,EAAA,OAAAjB,WAAA;EACGqB,MAAAO,MACzB9B,aAAa8B,EAACC,OAAO9B,UAAWC,SAAS;EAACiB,EAAA,KAAAjB;EAAAiB,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAE5C,IAAI,CAHW9C,YAAYkD,EAGtBS,GAAM,OAAS;CAAK,IAAAP;CAAA,IAAAN,EAAA,OAAA9B,YAAA;EAIrBoC,KAAA,oBAAC,wBAAD,EAAoCpC,WAAU,CAAA;EAAI8B,EAAA,KAAA9B;EAAA8B,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,IAAAc;CAAA,IAAAd,EAAA,OAAAjB,aAAAiB,EAAA,OAAAM,IAAA;EADpDQ,KAAA,oBAAC,qBAAD;GAAyB/B,IAAAA;GACvBuB,UAAAA;EADkB,CAAA;EAEEN,EAAA,KAAAjB;EAAAiB,EAAA,KAAAM;EAAAN,EAAA,KAAAc;CAAA,OAAAA,KAAAd,EAAA;CAAA,OAFtBc;AAEsB,IAGzBvC,MAAMC,SACLD,KAAKQ,cAAcP,KAAKO,aACxBT,iBAAiBC,KAAKL,YAAYM,KAAKN,UAAU,CACrD;AAEFwC,oCAAoCD,cAClC;AAEF,MAAMM,gCAEA,EAAE5C,eAAe;CACrB,MAAM6C,aAAa9D,YACjBC,oBAAoBwD,MAAMA,EAAEC,OAAO9B,SAASK,KAAKd,YAAYA,QAAQgB,EAAE,CAAC,CAC1E;CAEA,OAAOrC,cAAc;EACnB,IAAIgE,WAAWC,WAAW,GAAG,OAAO;EACpC,OAAOD,WAAW7B,KAAKJ,WAAWyB,UAChC,oBAAC,wBAAD;GAA+CA;GAC7C,UAAA,oBAAC,4BAAD;IACE,eAAeU,QAAQA,IAAIN,OAAOvC,QAAQ,EAAEmC,MAAM,CAAC,CAAC,CAACW,SAAS;IAE5DC,WAAAA,YACAjD,SAAS,EACP,IAAIE,UAAU;KACZ,OAAO+C,QAAQ;IACjB,EACF,CAAC;GAEuB,CAAA;EACN,GAZKrC,SAYL,CACzB;CACH,GAAG,CAACiC,YAAY7C,QAAQ,CAAC;AAC3B;;;;;;;;;;;;;;AAeA,MAAakD,+BAAiEtB,OAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAAC,MAAA,EAAA/B,YAAAC,aAAA4B;CAI7E,IAAI7B,YAAU;EAAA,IAAAkC;EAAA,IAAAJ,EAAA,OAAA9B,YAAA;GAEVkC,KAAA,oBAAC,8BAAD,EAAA,gBACS,oBAAC,wBAAD,EAAoClC,WAAU,CAAA,EAD1B,CAAA;GAEE8B,EAAA,KAAA9B;GAAA8B,EAAA,KAAAI;EAAA,OAAAA,KAAAJ,EAAA;EAAA,OAF/BI;CAE+B;CAElC,IAAAA;CAAA,IAAAJ,EAAA,OAAA7B,UAAA;EAECiC,KAAA,oBAAC,8BAAD,EAA+BjC,SAAF,CAAA;EAA0C6B,EAAA,KAAA7B;EAAA6B,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,OAAvEI;AAAuE;AAI3EiB,4BAA4BZ,cAAc;AAE1C,MAAazC,0BAA0BjB,KACrCsE,8BACC9C,MAAMC,SAAS;CACd,IAAID,KAAKJ,YAAYK,KAAKL,UACxB,OAAOI,KAAKJ,aAAaK,KAAKL;CAEhC,OAAOG,iBAAiBC,KAAKL,YAAaM,KAAKN,UAAW;AAC5D,CACF;AAzJgE,SAAAgC,MAAAS,GAAA;CAAA,OAG9BA,EAACtC,QAAQoB;AAAK;AAHgB,SAAAU,OAAAmB,KAAA;CAAA,OAIzBX,IAACtC,QAAQkD,SAAS7B;AAAU"}

package/dist/react/primitives/thread/ThreadSuggestions.d.ts

@@ -5,7 +5,7 @@ type SuggestionsComponentConfig = {
   /** Component used to render each suggestion */
   Suggestion: ComponentType;
 };
-declare namespace ThreadPrimitiveSuggestions {
+export declare namespace ThreadPrimitiveSuggestions {
   type Props = {
     /** @deprecated Use the children render function instead. */
     components: SuggestionsComponentConfig;
@@ -18,7 +18,7 @@ declare namespace ThreadPrimitiveSuggestions {
     components?: never;
   };
 }
-declare namespace ThreadPrimitiveSuggestionByIndex {
+export declare namespace ThreadPrimitiveSuggestionByIndex {
   type Props = {
     index: number;
     components: SuggestionsComponentConfig;
@@ -27,12 +27,11 @@ declare namespace ThreadPrimitiveSuggestionByIndex {
 /**
  * Renders a single suggestion at the specified index.
  */
-declare const ThreadPrimitiveSuggestionByIndex: FC<ThreadPrimitiveSuggestionByIndex.Props>;
+export declare const ThreadPrimitiveSuggestionByIndex: FC<ThreadPrimitiveSuggestionByIndex.Props>;
 /**
  * Renders all suggestions.
  */
-declare const ThreadPrimitiveSuggestionsImpl: FC<ThreadPrimitiveSuggestions.Props>;
-declare const ThreadPrimitiveSuggestions: import("react").NamedExoticComponent<ThreadPrimitiveSuggestions.Props>;
+export declare const ThreadPrimitiveSuggestionsImpl: FC<ThreadPrimitiveSuggestions.Props>;
+export declare const ThreadPrimitiveSuggestions: import("react").NamedExoticComponent<ThreadPrimitiveSuggestions.Props>;
 //#endregion
-export { ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl };
 //# sourceMappingURL=ThreadSuggestions.d.ts.map

package/dist/react/primitives/thread/ThreadSuggestions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ThreadSuggestions.d.ts","names":[],"sources":["../../../../src/react/primitives/thread/ThreadSuggestions.tsx"],"mappings":";;;KAWK;;EAEH,YAAY;;kBAGG;OACH;;IAGN,YAAY;IACZ;;;IAIA,WAAW;MAAS,YAAY;UAAsB;IACtD;;;kBAaS;OACH;IACV;IACA,YAAY;;;;;;cAOH,kCAAkC,GAAG,iCAAiC;;;;cAiDtE,gCAAgC,GAC3C,2BAA2B;cAkBhB,4CAA0B,qBAAA,2BAAA"}
+{"version":3,"file":"ThreadSuggestions.d.ts","names":[],"sources":["../../../../src/react/primitives/thread/ThreadSuggestions.tsx"],"mappings":";;;KAWK;;EAEH,YAAY;;yBAGG;OACH;;IAGN,YAAY;IACZ;;;IAIA,WAAW;MAAS,YAAY;UAAsB;IACtD;;;yBAaS;OACH;IACV;IACA,YAAY;;;;;;qBAOH,kCAAkC,GAAG,iCAAiC;;;;qBAiDtE,gCAAgC,GAC3C,2BAA2B;qBAkBhB,4CAA0B,qBAAA,2BAAA"}

package/dist/react/primitives/threadList/ThreadListItems.d.ts

@@ -4,7 +4,7 @@ import { ComponentType, FC, ReactNode } from "react";
 type ThreadListItemsComponentConfig = {
   ThreadListItem: ComponentType;
 };
-declare namespace ThreadListPrimitiveItems {
+export declare namespace ThreadListPrimitiveItems {
   type Props = {
     archived?: boolean | undefined;
   } & ({
@@ -19,7 +19,7 @@ declare namespace ThreadListPrimitiveItems {
     components?: never;
   });
 }
-declare namespace ThreadListPrimitiveItemByIndex {
+export declare namespace ThreadListPrimitiveItemByIndex {
   type Props = {
     index: number;
     archived?: boolean | undefined;
@@ -29,8 +29,7 @@ declare namespace ThreadListPrimitiveItemByIndex {
 /**
  * Renders a single thread list item at the specified index.
  */
-declare const ThreadListPrimitiveItemByIndex: FC<ThreadListPrimitiveItemByIndex.Props>;
-declare const ThreadListPrimitiveItems: FC<ThreadListPrimitiveItems.Props>;
+export declare const ThreadListPrimitiveItemByIndex: FC<ThreadListPrimitiveItemByIndex.Props>;
+export declare const ThreadListPrimitiveItems: FC<ThreadListPrimitiveItems.Props>;
 //#endregion
-export { ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems };
 //# sourceMappingURL=ThreadListItems.d.ts.map

package/dist/react/primitives/threadList/ThreadListItems.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ThreadListItems.d.ts","names":[],"sources":["../../../../src/react/primitives/threadList/ThreadListItems.tsx"],"mappings":";;;KAYK;EACH,gBAAgB;;kBAGD;OACH;IACV;;;IAII,YAAY;IACZ;;;IAIA,WAAW;MAAS,gBAAgB;UAA0B;IAC9D;;;kBAKS;OACH;IACV;IACA;IACA,YAAY;;;;;;cAOH,gCAAgC,GAAG,+BAA+B;cAwDlE,0BAA0B,GAAG,yBAAyB"}
+{"version":3,"file":"ThreadListItems.d.ts","names":[],"sources":["../../../../src/react/primitives/threadList/ThreadListItems.tsx"],"mappings":";;;KAYK;EACH,gBAAgB;;yBAGD;OACH;IACV;;;IAII,YAAY;IACZ;;;IAIA,WAAW;MAAS,gBAAgB;UAA0B;IAC9D;;;yBAKS;OACH;IACV;IACA;IACA,YAAY;;;;;;qBAOH,gCAAgC,GAAG,+BAA+B;qBAwDlE,0BAA0B,GAAG,yBAAyB"}

package/dist/react/primitives/threadListItem/ThreadListItemTitle.d.ts

@@ -1,11 +1,10 @@
 import { FC, ReactNode } from "react";
 //#region src/react/primitives/threadListItem/ThreadListItemTitle.d.ts
-declare namespace ThreadListItemPrimitiveTitle {
+export declare namespace ThreadListItemPrimitiveTitle {
   type Props = {
     fallback?: ReactNode;
   };
 }
-declare const ThreadListItemPrimitiveTitle: FC<ThreadListItemPrimitiveTitle.Props>;
+export declare const ThreadListItemPrimitiveTitle: FC<ThreadListItemPrimitiveTitle.Props>;
 //#endregion
-export { ThreadListItemPrimitiveTitle };
 //# sourceMappingURL=ThreadListItemTitle.d.ts.map

package/dist/react/primitives/threadListItem/ThreadListItemTitle.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ThreadListItemTitle.d.ts","names":[],"sources":["../../../../src/react/primitives/threadListItem/ThreadListItemTitle.tsx"],"mappings":";;kBAGiB;OACH;IACV,WAAW;;;cAIF,8BAA8B,GACzC,6BAA6B"}
+{"version":3,"file":"ThreadListItemTitle.d.ts","names":[],"sources":["../../../../src/react/primitives/threadListItem/ThreadListItemTitle.tsx"],"mappings":";;yBAGiB;OACH;IACV,WAAW;;;qBAIF,8BAA8B,GACzC,6BAA6B"}

package/dist/react/providers/AttachmentByIndexProvider.d.ts

@@ -1,11 +1,10 @@
 import { FC, PropsWithChildren } from "react";
 //#region src/react/providers/AttachmentByIndexProvider.d.ts
-declare const MessageAttachmentByIndexProvider: FC<PropsWithChildren<{
+export declare const MessageAttachmentByIndexProvider: FC<PropsWithChildren<{
   index: number;
 }>>;
-declare const ComposerAttachmentByIndexProvider: FC<PropsWithChildren<{
+export declare const ComposerAttachmentByIndexProvider: FC<PropsWithChildren<{
   index: number;
 }>>;
 //#endregion
-export { ComposerAttachmentByIndexProvider, MessageAttachmentByIndexProvider };
 //# sourceMappingURL=AttachmentByIndexProvider.d.ts.map

package/dist/react/providers/AttachmentByIndexProvider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AttachmentByIndexProvider.d.ts","names":[],"sources":["../../../src/react/providers/AttachmentByIndexProvider.tsx"],"mappings":";;cAGa,kCAAkC,GAC7C;EACE;;cAkBS,mCAAmC,GAC9C;EACE"}
+{"version":3,"file":"AttachmentByIndexProvider.d.ts","names":[],"sources":["../../../src/react/providers/AttachmentByIndexProvider.tsx"],"mappings":";;qBAGa,kCAAkC,GAC7C;EACE;;qBAkBS,mCAAmC,GAC9C;EACE"}

package/dist/react/providers/ChainOfThoughtByIndicesProvider.d.ts

@@ -1,9 +1,8 @@
 import { FC, PropsWithChildren } from "react";
 //#region src/react/providers/ChainOfThoughtByIndicesProvider.d.ts
-declare const ChainOfThoughtByIndicesProvider: FC<PropsWithChildren<{
+export declare const ChainOfThoughtByIndicesProvider: FC<PropsWithChildren<{
   startIndex: number;
   endIndex: number;
 }>>;
 //#endregion
-export { ChainOfThoughtByIndicesProvider };
 //# sourceMappingURL=ChainOfThoughtByIndicesProvider.d.ts.map

package/dist/react/providers/ChainOfThoughtByIndicesProvider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ChainOfThoughtByIndicesProvider.d.ts","names":[],"sources":["../../../src/react/providers/ChainOfThoughtByIndicesProvider.tsx"],"mappings":";;cAUa,iCAAiC,GAC5C;EACE;EACA"}
+{"version":3,"file":"ChainOfThoughtByIndicesProvider.d.ts","names":[],"sources":["../../../src/react/providers/ChainOfThoughtByIndicesProvider.tsx"],"mappings":";;qBAUa,iCAAiC,GAC5C;EACE;EACA"}

package/dist/react/providers/ChainOfThoughtPartByIndexProvider.d.ts

@@ -1,8 +1,7 @@
 import { FC, PropsWithChildren } from "react";
 //#region src/react/providers/ChainOfThoughtPartByIndexProvider.d.ts
-declare const ChainOfThoughtPartByIndexProvider: FC<PropsWithChildren<{
+export declare const ChainOfThoughtPartByIndexProvider: FC<PropsWithChildren<{
   index: number;
 }>>;
 //#endregion
-export { ChainOfThoughtPartByIndexProvider };
 //# sourceMappingURL=ChainOfThoughtPartByIndexProvider.d.ts.map

package/dist/react/providers/ChainOfThoughtPartByIndexProvider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ChainOfThoughtPartByIndexProvider.d.ts","names":[],"sources":["../../../src/react/providers/ChainOfThoughtPartByIndexProvider.tsx"],"mappings":";;cAGa,mCAAmC,GAC9C;EACE"}
+{"version":3,"file":"ChainOfThoughtPartByIndexProvider.d.ts","names":[],"sources":["../../../src/react/providers/ChainOfThoughtPartByIndexProvider.tsx"],"mappings":";;qBAGa,mCAAmC,GAC9C;EACE"}

package/dist/react/providers/MessageByIdProvider.d.ts

@@ -1,8 +1,7 @@
 import { FC, PropsWithChildren } from "react";
 //#region src/react/providers/MessageByIdProvider.d.ts
-declare const MessageByIdProvider: FC<PropsWithChildren<{
+export declare const MessageByIdProvider: FC<PropsWithChildren<{
   id: string;
 }>>;
 //#endregion
-export { MessageByIdProvider };
 //# sourceMappingURL=MessageByIdProvider.d.ts.map

package/dist/react/providers/MessageByIdProvider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageByIdProvider.d.ts","names":[],"sources":["../../../src/react/providers/MessageByIdProvider.tsx"],"mappings":";;cAGa,qBAAqB,GAChC;EACE"}
+{"version":3,"file":"MessageByIdProvider.d.ts","names":[],"sources":["../../../src/react/providers/MessageByIdProvider.tsx"],"mappings":";;qBAGa,qBAAqB,GAChC;EACE"}

package/dist/react/providers/MessageByIndexProvider.d.ts

@@ -1,8 +1,7 @@
 import { FC, PropsWithChildren } from "react";
 //#region src/react/providers/MessageByIndexProvider.d.ts
-declare const MessageByIndexProvider: FC<PropsWithChildren<{
+export declare const MessageByIndexProvider: FC<PropsWithChildren<{
   index: number;
 }>>;
 //#endregion
-export { MessageByIndexProvider };
 //# sourceMappingURL=MessageByIndexProvider.d.ts.map