@assistant-ui/react-markdown 0.2.4 → 0.2.6

package/dist/index.d.mts

@@ -1,7 +1,7 @@
 import * as react from 'react';
-import { ComponentType, FC } from 'react';
+import { ElementType, ComponentType, ForwardRefExoticComponent, RefAttributes, ElementRef, ComponentPropsWithoutRef, FC } from 'react';
 import { Options } from 'react-markdown';
-import { TextContentPartProps } from '@assistant-ui/react';
+import { Primitive } from '@radix-ui/react-primitive';
 
 type PreComponent = NonNullable<NonNullable<Options["components"]>["pre"]>;
 type CodeComponent = NonNullable<NonNullable<Options["components"]>["code"]>;
@@ -18,7 +18,11 @@ type SyntaxHighlighterProps = {
     code: string;
 };
 
+type MarkdownTextPrimitiveElement = ElementRef<typeof Primitive.div>;
+type PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;
 type MarkdownTextPrimitiveProps = Omit<Options, "components" | "children"> & {
+    containerProps?: Omit<PrimitiveDivProps, "children" | "asChild">;
+    containerComponent?: ElementType;
     components?: NonNullable<Options["components"]> & {
         SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;
         CodeHeader?: ComponentType<CodeHeaderProps>;
@@ -29,11 +33,13 @@ type MarkdownTextPrimitiveProps = Omit<Options, "components" | "children"> & {
     };
     smooth?: boolean;
 };
-declare const MarkdownTextPrimitive: FC<MarkdownTextPrimitiveProps>;
+declare const MarkdownTextPrimitive: ForwardRefExoticComponent<MarkdownTextPrimitiveProps> & RefAttributes<MarkdownTextPrimitiveElement>;
+
+declare const useIsMarkdownCodeBlock: () => boolean;
 
 type MakeMarkdownTextProps = MarkdownTextPrimitiveProps;
-declare const makeMarkdownText: ({ className, components: userComponents, ...rest }?: MakeMarkdownTextProps) => react.NamedExoticComponent<TextContentPartProps>;
+declare const makeMarkdownText: ({ className, components: userComponents, ...rest }?: MakeMarkdownTextProps) => react.NamedExoticComponent<{}>;
 
 declare const CodeHeader: FC<CodeHeaderProps>;
 
-export { CodeHeader, type CodeHeaderProps, type MakeMarkdownTextProps, MarkdownTextPrimitive, type MarkdownTextPrimitiveProps, type SyntaxHighlighterProps, makeMarkdownText };
+export { CodeHeader, type CodeHeaderProps, type MakeMarkdownTextProps, MarkdownTextPrimitive, type MarkdownTextPrimitiveProps, type SyntaxHighlighterProps, makeMarkdownText, useIsMarkdownCodeBlock };

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 import * as react from 'react';
-import { ComponentType, FC } from 'react';
+import { ElementType, ComponentType, ForwardRefExoticComponent, RefAttributes, ElementRef, ComponentPropsWithoutRef, FC } from 'react';
 import { Options } from 'react-markdown';
-import { TextContentPartProps } from '@assistant-ui/react';
+import { Primitive } from '@radix-ui/react-primitive';
 
 type PreComponent = NonNullable<NonNullable<Options["components"]>["pre"]>;
 type CodeComponent = NonNullable<NonNullable<Options["components"]>["code"]>;
@@ -18,7 +18,11 @@ type SyntaxHighlighterProps = {
     code: string;
 };
 
+type MarkdownTextPrimitiveElement = ElementRef<typeof Primitive.div>;
+type PrimitiveDivProps = ComponentPropsWithoutRef<typeof Primitive.div>;
 type MarkdownTextPrimitiveProps = Omit<Options, "components" | "children"> & {
+    containerProps?: Omit<PrimitiveDivProps, "children" | "asChild">;
+    containerComponent?: ElementType;
     components?: NonNullable<Options["components"]> & {
         SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;
         CodeHeader?: ComponentType<CodeHeaderProps>;
@@ -29,11 +33,13 @@ type MarkdownTextPrimitiveProps = Omit<Options, "components" | "children"> & {
     };
     smooth?: boolean;
 };
-declare const MarkdownTextPrimitive: FC<MarkdownTextPrimitiveProps>;
+declare const MarkdownTextPrimitive: ForwardRefExoticComponent<MarkdownTextPrimitiveProps> & RefAttributes<MarkdownTextPrimitiveElement>;
+
+declare const useIsMarkdownCodeBlock: () => boolean;
 
 type MakeMarkdownTextProps = MarkdownTextPrimitiveProps;
-declare const makeMarkdownText: ({ className, components: userComponents, ...rest }?: MakeMarkdownTextProps) => react.NamedExoticComponent<TextContentPartProps>;
+declare const makeMarkdownText: ({ className, components: userComponents, ...rest }?: MakeMarkdownTextProps) => react.NamedExoticComponent<{}>;
 
 declare const CodeHeader: FC<CodeHeaderProps>;
 
-export { CodeHeader, type CodeHeaderProps, type MakeMarkdownTextProps, MarkdownTextPrimitive, type MarkdownTextPrimitiveProps, type SyntaxHighlighterProps, makeMarkdownText };
+export { CodeHeader, type CodeHeaderProps, type MakeMarkdownTextProps, MarkdownTextPrimitive, type MarkdownTextPrimitiveProps, type SyntaxHighlighterProps, makeMarkdownText, useIsMarkdownCodeBlock };

package/dist/index.js

@@ -33,18 +33,23 @@ var src_exports = {};
 __export(src_exports, {
   CodeHeader: () => CodeHeader,
   MarkdownTextPrimitive: () => MarkdownTextPrimitive,
-  makeMarkdownText: () => makeMarkdownText
+  makeMarkdownText: () => makeMarkdownText,
+  useIsMarkdownCodeBlock: () => useIsMarkdownCodeBlock
 });
 module.exports = __toCommonJS(src_exports);
 
 // src/primitives/MarkdownText.tsx
 var import_react4 = require("@assistant-ui/react");
+var import_react5 = require("react");
 var import_react_markdown = __toESM(require("react-markdown"));
 
 // src/overrides/PreOverride.tsx
 var import_react = require("react");
 var import_jsx_runtime = require("react/jsx-runtime");
 var PreContext = (0, import_react.createContext)(null);
+var useIsMarkdownCodeBlock = () => {
+  return (0, import_react.useContext)(PreContext) !== null;
+};
 var PreOverride = ({ children, ...rest }) => {
   return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(PreContext.Provider, { value: rest, children });
 };
@@ -155,58 +160,74 @@ var CodeOverride = ({
 };
 
 // src/primitives/MarkdownText.tsx
+var import_internal = require("@assistant-ui/react/internal");
+var import_classnames2 = __toESM(require("classnames"));
 var import_jsx_runtime5 = require("react/jsx-runtime");
-var { useSmooth } = import_react4.INTERNAL;
-var MarkdownTextPrimitive = ({
-  smooth = true,
-  components: userComponents,
-  ...rest
-}) => {
-  const {
-    part: { text }
-  } = (0, import_react4.useContentPartText)();
-  const smoothText = useSmooth(text, smooth);
-  const {
-    pre = DefaultPre,
-    code = DefaultCode,
-    SyntaxHighlighter = DefaultCodeBlockContent,
-    CodeHeader: CodeHeader2 = DefaultCodeHeader,
-    by_language,
-    ...componentsRest
-  } = userComponents ?? {};
-  const components = {
-    ...componentsRest,
-    pre: PreOverride,
-    code: (0, import_react_use_callback_ref2.useCallbackRef)((props) => /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(
-      CodeOverride,
+var MarkdownTextPrimitive = (0, import_react5.forwardRef)(
+  ({
+    components: userComponents,
+    className,
+    containerProps,
+    containerComponent: Container = "div",
+    ...rest
+  }, forwardedRef, smooth = true) => {
+    const {
+      part: { text },
+      status
+    } = (0, import_internal.useSmooth)((0, import_react4.useContentPartText)(), smooth);
+    const {
+      pre = DefaultPre,
+      code = DefaultCode,
+      SyntaxHighlighter = DefaultCodeBlockContent,
+      CodeHeader: CodeHeader2 = DefaultCodeHeader,
+      by_language,
+      ...componentsRest
+    } = userComponents ?? {};
+    const components = {
+      ...componentsRest,
+      pre: PreOverride,
+      code: (0, import_react_use_callback_ref2.useCallbackRef)((props) => /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(
+        CodeOverride,
+        {
+          components: {
+            Pre: pre,
+            Code: code,
+            SyntaxHighlighter,
+            CodeHeader: CodeHeader2,
+            by_language
+          },
+          ...props
+        }
+      ))
+    };
+    return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(
+      Container,
       {
-        components: {
-          Pre: pre,
-          Code: code,
-          SyntaxHighlighter,
-          CodeHeader: CodeHeader2,
-          by_language
-        },
-        ...props
+        "data-status": status.type,
+        ...containerProps,
+        className: (0, import_classnames2.default)(className, containerProps?.className),
+        ref: forwardedRef,
+        children: /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(import_react_markdown.default, { components, ...rest, children: text })
       }
-    ))
-  };
-  return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(import_react_markdown.default, { components, ...rest, children: smoothText });
-};
+    );
+  }
+);
+MarkdownTextPrimitive.displayName = "MarkdownTextPrimitive";
 
 // src/ui/markdown-text.tsx
-var import_react7 = require("react");
+var import_react8 = require("react");
 
 // src/ui/code-header.tsx
 var import_lucide_react = require("lucide-react");
-var import_react6 = require("@assistant-ui/react");
+var import_react7 = require("@assistant-ui/react");
+var import_internal2 = require("@assistant-ui/react/internal");
 
 // src/ui/useCopyToClipboard.tsx
-var import_react5 = require("react");
+var import_react6 = require("react");
 var useCopyToClipboard = ({
   copiedDuration = 3e3
 } = {}) => {
-  const [isCopied, setIsCopied] = (0, import_react5.useState)(false);
+  const [isCopied, setIsCopied] = (0, import_react6.useState)(false);
   const copyToClipboard = (value) => {
     if (!value) return;
     navigator.clipboard.writeText(value).then(() => {
@@ -219,13 +240,12 @@ var useCopyToClipboard = ({
 
 // src/ui/code-header.tsx
 var import_jsx_runtime6 = require("react/jsx-runtime");
-var { TooltipIconButton } = import_react6.INTERNAL;
 var CodeHeader = ({ language, code }) => {
   const {
     strings: {
       code: { header: { copy: { tooltip = "Copy" } = {} } = {} } = {}
     } = {}
-  } = (0, import_react6.useThreadConfig)();
+  } = (0, import_react7.useThreadConfig)();
   const { isCopied, copyToClipboard } = useCopyToClipboard();
   const onCopy = () => {
     if (!code || isCopied) return;
@@ -233,7 +253,7 @@ var CodeHeader = ({ language, code }) => {
   };
   return /* @__PURE__ */ (0, import_jsx_runtime6.jsxs)("div", { className: "aui-code-header-root", children: [
     /* @__PURE__ */ (0, import_jsx_runtime6.jsx)("span", { className: "aui-code-header-language", children: language }),
-    /* @__PURE__ */ (0, import_jsx_runtime6.jsxs)(TooltipIconButton, { tooltip, onClick: onCopy, children: [
+    /* @__PURE__ */ (0, import_jsx_runtime6.jsxs)(import_internal2.TooltipIconButton, { tooltip, onClick: onCopy, children: [
       !isCopied && /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(import_lucide_react.CopyIcon, {}),
       isCopied && /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(import_lucide_react.CheckIcon, {})
     ] })
@@ -241,7 +261,8 @@ var CodeHeader = ({ language, code }) => {
 };
 
 // src/ui/markdown-text.tsx
-var import_classnames2 = __toESM(require("classnames"));
+var import_classnames3 = __toESM(require("classnames"));
+var import_internal3 = require("@assistant-ui/react/internal");
 var import_jsx_runtime7 = require("react/jsx-runtime");
 var makeMarkdownText = ({
   className,
@@ -252,29 +273,29 @@ var makeMarkdownText = ({
     ...userComponents,
     CodeHeader: userComponents?.CodeHeader ?? CodeHeader
   };
-  const MarkdownTextImpl = ({ status }) => {
+  const MarkdownTextImpl = () => {
+    const status = (0, import_internal3.useSmoothStatus)();
     return /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
-      "div",
+      MarkdownTextPrimitive,
       {
-        className: (0, import_classnames2.default)(
+        components,
+        ...rest,
+        className: (0, import_classnames3.default)(
           "aui-md-root",
-          status.type === "running" && "aui-md-in-progress",
+          status.type === "running" && "aui-md-running",
           className
-        ),
-        children: /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(MarkdownTextPrimitive, { components, ...rest })
+        )
       }
     );
   };
   MarkdownTextImpl.displayName = "MarkdownText";
-  return (0, import_react7.memo)(
-    MarkdownTextImpl,
-    (prev, next) => prev.status.type === next.status.type
-  );
+  return (0, import_react8.memo)((0, import_internal3.withSmoothContextProvider)(MarkdownTextImpl), () => true);
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   CodeHeader,
   MarkdownTextPrimitive,
-  makeMarkdownText
+  makeMarkdownText,
+  useIsMarkdownCodeBlock
 });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/primitives/MarkdownText.tsx","../src/overrides/PreOverride.tsx","../src/overrides/defaultComponents.tsx","../src/overrides/CodeOverride.tsx","../src/overrides/CodeBlock.tsx","../src/overrides/withDefaults.tsx","../src/ui/markdown-text.tsx","../src/ui/code-header.tsx","../src/ui/useCopyToClipboard.tsx"],"sourcesContent":["export {\n  MarkdownTextPrimitive,\n  type MarkdownTextPrimitiveProps,\n} from \"./primitives/MarkdownText\";\n\nexport type {\n  CodeHeaderProps,\n  SyntaxHighlighterProps,\n} from \"./overrides/types\";\n\nexport {\n  makeMarkdownText,\n  type MakeMarkdownTextProps,\n} from \"./ui/markdown-text\";\n\nexport { CodeHeader } from \"./ui/code-header\";\n","\"use client\";\n\nimport { INTERNAL, useContentPartText } from \"@assistant-ui/react\";\nimport type { ComponentType, FC } 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\";\n\nconst { useSmooth } = INTERNAL;\n\nexport type MarkdownTextPrimitiveProps = Omit<\n  Options,\n  \"components\" | \"children\"\n> & {\n  components?: NonNullable<Options[\"components\"]> & {\n    SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n    CodeHeader?: ComponentType<CodeHeaderProps>;\n    by_language?: Record<\n      string,\n      {\n        CodeHeader?: ComponentType<CodeHeaderProps>;\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n      }\n    >;\n  };\n  smooth?: boolean;\n};\nexport const MarkdownTextPrimitive: FC<MarkdownTextPrimitiveProps> = ({\n  smooth = true,\n  components: userComponents,\n  ...rest\n}) => {\n  const {\n    part: { text },\n  } = useContentPartText();\n  const smoothText = useSmooth(text, smooth); // TODO loading indicator disappears before smooth animation ends\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: typeof userComponents = {\n    ...componentsRest,\n    pre: PreOverride,\n    code: useCallbackRef((props) => (\n      <CodeOverride\n        components={{\n          Pre: pre,\n          Code: code,\n          SyntaxHighlighter,\n          CodeHeader,\n          by_language,\n        }}\n        {...props}\n      />\n    )),\n  };\n\n  return (\n    <ReactMarkdown components={components} {...rest}>\n      {smoothText}\n    </ReactMarkdown>\n  );\n};\n","import { createContext, ComponentPropsWithoutRef } from \"react\";\nimport { PreComponent } from \"./types\";\n\nexport const PreContext = createContext<Omit<\n  ComponentPropsWithoutRef<PreComponent>,\n  \"children\"\n> | null>(null);\n\nexport const PreOverride: PreComponent = ({ children, ...rest }) => {\n  return <PreContext.Provider value={rest}>{children}</PreContext.Provider>;\n};\n","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","import { ComponentPropsWithoutRef, ComponentType, FC, useContext } from \"react\";\nimport { PreContext } from \"./PreOverride\";\nimport {\n  CodeComponent,\n  CodeHeaderProps,\n  PreComponent,\n  SyntaxHighlighterProps,\n} from \"./types\";\nimport { DefaultCodeBlock } from \"./CodeBlock\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { withDefaultProps } from \"./withDefaults\";\nimport { DefaultCodeBlockContent } from \"./defaultComponents\";\n\nconst CodeBlockOverride: FC<CodeOverrideProps> = ({\n  components: {\n    Pre,\n    Code,\n    SyntaxHighlighter: FallbackSyntaxHighlighter,\n    CodeHeader: FallbackCodeHeader,\n    by_language = {},\n  },\n  children,\n  ...codeProps\n}) => {\n  const preProps = useContext(PreContext)!;\n  const getPreProps = withDefaultProps<any>(preProps);\n  const WrappedPre: PreComponent = useCallbackRef((props) => (\n    <Pre {...getPreProps(props)} />\n  ));\n\n  const getCodeProps = withDefaultProps<any>(codeProps);\n  const WrappedCode: CodeComponent = useCallbackRef((props) => (\n    <Code {...getCodeProps(props)} />\n  ));\n\n  const language = /language-(\\w+)/.exec(codeProps.className || \"\")?.[1] ?? \"\";\n\n  // if the code content is not string (due to rehype plugins), return a default code block\n  if (typeof children !== \"string\") {\n    return (\n      <DefaultCodeBlockContent\n        components={{ Pre: WrappedPre, Code: WrappedCode }}\n        code={children}\n      />\n    );\n  }\n\n  const SyntaxHighlighter: ComponentType<SyntaxHighlighterProps> =\n    by_language[language]?.SyntaxHighlighter ?? FallbackSyntaxHighlighter;\n\n  const CodeHeader: ComponentType<CodeHeaderProps> =\n    by_language[language]?.CodeHeader ?? FallbackCodeHeader;\n\n  return (\n    <DefaultCodeBlock\n      components={{\n        Pre: WrappedPre,\n        Code: WrappedCode,\n        SyntaxHighlighter,\n        CodeHeader,\n      }}\n      language={language || \"unknown\"}\n      code={children}\n    />\n  );\n};\n\nexport type CodeOverrideProps = ComponentPropsWithoutRef<CodeComponent> & {\n  components: {\n    Pre: PreComponent;\n    Code: CodeComponent;\n    CodeHeader: ComponentType<CodeHeaderProps>;\n    SyntaxHighlighter: ComponentType<SyntaxHighlighterProps>;\n    by_language?: Record<\n      string,\n      {\n        CodeHeader?: ComponentType<CodeHeaderProps>;\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n      }\n    >;\n  };\n};\n\nexport const CodeOverride: FC<CodeOverrideProps> = ({\n  components,\n  ...props\n}) => {\n  const preProps = useContext(PreContext);\n  if (!preProps) return <components.Code {...(props as any)} />;\n  return <CodeBlockOverride components={components} {...props} />;\n};\n","import { ComponentType, FC, useMemo } from \"react\";\n\nimport {\n  CodeComponent,\n  CodeHeaderProps,\n  PreComponent,\n  SyntaxHighlighterProps,\n} from \"./types\";\nimport { DefaultCodeBlockContent } from \"./defaultComponents\";\n\nexport type CodeBlockProps = {\n  language: string;\n  code: string;\n  components: {\n    Pre: PreComponent;\n    Code: CodeComponent;\n    CodeHeader: ComponentType<CodeHeaderProps>;\n    SyntaxHighlighter: ComponentType<SyntaxHighlighterProps>;\n  };\n};\n\nexport const DefaultCodeBlock: FC<CodeBlockProps> = ({\n  components: { Pre, Code, SyntaxHighlighter, CodeHeader },\n  language,\n  code,\n}) => {\n  const components = useMemo(() => ({ Pre, Code }), [Pre, Code]);\n\n  const SH = !!language ? SyntaxHighlighter : DefaultCodeBlockContent;\n\n  return (\n    <>\n      <CodeHeader language={language} code={code} />\n      <SH\n        components={components}\n        language={language ?? \"unknown\"}\n        code={code}\n      />\n    </>\n  );\n};\n","import classNames from \"classnames\";\n\nexport const withDefaultProps =\n  <TProps extends { className?: string | undefined }>({\n    className,\n    ...defaultProps\n  }: Partial<TProps>) =>\n  ({ className: classNameProp, ...props }: TProps) => {\n    return {\n      className: classNames(className, classNameProp),\n      ...defaultProps,\n      ...props,\n    } as TProps;\n  };\n","import { TextContentPartProps } from \"@assistant-ui/react\";\nimport { FC, memo } from \"react\";\nimport { CodeHeader } from \"./code-header\";\nimport classNames from \"classnames\";\nimport {\n  MarkdownTextPrimitive,\n  MarkdownTextPrimitiveProps,\n} from \"../primitives/MarkdownText\";\n\nexport type MakeMarkdownTextProps = MarkdownTextPrimitiveProps;\n\nexport const makeMarkdownText = ({\n  className,\n  components: userComponents,\n  ...rest\n}: MakeMarkdownTextProps = {}) => {\n  const components = {\n    ...userComponents,\n    CodeHeader: userComponents?.CodeHeader ?? CodeHeader,\n  };\n\n  const MarkdownTextImpl: FC<TextContentPartProps> = ({ status }) => {\n    return (\n      <div\n        className={classNames(\n          \"aui-md-root\",\n          status.type === \"running\" && \"aui-md-in-progress\",\n          className,\n        )}\n      >\n        <MarkdownTextPrimitive components={components} {...rest} />\n      </div>\n    );\n  };\n  MarkdownTextImpl.displayName = \"MarkdownText\";\n\n  return memo(\n    MarkdownTextImpl,\n    (prev, next) => prev.status.type === next.status.type,\n  );\n};\n","import { FC } from \"react\";\nimport { CheckIcon, CopyIcon } from \"lucide-react\";\nimport { useThreadConfig, INTERNAL } from \"@assistant-ui/react\";\n\nimport { CodeHeaderProps } from \"../overrides/types\";\nimport { useCopyToClipboard } from \"./useCopyToClipboard\";\n\nconst { TooltipIconButton } = INTERNAL;\n\nexport const CodeHeader: FC<CodeHeaderProps> = ({ language, code }) => {\n  const {\n    strings: {\n      code: { header: { copy: { tooltip = \"Copy\" } = {} } = {} } = {},\n    } = {},\n  } = useThreadConfig();\n\n  const { isCopied, copyToClipboard } = useCopyToClipboard();\n  const onCopy = () => {\n    if (!code || isCopied) return;\n    copyToClipboard(code);\n  };\n\n  return (\n    <div className=\"aui-code-header-root\">\n      <span className=\"aui-code-header-language\">{language}</span>\n      <TooltipIconButton tooltip={tooltip} onClick={onCopy}>\n        {!isCopied && <CopyIcon />}\n        {isCopied && <CheckIcon />}\n      </TooltipIconButton>\n    </div>\n  );\n};\n","import { useState } from \"react\";\nimport { UseActionBarCopyProps } from \"@assistant-ui/react\";\n\nexport type useCopyToClipboardProps = UseActionBarCopyProps;\n\nexport const useCopyToClipboard = ({\n  copiedDuration = 3000,\n}: useCopyToClipboardProps = {}) => {\n  const [isCopied, setIsCopied] = useState<boolean>(false);\n\n  const copyToClipboard = (value: string) => {\n    if (!value) return;\n\n    navigator.clipboard.writeText(value).then(() => {\n      setIsCopied(true);\n      setTimeout(() => setIsCopied(false), copiedDuration);\n    });\n  };\n\n  return { isCopied, copyToClipboard };\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACEA,IAAAA,gBAA6C;AAE7C,4BAA4C;;;ACJ5C,mBAAwD;AAS/C;AANF,IAAM,iBAAa,4BAGhB,IAAI;AAEP,IAAM,cAA4B,CAAC,EAAE,UAAU,GAAG,KAAK,MAAM;AAClE,SAAO,4CAAC,WAAW,UAAX,EAAoB,OAAO,MAAO,UAAS;AACrD;;;ACNE,IAAAC,sBAAA;AADK,IAAM,aAA2B,CAAC,EAAE,MAAM,GAAG,KAAK,MACvD,6CAAC,SAAK,GAAG,MAAM;AAGV,IAAM,cAA6B,CAAC,EAAE,MAAM,GAAG,KAAK,MACzD,6CAAC,UAAM,GAAG,MAAM;AAGX,IAAM,0BAGR,CAAC,EAAE,YAAY,EAAE,KAAK,KAAK,GAAG,KAAK,MACtC,6CAAC,OACC,uDAAC,QAAM,gBAAK,GACd;AAGK,IAAM,oBAAoD,MAAM;;;AFPvE,IAAAC,iCAA+B;;;AGb/B,IAAAC,gBAAwE;;;ACAxE,IAAAC,gBAA2C;AA+BvC,IAAAC,sBAAA;AAVG,IAAM,mBAAuC,CAAC;AAAA,EACnD,YAAY,EAAE,KAAK,MAAM,mBAAmB,YAAAC,YAAW;AAAA,EACvD;AAAA,EACA;AACF,MAAM;AACJ,QAAM,iBAAa,uBAAQ,OAAO,EAAE,KAAK,KAAK,IAAI,CAAC,KAAK,IAAI,CAAC;AAE7D,QAAM,KAAK,CAAC,CAAC,WAAW,oBAAoB;AAE5C,SACE,8EACE;AAAA,iDAACA,aAAA,EAAW,UAAoB,MAAY;AAAA,IAC5C;AAAA,MAAC;AAAA;AAAA,QACC;AAAA,QACA,UAAU,YAAY;AAAA,QACtB;AAAA;AAAA,IACF;AAAA,KACF;AAEJ;;;AD/BA,oCAA+B;;;AET/B,wBAAuB;AAEhB,IAAM,mBACX,CAAoD;AAAA,EAClD;AAAA,EACA,GAAG;AACL,MACA,CAAC,EAAE,WAAW,eAAe,GAAG,MAAM,MAAc;AAClD,SAAO;AAAA,IACL,eAAW,kBAAAC,SAAW,WAAW,aAAa;AAAA,IAC9C,GAAG;AAAA,IACH,GAAG;AAAA,EACL;AACF;;;AFcE,IAAAC,sBAAA;AAdJ,IAAM,oBAA2C,CAAC;AAAA,EAChD,YAAY;AAAA,IACV;AAAA,IACA;AAAA,IACA,mBAAmB;AAAA,IACnB,YAAY;AAAA,IACZ,cAAc,CAAC;AAAA,EACjB;AAAA,EACA;AAAA,EACA,GAAG;AACL,MAAM;AACJ,QAAM,eAAW,0BAAW,UAAU;AACtC,QAAM,cAAc,iBAAsB,QAAQ;AAClD,QAAM,iBAA2B,8CAAe,CAAC,UAC/C,6CAAC,OAAK,GAAG,YAAY,KAAK,GAAG,CAC9B;AAED,QAAM,eAAe,iBAAsB,SAAS;AACpD,QAAM,kBAA6B,8CAAe,CAAC,UACjD,6CAAC,QAAM,GAAG,aAAa,KAAK,GAAG,CAChC;AAED,QAAM,WAAW,iBAAiB,KAAK,UAAU,aAAa,EAAE,IAAI,CAAC,KAAK;AAG1E,MAAI,OAAO,aAAa,UAAU;AAChC,WACE;AAAA,MAAC;AAAA;AAAA,QACC,YAAY,EAAE,KAAK,YAAY,MAAM,YAAY;AAAA,QACjD,MAAM;AAAA;AAAA,IACR;AAAA,EAEJ;AAEA,QAAM,oBACJ,YAAY,QAAQ,GAAG,qBAAqB;AAE9C,QAAMC,cACJ,YAAY,QAAQ,GAAG,cAAc;AAEvC,SACE;AAAA,IAAC;AAAA;AAAA,MACC,YAAY;AAAA,QACV,KAAK;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA,YAAAA;AAAA,MACF;AAAA,MACA,UAAU,YAAY;AAAA,MACtB,MAAM;AAAA;AAAA,EACR;AAEJ;AAkBO,IAAM,eAAsC,CAAC;AAAA,EAClD;AAAA,EACA,GAAG;AACL,MAAM;AACJ,QAAM,eAAW,0BAAW,UAAU;AACtC,MAAI,CAAC,SAAU,QAAO,6CAAC,WAAW,MAAX,EAAiB,GAAI,OAAe;AAC3D,SAAO,6CAAC,qBAAkB,YAAyB,GAAG,OAAO;AAC/D;;;AHjCM,IAAAC,sBAAA;AAzCN,IAAM,EAAE,UAAU,IAAI;AAmBf,IAAM,wBAAwD,CAAC;AAAA,EACpE,SAAS;AAAA,EACT,YAAY;AAAA,EACZ,GAAG;AACL,MAAM;AACJ,QAAM;AAAA,IACJ,MAAM,EAAE,KAAK;AAAA,EACf,QAAI,kCAAmB;AACvB,QAAM,aAAa,UAAU,MAAM,MAAM;AAEzC,QAAM;AAAA,IACJ,MAAM;AAAA,IACN,OAAO;AAAA,IACP,oBAAoB;AAAA,IACpB,YAAAC,cAAa;AAAA,IACb;AAAA,IACA,GAAG;AAAA,EACL,IAAI,kBAAkB,CAAC;AACvB,QAAM,aAAoC;AAAA,IACxC,GAAG;AAAA,IACH,KAAK;AAAA,IACL,UAAM,+CAAe,CAAC,UACpB;AAAA,MAAC;AAAA;AAAA,QACC,YAAY;AAAA,UACV,KAAK;AAAA,UACL,MAAM;AAAA,UACN;AAAA,UACA,YAAAA;AAAA,UACA;AAAA,QACF;AAAA,QACC,GAAG;AAAA;AAAA,IACN,CACD;AAAA,EACH;AAEA,SACE,6CAAC,sBAAAC,SAAA,EAAc,YAAyB,GAAG,MACxC,sBACH;AAEJ;;;AM1EA,IAAAC,gBAAyB;;;ACAzB,0BAAoC;AACpC,IAAAC,gBAA0C;;;ACF1C,IAAAC,gBAAyB;AAKlB,IAAM,qBAAqB,CAAC;AAAA,EACjC,iBAAiB;AACnB,IAA6B,CAAC,MAAM;AAClC,QAAM,CAAC,UAAU,WAAW,QAAI,wBAAkB,KAAK;AAEvD,QAAM,kBAAkB,CAAC,UAAkB;AACzC,QAAI,CAAC,MAAO;AAEZ,cAAU,UAAU,UAAU,KAAK,EAAE,KAAK,MAAM;AAC9C,kBAAY,IAAI;AAChB,iBAAW,MAAM,YAAY,KAAK,GAAG,cAAc;AAAA,IACrD,CAAC;AAAA,EACH;AAEA,SAAO,EAAE,UAAU,gBAAgB;AACrC;;;ADIM,IAAAC,sBAAA;AAjBN,IAAM,EAAE,kBAAkB,IAAI;AAEvB,IAAM,aAAkC,CAAC,EAAE,UAAU,KAAK,MAAM;AACrE,QAAM;AAAA,IACJ,SAAS;AAAA,MACP,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,OAAO,IAAI,CAAC,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC;AAAA,IAChE,IAAI,CAAC;AAAA,EACP,QAAI,+BAAgB;AAEpB,QAAM,EAAE,UAAU,gBAAgB,IAAI,mBAAmB;AACzD,QAAM,SAAS,MAAM;AACnB,QAAI,CAAC,QAAQ,SAAU;AACvB,oBAAgB,IAAI;AAAA,EACtB;AAEA,SACE,8CAAC,SAAI,WAAU,wBACb;AAAA,iDAAC,UAAK,WAAU,4BAA4B,oBAAS;AAAA,IACrD,8CAAC,qBAAkB,SAAkB,SAAS,QAC3C;AAAA,OAAC,YAAY,6CAAC,gCAAS;AAAA,MACvB,YAAY,6CAAC,iCAAU;AAAA,OAC1B;AAAA,KACF;AAEJ;;;AD5BA,IAAAC,qBAAuB;AA2Bf,IAAAC,sBAAA;AAnBD,IAAM,mBAAmB,CAAC;AAAA,EAC/B;AAAA,EACA,YAAY;AAAA,EACZ,GAAG;AACL,IAA2B,CAAC,MAAM;AAChC,QAAM,aAAa;AAAA,IACjB,GAAG;AAAA,IACH,YAAY,gBAAgB,cAAc;AAAA,EAC5C;AAEA,QAAM,mBAA6C,CAAC,EAAE,OAAO,MAAM;AACjE,WACE;AAAA,MAAC;AAAA;AAAA,QACC,eAAW,mBAAAC;AAAA,UACT;AAAA,UACA,OAAO,SAAS,aAAa;AAAA,UAC7B;AAAA,QACF;AAAA,QAEA,uDAAC,yBAAsB,YAAyB,GAAG,MAAM;AAAA;AAAA,IAC3D;AAAA,EAEJ;AACA,mBAAiB,cAAc;AAE/B,aAAO;AAAA,IACL;AAAA,IACA,CAAC,MAAM,SAAS,KAAK,OAAO,SAAS,KAAK,OAAO;AAAA,EACnD;AACF;","names":["import_react","import_jsx_runtime","import_react_use_callback_ref","import_react","import_react","import_jsx_runtime","CodeHeader","classNames","import_jsx_runtime","CodeHeader","import_jsx_runtime","CodeHeader","ReactMarkdown","import_react","import_react","import_react","import_jsx_runtime","import_classnames","import_jsx_runtime","classNames"]}
+{"version":3,"sources":["../src/index.ts","../src/primitives/MarkdownText.tsx","../src/overrides/PreOverride.tsx","../src/overrides/defaultComponents.tsx","../src/overrides/CodeOverride.tsx","../src/overrides/CodeBlock.tsx","../src/overrides/withDefaults.tsx","../src/ui/markdown-text.tsx","../src/ui/code-header.tsx","../src/ui/useCopyToClipboard.tsx"],"sourcesContent":["export {\n  MarkdownTextPrimitive,\n  type MarkdownTextPrimitiveProps,\n} from \"./primitives/MarkdownText\";\n\nexport type {\n  CodeHeaderProps,\n  SyntaxHighlighterProps,\n} from \"./overrides/types\";\n\nexport { useIsMarkdownCodeBlock } from \"./overrides/PreOverride\";\n\nexport {\n  makeMarkdownText,\n  type MakeMarkdownTextProps,\n} from \"./ui/markdown-text\";\n\nexport { CodeHeader } from \"./ui/code-header\";\n","\"use client\";\n\nimport { useContentPartText } from \"@assistant-ui/react\";\nimport {\n  ElementRef,\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 { useSmooth } from \"@assistant-ui/react/internal\";\nimport { Primitive } from \"@radix-ui/react-primitive\";\nimport classNames from \"classnames\";\n\ntype MarkdownTextPrimitiveElement = ElementRef<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\">;\n  containerComponent?: ElementType;\n  components?: NonNullable<Options[\"components\"]> & {\n    SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n    CodeHeader?: ComponentType<CodeHeaderProps>;\n    by_language?: Record<\n      string,\n      {\n        CodeHeader?: ComponentType<CodeHeaderProps>;\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n      }\n    >;\n  };\n  smooth?: boolean;\n};\n\nexport const MarkdownTextPrimitive: ForwardRefExoticComponent<MarkdownTextPrimitiveProps> &\n  RefAttributes<MarkdownTextPrimitiveElement> = forwardRef<\n  MarkdownTextPrimitiveElement,\n  MarkdownTextPrimitiveProps\n>(\n  (\n    {\n      components: userComponents,\n      className,\n      containerProps,\n      containerComponent: Container = \"div\",\n      ...rest\n    },\n    forwardedRef,\n    smooth = true,\n  ) => {\n    const {\n      part: { text },\n      status,\n    } = 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: typeof userComponents = {\n      ...componentsRest,\n      pre: PreOverride,\n      code: useCallbackRef((props) => (\n        <CodeOverride\n          components={{\n            Pre: pre,\n            Code: code,\n            SyntaxHighlighter,\n            CodeHeader,\n            by_language,\n          }}\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","import { createContext, ComponentPropsWithoutRef, useContext } from \"react\";\nimport { PreComponent } from \"./types\";\n\nexport const PreContext = createContext<Omit<\n  ComponentPropsWithoutRef<PreComponent>,\n  \"children\"\n> | null>(null);\n\nexport const useIsMarkdownCodeBlock = () => {\n  return useContext(PreContext) !== null;\n};\n\nexport const PreOverride: PreComponent = ({ children, ...rest }) => {\n  return <PreContext.Provider value={rest}>{children}</PreContext.Provider>;\n};\n","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","import { ComponentPropsWithoutRef, ComponentType, FC, useContext } from \"react\";\nimport { PreContext } from \"./PreOverride\";\nimport {\n  CodeComponent,\n  CodeHeaderProps,\n  PreComponent,\n  SyntaxHighlighterProps,\n} from \"./types\";\nimport { DefaultCodeBlock } from \"./CodeBlock\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { withDefaultProps } from \"./withDefaults\";\nimport { DefaultCodeBlockContent } from \"./defaultComponents\";\n\nconst CodeBlockOverride: FC<CodeOverrideProps> = ({\n  components: {\n    Pre,\n    Code,\n    SyntaxHighlighter: FallbackSyntaxHighlighter,\n    CodeHeader: FallbackCodeHeader,\n    by_language = {},\n  },\n  children,\n  ...codeProps\n}) => {\n  const preProps = useContext(PreContext)!;\n  const getPreProps = withDefaultProps<any>(preProps);\n  const WrappedPre: PreComponent = useCallbackRef((props) => (\n    <Pre {...getPreProps(props)} />\n  ));\n\n  const getCodeProps = withDefaultProps<any>(codeProps);\n  const WrappedCode: CodeComponent = useCallbackRef((props) => (\n    <Code {...getCodeProps(props)} />\n  ));\n\n  const language = /language-(\\w+)/.exec(codeProps.className || \"\")?.[1] ?? \"\";\n\n  // if the code content is not string (due to rehype plugins), return a default code block\n  if (typeof children !== \"string\") {\n    return (\n      <DefaultCodeBlockContent\n        components={{ Pre: WrappedPre, Code: WrappedCode }}\n        code={children}\n      />\n    );\n  }\n\n  const SyntaxHighlighter: ComponentType<SyntaxHighlighterProps> =\n    by_language[language]?.SyntaxHighlighter ?? FallbackSyntaxHighlighter;\n\n  const CodeHeader: ComponentType<CodeHeaderProps> =\n    by_language[language]?.CodeHeader ?? FallbackCodeHeader;\n\n  return (\n    <DefaultCodeBlock\n      components={{\n        Pre: WrappedPre,\n        Code: WrappedCode,\n        SyntaxHighlighter,\n        CodeHeader,\n      }}\n      language={language || \"unknown\"}\n      code={children}\n    />\n  );\n};\n\nexport type CodeOverrideProps = ComponentPropsWithoutRef<CodeComponent> & {\n  components: {\n    Pre: PreComponent;\n    Code: CodeComponent;\n    CodeHeader: ComponentType<CodeHeaderProps>;\n    SyntaxHighlighter: ComponentType<SyntaxHighlighterProps>;\n    by_language?: Record<\n      string,\n      {\n        CodeHeader?: ComponentType<CodeHeaderProps>;\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n      }\n    >;\n  };\n};\n\nexport const CodeOverride: FC<CodeOverrideProps> = ({\n  components,\n  ...props\n}) => {\n  const preProps = useContext(PreContext);\n  if (!preProps) return <components.Code {...(props as any)} />;\n  return <CodeBlockOverride components={components} {...props} />;\n};\n","import { ComponentType, FC, useMemo } from \"react\";\n\nimport {\n  CodeComponent,\n  CodeHeaderProps,\n  PreComponent,\n  SyntaxHighlighterProps,\n} from \"./types\";\nimport { DefaultCodeBlockContent } from \"./defaultComponents\";\n\nexport type CodeBlockProps = {\n  language: string;\n  code: string;\n  components: {\n    Pre: PreComponent;\n    Code: CodeComponent;\n    CodeHeader: ComponentType<CodeHeaderProps>;\n    SyntaxHighlighter: ComponentType<SyntaxHighlighterProps>;\n  };\n};\n\nexport const DefaultCodeBlock: FC<CodeBlockProps> = ({\n  components: { Pre, Code, SyntaxHighlighter, CodeHeader },\n  language,\n  code,\n}) => {\n  const components = useMemo(() => ({ Pre, Code }), [Pre, Code]);\n\n  const SH = !!language ? SyntaxHighlighter : DefaultCodeBlockContent;\n\n  return (\n    <>\n      <CodeHeader language={language} code={code} />\n      <SH\n        components={components}\n        language={language ?? \"unknown\"}\n        code={code}\n      />\n    </>\n  );\n};\n","import classNames from \"classnames\";\n\nexport const withDefaultProps =\n  <TProps extends { className?: string | undefined }>({\n    className,\n    ...defaultProps\n  }: Partial<TProps>) =>\n  ({ className: classNameProp, ...props }: TProps) => {\n    return {\n      className: classNames(className, classNameProp),\n      ...defaultProps,\n      ...props,\n    } as TProps;\n  };\n","import { FC, memo } from \"react\";\nimport { CodeHeader } from \"./code-header\";\nimport classNames from \"classnames\";\nimport {\n  MarkdownTextPrimitive,\n  MarkdownTextPrimitiveProps,\n} from \"../primitives/MarkdownText\";\nimport {\n  withSmoothContextProvider,\n  useSmoothStatus,\n} from \"@assistant-ui/react/internal\";\n\nexport type MakeMarkdownTextProps = MarkdownTextPrimitiveProps;\n\nexport const makeMarkdownText = ({\n  className,\n  components: userComponents,\n  ...rest\n}: MakeMarkdownTextProps = {}) => {\n  const components = {\n    ...userComponents,\n    CodeHeader: userComponents?.CodeHeader ?? CodeHeader,\n  };\n\n  const MarkdownTextImpl: FC = () => {\n    const status = useSmoothStatus();\n    return (\n      <MarkdownTextPrimitive\n        components={components}\n        {...rest}\n        className={classNames(\n          \"aui-md-root\",\n          status.type === \"running\" && \"aui-md-running\",\n          className,\n        )}\n      />\n    );\n  };\n  MarkdownTextImpl.displayName = \"MarkdownText\";\n\n  return memo(withSmoothContextProvider(MarkdownTextImpl), () => true);\n};\n","import { FC } from \"react\";\nimport { CheckIcon, CopyIcon } from \"lucide-react\";\nimport { useThreadConfig } from \"@assistant-ui/react\";\nimport { TooltipIconButton } from \"@assistant-ui/react/internal\";\n\nimport { CodeHeaderProps } from \"../overrides/types\";\nimport { useCopyToClipboard } from \"./useCopyToClipboard\";\n\nexport const CodeHeader: FC<CodeHeaderProps> = ({ language, code }) => {\n  const {\n    strings: {\n      code: { header: { copy: { tooltip = \"Copy\" } = {} } = {} } = {},\n    } = {},\n  } = useThreadConfig();\n\n  const { isCopied, copyToClipboard } = useCopyToClipboard();\n  const onCopy = () => {\n    if (!code || isCopied) return;\n    copyToClipboard(code);\n  };\n\n  return (\n    <div className=\"aui-code-header-root\">\n      <span className=\"aui-code-header-language\">{language}</span>\n      <TooltipIconButton tooltip={tooltip} onClick={onCopy}>\n        {!isCopied && <CopyIcon />}\n        {isCopied && <CheckIcon />}\n      </TooltipIconButton>\n    </div>\n  );\n};\n","import { useState } from \"react\";\nimport { UseActionBarCopyProps } from \"@assistant-ui/react\";\n\nexport type useCopyToClipboardProps = UseActionBarCopyProps;\n\nexport const useCopyToClipboard = ({\n  copiedDuration = 3000,\n}: useCopyToClipboardProps = {}) => {\n  const [isCopied, setIsCopied] = useState<boolean>(false);\n\n  const copyToClipboard = (value: string) => {\n    if (!value) return;\n\n    navigator.clipboard.writeText(value).then(() => {\n      setIsCopied(true);\n      setTimeout(() => setIsCopied(false), copiedDuration);\n    });\n  };\n\n  return { isCopied, copyToClipboard };\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACEA,IAAAA,gBAAmC;AACnC,IAAAA,gBAQO;AACP,4BAA4C;;;ACZ5C,mBAAoE;AAa3D;AAVF,IAAM,iBAAa,4BAGhB,IAAI;AAEP,IAAM,yBAAyB,MAAM;AAC1C,aAAO,yBAAW,UAAU,MAAM;AACpC;AAEO,IAAM,cAA4B,CAAC,EAAE,UAAU,GAAG,KAAK,MAAM;AAClE,SAAO,4CAAC,WAAW,UAAX,EAAoB,OAAO,MAAO,UAAS;AACrD;;;ACVE,IAAAC,sBAAA;AADK,IAAM,aAA2B,CAAC,EAAE,MAAM,GAAG,KAAK,MACvD,6CAAC,SAAK,GAAG,MAAM;AAGV,IAAM,cAA6B,CAAC,EAAE,MAAM,GAAG,KAAK,MACzD,6CAAC,UAAM,GAAG,MAAM;AAGX,IAAM,0BAGR,CAAC,EAAE,YAAY,EAAE,KAAK,KAAK,GAAG,KAAK,MACtC,6CAAC,OACC,uDAAC,QAAM,gBAAK,GACd;AAGK,IAAM,oBAAoD,MAAM;;;AFCvE,IAAAC,iCAA+B;;;AGrB/B,IAAAC,gBAAwE;;;ACAxE,IAAAC,gBAA2C;AA+BvC,IAAAC,sBAAA;AAVG,IAAM,mBAAuC,CAAC;AAAA,EACnD,YAAY,EAAE,KAAK,MAAM,mBAAmB,YAAAC,YAAW;AAAA,EACvD;AAAA,EACA;AACF,MAAM;AACJ,QAAM,iBAAa,uBAAQ,OAAO,EAAE,KAAK,KAAK,IAAI,CAAC,KAAK,IAAI,CAAC;AAE7D,QAAM,KAAK,CAAC,CAAC,WAAW,oBAAoB;AAE5C,SACE,8EACE;AAAA,iDAACA,aAAA,EAAW,UAAoB,MAAY;AAAA,IAC5C;AAAA,MAAC;AAAA;AAAA,QACC;AAAA,QACA,UAAU,YAAY;AAAA,QACtB;AAAA;AAAA,IACF;AAAA,KACF;AAEJ;;;AD/BA,oCAA+B;;;AET/B,wBAAuB;AAEhB,IAAM,mBACX,CAAoD;AAAA,EAClD;AAAA,EACA,GAAG;AACL,MACA,CAAC,EAAE,WAAW,eAAe,GAAG,MAAM,MAAc;AAClD,SAAO;AAAA,IACL,eAAW,kBAAAC,SAAW,WAAW,aAAa;AAAA,IAC9C,GAAG;AAAA,IACH,GAAG;AAAA,EACL;AACF;;;AFcE,IAAAC,sBAAA;AAdJ,IAAM,oBAA2C,CAAC;AAAA,EAChD,YAAY;AAAA,IACV;AAAA,IACA;AAAA,IACA,mBAAmB;AAAA,IACnB,YAAY;AAAA,IACZ,cAAc,CAAC;AAAA,EACjB;AAAA,EACA;AAAA,EACA,GAAG;AACL,MAAM;AACJ,QAAM,eAAW,0BAAW,UAAU;AACtC,QAAM,cAAc,iBAAsB,QAAQ;AAClD,QAAM,iBAA2B,8CAAe,CAAC,UAC/C,6CAAC,OAAK,GAAG,YAAY,KAAK,GAAG,CAC9B;AAED,QAAM,eAAe,iBAAsB,SAAS;AACpD,QAAM,kBAA6B,8CAAe,CAAC,UACjD,6CAAC,QAAM,GAAG,aAAa,KAAK,GAAG,CAChC;AAED,QAAM,WAAW,iBAAiB,KAAK,UAAU,aAAa,EAAE,IAAI,CAAC,KAAK;AAG1E,MAAI,OAAO,aAAa,UAAU;AAChC,WACE;AAAA,MAAC;AAAA;AAAA,QACC,YAAY,EAAE,KAAK,YAAY,MAAM,YAAY;AAAA,QACjD,MAAM;AAAA;AAAA,IACR;AAAA,EAEJ;AAEA,QAAM,oBACJ,YAAY,QAAQ,GAAG,qBAAqB;AAE9C,QAAMC,cACJ,YAAY,QAAQ,GAAG,cAAc;AAEvC,SACE;AAAA,IAAC;AAAA;AAAA,MACC,YAAY;AAAA,QACV,KAAK;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA,YAAAA;AAAA,MACF;AAAA,MACA,UAAU,YAAY;AAAA,MACtB,MAAM;AAAA;AAAA,EACR;AAEJ;AAkBO,IAAM,eAAsC,CAAC;AAAA,EAClD;AAAA,EACA,GAAG;AACL,MAAM;AACJ,QAAM,eAAW,0BAAW,UAAU;AACtC,MAAI,CAAC,SAAU,QAAO,6CAAC,WAAW,MAAX,EAAiB,GAAI,OAAe;AAC3D,SAAO,6CAAC,qBAAkB,YAAyB,GAAG,OAAO;AAC/D;;;AHnEA,sBAA0B;AAE1B,IAAAC,qBAAuB;AA0Df,IAAAC,sBAAA;AAjCD,IAAM,4BACmC;AAAA,EAI9C,CACE;AAAA,IACE,YAAY;AAAA,IACZ;AAAA,IACA;AAAA,IACA,oBAAoB,YAAY;AAAA,IAChC,GAAG;AAAA,EACL,GACA,cACA,SAAS,SACN;AACH,UAAM;AAAA,MACJ,MAAM,EAAE,KAAK;AAAA,MACb;AAAA,IACF,QAAI,+BAAU,kCAAmB,GAAG,MAAM;AAE1C,UAAM;AAAA,MACJ,MAAM;AAAA,MACN,OAAO;AAAA,MACP,oBAAoB;AAAA,MACpB,YAAAC,cAAa;AAAA,MACb;AAAA,MACA,GAAG;AAAA,IACL,IAAI,kBAAkB,CAAC;AACvB,UAAM,aAAoC;AAAA,MACxC,GAAG;AAAA,MACH,KAAK;AAAA,MACL,UAAM,+CAAe,CAAC,UACpB;AAAA,QAAC;AAAA;AAAA,UACC,YAAY;AAAA,YACV,KAAK;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA,YAAAA;AAAA,YACA;AAAA,UACF;AAAA,UACC,GAAG;AAAA;AAAA,MACN,CACD;AAAA,IACH;AAEA,WACE;AAAA,MAAC;AAAA;AAAA,QACC,eAAa,OAAO;AAAA,QACnB,GAAG;AAAA,QACJ,eAAW,mBAAAC,SAAW,WAAW,gBAAgB,SAAS;AAAA,QAC1D,KAAK;AAAA,QAEL,uDAAC,sBAAAC,SAAA,EAAc,YAAyB,GAAG,MACxC,gBACH;AAAA;AAAA,IACF;AAAA,EAEJ;AACF;AAEA,sBAAsB,cAAc;;;AM/GpC,IAAAC,gBAAyB;;;ACCzB,0BAAoC;AACpC,IAAAC,gBAAgC;AAChC,IAAAC,mBAAkC;;;ACHlC,IAAAC,gBAAyB;AAKlB,IAAM,qBAAqB,CAAC;AAAA,EACjC,iBAAiB;AACnB,IAA6B,CAAC,MAAM;AAClC,QAAM,CAAC,UAAU,WAAW,QAAI,wBAAkB,KAAK;AAEvD,QAAM,kBAAkB,CAAC,UAAkB;AACzC,QAAI,CAAC,MAAO;AAEZ,cAAU,UAAU,UAAU,KAAK,EAAE,KAAK,MAAM;AAC9C,kBAAY,IAAI;AAChB,iBAAW,MAAM,YAAY,KAAK,GAAG,cAAc;AAAA,IACrD,CAAC;AAAA,EACH;AAEA,SAAO,EAAE,UAAU,gBAAgB;AACrC;;;ADGM,IAAAC,sBAAA;AAfC,IAAM,aAAkC,CAAC,EAAE,UAAU,KAAK,MAAM;AACrE,QAAM;AAAA,IACJ,SAAS;AAAA,MACP,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,OAAO,IAAI,CAAC,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC;AAAA,IAChE,IAAI,CAAC;AAAA,EACP,QAAI,+BAAgB;AAEpB,QAAM,EAAE,UAAU,gBAAgB,IAAI,mBAAmB;AACzD,QAAM,SAAS,MAAM;AACnB,QAAI,CAAC,QAAQ,SAAU;AACvB,oBAAgB,IAAI;AAAA,EACtB;AAEA,SACE,8CAAC,SAAI,WAAU,wBACb;AAAA,iDAAC,UAAK,WAAU,4BAA4B,oBAAS;AAAA,IACrD,8CAAC,sCAAkB,SAAkB,SAAS,QAC3C;AAAA,OAAC,YAAY,6CAAC,gCAAS;AAAA,MACvB,YAAY,6CAAC,iCAAU;AAAA,OAC1B;AAAA,KACF;AAEJ;;;AD5BA,IAAAC,qBAAuB;AAKvB,IAAAC,mBAGO;AAiBD,IAAAC,sBAAA;AAbC,IAAM,mBAAmB,CAAC;AAAA,EAC/B;AAAA,EACA,YAAY;AAAA,EACZ,GAAG;AACL,IAA2B,CAAC,MAAM;AAChC,QAAM,aAAa;AAAA,IACjB,GAAG;AAAA,IACH,YAAY,gBAAgB,cAAc;AAAA,EAC5C;AAEA,QAAM,mBAAuB,MAAM;AACjC,UAAM,aAAS,kCAAgB;AAC/B,WACE;AAAA,MAAC;AAAA;AAAA,QACC;AAAA,QACC,GAAG;AAAA,QACJ,eAAW,mBAAAC;AAAA,UACT;AAAA,UACA,OAAO,SAAS,aAAa;AAAA,UAC7B;AAAA,QACF;AAAA;AAAA,IACF;AAAA,EAEJ;AACA,mBAAiB,cAAc;AAE/B,aAAO,wBAAK,4CAA0B,gBAAgB,GAAG,MAAM,IAAI;AACrE;","names":["import_react","import_jsx_runtime","import_react_use_callback_ref","import_react","import_react","import_jsx_runtime","CodeHeader","classNames","import_jsx_runtime","CodeHeader","import_classnames","import_jsx_runtime","CodeHeader","classNames","ReactMarkdown","import_react","import_react","import_internal","import_react","import_jsx_runtime","import_classnames","import_internal","import_jsx_runtime","classNames"]}

package/dist/index.mjs

@@ -1,13 +1,19 @@
 "use client";
 
 // src/primitives/MarkdownText.tsx
-import { INTERNAL, useContentPartText } from "@assistant-ui/react";
+import { useContentPartText } from "@assistant-ui/react";
+import {
+  forwardRef
+} from "react";
 import ReactMarkdown from "react-markdown";
 
 // src/overrides/PreOverride.tsx
-import { createContext } from "react";
+import { createContext, useContext } from "react";
 import { jsx } from "react/jsx-runtime";
 var PreContext = createContext(null);
+var useIsMarkdownCodeBlock = () => {
+  return useContext(PreContext) !== null;
+};
 var PreOverride = ({ children, ...rest }) => {
   return /* @__PURE__ */ jsx(PreContext.Provider, { value: rest, children });
 };
@@ -23,7 +29,7 @@ var DefaultCodeHeader = () => null;
 import { useCallbackRef as useCallbackRef2 } from "@radix-ui/react-use-callback-ref";
 
 // src/overrides/CodeOverride.tsx
-import { useContext } from "react";
+import { useContext as useContext2 } from "react";
 
 // src/overrides/CodeBlock.tsx
 import { useMemo } from "react";
@@ -77,7 +83,7 @@ var CodeBlockOverride = ({
   children,
   ...codeProps
 }) => {
-  const preProps = useContext(PreContext);
+  const preProps = useContext2(PreContext);
   const getPreProps = withDefaultProps(preProps);
   const WrappedPre = useCallbackRef((props) => /* @__PURE__ */ jsx4(Pre, { ...getPreProps(props) }));
   const getCodeProps = withDefaultProps(codeProps);
@@ -112,57 +118,73 @@ var CodeOverride = ({
   components,
   ...props
 }) => {
-  const preProps = useContext(PreContext);
+  const preProps = useContext2(PreContext);
   if (!preProps) return /* @__PURE__ */ jsx4(components.Code, { ...props });
   return /* @__PURE__ */ jsx4(CodeBlockOverride, { components, ...props });
 };
 
 // src/primitives/MarkdownText.tsx
+import { useSmooth } from "@assistant-ui/react/internal";
+import classNames2 from "classnames";
 import { jsx as jsx5 } from "react/jsx-runtime";
-var { useSmooth } = INTERNAL;
-var MarkdownTextPrimitive = ({
-  smooth = true,
-  components: userComponents,
-  ...rest
-}) => {
-  const {
-    part: { text }
-  } = useContentPartText();
-  const smoothText = useSmooth(text, smooth);
-  const {
-    pre = DefaultPre,
-    code = DefaultCode,
-    SyntaxHighlighter = DefaultCodeBlockContent,
-    CodeHeader: CodeHeader2 = DefaultCodeHeader,
-    by_language,
-    ...componentsRest
-  } = userComponents ?? {};
-  const components = {
-    ...componentsRest,
-    pre: PreOverride,
-    code: useCallbackRef2((props) => /* @__PURE__ */ jsx5(
-      CodeOverride,
+var MarkdownTextPrimitive = forwardRef(
+  ({
+    components: userComponents,
+    className,
+    containerProps,
+    containerComponent: Container = "div",
+    ...rest
+  }, forwardedRef, smooth = true) => {
+    const {
+      part: { text },
+      status
+    } = useSmooth(useContentPartText(), smooth);
+    const {
+      pre = DefaultPre,
+      code = DefaultCode,
+      SyntaxHighlighter = DefaultCodeBlockContent,
+      CodeHeader: CodeHeader2 = DefaultCodeHeader,
+      by_language,
+      ...componentsRest
+    } = userComponents ?? {};
+    const components = {
+      ...componentsRest,
+      pre: PreOverride,
+      code: useCallbackRef2((props) => /* @__PURE__ */ jsx5(
+        CodeOverride,
+        {
+          components: {
+            Pre: pre,
+            Code: code,
+            SyntaxHighlighter,
+            CodeHeader: CodeHeader2,
+            by_language
+          },
+          ...props
+        }
+      ))
+    };
+    return /* @__PURE__ */ jsx5(
+      Container,
       {
-        components: {
-          Pre: pre,
-          Code: code,
-          SyntaxHighlighter,
-          CodeHeader: CodeHeader2,
-          by_language
-        },
-        ...props
+        "data-status": status.type,
+        ...containerProps,
+        className: classNames2(className, containerProps?.className),
+        ref: forwardedRef,
+        children: /* @__PURE__ */ jsx5(ReactMarkdown, { components, ...rest, children: text })
       }
-    ))
-  };
-  return /* @__PURE__ */ jsx5(ReactMarkdown, { components, ...rest, children: smoothText });
-};
+    );
+  }
+);
+MarkdownTextPrimitive.displayName = "MarkdownTextPrimitive";
 
 // src/ui/markdown-text.tsx
 import { memo } from "react";
 
 // src/ui/code-header.tsx
 import { CheckIcon, CopyIcon } from "lucide-react";
-import { useThreadConfig, INTERNAL as INTERNAL2 } from "@assistant-ui/react";
+import { useThreadConfig } from "@assistant-ui/react";
+import { TooltipIconButton } from "@assistant-ui/react/internal";
 
 // src/ui/useCopyToClipboard.tsx
 import { useState } from "react";
@@ -182,7 +204,6 @@ var useCopyToClipboard = ({
 
 // src/ui/code-header.tsx
 import { jsx as jsx6, jsxs as jsxs2 } from "react/jsx-runtime";
-var { TooltipIconButton } = INTERNAL2;
 var CodeHeader = ({ language, code }) => {
   const {
     strings: {
@@ -204,7 +225,11 @@ var CodeHeader = ({ language, code }) => {
 };
 
 // src/ui/markdown-text.tsx
-import classNames2 from "classnames";
+import classNames3 from "classnames";
+import {
+  withSmoothContextProvider,
+  useSmoothStatus
+} from "@assistant-ui/react/internal";
 import { jsx as jsx7 } from "react/jsx-runtime";
 var makeMarkdownText = ({
   className,
@@ -215,28 +240,28 @@ var makeMarkdownText = ({
     ...userComponents,
     CodeHeader: userComponents?.CodeHeader ?? CodeHeader
   };
-  const MarkdownTextImpl = ({ status }) => {
+  const MarkdownTextImpl = () => {
+    const status = useSmoothStatus();
     return /* @__PURE__ */ jsx7(
-      "div",
+      MarkdownTextPrimitive,
       {
-        className: classNames2(
+        components,
+        ...rest,
+        className: classNames3(
           "aui-md-root",
-          status.type === "running" && "aui-md-in-progress",
+          status.type === "running" && "aui-md-running",
           className
-        ),
-        children: /* @__PURE__ */ jsx7(MarkdownTextPrimitive, { components, ...rest })
+        )
       }
     );
   };
   MarkdownTextImpl.displayName = "MarkdownText";
-  return memo(
-    MarkdownTextImpl,
-    (prev, next) => prev.status.type === next.status.type
-  );
+  return memo(withSmoothContextProvider(MarkdownTextImpl), () => true);
 };
 export {
   CodeHeader,
   MarkdownTextPrimitive,
-  makeMarkdownText
+  makeMarkdownText,
+  useIsMarkdownCodeBlock
 };
 //# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/primitives/MarkdownText.tsx","../src/overrides/PreOverride.tsx","../src/overrides/defaultComponents.tsx","../src/overrides/CodeOverride.tsx","../src/overrides/CodeBlock.tsx","../src/overrides/withDefaults.tsx","../src/ui/markdown-text.tsx","../src/ui/code-header.tsx","../src/ui/useCopyToClipboard.tsx"],"sourcesContent":["\"use client\";\n\nimport { INTERNAL, useContentPartText } from \"@assistant-ui/react\";\nimport type { ComponentType, FC } 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\";\n\nconst { useSmooth } = INTERNAL;\n\nexport type MarkdownTextPrimitiveProps = Omit<\n  Options,\n  \"components\" | \"children\"\n> & {\n  components?: NonNullable<Options[\"components\"]> & {\n    SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n    CodeHeader?: ComponentType<CodeHeaderProps>;\n    by_language?: Record<\n      string,\n      {\n        CodeHeader?: ComponentType<CodeHeaderProps>;\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n      }\n    >;\n  };\n  smooth?: boolean;\n};\nexport const MarkdownTextPrimitive: FC<MarkdownTextPrimitiveProps> = ({\n  smooth = true,\n  components: userComponents,\n  ...rest\n}) => {\n  const {\n    part: { text },\n  } = useContentPartText();\n  const smoothText = useSmooth(text, smooth); // TODO loading indicator disappears before smooth animation ends\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: typeof userComponents = {\n    ...componentsRest,\n    pre: PreOverride,\n    code: useCallbackRef((props) => (\n      <CodeOverride\n        components={{\n          Pre: pre,\n          Code: code,\n          SyntaxHighlighter,\n          CodeHeader,\n          by_language,\n        }}\n        {...props}\n      />\n    )),\n  };\n\n  return (\n    <ReactMarkdown components={components} {...rest}>\n      {smoothText}\n    </ReactMarkdown>\n  );\n};\n","import { createContext, ComponentPropsWithoutRef } from \"react\";\nimport { PreComponent } from \"./types\";\n\nexport const PreContext = createContext<Omit<\n  ComponentPropsWithoutRef<PreComponent>,\n  \"children\"\n> | null>(null);\n\nexport const PreOverride: PreComponent = ({ children, ...rest }) => {\n  return <PreContext.Provider value={rest}>{children}</PreContext.Provider>;\n};\n","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","import { ComponentPropsWithoutRef, ComponentType, FC, useContext } from \"react\";\nimport { PreContext } from \"./PreOverride\";\nimport {\n  CodeComponent,\n  CodeHeaderProps,\n  PreComponent,\n  SyntaxHighlighterProps,\n} from \"./types\";\nimport { DefaultCodeBlock } from \"./CodeBlock\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { withDefaultProps } from \"./withDefaults\";\nimport { DefaultCodeBlockContent } from \"./defaultComponents\";\n\nconst CodeBlockOverride: FC<CodeOverrideProps> = ({\n  components: {\n    Pre,\n    Code,\n    SyntaxHighlighter: FallbackSyntaxHighlighter,\n    CodeHeader: FallbackCodeHeader,\n    by_language = {},\n  },\n  children,\n  ...codeProps\n}) => {\n  const preProps = useContext(PreContext)!;\n  const getPreProps = withDefaultProps<any>(preProps);\n  const WrappedPre: PreComponent = useCallbackRef((props) => (\n    <Pre {...getPreProps(props)} />\n  ));\n\n  const getCodeProps = withDefaultProps<any>(codeProps);\n  const WrappedCode: CodeComponent = useCallbackRef((props) => (\n    <Code {...getCodeProps(props)} />\n  ));\n\n  const language = /language-(\\w+)/.exec(codeProps.className || \"\")?.[1] ?? \"\";\n\n  // if the code content is not string (due to rehype plugins), return a default code block\n  if (typeof children !== \"string\") {\n    return (\n      <DefaultCodeBlockContent\n        components={{ Pre: WrappedPre, Code: WrappedCode }}\n        code={children}\n      />\n    );\n  }\n\n  const SyntaxHighlighter: ComponentType<SyntaxHighlighterProps> =\n    by_language[language]?.SyntaxHighlighter ?? FallbackSyntaxHighlighter;\n\n  const CodeHeader: ComponentType<CodeHeaderProps> =\n    by_language[language]?.CodeHeader ?? FallbackCodeHeader;\n\n  return (\n    <DefaultCodeBlock\n      components={{\n        Pre: WrappedPre,\n        Code: WrappedCode,\n        SyntaxHighlighter,\n        CodeHeader,\n      }}\n      language={language || \"unknown\"}\n      code={children}\n    />\n  );\n};\n\nexport type CodeOverrideProps = ComponentPropsWithoutRef<CodeComponent> & {\n  components: {\n    Pre: PreComponent;\n    Code: CodeComponent;\n    CodeHeader: ComponentType<CodeHeaderProps>;\n    SyntaxHighlighter: ComponentType<SyntaxHighlighterProps>;\n    by_language?: Record<\n      string,\n      {\n        CodeHeader?: ComponentType<CodeHeaderProps>;\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n      }\n    >;\n  };\n};\n\nexport const CodeOverride: FC<CodeOverrideProps> = ({\n  components,\n  ...props\n}) => {\n  const preProps = useContext(PreContext);\n  if (!preProps) return <components.Code {...(props as any)} />;\n  return <CodeBlockOverride components={components} {...props} />;\n};\n","import { ComponentType, FC, useMemo } from \"react\";\n\nimport {\n  CodeComponent,\n  CodeHeaderProps,\n  PreComponent,\n  SyntaxHighlighterProps,\n} from \"./types\";\nimport { DefaultCodeBlockContent } from \"./defaultComponents\";\n\nexport type CodeBlockProps = {\n  language: string;\n  code: string;\n  components: {\n    Pre: PreComponent;\n    Code: CodeComponent;\n    CodeHeader: ComponentType<CodeHeaderProps>;\n    SyntaxHighlighter: ComponentType<SyntaxHighlighterProps>;\n  };\n};\n\nexport const DefaultCodeBlock: FC<CodeBlockProps> = ({\n  components: { Pre, Code, SyntaxHighlighter, CodeHeader },\n  language,\n  code,\n}) => {\n  const components = useMemo(() => ({ Pre, Code }), [Pre, Code]);\n\n  const SH = !!language ? SyntaxHighlighter : DefaultCodeBlockContent;\n\n  return (\n    <>\n      <CodeHeader language={language} code={code} />\n      <SH\n        components={components}\n        language={language ?? \"unknown\"}\n        code={code}\n      />\n    </>\n  );\n};\n","import classNames from \"classnames\";\n\nexport const withDefaultProps =\n  <TProps extends { className?: string | undefined }>({\n    className,\n    ...defaultProps\n  }: Partial<TProps>) =>\n  ({ className: classNameProp, ...props }: TProps) => {\n    return {\n      className: classNames(className, classNameProp),\n      ...defaultProps,\n      ...props,\n    } as TProps;\n  };\n","import { TextContentPartProps } from \"@assistant-ui/react\";\nimport { FC, memo } from \"react\";\nimport { CodeHeader } from \"./code-header\";\nimport classNames from \"classnames\";\nimport {\n  MarkdownTextPrimitive,\n  MarkdownTextPrimitiveProps,\n} from \"../primitives/MarkdownText\";\n\nexport type MakeMarkdownTextProps = MarkdownTextPrimitiveProps;\n\nexport const makeMarkdownText = ({\n  className,\n  components: userComponents,\n  ...rest\n}: MakeMarkdownTextProps = {}) => {\n  const components = {\n    ...userComponents,\n    CodeHeader: userComponents?.CodeHeader ?? CodeHeader,\n  };\n\n  const MarkdownTextImpl: FC<TextContentPartProps> = ({ status }) => {\n    return (\n      <div\n        className={classNames(\n          \"aui-md-root\",\n          status.type === \"running\" && \"aui-md-in-progress\",\n          className,\n        )}\n      >\n        <MarkdownTextPrimitive components={components} {...rest} />\n      </div>\n    );\n  };\n  MarkdownTextImpl.displayName = \"MarkdownText\";\n\n  return memo(\n    MarkdownTextImpl,\n    (prev, next) => prev.status.type === next.status.type,\n  );\n};\n","import { FC } from \"react\";\nimport { CheckIcon, CopyIcon } from \"lucide-react\";\nimport { useThreadConfig, INTERNAL } from \"@assistant-ui/react\";\n\nimport { CodeHeaderProps } from \"../overrides/types\";\nimport { useCopyToClipboard } from \"./useCopyToClipboard\";\n\nconst { TooltipIconButton } = INTERNAL;\n\nexport const CodeHeader: FC<CodeHeaderProps> = ({ language, code }) => {\n  const {\n    strings: {\n      code: { header: { copy: { tooltip = \"Copy\" } = {} } = {} } = {},\n    } = {},\n  } = useThreadConfig();\n\n  const { isCopied, copyToClipboard } = useCopyToClipboard();\n  const onCopy = () => {\n    if (!code || isCopied) return;\n    copyToClipboard(code);\n  };\n\n  return (\n    <div className=\"aui-code-header-root\">\n      <span className=\"aui-code-header-language\">{language}</span>\n      <TooltipIconButton tooltip={tooltip} onClick={onCopy}>\n        {!isCopied && <CopyIcon />}\n        {isCopied && <CheckIcon />}\n      </TooltipIconButton>\n    </div>\n  );\n};\n","import { useState } from \"react\";\nimport { UseActionBarCopyProps } from \"@assistant-ui/react\";\n\nexport type useCopyToClipboardProps = UseActionBarCopyProps;\n\nexport const useCopyToClipboard = ({\n  copiedDuration = 3000,\n}: useCopyToClipboardProps = {}) => {\n  const [isCopied, setIsCopied] = useState<boolean>(false);\n\n  const copyToClipboard = (value: string) => {\n    if (!value) return;\n\n    navigator.clipboard.writeText(value).then(() => {\n      setIsCopied(true);\n      setTimeout(() => setIsCopied(false), copiedDuration);\n    });\n  };\n\n  return { isCopied, copyToClipboard };\n};\n"],"mappings":";;;AAEA,SAAS,UAAU,0BAA0B;AAE7C,OAAO,mBAAqC;;;ACJ5C,SAAS,qBAA+C;AAS/C;AANF,IAAM,aAAa,cAGhB,IAAI;AAEP,IAAM,cAA4B,CAAC,EAAE,UAAU,GAAG,KAAK,MAAM;AAClE,SAAO,oBAAC,WAAW,UAAX,EAAoB,OAAO,MAAO,UAAS;AACrD;;;ACNE,gBAAAA,YAAA;AADK,IAAM,aAA2B,CAAC,EAAE,MAAM,GAAG,KAAK,MACvD,gBAAAA,KAAC,SAAK,GAAG,MAAM;AAGV,IAAM,cAA6B,CAAC,EAAE,MAAM,GAAG,KAAK,MACzD,gBAAAA,KAAC,UAAM,GAAG,MAAM;AAGX,IAAM,0BAGR,CAAC,EAAE,YAAY,EAAE,KAAK,KAAK,GAAG,KAAK,MACtC,gBAAAA,KAAC,OACC,0BAAAA,KAAC,QAAM,gBAAK,GACd;AAGK,IAAM,oBAAoD,MAAM;;;AFPvE,SAAS,kBAAAC,uBAAsB;;;AGb/B,SAAsD,kBAAkB;;;ACAxE,SAA4B,eAAe;AA+BvC,mBACE,OAAAC,MADF;AAVG,IAAM,mBAAuC,CAAC;AAAA,EACnD,YAAY,EAAE,KAAK,MAAM,mBAAmB,YAAAC,YAAW;AAAA,EACvD;AAAA,EACA;AACF,MAAM;AACJ,QAAM,aAAa,QAAQ,OAAO,EAAE,KAAK,KAAK,IAAI,CAAC,KAAK,IAAI,CAAC;AAE7D,QAAM,KAAK,CAAC,CAAC,WAAW,oBAAoB;AAE5C,SACE,iCACE;AAAA,oBAAAD,KAACC,aAAA,EAAW,UAAoB,MAAY;AAAA,IAC5C,gBAAAD;AAAA,MAAC;AAAA;AAAA,QACC;AAAA,QACA,UAAU,YAAY;AAAA,QACtB;AAAA;AAAA,IACF;AAAA,KACF;AAEJ;;;AD/BA,SAAS,sBAAsB;;;AET/B,OAAO,gBAAgB;AAEhB,IAAM,mBACX,CAAoD;AAAA,EAClD;AAAA,EACA,GAAG;AACL,MACA,CAAC,EAAE,WAAW,eAAe,GAAG,MAAM,MAAc;AAClD,SAAO;AAAA,IACL,WAAW,WAAW,WAAW,aAAa;AAAA,IAC9C,GAAG;AAAA,IACH,GAAG;AAAA,EACL;AACF;;;AFcE,gBAAAE,YAAA;AAdJ,IAAM,oBAA2C,CAAC;AAAA,EAChD,YAAY;AAAA,IACV;AAAA,IACA;AAAA,IACA,mBAAmB;AAAA,IACnB,YAAY;AAAA,IACZ,cAAc,CAAC;AAAA,EACjB;AAAA,EACA;AAAA,EACA,GAAG;AACL,MAAM;AACJ,QAAM,WAAW,WAAW,UAAU;AACtC,QAAM,cAAc,iBAAsB,QAAQ;AAClD,QAAM,aAA2B,eAAe,CAAC,UAC/C,gBAAAA,KAAC,OAAK,GAAG,YAAY,KAAK,GAAG,CAC9B;AAED,QAAM,eAAe,iBAAsB,SAAS;AACpD,QAAM,cAA6B,eAAe,CAAC,UACjD,gBAAAA,KAAC,QAAM,GAAG,aAAa,KAAK,GAAG,CAChC;AAED,QAAM,WAAW,iBAAiB,KAAK,UAAU,aAAa,EAAE,IAAI,CAAC,KAAK;AAG1E,MAAI,OAAO,aAAa,UAAU;AAChC,WACE,gBAAAA;AAAA,MAAC;AAAA;AAAA,QACC,YAAY,EAAE,KAAK,YAAY,MAAM,YAAY;AAAA,QACjD,MAAM;AAAA;AAAA,IACR;AAAA,EAEJ;AAEA,QAAM,oBACJ,YAAY,QAAQ,GAAG,qBAAqB;AAE9C,QAAMC,cACJ,YAAY,QAAQ,GAAG,cAAc;AAEvC,SACE,gBAAAD;AAAA,IAAC;AAAA;AAAA,MACC,YAAY;AAAA,QACV,KAAK;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA,YAAAC;AAAA,MACF;AAAA,MACA,UAAU,YAAY;AAAA,MACtB,MAAM;AAAA;AAAA,EACR;AAEJ;AAkBO,IAAM,eAAsC,CAAC;AAAA,EAClD;AAAA,EACA,GAAG;AACL,MAAM;AACJ,QAAM,WAAW,WAAW,UAAU;AACtC,MAAI,CAAC,SAAU,QAAO,gBAAAD,KAAC,WAAW,MAAX,EAAiB,GAAI,OAAe;AAC3D,SAAO,gBAAAA,KAAC,qBAAkB,YAAyB,GAAG,OAAO;AAC/D;;;AHjCM,gBAAAE,YAAA;AAzCN,IAAM,EAAE,UAAU,IAAI;AAmBf,IAAM,wBAAwD,CAAC;AAAA,EACpE,SAAS;AAAA,EACT,YAAY;AAAA,EACZ,GAAG;AACL,MAAM;AACJ,QAAM;AAAA,IACJ,MAAM,EAAE,KAAK;AAAA,EACf,IAAI,mBAAmB;AACvB,QAAM,aAAa,UAAU,MAAM,MAAM;AAEzC,QAAM;AAAA,IACJ,MAAM;AAAA,IACN,OAAO;AAAA,IACP,oBAAoB;AAAA,IACpB,YAAAC,cAAa;AAAA,IACb;AAAA,IACA,GAAG;AAAA,EACL,IAAI,kBAAkB,CAAC;AACvB,QAAM,aAAoC;AAAA,IACxC,GAAG;AAAA,IACH,KAAK;AAAA,IACL,MAAMC,gBAAe,CAAC,UACpB,gBAAAF;AAAA,MAAC;AAAA;AAAA,QACC,YAAY;AAAA,UACV,KAAK;AAAA,UACL,MAAM;AAAA,UACN;AAAA,UACA,YAAAC;AAAA,UACA;AAAA,QACF;AAAA,QACC,GAAG;AAAA;AAAA,IACN,CACD;AAAA,EACH;AAEA,SACE,gBAAAD,KAAC,iBAAc,YAAyB,GAAG,MACxC,sBACH;AAEJ;;;AM1EA,SAAa,YAAY;;;ACAzB,SAAS,WAAW,gBAAgB;AACpC,SAAS,iBAAiB,YAAAG,iBAAgB;;;ACF1C,SAAS,gBAAgB;AAKlB,IAAM,qBAAqB,CAAC;AAAA,EACjC,iBAAiB;AACnB,IAA6B,CAAC,MAAM;AAClC,QAAM,CAAC,UAAU,WAAW,IAAI,SAAkB,KAAK;AAEvD,QAAM,kBAAkB,CAAC,UAAkB;AACzC,QAAI,CAAC,MAAO;AAEZ,cAAU,UAAU,UAAU,KAAK,EAAE,KAAK,MAAM;AAC9C,kBAAY,IAAI;AAChB,iBAAW,MAAM,YAAY,KAAK,GAAG,cAAc;AAAA,IACrD,CAAC;AAAA,EACH;AAEA,SAAO,EAAE,UAAU,gBAAgB;AACrC;;;ADIM,gBAAAC,MACA,QAAAC,aADA;AAjBN,IAAM,EAAE,kBAAkB,IAAIC;AAEvB,IAAM,aAAkC,CAAC,EAAE,UAAU,KAAK,MAAM;AACrE,QAAM;AAAA,IACJ,SAAS;AAAA,MACP,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,OAAO,IAAI,CAAC,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC;AAAA,IAChE,IAAI,CAAC;AAAA,EACP,IAAI,gBAAgB;AAEpB,QAAM,EAAE,UAAU,gBAAgB,IAAI,mBAAmB;AACzD,QAAM,SAAS,MAAM;AACnB,QAAI,CAAC,QAAQ,SAAU;AACvB,oBAAgB,IAAI;AAAA,EACtB;AAEA,SACE,gBAAAD,MAAC,SAAI,WAAU,wBACb;AAAA,oBAAAD,KAAC,UAAK,WAAU,4BAA4B,oBAAS;AAAA,IACrD,gBAAAC,MAAC,qBAAkB,SAAkB,SAAS,QAC3C;AAAA,OAAC,YAAY,gBAAAD,KAAC,YAAS;AAAA,MACvB,YAAY,gBAAAA,KAAC,aAAU;AAAA,OAC1B;AAAA,KACF;AAEJ;;;AD5BA,OAAOG,iBAAgB;AA2Bf,gBAAAC,YAAA;AAnBD,IAAM,mBAAmB,CAAC;AAAA,EAC/B;AAAA,EACA,YAAY;AAAA,EACZ,GAAG;AACL,IAA2B,CAAC,MAAM;AAChC,QAAM,aAAa;AAAA,IACjB,GAAG;AAAA,IACH,YAAY,gBAAgB,cAAc;AAAA,EAC5C;AAEA,QAAM,mBAA6C,CAAC,EAAE,OAAO,MAAM;AACjE,WACE,gBAAAA;AAAA,MAAC;AAAA;AAAA,QACC,WAAWC;AAAA,UACT;AAAA,UACA,OAAO,SAAS,aAAa;AAAA,UAC7B;AAAA,QACF;AAAA,QAEA,0BAAAD,KAAC,yBAAsB,YAAyB,GAAG,MAAM;AAAA;AAAA,IAC3D;AAAA,EAEJ;AACA,mBAAiB,cAAc;AAE/B,SAAO;AAAA,IACL;AAAA,IACA,CAAC,MAAM,SAAS,KAAK,OAAO,SAAS,KAAK,OAAO;AAAA,EACnD;AACF;","names":["jsx","useCallbackRef","jsx","CodeHeader","jsx","CodeHeader","jsx","CodeHeader","useCallbackRef","INTERNAL","jsx","jsxs","INTERNAL","classNames","jsx","classNames"]}
+{"version":3,"sources":["../src/primitives/MarkdownText.tsx","../src/overrides/PreOverride.tsx","../src/overrides/defaultComponents.tsx","../src/overrides/CodeOverride.tsx","../src/overrides/CodeBlock.tsx","../src/overrides/withDefaults.tsx","../src/ui/markdown-text.tsx","../src/ui/code-header.tsx","../src/ui/useCopyToClipboard.tsx"],"sourcesContent":["\"use client\";\n\nimport { useContentPartText } from \"@assistant-ui/react\";\nimport {\n  ElementRef,\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 { useSmooth } from \"@assistant-ui/react/internal\";\nimport { Primitive } from \"@radix-ui/react-primitive\";\nimport classNames from \"classnames\";\n\ntype MarkdownTextPrimitiveElement = ElementRef<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\">;\n  containerComponent?: ElementType;\n  components?: NonNullable<Options[\"components\"]> & {\n    SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n    CodeHeader?: ComponentType<CodeHeaderProps>;\n    by_language?: Record<\n      string,\n      {\n        CodeHeader?: ComponentType<CodeHeaderProps>;\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n      }\n    >;\n  };\n  smooth?: boolean;\n};\n\nexport const MarkdownTextPrimitive: ForwardRefExoticComponent<MarkdownTextPrimitiveProps> &\n  RefAttributes<MarkdownTextPrimitiveElement> = forwardRef<\n  MarkdownTextPrimitiveElement,\n  MarkdownTextPrimitiveProps\n>(\n  (\n    {\n      components: userComponents,\n      className,\n      containerProps,\n      containerComponent: Container = \"div\",\n      ...rest\n    },\n    forwardedRef,\n    smooth = true,\n  ) => {\n    const {\n      part: { text },\n      status,\n    } = 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: typeof userComponents = {\n      ...componentsRest,\n      pre: PreOverride,\n      code: useCallbackRef((props) => (\n        <CodeOverride\n          components={{\n            Pre: pre,\n            Code: code,\n            SyntaxHighlighter,\n            CodeHeader,\n            by_language,\n          }}\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","import { createContext, ComponentPropsWithoutRef, useContext } from \"react\";\nimport { PreComponent } from \"./types\";\n\nexport const PreContext = createContext<Omit<\n  ComponentPropsWithoutRef<PreComponent>,\n  \"children\"\n> | null>(null);\n\nexport const useIsMarkdownCodeBlock = () => {\n  return useContext(PreContext) !== null;\n};\n\nexport const PreOverride: PreComponent = ({ children, ...rest }) => {\n  return <PreContext.Provider value={rest}>{children}</PreContext.Provider>;\n};\n","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","import { ComponentPropsWithoutRef, ComponentType, FC, useContext } from \"react\";\nimport { PreContext } from \"./PreOverride\";\nimport {\n  CodeComponent,\n  CodeHeaderProps,\n  PreComponent,\n  SyntaxHighlighterProps,\n} from \"./types\";\nimport { DefaultCodeBlock } from \"./CodeBlock\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { withDefaultProps } from \"./withDefaults\";\nimport { DefaultCodeBlockContent } from \"./defaultComponents\";\n\nconst CodeBlockOverride: FC<CodeOverrideProps> = ({\n  components: {\n    Pre,\n    Code,\n    SyntaxHighlighter: FallbackSyntaxHighlighter,\n    CodeHeader: FallbackCodeHeader,\n    by_language = {},\n  },\n  children,\n  ...codeProps\n}) => {\n  const preProps = useContext(PreContext)!;\n  const getPreProps = withDefaultProps<any>(preProps);\n  const WrappedPre: PreComponent = useCallbackRef((props) => (\n    <Pre {...getPreProps(props)} />\n  ));\n\n  const getCodeProps = withDefaultProps<any>(codeProps);\n  const WrappedCode: CodeComponent = useCallbackRef((props) => (\n    <Code {...getCodeProps(props)} />\n  ));\n\n  const language = /language-(\\w+)/.exec(codeProps.className || \"\")?.[1] ?? \"\";\n\n  // if the code content is not string (due to rehype plugins), return a default code block\n  if (typeof children !== \"string\") {\n    return (\n      <DefaultCodeBlockContent\n        components={{ Pre: WrappedPre, Code: WrappedCode }}\n        code={children}\n      />\n    );\n  }\n\n  const SyntaxHighlighter: ComponentType<SyntaxHighlighterProps> =\n    by_language[language]?.SyntaxHighlighter ?? FallbackSyntaxHighlighter;\n\n  const CodeHeader: ComponentType<CodeHeaderProps> =\n    by_language[language]?.CodeHeader ?? FallbackCodeHeader;\n\n  return (\n    <DefaultCodeBlock\n      components={{\n        Pre: WrappedPre,\n        Code: WrappedCode,\n        SyntaxHighlighter,\n        CodeHeader,\n      }}\n      language={language || \"unknown\"}\n      code={children}\n    />\n  );\n};\n\nexport type CodeOverrideProps = ComponentPropsWithoutRef<CodeComponent> & {\n  components: {\n    Pre: PreComponent;\n    Code: CodeComponent;\n    CodeHeader: ComponentType<CodeHeaderProps>;\n    SyntaxHighlighter: ComponentType<SyntaxHighlighterProps>;\n    by_language?: Record<\n      string,\n      {\n        CodeHeader?: ComponentType<CodeHeaderProps>;\n        SyntaxHighlighter?: ComponentType<SyntaxHighlighterProps>;\n      }\n    >;\n  };\n};\n\nexport const CodeOverride: FC<CodeOverrideProps> = ({\n  components,\n  ...props\n}) => {\n  const preProps = useContext(PreContext);\n  if (!preProps) return <components.Code {...(props as any)} />;\n  return <CodeBlockOverride components={components} {...props} />;\n};\n","import { ComponentType, FC, useMemo } from \"react\";\n\nimport {\n  CodeComponent,\n  CodeHeaderProps,\n  PreComponent,\n  SyntaxHighlighterProps,\n} from \"./types\";\nimport { DefaultCodeBlockContent } from \"./defaultComponents\";\n\nexport type CodeBlockProps = {\n  language: string;\n  code: string;\n  components: {\n    Pre: PreComponent;\n    Code: CodeComponent;\n    CodeHeader: ComponentType<CodeHeaderProps>;\n    SyntaxHighlighter: ComponentType<SyntaxHighlighterProps>;\n  };\n};\n\nexport const DefaultCodeBlock: FC<CodeBlockProps> = ({\n  components: { Pre, Code, SyntaxHighlighter, CodeHeader },\n  language,\n  code,\n}) => {\n  const components = useMemo(() => ({ Pre, Code }), [Pre, Code]);\n\n  const SH = !!language ? SyntaxHighlighter : DefaultCodeBlockContent;\n\n  return (\n    <>\n      <CodeHeader language={language} code={code} />\n      <SH\n        components={components}\n        language={language ?? \"unknown\"}\n        code={code}\n      />\n    </>\n  );\n};\n","import classNames from \"classnames\";\n\nexport const withDefaultProps =\n  <TProps extends { className?: string | undefined }>({\n    className,\n    ...defaultProps\n  }: Partial<TProps>) =>\n  ({ className: classNameProp, ...props }: TProps) => {\n    return {\n      className: classNames(className, classNameProp),\n      ...defaultProps,\n      ...props,\n    } as TProps;\n  };\n","import { FC, memo } from \"react\";\nimport { CodeHeader } from \"./code-header\";\nimport classNames from \"classnames\";\nimport {\n  MarkdownTextPrimitive,\n  MarkdownTextPrimitiveProps,\n} from \"../primitives/MarkdownText\";\nimport {\n  withSmoothContextProvider,\n  useSmoothStatus,\n} from \"@assistant-ui/react/internal\";\n\nexport type MakeMarkdownTextProps = MarkdownTextPrimitiveProps;\n\nexport const makeMarkdownText = ({\n  className,\n  components: userComponents,\n  ...rest\n}: MakeMarkdownTextProps = {}) => {\n  const components = {\n    ...userComponents,\n    CodeHeader: userComponents?.CodeHeader ?? CodeHeader,\n  };\n\n  const MarkdownTextImpl: FC = () => {\n    const status = useSmoothStatus();\n    return (\n      <MarkdownTextPrimitive\n        components={components}\n        {...rest}\n        className={classNames(\n          \"aui-md-root\",\n          status.type === \"running\" && \"aui-md-running\",\n          className,\n        )}\n      />\n    );\n  };\n  MarkdownTextImpl.displayName = \"MarkdownText\";\n\n  return memo(withSmoothContextProvider(MarkdownTextImpl), () => true);\n};\n","import { FC } from \"react\";\nimport { CheckIcon, CopyIcon } from \"lucide-react\";\nimport { useThreadConfig } from \"@assistant-ui/react\";\nimport { TooltipIconButton } from \"@assistant-ui/react/internal\";\n\nimport { CodeHeaderProps } from \"../overrides/types\";\nimport { useCopyToClipboard } from \"./useCopyToClipboard\";\n\nexport const CodeHeader: FC<CodeHeaderProps> = ({ language, code }) => {\n  const {\n    strings: {\n      code: { header: { copy: { tooltip = \"Copy\" } = {} } = {} } = {},\n    } = {},\n  } = useThreadConfig();\n\n  const { isCopied, copyToClipboard } = useCopyToClipboard();\n  const onCopy = () => {\n    if (!code || isCopied) return;\n    copyToClipboard(code);\n  };\n\n  return (\n    <div className=\"aui-code-header-root\">\n      <span className=\"aui-code-header-language\">{language}</span>\n      <TooltipIconButton tooltip={tooltip} onClick={onCopy}>\n        {!isCopied && <CopyIcon />}\n        {isCopied && <CheckIcon />}\n      </TooltipIconButton>\n    </div>\n  );\n};\n","import { useState } from \"react\";\nimport { UseActionBarCopyProps } from \"@assistant-ui/react\";\n\nexport type useCopyToClipboardProps = UseActionBarCopyProps;\n\nexport const useCopyToClipboard = ({\n  copiedDuration = 3000,\n}: useCopyToClipboardProps = {}) => {\n  const [isCopied, setIsCopied] = useState<boolean>(false);\n\n  const copyToClipboard = (value: string) => {\n    if (!value) return;\n\n    navigator.clipboard.writeText(value).then(() => {\n      setIsCopied(true);\n      setTimeout(() => setIsCopied(false), copiedDuration);\n    });\n  };\n\n  return { isCopied, copyToClipboard };\n};\n"],"mappings":";;;AAEA,SAAS,0BAA0B;AACnC;AAAA,EAGE;AAAA,OAKK;AACP,OAAO,mBAAqC;;;ACZ5C,SAAS,eAAyC,kBAAkB;AAa3D;AAVF,IAAM,aAAa,cAGhB,IAAI;AAEP,IAAM,yBAAyB,MAAM;AAC1C,SAAO,WAAW,UAAU,MAAM;AACpC;AAEO,IAAM,cAA4B,CAAC,EAAE,UAAU,GAAG,KAAK,MAAM;AAClE,SAAO,oBAAC,WAAW,UAAX,EAAoB,OAAO,MAAO,UAAS;AACrD;;;ACVE,gBAAAA,YAAA;AADK,IAAM,aAA2B,CAAC,EAAE,MAAM,GAAG,KAAK,MACvD,gBAAAA,KAAC,SAAK,GAAG,MAAM;AAGV,IAAM,cAA6B,CAAC,EAAE,MAAM,GAAG,KAAK,MACzD,gBAAAA,KAAC,UAAM,GAAG,MAAM;AAGX,IAAM,0BAGR,CAAC,EAAE,YAAY,EAAE,KAAK,KAAK,GAAG,KAAK,MACtC,gBAAAA,KAAC,OACC,0BAAAA,KAAC,QAAM,gBAAK,GACd;AAGK,IAAM,oBAAoD,MAAM;;;AFCvE,SAAS,kBAAAC,uBAAsB;;;AGrB/B,SAAsD,cAAAC,mBAAkB;;;ACAxE,SAA4B,eAAe;AA+BvC,mBACE,OAAAC,MADF;AAVG,IAAM,mBAAuC,CAAC;AAAA,EACnD,YAAY,EAAE,KAAK,MAAM,mBAAmB,YAAAC,YAAW;AAAA,EACvD;AAAA,EACA;AACF,MAAM;AACJ,QAAM,aAAa,QAAQ,OAAO,EAAE,KAAK,KAAK,IAAI,CAAC,KAAK,IAAI,CAAC;AAE7D,QAAM,KAAK,CAAC,CAAC,WAAW,oBAAoB;AAE5C,SACE,iCACE;AAAA,oBAAAD,KAACC,aAAA,EAAW,UAAoB,MAAY;AAAA,IAC5C,gBAAAD;AAAA,MAAC;AAAA;AAAA,QACC;AAAA,QACA,UAAU,YAAY;AAAA,QACtB;AAAA;AAAA,IACF;AAAA,KACF;AAEJ;;;AD/BA,SAAS,sBAAsB;;;AET/B,OAAO,gBAAgB;AAEhB,IAAM,mBACX,CAAoD;AAAA,EAClD;AAAA,EACA,GAAG;AACL,MACA,CAAC,EAAE,WAAW,eAAe,GAAG,MAAM,MAAc;AAClD,SAAO;AAAA,IACL,WAAW,WAAW,WAAW,aAAa;AAAA,IAC9C,GAAG;AAAA,IACH,GAAG;AAAA,EACL;AACF;;;AFcE,gBAAAE,YAAA;AAdJ,IAAM,oBAA2C,CAAC;AAAA,EAChD,YAAY;AAAA,IACV;AAAA,IACA;AAAA,IACA,mBAAmB;AAAA,IACnB,YAAY;AAAA,IACZ,cAAc,CAAC;AAAA,EACjB;AAAA,EACA;AAAA,EACA,GAAG;AACL,MAAM;AACJ,QAAM,WAAWC,YAAW,UAAU;AACtC,QAAM,cAAc,iBAAsB,QAAQ;AAClD,QAAM,aAA2B,eAAe,CAAC,UAC/C,gBAAAD,KAAC,OAAK,GAAG,YAAY,KAAK,GAAG,CAC9B;AAED,QAAM,eAAe,iBAAsB,SAAS;AACpD,QAAM,cAA6B,eAAe,CAAC,UACjD,gBAAAA,KAAC,QAAM,GAAG,aAAa,KAAK,GAAG,CAChC;AAED,QAAM,WAAW,iBAAiB,KAAK,UAAU,aAAa,EAAE,IAAI,CAAC,KAAK;AAG1E,MAAI,OAAO,aAAa,UAAU;AAChC,WACE,gBAAAA;AAAA,MAAC;AAAA;AAAA,QACC,YAAY,EAAE,KAAK,YAAY,MAAM,YAAY;AAAA,QACjD,MAAM;AAAA;AAAA,IACR;AAAA,EAEJ;AAEA,QAAM,oBACJ,YAAY,QAAQ,GAAG,qBAAqB;AAE9C,QAAME,cACJ,YAAY,QAAQ,GAAG,cAAc;AAEvC,SACE,gBAAAF;AAAA,IAAC;AAAA;AAAA,MACC,YAAY;AAAA,QACV,KAAK;AAAA,QACL,MAAM;AAAA,QACN;AAAA,QACA,YAAAE;AAAA,MACF;AAAA,MACA,UAAU,YAAY;AAAA,MACtB,MAAM;AAAA;AAAA,EACR;AAEJ;AAkBO,IAAM,eAAsC,CAAC;AAAA,EAClD;AAAA,EACA,GAAG;AACL,MAAM;AACJ,QAAM,WAAWD,YAAW,UAAU;AACtC,MAAI,CAAC,SAAU,QAAO,gBAAAD,KAAC,WAAW,MAAX,EAAiB,GAAI,OAAe;AAC3D,SAAO,gBAAAA,KAAC,qBAAkB,YAAyB,GAAG,OAAO;AAC/D;;;AHnEA,SAAS,iBAAiB;AAE1B,OAAOG,iBAAgB;AA0Df,gBAAAC,YAAA;AAjCD,IAAM,wBACmC;AAAA,EAI9C,CACE;AAAA,IACE,YAAY;AAAA,IACZ;AAAA,IACA;AAAA,IACA,oBAAoB,YAAY;AAAA,IAChC,GAAG;AAAA,EACL,GACA,cACA,SAAS,SACN;AACH,UAAM;AAAA,MACJ,MAAM,EAAE,KAAK;AAAA,MACb;AAAA,IACF,IAAI,UAAU,mBAAmB,GAAG,MAAM;AAE1C,UAAM;AAAA,MACJ,MAAM;AAAA,MACN,OAAO;AAAA,MACP,oBAAoB;AAAA,MACpB,YAAAC,cAAa;AAAA,MACb;AAAA,MACA,GAAG;AAAA,IACL,IAAI,kBAAkB,CAAC;AACvB,UAAM,aAAoC;AAAA,MACxC,GAAG;AAAA,MACH,KAAK;AAAA,MACL,MAAMC,gBAAe,CAAC,UACpB,gBAAAF;AAAA,QAAC;AAAA;AAAA,UACC,YAAY;AAAA,YACV,KAAK;AAAA,YACL,MAAM;AAAA,YACN;AAAA,YACA,YAAAC;AAAA,YACA;AAAA,UACF;AAAA,UACC,GAAG;AAAA;AAAA,MACN,CACD;AAAA,IACH;AAEA,WACE,gBAAAD;AAAA,MAAC;AAAA;AAAA,QACC,eAAa,OAAO;AAAA,QACnB,GAAG;AAAA,QACJ,WAAWD,YAAW,WAAW,gBAAgB,SAAS;AAAA,QAC1D,KAAK;AAAA,QAEL,0BAAAC,KAAC,iBAAc,YAAyB,GAAG,MACxC,gBACH;AAAA;AAAA,IACF;AAAA,EAEJ;AACF;AAEA,sBAAsB,cAAc;;;AM/GpC,SAAa,YAAY;;;ACCzB,SAAS,WAAW,gBAAgB;AACpC,SAAS,uBAAuB;AAChC,SAAS,yBAAyB;;;ACHlC,SAAS,gBAAgB;AAKlB,IAAM,qBAAqB,CAAC;AAAA,EACjC,iBAAiB;AACnB,IAA6B,CAAC,MAAM;AAClC,QAAM,CAAC,UAAU,WAAW,IAAI,SAAkB,KAAK;AAEvD,QAAM,kBAAkB,CAAC,UAAkB;AACzC,QAAI,CAAC,MAAO;AAEZ,cAAU,UAAU,UAAU,KAAK,EAAE,KAAK,MAAM;AAC9C,kBAAY,IAAI;AAChB,iBAAW,MAAM,YAAY,KAAK,GAAG,cAAc;AAAA,IACrD,CAAC;AAAA,EACH;AAEA,SAAO,EAAE,UAAU,gBAAgB;AACrC;;;ADGM,gBAAAG,MACA,QAAAC,aADA;AAfC,IAAM,aAAkC,CAAC,EAAE,UAAU,KAAK,MAAM;AACrE,QAAM;AAAA,IACJ,SAAS;AAAA,MACP,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,OAAO,IAAI,CAAC,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC;AAAA,IAChE,IAAI,CAAC;AAAA,EACP,IAAI,gBAAgB;AAEpB,QAAM,EAAE,UAAU,gBAAgB,IAAI,mBAAmB;AACzD,QAAM,SAAS,MAAM;AACnB,QAAI,CAAC,QAAQ,SAAU;AACvB,oBAAgB,IAAI;AAAA,EACtB;AAEA,SACE,gBAAAA,MAAC,SAAI,WAAU,wBACb;AAAA,oBAAAD,KAAC,UAAK,WAAU,4BAA4B,oBAAS;AAAA,IACrD,gBAAAC,MAAC,qBAAkB,SAAkB,SAAS,QAC3C;AAAA,OAAC,YAAY,gBAAAD,KAAC,YAAS;AAAA,MACvB,YAAY,gBAAAA,KAAC,aAAU;AAAA,OAC1B;AAAA,KACF;AAEJ;;;AD5BA,OAAOE,iBAAgB;AAKvB;AAAA,EACE;AAAA,EACA;AAAA,OACK;AAiBD,gBAAAC,YAAA;AAbC,IAAM,mBAAmB,CAAC;AAAA,EAC/B;AAAA,EACA,YAAY;AAAA,EACZ,GAAG;AACL,IAA2B,CAAC,MAAM;AAChC,QAAM,aAAa;AAAA,IACjB,GAAG;AAAA,IACH,YAAY,gBAAgB,cAAc;AAAA,EAC5C;AAEA,QAAM,mBAAuB,MAAM;AACjC,UAAM,SAAS,gBAAgB;AAC/B,WACE,gBAAAA;AAAA,MAAC;AAAA;AAAA,QACC;AAAA,QACC,GAAG;AAAA,QACJ,WAAWC;AAAA,UACT;AAAA,UACA,OAAO,SAAS,aAAa;AAAA,UAC7B;AAAA,QACF;AAAA;AAAA,IACF;AAAA,EAEJ;AACA,mBAAiB,cAAc;AAE/B,SAAO,KAAK,0BAA0B,gBAAgB,GAAG,MAAM,IAAI;AACrE;","names":["jsx","useCallbackRef","useContext","jsx","CodeHeader","jsx","useContext","CodeHeader","classNames","jsx","CodeHeader","useCallbackRef","jsx","jsxs","classNames","jsx","classNames"]}

package/dist/styles/markdown.css

@@ -4,12 +4,12 @@
     opacity: .5;
   }
 }
-:where(.aui-md-in-progress:empty)::after,
-:where(.aui-md-in-progress > :not(ol):not(ul):not(pre):last-child)::after,
-:where(.aui-md-in-progress > pre:last-child code)::after,
-:where(.aui-md-in-progress > :is(ol, ul):last-child > li:last-child:not(:has(* > li)))::after,
-:where(.aui-md-in-progress > :is(ol, ul):last-child > li:last-child > :is(ol, ul):last-child > li:last-child:not(:has(* > li)))::after,
-:where(.aui-md-in-progress > :is(ol, ul):last-child > li:last-child > :is(ol, ul):last-child > li:last-child > :is(ol, ul):last-child > li:last-child)::after {
+:where(.aui-md-running):empty::after,
+:where(.aui-md-running) > :where(:not(ol):not(ul):not(pre)):last-child::after,
+:where(.aui-md-running) > pre:last-child code::after,
+:where(.aui-md-running) > :where(:is(ol, ul):last-child) > :where(li:last-child:not(:has(* > li)))::after,
+:where(.aui-md-running) > :where(:is(ol, ul):last-child) > :where(li:last-child) > :where(:is(ol, ul):last-child) > :where(li:last-child:not(:has(* > li)))::after,
+:where(.aui-md-running) > :where(:is(ol, ul):last-child) > :where(li:last-child) > :where(:is(ol, ul):last-child) > :where(li:last-child) > :where(:is(ol, ul):last-child) > :where(li:last-child)::after {
   animation: aui-pulse 2s cubic-bezier(0.4, 0, 0.6, 1) infinite;
   font-family:
     ui-sans-serif,
@@ -22,23 +22,23 @@
   --aui-content: "\25cf";
   content: var(--aui-content);
 }
-:where(.aui-md-in-progress:empty):where([dir=ltr], [dir=ltr] *)::after,
-:where(.aui-md-in-progress > :not(ol):not(ul):not(pre):last-child):where([dir=ltr], [dir=ltr] *)::after,
-:where(.aui-md-in-progress > pre:last-child code):where([dir=ltr], [dir=ltr] *)::after,
-:where(.aui-md-in-progress > :is(ol, ul):last-child > li:last-child:not(:has(* > li))):where([dir=ltr], [dir=ltr] *)::after,
-:where(.aui-md-in-progress > :is(ol, ul):last-child > li:last-child > :is(ol, ul):last-child > li:last-child:not(:has(* > li))):where([dir=ltr], [dir=ltr] *)::after,
-:where(.aui-md-in-progress > :is(ol, ul):last-child > li:last-child > :is(ol, ul):last-child > li:last-child > :is(ol, ul):last-child > li:last-child):where([dir=ltr], [dir=ltr] *)::after {
+:where(.aui-md-running):empty:where([dir=ltr], [dir=ltr] *)::after,
+:where(.aui-md-running) > :where(:not(ol):not(ul):not(pre)):last-child:where([dir=ltr], [dir=ltr] *)::after,
+:where(.aui-md-running) > pre:last-child code:where([dir=ltr], [dir=ltr] *)::after,
+:where(.aui-md-running) > :where(:is(ol, ul):last-child) > :where(li:last-child:not(:has(* > li))):where([dir=ltr], [dir=ltr] *)::after,
+:where(.aui-md-running) > :where(:is(ol, ul):last-child) > :where(li:last-child) > :where(:is(ol, ul):last-child) > :where(li:last-child:not(:has(* > li))):where([dir=ltr], [dir=ltr] *)::after,
+:where(.aui-md-running) > :where(:is(ol, ul):last-child) > :where(li:last-child) > :where(:is(ol, ul):last-child) > :where(li:last-child) > :where(:is(ol, ul):last-child) > :where(li:last-child):where([dir=ltr], [dir=ltr] *)::after {
   margin-left: 0.25rem;
 }
-:where(.aui-md-in-progress:empty):where([dir=rtl], [dir=rtl] *)::after,
-:where(.aui-md-in-progress > :not(ol):not(ul):not(pre):last-child):where([dir=rtl], [dir=rtl] *)::after,
-:where(.aui-md-in-progress > pre:last-child code):where([dir=rtl], [dir=rtl] *)::after,
-:where(.aui-md-in-progress > :is(ol, ul):last-child > li:last-child:not(:has(* > li))):where([dir=rtl], [dir=rtl] *)::after,
-:where(.aui-md-in-progress > :is(ol, ul):last-child > li:last-child > :is(ol, ul):last-child > li:last-child:not(:has(* > li))):where([dir=rtl], [dir=rtl] *)::after,
-:where(.aui-md-in-progress > :is(ol, ul):last-child > li:last-child > :is(ol, ul):last-child > li:last-child > :is(ol, ul):last-child > li:last-child):where([dir=rtl], [dir=rtl] *)::after {
+:where(.aui-md-running):empty:where([dir=rtl], [dir=rtl] *)::after,
+:where(.aui-md-running) > :where(:not(ol):not(ul):not(pre)):last-child:where([dir=rtl], [dir=rtl] *)::after,
+:where(.aui-md-running) > pre:last-child code:where([dir=rtl], [dir=rtl] *)::after,
+:where(.aui-md-running) > :where(:is(ol, ul):last-child) > :where(li:last-child:not(:has(* > li))):where([dir=rtl], [dir=rtl] *)::after,
+:where(.aui-md-running) > :where(:is(ol, ul):last-child) > :where(li:last-child) > :where(:is(ol, ul):last-child) > :where(li:last-child:not(:has(* > li))):where([dir=rtl], [dir=rtl] *)::after,
+:where(.aui-md-running) > :where(:is(ol, ul):last-child) > :where(li:last-child) > :where(:is(ol, ul):last-child) > :where(li:last-child) > :where(:is(ol, ul):last-child) > :where(li:last-child):where([dir=rtl], [dir=rtl] *)::after {
   margin-right: 0.25rem;
 }
-:where(.aui-md-root) h1 {
+.aui-md-root h1 {
   margin-bottom: 2rem;
   scroll-margin: 5rem;
   font-size: 2.25rem;
@@ -46,10 +46,10 @@
   font-weight: 800;
   letter-spacing: -0.025em;
 }
-:where(.aui-md-root) h1:last-child {
+.aui-md-root h1:last-child {
   margin-bottom: 0px;
 }
-:where(.aui-md-root) h2 {
+.aui-md-root h2 {
   margin-bottom: 1rem;
   margin-top: 2rem;
   scroll-margin: 5rem;
@@ -58,13 +58,13 @@
   font-weight: 600;
   letter-spacing: -0.025em;
 }
-:where(.aui-md-root) h2:first-child {
+.aui-md-root h2:first-child {
   margin-top: 0px;
 }
-:where(.aui-md-root) h2:last-child {
+.aui-md-root h2:last-child {
   margin-bottom: 0px;
 }
-:where(.aui-md-root) h3 {
+.aui-md-root h3 {
   margin-bottom: 1rem;
   margin-top: 1.5rem;
   scroll-margin: 5rem;
@@ -73,13 +73,13 @@
   font-weight: 600;
   letter-spacing: -0.025em;
 }
-:where(.aui-md-root) h3:first-child {
+.aui-md-root h3:first-child {
   margin-top: 0px;
 }
-:where(.aui-md-root) h3:last-child {
+.aui-md-root h3:last-child {
   margin-bottom: 0px;
 }
-:where(.aui-md-root) h4 {
+.aui-md-root h4 {
   margin-bottom: 1rem;
   margin-top: 1.5rem;
   scroll-margin: 5rem;
@@ -88,82 +88,82 @@
   font-weight: 600;
   letter-spacing: -0.025em;
 }
-:where(.aui-md-root) h4:first-child {
+.aui-md-root h4:first-child {
   margin-top: 0px;
 }
-:where(.aui-md-root) h4:last-child {
+.aui-md-root h4:last-child {
   margin-bottom: 0px;
 }
-:where(.aui-md-root) h5 {
+.aui-md-root h5 {
   margin-top: 1rem;
   margin-bottom: 1rem;
   font-size: 1.125rem;
   line-height: 1.75rem;
   font-weight: 600;
 }
-:where(.aui-md-root) h5:first-child {
+.aui-md-root h5:first-child {
   margin-top: 0px;
 }
-:where(.aui-md-root) h5:last-child {
+.aui-md-root h5:last-child {
   margin-bottom: 0px;
 }
-:where(.aui-md-root) h6 {
+.aui-md-root h6 {
   margin-top: 1rem;
   margin-bottom: 1rem;
   font-weight: 600;
 }
-:where(.aui-md-root) h6:first-child {
+.aui-md-root h6:first-child {
   margin-top: 0px;
 }
-:where(.aui-md-root) h6:last-child {
+.aui-md-root h6:last-child {
   margin-bottom: 0px;
 }
-:where(.aui-md-root) p {
+.aui-md-root p {
   margin-bottom: 1.25rem;
   margin-top: 1.25rem;
   line-height: 1.75rem;
 }
-:where(.aui-md-root) p:first-child {
+.aui-md-root p:first-child {
   margin-top: 0px;
 }
-:where(.aui-md-root) p:last-child {
+.aui-md-root p:last-child {
   margin-bottom: 0px;
 }
-:where(.aui-md-root) a {
+.aui-md-root a {
   font-weight: 500;
   color: hsl(var(--aui-primary));
   text-decoration-line: underline;
   text-underline-offset: 4px;
 }
-:where(.aui-md-root) blockquote {
+.aui-md-root blockquote {
   border-left-width: 2px;
   padding-left: 1.5rem;
   font-style: italic;
 }
-:where(.aui-md-root) ul {
+.aui-md-root ul {
   margin-top: 1.25rem;
   margin-bottom: 1.25rem;
   margin-left: 1.5rem;
   list-style-type: disc;
 }
-:where(.aui-md-root) ul > li {
+.aui-md-root ul > li {
   margin-top: 0.5rem;
 }
-:where(.aui-md-root) ol {
+.aui-md-root ol {
   margin-top: 1.25rem;
   margin-bottom: 1.25rem;
   margin-left: 1.5rem;
   list-style-type: decimal;
 }
-:where(.aui-md-root) ol > li {
+.aui-md-root ol > li {
   margin-top: 0.5rem;
 }
-:where(.aui-md-root) hr {
+.aui-md-root hr {
   margin-top: 1.25rem;
   margin-bottom: 1.25rem;
   border-bottom-width: 1px;
 }
-:where(.aui-md-root) table {
+.aui-md-root table {
   margin-top: 1.25rem;
   margin-bottom: 1.25rem;
   width: 100%;
@@ -173,7 +173,7 @@
   border-spacing: var(--aui-border-spacing-x) var(--aui-border-spacing-y);
   overflow-y: auto;
 }
-:where(.aui-md-root) th {
+.aui-md-root th {
   background-color: hsl(var(--aui-muted));
   padding-left: 1rem;
   padding-right: 1rem;
@@ -182,19 +182,19 @@
   text-align: left;
   font-weight: 700;
 }
-:where(.aui-md-root) th:first-child {
+.aui-md-root th:first-child {
   border-top-left-radius: var(--aui-radius);
 }
-:where(.aui-md-root) th:last-child {
+.aui-md-root th:last-child {
   border-top-right-radius: var(--aui-radius);
 }
-:where(.aui-md-root) th[align=center] {
+.aui-md-root th[align=center] {
   text-align: center;
 }
-:where(.aui-md-root) th[align=right] {
+.aui-md-root th[align=right] {
   text-align: right;
 }
-:where(.aui-md-root) td {
+.aui-md-root td {
   border-bottom-width: 1px;
   border-left-width: 1px;
   padding-left: 1rem;
@@ -203,35 +203,35 @@
   padding-bottom: 0.5rem;
   text-align: left;
 }
-:where(.aui-md-root) td:last-child {
+.aui-md-root td:last-child {
   border-right-width: 1px;
 }
-:where(.aui-md-root) td[align=center] {
+.aui-md-root td[align=center] {
   text-align: center;
 }
-:where(.aui-md-root) td[align=right] {
+.aui-md-root td[align=right] {
   text-align: right;
 }
-:where(.aui-md-root) tr {
+.aui-md-root tr {
   margin: 0px;
   border-bottom-width: 1px;
   padding: 0px;
 }
-:where(.aui-md-root) tr:first-child {
+.aui-md-root tr:first-child {
   border-top-width: 1px;
 }
-:where(.aui-md-root) tr:last-child > td:first-child {
+.aui-md-root tr:last-child > td:first-child {
   border-bottom-left-radius: var(--aui-radius);
 }
-:where(.aui-md-root) tr:last-child > td:last-child {
+.aui-md-root tr:last-child > td:last-child {
   border-bottom-right-radius: var(--aui-radius);
 }
-:where(.aui-md-root) sup > a {
+.aui-md-root sup > a {
   font-size: 0.75rem;
   line-height: 1rem;
   text-decoration-line: none;
 }
-:where(.aui-md-root) pre {
+.aui-md-root pre {
   overflow-x: auto;
   border-bottom-right-radius: var(--aui-radius);
   border-bottom-left-radius: var(--aui-radius);
@@ -240,8 +240,8 @@
   --aui-text-opacity: 1;
   color: rgb(255 255 255 / var(--aui-text-opacity));
 }
-:where(.aui-md-root) > code,
-:where(.aui-md-root) :not(:where(pre)) code {
+.aui-md-root > code,
+.aui-md-root :not(:where(pre)) code {
   border-radius: 0.25rem;
   border-width: 1px;
   background-color: hsl(var(--aui-muted));

package/dist/styles/tailwindcss/markdown.css

@@ -1,107 +1,100 @@
-/* in progress indicator */
-:where(.aui-md-in-progress:empty)::after,
-:where(.aui-md-in-progress > :not(ol):not(ul):not(pre):last-child)::after,
-:where(.aui-md-in-progress > pre:last-child code)::after,
-:where(
-    .aui-md-in-progress
-      > :is(ol, ul):last-child
-      > li:last-child:not(:has(* > li))
-  )::after,
-:where(
-    .aui-md-in-progress
-      > :is(ol, ul):last-child
-      > li:last-child
-      > :is(ol, ul):last-child
-      > li:last-child:not(:has(* > li))
-  )::after,
-:where(
-    .aui-md-in-progress
-      > :is(ol, ul):last-child
-      > li:last-child
-      > :is(ol, ul):last-child
-      > li:last-child
-      > :is(ol, ul):last-child
-      > li:last-child
-  )::after {
+/* running indicator */
+:where(.aui-md-running):empty::after,
+:where(.aui-md-running) > :where(:not(ol):not(ul):not(pre)):last-child::after,
+:where(.aui-md-running) > pre:last-child code::after,
+:where(.aui-md-running)
+  > :where(:is(ol, ul):last-child)
+  > :where(li:last-child:not(:has(* > li)))::after,
+:where(.aui-md-running)
+  > :where(:is(ol, ul):last-child)
+  > :where(li:last-child)
+  > :where(:is(ol, ul):last-child)
+  > :where(li:last-child:not(:has(* > li)))::after,
+:where(.aui-md-running)
+  > :where(:is(ol, ul):last-child)
+  > :where(li:last-child)
+  > :where(:is(ol, ul):last-child)
+  > :where(li:last-child)
+  > :where(:is(ol, ul):last-child)
+  > :where(li:last-child)::after {
   @apply animate-pulse font-sans content-['\25CF'] ltr:ml-1 rtl:mr-1;
 }
-
 /* typography */
 
-:where(.aui-md-root) h1 {
+.aui-md-root h1 {
   @apply mb-8 scroll-m-20 text-4xl font-extrabold tracking-tight last:mb-0;
 }
 
-:where(.aui-md-root) h2 {
+.aui-md-root h2 {
   @apply mb-4 mt-8 scroll-m-20 text-3xl font-semibold tracking-tight first:mt-0 last:mb-0;
 }
 
-:where(.aui-md-root) h3 {
+.aui-md-root h3 {
   @apply mb-4 mt-6 scroll-m-20 text-2xl font-semibold tracking-tight first:mt-0 last:mb-0;
 }
 
-:where(.aui-md-root) h4 {
+.aui-md-root h4 {
   @apply mb-4 mt-6 scroll-m-20 text-xl font-semibold tracking-tight first:mt-0 last:mb-0;
 }
 
-:where(.aui-md-root) h5 {
+.aui-md-root h5 {
   @apply my-4 text-lg font-semibold first:mt-0 last:mb-0;
 }
 
-:where(.aui-md-root) h6 {
+.aui-md-root h6 {
   @apply my-4 font-semibold first:mt-0 last:mb-0;
 }
 
-:where(.aui-md-root) p {
+.aui-md-root p {
   @apply mb-5 mt-5 leading-7 first:mt-0 last:mb-0;
 }
 
-:where(.aui-md-root) a {
+.aui-md-root a {
   @apply text-aui-primary font-medium underline underline-offset-4;
 }
 
-:where(.aui-md-root) blockquote {
+.aui-md-root blockquote {
   @apply border-l-2 pl-6 italic;
 }
 
-:where(.aui-md-root) ul {
+.aui-md-root ul {
   @apply my-5 ml-6 list-disc [&>li]:mt-2;
 }
 
-:where(.aui-md-root) ol {
+.aui-md-root ol {
   @apply my-5 ml-6 list-decimal [&>li]:mt-2;
 }
 
-:where(.aui-md-root) hr {
+.aui-md-root hr {
   @apply my-5 border-b;
 }
 
-:where(.aui-md-root) table {
+.aui-md-root table {
   @apply my-5 w-full border-separate border-spacing-0 overflow-y-auto;
 }
 
-:where(.aui-md-root) th {
+.aui-md-root th {
   @apply bg-aui-muted px-4 py-2 text-left font-bold first:rounded-tl-lg last:rounded-tr-lg [&[align=center]]:text-center [&[align=right]]:text-right;
 }
 
-:where(.aui-md-root) td {
+.aui-md-root td {
   @apply border-b border-l px-4 py-2 text-left last:border-r [&[align=center]]:text-center [&[align=right]]:text-right;
 }
 
-:where(.aui-md-root) tr {
+.aui-md-root tr {
   @apply m-0 border-b p-0 first:border-t [&:last-child>td:first-child]:rounded-bl-lg [&:last-child>td:last-child]:rounded-br-lg;
 }
 
-:where(.aui-md-root) sup {
+.aui-md-root sup {
   @apply [&>a]:text-xs [&>a]:no-underline;
 }
 
-:where(.aui-md-root) pre {
+.aui-md-root pre {
   @apply overflow-x-auto rounded-b-lg bg-black p-4 text-white;
 }
 
-:where(.aui-md-root) > code,
-:where(.aui-md-root) :not(:where(pre)) code {
+.aui-md-root > code,
+.aui-md-root :not(:where(pre)) code {
   @apply bg-aui-muted rounded border font-semibold;
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-markdown",
-  "version": "0.2.4",
+  "version": "0.2.6",
   "license": "MIT",
   "exports": {
     ".": {
@@ -37,13 +37,14 @@
   ],
   "sideEffects": false,
   "dependencies": {
+    "@radix-ui/react-primitive": "^2.0.0",
     "@radix-ui/react-use-callback-ref": "^1.1.0",
     "classnames": "^2.5.1",
-    "lucide-react": "^0.414.0",
+    "lucide-react": "^0.416.0",
     "react-markdown": "^9.0.1"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.5.9",
+    "@assistant-ui/react": "^0.5.20",
     "@types/react": "*",
     "react": "^18",
     "tailwindcss": "^3.4.4"
@@ -61,10 +62,10 @@
     "autoprefixer": "^10.4.19",
     "eslint": "^8",
     "eslint-config-next": "14.2.5",
-    "postcss": "^8.4.39",
-    "tailwindcss": "^3.4.6",
+    "postcss": "^8.4.40",
+    "tailwindcss": "^3.4.7",
     "tailwindcss-animate": "^1.0.7",
-    "tsup": "8.2.2",
+    "tsup": "8.2.3",
     "tsx": "^4.16.2",
     "@assistant-ui/tailwindcss-transformer": "^0.1.0",
     "@assistant-ui/tsconfig": "0.0.0"