diff --git a/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/setup-provider.tsx b/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/setup-provider.tsx
index ed56943ed0..123c82553d 100644
--- a/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/setup-provider.tsx
+++ b/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/setup-provider.tsx
@@ -80,6 +80,12 @@ export function setupAIProvider(
           workspaceId,
           docId,
           promptName,
+        }).then(sessionId => {
+          return updateChatSession({
+            sessionId,
+            client,
+            promptName,
+          });
         }),
         promptName,
       });