From 52b2bf04fb8ad0fbafa91fc9f24fbe1ee7dcd631 Mon Sep 17 00:00:00 2001
From: Timeless0911 <1604889533@qq.com>
Date: Mon, 4 Aug 2025 15:56:29 +0800
Subject: [PATCH 1/2] docs: add llms copy button

---
 website/theme/index.tsx | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/website/theme/index.tsx b/website/theme/index.tsx
index f3362ee1c..f05b329b2 100644
--- a/website/theme/index.tsx
+++ b/website/theme/index.tsx
@@ -1,4 +1,7 @@
-import { Layout as BaseLayout } from '@rspress/core/theme';
+import {
+  Layout as BaseLayout,
+  getCustomMDXComponent as basicGetCustomMDXComponent,
+} from '@rspress/core/theme';
 import {
   Search as PluginAlgoliaSearch,
   ZH_LOCALES,
@@ -7,6 +10,31 @@ import { NavIcon } from '@rstack-dev/doc-ui/nav-icon';
 import { HomeLayout } from './pages';
 import './index.scss';
 import { useLang } from '@rspress/core/runtime';
+import {
+  LlmsContainer,
+  LlmsCopyButton,
+  LlmsViewOptions,
+} from '@rspress/plugin-llms/runtime';
+
+export function getCustomMDXComponent() {
+  const { h1: H1, ...mdxComponents } = basicGetCustomMDXComponent();
+
+  const MyH1 = ({ ...props }) => {
+    return (
+      <>
+        <H1 {...props} />
+        <LlmsContainer>
+          <LlmsCopyButton />
+          <LlmsViewOptions />
+        </LlmsContainer>
+      </>
+    );
+  };
+  return {
+    ...mdxComponents,
+    h1: MyH1,
+  };
+}
 
 const Layout = () => {
   return <BaseLayout beforeNavTitle={<NavIcon />} />;

From 425b1ecb310edd6be6be9b700f51ead1038b53c2 Mon Sep 17 00:00:00 2001
From: Timeless0911 <1604889533@qq.com>
Date: Mon, 4 Aug 2025 15:57:41 +0800
Subject: [PATCH 2/2] chore: update

---
 website/theme/index.tsx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/website/theme/index.tsx b/website/theme/index.tsx
index f05b329b2..02cc95e8a 100644
--- a/website/theme/index.tsx
+++ b/website/theme/index.tsx
@@ -1,6 +1,6 @@
 import {
+  getCustomMDXComponent as BaseGetCustomMDXComponent,
   Layout as BaseLayout,
-  getCustomMDXComponent as basicGetCustomMDXComponent,
 } from '@rspress/core/theme';
 import {
   Search as PluginAlgoliaSearch,
@@ -17,7 +17,7 @@ import {
 } from '@rspress/plugin-llms/runtime';
 
 export function getCustomMDXComponent() {
-  const { h1: H1, ...mdxComponents } = basicGetCustomMDXComponent();
+  const { h1: H1, ...mdxComponents } = BaseGetCustomMDXComponent();
 
   const MyH1 = ({ ...props }) => {
     return (