diff --git a/packages/presets/src/ai/_common/icons.ts b/packages/presets/src/ai/_common/icons.ts
index a22b5e609a7d..ba93b90e00b5 100644
--- a/packages/presets/src/ai/_common/icons.ts
+++ b/packages/presets/src/ai/_common/icons.ts
@@ -891,6 +891,28 @@ export const CopyIcon = html`<svg
   </defs>
 </svg> `;
 
+export const RetryIcon = html`<svg
+  width="20"
+  height="20"
+  viewBox="0 0 20 20"
+  fill="none"
+  xmlns="http://www.w3.org/2000/svg"
+>
+  <g clip-path="url(#clip0_4361_69948)">
+    <path
+      fill-rule="evenodd"
+      clip-rule="evenodd"
+      d="M2.70837 9.99998C2.70837 5.9729 5.97296 2.70831 10 2.70831C11.8672 2.70831 13.5717 3.41091 14.8613 4.56514L14.8633 4.56693L16.0417 5.63077V3.33331C16.0417 2.98814 16.3215 2.70831 16.6667 2.70831C17.0119 2.70831 17.2917 2.98814 17.2917 3.33331V7.03702C17.2917 7.38219 17.0119 7.66202 16.6667 7.66202H12.963C12.6178 7.66202 12.338 7.38219 12.338 7.03702C12.338 6.69184 12.6178 6.41202 12.963 6.41202H15.0417L14.0277 5.49656C14.0273 5.49626 14.027 5.49597 14.0267 5.49567C12.9575 4.53915 11.5473 3.95831 10 3.95831C6.66332 3.95831 3.95837 6.66326 3.95837 9.99998C3.95837 13.3367 6.66332 16.0416 10 16.0416C12.8745 16.0416 15.2815 14.0336 15.892 11.3432C15.9684 11.0065 16.3032 10.7956 16.6398 10.872C16.9764 10.9483 17.1874 11.2831 17.111 11.6198C16.3742 14.8671 13.471 17.2916 10 17.2916C5.97296 17.2916 2.70837 14.0271 2.70837 9.99998Z"
+      fill="#77757D"
+    />
+  </g>
+  <defs>
+    <clipPath id="clip0_4361_69948">
+      <rect width="20" height="20" fill="white" />
+    </clipPath>
+  </defs>
+</svg> `;
+
 export const MoreIcon = html`<svg
   width="20"
   height="20"
diff --git a/packages/presets/src/ai/actions/types.ts b/packages/presets/src/ai/actions/types.ts
index 8907a6d10c3d..eb17d57e2fe9 100644
--- a/packages/presets/src/ai/actions/types.ts
+++ b/packages/presets/src/ai/actions/types.ts
@@ -26,10 +26,11 @@ export const textTones = [
 declare global {
   namespace BlockSuitePresets {
     interface AITextActionOptions {
-      input: string;
+      input?: string;
       stream?: boolean;
       attachments?: (string | File | Blob)[]; // blob could only be strings for the moments (url or data urls)
       signal?: AbortSignal;
+      retry?: boolean;
 
       // action's context
       docId: string;
diff --git a/packages/presets/src/ai/chat-panel/actions/copy-more.ts b/packages/presets/src/ai/chat-panel/actions/copy-more.ts
index bfccf2ee18ca..1de265ca2215 100644
--- a/packages/presets/src/ai/chat-panel/actions/copy-more.ts
+++ b/packages/presets/src/ai/chat-panel/actions/copy-more.ts
@@ -4,14 +4,16 @@ import type {
   TextSelection,
 } from '@blocksuite/block-std';
 import { WithDisposable } from '@blocksuite/block-std';
-import { createButtonPopper, Tooltip } from '@blocksuite/blocks';
+import { type AIError, createButtonPopper, Tooltip } from '@blocksuite/blocks';
 import { noop } from '@blocksuite/global/utils';
 import { css, html, LitElement, nothing, type PropertyValues } from 'lit';
 import { customElement, property, query, state } from 'lit/decorators.js';
 import { repeat } from 'lit/directives/repeat.js';
 
-import { CopyIcon, MoreIcon } from '../../_common/icons.js';
+import { CopyIcon, MoreIcon, RetryIcon } from '../../_common/icons.js';
+import { AIProvider } from '../../provider.js';
 import { copyText } from '../../utils/editor-actions.js';
+import type { ChatItem, ChatMessage, ChatStatus } from '../index.js';
 import { PageEditorActions } from './actions-handle.js';
 
 noop(Tooltip);
@@ -83,6 +85,24 @@ export class ChatCopyMore extends WithDisposable(LitElement) {
   @property({ attribute: false })
   curBlockSelections?: BlockSelection[];
 
+  @property({ attribute: false })
+  items!: ChatItem[];
+
+  @property({ attribute: false })
+  updateItems!: (items: ChatItem[]) => void;
+
+  @property({ attribute: false })
+  updateStatus!: (status: ChatStatus) => void;
+
+  @property({ attribute: false })
+  updateError!: (error: AIError | null) => void;
+
+  @property({ attribute: false })
+  abortController!: AbortController | null;
+
+  @property({ attribute: false })
+  updateAbortController!: (abortController: AbortController | null) => void;
+
   @state()
   private _showMoreMenu = false;
 
@@ -111,6 +131,52 @@ export class ChatCopyMore extends WithDisposable(LitElement) {
     }
   }
 
+  private async _retry() {
+    const { doc } = this.host;
+    try {
+      const abortController = new AbortController();
+      const items = [...this.items];
+      const last = items[items.length - 1];
+      if ('content' in last) {
+        last.content = '';
+        last.createdAt = new Date().toISOString();
+      }
+
+      this.updateItems(items);
+      this.updateStatus('loading');
+      this.updateError(null);
+
+      const stream = AIProvider.actions.chat?.({
+        retry: true,
+        docId: doc.id,
+        workspaceId: doc.collection.id,
+        host: this.host,
+        stream: true,
+        signal: abortController.signal,
+        where: 'chat-panel',
+        control: 'chat-send',
+      });
+
+      if (stream) {
+        this.updateAbortController(abortController);
+        for await (const text of stream) {
+          this.updateStatus('transmitting');
+          const items = [...this.items];
+          const last = items[items.length - 1] as ChatMessage;
+          last.content += text;
+          this.updateItems(items);
+        }
+
+        this.updateStatus('success');
+      }
+    } catch (error) {
+      this.updateStatus('error');
+      this.updateError(error as AIError);
+    } finally {
+      this.updateAbortController(null);
+    }
+  }
+
   override render() {
     const { host, content, isLast } = this;
     return html`<style>
@@ -123,6 +189,12 @@ export class ChatCopyMore extends WithDisposable(LitElement) {
           ${CopyIcon}
           <affine-tooltip>Copy</affine-tooltip>
         </div>
+        ${isLast
+          ? html`<div @click=${() => this._retry()}>
+              ${RetryIcon}
+              <affine-tooltip>Retry</affine-tooltip>
+            </div>`
+          : nothing}
         ${isLast
           ? nothing
           : html`<div class="more-button" @click=${this._toggle}>
diff --git a/packages/presets/src/ai/chat-panel/chat-panel-input.ts b/packages/presets/src/ai/chat-panel/chat-panel-input.ts
index a403913369b2..dcd32407a63b 100644
--- a/packages/presets/src/ai/chat-panel/chat-panel-input.ts
+++ b/packages/presets/src/ai/chat-panel/chat-panel-input.ts
@@ -160,8 +160,11 @@ export class ChatPanelInput extends WithDisposable(LitElement) {
   @state()
   focused = false;
 
-  @state()
-  abortController?: AbortController;
+  @property({ attribute: false })
+  abortController!: AbortController | null;
+
+  @property({ attribute: false })
+  updateAbortController!: (abortController: AbortController | null) => void;
 
   send = async () => {
     if (this.status === 'loading' || this.status === 'transmitting') return;
@@ -205,7 +208,7 @@ export class ChatPanelInput extends WithDisposable(LitElement) {
       });
 
       if (stream) {
-        this.abortController = abortController;
+        this.updateAbortController(abortController);
 
         for await (const text of stream) {
           this.updateStatus('transmitting');
@@ -221,7 +224,7 @@ export class ChatPanelInput extends WithDisposable(LitElement) {
       this.updateStatus('error');
       this.updateError(error as AIError);
     } finally {
-      this.abortController = undefined;
+      this.updateAbortController(null);
     }
   };
 
diff --git a/packages/presets/src/ai/chat-panel/chat-panel-messages.ts b/packages/presets/src/ai/chat-panel/chat-panel-messages.ts
index 5ad0bdafc197..377a5979a3e3 100644
--- a/packages/presets/src/ai/chat-panel/chat-panel-messages.ts
+++ b/packages/presets/src/ai/chat-panel/chat-panel-messages.ts
@@ -150,6 +150,21 @@ export class ChatPanelMessages extends WithDisposable(ShadowlessElement) {
   @property({ attribute: false })
   isLoading!: boolean;
 
+  @property({ attribute: false })
+  updateItems!: (items: ChatItem[]) => void;
+
+  @property({ attribute: false })
+  updateStatus!: (status: ChatStatus) => void;
+
+  @property({ attribute: false })
+  updateError!: (error: AIError | null) => void;
+
+  @property({ attribute: false })
+  abortController!: AbortController | null;
+
+  @property({ attribute: false })
+  updateAbortController!: (abortController: AbortController | null) => void;
+
   @query('.chat-panel-messages')
   messagesContainer!: HTMLDivElement;
 
@@ -359,6 +374,12 @@ export class ChatPanelMessages extends WithDisposable(ShadowlessElement) {
         .isLast=${isLast}
         .curTextSelection=${this._currentTextSelection}
         .curBlockSelections=${this._currentBlockSelections}
+        .items=${this.items}
+        .abortController=${this.abortController}
+        .updateItems=${this.updateItems}
+        .updateStatus=${this.updateStatus}
+        .updateError=${this.updateError}
+        .updateAbortController=${this.updateAbortController}
       ></chat-copy-more>
       ${isLast
         ? html`<div class="actions-container">
diff --git a/packages/presets/src/ai/chat-panel/index.ts b/packages/presets/src/ai/chat-panel/index.ts
index a207d8040371..bab362814968 100644
--- a/packages/presets/src/ai/chat-panel/index.ts
+++ b/packages/presets/src/ai/chat-panel/index.ts
@@ -112,6 +112,9 @@ export class ChatPanel extends WithDisposable(ShadowlessElement) {
   @state()
   isLoading = false;
 
+  @state()
+  abortController: AbortController | null = null;
+
   private _chatMessages: Ref<ChatPanelMessages> =
     createRef<ChatPanelMessages>();
 
@@ -192,6 +195,10 @@ export class ChatPanel extends WithDisposable(ShadowlessElement) {
     this.error = error;
   };
 
+  updateAbortController = (abortController: AbortController | null) => {
+    this.abortController = abortController;
+  };
+
   scrollToDown() {
     requestAnimationFrame(() => this._chatMessages.value?.scrollToDown());
   }
@@ -206,14 +213,21 @@ export class ChatPanel extends WithDisposable(ShadowlessElement) {
         .status=${this.status}
         .error=${this.error}
         .isLoading=${this.isLoading}
+        .updateItems=${this.updateItems}
+        .updateStatus=${this.updateStatus}
+        .updateError=${this.updateError}
+        .abortController=${this.abortController}
+        .updateAbortController=${this.updateAbortController}
       ></chat-panel-messages>
       <chat-panel-input
         .host=${this.host}
         .items=${this.items}
         .updateItems=${this.updateItems}
+        .updateStatus=${this.updateStatus}
+        .abortController=${this.abortController}
+        .updateAbortController=${this.updateAbortController}
         .addToItems=${this.addToItems}
         .status=${this.status}
-        .updateStatus=${this.updateStatus}
         .error=${this.error}
         .updateError=${this.updateError}
       ></chat-panel-input>
diff --git a/packages/presets/src/ai/provider.ts b/packages/presets/src/ai/provider.ts
index 34857581e7a6..6c8bca7a147a 100644
--- a/packages/presets/src/ai/provider.ts
+++ b/packages/presets/src/ai/provider.ts
@@ -59,6 +59,8 @@ export class AIProvider {
     // add more if needed
   };
 
+  static LAST_ACTION_SESSIONID = '';
+
   static MAX_LOCAL_HISTORY = 10;
   // track the history of triggered actions (in memory only)
   private readonly actionHistory: {