@assistant-ui/react 0.14.12 → 0.14.14

package/dist/unstable/useMentionAdapter.js.map

@@ -1 +1 @@
-{"version":3,"file":"useMentionAdapter.js","names":[],"sources":["../../src/unstable/useMentionAdapter.ts"],"sourcesContent":["\"use client\";\n\nimport { useMemo, type FC } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\nimport type {\n  Unstable_DirectiveFormatter,\n  Unstable_TriggerAdapter,\n  Unstable_TriggerCategory,\n  Unstable_TriggerItem,\n} from \"@assistant-ui/core\";\nimport { unstable_defaultDirectiveFormatter } from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\n\n/** Icon component shape consumed by `ComposerTriggerPopover`'s `iconMap`. */\nexport type Unstable_IconComponent = FC<{ className?: string }>;\n\nexport type Unstable_Mention = {\n  readonly id: string;\n  readonly type: string;\n  readonly label: string;\n  readonly description?: string | undefined;\n  /** Shortcut for `metadata.icon`; merged with `metadata` if both are given. */\n  readonly icon?: string | undefined;\n  readonly metadata?: ReadonlyJSONObject | undefined;\n};\n\nexport type Unstable_MentionCategory = {\n  readonly id: string;\n  readonly label: string;\n  readonly items: readonly Unstable_Mention[];\n};\n\nexport type Unstable_ModelContextToolsOptions = {\n  /** Wrap tools in a dedicated category (drill-down mode). */\n  readonly category?: { readonly id: string; readonly label: string };\n  /** Format tool name for display. */\n  readonly formatLabel?: (toolName: string) => string;\n  /** Default icon key for each tool. */\n  readonly icon?: string;\n};\n\nexport type Unstable_UseMentionAdapterOptions = {\n  /** Flat mention list. Ignored when `categories` is set. */\n  readonly items?: readonly Unstable_Mention[];\n  /** Categorized mentions for drill-down navigation. */\n  readonly categories?: readonly Unstable_MentionCategory[];\n  /**\n   * How tools registered via `useAssistantTool` integrate.\n   * - `false`: exclude.\n   * - `true`: include (default when no `items`/`categories`; as a category\n   *   if `categories` is set, flat otherwise).\n   * - object: explicit config.\n   *\n   * Omitted → defaults to `true` iff neither `items` nor `categories`.\n   */\n  readonly includeModelContextTools?:\n    | boolean\n    | Unstable_ModelContextToolsOptions;\n  /** Directive formatter. @default unstable_defaultDirectiveFormatter */\n  readonly formatter?: Unstable_DirectiveFormatter;\n  /** Fires after an item is inserted into the composer. */\n  readonly onInserted?: (item: Unstable_TriggerItem) => void;\n  /** Maps `metadata.icon` / `category.id` string keys to React components. */\n  readonly iconMap?: Record<string, Unstable_IconComponent>;\n  /** Fallback icon when no entry in `iconMap` matches. */\n  readonly fallbackIcon?: Unstable_IconComponent;\n};\n\nexport type Unstable_MentionDirective = {\n  readonly formatter: Unstable_DirectiveFormatter;\n  readonly onInserted?: ((item: Unstable_TriggerItem) => void) | undefined;\n};\n\n/**\n * @deprecated Under active development and might change without notice.\n *\n * Creates a spreadable `{ adapter, directive }` bundle for `@` mentions.\n * Supports tools registered via `useAssistantTool`, explicit items, or both —\n * flat or categorized.\n *\n * @example\n * ```tsx\n * const mention = unstable_useMentionAdapter();\n * <ComposerTriggerPopover char=\"@\" {...mention} />\n * ```\n */\nexport function unstable_useMentionAdapter(\n  options?: Unstable_UseMentionAdapterOptions,\n): {\n  adapter: Unstable_TriggerAdapter;\n  directive: Unstable_MentionDirective;\n  iconMap?: Record<string, Unstable_IconComponent>;\n  fallbackIcon?: Unstable_IconComponent;\n} {\n  const aui = useAui();\n\n  const items = options?.items;\n  const categories = options?.categories;\n  const includeTools =\n    options?.includeModelContextTools ?? (!items && !categories);\n  const toolsConfig =\n    typeof includeTools === \"object\" ? includeTools : undefined;\n  const wantsTools = includeTools !== false;\n  const formatter = options?.formatter;\n  const onInserted = options?.onInserted;\n\n  const adapter = useMemo<Unstable_TriggerAdapter>(() => {\n    const getModelContextTools = (): Unstable_TriggerItem[] => {\n      if (!wantsTools) return [];\n      const ctx = aui.thread().getModelContext();\n      const tools = ctx.tools;\n      if (!tools) return [];\n      const formatLabel = toolsConfig?.formatLabel;\n      const defaultIcon = toolsConfig?.icon;\n      return Object.entries(tools).map(([name, tool]) =>\n        toTriggerItem({\n          id: name,\n          type: \"tool\",\n          label: formatLabel ? formatLabel(name) : name,\n          description: tool.description ?? undefined,\n          icon: defaultIcon,\n        }),\n      );\n    };\n\n    // Categorized: drill-down mode\n    if (categories && categories.length > 0) {\n      const groups = categories.map((cat) => ({\n        id: cat.id,\n        label: cat.label,\n        items: cat.items.map(toTriggerItem),\n      }));\n\n      let toolCategory: {\n        id: string;\n        label: string;\n        items: Unstable_TriggerItem[];\n      } | null = null;\n      if (wantsTools) {\n        const toolItems = getModelContextTools();\n        if (toolItems.length > 0) {\n          toolCategory = {\n            id: toolsConfig?.category?.id ?? \"tools\",\n            label: toolsConfig?.category?.label ?? \"Tools\",\n            items: toolItems,\n          };\n        }\n      }\n      const allGroups = toolCategory ? [...groups, toolCategory] : groups;\n\n      return {\n        categories: () => allGroups.map(({ id, label }) => ({ id, label })),\n        categoryItems: (id) => allGroups.find((g) => g.id === id)?.items ?? [],\n        search: (query) => {\n          const lower = query.toLowerCase();\n          return allGroups\n            .flatMap((g) => g.items)\n            .filter((item) => matchesQuery(item, lower));\n        },\n      };\n    }\n\n    // Flat: items + (optionally) tools, all in one search pool\n    const flatItems = (items ?? []).map(toTriggerItem);\n    const getFlatPool = (): Unstable_TriggerItem[] => {\n      if (!wantsTools) return flatItems;\n      const toolItems = getModelContextTools();\n      // Dedupe by id — explicit items win.\n      const seen = new Set(flatItems.map((i) => i.id));\n      return [...flatItems, ...toolItems.filter((t) => !seen.has(t.id))];\n    };\n\n    return {\n      categories: (): readonly Unstable_TriggerCategory[] => [],\n      categoryItems: () => [],\n      search: (query) => {\n        const lower = query.toLowerCase();\n        return getFlatPool().filter((item) => matchesQuery(item, lower));\n      },\n    };\n  }, [aui, items, categories, wantsTools, toolsConfig]);\n\n  const directive = useMemo<Unstable_MentionDirective>(\n    () => ({\n      formatter: formatter ?? unstable_defaultDirectiveFormatter,\n      ...(onInserted ? { onInserted } : {}),\n    }),\n    [formatter, onInserted],\n  );\n\n  return {\n    adapter,\n    directive,\n    ...(options?.iconMap ? { iconMap: options.iconMap } : {}),\n    ...(options?.fallbackIcon ? { fallbackIcon: options.fallbackIcon } : {}),\n  };\n}\n\nfunction toTriggerItem(m: Unstable_Mention): Unstable_TriggerItem {\n  const metadata =\n    m.icon !== undefined ? { ...(m.metadata ?? {}), icon: m.icon } : m.metadata;\n  return {\n    id: m.id,\n    type: m.type,\n    label: m.label,\n    ...(m.description !== undefined ? { description: m.description } : {}),\n    ...(metadata !== undefined ? { metadata } : {}),\n  };\n}\n\nfunction matchesQuery(item: Unstable_TriggerItem, lower: string): boolean {\n  if (!lower) return true;\n  if (item.id.toLowerCase().includes(lower)) return true;\n  if (item.label.toLowerCase().includes(lower)) return true;\n  if (item.description?.toLowerCase().includes(lower)) return true;\n  return false;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAsFA,SAAgB,2BACd,SAMA;CACA,MAAM,MAAM,OAAO;CAEnB,MAAM,QAAQ,SAAS;CACvB,MAAM,aAAa,SAAS;CAC5B,MAAM,eACJ,SAAS,6BAA6B,CAAC,SAAS,CAAC;CACnD,MAAM,cACJ,OAAO,iBAAiB,WAAW,eAAe,KAAA;CACpD,MAAM,aAAa,iBAAiB;CACpC,MAAM,YAAY,SAAS;CAC3B,MAAM,aAAa,SAAS;CAsF5B,OAAO;EACL,SArFc,cAAuC;GACrD,MAAM,6BAAqD;IACzD,IAAI,CAAC,YAAY,OAAO,CAAC;IAEzB,MAAM,QADM,IAAI,OAAO,EAAE,gBACT,EAAE;IAClB,IAAI,CAAC,OAAO,OAAO,CAAC;IACpB,MAAM,cAAc,aAAa;IACjC,MAAM,cAAc,aAAa;IACjC,OAAO,OAAO,QAAQ,KAAK,EAAE,KAAK,CAAC,MAAM,UACvC,cAAc;KACZ,IAAI;KACJ,MAAM;KACN,OAAO,cAAc,YAAY,IAAI,IAAI;KACzC,aAAa,KAAK,eAAe,KAAA;KACjC,MAAM;IACR,CAAC,CACH;GACF;GAGA,IAAI,cAAc,WAAW,SAAS,GAAG;IACvC,MAAM,SAAS,WAAW,KAAK,SAAS;KACtC,IAAI,IAAI;KACR,OAAO,IAAI;KACX,OAAO,IAAI,MAAM,IAAI,aAAa;IACpC,EAAE;IAEF,IAAI,eAIO;IACX,IAAI,YAAY;KACd,MAAM,YAAY,qBAAqB;KACvC,IAAI,UAAU,SAAS,GACrB,eAAe;MACb,IAAI,aAAa,UAAU,MAAM;MACjC,OAAO,aAAa,UAAU,SAAS;MACvC,OAAO;KACT;IAEJ;IACA,MAAM,YAAY,eAAe,CAAC,GAAG,QAAQ,YAAY,IAAI;IAE7D,OAAO;KACL,kBAAkB,UAAU,KAAK,EAAE,IAAI,aAAa;MAAE;MAAI;KAAM,EAAE;KAClE,gBAAgB,OAAO,UAAU,MAAM,MAAM,EAAE,OAAO,EAAE,GAAG,SAAS,CAAC;KACrE,SAAS,UAAU;MACjB,MAAM,QAAQ,MAAM,YAAY;MAChC,OAAO,UACJ,SAAS,MAAM,EAAE,KAAK,EACtB,QAAQ,SAAS,aAAa,MAAM,KAAK,CAAC;KAC/C;IACF;GACF;GAGA,MAAM,aAAa,SAAS,CAAC,GAAG,IAAI,aAAa;GACjD,MAAM,oBAA4C;IAChD,IAAI,CAAC,YAAY,OAAO;IACxB,MAAM,YAAY,qBAAqB;IAEvC,MAAM,OAAO,IAAI,IAAI,UAAU,KAAK,MAAM,EAAE,EAAE,CAAC;IAC/C,OAAO,CAAC,GAAG,WAAW,GAAG,UAAU,QAAQ,MAAM,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC,CAAC;GACnE;GAEA,OAAO;IACL,kBAAuD,CAAC;IACxD,qBAAqB,CAAC;IACtB,SAAS,UAAU;KACjB,MAAM,QAAQ,MAAM,YAAY;KAChC,OAAO,YAAY,EAAE,QAAQ,SAAS,aAAa,MAAM,KAAK,CAAC;IACjE;GACF;EACF,GAAG;GAAC;GAAK;GAAO;GAAY;GAAY;EAAW,CAW3C;EACN,WAVgB,eACT;GACL,WAAW,aAAa;GACxB,GAAI,aAAa,EAAE,WAAW,IAAI,CAAC;EACrC,IACA,CAAC,WAAW,UAAU,CAKd;EACR,GAAI,SAAS,UAAU,EAAE,SAAS,QAAQ,QAAQ,IAAI,CAAC;EACvD,GAAI,SAAS,eAAe,EAAE,cAAc,QAAQ,aAAa,IAAI,CAAC;CACxE;AACF;AAEA,SAAS,cAAc,GAA2C;CAChE,MAAM,WACJ,EAAE,SAAS,KAAA,IAAY;EAAE,GAAI,EAAE,YAAY,CAAC;EAAI,MAAM,EAAE;CAAK,IAAI,EAAE;CACrE,OAAO;EACL,IAAI,EAAE;EACN,MAAM,EAAE;EACR,OAAO,EAAE;EACT,GAAI,EAAE,gBAAgB,KAAA,IAAY,EAAE,aAAa,EAAE,YAAY,IAAI,CAAC;EACpE,GAAI,aAAa,KAAA,IAAY,EAAE,SAAS,IAAI,CAAC;CAC/C;AACF;AAEA,SAAS,aAAa,MAA4B,OAAwB;CACxE,IAAI,CAAC,OAAO,OAAO;CACnB,IAAI,KAAK,GAAG,YAAY,EAAE,SAAS,KAAK,GAAG,OAAO;CAClD,IAAI,KAAK,MAAM,YAAY,EAAE,SAAS,KAAK,GAAG,OAAO;CACrD,IAAI,KAAK,aAAa,YAAY,EAAE,SAAS,KAAK,GAAG,OAAO;CAC5D,OAAO;AACT"}
+{"version":3,"file":"useMentionAdapter.js","names":[],"sources":["../../src/unstable/useMentionAdapter.ts"],"sourcesContent":["\"use client\";\n\nimport { useMemo, type FC } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\nimport type {\n  Unstable_DirectiveFormatter,\n  Unstable_TriggerAdapter,\n  Unstable_TriggerCategory,\n  Unstable_TriggerItem,\n} from \"@assistant-ui/core\";\nimport { unstable_defaultDirectiveFormatter } from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\n\n/** Icon component shape consumed by `ComposerTriggerPopover`'s `iconMap`. */\nexport type Unstable_IconComponent = FC<{ className?: string }>;\n\nexport type Unstable_Mention = {\n  readonly id: string;\n  readonly type: string;\n  readonly label: string;\n  readonly description?: string | undefined;\n  /** Shortcut for `metadata.icon`; merged with `metadata` if both are given. */\n  readonly icon?: string | undefined;\n  readonly metadata?: ReadonlyJSONObject | undefined;\n};\n\nexport type Unstable_MentionCategory = {\n  readonly id: string;\n  readonly label: string;\n  readonly items: readonly Unstable_Mention[];\n};\n\nexport type Unstable_ModelContextToolsOptions = {\n  /** Wrap tools in a dedicated category (drill-down mode). */\n  readonly category?: { readonly id: string; readonly label: string };\n  /** Format tool name for display. */\n  readonly formatLabel?: (toolName: string) => string;\n  /** Default icon key for each tool. */\n  readonly icon?: string;\n};\n\nexport type Unstable_UseMentionAdapterOptions = {\n  /** Flat mention list. Ignored when `categories` is set. */\n  readonly items?: readonly Unstable_Mention[];\n  /** Categorized mentions for drill-down navigation. */\n  readonly categories?: readonly Unstable_MentionCategory[];\n  /**\n   * How tools registered in model context integrate.\n   * - `false`: exclude.\n   * - `true`: include (default when no `items`/`categories`; as a category\n   *   if `categories` is set, flat otherwise).\n   * - object: explicit config.\n   *\n   * Omitted → defaults to `true` iff neither `items` nor `categories`.\n   */\n  readonly includeModelContextTools?:\n    | boolean\n    | Unstable_ModelContextToolsOptions;\n  /** Directive formatter. @default unstable_defaultDirectiveFormatter */\n  readonly formatter?: Unstable_DirectiveFormatter;\n  /** Fires after an item is inserted into the composer. */\n  readonly onInserted?: (item: Unstable_TriggerItem) => void;\n  /** Maps `metadata.icon` / `category.id` string keys to React components. */\n  readonly iconMap?: Record<string, Unstable_IconComponent>;\n  /** Fallback icon when no entry in `iconMap` matches. */\n  readonly fallbackIcon?: Unstable_IconComponent;\n};\n\nexport type Unstable_MentionDirective = {\n  readonly formatter: Unstable_DirectiveFormatter;\n  readonly onInserted?: ((item: Unstable_TriggerItem) => void) | undefined;\n};\n\n/**\n * @deprecated Under active development and might change without notice.\n *\n * Creates a spreadable `{ adapter, directive }` bundle for `@` mentions.\n * Supports tools registered in model context, explicit items, or both —\n * flat or categorized.\n *\n * @example\n * ```tsx\n * const mention = unstable_useMentionAdapter();\n * <ComposerTriggerPopover char=\"@\" {...mention} />\n * ```\n */\nexport function unstable_useMentionAdapter(\n  options?: Unstable_UseMentionAdapterOptions,\n): {\n  adapter: Unstable_TriggerAdapter;\n  directive: Unstable_MentionDirective;\n  iconMap?: Record<string, Unstable_IconComponent>;\n  fallbackIcon?: Unstable_IconComponent;\n} {\n  const aui = useAui();\n\n  const items = options?.items;\n  const categories = options?.categories;\n  const includeTools =\n    options?.includeModelContextTools ?? (!items && !categories);\n  const toolsConfig =\n    typeof includeTools === \"object\" ? includeTools : undefined;\n  const wantsTools = includeTools !== false;\n  const formatter = options?.formatter;\n  const onInserted = options?.onInserted;\n\n  const adapter = useMemo<Unstable_TriggerAdapter>(() => {\n    const getModelContextTools = (): Unstable_TriggerItem[] => {\n      if (!wantsTools) return [];\n      const ctx = aui.thread().getModelContext();\n      const tools = ctx.tools;\n      if (!tools) return [];\n      const formatLabel = toolsConfig?.formatLabel;\n      const defaultIcon = toolsConfig?.icon;\n      return Object.entries(tools).map(([name, tool]) =>\n        toTriggerItem({\n          id: name,\n          type: \"tool\",\n          label: formatLabel ? formatLabel(name) : name,\n          description: tool.description ?? undefined,\n          icon: defaultIcon,\n        }),\n      );\n    };\n\n    // Categorized: drill-down mode\n    if (categories && categories.length > 0) {\n      const groups = categories.map((cat) => ({\n        id: cat.id,\n        label: cat.label,\n        items: cat.items.map(toTriggerItem),\n      }));\n\n      let toolCategory: {\n        id: string;\n        label: string;\n        items: Unstable_TriggerItem[];\n      } | null = null;\n      if (wantsTools) {\n        const toolItems = getModelContextTools();\n        if (toolItems.length > 0) {\n          toolCategory = {\n            id: toolsConfig?.category?.id ?? \"tools\",\n            label: toolsConfig?.category?.label ?? \"Tools\",\n            items: toolItems,\n          };\n        }\n      }\n      const allGroups = toolCategory ? [...groups, toolCategory] : groups;\n\n      return {\n        categories: () => allGroups.map(({ id, label }) => ({ id, label })),\n        categoryItems: (id) => allGroups.find((g) => g.id === id)?.items ?? [],\n        search: (query) => {\n          const lower = query.toLowerCase();\n          return allGroups\n            .flatMap((g) => g.items)\n            .filter((item) => matchesQuery(item, lower));\n        },\n      };\n    }\n\n    // Flat: items + (optionally) tools, all in one search pool\n    const flatItems = (items ?? []).map(toTriggerItem);\n    const getFlatPool = (): Unstable_TriggerItem[] => {\n      if (!wantsTools) return flatItems;\n      const toolItems = getModelContextTools();\n      // Dedupe by id — explicit items win.\n      const seen = new Set(flatItems.map((i) => i.id));\n      return [...flatItems, ...toolItems.filter((t) => !seen.has(t.id))];\n    };\n\n    return {\n      categories: (): readonly Unstable_TriggerCategory[] => [],\n      categoryItems: () => [],\n      search: (query) => {\n        const lower = query.toLowerCase();\n        return getFlatPool().filter((item) => matchesQuery(item, lower));\n      },\n    };\n  }, [aui, items, categories, wantsTools, toolsConfig]);\n\n  const directive = useMemo<Unstable_MentionDirective>(\n    () => ({\n      formatter: formatter ?? unstable_defaultDirectiveFormatter,\n      ...(onInserted ? { onInserted } : {}),\n    }),\n    [formatter, onInserted],\n  );\n\n  return {\n    adapter,\n    directive,\n    ...(options?.iconMap ? { iconMap: options.iconMap } : {}),\n    ...(options?.fallbackIcon ? { fallbackIcon: options.fallbackIcon } : {}),\n  };\n}\n\nfunction toTriggerItem(m: Unstable_Mention): Unstable_TriggerItem {\n  const metadata =\n    m.icon !== undefined ? { ...(m.metadata ?? {}), icon: m.icon } : m.metadata;\n  return {\n    id: m.id,\n    type: m.type,\n    label: m.label,\n    ...(m.description !== undefined ? { description: m.description } : {}),\n    ...(metadata !== undefined ? { metadata } : {}),\n  };\n}\n\nfunction matchesQuery(item: Unstable_TriggerItem, lower: string): boolean {\n  if (!lower) return true;\n  if (item.id.toLowerCase().includes(lower)) return true;\n  if (item.label.toLowerCase().includes(lower)) return true;\n  if (item.description?.toLowerCase().includes(lower)) return true;\n  return false;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAsFA,SAAgB,2BACd,SAMA;CACA,MAAM,MAAM,OAAO;CAEnB,MAAM,QAAQ,SAAS;CACvB,MAAM,aAAa,SAAS;CAC5B,MAAM,eACJ,SAAS,6BAA6B,CAAC,SAAS,CAAC;CACnD,MAAM,cACJ,OAAO,iBAAiB,WAAW,eAAe,KAAA;CACpD,MAAM,aAAa,iBAAiB;CACpC,MAAM,YAAY,SAAS;CAC3B,MAAM,aAAa,SAAS;CAsF5B,OAAO;EACL,SArFc,cAAuC;GACrD,MAAM,6BAAqD;IACzD,IAAI,CAAC,YAAY,OAAO,CAAC;IAEzB,MAAM,QADM,IAAI,OAAO,EAAE,gBACT,EAAE;IAClB,IAAI,CAAC,OAAO,OAAO,CAAC;IACpB,MAAM,cAAc,aAAa;IACjC,MAAM,cAAc,aAAa;IACjC,OAAO,OAAO,QAAQ,KAAK,EAAE,KAAK,CAAC,MAAM,UACvC,cAAc;KACZ,IAAI;KACJ,MAAM;KACN,OAAO,cAAc,YAAY,IAAI,IAAI;KACzC,aAAa,KAAK,eAAe,KAAA;KACjC,MAAM;IACR,CAAC,CACH;GACF;GAGA,IAAI,cAAc,WAAW,SAAS,GAAG;IACvC,MAAM,SAAS,WAAW,KAAK,SAAS;KACtC,IAAI,IAAI;KACR,OAAO,IAAI;KACX,OAAO,IAAI,MAAM,IAAI,aAAa;IACpC,EAAE;IAEF,IAAI,eAIO;IACX,IAAI,YAAY;KACd,MAAM,YAAY,qBAAqB;KACvC,IAAI,UAAU,SAAS,GACrB,eAAe;MACb,IAAI,aAAa,UAAU,MAAM;MACjC,OAAO,aAAa,UAAU,SAAS;MACvC,OAAO;KACT;IAEJ;IACA,MAAM,YAAY,eAAe,CAAC,GAAG,QAAQ,YAAY,IAAI;IAE7D,OAAO;KACL,kBAAkB,UAAU,KAAK,EAAE,IAAI,aAAa;MAAE;MAAI;KAAM,EAAE;KAClE,gBAAgB,OAAO,UAAU,MAAM,MAAM,EAAE,OAAO,EAAE,GAAG,SAAS,CAAC;KACrE,SAAS,UAAU;MACjB,MAAM,QAAQ,MAAM,YAAY;MAChC,OAAO,UACJ,SAAS,MAAM,EAAE,KAAK,EACtB,QAAQ,SAAS,aAAa,MAAM,KAAK,CAAC;KAC/C;IACF;GACF;GAGA,MAAM,aAAa,SAAS,CAAC,GAAG,IAAI,aAAa;GACjD,MAAM,oBAA4C;IAChD,IAAI,CAAC,YAAY,OAAO;IACxB,MAAM,YAAY,qBAAqB;IAEvC,MAAM,OAAO,IAAI,IAAI,UAAU,KAAK,MAAM,EAAE,EAAE,CAAC;IAC/C,OAAO,CAAC,GAAG,WAAW,GAAG,UAAU,QAAQ,MAAM,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC,CAAC;GACnE;GAEA,OAAO;IACL,kBAAuD,CAAC;IACxD,qBAAqB,CAAC;IACtB,SAAS,UAAU;KACjB,MAAM,QAAQ,MAAM,YAAY;KAChC,OAAO,YAAY,EAAE,QAAQ,SAAS,aAAa,MAAM,KAAK,CAAC;IACjE;GACF;EACF,GAAG;GAAC;GAAK;GAAO;GAAY;GAAY;EAAW,CAW3C;EACN,WAVgB,eACT;GACL,WAAW,aAAa;GACxB,GAAI,aAAa,EAAE,WAAW,IAAI,CAAC;EACrC,IACA,CAAC,WAAW,UAAU,CAKd;EACR,GAAI,SAAS,UAAU,EAAE,SAAS,QAAQ,QAAQ,IAAI,CAAC;EACvD,GAAI,SAAS,eAAe,EAAE,cAAc,QAAQ,aAAa,IAAI,CAAC;CACxE;AACF;AAEA,SAAS,cAAc,GAA2C;CAChE,MAAM,WACJ,EAAE,SAAS,KAAA,IAAY;EAAE,GAAI,EAAE,YAAY,CAAC;EAAI,MAAM,EAAE;CAAK,IAAI,EAAE;CACrE,OAAO;EACL,IAAI,EAAE;EACN,MAAM,EAAE;EACR,OAAO,EAAE;EACT,GAAI,EAAE,gBAAgB,KAAA,IAAY,EAAE,aAAa,EAAE,YAAY,IAAI,CAAC;EACpE,GAAI,aAAa,KAAA,IAAY,EAAE,SAAS,IAAI,CAAC;CAC/C;AACF;AAEA,SAAS,aAAa,MAA4B,OAAwB;CACxE,IAAI,CAAC,OAAO,OAAO;CACnB,IAAI,KAAK,GAAG,YAAY,EAAE,SAAS,KAAK,GAAG,OAAO;CAClD,IAAI,KAAK,MAAM,YAAY,EAAE,SAAS,KAAK,GAAG,OAAO;CACrD,IAAI,KAAK,aAAa,YAAY,EAAE,SAAS,KAAK,GAAG,OAAO;CAC5D,OAAO;AACT"}

package/dist/utils/useToolArgsFieldStatus.d.ts

@@ -1,5 +1,7 @@
 //#region src/utils/useToolArgsFieldStatus.d.ts
 declare const useToolArgsFieldStatus: (fieldPath: (string | number)[]) => {
+  type: string;
+} | {
   readonly type: "running";
 } | {
   readonly type: "complete";
@@ -7,8 +9,6 @@ declare const useToolArgsFieldStatus: (fieldPath: (string | number)[]) => {
   readonly type: "incomplete";
   readonly reason: "cancelled" | "length" | "content-filter" | "other" | "error";
   readonly error?: unknown;
-} | {
-  type: string;
 };
 //#endregion
 export { useToolArgsFieldStatus };

package/dist/utils/useToolArgsFieldStatus.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useToolArgsFieldStatus.d.ts","names":[],"sources":["../../src/utils/useToolArgsFieldStatus.ts"],"mappings":";cAKa,sBAAA,GAA0B,SAAA;EAAA"}
+{"version":3,"file":"useToolArgsFieldStatus.d.ts","names":[],"sources":["../../src/utils/useToolArgsFieldStatus.ts"],"mappings":";cAKa,sBAAA,GAA0B,SAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react",
-  "version": "0.14.12",
+  "version": "0.14.14",
   "description": "Open-source TypeScript/React library for building production-grade AI chat experiences",
   "keywords": [
     "radix-ui",
@@ -48,7 +48,7 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.2.8",
+    "@assistant-ui/core": "^0.2.10",
     "@assistant-ui/store": "^0.2.13",
     "@assistant-ui/tap": "^0.5.14",
     "@radix-ui/primitive": "^1.1.3",
@@ -57,8 +57,8 @@
     "@radix-ui/react-primitive": "^2.1.4",
     "@radix-ui/react-use-callback-ref": "^1.1.1",
     "@radix-ui/react-use-escape-keydown": "^1.1.1",
-    "assistant-cloud": "^0.1.30",
-    "assistant-stream": "^0.3.18",
+    "assistant-cloud": "^0.1.31",
+    "assistant-stream": "^0.3.20",
     "nanoid": "^5.1.11",
     "radix-ui": "^1.4.3",
     "react-textarea-autosize": "^8.5.9",
@@ -90,7 +90,7 @@
     "react": "^19.2.6",
     "react-dom": "^19.2.6",
     "vitest": "^4.1.7",
-    "@assistant-ui/x-buildutils": "0.0.10"
+    "@assistant-ui/x-buildutils": "0.0.11"
   },
   "publishConfig": {
     "access": "public",

package/src/client/InMemoryThreadList.ts

@@ -24,6 +24,7 @@ type ThreadData = {
   id: string;
   title?: string;
   status: "regular" | "archived";
+  custom?: Record<string, unknown> | undefined;
 };
 
 // ThreadListItem Client
@@ -31,11 +32,19 @@ const ThreadListItemClient = resource(
   (props: {
     data: ThreadData;
     onSwitchTo: () => void;
+    onUpdateCustom: (custom: Record<string, unknown> | undefined) => void;
     onArchive: () => void;
     onUnarchive: () => void;
     onDelete: () => void;
   }): ClientOutput<"threadListItem"> => {
-    const { data, onSwitchTo, onArchive, onUnarchive, onDelete } = props;
+    const {
+      data,
+      onSwitchTo,
+      onUpdateCustom,
+      onArchive,
+      onUnarchive,
+      onDelete,
+    } = props;
     const state = tapMemo(
       () => ({
         id: data.id,
@@ -43,14 +52,16 @@ const ThreadListItemClient = resource(
         externalId: undefined,
         title: data.title,
         status: data.status,
+        custom: data.custom,
       }),
-      [data.id, data.title, data.status],
+      [data.id, data.title, data.status, data.custom],
     );
 
     return {
       getState: () => state,
       switchTo: onSwitchTo,
       rename: () => {},
+      updateCustom: onUpdateCustom,
       archive: onArchive,
       unarchive: onUnarchive,
       delete: onDelete,
@@ -96,6 +107,15 @@ export const InMemoryThreadList = resource(
       );
     };
 
+    const handleUpdateCustom = (
+      threadId: string,
+      custom: Record<string, unknown> | undefined,
+    ) => {
+      setThreads((prev) =>
+        prev.map((t) => (t.id === threadId ? { ...t, custom } : t)),
+      );
+    };
+
     const handleDelete = (threadId: string) => {
       setThreads((prev) => prev.filter((t) => t.id !== threadId));
       if (mainThreadId === threadId) {
@@ -122,6 +142,7 @@ export const InMemoryThreadList = resource(
             ThreadListItemClient({
               data: t,
               onSwitchTo: () => handleSwitchToThread(t.id),
+              onUpdateCustom: (custom) => handleUpdateCustom(t.id, custom),
               onArchive: () => handleArchive(t.id),
               onUnarchive: () => handleUnarchive(t.id),
               onDelete: () => handleDelete(t.id),

package/src/client/SingleThreadList.ts

@@ -1,4 +1,4 @@
-import { resource, tapMemo } from "@assistant-ui/tap";
+import { resource, tapMemo, tapState } from "@assistant-ui/tap";
 import {
   type ClientElement,
   type ClientOutput,
@@ -9,6 +9,8 @@ const RESOLVED_PROMISE = Promise.resolve();
 const THREAD_ID = "default";
 
 const SingleThreadListItem = resource((): ClientOutput<"threadListItem"> => {
+  const [custom, setCustom] = tapState<Record<string, unknown> | undefined>();
+
   return {
     getState: () => ({
       id: THREAD_ID,
@@ -16,9 +18,11 @@ const SingleThreadListItem = resource((): ClientOutput<"threadListItem"> => {
       externalId: undefined,
       title: undefined,
       status: "regular",
+      custom,
     }),
     switchTo: () => {},
     rename: () => {},
+    updateCustom: setCustom,
     archive: () => {},
     unarchive: () => {},
     delete: () => {},

package/src/index.ts

@@ -187,8 +187,18 @@ export {
   useInlineRender,
   type Toolkit,
   type ToolDefinition,
-  type ToolkitDeclaration,
-  type ToolkitDeclarationDefinition,
+  type ToolCallText,
+  type ToolkitDefinition,
+  type ToolkitDefinitionEntry,
+  defineToolkit,
+  stubTool,
+  useAuiToolOverrides,
+  hitl,
+  hitlTool,
+  providerTool,
+  type ProviderToolConfig,
+  defineMcpToolkit,
+  type McpToolkitDefinition,
   Tools,
   DataRenderers,
   Interactables,

package/src/legacy-runtime/runtime-cores/assistant-transport/replayBoundaryStream.test.ts

@@ -0,0 +1,426 @@
+// @vitest-environment jsdom
+
+import {
+  AssistantMessageAccumulator,
+  DataStreamDecoder,
+} from "assistant-stream";
+import { act, renderHook } from "@testing-library/react";
+import { describe, expect, it, vi } from "vitest";
+import {
+  createReplayBoundaryStream,
+  REPLAY_CONTENT_LENGTH_HEADER,
+  useReplayRenderWait,
+} from "./replayBoundaryStream";
+
+const encoder = new TextEncoder();
+const decoder = new TextDecoder();
+
+const createBody = (chunks: readonly string[]) =>
+  new ReadableStream<Uint8Array>({
+    start(controller) {
+      for (const chunk of chunks) {
+        controller.enqueue(encoder.encode(chunk));
+      }
+      controller.close();
+    },
+  });
+
+const createResponse = (
+  chunks: readonly string[],
+  replayContentLength?: number | string,
+) =>
+  new Response(createBody(chunks), {
+    headers:
+      replayContentLength === undefined
+        ? undefined
+        : { [REPLAY_CONTENT_LENGTH_HEADER]: String(replayContentLength) },
+  });
+
+const createRenderWait = () => {
+  const pending: Array<() => void> = [];
+  const waitForRender = vi.fn(
+    () =>
+      new Promise<void>((resolve) => {
+        pending.push(resolve);
+      }),
+  );
+
+  const releaseNext = async () => {
+    for (let i = 0; pending.length === 0 && i < 10; i++) {
+      await Promise.resolve();
+    }
+
+    const resolve = pending.shift();
+    expect(resolve).toBeDefined();
+    resolve!();
+    await Promise.resolve();
+  };
+
+  return { waitForRender, releaseNext };
+};
+
+const readText = async (stream: ReadableStream<Uint8Array>) => {
+  const reader = stream.getReader();
+  const chunks: string[] = [];
+
+  while (true) {
+    const { done, value } = await reader.read();
+    if (done) break;
+    chunks.push(decoder.decode(value, { stream: true }));
+  }
+  chunks.push(decoder.decode());
+
+  return chunks.join("");
+};
+
+describe("useReplayRenderWait", () => {
+  it("resolves after its own render ticket commits", async () => {
+    vi.useFakeTimers();
+
+    try {
+      const { result } = renderHook(() => useReplayRenderWait());
+
+      let resolved = false;
+      const wait = result.current().then(() => {
+        resolved = true;
+      });
+
+      await Promise.resolve();
+      expect(resolved).toBe(false);
+
+      await act(async () => {
+        vi.runOnlyPendingTimers();
+      });
+      await wait;
+
+      expect(resolved).toBe(true);
+    } finally {
+      vi.useRealTimers();
+    }
+  });
+});
+
+describe("createReplayBoundaryStream", () => {
+  it("short-circuits responses without a valid replay content length", async () => {
+    const setReplaying = vi.fn();
+    const waitForRender = vi.fn();
+
+    for (const replayContentLength of [undefined, "abc", "3.5", "-1"]) {
+      setReplaying.mockClear();
+      waitForRender.mockClear();
+
+      const body = await createReplayBoundaryStream(
+        createResponse(["live"], replayContentLength),
+        {
+          setReplaying,
+          waitForRender,
+        },
+      );
+
+      expect(await readText(body)).toBe("live");
+      expect(setReplaying).not.toHaveBeenCalled();
+      expect(waitForRender).not.toHaveBeenCalled();
+    }
+  });
+
+  it("pauses at the replay boundary before releasing live bytes", async () => {
+    const { waitForRender, releaseNext } = createRenderWait();
+    const setReplaying = vi.fn();
+    const replayPrefix = '0:"hi"\nb:{"toolCallId":"call-1","toolName":"te';
+    const liveSuffix = 'st"}\n';
+    const replayContentLength = encoder.encode(replayPrefix).byteLength;
+
+    const streamPromise = createReplayBoundaryStream(
+      createResponse([replayPrefix + liveSuffix], replayContentLength),
+      { setReplaying, waitForRender },
+    );
+
+    expect(setReplaying).toHaveBeenCalledWith(true);
+    expect(waitForRender).toHaveBeenCalledTimes(1);
+
+    await releaseNext();
+    const stream = await streamPromise;
+    const reader = stream.getReader();
+
+    await expect(reader.read()).resolves.toMatchObject({
+      done: false,
+      value: encoder.encode(replayPrefix),
+    });
+    expect(waitForRender).toHaveBeenCalledTimes(2);
+    expect(setReplaying).toHaveBeenCalledTimes(1);
+
+    let liveReadResolved = false;
+    const liveRead = reader.read().then((read) => {
+      liveReadResolved = true;
+      return read;
+    });
+    await Promise.resolve();
+    expect(liveReadResolved).toBe(false);
+
+    await releaseNext();
+    expect(setReplaying).toHaveBeenLastCalledWith(false);
+    expect(waitForRender).toHaveBeenCalledTimes(3);
+    await Promise.resolve();
+    expect(liveReadResolved).toBe(false);
+
+    await releaseNext();
+    await expect(liveRead).resolves.toMatchObject({
+      done: false,
+      value: encoder.encode(liveSuffix),
+    });
+  });
+
+  it("pauses when a chunk ends exactly at the replay boundary", async () => {
+    const { waitForRender, releaseNext } = createRenderWait();
+    const setReplaying = vi.fn();
+    const replayPrefix = "replay";
+    const liveSuffix = "live";
+    const replayContentLength = encoder.encode(replayPrefix).byteLength;
+
+    const streamPromise = createReplayBoundaryStream(
+      createResponse([replayPrefix, liveSuffix], replayContentLength),
+      { setReplaying, waitForRender },
+    );
+
+    await releaseNext();
+    const stream = await streamPromise;
+    const reader = stream.getReader();
+
+    await expect(reader.read()).resolves.toMatchObject({
+      done: false,
+      value: encoder.encode(replayPrefix),
+    });
+    expect(setReplaying).toHaveBeenCalledTimes(1);
+
+    let liveReadResolved = false;
+    const liveRead = reader.read().then((read) => {
+      liveReadResolved = true;
+      return read;
+    });
+    await Promise.resolve();
+    expect(liveReadResolved).toBe(false);
+
+    await releaseNext();
+    expect(setReplaying).toHaveBeenLastCalledWith(false);
+    await releaseNext();
+
+    await expect(liveRead).resolves.toMatchObject({
+      done: false,
+      value: encoder.encode(liveSuffix),
+    });
+  });
+
+  it("accumulates replay bytes across chunks before splitting live bytes", async () => {
+    const { waitForRender, releaseNext } = createRenderWait();
+    const setReplaying = vi.fn();
+    const firstReplayChunk = "part1";
+    const secondReplayChunk = "part2";
+    const firstLiveChunk = "live1";
+    const secondLiveChunk = "live2";
+    const replayContentLength = encoder.encode(
+      firstReplayChunk + secondReplayChunk,
+    ).byteLength;
+
+    const streamPromise = createReplayBoundaryStream(
+      createResponse(
+        [firstReplayChunk, secondReplayChunk + firstLiveChunk, secondLiveChunk],
+        replayContentLength,
+      ),
+      { setReplaying, waitForRender },
+    );
+
+    await releaseNext();
+    const stream = await streamPromise;
+    const reader = stream.getReader();
+
+    await expect(reader.read()).resolves.toMatchObject({
+      done: false,
+      value: encoder.encode(firstReplayChunk),
+    });
+    await expect(reader.read()).resolves.toMatchObject({
+      done: false,
+      value: encoder.encode(secondReplayChunk),
+    });
+    expect(setReplaying).toHaveBeenCalledTimes(1);
+
+    let liveReadResolved = false;
+    const liveRead = reader.read().then((read) => {
+      liveReadResolved = true;
+      return read;
+    });
+    await Promise.resolve();
+    expect(liveReadResolved).toBe(false);
+
+    await releaseNext();
+    expect(setReplaying).toHaveBeenLastCalledWith(false);
+    await releaseNext();
+
+    await expect(liveRead).resolves.toMatchObject({
+      done: false,
+      value: encoder.encode(firstLiveChunk),
+    });
+    await expect(reader.read()).resolves.toMatchObject({
+      done: false,
+      value: encoder.encode(secondLiveChunk),
+    });
+  });
+
+  it("clears replaying when the stream ends before the boundary", async () => {
+    const { waitForRender, releaseNext } = createRenderWait();
+    const setReplaying = vi.fn();
+    const streamPromise = createReplayBoundaryStream(
+      createResponse(["hi"], 10),
+      {
+        setReplaying,
+        waitForRender,
+      },
+    );
+
+    await releaseNext();
+    const stream = await streamPromise;
+    const text = readText(stream);
+    await releaseNext();
+    await releaseNext();
+
+    await expect(text).resolves.toBe("hi");
+    expect(setReplaying).toHaveBeenLastCalledWith(false);
+  });
+
+  it("clears replaying when the gated stream is cancelled", async () => {
+    const { waitForRender, releaseNext } = createRenderWait();
+    const setReplaying = vi.fn();
+    let cancelled = false;
+    const body = new ReadableStream<Uint8Array>({
+      start(controller) {
+        controller.enqueue(encoder.encode("hi"));
+      },
+      cancel() {
+        cancelled = true;
+      },
+    });
+    const streamPromise = createReplayBoundaryStream(
+      new Response(body, { headers: { [REPLAY_CONTENT_LENGTH_HEADER]: "10" } }),
+      { setReplaying, waitForRender },
+    );
+
+    await releaseNext();
+    const stream = await streamPromise;
+    await stream.cancel("done");
+
+    expect(setReplaying).toHaveBeenLastCalledWith(false);
+    expect(cancelled).toBe(true);
+  });
+
+  it("does not wait for replay completion after cancellation unblocks a read", async () => {
+    const { waitForRender, releaseNext } = createRenderWait();
+    const setReplaying = vi.fn();
+    let cancelled = false;
+    const body = new ReadableStream<Uint8Array>({
+      cancel() {
+        cancelled = true;
+      },
+    });
+    const streamPromise = createReplayBoundaryStream(
+      new Response(body, { headers: { [REPLAY_CONTENT_LENGTH_HEADER]: "10" } }),
+      { setReplaying, waitForRender },
+    );
+
+    await releaseNext();
+    const stream = await streamPromise;
+    const reader = stream.getReader();
+    const read = reader.read();
+
+    await Promise.resolve();
+    expect(waitForRender).toHaveBeenCalledTimes(1);
+
+    await reader.cancel("done");
+    await expect(read).resolves.toMatchObject({ done: true });
+
+    expect(waitForRender).toHaveBeenCalledTimes(1);
+    expect(setReplaying).toHaveBeenLastCalledWith(false);
+    expect(cancelled).toBe(true);
+  });
+
+  it("does not clear replaying twice when cancelled during replay completion", async () => {
+    const { waitForRender, releaseNext } = createRenderWait();
+    const setReplaying = vi.fn();
+    const replayStr = "replay";
+    const replayContentLength = encoder.encode(replayStr).byteLength;
+
+    const streamPromise = createReplayBoundaryStream(
+      createResponse([replayStr], replayContentLength),
+      { setReplaying, waitForRender },
+    );
+
+    await releaseNext();
+    const stream = await streamPromise;
+    const reader = stream.getReader();
+
+    await expect(reader.read()).resolves.toMatchObject({
+      done: false,
+      value: encoder.encode(replayStr),
+    });
+    expect(waitForRender).toHaveBeenCalledTimes(2);
+
+    const cancel = reader.cancel("done");
+    await Promise.resolve();
+    expect(setReplaying).toHaveBeenCalledTimes(1);
+
+    await releaseNext();
+    expect(setReplaying).toHaveBeenCalledTimes(2);
+    expect(setReplaying).toHaveBeenLastCalledWith(false);
+
+    await releaseNext();
+    await cancel;
+    expect(setReplaying).toHaveBeenCalledTimes(2);
+  });
+
+  it("lets data-stream parsing commit replayed text before live tool calls", async () => {
+    const { waitForRender, releaseNext } = createRenderWait();
+    const setReplaying = vi.fn();
+    const replayPrefix = '0:"hi"\nb:{"toolCallId":"call-1","toolName":"te';
+    const liveSuffix = 'st"}\n';
+    const replayContentLength = encoder.encode(replayPrefix).byteLength;
+
+    const streamPromise = createReplayBoundaryStream(
+      createResponse([replayPrefix + liveSuffix], replayContentLength),
+      { setReplaying, waitForRender },
+    );
+
+    await releaseNext();
+    const messages = (await streamPromise)
+      .pipeThrough(new DataStreamDecoder())
+      .pipeThrough(new AssistantMessageAccumulator({ throttle: true }));
+    const reader = messages.getReader();
+
+    let sawReplayedText = false;
+    while (!sawReplayedText) {
+      const replayedMessage = await reader.read();
+      expect(replayedMessage.done).toBe(false);
+      expect(
+        replayedMessage.value?.parts.some((part) => part.type === "tool-call"),
+      ).toBe(false);
+      sawReplayedText =
+        replayedMessage.value?.parts.some(
+          (part) => part.type === "text" && part.text === "hi",
+        ) ?? false;
+    }
+    expect(setReplaying).toHaveBeenCalledTimes(1);
+
+    const liveMessagePromise = reader.read();
+    await releaseNext();
+    expect(setReplaying).toHaveBeenLastCalledWith(false);
+    await releaseNext();
+
+    let liveMessage = await liveMessagePromise;
+    while (
+      !liveMessage.done &&
+      !liveMessage.value?.parts.some(
+        (part) => part.type === "tool-call" && part.toolName === "test",
+      )
+    ) {
+      liveMessage = await reader.read();
+    }
+    expect(liveMessage.done).toBe(false);
+  });
+});