@liveblocks/react-ui 2.25.0-aiprivatebeta8 → 3.0.0

package/dist/primitives/AiMessage/index.cjs

@@ -0,0 +1,77 @@
+'use strict';
+
+var jsxRuntime = require('react/jsx-runtime');
+var reactSlot = require('@radix-ui/react-slot');
+var react = require('react');
+var ErrorBoundary = require('../../utils/ErrorBoundary.cjs');
+var Markdown = require('../Markdown.cjs');
+var toolInvocation = require('./tool-invocation.cjs');
+
+const AI_MESSAGE_CONTENT_NAME = "AiMessageContent";
+const defaultMessageContentComponents = {
+  TextPart: ({ part }) => {
+    return /* @__PURE__ */ jsxRuntime.jsx(Markdown.Markdown, {
+      content: part.text
+    });
+  },
+  ReasoningPart: ({ part }) => {
+    return /* @__PURE__ */ jsxRuntime.jsx(Markdown.Markdown, {
+      content: part.text
+    });
+  },
+  ToolInvocationPart: ({ part, message }) => {
+    return /* @__PURE__ */ jsxRuntime.jsx(ErrorBoundary.ErrorBoundary, {
+      fallback: null,
+      children: /* @__PURE__ */ jsxRuntime.jsx(toolInvocation.AiMessageToolInvocation, {
+        part,
+        message
+      })
+    });
+  }
+};
+const AiMessageContent = react.forwardRef(
+  ({ message, components, asChild, ...props }, forwardedRef) => {
+    const Component = asChild ? reactSlot.Slot : "div";
+    const { TextPart, ReasoningPart, ToolInvocationPart } = react.useMemo(
+      () => ({ ...defaultMessageContentComponents, ...components }),
+      [components]
+    );
+    const content = message.content ?? message.contentSoFar;
+    const numParts = content.length;
+    const isGenerating = message.role === "assistant" && message.status === "generating";
+    return /* @__PURE__ */ jsxRuntime.jsx(Component, {
+      ...props,
+      ref: forwardedRef,
+      children: content.map((part, index) => {
+        const isStreaming = isGenerating && index === numParts - 1;
+        const extra = { index, isStreaming };
+        switch (part.type) {
+          case "text":
+            return /* @__PURE__ */ jsxRuntime.jsx(TextPart, {
+              part,
+              ...extra
+            }, index);
+          case "reasoning":
+            return /* @__PURE__ */ jsxRuntime.jsx(ReasoningPart, {
+              part,
+              ...extra
+            }, index);
+          case "tool-invocation":
+            return /* @__PURE__ */ jsxRuntime.jsx(ToolInvocationPart, {
+              part,
+              ...extra,
+              message
+            }, index);
+          default:
+            return null;
+        }
+      })
+    });
+  }
+);
+if (process.env.NODE_ENV !== "production") {
+  AiMessageContent.displayName = AI_MESSAGE_CONTENT_NAME;
+}
+
+exports.Content = AiMessageContent;
+//# sourceMappingURL=index.cjs.map

package/dist/primitives/AiMessage/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","sources":["../../../src/primitives/AiMessage/index.tsx"],"sourcesContent":["import { Slot } from \"@radix-ui/react-slot\";\nimport { forwardRef, useMemo } from \"react\";\n\nimport { ErrorBoundary } from \"../../utils/ErrorBoundary\";\nimport { Markdown } from \"../Markdown\";\nimport { AiMessageToolInvocation } from \"./tool-invocation\";\nimport type {\n  AiMessageContentComponents,\n  AiMessageContentProps,\n} from \"./types\";\n\nconst AI_MESSAGE_CONTENT_NAME = \"AiMessageContent\";\n\nconst defaultMessageContentComponents: AiMessageContentComponents = {\n  TextPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  ReasoningPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  ToolInvocationPart: ({ part, message }) => {\n    return (\n      <ErrorBoundary fallback={null}>\n        <AiMessageToolInvocation part={part} message={message} />\n      </ErrorBoundary>\n    );\n  },\n};\n\n/**\n * --------------------------------------------------------------------------\n * @private The API for this component is not yet stable.\n * --------------------------------------------------------------------------\n *\n * Primitive to help display an user or assistant message’s content, which is\n * an array of parts.\n *\n * @example\n * <AiMessage.Content message={message} components={{ TextPart }} />\n */\nconst AiMessageContent = forwardRef<HTMLDivElement, AiMessageContentProps>(\n  ({ message, components, asChild, ...props }, forwardedRef) => {\n    const Component = asChild ? Slot : \"div\";\n    const { TextPart, ReasoningPart, ToolInvocationPart } = useMemo(\n      () => ({ ...defaultMessageContentComponents, ...components }),\n      [components]\n    );\n\n    const content = message.content ?? message.contentSoFar;\n    const numParts = content.length;\n    const isGenerating =\n      message.role === \"assistant\" && message.status === \"generating\";\n    return (\n      <Component {...props} ref={forwardedRef}>\n        {content.map((part, index) => {\n          // A part is considered to be still \"streaming in\" if it's the last\n          // part in the content array, and the message is in \"generating\"\n          // state.\n          const isStreaming = isGenerating && index === numParts - 1;\n          const extra = { index, isStreaming };\n          switch (part.type) {\n            case \"text\":\n              return <TextPart key={index} part={part} {...extra} />;\n            case \"reasoning\":\n              return <ReasoningPart key={index} part={part} {...extra} />;\n            case \"tool-invocation\":\n              return (\n                <ToolInvocationPart\n                  key={index}\n                  part={part}\n                  {...extra}\n                  message={message}\n                />\n              );\n            default:\n              return null;\n          }\n        })}\n      </Component>\n    );\n  }\n);\n\nif (process.env.NODE_ENV !== \"production\") {\n  AiMessageContent.displayName = AI_MESSAGE_CONTENT_NAME;\n}\n\n// NOTE: Every export from this file will be available publicly as AiMessage.*\nexport { AiMessageContent as Content };\n"],"names":["jsx","Markdown","ErrorBoundary","AiMessageToolInvocation","forwardRef","Slot","useMemo"],"mappings":";;;;;;;;;AAWA,MAAM,uBAA0B,GAAA,kBAAA,CAAA;AAEhC,MAAM,+BAA8D,GAAA;AAAA,EAClE,QAAU,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AACtB,IAAA,uBAAQA,cAAA,CAAAC,iBAAA,EAAA;AAAA,MAAS,SAAS,IAAK,CAAA,IAAA;AAAA,KAAM,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,aAAe,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AAC3B,IAAA,uBAAQD,cAAA,CAAAC,iBAAA,EAAA;AAAA,MAAS,SAAS,IAAK,CAAA,IAAA;AAAA,KAAM,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,kBAAoB,EAAA,CAAC,EAAE,IAAA,EAAM,SAAc,KAAA;AACzC,IAAA,uBACGD,cAAA,CAAAE,2BAAA,EAAA;AAAA,MAAc,QAAU,EAAA,IAAA;AAAA,MACvB,QAAC,kBAAAF,cAAA,CAAAG,sCAAA,EAAA;AAAA,QAAwB,IAAA;AAAA,QAAY,OAAA;AAAA,OAAkB,CAAA;AAAA,KACzD,CAAA,CAAA;AAAA,GAEJ;AACF,CAAA,CAAA;AAaA,MAAM,gBAAmB,GAAAC,gBAAA;AAAA,EACvB,CAAC,EAAE,OAAA,EAAS,YAAY,OAAY,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AAC5D,IAAM,MAAA,SAAA,GAAY,UAAUC,cAAO,GAAA,KAAA,CAAA;AACnC,IAAA,MAAM,EAAE,QAAA,EAAU,aAAe,EAAA,kBAAA,EAAuB,GAAAC,aAAA;AAAA,MACtD,OAAO,EAAE,GAAG,+BAAA,EAAiC,GAAG,UAAW,EAAA,CAAA;AAAA,MAC3D,CAAC,UAAU,CAAA;AAAA,KACb,CAAA;AAEA,IAAM,MAAA,OAAA,GAAU,OAAQ,CAAA,OAAA,IAAW,OAAQ,CAAA,YAAA,CAAA;AAC3C,IAAA,MAAM,WAAW,OAAQ,CAAA,MAAA,CAAA;AACzB,IAAA,MAAM,YACJ,GAAA,OAAA,CAAQ,IAAS,KAAA,WAAA,IAAe,QAAQ,MAAW,KAAA,YAAA,CAAA;AACrD,IAAA,uBACGN,cAAA,CAAA,SAAA,EAAA;AAAA,MAAW,GAAG,KAAA;AAAA,MAAO,GAAK,EAAA,YAAA;AAAA,MACxB,QAAQ,EAAA,OAAA,CAAA,GAAA,CAAI,CAAC,IAAA,EAAM,KAAU,KAAA;AAI5B,QAAM,MAAA,WAAA,GAAc,YAAgB,IAAA,KAAA,KAAU,QAAW,GAAA,CAAA,CAAA;AACzD,QAAM,MAAA,KAAA,GAAQ,EAAE,KAAA,EAAO,WAAY,EAAA,CAAA;AACnC,QAAA,QAAQ,KAAK,IAAM;AAAA,UACjB,KAAK,MAAA;AACH,YAAA,uBAAQA,cAAA,CAAA,QAAA,EAAA;AAAA,cAAqB,IAAA;AAAA,cAAa,GAAG,KAAA;AAAA,aAAA,EAAvB,KAA8B,CAAA,CAAA;AAAA,UACtD,KAAK,WAAA;AACH,YAAA,uBAAQA,cAAA,CAAA,aAAA,EAAA;AAAA,cAA0B,IAAA;AAAA,cAAa,GAAG,KAAA;AAAA,aAAA,EAAvB,KAA8B,CAAA,CAAA;AAAA,UAC3D,KAAK,iBAAA;AACH,YAAA,uBACGA,cAAA,CAAA,kBAAA,EAAA;AAAA,cAEC,IAAA;AAAA,cACC,GAAG,KAAA;AAAA,cACJ,OAAA;AAAA,aAAA,EAHK,KAIP,CAAA,CAAA;AAAA,UAEJ;AACE,YAAO,OAAA,IAAA,CAAA;AAAA,SACX;AAAA,OACD,CAAA;AAAA,KACH,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAEA,IAAI,OAAA,CAAQ,GAAI,CAAA,QAAA,KAAa,YAAc,EAAA;AACzC,EAAA,gBAAA,CAAiB,WAAc,GAAA,uBAAA,CAAA;AACjC;;;;"}

package/dist/primitives/AiMessage/index.js

@@ -0,0 +1,75 @@
+import { jsx } from 'react/jsx-runtime';
+import { Slot } from '@radix-ui/react-slot';
+import { forwardRef, useMemo } from 'react';
+import { ErrorBoundary } from '../../utils/ErrorBoundary.js';
+import { Markdown } from '../Markdown.js';
+import { AiMessageToolInvocation } from './tool-invocation.js';
+
+const AI_MESSAGE_CONTENT_NAME = "AiMessageContent";
+const defaultMessageContentComponents = {
+  TextPart: ({ part }) => {
+    return /* @__PURE__ */ jsx(Markdown, {
+      content: part.text
+    });
+  },
+  ReasoningPart: ({ part }) => {
+    return /* @__PURE__ */ jsx(Markdown, {
+      content: part.text
+    });
+  },
+  ToolInvocationPart: ({ part, message }) => {
+    return /* @__PURE__ */ jsx(ErrorBoundary, {
+      fallback: null,
+      children: /* @__PURE__ */ jsx(AiMessageToolInvocation, {
+        part,
+        message
+      })
+    });
+  }
+};
+const AiMessageContent = forwardRef(
+  ({ message, components, asChild, ...props }, forwardedRef) => {
+    const Component = asChild ? Slot : "div";
+    const { TextPart, ReasoningPart, ToolInvocationPart } = useMemo(
+      () => ({ ...defaultMessageContentComponents, ...components }),
+      [components]
+    );
+    const content = message.content ?? message.contentSoFar;
+    const numParts = content.length;
+    const isGenerating = message.role === "assistant" && message.status === "generating";
+    return /* @__PURE__ */ jsx(Component, {
+      ...props,
+      ref: forwardedRef,
+      children: content.map((part, index) => {
+        const isStreaming = isGenerating && index === numParts - 1;
+        const extra = { index, isStreaming };
+        switch (part.type) {
+          case "text":
+            return /* @__PURE__ */ jsx(TextPart, {
+              part,
+              ...extra
+            }, index);
+          case "reasoning":
+            return /* @__PURE__ */ jsx(ReasoningPart, {
+              part,
+              ...extra
+            }, index);
+          case "tool-invocation":
+            return /* @__PURE__ */ jsx(ToolInvocationPart, {
+              part,
+              ...extra,
+              message
+            }, index);
+          default:
+            return null;
+        }
+      })
+    });
+  }
+);
+if (process.env.NODE_ENV !== "production") {
+  AiMessageContent.displayName = AI_MESSAGE_CONTENT_NAME;
+}
+
+export { AiMessageContent as Content };
+//# sourceMappingURL=index.js.map

package/dist/primitives/AiMessage/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sources":["../../../src/primitives/AiMessage/index.tsx"],"sourcesContent":["import { Slot } from \"@radix-ui/react-slot\";\nimport { forwardRef, useMemo } from \"react\";\n\nimport { ErrorBoundary } from \"../../utils/ErrorBoundary\";\nimport { Markdown } from \"../Markdown\";\nimport { AiMessageToolInvocation } from \"./tool-invocation\";\nimport type {\n  AiMessageContentComponents,\n  AiMessageContentProps,\n} from \"./types\";\n\nconst AI_MESSAGE_CONTENT_NAME = \"AiMessageContent\";\n\nconst defaultMessageContentComponents: AiMessageContentComponents = {\n  TextPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  ReasoningPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  ToolInvocationPart: ({ part, message }) => {\n    return (\n      <ErrorBoundary fallback={null}>\n        <AiMessageToolInvocation part={part} message={message} />\n      </ErrorBoundary>\n    );\n  },\n};\n\n/**\n * --------------------------------------------------------------------------\n * @private The API for this component is not yet stable.\n * --------------------------------------------------------------------------\n *\n * Primitive to help display an user or assistant message’s content, which is\n * an array of parts.\n *\n * @example\n * <AiMessage.Content message={message} components={{ TextPart }} />\n */\nconst AiMessageContent = forwardRef<HTMLDivElement, AiMessageContentProps>(\n  ({ message, components, asChild, ...props }, forwardedRef) => {\n    const Component = asChild ? Slot : \"div\";\n    const { TextPart, ReasoningPart, ToolInvocationPart } = useMemo(\n      () => ({ ...defaultMessageContentComponents, ...components }),\n      [components]\n    );\n\n    const content = message.content ?? message.contentSoFar;\n    const numParts = content.length;\n    const isGenerating =\n      message.role === \"assistant\" && message.status === \"generating\";\n    return (\n      <Component {...props} ref={forwardedRef}>\n        {content.map((part, index) => {\n          // A part is considered to be still \"streaming in\" if it's the last\n          // part in the content array, and the message is in \"generating\"\n          // state.\n          const isStreaming = isGenerating && index === numParts - 1;\n          const extra = { index, isStreaming };\n          switch (part.type) {\n            case \"text\":\n              return <TextPart key={index} part={part} {...extra} />;\n            case \"reasoning\":\n              return <ReasoningPart key={index} part={part} {...extra} />;\n            case \"tool-invocation\":\n              return (\n                <ToolInvocationPart\n                  key={index}\n                  part={part}\n                  {...extra}\n                  message={message}\n                />\n              );\n            default:\n              return null;\n          }\n        })}\n      </Component>\n    );\n  }\n);\n\nif (process.env.NODE_ENV !== \"production\") {\n  AiMessageContent.displayName = AI_MESSAGE_CONTENT_NAME;\n}\n\n// NOTE: Every export from this file will be available publicly as AiMessage.*\nexport { AiMessageContent as Content };\n"],"names":[],"mappings":";;;;;;;AAWA,MAAM,uBAA0B,GAAA,kBAAA,CAAA;AAEhC,MAAM,+BAA8D,GAAA;AAAA,EAClE,QAAU,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AACtB,IAAA,uBAAQ,GAAA,CAAA,QAAA,EAAA;AAAA,MAAS,SAAS,IAAK,CAAA,IAAA;AAAA,KAAM,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,aAAe,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AAC3B,IAAA,uBAAQ,GAAA,CAAA,QAAA,EAAA;AAAA,MAAS,SAAS,IAAK,CAAA,IAAA;AAAA,KAAM,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,kBAAoB,EAAA,CAAC,EAAE,IAAA,EAAM,SAAc,KAAA;AACzC,IAAA,uBACG,GAAA,CAAA,aAAA,EAAA;AAAA,MAAc,QAAU,EAAA,IAAA;AAAA,MACvB,QAAC,kBAAA,GAAA,CAAA,uBAAA,EAAA;AAAA,QAAwB,IAAA;AAAA,QAAY,OAAA;AAAA,OAAkB,CAAA;AAAA,KACzD,CAAA,CAAA;AAAA,GAEJ;AACF,CAAA,CAAA;AAaA,MAAM,gBAAmB,GAAA,UAAA;AAAA,EACvB,CAAC,EAAE,OAAA,EAAS,YAAY,OAAY,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AAC5D,IAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,KAAA,CAAA;AACnC,IAAA,MAAM,EAAE,QAAA,EAAU,aAAe,EAAA,kBAAA,EAAuB,GAAA,OAAA;AAAA,MACtD,OAAO,EAAE,GAAG,+BAAA,EAAiC,GAAG,UAAW,EAAA,CAAA;AAAA,MAC3D,CAAC,UAAU,CAAA;AAAA,KACb,CAAA;AAEA,IAAM,MAAA,OAAA,GAAU,OAAQ,CAAA,OAAA,IAAW,OAAQ,CAAA,YAAA,CAAA;AAC3C,IAAA,MAAM,WAAW,OAAQ,CAAA,MAAA,CAAA;AACzB,IAAA,MAAM,YACJ,GAAA,OAAA,CAAQ,IAAS,KAAA,WAAA,IAAe,QAAQ,MAAW,KAAA,YAAA,CAAA;AACrD,IAAA,uBACG,GAAA,CAAA,SAAA,EAAA;AAAA,MAAW,GAAG,KAAA;AAAA,MAAO,GAAK,EAAA,YAAA;AAAA,MACxB,QAAQ,EAAA,OAAA,CAAA,GAAA,CAAI,CAAC,IAAA,EAAM,KAAU,KAAA;AAI5B,QAAM,MAAA,WAAA,GAAc,YAAgB,IAAA,KAAA,KAAU,QAAW,GAAA,CAAA,CAAA;AACzD,QAAM,MAAA,KAAA,GAAQ,EAAE,KAAA,EAAO,WAAY,EAAA,CAAA;AACnC,QAAA,QAAQ,KAAK,IAAM;AAAA,UACjB,KAAK,MAAA;AACH,YAAA,uBAAQ,GAAA,CAAA,QAAA,EAAA;AAAA,cAAqB,IAAA;AAAA,cAAa,GAAG,KAAA;AAAA,aAAA,EAAvB,KAA8B,CAAA,CAAA;AAAA,UACtD,KAAK,WAAA;AACH,YAAA,uBAAQ,GAAA,CAAA,aAAA,EAAA;AAAA,cAA0B,IAAA;AAAA,cAAa,GAAG,KAAA;AAAA,aAAA,EAAvB,KAA8B,CAAA,CAAA;AAAA,UAC3D,KAAK,iBAAA;AACH,YAAA,uBACG,GAAA,CAAA,kBAAA,EAAA;AAAA,cAEC,IAAA;AAAA,cACC,GAAG,KAAA;AAAA,cACJ,OAAA;AAAA,aAAA,EAHK,KAIP,CAAA,CAAA;AAAA,UAEJ;AACE,YAAO,OAAA,IAAA,CAAA;AAAA,SACX;AAAA,OACD,CAAA;AAAA,KACH,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAEA,IAAI,OAAA,CAAQ,GAAI,CAAA,QAAA,KAAa,YAAc,EAAA;AACzC,EAAA,gBAAA,CAAiB,WAAc,GAAA,uBAAA,CAAA;AACjC;;;;"}

package/dist/primitives/AiMessage/tool-invocation.cjs

@@ -0,0 +1,70 @@
+'use strict';
+
+var jsxRuntime = require('react/jsx-runtime');
+var core = require('@liveblocks/core');
+var react = require('@liveblocks/react');
+var _private = require('@liveblocks/react/_private');
+var react$1 = require('react');
+var contexts = require('./contexts.cjs');
+
+function StableRenderFn(props) {
+  return props.renderFn(props.props);
+}
+function AiMessageToolInvocation({
+  message,
+  part
+}) {
+  const client = react.useClient();
+  const ai = client[core.kInternal].ai;
+  const tool = _private.useSignal(ai.signals.getTool\u03A3(part.name, message.chatId));
+  const respond = react$1.useCallback(
+    (result) => {
+      if (message.status !== "awaiting-tool") {
+      } else if (part.stage === "receiving") {
+      } else if (part.stage === "executed") {
+        console.log(
+          `Ignoring respond(): tool '${part.name}' (${part.invocationId}) has already executed`
+        );
+      } else {
+        ai.setToolResult(
+          message.chatId,
+          message.id,
+          part.invocationId,
+          result ?? { data: {} }
+        );
+      }
+    },
+    [
+      ai,
+      message.chatId,
+      message.id,
+      message.status,
+      part.invocationId,
+      part.name,
+      part.stage
+    ]
+  );
+  const props = react$1.useMemo(() => {
+    const { type: _, ...rest } = part;
+    return {
+      ...rest,
+      respond,
+      types: void 0,
+      [core.kInternal]: {
+        execute: tool?.execute
+      }
+    };
+  }, [part, respond, tool?.execute]);
+  if (tool?.render === void 0)
+    return null;
+  return /* @__PURE__ */ jsxRuntime.jsx(contexts.AiToolInvocationContext.Provider, {
+    value: props,
+    children: /* @__PURE__ */ jsxRuntime.jsx(StableRenderFn, {
+      renderFn: tool.render,
+      props
+    })
+  });
+}
+
+exports.AiMessageToolInvocation = AiMessageToolInvocation;
+//# sourceMappingURL=tool-invocation.cjs.map

package/dist/primitives/AiMessage/tool-invocation.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-invocation.cjs","sources":["../../../src/primitives/AiMessage/tool-invocation.tsx"],"sourcesContent":["import type {\n  AiChatMessage,\n  AiToolInvocationPart,\n  AiToolInvocationProps,\n  JsonObject,\n  ToolResultResponse,\n} from \"@liveblocks/core\";\nimport { kInternal } from \"@liveblocks/core\";\nimport { useClient } from \"@liveblocks/react\";\nimport { useSignal } from \"@liveblocks/react/_private\";\nimport { type FunctionComponent, useCallback, useMemo } from \"react\";\n\nimport { AiToolInvocationContext } from \"./contexts\";\n\ntype OpaqueAiToolInvocationProps = AiToolInvocationProps<\n  JsonObject,\n  JsonObject\n>;\n\nfunction StableRenderFn(props: {\n  renderFn: FunctionComponent<OpaqueAiToolInvocationProps>;\n  props: OpaqueAiToolInvocationProps;\n}) {\n  return props.renderFn(props.props);\n}\n\n/**\n * @internal\n *\n * This could become publicly exposed as <AiMessage.ToolInvocation /> in the future,\n * but because namespace exports can't be marked `@internal`, we're keeping it in its\n * own file for now.\n */\nexport function AiMessageToolInvocation({\n  message,\n  part,\n}: {\n  message: AiChatMessage;\n  part: AiToolInvocationPart;\n}) {\n  const client = useClient();\n  const ai = client[kInternal].ai;\n  const tool = useSignal(ai.signals.getToolΣ(part.name, message.chatId));\n\n  const respond = useCallback(\n    (result: ToolResultResponse | undefined) => {\n      if (message.status !== \"awaiting-tool\") {\n        // console.log(\"Ignoring respond(): message not awaiting tool result\");\n      } else if (part.stage === \"receiving\") {\n        // console.log(\n        //   `Ignoring respond(): tool '${part.name}' (${part.invocationId}) is still receiving`\n        // );\n      } else if (part.stage === \"executed\") {\n        console.log(\n          `Ignoring respond(): tool '${part.name}' (${part.invocationId}) has already executed`\n        );\n      } else {\n        ai.setToolResult(\n          message.chatId,\n          message.id,\n          part.invocationId,\n          result ?? { data: {} }\n          // TODO Pass in AiGenerationOptions here?\n        );\n      }\n    },\n    [\n      ai,\n      message.chatId,\n      message.id,\n      message.status,\n      part.invocationId,\n      part.name,\n      part.stage,\n    ]\n  );\n\n  const props = useMemo(() => {\n    const { type: _, ...rest } = part;\n    return {\n      ...rest,\n      respond,\n      types: undefined as never,\n      [kInternal]: {\n        execute: tool?.execute,\n      },\n    };\n  }, [part, respond, tool?.execute]);\n\n  if (tool?.render === undefined) return null;\n  return (\n    <AiToolInvocationContext.Provider value={props}>\n      <StableRenderFn\n        renderFn={tool.render as FunctionComponent<OpaqueAiToolInvocationProps>}\n        props={props}\n      />\n    </AiToolInvocationContext.Provider>\n  );\n}\n"],"names":["useClient","kInternal","useSignal","useCallback","useMemo","jsx","AiToolInvocationContext"],"mappings":";;;;;;;;;AAmBA,SAAS,eAAe,KAGrB,EAAA;AACD,EAAO,OAAA,KAAA,CAAM,QAAS,CAAA,KAAA,CAAM,KAAK,CAAA,CAAA;AACnC,CAAA;AASO,SAAS,uBAAwB,CAAA;AAAA,EACtC,OAAA;AAAA,EACA,IAAA;AACF,CAGG,EAAA;AACD,EAAA,MAAM,SAASA,eAAU,EAAA,CAAA;AACzB,EAAM,MAAA,EAAA,GAAK,OAAOC,cAAW,CAAA,CAAA,EAAA,CAAA;AAC7B,EAAM,MAAA,IAAA,GAAOC,mBAAU,EAAG,CAAA,OAAA,CAAQ,cAAS,IAAK,CAAA,IAAA,EAAM,OAAQ,CAAA,MAAM,CAAC,CAAA,CAAA;AAErE,EAAA,MAAM,OAAU,GAAAC,mBAAA;AAAA,IACd,CAAC,MAA2C,KAAA;AAC1C,MAAI,IAAA,OAAA,CAAQ,WAAW,eAAiB,EAAA;AAAA,OAExC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,WAAa,EAAA;AAAA,OAIvC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,UAAY,EAAA;AACpC,QAAQ,OAAA,CAAA,GAAA;AAAA,UACN,CAAA,0BAAA,EAA6B,IAAK,CAAA,IAAA,CAAA,GAAA,EAAU,IAAK,CAAA,YAAA,CAAA,sBAAA,CAAA;AAAA,SACnD,CAAA;AAAA,OACK,MAAA;AACL,QAAG,EAAA,CAAA,aAAA;AAAA,UACD,OAAQ,CAAA,MAAA;AAAA,UACR,OAAQ,CAAA,EAAA;AAAA,UACR,IAAK,CAAA,YAAA;AAAA,UACL,MAAU,IAAA,EAAE,IAAM,EAAA,EAAG,EAAA;AAAA,SAEvB,CAAA;AAAA,OACF;AAAA,KACF;AAAA,IACA;AAAA,MACE,EAAA;AAAA,MACA,OAAQ,CAAA,MAAA;AAAA,MACR,OAAQ,CAAA,EAAA;AAAA,MACR,OAAQ,CAAA,MAAA;AAAA,MACR,IAAK,CAAA,YAAA;AAAA,MACL,IAAK,CAAA,IAAA;AAAA,MACL,IAAK,CAAA,KAAA;AAAA,KACP;AAAA,GACF,CAAA;AAEA,EAAM,MAAA,KAAA,GAAQC,gBAAQ,MAAM;AAC1B,IAAA,MAAM,EAAE,IAAA,EAAM,CAAM,EAAA,GAAA,IAAA,EAAS,GAAA,IAAA,CAAA;AAC7B,IAAO,OAAA;AAAA,MACL,GAAG,IAAA;AAAA,MACH,OAAA;AAAA,MACA,KAAO,EAAA,KAAA,CAAA;AAAA,MACP,CAACH,cAAY,GAAA;AAAA,QACX,SAAS,IAAM,EAAA,OAAA;AAAA,OACjB;AAAA,KACF,CAAA;AAAA,KACC,CAAC,IAAA,EAAM,OAAS,EAAA,IAAA,EAAM,OAAO,CAAC,CAAA,CAAA;AAEjC,EAAA,IAAI,MAAM,MAAW,KAAA,KAAA,CAAA;AAAW,IAAO,OAAA,IAAA,CAAA;AACvC,EACE,uBAAAI,cAAA,CAACC,iCAAwB,QAAxB,EAAA;AAAA,IAAiC,KAAO,EAAA,KAAA;AAAA,IACvC,QAAC,kBAAAD,cAAA,CAAA,cAAA,EAAA;AAAA,MACC,UAAU,IAAK,CAAA,MAAA;AAAA,MACf,KAAA;AAAA,KACF,CAAA;AAAA,GACF,CAAA,CAAA;AAEJ;;;;"}

package/dist/primitives/AiMessage/tool-invocation.js

@@ -0,0 +1,68 @@
+import { jsx } from 'react/jsx-runtime';
+import { kInternal } from '@liveblocks/core';
+import { useClient } from '@liveblocks/react';
+import { useSignal } from '@liveblocks/react/_private';
+import { useCallback, useMemo } from 'react';
+import { AiToolInvocationContext } from './contexts.js';
+
+function StableRenderFn(props) {
+  return props.renderFn(props.props);
+}
+function AiMessageToolInvocation({
+  message,
+  part
+}) {
+  const client = useClient();
+  const ai = client[kInternal].ai;
+  const tool = useSignal(ai.signals.getTool\u03A3(part.name, message.chatId));
+  const respond = useCallback(
+    (result) => {
+      if (message.status !== "awaiting-tool") {
+      } else if (part.stage === "receiving") {
+      } else if (part.stage === "executed") {
+        console.log(
+          `Ignoring respond(): tool '${part.name}' (${part.invocationId}) has already executed`
+        );
+      } else {
+        ai.setToolResult(
+          message.chatId,
+          message.id,
+          part.invocationId,
+          result ?? { data: {} }
+        );
+      }
+    },
+    [
+      ai,
+      message.chatId,
+      message.id,
+      message.status,
+      part.invocationId,
+      part.name,
+      part.stage
+    ]
+  );
+  const props = useMemo(() => {
+    const { type: _, ...rest } = part;
+    return {
+      ...rest,
+      respond,
+      types: void 0,
+      [kInternal]: {
+        execute: tool?.execute
+      }
+    };
+  }, [part, respond, tool?.execute]);
+  if (tool?.render === void 0)
+    return null;
+  return /* @__PURE__ */ jsx(AiToolInvocationContext.Provider, {
+    value: props,
+    children: /* @__PURE__ */ jsx(StableRenderFn, {
+      renderFn: tool.render,
+      props
+    })
+  });
+}
+
+export { AiMessageToolInvocation };
+//# sourceMappingURL=tool-invocation.js.map

package/dist/primitives/AiMessage/tool-invocation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-invocation.js","sources":["../../../src/primitives/AiMessage/tool-invocation.tsx"],"sourcesContent":["import type {\n  AiChatMessage,\n  AiToolInvocationPart,\n  AiToolInvocationProps,\n  JsonObject,\n  ToolResultResponse,\n} from \"@liveblocks/core\";\nimport { kInternal } from \"@liveblocks/core\";\nimport { useClient } from \"@liveblocks/react\";\nimport { useSignal } from \"@liveblocks/react/_private\";\nimport { type FunctionComponent, useCallback, useMemo } from \"react\";\n\nimport { AiToolInvocationContext } from \"./contexts\";\n\ntype OpaqueAiToolInvocationProps = AiToolInvocationProps<\n  JsonObject,\n  JsonObject\n>;\n\nfunction StableRenderFn(props: {\n  renderFn: FunctionComponent<OpaqueAiToolInvocationProps>;\n  props: OpaqueAiToolInvocationProps;\n}) {\n  return props.renderFn(props.props);\n}\n\n/**\n * @internal\n *\n * This could become publicly exposed as <AiMessage.ToolInvocation /> in the future,\n * but because namespace exports can't be marked `@internal`, we're keeping it in its\n * own file for now.\n */\nexport function AiMessageToolInvocation({\n  message,\n  part,\n}: {\n  message: AiChatMessage;\n  part: AiToolInvocationPart;\n}) {\n  const client = useClient();\n  const ai = client[kInternal].ai;\n  const tool = useSignal(ai.signals.getToolΣ(part.name, message.chatId));\n\n  const respond = useCallback(\n    (result: ToolResultResponse | undefined) => {\n      if (message.status !== \"awaiting-tool\") {\n        // console.log(\"Ignoring respond(): message not awaiting tool result\");\n      } else if (part.stage === \"receiving\") {\n        // console.log(\n        //   `Ignoring respond(): tool '${part.name}' (${part.invocationId}) is still receiving`\n        // );\n      } else if (part.stage === \"executed\") {\n        console.log(\n          `Ignoring respond(): tool '${part.name}' (${part.invocationId}) has already executed`\n        );\n      } else {\n        ai.setToolResult(\n          message.chatId,\n          message.id,\n          part.invocationId,\n          result ?? { data: {} }\n          // TODO Pass in AiGenerationOptions here?\n        );\n      }\n    },\n    [\n      ai,\n      message.chatId,\n      message.id,\n      message.status,\n      part.invocationId,\n      part.name,\n      part.stage,\n    ]\n  );\n\n  const props = useMemo(() => {\n    const { type: _, ...rest } = part;\n    return {\n      ...rest,\n      respond,\n      types: undefined as never,\n      [kInternal]: {\n        execute: tool?.execute,\n      },\n    };\n  }, [part, respond, tool?.execute]);\n\n  if (tool?.render === undefined) return null;\n  return (\n    <AiToolInvocationContext.Provider value={props}>\n      <StableRenderFn\n        renderFn={tool.render as FunctionComponent<OpaqueAiToolInvocationProps>}\n        props={props}\n      />\n    </AiToolInvocationContext.Provider>\n  );\n}\n"],"names":[],"mappings":";;;;;;;AAmBA,SAAS,eAAe,KAGrB,EAAA;AACD,EAAO,OAAA,KAAA,CAAM,QAAS,CAAA,KAAA,CAAM,KAAK,CAAA,CAAA;AACnC,CAAA;AASO,SAAS,uBAAwB,CAAA;AAAA,EACtC,OAAA;AAAA,EACA,IAAA;AACF,CAGG,EAAA;AACD,EAAA,MAAM,SAAS,SAAU,EAAA,CAAA;AACzB,EAAM,MAAA,EAAA,GAAK,OAAO,SAAW,CAAA,CAAA,EAAA,CAAA;AAC7B,EAAM,MAAA,IAAA,GAAO,UAAU,EAAG,CAAA,OAAA,CAAQ,cAAS,IAAK,CAAA,IAAA,EAAM,OAAQ,CAAA,MAAM,CAAC,CAAA,CAAA;AAErE,EAAA,MAAM,OAAU,GAAA,WAAA;AAAA,IACd,CAAC,MAA2C,KAAA;AAC1C,MAAI,IAAA,OAAA,CAAQ,WAAW,eAAiB,EAAA;AAAA,OAExC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,WAAa,EAAA;AAAA,OAIvC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,UAAY,EAAA;AACpC,QAAQ,OAAA,CAAA,GAAA;AAAA,UACN,CAAA,0BAAA,EAA6B,IAAK,CAAA,IAAA,CAAA,GAAA,EAAU,IAAK,CAAA,YAAA,CAAA,sBAAA,CAAA;AAAA,SACnD,CAAA;AAAA,OACK,MAAA;AACL,QAAG,EAAA,CAAA,aAAA;AAAA,UACD,OAAQ,CAAA,MAAA;AAAA,UACR,OAAQ,CAAA,EAAA;AAAA,UACR,IAAK,CAAA,YAAA;AAAA,UACL,MAAU,IAAA,EAAE,IAAM,EAAA,EAAG,EAAA;AAAA,SAEvB,CAAA;AAAA,OACF;AAAA,KACF;AAAA,IACA;AAAA,MACE,EAAA;AAAA,MACA,OAAQ,CAAA,MAAA;AAAA,MACR,OAAQ,CAAA,EAAA;AAAA,MACR,OAAQ,CAAA,MAAA;AAAA,MACR,IAAK,CAAA,YAAA;AAAA,MACL,IAAK,CAAA,IAAA;AAAA,MACL,IAAK,CAAA,KAAA;AAAA,KACP;AAAA,GACF,CAAA;AAEA,EAAM,MAAA,KAAA,GAAQ,QAAQ,MAAM;AAC1B,IAAA,MAAM,EAAE,IAAA,EAAM,CAAM,EAAA,GAAA,IAAA,EAAS,GAAA,IAAA,CAAA;AAC7B,IAAO,OAAA;AAAA,MACL,GAAG,IAAA;AAAA,MACH,OAAA;AAAA,MACA,KAAO,EAAA,KAAA,CAAA;AAAA,MACP,CAAC,SAAY,GAAA;AAAA,QACX,SAAS,IAAM,EAAA,OAAA;AAAA,OACjB;AAAA,KACF,CAAA;AAAA,KACC,CAAC,IAAA,EAAM,OAAS,EAAA,IAAA,EAAM,OAAO,CAAC,CAAA,CAAA;AAEjC,EAAA,IAAI,MAAM,MAAW,KAAA,KAAA,CAAA;AAAW,IAAO,OAAA,IAAA,CAAA;AACvC,EACE,uBAAA,GAAA,CAAC,wBAAwB,QAAxB,EAAA;AAAA,IAAiC,KAAO,EAAA,KAAA;AAAA,IACvC,QAAC,kBAAA,GAAA,CAAA,cAAA,EAAA;AAAA,MACC,UAAU,IAAK,CAAA,MAAA;AAAA,MACf,KAAA;AAAA,KACF,CAAA;AAAA,GACF,CAAA,CAAA;AAEJ;;;;"}

package/dist/primitives/{internal/Collapsible → Collapsible}/index.cjs

@@ -3,29 +3,45 @@
 var jsxRuntime = require('react/jsx-runtime');
 var reactSlot = require('@radix-ui/react-slot');
 var react = require('react');
+var useControllableState = require('../../utils/use-controllable-state.cjs');
 
-const RootContext = react.createContext(null);
-const Root = react.forwardRef(
-  ({ open, onOpenChange, disabled = false, asChild, ...props }, forwardedRef) => {
+const COLLAPSIBLE_ROOT_NAME = "CollapsibleRoot";
+const COLLAPSIBLE_TRIGGER_NAME = "CollapsibleTrigger";
+const COLLAPSIBLE_CONTENT_NAME = "CollapsibleContent";
+const CollapsibleContext = react.createContext(null);
+const CollapsibleRoot = react.forwardRef(
+  ({
+    open: controlledOpen,
+    onOpenChange: controlledOnOpenChange,
+    defaultOpen,
+    disabled = false,
+    asChild,
+    ...props
+  }, forwardedRef) => {
+    const [isOpen, onOpenChange] = useControllableState.useControllableState(
+      defaultOpen ?? true,
+      controlledOpen,
+      controlledOnOpenChange
+    );
     const Component = asChild ? reactSlot.Slot : "div";
     const id = react.useId();
-    return /* @__PURE__ */ jsxRuntime.jsx(RootContext.Provider, {
-      value: { open, onOpenChange, disabled, contentId: id },
+    return /* @__PURE__ */ jsxRuntime.jsx(CollapsibleContext.Provider, {
+      value: { open: isOpen, onOpenChange, disabled, contentId: id },
       children: /* @__PURE__ */ jsxRuntime.jsx(Component, {
         ...props,
         ref: forwardedRef,
-        "data-state": open ? "open" : "closed",
+        "data-state": isOpen ? "open" : "closed",
         "data-disabled": disabled ? "" : void 0
       })
     });
   }
 );
-const Trigger = react.forwardRef(
+const CollapsibleTrigger = react.forwardRef(
   ({ onClick, asChild, ...props }, forwardedRef) => {
     const Component = asChild ? reactSlot.Slot : "button";
-    const context = react.useContext(RootContext);
+    const context = react.useContext(CollapsibleContext);
     if (!context) {
-      throw new Error("Collapsible.Trigger must be a descendant of Root");
+      throw new Error("Collapsible.Root is missing from the React tree.");
     }
     const { open, disabled, contentId, onOpenChange } = context;
     return /* @__PURE__ */ jsxRuntime.jsx(Component, {
@@ -48,14 +64,15 @@ const Trigger = react.forwardRef(
     });
   }
 );
-const Content = react.forwardRef(
+const CollapsibleContent = react.forwardRef(
   ({ asChild, ...props }, forwardedRef) => {
     const Component = asChild ? reactSlot.Slot : "div";
-    const rootContext = react.useContext(RootContext);
+    const context = react.useContext(CollapsibleContext);
     const divRef = react.useRef(null);
-    if (!rootContext)
-      throw new Error("Missing RootContext Provider");
-    const { open, onOpenChange, disabled, contentId } = rootContext;
+    if (!context) {
+      throw new Error("Collapsible.Root is missing from the React tree.");
+    }
+    const { open, onOpenChange, disabled, contentId } = context;
     react.useEffect(() => {
       const element = divRef.current;
       if (element === null)
@@ -98,8 +115,13 @@ const Content = react.forwardRef(
     });
   }
 );
+if (process.env.NODE_ENV !== "production") {
+  CollapsibleContent.displayName = COLLAPSIBLE_CONTENT_NAME;
+  CollapsibleRoot.displayName = COLLAPSIBLE_ROOT_NAME;
+  CollapsibleTrigger.displayName = COLLAPSIBLE_TRIGGER_NAME;
+}
 
-exports.Content = Content;
-exports.Root = Root;
-exports.Trigger = Trigger;
+exports.Content = CollapsibleContent;
+exports.Root = CollapsibleRoot;
+exports.Trigger = CollapsibleTrigger;
 //# sourceMappingURL=index.cjs.map

package/dist/primitives/Collapsible/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","sources":["../../../src/primitives/Collapsible/index.tsx"],"sourcesContent":["import { Slot } from \"@radix-ui/react-slot\";\nimport {\n  createContext,\n  forwardRef,\n  useContext,\n  useEffect,\n  useId,\n  useImperativeHandle,\n  useRef,\n} from \"react\";\n\nimport { useControllableState } from \"../../utils/use-controllable-state\";\nimport type { ContentProps, RootProps, TriggerProps } from \"./types\";\n\nconst COLLAPSIBLE_ROOT_NAME = \"CollapsibleRoot\";\nconst COLLAPSIBLE_TRIGGER_NAME = \"CollapsibleTrigger\";\nconst COLLAPSIBLE_CONTENT_NAME = \"CollapsibleContent\";\n\nconst CollapsibleContext = createContext<{\n  open?: boolean;\n  onOpenChange: (open: boolean) => void;\n  disabled: boolean;\n  contentId: string;\n} | null>(null);\n\n/* -------------------------------------------------------------------------------------------------\n * Root\n * -----------------------------------------------------------------------------------------------*/\n\nconst CollapsibleRoot = forwardRef<HTMLDivElement, RootProps>(\n  (\n    {\n      open: controlledOpen,\n      onOpenChange: controlledOnOpenChange,\n      defaultOpen,\n      disabled = false,\n      asChild,\n      ...props\n    },\n    forwardedRef\n  ) => {\n    const [isOpen, onOpenChange] = useControllableState(\n      defaultOpen ?? true,\n      controlledOpen,\n      controlledOnOpenChange\n    );\n    const Component = asChild ? Slot : \"div\";\n    const id = useId();\n\n    return (\n      <CollapsibleContext.Provider\n        value={{ open: isOpen, onOpenChange, disabled, contentId: id }}\n      >\n        <Component\n          {...props}\n          ref={forwardedRef}\n          data-state={isOpen ? \"open\" : \"closed\"}\n          data-disabled={disabled ? \"\" : undefined}\n        />\n      </CollapsibleContext.Provider>\n    );\n  }\n);\n\n/* -------------------------------------------------------------------------------------------------\n * Trigger\n * -----------------------------------------------------------------------------------------------*/\n\nconst CollapsibleTrigger = forwardRef<HTMLButtonElement, TriggerProps>(\n  ({ onClick, asChild, ...props }, forwardedRef) => {\n    const Component = asChild ? Slot : \"button\";\n    const context = useContext(CollapsibleContext);\n\n    if (!context) {\n      throw new Error(\"Collapsible.Root is missing from the React tree.\");\n    }\n\n    const { open, disabled, contentId, onOpenChange } = context;\n\n    return (\n      <Component\n        {...props}\n        ref={forwardedRef}\n        type=\"button\"\n        aria-controls={contentId}\n        aria-expanded={open || false}\n        data-state={open ? \"open\" : \"closed\"}\n        data-disabled={disabled ? \"\" : undefined}\n        disabled={disabled}\n        onClick={(event) => {\n          onClick?.(event);\n          if (event.defaultPrevented) return;\n          if (disabled) return;\n          onOpenChange(!open);\n        }}\n      />\n    );\n  }\n);\n\n/* -------------------------------------------------------------------------------------------------\n * Content\n * -----------------------------------------------------------------------------------------------*/\n\nconst CollapsibleContent = forwardRef<HTMLDivElement, ContentProps>(\n  ({ asChild, ...props }, forwardedRef) => {\n    const Component = asChild ? Slot : \"div\";\n    const context = useContext(CollapsibleContext);\n    const divRef = useRef<HTMLDivElement>(null);\n\n    if (!context) {\n      throw new Error(\"Collapsible.Root is missing from the React tree.\");\n    }\n\n    const { open, onOpenChange, disabled, contentId } = context;\n\n    useEffect(() => {\n      const element = divRef.current;\n      if (element === null) return;\n\n      const isHiddenUntilFoundSupported = \"onbeforematch\" in document.body;\n      if (!isHiddenUntilFoundSupported) return;\n\n      function handleBeforeMatch() {\n        onOpenChange(true);\n      }\n\n      // https://developer.chrome.com/articles/hidden-until-found/\n      element.addEventListener(\"beforematch\", handleBeforeMatch);\n      return () => {\n        element.removeEventListener(\"beforematch\", handleBeforeMatch);\n      };\n    }, [onOpenChange]);\n\n    // Passing `string` to `hidden` in JSX is not currently supported: https://github.com/facebook/react/issues/24740\n    useEffect(() => {\n      const element = divRef.current;\n      if (element === null) return;\n\n      if (open) return;\n\n      const isHiddenUntilFoundSupported = \"onbeforematch\" in document.body;\n      if (!isHiddenUntilFoundSupported) return;\n\n      element.setAttribute(\"hidden\", \"until-found\");\n      return () => {\n        element.removeAttribute(\"hidden\");\n      };\n    }, [open]);\n\n    useImperativeHandle<\n      HTMLDivElement | null,\n      HTMLDivElement | null\n    >(forwardedRef, () => {\n      return divRef.current;\n    }, []);\n\n    return (\n      <Component\n        {...props}\n        ref={divRef}\n        data-state={open ? \"open\" : \"closed\"}\n        data-disabled={disabled ? \"\" : undefined}\n        id={contentId}\n        hidden={!open}\n      />\n    );\n  }\n);\n\nif (process.env.NODE_ENV !== \"production\") {\n  CollapsibleContent.displayName = COLLAPSIBLE_CONTENT_NAME;\n  CollapsibleRoot.displayName = COLLAPSIBLE_ROOT_NAME;\n  CollapsibleTrigger.displayName = COLLAPSIBLE_TRIGGER_NAME;\n}\n\n// NOTE: Every export from this file will be available publicly as Collapsible.*\nexport {\n  CollapsibleContent as Content,\n  CollapsibleRoot as Root,\n  CollapsibleTrigger as Trigger,\n};\n"],"names":["createContext","forwardRef","useControllableState","Slot","useId","jsx","useContext","useRef","useEffect","useImperativeHandle"],"mappings":";;;;;;;AAcA,MAAM,qBAAwB,GAAA,iBAAA,CAAA;AAC9B,MAAM,wBAA2B,GAAA,oBAAA,CAAA;AACjC,MAAM,wBAA2B,GAAA,oBAAA,CAAA;AAEjC,MAAM,kBAAA,GAAqBA,oBAKjB,IAAI,CAAA,CAAA;AAMd,MAAM,eAAkB,GAAAC,gBAAA;AAAA,EACtB,CACE;AAAA,IACE,IAAM,EAAA,cAAA;AAAA,IACN,YAAc,EAAA,sBAAA;AAAA,IACd,WAAA;AAAA,IACA,QAAW,GAAA,KAAA;AAAA,IACX,OAAA;AAAA,IACG,GAAA,KAAA;AAAA,KAEL,YACG,KAAA;AACH,IAAM,MAAA,CAAC,MAAQ,EAAA,YAAY,CAAI,GAAAC,yCAAA;AAAA,MAC7B,WAAe,IAAA,IAAA;AAAA,MACf,cAAA;AAAA,MACA,sBAAA;AAAA,KACF,CAAA;AACA,IAAM,MAAA,SAAA,GAAY,UAAUC,cAAO,GAAA,KAAA,CAAA;AACnC,IAAA,MAAM,KAAKC,WAAM,EAAA,CAAA;AAEjB,IACE,uBAAAC,cAAA,CAAC,mBAAmB,QAAnB,EAAA;AAAA,MACC,OAAO,EAAE,IAAA,EAAM,QAAQ,YAAc,EAAA,QAAA,EAAU,WAAW,EAAG,EAAA;AAAA,MAE7D,QAAC,kBAAAA,cAAA,CAAA,SAAA,EAAA;AAAA,QACE,GAAG,KAAA;AAAA,QACJ,GAAK,EAAA,YAAA;AAAA,QACL,YAAA,EAAY,SAAS,MAAS,GAAA,QAAA;AAAA,QAC9B,eAAA,EAAe,WAAW,EAAK,GAAA,KAAA,CAAA;AAAA,OACjC,CAAA;AAAA,KACF,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAMA,MAAM,kBAAqB,GAAAJ,gBAAA;AAAA,EACzB,CAAC,EAAE,OAAA,EAAS,OAAY,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AAChD,IAAM,MAAA,SAAA,GAAY,UAAUE,cAAO,GAAA,QAAA,CAAA;AACnC,IAAM,MAAA,OAAA,GAAUG,iBAAW,kBAAkB,CAAA,CAAA;AAE7C,IAAA,IAAI,CAAC,OAAS,EAAA;AACZ,MAAM,MAAA,IAAI,MAAM,kDAAkD,CAAA,CAAA;AAAA,KACpE;AAEA,IAAA,MAAM,EAAE,IAAA,EAAM,QAAU,EAAA,SAAA,EAAW,cAAiB,GAAA,OAAA,CAAA;AAEpD,IAAA,uBACGD,cAAA,CAAA,SAAA,EAAA;AAAA,MACE,GAAG,KAAA;AAAA,MACJ,GAAK,EAAA,YAAA;AAAA,MACL,IAAK,EAAA,QAAA;AAAA,MACL,eAAe,EAAA,SAAA;AAAA,MACf,iBAAe,IAAQ,IAAA,KAAA;AAAA,MACvB,YAAA,EAAY,OAAO,MAAS,GAAA,QAAA;AAAA,MAC5B,eAAA,EAAe,WAAW,EAAK,GAAA,KAAA,CAAA;AAAA,MAC/B,QAAA;AAAA,MACA,OAAA,EAAS,CAAC,KAAU,KAAA;AAClB,QAAA,OAAA,GAAU,KAAK,CAAA,CAAA;AACf,QAAA,IAAI,KAAM,CAAA,gBAAA;AAAkB,UAAA,OAAA;AAC5B,QAAI,IAAA,QAAA;AAAU,UAAA,OAAA;AACd,QAAA,YAAA,CAAa,CAAC,IAAI,CAAA,CAAA;AAAA,OACpB;AAAA,KACF,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAMA,MAAM,kBAAqB,GAAAJ,gBAAA;AAAA,EACzB,CAAC,EAAE,OAAY,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AACvC,IAAM,MAAA,SAAA,GAAY,UAAUE,cAAO,GAAA,KAAA,CAAA;AACnC,IAAM,MAAA,OAAA,GAAUG,iBAAW,kBAAkB,CAAA,CAAA;AAC7C,IAAM,MAAA,MAAA,GAASC,aAAuB,IAAI,CAAA,CAAA;AAE1C,IAAA,IAAI,CAAC,OAAS,EAAA;AACZ,MAAM,MAAA,IAAI,MAAM,kDAAkD,CAAA,CAAA;AAAA,KACpE;AAEA,IAAA,MAAM,EAAE,IAAA,EAAM,YAAc,EAAA,QAAA,EAAU,WAAc,GAAA,OAAA,CAAA;AAEpD,IAAAC,eAAA,CAAU,MAAM;AACd,MAAA,MAAM,UAAU,MAAO,CAAA,OAAA,CAAA;AACvB,MAAA,IAAI,OAAY,KAAA,IAAA;AAAM,QAAA,OAAA;AAEtB,MAAM,MAAA,2BAAA,GAA8B,mBAAmB,QAAS,CAAA,IAAA,CAAA;AAChE,MAAA,IAAI,CAAC,2BAAA;AAA6B,QAAA,OAAA;AAElC,MAAA,SAAS,iBAAoB,GAAA;AAC3B,QAAA,YAAA,CAAa,IAAI,CAAA,CAAA;AAAA,OACnB;AAGA,MAAQ,OAAA,CAAA,gBAAA,CAAiB,eAAe,iBAAiB,CAAA,CAAA;AACzD,MAAA,OAAO,MAAM;AACX,QAAQ,OAAA,CAAA,mBAAA,CAAoB,eAAe,iBAAiB,CAAA,CAAA;AAAA,OAC9D,CAAA;AAAA,KACF,EAAG,CAAC,YAAY,CAAC,CAAA,CAAA;AAGjB,IAAAA,eAAA,CAAU,MAAM;AACd,MAAA,MAAM,UAAU,MAAO,CAAA,OAAA,CAAA;AACvB,MAAA,IAAI,OAAY,KAAA,IAAA;AAAM,QAAA,OAAA;AAEtB,MAAI,IAAA,IAAA;AAAM,QAAA,OAAA;AAEV,MAAM,MAAA,2BAAA,GAA8B,mBAAmB,QAAS,CAAA,IAAA,CAAA;AAChE,MAAA,IAAI,CAAC,2BAAA;AAA6B,QAAA,OAAA;AAElC,MAAQ,OAAA,CAAA,YAAA,CAAa,UAAU,aAAa,CAAA,CAAA;AAC5C,MAAA,OAAO,MAAM;AACX,QAAA,OAAA,CAAQ,gBAAgB,QAAQ,CAAA,CAAA;AAAA,OAClC,CAAA;AAAA,KACF,EAAG,CAAC,IAAI,CAAC,CAAA,CAAA;AAET,IAAAC,yBAAA,CAGE,cAAc,MAAM;AACpB,MAAA,OAAO,MAAO,CAAA,OAAA,CAAA;AAAA,KAChB,EAAG,EAAE,CAAA,CAAA;AAEL,IAAA,uBACGJ,cAAA,CAAA,SAAA,EAAA;AAAA,MACE,GAAG,KAAA;AAAA,MACJ,GAAK,EAAA,MAAA;AAAA,MACL,YAAA,EAAY,OAAO,MAAS,GAAA,QAAA;AAAA,MAC5B,eAAA,EAAe,WAAW,EAAK,GAAA,KAAA,CAAA;AAAA,MAC/B,EAAI,EAAA,SAAA;AAAA,MACJ,QAAQ,CAAC,IAAA;AAAA,KACX,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAEA,IAAI,OAAA,CAAQ,GAAI,CAAA,QAAA,KAAa,YAAc,EAAA;AACzC,EAAA,kBAAA,CAAmB,WAAc,GAAA,wBAAA,CAAA;AACjC,EAAA,eAAA,CAAgB,WAAc,GAAA,qBAAA,CAAA;AAC9B,EAAA,kBAAA,CAAmB,WAAc,GAAA,wBAAA,CAAA;AACnC;;;;;;"}

package/dist/primitives/{internal/Collapsible → Collapsible}/index.js

@@ -1,29 +1,45 @@
 import { jsx } from 'react/jsx-runtime';
 import { Slot } from '@radix-ui/react-slot';
 import { createContext, forwardRef, useId, useContext, useRef, useEffect, useImperativeHandle } from 'react';
+import { useControllableState } from '../../utils/use-controllable-state.js';
 
-const RootContext = createContext(null);
-const Root = forwardRef(
-  ({ open, onOpenChange, disabled = false, asChild, ...props }, forwardedRef) => {
+const COLLAPSIBLE_ROOT_NAME = "CollapsibleRoot";
+const COLLAPSIBLE_TRIGGER_NAME = "CollapsibleTrigger";
+const COLLAPSIBLE_CONTENT_NAME = "CollapsibleContent";
+const CollapsibleContext = createContext(null);
+const CollapsibleRoot = forwardRef(
+  ({
+    open: controlledOpen,
+    onOpenChange: controlledOnOpenChange,
+    defaultOpen,
+    disabled = false,
+    asChild,
+    ...props
+  }, forwardedRef) => {
+    const [isOpen, onOpenChange] = useControllableState(
+      defaultOpen ?? true,
+      controlledOpen,
+      controlledOnOpenChange
+    );
     const Component = asChild ? Slot : "div";
     const id = useId();
-    return /* @__PURE__ */ jsx(RootContext.Provider, {
-      value: { open, onOpenChange, disabled, contentId: id },
+    return /* @__PURE__ */ jsx(CollapsibleContext.Provider, {
+      value: { open: isOpen, onOpenChange, disabled, contentId: id },
       children: /* @__PURE__ */ jsx(Component, {
         ...props,
         ref: forwardedRef,
-        "data-state": open ? "open" : "closed",
+        "data-state": isOpen ? "open" : "closed",
         "data-disabled": disabled ? "" : void 0
       })
     });
   }
 );
-const Trigger = forwardRef(
+const CollapsibleTrigger = forwardRef(
   ({ onClick, asChild, ...props }, forwardedRef) => {
     const Component = asChild ? Slot : "button";
-    const context = useContext(RootContext);
+    const context = useContext(CollapsibleContext);
     if (!context) {
-      throw new Error("Collapsible.Trigger must be a descendant of Root");
+      throw new Error("Collapsible.Root is missing from the React tree.");
     }
     const { open, disabled, contentId, onOpenChange } = context;
     return /* @__PURE__ */ jsx(Component, {
@@ -46,14 +62,15 @@ const Trigger = forwardRef(
     });
   }
 );
-const Content = forwardRef(
+const CollapsibleContent = forwardRef(
   ({ asChild, ...props }, forwardedRef) => {
     const Component = asChild ? Slot : "div";
-    const rootContext = useContext(RootContext);
+    const context = useContext(CollapsibleContext);
     const divRef = useRef(null);
-    if (!rootContext)
-      throw new Error("Missing RootContext Provider");
-    const { open, onOpenChange, disabled, contentId } = rootContext;
+    if (!context) {
+      throw new Error("Collapsible.Root is missing from the React tree.");
+    }
+    const { open, onOpenChange, disabled, contentId } = context;
     useEffect(() => {
       const element = divRef.current;
       if (element === null)
@@ -96,6 +113,11 @@ const Content = forwardRef(
     });
   }
 );
+if (process.env.NODE_ENV !== "production") {
+  CollapsibleContent.displayName = COLLAPSIBLE_CONTENT_NAME;
+  CollapsibleRoot.displayName = COLLAPSIBLE_ROOT_NAME;
+  CollapsibleTrigger.displayName = COLLAPSIBLE_TRIGGER_NAME;
+}
 
-export { Content, Root, Trigger };
+export { CollapsibleContent as Content, CollapsibleRoot as Root, CollapsibleTrigger as Trigger };
 //# sourceMappingURL=index.js.map

package/dist/primitives/Collapsible/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sources":["../../../src/primitives/Collapsible/index.tsx"],"sourcesContent":["import { Slot } from \"@radix-ui/react-slot\";\nimport {\n  createContext,\n  forwardRef,\n  useContext,\n  useEffect,\n  useId,\n  useImperativeHandle,\n  useRef,\n} from \"react\";\n\nimport { useControllableState } from \"../../utils/use-controllable-state\";\nimport type { ContentProps, RootProps, TriggerProps } from \"./types\";\n\nconst COLLAPSIBLE_ROOT_NAME = \"CollapsibleRoot\";\nconst COLLAPSIBLE_TRIGGER_NAME = \"CollapsibleTrigger\";\nconst COLLAPSIBLE_CONTENT_NAME = \"CollapsibleContent\";\n\nconst CollapsibleContext = createContext<{\n  open?: boolean;\n  onOpenChange: (open: boolean) => void;\n  disabled: boolean;\n  contentId: string;\n} | null>(null);\n\n/* -------------------------------------------------------------------------------------------------\n * Root\n * -----------------------------------------------------------------------------------------------*/\n\nconst CollapsibleRoot = forwardRef<HTMLDivElement, RootProps>(\n  (\n    {\n      open: controlledOpen,\n      onOpenChange: controlledOnOpenChange,\n      defaultOpen,\n      disabled = false,\n      asChild,\n      ...props\n    },\n    forwardedRef\n  ) => {\n    const [isOpen, onOpenChange] = useControllableState(\n      defaultOpen ?? true,\n      controlledOpen,\n      controlledOnOpenChange\n    );\n    const Component = asChild ? Slot : \"div\";\n    const id = useId();\n\n    return (\n      <CollapsibleContext.Provider\n        value={{ open: isOpen, onOpenChange, disabled, contentId: id }}\n      >\n        <Component\n          {...props}\n          ref={forwardedRef}\n          data-state={isOpen ? \"open\" : \"closed\"}\n          data-disabled={disabled ? \"\" : undefined}\n        />\n      </CollapsibleContext.Provider>\n    );\n  }\n);\n\n/* -------------------------------------------------------------------------------------------------\n * Trigger\n * -----------------------------------------------------------------------------------------------*/\n\nconst CollapsibleTrigger = forwardRef<HTMLButtonElement, TriggerProps>(\n  ({ onClick, asChild, ...props }, forwardedRef) => {\n    const Component = asChild ? Slot : \"button\";\n    const context = useContext(CollapsibleContext);\n\n    if (!context) {\n      throw new Error(\"Collapsible.Root is missing from the React tree.\");\n    }\n\n    const { open, disabled, contentId, onOpenChange } = context;\n\n    return (\n      <Component\n        {...props}\n        ref={forwardedRef}\n        type=\"button\"\n        aria-controls={contentId}\n        aria-expanded={open || false}\n        data-state={open ? \"open\" : \"closed\"}\n        data-disabled={disabled ? \"\" : undefined}\n        disabled={disabled}\n        onClick={(event) => {\n          onClick?.(event);\n          if (event.defaultPrevented) return;\n          if (disabled) return;\n          onOpenChange(!open);\n        }}\n      />\n    );\n  }\n);\n\n/* -------------------------------------------------------------------------------------------------\n * Content\n * -----------------------------------------------------------------------------------------------*/\n\nconst CollapsibleContent = forwardRef<HTMLDivElement, ContentProps>(\n  ({ asChild, ...props }, forwardedRef) => {\n    const Component = asChild ? Slot : \"div\";\n    const context = useContext(CollapsibleContext);\n    const divRef = useRef<HTMLDivElement>(null);\n\n    if (!context) {\n      throw new Error(\"Collapsible.Root is missing from the React tree.\");\n    }\n\n    const { open, onOpenChange, disabled, contentId } = context;\n\n    useEffect(() => {\n      const element = divRef.current;\n      if (element === null) return;\n\n      const isHiddenUntilFoundSupported = \"onbeforematch\" in document.body;\n      if (!isHiddenUntilFoundSupported) return;\n\n      function handleBeforeMatch() {\n        onOpenChange(true);\n      }\n\n      // https://developer.chrome.com/articles/hidden-until-found/\n      element.addEventListener(\"beforematch\", handleBeforeMatch);\n      return () => {\n        element.removeEventListener(\"beforematch\", handleBeforeMatch);\n      };\n    }, [onOpenChange]);\n\n    // Passing `string` to `hidden` in JSX is not currently supported: https://github.com/facebook/react/issues/24740\n    useEffect(() => {\n      const element = divRef.current;\n      if (element === null) return;\n\n      if (open) return;\n\n      const isHiddenUntilFoundSupported = \"onbeforematch\" in document.body;\n      if (!isHiddenUntilFoundSupported) return;\n\n      element.setAttribute(\"hidden\", \"until-found\");\n      return () => {\n        element.removeAttribute(\"hidden\");\n      };\n    }, [open]);\n\n    useImperativeHandle<\n      HTMLDivElement | null,\n      HTMLDivElement | null\n    >(forwardedRef, () => {\n      return divRef.current;\n    }, []);\n\n    return (\n      <Component\n        {...props}\n        ref={divRef}\n        data-state={open ? \"open\" : \"closed\"}\n        data-disabled={disabled ? \"\" : undefined}\n        id={contentId}\n        hidden={!open}\n      />\n    );\n  }\n);\n\nif (process.env.NODE_ENV !== \"production\") {\n  CollapsibleContent.displayName = COLLAPSIBLE_CONTENT_NAME;\n  CollapsibleRoot.displayName = COLLAPSIBLE_ROOT_NAME;\n  CollapsibleTrigger.displayName = COLLAPSIBLE_TRIGGER_NAME;\n}\n\n// NOTE: Every export from this file will be available publicly as Collapsible.*\nexport {\n  CollapsibleContent as Content,\n  CollapsibleRoot as Root,\n  CollapsibleTrigger as Trigger,\n};\n"],"names":[],"mappings":";;;;;AAcA,MAAM,qBAAwB,GAAA,iBAAA,CAAA;AAC9B,MAAM,wBAA2B,GAAA,oBAAA,CAAA;AACjC,MAAM,wBAA2B,GAAA,oBAAA,CAAA;AAEjC,MAAM,kBAAA,GAAqB,cAKjB,IAAI,CAAA,CAAA;AAMd,MAAM,eAAkB,GAAA,UAAA;AAAA,EACtB,CACE;AAAA,IACE,IAAM,EAAA,cAAA;AAAA,IACN,YAAc,EAAA,sBAAA;AAAA,IACd,WAAA;AAAA,IACA,QAAW,GAAA,KAAA;AAAA,IACX,OAAA;AAAA,IACG,GAAA,KAAA;AAAA,KAEL,YACG,KAAA;AACH,IAAM,MAAA,CAAC,MAAQ,EAAA,YAAY,CAAI,GAAA,oBAAA;AAAA,MAC7B,WAAe,IAAA,IAAA;AAAA,MACf,cAAA;AAAA,MACA,sBAAA;AAAA,KACF,CAAA;AACA,IAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,KAAA,CAAA;AACnC,IAAA,MAAM,KAAK,KAAM,EAAA,CAAA;AAEjB,IACE,uBAAA,GAAA,CAAC,mBAAmB,QAAnB,EAAA;AAAA,MACC,OAAO,EAAE,IAAA,EAAM,QAAQ,YAAc,EAAA,QAAA,EAAU,WAAW,EAAG,EAAA;AAAA,MAE7D,QAAC,kBAAA,GAAA,CAAA,SAAA,EAAA;AAAA,QACE,GAAG,KAAA;AAAA,QACJ,GAAK,EAAA,YAAA;AAAA,QACL,YAAA,EAAY,SAAS,MAAS,GAAA,QAAA;AAAA,QAC9B,eAAA,EAAe,WAAW,EAAK,GAAA,KAAA,CAAA;AAAA,OACjC,CAAA;AAAA,KACF,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAMA,MAAM,kBAAqB,GAAA,UAAA;AAAA,EACzB,CAAC,EAAE,OAAA,EAAS,OAAY,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AAChD,IAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,QAAA,CAAA;AACnC,IAAM,MAAA,OAAA,GAAU,WAAW,kBAAkB,CAAA,CAAA;AAE7C,IAAA,IAAI,CAAC,OAAS,EAAA;AACZ,MAAM,MAAA,IAAI,MAAM,kDAAkD,CAAA,CAAA;AAAA,KACpE;AAEA,IAAA,MAAM,EAAE,IAAA,EAAM,QAAU,EAAA,SAAA,EAAW,cAAiB,GAAA,OAAA,CAAA;AAEpD,IAAA,uBACG,GAAA,CAAA,SAAA,EAAA;AAAA,MACE,GAAG,KAAA;AAAA,MACJ,GAAK,EAAA,YAAA;AAAA,MACL,IAAK,EAAA,QAAA;AAAA,MACL,eAAe,EAAA,SAAA;AAAA,MACf,iBAAe,IAAQ,IAAA,KAAA;AAAA,MACvB,YAAA,EAAY,OAAO,MAAS,GAAA,QAAA;AAAA,MAC5B,eAAA,EAAe,WAAW,EAAK,GAAA,KAAA,CAAA;AAAA,MAC/B,QAAA;AAAA,MACA,OAAA,EAAS,CAAC,KAAU,KAAA;AAClB,QAAA,OAAA,GAAU,KAAK,CAAA,CAAA;AACf,QAAA,IAAI,KAAM,CAAA,gBAAA;AAAkB,UAAA,OAAA;AAC5B,QAAI,IAAA,QAAA;AAAU,UAAA,OAAA;AACd,QAAA,YAAA,CAAa,CAAC,IAAI,CAAA,CAAA;AAAA,OACpB;AAAA,KACF,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAMA,MAAM,kBAAqB,GAAA,UAAA;AAAA,EACzB,CAAC,EAAE,OAAY,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AACvC,IAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,KAAA,CAAA;AACnC,IAAM,MAAA,OAAA,GAAU,WAAW,kBAAkB,CAAA,CAAA;AAC7C,IAAM,MAAA,MAAA,GAAS,OAAuB,IAAI,CAAA,CAAA;AAE1C,IAAA,IAAI,CAAC,OAAS,EAAA;AACZ,MAAM,MAAA,IAAI,MAAM,kDAAkD,CAAA,CAAA;AAAA,KACpE;AAEA,IAAA,MAAM,EAAE,IAAA,EAAM,YAAc,EAAA,QAAA,EAAU,WAAc,GAAA,OAAA,CAAA;AAEpD,IAAA,SAAA,CAAU,MAAM;AACd,MAAA,MAAM,UAAU,MAAO,CAAA,OAAA,CAAA;AACvB,MAAA,IAAI,OAAY,KAAA,IAAA;AAAM,QAAA,OAAA;AAEtB,MAAM,MAAA,2BAAA,GAA8B,mBAAmB,QAAS,CAAA,IAAA,CAAA;AAChE,MAAA,IAAI,CAAC,2BAAA;AAA6B,QAAA,OAAA;AAElC,MAAA,SAAS,iBAAoB,GAAA;AAC3B,QAAA,YAAA,CAAa,IAAI,CAAA,CAAA;AAAA,OACnB;AAGA,MAAQ,OAAA,CAAA,gBAAA,CAAiB,eAAe,iBAAiB,CAAA,CAAA;AACzD,MAAA,OAAO,MAAM;AACX,QAAQ,OAAA,CAAA,mBAAA,CAAoB,eAAe,iBAAiB,CAAA,CAAA;AAAA,OAC9D,CAAA;AAAA,KACF,EAAG,CAAC,YAAY,CAAC,CAAA,CAAA;AAGjB,IAAA,SAAA,CAAU,MAAM;AACd,MAAA,MAAM,UAAU,MAAO,CAAA,OAAA,CAAA;AACvB,MAAA,IAAI,OAAY,KAAA,IAAA;AAAM,QAAA,OAAA;AAEtB,MAAI,IAAA,IAAA;AAAM,QAAA,OAAA;AAEV,MAAM,MAAA,2BAAA,GAA8B,mBAAmB,QAAS,CAAA,IAAA,CAAA;AAChE,MAAA,IAAI,CAAC,2BAAA;AAA6B,QAAA,OAAA;AAElC,MAAQ,OAAA,CAAA,YAAA,CAAa,UAAU,aAAa,CAAA,CAAA;AAC5C,MAAA,OAAO,MAAM;AACX,QAAA,OAAA,CAAQ,gBAAgB,QAAQ,CAAA,CAAA;AAAA,OAClC,CAAA;AAAA,KACF,EAAG,CAAC,IAAI,CAAC,CAAA,CAAA;AAET,IAAA,mBAAA,CAGE,cAAc,MAAM;AACpB,MAAA,OAAO,MAAO,CAAA,OAAA,CAAA;AAAA,KAChB,EAAG,EAAE,CAAA,CAAA;AAEL,IAAA,uBACG,GAAA,CAAA,SAAA,EAAA;AAAA,MACE,GAAG,KAAA;AAAA,MACJ,GAAK,EAAA,MAAA;AAAA,MACL,YAAA,EAAY,OAAO,MAAS,GAAA,QAAA;AAAA,MAC5B,eAAA,EAAe,WAAW,EAAK,GAAA,KAAA,CAAA;AAAA,MAC/B,EAAI,EAAA,SAAA;AAAA,MACJ,QAAQ,CAAC,IAAA;AAAA,KACX,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAEA,IAAI,OAAA,CAAQ,GAAI,CAAA,QAAA,KAAa,YAAc,EAAA;AACzC,EAAA,kBAAA,CAAmB,WAAc,GAAA,wBAAA,CAAA;AACjC,EAAA,eAAA,CAAgB,WAAc,GAAA,qBAAA,CAAA;AAC9B,EAAA,kBAAA,CAAmB,WAAc,GAAA,wBAAA,CAAA;AACnC;;;;"}

package/dist/primitives/Comment/index.cjs

@@ -32,11 +32,11 @@ const CommentLink = react.forwardRef(
   }
 );
 const defaultBodyComponents = {
-  Mention: ({ userId }) => {
+  Mention: ({ mention }) => {
     return /* @__PURE__ */ jsxRuntime.jsxs(CommentMention, {
       children: [
         constants.MENTION_CHARACTER,
-        userId
+        mention.id
       ]
     });
   },
@@ -68,8 +68,9 @@ const CommentBody = react.forwardRef(
               style: { minHeight: "1lh" },
               children: block.children.map((inline, index2) => {
                 if (utils.isCommentBodyMention(inline)) {
-                  return inline.id ? /* @__PURE__ */ jsxRuntime.jsx(Mention, {
-                    userId: inline.id
+                  const { type: _, ...mention } = inline;
+                  return mention.id ? /* @__PURE__ */ jsxRuntime.jsx(Mention, {
+                    mention
                   }, index2) : null;
                 }
                 if (utils.isCommentBodyLink(inline)) {