diff --git a/.changeset/grumpy-peas-hammer.md b/.changeset/grumpy-peas-hammer.md
new file mode 100644
index 0000000000..abc1f9a185
--- /dev/null
+++ b/.changeset/grumpy-peas-hammer.md
@@ -0,0 +1,6 @@
+---
+"@gitbook/embed": minor
+"gitbook": patch
+---
+
+Improve Docs Embed with separate Assistant and Docs tabs
diff --git a/bun.lock b/bun.lock
index 2f67c7d8cb..bdd7190479 100644
--- a/bun.lock
+++ b/bun.lock
@@ -1,6 +1,5 @@
 {
   "lockfileVersion": 1,
-  "configVersion": 0,
   "workspaces": {
     "": {
       "name": "gitbook",
@@ -346,7 +345,7 @@
     "react-dom": "catalog:",
   },
   "catalog": {
-    "@gitbook/api": "0.151.0",
+    "@gitbook/api": "0.152.0",
     "@scalar/api-client-react": "^1.3.46",
     "@tsconfig/node20": "^20.1.6",
     "@tsconfig/strictest": "^2.0.6",
@@ -727,7 +726,7 @@
 
     "@fortawesome/fontawesome-svg-core": ["@fortawesome/fontawesome-svg-core@6.6.0", "", { "dependencies": { "@fortawesome/fontawesome-common-types": "6.6.0" } }, "sha512-KHwPkCk6oRT4HADE7smhfsKudt9N/9lm6EJ5BVg0tD1yPA5hht837fB87F8pn15D8JfTqQOjhKTktwmLMiD7Kg=="],
 
-    "@gitbook/api": ["@gitbook/api@0.151.0", "", { "dependencies": { "event-iterator": "^2.0.0", "eventsource-parser": "^3.0.0" } }, "sha512-5d9+rZ2u6CKKIiVHO1Toyk+7wHtTOXmP0+sVIE3teRkceX4z5FGIIpa4XsFeKC9XeosncvehdshPaOGQtSDpTQ=="],
+    "@gitbook/api": ["@gitbook/api@0.152.0", "", { "dependencies": { "event-iterator": "^2.0.0", "eventsource-parser": "^3.0.0" } }, "sha512-BC+SXnBybtzKYY3aUYDOjwz17uQwMM1kTNiij8OTKZyFByyWcz1FifJGNmKHmDG3XeZLJtCi3nOTEIcGIdG1VA=="],
 
     "@gitbook/browser-types": ["@gitbook/browser-types@workspace:packages/browser-types"],
 
diff --git a/package.json b/package.json
index b83bbd2e9a..6c99ae20bf 100644
--- a/package.json
+++ b/package.json
@@ -41,7 +41,7 @@
         "catalog": {
             "@tsconfig/strictest": "^2.0.6",
             "@tsconfig/node20": "^20.1.6",
-            "@gitbook/api": "0.151.0",
+            "@gitbook/api": "0.152.0",
             "@scalar/api-client-react": "^1.3.46",
             "@types/react": "^19.0.0",
             "@types/react-dom": "^19.0.0",
diff --git a/packages/embed/README.md b/packages/embed/README.md
index 45efed3ac8..3c52de77d3 100644
--- a/packages/embed/README.md
+++ b/packages/embed/README.md
@@ -1,24 +1,61 @@
-# `@gitbook/embed`
+# GitBook Docs Embed (`@gitbook/embed`)
 
-Embed the GitBook Docs Assistant in your product or website.
+Embed your GitBook docs in your product or website.
+
+The Docs Embed can contain two tabs:
+- **Assistant**: The [GitBook Assistant](https://gitbook.com/docs/publishing-documentation/gitbook-ai-assistant) - an AI-powered chat interface to help users find answers
+- **Docs**: A browser for navigating your documentation site
+
+The embed is set up automatically based on your site's configuration. You can optionally customize and override the configuration with custom actions, tools, suggested questions, [Authenticated Access](https://gitbook.com/docs/publishing-documentation/authenticated-access), and more. See the [Configuration](#configuration) section for all available options.
 
 # Usage
 
-## As a script from your docs site
+## As a standalone script from your docs site
 
-All GitBook docs site includes a script to easily embed the docs assistant as a widget on your website.
+All GitBook docs sites include a script to easily add the Docs Embed as a widget on your website.
 
 The script is served at `https://docs.company.com/~gitbook/embed/script.js`.
 
-You can find the embed script from your docs site settings, or you can copy the following and replace the `docs.company.com` by your docs site hostname.
+You can find the embed script from your docs site settings, or you can copy the following and replace `docs.company.com` with your docs site hostname.
 
 ```html
 <script src="https://docs.company.com/~gitbook/embed/script.js"></script>
 <script>
+// Initialize with Authenticated Access (optional)
+window.GitBook('init', 
+    { siteURL: 'https://docs.company.com' },
+    { visitor: { token: 'your-jwt-token' } }
+);
 window.GitBook('show');
 </script>
 ```
 
+The standalone script provides a global `GitBook` function. See the [API Reference](#api-reference) section for all available methods.
+
+### Example: Configuring the widget
+
+```javascript
+GitBook('configure', {
+    button: {
+        label: 'Ask',
+        icon: 'assistant' // 'assistant' | 'sparkle' | 'circle-question' | 'book'
+    },
+    tabs: ['assistant', 'docs'],
+    actions: [
+        {
+            icon: 'circle-question',
+            label: 'Contact Support',
+            onClick: () => window.open('https://support.example.com', '_blank')
+        }
+    ],
+    greeting: { title: 'Welcome!', subtitle: 'How can I help?' },
+    suggestions: ['What is GitBook?', 'How do I get started?'],
+    tools: [/* ... */]
+});
+```
+
+See the [Configuration](#configuration) section for all available options.
+
 ## As a package from NPM
 
 Install the package: `npm install @gitbook/embed` and import it in your web application:
@@ -30,10 +67,46 @@ const gitbook = createGitBook({
     siteURL: 'https://docs.company.com'
 });
 
+// Create an iframe and get its URL
 const iframe = document.createElement('iframe');
-iframe.src = gitbook.getFrameURL();
+iframe.src = gitbook.getFrameURL({
+    visitor: {
+        token: 'your-jwt-token', // Optional: for Adaptive Content or Authenticated Access
+        unsignedClaims: { // Optional: custom claims for dynamic expressions
+            userId: '123',
+            plan: 'premium'
+        }
+    }
+});
 
+// Create a frame client to communicate with the iframe
 const frame = gitbook.createFrame(iframe);
+
+// Use the frame client methods
+frame.navigateToPage('/getting-started'); // Navigate to a page in the docs tab
+frame.navigateToAssistant(); // Switch to the assistant tab
+frame.postUserMessage('How do I get started?');
+frame.clearChat();
+
+// Configure the embed (see Configuration section for all options)
+frame.configure({
+    tabs: ['assistant', 'docs'],
+    actions: [
+        {
+            icon: 'circle-question',
+            label: 'Contact Support',
+            onClick: () => window.open('https://support.example.com', '_blank')
+        }
+    ],
+    greeting: { title: 'Welcome!', subtitle: 'How can I help?' },
+    suggestions: ['What is GitBook?', 'How do I get started?'],
+    tools: [/* ... */]
+});
+
+// Listen to events
+frame.on('close', () => {
+    console.log('Frame closed');
+});
 ```
 
 ## As React components
@@ -41,9 +114,296 @@ const frame = gitbook.createFrame(iframe);
 After installing the NPM package, you can import prebuilt React components:
 
 ```tsx
-import { GitBookProvider, GitBookAssistantFrame } from '@gitbook/embed/react';
+import { GitBookProvider, GitBookFrame } from '@gitbook/embed/react';
 
 <GitBookProvider siteURL="https://docs.company.com">
-    <GitBookAssistantFrame />
+    <GitBookFrame
+        visitor={{
+            token: 'your-jwt-token', // Optional: for Adaptive Content or Authenticated Access
+            unsignedClaims: { userId: '123' } // Optional: custom claims for dynamic expressions
+        }}
+        tabs={['assistant', 'docs']}
+        greeting={{ title: 'Welcome!', subtitle: 'How can I help?' }}
+        suggestions={['What is GitBook?', 'How do I get started?']}
+        actions={[
+            {
+                icon: 'circle-question',
+                label: 'Contact Support',
+                onClick: () => window.open('https://support.example.com', '_blank')
+            }
+        ]}
+        tools={[/* ... */]}
+    />
 </GitBookProvider>
 ```
+
+You can also use the `useGitBook` hook to access the client:
+
+```tsx
+import { useGitBook } from '@gitbook/embed/react';
+
+function MyComponent() {
+    const gitbook = useGitBook();
+    const frameURL = gitbook.getFrameURL({ visitor: { token: '...' } });
+    // ...
+}
+```
+
+# API Reference
+
+## Method Comparison
+
+| Method | Standalone Script | NPM Package | React Components |
+|--------|------------------|-------------|------------------|
+| **Initialize** | `GitBook('init', options, frameOptions)` | `createGitBook(options)` | `<GitBookProvider siteURL="...">` |
+| **Get frame URL** | ❌ (handled internally) | `client.getFrameURL(options)` | `useGitBook().getFrameURL(options)` |
+| **Create frame client** | ❌ (handled internally) | `client.createFrame(iframe)` | `useGitBook().createFrame(iframe)` |
+| **Show/Hide widget** | `GitBook('show')` / `GitBook('hide')` | ❌ | ❌ |
+| **Open/Close window** | `GitBook('open')` / `GitBook('close')` / `GitBook('toggle')` | ❌ | ❌ |
+| **Navigate to page** | `GitBook('navigateToPage', path)` | `frame.navigateToPage(path)` | Via frame client |
+| **Navigate to assistant** | `GitBook('navigateToAssistant')` | `frame.navigateToAssistant()` | Via frame client |
+| **Post message** | `GitBook('postUserMessage', message)` | `frame.postUserMessage(message)` | Via frame client |
+| **Clear chat** | `GitBook('clearChat')` | `frame.clearChat()` | Via frame client |
+| **Configure** | `GitBook('configure', settings)` | `frame.configure(settings)` | Props on `<GitBookFrame>` |
+| **Event listeners** | ❌ | `frame.on(event, listener)` | Via frame client |
+| **Unload** | `GitBook('unload')` | ❌ | ❌ |
+
+## Method Signatures
+
+### Standalone Script
+
+- `GitBook('init', options: { siteURL: string }, frameOptions?: { visitor?: {...} })` - Initialize widget
+- `GitBook('show')` - Show widget button
+- `GitBook('hide')` - Hide widget button
+- `GitBook('open')` - Open widget window
+- `GitBook('close')` - Close widget window
+- `GitBook('toggle')` - Toggle widget window
+- `GitBook('navigateToPage', path: string)` - Navigate to page
+- `GitBook('navigateToAssistant')` - Navigate to assistant tab
+- `GitBook('postUserMessage', message: string)` - Post message to chat
+- `GitBook('clearChat')` - Clear chat history
+- `GitBook('configure', settings: {...})` - Configure widget
+- `GitBook('unload')` - Unload widget
+
+### NPM Package
+
+**Client Factory:**
+- `createGitBook(options: { siteURL: string })` → `GitBookClient`
+- `client.getFrameURL(options?: { visitor?: {...} })` → `string`
+- `client.createFrame(iframe: HTMLIFrameElement)` → `GitBookFrameClient`
+
+**Frame Client:**
+- `frame.navigateToPage(path: string)` → `void`
+- `frame.navigateToAssistant()` → `void`
+- `frame.postUserMessage(message: string)` → `void`
+- `frame.clearChat()` → `void`
+- `frame.configure(settings: Partial<GitBookEmbeddableConfiguration>)` → `void`
+- `frame.on(event: string, listener: Function)` → `() => void` (unsubscribe)
+
+### React Components
+
+**Components:**
+- `<GitBookProvider siteURL: string>` - Provider component
+- `<GitBookFrame {...props}>` - Frame component (accepts all config options as props)
+
+**Hooks:**
+- `useGitBook()` → `GitBookClient` (must be used within `<GitBookProvider>`)
+
+# Configuration
+
+Configuration options are available across usage methods as follows:
+- **Standalone script**: via `GitBook('configure', {...})`
+- **NPM package**: via `frame.configure({...})`
+- **React components**: via props on `<GitBookFrame>`
+
+### `tabs`
+
+Available in: Standalone script, NPM package, React components
+
+Override which tabs are displayed. Defaults to your site's configuration.
+
+- **Type**: `('assistant' | 'docs')[]`
+- **Options**:
+  - `['assistant', 'docs']` - Show both tabs
+  - `['assistant']` - Show only the assistant tab
+  - `['docs']` - Show only the docs tab
+
+```javascript
+tabs: ['assistant', 'docs']
+```
+
+### `actions`
+
+Available in: Standalone script, NPM package, React components
+
+Custom action buttons rendered in the sidebar alongside tabs. Each action button triggers a callback when clicked.
+
+- **Type**: `GitBookEmbeddableActionDefinition[]`
+- **Properties**:
+  - `icon`: `string` - Icon name (e.g., `'circle-question'`, `'book'`, `'sparkle'`, `'rocket'`, `'assistant'`)
+  - `label`: `string` - Button label text
+  - `onClick`: `() => void | Promise<void>` - Callback function when clicked
+
+```javascript
+actions: [
+    {
+        icon: 'circle-question',
+        label: 'Contact Support',
+        onClick: () => window.open('https://support.example.com', '_blank')
+    },
+    {
+        icon: 'rocket',
+        label: 'Get started',
+        onClick: () => {
+            GitBook('navigateToPage', '/getting-started');
+        }
+    }
+]
+```
+
+### `greeting`
+
+Available in: Standalone script, NPM package, React components
+
+Welcome message displayed in the [Assistant](https://gitbook.com/docs/publishing-documentation/gitbook-ai-assistant) tab.
+
+- **Type**: `{ title: string, subtitle: string }`
+
+```javascript
+greeting: {
+    title: 'Welcome!',
+    subtitle: 'How can I help you today?'
+}
+```
+
+### `suggestions`
+
+Available in: Standalone script, NPM package, React components
+
+Suggested questions displayed in the [Assistant](https://gitbook.com/docs/publishing-documentation/gitbook-ai-assistant) welcome screen.
+
+- **Type**: `string[]`
+
+```javascript
+suggestions: [
+    'What is GitBook?',
+    'How do I get started?',
+    'What can you do?'
+]
+```
+
+### `tools`
+
+Available in: Standalone script, NPM package, React components
+
+Custom AI tools to extend the [Assistant](https://gitbook.com/docs/publishing-documentation/gitbook-ai-assistant). Tools allow the assistant to execute functions and integrate with your own systems.
+
+**Note**: In addition to custom tools, the Assistant will always have access to any [MCP servers you define](https://gitbook.com/docs/publishing-documentation/gitbook-ai-assistant#extend-gitbook-assistant-with-mcp-servers) in your site's AI settings.
+
+- **Type**: `GitBookToolDefinition[]`
+- **Properties**:
+  - `name`: `string` - Unique tool identifier
+  - `description`: `string` - Description of what the tool does (used by the AI to decide when and how to use it).
+  - `inputSchema`: `object` - JSON schema defining the tool's input parameters
+  - `execute`: `(input: object) => Promise<{ output: any, summary: string }>` - Async function that executes the tool.
+    - `output`: The result of the tool execution, provided to the AI to continue working with. Not shown to the user.
+    - `summary`: The visual summary of the tool execution, shown in the user's chat window.
+  - `confirmation`: `{ icon?: string, label: string }` (optional) - Confirmation button shown before execution, useful for actions that require the user's express approval.
+
+```javascript
+tools: [
+    {
+        name: 'get_user_info',
+        description: 'Get information about the current user',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                userId: { 
+                    type: 'string', 
+                    description: 'The user ID to look up' 
+                }
+            },
+            required: ['userId']
+        },
+        execute: async (input) => {
+            const user = await fetch(`/api/users/${input.userId}`).then(r => r.json());
+            return {
+                output: { name: user.name, plan: user.plan },
+                summary: `Retrieved info for user ${user.name}`
+            };
+        }
+    },
+    {
+        name: 'create_ticket',
+        description: 'Create a support ticket',
+        confirmation: {
+            icon: 'circle-question',
+            label: 'Create support ticket?'
+        },
+        inputSchema: {
+            type: 'object',
+            properties: {
+                subject: { type: 'string' },
+                description: { type: 'string' }
+            },
+            required: ['subject', 'description']
+        },
+        execute: async (input) => {
+            const ticket = await fetch('/api/tickets', {
+                method: 'POST',
+                body: JSON.stringify(input)
+            }).then(r => r.json());
+            return {
+                output: { ticketId: ticket.id },
+                summary: `Created ticket #${ticket.id}`
+            };
+        }
+    }
+]
+```
+
+### `visitor` (Authenticated Access)
+
+Available in: Standalone script (via `init`), NPM package (via `getFrameURL()`), React components (as prop)
+
+[Authenticated Access](https://gitbook.com/docs/publishing-documentation/authenticated-access) options passed when creating the frame URL. Used for [Adaptive Content](https://gitbook.com/docs/publishing-documentation/adaptive-content) and [Authenticated Access](https://gitbook.com/docs/publishing-documentation/authenticated-access).
+
+**Note**: This is not a configuration option but rather a parameter when initializing the frame or creating the frame URL.
+
+**Standalone script**: Pass as the second argument to `GitBook('init', options, frameOptions)`
+**NPM package**: Pass to `getFrameURL({ visitor: {...} })`
+**React components**: Pass as the `visitor` prop on `<GitBookFrame>`
+
+- **Type**: `{ token?: string, unsignedClaims?: Record<string, unknown> }`
+- **Properties**:
+  - `token`: `string` (optional) - Signed JWT token for [Adaptive Content](https://gitbook.com/docs/publishing-documentation/adaptive-content) or [Authenticated Access](https://gitbook.com/docs/publishing-documentation/authenticated-access)
+  - `unsignedClaims`: `Record<string, unknown>` (optional) - Unsigned claims that can be used in dynamic expressions via `visitor.claims.unsigned.<claim-name>`
+
+```javascript
+visitor: {
+    token: 'your-jwt-token',
+    unsignedClaims: {
+        userId: '123',
+        plan: 'premium',
+        role: 'admin'
+    }
+}
+```
+
+### `button`
+
+Available in: Standalone script only
+
+Configure the widget button for the standalone script. This option is not available when using the NPM package or React components, since they can be customized completely.
+
+- **Type**: `{ label: string, icon: 'assistant' | 'sparkle' | 'circle-question' | 'book' }`
+- **Properties**:
+  - `label`: `string` - Button label text
+  - `icon`: `'assistant' | 'sparkle' | 'circle-question' | 'book'` - Icon displayed on the button
+
+```javascript
+button: {
+    label: 'Ask',
+    icon: 'assistant'
+}
+```
diff --git a/packages/embed/src/client/createGitBook.ts b/packages/embed/src/client/createGitBook.ts
index bcd57a0961..fc39c71bb3 100644
--- a/packages/embed/src/client/createGitBook.ts
+++ b/packages/embed/src/client/createGitBook.ts
@@ -40,7 +40,7 @@ export function createGitBook(options: CreateGitBookOptions) {
     const client: GitBookClient = {
         getFrameURL: (frameOptions) => {
             const url = new URL(options.siteURL);
-            url.pathname = `${url.pathname.endsWith('/') ? url.pathname : `${url.pathname}/`}~gitbook/embed/assistant`;
+            url.pathname = `${url.pathname.endsWith('/') ? url.pathname : `${url.pathname}/`}~gitbook/embed`;
 
             if (frameOptions.visitor?.token) {
                 url.searchParams.set('token', frameOptions.visitor.token);
diff --git a/packages/embed/src/client/createGitBookFrame.ts b/packages/embed/src/client/createGitBookFrame.ts
index baaabf647c..84908521b8 100644
--- a/packages/embed/src/client/createGitBookFrame.ts
+++ b/packages/embed/src/client/createGitBookFrame.ts
@@ -64,8 +64,9 @@ export function createGitBookFrame(iframe: HTMLIFrameElement): GitBookFrameClien
     const events = new Map<string, Array<(...args: any[]) => void>>();
 
     const configuration: GitBookEmbeddableConfiguration = {
-        buttons: [],
-        welcomeMessage: '',
+        tabs: ['assistant', 'docs'],
+        actions: [],
+        greeting: { title: '', subtitle: '' },
         suggestions: [],
         tools: [],
     };
diff --git a/packages/embed/src/client/protocol.ts b/packages/embed/src/client/protocol.ts
index f19b400e37..06c056f9f7 100644
--- a/packages/embed/src/client/protocol.ts
+++ b/packages/embed/src/client/protocol.ts
@@ -23,7 +23,7 @@ export type GitBookToolDefinition = AIToolDefinition & {
 /**
  * Custom button definition to be passed to the embeddable GitBook.
  */
-export type GitBookEmbeddableButtonDefinition = {
+export type GitBookEmbeddableActionDefinition = {
     /**
      * Icon to be displayed in the button.
      */
@@ -41,16 +41,20 @@ export type GitBookEmbeddableButtonDefinition = {
 };
 
 /**
- * Overall configuration for the layout of the embeddable GitBook.
+ * Overall configuration for the layout of the GitBook embed.
  */
 export type GitBookEmbeddableConfiguration = {
-    /**
-     * Buttons to be displayed in the header of the embeddable GitBook.
-     */
-    buttons: GitBookEmbeddableButtonDefinition[];
+    /** Tabs to display in the embed (if enabled on the site). */
+    tabs: ('assistant' | 'docs')[];
+
+    /** Additional buttons to be displayed in the header of the GitBook embed. */
+    actions: GitBookEmbeddableActionDefinition[];
 
     /** Message to be displayed in the welcome page. */
-    welcomeMessage: string;
+    greeting: {
+        title: string;
+        subtitle: string;
+    };
 
     /** Suggestions of questions to be displayed in the welcome page. */
     suggestions: string[];
diff --git a/packages/embed/src/index.ts b/packages/embed/src/index.ts
index 4f1cce44fa..cbfb347dcd 100644
--- a/packages/embed/src/index.ts
+++ b/packages/embed/src/index.ts
@@ -1 +1,2 @@
 export * from './client';
+export * from './standalone';
diff --git a/packages/embed/src/react/GitBookFrame.tsx b/packages/embed/src/react/GitBookFrame.tsx
index cd0d8f65d6..b9f527fea3 100644
--- a/packages/embed/src/react/GitBookFrame.tsx
+++ b/packages/embed/src/react/GitBookFrame.tsx
@@ -17,7 +17,7 @@ export type GitBookFrameProps = {
  * Render a frame with the GitBook Assistant in it.
  */
 export function GitBookFrame(props: GitBookFrameProps) {
-    const { className, visitor, buttons, welcomeMessage, suggestions, tools } = props;
+    const { className, visitor, actions, greeting, suggestions, tools } = props;
 
     const frameRef = useRef<HTMLIFrameElement>(null);
     const gitbook = useGitBook();
@@ -33,12 +33,13 @@ export function GitBookFrame(props: GitBookFrameProps) {
 
     useEffect(() => {
         gitbookFrame?.configure({
-            buttons,
-            welcomeMessage,
+            tabs: ['assistant', 'docs'],
+            actions,
+            greeting,
             suggestions,
             tools,
         });
-    }, [gitbookFrame, buttons, welcomeMessage, suggestions, tools]);
+    }, [gitbookFrame, actions, greeting, suggestions, tools]);
 
     return (
         <iframe
diff --git a/packages/embed/src/standalone/index.ts b/packages/embed/src/standalone/index.ts
index e9c7135b54..754ecb3cd1 100644
--- a/packages/embed/src/standalone/index.ts
+++ b/packages/embed/src/standalone/index.ts
@@ -31,24 +31,50 @@ type StandaloneCalls =
     // Clear the chat
     | ['clearChat']
     // Configure the embed
-    | ['configure', Partial<GitBookEmbeddableConfiguration>]
+    | ['configure', Partial<GitBookEmbeddableConfiguration & StandaloneConfiguration>]
     // Navigate to a page
     | ['navigateToPage', string]
     // Navigate to the assistant
     | ['navigateToAssistant'];
 
+type StandaloneConfiguration = {
+    /** Configure the button to open the embed */
+    button: {
+        /** Label to be displayed in the button. */
+        label: string;
+        /** Icon to be displayed in the button. */
+        icon: 'assistant' | 'sparkle' | 'help' | 'book';
+    };
+};
+
 export type GitBookStandalone = ((...args: StandaloneCalls) => void) & {
     q?: StandaloneCalls[];
 };
 
+let widgetIframe: HTMLIFrameElement | undefined;
+let _client: GitBookClient | undefined;
+let _frame: GitBookFrameClient | undefined;
+let frameOptions: GetFrameURLOptions | undefined;
+let frameConfiguration: GitBookEmbeddableConfiguration & StandaloneConfiguration = {
+    button: {
+        label: 'Ask',
+        icon: 'assistant',
+    },
+    actions: [],
+    greeting: { title: '', subtitle: '' },
+    suggestions: [],
+    tools: [],
+    tabs: ['assistant', 'docs'],
+};
+
 const widgetButton = document.createElement('button');
 widgetButton.id = 'gitbook-widget-button';
 widgetButton.addEventListener('click', () => {
     GitBook('toggle');
 });
 widgetButton.innerHTML = `
-    <span id="gitbook-widget-button-icon"></span>
-    <span id="gitbook-widget-button-label">Ask</span>
+    <span id="gitbook-widget-button-icon" data-icon="${frameConfiguration.button.icon}"></span>
+    <span id="gitbook-widget-button-label">${frameConfiguration.button.label}</span>
 `;
 
 const widgetWindow = document.createElement('div');
@@ -58,17 +84,6 @@ widgetWindow.classList.add('hidden');
 document.body.appendChild(widgetButton);
 document.body.appendChild(widgetWindow);
 
-let widgetIframe: HTMLIFrameElement | undefined;
-let _client: GitBookClient | undefined;
-let _frame: GitBookFrameClient | undefined;
-let frameOptions: GetFrameURLOptions | undefined;
-let frameConfiguration: GitBookEmbeddableConfiguration = {
-    buttons: [],
-    welcomeMessage: '',
-    suggestions: [],
-    tools: [],
-};
-
 function getClient() {
     if (!_client) {
         throw new Error(
@@ -135,27 +150,31 @@ const GitBook = (...args: StandaloneCalls) => {
         case 'postUserMessage':
             getIframe().frame.postUserMessage(args[1]);
             break;
-        case 'configure':
+        case 'configure': {
+            const settings = args[1];
             frameConfiguration = {
                 ...frameConfiguration,
-                ...args[1],
+                ...settings,
             };
+            // Update the button label and icon
+            if (settings.button?.label) {
+                const label = widgetButton.querySelector('#gitbook-widget-button-label');
+                if (label) {
+                    label.textContent = settings.button.label;
+                }
+            }
+            if (settings.button?.icon) {
+                const icon = widgetButton.querySelector('#gitbook-widget-button-icon');
+                if (icon) {
+                    icon.setAttribute('data-icon', settings.button.icon);
+                }
+            }
+
             getIframe().frame.configure({
                 ...frameConfiguration,
-                buttons: [
-                    ...frameConfiguration.buttons,
-
-                    // Always include a close button
-                    {
-                        icon: 'close',
-                        label: 'Close',
-                        onClick: () => {
-                            GitBook('close');
-                        },
-                    },
-                ],
             });
             break;
+        }
         case 'clearChat':
             getIframe().frame.clearChat();
             break;
diff --git a/packages/embed/src/standalone/style.css b/packages/embed/src/standalone/style.css
index e66a0dfc9d..f6d02715be 100644
--- a/packages/embed/src/standalone/style.css
+++ b/packages/embed/src/standalone/style.css
@@ -113,6 +113,18 @@
     background-color: currentColor;
 }
 
+#gitbook-widget-button-icon[data-icon="sparkle"] {
+    mask-image: url("https://ka-p.fontawesome.com/releases/v6.6.0/svgs/regular/sparkle.svg?v=2&token=a463935e93");
+}
+
+#gitbook-widget-button-icon[data-icon="help"] {
+    mask-image: url("https://ka-p.fontawesome.com/releases/v6.6.0/svgs/regular/circle-question.svg?v=2&token=a463935e93");
+}
+
+#gitbook-widget-button-icon[data-icon="book"] {
+    mask-image: url("https://ka-p.fontawesome.com/releases/v6.6.0/svgs/regular/book-open.svg?v=2&token=a463935e93");
+}
+
 #gitbook-widget-button.open #gitbook-widget-button-icon {
     mask-image: url('https://ka-p.fontawesome.com/releases/v6.6.0/svgs/regular/close.svg?v=2&token=a463935e93');
 }
diff --git a/packages/gitbook/e2e/internal.spec.ts b/packages/gitbook/e2e/internal.spec.ts
index 18ce044678..d09ba95db7 100644
--- a/packages/gitbook/e2e/internal.spec.ts
+++ b/packages/gitbook/e2e/internal.spec.ts
@@ -9,6 +9,7 @@ import {
     CustomizationSidebarListStyle,
     CustomizationThemeMode,
 } from '@gitbook/api';
+import type { GitBookStandalone } from '@gitbook/embed';
 import { expect } from '@playwright/test';
 import jwt from 'jsonwebtoken';
 
@@ -170,7 +171,7 @@ const searchTestCases: Test[] = [
             await expect(page.getByTestId('ai-chat-input')).toBeFocused();
             // Override text content for visual consistency in screenshots
             await page.evaluate(() => {
-                const greeting = document.querySelector('[data-testid="ai-chat-time-greeting"]');
+                const greeting = document.querySelector('[data-testid="ai-chat-greeting-title"]');
                 if (greeting) {
                     greeting.textContent = 'Good morning';
                 }
@@ -191,7 +192,7 @@ const searchTestCases: Test[] = [
             await expect(page.getByTestId('ai-chat-input')).toBeFocused();
             // Override text content for visual consistency in screenshots
             await page.evaluate(() => {
-                const greeting = document.querySelector('[data-testid="ai-chat-time-greeting"]');
+                const greeting = document.querySelector('[data-testid="ai-chat-greeting-title"]');
                 if (greeting) {
                     greeting.textContent = 'Good morning';
                 }
@@ -212,7 +213,7 @@ const searchTestCases: Test[] = [
             await expect(page.getByTestId('ai-chat-input')).toBeFocused();
             // Override text content for visual consistency in screenshots
             await page.evaluate(() => {
-                const greeting = document.querySelector('[data-testid="ai-chat-time-greeting"]');
+                const greeting = document.querySelector('[data-testid="ai-chat-greeting-title"]');
                 if (greeting) {
                     greeting.textContent = 'Good morning';
                 }
@@ -1952,6 +1953,299 @@ const testCases: TestsCase[] = [
             ]),
         ],
     },
+    {
+        name: 'Docs Embed - GitBook (Assistant + Docs)',
+        contentBaseURL: 'https://gitbook.com/docs/~gitbook/embed/demo/',
+        tests: [
+            {
+                name: 'Basic functionality',
+                url: '',
+                run: async (page) => {
+                    const button = page.locator('#gitbook-widget-button');
+                    await expect(button).toBeVisible();
+                    await expect(button.locator('#gitbook-widget-button-label')).toHaveText('Ask');
+                    await expect(button.locator('#gitbook-widget-button-icon')).toHaveAttribute(
+                        'data-icon',
+                        'assistant'
+                    );
+
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(iframe.getByTestId('embed-tab-assistant')).toBeVisible();
+                    await expect(iframe.getByTestId('embed-tab-docs')).toBeVisible();
+
+                    await button.click(); // Toggle the window off
+                    await expect(page.locator('#gitbook-widget-window')).not.toBeVisible();
+                    await button.click(); // Toggle the window on
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                },
+            },
+            {
+                name: 'Switch between tabs',
+                url: '',
+                run: async (page) => {
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await iframe.getByTestId('embed-tab-docs').click(); // Switch to docs tab
+                    await expect(iframe.getByTestId('embed-docs-page')).toBeVisible();
+
+                    await iframe.getByTestId('embed-tab-assistant').click(); // Switch to assistant tab
+                    await expect(iframe.getByTestId('ai-chat')).toBeVisible();
+                },
+            },
+            {
+                name: 'API - navigateToPage',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate(() => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('navigateToPage', '/getting-started/quickstart');
+                    });
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(iframe.getByTestId('embed-docs-page')).toBeVisible();
+                    await expect(iframe.owner()).toHaveAttribute(
+                        'src',
+                        expect.stringContaining('getting-started/quickstart')
+                    );
+                },
+            },
+            {
+                name: 'API - postUserMessage',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate((aiPrompt) => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('postUserMessage', aiPrompt);
+                    }, AI_PROMPT);
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(iframe.getByTestId('ai-chat')).toBeVisible();
+                    await expect(iframe.getByTestId('ai-chat-message-user').first()).toHaveText(
+                        AI_PROMPT
+                    );
+                },
+            },
+            {
+                name: 'Configuration - Change standalone button label and icon',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate(() => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('configure', {
+                            button: {
+                                label: 'Docs',
+                                icon: 'book',
+                            },
+                        });
+                    });
+                    await expect(page.locator('#gitbook-widget-button-label')).toHaveText('Docs');
+                    await expect(page.locator('#gitbook-widget-button-icon')).toHaveAttribute(
+                        'data-icon',
+                        'book'
+                    );
+                },
+            },
+            {
+                name: 'Configuration - Suggested questions',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate(() => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('configure', {
+                            suggestions: [
+                                'What is GitBook?',
+                                'How do I get started?',
+                                'What can you do?',
+                            ],
+                        });
+                    });
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(
+                        iframe.getByTestId('ai-chat-suggested-question').nth(0)
+                    ).toHaveText('What is GitBook?');
+                    await expect(
+                        iframe.getByTestId('ai-chat-suggested-question').nth(1)
+                    ).toHaveText('How do I get started?');
+                    await expect(
+                        iframe.getByTestId('ai-chat-suggested-question').nth(2)
+                    ).toHaveText('What can you do?');
+                },
+            },
+            {
+                name: 'Configuration - Custom action buttons',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate((aiPrompt) => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('configure', {
+                            actions: [
+                                {
+                                    label: 'Open internal link',
+                                    icon: 'bolt',
+                                    onClick: () =>
+                                        (window.GitBook as unknown as GitBookStandalone)(
+                                            'navigateToPage',
+                                            '/getting-started/quickstart'
+                                        ),
+                                },
+                                {
+                                    label: 'Open external link',
+                                    icon: 'sparkle',
+                                    onClick: () => {
+                                        window.open('https://gitbook.com', '_blank');
+                                    },
+                                },
+                                {
+                                    label: 'Post message',
+                                    icon: 'message',
+                                    onClick: () =>
+                                        (window.GitBook as unknown as GitBookStandalone)(
+                                            'postUserMessage',
+                                            aiPrompt
+                                        ),
+                                },
+                                {
+                                    label: 'Close',
+                                    icon: 'xmark',
+                                    onClick: () =>
+                                        (window.GitBook as unknown as GitBookStandalone)('close'),
+                                },
+                            ],
+                        });
+                    }, AI_PROMPT);
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(iframe.getByTestId('embed-action')).toHaveCount(4);
+                    const actions = iframe.getByTestId('embed-action');
+
+                    await expect(actions.nth(0)).toHaveAccessibleName('Open internal link');
+                    await actions.nth(0).click();
+                    await expect(iframe.getByTestId('embed-docs-page')).toBeVisible();
+                    await expect(iframe.owner()).toHaveAttribute(
+                        'src',
+                        expect.stringContaining('getting-started/quickstart')
+                    );
+
+                    await expect(actions.nth(1)).toHaveAccessibleName('Open external link');
+                    // Intercept the new page event without navigating
+                    const [newPage] = await Promise.all([
+                        page.context().waitForEvent('page', { timeout: 5000 }),
+                        actions.nth(1).click(),
+                    ]);
+                    // Verify the new page would have opened with the expected URL
+                    expect(newPage.url()).toContain('gitbook.com');
+                    // Close it immediately to avoid navigation
+                    await newPage.close();
+
+                    await expect(actions.nth(2)).toHaveAccessibleName('Post message');
+                    await actions.nth(2).click();
+                    await expect(iframe.getByTestId('ai-chat')).toBeVisible();
+                    await expect(iframe.getByTestId('ai-chat-message-user').first()).toHaveText(
+                        AI_PROMPT
+                    );
+
+                    await expect(actions.nth(3)).toHaveAccessibleName('Close');
+                    await actions.nth(3).click();
+                    await expect(page.locator('#gitbook-widget-window')).not.toBeVisible();
+                    await page.locator('#gitbook-widget-button').click();
+                },
+            },
+            {
+                name: 'Configuration - Custom tools',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate(() => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('configure', {
+                            tools: [
+                                {
+                                    name: 'contact_support',
+                                    description: 'Contact support on behalf of the user',
+                                    execute: async () => {
+                                        return {
+                                            output: { message: 'Support message' },
+                                            summary: {
+                                                text: 'Contacted support',
+                                                icon: 'circle-question',
+                                            },
+                                        };
+                                    },
+                                    confirmation: {
+                                        icon: 'circle-question',
+                                        label: 'Contact support',
+                                    },
+                                },
+                            ],
+                        });
+                        GitBook(
+                            'postUserMessage',
+                            'I want to contact support. Call the tool directly without a preamble. Do not respond with anything else.'
+                        );
+                    });
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(iframe.getByTestId('ai-chat-message-user').first()).toHaveText(
+                        'I want to contact support. Call the tool directly without a preamble. Do not respond with anything else.'
+                    );
+                    const toolConfirmation = iframe
+                        .getByTestId('ai-chat-tool-confirmation')
+                        .first();
+                    await expect(toolConfirmation).toBeVisible({
+                        timeout: 30000,
+                    });
+                    await page.waitForTimeout(10000);
+                },
+            },
+        ],
+    },
+    {
+        name: 'Docs Embed - Docs Only',
+        contentBaseURL: 'https://gitbook.gitbook.io/test-gitbook-open/~gitbook/embed/demo/',
+        tests: [
+            {
+                name: 'Docs only',
+                url: '',
+                run: async (page) => {
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(iframe.getByTestId('embed-docs-page')).toBeVisible();
+                },
+            },
+            {
+                name: 'Table of contents',
+                url: '',
+                run: async (page) => {
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(iframe.getByTestId('embed-docs-page')).toBeVisible();
+                    const tocButton = iframe.getByTestId('toc-button');
+                    await expect(tocButton).toBeVisible();
+                    await tocButton.click();
+                    await expect(iframe.getByTestId('table-of-contents')).toBeVisible();
+                },
+            },
+            {
+                name: 'Open in new tab',
+                url: '',
+                run: async (page) => {
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(iframe.getByTestId('embed-docs-page')).toBeVisible();
+                    const openInNewTabButton = iframe.getByTestId(
+                        'embed-docs-page-open-in-new-tab'
+                    );
+                    await expect(openInNewTabButton).toBeVisible();
+                    // Intercept the new page event without navigating
+                    const [newPage] = await Promise.all([
+                        page.context().waitForEvent('page', { timeout: 5000 }),
+                        openInNewTabButton.click(),
+                    ]);
+                    // Verify the new page would have opened with the expected URL
+                    expect(newPage.url()).toContain('gitbook.gitbook.io');
+                    // Close it immediately to avoid navigation
+                    await newPage.close();
+                },
+            },
+        ],
+    },
 ];
 
 runTestCases(testCases);
diff --git a/packages/gitbook/e2e/util.ts b/packages/gitbook/e2e/util.ts
index 6b62dea8c0..92538ab52b 100644
--- a/packages/gitbook/e2e/util.ts
+++ b/packages/gitbook/e2e/util.ts
@@ -454,7 +454,7 @@ export async function waitForIcons(page: Page) {
  */
 async function waitForTOCScrolling(page: Page) {
     const viewport = await page.viewportSize();
-    if (viewport && viewport.width >= 1024) {
+    if (viewport && viewport.width >= 1024 && !page.url().includes('~gitbook')) {
         const toc = page.getByTestId('table-of-contents');
         await expect(toc).toBeVisible();
         await page.evaluate(() => {
diff --git a/packages/gitbook/src/app/sites/dynamic/[mode]/[siteURL]/[siteData]/~gitbook/embed/page.tsx b/packages/gitbook/src/app/sites/dynamic/[mode]/[siteURL]/[siteData]/~gitbook/embed/page.tsx
new file mode 100644
index 0000000000..b847b5cf7a
--- /dev/null
+++ b/packages/gitbook/src/app/sites/dynamic/[mode]/[siteURL]/[siteData]/~gitbook/embed/page.tsx
@@ -0,0 +1,21 @@
+import type { RouteLayoutParams } from '@/app/utils';
+import { getEmbeddableDynamicContext } from '@/lib/embeddable';
+import { CustomizationAIMode } from '@gitbook/api';
+import { redirect } from 'next/navigation';
+
+type PageProps = {
+    params: Promise<RouteLayoutParams>;
+};
+
+export default async function Page(props: PageProps) {
+    const params = await props.params;
+    const { context } = await getEmbeddableDynamicContext(params);
+    const baseURL = context.linker.toPathInSite('~gitbook/embed/');
+
+    // If assistant is enabled, redirect to assistant, otherwise to docs
+    if (context.customization.ai.mode === CustomizationAIMode.Assistant) {
+        redirect(`${baseURL}/assistant`);
+    } else {
+        redirect(`${baseURL}/page/`);
+    }
+}
diff --git a/packages/gitbook/src/app/sites/static/[mode]/[siteURL]/[siteData]/~gitbook/embed/assistant/page.tsx b/packages/gitbook/src/app/sites/static/[mode]/[siteURL]/[siteData]/~gitbook/embed/assistant/page.tsx
index 63e8bf316e..161e66c6e9 100644
--- a/packages/gitbook/src/app/sites/static/[mode]/[siteURL]/[siteData]/~gitbook/embed/assistant/page.tsx
+++ b/packages/gitbook/src/app/sites/static/[mode]/[siteURL]/[siteData]/~gitbook/embed/assistant/page.tsx
@@ -1,7 +1,23 @@
+import type { RouteParams } from '@/app/utils';
 import { EmbeddableAssistantPage } from '@/components/Embeddable';
+import { getEmbeddableStaticContext } from '@/lib/embeddable';
+import { CustomizationAIMode } from '@gitbook/api';
+import { redirect } from 'next/navigation';
 
 export const dynamic = 'force-static';
 
-export default async function Page() {
+type PageProps = {
+    params: Promise<RouteParams>;
+};
+
+export default async function Page(props: PageProps) {
+    const params = await props.params;
+    const { context } = await getEmbeddableStaticContext(params);
+
+    // If the assistant is not enabled, redirect to the docs
+    if (context.customization.ai.mode !== CustomizationAIMode.Assistant) {
+        redirect(`${context.linker.toPathInSite('~gitbook/embed/page/')}`);
+    }
+
     return <EmbeddableAssistantPage />;
 }
diff --git a/packages/gitbook/src/app/sites/static/[mode]/[siteURL]/[siteData]/~gitbook/embed/page.tsx b/packages/gitbook/src/app/sites/static/[mode]/[siteURL]/[siteData]/~gitbook/embed/page.tsx
new file mode 100644
index 0000000000..2f667d2802
--- /dev/null
+++ b/packages/gitbook/src/app/sites/static/[mode]/[siteURL]/[siteData]/~gitbook/embed/page.tsx
@@ -0,0 +1,23 @@
+import type { RouteLayoutParams } from '@/app/utils';
+import { getEmbeddableStaticContext } from '@/lib/embeddable';
+import { CustomizationAIMode } from '@gitbook/api';
+import { redirect } from 'next/navigation';
+
+export const dynamic = 'force-static';
+
+type PageProps = {
+    params: Promise<RouteLayoutParams>;
+};
+
+export default async function Page(props: PageProps) {
+    const params = await props.params;
+    const { context } = await getEmbeddableStaticContext(params);
+    const baseURL = context.linker.toPathInSite('~gitbook/embed/');
+
+    // If assistant is enabled, redirect to assistant, otherwise to docs
+    if (context.customization.ai.mode === CustomizationAIMode.Assistant) {
+        redirect(`${baseURL}/assistant`);
+    } else {
+        redirect(`${baseURL}/page/`);
+    }
+}
diff --git a/packages/gitbook/src/components/AI/server-actions/AIToolCallsSummary.tsx b/packages/gitbook/src/components/AI/server-actions/AIToolCallsSummary.tsx
index 731e6480ff..b4ec0f4a60 100644
--- a/packages/gitbook/src/components/AI/server-actions/AIToolCallsSummary.tsx
+++ b/packages/gitbook/src/components/AI/server-actions/AIToolCallsSummary.tsx
@@ -39,7 +39,10 @@ function ToolCallSummary(props: { toolCall: AIToolCall; context: GitBookSiteCont
     const { toolCall, context } = props;
 
     return (
-        <div className="mt-2 flex origin-top-left animate-blur-in-slow items-start gap-2 text-sm text-tint-subtle">
+        <div
+            data-testid="ai-chat-tool-summary"
+            className="mt-2 flex origin-top-left animate-blur-in-slow items-start gap-2 text-sm text-tint-subtle"
+        >
             <Icon
                 icon={getIconForToolCall(toolCall)}
                 className="mt-1 size-3 shrink-0 text-tint-subtle/8"
@@ -160,7 +163,7 @@ async function DescriptionForSearchToolCall(props: {
     const hasResults = toolCall.results.length > 0;
 
     return (
-        <details className="-ml-5 group flex w-full flex-col">
+        <details className={tcls('-ml-5 group flex w-full flex-col')}>
             <summary
                 className={tcls(
                     '-mx-2 flex list-none items-center gap-2 circular-corners:rounded-2xl rounded-corners:rounded-md pr-4 pl-7 transition-colors marker:hidden',
diff --git a/packages/gitbook/src/components/AI/useAI.tsx b/packages/gitbook/src/components/AI/useAI.tsx
index 29df1f2356..d1b7c92fcc 100644
--- a/packages/gitbook/src/components/AI/useAI.tsx
+++ b/packages/gitbook/src/components/AI/useAI.tsx
@@ -88,6 +88,7 @@ export function useAI(): AIContext {
                 <AIChatIcon
                     state={chat.loading ? 'thinking' : 'default'}
                     trademark={config.trademark}
+                    className="size-4"
                 />
             ),
             open: (query?: string) => {
diff --git a/packages/gitbook/src/components/AI/useAIChat.tsx b/packages/gitbook/src/components/AI/useAIChat.tsx
index 55317f8d76..8cb39e25df 100644
--- a/packages/gitbook/src/components/AI/useAIChat.tsx
+++ b/packages/gitbook/src/components/AI/useAIChat.tsx
@@ -96,6 +96,8 @@ export type AIChatController = {
     postMessage: (input: { message: string }) => void;
     /** Clear the conversation */
     clear: () => void;
+    /** Register an event listener */
+    on: (event: 'postMessage', listener: (input: { message: string }) => void) => () => void;
 };
 
 const AIChatControllerContext = React.createContext<AIChatController | null>(null);
@@ -137,6 +139,11 @@ export function AIChatProvider(props: {
     const [, setSearchState] = useSearch();
     const language = useLanguage();
 
+    // Event listeners storage
+    const eventsRef = React.useRef<Map<'postMessage', Array<(input: { message: string }) => void>>>(
+        new Map()
+    );
+
     // Open AI chat and sync with search state
     const onOpen = React.useCallback(() => {
         const { initialQuery } = globalState.getState();
@@ -379,8 +386,18 @@ export function AIChatProvider(props: {
                 }));
             }
 
+            // Defer event listeners to next tick so React can process state updates first
+            setTimeout(() => {
+                const listeners = eventsRef.current.get('postMessage') || [];
+                listeners.forEach((listener) => listener(input));
+            }, 0);
+
             if (query === input.message) {
                 // Return early if the message is the same as the previous message
+                globalState.setState((state) => ({
+                    ...state,
+                    opened: true,
+                }));
                 return;
             }
 
@@ -440,14 +457,31 @@ export function AIChatProvider(props: {
         }));
     }, [setSearchState]);
 
+    const onEvent = React.useCallback(
+        (event: 'postMessage', listener: (input: { message: string }) => void) => {
+            const listeners = eventsRef.current.get(event) || [];
+            listeners.push(listener);
+            eventsRef.current.set(event, listeners);
+            return () => {
+                const currentListeners = eventsRef.current.get(event) || [];
+                eventsRef.current.set(
+                    event,
+                    currentListeners.filter((l) => l !== listener)
+                );
+            };
+        },
+        []
+    );
+
     const controller = React.useMemo(() => {
         return {
             open: onOpen,
             close: onClose,
             clear: onClear,
             postMessage: onPostMessage,
+            on: onEvent,
         };
-    }, [onOpen, onClose, onClear, onPostMessage]);
+    }, [onOpen, onClose, onClear, onPostMessage, onEvent]);
 
     return (
         <AIChatControllerContext.Provider value={controller}>
diff --git a/packages/gitbook/src/components/AIChat/AIChat.tsx b/packages/gitbook/src/components/AIChat/AIChat.tsx
index aa3ca9787e..799b76e861 100644
--- a/packages/gitbook/src/components/AIChat/AIChat.tsx
+++ b/packages/gitbook/src/components/AIChat/AIChat.tsx
@@ -19,6 +19,7 @@ import {
     EmbeddableFrameButtons,
     EmbeddableFrameHeader,
     EmbeddableFrameHeaderMain,
+    EmbeddableFrameMain,
     EmbeddableFrameSubtitle,
     EmbeddableFrameTitle,
 } from '../Embeddable/EmbeddableFrame';
@@ -69,7 +70,6 @@ export function AIChat() {
 
     return (
         <div
-            data-testid="ai-chat"
             className={tcls(
                 'ai-chat inset-y-0 right-0 z-40 mx-auto flex max-w-3xl scroll-mt-36 px-4 py-4 transition-[width,opacity,margin,display] transition-discrete duration-300 sm:px-6 lg:fixed lg:w-80 lg:p-0 xl:w-96',
                 chat.opened
@@ -78,29 +78,31 @@ export function AIChat() {
             )}
         >
             <EmbeddableFrame className="relative shrink-0 border-tint-subtle border-l to-tint-base transition-all duration-300 max-lg:circular-corners:rounded-3xl max-lg:rounded-corners:rounded-md max-lg:border lg:w-80 xl:w-96">
-                <EmbeddableFrameHeader>
-                    <AIChatDynamicIcon trademark={config.trademark} />
-                    <EmbeddableFrameHeaderMain>
-                        <EmbeddableFrameTitle>
-                            {getAIChatName(language, config.trademark)}
-                        </EmbeddableFrameTitle>
-                        <AIChatSubtitle chat={chat} />
-                    </EmbeddableFrameHeaderMain>
-                    <EmbeddableFrameButtons>
-                        <AIChatControlButton />
-                        <Button
-                            onClick={() => chatController.close()}
-                            iconOnly
-                            icon="close"
-                            label={tString(language, 'close')}
-                            variant="blank"
-                            size="default"
-                        />
-                    </EmbeddableFrameButtons>
-                </EmbeddableFrameHeader>
-                <EmbeddableFrameBody>
-                    <AIChatBody chatController={chatController} chat={chat} />
-                </EmbeddableFrameBody>
+                <EmbeddableFrameMain data-testid="ai-chat">
+                    <EmbeddableFrameHeader>
+                        <AIChatDynamicIcon trademark={config.trademark} />
+                        <EmbeddableFrameHeaderMain>
+                            <EmbeddableFrameTitle>
+                                {getAIChatName(language, config.trademark)}
+                            </EmbeddableFrameTitle>
+                            <AIChatSubtitle chat={chat} />
+                        </EmbeddableFrameHeaderMain>
+                        <EmbeddableFrameButtons>
+                            <AIChatControlButton />
+                            <Button
+                                onClick={() => chatController.close()}
+                                iconOnly
+                                icon="close"
+                                label={tString(language, 'close')}
+                                variant="blank"
+                                size="default"
+                            />
+                        </EmbeddableFrameButtons>
+                    </EmbeddableFrameHeader>
+                    <EmbeddableFrameBody>
+                        <AIChatBody chatController={chatController} chat={chat} />
+                    </EmbeddableFrameBody>
+                </EmbeddableFrameMain>
             </EmbeddableFrame>
         </div>
     );
@@ -229,13 +231,14 @@ export function AIChatBody(props: {
                                 <h5
                                     className="animate-blur-in-slow font-bold text-lg text-tint-strong [@container(min-height:400px)]:text-center"
                                     style={{ animationDelay: '.5s' }}
-                                    data-testid="ai-chat-time-greeting"
+                                    data-testid="ai-chat-greeting-title"
                                 >
                                     {timeGreeting}
                                 </h5>
                                 <p
                                     className="animate-blur-in-slow text-tint [@container(min-height:400px)]:text-center"
                                     style={{ animationDelay: '.6s' }}
+                                    data-testid="ai-chat-greeting-subtitle"
                                 >
                                     {t(language, 'ai_chat_assistant_description')}
                                 </p>
diff --git a/packages/gitbook/src/components/AIChat/AIChatControlButton.tsx b/packages/gitbook/src/components/AIChat/AIChatControlButton.tsx
index f909b62ab9..61d5f8e247 100644
--- a/packages/gitbook/src/components/AIChat/AIChatControlButton.tsx
+++ b/packages/gitbook/src/components/AIChat/AIChatControlButton.tsx
@@ -1,10 +1,9 @@
 'use client';
 
 import { useLanguage } from '@/intl/client';
-import { t, tString } from '@/intl/translate';
-import { Icon } from '@gitbook/icons';
+import { t } from '@/intl/translate';
 import { useAIChatController, useAIChatState } from '../AI';
-import { Button, DropdownMenu, DropdownMenuItem } from '../primitives';
+import { Button } from '../primitives';
 
 /**
  * Button to control the chat (clear, etc.)
@@ -15,26 +14,16 @@ export function AIChatControlButton() {
     const chatController = useAIChatController();
 
     return chat.messages.length > 0 ? (
-        <DropdownMenu
-            button={
-                <Button
-                    onClick={() => {}}
-                    iconOnly
-                    icon="ellipsis"
-                    label={tString(language, 'actions')}
-                    variant="blank"
-                    size="default"
-                />
-            }
-        >
-            <DropdownMenuItem
-                onClick={() => {
-                    chatController.clear();
-                }}
-            >
-                <Icon icon="broom-wide" className="size-3 shrink-0 text-tint-subtle" />
-                {t(language, 'ai_chat_clear_conversation')}
-            </DropdownMenuItem>
-        </DropdownMenu>
+        <Button
+            onClick={() => {
+                chatController.clear();
+            }}
+            iconOnly
+            icon="trash-can"
+            label={t(language, 'ai_chat_clear_conversation')}
+            variant="blank"
+            size="default"
+            className="animate-blur-in-slow"
+        />
     ) : null;
 }
diff --git a/packages/gitbook/src/components/AIChat/AIChatSuggestedQuestions.tsx b/packages/gitbook/src/components/AIChat/AIChatSuggestedQuestions.tsx
index cbc31bff64..afd603b18e 100644
--- a/packages/gitbook/src/components/AIChat/AIChatSuggestedQuestions.tsx
+++ b/packages/gitbook/src/components/AIChat/AIChatSuggestedQuestions.tsx
@@ -17,9 +17,13 @@ export default function AIChatSuggestedQuestions(props: {
     } = props;
 
     return (
-        <div className="flex flex-col items-start gap-2 self-start">
+        <div
+            className="flex flex-col items-start gap-2 self-start"
+            data-testid="ai-chat-suggested-questions"
+        >
             {suggestions.map((question, index) => (
                 <Button
+                    data-testid="ai-chat-suggested-question"
                     key={question}
                     variant="blank"
                     size="medium"
diff --git a/packages/gitbook/src/components/AIChat/AIChatToolConfirmations.tsx b/packages/gitbook/src/components/AIChat/AIChatToolConfirmations.tsx
index c0aeb371d6..292830f37a 100644
--- a/packages/gitbook/src/components/AIChat/AIChatToolConfirmations.tsx
+++ b/packages/gitbook/src/components/AIChat/AIChatToolConfirmations.tsx
@@ -37,6 +37,7 @@ export function AIChatToolConfirmations(props: {
                     style={{ animationDelay: `${0.5 + index * 0.1}s` }}
                 >
                     <Button
+                        data-testid="ai-chat-tool-confirmation"
                         onClick={() => {
                             tool.confirm();
                         }}
diff --git a/packages/gitbook/src/components/DocumentView/Update.tsx b/packages/gitbook/src/components/DocumentView/Update.tsx
index ff10fba45c..38c80da521 100644
--- a/packages/gitbook/src/components/DocumentView/Update.tsx
+++ b/packages/gitbook/src/components/DocumentView/Update.tsx
@@ -35,7 +35,7 @@ export function Update(props: BlockProps<DocumentBlockUpdate>) {
         short: formatDateShort(parsedDate),
     }[dateFormat];
 
-    const isSticky = parentUpdates.data.sticky;
+    const isSticky = true; // TODO: use parentUpdates.data.sticky once implemented
 
     return (
         <div
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableAIChat.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableAIChat.tsx
index 655e692cd6..41f1bbd3f6 100644
--- a/packages/gitbook/src/components/Embeddable/EmbeddableAIChat.tsx
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableAIChat.tsx
@@ -4,7 +4,6 @@ import { useAI, useAIChatController, useAIChatState } from '@/components/AI';
 import {
     AIChatBody,
     AIChatControlButton,
-    AIChatDynamicIcon,
     AIChatSubtitle,
     getAIChatName,
 } from '@/components/AIChat';
@@ -18,9 +17,15 @@ import {
     EmbeddableFrameButtons,
     EmbeddableFrameHeader,
     EmbeddableFrameHeaderMain,
+    EmbeddableFrameMain,
+    EmbeddableFrameSidebar,
     EmbeddableFrameTitle,
 } from './EmbeddableFrame';
-import { EmbeddableIframeButtons, useEmbeddableConfiguration } from './EmbeddableIframeAPI';
+import {
+    EmbeddableIframeButtons,
+    EmbeddableIframeTabs,
+    useEmbeddableConfiguration,
+} from './EmbeddableIframeAPI';
 
 /**
  * Embeddable AI chat window in an iframe.
@@ -32,6 +37,10 @@ export function EmbeddableAIChat() {
     const configuration = useEmbeddableConfiguration();
     const language = useLanguage();
 
+    React.useEffect(() => {
+        chatController.open();
+    }, [chatController]);
+
     // Track the view of the AI chat
     const trackEvent = useTrackEvent();
     React.useEffect(() => {
@@ -48,26 +57,30 @@ export function EmbeddableAIChat() {
 
     return (
         <EmbeddableFrame>
-            <EmbeddableFrameHeader>
-                <AIChatDynamicIcon trademark={config.trademark} />
-                <EmbeddableFrameHeaderMain>
-                    <EmbeddableFrameTitle>
-                        {getAIChatName(language, config.trademark)}
-                    </EmbeddableFrameTitle>
-                    <AIChatSubtitle chat={chat} />
-                </EmbeddableFrameHeaderMain>
-                <EmbeddableFrameButtons>
-                    <AIChatControlButton />
-                    <EmbeddableIframeButtons />
-                </EmbeddableFrameButtons>
-            </EmbeddableFrameHeader>
-            <EmbeddableFrameBody>
-                <AIChatBody
-                    chatController={chatController}
-                    chat={chat}
-                    suggestions={configuration.suggestions}
-                />
-            </EmbeddableFrameBody>
+            <EmbeddableFrameSidebar>
+                <EmbeddableIframeTabs active="assistant" />
+                <EmbeddableIframeButtons />
+            </EmbeddableFrameSidebar>
+            <EmbeddableFrameMain data-testid="ai-chat">
+                <EmbeddableFrameHeader>
+                    <EmbeddableFrameHeaderMain>
+                        <EmbeddableFrameTitle>
+                            {getAIChatName(language, config.trademark)}
+                        </EmbeddableFrameTitle>
+                        <AIChatSubtitle chat={chat} />
+                    </EmbeddableFrameHeaderMain>
+                    <EmbeddableFrameButtons>
+                        <AIChatControlButton />
+                    </EmbeddableFrameButtons>
+                </EmbeddableFrameHeader>
+                <EmbeddableFrameBody>
+                    <AIChatBody
+                        chatController={chatController}
+                        chat={chat}
+                        suggestions={configuration.suggestions}
+                    />
+                </EmbeddableFrameBody>
+            </EmbeddableFrameMain>
         </EmbeddableFrame>
     );
 }
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableDocsPage.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableDocsPage.tsx
index ee8b26a536..87de3e7d4a 100644
--- a/packages/gitbook/src/components/Embeddable/EmbeddableDocsPage.tsx
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableDocsPage.tsx
@@ -1,18 +1,24 @@
 import { type PagePathParams, getSitePageData } from '@/components/SitePage';
-
-import { PageBody } from '@/components/PageBody';
 import type { GitBookSiteContext } from '@/lib/context';
 import { SiteInsightsDisplayContext } from '@gitbook/api';
 import type { Metadata } from 'next';
-import { Button } from '../primitives';
+import { HeaderMobileMenu } from '../Header/HeaderMobileMenu';
+import { PageBody } from '../PageBody';
+import { SiteSectionTabs, encodeClientSiteSections } from '../SiteSections';
+import { TableOfContents } from '../TableOfContents';
+import { ScrollContainer } from '../primitives/ScrollContainer';
+import { EmbeddableDocsPageControlButtons } from './EmbeddableDocsPageControlButtons';
 import {
     EmbeddableFrame,
     EmbeddableFrameBody,
     EmbeddableFrameButtons,
     EmbeddableFrameHeader,
     EmbeddableFrameHeaderMain,
+    EmbeddableFrameMain,
+    EmbeddableFrameSidebar,
+    EmbeddableFrameTitle,
 } from './EmbeddableFrame';
-import { EmbeddableIframeButtons } from './EmbeddableIframeAPI';
+import { EmbeddableIframeButtons, EmbeddableIframeTabs } from './EmbeddableIframeAPI';
 
 export const dynamic = 'force-static';
 
@@ -33,32 +39,54 @@ export async function EmbeddableDocsPage(props: EmbeddableDocsPageProps) {
 
     return (
         <EmbeddableFrame className="site-background">
-            <EmbeddableFrameHeader>
-                <EmbeddableFrameHeaderMain>
-                    <Button
-                        href={context.linker.toPathInSite('~gitbook/embed/assistant')}
-                        size="default"
-                        variant="blank"
-                        icon="arrow-left"
-                        label="Back"
-                    />
-                </EmbeddableFrameHeaderMain>
-                <EmbeddableFrameButtons>
-                    <EmbeddableIframeButtons />
-                </EmbeddableFrameButtons>
-            </EmbeddableFrameHeader>
-            <EmbeddableFrameBody>
-                <div className="flex-1 overflow-auto p-6">
-                    <PageBody
-                        context={context}
-                        page={page}
-                        ancestors={ancestors}
-                        document={document}
-                        withPageFeedback={withPageFeedback}
-                        insightsDisplayContext={SiteInsightsDisplayContext.Embed}
-                    />
+            <EmbeddableFrameSidebar>
+                <EmbeddableIframeTabs active="docs" />
+                <EmbeddableIframeButtons />
+            </EmbeddableFrameSidebar>
+            <EmbeddableFrameMain data-testid="embed-docs-page">
+                <div className="relative flex not-hydrated:animate-blur-in-slow flex-col">
+                    <EmbeddableFrameHeader>
+                        <HeaderMobileMenu className="-ml-2 page-no-toc:hidden" />
+                        <EmbeddableFrameHeaderMain>
+                            <EmbeddableFrameTitle>{context.site.title}</EmbeddableFrameTitle>
+                        </EmbeddableFrameHeaderMain>
+                        <EmbeddableFrameButtons>
+                            <EmbeddableDocsPageControlButtons
+                                href={context.linker
+                                    .toPathForPage({
+                                        pages: context.revision.pages,
+                                        page,
+                                    })
+                                    .replace(/~gitbook\/embed\/page\/?/, '')}
+                            />
+                        </EmbeddableFrameButtons>
+                    </EmbeddableFrameHeader>
+                    {context.sections ? (
+                        <SiteSectionTabs
+                            className="-mt-2 border-tint-subtle border-b"
+                            sections={encodeClientSiteSections(context, context.sections)}
+                        />
+                    ) : null}
                 </div>
-            </EmbeddableFrameBody>
+                <EmbeddableFrameBody>
+                    <ScrollContainer
+                        orientation="vertical"
+                        className="not-hydrated:animate-blur-in-slow"
+                        contentClassName="p-4"
+                        fadeEdges={context.sections ? [] : ['leading']}
+                    >
+                        <TableOfContents className="pt-0" context={context} />
+                        <PageBody
+                            context={context}
+                            page={page}
+                            ancestors={ancestors}
+                            document={document}
+                            withPageFeedback={withPageFeedback}
+                            insightsDisplayContext={SiteInsightsDisplayContext.Embed}
+                        />
+                    </ScrollContainer>
+                </EmbeddableFrameBody>
+            </EmbeddableFrameMain>
         </EmbeddableFrame>
     );
 }
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableDocsPageControlButtons.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableDocsPageControlButtons.tsx
new file mode 100644
index 0000000000..722b5018bd
--- /dev/null
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableDocsPageControlButtons.tsx
@@ -0,0 +1,22 @@
+'use client';
+
+import { tString, useLanguage } from '@/intl/client';
+import { Button } from '../primitives';
+
+export function EmbeddableDocsPageControlButtons(props: { href: string }) {
+    const { href } = props;
+    const language = useLanguage();
+
+    return (
+        <Button
+            icon="arrow-up-right-from-square"
+            data-testid="embed-docs-page-open-in-new-tab"
+            label={tString(language, 'open_in_new_tab')}
+            href={href}
+            target="_blank"
+            iconOnly
+            variant="blank"
+            size="default"
+        />
+    );
+}
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableFrame.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableFrame.tsx
index f46fa37e96..3dd1d63a73 100644
--- a/packages/gitbook/src/components/Embeddable/EmbeddableFrame.tsx
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableFrame.tsx
@@ -17,7 +17,7 @@ export const EmbeddableFrame = React.forwardRef<HTMLDivElement, EmbeddableFrameP
             <div
                 {...divProps}
                 className={tcls(
-                    'flex h-full grow flex-col overflow-hidden bg-radial-[circle_at_bottom] from-primary to-50% to-transparent text-sm text-tint',
+                    'flex h-full grow overflow-hidden bg-radial-[circle_at_bottom] from-primary to-50% to-transparent text-sm text-tint',
                     divProps.className
                 )}
                 ref={ref}
@@ -28,13 +28,23 @@ export const EmbeddableFrame = React.forwardRef<HTMLDivElement, EmbeddableFrameP
     }
 );
 
+export function EmbeddableFrameMain(props: React.ComponentProps<'div'>) {
+    const { children, ...rest } = props;
+
+    return (
+        <div className="flex flex-1 flex-col overflow-hidden" {...rest}>
+            {children}
+        </div>
+    );
+}
+
 export function EmbeddableFrameHeader(props: {
     children: React.ReactNode;
 }) {
     const { children } = props;
 
     return (
-        <div className="relative z-10 flex animate-fade-in-slow select-none items-center gap-2 px-4 py-2 text-tint-strong">
+        <div className="relative z-10 flex not-hydrated:animate-blur-in-slow select-none items-center gap-2 px-4 py-2.5 text-tint-strong">
             {children}
         </div>
     );
@@ -45,7 +55,7 @@ export function EmbeddableFrameHeaderMain(props: {
 }) {
     const { children } = props;
 
-    return <div className="flex flex-1 flex-col">{children}</div>;
+    return <div className="flex h-8 flex-1 flex-col justify-center">{children}</div>;
 }
 
 export function EmbeddableFrameBody(props: {
@@ -82,10 +92,21 @@ export function EmbeddableFrameSubtitle(props: {
     );
 }
 
+export function EmbeddableFrameSidebar(props: { children: React.ReactNode }) {
+    const { children } = props;
+
+    return (
+        <div className="flex w-13 shrink-0 origin-top not-hydrated:animate-blur-in-slow flex-col gap-2 overflow-hidden border-tint-solid/3 border-r bg-tint-solid/1 p-2 transition-all transition-discrete duration-300 empty:hidden empty:w-0 empty:px-0">
+            {children}
+        </div>
+    );
+}
+
 export function EmbeddableFrameButtons(props: {
+    className?: string;
     children: React.ReactNode;
 }) {
-    const { children } = props;
+    const { children, className } = props;
 
-    return <div className="-mr-2 ml-auto flex gap-2">{children}</div>;
+    return <div className={tcls('flex gap-2', className)}>{children}</div>;
 }
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableIframeAPI.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableIframeAPI.tsx
index f37dd4c55f..1d9fbb495b 100644
--- a/packages/gitbook/src/components/Embeddable/EmbeddableIframeAPI.tsx
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableIframeAPI.tsx
@@ -4,17 +4,23 @@ import type { GitBookEmbeddableConfiguration, ParentToFrameMessage } from '@gitb
 import { createChannel } from 'bidc';
 import React from 'react';
 
-import { useAIChatController } from '@/components/AI';
+import { useAI, useAIChatController } from '@/components/AI';
+import { CustomizationAIMode } from '@gitbook/api';
 import { useRouter } from 'next/navigation';
 import { createStore, useStore } from 'zustand';
 import { integrationsAssistantTools } from '../Integrations';
 import { Button } from '../primitives';
 
-const embeddableConfiguration = createStore<GitBookEmbeddableConfiguration>(() => ({
-    buttons: [],
-    welcomeMessage: '',
+const embeddableConfiguration = createStore<
+    GitBookEmbeddableConfiguration & { baseURL: string; siteTitle: string }
+>(() => ({
+    tabs: [],
+    actions: [],
+    greeting: { title: '', subtitle: '' },
     suggestions: [],
     tools: [],
+    baseURL: '',
+    siteTitle: '',
 }));
 
 /**
@@ -22,12 +28,24 @@ const embeddableConfiguration = createStore<GitBookEmbeddableConfiguration>(() =
  */
 export function EmbeddableIframeAPI(props: {
     baseURL: string;
+    siteTitle: string;
 }) {
-    const { baseURL } = props;
+    const { baseURL, siteTitle } = props;
 
     const router = useRouter();
     const chatController = useAIChatController();
 
+    React.useEffect(() => {
+        embeddableConfiguration.setState({ baseURL });
+        embeddableConfiguration.setState({ siteTitle });
+    }, [baseURL, siteTitle]);
+
+    React.useEffect(() => {
+        return chatController.on('postMessage', () => {
+            router.push(`${baseURL}/assistant`);
+        });
+    }, [router, baseURL, chatController]);
+
     React.useEffect(() => {
         if (window.parent === window) {
             return;
@@ -50,6 +68,7 @@ export function EmbeddableIframeAPI(props: {
                     chatController.postMessage({
                         message: message.message,
                     });
+                    router.push(`${baseURL}/assistant`);
                     break;
                 }
                 case 'configure': {
@@ -82,7 +101,9 @@ export function EmbeddableIframeAPI(props: {
 /**
  * Hook to get the configuration from the parent window.
  */
-export function useEmbeddableConfiguration<T = GitBookEmbeddableConfiguration>(
+export function useEmbeddableConfiguration<
+    T = GitBookEmbeddableConfiguration & { baseURL: string; siteTitle: string },
+>(
     // @ts-expect-error - This is a workaround to allow the function to be optional.
     fn: (state: GitBookEmbeddableConfiguration) => T = (state) => state
 ) {
@@ -93,23 +114,109 @@ export function useEmbeddableConfiguration<T = GitBookEmbeddableConfiguration>(
  * Display the buttons defined by the parent window.
  */
 export function EmbeddableIframeButtons() {
-    const buttons = useEmbeddableConfiguration((state) => state.buttons);
+    const actions = useEmbeddableConfiguration((state) => state.actions);
 
     return (
         <>
-            {buttons.map((button) => (
+            {actions.length > 0 && (
+                <hr className="my-2 border-0 border-tint-subtle border-b first:hidden" />
+            )}
+            {actions.map((action, index) => (
                 <Button
-                    key={button.label}
+                    data-testid="embed-action"
+                    key={action.label}
                     size="default"
                     variant="blank"
-                    icon={button.icon}
-                    label={button.label}
+                    icon={action?.icon ?? 'square-question'}
+                    label={action?.label}
                     iconOnly
+                    className="not-hydrated:animate-blur-in-slow [&_.button-leading-icon]:size-5"
+                    disabled={!action.onClick}
                     onClick={() => {
-                        button.onClick();
+                        action.onClick?.();
                     }}
+                    tooltipProps={{
+                        contentProps: {
+                            side: 'right',
+                        },
+                    }}
+                    style={{ animationDelay: `${index * 100}ms` }}
                 />
             ))}
         </>
     );
 }
+
+export function EmbeddableIframeTabs(props: { active?: string }) {
+    const { active = 'assistant' } = props;
+    let { tabs: configuredTabs, actions } = useEmbeddableConfiguration();
+
+    if (configuredTabs.length === 0) {
+        configuredTabs = ['assistant', 'docs'];
+    }
+
+    const { assistants, config } = useAI();
+
+    const router = useRouter();
+    const { baseURL, siteTitle } = useEmbeddableConfiguration();
+
+    // Override the active tab if it doesn't match the configured tabs
+    React.useEffect(() => {
+        if (active === 'assistant' && !configuredTabs.includes('assistant')) {
+            router.replace(`${baseURL}/page`);
+        } else if (active === 'docs' && !configuredTabs.includes('docs')) {
+            router.replace(`${baseURL}/assistant`);
+        }
+    }, [configuredTabs, baseURL, router, active]);
+
+    const tabs = [
+        config.aiMode === CustomizationAIMode.Assistant &&
+        assistants[0] &&
+        configuredTabs.includes('assistant')
+            ? {
+                  key: 'assistant',
+                  label: assistants[0].label,
+                  icon: assistants[0].icon,
+                  onClick: () => {
+                      router.push(`${baseURL}/assistant`);
+                  },
+              }
+            : null,
+        configuredTabs.includes('docs')
+            ? {
+                  key: 'docs',
+                  label: siteTitle,
+                  icon: 'book-open',
+                  onClick: () => {
+                      router.push(`${baseURL}/page/`);
+                  },
+              }
+            : null,
+    ].filter((tab) => tab !== null);
+
+    return (
+        <>
+            {tabs.length > 1 || actions.length > 0
+                ? tabs.map((tab) => (
+                      <Button
+                          key={tab.key}
+                          data-testid={`embed-tab-${tab.key}`}
+                          label={tab.label}
+                          size="default"
+                          variant="blank"
+                          icon={tab.icon}
+                          active={tab.key === active}
+                          className="not-hydrated:animate-blur-in-slow [&_.button-leading-icon]:size-5"
+                          iconOnly
+                          onClick={tab.onClick}
+                          tooltipProps={{
+                              contentProps: {
+                                  side: 'right',
+                              },
+                          }}
+                      />
+                  ))
+                : null}
+        </>
+    );
+}
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableRootLayout.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableRootLayout.tsx
index 104d50a3a6..b48fff14dd 100644
--- a/packages/gitbook/src/components/Embeddable/EmbeddableRootLayout.tsx
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableRootLayout.tsx
@@ -7,8 +7,10 @@ import {
 } from '@/components/SiteLayout';
 import type { VisitorAuthClaims } from '@/lib/adaptive';
 import type { GitBookSiteContext } from '@/lib/context';
-import { CustomizationAIMode } from '@gitbook/api';
+import { SiteInsightsTrademarkPlacement } from '@gitbook/api';
 import { SpaceLayoutServerContext } from '../SpaceLayout';
+import { TrademarkLink } from '../TableOfContents/Trademark';
+import { NavigationLoader } from '../primitives/NavigationLoader';
 import { EmbeddableIframeAPI } from './EmbeddableIframeAPI';
 
 type EmbeddableRootLayoutProps = {
@@ -29,12 +31,16 @@ export async function EmbeddableRootLayout({
     return (
         <CustomizationRootLayout context={context}>
             <SiteLayoutClientContexts
-                forcedTheme={context.customization.themes.default}
+                forcedTheme={
+                    context.customization.themes.toggeable
+                        ? undefined
+                        : context.customization.themes.default
+                }
                 externalLinksTarget={context.customization.externalLinks.target}
                 contextId={context.contextId}
             >
                 <AIContextProvider
-                    aiMode={CustomizationAIMode.Assistant}
+                    aiMode={context.customization.ai.mode}
                     trademark={context.customization.trademark.enabled}
                 >
                     <SpaceLayoutServerContext
@@ -46,9 +52,20 @@ export async function EmbeddableRootLayout({
                             asEmbeddable: true,
                         }}
                     >
-                        <div className="fixed inset-0 flex flex-col">{children}</div>
+                        <NavigationLoader />
+                        <div className="fixed inset-0 flex flex-col">
+                            {children}
+                            {context.customization.trademark.enabled ? (
+                                <TrademarkLink
+                                    className="rounded-none border-tint-solid/3 border-t bg-tint-solid/1 px-4 py-2.5 text-tint/8 ring-0"
+                                    context={context}
+                                    placement={SiteInsightsTrademarkPlacement.Embed}
+                                />
+                            ) : null}
+                        </div>
                         <EmbeddableIframeAPI
-                            baseURL={context.linker.toPathInSpace('~gitbook/embed/')}
+                            baseURL={context.linker.toPathInSite('~gitbook/embed/')}
+                            siteTitle={context.site.title}
                         />
                     </SpaceLayoutServerContext>
                 </AIContextProvider>
diff --git a/packages/gitbook/src/components/Header/HeaderMobileMenu.tsx b/packages/gitbook/src/components/Header/HeaderMobileMenu.tsx
index 3b0ed49dfc..dea4ff836e 100644
--- a/packages/gitbook/src/components/Header/HeaderMobileMenu.tsx
+++ b/packages/gitbook/src/components/Header/HeaderMobileMenu.tsx
@@ -1,13 +1,11 @@
 'use client';
-
-import { Icon } from '@gitbook/icons';
 import { usePathname } from 'next/navigation';
-import { useEffect, useRef } from 'react';
+import { useEffect, useRef, useState } from 'react';
 
 import { tString, useLanguage } from '@/intl/client';
-import { tcls } from '@/lib/tailwind';
 
 import { useScrollListener } from '../hooks/useScrollListener';
+import { Button, type ButtonProps } from '../primitives';
 
 const globalClassName = 'navigation-open';
 
@@ -16,18 +14,22 @@ const SCROLL_DISTANCE = 320;
 /**
  * Button to show/hide the table of content on mobile.
  */
-export function HeaderMobileMenu(props: Partial<React.ButtonHTMLAttributes<HTMLButtonElement>>) {
+export function HeaderMobileMenu(props: ButtonProps) {
     const language = useLanguage();
 
     const pathname = usePathname();
     const hasScrollRef = useRef(false);
 
+    const [isOpen, setIsOpen] = useState(false);
+
     const toggleNavigation = () => {
         if (!hasScrollRef.current && document.body.classList.contains(globalClassName)) {
             document.body.classList.remove(globalClassName);
+            setIsOpen(false);
         } else {
             document.body.classList.add(globalClassName);
             window.scrollTo(0, 0);
+            setIsOpen(true);
         }
     };
 
@@ -42,16 +44,16 @@ export function HeaderMobileMenu(props: Partial<React.ButtonHTMLAttributes<HTMLB
     }, [pathname]);
 
     return (
-        <button
-            {...props}
-            aria-label={tString(language, 'table_of_contents_button_label')}
+        <Button
+            data-testid="toc-button"
+            icon="bars"
+            iconOnly
+            variant="blank"
+            size="default"
+            label={tString(language, 'table_of_contents_button_label')}
             onClick={toggleNavigation}
-            className={tcls(
-                'flex flex-row items-center rounded-sm straight-corners:rounded-xs px-2 py-1',
-                props.className
-            )}
-        >
-            <Icon icon="bars" className="size-4 text-inherit" />
-        </button>
+            active={isOpen}
+            {...props}
+        />
     );
 }
diff --git a/packages/gitbook/src/components/SiteSections/SiteSectionTabs.tsx b/packages/gitbook/src/components/SiteSections/SiteSectionTabs.tsx
index d2ea495a5f..e176a3f826 100644
--- a/packages/gitbook/src/components/SiteSections/SiteSectionTabs.tsx
+++ b/packages/gitbook/src/components/SiteSections/SiteSectionTabs.tsx
@@ -33,6 +33,7 @@ export function SiteSectionTabs(props: {
         children,
     } = props;
 
+    const containerRef = React.useRef<HTMLDivElement>(null);
     const currentTriggerRef = React.useRef<HTMLButtonElement | null>(null);
     const [offset, setOffset] = React.useState<number | null>(null);
     const [value, setValue] = React.useState<string | undefined>();
@@ -41,12 +42,15 @@ export function SiteSectionTabs(props: {
 
     React.useEffect(() => {
         const trigger = currentTriggerRef.current;
-        if (!value || !trigger) {
+        const container = containerRef.current;
+        if (!value || !trigger || !container) {
             return;
         }
 
-        const triggerWidth = trigger.getBoundingClientRect().width;
-        const triggerLeft = trigger.getBoundingClientRect().left;
+        const triggerWidth = trigger.getBoundingClientRect().width - SCREEN_OFFSET;
+        const triggerLeft =
+            trigger.getBoundingClientRect().left -
+            (window.innerWidth - container.getBoundingClientRect().width) / 2;
         setOffset(triggerLeft + triggerWidth / 2);
     }, [value]);
 
@@ -58,6 +62,7 @@ export function SiteSectionTabs(props: {
                 'page-default-width:2xl:px-[calc((100%-1536px+4rem)/2)]',
                 className
             )}
+            ref={containerRef}
             value={value}
             onValueChange={setValue}
             skipDelayDuration={500}
@@ -141,14 +146,14 @@ export function SiteSectionTabs(props: {
             {children}
 
             <div
-                className="fixed top-full left-0 z-20 flex w-full"
+                className="absolute top-full left-0 z-20 flex w-full"
                 style={{
                     padding: `0 ${SCREEN_OFFSET}px 0 ${SCREEN_OFFSET}px`,
                 }}
             >
                 <NavigationMenu.Viewport
                     className={tcls(
-                        'relative origin-top overflow-auto circular-corners:rounded-3xl rounded-corners:rounded-xl border border-tint bg-tint-base shadow-lg transition-transform duration-250 ease-in-out',
+                        'relative origin-top overflow-auto circular-corners:rounded-3xl rounded-corners:rounded-xl border border-tint bg-tint-base shadow-lg ease-in-out',
                         '-mt-0.5 w-full md:w-max',
                         'max-h-[calc(100vh-8rem)] data-[state=closed]:animate-scale-out data-[state=open]:animate-scale-in',
                         "[&:not([style*='--radix-navigation-menu-viewport-width'])]:hidden" // The viewport width is only calculated once it's triggered, and can take a while. We hide the viewport until it's ready.
@@ -156,7 +161,7 @@ export function SiteSectionTabs(props: {
                     style={{
                         translate:
                             !isMobile && offset
-                                ? `clamp(0px, calc(${offset}px - ${SCREEN_OFFSET}px - 50%), calc(100vw - var(--radix-navigation-menu-viewport-width, 0px) - ${SCREEN_OFFSET * 3}px)) 0 0`
+                                ? `clamp(0px, calc(${offset}px - var(--radix-navigation-menu-viewport-width, 0px)/2), calc(100vw - var(--radix-navigation-menu-viewport-width, 0px) - ${SCREEN_OFFSET * 3}px)) 0 0`
                                 : '0 0 0', // TranslateZ is needed to force a stacking context, fixing a rendering bug on Safari
                         display: offset === null ? 'none' : undefined,
                     }}
diff --git a/packages/gitbook/src/components/TableOfContents/TableOfContents.tsx b/packages/gitbook/src/components/TableOfContents/TableOfContents.tsx
index 632135a283..729507d3d0 100644
--- a/packages/gitbook/src/components/TableOfContents/TableOfContents.tsx
+++ b/packages/gitbook/src/components/TableOfContents/TableOfContents.tsx
@@ -13,8 +13,9 @@ export async function TableOfContents(props: {
     context: GitBookSiteContext;
     header?: React.ReactNode; // Displayed outside the scrollable TOC as a sticky header
     innerHeader?: React.ReactNode; // Displayed outside the scrollable TOC, directly above the page list
+    className?: string;
 }) {
-    const { innerHeader, context, header } = props;
+    const { innerHeader, context, header, className } = props;
     const { customization, revision } = context;
 
     const pages = await encodeClientTableOfContents(context, revision.pages, revision.pages);
@@ -74,7 +75,8 @@ export async function TableOfContents(props: {
                     'gap-4',
 
                     'navigation-open:border-b',
-                    'border-tint-subtle'
+                    'border-tint-subtle',
+                    className
                 )}
             >
                 {header && header}
diff --git a/packages/gitbook/src/components/TableOfContents/Trademark.tsx b/packages/gitbook/src/components/TableOfContents/Trademark.tsx
index bd077d84cf..f5dbe8387d 100644
--- a/packages/gitbook/src/components/TableOfContents/Trademark.tsx
+++ b/packages/gitbook/src/components/TableOfContents/Trademark.tsx
@@ -13,7 +13,9 @@ import { Link } from '../primitives';
 export function Trademark(props: {
     context: GitBookSpaceContext;
     placement: SiteInsightsTrademarkPlacement;
+    className?: string;
 }) {
+    const { className, ...rest } = props;
     return (
         <div
             className={tcls(
@@ -55,10 +57,12 @@ export function Trademark(props: {
                 '[html.sidebar-filled.theme-bold.tint_&]:before:to-tint-subtle',
                 '[html.sidebar-filled.theme-muted_&]:before:to-tint-base',
                 '[html.sidebar-filled.theme-bold.tint_&]:before:to-tint-base',
-                'page-no-toc:before:to-transparent!'
+                'page-no-toc:before:to-transparent!',
+
+                className
             )}
         >
-            <TrademarkLink {...props} />
+            <TrademarkLink {...rest} />
         </div>
     );
 }
@@ -69,8 +73,9 @@ export function Trademark(props: {
 export function TrademarkLink(props: {
     context: GitBookSpaceContext;
     placement: SiteInsightsTrademarkPlacement;
+    className?: string;
 }) {
-    const { context, placement } = props;
+    const { context, placement, className } = props;
     const { space } = context;
     const language = getSpaceLanguage(context);
 
@@ -85,9 +90,6 @@ export function TrademarkLink(props: {
             href={url.toString()}
             className={tcls(
                 'text-sm',
-                // 'lg:max-xl:page-no-toc:text-xs',
-                // 'lg:max-xl:page-no-toc:px-3',
-                // 'lg:max-xl:page-no-toc:py-3',
                 'font-semibold',
                 'text-tint',
 
@@ -113,29 +115,17 @@ export function TrademarkLink(props: {
                 'ring-tint-subtle',
 
                 'transition-colors',
-                'pointer-events-auto'
+                'pointer-events-auto',
+
+                className
             )}
             insights={{
                 type: 'trademark_click',
                 placement,
             }}
         >
-            <Icon
-                icon="gitbook"
-                className={tcls(
-                    'size-5',
-                    // 'lg:max-xl:page-no-toc:size-4',
-                    'shrink-0'
-                )}
-            />
-            <span
-                className={tcls(
-                    'ml-3'
-                    // 'lg:max-xl:page-no-toc:ml-2'
-                )}
-            >
-                {t(language, 'powered_by_gitbook')}
-            </span>
+            <Icon icon="gitbook" className={tcls('size-5', 'shrink-0')} />
+            <span className={tcls('ml-3')}>{t(language, 'powered_by_gitbook')}</span>
         </Link>
     );
 }
diff --git a/packages/gitbook/src/components/primitives/Button.tsx b/packages/gitbook/src/components/primitives/Button.tsx
index 4d649d3bed..79bb7e48a1 100644
--- a/packages/gitbook/src/components/primitives/Button.tsx
+++ b/packages/gitbook/src/components/primitives/Button.tsx
@@ -7,7 +7,7 @@ import { type ClassValue, tcls } from '@/lib/tailwind';
 import { Icon, type IconName } from '@gitbook/icons';
 import { Link, type LinkInsightsProps } from './Link';
 import { useClassnames } from './StyleProvider';
-import { Tooltip } from './Tooltip';
+import { Tooltip, type TooltipProps } from './Tooltip';
 
 export type ButtonProps = {
     href?: string;
@@ -20,6 +20,7 @@ export type ButtonProps = {
     trailing?: React.ReactNode;
     children?: React.ReactNode;
     active?: boolean;
+    tooltipProps?: TooltipProps;
 } & LinkInsightsProps &
     React.HTMLAttributes<HTMLElement>;
 
@@ -112,6 +113,7 @@ export const Button = React.forwardRef<
             active,
             trailing,
             disabled,
+            tooltipProps,
             ...rest
         },
         ref
@@ -133,18 +135,29 @@ export const Button = React.forwardRef<
         );
         const buttonOnlyClassNames = useClassnames(['ButtonStyles']);
 
+        let iconElement = null;
+        if (icon) {
+            if (React.isValidElement(icon)) {
+                type IconElement = React.ReactElement<React.SVGProps<SVGSVGElement>>;
+                iconElement = React.cloneElement(icon as IconElement, {
+                    className: tcls(
+                        'button-leading-icon size-[1em] shrink-0',
+                        (icon as IconElement).props.className
+                    ),
+                });
+            } else {
+                iconElement = (
+                    <Icon
+                        icon={icon as IconName}
+                        className={tcls('button-leading-icon size-[1em] shrink-0')}
+                    />
+                );
+            }
+        }
+
         const content = (
             <>
-                {icon ? (
-                    typeof icon === 'string' ? (
-                        <Icon
-                            icon={icon as IconName}
-                            className={tcls('button-leading-icon size-[1em] shrink-0')}
-                        />
-                    ) : (
-                        icon
-                    )
-                ) : null}
+                {iconElement}
                 {iconOnly || (!children && !label) ? null : (
                     <span className="button-content truncate">{children ?? label}</span>
                 )}
@@ -184,9 +197,13 @@ export const Button = React.forwardRef<
 
         return (children || iconOnly) && label ? (
             <Tooltip
-                rootProps={{ open: disabled === true ? false : undefined }}
+                rootProps={{
+                    open: disabled === true ? false : undefined,
+                    ...tooltipProps?.rootProps,
+                }}
                 label={label}
-                triggerProps={{ disabled }}
+                triggerProps={{ disabled, ...tooltipProps?.triggerProps }}
+                contentProps={{ ...tooltipProps?.contentProps }}
             >
                 {button}
             </Tooltip>
diff --git a/packages/gitbook/src/components/primitives/Link.tsx b/packages/gitbook/src/components/primitives/Link.tsx
index 80ae283a2d..13bf3bd68f 100644
--- a/packages/gitbook/src/components/primitives/Link.tsx
+++ b/packages/gitbook/src/components/primitives/Link.tsx
@@ -78,8 +78,8 @@ export function Link(props: LinkProps) {
 
     const onClick = (event: React.MouseEvent<HTMLAnchorElement>) => {
         const isExternalWithOrigin = isExternalLink(href, window.location.origin);
-        // Only trigger navigation context for internal links without modifier keys (i.e. open in new tab).
-        if (!isExternal && !event.ctrlKey && !event.metaKey) {
+        // Only trigger navigation context for internal links in the same window without modifier keys (i.e. open in new tab).
+        if (!isExternal && target !== '_blank' && !event.ctrlKey && !event.metaKey) {
             onNavigationClick(href);
         }
 
diff --git a/packages/gitbook/src/components/primitives/Tooltip.tsx b/packages/gitbook/src/components/primitives/Tooltip.tsx
index 7e2a710df2..1f9f389b46 100644
--- a/packages/gitbook/src/components/primitives/Tooltip.tsx
+++ b/packages/gitbook/src/components/primitives/Tooltip.tsx
@@ -3,12 +3,18 @@
 import { tcls } from '@/lib/tailwind';
 import * as RadixTooltip from '@radix-ui/react-tooltip';
 
+export type TooltipProps = {
+    rootProps?: RadixTooltip.TooltipProps;
+    triggerProps?: RadixTooltip.TooltipTriggerProps;
+    contentProps?: RadixTooltip.TooltipContentProps;
+};
+
 export function Tooltip(props: {
     children: React.ReactNode;
     label?: string | React.ReactNode;
-    triggerProps?: RadixTooltip.TooltipTriggerProps;
-    contentProps?: RadixTooltip.TooltipContentProps;
-    rootProps?: RadixTooltip.TooltipProps;
+    triggerProps?: TooltipProps['triggerProps'];
+    contentProps?: TooltipProps['contentProps'];
+    rootProps?: TooltipProps['rootProps'];
     arrow?: boolean;
     className?: string;
 }) {
@@ -30,6 +36,7 @@ export function Tooltip(props: {
             <RadixTooltip.Portal>
                 <RadixTooltip.Content
                     sideOffset={4}
+                    collisionPadding={8}
                     className={tcls(
                         'z-50 max-w-xs animate-scale-in circular-corners:rounded-2xl rounded-corners:rounded-md bg-tint-12 px-2 py-1 text-contrast-tint-12 text-sm',
                         className
diff --git a/packages/gitbook/src/lib/embeddable.ts b/packages/gitbook/src/lib/embeddable.ts
index 339d806ad1..68f6c1e4bc 100644
--- a/packages/gitbook/src/lib/embeddable.ts
+++ b/packages/gitbook/src/lib/embeddable.ts
@@ -55,5 +55,16 @@ export function getEmbeddableLinker(linker: GitBookLinker): GitBookLinker {
                 spaceBasePath: joinPath(override.spaceBasePath, '~gitbook/embed/page'),
             });
         },
+
+        toLinkForContent(rawURL: string): string {
+            const result = linker.toLinkForContent(rawURL);
+            // If the link is not relative or already an embed, return it as is
+            if (result.includes('~gitbook') || !result.startsWith('/')) {
+                return result;
+            }
+
+            // If the link is relative, assume it's a section link and append the embed path
+            return joinPath(result, '~gitbook/embed/page');
+        },
     };
 }
diff --git a/packages/gitbook/src/middleware.ts b/packages/gitbook/src/middleware.ts
index 979276e22a..f5b7f11136 100644
--- a/packages/gitbook/src/middleware.ts
+++ b/packages/gitbook/src/middleware.ts
@@ -612,6 +612,7 @@ function encodePathInSiteContent(rawPathname: string): {
     }
 
     switch (pathname) {
+        case '~gitbook/embed':
         case '~gitbook/embed/assistant':
         case '~gitbook/icon':
             return { pathname };