From 8cbefba34143b27361a92881ae083b36e87fe37f Mon Sep 17 00:00:00 2001
From: akumatus <akumatus@gmail.com>
Date: Mon, 12 May 2025 03:47:05 +0000
Subject: [PATCH] fix(core): cannot abort AI chat immediately after sending a
 message (#12209)

Close [AI-118](https://linear.app/affine-design/issue/AI-118)

<!-- This is an auto-generated comment: release notes by coderabbit.ai -->

## Summary by CodeRabbit

- **New Features**
  - The abort button in the chat input is now available during both "transmitting" and "loading" states, giving users more control to cancel ongoing actions.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->
---
 .../src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/frontend/core/src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts b/packages/frontend/core/src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts
index af2e1dbf2..d4a3a8e0b 100644
--- a/packages/frontend/core/src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts
+++ b/packages/frontend/core/src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts
@@ -426,7 +426,7 @@ export class AIChatInput extends SignalWatcher(WithDisposable(LitElement)) {
             ? html`<span class="chat-input-icon-label">Reason</span>`
             : nothing}
         </div>
-        ${status === 'transmitting'
+        ${status === 'transmitting' || status === 'loading'
           ? html`<div @click=${this._handleAbort} data-testid="chat-panel-stop">
               ${ChatAbortIcon}
             </div>`