From bd30a73db8f39011b8cfb728b135b4aa43fe70f8 Mon Sep 17 00:00:00 2001
From: yoyoyohamapi <8338436+yoyoyohamapi@users.noreply.github.com>
Date: Tue, 1 Apr 2025 12:11:28 +0000
Subject: [PATCH] fix(core): do not show AI avatar during generating (#11363)

> CLOSE BS-2989
---
 .../src/blocksuite/ai/chat-panel/message/assistant.ts  | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/packages/frontend/core/src/blocksuite/ai/chat-panel/message/assistant.ts b/packages/frontend/core/src/blocksuite/ai/chat-panel/message/assistant.ts
index 7164ca4e8c..e903d434bc 100644
--- a/packages/frontend/core/src/blocksuite/ai/chat-panel/message/assistant.ts
+++ b/packages/frontend/core/src/blocksuite/ai/chat-panel/message/assistant.ts
@@ -70,10 +70,6 @@ export class ChatMessageAssistant extends WithDisposable(ShadowlessElement) {
   renderContent() {
     const { host, item, isLast, status, error } = this;
 
-    if (isLast && status === 'loading') {
-      return html`<ai-loading></ai-loading>`;
-    }
-
     const state = isLast
       ? status !== 'loading' && status !== 'transmitting'
         ? 'finished'
@@ -143,6 +139,12 @@ export class ChatMessageAssistant extends WithDisposable(ShadowlessElement) {
   }
 
   protected override render() {
+    const { isLast, status } = this;
+
+    if (isLast && status === 'loading') {
+      return html`<ai-loading></ai-loading>`;
+    }
+
     return html`
       ${this.renderHeader()}
       <div class="item-wrapper">${this.renderContent()}</div>