diff --git a/packages/frontend/core/src/blocksuite/ai/components/ai-chat-composer/ai-chat-composer.ts b/packages/frontend/core/src/blocksuite/ai/components/ai-chat-composer/ai-chat-composer.ts
index da6d56dd7..4fa2b156e 100644
--- a/packages/frontend/core/src/blocksuite/ai/components/ai-chat-composer/ai-chat-composer.ts
+++ b/packages/frontend/core/src/blocksuite/ai/components/ai-chat-composer/ai-chat-composer.ts
@@ -427,9 +427,9 @@ export class AIChatComposer extends SignalWatcher(
       html,
       state: 'finished',
     };
-    await this.addChip(chip, true);
-    await Promise.all(
-      docs.map(docId =>
+    await Promise.all([
+      this.addChip(chip, true),
+      ...docs.map(docId =>
         this.addChip(
           {
             docId,
@@ -437,10 +437,8 @@ export class AIChatComposer extends SignalWatcher(
           },
           true
         )
-      )
-    );
-    await Promise.all(
-      attachments.map(attachment =>
+      ),
+      ...attachments.map(attachment =>
         this.addChip(
           {
             sourceId: attachment.sourceId,
@@ -449,8 +447,8 @@ export class AIChatComposer extends SignalWatcher(
           },
           true
         )
-      )
-    );
+      ),
+    ]);
   };
 
   private readonly removeSelectedContextChip = async () => {