@assistant-ui/react-markdown 0.7.13 → 0.7.15

package/dist/overrides/defaultComponents.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/overrides/defaultComponents.tsx"],"sourcesContent":["import type { ComponentType, ReactNode } from \"react\";\nimport { Element } from \"hast\";\nimport { PreComponent, CodeComponent, CodeHeaderProps } from \"./types\";\n\nexport const DefaultPre: PreComponent = ({ node, ...rest }) => (\n  <pre {...rest} />\n);\n\nexport const DefaultCode: CodeComponent = ({ node, ...rest }) => (\n  <code {...rest} />\n);\n\nexport const DefaultCodeBlockContent: ComponentType<{\n  components: { Pre: PreComponent; Code: CodeComponent };\n  code: string | ReactNode | undefined;\n  node: Element | undefined;\n}> = ({ components: { Pre, Code }, code, node }) => (\n  <Pre node={node}>\n    <Code node={node}>{code}</Code>\n  </Pre>\n);\n\nexport const DefaultCodeHeader: ComponentType<CodeHeaderProps> = () => null;\n"],"mappings":";AAKE;AADK,IAAM,aAA2B,CAAC,EAAE,MAAM,GAAG,KAAK,MACvD,oBAAC,SAAK,GAAG,MAAM;AAGV,IAAM,cAA6B,CAAC,EAAE,MAAM,GAAG,KAAK,MACzD,oBAAC,UAAM,GAAG,MAAM;AAGX,IAAM,0BAIR,CAAC,EAAE,YAAY,EAAE,KAAK,KAAK,GAAG,MAAM,KAAK,MAC5C,oBAAC,OAAI,MACH,8BAAC,QAAK,MAAa,gBAAK,GAC1B;AAGK,IAAM,oBAAoD,MAAM;","names":[]}
+{"version":3,"sources":["../../src/overrides/defaultComponents.tsx"],"sourcesContent":["import type { ComponentType, ReactNode } from \"react\";\nimport { PreComponent, CodeComponent, CodeHeaderProps } from \"./types\";\n\nexport const DefaultPre: PreComponent = ({ node, ...rest }) => (\n  <pre {...rest} />\n);\n\nexport const DefaultCode: CodeComponent = ({ node, ...rest }) => (\n  <code {...rest} />\n);\n\nexport const DefaultCodeBlockContent: ComponentType<{\n  components: { Pre: PreComponent; Code: CodeComponent };\n  code: string | ReactNode | undefined;\n}> = ({ components: { Pre, Code }, code }) => (\n  <Pre>\n    <Code>{code}</Code>\n  </Pre>\n);\n\nexport const DefaultCodeHeader: ComponentType<CodeHeaderProps> = () => null;\n"],"mappings":";AAIE;AADK,IAAM,aAA2B,CAAC,EAAE,MAAM,GAAG,KAAK,MACvD,oBAAC,SAAK,GAAG,MAAM;AAGV,IAAM,cAA6B,CAAC,EAAE,MAAM,GAAG,KAAK,MACzD,oBAAC,UAAM,GAAG,MAAM;AAGX,IAAM,0BAGR,CAAC,EAAE,YAAY,EAAE,KAAK,KAAK,GAAG,KAAK,MACtC,oBAAC,OACC,8BAAC,QAAM,gBAAK,GACd;AAGK,IAAM,oBAAoD,MAAM;","names":[]}

package/dist/primitives/MarkdownText.d.ts

@@ -1,8 +1,7 @@
-import { ComponentRef, ElementType, ForwardRefExoticComponent, RefAttributes, type ComponentPropsWithoutRef, type ComponentType } from "react";
+import { ElementType, ForwardRefExoticComponent, RefAttributes, type ComponentPropsWithoutRef, type ComponentType } from "react";
 import { type Options } from "react-markdown";
 import { SyntaxHighlighterProps, CodeHeaderProps } from "../overrides/types";
 import { Primitive } from "@radix-ui/react-primitive";
-type MarkdownTextPrimitiveElement = ComponentRef<typeof Primitive.div>;
 type PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;
 export type MarkdownTextPrimitiveProps = Omit<Options, "components" | "children"> & {
     containerProps?: Omit<PrimitiveDivProps, "children" | "asChild"> | undefined;
@@ -21,6 +20,6 @@ export type MarkdownTextPrimitiveProps = Omit<Options, "components" | "children"
     }> | undefined;
     smooth?: boolean | undefined;
 };
-export declare const MarkdownTextPrimitive: ForwardRefExoticComponent<MarkdownTextPrimitiveProps> & RefAttributes<MarkdownTextPrimitiveElement>;
+export declare const MarkdownTextPrimitive: ForwardRefExoticComponent<MarkdownTextPrimitiveProps> & RefAttributes<HTMLDivElement>;
 export {};
 //# sourceMappingURL=MarkdownText.d.ts.map

package/dist/primitives/MarkdownText.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MarkdownText.d.ts","sourceRoot":"","sources":["../../src/primitives/MarkdownText.tsx"],"names":[],"mappings":"AAGA,OAAO,EACL,YAAY,EACZ,WAAW,EAEX,yBAAyB,EACzB,aAAa,EACb,KAAK,wBAAwB,EAC7B,KAAK,aAAa,EACnB,MAAM,OAAO,CAAC;AACf,OAAsB,EAAE,KAAK,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAC7D,OAAO,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAU7E,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAKtD,KAAK,4BAA4B,GAAG,YAAY,CAAC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;AACvE,KAAK,iBAAiB,GAAG,wBAAwB,CAAC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;AAExE,MAAM,MAAM,0BAA0B,GAAG,IAAI,CAC3C,OAAO,EACP,YAAY,GAAG,UAAU,CAC1B,GAAG;IACF,cAAc,CAAC,EAAE,IAAI,CAAC,iBAAiB,EAAE,UAAU,GAAG,SAAS,CAAC,GAAG,SAAS,CAAC;IAC7E,kBAAkB,CAAC,EAAE,WAAW,GAAG,SAAS,CAAC;IAC7C,UAAU,CAAC,EACP,CAAC,WAAW,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,GAAG;QACpC,iBAAiB,CAAC,EAAE,aAAa,CAAC,sBAAsB,CAAC,GAAG,SAAS,CAAC;QACtE,UAAU,CAAC,EAAE,aAAa,CAAC,eAAe,CAAC,GAAG,SAAS,CAAC;QACxD;;YAEI;QACJ,WAAW,CAAC,EAAE,SAAS,CAAC;KACzB,CAAC,GACF,SAAS,CAAC;IACd,oBAAoB,CAAC,EACjB,MAAM,CACJ,MAAM,EACN;QACE,UAAU,CAAC,EAAE,aAAa,CAAC,eAAe,CAAC,GAAG,SAAS,CAAC;QACxD,iBAAiB,CAAC,EAAE,aAAa,CAAC,sBAAsB,CAAC,GAAG,SAAS,CAAC;KACvE,CACF,GACD,SAAS,CAAC;IACd,MAAM,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CAC9B,CAAC;AAEF,eAAO,MAAM,qBAAqB,EAAE,yBAAyB,CAAC,0BAA0B,CAAC,GACvF,aAAa,CAAC,4BAA4B,CAwD3C,CAAC"}
+{"version":3,"file":"MarkdownText.d.ts","sourceRoot":"","sources":["../../src/primitives/MarkdownText.tsx"],"names":[],"mappings":"AAGA,OAAO,EAEL,WAAW,EAGX,yBAAyB,EACzB,aAAa,EAEb,KAAK,wBAAwB,EAC7B,KAAK,aAAa,EACnB,MAAM,OAAO,CAAC;AACf,OAAsB,EAAE,KAAK,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAC7D,OAAO,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAU7E,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAMtD,KAAK,iBAAiB,GAAG,wBAAwB,CAAC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;AAExE,MAAM,MAAM,0BAA0B,GAAG,IAAI,CAC3C,OAAO,EACP,YAAY,GAAG,UAAU,CAC1B,GAAG;IACF,cAAc,CAAC,EAAE,IAAI,CAAC,iBAAiB,EAAE,UAAU,GAAG,SAAS,CAAC,GAAG,SAAS,CAAC;IAC7E,kBAAkB,CAAC,EAAE,WAAW,GAAG,SAAS,CAAC;IAC7C,UAAU,CAAC,EACP,CAAC,WAAW,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,GAAG;QACpC,iBAAiB,CAAC,EAAE,aAAa,CAAC,sBAAsB,CAAC,GAAG,SAAS,CAAC;QACtE,UAAU,CAAC,EAAE,aAAa,CAAC,eAAe,CAAC,GAAG,SAAS,CAAC;QACxD;;YAEI;QACJ,WAAW,CAAC,EAAE,SAAS,CAAC;KACzB,CAAC,GACF,SAAS,CAAC;IACd,oBAAoB,CAAC,EACjB,MAAM,CACJ,MAAM,EACN;QACE,UAAU,CAAC,EAAE,aAAa,CAAC,eAAe,CAAC,GAAG,SAAS,CAAC;QACxD,iBAAiB,CAAC,EAAE,aAAa,CAAC,sBAAsB,CAAC,GAAG,SAAS,CAAC;KACvE,CACF,GACD,SAAS,CAAC;IACd,MAAM,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CAC9B,CAAC;AAqFF,eAAO,MAAM,qBAAqB,uFAEjC,CAAC"}

package/dist/primitives/MarkdownText.js

@@ -43,43 +43,61 @@ var import_react_use_callback_ref = require("@radix-ui/react-use-callback-ref");
 var import_CodeOverride = require("../overrides/CodeOverride.js");
 var import_classnames = __toESM(require("classnames"));
 var import_jsx_runtime = require("react/jsx-runtime");
-var { useSmooth } = import_react.INTERNAL;
-var MarkdownTextPrimitive = (0, import_react2.forwardRef)(
-  ({
-    components: userComponents,
-    componentsByLanguage = userComponents?.by_language,
-    className,
-    containerProps,
-    containerComponent: Container = "div",
-    smooth = true,
-    ...rest
-  }, forwardedRef) => {
-    const { text, status } = useSmooth((0, import_react.useContentPartText)(), smooth);
+var { useSmooth, useSmoothStatus, withSmoothContextProvider } = import_react.INTERNAL;
+var MarkdownTextInner = ({
+  components: userComponents,
+  componentsByLanguage = userComponents?.by_language,
+  smooth = true,
+  ...rest
+}) => {
+  const { text } = useSmooth((0, import_react.useContentPartText)(), smooth);
+  const {
+    pre = import_defaultComponents.DefaultPre,
+    code = import_defaultComponents.DefaultCode,
+    SyntaxHighlighter = import_defaultComponents.DefaultCodeBlockContent,
+    CodeHeader = import_defaultComponents.DefaultCodeHeader
+  } = userComponents ?? {};
+  const useCodeOverrideComponents = (0, import_react2.useMemo)(() => {
+    return {
+      Pre: pre,
+      Code: code,
+      SyntaxHighlighter,
+      CodeHeader
+    };
+  }, [pre, code, SyntaxHighlighter, CodeHeader]);
+  const CodeComponent = (0, import_react_use_callback_ref.useCallbackRef)((props) => /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
+    import_CodeOverride.CodeOverride,
+    {
+      components: useCodeOverrideComponents,
+      componentsByLanguage,
+      ...props
+    }
+  ));
+  const components = (0, import_react2.useMemo)(() => {
     const {
-      pre = import_defaultComponents.DefaultPre,
-      code = import_defaultComponents.DefaultCode,
-      SyntaxHighlighter = import_defaultComponents.DefaultCodeBlockContent,
-      CodeHeader = import_defaultComponents.DefaultCodeHeader,
+      pre: pre2 = import_defaultComponents.DefaultPre,
+      code: code2 = import_defaultComponents.DefaultCode,
+      SyntaxHighlighter: SyntaxHighlighter2 = import_defaultComponents.DefaultCodeBlockContent,
+      CodeHeader: CodeHeader2 = import_defaultComponents.DefaultCodeHeader,
       by_language,
       ...componentsRest
     } = userComponents ?? {};
-    const components = {
+    return {
       ...componentsRest,
       pre: import_PreOverride.PreOverride,
-      code: (0, import_react_use_callback_ref.useCallbackRef)((props) => /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
-        import_CodeOverride.CodeOverride,
-        {
-          components: {
-            Pre: pre,
-            Code: code,
-            SyntaxHighlighter,
-            CodeHeader
-          },
-          componentsByLanguage,
-          ...props
-        }
-      ))
+      code: CodeComponent
     };
+  }, [CodeComponent, userComponents, componentsByLanguage]);
+  return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(import_react_markdown.default, { components, ...rest, children: text });
+};
+var MarkdownTextPrimitiveImpl = (0, import_react2.forwardRef)(
+  ({
+    className,
+    containerProps,
+    containerComponent: Container = "div",
+    ...rest
+  }, forwardedRef) => {
+    const status = useSmoothStatus();
     return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
       Container,
       {
@@ -87,12 +105,15 @@ var MarkdownTextPrimitive = (0, import_react2.forwardRef)(
         ...containerProps,
         className: (0, import_classnames.default)(className, containerProps?.className),
         ref: forwardedRef,
-        children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(import_react_markdown.default, { components, ...rest, children: text })
+        children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(MarkdownTextInner, { ...rest })
       }
     );
   }
 );
-MarkdownTextPrimitive.displayName = "MarkdownTextPrimitive";
+MarkdownTextPrimitiveImpl.displayName = "MarkdownTextPrimitive";
+var MarkdownTextPrimitive = withSmoothContextProvider(
+  MarkdownTextPrimitiveImpl
+);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   MarkdownTextPrimitive

package/dist/primitives/MarkdownText.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/primitives/MarkdownText.tsx"],"sourcesContent":["\"use client\";\n\nimport { INTERNAL, useContentPartText } from \"@assistant-ui/react\";\nimport {\n  ComponentRef,\n  ElementType,\n  forwardRef,\n  ForwardRefExoticComponent,\n  RefAttributes,\n  type ComponentPropsWithoutRef,\n  type ComponentType,\n} from \"react\";\nimport ReactMarkdown, { type Options } from \"react-markdown\";\nimport { SyntaxHighlighterProps, CodeHeaderProps } from \"../overrides/types\";\nimport { PreOverride } from \"../overrides/PreOverride\";\nimport {\n  DefaultPre,\n  DefaultCode,\n  DefaultCodeBlockContent,\n  DefaultCodeHeader,\n} from \"../overrides/defaultComponents\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { CodeOverride } from \"../overrides/CodeOverride\";\nimport { Primitive } from \"@radix-ui/react-primitive\";\nimport classNames from \"classnames\";\n\nconst { useSmooth } = INTERNAL;\n\ntype MarkdownTextPrimitiveElement = ComponentRef<typeof Primitive.div>;\ntype PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;\n\nexport type MarkdownTextPrimitiveProps = Omit<\n  Options,\n  \"components\" | \"children\"\n> & {\n  containerProps?: Omit<PrimitiveDivProps, \"children\" | \"asChild\"> | undefined;\n  containerComponent?: ElementType | undefined;\n  components?:\n    | (NonNullable<Options[\"components\"]> & {\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps> | undefined;\n        CodeHeader?: ComponentType<CodeHeaderProps> | undefined;\n        /**\n         * @deprecated Use `componentsByLanguage` instead of `components.by_language`. This will be removed in the next major version.\n         **/\n        by_language?: undefined;\n      })\n    | undefined;\n  componentsByLanguage?:\n    | Record<\n        string,\n        {\n          CodeHeader?: ComponentType<CodeHeaderProps> | undefined;\n          SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps> | undefined;\n        }\n      >\n    | undefined;\n  smooth?: boolean | undefined;\n};\n\nexport const MarkdownTextPrimitive: ForwardRefExoticComponent<MarkdownTextPrimitiveProps> &\n  RefAttributes<MarkdownTextPrimitiveElement> = forwardRef<\n  MarkdownTextPrimitiveElement,\n  MarkdownTextPrimitiveProps\n>(\n  (\n    {\n      components: userComponents,\n      componentsByLanguage = userComponents?.by_language,\n      className,\n      containerProps,\n      containerComponent: Container = \"div\",\n      smooth = true,\n      ...rest\n    },\n    forwardedRef,\n  ) => {\n    const { text, status } = useSmooth(useContentPartText(), smooth);\n\n    const {\n      pre = DefaultPre,\n      code = DefaultCode,\n      SyntaxHighlighter = DefaultCodeBlockContent,\n      CodeHeader = DefaultCodeHeader,\n      by_language,\n      ...componentsRest\n    } = userComponents ?? {};\n    const components: Options[\"components\"] = {\n      ...componentsRest,\n      pre: PreOverride,\n      code: useCallbackRef((props) => (\n        <CodeOverride\n          components={{\n            Pre: pre,\n            Code: code,\n            SyntaxHighlighter,\n            CodeHeader,\n          }}\n          componentsByLanguage={componentsByLanguage}\n          {...props}\n        />\n      )),\n    };\n\n    return (\n      <Container\n        data-status={status.type}\n        {...containerProps}\n        className={classNames(className, containerProps?.className)}\n        ref={forwardedRef}\n      >\n        <ReactMarkdown components={components} {...rest}>\n          {text}\n        </ReactMarkdown>\n      </Container>\n    );\n  },\n);\n\nMarkdownTextPrimitive.displayName = \"MarkdownTextPrimitive\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAEA,mBAA6C;AAC7C,IAAAA,gBAQO;AACP,4BAA4C;AAE5C,yBAA4B;AAC5B,+BAKO;AACP,oCAA+B;AAC/B,0BAA6B;AAE7B,wBAAuB;AAkEf;AAhER,IAAM,EAAE,UAAU,IAAI;AAiCf,IAAM,4BACmC;AAAA,EAI9C,CACE;AAAA,IACE,YAAY;AAAA,IACZ,uBAAuB,gBAAgB;AAAA,IACvC;AAAA,IACA;AAAA,IACA,oBAAoB,YAAY;AAAA,IAChC,SAAS;AAAA,IACT,GAAG;AAAA,EACL,GACA,iBACG;AACH,UAAM,EAAE,MAAM,OAAO,IAAI,cAAU,iCAAmB,GAAG,MAAM;AAE/D,UAAM;AAAA,MACJ,MAAM;AAAA,MACN,OAAO;AAAA,MACP,oBAAoB;AAAA,MACpB,aAAa;AAAA,MACb;AAAA,MACA,GAAG;AAAA,IACL,IAAI,kBAAkB,CAAC;AACvB,UAAM,aAAoC;AAAA,MACxC,GAAG;AAAA,MACH,KAAK;AAAA,MACL,UAAM,8CAAe,CAAC,UACpB;AAAA,QAAC;AAAA;AAAA,UACC,YAAY;AAAA,YACV,KAAK;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF;AAAA,UACA;AAAA,UACC,GAAG;AAAA;AAAA,MACN,CACD;AAAA,IACH;AAEA,WACE;AAAA,MAAC;AAAA;AAAA,QACC,eAAa,OAAO;AAAA,QACnB,GAAG;AAAA,QACJ,eAAW,kBAAAC,SAAW,WAAW,gBAAgB,SAAS;AAAA,QAC1D,KAAK;AAAA,QAEL,sDAAC,sBAAAC,SAAA,EAAc,YAAyB,GAAG,MACxC,gBACH;AAAA;AAAA,IACF;AAAA,EAEJ;AACF;AAEA,sBAAsB,cAAc;","names":["import_react","classNames","ReactMarkdown"]}
+{"version":3,"sources":["../../src/primitives/MarkdownText.tsx"],"sourcesContent":["\"use client\";\n\nimport { INTERNAL, useContentPartText } from \"@assistant-ui/react\";\nimport {\n  ComponentRef,\n  ElementType,\n  FC,\n  forwardRef,\n  ForwardRefExoticComponent,\n  RefAttributes,\n  useMemo,\n  type ComponentPropsWithoutRef,\n  type ComponentType,\n} from \"react\";\nimport ReactMarkdown, { type Options } from \"react-markdown\";\nimport { SyntaxHighlighterProps, CodeHeaderProps } from \"../overrides/types\";\nimport { PreOverride } from \"../overrides/PreOverride\";\nimport {\n  DefaultPre,\n  DefaultCode,\n  DefaultCodeBlockContent,\n  DefaultCodeHeader,\n} from \"../overrides/defaultComponents\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { CodeOverride } from \"../overrides/CodeOverride\";\nimport { Primitive } from \"@radix-ui/react-primitive\";\nimport classNames from \"classnames\";\n\nconst { useSmooth, useSmoothStatus, withSmoothContextProvider } = INTERNAL;\n\ntype MarkdownTextPrimitiveElement = ComponentRef<typeof Primitive.div>;\ntype PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;\n\nexport type MarkdownTextPrimitiveProps = Omit<\n  Options,\n  \"components\" | \"children\"\n> & {\n  containerProps?: Omit<PrimitiveDivProps, \"children\" | \"asChild\"> | undefined;\n  containerComponent?: ElementType | undefined;\n  components?:\n    | (NonNullable<Options[\"components\"]> & {\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps> | undefined;\n        CodeHeader?: ComponentType<CodeHeaderProps> | undefined;\n        /**\n         * @deprecated Use `componentsByLanguage` instead of `components.by_language`. This will be removed in the next major version.\n         **/\n        by_language?: undefined;\n      })\n    | undefined;\n  componentsByLanguage?:\n    | Record<\n        string,\n        {\n          CodeHeader?: ComponentType<CodeHeaderProps> | undefined;\n          SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps> | undefined;\n        }\n      >\n    | undefined;\n  smooth?: boolean | undefined;\n};\n\nconst MarkdownTextInner: FC<MarkdownTextPrimitiveProps> = ({\n  components: userComponents,\n  componentsByLanguage = userComponents?.by_language,\n  smooth = true,\n  ...rest\n}) => {\n  const { text } = useSmooth(useContentPartText(), smooth);\n\n  const {\n    pre = DefaultPre,\n    code = DefaultCode,\n    SyntaxHighlighter = DefaultCodeBlockContent,\n    CodeHeader = DefaultCodeHeader,\n  } = userComponents ?? {};\n  const useCodeOverrideComponents = useMemo(() => {\n    return {\n      Pre: pre,\n      Code: code,\n      SyntaxHighlighter,\n      CodeHeader,\n    };\n  }, [pre, code, SyntaxHighlighter, CodeHeader]);\n  const CodeComponent = useCallbackRef((props) => (\n    <CodeOverride\n      components={useCodeOverrideComponents}\n      componentsByLanguage={componentsByLanguage}\n      {...props}\n    />\n  ));\n\n  const components: Options[\"components\"] = useMemo(() => {\n    const {\n      pre = DefaultPre,\n      code = DefaultCode,\n      SyntaxHighlighter = DefaultCodeBlockContent,\n      CodeHeader = DefaultCodeHeader,\n      by_language,\n      ...componentsRest\n    } = userComponents ?? {};\n    return {\n      ...componentsRest,\n      pre: PreOverride,\n      code: CodeComponent,\n    };\n  }, [CodeComponent, userComponents, componentsByLanguage]);\n\n  return (\n    <ReactMarkdown components={components} {...rest}>\n      {text}\n    </ReactMarkdown>\n  );\n};\n\nconst MarkdownTextPrimitiveImpl: ForwardRefExoticComponent<MarkdownTextPrimitiveProps> &\n  RefAttributes<MarkdownTextPrimitiveElement> = forwardRef<\n  MarkdownTextPrimitiveElement,\n  MarkdownTextPrimitiveProps\n>(\n  (\n    {\n      className,\n      containerProps,\n      containerComponent: Container = \"div\",\n      ...rest\n    },\n    forwardedRef,\n  ) => {\n    const status = useSmoothStatus();\n    return (\n      <Container\n        data-status={status.type}\n        {...containerProps}\n        className={classNames(className, containerProps?.className)}\n        ref={forwardedRef}\n      >\n        <MarkdownTextInner {...rest}></MarkdownTextInner>\n      </Container>\n    );\n  },\n);\n\nMarkdownTextPrimitiveImpl.displayName = \"MarkdownTextPrimitive\";\n\nexport const MarkdownTextPrimitive = withSmoothContextProvider(\n  MarkdownTextPrimitiveImpl,\n);\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAEA,mBAA6C;AAC7C,IAAAA,gBAUO;AACP,4BAA4C;AAE5C,yBAA4B;AAC5B,+BAKO;AACP,oCAA+B;AAC/B,0BAA6B;AAE7B,wBAAuB;AA0DnB;AAxDJ,IAAM,EAAE,WAAW,iBAAiB,0BAA0B,IAAI;AAiClE,IAAM,oBAAoD,CAAC;AAAA,EACzD,YAAY;AAAA,EACZ,uBAAuB,gBAAgB;AAAA,EACvC,SAAS;AAAA,EACT,GAAG;AACL,MAAM;AACJ,QAAM,EAAE,KAAK,IAAI,cAAU,iCAAmB,GAAG,MAAM;AAEvD,QAAM;AAAA,IACJ,MAAM;AAAA,IACN,OAAO;AAAA,IACP,oBAAoB;AAAA,IACpB,aAAa;AAAA,EACf,IAAI,kBAAkB,CAAC;AACvB,QAAM,gCAA4B,uBAAQ,MAAM;AAC9C,WAAO;AAAA,MACL,KAAK;AAAA,MACL,MAAM;AAAA,MACN;AAAA,MACA;AAAA,IACF;AAAA,EACF,GAAG,CAAC,KAAK,MAAM,mBAAmB,UAAU,CAAC;AAC7C,QAAM,oBAAgB,8CAAe,CAAC,UACpC;AAAA,IAAC;AAAA;AAAA,MACC,YAAY;AAAA,MACZ;AAAA,MACC,GAAG;AAAA;AAAA,EACN,CACD;AAED,QAAM,iBAAoC,uBAAQ,MAAM;AACtD,UAAM;AAAA,MACJ,KAAAC,OAAM;AAAA,MACN,MAAAC,QAAO;AAAA,MACP,mBAAAC,qBAAoB;AAAA,MACpB,YAAAC,cAAa;AAAA,MACb;AAAA,MACA,GAAG;AAAA,IACL,IAAI,kBAAkB,CAAC;AACvB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,KAAK;AAAA,MACL,MAAM;AAAA,IACR;AAAA,EACF,GAAG,CAAC,eAAe,gBAAgB,oBAAoB,CAAC;AAExD,SACE,4CAAC,sBAAAC,SAAA,EAAc,YAAyB,GAAG,MACxC,gBACH;AAEJ;AAEA,IAAM,gCAC0C;AAAA,EAI9C,CACE;AAAA,IACE;AAAA,IACA;AAAA,IACA,oBAAoB,YAAY;AAAA,IAChC,GAAG;AAAA,EACL,GACA,iBACG;AACH,UAAM,SAAS,gBAAgB;AAC/B,WACE;AAAA,MAAC;AAAA;AAAA,QACC,eAAa,OAAO;AAAA,QACnB,GAAG;AAAA,QACJ,eAAW,kBAAAC,SAAW,WAAW,gBAAgB,SAAS;AAAA,QAC1D,KAAK;AAAA,QAEL,sDAAC,qBAAmB,GAAG,MAAM;AAAA;AAAA,IAC/B;AAAA,EAEJ;AACF;AAEA,0BAA0B,cAAc;AAEjC,IAAM,wBAAwB;AAAA,EACnC;AACF;","names":["import_react","pre","code","SyntaxHighlighter","CodeHeader","ReactMarkdown","classNames"]}

package/dist/primitives/MarkdownText.mjs

@@ -3,7 +3,8 @@
 // src/primitives/MarkdownText.tsx
 import { INTERNAL, useContentPartText } from "@assistant-ui/react";
 import {
-  forwardRef
+  forwardRef,
+  useMemo
 } from "react";
 import ReactMarkdown from "react-markdown";
 import { PreOverride } from "../overrides/PreOverride.mjs";
@@ -17,43 +18,61 @@ import { useCallbackRef } from "@radix-ui/react-use-callback-ref";
 import { CodeOverride } from "../overrides/CodeOverride.mjs";
 import classNames from "classnames";
 import { jsx } from "react/jsx-runtime";
-var { useSmooth } = INTERNAL;
-var MarkdownTextPrimitive = forwardRef(
-  ({
-    components: userComponents,
-    componentsByLanguage = userComponents?.by_language,
-    className,
-    containerProps,
-    containerComponent: Container = "div",
-    smooth = true,
-    ...rest
-  }, forwardedRef) => {
-    const { text, status } = useSmooth(useContentPartText(), smooth);
+var { useSmooth, useSmoothStatus, withSmoothContextProvider } = INTERNAL;
+var MarkdownTextInner = ({
+  components: userComponents,
+  componentsByLanguage = userComponents?.by_language,
+  smooth = true,
+  ...rest
+}) => {
+  const { text } = useSmooth(useContentPartText(), smooth);
+  const {
+    pre = DefaultPre,
+    code = DefaultCode,
+    SyntaxHighlighter = DefaultCodeBlockContent,
+    CodeHeader = DefaultCodeHeader
+  } = userComponents ?? {};
+  const useCodeOverrideComponents = useMemo(() => {
+    return {
+      Pre: pre,
+      Code: code,
+      SyntaxHighlighter,
+      CodeHeader
+    };
+  }, [pre, code, SyntaxHighlighter, CodeHeader]);
+  const CodeComponent = useCallbackRef((props) => /* @__PURE__ */ jsx(
+    CodeOverride,
+    {
+      components: useCodeOverrideComponents,
+      componentsByLanguage,
+      ...props
+    }
+  ));
+  const components = useMemo(() => {
     const {
-      pre = DefaultPre,
-      code = DefaultCode,
-      SyntaxHighlighter = DefaultCodeBlockContent,
-      CodeHeader = DefaultCodeHeader,
+      pre: pre2 = DefaultPre,
+      code: code2 = DefaultCode,
+      SyntaxHighlighter: SyntaxHighlighter2 = DefaultCodeBlockContent,
+      CodeHeader: CodeHeader2 = DefaultCodeHeader,
       by_language,
       ...componentsRest
     } = userComponents ?? {};
-    const components = {
+    return {
       ...componentsRest,
       pre: PreOverride,
-      code: useCallbackRef((props) => /* @__PURE__ */ jsx(
-        CodeOverride,
-        {
-          components: {
-            Pre: pre,
-            Code: code,
-            SyntaxHighlighter,
-            CodeHeader
-          },
-          componentsByLanguage,
-          ...props
-        }
-      ))
+      code: CodeComponent
     };
+  }, [CodeComponent, userComponents, componentsByLanguage]);
+  return /* @__PURE__ */ jsx(ReactMarkdown, { components, ...rest, children: text });
+};
+var MarkdownTextPrimitiveImpl = forwardRef(
+  ({
+    className,
+    containerProps,
+    containerComponent: Container = "div",
+    ...rest
+  }, forwardedRef) => {
+    const status = useSmoothStatus();
     return /* @__PURE__ */ jsx(
       Container,
       {
@@ -61,12 +80,15 @@ var MarkdownTextPrimitive = forwardRef(
         ...containerProps,
         className: classNames(className, containerProps?.className),
         ref: forwardedRef,
-        children: /* @__PURE__ */ jsx(ReactMarkdown, { components, ...rest, children: text })
+        children: /* @__PURE__ */ jsx(MarkdownTextInner, { ...rest })
       }
     );
   }
 );
-MarkdownTextPrimitive.displayName = "MarkdownTextPrimitive";
+MarkdownTextPrimitiveImpl.displayName = "MarkdownTextPrimitive";
+var MarkdownTextPrimitive = withSmoothContextProvider(
+  MarkdownTextPrimitiveImpl
+);
 export {
   MarkdownTextPrimitive
 };

package/dist/primitives/MarkdownText.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/primitives/MarkdownText.tsx"],"sourcesContent":["\"use client\";\n\nimport { INTERNAL, useContentPartText } from \"@assistant-ui/react\";\nimport {\n  ComponentRef,\n  ElementType,\n  forwardRef,\n  ForwardRefExoticComponent,\n  RefAttributes,\n  type ComponentPropsWithoutRef,\n  type ComponentType,\n} from \"react\";\nimport ReactMarkdown, { type Options } from \"react-markdown\";\nimport { SyntaxHighlighterProps, CodeHeaderProps } from \"../overrides/types\";\nimport { PreOverride } from \"../overrides/PreOverride\";\nimport {\n  DefaultPre,\n  DefaultCode,\n  DefaultCodeBlockContent,\n  DefaultCodeHeader,\n} from \"../overrides/defaultComponents\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { CodeOverride } from \"../overrides/CodeOverride\";\nimport { Primitive } from \"@radix-ui/react-primitive\";\nimport classNames from \"classnames\";\n\nconst { useSmooth } = INTERNAL;\n\ntype MarkdownTextPrimitiveElement = ComponentRef<typeof Primitive.div>;\ntype PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;\n\nexport type MarkdownTextPrimitiveProps = Omit<\n  Options,\n  \"components\" | \"children\"\n> & {\n  containerProps?: Omit<PrimitiveDivProps, \"children\" | \"asChild\"> | undefined;\n  containerComponent?: ElementType | undefined;\n  components?:\n    | (NonNullable<Options[\"components\"]> & {\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps> | undefined;\n        CodeHeader?: ComponentType<CodeHeaderProps> | undefined;\n        /**\n         * @deprecated Use `componentsByLanguage` instead of `components.by_language`. This will be removed in the next major version.\n         **/\n        by_language?: undefined;\n      })\n    | undefined;\n  componentsByLanguage?:\n    | Record<\n        string,\n        {\n          CodeHeader?: ComponentType<CodeHeaderProps> | undefined;\n          SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps> | undefined;\n        }\n      >\n    | undefined;\n  smooth?: boolean | undefined;\n};\n\nexport const MarkdownTextPrimitive: ForwardRefExoticComponent<MarkdownTextPrimitiveProps> &\n  RefAttributes<MarkdownTextPrimitiveElement> = forwardRef<\n  MarkdownTextPrimitiveElement,\n  MarkdownTextPrimitiveProps\n>(\n  (\n    {\n      components: userComponents,\n      componentsByLanguage = userComponents?.by_language,\n      className,\n      containerProps,\n      containerComponent: Container = \"div\",\n      smooth = true,\n      ...rest\n    },\n    forwardedRef,\n  ) => {\n    const { text, status } = useSmooth(useContentPartText(), smooth);\n\n    const {\n      pre = DefaultPre,\n      code = DefaultCode,\n      SyntaxHighlighter = DefaultCodeBlockContent,\n      CodeHeader = DefaultCodeHeader,\n      by_language,\n      ...componentsRest\n    } = userComponents ?? {};\n    const components: Options[\"components\"] = {\n      ...componentsRest,\n      pre: PreOverride,\n      code: useCallbackRef((props) => (\n        <CodeOverride\n          components={{\n            Pre: pre,\n            Code: code,\n            SyntaxHighlighter,\n            CodeHeader,\n          }}\n          componentsByLanguage={componentsByLanguage}\n          {...props}\n        />\n      )),\n    };\n\n    return (\n      <Container\n        data-status={status.type}\n        {...containerProps}\n        className={classNames(className, containerProps?.className)}\n        ref={forwardedRef}\n      >\n        <ReactMarkdown components={components} {...rest}>\n          {text}\n        </ReactMarkdown>\n      </Container>\n    );\n  },\n);\n\nMarkdownTextPrimitive.displayName = \"MarkdownTextPrimitive\";\n"],"mappings":";;;AAEA,SAAS,UAAU,0BAA0B;AAC7C;AAAA,EAGE;AAAA,OAKK;AACP,OAAO,mBAAqC;AAE5C,SAAS,mBAAmB;AAC5B;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,sBAAsB;AAC/B,SAAS,oBAAoB;AAE7B,OAAO,gBAAgB;AAkEf;AAhER,IAAM,EAAE,UAAU,IAAI;AAiCf,IAAM,wBACmC;AAAA,EAI9C,CACE;AAAA,IACE,YAAY;AAAA,IACZ,uBAAuB,gBAAgB;AAAA,IACvC;AAAA,IACA;AAAA,IACA,oBAAoB,YAAY;AAAA,IAChC,SAAS;AAAA,IACT,GAAG;AAAA,EACL,GACA,iBACG;AACH,UAAM,EAAE,MAAM,OAAO,IAAI,UAAU,mBAAmB,GAAG,MAAM;AAE/D,UAAM;AAAA,MACJ,MAAM;AAAA,MACN,OAAO;AAAA,MACP,oBAAoB;AAAA,MACpB,aAAa;AAAA,MACb;AAAA,MACA,GAAG;AAAA,IACL,IAAI,kBAAkB,CAAC;AACvB,UAAM,aAAoC;AAAA,MACxC,GAAG;AAAA,MACH,KAAK;AAAA,MACL,MAAM,eAAe,CAAC,UACpB;AAAA,QAAC;AAAA;AAAA,UACC,YAAY;AAAA,YACV,KAAK;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA;AAAA,UACF;AAAA,UACA;AAAA,UACC,GAAG;AAAA;AAAA,MACN,CACD;AAAA,IACH;AAEA,WACE;AAAA,MAAC;AAAA;AAAA,QACC,eAAa,OAAO;AAAA,QACnB,GAAG;AAAA,QACJ,WAAW,WAAW,WAAW,gBAAgB,SAAS;AAAA,QAC1D,KAAK;AAAA,QAEL,8BAAC,iBAAc,YAAyB,GAAG,MACxC,gBACH;AAAA;AAAA,IACF;AAAA,EAEJ;AACF;AAEA,sBAAsB,cAAc;","names":[]}
+{"version":3,"sources":["../../src/primitives/MarkdownText.tsx"],"sourcesContent":["\"use client\";\n\nimport { INTERNAL, useContentPartText } from \"@assistant-ui/react\";\nimport {\n  ComponentRef,\n  ElementType,\n  FC,\n  forwardRef,\n  ForwardRefExoticComponent,\n  RefAttributes,\n  useMemo,\n  type ComponentPropsWithoutRef,\n  type ComponentType,\n} from \"react\";\nimport ReactMarkdown, { type Options } from \"react-markdown\";\nimport { SyntaxHighlighterProps, CodeHeaderProps } from \"../overrides/types\";\nimport { PreOverride } from \"../overrides/PreOverride\";\nimport {\n  DefaultPre,\n  DefaultCode,\n  DefaultCodeBlockContent,\n  DefaultCodeHeader,\n} from \"../overrides/defaultComponents\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { CodeOverride } from \"../overrides/CodeOverride\";\nimport { Primitive } from \"@radix-ui/react-primitive\";\nimport classNames from \"classnames\";\n\nconst { useSmooth, useSmoothStatus, withSmoothContextProvider } = INTERNAL;\n\ntype MarkdownTextPrimitiveElement = ComponentRef<typeof Primitive.div>;\ntype PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;\n\nexport type MarkdownTextPrimitiveProps = Omit<\n  Options,\n  \"components\" | \"children\"\n> & {\n  containerProps?: Omit<PrimitiveDivProps, \"children\" | \"asChild\"> | undefined;\n  containerComponent?: ElementType | undefined;\n  components?:\n    | (NonNullable<Options[\"components\"]> & {\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps> | undefined;\n        CodeHeader?: ComponentType<CodeHeaderProps> | undefined;\n        /**\n         * @deprecated Use `componentsByLanguage` instead of `components.by_language`. This will be removed in the next major version.\n         **/\n        by_language?: undefined;\n      })\n    | undefined;\n  componentsByLanguage?:\n    | Record<\n        string,\n        {\n          CodeHeader?: ComponentType<CodeHeaderProps> | undefined;\n          SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps> | undefined;\n        }\n      >\n    | undefined;\n  smooth?: boolean | undefined;\n};\n\nconst MarkdownTextInner: FC<MarkdownTextPrimitiveProps> = ({\n  components: userComponents,\n  componentsByLanguage = userComponents?.by_language,\n  smooth = true,\n  ...rest\n}) => {\n  const { text } = useSmooth(useContentPartText(), smooth);\n\n  const {\n    pre = DefaultPre,\n    code = DefaultCode,\n    SyntaxHighlighter = DefaultCodeBlockContent,\n    CodeHeader = DefaultCodeHeader,\n  } = userComponents ?? {};\n  const useCodeOverrideComponents = useMemo(() => {\n    return {\n      Pre: pre,\n      Code: code,\n      SyntaxHighlighter,\n      CodeHeader,\n    };\n  }, [pre, code, SyntaxHighlighter, CodeHeader]);\n  const CodeComponent = useCallbackRef((props) => (\n    <CodeOverride\n      components={useCodeOverrideComponents}\n      componentsByLanguage={componentsByLanguage}\n      {...props}\n    />\n  ));\n\n  const components: Options[\"components\"] = useMemo(() => {\n    const {\n      pre = DefaultPre,\n      code = DefaultCode,\n      SyntaxHighlighter = DefaultCodeBlockContent,\n      CodeHeader = DefaultCodeHeader,\n      by_language,\n      ...componentsRest\n    } = userComponents ?? {};\n    return {\n      ...componentsRest,\n      pre: PreOverride,\n      code: CodeComponent,\n    };\n  }, [CodeComponent, userComponents, componentsByLanguage]);\n\n  return (\n    <ReactMarkdown components={components} {...rest}>\n      {text}\n    </ReactMarkdown>\n  );\n};\n\nconst MarkdownTextPrimitiveImpl: ForwardRefExoticComponent<MarkdownTextPrimitiveProps> &\n  RefAttributes<MarkdownTextPrimitiveElement> = forwardRef<\n  MarkdownTextPrimitiveElement,\n  MarkdownTextPrimitiveProps\n>(\n  (\n    {\n      className,\n      containerProps,\n      containerComponent: Container = \"div\",\n      ...rest\n    },\n    forwardedRef,\n  ) => {\n    const status = useSmoothStatus();\n    return (\n      <Container\n        data-status={status.type}\n        {...containerProps}\n        className={classNames(className, containerProps?.className)}\n        ref={forwardedRef}\n      >\n        <MarkdownTextInner {...rest}></MarkdownTextInner>\n      </Container>\n    );\n  },\n);\n\nMarkdownTextPrimitiveImpl.displayName = \"MarkdownTextPrimitive\";\n\nexport const MarkdownTextPrimitive = withSmoothContextProvider(\n  MarkdownTextPrimitiveImpl,\n);\n"],"mappings":";;;AAEA,SAAS,UAAU,0BAA0B;AAC7C;AAAA,EAIE;AAAA,EAGA;AAAA,OAGK;AACP,OAAO,mBAAqC;AAE5C,SAAS,mBAAmB;AAC5B;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,sBAAsB;AAC/B,SAAS,oBAAoB;AAE7B,OAAO,gBAAgB;AA0DnB;AAxDJ,IAAM,EAAE,WAAW,iBAAiB,0BAA0B,IAAI;AAiClE,IAAM,oBAAoD,CAAC;AAAA,EACzD,YAAY;AAAA,EACZ,uBAAuB,gBAAgB;AAAA,EACvC,SAAS;AAAA,EACT,GAAG;AACL,MAAM;AACJ,QAAM,EAAE,KAAK,IAAI,UAAU,mBAAmB,GAAG,MAAM;AAEvD,QAAM;AAAA,IACJ,MAAM;AAAA,IACN,OAAO;AAAA,IACP,oBAAoB;AAAA,IACpB,aAAa;AAAA,EACf,IAAI,kBAAkB,CAAC;AACvB,QAAM,4BAA4B,QAAQ,MAAM;AAC9C,WAAO;AAAA,MACL,KAAK;AAAA,MACL,MAAM;AAAA,MACN;AAAA,MACA;AAAA,IACF;AAAA,EACF,GAAG,CAAC,KAAK,MAAM,mBAAmB,UAAU,CAAC;AAC7C,QAAM,gBAAgB,eAAe,CAAC,UACpC;AAAA,IAAC;AAAA;AAAA,MACC,YAAY;AAAA,MACZ;AAAA,MACC,GAAG;AAAA;AAAA,EACN,CACD;AAED,QAAM,aAAoC,QAAQ,MAAM;AACtD,UAAM;AAAA,MACJ,KAAAA,OAAM;AAAA,MACN,MAAAC,QAAO;AAAA,MACP,mBAAAC,qBAAoB;AAAA,MACpB,YAAAC,cAAa;AAAA,MACb;AAAA,MACA,GAAG;AAAA,IACL,IAAI,kBAAkB,CAAC;AACvB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,KAAK;AAAA,MACL,MAAM;AAAA,IACR;AAAA,EACF,GAAG,CAAC,eAAe,gBAAgB,oBAAoB,CAAC;AAExD,SACE,oBAAC,iBAAc,YAAyB,GAAG,MACxC,gBACH;AAEJ;AAEA,IAAM,4BAC0C;AAAA,EAI9C,CACE;AAAA,IACE;AAAA,IACA;AAAA,IACA,oBAAoB,YAAY;AAAA,IAChC,GAAG;AAAA,EACL,GACA,iBACG;AACH,UAAM,SAAS,gBAAgB;AAC/B,WACE;AAAA,MAAC;AAAA;AAAA,QACC,eAAa,OAAO;AAAA,QACnB,GAAG;AAAA,QACJ,WAAW,WAAW,WAAW,gBAAgB,SAAS;AAAA,QAC1D,KAAK;AAAA,QAEL,8BAAC,qBAAmB,GAAG,MAAM;AAAA;AAAA,IAC/B;AAAA,EAEJ;AACF;AAEA,0BAA0B,cAAc;AAEjC,IAAM,wBAAwB;AAAA,EACnC;AACF;","names":["pre","code","SyntaxHighlighter","CodeHeader"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-markdown",
-  "version": "0.7.13",
+  "version": "0.7.15",
   "license": "MIT",
   "exports": {
     ".": {
@@ -41,7 +41,7 @@
     "react-markdown": "^9.0.3"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.7.61",
+    "@assistant-ui/react": "^0.7.62",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc",
     "tailwindcss": "^3.4.4"
@@ -65,7 +65,7 @@
     "tailwindcss": "^3.4.17",
     "tailwindcss-animate": "^1.0.7",
     "tsx": "^4.19.2",
-    "@assistant-ui/react": "0.7.61",
+    "@assistant-ui/react": "0.7.62",
     "@assistant-ui/tailwindcss-transformer": "0.1.0",
     "@assistant-ui/tsbuildutils": "^0.0.1",
     "@assistant-ui/tsconfig": "0.0.0"

package/src/index.ts

@@ -16,3 +16,5 @@ export {
 } from "./ui/markdown-text";
 
 export { CodeHeader } from "./ui/code-header";
+
+export { memoizeMarkdownComponents as unstable_memoizeMarkdownComponents } from "./memoization";

package/src/memoization.tsx

@@ -0,0 +1,47 @@
+import { Element } from "hast";
+import { ComponentProps, ComponentType, ElementType, memo } from "react";
+import { CodeHeaderProps, SyntaxHighlighterProps } from "./overrides/types";
+
+type Components = {
+  [Key in Extract<ElementType, string>]?: ComponentType<ComponentProps<Key>>;
+} & {
+  SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps> | undefined;
+  CodeHeader?: ComponentType<CodeHeaderProps> | undefined;
+};
+
+const areChildrenEqual = (prev: string | unknown, next: string | unknown) => {
+  if (typeof prev === "string") return prev === next;
+  return JSON.stringify(prev) === JSON.stringify(next);
+};
+
+export const areNodesEqual = (
+  prev: Element | undefined,
+  next: Element | undefined,
+) => {
+  if (!prev || !next) return prev === next;
+  const isEqual =
+    JSON.stringify(prev?.properties) === JSON.stringify(next?.properties) &&
+    areChildrenEqual(prev?.children, next?.children);
+  return isEqual;
+};
+
+export const memoCompareNodes = (
+  prev: { node?: Element | undefined },
+  next: { node?: Element | undefined },
+) => {
+  return areNodesEqual(prev.node, next.node);
+};
+
+export const memoizeMarkdownComponents = (components: Components = {}) => {
+  return Object.fromEntries(
+    Object.entries(components ?? {}).map(([key, value]) => {
+      if (!value) return [key, value];
+
+      const Component = value as ComponentType;
+      const WithoutNode = ({ node, ...props }: { node?: Element }) => {
+        return <Component {...props} />;
+      };
+      return [key, memo(WithoutNode, memoCompareNodes)];
+    }),
+  );
+};

package/src/overrides/CodeBlock.tsx

@@ -7,7 +7,6 @@ import {
   SyntaxHighlighterProps,
 } from "./types";
 import { DefaultCodeBlockContent } from "./defaultComponents";
-import { Element } from "hast";
 
 export type CodeBlockProps = {
   language: string;
@@ -18,14 +17,12 @@ export type CodeBlockProps = {
     CodeHeader: ComponentType<CodeHeaderProps>;
     SyntaxHighlighter: ComponentType<SyntaxHighlighterProps>;
   };
-  node: Element | undefined;
 };
 
 export const DefaultCodeBlock: FC<CodeBlockProps> = ({
   components: { Pre, Code, SyntaxHighlighter, CodeHeader },
   language,
   code,
-  node,
 }) => {
   const components = useMemo(() => ({ Pre, Code }), [Pre, Code]);
 
@@ -38,7 +35,6 @@ export const DefaultCodeBlock: FC<CodeBlockProps> = ({
         components={components}
         language={language ?? "unknown"}
         code={code}
-        node={node}
       />
     </>
   );

package/src/overrides/CodeOverride.tsx

@@ -1,5 +1,11 @@
-import { ComponentPropsWithoutRef, ComponentType, FC, useContext } from "react";
-import { PreContext } from "./PreOverride";
+import {
+  ComponentPropsWithoutRef,
+  ComponentType,
+  FC,
+  memo,
+  useContext,
+} from "react";
+import { PreContext, useIsMarkdownCodeBlock } from "./PreOverride";
 import {
   CodeComponent,
   CodeHeaderProps,
@@ -10,6 +16,7 @@ import { DefaultCodeBlock } from "./CodeBlock";
 import { useCallbackRef } from "@radix-ui/react-use-callback-ref";
 import { withDefaultProps } from "./withDefaults";
 import { DefaultCodeBlockContent } from "./defaultComponents";
+import { memoCompareNodes } from "../memoization";
 
 const CodeBlockOverride: FC<CodeOverrideProps> = ({
   components: {
@@ -20,7 +27,6 @@ const CodeBlockOverride: FC<CodeOverrideProps> = ({
   },
   componentsByLanguage = {},
   children,
-  node,
   ...codeProps
 }) => {
   const preProps = useContext(PreContext)!;
@@ -42,7 +48,6 @@ const CodeBlockOverride: FC<CodeOverrideProps> = ({
       <DefaultCodeBlockContent
         components={{ Pre: WrappedPre, Code: WrappedCode }}
         code={children}
-        node={node}
       />
     );
   }
@@ -64,7 +69,6 @@ const CodeBlockOverride: FC<CodeOverrideProps> = ({
       }}
       language={language || "unknown"}
       code={children}
-      node={node}
     />
   );
 };
@@ -87,13 +91,13 @@ export type CodeOverrideProps = ComponentPropsWithoutRef<CodeComponent> & {
     | undefined;
 };
 
-export const CodeOverride: FC<CodeOverrideProps> = ({
+const CodeOverrideImpl: FC<CodeOverrideProps> = ({
   components,
   componentsByLanguage,
   ...props
 }) => {
-  const preProps = useContext(PreContext);
-  if (!preProps) return <components.Code {...props} />;
+  const isCodeBlock = useIsMarkdownCodeBlock();
+  if (!isCodeBlock) return <components.Code {...props} />;
   return (
     <CodeBlockOverride
       components={components}
@@ -102,3 +106,11 @@ export const CodeOverride: FC<CodeOverrideProps> = ({
     />
   );
 };
+
+export const CodeOverride = memo(CodeOverrideImpl, (prev, next) => {
+  const isEqual =
+    prev.components === next.components &&
+    prev.componentsByLanguage === next.componentsByLanguage &&
+    memoCompareNodes(prev, next);
+  return isEqual;
+});

package/src/overrides/PreOverride.tsx

@@ -1,5 +1,11 @@
-import { createContext, ComponentPropsWithoutRef, useContext } from "react";
+import {
+  createContext,
+  ComponentPropsWithoutRef,
+  useContext,
+  memo,
+} from "react";
 import { PreComponent } from "./types";
+import { memoCompareNodes } from "../memoization";
 
 export const PreContext = createContext<Omit<
   ComponentPropsWithoutRef<PreComponent>,
@@ -10,6 +16,8 @@ export const useIsMarkdownCodeBlock = () => {
   return useContext(PreContext) !== null;
 };
 
-export const PreOverride: PreComponent = ({ children, ...rest }) => {
+const PreOverrideImpl: PreComponent = ({ children, ...rest }) => {
   return <PreContext.Provider value={rest}>{children}</PreContext.Provider>;
 };
+
+export const PreOverride = memo(PreOverrideImpl, memoCompareNodes);

package/src/overrides/defaultComponents.tsx

@@ -1,5 +1,4 @@
 import type { ComponentType, ReactNode } from "react";
-import { Element } from "hast";
 import { PreComponent, CodeComponent, CodeHeaderProps } from "./types";
 
 export const DefaultPre: PreComponent = ({ node, ...rest }) => (
@@ -13,10 +12,9 @@ export const DefaultCode: CodeComponent = ({ node, ...rest }) => (
 export const DefaultCodeBlockContent: ComponentType<{
   components: { Pre: PreComponent; Code: CodeComponent };
   code: string | ReactNode | undefined;
-  node: Element | undefined;
-}> = ({ components: { Pre, Code }, code, node }) => (
-  <Pre node={node}>
-    <Code node={node}>{code}</Code>
+}> = ({ components: { Pre, Code }, code }) => (
+  <Pre>
+    <Code>{code}</Code>
   </Pre>
 );