fix(core): comment empty style issue (#13208)

fix BS-3618

#### PR Dependency Tree


* **PR #13208** 👈

This tree was auto-generated by
[Charcoal](https://github.com/danerwilliams/charcoal)

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

## Summary by CodeRabbit

* **Style**
* Improved the appearance of the empty state in the comment sidebar by
centering the text and adjusting line spacing for better readability.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->

.github/actions/setup-version/action.yml

@@ -4,9 +4,15 @@ inputs:
   app-version:
     description: 'App Version'
     required: true
+  ios-app-version:
+    description: 'iOS App Store Version (Optional, use App version if empty)'
+    required: false
+    type: string
 runs:
   using: 'composite'
   steps:
     - name: 'Write Version'
       shell: bash
+      env:
+        IOS_APP_VERSION: ${{ inputs.ios-app-version }}
       run: ./scripts/set-version.sh ${{ inputs.app-version }}

.github/workflows/release-mobile.yml

@@ -12,6 +12,9 @@ on:
       build-type:
         type: string
         required: true
+      ios-app-version:
+        type: string
+        required: false
 
 env:
   BUILD_TYPE: ${{ inputs.build-type }}
@@ -78,7 +81,7 @@ jobs:
           path: packages/frontend/apps/android/dist
 
   ios:
-    runs-on: ${{ github.ref_name == 'canary' && 'macos-latest' || 'blaze/macos-14' }}
+    runs-on: 'macos-15'
     needs:
       - build-ios-web
     steps:
@@ -87,6 +90,7 @@ jobs:
         uses: ./.github/actions/setup-version
         with:
           app-version: ${{ inputs.app-version }}
+          ios-app-version: ${{ inputs.ios-app-version }}
       - name: 'Update Code Sign Identity'
         shell: bash
         run: ./packages/frontend/apps/ios/update_code_sign_identity.sh
@@ -106,7 +110,7 @@ jobs:
           enableScripts: false
       - uses: maxim-lobanov/setup-xcode@v1
         with:
-          xcode-version: 16.2
+          xcode-version: 16.4
       - name: Install Swiftformat
         run: brew install swiftformat
       - name: Cap sync

.github/workflows/release.yml

@@ -21,6 +21,10 @@ on:
         required: true
         type: boolean
         default: false
+      ios-app-version:
+        description: 'iOS App Store Version (Optional, use tag version if empty)'
+        required: false
+        type: string
 
 permissions:
   contents: write
@@ -117,3 +121,4 @@ jobs:
       build-type: ${{ needs.prepare.outputs.BUILD_TYPE }}
       app-version: ${{ needs.prepare.outputs.APP_VERSION }}
       git-short-hash: ${{ needs.prepare.outputs.GIT_SHORT_HASH }}
+      ios-app-version: ${{ inputs.ios-app-version }}

blocksuite/affine/all/package.json

@@ -266,6 +266,7 @@
     "./components/toggle-button": "./src/components/toggle-button.ts",
     "./components/toggle-switch": "./src/components/toggle-switch.ts",
     "./components/toolbar": "./src/components/toolbar.ts",
+    "./components/tooltip": "./src/components/tooltip.ts",
     "./components/view-dropdown-menu": "./src/components/view-dropdown-menu.ts",
     "./components/tooltip-content-with-shortcut": "./src/components/tooltip-content-with-shortcut.ts",
     "./components/resource": "./src/components/resource.ts",

blocksuite/affine/all/src/components/tooltip.ts

@@ -0,0 +1 @@
+export * from '@blocksuite/affine-components/tooltip';

blocksuite/affine/components/package.json

@@ -73,7 +73,8 @@
     "./edgeless-line-styles-panel": "./src/edgeless-line-styles-panel/index.ts",
     "./edgeless-shape-color-picker": "./src/edgeless-shape-color-picker/index.ts",
     "./open-doc-dropdown-menu": "./src/open-doc-dropdown-menu/index.ts",
-    "./slider": "./src/slider/index.ts"
+    "./slider": "./src/slider/index.ts",
+    "./tooltip": "./src/tooltip/index.ts"
   },
   "files": [
     "src",

blocksuite/affine/components/src/context-menu/sub-menu.ts

@@ -85,6 +85,8 @@ export class MenuSubMenu extends MenuFocusable {
         .catch(err => console.error(err));
     });
     this.menu.openSubMenu(menu);
+    // in case that the menu is not closed, but the component is removed,
+    this.disposables.add(unsub);
   }
 
   protected override render(): unknown {

blocksuite/affine/components/src/icons/loading.ts

@@ -18,6 +18,7 @@ export const LoadingIcon = ({
     viewBox="0 0 24 24"
     xmlns="http://www.w3.org/2000/svg"
     fill="none"
+    style="fill: none;"
   >
     <style>
       .spinner {

blocksuite/affine/components/src/toolbar/index.ts

@@ -1,3 +1,4 @@
+import { effects as tooltipEffects } from '../tooltip/effect.js';
 import { EditorIconButton } from './icon-button.js';
 import {
   EditorMenuAction,
@@ -6,7 +7,6 @@ import {
 } from './menu-button.js';
 import { EditorToolbarSeparator } from './separator.js';
 import { EditorToolbar } from './toolbar.js';
-import { Tooltip } from './tooltip.js';
 
 export { EditorChevronDown } from './chevron-down.js';
 export { ToolbarMoreMenuConfigExtension } from './config.js';
@@ -20,7 +20,6 @@ export { MenuContext } from './menu-context.js';
 export { EditorToolbarSeparator } from './separator.js';
 export { darkToolbarStyles, lightToolbarStyles } from './styles.js';
 export { EditorToolbar } from './toolbar.js';
-export { Tooltip } from './tooltip.js';
 export type {
   AdvancedMenuItem,
   FatMenuItems,
@@ -38,11 +37,12 @@ export {
 } from './utils.js';
 
 export function effects() {
+  tooltipEffects();
+
   customElements.define('editor-toolbar-separator', EditorToolbarSeparator);
   customElements.define('editor-toolbar', EditorToolbar);
   customElements.define('editor-icon-button', EditorIconButton);
   customElements.define('editor-menu-button', EditorMenuButton);
   customElements.define('editor-menu-content', EditorMenuContent);
   customElements.define('editor-menu-action', EditorMenuAction);
-  customElements.define('affine-tooltip', Tooltip);
 }

blocksuite/affine/components/src/tooltip/effect.ts

@@ -0,0 +1,7 @@
+import { Tooltip } from './tooltip.js';
+
+export function effects() {
+  if (!customElements.get('affine-tooltip')) {
+    customElements.define('affine-tooltip', Tooltip);
+  }
+}

blocksuite/affine/components/src/tooltip/index.ts

@@ -0,0 +1,2 @@
+export { effects } from './effect.js';
+export { Tooltip } from './tooltip.js';

blocksuite/affine/gfx/template/src/toolbar/template-tool-button.ts

@@ -116,6 +116,7 @@ export class EdgelessTemplateButton extends EdgelessToolbarToolMixin(
   `;
 
   private _cleanup: (() => void) | null = null;
+  private _autoUpdateCleanup: (() => void) | null = null;
 
   private _prevTool: ToolOptionWithType | null = null;
 
@@ -128,6 +129,11 @@ export class EdgelessTemplateButton extends EdgelessToolbarToolMixin(
     return [TemplateCard1[theme], TemplateCard2[theme], TemplateCard3[theme]];
   }
 
+  override connectedCallback() {
+    super.connectedCallback();
+    this.disposables.add(() => this._autoUpdateCleanup?.());
+  }
+
   private _closePanel() {
     if (this._openedPanel) {
       this._openedPanel.remove();
@@ -175,8 +181,8 @@ export class EdgelessTemplateButton extends EdgelessToolbarToolMixin(
 
     requestAnimationFrame(() => {
       const arrowEl = panel.renderRoot.querySelector('.arrow') as HTMLElement;
-
-      autoUpdate(this, panel, () => {
+      this._autoUpdateCleanup?.();
+      this._autoUpdateCleanup = autoUpdate(this, panel, () => {
         computePosition(this, panel, {
           placement: 'top',
           middleware: [offset(20), arrow({ element: arrowEl }), shift()],

blocksuite/affine/inlines/comment/src/inline-comment.ts

@@ -22,8 +22,11 @@ import { isEqual } from 'lodash-es';
 })
 export class InlineComment extends WithDisposable(ShadowlessElement) {
   static override styles = css`
+    inline-comment {
+      display: inline;
+    }
+
     inline-comment.unresolved {
-      display: inline-block;
       background-color: ${unsafeCSSVarV2('block/comment/highlightDefault')};
       border-bottom: 2px solid
         ${unsafeCSSVarV2('block/comment/highlightUnderline')};

blocksuite/framework/std/src/inline/utils/attribute-renderer.ts

@@ -30,9 +30,9 @@ function inlineTextStyles(
   }
 
   return styleMap({
-    'font-weight': props.bold ? 'bold' : 'normal',
-    'font-style': props.italic ? 'italic' : 'normal',
-    'text-decoration': textDecorations.length > 0 ? textDecorations : 'none',
+    'font-weight': props.bold ? 'bold' : 'inherit',
+    'font-style': props.italic ? 'italic' : 'inherit',
+    'text-decoration': textDecorations.length > 0 ? textDecorations : 'inherit',
     ...inlineCodeStyle,
   });
 }

packages/backend/server/src/__tests__/__snapshots__/copilot.spec.ts.md

@@ -372,3 +372,66 @@ Generated by [AVA](https://avajs.dev).
       [assistant]: Quantum computing uses quantum mechanics principles.`,
       promptName: 'Summary as title',
     }
+
+## should handle copilot cron jobs correctly
+
+> daily job scheduling calls
+
+    [
+      {
+        args: [
+          'copilot.session.cleanupEmptySessions',
+          {},
+          {
+            jobId: 'daily-copilot-cleanup-empty-sessions',
+          },
+        ],
+      },
+      {
+        args: [
+          'copilot.session.generateMissingTitles',
+          {},
+          {
+            jobId: 'daily-copilot-generate-missing-titles',
+          },
+        ],
+      },
+    ]
+
+> cleanup empty sessions calls
+
+    [
+      {
+        args: [
+          'Date',
+        ],
+      },
+    ]
+
+> title generation calls
+
+    {
+      jobCalls: [
+        {
+          args: [
+            'copilot.session.generateTitle',
+            {
+              sessionId: 'session1',
+            },
+          ],
+        },
+        {
+          args: [
+            'copilot.session.generateTitle',
+            {
+              sessionId: 'session2',
+            },
+          ],
+        },
+      ],
+      modelCalls: [
+        {
+          args: [],
+        },
+      ],
+    }

packages/backend/server/src/__tests__/copilot-provider.spec.ts

@@ -207,6 +207,7 @@ const retry = async (
       try {
         await callback(t);
       } catch (e) {
+        console.error(`Error during ${action}:`, e);
         t.log(`Error during ${action}:`, e);
         throw e;
       }
@@ -350,10 +351,10 @@ The term **“CRDT”** was first introduced by Marc Shapiro, Nuno Preguiça, Ca
         params: {
           files: [
             {
-              blobId: 'euclidean_distance',
-              fileName: 'euclidean_distance.rs',
-              fileType: 'text/rust',
-              fileContent: TestAssets.Code,
+              blobId: 'todo_md',
+              fileName: 'todo.md',
+              fileType: 'text/markdown',
+              fileContent: TestAssets.TODO,
             },
           ],
         },
@@ -475,6 +476,7 @@ The term **“CRDT”** was first introduced by Marc Shapiro, Nuno Preguiça, Ca
         },
       },
     ],
+    config: { model: 'gemini-2.5-pro' },
     verifier: (t: ExecutionContext<Tester>, result: string) => {
       t.notThrows(() => {
         TranscriptionResponseSchema.parse(JSON.parse(result));
@@ -483,6 +485,34 @@ The term **“CRDT”** was first introduced by Marc Shapiro, Nuno Preguiça, Ca
     type: 'structured' as const,
     prefer: CopilotProviderType.Gemini,
   },
+  {
+    promptName: ['Conversation Summary'],
+    messages: [
+      {
+        role: 'user' as const,
+        content: '',
+        params: {
+          messages: [
+            { role: 'user', content: 'what is single source of truth?' },
+            { role: 'assistant', content: TestAssets.SSOT },
+          ],
+          focus: 'technical decisions',
+          length: 'comprehensive',
+        },
+      },
+    ],
+    verifier: (t: ExecutionContext<Tester>, result: string) => {
+      assertNotWrappedInCodeBlock(t, result);
+      const cleared = result.toLowerCase();
+      t.assert(
+        cleared.includes('single source of truth') ||
+          /single.*source/.test(cleared) ||
+          cleared.includes('ssot'),
+        'should include original keyword'
+      );
+    },
+    type: 'text' as const,
+  },
   {
     promptName: [
       'Summary',
@@ -668,11 +698,12 @@ for (const {
         t.truthy(provider, 'should have provider');
         await retry(`action: ${promptName}`, t, async t => {
           const finalConfig = Object.assign({}, prompt.config, config);
+          const modelId = finalConfig.model || prompt.model;
 
           switch (type) {
             case 'text': {
               const result = await provider.text(
-                { modelId: prompt.model },
+                { modelId },
                 [
                   ...prompt.finish(
                     messages.reduce(
@@ -691,7 +722,7 @@ for (const {
             }
             case 'structured': {
               const result = await provider.structure(
-                { modelId: prompt.model },
+                { modelId },
                 [
                   ...prompt.finish(
                     messages.reduce(
@@ -710,7 +741,7 @@ for (const {
             case 'object': {
               const streamObjects: StreamObject[] = [];
               for await (const chunk of provider.streamObject(
-                { modelId: prompt.model },
+                { modelId },
                 [
                   ...prompt.finish(
                     messages.reduce(
@@ -742,7 +773,7 @@ for (const {
                 });
               }
               const stream = provider.streamImages(
-                { modelId: prompt.model },
+                { modelId },
                 [
                   ...prompt.finish(
                     finalMessage.reduce(

packages/backend/server/src/__tests__/copilot.e2e.ts

@@ -290,6 +290,7 @@ test('should fork session correctly', async t => {
 
   const assertForkSession = async (
     workspaceId: string,
+    docId: string,
     sessionId: string,
     lastMessageId: string | undefined,
     error: string,
@@ -300,13 +301,7 @@ test('should fork session correctly', async t => {
     }
   ) =>
     await asserter(
-      forkCopilotSession(
-        app,
-        workspaceId,
-        randomUUID(),
-        sessionId,
-        lastMessageId
-      )
+      forkCopilotSession(app, workspaceId, docId, sessionId, lastMessageId)
     );
 
   // prepare session
@@ -330,6 +325,7 @@ test('should fork session correctly', async t => {
     // should be able to fork session
     forkedSessionId = await assertForkSession(
       id,
+      docId,
       sessionId,
       latestMessageId!,
       'should be able to fork session with cloud workspace that user can access'
@@ -340,6 +336,7 @@ test('should fork session correctly', async t => {
   {
     forkedSessionId = await assertForkSession(
       id,
+      docId,
       sessionId,
       undefined,
       'should be able to fork session without latestMessageId'
@@ -348,18 +345,25 @@ test('should fork session correctly', async t => {
 
   // should not be able to fork session with wrong latestMessageId
   {
-    await assertForkSession(id, sessionId, 'wrong-message-id', '', async x => {
-      await t.throwsAsync(
-        x,
-        { instanceOf: Error },
-        'should not able to fork session with wrong latestMessageId'
-      );
-    });
+    await assertForkSession(
+      id,
+      docId,
+      sessionId,
+      'wrong-message-id',
+      '',
+      async x => {
+        await t.throwsAsync(
+          x,
+          { instanceOf: Error },
+          'should not able to fork session with wrong latestMessageId'
+        );
+      }
+    );
   }
 
   {
     const u2 = await app.signupV1();
-    await assertForkSession(id, sessionId, randomUUID(), '', async x => {
+    await assertForkSession(id, docId, sessionId, randomUUID(), '', async x => {
       await t.throwsAsync(
         x,
         { instanceOf: Error },
@@ -371,7 +375,7 @@ test('should fork session correctly', async t => {
     const inviteId = await inviteUser(app, id, u2.email);
     await app.switchUser(u2);
     await acceptInviteById(app, id, inviteId, false);
-    await assertForkSession(id, sessionId, randomUUID(), '', async x => {
+    await assertForkSession(id, docId, sessionId, randomUUID(), '', async x => {
       await t.throwsAsync(
         x,
         { instanceOf: Error },
@@ -389,6 +393,7 @@ test('should fork session correctly', async t => {
     await app.switchUser(u2);
     await assertForkSession(
       id,
+      docId,
       forkedSessionId,
       latestMessageId!,
       'should able to fork a forked session created by other user'

packages/backend/server/src/__tests__/copilot.spec.ts

@@ -18,6 +18,7 @@ import {
 } from '../models';
 import { CopilotModule } from '../plugins/copilot';
 import { CopilotContextService } from '../plugins/copilot/context';
+import { CopilotCronJobs } from '../plugins/copilot/cron';
 import {
   CopilotEmbeddingJob,
   MockEmbeddingClient,
@@ -77,6 +78,7 @@ type Context = {
   jobs: CopilotEmbeddingJob;
   storage: CopilotStorage;
   workflow: CopilotWorkflowService;
+  cronJobs: CopilotCronJobs;
   executors: {
     image: CopilotChatImageExecutor;
     text: CopilotChatTextExecutor;
@@ -137,6 +139,7 @@ test.before(async t => {
   const jobs = module.get(CopilotEmbeddingJob);
   const transcript = module.get(CopilotTranscriptionService);
   const workspaceEmbedding = module.get(CopilotWorkspaceService);
+  const cronJobs = module.get(CopilotCronJobs);
 
   t.context.module = module;
   t.context.auth = auth;
@@ -153,6 +156,7 @@ test.before(async t => {
   t.context.jobs = jobs;
   t.context.transcript = transcript;
   t.context.workspaceEmbedding = workspaceEmbedding;
+  t.context.cronJobs = cronJobs;
 
   t.context.executors = {
     image: module.get(CopilotChatImageExecutor),
@@ -1931,3 +1935,71 @@ test('should handle generateSessionTitle correctly under various conditions', as
     );
   }
 });
+
+test('should handle copilot cron jobs correctly', async t => {
+  const { cronJobs, copilotSession } = t.context;
+
+  // mock calls
+  const mockCleanupResult = { removed: 2, cleaned: 3 };
+  const mockSessions = [
+    { id: 'session1', _count: { messages: 1 } },
+    { id: 'session2', _count: { messages: 2 } },
+  ];
+  const cleanupStub = Sinon.stub(
+    copilotSession,
+    'cleanupEmptySessions'
+  ).resolves(mockCleanupResult);
+  const toBeGenerateStub = Sinon.stub(
+    copilotSession,
+    'toBeGenerateTitle'
+  ).resolves(mockSessions);
+  const jobAddStub = Sinon.stub(cronJobs['jobs'], 'add').resolves();
+
+  // daily cleanup job scheduling
+  {
+    await cronJobs.dailyCleanupJob();
+    t.snapshot(
+      jobAddStub.getCalls().map(call => ({
+        args: call.args,
+      })),
+      'daily job scheduling calls'
+    );
+
+    jobAddStub.reset();
+    cleanupStub.reset();
+    toBeGenerateStub.reset();
+  }
+
+  // cleanup empty sessions
+  {
+    // mock
+    cleanupStub.resolves(mockCleanupResult);
+    toBeGenerateStub.resolves(mockSessions);
+
+    await cronJobs.cleanupEmptySessions();
+    t.snapshot(
+      cleanupStub.getCalls().map(call => ({
+        args: call.args.map(arg => (arg instanceof Date ? 'Date' : arg)), // Replace Date with string for stable snapshot
+      })),
+      'cleanup empty sessions calls'
+    );
+  }
+
+  // generate missing titles
+  await cronJobs.generateMissingTitles();
+  t.snapshot(
+    {
+      modelCalls: toBeGenerateStub.getCalls().map(call => ({
+        args: call.args,
+      })),
+      jobCalls: jobAddStub.getCalls().map(call => ({
+        args: call.args,
+      })),
+    },
+    'title generation calls'
+  );
+
+  cleanupStub.restore();
+  toBeGenerateStub.restore();
+  jobAddStub.restore();
+});

packages/backend/server/src/__tests__/e2e/doc/resolver.spec.ts

@@ -99,3 +99,56 @@ e2e(
     t.is(result2.workspace.doc.public, true);
   }
 );
+
+e2e('should get doc with title and summary', async t => {
+  const owner = await app.signup();
+
+  const workspace = await app.create(Mockers.Workspace, {
+    owner: { id: owner.id },
+  });
+
+  const docSnapshot = await app.create(Mockers.DocSnapshot, {
+    workspaceId: workspace.id,
+    user: owner,
+  });
+  const doc = await app.create(Mockers.DocMeta, {
+    workspaceId: workspace.id,
+    docId: docSnapshot.id,
+    title: 'doc1',
+    summary: 'summary1',
+  });
+
+  const result = await app.gql({
+    query: getWorkspacePageByIdQuery,
+    variables: { workspaceId: workspace.id, pageId: doc.docId },
+  });
+
+  t.is(result.workspace.doc.title, doc.title);
+  t.is(result.workspace.doc.summary, doc.summary);
+});
+
+e2e('should get doc with title and null summary', async t => {
+  const owner = await app.signup();
+
+  const workspace = await app.create(Mockers.Workspace, {
+    owner: { id: owner.id },
+  });
+
+  const docSnapshot = await app.create(Mockers.DocSnapshot, {
+    workspaceId: workspace.id,
+    user: owner,
+  });
+  const doc = await app.create(Mockers.DocMeta, {
+    workspaceId: workspace.id,
+    docId: docSnapshot.id,
+    title: 'doc1',
+  });
+
+  const result = await app.gql({
+    query: getWorkspacePageByIdQuery,
+    variables: { workspaceId: workspace.id, pageId: doc.docId },
+  });
+
+  t.is(result.workspace.doc.title, doc.title);
+  t.is(result.workspace.doc.summary, null);
+});

packages/backend/server/src/__tests__/mocks/copilot.mock.ts

@@ -111,6 +111,19 @@ export class MockCopilotProvider extends OpenAIProvider {
         },
       ],
     },
+    {
+      id: 'gemini-2.5-pro',
+      capabilities: [
+        {
+          input: [ModelInputType.Text, ModelInputType.Image],
+          output: [
+            ModelOutputType.Text,
+            ModelOutputType.Object,
+            ModelOutputType.Structured,
+          ],
+        },
+      ],
+    },
   ];
 
   override async text(

packages/backend/server/src/__tests__/models/__snapshots__/copilot-session.spec.ts.md

@@ -565,3 +565,65 @@ Generated by [AVA](https://avajs.dev).
         workspaceSessionExists: true,
       },
     }
+
+## should cleanup empty sessions correctly
+
+> cleanup empty sessions results
+
+    {
+      cleanupResult: {
+        cleaned: 0,
+        removed: 0,
+      },
+      remainingSessions: [
+        {
+          deleted: false,
+          pinned: false,
+          type: 'zeroCost',
+        },
+        {
+          deleted: false,
+          pinned: false,
+          type: 'zeroCost',
+        },
+        {
+          deleted: false,
+          pinned: false,
+          type: 'noMessages',
+        },
+        {
+          deleted: false,
+          pinned: false,
+          type: 'noMessages',
+        },
+        {
+          deleted: false,
+          pinned: false,
+          type: 'recent',
+        },
+        {
+          deleted: false,
+          pinned: false,
+          type: 'withMessages',
+        },
+      ],
+    }
+
+## should get sessions for title generation correctly
+
+> sessions for title generation results
+
+    {
+      onlyValidSessionsReturned: true,
+      sessions: [
+        {
+          assistantMessageCount: 1,
+          isValid: true,
+        },
+        {
+          assistantMessageCount: 2,
+          isValid: true,
+        },
+      ],
+      total: 2,
+    }

packages/backend/server/src/__tests__/models/copilot-session.spec.ts

@@ -917,3 +917,178 @@ test('should handle fork and session attachment operations', async t => {
     'attach and detach operation results'
   );
 });
+
+test('should cleanup empty sessions correctly', async t => {
+  const { copilotSession, db } = t.context;
+  await createTestPrompts(copilotSession, db);
+
+  const oneDayAgo = new Date(Date.now() - 24 * 60 * 60 * 1000);
+  const twoHoursAgo = new Date(Date.now() - 2 * 60 * 60 * 1000);
+
+  // should be deleted
+  const neverUsedSessionIds: string[] = [randomUUID(), randomUUID()];
+  await Promise.all(
+    neverUsedSessionIds.map(async id => {
+      await createTestSession(t, { sessionId: id });
+      await db.aiSession.update({
+        where: { id },
+        data: { messageCost: 0, updatedAt: oneDayAgo },
+      });
+    })
+  );
+
+  // should be marked as deleted
+  const emptySessionIds: string[] = [randomUUID(), randomUUID()];
+  await Promise.all(
+    emptySessionIds.map(async id => {
+      await createTestSession(t, { sessionId: id });
+      await db.aiSession.update({
+        where: { id },
+        data: { messageCost: 100, updatedAt: oneDayAgo },
+      });
+    })
+  );
+
+  // should not be affected
+  const recentSessionId = randomUUID();
+  await createTestSession(t, { sessionId: recentSessionId });
+  await db.aiSession.update({
+    where: { id: recentSessionId },
+    data: { messageCost: 0, updatedAt: twoHoursAgo },
+  });
+
+  // Create session with messages (should not be affected)
+  const sessionWithMsgId = randomUUID();
+  await createSessionWithMessages(
+    t,
+    { sessionId: sessionWithMsgId },
+    'test message'
+  );
+
+  const result = await copilotSession.cleanupEmptySessions(oneDayAgo);
+
+  const remainingSessions = await db.aiSession.findMany({
+    where: {
+      id: {
+        in: [
+          ...neverUsedSessionIds,
+          ...emptySessionIds,
+          recentSessionId,
+          sessionWithMsgId,
+        ],
+      },
+    },
+    select: { id: true, deletedAt: true, pinned: true },
+  });
+
+  t.snapshot(
+    {
+      cleanupResult: result,
+      remainingSessions: remainingSessions.map(s => ({
+        deleted: !!s.deletedAt,
+        pinned: s.pinned,
+        type: neverUsedSessionIds.includes(s.id)
+          ? 'zeroCost'
+          : emptySessionIds.includes(s.id)
+            ? 'noMessages'
+            : s.id === recentSessionId
+              ? 'recent'
+              : 'withMessages',
+      })),
+    },
+    'cleanup empty sessions results'
+  );
+});
+
+test('should get sessions for title generation correctly', async t => {
+  const { copilotSession, db } = t.context;
+  await createTestPrompts(copilotSession, db);
+
+  // create valid sessions with messages
+  const sessionIds: string[] = [randomUUID(), randomUUID()];
+  await Promise.all(
+    sessionIds.map(async (id, index) => {
+      await createTestSession(t, { sessionId: id });
+      await db.aiSession.update({
+        where: { id },
+        data: {
+          updatedAt: new Date(Date.now() - index * 1000),
+          messages: {
+            create: Array.from({ length: index + 1 }, (_, i) => ({
+              role: 'assistant',
+              content: `assistant message ${i}`,
+            })),
+          },
+        },
+      });
+    })
+  );
+
+  // create excluded sessions
+  const excludedSessions = [
+    {
+      reason: 'hasTitle',
+      setupFn: async (id: string) => {
+        await createTestSession(t, { sessionId: id });
+        await db.aiSession.update({
+          where: { id },
+          data: { title: 'Existing Title' },
+        });
+      },
+    },
+    {
+      reason: 'isDeleted',
+      setupFn: async (id: string) => {
+        await createTestSession(t, { sessionId: id });
+        await db.aiSession.update({
+          where: { id },
+          data: { deletedAt: new Date() },
+        });
+      },
+    },
+    {
+      reason: 'noMessages',
+      setupFn: async (id: string) => {
+        await createTestSession(t, { sessionId: id });
+      },
+    },
+    {
+      reason: 'isAction',
+      setupFn: async (id: string) => {
+        await createTestSession(t, {
+          sessionId: id,
+          promptName: TEST_PROMPTS.ACTION,
+        });
+      },
+    },
+    {
+      reason: 'noAssistantMessages',
+      setupFn: async (id: string) => {
+        await createTestSession(t, { sessionId: id });
+        await db.aiSessionMessage.create({
+          data: { sessionId: id, role: 'user', content: 'User message only' },
+        });
+      },
+    },
+  ];
+
+  await Promise.all(
+    excludedSessions.map(async session => {
+      await session.setupFn(randomUUID());
+    })
+  );
+
+  const result = await copilotSession.toBeGenerateTitle();
+
+  t.snapshot(
+    {
+      total: result.length,
+      sessions: result.map(s => ({
+        assistantMessageCount: s._count.messages,
+        isValid: sessionIds.includes(s.id),
+      })),
+      onlyValidSessionsReturned: result.every(s => sessionIds.includes(s.id)),
+    },
+    'sessions for title generation results'
+  );
+});

packages/backend/server/src/__tests__/models/doc.spec.ts

@@ -669,7 +669,10 @@ test('should get doc info', async t => {
   };
 
   await t.context.doc.upsert(snapshot);
-  await t.context.doc.upsertMeta(workspace.id, docId);
+  await t.context.doc.upsertMeta(workspace.id, docId, {
+    title: 'test title',
+    summary: 'test summary',
+  });
 
   const docInfo = await t.context.doc.getDocInfo(workspace.id, docId);
 
@@ -679,6 +682,8 @@ test('should get doc info', async t => {
     updatedAt: new Date(snapshot.timestamp),
     creatorId: user.id,
     lastUpdaterId: user.id,
+    title: 'test title',
+    summary: 'test summary',
   });
 });
 

packages/backend/server/src/__tests__/utils/copilot.ts

@@ -433,7 +433,7 @@ export async function submitAudioTranscription(
   for (const [idx, buffer] of content.entries()) {
     resp = resp.attach(idx.toString(), buffer, {
       filename: fileName,
-      contentType: 'application/octet-stream',
+      contentType: 'audio/opus',
     });
   }
 

packages/backend/server/src/app.module.ts

@@ -36,6 +36,7 @@ import { DocRendererModule } from './core/doc-renderer';
 import { DocServiceModule } from './core/doc-service';
 import { FeatureModule } from './core/features';
 import { MailModule } from './core/mail';
+import { MonitorModule } from './core/monitor';
 import { NotificationModule } from './core/notification';
 import { PermissionModule } from './core/permission';
 import { QuotaModule } from './core/quota';
@@ -112,6 +113,8 @@ export const FunctionalityModules = [
   WebSocketModule,
   JobModule.forRoot(),
   ModelsModule,
+  ScheduleModule.forRoot(),
+  MonitorModule,
 ];
 
 export class AppModuleBuilder {
@@ -151,12 +154,8 @@ export function buildAppModule(env: Env) {
     // basic
     .use(...FunctionalityModules)
 
-    // enable schedule module on graphql server and doc service
-    .useIf(
-      () => env.flavors.graphql || env.flavors.doc,
-      ScheduleModule.forRoot(),
-      IndexerModule
-    )
+    // enable indexer module on graphql server and doc service
+    .useIf(() => env.flavors.graphql || env.flavors.doc, IndexerModule)
 
     // auth
     .use(UserModule, AuthModule, PermissionModule)

packages/backend/server/src/base/error/def.ts

@@ -653,12 +653,19 @@ export const USER_FRIENDLY_ERRORS = {
   },
   no_copilot_provider_available: {
     type: 'internal_server_error',
-    message: `No copilot provider available.`,
+    args: { modelId: 'string' },
+    message: ({ modelId }) => `No copilot provider available: ${modelId}`,
   },
   copilot_failed_to_generate_text: {
     type: 'internal_server_error',
     message: `Failed to generate text.`,
   },
+  copilot_failed_to_generate_embedding: {
+    type: 'internal_server_error',
+    args: { provider: 'string', message: 'string' },
+    message: ({ provider, message }) =>
+      `Failed to generate embedding with ${provider}: ${message}`,
+  },
   copilot_failed_to_create_message: {
     type: 'internal_server_error',
     message: `Failed to create chat message.`,

packages/backend/server/src/base/error/errors.gen.ts

@@ -668,10 +668,14 @@ export class CopilotSessionDeleted extends UserFriendlyError {
     super('action_forbidden', 'copilot_session_deleted', message);
   }
 }
+@ObjectType()
+class NoCopilotProviderAvailableDataType {
+  @Field() modelId!: string
+}
 
 export class NoCopilotProviderAvailable extends UserFriendlyError {
-  constructor(message?: string) {
-    super('internal_server_error', 'no_copilot_provider_available', message);
+  constructor(args: NoCopilotProviderAvailableDataType, message?: string | ((args: NoCopilotProviderAvailableDataType) => string)) {
+    super('internal_server_error', 'no_copilot_provider_available', message, args);
   }
 }
 
@@ -680,6 +684,17 @@ export class CopilotFailedToGenerateText extends UserFriendlyError {
     super('internal_server_error', 'copilot_failed_to_generate_text', message);
   }
 }
+@ObjectType()
+class CopilotFailedToGenerateEmbeddingDataType {
+  @Field() provider!: string
+  @Field() message!: string
+}
+
+export class CopilotFailedToGenerateEmbedding extends UserFriendlyError {
+  constructor(args: CopilotFailedToGenerateEmbeddingDataType, message?: string | ((args: CopilotFailedToGenerateEmbeddingDataType) => string)) {
+    super('internal_server_error', 'copilot_failed_to_generate_embedding', message, args);
+  }
+}
 
 export class CopilotFailedToCreateMessage extends UserFriendlyError {
   constructor(message?: string) {
@@ -1179,6 +1194,7 @@ export enum ErrorNames {
   COPILOT_SESSION_DELETED,
   NO_COPILOT_PROVIDER_AVAILABLE,
   COPILOT_FAILED_TO_GENERATE_TEXT,
+  COPILOT_FAILED_TO_GENERATE_EMBEDDING,
   COPILOT_FAILED_TO_CREATE_MESSAGE,
   UNSPLASH_IS_NOT_CONFIGURED,
   COPILOT_ACTION_TAKEN,
@@ -1239,5 +1255,5 @@ registerEnumType(ErrorNames, {
 export const ErrorDataUnionType = createUnionType({
   name: 'ErrorDataUnion',
   types: () =>
-    [GraphqlBadRequestDataType, HttpRequestErrorDataType, QueryTooLongDataType, ValidationErrorDataType, WrongSignInCredentialsDataType, UnknownOauthProviderDataType, InvalidOauthCallbackCodeDataType, MissingOauthQueryParameterDataType, InvalidOauthResponseDataType, InvalidEmailDataType, InvalidPasswordLengthDataType, WorkspacePermissionNotFoundDataType, SpaceNotFoundDataType, MemberNotFoundInSpaceDataType, NotInSpaceDataType, AlreadyInSpaceDataType, SpaceAccessDeniedDataType, SpaceOwnerNotFoundDataType, SpaceShouldHaveOnlyOneOwnerDataType, DocNotFoundDataType, DocActionDeniedDataType, DocUpdateBlockedDataType, VersionRejectedDataType, InvalidHistoryTimestampDataType, DocHistoryNotFoundDataType, BlobNotFoundDataType, ExpectToGrantDocUserRolesDataType, ExpectToRevokeDocUserRolesDataType, ExpectToUpdateDocUserRoleDataType, NoMoreSeatDataType, UnsupportedSubscriptionPlanDataType, SubscriptionAlreadyExistsDataType, SubscriptionNotExistsDataType, SameSubscriptionRecurringDataType, SubscriptionPlanNotFoundDataType, CopilotDocNotFoundDataType, CopilotMessageNotFoundDataType, CopilotPromptNotFoundDataType, CopilotProviderNotSupportedDataType, CopilotProviderSideErrorDataType, CopilotInvalidContextDataType, CopilotContextFileNotSupportedDataType, CopilotFailedToModifyContextDataType, CopilotFailedToMatchContextDataType, CopilotFailedToMatchGlobalContextDataType, CopilotFailedToAddWorkspaceFileEmbeddingDataType, RuntimeConfigNotFoundDataType, InvalidRuntimeConfigTypeDataType, InvalidLicenseToActivateDataType, InvalidLicenseUpdateParamsDataType, UnsupportedClientVersionDataType, MentionUserDocAccessDeniedDataType, InvalidAppConfigDataType, InvalidAppConfigInputDataType, InvalidSearchProviderRequestDataType, InvalidIndexerInputDataType] as const,
+    [GraphqlBadRequestDataType, HttpRequestErrorDataType, QueryTooLongDataType, ValidationErrorDataType, WrongSignInCredentialsDataType, UnknownOauthProviderDataType, InvalidOauthCallbackCodeDataType, MissingOauthQueryParameterDataType, InvalidOauthResponseDataType, InvalidEmailDataType, InvalidPasswordLengthDataType, WorkspacePermissionNotFoundDataType, SpaceNotFoundDataType, MemberNotFoundInSpaceDataType, NotInSpaceDataType, AlreadyInSpaceDataType, SpaceAccessDeniedDataType, SpaceOwnerNotFoundDataType, SpaceShouldHaveOnlyOneOwnerDataType, DocNotFoundDataType, DocActionDeniedDataType, DocUpdateBlockedDataType, VersionRejectedDataType, InvalidHistoryTimestampDataType, DocHistoryNotFoundDataType, BlobNotFoundDataType, ExpectToGrantDocUserRolesDataType, ExpectToRevokeDocUserRolesDataType, ExpectToUpdateDocUserRoleDataType, NoMoreSeatDataType, UnsupportedSubscriptionPlanDataType, SubscriptionAlreadyExistsDataType, SubscriptionNotExistsDataType, SameSubscriptionRecurringDataType, SubscriptionPlanNotFoundDataType, NoCopilotProviderAvailableDataType, CopilotFailedToGenerateEmbeddingDataType, CopilotDocNotFoundDataType, CopilotMessageNotFoundDataType, CopilotPromptNotFoundDataType, CopilotProviderNotSupportedDataType, CopilotProviderSideErrorDataType, CopilotInvalidContextDataType, CopilotContextFileNotSupportedDataType, CopilotFailedToModifyContextDataType, CopilotFailedToMatchContextDataType, CopilotFailedToMatchGlobalContextDataType, CopilotFailedToAddWorkspaceFileEmbeddingDataType, RuntimeConfigNotFoundDataType, InvalidRuntimeConfigTypeDataType, InvalidLicenseToActivateDataType, InvalidLicenseUpdateParamsDataType, UnsupportedClientVersionDataType, MentionUserDocAccessDeniedDataType, InvalidAppConfigDataType, InvalidAppConfigInputDataType, InvalidSearchProviderRequestDataType, InvalidIndexerInputDataType] as const,
 });

packages/backend/server/src/base/metrics/metrics.ts

@@ -59,7 +59,9 @@ export type KnownMetricScopes =
   | 'mail'
   | 'ai'
   | 'event'
-  | 'queue';
+  | 'queue'
+  | 'storage'
+  | 'process';
 
 const metricCreators: MetricCreators = {
   counter(meter: Meter, name: string, opts?: MetricOptions) {

packages/backend/server/src/core/doc/adapters/workspace.ts

@@ -100,7 +100,7 @@ export class PgWorkspaceDocStorageAdapter extends DocStorageAdapter {
             {
               // keep it simple to let all update merged in one job
               jobId: `doc:merge-pending-updates:${workspaceId}:${docId}`,
-              delay: 30 * 1000 /* 30s */,
+              delay: 5 * 1000 /* 5s */,
               priority: 100,
             }
           );

packages/backend/server/src/core/monitor/index.ts

@@ -0,0 +1,9 @@
+import { Global, Module } from '@nestjs/common';
+
+import { MonitorService } from './service';
+
+@Global()
+@Module({
+  providers: [MonitorService],
+})
+export class MonitorModule {}

packages/backend/server/src/core/monitor/service.ts

@@ -0,0 +1,28 @@
+import { Injectable, Logger } from '@nestjs/common';
+import { Cron, CronExpression } from '@nestjs/schedule';
+
+import { metrics } from '../../base';
+
+@Injectable()
+export class MonitorService {
+  protected logger = new Logger(MonitorService.name);
+
+  @Cron(CronExpression.EVERY_MINUTE)
+  async monitor() {
+    const memoryUsage = process.memoryUsage();
+    this.logger.log(
+      `memory usage: rss: ${memoryUsage.rss}, heapTotal: ${memoryUsage.heapTotal}, heapUsed: ${memoryUsage.heapUsed}, external: ${memoryUsage.external}, arrayBuffers: ${memoryUsage.arrayBuffers}`
+    );
+    metrics.process.gauge('node_process_rss').record(memoryUsage.rss);
+    metrics.process
+      .gauge('node_process_heap_total')
+      .record(memoryUsage.heapTotal);
+    metrics.process
+      .gauge('node_process_heap_used')
+      .record(memoryUsage.heapUsed);
+    metrics.process.gauge('node_process_external').record(memoryUsage.external);
+    metrics.process
+      .gauge('node_process_array_buffers')
+      .record(memoryUsage.arrayBuffers);
+  }
+}

packages/backend/server/src/core/storage/wrappers/comment-attachment.ts

@@ -4,6 +4,7 @@ import {
   autoMetadata,
   Config,
   EventBus,
+  metrics,
   OnEvent,
   type StorageProvider,
   StorageProviderFactory,
@@ -69,15 +70,23 @@ export class CommentAttachmentStorage {
       blob,
       meta
     );
+    const mime = meta.contentType ?? 'application/octet-stream';
+    const size = blob.length;
     await this.models.commentAttachment.upsert({
       workspaceId,
       docId,
       key,
       name,
-      mime: meta.contentType ?? 'application/octet-stream',
-      size: blob.length,
+      mime,
+      size,
       createdBy: userId,
     });
+
+    metrics.storage.histogram('comment_attachment_size').record(size, { mime });
+    metrics.storage.counter('comment_attachment_total').add(1, { mime });
+    this.logger.log(
+      `uploaded comment attachment ${workspaceId}/${docId}/${key} with size ${size}, mime: ${mime}, name: ${name}, user: ${userId}`
+    );
   }
 
   async get(

packages/backend/server/src/core/workspaces/resolvers/doc.ts

@@ -79,6 +79,9 @@ class DocType {
 
   @Field(() => String, { nullable: true })
   title?: string | null;
+
+  @Field(() => String, { nullable: true })
+  summary?: string | null;
 }
 
 @InputType()
@@ -250,10 +253,11 @@ export class WorkspaceDocResolver {
     deprecationReason: 'use [WorkspaceType.doc] instead',
   })
   async publicPage(
+    @CurrentUser() me: CurrentUser,
     @Parent() workspace: WorkspaceType,
     @Args('pageId') pageId: string
   ) {
-    return this.doc(workspace, pageId);
+    return this.doc(me, workspace, pageId);
   }
 
   @ResolveField(() => PaginatedDocType)
@@ -294,11 +298,14 @@ export class WorkspaceDocResolver {
     complexity: 2,
   })
   async doc(
+    @CurrentUser() me: CurrentUser,
     @Parent() workspace: WorkspaceType,
     @Args('docId') docId: string
   ): Promise<DocType> {
     const doc = await this.models.doc.getDocInfo(workspace.id, docId);
     if (doc) {
+      // check if doc is readable
+      await this.ac.user(me.id).doc(workspace.id, docId).assert('Doc.Read');
       return doc;
     }
 

packages/backend/server/src/models/copilot-context.ts

@@ -165,6 +165,13 @@ export class CopilotContextModel extends BaseModel {
     fileId: string,
     embeddings: Embedding[]
   ) {
+    if (embeddings.length === 0) {
+      this.logger.warn(
+        `No embeddings provided for contextId: ${contextId}, fileId: ${fileId}. Skipping insertion.`
+      );
+      return;
+    }
+
     const values = this.processEmbeddings(contextId, fileId, embeddings);
 
     await this.db.$executeRaw`
@@ -204,6 +211,13 @@ export class CopilotContextModel extends BaseModel {
     docId: string,
     embeddings: Embedding[]
   ) {
+    if (embeddings.length === 0) {
+      this.logger.warn(
+        `No embeddings provided for workspaceId: ${workspaceId}, docId: ${docId}. Skipping insertion.`
+      );
+      return;
+    }
+
     const values = this.processEmbeddings(
       workspaceId,
       docId,

packages/backend/server/src/models/copilot-session.ts

@@ -582,4 +582,56 @@ export class CopilotSessionModel extends BaseModel {
       .map(({ messageCost, prompt: { action } }) => (action ? 1 : messageCost))
       .reduce((prev, cost) => prev + cost, 0);
   }
+
+  @Transactional()
+  async cleanupEmptySessions(earlyThen: Date) {
+    // delete never used sessions
+    const { count: removed } = await this.db.aiSession.deleteMany({
+      where: {
+        messageCost: 0,
+        deletedAt: null,
+        // filter session updated more than 24 hours ago
+        updatedAt: { lt: earlyThen },
+      },
+    });
+
+    // mark empty sessions as deleted
+    const { count: cleaned } = await this.db.aiSession.updateMany({
+      where: {
+        deletedAt: null,
+        messages: { none: {} },
+        // filter session updated more than 24 hours ago
+        updatedAt: { lt: earlyThen },
+      },
+      data: {
+        deletedAt: new Date(),
+        pinned: false,
+      },
+    });
+
+    return { removed, cleaned };
+  }
+
+  @Transactional()
+  async toBeGenerateTitle() {
+    const sessions = await this.db.aiSession
+      .findMany({
+        where: {
+          title: null,
+          deletedAt: null,
+          messages: { some: {} },
+          // only generate titles for non-actions sessions
+          prompt: { action: null },
+        },
+        select: {
+          id: true,
+          // count assistant messages
+          _count: { select: { messages: { where: { role: 'assistant' } } } },
+        },
+        orderBy: { updatedAt: 'desc' },
+      })
+      .then(s => s.filter(s => s._count.messages > 0));
+
+    return sessions;
+  }
 }

packages/backend/server/src/models/copilot-workspace.ts

@@ -283,6 +283,13 @@ export class CopilotWorkspaceConfigModel extends BaseModel {
     fileId: string,
     embeddings: Embedding[]
   ) {
+    if (embeddings.length === 0) {
+      this.logger.warn(
+        `No embeddings provided for workspaceId: ${workspaceId}, fileId: ${fileId}. Skipping insertion.`
+      );
+      return;
+    }
+
     const values = this.processEmbeddings(workspaceId, fileId, embeddings);
     await this.db.$executeRaw`
           INSERT INTO "ai_workspace_file_embeddings"

packages/backend/server/src/models/doc.ts

@@ -558,6 +558,8 @@ export class DocModel extends BaseModel {
         mode: PublicDocMode;
         public: boolean;
         defaultRole: DocRole;
+        title: string | null;
+        summary: string | null;
         createdAt: Date;
         updatedAt: Date;
         creatorId?: string;
@@ -570,6 +572,8 @@ export class DocModel extends BaseModel {
      "workspace_pages"."mode" as "mode",
      "workspace_pages"."public" as "public",
      "workspace_pages"."defaultRole" as "defaultRole",
+     "workspace_pages"."title" as "title",
+     "workspace_pages"."summary" as "summary",
      "snapshots"."created_at" as "createdAt",
      "snapshots"."updated_at" as "updatedAt",
      "snapshots"."created_by" as "creatorId",

packages/backend/server/src/plugins/copilot/context/service.ts

@@ -125,7 +125,10 @@ export class CopilotContextService implements OnApplicationBootstrap {
 
   async get(id: string): Promise<ContextSession> {
     if (!this.embeddingClient) {
-      throw new NoCopilotProviderAvailable('embedding client not configured');
+      throw new NoCopilotProviderAvailable(
+        { modelId: 'embedding' },
+        'embedding client not configured'
+      );
     }
 
     const context = await this.getCachedSession(id);

packages/backend/server/src/plugins/copilot/controller.ts

@@ -124,7 +124,7 @@ export class CopilotController implements BeforeApplicationShutdown {
       modelId: model,
     });
     if (!provider) {
-      throw new NoCopilotProviderAvailable();
+      throw new NoCopilotProviderAvailable({ modelId: model });
     }
 
     return { provider, model, hasAttachment };
@@ -299,6 +299,13 @@ export class CopilotController implements BeforeApplicationShutdown {
       this.ongoingStreamCount$.next(this.ongoingStreamCount$.value + 1);
 
       const { signal, onConnectionClosed } = getSignal(req);
+      let endBeforePromiseResolve = false;
+      onConnectionClosed(isAborted => {
+        if (isAborted) {
+          endBeforePromiseResolve = true;
+        }
+      });
+
       const { messageId, reasoning, webSearch } = ChatQuerySchema.parse(query);
 
       const source$ = from(
@@ -322,21 +329,21 @@ export class CopilotController implements BeforeApplicationShutdown {
             shared$.pipe(
               reduce((acc, chunk) => acc + chunk, ''),
               tap(buffer => {
-                onConnectionClosed(isAborted => {
-                  session.push({
-                    role: 'assistant',
-                    content: isAborted ? '> Request aborted' : buffer,
-                    createdAt: new Date(),
-                  });
-                  void session
-                    .save()
-                    .catch(err =>
-                      this.logger.error(
-                        'Failed to save session in sse stream',
-                        err
-                      )
-                    );
+                session.push({
+                  role: 'assistant',
+                  content: endBeforePromiseResolve
+                    ? '> Request aborted'
+                    : buffer,
+                  createdAt: new Date(),
                 });
+                void session
+                  .save()
+                  .catch(err =>
+                    this.logger.error(
+                      'Failed to save session in sse stream',
+                      err
+                    )
+                  );
               }),
               ignoreElements()
             )
@@ -384,6 +391,13 @@ export class CopilotController implements BeforeApplicationShutdown {
       this.ongoingStreamCount$.next(this.ongoingStreamCount$.value + 1);
 
       const { signal, onConnectionClosed } = getSignal(req);
+      let endBeforePromiseResolve = false;
+      onConnectionClosed(isAborted => {
+        if (isAborted) {
+          endBeforePromiseResolve = true;
+        }
+      });
+
       const { messageId, reasoning, webSearch } = ChatQuerySchema.parse(query);
 
       const source$ = from(
@@ -407,25 +421,25 @@ export class CopilotController implements BeforeApplicationShutdown {
             shared$.pipe(
               reduce((acc, chunk) => acc.concat([chunk]), [] as StreamObject[]),
               tap(result => {
-                onConnectionClosed(isAborted => {
-                  const parser = new StreamObjectParser();
-                  const streamObjects = parser.mergeTextDelta(result);
-                  const content = parser.mergeContent(streamObjects);
-                  session.push({
-                    role: 'assistant',
-                    content: isAborted ? '> Request aborted' : content,
-                    streamObjects: isAborted ? null : streamObjects,
-                    createdAt: new Date(),
-                  });
-                  void session
-                    .save()
-                    .catch(err =>
-                      this.logger.error(
-                        'Failed to save session in sse stream',
-                        err
-                      )
-                    );
+                const parser = new StreamObjectParser();
+                const streamObjects = parser.mergeTextDelta(result);
+                const content = parser.mergeContent(streamObjects);
+                session.push({
+                  role: 'assistant',
+                  content: endBeforePromiseResolve
+                    ? '> Request aborted'
+                    : content,
+                  streamObjects: endBeforePromiseResolve ? null : streamObjects,
+                  createdAt: new Date(),
                 });
+                void session
+                  .save()
+                  .catch(err =>
+                    this.logger.error(
+                      'Failed to save session in sse stream',
+                      err
+                    )
+                  );
               }),
               ignoreElements()
             )
@@ -477,6 +491,13 @@ export class CopilotController implements BeforeApplicationShutdown {
       this.ongoingStreamCount$.next(this.ongoingStreamCount$.value + 1);
 
       const { signal, onConnectionClosed } = getSignal(req);
+      let endBeforePromiseResolve = false;
+      onConnectionClosed(isAborted => {
+        if (isAborted) {
+          endBeforePromiseResolve = true;
+        }
+      });
+
       const source$ = from(
         this.workflow.runGraph(params, session.model, {
           ...session.config.promptConfig,
@@ -526,21 +547,21 @@ export class CopilotController implements BeforeApplicationShutdown {
                 return acc;
               }, ''),
               tap(content => {
-                onConnectionClosed(isAborted => {
-                  session.push({
-                    role: 'assistant',
-                    content: isAborted ? '> Request aborted' : content,
-                    createdAt: new Date(),
-                  });
-                  void session
-                    .save()
-                    .catch(err =>
-                      this.logger.error(
-                        'Failed to save session in sse stream',
-                        err
-                      )
-                    );
+                session.push({
+                  role: 'assistant',
+                  content: endBeforePromiseResolve
+                    ? '> Request aborted'
+                    : content,
+                  createdAt: new Date(),
                 });
+                void session
+                  .save()
+                  .catch(err =>
+                    this.logger.error(
+                      'Failed to save session in sse stream',
+                      err
+                    )
+                  );
               }),
               ignoreElements()
             )
@@ -604,6 +625,13 @@ export class CopilotController implements BeforeApplicationShutdown {
       this.ongoingStreamCount$.next(this.ongoingStreamCount$.value + 1);
 
       const { signal, onConnectionClosed } = getSignal(req);
+      let endBeforePromiseResolve = false;
+      onConnectionClosed(isAborted => {
+        if (isAborted) {
+          endBeforePromiseResolve = true;
+        }
+      });
+
       const source$ = from(
         provider.streamImages(
           {
@@ -639,22 +667,20 @@ export class CopilotController implements BeforeApplicationShutdown {
             shared$.pipe(
               reduce((acc, chunk) => acc.concat([chunk]), [] as string[]),
               tap(attachments => {
-                onConnectionClosed(isAborted => {
-                  session.push({
-                    role: 'assistant',
-                    content: isAborted ? '> Request aborted' : '',
-                    attachments: isAborted ? [] : attachments,
-                    createdAt: new Date(),
-                  });
-                  void session
-                    .save()
-                    .catch(err =>
-                      this.logger.error(
-                        'Failed to save session in sse stream',
-                        err
-                      )
-                    );
+                session.push({
+                  role: 'assistant',
+                  content: endBeforePromiseResolve ? '> Request aborted' : '',
+                  attachments: endBeforePromiseResolve ? [] : attachments,
+                  createdAt: new Date(),
                 });
+                void session
+                  .save()
+                  .catch(err =>
+                    this.logger.error(
+                      'Failed to save session in sse stream',
+                      err
+                    )
+                  );
               }),
               ignoreElements()
             )

packages/backend/server/src/plugins/copilot/cron.ts

@@ -0,0 +1,71 @@
+import { Injectable, Logger } from '@nestjs/common';
+import { Cron, CronExpression } from '@nestjs/schedule';
+
+import { JobQueue, OneDay, OnJob } from '../../base';
+import { Models } from '../../models';
+
+declare global {
+  interface Jobs {
+    'copilot.session.cleanupEmptySessions': {};
+    'copilot.session.generateMissingTitles': {};
+  }
+}
+
+@Injectable()
+export class CopilotCronJobs {
+  private readonly logger = new Logger(CopilotCronJobs.name);
+
+  constructor(
+    private readonly models: Models,
+    private readonly jobs: JobQueue
+  ) {}
+
+  @Cron(CronExpression.EVERY_DAY_AT_MIDNIGHT)
+  async dailyCleanupJob() {
+    await this.jobs.add(
+      'copilot.session.cleanupEmptySessions',
+      {},
+      { jobId: 'daily-copilot-cleanup-empty-sessions' }
+    );
+
+    await this.jobs.add(
+      'copilot.session.generateMissingTitles',
+      {},
+      { jobId: 'daily-copilot-generate-missing-titles' }
+    );
+  }
+
+  async triggerGenerateMissingTitles() {
+    await this.jobs.add(
+      'copilot.session.generateMissingTitles',
+      {},
+      { jobId: 'trigger-copilot-generate-missing-titles' }
+    );
+  }
+
+  @OnJob('copilot.session.cleanupEmptySessions')
+  async cleanupEmptySessions() {
+    const { removed, cleaned } =
+      await this.models.copilotSession.cleanupEmptySessions(
+        new Date(Date.now() - OneDay)
+      );
+
+    this.logger.log(
+      `Cleanup completed: ${removed} sessions deleted, ${cleaned} sessions marked as deleted`
+    );
+  }
+
+  @OnJob('copilot.session.generateMissingTitles')
+  async generateMissingTitles() {
+    const sessions = await this.models.copilotSession.toBeGenerateTitle();
+
+    for (const session of sessions) {
+      await this.jobs.add('copilot.session.generateTitle', {
+        sessionId: session.id,
+      });
+    }
+    this.logger.log(
+      `Scheduled title generation for ${sessions.length} sessions`
+    );
+  }
+}

packages/backend/server/src/plugins/copilot/embedding/client.ts

@@ -5,6 +5,7 @@ import {
   CopilotPromptNotFound,
   CopilotProviderNotSupported,
 } from '../../../base';
+import { CopilotFailedToGenerateEmbedding } from '../../../base/error/errors.gen';
 import { ChunkSimilarity, Embedding } from '../../../models';
 import { PromptService } from '../prompt';
 import {
@@ -74,6 +75,12 @@ class ProductionEmbeddingClient extends EmbeddingClient {
       input,
       { dimensions: EMBEDDING_DIMENSIONS }
     );
+    if (embeddings.length !== input.length) {
+      throw new CopilotFailedToGenerateEmbedding({
+        provider: provider.type,
+        message: `Expected ${input.length} embeddings, got ${embeddings.length}`,
+      });
+    }
 
     return Array.from(embeddings.entries()).map(([index, embedding]) => ({
       index,

packages/backend/server/src/plugins/copilot/index.ts

@@ -15,6 +15,7 @@ import {
   CopilotContextService,
 } from './context';
 import { CopilotController } from './controller';
+import { CopilotCronJobs } from './cron';
 import { CopilotEmbeddingJob } from './embedding';
 import { ChatMessageCache } from './message';
 import { PromptService } from './prompt';
@@ -64,6 +65,8 @@ import {
     CopilotContextResolver,
     CopilotContextService,
     CopilotEmbeddingJob,
+    // cron jobs
+    CopilotCronJobs,
     // transcription
     CopilotTranscriptionService,
     CopilotTranscriptionResolver,

packages/backend/server/src/plugins/copilot/prompt/prompts.ts

@@ -303,7 +303,8 @@ const textActions: Prompt[] = [
   {
     name: 'Transcript audio',
     action: 'Transcript audio',
-    model: 'gemini-2.5-flash',
+    model: 'gemini-2.5-pro',
+    optionalModels: ['gemini-2.5-flash', 'gemini-2.5-pro'],
     messages: [
       {
         role: 'system',
@@ -366,6 +367,31 @@ Convert a multi-speaker audio recording into a structured JSON format by transcr
       requireAttachment: true,
     },
   },
+  {
+    name: 'Conversation Summary',
+    action: 'Conversation Summary',
+    model: 'gpt-4.1-2025-04-14',
+    messages: [
+      {
+        role: 'system',
+        content: `You are an expert conversation summarizer. Your job is to distill long dialogues into clear, compact summaries that preserve every key decision, fact, and open question. When asked, always:
+• Honor any explicit “focus” the user gives you.
+• Match the desired length style:
+  - “brief” → 1-2 sentences
+  - “detailed” → ≈ 5 sentences or short bullet list
+  - “comprehensive” → full paragraph(s) covering all salient points.
+• Write in neutral, third-person prose and never add new information.
+Return only the summary text—no headings, labels, or commentary.`,
+      },
+      {
+        role: 'user',
+        content: `Summarize the conversation below so it can be carried forward without loss.\n\nFocus: {{focus}}\nDesired length: {{length}}\n\nConversation:\n{{#messages}}\n{{role}}: {{content}}\n{{/messages}}`,
+      },
+    ],
+    config: {
+      requireContent: false,
+    },
+  },
   {
     name: 'Summary',
     action: 'Summary',
@@ -1598,6 +1624,166 @@ const imageActions: Prompt[] = [
   },
 ];
 
+const modelActions: Prompt[] = [
+  {
+    name: 'Apply Updates',
+    action: 'Apply Updates',
+    model: 'claude-sonnet-4@20250514',
+    messages: [
+      {
+        role: 'user',
+        content: `
+You are a Markdown document update engine.
+
+You will be given:
+
+1. content: The original Markdown document
+   - The content is structured into blocks.
+   - Each block starts with a comment like <!-- block_id=... flavour=... --> and contains the block's content.
+   - The content is {{content}}
+
+2. op: A description of the edit intention
+   - This describes the semantic meaning of the edit, such as "Bold the first paragraph".
+   - The op is {{op}}
+
+3. updates: A Markdown snippet
+   - The updates is {{updates}}
+   - This represents the block-level changes to apply to the original Markdown.
+   - The update may:
+     - **Replace** an existing block (same block_id, new content)
+     - **Delete** block(s) using <!-- delete block BLOCK_ID -->
+     - **Insert** new block(s) with a new unique block_id
+   - When performing deletions, the update will include **surrounding context blocks** (or use <!-- existing blocks -->) to help you determine where and what to delete.
+
+Your task:
+- Apply the update in <updates> to the document in <code>, following the intent described in <op>.
+- Preserve all block_id and flavour comments.
+- Maintain the original block order unless the update clearly appends new blocks.
+- Do not remove or alter unrelated blocks.
+- Output only the fully updated Markdown content. Do not wrap the content in \`\`\`markdown.
+
+---
+
+✍️ Examples
+
+✅ Replacement (modifying an existing block)
+
+<code>
+<!-- block_id=101 flavour=paragraph -->
+## Introduction
+
+<!-- block_id=102 flavour=paragraph -->
+This document provides an overview of the system architecture and its components.
+</code>
+
+<op>
+Make the introduction more formal.
+</op>
+
+<updates>
+<!-- block_id=102 flavour=paragraph -->
+This document outlines the architectural design and individual components of the system in detail.
+</updates>
+
+Expected Output:
+<!-- block_id=101 flavour=paragraph -->
+## Introduction
+
+<!-- block_id=102 flavour=paragraph -->
+This document outlines the architectural design and individual components of the system in detail.
+
+---
+
+➕ Insertion (adding new content)
+
+<code>
+<!-- block_id=201 flavour=paragraph -->
+# Project Summary
+
+<!-- block_id=202 flavour=paragraph -->
+This project aims to build a collaborative text editing tool.
+</code>
+
+<op>
+Add a disclaimer section at the end.
+</op>
+
+<updates>
+<!-- block_id=new-301 flavour=paragraph -->
+## Disclaimer
+
+<!-- block_id=new-302 flavour=paragraph -->
+This document is subject to change. Do not distribute externally.
+</updates>
+
+Expected Output:
+<!-- block_id=201 flavour=paragraph -->
+# Project Summary
+
+<!-- block_id=202 flavour=paragraph -->
+This project aims to build a collaborative text editing tool.
+
+<!-- block_id=new-301 flavour=paragraph -->
+## Disclaimer
+
+<!-- block_id=new-302 flavour=paragraph -->
+This document is subject to change. Do not distribute externally.
+
+---
+
+❌ Deletion (removing blocks)
+
+<code>
+<!-- block_id=401 flavour=paragraph -->
+## Author
+
+<!-- block_id=402 flavour=paragraph -->
+Written by the AI team at OpenResearch.
+
+<!-- block_id=403 flavour=paragraph -->
+## Experimental Section
+
+<!-- block_id=404 flavour=paragraph -->
+The following section is still under development and may change without notice.
+
+<!-- block_id=405 flavour=paragraph -->
+## License
+
+<!-- block_id=406 flavour=paragraph -->
+This document is licensed under CC BY-NC 4.0.
+</code>
+
+<op>
+Remove the experimental section.
+</op>
+
+<updates>
+<!-- delete block_id=403 -->
+<!-- delete block_id=404 -->
+</updates>
+
+Expected Output:
+<!-- block_id=401 flavour=paragraph -->
+## Author
+
+<!-- block_id=402 flavour=paragraph -->
+Written by the AI team at OpenResearch.
+
+<!-- block_id=405 flavour=paragraph -->
+## License
+
+<!-- block_id=406 flavour=paragraph -->
+This document is licensed under CC BY-NC 4.0.
+
+---
+
+Now apply the \`updates\` to the \`content\`, following the intent in \`op\`, and return the updated Markdown.
+`,
+      },
+    ],
+  },
+];
+
 const CHAT_PROMPT: Omit<Prompt, 'name'> = {
   model: 'claude-sonnet-4@20250514',
   optionalModels: [
@@ -1770,11 +1956,75 @@ const chat: Prompt[] = [
   },
 ];
 
+const artifactActions: Prompt[] = [
+  {
+    name: 'Code Artifact',
+    model: 'claude-sonnet-4@20250514',
+    messages: [
+      {
+        role: 'system',
+        content: `
+        When sent new notes, respond ONLY with the contents of the html file.
+        DO NOT INCLUDE ANY OTHER TEXT, EXPLANATIONS, APOLOGIES, OR INTRODUCTORY/CLOSING PHRASES.
+        IF USER DOES NOT SPECIFY A STYLE, FOLLOW THE DEFAULT STYLE.
+        <generate_guide>
+        - The results should be a single HTML file.
+        - Use tailwindcss to style the website
+        - Put any additional CSS styles in a style tag and any JavaScript in a script tag.
+        - Use unpkg or skypack to import any required dependencies.
+        - Use Google fonts to pull in any open source fonts you require.
+        - Use lucide icons for any icons.
+        - If you have any images, load them from Unsplash or use solid colored rectangles.
+        </generate_guide>
+        
+        <DO_NOT_USE_COLORS>
+        - DO NOT USE ANY COLORS
+        </DO_NOT_USE_COLORS>
+        <DO_NOT_USE_GRADIENTS>
+        - DO NOT USE ANY GRADIENTS
+        </DO_NOT_USE_GRADIENTS>
+        
+        <COLOR_THEME>
+          - --affine-blue-300: #93e2fd
+          - --affine-blue-400: #60cffa
+          - --affine-blue-500: #3ab5f7
+          - --affine-blue-600: #1e96eb
+          - --affine-blue-700: #1e67af
+          - --affine-text-primary-color: #121212
+          - --affine-text-secondary-color: #8e8d91
+          - --affine-text-disable-color: #a9a9ad
+          - --affine-background-overlay-panel-color: #fbfbfc
+          - --affine-background-secondary-color: #f4f4f5
+          - --affine-background-primary-color: #fff
+        </COLOR_THEME>
+        <default_style_guide>
+        - MUST USE White and Blue(#1e96eb) as the primary color
+        - KEEP THE DEFAULT STYLE SIMPLE AND CLEAN
+        - DO NOT USE ANY COMPLEX STYLES
+        - DO NOT USE ANY GRADIENTS
+        - USE LESS SHADOWS
+        - USE RADIUS 4px or 8px for rounded corners
+        - USE 12px or 16px for padding
+        - Use the tailwind color gray, zinc, slate, neutral much more.
+        - Use 0.5px border should be better 
+        </default_style_guide>
+        `,
+      },
+      {
+        role: 'user',
+        content: '{{content}}',
+      },
+    ],
+  },
+];
+
 export const prompts: Prompt[] = [
   ...textActions,
   ...imageActions,
+  ...modelActions,
   ...chat,
   ...workflows,
+  ...artifactActions,
 ];
 
 export async function refreshPrompts(db: PrismaClient) {

packages/backend/server/src/plugins/copilot/providers/morph.ts

@@ -37,6 +37,24 @@ export class MorphProvider extends CopilotProvider<MorphConfig> {
         },
       ],
     },
+    {
+      id: 'morph-v3-fast',
+      capabilities: [
+        {
+          input: [ModelInputType.Text],
+          output: [ModelOutputType.Text],
+        },
+      ],
+    },
+    {
+      id: 'morph-v3-large',
+      capabilities: [
+        {
+          input: [ModelInputType.Text],
+          output: [ModelOutputType.Text],
+        },
+      ],
+    },
   ];
 
   #instance!: VercelOpenAICompatibleProvider;

packages/backend/server/src/plugins/copilot/providers/provider.ts

@@ -21,6 +21,7 @@ import {
   buildDocKeywordSearchGetter,
   buildDocSearchGetter,
   createCodeArtifactTool,
+  createConversationSummaryTool,
   createDocComposeTool,
   createDocEditTool,
   createDocKeywordSearchTool,
@@ -139,6 +140,11 @@ export abstract class CopilotProvider<C = any> {
     if (options?.tools?.length) {
       this.logger.debug(`getTools: ${JSON.stringify(options.tools)}`);
       const ac = this.moduleRef.get(AccessController, { strict: false });
+      const docReader = this.moduleRef.get(DocReader, { strict: false });
+      const models = this.moduleRef.get(Models, { strict: false });
+      const prompt = this.moduleRef.get(PromptService, {
+        strict: false,
+      });
 
       for (const tool of options.tools) {
         const toolDef = this.getProviderSpecificTools(tool, model);
@@ -150,11 +156,23 @@ export abstract class CopilotProvider<C = any> {
           continue;
         }
         switch (tool) {
+          case 'codeArtifact': {
+            tools.code_artifact = createCodeArtifactTool(prompt, this.factory);
+            break;
+          }
+          case 'conversationSummary': {
+            tools.conversation_summary = createConversationSummaryTool(
+              options.session,
+              prompt,
+              this.factory
+            );
+            break;
+          }
           case 'docEdit': {
-            const doc = this.moduleRef.get(DocReader, { strict: false });
-            const getDocContent = buildContentGetter(ac, doc);
+            const getDocContent = buildContentGetter(ac, docReader);
             tools.doc_edit = createDocEditTool(
               this.factory,
+              prompt,
               getDocContent.bind(null, options)
             );
             break;
@@ -163,11 +181,15 @@ export abstract class CopilotProvider<C = any> {
             const context = this.moduleRef.get(CopilotContextService, {
               strict: false,
             });
-
             const docContext = options.session
               ? await context.getBySessionId(options.session)
               : null;
-            const searchDocs = buildDocSearchGetter(ac, context, docContext);
+            const searchDocs = buildDocSearchGetter(
+              ac,
+              context,
+              docContext,
+              models
+            );
             tools.doc_semantic_search = createDocSemanticSearchTool(
               searchDocs.bind(null, options)
             );
@@ -175,9 +197,6 @@ export abstract class CopilotProvider<C = any> {
           }
           case 'docKeywordSearch': {
             if (this.AFFiNEConfig.indexer.enabled) {
-              const ac = this.moduleRef.get(AccessController, {
-                strict: false,
-              });
               const indexerService = this.moduleRef.get(IndexerService, {
                 strict: false,
               });
@@ -192,9 +211,6 @@ export abstract class CopilotProvider<C = any> {
             break;
           }
           case 'docRead': {
-            const ac = this.moduleRef.get(AccessController, { strict: false });
-            const models = this.moduleRef.get(Models, { strict: false });
-            const docReader = this.moduleRef.get(DocReader, { strict: false });
             const getDoc = buildDocContentGetter(ac, docReader, models);
             tools.doc_read = createDocReadTool(getDoc.bind(null, options));
             break;
@@ -205,23 +221,7 @@ export abstract class CopilotProvider<C = any> {
             break;
           }
           case 'docCompose': {
-            const promptService = this.moduleRef.get(PromptService, {
-              strict: false,
-            });
-            tools.doc_compose = createDocComposeTool(
-              promptService,
-              this.factory
-            );
-            break;
-          }
-          case 'codeArtifact': {
-            const promptService = this.moduleRef.get(PromptService, {
-              strict: false,
-            });
-            tools.code_artifact = createCodeArtifactTool(
-              promptService,
-              this.factory
-            );
+            tools.doc_compose = createDocComposeTool(prompt, this.factory);
             break;
           }
         }

packages/backend/server/src/plugins/copilot/providers/types.ts

@@ -60,6 +60,8 @@ export const VertexSchema: JSONSchema = {
 export const PromptConfigStrictSchema = z.object({
   tools: z
     .enum([
+      'codeArtifact',
+      'conversationSummary',
       // work with morph
       'docEdit',
       // work with indexer
@@ -71,7 +73,6 @@ export const PromptConfigStrictSchema = z.object({
       'webSearch',
       // artifact tools
       'docCompose',
-      'codeArtifact',
     ])
     .array()
     .nullable()

packages/backend/server/src/plugins/copilot/providers/utils.ts

@@ -6,20 +6,10 @@ import {
   ImagePart,
   TextPart,
   TextStreamPart,
-  ToolSet,
 } from 'ai';
 import { ZodType } from 'zod';
 
-import {
-  createCodeArtifactTool,
-  createDocComposeTool,
-  createDocEditTool,
-  createDocKeywordSearchTool,
-  createDocReadTool,
-  createDocSemanticSearchTool,
-  createExaCrawlTool,
-  createExaSearchTool,
-} from '../tools';
+import { CustomAITools } from '../tools';
 import { PromptMessage, StreamObject } from './types';
 
 type ChatMessage = CoreUserMessage | CoreAssistantMessage;
@@ -385,17 +375,6 @@ export class CitationParser {
   }
 }
 
-export interface CustomAITools extends ToolSet {
-  doc_edit: ReturnType<typeof createDocEditTool>;
-  doc_semantic_search: ReturnType<typeof createDocSemanticSearchTool>;
-  doc_keyword_search: ReturnType<typeof createDocKeywordSearchTool>;
-  doc_read: ReturnType<typeof createDocReadTool>;
-  doc_compose: ReturnType<typeof createDocComposeTool>;
-  web_search_exa: ReturnType<typeof createExaSearchTool>;
-  web_crawl_exa: ReturnType<typeof createExaCrawlTool>;
-  code_artifact: ReturnType<typeof createCodeArtifactTool>;
-}
-
 type ChunkType = TextStreamPart<CustomAITools>['type'];
 
 export function toError(error: unknown): Error {
@@ -451,6 +430,10 @@ export class TextStreamParser {
         );
         result = this.addPrefix(result);
         switch (chunk.toolName) {
+          case 'conversation_summary': {
+            result += `\nSummarizing context\n`;
+            break;
+          }
           case 'web_search_exa': {
             result += `\nSearching the web "${chunk.args.query}"\n`;
             break;
@@ -489,10 +472,18 @@ export class TextStreamParser {
         result = this.addPrefix(result);
         switch (chunk.toolName) {
           case 'doc_edit': {
-            if (chunk.result && typeof chunk.result === 'object') {
-              result += `\n${chunk.result.result}\n`;
+            if (
+              chunk.result &&
+              typeof chunk.result === 'object' &&
+              Array.isArray(chunk.result.result)
+            ) {
+              result += chunk.result.result
+                .map(item => {
+                  return `\n${item.changedContent}\n`;
+                })
+                .join('');
               this.docEditFootnotes[this.docEditFootnotes.length - 1].result =
-                chunk.result.result;
+                result;
             } else {
               this.docEditFootnotes.pop();
             }

packages/backend/server/src/plugins/copilot/resolver.ts

@@ -23,6 +23,7 @@ import {
   CallMetric,
   CopilotDocNotFound,
   CopilotFailedToCreateMessage,
+  CopilotProviderSideError,
   CopilotSessionNotFound,
   type FileUpload,
   paginate,
@@ -31,14 +32,18 @@ import {
   RequestMutex,
   Throttle,
   TooManyRequest,
+  UserFriendlyError,
 } from '../../base';
 import { CurrentUser } from '../../core/auth';
 import { Admin } from '../../core/common';
+import { DocReader } from '../../core/doc';
 import { AccessController } from '../../core/permission';
 import { UserType } from '../../core/user';
 import type { ListSessionOptions, UpdateChatSession } from '../../models';
+import { CopilotCronJobs } from './cron';
 import { PromptService } from './prompt';
 import { PromptMessage, StreamObject } from './providers';
+import { CopilotProviderFactory } from './providers/factory';
 import { ChatSessionService } from './session';
 import { CopilotStorage } from './storage';
 import { type ChatHistory, type ChatMessage, SubmittedMessage } from './types';
@@ -396,7 +401,9 @@ export class CopilotResolver {
     private readonly ac: AccessController,
     private readonly mutex: RequestMutex,
     private readonly chatSession: ChatSessionService,
-    private readonly storage: CopilotStorage
+    private readonly storage: CopilotStorage,
+    private readonly docReader: DocReader,
+    private readonly providerFactory: CopilotProviderFactory
   ) {}
 
   @ResolveField(() => CopilotQuotaType, {
@@ -724,6 +731,65 @@ export class CopilotResolver {
     }
   }
 
+  @Query(() => String, {
+    description:
+      'Apply updates to a doc using LLM and return the merged markdown.',
+  })
+  async applyDocUpdates(
+    @CurrentUser() user: CurrentUser,
+    @Args({ name: 'workspaceId', type: () => String })
+    workspaceId: string,
+    @Args({ name: 'docId', type: () => String })
+    docId: string,
+    @Args({ name: 'op', type: () => String })
+    op: string,
+    @Args({ name: 'updates', type: () => String })
+    updates: string
+  ): Promise<string> {
+    await this.assertPermission(user, { workspaceId, docId });
+
+    const docContent = await this.docReader.getDocMarkdown(
+      workspaceId,
+      docId,
+      true
+    );
+    if (!docContent || !docContent.markdown) {
+      throw new NotFoundException('Doc not found or empty');
+    }
+
+    const markdown = docContent.markdown.trim();
+
+    // Get LLM provider
+    const provider =
+      await this.providerFactory.getProviderByModel('morph-v3-large');
+    if (!provider) {
+      throw new BadRequestException('No LLM provider available');
+    }
+
+    try {
+      return await provider.text(
+        { modelId: 'morph-v3-large' },
+        [
+          {
+            role: 'user',
+            content: `<instruction>${op}</instruction>\n<code>${markdown}</code>\n<update>${updates}</update>`,
+          },
+        ],
+        { reasoning: false }
+      );
+    } catch (e: any) {
+      if (e instanceof UserFriendlyError) {
+        throw e;
+      } else {
+        throw new CopilotProviderSideError({
+          provider: provider.type,
+          kind: 'unexpected_response',
+          message: e?.message || 'Unexpected apply response',
+        });
+      }
+    }
+  }
+
   private transformToSessionType(
     session: Omit<ChatHistory, 'messages'>
   ): CopilotSessionType {
@@ -773,7 +839,18 @@ class CreateCopilotPromptInput {
 @Admin()
 @Resolver(() => String)
 export class PromptsManagementResolver {
-  constructor(private readonly promptService: PromptService) {}
+  constructor(
+    private readonly cron: CopilotCronJobs,
+    private readonly promptService: PromptService
+  ) {}
+
+  @Query(() => Boolean, {
+    description: 'Trigger generate missing titles cron job',
+  })
+  async triggerGenerateTitleCron() {
+    await this.cron.triggerGenerateMissingTitles();
+    return true;
+  }
 
   @Query(() => [CopilotPromptType], {
     description: 'List all copilot prompts',

packages/backend/server/src/plugins/copilot/session.ts

@@ -507,6 +507,8 @@ export class ChatSessionService {
     return await this.models.copilotSession.fork({
       ...session,
       userId: options.userId,
+      // docId can be changed in fork
+      docId: options.docId,
       sessionId: randomUUID(),
       parentSessionId: options.sessionId,
       messages,
@@ -569,7 +571,7 @@ export class ChatSessionService {
       });
 
     if (!provider) {
-      throw new NoCopilotProviderAvailable();
+      throw new NoCopilotProviderAvailable({ modelId: prompt.model });
     }
 
     return provider.text(cond, [...prompt.finish({}), msg], config);

packages/backend/server/src/plugins/copilot/tools/code-artifact.ts

@@ -5,9 +5,7 @@ import { z } from 'zod';
 import type { PromptService } from '../prompt';
 import type { CopilotProviderFactory } from '../providers';
 import { toolError } from './error';
-
 const logger = new Logger('CodeArtifactTool');
-
 /**
  * A copilot tool that produces a completely self-contained HTML artifact.
  * The returned HTML must include <style> and <script> tags directly so that
@@ -37,23 +35,20 @@ export const createCodeArtifactTool = (
     }),
     execute: async ({ title, userPrompt }) => {
       try {
-        const prompt = await promptService.get('Make it real with text');
+        const prompt = await promptService.get('Code Artifact');
         if (!prompt) {
           throw new Error('Prompt not found');
         }
-
         const provider = await factory.getProviderByModel(prompt.model);
         if (!provider) {
           throw new Error('Provider not found');
         }
-
         const content = await provider.text(
           {
             modelId: prompt.model,
           },
-          [...prompt.finish({}), { role: 'user', content: userPrompt }]
+          prompt.finish({ content: userPrompt })
         );
-
         // Remove surrounding ``` or ```html fences if present
         let stripped = content.trim();
         if (stripped.startsWith('```')) {
@@ -65,7 +60,6 @@ export const createCodeArtifactTool = (
             stripped = stripped.slice(0, -3);
           }
         }
-
         return {
           title,
           html: stripped,

packages/backend/server/src/plugins/copilot/tools/conversation-summary.ts

@@ -0,0 +1,76 @@
+import { Logger } from '@nestjs/common';
+import { tool } from 'ai';
+import { z } from 'zod';
+
+import type { PromptService } from '../prompt';
+import type { CopilotProviderFactory } from '../providers';
+import { toolError } from './error';
+
+const logger = new Logger('ConversationSummaryTool');
+
+export const createConversationSummaryTool = (
+  sessionId: string | undefined,
+  promptService: PromptService,
+  factory: CopilotProviderFactory
+) => {
+  return tool({
+    description:
+      'Create a concise, AI-generated summary of the conversation so far—capturing key topics, decisions, and critical details. Use this tool whenever the context becomes lengthy to preserve essential information that might otherwise be lost to truncation in future turns.',
+    parameters: z.object({
+      focus: z
+        .string()
+        .optional()
+        .describe(
+          'Optional focus area for the summary (e.g., "technical decisions", "user requirements", "project status")'
+        ),
+      length: z
+        .enum(['brief', 'detailed', 'comprehensive'])
+        .default('detailed')
+        .describe(
+          'The desired length of the summary: brief (1-2 sentences), detailed (paragraph), comprehensive (multiple paragraphs)'
+        ),
+    }),
+    execute: async ({ focus, length }, { messages }) => {
+      try {
+        if (!messages || messages.length === 0) {
+          return toolError(
+            'No Conversation Context',
+            'No messages available to summarize'
+          );
+        }
+
+        const prompt = await promptService.get('Conversation Summary');
+        const provider = await factory.getProviderByModel(prompt?.model || '');
+
+        if (!prompt || !provider) {
+          return toolError(
+            'Prompt Not Found',
+            'Failed to summarize conversation.'
+          );
+        }
+
+        const summary = await provider.text(
+          { modelId: prompt.model },
+          prompt.finish({
+            messages: messages.map(m => ({
+              ...m,
+              content: m.content.toString(),
+            })),
+            focus: focus || 'general',
+            length,
+          })
+        );
+
+        return {
+          focusArea: focus || 'general',
+          messageCount: messages.length,
+          summary,
+          timestamp: new Date().toISOString(),
+        };
+      } catch (err: any) {
+        logger.error(`Failed to summarize conversation (${sessionId})`, err);
+        return toolError('Conversation Summary Failed', err.message);
+      }
+    },
+  });
+};

packages/backend/server/src/plugins/copilot/tools/doc-edit.ts

@@ -3,6 +3,7 @@ import { z } from 'zod';
 
 import { DocReader } from '../../../core/doc';
 import { AccessController } from '../../../core/permission';
+import { type PromptService } from '../prompt';
 import type { CopilotChatOptions, CopilotProviderFactory } from '../providers';
 
 export const buildContentGetter = (ac: AccessController, doc: DocReader) => {
@@ -24,14 +25,20 @@ export const buildContentGetter = (ac: AccessController, doc: DocReader) => {
 
 export const createDocEditTool = (
   factory: CopilotProviderFactory,
+  prompt: PromptService,
   getContent: (targetId?: string) => Promise<string | undefined>
 ) => {
   return tool({
     description: `
-Use this tool to propose an edit to a structured Markdown document with identifiable blocks. Each block begins with a comment like <!-- block_id=... -->, and represents a unit of editable content such as a heading, paragraph, list, or code snippet.
+Use this tool to propose an edit to a structured Markdown document with identifiable blocks. 
+Each block begins with a comment like <!-- block_id=... -->, and represents a unit of editable content such as a heading, paragraph, list, or code snippet.
 This will be read by a less intelligent model, which will quickly apply the edit. You should make it clear what the edit is, while also minimizing the unchanged code you write.
 
-Your task is to return a list of block-level changes needed to fulfill the user's intent. Each change should correspond to a specific user instruction and be represented by one of the following operations:
+If you receive a markdown without block_id comments, you should call \`doc_read\` tool to get the content.
+
+Your task is to return a list of block-level changes needed to fulfill the user's intent. **Each change in code_edit must be completely independent: each code_edit entry should only perform a single, isolated change, and must not include the effects of other changes. For example, the updates for a delete operation should only show the context related to the deletion, and must not include any content modified by other operations (such as bolding or insertion). This ensures that each change can be applied independently and in any order.**
+
+Each change should correspond to a specific user instruction and be represented by one of the following operations:
 
 replace: Replace the content of a block with updated Markdown.
 
@@ -41,83 +48,75 @@ insert: Add a new block, and specify its block_id and content.
 
 Important Instructions:
 - Use the existing block structure as-is. Do not reformat or reorder blocks unless explicitly asked.
-- Always preserve block_id and type in your replacements.
-- When replacing a block, use the full new block including <!-- block_id=... type=... --> and the updated content.
 - When inserting, follow the same format as a replacement, but ensure the new block_id does not conflict with existing IDs.
+- When replacing content, always keep the original block_id unchanged.
+- When deleting content, only use the format <!-- delete block_id=xxx -->, and only for valid block_id present in the original <code> content.
 - Each list item should be a block.
-- Use <!-- existing blocks ... --> for unchanged sections.
-- If you plan on deleting a section, you must provide surrounding context to indicate the deletion.
+- Your task is to return a list of block-level changes needed to fulfill the user's intent. 
+- **Each change in code_edit must be completely independent: each code_edit entry should only perform a single, isolated change, and must not include the effects of other changes. For example, the updates for a delete operation should only show the context related to the deletion, and must not include any content modified by other operations (such as bolding or insertion). This ensures that each change can be applied independently and in any order.**
 
-Example Input Document:
-\`\`\`md
-<!-- block_id=block-001 type=paragraph -->
-# My Holiday Plan
+Original Content:
+\`\`\`markdown
+<!-- block_id=001 flavour=paragraph -->
+# Andriy Shevchenko
 
-<!-- block_id=block-002 type=paragraph -->
-I plan to travel to Paris, France, where I will visit the Eiffel Tower, the Louvre, and the Champs-Élysées.
+<!-- block_id=002 flavour=paragraph -->
+## Player Profile
 
-<!-- block_id=block-003 type=paragraph -->
-I love Paris.
+<!-- block_id=003 flavour=paragraph -->
+Andriy Shevchenko is a legendary Ukrainian striker, best known for his time at AC Milan and Dynamo Kyiv. He won the Ballon d'Or in 2004.
 
-<!-- block_id=block-004 type=paragraph -->
-## Reason for the delay
+<!-- block_id=004 flavour=paragraph -->
+## Career Overview
 
-<!-- block_id=block-005 type=paragraph -->
-This plan has been brewing for a long time, but I always postponed it because I was too busy with work.
-
-<!-- block_id=block-006 type=paragraph -->
-## Trip Steps
-
-<!-- block_id=block-007 type=list -->
-- Book flight tickets
-<!-- block_id=block-008 type=list -->
-- Reserve a hotel
-<!-- block_id=block-009 type=list -->
-- Prepare visa documents
-<!-- block_id=block-010 type=list -->
-- Plan the itinerary
-
-<!-- block_id=block-011 type=paragraph -->
-Additionally, I plan to learn some basic French to make communication easier during the trip.
+<!-- block_id=005 flavour=list -->
+- Born in 1976 in Ukraine.
+<!-- block_id=006 flavour=list -->
+- Rose to fame at Dynamo Kyiv in the 1990s.
+<!-- block_id=007 flavour=list -->
+- Starred at AC Milan (1999–2006), scoring over 170 goals.
+<!-- block_id=008 flavour=list -->
+- Played for Chelsea (2006–2009) before returning to Kyiv.
+<!-- block_id=009 flavour=list -->
+- Coached Ukraine national team, reaching Euro 2020 quarter-finals.
 \`\`\`
 
-Example User Request:
-
+User Request：
 \`\`\`
-Translate the trip steps to Chinese, remove the reason for the delay, and bold the final paragraph.
+Bold the player’s name in the intro, add a summary section at the end, and remove the career overview.
 \`\`\`
 
-Expected Output:
-
-\`\`\`md
-<!-- existing blocks ... -->
-
-<!-- block_id=block-002 type=paragraph -->
-I plan to travel to Paris, France, where I will visit the Eiffel Tower, the Louvre, and the Champs-Élysées.
-
-<!-- block_id=block-003 type=paragraph -->
-I love Paris.
-
-<!-- delete block-004 -->
-
-<!-- delete block-005 -->
-
-<!-- block_id=block-006 type=paragraph -->
-## Trip Steps
-
-<!-- block_id=block-007 type=list -->
-- 订机票
-<!-- block_id=block-008 type=list -->
-- 预定酒店
-<!-- block_id=block-009 type=list -->
-- 准备签证材料
-<!-- block_id=block-010 type=list -->
-- 规划行程
-
-<!-- existing blocks ... -->
-
-<!-- block_id=block-011 type=paragraph -->
-**Additionally, I plan to learn some basic French to make communication easier during the trip.**
+Example response:
+\`\`\`json
+[
+  {
+    "op": "Bold the player's name in the introduction",
+    "updates": "
+      <!-- block_id=003 flavour=paragraph -->
+      **Andriy Shevchenko** is a legendary Ukrainian striker, best known for his time at AC Milan and Dynamo Kyiv. He won the Ballon d'Or in 2004.
+    "
+  },
+  {
+    "op": "Add a summary section at the end",
+    "updates": "
+      <!-- block_id=new-abc123 flavour=paragraph -->
+      ## Summary
+      <!-- block_id=new-def456 flavour=paragraph -->
+      Shevchenko is celebrated as one of the greatest Ukrainian footballers of all time. Known for his composure, strength, and goal-scoring instinct, he left a lasting legacy both on and off the pitch.
+    "
+  },
+  {
+    "op": "Delete the career overview section",
+    "updates": "
+      <!-- delete block_id=004 -->
+      <!-- delete block_id=005 -->
+      <!-- delete block_id=006 -->
+      <!-- delete block_id=007 -->
+      <!-- delete block_id=008 -->
+      <!-- delete block_id=009 -->
+    "
+  }
+]
 \`\`\`
 You should specify the following arguments before the others: [doc_id], [origin_content]
 
@@ -144,14 +143,32 @@ You should specify the following arguments before the others: [doc_id], [origin_
         ),
 
       code_edit: z
-        .string()
+        .array(
+          z.object({
+            op: z
+              .string()
+              .describe(
+                'A short description of the change, such as "Bold intro name"'
+              ),
+            updates: z
+              .string()
+              .describe(
+                'Markdown block fragments that represent the change, including the block_id and type'
+              ),
+          })
+        )
         .describe(
-          'Specify only the necessary Markdown block-level changes. Return a list of inserted, replaced, or deleted blocks. Each block must start with its <!-- block_id=... type=... --> comment. Use <!-- existing blocks ... --> for unchanged sections.If you plan on deleting a section, you must provide surrounding context to indicate the deletion.'
+          'An array of independent semantic changes to apply to the document.'
         ),
     }),
     execute: async ({ doc_id, origin_content, code_edit }) => {
       try {
-        const provider = await factory.getProviderByModel('morph-v2');
+        const applyPrompt = await prompt.get('Apply Updates');
+        if (!applyPrompt) {
+          return 'Prompt not found';
+        }
+        const model = applyPrompt.model;
+        const provider = await factory.getProviderByModel(model);
         if (!provider) {
           return 'Editing docs is not supported';
         }
@@ -160,14 +177,27 @@ You should specify the following arguments before the others: [doc_id], [origin_
         if (!content) {
           return 'Doc not found or doc is empty';
         }
-        const result = await provider.text({ modelId: 'morph-v2' }, [
-          {
-            role: 'user',
-            content: `<code>${content}</code>\n<update>${code_edit}</update>`,
-          },
-        ]);
 
-        return { result, content };
+        const changedContents = await Promise.all(
+          code_edit.map(async edit => {
+            return await provider.text({ modelId: model }, [
+              ...applyPrompt.finish({
+                content,
+                op: edit.op,
+                updates: edit.updates,
+              }),
+            ]);
+          })
+        );
+
+        return {
+          result: changedContents.map((changedContent, index) => ({
+            op: code_edit[index].op,
+            updates: code_edit[index].updates,
+            originalContent: content,
+            changedContent,
+          })),
+        };
       } catch {
         return 'Failed to apply edit to the doc';
       }

packages/backend/server/src/plugins/copilot/tools/doc-semantic-search.ts

@@ -2,7 +2,7 @@ import { tool } from 'ai';
 import { z } from 'zod';
 
 import type { AccessController } from '../../../core/permission';
-import type { ChunkSimilarity } from '../../../models';
+import type { ChunkSimilarity, Models } from '../../../models';
 import type { CopilotContextService } from '../context';
 import type { ContextSession } from '../context/session';
 import type { CopilotChatOptions } from '../providers';
@@ -11,7 +11,8 @@ import { toolError } from './error';
 export const buildDocSearchGetter = (
   ac: AccessController,
   context: CopilotContextService,
-  docContext: ContextSession | null
+  docContext: ContextSession | null,
+  models: Models
 ) => {
   const searchDocs = async (
     options: CopilotChatOptions,
@@ -45,7 +46,24 @@ export const buildDocSearchGetter = (
     }
     if (!docChunks.length && !fileChunks.length)
       return `No results found for "${query}".`;
-    return [...fileChunks, ...docChunks];
+
+    const docMetas = await models.doc
+      .findAuthors(
+        docChunks.map(c => ({
+          // oxlint-disable-next-line no-non-null-assertion
+          workspaceId: options.workspace!,
+          docId: c.docId,
+        }))
+      )
+      .then(docs => new Map(docs.filter(d => !!d).map(doc => [doc.id, doc])));
+
+    return [
+      ...fileChunks,
+      ...docChunks.map(c => ({
+        ...c,
+        ...docMetas.get(c.docId),
+      })),
+    ] as ChunkSimilarity[];
   };
   return searchDocs;
 };

packages/backend/server/src/plugins/copilot/tools/index.ts

@@ -1,4 +1,29 @@
+import { ToolSet } from 'ai';
+
+import { createCodeArtifactTool } from './code-artifact';
+import { createConversationSummaryTool } from './conversation-summary';
+import { createDocComposeTool } from './doc-compose';
+import { createDocEditTool } from './doc-edit';
+import { createDocKeywordSearchTool } from './doc-keyword-search';
+import { createDocReadTool } from './doc-read';
+import { createDocSemanticSearchTool } from './doc-semantic-search';
+import { createExaCrawlTool } from './exa-crawl';
+import { createExaSearchTool } from './exa-search';
+
+export interface CustomAITools extends ToolSet {
+  code_artifact: ReturnType<typeof createCodeArtifactTool>;
+  conversation_summary: ReturnType<typeof createConversationSummaryTool>;
+  doc_edit: ReturnType<typeof createDocEditTool>;
+  doc_semantic_search: ReturnType<typeof createDocSemanticSearchTool>;
+  doc_keyword_search: ReturnType<typeof createDocKeywordSearchTool>;
+  doc_read: ReturnType<typeof createDocReadTool>;
+  doc_compose: ReturnType<typeof createDocComposeTool>;
+  web_search_exa: ReturnType<typeof createExaSearchTool>;
+  web_crawl_exa: ReturnType<typeof createExaCrawlTool>;
+}
+
 export * from './code-artifact';
+export * from './conversation-summary';
 export * from './doc-compose';
 export * from './doc-edit';
 export * from './doc-keyword-search';

packages/backend/server/src/plugins/copilot/transcript/service.ts

@@ -171,7 +171,7 @@ export class CopilotTranscriptionService {
     );
 
     if (!provider) {
-      throw new NoCopilotProviderAvailable();
+      throw new NoCopilotProviderAvailable({ modelId });
     }
 
     return provider;

packages/backend/server/src/plugins/indexer/providers/elasticsearch.ts

@@ -140,10 +140,13 @@ export class ElasticsearchProvider extends SearchProvider {
     const result = await this.request(
       'POST',
       url.toString(),
-      JSON.stringify({ query })
+      JSON.stringify({ query }),
+      'application/json',
+      // ignore 409 error: version_conflict_engine_exception, version conflict, required seqNo [255898790], primary term [3]. current document has seqNo [256133002] and primary term [3]
+      [409]
     );
     this.logger.debug(
-      `deleted by query ${table} ${JSON.stringify(query)} in ${Date.now() - start}ms, result: ${JSON.stringify(result)}`
+      `deleted by query ${table} ${JSON.stringify(query)} in ${Date.now() - start}ms, result: ${JSON.stringify(result).substring(0, 500)}`
     );
   }
 
@@ -264,7 +267,8 @@ export class ElasticsearchProvider extends SearchProvider {
     method: 'POST' | 'PUT',
     url: string,
     body: string,
-    contentType = 'application/json'
+    contentType = 'application/json',
+    ignoreErrorStatus?: number[]
   ) {
     const headers = {
       'Content-Type': contentType,
@@ -280,6 +284,10 @@ export class ElasticsearchProvider extends SearchProvider {
       headers,
     });
     const data = await response.json();
+    if (ignoreErrorStatus?.includes(response.status)) {
+      return data;
+    }
+
     // handle error, status >= 400
     // {
     //   "error": {

packages/backend/server/src/schema.gql

@@ -291,6 +291,11 @@ type CopilotFailedToAddWorkspaceFileEmbeddingDataType {
   message: String!
 }
 
+type CopilotFailedToGenerateEmbeddingDataType {
+  message: String!
+  provider: String!
+}
+
 type CopilotFailedToMatchContextDataType {
   content: String!
   contextId: String!
@@ -595,6 +600,7 @@ type DocType {
   mode: PublicDocMode!
   permissions: DocPermissions!
   public: Boolean!
+  summary: String
   title: String
   updatedAt: DateTime
   workspaceId: String!
@@ -615,7 +621,7 @@ type EditorType {
   name: String!
 }
 
-union ErrorDataUnion = AlreadyInSpaceDataType | BlobNotFoundDataType | CopilotContextFileNotSupportedDataType | CopilotDocNotFoundDataType | CopilotFailedToAddWorkspaceFileEmbeddingDataType | CopilotFailedToMatchContextDataType | CopilotFailedToMatchGlobalContextDataType | CopilotFailedToModifyContextDataType | CopilotInvalidContextDataType | CopilotMessageNotFoundDataType | CopilotPromptNotFoundDataType | CopilotProviderNotSupportedDataType | CopilotProviderSideErrorDataType | DocActionDeniedDataType | DocHistoryNotFoundDataType | DocNotFoundDataType | DocUpdateBlockedDataType | ExpectToGrantDocUserRolesDataType | ExpectToRevokeDocUserRolesDataType | ExpectToUpdateDocUserRoleDataType | GraphqlBadRequestDataType | HttpRequestErrorDataType | InvalidAppConfigDataType | InvalidAppConfigInputDataType | InvalidEmailDataType | InvalidHistoryTimestampDataType | InvalidIndexerInputDataType | InvalidLicenseToActivateDataType | InvalidLicenseUpdateParamsDataType | InvalidOauthCallbackCodeDataType | InvalidOauthResponseDataType | InvalidPasswordLengthDataType | InvalidRuntimeConfigTypeDataType | InvalidSearchProviderRequestDataType | MemberNotFoundInSpaceDataType | MentionUserDocAccessDeniedDataType | MissingOauthQueryParameterDataType | NoMoreSeatDataType | NotInSpaceDataType | QueryTooLongDataType | RuntimeConfigNotFoundDataType | SameSubscriptionRecurringDataType | SpaceAccessDeniedDataType | SpaceNotFoundDataType | SpaceOwnerNotFoundDataType | SpaceShouldHaveOnlyOneOwnerDataType | SubscriptionAlreadyExistsDataType | SubscriptionNotExistsDataType | SubscriptionPlanNotFoundDataType | UnknownOauthProviderDataType | UnsupportedClientVersionDataType | UnsupportedSubscriptionPlanDataType | ValidationErrorDataType | VersionRejectedDataType | WorkspacePermissionNotFoundDataType | WrongSignInCredentialsDataType
+union ErrorDataUnion = AlreadyInSpaceDataType | BlobNotFoundDataType | CopilotContextFileNotSupportedDataType | CopilotDocNotFoundDataType | CopilotFailedToAddWorkspaceFileEmbeddingDataType | CopilotFailedToGenerateEmbeddingDataType | CopilotFailedToMatchContextDataType | CopilotFailedToMatchGlobalContextDataType | CopilotFailedToModifyContextDataType | CopilotInvalidContextDataType | CopilotMessageNotFoundDataType | CopilotPromptNotFoundDataType | CopilotProviderNotSupportedDataType | CopilotProviderSideErrorDataType | DocActionDeniedDataType | DocHistoryNotFoundDataType | DocNotFoundDataType | DocUpdateBlockedDataType | ExpectToGrantDocUserRolesDataType | ExpectToRevokeDocUserRolesDataType | ExpectToUpdateDocUserRoleDataType | GraphqlBadRequestDataType | HttpRequestErrorDataType | InvalidAppConfigDataType | InvalidAppConfigInputDataType | InvalidEmailDataType | InvalidHistoryTimestampDataType | InvalidIndexerInputDataType | InvalidLicenseToActivateDataType | InvalidLicenseUpdateParamsDataType | InvalidOauthCallbackCodeDataType | InvalidOauthResponseDataType | InvalidPasswordLengthDataType | InvalidRuntimeConfigTypeDataType | InvalidSearchProviderRequestDataType | MemberNotFoundInSpaceDataType | MentionUserDocAccessDeniedDataType | MissingOauthQueryParameterDataType | NoCopilotProviderAvailableDataType | NoMoreSeatDataType | NotInSpaceDataType | QueryTooLongDataType | RuntimeConfigNotFoundDataType | SameSubscriptionRecurringDataType | SpaceAccessDeniedDataType | SpaceNotFoundDataType | SpaceOwnerNotFoundDataType | SpaceShouldHaveOnlyOneOwnerDataType | SubscriptionAlreadyExistsDataType | SubscriptionNotExistsDataType | SubscriptionPlanNotFoundDataType | UnknownOauthProviderDataType | UnsupportedClientVersionDataType | UnsupportedSubscriptionPlanDataType | ValidationErrorDataType | VersionRejectedDataType | WorkspacePermissionNotFoundDataType | WrongSignInCredentialsDataType
 
 enum ErrorNames {
   ACCESS_DENIED
@@ -644,6 +650,7 @@ enum ErrorNames {
   COPILOT_EMBEDDING_UNAVAILABLE
   COPILOT_FAILED_TO_ADD_WORKSPACE_FILE_EMBEDDING
   COPILOT_FAILED_TO_CREATE_MESSAGE
+  COPILOT_FAILED_TO_GENERATE_EMBEDDING
   COPILOT_FAILED_TO_GENERATE_TEXT
   COPILOT_FAILED_TO_MATCH_CONTEXT
   COPILOT_FAILED_TO_MATCH_GLOBAL_CONTEXT
@@ -1335,6 +1342,10 @@ type Mutation {
   verifyEmail(token: String!): Boolean!
 }
 
+type NoCopilotProviderAvailableDataType {
+  modelId: String!
+}
+
 type NoMoreSeatDataType {
   spaceId: String!
 }
@@ -1507,6 +1518,9 @@ type PublicUserType {
 type Query {
   """get the whole app configuration"""
   appConfig: JSONObject!
+
+  """Apply updates to a doc using LLM and return the merged markdown."""
+  applyDocUpdates(docId: String!, op: String!, updates: String!, workspaceId: String!): String!
   collectAllBlobSizes: WorkspaceBlobSizes! @deprecated(reason: "use `user.quotaUsage` instead")
 
   """Get current user"""

packages/common/graphql/src/fetcher.ts

@@ -82,6 +82,10 @@ export type RequestOptions<Q extends GraphQLQuery> = QueryVariablesOption<Q> & {
    * @default 15000
    */
   timeout?: number;
+  /**
+   * Abort signal
+   */
+  signal?: AbortSignal;
 };
 
 export type QueryOptions<Q extends GraphQLQuery> = RequestOptions<Q> & {
@@ -207,6 +211,7 @@ export const gqlFetcherFactory = (
         headers,
         body: isFormData ? body : JSON.stringify(body),
         timeout: options.timeout,
+        signal: options.signal,
       })
     ).then(async res => {
       if (res.headers.get('content-type')?.startsWith('application/json')) {

packages/common/graphql/src/graphql/copilot-apply-doc-updates.gql

@@ -0,0 +1,3 @@
+query applyDocUpdates($workspaceId: String!, $docId: String!, $op: String!, $updates: String!) {
+  applyDocUpdates(workspaceId: $workspaceId, docId: $docId, op: $op, updates: $updates)
+} 

packages/common/graphql/src/graphql/copilot-session-list-recent.gql

@@ -3,15 +3,17 @@
 query getCopilotRecentSessions(
   $workspaceId: String!
   $limit: Int = 10
+  $offset: Int = 0
 ) {
   currentUser {
     copilot(workspaceId: $workspaceId) {
       chats(
-        pagination: { first: $limit }
+        pagination: { first: $limit, offset: $offset }
         options: {
+          action: false
           fork: false
           sessionOrder: desc
-          withMessages: true
+          withMessages: false
         }
       ) {
         ...PaginatedCopilotChats

packages/common/graphql/src/graphql/get-workspace-page-by-id.gql

@@ -5,6 +5,8 @@ query getWorkspacePageById($workspaceId: String!, $pageId: String!) {
       mode
       defaultRole
       public
+      title
+      summary
     }
   }
 }

packages/common/graphql/src/graphql/index.ts

@@ -555,6 +555,19 @@ export const uploadCommentAttachmentMutation = {
   file: true,
 };
 
+export const applyDocUpdatesQuery = {
+  id: 'applyDocUpdatesQuery' as const,
+  op: 'applyDocUpdates',
+  query: `query applyDocUpdates($workspaceId: String!, $docId: String!, $op: String!, $updates: String!) {
+  applyDocUpdates(
+    workspaceId: $workspaceId
+    docId: $docId
+    op: $op
+    updates: $updates
+  )
+}`,
+};
+
 export const addContextCategoryMutation = {
   id: 'addContextCategoryMutation' as const,
   op: 'addContextCategory',
@@ -1068,12 +1081,12 @@ ${paginatedCopilotChatsFragment}`,
 export const getCopilotRecentSessionsQuery = {
   id: 'getCopilotRecentSessionsQuery' as const,
   op: 'getCopilotRecentSessions',
-  query: `query getCopilotRecentSessions($workspaceId: String!, $limit: Int = 10) {
+  query: `query getCopilotRecentSessions($workspaceId: String!, $limit: Int = 10, $offset: Int = 0) {
   currentUser {
     copilot(workspaceId: $workspaceId) {
       chats(
-        pagination: {first: $limit}
-        options: {fork: false, sessionOrder: desc, withMessages: true}
+        pagination: {first: $limit, offset: $offset}
+        options: {action: false, fork: false, sessionOrder: desc, withMessages: false}
       ) {
         ...PaginatedCopilotChats
       }
@@ -1584,6 +1597,8 @@ export const getWorkspacePageByIdQuery = {
       mode
       defaultRole
       public
+      title
+      summary
     }
   }
 }`,

packages/common/graphql/src/schema.ts

@@ -375,6 +375,12 @@ export interface CopilotFailedToAddWorkspaceFileEmbeddingDataType {
   message: Scalars['String']['output'];
 }
 
+export interface CopilotFailedToGenerateEmbeddingDataType {
+  __typename?: 'CopilotFailedToGenerateEmbeddingDataType';
+  message: Scalars['String']['output'];
+  provider: Scalars['String']['output'];
+}
+
 export interface CopilotFailedToMatchContextDataType {
   __typename?: 'CopilotFailedToMatchContextDataType';
   content: Scalars['String']['output'];
@@ -703,6 +709,7 @@ export interface DocType {
   mode: PublicDocMode;
   permissions: DocPermissions;
   public: Scalars['Boolean']['output'];
+  summary: Maybe<Scalars['String']['output']>;
   title: Maybe<Scalars['String']['output']>;
   updatedAt: Maybe<Scalars['DateTime']['output']>;
   workspaceId: Scalars['String']['output'];
@@ -736,6 +743,7 @@ export type ErrorDataUnion =
   | CopilotContextFileNotSupportedDataType
   | CopilotDocNotFoundDataType
   | CopilotFailedToAddWorkspaceFileEmbeddingDataType
+  | CopilotFailedToGenerateEmbeddingDataType
   | CopilotFailedToMatchContextDataType
   | CopilotFailedToMatchGlobalContextDataType
   | CopilotFailedToModifyContextDataType
@@ -768,6 +776,7 @@ export type ErrorDataUnion =
   | MemberNotFoundInSpaceDataType
   | MentionUserDocAccessDeniedDataType
   | MissingOauthQueryParameterDataType
+  | NoCopilotProviderAvailableDataType
   | NoMoreSeatDataType
   | NotInSpaceDataType
   | QueryTooLongDataType
@@ -815,6 +824,7 @@ export enum ErrorNames {
   COPILOT_EMBEDDING_UNAVAILABLE = 'COPILOT_EMBEDDING_UNAVAILABLE',
   COPILOT_FAILED_TO_ADD_WORKSPACE_FILE_EMBEDDING = 'COPILOT_FAILED_TO_ADD_WORKSPACE_FILE_EMBEDDING',
   COPILOT_FAILED_TO_CREATE_MESSAGE = 'COPILOT_FAILED_TO_CREATE_MESSAGE',
+  COPILOT_FAILED_TO_GENERATE_EMBEDDING = 'COPILOT_FAILED_TO_GENERATE_EMBEDDING',
   COPILOT_FAILED_TO_GENERATE_TEXT = 'COPILOT_FAILED_TO_GENERATE_TEXT',
   COPILOT_FAILED_TO_MATCH_CONTEXT = 'COPILOT_FAILED_TO_MATCH_CONTEXT',
   COPILOT_FAILED_TO_MATCH_GLOBAL_CONTEXT = 'COPILOT_FAILED_TO_MATCH_GLOBAL_CONTEXT',
@@ -1880,6 +1890,11 @@ export interface MutationVerifyEmailArgs {
   token: Scalars['String']['input'];
 }
 
+export interface NoCopilotProviderAvailableDataType {
+  __typename?: 'NoCopilotProviderAvailableDataType';
+  modelId: Scalars['String']['output'];
+}
+
 export interface NoMoreSeatDataType {
   __typename?: 'NoMoreSeatDataType';
   spaceId: Scalars['String']['output'];
@@ -2058,6 +2073,8 @@ export interface Query {
   __typename?: 'Query';
   /** get the whole app configuration */
   appConfig: Scalars['JSONObject']['output'];
+  /** Apply updates to a doc using LLM and return the merged markdown. */
+  applyDocUpdates: Scalars['String']['output'];
   /** @deprecated use `user.quotaUsage` instead */
   collectAllBlobSizes: WorkspaceBlobSizes;
   /** Get current user */
@@ -2105,6 +2122,13 @@ export interface Query {
   workspaces: Array<WorkspaceType>;
 }
 
+export interface QueryApplyDocUpdatesArgs {
+  docId: Scalars['String']['input'];
+  op: Scalars['String']['input'];
+  updates: Scalars['String']['input'];
+  workspaceId: Scalars['String']['input'];
+}
+
 export interface QueryErrorArgs {
   name: ErrorNames;
 }
@@ -3494,6 +3518,18 @@ export type UploadCommentAttachmentMutation = {
   uploadCommentAttachment: string;
 };
 
+export type ApplyDocUpdatesQueryVariables = Exact<{
+  workspaceId: Scalars['String']['input'];
+  docId: Scalars['String']['input'];
+  op: Scalars['String']['input'];
+  updates: Scalars['String']['input'];
+}>;
+
+export type ApplyDocUpdatesQuery = {
+  __typename?: 'Query';
+  applyDocUpdates: string;
+};
+
 export type AddContextCategoryMutationVariables = Exact<{
   options: AddContextCategoryInput;
 }>;
@@ -4350,6 +4386,7 @@ export type GetCopilotSessionQuery = {
 export type GetCopilotRecentSessionsQueryVariables = Exact<{
   workspaceId: Scalars['String']['input'];
   limit?: InputMaybe<Scalars['Int']['input']>;
+  offset?: InputMaybe<Scalars['Int']['input']>;
 }>;
 
 export type GetCopilotRecentSessionsQuery = {
@@ -5147,6 +5184,8 @@ export type GetWorkspacePageByIdQuery = {
       mode: PublicDocMode;
       defaultRole: DocRole;
       public: boolean;
+      title: string | null;
+      summary: string | null;
     };
   };
 };
@@ -6130,6 +6169,11 @@ export type Queries =
       variables: ListCommentsQueryVariables;
       response: ListCommentsQuery;
     }
+  | {
+      name: 'applyDocUpdatesQuery';
+      variables: ApplyDocUpdatesQueryVariables;
+      response: ApplyDocUpdatesQuery;
+    }
   | {
       name: 'listContextObjectQuery';
       variables: ListContextObjectQueryVariables;

packages/frontend/apps/ios/App/App.xcodeproj/project.pbxproj

@@ -541,7 +541,7 @@
 					"$(inherited)",
 					"$(PROJECT_DIR)",
 				);
-				MARKETING_VERSION = 0.22.2;
+				MARKETING_VERSION = 0.23.1;
 				OTHER_SWIFT_FLAGS = "$(inherited) \"-D\" \"COCOAPODS\" \"-DDEBUG\"";
 				PRODUCT_BUNDLE_IDENTIFIER = app.affine.pro;
 				PRODUCT_NAME = "$(TARGET_NAME)";
@@ -577,7 +577,7 @@
 					"$(inherited)",
 					"$(PROJECT_DIR)",
 				);
-				MARKETING_VERSION = 0.22.2;
+				MARKETING_VERSION = 0.23.1;
 				ONLY_ACTIVE_ARCH = NO;
 				PRODUCT_BUNDLE_IDENTIFIER = app.affine.pro;
 				PRODUCT_NAME = "$(TARGET_NAME)";

packages/frontend/component/src/theme/global.css

@@ -6,6 +6,7 @@
 
 :root {
   --noise-background: url(./noise.avif);
+  text-autospace: normal;
 }
 
 html,

packages/frontend/core/src/blocksuite/ai/actions/answer-renderer.ts

@@ -1,5 +1,6 @@
 import type { MindmapElementModel } from '@blocksuite/affine/model';
 import type { EditorHost } from '@blocksuite/affine/std';
+import { ThemeProvider } from '@blocksuite/affine-shared/services';
 
 import { createAIScrollableTextRenderer } from '../components/ai-scrollable-text-renderer';
 import {
@@ -52,5 +53,11 @@ export function actionToAnswerRenderer<
     return createImageRenderer(host, { height: 300 });
   }
 
-  return createAIScrollableTextRenderer(host, {}, 320, true);
+  return createAIScrollableTextRenderer(
+    {
+      theme: host.std.get(ThemeProvider).app$,
+    },
+    320,
+    true
+  );
 }

packages/frontend/core/src/blocksuite/ai/actions/types.ts

@@ -348,6 +348,12 @@ declare global {
         files?: ContextMatchedFileChunk[];
         docs?: ContextMatchedDocChunk[];
       }>;
+      applyDocUpdates: (
+        workspaceId: string,
+        docId: string,
+        op: string,
+        updates: string
+      ) => Promise<string>;
     }
 
     // TODO(@Peng): should be refactored to get rid of implement details (like messages, action, role, etc.)

packages/frontend/core/src/blocksuite/ai/ai-panel.ts

@@ -11,6 +11,7 @@ import {
 } from '@blocksuite/affine/shared/utils';
 import type { EditorHost } from '@blocksuite/affine/std';
 import { GfxControllerIdentifier } from '@blocksuite/affine/std/gfx';
+import { ThemeProvider } from '@blocksuite/affine-shared/services';
 import {
   ChatWithAiIcon,
   DeleteIcon,
@@ -306,7 +307,13 @@ export function buildAIPanelConfig(
   const ctx = new AIContext();
   const searchService = framework.get(AINetworkSearchService);
   return {
-    answerRenderer: createAIScrollableTextRenderer(panel.host, {}, 320, true),
+    answerRenderer: createAIScrollableTextRenderer(
+      {
+        theme: panel.host.std.get(ThemeProvider).app$,
+      },
+      320,
+      true
+    ),
     finishStateConfig: buildFinishConfig(panel, 'chat', ctx),
     generatingStateConfig: buildGeneratingConfig(),
     errorStateConfig: buildErrorConfig(panel),

packages/frontend/core/src/blocksuite/ai/chat-panel/index.ts

@@ -20,16 +20,15 @@ import { property, state } from 'lit/decorators.js';
 import { keyed } from 'lit/directives/keyed.js';
 
 import { AffineIcon } from '../_common/icons';
-import type {
-  DocDisplayConfig,
-  SearchMenuConfig,
-} from '../components/ai-chat-chips';
+import type { SearchMenuConfig } from '../components/ai-chat-add-context';
+import type { DocDisplayConfig } from '../components/ai-chat-chips';
 import type { ChatContextValue } from '../components/ai-chat-content';
 import type {
   AINetworkSearchConfig,
   AIPlaygroundConfig,
   AIReasoningConfig,
 } from '../components/ai-chat-input';
+import type { ChatStatus } from '../components/ai-chat-messages';
 import { createPlaygroundModal } from '../components/playground/modal';
 import { AIProvider } from '../provider';
 import type { AppSidebarConfig } from './chat-config';
@@ -138,6 +137,9 @@ export class ChatPanel extends SignalWatcher(
   @state()
   accessor embeddingProgress: [number, number] = [0, 0];
 
+  @state()
+  accessor status: ChatStatus = 'idle';
+
   private isSidebarOpen: Signal<boolean | undefined> = signal(false);
 
   private sidebarWidth: Signal<number | undefined> = signal(undefined);
@@ -171,6 +173,7 @@ export class ChatPanel extends SignalWatcher(
         .session=${this.session}
         .workspaceId=${this.doc.workspace.id}
         .docId=${this.doc.id}
+        .status=${this.status}
         .onNewSession=${this.newSession}
         .onTogglePin=${this.togglePin}
         .onOpenSession=${this.openSession}
@@ -359,6 +362,7 @@ export class ChatPanel extends SignalWatcher(
   private readonly onContextChange = async (
     context: Partial<ChatContextValue>
   ) => {
+    this.status = context.status ?? 'idle';
     if (context.status === 'success') {
       await this.rebindSession();
     }
@@ -379,6 +383,7 @@ export class ChatPanel extends SignalWatcher(
         .affineFeatureFlagService=${this.affineFeatureFlagService}
         .affineThemeService=${this.affineThemeService}
         .notificationService=${this.notificationService}
+        .affineWorkspaceDialogService=${this.affineWorkspaceDialogService}
       ></playground-content>
     `;
 

packages/frontend/core/src/blocksuite/ai/chat-panel/message/assistant.ts

@@ -3,8 +3,11 @@ import type { AppThemeService } from '@affine/core/modules/theme';
 import type { CopilotChatHistoryFragment } from '@affine/graphql';
 import { WithDisposable } from '@blocksuite/affine/global/lit';
 import { isInsidePageEditor } from '@blocksuite/affine/shared/utils';
-import type { EditorHost } from '@blocksuite/affine/std';
-import { ShadowlessElement } from '@blocksuite/affine/std';
+import {
+  type BlockStdScope,
+  type EditorHost,
+  ShadowlessElement,
+} from '@blocksuite/affine/std';
 import type { ExtensionType } from '@blocksuite/affine/store';
 import type { NotificationService } from '@blocksuite/affine-shared/services';
 import type { Signal } from '@preact/signals-core';
@@ -15,6 +18,7 @@ import {
   EdgelessEditorActions,
   PageEditorActions,
 } from '../../_common/chat-actions-handle';
+import type { DocDisplayConfig } from '../../components/ai-chat-chips';
 import {
   type ChatMessage,
   type ChatStatus,
@@ -37,6 +41,9 @@ export class ChatMessageAssistant extends WithDisposable(ShadowlessElement) {
   @property({ attribute: false })
   accessor host: EditorHost | null | undefined;
 
+  @property({ attribute: false })
+  accessor std: BlockStdScope | null | undefined;
+
   @property({ attribute: false })
   accessor item!: ChatMessage;
 
@@ -73,6 +80,12 @@ export class ChatMessageAssistant extends WithDisposable(ShadowlessElement) {
   @property({ attribute: false })
   accessor notificationService!: NotificationService;
 
+  @property({ attribute: false })
+  accessor independentMode: boolean | undefined;
+
+  @property({ attribute: false })
+  accessor docDisplayService!: DocDisplayConfig;
+
   get state() {
     const { isLast, status } = this;
     return isLast
@@ -124,6 +137,7 @@ export class ChatMessageAssistant extends WithDisposable(ShadowlessElement) {
   private renderStreamObjects(answer: StreamObject[]) {
     return html`<chat-content-stream-objects
       .host=${this.host}
+      .std=${this.std}
       .answer=${answer}
       .state=${this.state}
       .width=${this.width}
@@ -131,6 +145,7 @@ export class ChatMessageAssistant extends WithDisposable(ShadowlessElement) {
       .affineFeatureFlagService=${this.affineFeatureFlagService}
       .notificationService=${this.notificationService}
       .theme=${this.affineThemeService.appTheme.themeSignal}
+      .docDisplayService=${this.docDisplayService}
     ></chat-content-stream-objects>`;
   }
 
@@ -168,7 +183,7 @@ export class ChatMessageAssistant extends WithDisposable(ShadowlessElement) {
         : EdgelessEditorActions
       : null;
 
-    const showActions = host && !!markdown;
+    const showActions = host && !!markdown && !this.independentMode;
 
     return html`
       <chat-copy-more

packages/frontend/core/src/blocksuite/ai/components/ai-chat-add-context/ai-chat-add-context.ts

@@ -0,0 +1,101 @@
+import { createLitPortal } from '@blocksuite/affine/components/portal';
+import { SignalWatcher, WithDisposable } from '@blocksuite/affine/global/lit';
+import { ShadowlessElement } from '@blocksuite/affine/std';
+import { PlusIcon } from '@blocksuite/icons/lit';
+import { flip, offset } from '@floating-ui/dom';
+import { css, html } from 'lit';
+import { property, query } from 'lit/decorators.js';
+
+import type { ChatChip, DocDisplayConfig } from '../ai-chat-chips';
+import type { SearchMenuConfig } from './type';
+
+export class AIChatAddContext extends SignalWatcher(
+  WithDisposable(ShadowlessElement)
+) {
+  static override styles = css`
+    .ai-chat-add-context {
+      display: flex;
+      flex-shrink: 0;
+      flex-grow: 0;
+      align-items: center;
+      justify-content: center;
+      cursor: pointer;
+    }
+  `;
+
+  @property({ attribute: false })
+  accessor docId: string | undefined;
+
+  @property({ attribute: false })
+  accessor independentMode: boolean | undefined;
+
+  @property({ attribute: false })
+  accessor addChip!: (chip: ChatChip) => Promise<void>;
+
+  @property({ attribute: false })
+  accessor addImages!: (images: File[]) => void;
+
+  @property({ attribute: false })
+  accessor docDisplayConfig!: DocDisplayConfig;
+
+  @property({ attribute: false })
+  accessor searchMenuConfig!: SearchMenuConfig;
+
+  @property({ attribute: false })
+  accessor portalContainer: HTMLElement | null = null;
+
+  @query('.ai-chat-add-context')
+  accessor addButton!: HTMLDivElement;
+
+  private abortController: AbortController | null = null;
+
+  override render() {
+    return html`
+      <div
+        class="ai-chat-add-context"
+        data-testid="chat-panel-with-button"
+        @click=${this.toggleAddDocMenu}
+      >
+        ${PlusIcon()}
+      </div>
+    `;
+  }
+
+  private readonly toggleAddDocMenu = () => {
+    if (this.abortController) {
+      this.abortController.abort();
+      return;
+    }
+
+    this.abortController = new AbortController();
+    this.abortController.signal.addEventListener('abort', () => {
+      this.abortController = null;
+    });
+
+    createLitPortal({
+      template: html`
+        <chat-panel-add-popover
+          .docId=${this.docId}
+          .independentMode=${this.independentMode}
+          .addChip=${this.addChip}
+          .addImages=${this.addImages}
+          .searchMenuConfig=${this.searchMenuConfig}
+          .docDisplayConfig=${this.docDisplayConfig}
+          .abortController=${this.abortController}
+        ></chat-panel-add-popover>
+      `,
+      portalStyles: {
+        zIndex: 'var(--affine-z-index-popover)',
+      },
+      container: this.portalContainer ?? document.body,
+      computePosition: {
+        referenceElement: this.addButton,
+        placement: 'top-start',
+        middleware: [offset({ crossAxis: -30, mainAxis: 8 }), flip()],
+        autoUpdate: { animationFrame: true },
+      },
+      abortController: this.abortController,
+      closeOnClickAway: true,
+    });
+  };
+}

packages/frontend/core/src/blocksuite/ai/components/ai-chat-add-context/index.ts

@@ -0,0 +1,2 @@
+export * from './ai-chat-add-context';
+export * from './type';

packages/frontend/core/src/blocksuite/ai/components/ai-chat-add-context/type.ts

@@ -0,0 +1,24 @@
+import type {
+  SearchCollectionMenuAction,
+  SearchDocMenuAction,
+  SearchTagMenuAction,
+} from '@affine/core/modules/search-menu/services';
+import type { LinkedMenuGroup } from '@blocksuite/affine/widgets/linked-doc';
+
+export interface SearchMenuConfig {
+  getDocMenuGroup: (
+    query: string,
+    action: SearchDocMenuAction,
+    abortSignal: AbortSignal
+  ) => LinkedMenuGroup;
+  getTagMenuGroup: (
+    query: string,
+    action: SearchTagMenuAction,
+    abortSignal: AbortSignal
+  ) => LinkedMenuGroup;
+  getCollectionMenuGroup: (
+    query: string,
+    action: SearchCollectionMenuAction,
+    abortSignal: AbortSignal
+  ) => LinkedMenuGroup;
+}

packages/frontend/core/src/blocksuite/ai/components/ai-chat-chips/add-popover.ts

@@ -1,7 +1,7 @@
 import { toast } from '@affine/component';
 import type { TagMeta } from '@affine/core/components/page-list';
 import type { CollectionMeta } from '@affine/core/modules/collection';
-import track from '@affine/track';
+import track, { type EventArgs } from '@affine/track';
 import { SignalWatcher, WithDisposable } from '@blocksuite/affine/global/lit';
 import { scrollbarStyle } from '@blocksuite/affine/shared/styles';
 import { unsafeCSSVar, unsafeCSSVarV2 } from '@blocksuite/affine/shared/theme';
@@ -21,8 +21,8 @@ import { css, html, type TemplateResult } from 'lit';
 import { property, query, state } from 'lit/decorators.js';
 import { repeat } from 'lit/directives/repeat.js';
 
-import { MAX_IMAGE_COUNT } from '../ai-chat-input';
-import type { ChatChip, DocDisplayConfig, SearchMenuConfig } from './type';
+import type { SearchMenuConfig } from '../ai-chat-add-context';
+import type { ChatChip, DocDisplayConfig } from './type';
 
 enum AddPopoverMode {
   Default = 'default',
@@ -120,6 +120,12 @@ export class ChatPanelAddPopover extends SignalWatcher(
 
   private accessor _query = '';
 
+  @property({ attribute: false })
+  accessor independentMode: boolean | undefined;
+
+  @property({ attribute: false })
+  accessor docId: string | undefined;
+
   @state()
   private accessor _searchGroups: MenuGroup[] = [];
 
@@ -165,35 +171,31 @@ export class ChatPanelAddPopover extends SignalWatcher(
     const files = await openFilesWith();
     if (!files || files.length === 0) return;
 
+    this.abortController.abort();
     const images = files.filter(file => file.type.startsWith('image/'));
     if (images.length > 0) {
       this.addImages(images);
     }
 
     const others = files.filter(file => !file.type.startsWith('image/'));
-    for (const file of others) {
+    const addChipPromises = others.map(async file => {
       if (file.size > 50 * 1024 * 1024) {
         toast(`${file.name} is too large, please upload a file less than 50MB`);
-      } else {
-        await this.addChip({
-          file,
-          state: 'processing',
-        });
+        return;
       }
-    }
+      await this.addChip({
+        file,
+        state: 'processing',
+      });
+    });
+    await Promise.all(addChipPromises);
     this._track('file');
-    this.abortController.abort();
   };
 
   private readonly _addImageChip = async () => {
-    if (this.isImageUploadDisabled) return;
-
     const images = await openFilesWith('Images');
     if (!images) return;
-    if (this.uploadImageCount + images.length > MAX_IMAGE_COUNT) {
-      toast(`You can only upload up to ${MAX_IMAGE_COUNT} images`);
-      return;
-    }
+    this.abortController.abort();
     this.addImages(images);
   };
 
@@ -289,9 +291,6 @@ export class ChatPanelAddPopover extends SignalWatcher(
   @property({ attribute: 'data-testid', reflect: true })
   accessor testId: string = 'ai-search-input';
 
-  @property({ attribute: false })
-  accessor isImageUploadDisabled!: boolean;
-
   @property({ attribute: false })
   accessor uploadImageCount!: number;
 
@@ -498,31 +497,32 @@ export class ChatPanelAddPopover extends SignalWatcher(
   }
 
   private readonly _addDocChip = async (meta: DocMeta) => {
+    this.abortController.abort();
     await this.addChip({
       docId: meta.id,
       state: 'processing',
     });
     const mode = this.docDisplayConfig.getDocPrimaryMode(meta.id);
-    this._track('doc', mode);
-    this.abortController.abort();
+    const method = meta.id === this.docId ? 'cur-doc' : 'doc';
+    this._track(method, mode);
   };
 
   private readonly _addTagChip = async (tag: TagMeta) => {
+    this.abortController.abort();
     await this.addChip({
       tagId: tag.id,
       state: 'processing',
     });
     this._track('tags');
-    this.abortController.abort();
   };
 
   private readonly _addCollectionChip = async (collection: CollectionMeta) => {
+    this.abortController.abort();
     await this.addChip({
       collectionId: collection.id,
       state: 'processing',
     });
     this._track('collections');
-    this.abortController.abort();
   };
 
   private readonly _handleKeyDown = (event: KeyboardEvent) => {
@@ -568,10 +568,13 @@ export class ChatPanelAddPopover extends SignalWatcher(
   }
 
   private _track(
-    method: 'doc' | 'file' | 'tags' | 'collections',
+    method: EventArgs['addEmbeddingDoc']['method'],
     type?: 'page' | 'edgeless'
   ) {
-    track.$.chatPanel.chatPanelInput.addEmbeddingDoc({
+    const page = this.independentMode
+      ? track.$.intelligence
+      : track.$.chatPanel;
+    page.chatPanelInput.addEmbeddingDoc({
       control: 'addButton',
       method,
       type,

packages/frontend/core/src/blocksuite/ai/components/ai-chat-chips/chat-panel-chips.ts

@@ -3,7 +3,7 @@ import { createLitPortal } from '@blocksuite/affine/components/portal';
 import { SignalWatcher, WithDisposable } from '@blocksuite/affine/global/lit';
 import { unsafeCSSVarV2 } from '@blocksuite/affine/shared/theme';
 import { ShadowlessElement } from '@blocksuite/affine/std';
-import { MoreVerticalIcon, PlusIcon } from '@blocksuite/icons/lit';
+import { MoreVerticalIcon } from '@blocksuite/icons/lit';
 import { flip, offset } from '@floating-ui/dom';
 import { computed, type Signal, signal } from '@preact/signals-core';
 import { css, html, nothing, type PropertyValues } from 'lit';
@@ -11,16 +11,7 @@ import { property, query, state } from 'lit/decorators.js';
 import { repeat } from 'lit/directives/repeat.js';
 import { isEqual } from 'lodash-es';
 
-import { AIProvider } from '../../provider';
-import type {
-  ChatChip,
-  CollectionChip,
-  DocChip,
-  DocDisplayConfig,
-  FileChip,
-  SearchMenuConfig,
-  TagChip,
-} from './type';
+import type { ChatChip, DocChip, DocDisplayConfig, FileChip } from './type';
 import {
   estimateTokenCount,
   getChipKey,
@@ -39,44 +30,46 @@ export class ChatPanelChips extends SignalWatcher(
   WithDisposable(ShadowlessElement)
 ) {
   static override styles = css`
-    .chips-wrapper {
+    .ai-chat-panel-chips {
       display: flex;
       gap: 8px;
       align-items: center;
       flex-wrap: wrap;
       padding: 4px 12px;
-    }
-    .add-button,
-    .collapse-button,
-    .more-candidate-button {
-      display: flex;
-      flex-shrink: 0;
-      flex-grow: 0;
-      align-items: center;
-      justify-content: center;
-      width: 24px;
-      height: 24px;
-      border: 0.5px solid ${unsafeCSSVarV2('layer/insideBorder/border')};
-      border-radius: 4px;
-      box-sizing: border-box;
-      cursor: pointer;
-      font-size: 12px;
-      color: ${unsafeCSSVarV2('icon/primary')};
-    }
-    .add-button:hover,
-    .collapse-button:hover,
-    .more-candidate-button:hover {
-      background-color: ${unsafeCSSVarV2('layer/background/hoverOverlay')};
-    }
-    .more-candidate-button {
-      border-width: 1px;
-      border-style: dashed;
-      border-color: ${unsafeCSSVarV2('icon/tertiary')};
-      background: ${unsafeCSSVarV2('layer/background/secondary')};
-      color: ${unsafeCSSVarV2('icon/secondary')};
-    }
-    .more-candidate-button svg {
-      color: ${unsafeCSSVarV2('icon/secondary')};
+
+      .collapse-button,
+      .more-candidate-button {
+        display: flex;
+        flex-shrink: 0;
+        flex-grow: 0;
+        align-items: center;
+        justify-content: center;
+        width: 24px;
+        height: 24px;
+        border: 0.5px solid ${unsafeCSSVarV2('layer/insideBorder/border')};
+        border-radius: 4px;
+        box-sizing: border-box;
+        cursor: pointer;
+        font-size: 12px;
+        color: ${unsafeCSSVarV2('icon/primary')};
+      }
+
+      .collapse-button:hover,
+      .more-candidate-button:hover {
+        background-color: ${unsafeCSSVarV2('layer/background/hoverOverlay')};
+      }
+
+      .more-candidate-button {
+        border-width: 1px;
+        border-style: dashed;
+        border-color: ${unsafeCSSVarV2('icon/tertiary')};
+        background: ${unsafeCSSVarV2('layer/background/secondary')};
+        color: ${unsafeCSSVarV2('icon/secondary')};
+      }
+
+      .more-candidate-button svg {
+        color: ${unsafeCSSVarV2('icon/secondary')};
+      }
     }
   `;
 
@@ -86,38 +79,38 @@ export class ChatPanelChips extends SignalWatcher(
   accessor chips!: ChatChip[];
 
   @property({ attribute: false })
-  accessor createContextId!: () => Promise<string | undefined>;
+  accessor isCollapsed!: boolean;
 
   @property({ attribute: false })
-  accessor updateChips!: (chips: ChatChip[]) => void;
+  accessor independentMode: boolean | undefined;
 
   @property({ attribute: false })
-  accessor addImages!: (images: File[]) => void;
+  accessor addChip!: (chip: ChatChip) => Promise<void>;
 
   @property({ attribute: false })
-  accessor pollContextDocsAndFiles!: () => void;
+  accessor updateChip!: (
+    chip: ChatChip,
+    options: Partial<DocChip | FileChip>
+  ) => void;
+
+  @property({ attribute: false })
+  accessor removeChip!: (chip: ChatChip) => Promise<void>;
+
+  @property({ attribute: false })
+  accessor toggleCollapse!: () => void;
 
   @property({ attribute: false })
   accessor docDisplayConfig!: DocDisplayConfig;
 
-  @property({ attribute: false })
-  accessor searchMenuConfig!: SearchMenuConfig;
-
   @property({ attribute: false })
   accessor portalContainer: HTMLElement | null = null;
 
   @property({ attribute: 'data-testid', reflect: true })
   accessor testId = 'chat-panel-chips';
 
-  @query('.add-button')
-  accessor addButton!: HTMLDivElement;
-
   @query('.more-candidate-button')
   accessor moreCandidateButton!: HTMLDivElement;
 
-  @state()
-  accessor isCollapsed = false;
-
   @state()
   accessor referenceDocs: Signal<
     Array<{
@@ -144,14 +137,7 @@ export class ChatPanelChips extends SignalWatcher(
     const isCollapsed = this.isCollapsed && allChips.length > 1;
     const chips = isCollapsed ? allChips.slice(0, 1) : allChips;
 
-    return html`<div class="chips-wrapper">
-      <div
-        class="add-button"
-        data-testid="chat-panel-with-button"
-        @click=${this._toggleAddDocMenu}
-      >
-        ${PlusIcon()}
-      </div>
+    return html`<div class="ai-chat-panel-chips">
       ${repeat(
         chips,
         chip => getChipKey(chip),
@@ -159,9 +145,10 @@ export class ChatPanelChips extends SignalWatcher(
           if (isDocChip(chip)) {
             return html`<chat-panel-doc-chip
               .chip=${chip}
-              .addChip=${this._addChip}
-              .updateChip=${this._updateChip}
-              .removeChip=${this._removeChip}
+              .independentMode=${this.independentMode}
+              .addChip=${this.addChip}
+              .updateChip=${this.updateChip}
+              .removeChip=${this.removeChip}
               .checkTokenLimit=${this._checkTokenLimit}
               .docDisplayConfig=${this.docDisplayConfig}
             ></chat-panel-doc-chip>`;
@@ -169,7 +156,7 @@ export class ChatPanelChips extends SignalWatcher(
           if (isFileChip(chip)) {
             return html`<chat-panel-file-chip
               .chip=${chip}
-              .removeChip=${this._removeChip}
+              .removeChip=${this.removeChip}
             ></chat-panel-file-chip>`;
           }
           if (isTagChip(chip)) {
@@ -180,7 +167,7 @@ export class ChatPanelChips extends SignalWatcher(
             return html`<chat-panel-tag-chip
               .chip=${chip}
               .tag=${tag}
-              .removeChip=${this._removeChip}
+              .removeChip=${this.removeChip}
             ></chat-panel-tag-chip>`;
           }
           if (isCollectionChip(chip)) {
@@ -193,7 +180,7 @@ export class ChatPanelChips extends SignalWatcher(
             return html`<chat-panel-collection-chip
               .chip=${chip}
               .collection=${collection}
-              .removeChip=${this._removeChip}
+              .removeChip=${this.removeChip}
             ></chat-panel-collection-chip>`;
           }
           return null;
@@ -208,7 +195,7 @@ export class ChatPanelChips extends SignalWatcher(
           </div>`
         : nothing}
       ${isCollapsed
-        ? html`<div class="collapse-button" @click=${this._toggleCollapse}>
+        ? html`<div class="collapse-button" @click=${this.toggleCollapse}>
             +${allChips.length - 1}
           </div>`
         : nothing}
@@ -227,14 +214,6 @@ export class ChatPanelChips extends SignalWatcher(
   }
 
   protected override updated(_changedProperties: PropertyValues): void {
-    if (
-      _changedProperties.has('chatContextValue') &&
-      _changedProperties.get('chatContextValue')?.status === 'loading' &&
-      this.isCollapsed === false
-    ) {
-      this.isCollapsed = true;
-    }
-
     if (_changedProperties.has('chips')) {
       this._updateReferenceDocs();
     }
@@ -245,46 +224,6 @@ export class ChatPanelChips extends SignalWatcher(
     this._cleanup?.();
   }
 
-  private readonly _toggleCollapse = () => {
-    this.isCollapsed = !this.isCollapsed;
-  };
-
-  private readonly _toggleAddDocMenu = () => {
-    if (this._abortController) {
-      this._abortController.abort();
-      return;
-    }
-
-    this._abortController = new AbortController();
-    this._abortController.signal.addEventListener('abort', () => {
-      this._abortController = null;
-    });
-
-    createLitPortal({
-      template: html`
-        <chat-panel-add-popover
-          .addChip=${this._addChip}
-          .addImages=${this.addImages}
-          .searchMenuConfig=${this.searchMenuConfig}
-          .docDisplayConfig=${this.docDisplayConfig}
-          .abortController=${this._abortController}
-        ></chat-panel-add-popover>
-      `,
-      portalStyles: {
-        zIndex: 'var(--affine-z-index-popover)',
-      },
-      container: this.portalContainer ?? document.body,
-      computePosition: {
-        referenceElement: this.addButton,
-        placement: 'top-start',
-        middleware: [offset({ crossAxis: -30, mainAxis: 8 }), flip()],
-        autoUpdate: { animationFrame: true },
-      },
-      abortController: this._abortController,
-      closeOnClickAway: true,
-    });
-  };
-
   private readonly _toggleMoreCandidatesMenu = () => {
     if (this._abortController) {
       this._abortController.abort();
@@ -303,7 +242,7 @@ export class ChatPanelChips extends SignalWatcher(
     createLitPortal({
       template: html`
         <chat-panel-candidates-popover
-          .addChip=${this._addChip}
+          .addChip=${this.addChip}
           .referenceDocs=${referenceDocs}
           .docDisplayConfig=${this.docDisplayConfig}
           .abortController=${this._abortController}
@@ -324,190 +263,6 @@ export class ChatPanelChips extends SignalWatcher(
     });
   };
 
-  private readonly _addChip = async (chip: ChatChip) => {
-    this.isCollapsed = false;
-    // remove the chip if it already exists
-    const chips = this._omitChip(this.chips, chip);
-    this.updateChips([...chips, chip]);
-    if (chips.length < this.chips.length) {
-      await this._removeFromContext(chip);
-    }
-    await this._addToContext(chip);
-    this.pollContextDocsAndFiles();
-  };
-
-  private readonly _updateChip = (
-    chip: ChatChip,
-    options: Partial<DocChip | FileChip>
-  ) => {
-    const index = this._findChipIndex(this.chips, chip);
-    if (index === -1) {
-      return;
-    }
-    const nextChip: ChatChip = {
-      ...chip,
-      ...options,
-    };
-    this.updateChips([
-      ...this.chips.slice(0, index),
-      nextChip,
-      ...this.chips.slice(index + 1),
-    ]);
-  };
-
-  private readonly _removeChip = async (chip: ChatChip) => {
-    const chips = this._omitChip(this.chips, chip);
-    this.updateChips(chips);
-    if (chips.length < this.chips.length) {
-      await this._removeFromContext(chip);
-    }
-  };
-
-  private readonly _addToContext = async (chip: ChatChip) => {
-    if (isDocChip(chip)) {
-      return await this._addDocToContext(chip);
-    }
-    if (isFileChip(chip)) {
-      return await this._addFileToContext(chip);
-    }
-    if (isTagChip(chip)) {
-      return await this._addTagToContext(chip);
-    }
-    if (isCollectionChip(chip)) {
-      return await this._addCollectionToContext(chip);
-    }
-    return null;
-  };
-
-  private readonly _addDocToContext = async (chip: DocChip) => {
-    try {
-      const contextId = await this.createContextId();
-      if (!contextId || !AIProvider.context) {
-        throw new Error('Context not found');
-      }
-      await AIProvider.context.addContextDoc({
-        contextId,
-        docId: chip.docId,
-      });
-    } catch (e) {
-      this._updateChip(chip, {
-        state: 'failed',
-        tooltip: e instanceof Error ? e.message : 'Add context doc error',
-      });
-    }
-  };
-
-  private readonly _addFileToContext = async (chip: FileChip) => {
-    try {
-      const contextId = await this.createContextId();
-      if (!contextId || !AIProvider.context) {
-        throw new Error('Context not found');
-      }
-      const contextFile = await AIProvider.context.addContextFile(chip.file, {
-        contextId,
-      });
-      this._updateChip(chip, {
-        state: contextFile.status,
-        blobId: contextFile.blobId,
-        fileId: contextFile.id,
-      });
-    } catch (e) {
-      this._updateChip(chip, {
-        state: 'failed',
-        tooltip: e instanceof Error ? e.message : 'Add context file error',
-      });
-    }
-  };
-
-  private readonly _addTagToContext = async (chip: TagChip) => {
-    try {
-      const contextId = await this.createContextId();
-      if (!contextId || !AIProvider.context) {
-        throw new Error('Context not found');
-      }
-      // TODO: server side docIds calculation
-      const docIds = this.docDisplayConfig.getTagPageIds(chip.tagId);
-      await AIProvider.context.addContextTag({
-        contextId,
-        tagId: chip.tagId,
-        docIds,
-      });
-      this._updateChip(chip, {
-        state: 'finished',
-      });
-    } catch (e) {
-      this._updateChip(chip, {
-        state: 'failed',
-        tooltip: e instanceof Error ? e.message : 'Add context tag error',
-      });
-    }
-  };
-
-  private readonly _addCollectionToContext = async (chip: CollectionChip) => {
-    try {
-      const contextId = await this.createContextId();
-      if (!contextId || !AIProvider.context) {
-        throw new Error('Context not found');
-      }
-      // TODO: server side docIds calculation
-      const docIds = this.docDisplayConfig.getCollectionPageIds(
-        chip.collectionId
-      );
-      await AIProvider.context.addContextCollection({
-        contextId,
-        collectionId: chip.collectionId,
-        docIds,
-      });
-      this._updateChip(chip, {
-        state: 'finished',
-      });
-    } catch (e) {
-      this._updateChip(chip, {
-        state: 'failed',
-        tooltip:
-          e instanceof Error ? e.message : 'Add context collection error',
-      });
-    }
-  };
-
-  private readonly _removeFromContext = async (
-    chip: ChatChip
-  ): Promise<boolean> => {
-    try {
-      const contextId = await this.createContextId();
-      if (!contextId || !AIProvider.context) {
-        return true;
-      }
-      if (isDocChip(chip)) {
-        return await AIProvider.context.removeContextDoc({
-          contextId,
-          docId: chip.docId,
-        });
-      }
-      if (isFileChip(chip) && chip.fileId) {
-        return await AIProvider.context.removeContextFile({
-          contextId,
-          fileId: chip.fileId,
-        });
-      }
-      if (isTagChip(chip)) {
-        return await AIProvider.context.removeContextTag({
-          contextId,
-          tagId: chip.tagId,
-        });
-      }
-      if (isCollectionChip(chip)) {
-        return await AIProvider.context.removeContextCollection({
-          contextId,
-          collectionId: chip.collectionId,
-        });
-      }
-      return true;
-    } catch {
-      return true;
-    }
-  };
-
   private readonly _checkTokenLimit = (
     newChip: DocChip,
     newTokenCount: number
@@ -544,44 +299,4 @@ export class ChatPanelChips extends SignalWatcher(
     this.referenceDocs = signal;
     this._cleanup = cleanup;
   };
-
-  private readonly _omitChip = (chips: ChatChip[], chip: ChatChip) => {
-    return chips.filter(item => {
-      if (isDocChip(chip)) {
-        return !isDocChip(item) || item.docId !== chip.docId;
-      }
-      if (isFileChip(chip)) {
-        return !isFileChip(item) || item.file !== chip.file;
-      }
-      if (isTagChip(chip)) {
-        return !isTagChip(item) || item.tagId !== chip.tagId;
-      }
-      if (isCollectionChip(chip)) {
-        return (
-          !isCollectionChip(item) || item.collectionId !== chip.collectionId
-        );
-      }
-      return true;
-    });
-  };
-
-  private readonly _findChipIndex = (chips: ChatChip[], chip: ChatChip) => {
-    return chips.findIndex(item => {
-      if (isDocChip(chip)) {
-        return isDocChip(item) && item.docId === chip.docId;
-      }
-      if (isFileChip(chip)) {
-        return isFileChip(item) && item.file === chip.file;
-      }
-      if (isTagChip(chip)) {
-        return isTagChip(item) && item.tagId === chip.tagId;
-      }
-      if (isCollectionChip(chip)) {
-        return (
-          isCollectionChip(item) && item.collectionId === chip.collectionId
-        );
-      }
-      return -1;
-    });
-  };
 }

packages/frontend/core/src/blocksuite/ai/components/ai-chat-chips/doc-chip.ts

@@ -18,6 +18,9 @@ export class ChatPanelDocChip extends SignalWatcher(
   @property({ attribute: false })
   accessor chip!: DocChip;
 
+  @property({ attribute: false })
+  accessor independentMode: boolean | undefined;
+
   @property({ attribute: false })
   accessor addChip!: (chip: DocChip) => void;
 
@@ -81,7 +84,10 @@ export class ChatPanelDocChip extends SignalWatcher(
         state: 'processing',
       });
       const mode = this.docDisplayConfig.getDocPrimaryMode(this.chip.docId);
-      track.$.chatPanel.chatPanelInput.addEmbeddingDoc({
+      const page = this.independentMode
+        ? track.$.intelligence
+        : track.$.chatPanel;
+      page.chatPanelInput.addEmbeddingDoc({
         control: 'addButton',
         method: 'suggestion',
         type: mode,

packages/frontend/core/src/blocksuite/ai/components/ai-chat-chips/type.ts

@@ -1,11 +1,5 @@
 import type { TagMeta } from '@affine/core/components/page-list';
-import type {
-  SearchCollectionMenuAction,
-  SearchDocMenuAction,
-  SearchTagMenuAction,
-} from '@affine/core/modules/search-menu/services';
 import type { DocMeta, Store } from '@blocksuite/affine/store';
-import type { LinkedMenuGroup } from '@blocksuite/affine/widgets/linked-doc';
 import type { Signal } from '@preact/signals-core';
 
 export type ChipState = 'candidate' | 'processing' | 'finished' | 'failed';
@@ -75,21 +69,3 @@ export interface DocDisplayConfig {
   };
   getCollectionPageIds: (collectionId: string) => string[];
 }
-
-export interface SearchMenuConfig {
-  getDocMenuGroup: (
-    query: string,
-    action: SearchDocMenuAction,
-    abortSignal: AbortSignal
-  ) => LinkedMenuGroup;
-  getTagMenuGroup: (
-    query: string,
-    action: SearchTagMenuAction,
-    abortSignal: AbortSignal
-  ) => LinkedMenuGroup;
-  getCollectionMenuGroup: (
-    query: string,
-    action: SearchCollectionMenuAction,
-    abortSignal: AbortSignal
-  ) => LinkedMenuGroup;
-}

packages/frontend/core/src/blocksuite/ai/components/ai-chat-chips/utils.ts

@@ -78,6 +78,42 @@ export function getChipKey(chip: ChatChip) {
   return null;
 }
 
+export function omitChip(chips: ChatChip[], chip: ChatChip) {
+  return chips.filter(item => {
+    if (isDocChip(chip)) {
+      return !isDocChip(item) || item.docId !== chip.docId;
+    }
+    if (isFileChip(chip)) {
+      return !isFileChip(item) || item.file !== chip.file;
+    }
+    if (isTagChip(chip)) {
+      return !isTagChip(item) || item.tagId !== chip.tagId;
+    }
+    if (isCollectionChip(chip)) {
+      return !isCollectionChip(item) || item.collectionId !== chip.collectionId;
+    }
+    return true;
+  });
+}
+
+export function findChipIndex(chips: ChatChip[], chip: ChatChip) {
+  return chips.findIndex(item => {
+    if (isDocChip(chip)) {
+      return isDocChip(item) && item.docId === chip.docId;
+    }
+    if (isFileChip(chip)) {
+      return isFileChip(item) && item.file === chip.file;
+    }
+    if (isTagChip(chip)) {
+      return isTagChip(item) && item.tagId === chip.tagId;
+    }
+    if (isCollectionChip(chip)) {
+      return isCollectionChip(item) && item.collectionId === chip.collectionId;
+    }
+    return -1;
+  });
+}
+
 export function estimateTokenCount(text: string): number {
   const chinese = text.match(/[\u4e00-\u9fa5]/g)?.length || 0;
   const english = text.replace(/[\u4e00-\u9fa5]/g, '');

packages/frontend/core/src/blocksuite/ai/components/ai-chat-composer/ai-chat-composer.ts

@@ -12,20 +12,28 @@ import type {
 import { SignalWatcher, WithDisposable } from '@blocksuite/affine/global/lit';
 import type { EditorHost } from '@blocksuite/affine/std';
 import { ShadowlessElement } from '@blocksuite/affine/std';
-import { css, html } from 'lit';
+import type { NotificationService } from '@blocksuite/affine-shared/services';
+import { css, html, type PropertyValues } from 'lit';
 import { property, state } from 'lit/decorators.js';
 
 import { AIProvider } from '../../provider';
+import type { SearchMenuConfig } from '../ai-chat-add-context';
 import type {
   ChatChip,
   CollectionChip,
   DocChip,
   DocDisplayConfig,
   FileChip,
-  SearchMenuConfig,
   TagChip,
 } from '../ai-chat-chips';
-import { isCollectionChip, isDocChip, isTagChip } from '../ai-chat-chips';
+import {
+  findChipIndex,
+  isCollectionChip,
+  isDocChip,
+  isFileChip,
+  isTagChip,
+  omitChip,
+} from '../ai-chat-chips';
 import type {
   AIChatInputContext,
   AINetworkSearchConfig,
@@ -51,7 +59,7 @@ export class AIChatComposer extends SignalWatcher(
   `;
 
   @property({ attribute: false })
-  accessor independentMode!: boolean;
+  accessor independentMode: boolean | undefined;
 
   @property({ attribute: false })
   accessor host: EditorHost | null | undefined;
@@ -77,9 +85,9 @@ export class AIChatComposer extends SignalWatcher(
   accessor updateContext!: (context: Partial<AIChatInputContext>) => void;
 
   @property({ attribute: false })
-  accessor onEmbeddingProgressChange!: (
-    count: Record<ContextEmbedStatus, number>
-  ) => void;
+  accessor onEmbeddingProgressChange:
+    | ((count: Record<ContextEmbedStatus, number>) => void)
+    | undefined;
 
   @property({ attribute: false })
   accessor docDisplayConfig!: DocDisplayConfig;
@@ -105,9 +113,15 @@ export class AIChatComposer extends SignalWatcher(
   @property({ attribute: false })
   accessor affineWorkspaceDialogService!: WorkspaceDialogService;
 
+  @property({ attribute: false })
+  accessor notificationService!: NotificationService;
+
   @state()
   accessor chips: ChatChip[] = [];
 
+  @state()
+  accessor isChipsCollapsed = false;
+
   @state()
   accessor embeddingCompleted = false;
 
@@ -121,11 +135,13 @@ export class AIChatComposer extends SignalWatcher(
     return html`
       <chat-panel-chips
         .chips=${this.chips}
-        .createContextId=${this._createContextId}
-        .updateChips=${this.updateChips}
-        .pollContextDocsAndFiles=${this._pollContextDocsAndFiles}
+        .isCollapsed=${this.isChipsCollapsed}
+        .independentMode=${this.independentMode}
+        .addChip=${this.addChip}
+        .updateChip=${this.updateChip}
+        .removeChip=${this.removeChip}
+        .toggleCollapse=${this.toggleChipsCollapse}
         .docDisplayConfig=${this.docDisplayConfig}
-        .searchMenuConfig=${this.searchMenuConfig}
         .portalContainer=${this.portalContainer}
         .addImages=${this.addImages}
       ></chat-panel-chips>
@@ -136,15 +152,18 @@ export class AIChatComposer extends SignalWatcher(
         .docId=${this.docId}
         .session=${this.session}
         .chips=${this.chips}
+        .addChip=${this.addChip}
+        .addImages=${this.addImages}
         .createSession=${this.createSession}
         .chatContextValue=${this.chatContextValue}
         .updateContext=${this.updateContext}
         .networkSearchConfig=${this.networkSearchConfig}
         .reasoningConfig=${this.reasoningConfig}
         .docDisplayConfig=${this.docDisplayConfig}
+        .searchMenuConfig=${this.searchMenuConfig}
+        .portalContainer=${this.portalContainer}
         .onChatSuccess=${this.onChatSuccess}
         .trackOptions=${this.trackOptions}
-        .addImages=${this.addImages}
       ></ai-chat-input>
       <div class="chat-panel-footer">
         <ai-chat-composer-tip
@@ -165,7 +184,7 @@ export class AIChatComposer extends SignalWatcher(
 
   override connectedCallback() {
     super.connectedCallback();
-    this._initComposer().catch(console.error);
+    this.initComposer().catch(console.error);
   }
 
   override disconnectedCallback() {
@@ -174,6 +193,17 @@ export class AIChatComposer extends SignalWatcher(
     this._abortPollEmbeddingStatus();
   }
 
+  protected override willUpdate(changedProperties: PropertyValues): void {
+    if (
+      changedProperties.has('chatContextValue') &&
+      changedProperties.get('chatContextValue')?.status !== 'loading' &&
+      this.chatContextValue.status === 'loading' &&
+      this.isChipsCollapsed === false
+    ) {
+      this.isChipsCollapsed = true;
+    }
+  }
+
   private readonly _getContextId = async () => {
     if (this._contextId) {
       return this._contextId;
@@ -190,7 +220,7 @@ export class AIChatComposer extends SignalWatcher(
     return this._contextId;
   };
 
-  private readonly _createContextId = async () => {
+  private readonly createContextId = async () => {
     if (this._contextId) {
       return this._contextId;
     }
@@ -205,7 +235,7 @@ export class AIChatComposer extends SignalWatcher(
     return this._contextId;
   };
 
-  private readonly _initChips = async () => {
+  private readonly initChips = async () => {
     // context not initialized
     const sessionId = this.session?.sessionId;
     const contextId = await this._getContextId();
@@ -275,14 +305,206 @@ export class AIChatComposer extends SignalWatcher(
     this.chips = chips;
   };
 
+  private readonly updateChip = (
+    chip: ChatChip,
+    options: Partial<DocChip | FileChip>
+  ) => {
+    const index = findChipIndex(this.chips, chip);
+    if (index === -1) {
+      return;
+    }
+    const nextChip: ChatChip = {
+      ...chip,
+      ...options,
+    };
+    this.updateChips([
+      ...this.chips.slice(0, index),
+      nextChip,
+      ...this.chips.slice(index + 1),
+    ]);
+  };
+
+  private readonly addChip = async (chip: ChatChip) => {
+    this.isChipsCollapsed = false;
+    // if already exists
+    const index = findChipIndex(this.chips, chip);
+    if (index !== -1) {
+      this.notificationService.toast('chip already exists');
+      return;
+    }
+    this.updateChips([...this.chips, chip]);
+    await this.addToContext(chip);
+    await this.pollContextDocsAndFiles();
+  };
+
+  private readonly removeChip = async (chip: ChatChip) => {
+    const chips = omitChip(this.chips, chip);
+    this.updateChips(chips);
+    await this.removeFromContext(chip);
+  };
+
+  private readonly addToContext = async (chip: ChatChip) => {
+    if (isDocChip(chip)) {
+      return await this.addDocToContext(chip);
+    }
+    if (isFileChip(chip)) {
+      return await this.addFileToContext(chip);
+    }
+    if (isTagChip(chip)) {
+      return await this.addTagToContext(chip);
+    }
+    if (isCollectionChip(chip)) {
+      return await this.addCollectionToContext(chip);
+    }
+    return null;
+  };
+
+  private readonly addDocToContext = async (chip: DocChip) => {
+    try {
+      const contextId = await this.createContextId();
+      if (!contextId || !AIProvider.context) {
+        throw new Error('Context not found');
+      }
+      await AIProvider.context.addContextDoc({
+        contextId,
+        docId: chip.docId,
+      });
+    } catch (e) {
+      this.updateChip(chip, {
+        state: 'failed',
+        tooltip: e instanceof Error ? e.message : 'Add context doc error',
+      });
+    }
+  };
+
+  private readonly addFileToContext = async (chip: FileChip) => {
+    try {
+      const contextId = await this.createContextId();
+      if (!contextId || !AIProvider.context) {
+        throw new Error('Context not found');
+      }
+      const contextFile = await AIProvider.context.addContextFile(chip.file, {
+        contextId,
+      });
+      this.updateChip(chip, {
+        state: contextFile.status,
+        blobId: contextFile.blobId,
+        fileId: contextFile.id,
+      });
+    } catch (e) {
+      this.updateChip(chip, {
+        state: 'failed',
+        tooltip: e instanceof Error ? e.message : 'Add context file error',
+      });
+    }
+  };
+
+  private readonly addTagToContext = async (chip: TagChip) => {
+    try {
+      const contextId = await this.createContextId();
+      if (!contextId || !AIProvider.context) {
+        throw new Error('Context not found');
+      }
+      // TODO: server side docIds calculation
+      const docIds = this.docDisplayConfig.getTagPageIds(chip.tagId);
+      await AIProvider.context.addContextTag({
+        contextId,
+        tagId: chip.tagId,
+        docIds,
+      });
+      this.updateChip(chip, {
+        state: 'finished',
+      });
+    } catch (e) {
+      this.updateChip(chip, {
+        state: 'failed',
+        tooltip: e instanceof Error ? e.message : 'Add context tag error',
+      });
+    }
+  };
+
+  private readonly addCollectionToContext = async (chip: CollectionChip) => {
+    try {
+      const contextId = await this.createContextId();
+      if (!contextId || !AIProvider.context) {
+        throw new Error('Context not found');
+      }
+      // TODO: server side docIds calculation
+      const docIds = this.docDisplayConfig.getCollectionPageIds(
+        chip.collectionId
+      );
+      await AIProvider.context.addContextCollection({
+        contextId,
+        collectionId: chip.collectionId,
+        docIds,
+      });
+      this.updateChip(chip, {
+        state: 'finished',
+      });
+    } catch (e) {
+      this.updateChip(chip, {
+        state: 'failed',
+        tooltip:
+          e instanceof Error ? e.message : 'Add context collection error',
+      });
+    }
+  };
+
+  private readonly removeFromContext = async (
+    chip: ChatChip
+  ): Promise<boolean> => {
+    try {
+      const contextId = await this.createContextId();
+      if (!contextId || !AIProvider.context) {
+        return true;
+      }
+      if (isDocChip(chip)) {
+        return await AIProvider.context.removeContextDoc({
+          contextId,
+          docId: chip.docId,
+        });
+      }
+      if (isFileChip(chip) && chip.fileId) {
+        return await AIProvider.context.removeContextFile({
+          contextId,
+          fileId: chip.fileId,
+        });
+      }
+      if (isTagChip(chip)) {
+        return await AIProvider.context.removeContextTag({
+          contextId,
+          tagId: chip.tagId,
+        });
+      }
+      if (isCollectionChip(chip)) {
+        return await AIProvider.context.removeContextCollection({
+          contextId,
+          collectionId: chip.collectionId,
+        });
+      }
+      return true;
+    } catch {
+      return true;
+    }
+  };
+
+  private readonly toggleChipsCollapse = () => {
+    this.isChipsCollapsed = !this.isChipsCollapsed;
+  };
+
   private readonly addImages = (images: File[]) => {
     const oldImages = this.chatContextValue.images;
+    if (oldImages.length + images.length > MAX_IMAGE_COUNT) {
+      this.notificationService.toast(
+        `You can only upload up to ${MAX_IMAGE_COUNT} images`
+      );
+    }
     this.updateContext({
       images: [...oldImages, ...images].slice(0, MAX_IMAGE_COUNT),
     });
   };
 
-  private readonly _pollContextDocsAndFiles = async () => {
+  private readonly pollContextDocsAndFiles = async () => {
     const sessionId = this.session?.sessionId;
     const contextId = await this._getContextId();
     if (!sessionId || !contextId || !AIProvider.context) {
@@ -302,7 +524,7 @@ export class AIChatComposer extends SignalWatcher(
     );
   };
 
-  private readonly _pollEmbeddingStatus = async () => {
+  private readonly pollEmbeddingStatus = async () => {
     if (this._pollEmbeddingStatusAbortController) {
       this._pollEmbeddingStatusAbortController.abort();
     }
@@ -317,12 +539,11 @@ export class AIChatComposer extends SignalWatcher(
             this.embeddingCompleted = false;
             return;
           }
+          const prevCompleted = this.embeddingCompleted;
           const completed = status.embedded === status.total;
           this.embeddingCompleted = completed;
-          if (completed) {
-            this.embeddingCompleted = true;
-          } else {
-            this.embeddingCompleted = false;
+          if (prevCompleted !== completed) {
+            this.requestUpdate();
           }
         },
         signal
@@ -383,7 +604,7 @@ export class AIChatComposer extends SignalWatcher(
       return chip;
     });
     this.updateChips(nextChips);
-    this.onEmbeddingProgressChange(count);
+    this.onEmbeddingProgressChange?.(count);
     if (count.processing === 0) {
       this._abortPoll();
     }
@@ -399,18 +620,18 @@ export class AIChatComposer extends SignalWatcher(
     this._pollEmbeddingStatusAbortController = null;
   };
 
-  private readonly _initComposer = async () => {
+  private readonly initComposer = async () => {
     const userId = (await AIProvider.userInfo)?.id;
     if (!userId || !this.session) return;
 
-    await this._initChips();
+    await this.initChips();
     const needPoll = this.chips.some(
       chip =>
         chip.state === 'processing' || isTagChip(chip) || isCollectionChip(chip)
     );
     if (needPoll) {
-      await this._pollContextDocsAndFiles();
+      await this.pollContextDocsAndFiles();
     }
-    await this._pollEmbeddingStatus();
+    await this.pollEmbeddingStatus();
   };
 }

packages/frontend/core/src/blocksuite/ai/components/ai-chat-content/ai-chat-content.ts

@@ -6,8 +6,7 @@ import type {
   CopilotChatHistoryFragment,
 } from '@affine/graphql';
 import { SignalWatcher, WithDisposable } from '@blocksuite/affine/global/lit';
-import type { EditorHost } from '@blocksuite/affine/std';
-import { ShadowlessElement } from '@blocksuite/affine/std';
+import { type EditorHost, ShadowlessElement } from '@blocksuite/affine/std';
 import type { ExtensionType } from '@blocksuite/affine/store';
 import type { NotificationService } from '@blocksuite/affine-shared/services';
 import { type Signal } from '@preact/signals-core';
@@ -26,7 +25,8 @@ import { styleMap } from 'lit/directives/style-map.js';
 import { HISTORY_IMAGE_ACTIONS } from '../../chat-panel/const';
 import { type AIChatParams, AIProvider } from '../../provider/ai-provider';
 import { extractSelectedContent } from '../../utils/extract';
-import type { DocDisplayConfig, SearchMenuConfig } from '../ai-chat-chips';
+import type { SearchMenuConfig } from '../ai-chat-add-context';
+import type { DocDisplayConfig } from '../ai-chat-chips';
 import type {
   AINetworkSearchConfig,
   AIReasoningConfig,
@@ -63,7 +63,7 @@ export class AIChatContent extends SignalWatcher(
       .ai-chat-title {
         background: var(--affine-background-primary-color);
         position: relative;
-        padding: 8px 0px;
+        padding: 8px var(--h-padding);
         width: 100%;
         height: 36px;
         display: flex;
@@ -80,7 +80,8 @@ export class AIChatContent extends SignalWatcher(
 
       ai-chat-messages {
         flex: 1;
-        overflow-y: hidden;
+        overflow-y: auto;
+        padding: 0 var(--h-padding);
         transition:
           flex-grow 0.32s cubic-bezier(0.07, 0.83, 0.46, 1),
           padding-top 0.32s ease,
@@ -99,24 +100,31 @@ export class AIChatContent extends SignalWatcher(
       container-name: chat-panel-split-view;
     }
     .chat-panel-main {
+      --h-padding: 8px;
       display: flex;
       flex-direction: column;
       justify-content: center;
       height: 100%;
       width: 100%;
-      padding: 8px 24px 0 24px;
+      padding: 8px calc(24px - var(--h-padding)) 0 calc(24px - var(--h-padding));
       max-width: 800px;
       margin: 0 auto;
     }
+
+    ai-chat-composer {
+      padding: 0 var(--h-padding);
+    }
+
     @container chat-panel-split-view (width < 540px) {
       .chat-panel-main {
-        padding: 8px 12px 0 12px;
+        padding: 8px calc(12px - var(--h-padding)) 0
+          calc(12px - var(--h-padding));
       }
     }
   `;
 
   @property({ attribute: false })
-  accessor independentMode!: boolean;
+  accessor independentMode: boolean | undefined;
 
   @property({ attribute: false })
   accessor onboardingOffsetY!: number;
@@ -169,9 +177,9 @@ export class AIChatContent extends SignalWatcher(
   accessor notificationService!: NotificationService;
 
   @property({ attribute: false })
-  accessor onEmbeddingProgressChange!: (
-    count: Record<ContextEmbedStatus, number>
-  ) => void;
+  accessor onEmbeddingProgressChange:
+    | ((count: Record<ContextEmbedStatus, number>) => void)
+    | undefined;
 
   @property({ attribute: false })
   accessor onContextChange!: (context: Partial<ChatContextValue>) => void;
@@ -211,14 +219,7 @@ export class AIChatContent extends SignalWatcher(
   }
 
   get showActions() {
-    if (this.docId) {
-      if (!this.session) {
-        return true;
-      }
-      return this.session.docId === this.docId;
-    } else {
-      return false;
-    }
+    return false;
   }
 
   private readonly updateHistory = async () => {
@@ -261,7 +262,7 @@ export class AIChatContent extends SignalWatcher(
   };
 
   private readonly updateActions = async () => {
-    if (!this.docId || !AIProvider.histories) {
+    if (!this.docId || !AIProvider.histories || !this.showActions) {
       return;
     }
     const actions = await AIProvider.histories.actions(
@@ -395,7 +396,7 @@ export class AIChatContent extends SignalWatcher(
       <ai-chat-messages
         class=${classMap({
           'ai-chat-messages': true,
-          'independent-mode': this.independentMode,
+          'independent-mode': !!this.independentMode,
           'no-message': this.messages.length === 0,
         })}
         ${ref(this.chatMessagesRef)}
@@ -416,6 +417,7 @@ export class AIChatContent extends SignalWatcher(
         .width=${this.width}
         .independentMode=${this.independentMode}
         .messages=${this.messages}
+        .docDisplayService=${this.docDisplayConfig}
       ></ai-chat-messages>
       <ai-chat-composer
         style=${styleMap({
@@ -436,6 +438,7 @@ export class AIChatContent extends SignalWatcher(
         .docDisplayConfig=${this.docDisplayConfig}
         .searchMenuConfig=${this.searchMenuConfig}
         .affineWorkspaceDialogService=${this.affineWorkspaceDialogService}
+        .notificationService=${this.notificationService}
         .trackOptions=${{
           where: 'chat-panel',
           control: 'chat-send',

packages/frontend/core/src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts

@@ -1,11 +1,9 @@
-import { toast } from '@affine/component';
 import type { CopilotChatHistoryFragment } from '@affine/graphql';
 import { SignalWatcher, WithDisposable } from '@blocksuite/affine/global/lit';
 import { unsafeCSSVar, unsafeCSSVarV2 } from '@blocksuite/affine/shared/theme';
-import { openFilesWith } from '@blocksuite/affine/shared/utils';
 import type { EditorHost } from '@blocksuite/affine/std';
 import { ShadowlessElement } from '@blocksuite/affine/std';
-import { ArrowUpBigIcon, CloseIcon, ImageIcon } from '@blocksuite/icons/lit';
+import { ArrowUpBigIcon, CloseIcon } from '@blocksuite/icons/lit';
 import { css, html, nothing } from 'lit';
 import { property, query, state } from 'lit/decorators.js';
 import { repeat } from 'lit/directives/repeat.js';
@@ -16,6 +14,7 @@ import { type AIError, AIProvider, type AISendParams } from '../../provider';
 import { reportResponse } from '../../utils/action-reporter';
 import { readBlobAsURL } from '../../utils/image';
 import { mergeStreamObjects } from '../../utils/stream-objects';
+import type { SearchMenuConfig } from '../ai-chat-add-context';
 import type { ChatChip, DocDisplayConfig } from '../ai-chat-chips/type';
 import { isDocChip } from '../ai-chat-chips/utils';
 import {
@@ -23,7 +22,6 @@ import {
   isChatMessage,
   StreamObjectSchema,
 } from '../ai-chat-messages';
-import { MAX_IMAGE_COUNT } from './const';
 import type {
   AIChatInputContext,
   AINetworkSearchConfig,
@@ -292,7 +290,7 @@ export class AIChatInput extends SignalWatcher(
   `;
 
   @property({ attribute: false })
-  accessor independentMode!: boolean;
+  accessor independentMode: boolean | undefined;
 
   @property({ attribute: false })
   accessor host: EditorHost | null | undefined;
@@ -335,6 +333,12 @@ export class AIChatInput extends SignalWatcher(
   @property({ attribute: false })
   accessor updateContext!: (context: Partial<AIChatInputContext>) => void;
 
+  @property({ attribute: false })
+  accessor addImages!: (images: File[]) => void;
+
+  @property({ attribute: false })
+  accessor addChip!: (chip: ChatChip) => Promise<void>;
+
   @property({ attribute: false })
   accessor networkSearchConfig!: AINetworkSearchConfig;
 
@@ -344,6 +348,9 @@ export class AIChatInput extends SignalWatcher(
   @property({ attribute: false })
   accessor docDisplayConfig!: DocDisplayConfig;
 
+  @property({ attribute: false })
+  accessor searchMenuConfig!: SearchMenuConfig;
+
   @property({ attribute: false })
   accessor isRootSession: boolean = true;
 
@@ -357,7 +364,7 @@ export class AIChatInput extends SignalWatcher(
   accessor testId = 'chat-panel-input-container';
 
   @property({ attribute: false })
-  accessor addImages!: (images: File[]) => void;
+  accessor portalContainer: HTMLElement | null = null;
 
   private get _isNetworkActive() {
     return (
@@ -370,10 +377,6 @@ export class AIChatInput extends SignalWatcher(
     return !!this.reasoningConfig.enabled.value;
   }
 
-  private get _isImageUploadDisabled() {
-    return this.chatContextValue.images.length >= MAX_IMAGE_COUNT;
-  }
-
   override connectedCallback() {
     super.connectedCallback();
     this._disposables.add(
@@ -453,14 +456,16 @@ export class AIChatInput extends SignalWatcher(
         data-testid="chat-panel-input"
       ></textarea>
       <div class="chat-panel-input-actions">
-        <div
-          class="chat-input-icon"
-          data-testid="chat-panel-input-image-upload"
-          aria-disabled=${this._isImageUploadDisabled}
-          @click=${this._uploadImageFiles}
-        >
-          ${ImageIcon()}
-          <affine-tooltip>Upload</affine-tooltip>
+        <div class="chat-input-icon">
+          <ai-chat-add-context
+            .docId=${this.docId}
+            .independentMode=${this.independentMode}
+            .addChip=${this.addChip}
+            .addImages=${this.addImages}
+            .docDisplayConfig=${this.docDisplayConfig}
+            .searchMenuConfig=${this.searchMenuConfig}
+            .portalContainer=${this.portalContainer}
+          ></ai-chat-add-context>
         </div>
         <div class="chat-input-footer-spacer"></div>
         <chat-input-preference
@@ -555,18 +560,6 @@ export class AIChatInput extends SignalWatcher(
     this.updateContext({ images: newImages });
   };
 
-  private readonly _uploadImageFiles = async (_e: MouseEvent) => {
-    if (this._isImageUploadDisabled) return;
-
-    const images = await openFilesWith('Images');
-    if (!images) return;
-    if (this.chatContextValue.images.length + images.length > MAX_IMAGE_COUNT) {
-      toast(`You can only upload up to ${MAX_IMAGE_COUNT} images`);
-      return;
-    }
-    this.addImages(images);
-  };
-
   private readonly _onTextareaSend = async (e: MouseEvent | KeyboardEvent) => {
     e.preventDefault();
     e.stopPropagation();

packages/frontend/core/src/blocksuite/ai/components/ai-chat-messages/ai-chat-messages.ts

@@ -6,8 +6,7 @@ import {
   type FeatureFlagService,
   type NotificationService,
 } from '@blocksuite/affine/shared/services';
-import type { EditorHost } from '@blocksuite/affine/std';
-import { ShadowlessElement } from '@blocksuite/affine/std';
+import { type EditorHost, ShadowlessElement } from '@blocksuite/affine/std';
 import type { BaseSelection, ExtensionType } from '@blocksuite/affine/store';
 import { ArrowDownBigIcon as ArrowDownIcon } from '@blocksuite/icons/lit';
 import type { Signal } from '@preact/signals-core';
@@ -21,6 +20,7 @@ import { AffineIcon } from '../../_common/icons';
 import { AIPreloadConfig } from '../../chat-panel/preload-config';
 import { type AIError, AIProvider, UnauthorizedError } from '../../provider';
 import { mergeStreamObjects } from '../../utils/stream-objects';
+import type { DocDisplayConfig } from '../ai-chat-chips';
 import { type ChatContextValue } from '../ai-chat-content/type';
 import type {
   AINetworkSearchConfig,
@@ -43,9 +43,8 @@ export class AIChatMessages extends WithDisposable(ShadowlessElement) {
       display: flex;
       flex-direction: column;
       gap: 24px;
-      height: 100%;
+      min-height: 100%;
       position: relative;
-      overflow-y: auto;
     }
 
     chat-panel-assistant-message,
@@ -152,7 +151,7 @@ export class AIChatMessages extends WithDisposable(ShadowlessElement) {
   accessor avatarUrl = '';
 
   @property({ attribute: false })
-  accessor independentMode!: boolean;
+  accessor independentMode: boolean | undefined;
 
   @property({ attribute: false })
   accessor messages!: HistoryMessage[];
@@ -204,6 +203,9 @@ export class AIChatMessages extends WithDisposable(ShadowlessElement) {
   @property({ attribute: false })
   accessor width: Signal<number | undefined> | undefined;
 
+  @property({ attribute: false })
+  accessor docDisplayService!: DocDisplayConfig;
+
   @query('.chat-panel-messages-container')
   accessor messagesContainer: HTMLDivElement | null = null;
 
@@ -277,7 +279,7 @@ export class AIChatMessages extends WithDisposable(ShadowlessElement) {
       <div
         class=${classMap({
           'chat-panel-messages-container': true,
-          'independent-mode': this.independentMode,
+          'independent-mode': !!this.independentMode,
         })}
         data-testid="chat-panel-messages-container"
         @scroll=${() => this._debouncedOnScroll()}
@@ -329,6 +331,8 @@ export class AIChatMessages extends WithDisposable(ShadowlessElement) {
                     .notificationService=${this.notificationService}
                     .retry=${() => this.retry()}
                     .width=${this.width}
+                    .independentMode=${this.independentMode}
+                    .docDisplayService=${this.docDisplayService}
                   ></chat-message-assistant>`;
                 } else if (isChatAction(item) && this.host) {
                   return html`<chat-message-action
@@ -437,6 +441,7 @@ export class AIChatMessages extends WithDisposable(ShadowlessElement) {
       const last = messages[messages.length - 1];
       if ('content' in last) {
         last.content = '';
+        last.streamObjects = [];
         last.createdAt = new Date().toISOString();
       }
       this.updateContext({

packages/frontend/core/src/blocksuite/ai/components/ai-chat-toolbar/ai-chat-toolbar.ts

@@ -15,6 +15,7 @@ import { css, html } from 'lit';
 import { property, query } from 'lit/decorators.js';
 
 import type { DocDisplayConfig } from '../ai-chat-chips';
+import type { ChatStatus } from '../ai-chat-messages';
 
 export class AIChatToolbar extends WithDisposable(ShadowlessElement) {
   @property({ attribute: false })
@@ -26,6 +27,9 @@ export class AIChatToolbar extends WithDisposable(ShadowlessElement) {
   @property({ attribute: false })
   accessor docId: string | undefined;
 
+  @property({ attribute: false })
+  accessor status!: ChatStatus;
+
   @property({ attribute: false })
   accessor onNewSession!: () => void;
 
@@ -49,6 +53,10 @@ export class AIChatToolbar extends WithDisposable(ShadowlessElement) {
 
   private abortController: AbortController | null = null;
 
+  get isGenerating() {
+    return this.status === 'transmitting' || this.status === 'loading';
+  }
+
   static override styles = css`
     .ai-chat-toolbar {
       display: flex;
@@ -72,6 +80,10 @@ export class AIChatToolbar extends WithDisposable(ShadowlessElement) {
           height: 16px;
           color: ${unsafeCSSVarV2('icon/primary')};
         }
+
+        &[data-disabled='true'] {
+          cursor: not-allowed;
+        }
       }
     }
   `;
@@ -84,7 +96,11 @@ export class AIChatToolbar extends WithDisposable(ShadowlessElement) {
           ${PlusIcon()}
           <affine-tooltip>New Chat</affine-tooltip>
         </div>
-        <div class="chat-toolbar-icon" @click=${this.onTogglePin}>
+        <div
+          class="chat-toolbar-icon"
+          @click=${this.onPinClick}
+          data-disabled=${this.isGenerating}
+        >
           ${pinned ? PinedIcon() : PinIcon()}
           <affine-tooltip>
             ${pinned ? 'Unpin this Chat' : 'Pin this Chat'}
@@ -101,6 +117,16 @@ export class AIChatToolbar extends WithDisposable(ShadowlessElement) {
     `;
   }
 
+  private readonly onPinClick = async () => {
+    if (this.isGenerating) {
+      this.notificationService.toast(
+        'Cannot pin a chat while generating an answer'
+      );
+      return;
+    }
+    await this.onTogglePin();
+  };
+
   private readonly unpinConfirm = async () => {
     if (this.session && this.session.pinned) {
       try {

packages/frontend/core/src/blocksuite/ai/components/ai-chat-toolbar/ai-session-history.ts

@@ -225,7 +225,7 @@ export class AISessionHistory extends WithDisposable(ShadowlessElement) {
               }}
             >
               <div class="ai-session-title">
-                ${session.sessionId}
+                ${session.title || 'New chat'}
                 <affine-tooltip .offsetX=${60}>
                   Click to open this chat
                 </affine-tooltip>

packages/frontend/core/src/blocksuite/ai/components/ai-message-content/stream-objects.ts

@@ -1,7 +1,11 @@
 import type { FeatureFlagService } from '@affine/core/modules/feature-flag';
 import { WithDisposable } from '@blocksuite/affine/global/lit';
 import type { ColorScheme } from '@blocksuite/affine/model';
-import { type EditorHost, ShadowlessElement } from '@blocksuite/affine/std';
+import {
+  type BlockStdScope,
+  type EditorHost,
+  ShadowlessElement,
+} from '@blocksuite/affine/std';
 import type { ExtensionType } from '@blocksuite/affine/store';
 import type { NotificationService } from '@blocksuite/affine-shared/services';
 import type { Signal } from '@preact/signals-core';
@@ -9,6 +13,7 @@ import { css, html, nothing } from 'lit';
 import { property } from 'lit/decorators.js';
 
 import type { AffineAIPanelState } from '../../widgets/ai-panel/type';
+import type { DocDisplayConfig } from '../ai-chat-chips';
 import type { StreamObject } from '../ai-chat-messages';
 
 export class ChatContentStreamObjects extends WithDisposable(
@@ -29,6 +34,9 @@ export class ChatContentStreamObjects extends WithDisposable(
   @property({ attribute: false })
   accessor host: EditorHost | null | undefined;
 
+  @property({ attribute: false })
+  accessor std: BlockStdScope | null | undefined;
+
   @property({ attribute: false })
   accessor state: AffineAIPanelState = 'finished';
 
@@ -47,6 +55,9 @@ export class ChatContentStreamObjects extends WithDisposable(
   @property({ attribute: false })
   accessor notificationService!: NotificationService;
 
+  @property({ attribute: false })
+  accessor docDisplayService!: DocDisplayConfig;
+
   private renderToolCall(streamObject: StreamObject) {
     if (streamObject.type !== 'tool-call') {
       return nothing;
@@ -70,7 +81,7 @@ export class ChatContentStreamObjects extends WithDisposable(
       case 'doc_compose':
         return html`
           <doc-compose-tool
-            .std=${this.host?.std}
+            .std=${this.std || this.host?.std}
             .data=${streamObject}
             .width=${this.width}
             .theme=${this.theme}
@@ -80,9 +91,10 @@ export class ChatContentStreamObjects extends WithDisposable(
       case 'code_artifact':
         return html`
           <code-artifact-tool
-            .std=${this.host?.std}
+            .std=${this.std || this.host?.std}
             .data=${streamObject}
             .width=${this.width}
+            .theme=${this.theme}
           ></code-artifact-tool>
         `;
       case 'doc_edit':
@@ -93,6 +105,21 @@ export class ChatContentStreamObjects extends WithDisposable(
             .notificationService=${this.notificationService}
           ></doc-edit-tool>
         `;
+      case 'doc_semantic_search':
+        return html`<doc-semantic-search-result
+          .data=${streamObject}
+          .width=${this.width}
+        ></doc-semantic-search-result>`;
+      case 'doc_keyword_search':
+        return html`<doc-keyword-search-result
+          .data=${streamObject}
+          .width=${this.width}
+        ></doc-keyword-search-result>`;
+      case 'doc_read':
+        return html`<doc-read-result
+          .data=${streamObject}
+          .width=${this.width}
+        ></doc-read-result>`;
       default: {
         const name = streamObject.toolName + ' tool calling';
         return html`
@@ -125,7 +152,7 @@ export class ChatContentStreamObjects extends WithDisposable(
       case 'doc_compose':
         return html`
           <doc-compose-tool
-            .std=${this.host?.std}
+            .std=${this.std || this.host?.std}
             .data=${streamObject}
             .width=${this.width}
             .theme=${this.theme}
@@ -135,7 +162,7 @@ export class ChatContentStreamObjects extends WithDisposable(
       case 'code_artifact':
         return html`
           <code-artifact-tool
-            .std=${this.host?.std}
+            .std=${this.std || this.host?.std}
             .data=${streamObject}
             .width=${this.width}
             .theme=${this.theme}
@@ -151,6 +178,22 @@ export class ChatContentStreamObjects extends WithDisposable(
             .notificationService=${this.notificationService}
           ></doc-edit-tool>
         `;
+      case 'doc_semantic_search':
+        return html`<doc-semantic-search-result
+          .data=${streamObject}
+          .width=${this.width}
+          .docDisplayService=${this.docDisplayService}
+        ></doc-semantic-search-result>`;
+      case 'doc_keyword_search':
+        return html`<doc-keyword-search-result
+          .data=${streamObject}
+          .width=${this.width}
+        ></doc-keyword-search-result>`;
+      case 'doc_read':
+        return html`<doc-read-result
+          .data=${streamObject}
+          .width=${this.width}
+        ></doc-read-result>`;
       default: {
         const name = streamObject.toolName + ' tool result';
         return html`

packages/frontend/core/src/blocksuite/ai/components/ai-scrollable-text-renderer.ts

@@ -1,6 +1,6 @@
 import { WithDisposable } from '@blocksuite/affine/global/lit';
 import { scrollbarStyle } from '@blocksuite/affine/shared/styles';
-import { type EditorHost, ShadowlessElement } from '@blocksuite/affine/std';
+import { ShadowlessElement } from '@blocksuite/affine/std';
 import type { PropertyValues } from 'lit';
 import { css, html } from 'lit';
 import { property, query } from 'lit/decorators.js';
@@ -81,9 +81,6 @@ export class AIScrollableTextRenderer extends WithDisposable(
   @property({ attribute: false })
   accessor answer!: string;
 
-  @property({ attribute: false })
-  accessor host!: EditorHost;
-
   @property({ attribute: false })
   accessor state: AffineAIPanelState | undefined;
 
@@ -101,19 +98,16 @@ export class AIScrollableTextRenderer extends WithDisposable(
 }
 
 export const createAIScrollableTextRenderer: (
-  host: EditorHost,
   textRendererOptions: TextRendererOptions,
   maxHeight: number,
   autoScroll: boolean
 ) => AffineAIPanelWidgetConfig['answerRenderer'] = (
-  host,
   textRendererOptions,
   maxHeight,
   autoScroll
 ) => {
   return (answer: string, state: AffineAIPanelState | undefined) => {
     return html`<ai-scrollable-text-renderer
-      .host=${host}
       .answer=${answer}
       .state=${state}
       .textRendererOptions=${textRendererOptions}

packages/frontend/core/src/blocksuite/ai/components/ai-tools/artifact-tool.ts

@@ -3,6 +3,7 @@ import { SignalWatcher, WithDisposable } from '@blocksuite/affine/global/lit';
 import type { ColorScheme } from '@blocksuite/affine/model';
 import { ShadowlessElement } from '@blocksuite/affine/std';
 import { type NotificationService } from '@blocksuite/affine-shared/services';
+import { unsafeCSSVar, unsafeCSSVarV2 } from '@blocksuite/affine-shared/theme';
 import type { Signal } from '@preact/signals-core';
 import {
   css,
@@ -29,12 +30,17 @@ export abstract class ArtifactTool<
 > extends SignalWatcher(WithDisposable(ShadowlessElement)) {
   static override styles = css`
     .artifact-tool-card {
-      cursor: pointer;
       margin: 8px 0;
-    }
+      padding: 10px 0;
 
-    .artifact-tool-card:hover {
-      opacity: 0.8;
+      .affine-embed-linked-doc-block {
+        box-shadow: ${unsafeCSSVar('buttonShadow')};
+        cursor: pointer;
+      }
+
+      .affine-embed-linked-doc-block:hover {
+        background-color: ${unsafeCSSVarV2('layer/background/hoverOverlay')};
+      }
     }
   `;
 
@@ -119,23 +125,24 @@ export abstract class ArtifactTool<
 
     return html`
       <div
-        class="affine-embed-linked-doc-block artifact-tool-card ${className ??
-        ''} horizontal"
+        class="artifact-tool-card ${className ?? ''}"
         @click=${this.onCardClick}
       >
-        <div class="affine-embed-linked-doc-content">
-          <div class="affine-embed-linked-doc-content-title">
-            <div class="affine-embed-linked-doc-content-title-icon">
-              ${resolvedIcon}
-            </div>
-            <div class="affine-embed-linked-doc-content-title-text">
-              ${title}
+        <div class="affine-embed-linked-doc-block horizontal">
+          <div class="affine-embed-linked-doc-content">
+            <div class="affine-embed-linked-doc-content-title">
+              <div class="affine-embed-linked-doc-content-title-icon">
+                ${resolvedIcon}
+              </div>
+              <div class="affine-embed-linked-doc-content-title-text">
+                ${title}
+              </div>
             </div>
           </div>
+          ${banner
+            ? html`<div class="affine-embed-linked-doc-banner">${banner}</div>`
+            : nothing}
         </div>
-        ${banner
-          ? html`<div class="affine-embed-linked-doc-banner">${banner}</div>`
-          : nothing}
       </div>
     `;
   }

packages/frontend/core/src/blocksuite/ai/components/ai-tools/artifacts-preview-panel.ts

@@ -62,7 +62,7 @@ export class ArtifactPreviewPanel extends WithDisposable(ShadowlessElement) {
       background-color: ${unsafeCSSVarV2('layer/background/overlayPanel')};
       box-shadow: ${unsafeCSSVar('overlayPanelShadow')};
       height: 100%;
-      overflow-y: auto;
+      overflow: hidden;
     }
 
     .artifact-panel-header {
@@ -71,9 +71,6 @@ export class ArtifactPreviewPanel extends WithDisposable(ShadowlessElement) {
       justify-content: flex-end;
       padding: 0 12px;
       height: 52px;
-      position: sticky;
-      z-index: 1;
-      top: 0;
       background: ${unsafeCSSVarV2('layer/background/overlayPanel')};
     }
 
@@ -105,7 +102,9 @@ export class ArtifactPreviewPanel extends WithDisposable(ShadowlessElement) {
     }
 
     .artifact-panel-content {
+      overflow-y: auto;
       height: calc(100% - 52px);
+      position: relative;
     }
 
     .artifact-panel-close:hover {

packages/frontend/core/src/blocksuite/ai/components/ai-tools/code-artifact.ts

@@ -352,6 +352,8 @@ export class CodeArtifactTool extends ArtifactTool<
 > {
   static override styles = css`
     .code-artifact-preview {
+      overflow: hidden;
+      position: absolute;
       padding: 0;
       width: 100%;
       height: 100%;
@@ -363,6 +365,11 @@ export class CodeArtifactTool extends ArtifactTool<
       height: 100%;
     }
 
+    .code-artifact-preview > code-highlighter {
+      height: 100%;
+      overflow: auto;
+    }
+
     .code-artifact-preview :is(.html-preview-iframe, .html-preview-container) {
       height: 100%;
     }

packages/frontend/core/src/blocksuite/ai/components/ai-tools/doc-compose.ts

@@ -4,7 +4,6 @@ import { getEmbedLinkedDocIcons } from '@blocksuite/affine/blocks/embed-doc';
 import { LoadingIcon } from '@blocksuite/affine/components/icons';
 import { RefNodeSlotsProvider } from '@blocksuite/affine/inlines/reference';
 import type { ColorScheme } from '@blocksuite/affine/model';
-import { NotificationProvider } from '@blocksuite/affine/shared/services';
 import { unsafeCSSVarV2 } from '@blocksuite/affine/shared/theme';
 import { MarkdownTransformer } from '@blocksuite/affine/widgets/linked-doc';
 import { CopyIcon, PageIcon, ToolIcon } from '@blocksuite/icons/lit';
@@ -47,7 +46,6 @@ export class DocComposeTool extends ArtifactTool<
   static override styles = css`
     .doc-compose-result-preview {
       padding: 24px;
-      height: 100%;
     }
 
     .doc-compose-result-preview-title {
@@ -111,7 +109,6 @@ export class DocComposeTool extends ArtifactTool<
 
   protected override getPreviewContent() {
     if (!this.std) return html``;
-    const std = this.std;
     const resultData = this.data;
     const title = this.data.args.title;
     const result = resultData.type === 'tool-result' ? resultData.result : null;
@@ -122,11 +119,11 @@ export class DocComposeTool extends ArtifactTool<
       ${successResult
         ? html`<text-renderer
             .answer=${successResult.markdown}
-            .host=${std.host}
-            .schema=${std.store.schema}
+            .schema=${this.std?.store.schema}
             .options=${{
               customHeading: true,
               extensions: getCustomPageEditorBlockSpecs(),
+              theme: this.theme,
             }}
           ></text-renderer>`
         : html`<div class="doc-compose-result-preview-loading">
@@ -161,7 +158,6 @@ export class DocComposeTool extends ArtifactTool<
           return;
         }
         const workspace = std.store.workspace;
-        const notificationService = std.get(NotificationProvider);
         const refNodeSlots = std.getOptional(RefNodeSlotsProvider);
         const docId = await MarkdownTransformer.importMarkdownToDoc({
           collection: workspace,
@@ -171,7 +167,7 @@ export class DocComposeTool extends ArtifactTool<
           extensions: getStoreManager().config.init().value.get('store'),
         });
         if (docId) {
-          const open = await notificationService.confirm({
+          const open = await this.notificationService.confirm({
             title: 'Open the doc you just created',
             message: 'Doc saved successfully! Would you like to open it now?',
             cancelText: 'Cancel',

packages/frontend/core/src/blocksuite/ai/components/ai-tools/doc-edit.ts

@@ -1,6 +1,8 @@
+import track from '@affine/track';
 import { WithDisposable } from '@blocksuite/affine/global/lit';
 import { unsafeCSSVar, unsafeCSSVarV2 } from '@blocksuite/affine/shared/theme';
 import { type EditorHost, ShadowlessElement } from '@blocksuite/affine/std';
+import { AIStarIconWithAnimation } from '@blocksuite/affine-components/icons';
 import type { NotificationService } from '@blocksuite/affine-shared/services';
 import {
   CloseIcon,
@@ -13,7 +15,9 @@ import {
 } from '@blocksuite/icons/lit';
 import { css, html, nothing } from 'lit';
 import { property, state } from 'lit/decorators.js';
+import { repeat } from 'lit/directives/repeat.js';
 
+import { AIProvider } from '../../provider';
 import { BlockDiffProvider } from '../../services/block-diff';
 import { diffMarkdown } from '../../utils/apply-model/markdown-diff';
 import { copyText } from '../../utils/editor-actions';
@@ -36,8 +40,12 @@ interface DocEditToolResult {
   };
   result:
     | {
-        result: string;
-        content: string;
+        result: {
+          op: string;
+          updates: string;
+          originalContent: string;
+          changedContent: string;
+        }[];
       }
     | ToolError
     | null;
@@ -198,31 +206,108 @@ export class DocEditTool extends WithDisposable(ShadowlessElement) {
   @state()
   accessor isCollapsed = false;
 
+  @state()
+  accessor applyingMap: Record<string, boolean> = {};
+
+  @state()
+  accessor acceptingMap: Record<string, boolean> = {};
+
   get blockDiffService() {
     return this.host?.std.getOptional(BlockDiffProvider);
   }
 
-  private async _handleApply(markdown: string) {
-    if (!this.host) {
-      return;
-    }
-    await this.blockDiffService?.apply(this.host.store, markdown);
+  get isBusy() {
+    return undefined;
   }
 
-  private async _handleReject(changedMarkdown: string) {
-    if (!this.host) {
+  isBusyForOp(op: string) {
+    return this.applyingMap[op] || this.acceptingMap[op];
+  }
+
+  private async _handleApply(op: string, updates: string) {
+    if (
+      !this.host ||
+      this.data.type !== 'tool-result' ||
+      this.isBusyForOp(op)
+    ) {
       return;
     }
-    this.blockDiffService?.setChangedMarkdown(changedMarkdown);
+    this.applyingMap = { ...this.applyingMap, [op]: true };
+    try {
+      const markdown = await AIProvider.context?.applyDocUpdates(
+        this.host.std.workspace.id,
+        this.data.args.doc_id,
+        op,
+        updates
+      );
+      if (!markdown) {
+        return;
+      }
+      track.applyModel.chat.$.apply({
+        instruction: this.data.args.instructions,
+        operation: op,
+      });
+      await this.blockDiffService?.apply(this.host.store, markdown);
+    } catch (error) {
+      this.notificationService.notify({
+        title: 'Failed to apply updates',
+        message: error instanceof Error ? error.message : 'Unknown error',
+        accent: 'error',
+        onClose: function (): void {},
+      });
+    } finally {
+      this.applyingMap = { ...this.applyingMap, [op]: false };
+    }
+  }
+
+  private async _handleReject(op: string) {
+    if (!this.host || this.data.type !== 'tool-result') {
+      return;
+    }
+    // TODO: set the rejected status
+    track.applyModel.chat.$.reject({
+      instruction: this.data.args.instructions,
+      operation: op,
+    });
+    this.blockDiffService?.setChangedMarkdown(null);
     this.blockDiffService?.rejectAll();
   }
 
-  private async _handleAccept(changedMarkdown: string) {
-    if (!this.host) {
+  private async _handleAccept(op: string, updates: string) {
+    if (
+      !this.host ||
+      this.data.type !== 'tool-result' ||
+      this.isBusyForOp(op)
+    ) {
       return;
     }
-    await this.blockDiffService?.apply(this.host.store, changedMarkdown);
-    await this.blockDiffService?.acceptAll(this.host.store);
+    this.acceptingMap = { ...this.acceptingMap, [op]: true };
+    try {
+      const changedMarkdown = await AIProvider.context?.applyDocUpdates(
+        this.host.std.workspace.id,
+        this.data.args.doc_id,
+        op,
+        updates
+      );
+      if (!changedMarkdown) {
+        return;
+      }
+      track.applyModel.chat.$.accept({
+        instruction: this.data.args.instructions,
+        operation: op,
+      });
+      await this.blockDiffService?.apply(this.host.store, changedMarkdown);
+      await this.blockDiffService?.acceptAll(this.host.store);
+    } catch (error) {
+      this.notificationService.notify({
+        title: 'Failed to apply updates',
+        message: error instanceof Error ? error.message : 'Unknown error',
+        accent: 'error',
+        onClose: function (): void {},
+      });
+    } finally {
+      this.acceptingMap = { ...this.acceptingMap, [op]: false };
+    }
   }
 
   private async _toggleCollapse() {
@@ -233,6 +318,7 @@ export class DocEditTool extends WithDisposable(ShadowlessElement) {
     if (!this.host) {
       return;
     }
+    track.applyModel.chat.$.copy();
     const success = await copyText(removeMarkdownComments(changedMarkdown));
     if (success) {
       this.notificationService.notify({
@@ -311,69 +397,84 @@ export class DocEditTool extends WithDisposable(ShadowlessElement) {
 
     const result = this.data.result;
 
-    if (result && 'result' in result && 'content' in result) {
-      const { result: changedMarkdown, content } = result;
-      const { instructions, doc_id: docId } = this.data.args;
+    if (result && 'result' in result && Array.isArray(result.result)) {
+      const { doc_id: docId } = this.data.args;
 
-      const diffs = diffMarkdown(content, changedMarkdown);
-
-      return html`
-        <div class="doc-edit-tool-result-wrapper">
-          <div class="doc-edit-tool-result-title">${instructions}</div>
-          <div
-            class="doc-edit-tool-result-card ${this.isCollapsed
-              ? 'collapsed'
-              : ''}"
-          >
-            <div class="doc-edit-tool-result-card-header">
-              <div class="doc-edit-tool-result-card-header-title">
-                ${PenIcon({
-                  style: `color: ${unsafeCSSVarV2('icon/activated')}`,
-                })}
-                ${docId}
-              </div>
-              <div class="doc-edit-tool-result-card-header-operations">
-                <span @click=${() => this._toggleCollapse()}
-                  >${this.isCollapsed
-                    ? ExpandFullIcon()
-                    : ExpandCloseIcon()}</span
-                >
-                <span @click=${() => this._handleCopy(changedMarkdown)}>
-                  ${CopyIcon()}
-                </span>
-                <button @click=${() => this._handleApply(changedMarkdown)}>
-                  Apply
-                </button>
-              </div>
-            </div>
-            <div class="doc-edit-tool-result-card-content">
-              <div class="doc-edit-tool-result-card-content-title">
-                ${this.renderBlockDiffs(diffs)}
-              </div>
-            </div>
-            <div class="doc-edit-tool-result-card-footer">
+      return repeat(
+        result.result,
+        change => change.op,
+        ({ op, updates, originalContent, changedContent }) => {
+          const diffs = diffMarkdown(originalContent, changedContent);
+          return html`
+            <div class="doc-edit-tool-result-wrapper">
+              <div class="doc-edit-tool-result-title">${op}</div>
               <div
-                class="doc-edit-tool-result-reject"
-                @click=${() => this._handleReject(changedMarkdown)}
+                class="doc-edit-tool-result-card ${this.isCollapsed
+                  ? 'collapsed'
+                  : ''}"
               >
-                ${CloseIcon({
-                  style: `color: ${unsafeCSSVarV2('icon/secondary')}`,
-                })}
-                Reject
-              </div>
-              <div
-                class="doc-edit-tool-result-accept"
-                @click=${() => this._handleAccept(changedMarkdown)}
-              >
-                ${DoneIcon({
-                  style: `color: ${unsafeCSSVarV2('icon/activated')}`,
-                })}
-                Accept
+                <div class="doc-edit-tool-result-card-header">
+                  <div class="doc-edit-tool-result-card-header-title">
+                    ${PenIcon({
+                      style: `color: ${unsafeCSSVarV2('icon/activated')}`,
+                    })}
+                    ${docId}
+                  </div>
+                  <div class="doc-edit-tool-result-card-header-operations">
+                    <span @click=${() => this._toggleCollapse()}
+                      >${this.isCollapsed
+                        ? ExpandFullIcon()
+                        : ExpandCloseIcon()}</span
+                    >
+                    <span @click=${() => this._handleCopy(changedContent)}>
+                      ${CopyIcon()}
+                    </span>
+                    <button
+                      @click=${() => this._handleApply(op, updates)}
+                      ?disabled=${this.isBusyForOp(op)}
+                    >
+                      ${this.applyingMap[op]
+                        ? AIStarIconWithAnimation
+                        : html`Apply`}
+                    </button>
+                  </div>
+                </div>
+                <div class="doc-edit-tool-result-card-content">
+                  <div class="doc-edit-tool-result-card-content-title">
+                    ${this.renderBlockDiffs(diffs)}
+                  </div>
+                </div>
+                <div class="doc-edit-tool-result-card-footer">
+                  <div
+                    class="doc-edit-tool-result-reject"
+                    @click=${() => this._handleReject(op)}
+                  >
+                    ${CloseIcon({
+                      style: `color: ${unsafeCSSVarV2('icon/secondary')}`,
+                    })}
+                    Reject
+                  </div>
+                  <button
+                    class="doc-edit-tool-result-accept"
+                    @click=${() => this._handleAccept(op, updates)}
+                    ?disabled=${this.isBusyForOp(op)}
+                    style="${this.isBusyForOp(op)
+                      ? 'pointer-events: none; opacity: 0.6;'
+                      : ''}"
+                  >
+                    ${this.acceptingMap[op]
+                      ? AIStarIconWithAnimation
+                      : DoneIcon({
+                          style: `color: ${unsafeCSSVarV2('icon/activated')}`,
+                        })}
+                    ${this.acceptingMap[op] ? 'Accepting...' : 'Accept'}
+                  </button>
+                </div>
               </div>
             </div>
-          </div>
-        </div>
-      `;
+          `;
+        }
+      );
     }
 
     return html`

packages/frontend/core/src/blocksuite/ai/components/ai-tools/doc-keyword-search-result.ts

@@ -0,0 +1,70 @@
+import { WithDisposable } from '@blocksuite/global/lit';
+import { PageIcon, SearchIcon } from '@blocksuite/icons/lit';
+import { ShadowlessElement } from '@blocksuite/std';
+import type { Signal } from '@preact/signals-core';
+import { html, nothing } from 'lit';
+import { property } from 'lit/decorators.js';
+
+import type { ToolResult } from './tool-result-card';
+
+interface DocKeywordSearchToolCall {
+  type: 'tool-call';
+  toolCallId: string;
+  toolName: string;
+  args: { query: string };
+}
+
+interface DocKeywordSearchToolResult {
+  type: 'tool-result';
+  toolCallId: string;
+  toolName: string;
+  args: { query: string };
+  result: Array<{
+    title: string;
+    docId: string;
+  }>;
+}
+
+export class DocKeywordSearchResult extends WithDisposable(ShadowlessElement) {
+  @property({ attribute: false })
+  accessor data!: DocKeywordSearchToolCall | DocKeywordSearchToolResult;
+
+  @property({ attribute: false })
+  accessor width: Signal<number | undefined> | undefined;
+
+  renderToolCall() {
+    return html`<tool-call-card
+      .name=${`Searching workspace documents for "${this.data.args.query}"`}
+      .icon=${SearchIcon()}
+      .width=${this.width}
+    ></tool-call-card>`;
+  }
+
+  renderToolResult() {
+    if (this.data.type !== 'tool-result') {
+      return nothing;
+    }
+    let results: ToolResult[] = [];
+    try {
+      results = this.data.result.map(item => ({
+        title: item.title,
+        icon: PageIcon(),
+      }));
+    } catch (err) {
+      console.error('Failed to parse result', err);
+    }
+    return html`<tool-result-card
+      .name=${`Found ${this.data.result.length} pages for "${this.data.args.query}"`}
+      .icon=${SearchIcon()}
+      .width=${this.width}
+      .results=${results}
+    ></tool-result-card>`;
+  }
+
+  protected override render() {
+    if (this.data.type === 'tool-call') {
+      return this.renderToolCall();
+    }
+    return this.renderToolResult();
+  }
+}

packages/frontend/core/src/blocksuite/ai/components/ai-tools/doc-read-result.ts

@@ -0,0 +1,70 @@
+import { WithDisposable } from '@blocksuite/global/lit';
+import { PageIcon, ViewIcon } from '@blocksuite/icons/lit';
+import { ShadowlessElement } from '@blocksuite/std';
+import type { Signal } from '@preact/signals-core';
+import { html, nothing } from 'lit';
+import { property } from 'lit/decorators.js';
+
+interface DocReadToolCall {
+  type: 'tool-call';
+  toolCallId: string;
+  toolName: string;
+  args: { doc_id: string };
+}
+
+interface DocReadToolResult {
+  type: 'tool-result';
+  toolCallId: string;
+  toolName: string;
+  args: { doc_id: string };
+  result: {
+    title: string;
+    markdown: string;
+  };
+}
+
+export class DocReadResult extends WithDisposable(ShadowlessElement) {
+  @property({ attribute: false })
+  accessor data!: DocReadToolCall | DocReadToolResult;
+
+  @property({ attribute: false })
+  accessor width: Signal<number | undefined> | undefined;
+
+  renderToolCall() {
+    // TODO: get document name by doc_id
+    return html`<tool-call-card
+      .name=${`Reading document`}
+      .icon=${ViewIcon()}
+      .width=${this.width}
+    ></tool-call-card>`;
+  }
+
+  renderToolResult() {
+    if (this.data.type !== 'tool-result') {
+      return nothing;
+    }
+    // TODO: better markdown rendering
+    return html`<tool-result-card
+      .name=${`Read "${this.data.result.title}"`}
+      .icon=${ViewIcon()}
+      .width=${this.width}
+      .results=${[
+        {
+          title: this.data.result.title,
+          icon: PageIcon(),
+          content: this.data.result.markdown,
+        },
+      ]}
+    ></tool-result-card>`;
+  }
+
+  protected override render() {
+    if (this.data.type === 'tool-call') {
+      return this.renderToolCall();
+    }
+    if (this.data.type === 'tool-result') {
+      return this.renderToolResult();
+    }
+    return nothing;
+  }
+}

packages/frontend/core/src/blocksuite/ai/components/ai-tools/doc-semantic-search-result.ts

@@ -0,0 +1,108 @@
+import { WithDisposable } from '@blocksuite/global/lit';
+import { AiEmbeddingIcon, PageIcon } from '@blocksuite/icons/lit';
+import { ShadowlessElement } from '@blocksuite/std';
+import type { Signal } from '@preact/signals-core';
+import { html, nothing } from 'lit';
+import { property } from 'lit/decorators.js';
+
+import type { DocDisplayConfig } from '../ai-chat-chips';
+
+interface DocSemanticSearchToolCall {
+  type: 'tool-call';
+  toolCallId: string;
+  toolName: string;
+  args: { query: string };
+}
+
+interface DocSemanticSearchToolResult {
+  type: 'tool-result';
+  toolCallId: string;
+  toolName: string;
+  args: { query: string };
+  result: Array<{
+    content: string;
+    docId: string;
+  }>;
+}
+
+function parseResultContent(content: string) {
+  const properties = [
+    'Title',
+    'Created at',
+    'Updated at',
+    'Created by',
+    'Updated by',
+  ];
+  try {
+    // A row starts with "Title: ${title}\n"
+    const title = content.match(/^Title:\s+(.*)\n/)?.[1];
+    // from first row that not starts with "${propertyName}:" to end of the content
+    const rows = content.split('\n');
+    const startIndex = rows.findIndex(
+      line => !properties.some(property => line.startsWith(`${property}:`))
+    );
+    const text = rows.slice(startIndex).join('\n');
+    return {
+      title,
+      content: text,
+      icon: PageIcon(),
+    };
+  } catch (error) {
+    console.error('Failed to parse result content', error);
+    return null;
+  }
+}
+
+export class DocSemanticSearchResult extends WithDisposable(ShadowlessElement) {
+  @property({ attribute: false })
+  accessor data!: DocSemanticSearchToolCall | DocSemanticSearchToolResult;
+
+  @property({ attribute: false })
+  accessor width: Signal<number | undefined> | undefined;
+
+  @property({ attribute: false })
+  accessor docDisplayService!: DocDisplayConfig;
+
+  renderToolCall() {
+    return html`<tool-call-card
+      .name=${`Finding semantically related pages for "${this.data.args.query}"`}
+      .icon=${AiEmbeddingIcon()}
+      .width=${this.width}
+    ></tool-call-card>`;
+  }
+
+  renderToolResult() {
+    if (this.data.type !== 'tool-result') {
+      return nothing;
+    }
+    return html`<tool-result-card
+      .name=${`Found semantically related pages for "${this.data.args.query}"`}
+      .icon=${AiEmbeddingIcon()}
+      .width=${this.width}
+      .results=${this.data.result
+        .map(result => ({
+          ...parseResultContent(result.content),
+          title: this.docDisplayService.getTitle(result.docId),
+        }))
+        .filter(Boolean)}
+    ></tool-result-card>`;
+  }
+
+  protected override render() {
+    const { data } = this;
+
+    if (data.type === 'tool-call') {
+      return this.renderToolCall();
+    }
+    if (data.type === 'tool-result') {
+      return this.renderToolResult();
+    }
+    return nothing;
+  }
+}
+
+declare global {
+  interface HTMLElementTagNameMap {
+    'doc-semantic-search-result': DocSemanticSearchResult;
+  }
+}

packages/frontend/core/src/blocksuite/ai/components/ai-tools/tool-result-card.ts

@@ -7,7 +7,7 @@ import { type Signal } from '@preact/signals-core';
 import { css, html, nothing, type TemplateResult } from 'lit';
 import { property, state } from 'lit/decorators.js';
 
-interface ToolResult {
+export interface ToolResult {
   title: string;
   icon?: string | TemplateResult<1>;
   content?: string;

packages/frontend/core/src/blocksuite/ai/components/copy-more.ts

@@ -1,5 +1,5 @@
 import type { CopilotChatHistoryFragment } from '@affine/graphql';
-import { Tooltip } from '@blocksuite/affine/components/toolbar';
+import { Tooltip } from '@blocksuite/affine/components/tooltip';
 import { WithDisposable } from '@blocksuite/affine/global/lit';
 import { noop } from '@blocksuite/affine/global/utils';
 import { unsafeCSSVarV2 } from '@blocksuite/affine/shared/theme';
@@ -160,6 +160,7 @@ export class ChatCopyMore extends WithDisposable(LitElement) {
           mainAxis: 0,
           crossAxis: -100,
         });
+        this.disposables.add(() => this._morePopper?.dispose());
       }
     }
   }

packages/frontend/core/src/blocksuite/ai/components/playground/chat.ts

@@ -1,3 +1,4 @@
+import type { WorkspaceDialogService } from '@affine/core/modules/dialogs';
 import type { FeatureFlagService } from '@affine/core/modules/feature-flag';
 import type { AppThemeService } from '@affine/core/modules/theme';
 import type {
@@ -19,7 +20,8 @@ import { throttle } from 'lodash-es';
 import type { AppSidebarConfig } from '../../chat-panel/chat-config';
 import { HISTORY_IMAGE_ACTIONS } from '../../chat-panel/const';
 import { AIProvider } from '../../provider';
-import type { DocDisplayConfig, SearchMenuConfig } from '../ai-chat-chips';
+import type { SearchMenuConfig } from '../ai-chat-add-context';
+import type { DocDisplayConfig } from '../ai-chat-chips';
 import type { ChatContextValue } from '../ai-chat-content';
 import type {
   AINetworkSearchConfig,
@@ -165,6 +167,9 @@ export class PlaygroundChat extends SignalWatcher(
   @property({ attribute: false })
   accessor affineThemeService!: AppThemeService;
 
+  @property({ attribute: false })
+  accessor affineWorkspaceDialogService!: WorkspaceDialogService;
+
   @property({ attribute: false })
   accessor notificationService!: NotificationService;
 
@@ -351,6 +356,8 @@ export class PlaygroundChat extends SignalWatcher(
         .playgroundConfig=${this.playgroundConfig}
         .docDisplayConfig=${this.docDisplayConfig}
         .searchMenuConfig=${this.searchMenuConfig}
+        .notificationService=${this.notificationService}
+        .affineWorkspaceDialogService=${this.affineWorkspaceDialogService}
       ></ai-chat-composer>
     </div>`;
   }