@anvia/react-ui 0.7.1 → 1.0.0-rc.10

package/dist/chunk-647LAYRV.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/thread-list/parts.tsx","../src/thread-list/index.ts"],"sourcesContent":["import {\n  forwardRef,\n  type KeyboardEvent,\n  type MouseEvent,\n  type ReactNode,\n  useCallback,\n  useMemo,\n  useRef,\n} from \"react\";\n\nimport {\n  InternalThreadListItemProvider,\n  InternalThreadListRootProvider,\n  type ThreadListRecord,\n  useInternalThreadListRoot,\n  useThreadList,\n  useThreadListItem,\n} from \"../contexts\";\nimport { composeRefs, type PrimitiveProps, renderPrimitive } from \"../primitives\";\n\ntype ThreadListRootProps = PrimitiveProps<\"div\">;\n\nconst ThreadListRoot = forwardRef<HTMLDivElement, ThreadListRootProps>(function ThreadListRoot(\n  { onKeyDown, ...props },\n  ref,\n) {\n  const threadList = useThreadList();\n  const rootRef = useRef<HTMLDivElement | null>(null);\n  const newButtonRef = useRef<HTMLButtonElement | null>(null);\n  const itemElements = useRef(new Map<string, HTMLDivElement>());\n  const triggerElements = useRef(new Map<string, HTMLButtonElement>());\n  const registerItem = useCallback((threadId: string, element: HTMLDivElement | null) => {\n    if (element === null) itemElements.current.delete(threadId);\n    else itemElements.current.set(threadId, element);\n  }, []);\n  const registerTrigger = useCallback((threadId: string, element: HTMLButtonElement | null) => {\n    if (element === null) triggerElements.current.delete(threadId);\n    else triggerElements.current.set(threadId, element);\n  }, []);\n  const rootContext = useMemo(\n    () => ({\n      rootRef,\n      newButtonRef,\n      itemElements,\n      triggerElements,\n      registerItem,\n      registerTrigger,\n    }),\n    [registerItem, registerTrigger],\n  );\n  const composedRef = useMemo(\n    () =>\n      composeRefs<HTMLDivElement>(ref, (node) => {\n        rootRef.current = node;\n      }),\n    [ref],\n  );\n\n  const handleKeyDown = useCallback(\n    (event: KeyboardEvent<HTMLDivElement>) => {\n      onKeyDown?.(event);\n      if (event.defaultPrevented || ![\"ArrowDown\", \"ArrowUp\", \"Home\", \"End\"].includes(event.key)) {\n        return;\n      }\n      const triggers = orderedEnabledTriggers(rootContext);\n      if (triggers.length === 0) {\n        return;\n      }\n      event.preventDefault();\n      const activeElement = document.activeElement;\n      const currentIndex =\n        activeElement instanceof HTMLButtonElement ? triggers.indexOf(activeElement) : -1;\n      const nextIndex = nextTriggerIndex(event.key, currentIndex, triggers.length);\n      triggers[nextIndex]?.focus();\n    },\n    [onKeyDown, rootContext],\n  );\n\n  return (\n    <InternalThreadListRootProvider value={rootContext}>\n      {renderPrimitive(\n        \"div\",\n        {\n          ...props,\n          onKeyDown: handleKeyDown,\n          \"data-state\": threadList.status ?? \"idle\",\n        } as PrimitiveProps<\"div\">,\n        composedRef,\n      )}\n    </InternalThreadListRootProvider>\n  );\n});\n\nconst ThreadListNew = forwardRef<HTMLButtonElement, PrimitiveProps<\"button\">>(\n  function ThreadListNew({ onClick, ...props }, ref) {\n    const threadList = useThreadList();\n    const root = useInternalThreadListRoot();\n    const composedRef = useMemo(\n      () =>\n        composeRefs<HTMLButtonElement>(ref, (node) => {\n          root.newButtonRef.current = node;\n        }),\n      [ref, root.newButtonRef],\n    );\n\n    const handleClick = useCallback(\n      (event: MouseEvent<HTMLButtonElement>) => {\n        onClick?.(event);\n        if (event.defaultPrevented || props.disabled) {\n          return;\n        }\n        void threadList.createThread();\n      },\n      [onClick, props.disabled, threadList],\n    );\n\n    return renderPrimitive(\n      \"button\",\n      {\n        ...props,\n        children: props.children ?? \"New chat\",\n        onClick: handleClick,\n        type: props.type ?? \"button\",\n        \"data-state\": props.disabled ? \"disabled\" : \"enabled\",\n      } as PrimitiveProps<\"button\">,\n      composedRef,\n    );\n  },\n);\n\ntype ThreadListItemsChildren = ReactNode | ((thread: ThreadListRecord) => ReactNode);\n\ntype ThreadListItemsProps = Omit<PrimitiveProps<\"div\">, \"children\"> & {\n  archived?: boolean;\n  children?: ThreadListItemsChildren;\n  keepMounted?: boolean;\n};\n\nconst ThreadListItems = forwardRef<HTMLDivElement, ThreadListItemsProps>(function ThreadListItems(\n  { archived = false, children, keepMounted = false, ...props },\n  ref,\n) {\n  const threadList = useThreadList();\n  const threads = threadList.threads.filter((thread) => Boolean(thread.archived) === archived);\n  const empty = threads.length === 0;\n  if (empty && !keepMounted) {\n    return null;\n  }\n\n  return renderPrimitive(\n    \"div\",\n    {\n      ...props,\n      children: threads.map((thread) => (\n        <InternalThreadListItemProvider\n          key={thread.id}\n          value={{ thread, active: thread.id === threadList.activeThreadId }}\n        >\n          {typeof children === \"function\" ? children(thread) : (children ?? <DefaultThreadItem />)}\n        </InternalThreadListItemProvider>\n      )),\n      \"data-state\": empty ? \"empty\" : \"populated\",\n    } as PrimitiveProps<\"div\">,\n    ref,\n  );\n});\n\ntype ThreadListEmptyProps = PrimitiveProps<\"div\"> & {\n  archived?: boolean;\n};\n\nconst ThreadListEmpty = forwardRef<HTMLDivElement, ThreadListEmptyProps>(function ThreadListEmpty(\n  { archived = false, ...props },\n  ref,\n) {\n  const threadList = useThreadList();\n  const hasVisibleThreads = threadList.threads.some(\n    (thread) => Boolean(thread.archived) === archived,\n  );\n  if (hasVisibleThreads) {\n    return null;\n  }\n\n  return renderPrimitive(\n    \"div\",\n    {\n      ...props,\n      children: props.children ?? \"No conversations.\",\n    } as PrimitiveProps<\"div\">,\n    ref,\n  );\n});\n\nconst ThreadListItemRoot = forwardRef<HTMLDivElement, PrimitiveProps<\"div\">>(\n  function ThreadListItemRoot(props, ref) {\n    const item = useThreadListItem();\n    const root = useInternalThreadListRoot();\n    const composedRef = useMemo(\n      () => composeRefs<HTMLDivElement>(ref, (node) => root.registerItem(item.thread.id, node)),\n      [item.thread.id, ref, root],\n    );\n\n    return renderPrimitive(\n      \"div\",\n      {\n        ...props,\n        \"data-state\": item.active ? \"active\" : \"inactive\",\n      } as PrimitiveProps<\"div\">,\n      composedRef,\n    );\n  },\n);\n\nconst ThreadListItemTrigger = forwardRef<HTMLButtonElement, PrimitiveProps<\"button\">>(\n  function ThreadListItemTrigger({ onClick, ...props }, ref) {\n    const threadList = useThreadList();\n    const item = useThreadListItem();\n    const root = useInternalThreadListRoot();\n    const disabled = props.disabled ?? false;\n    const composedRef = useMemo(\n      () =>\n        composeRefs<HTMLButtonElement>(ref, (node) => root.registerTrigger(item.thread.id, node)),\n      [item.thread.id, ref, root],\n    );\n\n    const handleClick = useCallback(\n      (event: MouseEvent<HTMLButtonElement>) => {\n        onClick?.(event);\n        if (event.defaultPrevented || disabled) {\n          return;\n        }\n        void threadList.switchThread(item.thread.id);\n      },\n      [disabled, item.thread.id, onClick, threadList],\n    );\n\n    return renderPrimitive(\n      \"button\",\n      {\n        ...props,\n        children: props.children ?? <ThreadListItemTitle />,\n        disabled,\n        onClick: handleClick,\n        type: props.type ?? \"button\",\n        \"aria-current\": item.active ? \"true\" : undefined,\n        \"data-state\": disabled ? \"disabled\" : \"enabled\",\n      } as PrimitiveProps<\"button\">,\n      composedRef,\n    );\n  },\n);\n\ntype ThreadListItemTitleProps = PrimitiveProps<\"span\"> & {\n  fallback?: ReactNode;\n};\n\nconst ThreadListItemTitle = forwardRef<HTMLSpanElement, ThreadListItemTitleProps>(\n  function ThreadListItemTitle({ fallback = \"New chat\", ...props }, ref) {\n    const { thread } = useThreadListItem();\n\n    return renderPrimitive(\n      \"span\",\n      {\n        ...props,\n        children: props.children ?? thread.title ?? fallback,\n      } as PrimitiveProps<\"span\">,\n      ref,\n    );\n  },\n);\n\ntype ThreadListItemActionKey = \"archiveThread\" | \"unarchiveThread\" | \"deleteThread\";\nfunction createThreadListItemAction(action: ThreadListItemActionKey, defaultLabel: string) {\n  return forwardRef<HTMLButtonElement, PrimitiveProps<\"button\">>(function ThreadListItemAction(\n    { onClick, ...props },\n    ref,\n  ) {\n    const threadList = useThreadList();\n    const { thread } = useThreadListItem();\n    const root = useInternalThreadListRoot();\n    const actionHandler = threadList[action];\n    const disabled = props.disabled ?? actionHandler === undefined;\n\n    const handleClick = useCallback(\n      (event: MouseEvent<HTMLButtonElement>) => {\n        onClick?.(event);\n        if (event.defaultPrevented || disabled) {\n          return;\n        }\n        focusAfterThreadItemRemoval(thread.id, root);\n        void actionHandler?.(thread.id);\n      },\n      [actionHandler, disabled, onClick, root, thread.id],\n    );\n\n    return renderPrimitive(\n      \"button\",\n      {\n        ...props,\n        children: props.children ?? defaultLabel,\n        disabled,\n        onClick: handleClick,\n        type: props.type ?? \"button\",\n        \"data-state\": disabled ? \"disabled\" : \"enabled\",\n      } as PrimitiveProps<\"button\">,\n      ref,\n    );\n  });\n}\n\nconst ThreadListItemArchive = createThreadListItemAction(\"archiveThread\", \"Archive\");\nconst ThreadListItemUnarchive = createThreadListItemAction(\"unarchiveThread\", \"Unarchive\");\nconst ThreadListItemDelete = createThreadListItemAction(\"deleteThread\", \"Delete\");\n\nfunction DefaultThreadItem(): ReactNode {\n  return (\n    <ThreadListItemRoot>\n      <ThreadListItemTrigger />\n    </ThreadListItemRoot>\n  );\n}\n\nfunction nextTriggerIndex(key: string, currentIndex: number, triggerCount: number): number {\n  if (key === \"Home\") {\n    return 0;\n  }\n  if (key === \"End\") {\n    return triggerCount - 1;\n  }\n  if (currentIndex === -1) {\n    return key === \"ArrowUp\" ? triggerCount - 1 : 0;\n  }\n  if (key === \"ArrowUp\") {\n    return (currentIndex - 1 + triggerCount) % triggerCount;\n  }\n  return (currentIndex + 1) % triggerCount;\n}\n\nfunction focusAfterThreadItemRemoval(\n  threadId: string,\n  rootContext: ReturnType<typeof useInternalThreadListRoot>,\n): void {\n  const root = rootContext.rootRef.current;\n  const currentItem = rootContext.itemElements.current.get(threadId);\n  if (\n    root === null ||\n    currentItem === undefined ||\n    !currentItem.contains(root.ownerDocument.activeElement)\n  ) {\n    return;\n  }\n\n  const triggers = orderedEnabledTriggers(rootContext);\n  const currentTrigger = rootContext.triggerElements.current.get(threadId);\n  const currentIndex = currentTrigger === undefined ? -1 : triggers.indexOf(currentTrigger);\n  const remainingTriggers = triggers.filter((trigger) => !currentItem.contains(trigger));\n  const fallback = rootContext.newButtonRef.current;\n  const focusTarget =\n    remainingTriggers[Math.min(Math.max(currentIndex, 0), remainingTriggers.length - 1)] ??\n    fallback ??\n    root;\n\n  if (focusTarget === root && !root.hasAttribute(\"tabindex\")) {\n    root.tabIndex = -1;\n  }\n  focusTarget.focus();\n}\n\nfunction orderedEnabledTriggers(\n  rootContext: ReturnType<typeof useInternalThreadListRoot>,\n): HTMLButtonElement[] {\n  return [...rootContext.triggerElements.current.values()]\n    .filter((trigger) => trigger.isConnected && !trigger.disabled)\n    .sort((left, right) =>\n      left.compareDocumentPosition(right) & Node.DOCUMENT_POSITION_FOLLOWING ? -1 : 1,\n    );\n}\n\nexport {\n  ThreadListEmpty,\n  ThreadListItemArchive,\n  ThreadListItemDelete,\n  ThreadListItemRoot,\n  ThreadListItems,\n  ThreadListItemTitle,\n  ThreadListItemTrigger,\n  ThreadListItemUnarchive,\n  ThreadListNew,\n  ThreadListRoot,\n};\n","import {\n  ThreadListEmpty,\n  ThreadListItemArchive,\n  ThreadListItemDelete,\n  ThreadListItemRoot,\n  ThreadListItems,\n  ThreadListItemTitle,\n  ThreadListItemTrigger,\n  ThreadListItemUnarchive,\n  ThreadListNew,\n  ThreadListRoot,\n} from \"./parts\";\n\nexport const ThreadListPrimitive = {\n  Root: ThreadListRoot,\n  New: ThreadListNew,\n  Items: ThreadListItems,\n  Empty: ThreadListEmpty,\n} as const;\n\nexport const ThreadListItemPrimitive = {\n  Root: ThreadListItemRoot,\n  Trigger: ThreadListItemTrigger,\n  Title: ThreadListItemTitle,\n  Archive: ThreadListItemArchive,\n  Unarchive: ThreadListItemUnarchive,\n  Delete: ThreadListItemDelete,\n} as const;\n\nexport type {\n  ThreadListController,\n  ThreadListItemContextValue,\n  ThreadListProviderProps,\n  ThreadListRecord,\n} from \"../contexts\";\nexport { ThreadListProvider, useThreadList, useThreadListItem } from \"../contexts\";\n"],"mappings":";;;;;;;;;;;;;AAAA;AAAA,EACE;AAAA,EAIA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAuEH;AAzDJ,IAAM,iBAAiB,WAAgD,SAASA,gBAC9E,EAAE,WAAW,GAAG,MAAM,GACtB,KACA;AACA,QAAM,aAAa,cAAc;AACjC,QAAM,UAAU,OAA8B,IAAI;AAClD,QAAM,eAAe,OAAiC,IAAI;AAC1D,QAAM,eAAe,OAAO,oBAAI,IAA4B,CAAC;AAC7D,QAAM,kBAAkB,OAAO,oBAAI,IAA+B,CAAC;AACnE,QAAM,eAAe,YAAY,CAAC,UAAkB,YAAmC;AACrF,QAAI,YAAY,KAAM,cAAa,QAAQ,OAAO,QAAQ;AAAA,QACrD,cAAa,QAAQ,IAAI,UAAU,OAAO;AAAA,EACjD,GAAG,CAAC,CAAC;AACL,QAAM,kBAAkB,YAAY,CAAC,UAAkB,YAAsC;AAC3F,QAAI,YAAY,KAAM,iBAAgB,QAAQ,OAAO,QAAQ;AAAA,QACxD,iBAAgB,QAAQ,IAAI,UAAU,OAAO;AAAA,EACpD,GAAG,CAAC,CAAC;AACL,QAAM,cAAc;AAAA,IAClB,OAAO;AAAA,MACL;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,IACA,CAAC,cAAc,eAAe;AAAA,EAChC;AACA,QAAM,cAAc;AAAA,IAClB,MACE,YAA4B,KAAK,CAAC,SAAS;AACzC,cAAQ,UAAU;AAAA,IACpB,CAAC;AAAA,IACH,CAAC,GAAG;AAAA,EACN;AAEA,QAAM,gBAAgB;AAAA,IACpB,CAAC,UAAyC;AACxC,kBAAY,KAAK;AACjB,UAAI,MAAM,oBAAoB,CAAC,CAAC,aAAa,WAAW,QAAQ,KAAK,EAAE,SAAS,MAAM,GAAG,GAAG;AAC1F;AAAA,MACF;AACA,YAAM,WAAW,uBAAuB,WAAW;AACnD,UAAI,SAAS,WAAW,GAAG;AACzB;AAAA,MACF;AACA,YAAM,eAAe;AACrB,YAAM,gBAAgB,SAAS;AAC/B,YAAM,eACJ,yBAAyB,oBAAoB,SAAS,QAAQ,aAAa,IAAI;AACjF,YAAM,YAAY,iBAAiB,MAAM,KAAK,cAAc,SAAS,MAAM;AAC3E,eAAS,SAAS,GAAG,MAAM;AAAA,IAC7B;AAAA,IACA,CAAC,WAAW,WAAW;AAAA,EACzB;AAEA,SACE,oBAAC,kCAA+B,OAAO,aACpC;AAAA,IACC;AAAA,IACA;AAAA,MACE,GAAG;AAAA,MACH,WAAW;AAAA,MACX,cAAc,WAAW,UAAU;AAAA,IACrC;AAAA,IACA;AAAA,EACF,GACF;AAEJ,CAAC;AAED,IAAM,gBAAgB;AAAA,EACpB,SAASC,eAAc,EAAE,SAAS,GAAG,MAAM,GAAG,KAAK;AACjD,UAAM,aAAa,cAAc;AACjC,UAAM,OAAO,0BAA0B;AACvC,UAAM,cAAc;AAAA,MAClB,MACE,YAA+B,KAAK,CAAC,SAAS;AAC5C,aAAK,aAAa,UAAU;AAAA,MAC9B,CAAC;AAAA,MACH,CAAC,KAAK,KAAK,YAAY;AAAA,IACzB;AAEA,UAAM,cAAc;AAAA,MAClB,CAAC,UAAyC;AACxC,kBAAU,KAAK;AACf,YAAI,MAAM,oBAAoB,MAAM,UAAU;AAC5C;AAAA,QACF;AACA,aAAK,WAAW,aAAa;AAAA,MAC/B;AAAA,MACA,CAAC,SAAS,MAAM,UAAU,UAAU;AAAA,IACtC;AAEA,WAAO;AAAA,MACL;AAAA,MACA;AAAA,QACE,GAAG;AAAA,QACH,UAAU,MAAM,YAAY;AAAA,QAC5B,SAAS;AAAA,QACT,MAAM,MAAM,QAAQ;AAAA,QACpB,cAAc,MAAM,WAAW,aAAa;AAAA,MAC9C;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAUA,IAAM,kBAAkB,WAAiD,SAASC,iBAChF,EAAE,WAAW,OAAO,UAAU,cAAc,OAAO,GAAG,MAAM,GAC5D,KACA;AACA,QAAM,aAAa,cAAc;AACjC,QAAM,UAAU,WAAW,QAAQ,OAAO,CAAC,WAAW,QAAQ,OAAO,QAAQ,MAAM,QAAQ;AAC3F,QAAM,QAAQ,QAAQ,WAAW;AACjC,MAAI,SAAS,CAAC,aAAa;AACzB,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,MACE,GAAG;AAAA,MACH,UAAU,QAAQ,IAAI,CAAC,WACrB;AAAA,QAAC;AAAA;AAAA,UAEC,OAAO,EAAE,QAAQ,QAAQ,OAAO,OAAO,WAAW,eAAe;AAAA,UAEhE,iBAAO,aAAa,aAAa,SAAS,MAAM,IAAK,YAAY,oBAAC,qBAAkB;AAAA;AAAA,QAHhF,OAAO;AAAA,MAId,CACD;AAAA,MACD,cAAc,QAAQ,UAAU;AAAA,IAClC;AAAA,IACA;AAAA,EACF;AACF,CAAC;AAMD,IAAM,kBAAkB,WAAiD,SAASC,iBAChF,EAAE,WAAW,OAAO,GAAG,MAAM,GAC7B,KACA;AACA,QAAM,aAAa,cAAc;AACjC,QAAM,oBAAoB,WAAW,QAAQ;AAAA,IAC3C,CAAC,WAAW,QAAQ,OAAO,QAAQ,MAAM;AAAA,EAC3C;AACA,MAAI,mBAAmB;AACrB,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,MACE,GAAG;AAAA,MACH,UAAU,MAAM,YAAY;AAAA,IAC9B;AAAA,IACA;AAAA,EACF;AACF,CAAC;AAED,IAAM,qBAAqB;AAAA,EACzB,SAASC,oBAAmB,OAAO,KAAK;AACtC,UAAM,OAAO,kBAAkB;AAC/B,UAAM,OAAO,0BAA0B;AACvC,UAAM,cAAc;AAAA,MAClB,MAAM,YAA4B,KAAK,CAAC,SAAS,KAAK,aAAa,KAAK,OAAO,IAAI,IAAI,CAAC;AAAA,MACxF,CAAC,KAAK,OAAO,IAAI,KAAK,IAAI;AAAA,IAC5B;AAEA,WAAO;AAAA,MACL;AAAA,MACA;AAAA,QACE,GAAG;AAAA,QACH,cAAc,KAAK,SAAS,WAAW;AAAA,MACzC;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,wBAAwB;AAAA,EAC5B,SAASC,uBAAsB,EAAE,SAAS,GAAG,MAAM,GAAG,KAAK;AACzD,UAAM,aAAa,cAAc;AACjC,UAAM,OAAO,kBAAkB;AAC/B,UAAM,OAAO,0BAA0B;AACvC,UAAM,WAAW,MAAM,YAAY;AACnC,UAAM,cAAc;AAAA,MAClB,MACE,YAA+B,KAAK,CAAC,SAAS,KAAK,gBAAgB,KAAK,OAAO,IAAI,IAAI,CAAC;AAAA,MAC1F,CAAC,KAAK,OAAO,IAAI,KAAK,IAAI;AAAA,IAC5B;AAEA,UAAM,cAAc;AAAA,MAClB,CAAC,UAAyC;AACxC,kBAAU,KAAK;AACf,YAAI,MAAM,oBAAoB,UAAU;AACtC;AAAA,QACF;AACA,aAAK,WAAW,aAAa,KAAK,OAAO,EAAE;AAAA,MAC7C;AAAA,MACA,CAAC,UAAU,KAAK,OAAO,IAAI,SAAS,UAAU;AAAA,IAChD;AAEA,WAAO;AAAA,MACL;AAAA,MACA;AAAA,QACE,GAAG;AAAA,QACH,UAAU,MAAM,YAAY,oBAAC,uBAAoB;AAAA,QACjD;AAAA,QACA,SAAS;AAAA,QACT,MAAM,MAAM,QAAQ;AAAA,QACpB,gBAAgB,KAAK,SAAS,SAAS;AAAA,QACvC,cAAc,WAAW,aAAa;AAAA,MACxC;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAMA,IAAM,sBAAsB;AAAA,EAC1B,SAASC,qBAAoB,EAAE,WAAW,YAAY,GAAG,MAAM,GAAG,KAAK;AACrE,UAAM,EAAE,OAAO,IAAI,kBAAkB;AAErC,WAAO;AAAA,MACL;AAAA,MACA;AAAA,QACE,GAAG;AAAA,QACH,UAAU,MAAM,YAAY,OAAO,SAAS;AAAA,MAC9C;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAGA,SAAS,2BAA2B,QAAiC,cAAsB;AACzF,SAAO,WAAwD,SAAS,qBACtE,EAAE,SAAS,GAAG,MAAM,GACpB,KACA;AACA,UAAM,aAAa,cAAc;AACjC,UAAM,EAAE,OAAO,IAAI,kBAAkB;AACrC,UAAM,OAAO,0BAA0B;AACvC,UAAM,gBAAgB,WAAW,MAAM;AACvC,UAAM,WAAW,MAAM,YAAY,kBAAkB;AAErD,UAAM,cAAc;AAAA,MAClB,CAAC,UAAyC;AACxC,kBAAU,KAAK;AACf,YAAI,MAAM,oBAAoB,UAAU;AACtC;AAAA,QACF;AACA,oCAA4B,OAAO,IAAI,IAAI;AAC3C,aAAK,gBAAgB,OAAO,EAAE;AAAA,MAChC;AAAA,MACA,CAAC,eAAe,UAAU,SAAS,MAAM,OAAO,EAAE;AAAA,IACpD;AAEA,WAAO;AAAA,MACL;AAAA,MACA;AAAA,QACE,GAAG;AAAA,QACH,UAAU,MAAM,YAAY;AAAA,QAC5B;AAAA,QACA,SAAS;AAAA,QACT,MAAM,MAAM,QAAQ;AAAA,QACpB,cAAc,WAAW,aAAa;AAAA,MACxC;AAAA,MACA;AAAA,IACF;AAAA,EACF,CAAC;AACH;AAEA,IAAM,wBAAwB,2BAA2B,iBAAiB,SAAS;AACnF,IAAM,0BAA0B,2BAA2B,mBAAmB,WAAW;AACzF,IAAM,uBAAuB,2BAA2B,gBAAgB,QAAQ;AAEhF,SAAS,oBAA+B;AACtC,SACE,oBAAC,sBACC,8BAAC,yBAAsB,GACzB;AAEJ;AAEA,SAAS,iBAAiB,KAAa,cAAsB,cAA8B;AACzF,MAAI,QAAQ,QAAQ;AAClB,WAAO;AAAA,EACT;AACA,MAAI,QAAQ,OAAO;AACjB,WAAO,eAAe;AAAA,EACxB;AACA,MAAI,iBAAiB,IAAI;AACvB,WAAO,QAAQ,YAAY,eAAe,IAAI;AAAA,EAChD;AACA,MAAI,QAAQ,WAAW;AACrB,YAAQ,eAAe,IAAI,gBAAgB;AAAA,EAC7C;AACA,UAAQ,eAAe,KAAK;AAC9B;AAEA,SAAS,4BACP,UACA,aACM;AACN,QAAM,OAAO,YAAY,QAAQ;AACjC,QAAM,cAAc,YAAY,aAAa,QAAQ,IAAI,QAAQ;AACjE,MACE,SAAS,QACT,gBAAgB,UAChB,CAAC,YAAY,SAAS,KAAK,cAAc,aAAa,GACtD;AACA;AAAA,EACF;AAEA,QAAM,WAAW,uBAAuB,WAAW;AACnD,QAAM,iBAAiB,YAAY,gBAAgB,QAAQ,IAAI,QAAQ;AACvE,QAAM,eAAe,mBAAmB,SAAY,KAAK,SAAS,QAAQ,cAAc;AACxF,QAAM,oBAAoB,SAAS,OAAO,CAAC,YAAY,CAAC,YAAY,SAAS,OAAO,CAAC;AACrF,QAAM,WAAW,YAAY,aAAa;AAC1C,QAAM,cACJ,kBAAkB,KAAK,IAAI,KAAK,IAAI,cAAc,CAAC,GAAG,kBAAkB,SAAS,CAAC,CAAC,KACnF,YACA;AAEF,MAAI,gBAAgB,QAAQ,CAAC,KAAK,aAAa,UAAU,GAAG;AAC1D,SAAK,WAAW;AAAA,EAClB;AACA,cAAY,MAAM;AACpB;AAEA,SAAS,uBACP,aACqB;AACrB,SAAO,CAAC,GAAG,YAAY,gBAAgB,QAAQ,OAAO,CAAC,EACpD,OAAO,CAAC,YAAY,QAAQ,eAAe,CAAC,QAAQ,QAAQ,EAC5D;AAAA,IAAK,CAAC,MAAM,UACX,KAAK,wBAAwB,KAAK,IAAI,KAAK,8BAA8B,KAAK;AAAA,EAChF;AACJ;;;AC3WO,IAAM,sBAAsB;AAAA,EACjC,MAAM;AAAA,EACN,KAAK;AAAA,EACL,OAAO;AAAA,EACP,OAAO;AACT;AAEO,IAAM,0BAA0B;AAAA,EACrC,MAAM;AAAA,EACN,SAAS;AAAA,EACT,OAAO;AAAA,EACP,SAAS;AAAA,EACT,WAAW;AAAA,EACX,QAAQ;AACV;","names":["ThreadListRoot","ThreadListNew","ThreadListItems","ThreadListEmpty","ThreadListItemRoot","ThreadListItemTrigger","ThreadListItemTitle"]}

package/dist/chunk-6XOJFLX4.js

@@ -0,0 +1,22 @@
+// src/message/elements.ts
+var messageIds = /* @__PURE__ */ new WeakMap();
+function registerMessageElement(element, messageId) {
+  messageIds.set(element, messageId);
+}
+function messageElementFromNode(node) {
+  let element = node?.nodeType === Node.ELEMENT_NODE ? node : node?.parentElement ?? void 0;
+  while (element !== void 0 && element !== null) {
+    const messageId = messageIds.get(element);
+    if (messageId !== void 0) {
+      return { element, messageId };
+    }
+    element = element.parentElement ?? void 0;
+  }
+  return void 0;
+}
+
+export {
+  registerMessageElement,
+  messageElementFromNode
+};
+//# sourceMappingURL=chunk-6XOJFLX4.js.map

package/dist/chunk-6XOJFLX4.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/message/elements.ts"],"sourcesContent":["const messageIds = new WeakMap<HTMLElement, string>();\n\nexport function registerMessageElement(element: HTMLElement, messageId: string): void {\n  messageIds.set(element, messageId);\n}\n\nexport function messageElementFromNode(\n  node: Node | null,\n): { element: HTMLElement; messageId: string } | undefined {\n  let element =\n    node?.nodeType === Node.ELEMENT_NODE\n      ? (node as HTMLElement)\n      : (node?.parentElement ?? undefined);\n\n  while (element !== undefined && element !== null) {\n    const messageId = messageIds.get(element);\n    if (messageId !== undefined) {\n      return { element, messageId };\n    }\n    element = element.parentElement ?? undefined;\n  }\n  return undefined;\n}\n"],"mappings":";AAAA,IAAM,aAAa,oBAAI,QAA6B;AAE7C,SAAS,uBAAuB,SAAsB,WAAyB;AACpF,aAAW,IAAI,SAAS,SAAS;AACnC;AAEO,SAAS,uBACd,MACyD;AACzD,MAAI,UACF,MAAM,aAAa,KAAK,eACnB,OACA,MAAM,iBAAiB;AAE9B,SAAO,YAAY,UAAa,YAAY,MAAM;AAChD,UAAM,YAAY,WAAW,IAAI,OAAO;AACxC,QAAI,cAAc,QAAW;AAC3B,aAAO,EAAE,SAAS,UAAU;AAAA,IAC9B;AACA,cAAU,QAAQ,iBAAiB;AAAA,EACrC;AACA,SAAO;AACT;","names":[]}

package/dist/chunk-CKA4CE26.js

@@ -0,0 +1,25 @@
+// src/format/message.ts
+function messageText(message) {
+  return message.parts.filter((part) => part.type === "text").map((part) => part.text).join("");
+}
+
+// src/format/value.ts
+function stringifyValue(value) {
+  if (value === void 0) {
+    return "";
+  }
+  if (typeof value === "string") {
+    return value;
+  }
+  try {
+    return JSON.stringify(value, null, 2);
+  } catch {
+    return String(value);
+  }
+}
+
+export {
+  messageText,
+  stringifyValue
+};
+//# sourceMappingURL=chunk-CKA4CE26.js.map

package/dist/chunk-CKA4CE26.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/format/message.ts","../src/format/value.ts"],"sourcesContent":["import type { UIMessage, UIMessagePart } from \"@anvia/client\";\n\nexport function messageText(message: UIMessage): string {\n  return message.parts\n    .filter((part): part is Extract<UIMessagePart, { type: \"text\" }> => part.type === \"text\")\n    .map((part) => part.text)\n    .join(\"\");\n}\n","export function stringifyValue(value: unknown): string {\n  if (value === undefined) {\n    return \"\";\n  }\n  if (typeof value === \"string\") {\n    return value;\n  }\n  try {\n    return JSON.stringify(value, null, 2);\n  } catch {\n    return String(value);\n  }\n}\n"],"mappings":";AAEO,SAAS,YAAY,SAA4B;AACtD,SAAO,QAAQ,MACZ,OAAO,CAAC,SAA2D,KAAK,SAAS,MAAM,EACvF,IAAI,CAAC,SAAS,KAAK,IAAI,EACvB,KAAK,EAAE;AACZ;;;ACPO,SAAS,eAAe,OAAwB;AACrD,MAAI,UAAU,QAAW;AACvB,WAAO;AAAA,EACT;AACA,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO;AAAA,EACT;AACA,MAAI;AACF,WAAO,KAAK,UAAU,OAAO,MAAM,CAAC;AAAA,EACtC,QAAQ;AACN,WAAO,OAAO,KAAK;AAAA,EACrB;AACF;","names":[]}

package/dist/{chunk-JWZDNM5D.js → chunk-CRXNFZRW.js}

@@ -1,10 +1,18 @@
+import {
+  registerMessageElement
+} from "./chunk-6XOJFLX4.js";
 import {
   StreamMarkdown
-} from "./chunk-ULGJLLZE.js";
+} from "./chunk-F6AKBUO3.js";
+import {
+  AttachmentPrimitive
+} from "./chunk-MISX4WUT.js";
 import {
-  Attachment
-} from "./chunk-SBYUKODG.js";
+  messageText,
+  stringifyValue
+} from "./chunk-CKA4CE26.js";
 import {
+  composeRefs,
   renderPrimitive
 } from "./chunk-T3ZFZ3NP.js";
 import {
@@ -14,32 +22,10 @@ import {
   useMessage,
   useMessagePart,
   useOptionalMessagePart
-} from "./chunk-43U2XQP2.js";
+} from "./chunk-SOG2GVE4.js";
 
 // src/message/actions.tsx
 import { forwardRef, useCallback, useEffect, useRef, useState } from "react";
-
-// src/format/message.ts
-function messageText(message) {
-  return message.parts.filter((part) => part.type === "text").map((part) => part.text).join("");
-}
-
-// src/format/value.ts
-function stringifyValue(value) {
-  if (value === void 0) {
-    return "";
-  }
-  if (typeof value === "string") {
-    return value;
-  }
-  try {
-    return JSON.stringify(value, null, 2);
-  } catch {
-    return String(value);
-  }
-}
-
-// src/message/actions.tsx
 import { Fragment, jsx, jsxs } from "react/jsx-runtime";
 var MessageActions = forwardRef(
   function MessageActions2(props, ref) {
@@ -50,8 +36,7 @@ var MessageActions = forwardRef(
         children: props.children ?? /* @__PURE__ */ jsxs(Fragment, { children: [
           /* @__PURE__ */ jsx(MessageCopy, {}),
           /* @__PURE__ */ jsx(MessageRegenerate, {})
-        ] }),
-        "data-anvia-message-actions": ""
+        ] })
       },
       ref
     );
@@ -97,7 +82,6 @@ var MessageCopy = forwardRef(function MessageCopy2({ onClick, ...props }, ref) {
       disabled,
       onClick: handleClick,
       type: props.type ?? "button",
-      "data-anvia-copy": "",
       "data-state": copyState
     },
     ref
@@ -108,7 +92,7 @@ var MessageRegenerate = forwardRef(
     const chat = useChatContext();
     const { message } = useMessage();
     const latestAssistantMessage = [...chat.messages].reverse().find((item) => item.role === "assistant");
-    const disabled = props.disabled ?? (chat.status === "streaming" || message.role !== "assistant" || latestAssistantMessage?.id !== message.id);
+    const disabled = props.disabled ?? (chat.status === "submitted" || chat.status === "streaming" || message.role !== "assistant" || latestAssistantMessage?.id !== message.id);
     const handleClick = useCallback(
       (event) => {
         onClick?.(event);
@@ -127,7 +111,6 @@ var MessageRegenerate = forwardRef(
         disabled,
         onClick: handleClick,
         type: props.type ?? "button",
-        "data-anvia-regenerate": "",
         "data-state": disabled ? "disabled" : "idle"
       },
       ref
@@ -139,17 +122,7 @@ var MessageRegenerate = forwardRef(
 import { forwardRef as forwardRef2 } from "react";
 import { jsx as jsx2 } from "react/jsx-runtime";
 var MessageEntity = forwardRef2(function MessageEntity2({ children, entity, ...props }, ref) {
-  return /* @__PURE__ */ jsx2(
-    "span",
-    {
-      ...props,
-      "data-anvia-message-entity": "",
-      "data-entity-id": entity.id,
-      "data-trigger-id": entity.triggerId,
-      ref,
-      children: children ?? entity.text
-    }
-  );
+  return /* @__PURE__ */ jsx2("span", { ...props, "data-role": "entity", ref, children: children ?? entity.text });
 });
 
 // src/message/parts.tsx
@@ -164,6 +137,55 @@ import {
 } from "react";
 import remarkGfm from "remark-gfm";
 
+// src/strict-json.ts
+function isComposerEntityData(value) {
+  return isComposerEntityDataWithAncestors(value, /* @__PURE__ */ new WeakSet());
+}
+function isComposerEntityDataWithAncestors(value, ancestors) {
+  if (value === null || typeof value === "string" || typeof value === "boolean") return true;
+  if (typeof value === "number") return Number.isFinite(value);
+  if (typeof value !== "object" || ancestors.has(value)) return false;
+  ancestors.add(value);
+  try {
+    if (Array.isArray(value)) {
+      if (Object.getPrototypeOf(value) !== Array.prototype || !hasOnlyDenseArrayProperties(value)) {
+        return false;
+      }
+      for (let index = 0; index < value.length; index += 1) {
+        if (!isComposerEntityDataWithAncestors(value[index], ancestors)) return false;
+      }
+      return true;
+    }
+    const prototype = Object.getPrototypeOf(value);
+    if (prototype !== Object.prototype && prototype !== null) return false;
+    for (const key of Reflect.ownKeys(value)) {
+      if (typeof key !== "string") return false;
+      const descriptor = Object.getOwnPropertyDescriptor(value, key);
+      if (descriptor === void 0 || !descriptor.enumerable || !("value" in descriptor) || !isComposerEntityDataWithAncestors(descriptor.value, ancestors)) {
+        return false;
+      }
+    }
+    return true;
+  } catch {
+    return false;
+  } finally {
+    ancestors.delete(value);
+  }
+}
+function hasOnlyDenseArrayProperties(value) {
+  const keys = Reflect.ownKeys(value);
+  if (keys.length !== value.length + 1 || keys.at(-1) !== "length") return false;
+  for (let index = 0; index < value.length; index += 1) {
+    const key = String(index);
+    if (keys[index] !== key) return false;
+    const descriptor = Object.getOwnPropertyDescriptor(value, key);
+    if (descriptor === void 0 || !descriptor.enumerable || !("value" in descriptor)) {
+      return false;
+    }
+  }
+  return true;
+}
+
 // src/entities.ts
 function messageTextLayout(parts) {
   const textParts = parts.filter(
@@ -184,9 +206,10 @@ function messageTextLayout(parts) {
 }
 function validComposerEntities(text, metadata) {
   const candidates = composerEntityCandidates(metadata);
-  const individuallyValid = candidates.flatMap(
-    (entity, index) => isValidComposerEntity(entity, text) ? [{ entity, index }] : []
-  );
+  const individuallyValid = candidates.flatMap((candidate, index) => {
+    const entity = parseComposerEntity(candidate, text);
+    return entity === void 0 ? [] : [{ entity, index }];
+  });
   individuallyValid.sort(
     (left, right) => left.entity.range.from - right.entity.range.from || left.entity.range.to - right.entity.range.to || left.index - right.index
   );
@@ -249,15 +272,28 @@ function composerEntityCandidates(metadata) {
   }
   return Array.isArray(metadata.composer.entities) ? metadata.composer.entities : [];
 }
-function isValidComposerEntity(value, text) {
+function parseComposerEntity(value, text) {
   if (!isRecord(value) || !isRecord(value.range)) {
-    return false;
+    return void 0;
   }
   if (!isNonEmptyString(value.id) || !isNonEmptyString(value.triggerId) || !isNonEmptyString(value.trigger) || !isNonEmptyString(value.label) || !isNonEmptyString(value.text)) {
-    return false;
+    return void 0;
   }
   const { from, to } = value.range;
-  return typeof from === "number" && typeof to === "number" && Number.isInteger(from) && Number.isInteger(to) && from >= 0 && to > from && to <= text.length && text.slice(from, to) === value.text;
+  if (typeof from !== "number" || typeof to !== "number" || !Number.isInteger(from) || !Number.isInteger(to) || from < 0 || to <= from || to > text.length || text.slice(from, to) !== value.text) {
+    return void 0;
+  }
+  if (value.data !== void 0 && !isComposerEntityData(value.data)) return void 0;
+  const entity = {
+    id: value.id,
+    triggerId: value.triggerId,
+    trigger: value.trigger,
+    label: value.label,
+    text: value.text,
+    range: { from, to }
+  };
+  if (value.data !== void 0) entity.data = value.data;
+  return entity;
 }
 function isNonEmptyString(value) {
   return typeof value === "string" && value.length > 0;
@@ -295,8 +331,7 @@ function messageEntityRehypeOptions() {
           type: "element",
           tagName: "span",
           properties: {
-            dataAnviaMessageEntity: "",
-            dataAnviaEntityIndex: entityIndex === void 0 ? "" : String(entityIndex)
+            dataEntityIndex: entityIndex === void 0 ? "" : String(entityIndex)
           },
           children: [{ type: "text", value: node.value ?? "" }]
         };
@@ -523,27 +558,29 @@ var MessageParts = forwardRef3(function MessageParts2({ children, filter, stream
     {
       ...props,
       children: renderMessagePartNodes(parts, children),
-      "data-anvia-message-parts": ""
+      "data-role": "parts"
     },
     ref
   );
 });
 var SmoothedMessageParts = forwardRef3(
   function SmoothedMessageParts2({ children, parts, stream, ...props }, ref) {
-    const smooth = useSmoothStreamItems(parts, {
+    const streamOptions = {
       adapter: messagePartStreamAdapter,
       isStreaming: stream.isStreaming,
-      resetKey: stream.resetKey,
-      ...stream.flushImmediately === void 0 ? {} : { flushImmediately: stream.flushImmediately }
-    });
+      resetKey: stream.resetKey
+    };
+    if (stream.flushImmediately !== void 0) {
+      Object.assign(streamOptions, { flushImmediately: stream.flushImmediately });
+    }
+    const smooth = useSmoothStreamItems(parts, streamOptions);
     return renderPrimitive(
       "div",
       {
         ...props,
         children: renderMessagePartNodes(smooth.items, children, smooth.liveItemKey),
-        "data-anvia-message-parts": "",
-        "data-draining": smooth.isDraining ? "" : void 0,
-        "data-streaming": stream.isStreaming ? "" : void 0
+        "data-role": "parts",
+        "data-state": smooth.isDraining ? "draining" : stream.isStreaming ? "streaming" : "idle"
       },
       ref
     );
@@ -569,8 +606,7 @@ var MessagePart = forwardRef3(function MessagePart2({ children, ...props }, ref)
     {
       ...props,
       children: renderedChildren,
-      "data-anvia-part": "",
-      "data-part": part.type
+      "data-role": part.type
     },
     ref
   );
@@ -594,19 +630,21 @@ var MessageText = forwardRef3(function MessageText2(props, ref) {
 var MessageTextContent = forwardRef3(
   function MessageTextContent2({ content, partId, stream, streamControlled, ...props }, ref) {
     const ownsStream = stream !== void 0 && !streamControlled && props.children === void 0;
-    const smooth = useSmoothStreamText(content, {
+    const streamOptions = {
       isStreaming: ownsStream ? stream.isStreaming : false,
-      resetKey: ownsStream ? stream.resetKey : partId,
-      ...ownsStream && stream.flushImmediately !== void 0 ? { flushImmediately: stream.flushImmediately } : {}
-    });
+      resetKey: ownsStream ? stream.resetKey : partId
+    };
+    if (ownsStream && stream.flushImmediately !== void 0) {
+      Object.assign(streamOptions, { flushImmediately: stream.flushImmediately });
+    }
+    const smooth = useSmoothStreamText(content, streamOptions);
     return renderPrimitive(
       "span",
       {
         ...props,
         children: props.children ?? (ownsStream ? smooth.text : content),
-        "data-anvia-text": "",
-        "data-draining": ownsStream && smooth.isDraining ? "" : void 0,
-        "data-streaming": ownsStream && stream.isStreaming ? "" : void 0
+        "data-role": "text",
+        "data-state": ownsStream ? smooth.isDraining ? "draining" : stream.isStreaming ? "streaming" : "idle" : void 0
       },
       ref
     );
@@ -656,11 +694,14 @@ var MessageMarkdownContent = forwardRef3(
     ...props
   }, ref) {
     const ownsStream = stream !== void 0 && !streamControlled;
-    const smooth = useSmoothStreamText(markdown, {
+    const streamOptions = {
       isStreaming: ownsStream ? stream.isStreaming : false,
-      resetKey: ownsStream ? stream.resetKey : resetKey,
-      ...ownsStream && stream.flushImmediately !== void 0 ? { flushImmediately: stream.flushImmediately } : {}
-    });
+      resetKey: ownsStream ? stream.resetKey : resetKey
+    };
+    if (ownsStream && stream.flushImmediately !== void 0) {
+      Object.assign(streamOptions, { flushImmediately: stream.flushImmediately });
+    }
+    const smooth = useSmoothStreamText(markdown, streamOptions);
     const renderedMarkdown = ownsStream ? smooth.text : markdown;
     const renderedEntities = useMemo(
       () => entities.length === 0 ? entities : validComposerEntities(renderedMarkdown, {
@@ -686,9 +727,7 @@ var MessageMarkdownContent = forwardRef3(
         ...props,
         components: renderedComponents,
         content: renderedMarkdown,
-        "data-anvia-markdown": "",
-        "data-draining": ownsStream && smooth.isDraining ? "" : void 0,
-        "data-streaming": ownsStream && stream.isStreaming ? "" : void 0,
+        "data-state": ownsStream && smooth.isDraining ? "draining" : ownsStream && stream.isStreaming ? "streaming" : "idle",
         live,
         ref,
         remarkPlugins: renderedRemarkPlugins,
@@ -705,20 +744,18 @@ function markdownComponents(components, entities, renderEntity) {
     ...components,
     span(spanProps) {
       const internalProps = spanProps;
-      const entityIndexValue = internalProps["data-anvia-entity-index"];
+      const entityIndexValue = internalProps["data-entity-index"];
       const entityIndex = typeof entityIndexValue === "string" && /^\d+$/.test(entityIndexValue) ? Number(entityIndexValue) : void 0;
       const entity = entityIndex === void 0 ? void 0 : entities[entityIndex];
       const consumerProps = { ...internalProps };
-      delete consumerProps["data-anvia-entity-index"];
+      delete consumerProps["data-entity-index"];
       if (entity !== void 0) {
         if (renderEntity !== void 0) {
           return renderEntity(entity);
         }
         const entityProps = {
           ...consumerProps,
-          "data-anvia-message-entity": "",
-          "data-entity-id": entity.id,
-          "data-trigger-id": entity.triggerId
+          "data-role": "entity"
         };
         if (consumerSpan !== void 0) {
           return createElement(consumerSpan, entityProps);
@@ -740,9 +777,8 @@ var MessageCodeBlock = forwardRef3(
       "pre",
       {
         ...props,
-        children: children ?? /* @__PURE__ */ jsx3("code", { "data-anvia-code": "", "data-language": language, children: code ?? "" }),
-        "data-anvia-code-block": "",
-        "data-language": language
+        "data-role": "code-block",
+        children: children ?? /* @__PURE__ */ jsx3("code", { className: language === void 0 ? void 0 : `language-${language}`, children: code ?? "" })
       },
       ref
     );
@@ -761,8 +797,7 @@ var MessageReasoning = forwardRef3(
         children: props.children ?? /* @__PURE__ */ jsxs2(Fragment2, { children: [
           /* @__PURE__ */ jsx3("summary", { children: "Reasoning" }),
           /* @__PURE__ */ jsx3("pre", { children: part.text })
-        ] }),
-        "data-anvia-reasoning": ""
+        ] })
       },
       ref
     );
@@ -779,7 +814,7 @@ var MessageTool = forwardRef3(function MessageTool2({ children, renderWhen = "al
     {
       ...props,
       children: renderedChildren,
-      "data-anvia-tool": "",
+      "data-role": "tool",
       "data-state": part.state
     },
     ref
@@ -795,8 +830,7 @@ var MessageToolName = forwardRef3(
       "span",
       {
         ...props,
-        children: props.children ?? part.toolName,
-        "data-anvia-tool-name": ""
+        children: props.children ?? part.toolName
       },
       ref
     );
@@ -813,7 +847,6 @@ var MessageToolStatus = forwardRef3(
       {
         ...props,
         children: props.children ?? toolStateLabel(part.state),
-        "data-anvia-tool-status": "",
         "data-state": part.state
       },
       ref
@@ -823,15 +856,14 @@ var MessageToolStatus = forwardRef3(
 var MessageToolInput = forwardRef3(
   function MessageToolInput2(props, ref) {
     const { part } = useMessagePart();
-    if (part.type !== "tool" || part.input === void 0) {
+    if (part.type !== "tool") {
       return null;
     }
     return renderPrimitive(
       "pre",
       {
         ...props,
-        children: props.children ?? stringifyValue(part.input),
-        "data-anvia-tool-input": ""
+        children: props.children ?? stringifyValue(part.input)
       },
       ref
     );
@@ -840,15 +872,14 @@ var MessageToolInput = forwardRef3(
 var MessageToolOutput = forwardRef3(
   function MessageToolOutput2(props, ref) {
     const { part } = useMessagePart();
-    if (part.type !== "tool" || part.output === void 0) {
+    if (part.type !== "tool" || part.state !== "output-available") {
       return null;
     }
     return renderPrimitive(
       "pre",
       {
         ...props,
-        children: props.children ?? stringifyValue(part.output),
-        "data-anvia-tool-output": ""
+        children: props.children ?? stringifyValue(part.output)
       },
       ref
     );
@@ -857,7 +888,7 @@ var MessageToolOutput = forwardRef3(
 var MessageToolError = forwardRef3(
   function MessageToolError2(props, ref) {
     const { part } = useMessagePart();
-    if (part.type !== "tool" || part.error === void 0) {
+    if (part.type !== "tool" || part.state !== "error") {
       return null;
     }
     return renderPrimitive(
@@ -865,7 +896,6 @@ var MessageToolError = forwardRef3(
       {
         ...props,
         children: props.children ?? part.error.message,
-        "data-anvia-tool-error": "",
         role: props.role ?? "alert"
       },
       ref
@@ -878,13 +908,12 @@ var MessageAttachment = forwardRef3(
     if (part.type !== "attachment") {
       return null;
     }
-    const renderedChildren = typeof children === "function" ? children(part.attachment) : children ?? /* @__PURE__ */ jsx3(Attachment.Root, {});
+    const renderedChildren = typeof children === "function" ? children(part.attachment) : children ?? /* @__PURE__ */ jsx3(AttachmentPrimitive.Root, {});
     return /* @__PURE__ */ jsx3(InternalAttachmentProvider, { value: { attachment: part.attachment }, children: renderPrimitive(
       "div",
       {
         ...props,
-        children: renderedChildren,
-        "data-anvia-message-attachment": ""
+        children: renderedChildren
       },
       ref
     ) });
@@ -900,9 +929,7 @@ var MessageData = forwardRef3(
       "pre",
       {
         ...props,
-        children: props.children ?? stringifyValue(part.data),
-        "data-anvia-data": "",
-        "data-name": part.name
+        children: props.children ?? stringifyValue(part.data)
       },
       ref
     );
@@ -919,7 +946,6 @@ var MessageError = forwardRef3(
       {
         ...props,
         children: props.children ?? part.error.message,
-        "data-anvia-error": "",
         role: props.role ?? "alert"
       },
       ref
@@ -959,18 +985,7 @@ function defaultCodeComponent({
   node: _node,
   ...props
 }) {
-  const language = /language-(\S+)/.exec(className ?? "")?.[1];
-  return /* @__PURE__ */ jsx3(
-    "code",
-    {
-      className,
-      "data-anvia-code": "",
-      "data-anvia-inline-code": language === void 0 ? "" : void 0,
-      "data-language": language,
-      ...props,
-      children
-    }
-  );
+  return /* @__PURE__ */ jsx3("code", { className, ...props, children });
 }
 function defaultPreComponent({
   children,
@@ -999,20 +1014,25 @@ function toolStateLabel(state) {
 }
 
 // src/message/root.tsx
-import { forwardRef as forwardRef4 } from "react";
+import { forwardRef as forwardRef4, useCallback as useCallback2, useMemo as useMemo2 } from "react";
 var MessageRoot = forwardRef4(function MessageRoot2({ children, ...props }, ref) {
   const { message } = useMessage();
   const renderedChildren = typeof children === "function" ? children(message) : children;
+  const registerElement = useCallback2(
+    (element) => {
+      if (element !== null) registerMessageElement(element, message.id);
+    },
+    [message.id]
+  );
+  const composedRef = useMemo2(() => composeRefs(ref, registerElement), [ref, registerElement]);
   return renderPrimitive(
     "article",
     {
       ...props,
       children: renderedChildren,
-      "data-anvia-message": "",
-      "data-anvia-message-id": message.id,
       "data-role": message.role
     },
-    ref
+    composedRef
   );
 });
 var MessageContent = forwardRef4(
@@ -1020,8 +1040,7 @@ var MessageContent = forwardRef4(
     return renderPrimitive(
       "div",
       {
-        ...props,
-        "data-anvia-message-content": ""
+        ...props
       },
       ref
     );
@@ -1029,7 +1048,7 @@ var MessageContent = forwardRef4(
 );
 
 // src/message/index.ts
-var Message = {
+var MessagePrimitive = {
   Root: MessageRoot,
   Content: MessageContent,
   Parts: MessageParts,
@@ -1054,7 +1073,8 @@ var Message = {
 };
 
 export {
+  isComposerEntityData,
   shiftComposerEntityRanges,
-  Message
+  MessagePrimitive
 };
-//# sourceMappingURL=chunk-JWZDNM5D.js.map
+//# sourceMappingURL=chunk-CRXNFZRW.js.map