diff --git a/packages/frontend/core/src/blocksuite/presets/ai/entries/code-toolbar/setup-code-toolbar.ts b/packages/frontend/core/src/blocksuite/presets/ai/entries/code-toolbar/setup-code-toolbar.ts
index c37064620d..9ff5375cb8 100644
--- a/packages/frontend/core/src/blocksuite/presets/ai/entries/code-toolbar/setup-code-toolbar.ts
+++ b/packages/frontend/core/src/blocksuite/presets/ai/entries/code-toolbar/setup-code-toolbar.ts
@@ -33,7 +33,7 @@ export function setupCodeToolbarEntry(codeToolbar: AffineCodeToolbarWidget) {
         name: 'Ask AI',
         tooltip: 'Ask AI',
         icon: AIStarIcon,
-        showWhen: () => true,
+        showWhen: CodeBlockComponent => !CodeBlockComponent.doc.readonly,
         render(codeBlock: CodeBlockComponent, onClick?: () => void) {
           return html`<ask-ai-button
             class="code-toolbar-button ask-ai"
diff --git a/packages/frontend/core/src/blocksuite/presets/ai/entries/image-toolbar/setup-image-toolbar.ts b/packages/frontend/core/src/blocksuite/presets/ai/entries/image-toolbar/setup-image-toolbar.ts
index b4f9e4a9ba..33a951516b 100644
--- a/packages/frontend/core/src/blocksuite/presets/ai/entries/image-toolbar/setup-image-toolbar.ts
+++ b/packages/frontend/core/src/blocksuite/presets/ai/entries/image-toolbar/setup-image-toolbar.ts
@@ -44,7 +44,7 @@ export function setupImageToolbarEntry(imageToolbar: AffineImageToolbarWidget) {
             }}
           ></ask-ai-button>`;
         },
-        showWhen: () => true,
+        showWhen: imageBlockComponent => !imageBlockComponent.doc.readonly,
       },
     ],
     0