fix(server): exclude outdated doc id style in embedding count (#13269)

fix AI-392
fix AI-393

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

## Summary by CodeRabbit

* **New Features**
* Improved filtering of outdated document ID styles in embedding status
reporting, ensuring more accurate counts of embedded documents.
* Stricter rate limiting applied to workspace embedding status queries
for enhanced system reliability.

* **Bug Fixes**
* Resolved issues with duplicate or outdated document IDs affecting
embedding status totals.

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

.github/workflows/release-desktop.yml

@@ -465,7 +465,7 @@ jobs:
           name: ${{ env.RELEASE_VERSION }}
           draft: ${{ inputs.build-type == 'stable' }}
           prerelease: ${{ inputs.build-type != 'stable' }}
-          tag_name: ${{ env.RELEASE_VERSION}}
+          tag_name: v${{ env.RELEASE_VERSION}}
           files: |
             ./release/*
             ./release/.env.example

.github/workflows/release.yml

@@ -34,6 +34,7 @@ permissions:
   packages: write
   security-events: write
   attestations: write
+  issues: write
 
 jobs:
   prepare:
@@ -74,6 +75,7 @@ jobs:
         with:
           secret: ${{ secrets.GITHUB_TOKEN }}
           approvers: forehalo,fengmk2
+          minimum-approvals: 1
           fail-on-denial: true
           issue-title: Please confirm to release docker image
           issue-body: |

blocksuite/affine/blocks/code/src/code-block-service.ts

@@ -39,6 +39,13 @@ export class CodeBlockHighlighter extends LifeCycleWatcher {
   private readonly _loadTheme = async (
     highlighter: HighlighterCore
   ): Promise<void> => {
+    // It is possible that by the time the highlighter is ready all instances
+    // have already been unmounted. In that case there is no need to load
+    // themes or update state.
+    if (CodeBlockHighlighter._refCount === 0) {
+      return;
+    }
+
     const config = this.std.getOptional(CodeBlockConfigExtension.identifier);
     const darkTheme = config?.theme?.dark ?? CODE_BLOCK_DEFAULT_DARK_THEME;
     const lightTheme = config?.theme?.light ?? CODE_BLOCK_DEFAULT_LIGHT_THEME;
@@ -78,14 +85,27 @@ export class CodeBlockHighlighter extends LifeCycleWatcher {
   override unmounted(): void {
     CodeBlockHighlighter._refCount--;
 
-    // Only dispose the shared highlighter when no instances are using it
-    if (
-      CodeBlockHighlighter._refCount === 0 &&
-      CodeBlockHighlighter._sharedHighlighter
-    ) {
-      CodeBlockHighlighter._sharedHighlighter.dispose();
+    // Dispose the shared highlighter **after** any in-flight creation finishes.
+    if (CodeBlockHighlighter._refCount !== 0) {
+      return;
+    }
+
+    const doDispose = (highlighter: HighlighterCore | null) => {
+      if (highlighter) {
+        highlighter.dispose();
+      }
       CodeBlockHighlighter._sharedHighlighter = null;
       CodeBlockHighlighter._highlighterPromise = null;
+    };
+
+    if (CodeBlockHighlighter._sharedHighlighter) {
+      // Highlighter already created – dispose immediately.
+      doDispose(CodeBlockHighlighter._sharedHighlighter);
+    } else if (CodeBlockHighlighter._highlighterPromise) {
+      // Highlighter still being created – wait for it, then dispose.
+      CodeBlockHighlighter._highlighterPromise
+        .then(doDispose)
+        .catch(console.error);
     }
   }
 }

blocksuite/affine/widgets/keyboard-toolbar/package.json

@@ -20,6 +20,7 @@
     "@blocksuite/affine-block-paragraph": "workspace:*",
     "@blocksuite/affine-block-surface": "workspace:*",
     "@blocksuite/affine-block-surface-ref": "workspace:*",
+    "@blocksuite/affine-block-table": "workspace:*",
     "@blocksuite/affine-components": "workspace:*",
     "@blocksuite/affine-ext-loader": "workspace:*",
     "@blocksuite/affine-fragment-doc-title": "workspace:*",

blocksuite/affine/widgets/keyboard-toolbar/src/config.ts

@@ -18,6 +18,7 @@ import {
 } from '@blocksuite/affine-block-paragraph';
 import { DefaultTool, getSurfaceBlock } from '@blocksuite/affine-block-surface';
 import { insertSurfaceRefBlockCommand } from '@blocksuite/affine-block-surface-ref';
+import { insertTableBlockCommand } from '@blocksuite/affine-block-table';
 import { toggleEmbedCardCreateModal } from '@blocksuite/affine-components/embed-card-modal';
 import { toast } from '@blocksuite/affine-components/toast';
 import { insertInlineLatex } from '@blocksuite/affine-inline-latex';
@@ -40,14 +41,20 @@ import {
   deleteSelectedModelsCommand,
   draftSelectedModelsCommand,
   duplicateSelectedModelsCommand,
+  focusBlockEnd,
   getBlockSelectionsCommand,
   getSelectedModelsCommand,
   getTextSelectionCommand,
 } from '@blocksuite/affine-shared/commands';
 import { REFERENCE_NODE } from '@blocksuite/affine-shared/consts';
+import {
+  FeatureFlagService,
+  TelemetryProvider,
+} from '@blocksuite/affine-shared/services';
 import type { AffineTextStyleAttributes } from '@blocksuite/affine-shared/types';
 import {
   createDefaultDoc,
+  isInsideBlockByFlavour,
   openSingleFileWith,
   type Signal,
 } from '@blocksuite/affine-shared/utils';
@@ -87,6 +94,7 @@ import {
   RedoIcon,
   RightTabIcon,
   StrikeThroughIcon,
+  TableIcon,
   TeXIcon,
   TextIcon,
   TodayIcon,
@@ -258,6 +266,62 @@ const textToolActionItems: KeyboardToolbarActionItem[] = [
         .run();
     },
   },
+  {
+    name: 'Table',
+    icon: TableIcon(),
+    showWhen: ({ std, rootComponent: { model } }) =>
+      std.store.schema.flavourSchemaMap.has('affine:table') &&
+      !isInsideBlockByFlavour(std.store, model, 'affine:edgeless-text'),
+    action: ({ std }) => {
+      std.command
+        .chain()
+        .pipe(getSelectedModelsCommand)
+        .pipe(insertTableBlockCommand, {
+          place: 'after',
+          removeEmptyLine: true,
+        })
+        .pipe(({ insertedTableBlockId }) => {
+          if (insertedTableBlockId) {
+            const telemetry = std.getOptional(TelemetryProvider);
+            telemetry?.track('BlockCreated', {
+              blockType: 'affine:table',
+            });
+          }
+        })
+        .run();
+    },
+  },
+  {
+    name: 'Callout',
+    icon: FontIcon(),
+    showWhen: ({ std, rootComponent: { model } }) => {
+      return (
+        std.get(FeatureFlagService).getFlag('enable_callout') &&
+        !isInsideBlockByFlavour(model.store, model, 'affine:edgeless-text')
+      );
+    },
+    action: ({ rootComponent: { model }, std }) => {
+      const { store } = model;
+      const parent = store.getParent(model);
+      if (!parent) return;
+
+      const index = parent.children.indexOf(model);
+      if (index === -1) return;
+      const calloutId = store.addBlock('affine:callout', {}, parent, index + 1);
+      if (!calloutId) return;
+      const paragraphId = store.addBlock('affine:paragraph', {}, calloutId);
+      if (!paragraphId) return;
+      std.host.updateComplete
+        .then(() => {
+          const paragraph = std.view.getBlock(paragraphId);
+          if (!paragraph) return;
+          std.command.exec(focusBlockEnd, {
+            focusBlock: paragraph,
+          });
+        })
+        .catch(console.error);
+    },
+  },
 ];
 
 const listToolActionItems: KeyboardToolbarActionItem[] = [

blocksuite/affine/widgets/keyboard-toolbar/tsconfig.json

@@ -17,6 +17,7 @@
     { "path": "../../blocks/paragraph" },
     { "path": "../../blocks/surface" },
     { "path": "../../blocks/surface-ref" },
+    { "path": "../../blocks/table" },
     { "path": "../../components" },
     { "path": "../../ext-loader" },
     { "path": "../../fragments/doc-title" },

blocksuite/affine/widgets/linked-doc/src/linked-doc-popover.ts

@@ -343,7 +343,18 @@ export class LinkedDocPopover extends SignalWatcher(
   override willUpdate() {
     if (!this.hasUpdated) {
       const updatePosition = throttle(() => {
-        this._position = getPopperPosition(this, this.context.startNativeRange);
+        this._position = getPopperPosition(
+          {
+            getBoundingClientRect: () => {
+              return {
+                ...this.getBoundingClientRect(),
+                // Workaround: the width of the popover is zero when it is not rendered
+                width: 280,
+              };
+            },
+          },
+          this.context.startNativeRange
+        );
       }, 10);
 
       this.disposables.addFromEvent(window, 'resize', updatePosition);

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

@@ -396,6 +396,15 @@ Generated by [AVA](https://avajs.dev).
           },
         ],
       },
+      {
+        args: [
+          'copilot.workspace.cleanupTrashedDocEmbeddings',
+          {},
+          {
+            jobId: 'daily-copilot-cleanup-trashed-doc-embeddings',
+          },
+        ],
+      },
     ]
 
 > cleanup empty sessions calls

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

@@ -461,6 +461,29 @@ test('should create message correctly', async t => {
         sessionId,
         undefined,
         undefined,
+        new File([new Uint8Array(pngData)], '1.png', { type: 'image/png' })
+      );
+      t.truthy(messageId, 'should be able to create message with blob');
+    }
+
+    // with attachments
+    {
+      const { id } = await createWorkspace(app);
+      const sessionId = await createCopilotSession(
+        app,
+        id,
+        randomUUID(),
+        textPromptName
+      );
+      const smallestPng =
+        'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAgAAAAIAQMAAAD+wSzIAAAABlBMVEX///+/v7+jQ3Y5AAAADklEQVQI12P4AIX8EAgALgAD/aNpbtEAAAAASUVORK5CYII';
+      const pngData = await fetch(smallestPng).then(res => res.arrayBuffer());
+      const messageId = await createCopilotMessage(
+        app,
+        sessionId,
+        undefined,
+        undefined,
+        undefined,
         [new File([new Uint8Array(pngData)], '1.png', { type: 'image/png' })]
       );
       t.truthy(messageId, 'should be able to create message with blobs');

packages/backend/server/src/__tests__/e2e/doc-service/__snapshots__/controller.spec.ts.md

@@ -13,74 +13,45 @@ Generated by [AVA](https://avajs.dev).
       ␊
       ␊
       ␊
-      ␊
-      ␊
       # You own your data, with no compromises␊
       ␊
-      ␊
       ## Local-first & Real-time collaborative␊
       ␊
-      ␊
       We love the idea proposed by Ink & Switch in the famous article about you owning your data, despite the cloud. Furthermore, AFFiNE is the first all-in-one workspace that keeps your data ownership with no compromises on real-time collaboration and editing experience.␊
       ␊
-      ␊
       AFFiNE is a local-first application upon CRDTs with real-time collaboration support. Your data is always stored locally while multiple nodes remain synced in real-time.␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       ### Blocks that assemble your next docs, tasks kanban or whiteboard␊
       ␊
-      ␊
       There is a large overlap of their atomic "building blocks" between these apps. They are neither open source nor have a plugin system like VS Code for contributors to customize. We want to have something that contains all the features we love and goes one step further.␊
       ␊
-      ␊
       We are building AFFiNE to be a fundamental open source platform that contains all the building blocks for docs, task management and visual collaboration, hoping you can shape your next workflow with us that can make your life better and also connect others, too.␊
       ␊
-      ␊
       If you want to learn more about the product design of AFFiNE, here goes the concepts:␊
       ␊
-      ␊
       To Shape, not to adapt. AFFiNE is built for individuals & teams who care about their data, who refuse vendor lock-in, and who want to have control over their essential tools.␊
       ␊
-      ␊
       ## A true canvas for blocks in any form␊
       ␊
-      ␊
       [Many editor apps](http://notion.so) claimed to be a canvas for productivity. Since _the Mother of All Demos,_ Douglas Engelbart, a creative and programable digital workspace has been a pursuit and an ultimate mission for generations of tool makers.␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       "We shape our tools and thereafter our tools shape us”. A lot of pioneers have inspired us a long the way, e.g.:␊
       ␊
-      ␊
       * Quip & Notion with their great concept of "everything is a block"␊
-      ␊
-      ␊
       * Trello with their Kanban␊
-      ␊
-      ␊
       * Airtable & Miro with their no-code programable datasheets␊
-      ␊
-      ␊
       * Miro & Whimiscal with their edgeless visual whiteboard␊
-      ␊
-      ␊
       * Remnote & Capacities with their object-based tag system␊
-      ␊
-      ␊
       For more details, please refer to our [RoadMap](https://docs.affine.pro/docs/core-concepts/roadmap)␊
       ␊
-      ␊
       ## Self Host␊
       ␊
-      ␊
       Self host AFFiNE␊
       ␊
-      ␊
       ||Title|Tag|␊
       |---|---|---|␊
       |Affine Development|Affine Development|<span data-affine-option data-value="AxSe-53xjX" data-option-color="var(--affine-tag-pink)">AFFiNE</span>|␊
@@ -91,16 +62,12 @@ Generated by [AVA](https://avajs.dev).
       |Miro & Whimiscal with their edgeless visual whiteboard|Miro & Whimiscal with their edgeless visual whiteboard|<span data-affine-option data-value="HgHsKOUINZ" data-option-color="var(--affine-tag-blue)">Reference</span>|␊
       |Remnote & Capacities with their object-based tag system|Remnote & Capacities with their object-based tag system||␊
       ␊
-      ␊
       ## Affine Development␊
       ␊
-      ␊
       For developer or installation guides, please go to [AFFiNE Development](https://docs.affine.pro/docs/development/quick-start)␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       `,
       title: 'Write, Draw, Plan all at Once.',
     }

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

@@ -89,3 +89,19 @@ Generated by [AVA](https://avajs.dev).
 > should not find docs to embed
 
     0
+
+## should filter outdated doc id style in embedding status
+
+> should include modern doc format
+
+    {
+      embedded: 0,
+      total: 1,
+    }
+
+> should count docs after filtering outdated
+
+    {
+      embedded: 1,
+      total: 1,
+    }

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

@@ -164,11 +164,14 @@ test('should insert embedding by doc id', async t => {
     );
 
     {
-      const ret = await t.context.copilotContext.hasWorkspaceEmbedding(
+      const ret = await t.context.copilotContext.listWorkspaceEmbedding(
         workspace.id,
         [docId]
       );
-      t.true(ret.has(docId), 'should return doc id when embedding is inserted');
+      t.true(
+        ret.includes(docId),
+        'should return doc id when embedding is inserted'
+      );
     }
 
     {
@@ -317,8 +320,8 @@ test('should merge doc status correctly', async t => {
 
     const hasEmbeddingStub = Sinon.stub(
       t.context.copilotContext,
-      'hasWorkspaceEmbedding'
-    ).resolves(new Set<string>());
+      'listWorkspaceEmbedding'
+    ).resolves([]);
 
     const stubResult = await t.context.copilotContext.mergeDocStatus(
       workspace.id,

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

@@ -214,6 +214,21 @@ test('should insert and search embedding', async t => {
     );
     t.false(results.includes(docId), 'docs containing `$` should be excluded');
   }
+
+  {
+    const docId = 'empty_doc';
+    await t.context.doc.upsert({
+      spaceId: workspace.id,
+      docId: docId,
+      blob: Uint8Array.from([0, 0]),
+      timestamp: Date.now(),
+      editorId: user.id,
+    });
+    const results = await t.context.copilotWorkspace.findDocsToEmbed(
+      workspace.id
+    );
+    t.false(results.includes(docId), 'empty documents should be excluded');
+  }
 });
 
 test('should check need to be embedded', async t => {
@@ -291,3 +306,50 @@ test('should check embedding table', async t => {
   //   t.false(ret, 'should return false when embedding table is not available');
   // }
 });
+
+test('should filter outdated doc id style in embedding status', async t => {
+  const docId = randomUUID();
+  const outdatedDocId = `${workspace.id}:space:${docId}`;
+
+  await t.context.doc.upsert({
+    spaceId: workspace.id,
+    docId,
+    blob: Uint8Array.from([1, 2, 3]),
+    timestamp: Date.now(),
+    editorId: user.id,
+  });
+
+  await t.context.doc.upsert({
+    spaceId: workspace.id,
+    docId: outdatedDocId,
+    blob: Uint8Array.from([1, 2, 3]),
+    timestamp: Date.now(),
+    editorId: user.id,
+  });
+
+  {
+    const status = await t.context.copilotWorkspace.getEmbeddingStatus(
+      workspace.id
+    );
+    t.snapshot(status, 'should include modern doc format');
+  }
+
+  {
+    await t.context.copilotContext.insertWorkspaceEmbedding(
+      workspace.id,
+      docId,
+      [
+        {
+          index: 0,
+          content: 'content',
+          embedding: Array.from({ length: 1024 }, () => 1),
+        },
+      ]
+    );
+
+    const status = await t.context.copilotWorkspace.getEmbeddingStatus(
+      workspace.id
+    );
+    t.snapshot(status, 'should count docs after filtering outdated');
+  }
+});

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

@@ -554,52 +554,73 @@ export async function createCopilotMessage(
   sessionId: string,
   content?: string,
   attachments?: string[],
+  blob?: File,
   blobs?: File[],
   params?: Record<string, string>
 ): Promise<string> {
-  let resp = app
-    .POST('/graphql')
-    .set({ 'x-request-id': 'test', 'x-operation-name': 'test' })
-    .field(
-      'operations',
-      JSON.stringify({
-        query: `
+  const gql = {
+    query: `
           mutation createCopilotMessage($options: CreateChatMessageInput!) {
             createCopilotMessage(options: $options)
           }
         `,
-        variables: {
-          options: { sessionId, content, attachments, blobs: [], params },
-        },
-      })
-    )
-    .field(
-      'map',
-      JSON.stringify(
-        Array.from<any>({ length: blobs?.length ?? 0 }).reduce(
-          (acc, _, idx) => {
-            acc[idx.toString()] = [`variables.options.blobs.${idx}`];
-            return acc;
-          },
-          {}
-        )
-      )
-    );
-  if (blobs && blobs.length) {
-    for (const [idx, file] of blobs.entries()) {
-      resp = resp.attach(
-        idx.toString(),
-        Buffer.from(await file.arrayBuffer()),
-        {
-          filename: file.name || `file${idx}`,
-          contentType: file.type || 'application/octet-stream',
-        }
+    variables: {
+      options: {
+        sessionId,
+        content,
+        attachments,
+        blob: null,
+        blobs: [],
+        params,
+      },
+    },
+  };
+
+  let resp = app
+    .POST('/graphql')
+    .set({ 'x-request-id': 'test', 'x-operation-name': 'test' });
+  if (blob || blobs) {
+    resp = resp.field('operations', JSON.stringify(gql));
+
+    if (blob) {
+      resp = resp.field(
+        'map',
+        JSON.stringify({ '0': ['variables.options.blob'] })
       );
+      resp = resp.attach('0', Buffer.from(await blob.arrayBuffer()), {
+        filename: blob.name || 'file',
+        contentType: blob.type || 'application/octet-stream',
+      });
+    } else if (blobs && blobs.length) {
+      resp = resp.field(
+        'map',
+        JSON.stringify(
+          Array.from<any>({ length: blobs?.length ?? 0 }).reduce(
+            (acc, _, idx) => {
+              acc[idx.toString()] = [`variables.options.blobs.${idx}`];
+              return acc;
+            },
+            {}
+          )
+        )
+      );
+      for (const [idx, file] of blobs.entries()) {
+        resp = resp.attach(
+          idx.toString(),
+          Buffer.from(await file.arrayBuffer()),
+          {
+            filename: file.name || `file${idx}`,
+            contentType: file.type || 'application/octet-stream',
+          }
+        );
+      }
     }
+  } else {
+    resp = resp.send(gql);
   }
 
   const res = await resp.expect(200);
-
+  console.log('createCopilotMessage', res.body);
   return res.body.data.createCopilotMessage;
 }
 

packages/backend/server/src/base/utils/promise.ts

@@ -1,3 +1,5 @@
+import { setTimeout } from 'node:timers/promises';
+
 import { defer as rxjsDefer, retry } from 'rxjs';
 
 export class RetryablePromise<T> extends Promise<T> {
@@ -48,3 +50,7 @@ export function defer(dispose: () => Promise<void>) {
     [Symbol.asyncDispose]: dispose,
   };
 }
+
+export function sleep(ms: number): Promise<void> {
+  return setTimeout(ms);
+}

packages/backend/server/src/core/doc/__tests__/__snapshots__/reader-from-database.spec.ts.md

@@ -13,74 +13,45 @@ Generated by [AVA](https://avajs.dev).
       ␊
       ␊
       ␊
-      ␊
-      ␊
       # You own your data, with no compromises␊
       ␊
-      ␊
       ## Local-first & Real-time collaborative␊
       ␊
-      ␊
       We love the idea proposed by Ink & Switch in the famous article about you owning your data, despite the cloud. Furthermore, AFFiNE is the first all-in-one workspace that keeps your data ownership with no compromises on real-time collaboration and editing experience.␊
       ␊
-      ␊
       AFFiNE is a local-first application upon CRDTs with real-time collaboration support. Your data is always stored locally while multiple nodes remain synced in real-time.␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       ### Blocks that assemble your next docs, tasks kanban or whiteboard␊
       ␊
-      ␊
       There is a large overlap of their atomic "building blocks" between these apps. They are neither open source nor have a plugin system like VS Code for contributors to customize. We want to have something that contains all the features we love and goes one step further.␊
       ␊
-      ␊
       We are building AFFiNE to be a fundamental open source platform that contains all the building blocks for docs, task management and visual collaboration, hoping you can shape your next workflow with us that can make your life better and also connect others, too.␊
       ␊
-      ␊
       If you want to learn more about the product design of AFFiNE, here goes the concepts:␊
       ␊
-      ␊
       To Shape, not to adapt. AFFiNE is built for individuals & teams who care about their data, who refuse vendor lock-in, and who want to have control over their essential tools.␊
       ␊
-      ␊
       ## A true canvas for blocks in any form␊
       ␊
-      ␊
       [Many editor apps](http://notion.so) claimed to be a canvas for productivity. Since _the Mother of All Demos,_ Douglas Engelbart, a creative and programable digital workspace has been a pursuit and an ultimate mission for generations of tool makers.␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       "We shape our tools and thereafter our tools shape us”. A lot of pioneers have inspired us a long the way, e.g.:␊
       ␊
-      ␊
       * Quip & Notion with their great concept of "everything is a block"␊
-      ␊
-      ␊
       * Trello with their Kanban␊
-      ␊
-      ␊
       * Airtable & Miro with their no-code programable datasheets␊
-      ␊
-      ␊
       * Miro & Whimiscal with their edgeless visual whiteboard␊
-      ␊
-      ␊
       * Remnote & Capacities with their object-based tag system␊
-      ␊
-      ␊
       For more details, please refer to our [RoadMap](https://docs.affine.pro/docs/core-concepts/roadmap)␊
       ␊
-      ␊
       ## Self Host␊
       ␊
-      ␊
       Self host AFFiNE␊
       ␊
-      ␊
       ||Title|Tag|␊
       |---|---|---|␊
       |Affine Development|Affine Development|<span data-affine-option data-value="AxSe-53xjX" data-option-color="var(--affine-tag-pink)">AFFiNE</span>|␊
@@ -91,16 +62,12 @@ Generated by [AVA](https://avajs.dev).
       |Miro & Whimiscal with their edgeless visual whiteboard|Miro & Whimiscal with their edgeless visual whiteboard|<span data-affine-option data-value="HgHsKOUINZ" data-option-color="var(--affine-tag-blue)">Reference</span>|␊
       |Remnote & Capacities with their object-based tag system|Remnote & Capacities with their object-based tag system||␊
       ␊
-      ␊
       ## Affine Development␊
       ␊
-      ␊
       For developer or installation guides, please go to [AFFiNE Development](https://docs.affine.pro/docs/development/quick-start)␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       `,
       title: 'Write, Draw, Plan all at Once.',
     }

packages/backend/server/src/core/doc/__tests__/__snapshots__/reader-from-rpc.spec.ts.md

@@ -13,74 +13,45 @@ Generated by [AVA](https://avajs.dev).
       ␊
       ␊
       ␊
-      ␊
-      ␊
       # You own your data, with no compromises␊
       ␊
-      ␊
       ## Local-first & Real-time collaborative␊
       ␊
-      ␊
       We love the idea proposed by Ink & Switch in the famous article about you owning your data, despite the cloud. Furthermore, AFFiNE is the first all-in-one workspace that keeps your data ownership with no compromises on real-time collaboration and editing experience.␊
       ␊
-      ␊
       AFFiNE is a local-first application upon CRDTs with real-time collaboration support. Your data is always stored locally while multiple nodes remain synced in real-time.␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       ### Blocks that assemble your next docs, tasks kanban or whiteboard␊
       ␊
-      ␊
       There is a large overlap of their atomic "building blocks" between these apps. They are neither open source nor have a plugin system like VS Code for contributors to customize. We want to have something that contains all the features we love and goes one step further.␊
       ␊
-      ␊
       We are building AFFiNE to be a fundamental open source platform that contains all the building blocks for docs, task management and visual collaboration, hoping you can shape your next workflow with us that can make your life better and also connect others, too.␊
       ␊
-      ␊
       If you want to learn more about the product design of AFFiNE, here goes the concepts:␊
       ␊
-      ␊
       To Shape, not to adapt. AFFiNE is built for individuals & teams who care about their data, who refuse vendor lock-in, and who want to have control over their essential tools.␊
       ␊
-      ␊
       ## A true canvas for blocks in any form␊
       ␊
-      ␊
       [Many editor apps](http://notion.so) claimed to be a canvas for productivity. Since _the Mother of All Demos,_ Douglas Engelbart, a creative and programable digital workspace has been a pursuit and an ultimate mission for generations of tool makers.␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       "We shape our tools and thereafter our tools shape us”. A lot of pioneers have inspired us a long the way, e.g.:␊
       ␊
-      ␊
       * Quip & Notion with their great concept of "everything is a block"␊
-      ␊
-      ␊
       * Trello with their Kanban␊
-      ␊
-      ␊
       * Airtable & Miro with their no-code programable datasheets␊
-      ␊
-      ␊
       * Miro & Whimiscal with their edgeless visual whiteboard␊
-      ␊
-      ␊
       * Remnote & Capacities with their object-based tag system␊
-      ␊
-      ␊
       For more details, please refer to our [RoadMap](https://docs.affine.pro/docs/core-concepts/roadmap)␊
       ␊
-      ␊
       ## Self Host␊
       ␊
-      ␊
       Self host AFFiNE␊
       ␊
-      ␊
       ||Title|Tag|␊
       |---|---|---|␊
       |Affine Development|Affine Development|<span data-affine-option data-value="AxSe-53xjX" data-option-color="var(--affine-tag-pink)">AFFiNE</span>|␊
@@ -91,16 +62,12 @@ Generated by [AVA](https://avajs.dev).
       |Miro & Whimiscal with their edgeless visual whiteboard|Miro & Whimiscal with their edgeless visual whiteboard|<span data-affine-option data-value="HgHsKOUINZ" data-option-color="var(--affine-tag-blue)">Reference</span>|␊
       |Remnote & Capacities with their object-based tag system|Remnote & Capacities with their object-based tag system||␊
       ␊
-      ␊
       ## Affine Development␊
       ␊
-      ␊
       For developer or installation guides, please go to [AFFiNE Development](https://docs.affine.pro/docs/development/quick-start)␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       `,
       title: 'Write, Draw, Plan all at Once.',
     }

packages/backend/server/src/core/mail/job.ts

@@ -1,7 +1,7 @@
 import { Injectable } from '@nestjs/common';
 import { getStreamAsBuffer } from 'get-stream';
 
-import { JOB_SIGNAL, OnJob } from '../../base';
+import { JOB_SIGNAL, OnJob, sleep } from '../../base';
 import { type MailName, MailProps, Renderers } from '../../mails';
 import { UserProps, WorkspaceProps } from '../../mails/components';
 import { Models } from '../../models';
@@ -34,7 +34,7 @@ type SendMailJob<Mail extends MailName = MailName, Props = MailProps<Mail>> = {
 
 declare global {
   interface Jobs {
-    'notification.sendMail': {
+    'notification.sendMail': { startTime: number } & {
       [K in MailName]: SendMailJob<K>;
     }[MailName];
   }
@@ -50,7 +50,12 @@ export class MailJob {
   ) {}
 
   @OnJob('notification.sendMail')
-  async sendMail({ name, to, props }: Jobs['notification.sendMail']) {
+  async sendMail({
+    startTime,
+    name,
+    to,
+    props,
+  }: Jobs['notification.sendMail']) {
     let options: Partial<SendOptions> = {};
 
     for (const key in props) {
@@ -100,8 +105,15 @@ export class MailJob {
       )),
       ...options,
     });
+    if (result === false) {
+      // wait for a while before retrying
+      const elapsed = Date.now() - startTime;
+      const retryDelay = Math.min(30 * 1000, Math.round(elapsed / 2000) * 1000);
+      await sleep(retryDelay);
+      return JOB_SIGNAL.Retry;
+    }
 
-    return result === false ? JOB_SIGNAL.Retry : undefined;
+    return undefined;
   }
 
   private async fetchWorkspaceProps(workspaceId: string) {

packages/backend/server/src/core/mail/mailer.ts

@@ -15,11 +15,14 @@ export class Mailer {
    *
    * @note never throw
    */
-  async trySend(command: Jobs['notification.sendMail']) {
+  async trySend(command: Omit<Jobs['notification.sendMail'], 'startTime'>) {
     return this.send(command, true);
   }
 
-  async send(command: Jobs['notification.sendMail'], suppressError = false) {
+  async send(
+    command: Omit<Jobs['notification.sendMail'], 'startTime'>,
+    suppressError = false
+  ) {
     if (!this.sender.configured) {
       if (suppressError) {
         return false;
@@ -28,7 +31,12 @@ export class Mailer {
     }
 
     try {
-      await this.queue.add('notification.sendMail', command);
+      await this.queue.add(
+        'notification.sendMail',
+        Object.assign({}, command, {
+          startTime: Date.now(),
+        }) as Jobs['notification.sendMail']
+      );
       return true;
     } catch {
       return false;

packages/backend/server/src/core/utils/__tests__/__snapshots__/blocksute.spec.ts.md

@@ -1376,74 +1376,45 @@ Generated by [AVA](https://avajs.dev).
       ␊
       ␊
       ␊
-      ␊
-      ␊
       # You own your data, with no compromises␊
       ␊
-      ␊
       ## Local-first & Real-time collaborative␊
       ␊
-      ␊
       We love the idea proposed by Ink & Switch in the famous article about you owning your data, despite the cloud. Furthermore, AFFiNE is the first all-in-one workspace that keeps your data ownership with no compromises on real-time collaboration and editing experience.␊
       ␊
-      ␊
       AFFiNE is a local-first application upon CRDTs with real-time collaboration support. Your data is always stored locally while multiple nodes remain synced in real-time.␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       ### Blocks that assemble your next docs, tasks kanban or whiteboard␊
       ␊
-      ␊
       There is a large overlap of their atomic "building blocks" between these apps. They are neither open source nor have a plugin system like VS Code for contributors to customize. We want to have something that contains all the features we love and goes one step further.␊
       ␊
-      ␊
       We are building AFFiNE to be a fundamental open source platform that contains all the building blocks for docs, task management and visual collaboration, hoping you can shape your next workflow with us that can make your life better and also connect others, too.␊
       ␊
-      ␊
       If you want to learn more about the product design of AFFiNE, here goes the concepts:␊
       ␊
-      ␊
       To Shape, not to adapt. AFFiNE is built for individuals & teams who care about their data, who refuse vendor lock-in, and who want to have control over their essential tools.␊
       ␊
-      ␊
       ## A true canvas for blocks in any form␊
       ␊
-      ␊
       [Many editor apps](http://notion.so) claimed to be a canvas for productivity. Since _the Mother of All Demos,_ Douglas Engelbart, a creative and programable digital workspace has been a pursuit and an ultimate mission for generations of tool makers.␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       "We shape our tools and thereafter our tools shape us”. A lot of pioneers have inspired us a long the way, e.g.:␊
       ␊
-      ␊
       * Quip & Notion with their great concept of "everything is a block"␊
-      ␊
-      ␊
       * Trello with their Kanban␊
-      ␊
-      ␊
       * Airtable & Miro with their no-code programable datasheets␊
-      ␊
-      ␊
       * Miro & Whimiscal with their edgeless visual whiteboard␊
-      ␊
-      ␊
       * Remnote & Capacities with their object-based tag system␊
-      ␊
-      ␊
       For more details, please refer to our [RoadMap](https://docs.affine.pro/docs/core-concepts/roadmap)␊
       ␊
-      ␊
       ## Self Host␊
       ␊
-      ␊
       Self host AFFiNE␊
       ␊
-      ␊
       ||Title|Tag|␊
       |---|---|---|␊
       |Affine Development|Affine Development|<span data-affine-option data-value="AxSe-53xjX" data-option-color="var(--affine-tag-pink)">AFFiNE</span>|␊
@@ -1454,16 +1425,12 @@ Generated by [AVA](https://avajs.dev).
       |Miro & Whimiscal with their edgeless visual whiteboard|Miro & Whimiscal with their edgeless visual whiteboard|<span data-affine-option data-value="HgHsKOUINZ" data-option-color="var(--affine-tag-blue)">Reference</span>|␊
       |Remnote & Capacities with their object-based tag system|Remnote & Capacities with their object-based tag system||␊
       ␊
-      ␊
       ## Affine Development␊
       ␊
-      ␊
       For developer or installation guides, please go to [AFFiNE Development](https://docs.affine.pro/docs/development/quick-start)␊
       ␊
       ␊
       ␊
-      ␊
-      ␊
       `,
       title: 'Write, Draw, Plan all at Once.',
     }
@@ -1476,113 +1443,80 @@ Generated by [AVA](https://avajs.dev).
       markdown: `<!-- block_id=FoPQcAyV_m flavour=affine:paragraph -->␊
       AFFiNE is an open source all in one workspace, an operating system for all the building blocks of your team wiki, knowledge management and digital assets and a better alternative to Notion and Miro.␊
       ␊
-      ␊
       <!-- block_id=oz48nn_zp8 flavour=affine:paragraph -->␊
       ␊
       ␊
-      ␊
       <!-- block_id=g8a-D9-jXS flavour=affine:paragraph -->␊
       # You own your data, with no compromises␊
       ␊
-      ␊
       <!-- block_id=J8lHN1GR_5 flavour=affine:paragraph -->␊
       ## Local-first & Real-time collaborative␊
       ␊
-      ␊
       <!-- block_id=xCuWdM0VLz flavour=affine:paragraph -->␊
       We love the idea proposed by Ink & Switch in the famous article about you owning your data, despite the cloud. Furthermore, AFFiNE is the first all-in-one workspace that keeps your data ownership with no compromises on real-time collaboration and editing experience.␊
       ␊
-      ␊
       <!-- block_id=zElMi0tViK flavour=affine:paragraph -->␊
       AFFiNE is a local-first application upon CRDTs with real-time collaboration support. Your data is always stored locally while multiple nodes remain synced in real-time.␊
       ␊
-      ␊
       <!-- block_id=Z4rK0OF9Wk flavour=affine:paragraph -->␊
       ␊
       ␊
-      ␊
       <!-- block_id=DQ0Ryb-SpW flavour=affine:paragraph -->␊
       ### Blocks that assemble your next docs, tasks kanban or whiteboard␊
       ␊
-      ␊
       <!-- block_id=HAZC3URZp_ flavour=affine:paragraph -->␊
       There is a large overlap of their atomic "building blocks" between these apps. They are neither open source nor have a plugin system like VS Code for contributors to customize. We want to have something that contains all the features we love and goes one step further.␊
       ␊
-      ␊
       <!-- block_id=0H87ypiuv8 flavour=affine:paragraph -->␊
       We are building AFFiNE to be a fundamental open source platform that contains all the building blocks for docs, task management and visual collaboration, hoping you can shape your next workflow with us that can make your life better and also connect others, too.␊
       ␊
-      ␊
       <!-- block_id=Sp4G1KD0Wn flavour=affine:paragraph -->␊
       If you want to learn more about the product design of AFFiNE, here goes the concepts:␊
       ␊
-      ␊
       <!-- block_id=RsUhDuEqXa flavour=affine:paragraph -->␊
       To Shape, not to adapt. AFFiNE is built for individuals & teams who care about their data, who refuse vendor lock-in, and who want to have control over their essential tools.␊
       ␊
-      ␊
       <!-- block_id=Z2HibKzAr- flavour=affine:paragraph -->␊
       ## A true canvas for blocks in any form␊
       ␊
-      ␊
       <!-- block_id=UwvWddamzM flavour=affine:paragraph -->␊
       [Many editor apps](http://notion.so) claimed to be a canvas for productivity. Since _the Mother of All Demos,_ Douglas Engelbart, a creative and programable digital workspace has been a pursuit and an ultimate mission for generations of tool makers.␊
       ␊
-      ␊
       <!-- block_id=g9xKUjhJj1 flavour=affine:paragraph -->␊
       ␊
       ␊
-      ␊
       <!-- block_id=wDTn4YJ4pm flavour=affine:paragraph -->␊
       "We shape our tools and thereafter our tools shape us”. A lot of pioneers have inspired us a long the way, e.g.:␊
       ␊
-      ␊
       <!-- block_id=xFrrdiP3-V flavour=affine:list -->␊
       * Quip & Notion with their great concept of "everything is a block"␊
-      ␊
-      ␊
       <!-- block_id=Tp9xyN4Okl flavour=affine:list -->␊
       * Trello with their Kanban␊
-      ␊
-      ␊
       <!-- block_id=K_4hUzKZFQ flavour=affine:list -->␊
       * Airtable & Miro with their no-code programable datasheets␊
-      ␊
-      ␊
       <!-- block_id=QwMzON2s7x flavour=affine:list -->␊
       * Miro & Whimiscal with their edgeless visual whiteboard␊
-      ␊
-      ␊
       <!-- block_id=FFVmit6u1T flavour=affine:list -->␊
       * Remnote & Capacities with their object-based tag system␊
-      ␊
-      ␊
       <!-- block_id=YqnG5O6AE6 flavour=affine:paragraph -->␊
       For more details, please refer to our [RoadMap](https://docs.affine.pro/docs/core-concepts/roadmap)␊
       ␊
-      ␊
       <!-- block_id=sbDTmZMZcq flavour=affine:paragraph -->␊
       ## Self Host␊
       ␊
-      ␊
       <!-- block_id=QVvitesfbj flavour=affine:paragraph -->␊
       Self host AFFiNE␊
       ␊
-      ␊
       <!-- block_id=U_GoHFD9At flavour=affine:database placeholder -->␊
-      ␊
       <!-- block_id=NyHXrMX3R1 flavour=affine:paragraph -->␊
       ## Affine Development␊
       ␊
-      ␊
       <!-- block_id=9-K49otbCv flavour=affine:paragraph -->␊
       For developer or installation guides, please go to [AFFiNE Development](https://docs.affine.pro/docs/development/quick-start)␊
       ␊
-      ␊
       <!-- block_id=faFteK9eG- flavour=affine:paragraph -->␊
       ␊
       ␊
-      ␊
       `,
       title: 'Write, Draw, Plan all at Once.',
     }

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

@@ -84,11 +84,17 @@ export class CopilotContextModel extends BaseModel {
   }
 
   async mergeDocStatus(workspaceId: string, docs: ContextDoc[]) {
-    const docIds = Array.from(new Set(docs.map(doc => doc.id)));
-    const finishedDoc = await this.hasWorkspaceEmbedding(workspaceId, docIds);
+    const canEmbedding = await this.checkEmbeddingAvailable();
+    const finishedDoc = canEmbedding
+      ? await this.listWorkspaceEmbedding(
+          workspaceId,
+          Array.from(new Set(docs.map(doc => doc.id)))
+        )
+      : [];
+    const finishedDocSet = new Set(finishedDoc);
 
     for (const doc of docs) {
-      const status = finishedDoc.has(doc.id)
+      const status = finishedDocSet.has(doc.id)
         ? ContextEmbedStatus.finished
         : undefined;
       // NOTE: when the document has not been synchronized to the server or is in the embedding queue
@@ -120,24 +126,17 @@ export class CopilotContextModel extends BaseModel {
     return Number(count) === 2;
   }
 
-  async hasWorkspaceEmbedding(workspaceId: string, docIds: string[]) {
-    const canEmbedding = await this.checkEmbeddingAvailable();
-    if (!canEmbedding) {
-      return new Set();
-    }
-
+  async listWorkspaceEmbedding(workspaceId: string, docIds?: string[]) {
     const existsIds = await this.db.aiWorkspaceEmbedding
-      .findMany({
+      .groupBy({
         where: {
           workspaceId,
-          docId: { in: docIds },
-        },
-        select: {
-          docId: true,
+          docId: docIds ? { in: docIds } : undefined,
         },
+        by: ['docId'],
       })
       .then(r => r.map(r => r.docId));
-    return new Set(existsIds);
+    return existsIds;
   }
 
   private processEmbeddings(

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

@@ -58,10 +58,12 @@ export class CopilotWorkspaceConfigModel extends BaseModel {
             ON id.workspace_id = s.workspace_id
                AND id.doc_id = s.guid
         WHERE s.workspace_id = ${workspaceId}
-          AND s.guid != s.workspace_id
+          AND s.guid <> s.workspace_id
           AND s.guid NOT LIKE '%$%'
+          AND s.guid NOT LIKE '%:settings:%'
           AND e.doc_id IS NULL
-          AND id.doc_id IS NULL;`;
+          AND id.doc_id IS NULL
+          AND s.blob <> E'\\\\x0000';`;
 
     return docIds.map(r => r.id);
   }
@@ -150,7 +152,7 @@ export class CopilotWorkspaceConfigModel extends BaseModel {
   }
 
   @Transactional()
-  async getWorkspaceEmbeddingStatus(workspaceId: string) {
+  async getEmbeddingStatus(workspaceId: string) {
     const ignoredDocIds = (await this.listIgnoredDocIds(workspaceId)).map(
       d => d.docId
     );
@@ -160,13 +162,19 @@ export class CopilotWorkspaceConfigModel extends BaseModel {
         { id: { notIn: ignoredDocIds } },
         { id: { not: workspaceId } },
         { id: { not: { contains: '$' } } },
+        { id: { not: { contains: ':settings:' } } },
+        { blob: { not: new Uint8Array([0, 0]) } },
       ],
     };
 
     const [docTotal, docEmbedded, fileTotal, fileEmbedded] = await Promise.all([
-      this.db.snapshot.count({ where: snapshotCondition }),
-      this.db.snapshot.count({
+      this.db.snapshot.findMany({
+        where: snapshotCondition,
+        select: { id: true },
+      }),
+      this.db.snapshot.findMany({
         where: { ...snapshotCondition, embedding: { some: {} } },
+        select: { id: true },
       }),
       this.db.aiWorkspaceFiles.count({ where: { workspaceId } }),
       this.db.aiWorkspaceFiles.count({
@@ -174,9 +182,23 @@ export class CopilotWorkspaceConfigModel extends BaseModel {
       }),
     ]);
 
+    const docTotalIds = docTotal.map(d => d.id);
+    const docTotalSet = new Set(docTotalIds);
+    const outdatedDocPrefix = `${workspaceId}:space:`;
+    const duplicateOutdatedDocSet = new Set(
+      docTotalIds
+        .filter(id => id.startsWith(outdatedDocPrefix))
+        .filter(id => docTotalSet.has(id.slice(outdatedDocPrefix.length)))
+    );
+
     return {
-      total: docTotal + fileTotal,
-      embedded: docEmbedded + fileEmbedded,
+      total:
+        docTotalIds.filter(id => !duplicateOutdatedDocSet.has(id)).length +
+        fileTotal,
+      embedded:
+        docEmbedded
+          .map(d => d.id)
+          .filter(id => !duplicateOutdatedDocSet.has(id)).length + fileEmbedded,
     };
   }
 

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

@@ -1,6 +1,6 @@
 import { Injectable } from '@nestjs/common';
 import { Transactional } from '@nestjs-cls/transactional';
-import { type Workspace } from '@prisma/client';
+import { Prisma, type Workspace } from '@prisma/client';
 
 import { EventBus } from '../base';
 import { BaseModel } from './base';
@@ -93,6 +93,19 @@ export class WorkspaceModel extends BaseModel {
     });
   }
 
+  async list<S extends Prisma.WorkspaceSelect>(
+    where: Prisma.WorkspaceWhereInput = {},
+    select?: S
+  ) {
+    return (await this.db.workspace.findMany({
+      where,
+      select,
+      orderBy: {
+        sid: 'asc',
+      },
+    })) as Prisma.WorkspaceGetPayload<{ select: S }>[];
+  }
+
   async delete(workspaceId: string) {
     const rawResult = await this.db.workspace.deleteMany({
       where: {

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

@@ -356,6 +356,7 @@ export class CopilotContextRootResolver {
     return false;
   }
 
+  @Throttle('strict')
   @Query(() => ContextWorkspaceEmbeddingStatus, {
     description: 'query workspace embedding status',
   })
@@ -372,9 +373,7 @@ export class CopilotContextRootResolver {
 
     if (this.context.canEmbedding) {
       const { total, embedded } =
-        await this.models.copilotWorkspace.getWorkspaceEmbeddingStatus(
-          workspaceId
-        );
+        await this.models.copilotWorkspace.getEmbeddingStatus(workspaceId);
       return { total, embedded };
     }
 

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

@@ -8,6 +8,7 @@ declare global {
   interface Jobs {
     'copilot.session.cleanupEmptySessions': {};
     'copilot.session.generateMissingTitles': {};
+    'copilot.workspace.cleanupTrashedDocEmbeddings': {};
   }
 }
 
@@ -20,6 +21,14 @@ export class CopilotCronJobs {
     private readonly jobs: JobQueue
   ) {}
 
+  async triggerCleanupTrashedDocEmbeddings() {
+    await this.jobs.add(
+      'copilot.workspace.cleanupTrashedDocEmbeddings',
+      {},
+      { jobId: 'daily-copilot-cleanup-trashed-doc-embeddings' }
+    );
+  }
+
   @Cron(CronExpression.EVERY_DAY_AT_MIDNIGHT)
   async dailyCleanupJob() {
     await this.jobs.add(
@@ -33,6 +42,12 @@ export class CopilotCronJobs {
       {},
       { jobId: 'daily-copilot-generate-missing-titles' }
     );
+
+    await this.jobs.add(
+      'copilot.workspace.cleanupTrashedDocEmbeddings',
+      {},
+      { jobId: 'daily-copilot-cleanup-trashed-doc-embeddings' }
+    );
   }
 
   async triggerGenerateMissingTitles() {
@@ -68,4 +83,18 @@ export class CopilotCronJobs {
       `Scheduled title generation for ${sessions.length} sessions`
     );
   }
+
+  @OnJob('copilot.workspace.cleanupTrashedDocEmbeddings')
+  async cleanupTrashedDocEmbeddings() {
+    const workspaces = await this.models.workspace.list(undefined, {
+      id: true,
+    });
+    for (const { id: workspaceId } of workspaces) {
+      await this.jobs.add(
+        'copilot.embedding.cleanupTrashedDocEmbeddings',
+        { workspaceId },
+        { jobId: `cleanup-trashed-doc-embeddings-${workspaceId}` }
+      );
+    }
+  }
 }

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

@@ -12,6 +12,7 @@ import {
   OnJob,
 } from '../../../base';
 import { DocReader } from '../../../core/doc';
+import { readAllDocIdsFromWorkspaceSnapshot } from '../../../core/utils/blocksuite';
 import { Models } from '../../../models';
 import { CopilotStorage } from '../storage';
 import { readStream } from '../utils';
@@ -134,10 +135,30 @@ export class CopilotEmbeddingJob {
     if (enableDocEmbedding) {
       const toBeEmbedDocIds =
         await this.models.copilotWorkspace.findDocsToEmbed(workspaceId);
+      if (!toBeEmbedDocIds.length) {
+        return;
+      }
+      // filter out trashed docs
+      const rootSnapshot = await this.models.doc.getSnapshot(
+        workspaceId,
+        workspaceId
+      );
+      if (!rootSnapshot) {
+        this.logger.warn(
+          `Root snapshot for workspace ${workspaceId} not found, skipping embedding.`
+        );
+        return;
+      }
+      const allDocIds = new Set(
+        readAllDocIdsFromWorkspaceSnapshot(rootSnapshot.blob)
+      );
       this.logger.log(
         `Trigger embedding for ${toBeEmbedDocIds.length} docs in workspace ${workspaceId}`
       );
-      for (const docId of toBeEmbedDocIds) {
+      const finalToBeEmbedDocIds = toBeEmbedDocIds.filter(docId =>
+        allDocIds.has(docId)
+      );
+      for (const docId of finalToBeEmbedDocIds) {
         await this.queue.add(
           'copilot.embedding.docs',
           {
@@ -337,6 +358,10 @@ export class CopilotEmbeddingJob {
     const signal = this.getWorkspaceSignal(workspaceId);
 
     try {
+      const hasNewDoc = await this.models.doc.exists(
+        workspaceId,
+        docId.split(':space:')[1] || ''
+      );
       const needEmbedding =
         await this.models.copilotWorkspace.checkDocNeedEmbedded(
           workspaceId,
@@ -352,8 +377,11 @@ export class CopilotEmbeddingJob {
           );
           return;
         }
-        const fragment = await this.getDocFragment(workspaceId, docId);
-        if (fragment) {
+        // if doc id deprecated, skip embedding and fulfill empty embedding
+        const fragment = !hasNewDoc
+          ? await this.getDocFragment(workspaceId, docId)
+          : undefined;
+        if (!hasNewDoc && fragment) {
           // fast fall for empty doc, journal is easily to create a empty doc
           if (fragment.summary.trim()) {
             const embeddings = await this.embeddingClient.getFileEmbeddings(
@@ -382,7 +410,7 @@ export class CopilotEmbeddingJob {
             );
             await this.fulfillEmptyEmbedding(workspaceId, docId);
           }
-        } else if (contextId) {
+        } else {
           this.logger.warn(
             `Doc ${docId} in workspace ${workspaceId} has no fragment, fulfilling empty embedding.`
           );
@@ -415,4 +443,39 @@ export class CopilotEmbeddingJob {
       );
     }
   }
+
+  @OnJob('copilot.embedding.cleanupTrashedDocEmbeddings')
+  async cleanupTrashedDocEmbeddings({
+    workspaceId,
+  }: Jobs['copilot.embedding.cleanupTrashedDocEmbeddings']) {
+    const workspace = await this.models.workspace.get(workspaceId);
+    if (!workspace) {
+      this.logger.warn(`workspace ${workspaceId} not found`);
+      return;
+    }
+
+    const snapshot = await this.models.doc.getSnapshot(
+      workspaceId,
+      workspaceId
+    );
+    if (!snapshot) {
+      this.logger.warn(`workspace snapshot ${workspaceId} not found`);
+      return;
+    }
+
+    const docIdsInWorkspace = readAllDocIdsFromWorkspaceSnapshot(snapshot.blob);
+    const docIdsInEmbedding =
+      await this.models.copilotContext.listWorkspaceEmbedding(workspaceId);
+    const docIdsInWorkspaceSet = new Set(docIdsInWorkspace);
+
+    const deletedDocIds = docIdsInEmbedding.filter(
+      docId => !docIdsInWorkspaceSet.has(docId)
+    );
+    for (const docId of deletedDocIds) {
+      await this.models.copilotContext.deleteWorkspaceEmbedding(
+        workspaceId,
+        docId
+      );
+    }
+  }
 }

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

@@ -61,6 +61,10 @@ declare global {
       fileId: string;
       fileName: string;
     };
+
+    'copilot.embedding.cleanupTrashedDocEmbeddings': {
+      workspaceId: string;
+    };
   }
 }
 

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

@@ -64,8 +64,8 @@ import {
     // context
     CopilotContextResolver,
     CopilotContextService,
+    // jobs
     CopilotEmbeddingJob,
-    // cron jobs
     CopilotCronJobs,
     // transcription
     CopilotTranscriptionService,

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

@@ -303,7 +303,7 @@ const textActions: Prompt[] = [
   {
     name: 'Transcript audio',
     action: 'Transcript audio',
-    model: 'gemini-2.5-pro',
+    model: 'gemini-2.5-flash',
     optionalModels: ['gemini-2.5-flash', 'gemini-2.5-pro'],
     messages: [
       {
@@ -334,6 +334,7 @@ Convert a multi-speaker audio recording into a structured JSON format by transcr
     config: {
       requireContent: false,
       requireAttachment: true,
+      maxRetries: 1,
     },
   },
   {

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

@@ -129,7 +129,16 @@ export abstract class GeminiProvider<T> extends CopilotProvider<T> {
         system,
         messages: msgs,
         schema,
+        providerOptions: {
+          google: {
+            thinkingConfig: {
+              thinkingBudget: -1,
+              includeThoughts: false,
+            },
+          },
+        },
         abortSignal: options.signal,
+        maxRetries: options.maxRetries || 3,
         experimental_repairText: async ({ text, error }) => {
           if (error instanceof JSONParseError) {
             // strange fixed response, temporarily replace it

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

@@ -37,7 +37,7 @@ import {
 import { CurrentUser } from '../../core/auth';
 import { Admin } from '../../core/common';
 import { DocReader } from '../../core/doc';
-import { AccessController } from '../../core/permission';
+import { AccessController, DocAction } from '../../core/permission';
 import { UserType } from '../../core/user';
 import type { ListSessionOptions, UpdateChatSession } from '../../models';
 import { CopilotCronJobs } from './cron';
@@ -143,6 +143,9 @@ class CreateChatMessageInput implements Omit<SubmittedMessage, 'content'> {
   @Field(() => [String], { nullable: true, deprecationReason: 'use blobs' })
   attachments!: string[] | undefined;
 
+  @Field(() => GraphQLUpload, { nullable: true })
+  blob!: Promise<FileUpload> | undefined;
+
   @Field(() => [GraphQLUpload], { nullable: true })
   blobs!: Promise<FileUpload>[] | undefined;
 
@@ -417,7 +420,8 @@ export class CopilotResolver {
 
   private async assertPermission(
     user: CurrentUser,
-    options: { workspaceId?: string | null; docId?: string | null }
+    options: { workspaceId?: string | null; docId?: string | null },
+    fallbackAction?: DocAction
   ) {
     const { workspaceId, docId } = options;
     if (!workspaceId) {
@@ -428,7 +432,7 @@ export class CopilotResolver {
         .user(user.id)
         .doc({ workspaceId, docId })
         .allowLocal()
-        .assert('Doc.Update');
+        .assert(fallbackAction ?? 'Doc.Update');
     } else {
       await this.ac
         .user(user.id)
@@ -507,7 +511,7 @@ export class CopilotResolver {
     if (!workspaceId) {
       return [];
     } else {
-      await this.assertPermission(user, { workspaceId, docId });
+      await this.assertPermission(user, { workspaceId, docId }, 'Doc.Read');
     }
 
     const histories = await this.chatSession.list(
@@ -537,7 +541,7 @@ export class CopilotResolver {
     if (!workspaceId) {
       return paginate([], 'updatedAt', pagination, 0);
     } else {
-      await this.assertPermission(user, { workspaceId, docId });
+      await this.assertPermission(user, { workspaceId, docId }, 'Doc.Read');
     }
 
     const finalOptions = Object.assign(
@@ -703,10 +707,13 @@ export class CopilotResolver {
     }
 
     const attachments: PromptMessage['attachments'] = options.attachments || [];
-    if (options.blobs) {
+    if (options.blob || options.blobs) {
       const { workspaceId } = session.config;
 
-      const blobs = await Promise.all(options.blobs);
+      const blobs = await Promise.all(
+        options.blob ? [options.blob] : options.blobs || []
+      );
+      delete options.blob;
       delete options.blobs;
 
       for (const blob of blobs) {
@@ -844,7 +851,7 @@ export class PromptsManagementResolver {
     private readonly promptService: PromptService
   ) {}
 
-  @Query(() => Boolean, {
+  @Mutation(() => Boolean, {
     description: 'Trigger generate missing titles cron job',
   })
   async triggerGenerateTitleCron() {
@@ -852,6 +859,14 @@ export class PromptsManagementResolver {
     return true;
   }
 
+  @Mutation(() => Boolean, {
+    description: 'Trigger cleanup of trashed doc embeddings',
+  })
+  async triggerCleanupTrashedDocEmbeddings() {
+    await this.cron.triggerCleanupTrashedDocEmbeddings();
+    return true;
+  }
+
   @Query(() => [CopilotPromptType], {
     description: 'List all copilot prompts',
   })

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

@@ -51,7 +51,15 @@ Important Instructions:
 - 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.
+- Each top-level list item should be a block. Like this:
+  \`\`\`markdown
+  <!-- block_id=001 flavour=affine:list -->
+  * Item 1
+    * SubItem 1
+  <!-- block_id=002 flavour=affine:list -->
+  1. Item 1
+    1. SubItem 1
+  \`\`\`
 - 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.**
 
@@ -142,24 +150,33 @@ You should specify the following arguments before the others: [doc_id], [origin_
           'A short, first-person description of the intended edit, clearly summarizing what I will change. For example: "I will translate the steps into English and delete the paragraph explaining the delay." This helps the downstream system understand the purpose of the changes.'
         ),
 
-      code_edit: z
-        .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(
-          'An array of independent semantic changes to apply to the document.'
-        ),
+      code_edit: z.preprocess(
+        val => {
+          // BACKGROUND: LLM sometimes returns a JSON string instead of an array.
+          if (typeof val === 'string') {
+            return JSON.parse(val);
+          }
+          return val;
+        },
+        z
+          .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(
+            'An array of independent semantic changes to apply to the document.'
+          )
+      ),
     }),
     execute: async ({ doc_id, origin_content, code_edit }) => {
       try {

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

@@ -1,4 +1,5 @@
 import { tool } from 'ai';
+import { omit } from 'lodash-es';
 import { z } from 'zod';
 
 import type { AccessController } from '../../../core/permission';
@@ -8,6 +9,32 @@ import type { ContextSession } from '../context/session';
 import type { CopilotChatOptions } from '../providers';
 import { toolError } from './error';
 
+const FILTER_PREFIX = [
+  'Title: ',
+  'Created at: ',
+  'Updated at: ',
+  'Created by: ',
+  'Updated by: ',
+];
+
+function clearEmbeddingChunk(chunk: ChunkSimilarity): ChunkSimilarity {
+  if (chunk.content) {
+    const lines = chunk.content.split('\n');
+    let maxLines = 5;
+    while (maxLines > 0 && lines.length > 0) {
+      if (FILTER_PREFIX.some(prefix => lines[0].startsWith(prefix))) {
+        lines.shift();
+        maxLines--;
+      } else {
+        // only process consecutive metadata rows
+        break;
+      }
+    }
+    return { ...chunk, content: lines.join('\n') };
+  }
+  return chunk;
+}
+
 export const buildDocSearchGetter = (
   ac: AccessController,
   context: CopilotContextService,
@@ -47,18 +74,37 @@ export const buildDocSearchGetter = (
     if (!docChunks.length && !fileChunks.length)
       return `No results found for "${query}".`;
 
+    const docIds = docChunks.map(c => ({
+      // oxlint-disable-next-line no-non-null-assertion
+      workspaceId: options.workspace!,
+      docId: c.docId,
+    }));
+    const docAuthors = await models.doc
+      .findAuthors(docIds)
+      .then(
+        docs =>
+          new Map(
+            docs
+              .filter(d => !!d)
+              .map(doc => [doc.id, omit(doc, ['id', 'workspaceId'])])
+          )
+      );
     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])));
+      .findMetas(docIds, { select: { title: true } })
+      .then(
+        docs =>
+          new Map(
+            docs
+              .filter(d => !!d)
+              .map(doc => [
+                doc.docId,
+                Object.assign({}, doc, docAuthors.get(doc.docId)),
+              ])
+          )
+      );
 
     return [
-      ...fileChunks,
+      ...fileChunks.map(clearEmbeddingChunk),
       ...docChunks.map(c => ({
         ...c,
         ...docMetas.get(c.docId),

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

@@ -15,7 +15,6 @@ import GraphQLUpload from 'graphql-upload/GraphQLUpload.mjs';
 
 import {
   CopilotTranscriptionAudioNotProvided,
-  CopilotTranscriptionJobNotFound,
   type FileUpload,
 } from '../../../base';
 import { CurrentUser } from '../../../core/auth';
@@ -74,7 +73,7 @@ const FinishedStatus: Set<AiJobStatus> = new Set([
 export class CopilotTranscriptionResolver {
   constructor(
     private readonly ac: AccessController,
-    private readonly service: CopilotTranscriptionService
+    private readonly transcript: CopilotTranscriptionService
   ) {}
 
   private handleJobResult(
@@ -122,7 +121,7 @@ export class CopilotTranscriptionResolver {
       throw new CopilotTranscriptionAudioNotProvided();
     }
 
-    const jobResult = await this.service.submitTranscriptionJob(
+    const jobResult = await this.transcript.submitJob(
       user.id,
       workspaceId,
       blobId,
@@ -144,19 +143,11 @@ export class CopilotTranscriptionResolver {
       .allowLocal()
       .assert('Workspace.Copilot');
 
-    const job = await this.service.queryTranscriptionJob(
+    const jobResult = await this.transcript.retryJob(
       user.id,
       workspaceId,
       jobId
     );
-    if (!job || !job.infos) {
-      throw new CopilotTranscriptionJobNotFound();
-    }
-
-    const jobResult = await this.service.executeTranscriptionJob(
-      job.id,
-      job.infos
-    );
 
     return this.handleJobResult(jobResult);
   }
@@ -166,7 +157,7 @@ export class CopilotTranscriptionResolver {
     @CurrentUser() user: CurrentUser,
     @Args('jobId') jobId: string
   ): Promise<TranscriptionResultType | null> {
-    const job = await this.service.claimTranscriptionJob(user.id, jobId);
+    const job = await this.transcript.claimJob(user.id, jobId);
     return this.handleJobResult(job);
   }
 
@@ -190,7 +181,7 @@ export class CopilotTranscriptionResolver {
       .allowLocal()
       .assert('Workspace.Copilot');
 
-    const job = await this.service.queryTranscriptionJob(
+    const job = await this.transcript.queryJob(
       user.id,
       copilot.workspaceId,
       jobId,

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

@@ -49,7 +49,17 @@ export class CopilotTranscriptionService {
     private readonly providerFactory: CopilotProviderFactory
   ) {}
 
-  async submitTranscriptionJob(
+  private async getModel(userId: string) {
+    const prompt = await this.prompt.get('Transcript audio');
+    const hasAccess = await this.models.userFeature.has(
+      userId,
+      'unlimited_copilot'
+    );
+    // choose the pro model if user has copilot plan
+    return prompt?.optionalModels[hasAccess ? 1 : 0];
+  }
+
+  async submitJob(
     userId: string,
     workspaceId: string,
     blobId: string,
@@ -78,12 +88,26 @@ export class CopilotTranscriptionService {
       infos.push({ url, mimeType: blob.mimetype });
     }
 
-    return await this.executeTranscriptionJob(jobId, infos);
+    const model = await this.getModel(userId);
+    return await this.executeJob(jobId, infos, model);
   }
 
-  async executeTranscriptionJob(
+  async retryJob(userId: string, workspaceId: string, jobId: string) {
+    const job = await this.queryJob(userId, workspaceId, jobId);
+    if (!job || !job.infos) {
+      throw new CopilotTranscriptionJobNotFound();
+    }
+
+    const model = await this.getModel(userId);
+    const jobResult = await this.executeJob(job.id, job.infos, model);
+
+    return jobResult;
+  }
+
+  async executeJob(
     jobId: string,
-    infos: AudioBlobInfos
+    infos: AudioBlobInfos,
+    modelId?: string
   ): Promise<TranscriptionJob> {
     const status = AiJobStatus.running;
     const success = await this.models.copilotJob.update(jobId, {
@@ -98,12 +122,13 @@ export class CopilotTranscriptionService {
     await this.job.add('copilot.transcript.submit', {
       jobId,
       infos,
+      modelId,
     });
 
     return { id: jobId, status };
   }
 
-  async claimTranscriptionJob(
+  async claimJob(
     userId: string,
     jobId: string
   ): Promise<TranscriptionJob | null> {
@@ -118,7 +143,7 @@ export class CopilotTranscriptionService {
     return null;
   }
 
-  async queryTranscriptionJob(
+  async queryJob(
     userId: string,
     workspaceId: string,
     jobId?: string,
@@ -181,14 +206,20 @@ export class CopilotTranscriptionService {
     promptName: string,
     message: Partial<PromptMessage>,
     schema?: ZodType<any>,
-    prefer?: CopilotProviderType
+    prefer?: CopilotProviderType,
+    modelId?: string
   ): Promise<string> {
     const prompt = await this.prompt.get(promptName);
     if (!prompt) {
       throw new CopilotPromptNotFound({ name: promptName });
     }
 
-    const cond = { modelId: prompt.model };
+    const cond = {
+      modelId:
+        modelId && prompt.optionalModels.includes(modelId)
+          ? modelId
+          : prompt.model,
+    };
     const msg = { role: 'user' as const, content: '', ...message };
     const config = Object.assign({}, prompt.config);
     if (schema) {
@@ -231,13 +262,19 @@ export class CopilotTranscriptionService {
     return `${hoursStr}:${minutesStr}:${secondsStr}`;
   }
 
-  private async callTranscript(url: string, mimeType: string, offset: number) {
+  private async callTranscript(
+    url: string,
+    mimeType: string,
+    offset: number,
+    modelId?: string
+  ) {
     // NOTE: Vertex provider not support transcription yet, we always use Gemini here
     const result = await this.chatWithPrompt(
       'Transcript audio',
       { attachments: [url], params: { mimetype: mimeType } },
       TranscriptionResponseSchema,
-      CopilotProviderType.Gemini
+      CopilotProviderType.Gemini,
+      modelId
     );
 
     const transcription = TranscriptionResponseSchema.parse(
@@ -256,6 +293,7 @@ export class CopilotTranscriptionService {
   async transcriptAudio({
     jobId,
     infos,
+    modelId,
     // @deprecated
     url,
     mimeType,
@@ -264,7 +302,7 @@ export class CopilotTranscriptionService {
       const blobInfos = this.mergeInfos(infos, url, mimeType);
       const transcriptions = await Promise.all(
         Array.from(blobInfos.entries()).map(([idx, { url, mimeType }]) =>
-          this.callTranscript(url, mimeType, idx * 10 * 60)
+          this.callTranscript(url, mimeType, idx * 10 * 60, modelId)
         )
       );
 

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

@@ -56,6 +56,7 @@ declare global {
     'copilot.transcript.submit': {
       jobId: string;
       infos?: AudioBlobInfos;
+      modelId?: string;
       /// @deprecated use `infos` instead
       url?: string;
       /// @deprecated use `infos` instead

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

@@ -103,6 +103,7 @@ export class CopilotWorkspaceEmbeddingConfigResolver {
 
     return ignoredDocs;
   }
+
   @Mutation(() => Number, {
     name: 'updateWorkspaceEmbeddingIgnoredDocs',
     complexity: 2,

packages/backend/server/src/schema.gql

@@ -457,6 +457,7 @@ type CopilotWorkspaceIgnoredDocTypeEdge {
 
 input CreateChatMessageInput {
   attachments: [String!]
+  blob: Upload
   blobs: [Upload!]
   content: String
   params: JSON
@@ -1297,6 +1298,12 @@ type Mutation {
   setBlob(blob: Upload!, workspaceId: String!): String!
   submitAudioTranscription(blob: Upload, blobId: String!, blobs: [Upload!], workspaceId: String!): TranscriptionResultType
 
+  """Trigger cleanup of trashed doc embeddings"""
+  triggerCleanupTrashedDocEmbeddings: Boolean!
+
+  """Trigger generate missing titles cron job"""
+  triggerGenerateTitleCron: Boolean!
+
   """update app configuration"""
   updateAppConfig(updates: [UpdateAppConfigInput!]!): JSONObject!
 

packages/common/graphql/src/schema.ts

@@ -569,6 +569,7 @@ export interface CopilotWorkspaceIgnoredDocTypeEdge {
 
 export interface CreateChatMessageInput {
   attachments?: InputMaybe<Array<Scalars['String']['input']>>;
+  blob?: InputMaybe<Scalars['Upload']['input']>;
   blobs?: InputMaybe<Array<Scalars['Upload']['input']>>;
   content?: InputMaybe<Scalars['String']['input']>;
   params?: InputMaybe<Scalars['JSON']['input']>;
@@ -1440,6 +1441,10 @@ export interface Mutation {
   sendVerifyEmail: Scalars['Boolean']['output'];
   setBlob: Scalars['String']['output'];
   submitAudioTranscription: Maybe<TranscriptionResultType>;
+  /** Trigger cleanup of trashed doc embeddings */
+  triggerCleanupTrashedDocEmbeddings: Scalars['Boolean']['output'];
+  /** Trigger generate missing titles cron job */
+  triggerGenerateTitleCron: Scalars['Boolean']['output'];
   /** update app configuration */
   updateAppConfig: Scalars['JSONObject']['output'];
   /** Update a comment content */

packages/common/reader/__tests__/__snapshots__/reader.spec.ts.snap

@@ -58,74 +58,45 @@ exports[`should parse page doc work 1`] = `
 
 
 
-
-
 # You own your data, with no compromises
 
-
 ## Local-first & Real-time collaborative
 
-
 We love the idea proposed by Ink & Switch in the famous article about you owning your data, despite the cloud. Furthermore, AFFiNE is the first all-in-one workspace that keeps your data ownership with no compromises on real-time collaboration and editing experience.
 
-
 AFFiNE is a local-first application upon CRDTs with real-time collaboration support. Your data is always stored locally while multiple nodes remain synced in real-time.
 
 
 
-
-
 ### Blocks that assemble your next docs, tasks kanban or whiteboard
 
-
 There is a large overlap of their atomic "building blocks" between these apps. They are neither open source nor have a plugin system like VS Code for contributors to customize. We want to have something that contains all the features we love and goes one step further.
 
-
 We are building AFFiNE to be a fundamental open source platform that contains all the building blocks for docs, task management and visual collaboration, hoping you can shape your next workflow with us that can make your life better and also connect others, too.
 
-
 If you want to learn more about the product design of AFFiNE, here goes the concepts:
 
-
 To Shape, not to adapt. AFFiNE is built for individuals & teams who care about their data, who refuse vendor lock-in, and who want to have control over their essential tools.
 
-
 ## A true canvas for blocks in any form
 
-
 [Many editor apps](http://notion.so) claimed to be a canvas for productivity. Since _the Mother of All Demos,_ Douglas Engelbart, a creative and programable digital workspace has been a pursuit and an ultimate mission for generations of tool makers.
 
 
 
-
-
 "We shape our tools and thereafter our tools shape us”. A lot of pioneers have inspired us a long the way, e.g.:
 
-
 * Quip & Notion with their great concept of "everything is a block"
-
-
 * Trello with their Kanban
-
-
 * Airtable & Miro with their no-code programable datasheets
-
-
 * Miro & Whimiscal with their edgeless visual whiteboard
-
-
 * Remnote & Capacities with their object-based tag system
-
-
 For more details, please refer to our [RoadMap](https://docs.affine.pro/docs/core-concepts/roadmap)
 
-
 ## Self Host
 
-
 Self host AFFiNE
 
-
 ||Title|Tag|
 |---|---|---|
 |Affine Development|Affine Development|<span data-affine-option data-value="AxSe-53xjX" data-option-color="var(--affine-tag-pink)">AFFiNE</span>|
@@ -136,16 +107,12 @@ Self host AFFiNE
 |Miro & Whimiscal with their edgeless visual whiteboard|Miro & Whimiscal with their edgeless visual whiteboard|<span data-affine-option data-value="HgHsKOUINZ" data-option-color="var(--affine-tag-blue)">Reference</span>|
 |Remnote & Capacities with their object-based tag system|Remnote & Capacities with their object-based tag system||
 
-
 ## Affine Development
 
-
 For developer or installation guides, please go to [AFFiNE Development](https://docs.affine.pro/docs/development/quick-start)
 
 
 
-
-
 ",
   "parsedBlock": {
     "children": [
@@ -322,7 +289,6 @@ For developer or installation guides, please go to [AFFiNE Development](https://
           {
             "children": [],
             "content": "* Quip & Notion with their great concept of "everything is a block"
-
 ",
             "flavour": "affine:list",
             "id": "xFrrdiP3-V",
@@ -331,7 +297,6 @@ For developer or installation guides, please go to [AFFiNE Development](https://
           {
             "children": [],
             "content": "* Trello with their Kanban
-
 ",
             "flavour": "affine:list",
             "id": "Tp9xyN4Okl",
@@ -340,7 +305,6 @@ For developer or installation guides, please go to [AFFiNE Development](https://
           {
             "children": [],
             "content": "* Airtable & Miro with their no-code programable datasheets
-
 ",
             "flavour": "affine:list",
             "id": "K_4hUzKZFQ",
@@ -349,7 +313,6 @@ For developer or installation guides, please go to [AFFiNE Development](https://
           {
             "children": [],
             "content": "* Miro & Whimiscal with their edgeless visual whiteboard
-
 ",
             "flavour": "affine:list",
             "id": "QwMzON2s7x",
@@ -358,7 +321,6 @@ For developer or installation guides, please go to [AFFiNE Development](https://
           {
             "children": [],
             "content": "* Remnote & Capacities with their object-based tag system
-
 ",
             "flavour": "affine:list",
             "id": "FFVmit6u1T",
@@ -427,77 +389,63 @@ For developer or installation guides, please go to [AFFiNE Development](https://
                 "Tag": "<span data-affine-option data-value="AxSe-53xjX" data-option-color="var(--affine-tag-pink)">AFFiNE</span>",
                 "Title": "Affine Development
 
-
 ",
                 "undefined": "Affine Development
 
-
 ",
               },
               {
                 "Tag": "<span data-affine-option data-value="0jh9gNw4Yl" data-option-color="var(--affine-tag-orange)">Developers</span>",
                 "Title": "For developers or installations guides, please go to AFFiNE Doc
 
-
 ",
                 "undefined": "For developers or installations guides, please go to AFFiNE Doc
 
-
 ",
               },
               {
                 "Tag": "<span data-affine-option data-value="HgHsKOUINZ" data-option-color="var(--affine-tag-blue)">Reference</span>",
                 "Title": "Quip & Notion with their great concept of "everything is a block"
 
-
 ",
                 "undefined": "Quip & Notion with their great concept of "everything is a block"
 
-
 ",
               },
               {
                 "Tag": "<span data-affine-option data-value="HgHsKOUINZ" data-option-color="var(--affine-tag-blue)">Reference</span>",
                 "Title": "Trello with their Kanban
 
-
 ",
                 "undefined": "Trello with their Kanban
 
-
 ",
               },
               {
                 "Tag": "<span data-affine-option data-value="HgHsKOUINZ" data-option-color="var(--affine-tag-blue)">Reference</span>",
                 "Title": "Airtable & Miro with their no-code programable datasheets
 
-
 ",
                 "undefined": "Airtable & Miro with their no-code programable datasheets
 
-
 ",
               },
               {
                 "Tag": "<span data-affine-option data-value="HgHsKOUINZ" data-option-color="var(--affine-tag-blue)">Reference</span>",
                 "Title": "Miro & Whimiscal with their edgeless visual whiteboard
 
-
 ",
                 "undefined": "Miro & Whimiscal with their edgeless visual whiteboard
 
-
 ",
               },
               {
                 "Tag": "",
                 "Title": "Remnote & Capacities with their object-based tag system
 
-
 ",
                 "undefined": "Remnote & Capacities with their object-based tag system
 
-
 ",
               },
             ],
@@ -559,113 +507,80 @@ exports[`should parse page doc work with ai editable 1`] = `
 "<!-- block_id=FoPQcAyV_m flavour=affine:paragraph -->
 AFFiNE is an open source all in one workspace, an operating system for all the building blocks of your team wiki, knowledge management and digital assets and a better alternative to Notion and Miro.
 
-
 <!-- block_id=oz48nn_zp8 flavour=affine:paragraph -->
 
 
-
 <!-- block_id=g8a-D9-jXS flavour=affine:paragraph -->
 # You own your data, with no compromises
 
-
 <!-- block_id=J8lHN1GR_5 flavour=affine:paragraph -->
 ## Local-first & Real-time collaborative
 
-
 <!-- block_id=xCuWdM0VLz flavour=affine:paragraph -->
 We love the idea proposed by Ink & Switch in the famous article about you owning your data, despite the cloud. Furthermore, AFFiNE is the first all-in-one workspace that keeps your data ownership with no compromises on real-time collaboration and editing experience.
 
-
 <!-- block_id=zElMi0tViK flavour=affine:paragraph -->
 AFFiNE is a local-first application upon CRDTs with real-time collaboration support. Your data is always stored locally while multiple nodes remain synced in real-time.
 
-
 <!-- block_id=Z4rK0OF9Wk flavour=affine:paragraph -->
 
 
-
 <!-- block_id=DQ0Ryb-SpW flavour=affine:paragraph -->
 ### Blocks that assemble your next docs, tasks kanban or whiteboard
 
-
 <!-- block_id=HAZC3URZp_ flavour=affine:paragraph -->
 There is a large overlap of their atomic "building blocks" between these apps. They are neither open source nor have a plugin system like VS Code for contributors to customize. We want to have something that contains all the features we love and goes one step further.
 
-
 <!-- block_id=0H87ypiuv8 flavour=affine:paragraph -->
 We are building AFFiNE to be a fundamental open source platform that contains all the building blocks for docs, task management and visual collaboration, hoping you can shape your next workflow with us that can make your life better and also connect others, too.
 
-
 <!-- block_id=Sp4G1KD0Wn flavour=affine:paragraph -->
 If you want to learn more about the product design of AFFiNE, here goes the concepts:
 
-
 <!-- block_id=RsUhDuEqXa flavour=affine:paragraph -->
 To Shape, not to adapt. AFFiNE is built for individuals & teams who care about their data, who refuse vendor lock-in, and who want to have control over their essential tools.
 
-
 <!-- block_id=Z2HibKzAr- flavour=affine:paragraph -->
 ## A true canvas for blocks in any form
 
-
 <!-- block_id=UwvWddamzM flavour=affine:paragraph -->
 [Many editor apps](http://notion.so) claimed to be a canvas for productivity. Since _the Mother of All Demos,_ Douglas Engelbart, a creative and programable digital workspace has been a pursuit and an ultimate mission for generations of tool makers.
 
-
 <!-- block_id=g9xKUjhJj1 flavour=affine:paragraph -->
 
 
-
 <!-- block_id=wDTn4YJ4pm flavour=affine:paragraph -->
 "We shape our tools and thereafter our tools shape us”. A lot of pioneers have inspired us a long the way, e.g.:
 
-
 <!-- block_id=xFrrdiP3-V flavour=affine:list -->
 * Quip & Notion with their great concept of "everything is a block"
-
-
 <!-- block_id=Tp9xyN4Okl flavour=affine:list -->
 * Trello with their Kanban
-
-
 <!-- block_id=K_4hUzKZFQ flavour=affine:list -->
 * Airtable & Miro with their no-code programable datasheets
-
-
 <!-- block_id=QwMzON2s7x flavour=affine:list -->
 * Miro & Whimiscal with their edgeless visual whiteboard
-
-
 <!-- block_id=FFVmit6u1T flavour=affine:list -->
 * Remnote & Capacities with their object-based tag system
-
-
 <!-- block_id=YqnG5O6AE6 flavour=affine:paragraph -->
 For more details, please refer to our [RoadMap](https://docs.affine.pro/docs/core-concepts/roadmap)
 
-
 <!-- block_id=sbDTmZMZcq flavour=affine:paragraph -->
 ## Self Host
 
-
 <!-- block_id=QVvitesfbj flavour=affine:paragraph -->
 Self host AFFiNE
 
-
 <!-- block_id=U_GoHFD9At flavour=affine:database placeholder -->
-
 <!-- block_id=NyHXrMX3R1 flavour=affine:paragraph -->
 ## Affine Development
 
-
 <!-- block_id=9-K49otbCv flavour=affine:paragraph -->
 For developer or installation guides, please go to [AFFiNE Development](https://docs.affine.pro/docs/development/quick-start)
 
-
 <!-- block_id=faFteK9eG- flavour=affine:paragraph -->
 
 
-
 "
 `;
 
@@ -673,122 +588,74 @@ exports[`should parse page full doc work with ai editable 1`] = `
 "<!-- block_id=T4qSXc13wz flavour=affine:paragraph -->
 # H1 text
 
-
 <!-- block_id=F5eByK8Fx_ flavour=affine:paragraph -->
 List all flavours in one document.
 
-
 <!-- block_id=6_-Ta2Hpsg flavour=affine:paragraph -->
 ## H2 ~ H6
 
-
 <!-- block_id=QLH8pCeJwr flavour=affine:paragraph -->
 ### H3
 
-
 <!-- block_id=eRseB5ilzP flavour=affine:paragraph -->
 #### H4 with emoji 😄
 
-
 <!-- block_id=xSEIo9I5jQ flavour=affine:paragraph -->
 ##### H5
 
-
 <!-- block_id=h4Fozi-Mvv flavour=affine:paragraph -->
 ###### H6
 
-
 <!-- block_id=U-Hd9O6FEZ flavour=affine:paragraph -->
 max is H6
 
-
 <!-- block_id=z2aCxUDpOc flavour=affine:paragraph -->
 ## List
 
-
 <!-- block_id=z5Zw7lMlD7 flavour=affine:list -->
 * item 1
-
-
 <!-- block_id=Opmt3x2Ao0 flavour=affine:list -->
 * item 2
-
-
-  * sub item 1
-  
-  
-  * sub item 2
-  
-  
-    * super sub item 1
-    
-    
-  * sub item 3
-  
-  
+    * sub item 1
+    * sub item 2
+        * super sub item 1
+    * sub item 3
 <!-- block_id=_EF3g4194w flavour=affine:list -->
 * item 3
-
-
 <!-- block_id=5u-T48lLVF flavour=affine:paragraph -->
 
 
-
 <!-- block_id=7urxrvhr-p flavour=affine:paragraph -->
 
 
-
 <!-- block_id=U-96XKGGz7 flavour=affine:paragraph -->
 
 
-
 <!-- block_id=hOvvRmDGqN flavour=affine:paragraph -->
 sort list
 
-
 <!-- block_id=hcqkMyvKnx flavour=affine:list -->
 1. item 1
-
-
 <!-- block_id=xUsDktnmuD flavour=affine:list -->
 1. item 2
-
-
 <!-- block_id=xa5tsLHHJN flavour=affine:list -->
 1. item 3
-
-
-  1. sub item 1
-  
-  
-  1. sub item 2
-  
-  
-    1. super item 1
-    
-    
-    1. super item 2
-    
-    
-  1. sub item 3
-  
-  
+    1. sub item 1
+    1. sub item 2
+        1. super item 1
+        1. super item 2
+    1. sub item 3
 <!-- block_id=BX05mQdxJ0 flavour=affine:list -->
 1. item 4
-
-
 <!-- block_id=VYzM3O17th flavour=affine:paragraph -->
 
 
-
 <!-- block_id=epKYpKt5vo flavour=affine:paragraph -->
 
 
-
 <!-- block_id=5Ghem19uGh flavour=affine:paragraph -->
 Table
 
-
 <!-- block_id=OXvH-s1Jx4 flavour=affine:table -->
 |c1|c2|c3|c4|
 |---|---|---|---|
@@ -796,176 +663,129 @@ Table
 ||||v4|
 ||v6||v5|
 
-
 <!-- block_id=j2F2hQ3zy9 flavour=affine:paragraph -->
 
 
-
 <!-- block_id=jLCRD2G_BC flavour=affine:paragraph -->
 
 
-
 <!-- block_id=794ZoPeBJM flavour=affine:paragraph -->
 Database
 
-
 <!-- block_id=xQ7rA57Qxz flavour=affine:database placeholder -->
-
 <!-- block_id=RbMSmluZYK flavour=affine:paragraph -->
 Code
 
-
 <!-- block_id=cJ6CMeUWMg flavour=affine:code -->
 \`\`\`javascript
 console.log('hello world');
 \`\`\`
 
-
 <!-- block_id=y1xVwkxlDm flavour=affine:paragraph -->
 
 
-
 <!-- block_id=BKy3zmm8SE flavour=affine:paragraph -->
 Image
 
-
 <!-- block_id=WFftQ-qXzr flavour=affine:image -->
 
 ![-HjsQksaVuEaM0KeTEbBYDgWVOmoVzrAeVK0kn4Jfr0=](blob://-HjsQksaVuEaM0KeTEbBYDgWVOmoVzrAeVK0kn4Jfr0=)
 
-
 <!-- block_id=F-RKpfxL1z flavour=affine:paragraph -->
 
 
-
 <!-- block_id=G3LSqjKv8M flavour=affine:paragraph -->
 File
 
-
 <!-- block_id=pO8JCsiK4z flavour=affine:attachment -->
 
 ![IrsiJ9XonFXF8fw9tLQXbSsBbUYFfzLgoFpodeidQOU=](blob://IrsiJ9XonFXF8fw9tLQXbSsBbUYFfzLgoFpodeidQOU=)
 
-
 <!-- block_id=dTKFqQhJuA flavour=affine:paragraph -->
 
 
-
 <!-- block_id=nwld7RMYvp flavour=affine:paragraph -->
 > foo bar quote text
 
-
 <!-- block_id=MwBD3BhRnf flavour=affine:paragraph -->
 
 
-
 <!-- block_id=pakOSAm6EU flavour=affine:paragraph -->
 
 
-
 <!-- block_id=95-NxAyFuo flavour=affine:divider -->
 
 ---
 
-
 <!-- block_id=r9EllTNiN1 flavour=affine:paragraph -->
 
 
-
 <!-- block_id=OpxZ1kYM40 flavour=affine:paragraph -->
 TeX
 
-
 <!-- block_id=gjFqI97IRc flavour=affine:paragraph -->
 
 
-
 <!-- block_id=KXBZ1_Pfdw flavour=affine:paragraph -->
 
 
-
 <!-- block_id=VHj5gMaGa7 flavour=affine:paragraph -->
 2025-06-18 13:15
 
-
 <!-- block_id=JwaUwzuQEH flavour=affine:paragraph -->
 
 
-
 <!-- block_id=_zu2kl56FY flavour=affine:database placeholder -->
-
 <!-- block_id=Kcbp6BLA-y flavour=affine:paragraph -->
 Mind Map
 
-
 <!-- block_id=R_g1tzqzAU flavour=affine:paragraph -->
 
 
-
 <!-- block_id=C8G82uLCz1 flavour=affine:paragraph -->
 
 
-
 <!-- block_id=J6gfR8YMGy flavour=affine:paragraph -->
 A Link
 
-
 <!-- block_id=yHky0s_H1v flavour=affine:embed-linked-doc -->
 
 [null](doc://FmHFPAPzp51JjFP89aZ-b)
 
-
 <!-- block_id=P7w3ka4Amo flavour=affine:paragraph -->
 Todo List
 
-
 <!-- block_id=WbeCXu6fcA flavour=affine:list -->
 - [ ] abc
-
-
 <!-- block_id=X_F5fw-MEn flavour=affine:list -->
 - [ ] edf
-
-
-  - [x] done1
-  
-  
+    - [x] done1
 <!-- block_id=sdw-couBVA flavour=affine:list -->
 - [ ] end
-
-
 <!-- block_id=COJiWGOVJu flavour=affine:paragraph -->
 
 
-
 <!-- block_id=shK7TY-Q3F flavour=affine:paragraph -->
 ~~delete text~~
 
-
 <!-- block_id=_NIj4pT_Iy flavour=affine:paragraph -->
 
 
-
 <!-- block_id=CaXXPfEt62 flavour=affine:paragraph -->
 **Bold text**
 
-
 <!-- block_id=1WFCwn1708 flavour=affine:paragraph -->
 
 
-
 <!-- block_id=25f19QUjQI flavour=affine:paragraph -->
 Underline
 
-
 <!-- block_id=GrS-y17iiw flavour=affine:paragraph -->
 
 
-
 <!-- block_id=dJm5C8KsEg flavour=affine:paragraph -->
 Youtube
 
-
 <!-- block_id=epfNja2Txk flavour=affine:embed-youtube -->
 
         <iframe
@@ -979,23 +799,18 @@ Youtube
           credentialless>
         </iframe>
 
-
 <!-- block_id=wNb6ZRJKMt flavour=affine:paragraph -->
 
 
-
 <!-- block_id=HqKjEGWF_s flavour=affine:paragraph -->
 ## end
 
-
 <!-- block_id=FOh_TJmcF1 flavour=affine:paragraph -->
 this is end
 
-
 <!-- block_id=ImCJN2Xint flavour=affine:paragraph -->
 
 
-
 "
 `;
 

packages/common/reader/src/doc-parser/parser.ts

@@ -22,9 +22,10 @@ export const parseBlockToMd = (
       block.content
         .split('\n')
         .map(line => padding + line)
+        .slice(0, -1)
         .join('\n') +
       '\n' +
-      block.children.map(b => parseBlockToMd(b, padding + '  ')).join('')
+      block.children.map(b => parseBlockToMd(b, padding + '    ')).join('')
     );
   } else {
     return block.children.map(b => parseBlockToMd(b, padding)).join('');
@@ -109,7 +110,7 @@ export function parseBlock(
           const checked = yBlock.get('prop:checked') as boolean;
           prefix = checked ? '- [x] ' : '- [ ] ';
         }
-        result.content = prefix + toMd() + '\n';
+        result.content = prefix + toMd();
         break;
       }
       case 'affine:code': {

packages/frontend/apps/android/App/app/src/main/AndroidManifest.xml

@@ -14,6 +14,7 @@
         android:label="@string/app_name"
         android:roundIcon="@mipmap/ic_launcher_round"
         android:supportsRtl="true"
+        android:usesCleartextTraffic="true"
         android:theme="@style/AppTheme">
 
         <activity

packages/frontend/apps/electron-renderer/src/app/app.tsx

@@ -10,6 +10,7 @@ import { Suspense } from 'react';
 import { RouterProvider } from 'react-router-dom';
 
 import { setupEffects } from './effects';
+import { DesktopLanguageSync } from './language-sync';
 import { DesktopThemeSync } from './theme-sync';
 
 const { frameworkProvider } = setupEffects();
@@ -46,6 +47,7 @@ export function App() {
           <I18nProvider>
             <AffineContext store={getCurrentStore()}>
               <DesktopThemeSync />
+              <DesktopLanguageSync />
               <RouterProvider
                 fallbackElement={<AppContainer fallback />}
                 router={router}

packages/frontend/apps/electron-renderer/src/app/language-sync.ts

@@ -0,0 +1,18 @@
+import { DesktopApiService } from '@affine/core/modules/desktop-api';
+import { I18nService } from '@affine/core/modules/i18n';
+import { useLiveData, useService } from '@toeverything/infra';
+import { useEffect } from 'react';
+
+export const DesktopLanguageSync = () => {
+  const i18nService = useService(I18nService);
+  const currentLanguage = useLiveData(i18nService.i18n.currentLanguageKey$);
+  const handler = useService(DesktopApiService).api.handler;
+
+  useEffect(() => {
+    handler.i18n.changeLanguage(currentLanguage ?? 'en').catch(err => {
+      console.error(err);
+    });
+  }, [currentLanguage, handler]);
+
+  return null;
+};

packages/frontend/apps/electron/package.json

@@ -33,6 +33,7 @@
   },
   "devDependencies": {
     "@affine-tools/utils": "workspace:*",
+    "@affine/i18n": "workspace:*",
     "@affine/native": "workspace:*",
     "@affine/nbstore": "workspace:*",
     "@electron-forge/cli": "^7.6.0",

packages/frontend/apps/electron/src/main/handlers.ts

@@ -1,3 +1,4 @@
+import { I18n } from '@affine/i18n';
 import { ipcMain } from 'electron';
 
 import { AFFINE_API_CHANNEL_NAME } from '../shared/type';
@@ -21,6 +22,12 @@ export const debugHandlers = {
   },
 };
 
+export const i18nHandlers = {
+  changeLanguage: async (_: Electron.IpcMainInvokeEvent, language: string) => {
+    return I18n.changeLanguage(language);
+  },
+};
+
 // Note: all of these handlers will be the single-source-of-truth for the apis exposed to the renderer process
 export const allHandlers = {
   debug: debugHandlers,
@@ -33,6 +40,7 @@ export const allHandlers = {
   worker: workerHandlers,
   recording: recordingHandlers,
   popup: popupHandlers,
+  i18n: i18nHandlers,
 };
 
 export const registerHandlers = () => {

packages/frontend/apps/electron/src/main/shared-storage/events.ts

@@ -1,25 +1,17 @@
 import type { MainEventRegister } from '../type';
-import { globalCacheStorage, globalStateStorage } from './storage';
+import { globalCacheUpdates$, globalStateUpdates$ } from './handlers';
 
 export const sharedStorageEvents = {
   onGlobalStateChanged: (
     fn: (state: Record<string, unknown | undefined>) => void
   ) => {
-    const subscription = globalStateStorage.watchAll().subscribe(updates => {
-      fn(updates);
-    });
-    return () => {
-      subscription.unsubscribe();
-    };
+    const subscription = globalStateUpdates$.subscribe(fn);
+    return () => subscription.unsubscribe();
   },
   onGlobalCacheChanged: (
     fn: (state: Record<string, unknown | undefined>) => void
   ) => {
-    const subscription = globalCacheStorage.watchAll().subscribe(updates => {
-      fn(updates);
-    });
-    return () => {
-      subscription.unsubscribe();
-    };
+    const subscription = globalCacheUpdates$.subscribe(fn);
+    return () => subscription.unsubscribe();
   },
 } satisfies Record<string, MainEventRegister>;

packages/frontend/apps/electron/src/main/shared-storage/handlers.ts

@@ -1,6 +1,22 @@
+import { Subject } from 'rxjs';
+
 import type { NamespaceHandlers } from '../type';
 import { globalCacheStorage, globalStateStorage } from './storage';
 
+// Subjects used by shared-storage/events.ts to broadcast updates to all renderer processes
+export const globalStateUpdates$ = new Subject<Record<string, any>>();
+export const globalCacheUpdates$ = new Subject<Record<string, any>>();
+
+// Revision maps; main generates the next value each time
+const globalStateRevisions = new Map<string, number>();
+const globalCacheRevisions = new Map<string, number>();
+
+function nextRev(revisions: Map<string, number>, key: string) {
+  const r = (revisions.get(key) ?? 0) + 1;
+  revisions.set(key, r);
+  return r;
+}
+
 export const sharedStorageHandlers = {
   getAllGlobalState: async () => {
     return globalStateStorage.all();
@@ -8,22 +24,36 @@ export const sharedStorageHandlers = {
   getAllGlobalCache: async () => {
     return globalCacheStorage.all();
   },
-  setGlobalState: async (_, key: string, value: any) => {
-    return globalStateStorage.set(key, value);
+
+  setGlobalState: async (_e, key: string, value: any, sourceId?: string) => {
+    const rev = nextRev(globalStateRevisions, key);
+    globalStateStorage.set(key, value);
+    globalStateUpdates$.next({ [key]: { v: value, r: rev, s: sourceId } });
   },
-  delGlobalState: async (_, key: string) => {
-    return globalStateStorage.del(key);
+  delGlobalState: async (_e, key: string, sourceId?: string) => {
+    const rev = nextRev(globalStateRevisions, key);
+    globalStateStorage.del(key);
+    globalStateUpdates$.next({ [key]: { v: undefined, r: rev, s: sourceId } });
   },
-  clearGlobalState: async () => {
-    return globalStateStorage.clear();
+  clearGlobalState: async (_e, sourceId?: string) => {
+    globalStateRevisions.clear();
+    globalStateStorage.clear();
+    globalStateUpdates$.next({ '*': { v: undefined, r: 0, s: sourceId } });
   },
-  setGlobalCache: async (_, key: string, value: any) => {
-    return globalCacheStorage.set(key, value);
+
+  setGlobalCache: async (_e, key: string, value: any, sourceId?: string) => {
+    const rev = nextRev(globalCacheRevisions, key);
+    globalCacheStorage.set(key, value);
+    globalCacheUpdates$.next({ [key]: { v: value, r: rev, s: sourceId } });
   },
-  delGlobalCache: async (_, key: string) => {
-    return globalCacheStorage.del(key);
+  delGlobalCache: async (_e, key: string, sourceId?: string) => {
+    const rev = nextRev(globalCacheRevisions, key);
+    globalCacheStorage.del(key);
+    globalCacheUpdates$.next({ [key]: { v: undefined, r: rev, s: sourceId } });
   },
-  clearGlobalCache: async () => {
-    return globalCacheStorage.clear();
+  clearGlobalCache: async (_e, sourceId?: string) => {
+    globalCacheRevisions.clear();
+    globalCacheStorage.clear();
+    globalCacheUpdates$.next({ '*': { v: undefined, r: 0, s: sourceId } });
   },
 } satisfies NamespaceHandlers;

packages/frontend/apps/electron/src/main/windows-manager/tab-views.ts

@@ -1,5 +1,6 @@
 import { join } from 'node:path';
 
+import { I18n } from '@affine/i18n';
 import {
   app,
   BrowserWindow,
@@ -822,42 +823,53 @@ export class WebContentViewsManager {
       },
     });
 
-    if (spellCheckSettings.enabled) {
-      view.webContents.on('context-menu', (_event, params) => {
-        const shouldShow =
-          params.misspelledWord && params.dictionarySuggestions.length > 0;
+    view.webContents.on('context-menu', (_event, params) => {
+      const menu = Menu.buildFromTemplate([
+        {
+          id: 'cut',
+          label: I18n['com.affine.context-menu.cut'](),
+          role: 'cut',
+          enabled: params.editFlags.canCut,
+        },
+        {
+          id: 'copy',
+          label: I18n['com.affine.context-menu.copy'](),
+          role: 'copy',
+          enabled: params.editFlags.canCopy,
+        },
+        {
+          id: 'paste',
+          label: I18n['com.affine.context-menu.paste'](),
+          role: 'paste',
+          enabled: params.editFlags.canPaste,
+        },
+      ]);
 
-        if (!shouldShow) {
-          return;
-        }
-        const menu = new Menu();
+      // Add each spelling suggestion
+      for (const suggestion of params.dictionarySuggestions) {
+        menu.append(
+          new MenuItem({
+            label: suggestion,
+            click: () => view.webContents.replaceMisspelling(suggestion),
+          })
+        );
+      }
 
-        // Add each spelling suggestion
-        for (const suggestion of params.dictionarySuggestions) {
-          menu.append(
-            new MenuItem({
-              label: suggestion,
-              click: () => view.webContents.replaceMisspelling(suggestion),
-            })
-          );
-        }
+      // Allow users to add the misspelled word to the dictionary
+      if (params.misspelledWord) {
+        menu.append(
+          new MenuItem({
+            label: 'Add to dictionary', // TODO: i18n
+            click: () =>
+              view.webContents.session.addWordToSpellCheckerDictionary(
+                params.misspelledWord
+              ),
+          })
+        );
+      }
 
-        // Allow users to add the misspelled word to the dictionary
-        if (params.misspelledWord) {
-          menu.append(
-            new MenuItem({
-              label: 'Add to dictionary', // TODO: i18n
-              click: () =>
-                view.webContents.session.addWordToSpellCheckerDictionary(
-                  params.misspelledWord
-                ),
-            })
-          );
-        }
-
-        menu.popup();
-      });
-    }
+      menu.popup();
+    });
 
     this.webViewsMap$.next(this.tabViewsMap.set(viewId, view));
     let unsub = () => {};

packages/frontend/apps/electron/src/preload/shared-storage.ts

@@ -6,6 +6,7 @@ import {
   AFFINE_EVENT_CHANNEL_NAME,
 } from '../shared/type';
 
+// Load persisted data from main process synchronously at preload time
 const initialGlobalState = ipcRenderer.sendSync(
   AFFINE_API_CHANNEL_NAME,
   'sharedStorage:getAllGlobalState'
@@ -15,6 +16,9 @@ const initialGlobalCache = ipcRenderer.sendSync(
   'sharedStorage:getAllGlobalCache'
 );
 
+// Unique id for this renderer instance, used to ignore self-originated broadcasts
+const CLIENT_ID: string = Math.random().toString(36).slice(2);
+
 function invokeWithCatch(key: string, ...args: any[]) {
   ipcRenderer.invoke(AFFINE_API_CHANNEL_NAME, key, ...args).catch(err => {
     console.error(`Failed to invoke ${key}`, err);
@@ -34,7 +38,23 @@ function createSharedStorageApi(
   memory.setAll(init);
   ipcRenderer.on(AFFINE_EVENT_CHANNEL_NAME, (_event, channel, updates) => {
     if (channel === `sharedStorage:${event}`) {
-      for (const [key, value] of Object.entries(updates)) {
+      for (const [key, raw] of Object.entries(updates)) {
+        // support both legacy plain value and new { v, r, s } structure
+        let value: any;
+        let source: string | undefined;
+
+        if (raw && typeof raw === 'object' && 'v' in raw) {
+          value = (raw as any).v;
+          source = (raw as any).s;
+        } else {
+          value = raw;
+        }
+
+        // Ignore our own broadcasts
+        if (source && source === CLIENT_ID) {
+          continue;
+        }
+
         if (value === undefined) {
           memory.del(key);
         } else {
@@ -47,11 +67,11 @@ function createSharedStorageApi(
   return {
     del(key: string) {
       memory.del(key);
-      invokeWithCatch(`sharedStorage:${api.del}`, key);
+      invokeWithCatch(`sharedStorage:${api.del}`, key, CLIENT_ID);
     },
     clear() {
       memory.clear();
-      invokeWithCatch(`sharedStorage:${api.clear}`);
+      invokeWithCatch(`sharedStorage:${api.clear}`, CLIENT_ID);
     },
     get<T>(key: string): T | undefined {
       return memory.get(key);
@@ -61,7 +81,7 @@ function createSharedStorageApi(
     },
     set(key: string, value: unknown) {
       memory.set(key, value);
-      invokeWithCatch(`sharedStorage:${api.set}`, key, value);
+      invokeWithCatch(`sharedStorage:${api.set}`, key, value, CLIENT_ID);
     },
     watch<T>(key: string, cb: (i: T | undefined) => void): () => void {
       const subscription = memory.watch(key).subscribe(i => cb(i as T));

packages/frontend/apps/electron/tsconfig.json

@@ -8,6 +8,7 @@
   "include": ["./src"],
   "references": [
     { "path": "../../../../tools/utils" },
+    { "path": "../../i18n" },
     { "path": "../../native" },
     { "path": "../../../common/nbstore" },
     { "path": "../../../common/infra" }

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

@@ -3,7 +3,7 @@
 	archiveVersion = 1;
 	classes = {
 	};
-	objectVersion = 56;
+	objectVersion = 77;
 	objects = {
 
 /* Begin PBXBuildFile section */
@@ -90,8 +90,6 @@
 /* Begin PBXFileSystemSynchronizedRootGroup section */
 		C45499AB2D140B5000E21978 /* NBStore */ = {
 			isa = PBXFileSystemSynchronizedRootGroup;
-			exceptions = (
-			);
 			path = NBStore;
 			sourceTree = "<group>";
 		};
@@ -339,9 +337,13 @@
 			);
 			inputFileListPaths = (
 			);
+			inputPaths = (
+			);
 			name = "[CP] Embed Pods Frameworks";
 			outputFileListPaths = (
 			);
+			outputPaths = (
+			);
 			runOnlyForDeploymentPostprocessing = 0;
 			shellPath = /bin/sh;
 			shellScript = "\"${PODS_ROOT}/Target Support Files/Pods-AFFiNE/Pods-AFFiNE-frameworks.sh\"\n";

packages/frontend/apps/ios/App/App.xcworkspace/xcshareddata/swiftpm/Package.resolved

@@ -27,15 +27,6 @@
         "version" : "1.1.6"
       }
     },
-    {
-      "identity" : "litext",
-      "kind" : "remoteSourceControl",
-      "location" : "https://github.com/Lakr233/Litext",
-      "state" : {
-        "revision" : "c37f3ab5826659854311e20d6c3942d4905b00b6",
-        "version" : "0.5.0"
-      }
-    },
     {
       "identity" : "lrucache",
       "kind" : "remoteSourceControl",
@@ -50,8 +41,8 @@
       "kind" : "remoteSourceControl",
       "location" : "https://github.com/Lakr233/MarkdownView",
       "state" : {
-        "revision" : "29a9da19d6dc21af4e629c423961b0f453ffe192",
-        "version" : "2.3.8"
+        "revision" : "446dba45be81c67d0717d19277367dcbe5b2fb12",
+        "version" : "3.1.9"
       }
     },
     {
@@ -68,8 +59,8 @@
       "kind" : "remoteSourceControl",
       "location" : "https://github.com/Lakr233/Splash",
       "state" : {
-        "revision" : "4d997712fe07f75695aacdf287aeb3b1f2c6ab88",
-        "version" : "0.17.0"
+        "revision" : "de9cde249fdb7a173a6e6b950ab18b11f6c2a557",
+        "version" : "0.18.0"
       }
     },
     {

packages/frontend/apps/ios/App/App/AffineViewController+AIButton.swift

@@ -13,23 +13,5 @@ extension AFFiNEViewController: IntelligentsButtonDelegate {
     // if it shows up then we are ready to go
     let controller = IntelligentsController()
     self.present(controller, animated: true)
-    //    IntelligentContext.shared.webView = webView
-    //    button.beginProgress()
-    //    IntelligentContext.shared.preparePresent { result in
-    //      DispatchQueue.main.async {
-    //        button.stopProgress()
-    //        switch result {
-    //        case .success:
-    //        case let .failure(failure):
-    //          let alert = UIAlertController(
-    //            title: "Error",
-    //            message: failure.localizedDescription,
-    //            preferredStyle: .alert
-    //          )
-    //          alert.addAction(UIAlertAction(title: "OK", style: .default))
-    //          self.present(alert, animated: true)
-    //        }
-    //      }
-    //    }
   }
 }

packages/frontend/apps/ios/App/App/AffineViewController.swift

@@ -64,12 +64,7 @@ class AFFiNEViewController: CAPBridgeViewController {
         switch result {
         case .failure: break
         case .success:
-          #if DEBUG
-          // only show the button in debug mode before we get done
           self.presentIntelligentsButton()
-          #else
-          break
-          #endif
         }
       }
     }

packages/frontend/apps/ios/App/App/Info.plist

@@ -69,5 +69,10 @@
 	</array>
 	<key>UIViewControllerBasedStatusBarAppearance</key>
 	<true/>
+  <key>NSAppTransportSecurity</key>
+  <dict>
+    <key>NSAllowsArbitraryLoads</key>
+    <true/>
+  </dict>
 </dict>
 </plist>

packages/frontend/apps/ios/App/Packages/AffineGraphQL/Sources/Operations/Queries/GetCopilotRecentSessionsQuery.graphql.swift

@@ -7,7 +7,7 @@ public class GetCopilotRecentSessionsQuery: GraphQLQuery {
   public static let operationName: String = "getCopilotRecentSessions"
   public static let operationDocument: ApolloAPI.OperationDocument = .init(
     definition: .init(
-      #"query getCopilotRecentSessions($workspaceId: String!, $limit: Int = 10) { currentUser { __typename copilot(workspaceId: $workspaceId) { __typename chats( pagination: { first: $limit } options: { fork: false, sessionOrder: desc, withMessages: true } ) { __typename ...PaginatedCopilotChats } } } }"#,
+      #"query getCopilotRecentSessions($workspaceId: String!, $limit: Int = 10) { currentUser { __typename copilot(workspaceId: $workspaceId) { __typename chats( pagination: { first: $limit } options: { fork: false, sessionOrder: desc, withMessages: false } ) { __typename ...PaginatedCopilotChats } } } }"#,
       fragments: [CopilotChatHistory.self, CopilotChatMessage.self, PaginatedCopilotChats.self]
     ))
 
@@ -69,7 +69,7 @@ public class GetCopilotRecentSessionsQuery: GraphQLQuery {
             "options": [
               "fork": false,
               "sessionOrder": "desc",
-              "withMessages": true
+              "withMessages": false
             ]
           ]),
         ] }

packages/frontend/apps/ios/App/Packages/AffineGraphQL/Sources/Operations/Queries/GetWorkspacePageByIdQuery.graphql.swift

@@ -7,7 +7,7 @@ public class GetWorkspacePageByIdQuery: GraphQLQuery {
   public static let operationName: String = "getWorkspacePageById"
   public static let operationDocument: ApolloAPI.OperationDocument = .init(
     definition: .init(
-      #"query getWorkspacePageById($workspaceId: String!, $pageId: String!) { workspace(id: $workspaceId) { __typename doc(docId: $pageId) { __typename id mode defaultRole public } } }"#
+      #"query getWorkspacePageById($workspaceId: String!, $pageId: String!) { workspace(id: $workspaceId) { __typename doc(docId: $pageId) { __typename id mode defaultRole public title summary } } }"#
     ))
 
   public var workspaceId: String
@@ -68,12 +68,16 @@ public class GetWorkspacePageByIdQuery: GraphQLQuery {
           .field("mode", GraphQLEnum<AffineGraphQL.PublicDocMode>.self),
           .field("defaultRole", GraphQLEnum<AffineGraphQL.DocRole>.self),
           .field("public", Bool.self),
+          .field("title", String?.self),
+          .field("summary", String?.self),
         ] }
 
         public var id: String { __data["id"] }
         public var mode: GraphQLEnum<AffineGraphQL.PublicDocMode> { __data["mode"] }
         public var defaultRole: GraphQLEnum<AffineGraphQL.DocRole> { __data["defaultRole"] }
         public var `public`: Bool { __data["public"] }
+        public var title: String? { __data["title"] }
+        public var summary: String? { __data["summary"] }
       }
     }
   }

packages/frontend/apps/ios/App/Packages/Intelligents/Package.swift

@@ -21,7 +21,7 @@ let package = Package(
     .package(url: "https://github.com/SwifterSwift/SwifterSwift.git", from: "6.0.0"),
     .package(url: "https://github.com/Recouse/EventSource", from: "0.1.4"),
     .package(url: "https://github.com/Lakr233/ListViewKit", from: "1.1.6"),
-    .package(url: "https://github.com/Lakr233/MarkdownView", exact: "2.3.8"),
+    .package(url: "https://github.com/Lakr233/MarkdownView", from: "3.1.9"),
   ],
   targets: [
     .target(name: "Intelligents", dependencies: [

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/ChatManager/ChatManager+CURD.swift

@@ -0,0 +1,33 @@
+//
+//  ChatManager+CURD.swift
+//  Intelligents
+//
+//  Created by 秋星桥 on 7/14/25.
+//
+
+import AffineGraphQL
+import Apollo
+import ApolloAPI
+import EventSource
+import Foundation
+import MarkdownParser
+import MarkdownView
+
+extension ChatManager {
+  func clearCurrentSession() {
+    guard let session = IntelligentContext.shared.currentSession else {
+      print("[-] no current session to clear")
+      return
+    }
+
+    let mutation = CleanupCopilotSessionMutation(input: .init(
+      docId: session.docId ?? "",
+      sessionIds: [session.id],
+      workspaceId: session.workspaceId
+    ))
+
+    QLService.shared.client.perform(mutation: mutation) { result in
+      print("[+] cleanup session result: \(result)")
+    }
+  }
+}

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/ChatManager/ChatManager+Stream.swift

@@ -24,53 +24,173 @@ private extension InputBoxData {
   }
 }
 
-extension ChatManager {
-  public func startUserRequest(
-    content: String,
-    inputBoxData: InputBoxData,
-    sessionId: String
-  ) {
+public extension ChatManager {
+  func startUserRequest(editorData: InputBoxData, sessionId: String) {
     append(sessionId: sessionId, UserMessageCellViewModel(
       id: .init(),
-      content: inputBoxData.text,
+      content: editorData.text,
       timestamp: .init()
     ))
     append(sessionId: sessionId, UserHintCellViewModel(
       id: .init(),
       timestamp: .init(),
-      imageAttachments: inputBoxData.imageAttachments,
-      fileAttachments: inputBoxData.fileAttachments,
-      docAttachments: inputBoxData.documentAttachments
+      imageAttachments: editorData.imageAttachments,
+      fileAttachments: editorData.fileAttachments,
+      docAttachments: editorData.documentAttachments
     ))
 
+    let viewModelId = append(sessionId: sessionId, AssistantMessageCellViewModel(
+      id: .init(),
+      content: "...",
+      timestamp: .init()
+    ))
+    scrollToBottomPublisher.send(sessionId)
+
+    guard let workspaceId = IntelligentContext.shared.currentWorkspaceId,
+          !workspaceId.isEmpty
+    else {
+      report(sessionId, ChatError.unknownError)
+      assertionFailure("Invalid workspace ID")
+      return
+    }
+
+    DispatchQueue.global().async {
+      self.prepareContext(
+        workspaceId: workspaceId,
+        sessionId: sessionId,
+        editorData: editorData,
+        viewModelId: viewModelId
+      )
+    }
+  }
+}
+
+private extension ChatManager {
+  func prepareContext(
+    workspaceId: String,
+    sessionId: String,
+    editorData: InputBoxData,
+    viewModelId: UUID
+  ) {
+    assert(!Thread.isMainThread)
+    let createContext = CreateCopilotContextMutation(
+      workspaceId: workspaceId,
+      sessionId: sessionId
+    )
+    QLService.shared.client.perform(mutation: createContext) { result in
+      DispatchQueue.main.async {
+        switch result {
+        case let .success(graphQLResult):
+          guard let contextId = graphQLResult.data?.createCopilotContext else {
+            self.report(sessionId, ChatError.invalidResponse)
+            return
+          }
+          print("[+] copilot context created: \(contextId)")
+
+          DispatchQueue.global().async {
+            let docAttachGroup = DispatchGroup()
+            for docAttach in editorData.documentAttachments {
+              let addDoc = AddContextDocMutation(
+                options: .init(
+                  contextId: contextId,
+                  docId: docAttach.documentID
+                )
+              )
+              docAttachGroup.enter()
+              QLService.shared.client.perform(mutation: addDoc) { result in
+                switch result {
+                case .success:
+                  print("[+] doc \(docAttach.documentID) added to context")
+                case let .failure(error):
+                  print("[-] addContextDoc failed: \(error)")
+                }
+                docAttachGroup.leave()
+              }
+            }
+
+            docAttachGroup.notify(queue: .global()) {
+              var contextSnippet = ""
+              if !editorData.documentAttachments.isEmpty {
+                let sem = DispatchSemaphore(value: 0)
+                let matchQuery = MatchContextQuery(
+                  contextId: .some(contextId),
+                  workspaceId: .some(workspaceId),
+                  content: editorData.text,
+                  limit: .none,
+                  scopedThreshold: .none,
+                  threshold: .none
+                )
+                QLService.shared.client.fetch(query: matchQuery) { result in
+                  switch result {
+                  case let .success(queryResult):
+                    let matches = queryResult.data?.currentUser?.copilot.contexts ?? []
+                    let matchDocs = matches.compactMap(\.matchWorkspaceDocs).flatMap(\.self)
+                    for context in matchDocs {
+                      contextSnippet += "<file docId=\"\(context.docId)\" chunk=\"\(context.chunk)\">\(context.content)</file>\n"
+                    }
+                  case let .failure(error):
+                    print("[-] matchContext failed: \(error)")
+                    // self.report(sessionId, error)
+                  }
+                  sem.signal()
+                }
+                sem.wait()
+              }
+              print("[+] context snippet prepared: \(contextSnippet)")
+              self.startCopilotResponse(
+                editorData: editorData,
+                contextSnippet: contextSnippet,
+                sessionId: sessionId,
+                viewModelId: viewModelId
+              )
+            }
+          }
+        case let .failure(error):
+          self.report(sessionId, error)
+          return
+        }
+      }
+    }
+  }
+
+  func startCopilotResponse(
+    editorData: InputBoxData,
+    contextSnippet: String,
+    sessionId: String,
+    viewModelId: UUID
+  ) {
+    assert(!Thread.isMainThread)
+
     let messageParameters: [String: AnyHashable] = [
       // packages/frontend/core/src/blocksuite/ai/provider/setup-provider.tsx
-      "docs": inputBoxData.documentAttachments.map(\.documentID), // affine doc
+      "docs": editorData.documentAttachments.map(\.documentID), // affine doc
       "files": [String](), // attachment in context, keep nil for now
-      "searchMode": inputBoxData.isSearchEnabled ? "MUST" : "AUTO",
+      "searchMode": editorData.isSearchEnabled ? "MUST" : "AUTO",
     ]
-    let uploadableAttachments: [GraphQLFile] = [
-      inputBoxData.fileAttachments.map { file -> GraphQLFile in
-        .init(
-          fieldName: file.name,
-          originalName: file.name,
-          data: file.data ?? .init()
-        )
+    let attachmentFieldName = "options.blobs"
+    var uploadableAttachments: [GraphQLFile] = [
+      editorData.fileAttachments.map { file -> GraphQLFile in
+        .init(fieldName: attachmentFieldName, originalName: file.name, data: file.data ?? .init())
       },
-      inputBoxData.imageAttachments.map { image -> GraphQLFile in
-        .init(
-          fieldName: image.hashValue.description,
-          originalName: "image.jpg",
-          data: image.imageData
-        )
+      editorData.imageAttachments.map { image -> GraphQLFile in
+        .init(fieldName: attachmentFieldName, originalName: "image.jpg", data: image.imageData)
       },
     ].flatMap(\.self)
     assert(uploadableAttachments.allSatisfy { !($0.data?.isEmpty ?? true) })
+    // in Apollo, filed name is handled as attached object to field when there is only one attachment
+    // to use array on our server, we need to append a dummy attachment
+    // which is ignored if data is empty and name is empty
+    if uploadableAttachments.count == 1 {
+      uploadableAttachments.append(.init(fieldName: attachmentFieldName, originalName: "", data: .init()))
+    }
     guard let input = try? CreateChatMessageInput(
-      content: .some(content),
+      attachments: [],
+      blobs: .some([]), // must have the placeholder
+      content: .some(contextSnippet.isEmpty ? editorData.text : "\(contextSnippet)\n\(editorData.text)"),
       params: .some(AffineGraphQL.JSON(_jsonValue: messageParameters)),
       sessionId: sessionId
     ) else {
+      report(sessionId, ChatError.unknownError)
       assertionFailure() // very unlikely to happen
       return
     }
@@ -83,11 +203,6 @@ extension ChatManager {
             self.report(sessionId, ChatError.invalidResponse)
             return
           }
-          let viewModelId = self.append(sessionId: sessionId, AssistantMessageCellViewModel(
-            id: .init(),
-            content: .init(),
-            timestamp: .init()
-          ))
           self.startStreamingResponse(
             sessionId: sessionId,
             messageId: messageIdentifier,
@@ -99,8 +214,10 @@ extension ChatManager {
       }
     }
   }
+}
 
-  private func startStreamingResponse(sessionId: String, messageId: String, applyingTo vmId: UUID) {
+private extension ChatManager {
+  func startStreamingResponse(sessionId: String, messageId: String, applyingTo vmId: UUID) {
     let base = IntelligentContext.shared.webViewMetadata[.currentServerBaseUrl] as? String
     guard let base, let url = URL(string: base) else {
       report(sessionId, ChatError.invalidServerConfiguration)
@@ -164,24 +281,11 @@ extension ChatManager {
     vmId: UUID
   ) {
     let result = MarkdownParser().parse(document)
-    var renderedContexts: [String: RenderedItem] = [:]
-    for (key, value) in result.mathContext {
-      let image = MathRenderer.renderToImage(
-        latex: value,
-        fontSize: MarkdownTheme.default.fonts.body.pointSize,
-        textColor: MarkdownTheme.default.colors.body
-      )?.withRenderingMode(.alwaysTemplate)
-      let renderedContext = RenderedItem(
-        image: image,
-        text: value
-      )
-      renderedContexts["math://\(key)"] = renderedContext
-    }
+    let content = MarkdownTextView.PreprocessContent(parserResult: result, theme: .default)
 
     with(sessionId: sessionId, vmId: vmId) { (viewModel: inout AssistantMessageCellViewModel) in
       viewModel.content = document
-      viewModel.documentBlocks = result.document
-      viewModel.documentRenderedContent = renderedContexts
+      viewModel.preprocessedContent = content
     }
   }
 }

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/ChatManager/ChatManager.swift

@@ -33,6 +33,12 @@ public class ChatManager: ObservableObject, @unchecked Sendable {
     closable.removeAll()
   }
 
+  public func clearAll() {
+    assert(Thread.isMainThread)
+    closeAll()
+    viewModels.removeAll()
+  }
+
   public func with(sessionId: String, _ action: (inout OrderedDictionary<MessageID, any ChatCellViewModel>) -> Void) {
     if Thread.isMainThread {
       if var sessionViewModels = viewModels[sessionId] {
@@ -59,8 +65,6 @@ public class ChatManager: ObservableObject, @unchecked Sendable {
           return
         }
         sessionViewModels[vmId] = vm
-      } else {
-        assertionFailure()
       }
     }
   }

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/Extension/BridgedWindowScript.swift

@@ -17,6 +17,7 @@ enum BridgedWindowScript: String {
   case getCurrentServerBaseUrl = "window.getCurrentServerBaseUrl()"
   case getCurrentWorkspaceId = "window.getCurrentWorkspaceId();"
   case getCurrentDocId = "window.getCurrentDocId();"
+  case getAiButtonFeatureFlag = "window.getAiButtonFeatureFlag();"
   case getCurrentI18nLocale = "window.getCurrentI18nLocale();"
   case createNewDocByMarkdownInCurrentWorkspace = "return await window.createNewDocByMarkdownInCurrentWorkspace(markdown, title);"
 

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/IntelligentContext/IntelligentContext+WebView.swift

@@ -17,6 +17,7 @@ extension IntelligentContext {
       (.currentWorkspaceId, .getCurrentWorkspaceId),
       (.currentServerBaseUrl, .getCurrentServerBaseUrl),
       (.currentI18nLocale, .getCurrentI18nLocale),
+      (.currentAiButtonFeatureFlag, .getAiButtonFeatureFlag),
     ]
     for (key, script) in keysAndScripts {
       DispatchQueue.main.async {

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/IntelligentContext/IntelligentContext.swift

@@ -40,6 +40,7 @@ public class IntelligentContext {
     case currentWorkspaceId
     case currentServerBaseUrl
     case currentI18nLocale
+    case currentAiButtonFeatureFlag
   }
 
   @Published public private(set) var currentSession: ChatSessionObject?
@@ -53,6 +54,7 @@ public class IntelligentContext {
   public enum IntelligentError: Error, LocalizedError {
     case loginRequired(String)
     case sessionCreationFailed(String)
+    case featureClosed
 
     public var errorDescription: String? {
       switch self {
@@ -60,6 +62,8 @@ public class IntelligentContext {
         "Login required: \(reason)"
       case let .sessionCreationFailed(reason):
         "Session creation failed: \(reason)"
+      case let .featureClosed:
+        "Intelligent feature closed"
       }
     }
   }
@@ -81,6 +85,11 @@ public class IntelligentContext {
       }
       webViewGroup.wait()
       webViewMetadata = webViewMetadataResult
+      
+      if webViewMetadataResult[.currentAiButtonFeatureFlag] as? Bool == false {
+        completion(.failure(IntelligentError.featureClosed))
+        return
+      }
 
       // Check required webView metadata
       guard let baseUrlString = webViewMetadataResult[.currentServerBaseUrl] as? String,

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/Interface/Controller/MainViewController/MainViewController+Input.swift

@@ -85,11 +85,7 @@ extension MainViewController: InputBoxDelegate {
     }
 
     ChatManager.shared.closeAll()
-    ChatManager.shared.startUserRequest(
-      content: inputData.text,
-      inputBoxData: inputData,
-      sessionId: currentSession.id
-    )
+    ChatManager.shared.startUserRequest(editorData: inputData, sessionId: currentSession.id)
   }
 
   private func showAlert(title: String, message: String) {

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/Interface/View/ChatCell/AssistantMessageCell.swift

@@ -21,11 +21,6 @@ class AssistantMessageCell: ChatBaseCell {
     contentView.addSubview(markdownView)
   }
 
-  override func prepareForReuse() {
-    super.prepareForReuse()
-    markdownView.prepareForReuse()
-  }
-
   override func configure(with viewModel: any ChatCellViewModel) {
     super.configure(with: viewModel)
 
@@ -33,10 +28,7 @@ class AssistantMessageCell: ChatBaseCell {
       assertionFailure()
       return
     }
-    markdownView.setMarkdown(
-      vm.documentBlocks,
-      renderedContent: vm.documentRenderedContent
-    )
+    markdownView.setMarkdown(vm.preprocessedContent)
   }
 
   override func layoutContentView(bounds: CGRect) {
@@ -53,10 +45,7 @@ class AssistantMessageCell: ChatBaseCell {
     markdownViewForSizeCalculation.frame = .init(
       x: 0, y: 0, width: width, height: .greatestFiniteMagnitude
     )
-    markdownViewForSizeCalculation.setMarkdown(
-      vm.documentBlocks,
-      renderedContent: vm.documentRenderedContent
-    )
+    markdownViewForSizeCalculation.setMarkdownManually(vm.preprocessedContent)
     let boundingSize = markdownViewForSizeCalculation.boundingSize(for: width)
     return ceil(boundingSize.height)
   }

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/Interface/View/ChatCell/ViewModel/CCVM+Assistant.swift

@@ -38,8 +38,7 @@ struct AssistantMessageCellViewModel: ChatCellViewModel {
   var citations: [CitationViewModel]?
   var actions: [MessageActionViewModel]?
 
-  var documentBlocks: [MarkdownBlockNode]
-  var documentRenderedContent: RenderContext
+  var preprocessedContent: MarkdownTextView.PreprocessContent
 
   init(
     id: UUID,
@@ -53,7 +52,7 @@ struct AssistantMessageCellViewModel: ChatCellViewModel {
     actions: [MessageActionViewModel]? = nil
   ) {
     // time expensive rendering should not happen here
-    assert(!Thread.isMainThread || content.isEmpty)
+    assert(!Thread.isMainThread || content.count < 10) // allow placeholder content
 
     self.id = id
     self.content = content
@@ -67,21 +66,10 @@ struct AssistantMessageCellViewModel: ChatCellViewModel {
 
     let parser = MarkdownParser()
     let parserResult = parser.parse(content)
-    documentBlocks = parserResult.document
-    var renderedContexts: [String: RenderedItem] = [:]
-    for (key, value) in parserResult.mathContext {
-      let image = MathRenderer.renderToImage(
-        latex: value,
-        fontSize: MarkdownTheme.default.fonts.body.pointSize,
-        textColor: MarkdownTheme.default.colors.body
-      )?.withRenderingMode(.alwaysTemplate)
-      let renderedContext = RenderedItem(
-        image: image,
-        text: value
-      )
-      renderedContexts["math://\(key)"] = renderedContext
-    }
-    documentRenderedContent = renderedContexts
+    preprocessedContent = MarkdownTextView.PreprocessContent(
+      parserResult: parserResult,
+      theme: .default,
+    )
   }
 }
 

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/Interface/View/InputBox/InputBoxFunctionBar.swift

@@ -6,8 +6,8 @@ private let unselectedColor: UIColor = .affineIconPrimary
 private let selectedColor: UIColor = .affineIconActivated
 
 private let configurableOptions: [ConfigurableOptions] = [
-  .networking,
-  .reasoning,
+//  .networking,
+//  .reasoning,
 ]
 enum ConfigurableOptions {
   case tool

packages/frontend/apps/ios/App/Packages/Intelligents/Sources/Intelligents/Interface/View/MainHeaderView/MainHeaderView.swift

@@ -22,11 +22,20 @@ class MainHeaderView: UIView {
     $0.textAlignment = .center
   }
 
+  private lazy var modelMenu = UIDeferredMenuElement.uncached { completion in
+    completion([])
+  }
+
   private lazy var dropdownButton = UIButton(type: .system).then {
     $0.imageView?.contentMode = .scaleAspectFit
     $0.setImage(UIImage.affineArrowDown, for: .normal)
     $0.tintColor = UIColor.affineIconPrimary
     $0.addTarget(self, action: #selector(dropdownButtonTapped), for: .touchUpInside)
+    $0.showsMenuAsPrimaryAction = true
+    $0.menu = UIMenu(options: [.displayInline], children: [
+      modelMenu,
+    ])
+    $0.isHidden = true
   }
 
   private lazy var centerStackView = UIStackView().then {
@@ -45,6 +54,13 @@ class MainHeaderView: UIView {
     $0.layer.cornerRadius = 8
     $0.addTarget(self, action: #selector(menuButtonTapped), for: .touchUpInside)
     $0.setContentHuggingPriority(.required, for: .horizontal)
+    $0.showsMenuAsPrimaryAction = true
+    $0.menu = .init(options: [.displayInline], children: [
+      UIAction(title: "Clear History", image: .affineBroom, handler: { _ in
+        ChatManager.shared.clearCurrentSession()
+        ChatManager.shared.clearAll()
+      }),
+    ])
   }
 
   private lazy var leftSpacerView = UIView()

packages/frontend/apps/ios/App/Podfile.lock

@@ -45,13 +45,13 @@ EXTERNAL SOURCES:
     :path: "../../../../../node_modules/capacitor-plugin-app-tracking-transparency"
 
 SPEC CHECKSUMS:
-  Capacitor: 03bc7cbdde6a629a8b910a9d7d78c3cc7ed09ea7
-  CapacitorApp: febecbb9582cb353aed037e18ec765141f880fe9
-  CapacitorBrowser: 6299776d496e968505464884d565992faa20444a
+  Capacitor: 106e7a4205f4618d582b886a975657c61179138d
+  CapacitorApp: d63334c052278caf5d81585d80b21905c6f93f39
+  CapacitorBrowser: 081852cf532acf77b9d2953f3a88fe5b9711fb06
   CapacitorCordova: 5967b9ba03915ef1d585469d6e31f31dc49be96f
-  CapacitorHaptics: 1f1e17041f435d8ead9ff2a34edd592c6aa6a8d6
-  CapacitorKeyboard: 09fd91dcde4f8a37313e7f11bde553ad1ed52036
-  CapacitorPluginAppTrackingTransparency: 92ae9c1cfb5cf477753db9269689332a686f675a
+  CapacitorHaptics: 70e47470fa1a6bd6338cd102552e3846b7f9a1b3
+  CapacitorKeyboard: 969647d0ca2e5c737d7300088e2517aa832434e2
+  CapacitorPluginAppTrackingTransparency: 2a2792623a5a72795f2e8f9ab3f1147573732fd8
   CryptoSwift: 967f37cea5a3294d9cce358f78861652155be483
 
 PODFILE CHECKSUM: 2c1e4be82121f2d9724ecf7e31dd14e165aeb082

packages/frontend/apps/ios/src/app.tsx

@@ -7,7 +7,6 @@ import { NavigationGestureProvider } from '@affine/core/mobile/modules/navigatio
 import { VirtualKeyboardProvider } from '@affine/core/mobile/modules/virtual-keyboard';
 import { router } from '@affine/core/mobile/router';
 import { configureCommonModules } from '@affine/core/modules';
-import { AIButtonProvider } from '@affine/core/modules/ai-button';
 import {
   AuthProvider,
   AuthService,
@@ -18,6 +17,7 @@ import {
   ValidatorProvider,
 } from '@affine/core/modules/cloud';
 import { DocsService } from '@affine/core/modules/doc';
+import { FeatureFlagService } from '@affine/core/modules/feature-flag';
 import { GlobalContextService } from '@affine/core/modules/global-context';
 import { I18nProvider } from '@affine/core/modules/i18n';
 import { LifecycleService } from '@affine/core/modules/lifecycle';
@@ -62,7 +62,6 @@ import { BlocksuiteMenuConfigProvider } from './bs-menu-config';
 import { ModalConfigProvider } from './modal-config';
 import { Auth } from './plugins/auth';
 import { Hashcash } from './plugins/hashcash';
-import { Intelligents } from './plugins/intelligents';
 import { NbStoreNativeDBApis } from './plugins/nbstore';
 import { writeEndpointToken } from './proxy';
 import { enableNavigationGesture$ } from './web-navigation-control';
@@ -162,14 +161,6 @@ framework.impl(HapticProvider, {
   selectionChanged: () => Haptics.selectionChanged(),
   selectionEnd: () => Haptics.selectionEnd(),
 });
-framework.impl(AIButtonProvider, {
-  presentAIButton: () => {
-    return Intelligents.presentIntelligentsButton();
-  },
-  dismissAIButton: () => {
-    return Intelligents.dismissIntelligentsButton();
-  },
-});
 framework.scope(ServerScope).override(AuthProvider, resolver => {
   const serverService = resolver.get(ServerService);
   const endpoint = serverService.server.baseUrl;
@@ -224,6 +215,10 @@ const frameworkProvider = framework.provider();
 (window as any).getCurrentI18nLocale = () => {
   return I18n.language;
 };
+(window as any).getAiButtonFeatureFlag = () => {
+  const featureFlagService = frameworkProvider.get(FeatureFlagService);
+  return featureFlagService.flags.enable_mobile_ai_button.value;
+};
 (window as any).getCurrentWorkspaceId = () => {
   const globalContextService = frameworkProvider.get(GlobalContextService);
   return globalContextService.globalContext.workspaceId.get();

packages/frontend/apps/ios/src/plugins/intelligents/definitions.ts

@@ -1,4 +0,0 @@
-export interface IntelligentsPlugin {
-  presentIntelligentsButton(): Promise<void>;
-  dismissIntelligentsButton(): Promise<void>;
-}

packages/frontend/apps/ios/src/plugins/intelligents/index.ts

@@ -1,8 +0,0 @@
-import { registerPlugin } from '@capacitor/core';
-
-import type { IntelligentsPlugin } from './definitions';
-
-const Intelligents = registerPlugin<IntelligentsPlugin>('Intelligents');
-
-export * from './definitions';
-export { Intelligents };

packages/frontend/component/package.json

@@ -34,6 +34,7 @@
     "@emotion/styled": "^11.14.0",
     "@radix-ui/react-avatar": "^1.1.2",
     "@radix-ui/react-collapsible": "^1.1.2",
+    "@radix-ui/react-context-menu": "^2.2.15",
     "@radix-ui/react-dialog": "^1.1.3",
     "@radix-ui/react-dropdown-menu": "^2.1.3",
     "@radix-ui/react-popover": "^1.1.3",

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

@@ -310,7 +310,10 @@ math {
 
 /* AI Block Diff */
 .ai-block-diff-deleted {
-  background-color: var(--aI-applyDeleteHighlight, #ffeaea) !important;
+  background-color: var(
+    --affine-v2-aI-applyDeleteHighlight,
+    #ffeaea
+  ) !important;
   border-radius: 4px !important;
   padding: 8px 0px !important;
   margin-bottom: 10px !important;

packages/frontend/component/src/ui/menu/desktop/context-menu.tsx

@@ -0,0 +1,58 @@
+import * as RadixContextMenu from '@radix-ui/react-context-menu';
+import clsx from 'clsx';
+import type { RefAttributes } from 'react';
+
+import * as styles from '../styles.css';
+import { DesktopMenuContext } from './context';
+import * as desktopStyles from './styles.css';
+
+export type ContextMenuProps = RadixContextMenu.ContextMenuProps &
+  RadixContextMenu.ContextMenuTriggerProps &
+  RefAttributes<HTMLSpanElement> & {
+    items: React.ReactNode;
+    contentProps?: RadixContextMenu.ContextMenuContentProps;
+  };
+
+const ContextMenuContextValue = {
+  type: 'context-menu',
+} as const;
+
+export const ContextMenu = ({
+  children,
+  onOpenChange,
+  dir,
+  modal,
+  items,
+  contentProps,
+  ...props
+}: ContextMenuProps) => {
+  return (
+    <DesktopMenuContext.Provider value={ContextMenuContextValue}>
+      <RadixContextMenu.Root
+        onOpenChange={onOpenChange}
+        dir={dir}
+        modal={modal}
+      >
+        <RadixContextMenu.Trigger {...props}>
+          {children}
+        </RadixContextMenu.Trigger>
+        <RadixContextMenu.Portal>
+          <RadixContextMenu.Content
+            className={clsx(
+              styles.menuContent,
+              desktopStyles.contentAnimation,
+              contentProps?.className
+            )}
+            style={{
+              zIndex: 'var(--affine-z-index-popover)',
+              ...contentProps?.style,
+            }}
+            {...contentProps}
+          >
+            {items}
+          </RadixContextMenu.Content>
+        </RadixContextMenu.Portal>
+      </RadixContextMenu.Root>
+    </DesktopMenuContext.Provider>
+  );
+};

packages/frontend/component/src/ui/menu/desktop/context.ts

@@ -0,0 +1,9 @@
+import { createContext } from 'react';
+
+interface DesktopMenuContextValue {
+  type: 'dropdown-menu' | 'context-menu';
+}
+
+export const DesktopMenuContext = createContext<DesktopMenuContextValue>({
+  type: 'dropdown-menu',
+});

packages/frontend/component/src/ui/menu/desktop/item.tsx

@@ -1,13 +1,30 @@
+import * as ContextMenu from '@radix-ui/react-context-menu';
 import * as DropdownMenu from '@radix-ui/react-dropdown-menu';
+import { useContext } from 'react';
 
 import type { MenuItemProps } from '../menu.types';
 import { useMenuItem } from '../use-menu-item';
+import { DesktopMenuContext } from './context';
 
 export const DesktopMenuItem = (props: MenuItemProps) => {
+  const { type } = useContext(DesktopMenuContext);
   const { className, children, otherProps } = useMenuItem(props);
-  return (
-    <DropdownMenu.Item className={className} {...otherProps}>
-      {children}
-    </DropdownMenu.Item>
-  );
+
+  if (type === 'dropdown-menu') {
+    return (
+      <DropdownMenu.Item className={className} {...otherProps}>
+        {children}
+      </DropdownMenu.Item>
+    );
+  }
+
+  if (type === 'context-menu') {
+    return (
+      <ContextMenu.Item className={className} {...otherProps}>
+        {children}
+      </ContextMenu.Item>
+    );
+  }
+
+  return null;
 };

packages/frontend/component/src/ui/menu/desktop/root.tsx

@@ -4,8 +4,13 @@ import React, { useCallback, useImperativeHandle, useState } from 'react';
 
 import type { MenuProps } from '../menu.types';
 import * as styles from '../styles.css';
+import { DesktopMenuContext } from './context';
 import * as desktopStyles from './styles.css';
 
+const MenuContextValue = {
+  type: 'dropdown-menu',
+} as const;
+
 export const DesktopMenu = ({
   children,
   items,
@@ -53,37 +58,39 @@ export const DesktopMenu = ({
 
   const ContentWrapper = noPortal ? React.Fragment : DropdownMenu.Portal;
   return (
-    <DropdownMenu.Root
-      modal={modal ?? false}
-      open={finalOpen}
-      onOpenChange={handleOpenChange}
-      {...rootOptions}
-    >
-      <DropdownMenu.Trigger
-        asChild
-        onClick={e => {
-          e.stopPropagation();
-          e.preventDefault();
-        }}
+    <DesktopMenuContext.Provider value={MenuContextValue}>
+      <DropdownMenu.Root
+        modal={modal ?? false}
+        open={finalOpen}
+        onOpenChange={handleOpenChange}
+        {...rootOptions}
       >
-        {children}
-      </DropdownMenu.Trigger>
-
-      <ContentWrapper {...portalOptions}>
-        <DropdownMenu.Content
-          className={clsx(
-            styles.menuContent,
-            desktopStyles.contentAnimation,
-            className
-          )}
-          sideOffset={4}
-          align="start"
-          style={{ zIndex: 'var(--affine-z-index-popover)', ...contentStyle }}
-          {...otherContentOptions}
+        <DropdownMenu.Trigger
+          asChild
+          onClick={e => {
+            e.stopPropagation();
+            e.preventDefault();
+          }}
         >
-          {items}
-        </DropdownMenu.Content>
-      </ContentWrapper>
-    </DropdownMenu.Root>
+          {children}
+        </DropdownMenu.Trigger>
+
+        <ContentWrapper {...portalOptions}>
+          <DropdownMenu.Content
+            className={clsx(
+              styles.menuContent,
+              desktopStyles.contentAnimation,
+              className
+            )}
+            sideOffset={4}
+            align="start"
+            style={{ zIndex: 'var(--affine-z-index-popover)', ...contentStyle }}
+            {...otherContentOptions}
+          >
+            {items}
+          </DropdownMenu.Content>
+        </ContentWrapper>
+      </DropdownMenu.Root>
+    </DesktopMenuContext.Provider>
   );
 };

packages/frontend/component/src/ui/menu/desktop/sub.tsx

@@ -1,11 +1,13 @@
 import { ArrowRightSmallIcon } from '@blocksuite/icons/rc';
+import * as ContextMenu from '@radix-ui/react-context-menu';
 import * as DropdownMenu from '@radix-ui/react-dropdown-menu';
 import clsx from 'clsx';
-import { useMemo } from 'react';
+import { useContext, useMemo } from 'react';
 
 import type { MenuSubProps } from '../menu.types';
 import * as styles from '../styles.css';
 import { useMenuItem } from '../use-menu-item';
+import { DesktopMenuContext } from './context';
 
 export const DesktopMenuSub = ({
   children: propsChildren,
@@ -19,12 +21,37 @@ export const DesktopMenuSub = ({
     ...otherSubContentOptions
   } = {},
 }: MenuSubProps) => {
+  const { type } = useContext(DesktopMenuContext);
   const { className, children, otherProps } = useMenuItem({
     children: propsChildren,
     suffixIcon: <ArrowRightSmallIcon />,
     ...triggerOptions,
   });
 
+  const contentClassName = useMemo(
+    () => clsx(styles.menuContent, subContentClassName),
+    [subContentClassName]
+  );
+
+  if (type === 'context-menu') {
+    return (
+      <ContextMenu.Sub defaultOpen={defaultOpen} {...otherSubOptions}>
+        <ContextMenu.SubTrigger className={className} {...otherProps}>
+          {children}
+        </ContextMenu.SubTrigger>
+        <ContextMenu.Portal {...portalOptions}>
+          <ContextMenu.SubContent
+            className={contentClassName}
+            style={{ zIndex: 'var(--affine-z-index-popover)', ...contentStyle }}
+            {...otherSubContentOptions}
+          >
+            {items}
+          </ContextMenu.SubContent>
+        </ContextMenu.Portal>
+      </ContextMenu.Sub>
+    );
+  }
+
   return (
     <DropdownMenu.Sub defaultOpen={defaultOpen} {...otherSubOptions}>
       <DropdownMenu.SubTrigger className={className} {...otherProps}>
@@ -32,10 +59,7 @@ export const DesktopMenuSub = ({
       </DropdownMenu.SubTrigger>
       <DropdownMenu.Portal {...portalOptions}>
         <DropdownMenu.SubContent
-          className={useMemo(
-            () => clsx(styles.menuContent, subContentClassName),
-            [subContentClassName]
-          )}
+          className={contentClassName}
           style={{ zIndex: 'var(--affine-z-index-popover)', ...contentStyle }}
           {...otherSubContentOptions}
         >

packages/frontend/component/src/ui/menu/index.ts

@@ -1,4 +1,5 @@
 export * from './menu.types';
+import { ContextMenu } from './desktop/context-menu';
 import { DesktopMenuItem } from './desktop/item';
 import { DesktopMenu } from './desktop/root';
 import { DesktopMenuSeparator } from './desktop/separator';
@@ -19,6 +20,7 @@ const MenuSub = BUILD_CONFIG.isMobileEdition ? MobileMenuSub : DesktopMenuSub;
 const Menu = BUILD_CONFIG.isMobileEdition ? MobileMenu : DesktopMenu;
 
 export {
+  ContextMenu,
   DesktopMenu,
   DesktopMenuItem,
   DesktopMenuSeparator,

packages/frontend/component/src/ui/safe-area/style.css.ts

@@ -1,4 +1,4 @@
-import { createVar, style } from '@vanilla-extract/css';
+import { createVar, fallbackVar, style } from '@vanilla-extract/css';
 
 export const topOffsetVar = createVar();
 export const bottomOffsetVar = createVar();
@@ -9,14 +9,13 @@ export const safeArea = style({
       paddingTop: `calc(${topOffsetVar} + 12px)`,
     },
     '&[data-bottom]': {
-      paddingBottom: `calc(${bottomOffsetVar} + 0px)`,
+      paddingBottom: `calc(${fallbackVar(bottomOffsetVar, '0px')} + 0px)`,
     },
     '&[data-standalone][data-top]': {
       paddingTop: `calc(env(safe-area-inset-top, 12px) + ${topOffsetVar})`,
     },
     '&[data-standalone][data-bottom]': {
-      // paddingBottom: 'env(safe-area-inset-bottom, 12px)',
-      paddingBottom: `calc(env(safe-area-inset-bottom, 0px) + ${bottomOffsetVar})`,
+      paddingBottom: `calc(env(safe-area-inset-bottom, 0px) + ${fallbackVar(bottomOffsetVar, '0px')})`,
     },
   },
 });

packages/frontend/core/package.json

@@ -16,6 +16,7 @@
     "@affine/graphql": "workspace:*",
     "@affine/i18n": "workspace:*",
     "@affine/nbstore": "workspace:*",
+    "@affine/reader": "workspace:*",
     "@affine/templates": "workspace:*",
     "@affine/track": "workspace:*",
     "@blocksuite/affine": "workspace:*",
@@ -34,6 +35,7 @@
     "@marsidev/react-turnstile": "^1.1.0",
     "@preact/signals-core": "^1.8.0",
     "@radix-ui/react-collapsible": "^1.1.2",
+    "@radix-ui/react-context-menu": "^2.1.15",
     "@radix-ui/react-dialog": "^1.1.3",
     "@radix-ui/react-popover": "^1.1.3",
     "@radix-ui/react-scroll-area": "^1.2.2",
@@ -82,6 +84,7 @@
     "react-transition-state": "^2.2.0",
     "react-virtuoso": "^4.12.3",
     "rxjs": "^7.8.1",
+    "semver": "^7.7.2",
     "ses": "^1.10.0",
     "shiki": "^3.7.0",
     "socket.io-client": "^4.8.1",
@@ -99,6 +102,7 @@
     "@types/bytes": "^3.1.5",
     "@types/image-blob-reduce": "^4.1.4",
     "@types/lodash-es": "^4.17.12",
+    "@types/semver": "^7",
     "@vanilla-extract/css": "^1.17.0",
     "fake-indexeddb": "^6.0.0",
     "lodash-es": "^4.17.21",

packages/frontend/core/src/__tests__/ai/utils/apply-model/apply-patch-to-doc.spec.ts

@@ -39,7 +39,8 @@ describe('applyPatchToDoc', () => {
     });
   });
 
-  it('should replace a block', async () => {
+  // FIXME: markdown parse error in test mode
+  it.skip('should replace a block', async () => {
     const host = affine`
     <affine-page id="page">
       <affine-note id="note">
@@ -73,7 +74,8 @@ describe('applyPatchToDoc', () => {
     });
   });
 
-  it('should insert a block at index', async () => {
+  // FIXME: markdown parse error in test mode
+  it.skip('should insert a block at index', async () => {
     const host = affine`
     <affine-page id="page">
       <affine-note id="note">

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

@@ -407,7 +407,8 @@ declare global {
       ) => Promise<CopilotChatHistoryFragment[] | undefined>;
       getRecentSessions: (
         workspaceId: string,
-        limit?: number
+        limit?: number,
+        offset?: number
       ) => Promise<AIRecentSession[] | undefined>;
       updateSession: (options: UpdateChatSessionInput) => Promise<string>;
     }

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

@@ -0,0 +1,186 @@
+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 { CopilotChatHistoryFragment } from '@affine/graphql';
+import { SignalWatcher, WithDisposable } from '@blocksuite/affine/global/lit';
+import { type NotificationService } from '@blocksuite/affine/shared/services';
+import { unsafeCSSVarV2 } from '@blocksuite/affine/shared/theme';
+import type { EditorHost } from '@blocksuite/affine/std';
+import { ShadowlessElement } from '@blocksuite/affine/std';
+import type { ExtensionType, Store } from '@blocksuite/affine/store';
+import { CenterPeekIcon } from '@blocksuite/icons/lit';
+import { css, html, nothing } from 'lit';
+import { property } from 'lit/decorators.js';
+
+import type { SearchMenuConfig } from '../components/ai-chat-add-context';
+import type { DocDisplayConfig } from '../components/ai-chat-chips';
+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 type { AppSidebarConfig } from './chat-config';
+
+export class AIChatPanelTitle extends SignalWatcher(
+  WithDisposable(ShadowlessElement)
+) {
+  static override styles = css`
+    .ai-chat-panel-title {
+      background: var(--affine-background-primary-color);
+      position: relative;
+      padding: 8px var(--h-padding, 16px);
+      width: 100%;
+      height: 36px;
+      display: flex;
+      justify-content: space-between;
+      align-items: center;
+      z-index: 1;
+
+      svg {
+        width: 18px;
+        height: 18px;
+        color: var(--affine-text-secondary-color);
+      }
+
+      .chat-panel-title-text {
+        font-size: 14px;
+        font-weight: 500;
+        color: var(--affine-text-secondary-color);
+      }
+
+      .chat-panel-playground {
+        cursor: pointer;
+        padding: 2px;
+        margin-left: 8px;
+        margin-right: auto;
+        display: flex;
+        justify-content: center;
+        align-items: center;
+      }
+
+      .chat-panel-playground:hover svg {
+        color: ${unsafeCSSVarV2('icon/activated')};
+      }
+    }
+  `;
+
+  @property({ attribute: false })
+  accessor host!: EditorHost;
+
+  @property({ attribute: false })
+  accessor doc!: Store;
+
+  @property({ attribute: false })
+  accessor playgroundConfig!: AIPlaygroundConfig;
+
+  @property({ attribute: false })
+  accessor appSidebarConfig!: AppSidebarConfig;
+
+  @property({ attribute: false })
+  accessor networkSearchConfig!: AINetworkSearchConfig;
+
+  @property({ attribute: false })
+  accessor reasoningConfig!: AIReasoningConfig;
+
+  @property({ attribute: false })
+  accessor searchMenuConfig!: SearchMenuConfig;
+
+  @property({ attribute: false })
+  accessor docDisplayConfig!: DocDisplayConfig;
+
+  @property({ attribute: false })
+  accessor extensions!: ExtensionType[];
+
+  @property({ attribute: false })
+  accessor affineFeatureFlagService!: FeatureFlagService;
+
+  @property({ attribute: false })
+  accessor affineWorkspaceDialogService!: WorkspaceDialogService;
+
+  @property({ attribute: false })
+  accessor affineThemeService!: AppThemeService;
+
+  @property({ attribute: false })
+  accessor notificationService!: NotificationService;
+
+  @property({ attribute: false })
+  accessor session!: CopilotChatHistoryFragment | null | undefined;
+
+  @property({ attribute: false })
+  accessor status!: ChatStatus;
+
+  @property({ attribute: false })
+  accessor embeddingProgress: [number, number] = [0, 0];
+
+  @property({ attribute: false })
+  accessor newSession!: () => void;
+
+  @property({ attribute: false })
+  accessor togglePin!: () => void;
+
+  @property({ attribute: false })
+  accessor openSession!: (sessionId: string) => void;
+
+  @property({ attribute: false })
+  accessor openDoc!: (docId: string, sessionId: string) => void;
+
+  private readonly openPlayground = () => {
+    const playgroundContent = html`
+      <playground-content
+        .host=${this.host}
+        .doc=${this.doc}
+        .networkSearchConfig=${this.networkSearchConfig}
+        .reasoningConfig=${this.reasoningConfig}
+        .playgroundConfig=${this.playgroundConfig}
+        .appSidebarConfig=${this.appSidebarConfig}
+        .searchMenuConfig=${this.searchMenuConfig}
+        .docDisplayConfig=${this.docDisplayConfig}
+        .extensions=${this.extensions}
+        .affineFeatureFlagService=${this.affineFeatureFlagService}
+        .affineThemeService=${this.affineThemeService}
+        .notificationService=${this.notificationService}
+        .affineWorkspaceDialogService=${this.affineWorkspaceDialogService}
+      ></playground-content>
+    `;
+
+    createPlaygroundModal(playgroundContent, 'AI Playground');
+  };
+
+  override render() {
+    const [done, total] = this.embeddingProgress;
+    const isEmbedding = total > 0 && done < total;
+
+    return html`
+      <div class="ai-chat-panel-title">
+        <div class="chat-panel-title-text">
+          ${isEmbedding
+            ? html`<span data-testid="chat-panel-embedding-progress"
+                >Embedding ${done}/${total}</span
+              >`
+            : 'AFFiNE AI'}
+        </div>
+        ${this.playgroundConfig.visible.value
+          ? html`
+              <div class="chat-panel-playground" @click=${this.openPlayground}>
+                ${CenterPeekIcon()}
+              </div>
+            `
+          : nothing}
+        <ai-chat-toolbar
+          .session=${this.session}
+          .workspaceId=${this.doc.workspace.id}
+          .docId=${this.doc.id}
+          .status=${this.status}
+          .onNewSession=${this.newSession}
+          .onTogglePin=${this.togglePin}
+          .onOpenSession=${this.openSession}
+          .onOpenDoc=${this.openDoc}
+          .docDisplayConfig=${this.docDisplayConfig}
+          .notificationService=${this.notificationService}
+        ></ai-chat-toolbar>
+      </div>
+    `;
+  }
+}

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

@@ -1,3 +1,4 @@
+import type { AIDraftService } from '@affine/core/modules/ai-button';
 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';
@@ -9,13 +10,11 @@ import type {
 } from '@affine/graphql';
 import { SignalWatcher, WithDisposable } from '@blocksuite/affine/global/lit';
 import { type NotificationService } from '@blocksuite/affine/shared/services';
-import { unsafeCSSVarV2 } from '@blocksuite/affine/shared/theme';
 import type { EditorHost } from '@blocksuite/affine/std';
 import { ShadowlessElement } from '@blocksuite/affine/std';
 import type { ExtensionType, Store } from '@blocksuite/affine/store';
-import { CenterPeekIcon } from '@blocksuite/icons/lit';
 import { type Signal, signal } from '@preact/signals-core';
-import { css, html, nothing, type PropertyValues } from 'lit';
+import { css, html, type PropertyValues } from 'lit';
 import { property, state } from 'lit/decorators.js';
 import { keyed } from 'lit/directives/keyed.js';
 
@@ -29,7 +28,6 @@ import type {
   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';
 
@@ -43,12 +41,13 @@ export class ChatPanel extends SignalWatcher(
 
       .chat-panel-container {
         height: 100%;
+        display: flex;
+        flex-direction: column;
       }
 
-      .chat-panel-title-text {
-        font-size: 14px;
-        font-weight: 500;
-        color: var(--affine-text-secondary-color);
+      ai-chat-content {
+        height: 0;
+        flex-grow: 1;
       }
 
       .chat-loading-container {
@@ -72,20 +71,6 @@ export class ChatPanel extends SignalWatcher(
         font-size: var(--affine-font-sm);
         color: var(--affine-text-secondary-color);
       }
-
-      .chat-panel-playground {
-        cursor: pointer;
-        padding: 2px;
-        margin-left: 8px;
-        margin-right: auto;
-        display: flex;
-        justify-content: center;
-        align-items: center;
-      }
-
-      .chat-panel-playground:hover svg {
-        color: ${unsafeCSSVarV2('icon/activated')};
-      }
     }
   `;
 
@@ -131,6 +116,9 @@ export class ChatPanel extends SignalWatcher(
   @property({ attribute: false })
   accessor notificationService!: NotificationService;
 
+  @property({ attribute: false })
+  accessor aiDraftService!: AIDraftService;
+
   @state()
   accessor session: CopilotChatHistoryFragment | null | undefined;
 
@@ -150,40 +138,6 @@ export class ChatPanel extends SignalWatcher(
     return this.session !== undefined;
   }
 
-  private get chatTitle() {
-    const [done, total] = this.embeddingProgress;
-    const isEmbedding = total > 0 && done < total;
-
-    return html`
-      <div class="chat-panel-title-text">
-        ${isEmbedding
-          ? html`<span data-testid="chat-panel-embedding-progress"
-              >Embedding ${done}/${total}</span
-            >`
-          : 'AFFiNE AI'}
-      </div>
-      ${this.playgroundConfig.visible.value
-        ? html`
-            <div class="chat-panel-playground" @click=${this.openPlayground}>
-              ${CenterPeekIcon()}
-            </div>
-          `
-        : nothing}
-      <ai-chat-toolbar
-        .session=${this.session}
-        .workspaceId=${this.doc.workspace.id}
-        .docId=${this.doc.id}
-        .status=${this.status}
-        .onNewSession=${this.newSession}
-        .onTogglePin=${this.togglePin}
-        .onOpenSession=${this.openSession}
-        .onOpenDoc=${this.openDoc}
-        .docDisplayConfig=${this.docDisplayConfig}
-        .notificationService=${this.notificationService}
-      ></ai-chat-toolbar>
-    `;
-  }
-
   private readonly getSessionIdFromUrl = () => {
     if (this.affineWorkbenchService) {
       const { workbench } = this.affineWorkbenchService;
@@ -368,28 +322,6 @@ export class ChatPanel extends SignalWatcher(
     }
   };
 
-  private readonly openPlayground = () => {
-    const playgroundContent = html`
-      <playground-content
-        .host=${this.host}
-        .doc=${this.doc}
-        .networkSearchConfig=${this.networkSearchConfig}
-        .reasoningConfig=${this.reasoningConfig}
-        .playgroundConfig=${this.playgroundConfig}
-        .appSidebarConfig=${this.appSidebarConfig}
-        .searchMenuConfig=${this.searchMenuConfig}
-        .docDisplayConfig=${this.docDisplayConfig}
-        .extensions=${this.extensions}
-        .affineFeatureFlagService=${this.affineFeatureFlagService}
-        .affineThemeService=${this.affineThemeService}
-        .notificationService=${this.notificationService}
-        .affineWorkspaceDialogService=${this.affineWorkspaceDialogService}
-      ></playground-content>
-    `;
-
-    createPlaygroundModal(playgroundContent, 'AI Playground');
-  };
-
   protected override updated(changedProperties: PropertyValues) {
     if (changedProperties.has('doc')) {
       if (this.session?.pinned) {
@@ -441,10 +373,31 @@ export class ChatPanel extends SignalWatcher(
     }
 
     return html`<div class="chat-panel-container">
+      <ai-chat-panel-title
+        .host=${this.host}
+        .doc=${this.doc}
+        .playgroundConfig=${this.playgroundConfig}
+        .appSidebarConfig=${this.appSidebarConfig}
+        .networkSearchConfig=${this.networkSearchConfig}
+        .reasoningConfig=${this.reasoningConfig}
+        .searchMenuConfig=${this.searchMenuConfig}
+        .docDisplayConfig=${this.docDisplayConfig}
+        .extensions=${this.extensions}
+        .affineFeatureFlagService=${this.affineFeatureFlagService}
+        .affineWorkspaceDialogService=${this.affineWorkspaceDialogService}
+        .affineThemeService=${this.affineThemeService}
+        .notificationService=${this.notificationService}
+        .session=${this.session}
+        .status=${this.status}
+        .embeddingProgress=${this.embeddingProgress}
+        .newSession=${this.newSession}
+        .togglePin=${this.togglePin}
+        .openSession=${this.openSession}
+        .openDoc=${this.openDoc}
+      ></ai-chat-panel-title>
       ${keyed(
         this.hasPinned ? this.session?.sessionId : this.doc.id,
         html`<ai-chat-content
-          .chatTitle=${this.chatTitle}
           .host=${this.host}
           .session=${this.session}
           .createSession=${this.createSession}
@@ -459,9 +412,11 @@ export class ChatPanel extends SignalWatcher(
           .affineWorkspaceDialogService=${this.affineWorkspaceDialogService}
           .affineThemeService=${this.affineThemeService}
           .notificationService=${this.notificationService}
+          .aiDraftService=${this.aiDraftService}
           .onEmbeddingProgressChange=${this.onEmbeddingProgressChange}
           .onContextChange=${this.onContextChange}
           .width=${this.sidebarWidth}
+          .onOpenDoc=${this.openDoc}
         ></ai-chat-content>`
       )}
     </div>`;

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

@@ -86,6 +86,9 @@ export class ChatMessageAssistant extends WithDisposable(ShadowlessElement) {
   @property({ attribute: false })
   accessor docDisplayService!: DocDisplayConfig;
 
+  @property({ attribute: false })
+  accessor onOpenDoc!: (docId: string, sessionId?: string) => void;
+
   get state() {
     const { isLast, status } = this;
     return isLast
@@ -146,6 +149,7 @@ export class ChatMessageAssistant extends WithDisposable(ShadowlessElement) {
       .notificationService=${this.notificationService}
       .theme=${this.affineThemeService.appTheme.themeSignal}
       .docDisplayService=${this.docDisplayService}
+      .onOpenDoc=${this.onOpenDoc}
     ></chat-content-stream-objects>`;
   }
 

packages/frontend/core/src/blocksuite/ai/components/ai-artifact-skeleton/index.ts

@@ -0,0 +1,196 @@
+import { unsafeCSSVarV2 } from '@blocksuite/affine-shared/theme';
+import { css, html, LitElement, nothing, type TemplateResult } from 'lit';
+import { property } from 'lit/decorators.js';
+
+/**
+ * ArtifactSkeleton
+ *
+ * A lightweight loading skeleton used while an artifact preview is fetching / processing.
+ * It mimics the layout of a document – an optional icon followed by several animated grey lines.
+ *
+ * Animation is implemented with pure CSS keyframes (no framer-motion dependency).
+ * Only a single prop is supported for now:
+ *   - `icon` – TemplateResult that will be rendered at the top-left position.
+ */
+export class ArtifactSkeleton extends LitElement {
+  /* ----- Styling --------------------------------------------------------------------------- */
+  static override styles = css`
+    :host {
+      /* The host is an inline-block so it can size to its contents. */
+      display: inline-block;
+      position: relative;
+      /* The size roughly follows the design used in the legacy React implementation. */
+      width: 250px;
+      height: 200px;
+      box-sizing: border-box;
+    }
+
+    /* Optional icon wrapper */
+    .icon {
+      position: absolute;
+      top: 10px;
+      left: 11px;
+      width: 32px;
+      height: 32px;
+
+      svg {
+        color: ${unsafeCSSVarV2('icon/activated')};
+        width: 100%;
+        height: 100%;
+      }
+    }
+
+    /* Base line style */
+    .line {
+      position: absolute;
+      left: 11px;
+      height: 10px;
+      border-radius: 6px;
+      background-color: ${unsafeCSSVarV2('layer/background/tertiary')};
+    }
+
+    /* Keyframes for each line – width cycles through a handful of values to create movement */
+    @keyframes line1Anim {
+      0%,
+      100% {
+        width: 98px;
+      }
+      25% {
+        width: 120px;
+      }
+      50% {
+        width: 85px;
+      }
+      75% {
+        width: 110px;
+      }
+    }
+    @keyframes line2Anim {
+      0%,
+      100% {
+        width: 195px;
+      }
+      30% {
+        width: 180px;
+      }
+      60% {
+        width: 210px;
+      }
+      80% {
+        width: 165px;
+      }
+    }
+    @keyframes line3Anim {
+      0%,
+      100% {
+        width: 163px;
+      }
+      40% {
+        width: 140px;
+      }
+      70% {
+        width: 180px;
+      }
+      90% {
+        width: 155px;
+      }
+    }
+    @keyframes line4Anim {
+      0%,
+      100% {
+        width: 107px;
+      }
+      20% {
+        width: 130px;
+      }
+      60% {
+        width: 90px;
+      }
+      85% {
+        width: 115px;
+      }
+    }
+    @keyframes line5Anim {
+      0%,
+      100% {
+        width: 134px;
+      }
+      35% {
+        width: 160px;
+      }
+      65% {
+        width: 120px;
+      }
+      80% {
+        width: 145px;
+      }
+    }
+    @keyframes line6Anim {
+      0%,
+      100% {
+        width: 154px;
+      }
+      30% {
+        width: 135px;
+      }
+      55% {
+        width: 175px;
+      }
+      75% {
+        width: 160px;
+      }
+    }
+
+    .line1 {
+      top: 48.5px;
+      animation: line1Anim 3.2s ease-in-out infinite;
+    }
+    .line2 {
+      top: 73.5px;
+      animation: line2Anim 4.1s ease-in-out infinite;
+    }
+    .line3 {
+      top: 98.5px;
+      animation: line3Anim 2.8s ease-in-out infinite;
+    }
+    .line4 {
+      top: 123.5px;
+      animation: line4Anim 3.7s ease-in-out infinite;
+    }
+    .line5 {
+      top: 148.5px;
+      animation: line5Anim 3.5s ease-in-out infinite;
+    }
+    .line6 {
+      top: 170.5px;
+      animation: line6Anim 4.3s ease-in-out infinite;
+    }
+  `;
+
+  /* ----- Public API ------------------------------------------------------------------------ */
+  /**
+   * Optional icon rendered at the top-left corner.
+   * It should be a lit `TemplateResult`, typically an inline SVG.
+   */
+  @property({ attribute: false })
+  accessor icon: TemplateResult | null = null;
+
+  /* ----- Render --------------------------------------------------------------------------- */
+  override render() {
+    return html`
+      ${this.icon ? html`<div class="icon">${this.icon}</div>` : nothing}
+      <div class="line line1"></div>
+      <div class="line line2"></div>
+      <div class="line line3"></div>
+      <div class="line line4"></div>
+      <div class="line line5"></div>
+      <div class="line line6"></div>
+    `;
+  }
+}
+
+declare global {
+  interface HTMLElementTagNameMap {
+    'artifact-skeleton': ArtifactSkeleton;
+  }
+}

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

@@ -1,5 +1,6 @@
 import './ai-chat-composer-tip';
 
+import type { AIDraftService } from '@affine/core/modules/ai-button';
 import type { WorkspaceDialogService } from '@affine/core/modules/dialogs';
 import type {
   ContextEmbedStatus,
@@ -116,6 +117,9 @@ export class AIChatComposer extends SignalWatcher(
   @property({ attribute: false })
   accessor notificationService!: NotificationService;
 
+  @property({ attribute: false })
+  accessor aiDraftService!: AIDraftService;
+
   @state()
   accessor chips: ChatChip[] = [];
 
@@ -161,6 +165,7 @@ export class AIChatComposer extends SignalWatcher(
         .reasoningConfig=${this.reasoningConfig}
         .docDisplayConfig=${this.docDisplayConfig}
         .searchMenuConfig=${this.searchMenuConfig}
+        .aiDraftService=${this.aiDraftService}
         .portalContainer=${this.portalContainer}
         .onChatSuccess=${this.onChatSuccess}
         .trackOptions=${this.trackOptions}

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

@@ -1,3 +1,5 @@
+import type { AIDraftService } from '@affine/core/modules/ai-button';
+import type { AIDraftState } from '@affine/core/modules/ai-button/services/ai-draft';
 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';
@@ -10,17 +12,12 @@ 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';
-import {
-  css,
-  html,
-  nothing,
-  type PropertyValues,
-  type TemplateResult,
-} from 'lit';
+import { css, html, type PropertyValues, type TemplateResult } from 'lit';
 import { property, state } from 'lit/decorators.js';
 import { classMap } from 'lit/directives/class-map.js';
 import { createRef, type Ref, ref } from 'lit/directives/ref.js';
 import { styleMap } from 'lit/directives/style-map.js';
+import { pick } from 'lodash-es';
 
 import { HISTORY_IMAGE_ACTIONS } from '../../chat-panel/const';
 import { type AIChatParams, AIProvider } from '../../provider/ai-provider';
@@ -60,24 +57,6 @@ export class AIChatContent extends SignalWatcher(
       justify-content: center;
       height: 100%;
 
-      .ai-chat-title {
-        background: var(--affine-background-primary-color);
-        position: relative;
-        padding: 8px var(--h-padding);
-        width: 100%;
-        height: 36px;
-        display: flex;
-        justify-content: space-between;
-        align-items: center;
-        z-index: 1;
-
-        svg {
-          width: 18px;
-          height: 18px;
-          color: var(--affine-text-secondary-color);
-        }
-      }
-
       ai-chat-messages {
         flex: 1;
         overflow-y: auto;
@@ -129,9 +108,6 @@ export class AIChatContent extends SignalWatcher(
   @property({ attribute: false })
   accessor onboardingOffsetY!: number;
 
-  @property({ attribute: false })
-  accessor chatTitle: TemplateResult<1> | undefined;
-
   @property({ attribute: false })
   accessor host: EditorHost | null | undefined;
 
@@ -176,6 +152,9 @@ export class AIChatContent extends SignalWatcher(
   @property({ attribute: false })
   accessor notificationService!: NotificationService;
 
+  @property({ attribute: false })
+  accessor aiDraftService!: AIDraftService;
+
   @property({ attribute: false })
   accessor onEmbeddingProgressChange:
     | ((count: Record<ContextEmbedStatus, number>) => void)
@@ -184,6 +163,9 @@ export class AIChatContent extends SignalWatcher(
   @property({ attribute: false })
   accessor onContextChange!: (context: Partial<ChatContextValue>) => void;
 
+  @property({ attribute: false })
+  accessor onOpenDoc!: (docId: string, sessionId?: string) => void;
+
   @property({ attribute: false })
   accessor width: Signal<number | undefined> | undefined;
 
@@ -287,6 +269,19 @@ export class AIChatContent extends SignalWatcher(
   private readonly updateContext = (context: Partial<ChatContextValue>) => {
     this.chatContextValue = { ...this.chatContextValue, ...context };
     this.onContextChange?.(context);
+    this.updateDraft(context).catch(console.error);
+  };
+
+  private readonly updateDraft = async (context: Partial<ChatContextValue>) => {
+    const draft: Partial<AIDraftState> = pick(context, [
+      'quote',
+      'images',
+      'markdown',
+    ]);
+    if (!Object.keys(draft).length) {
+      return;
+    }
+    await this.aiDraftService.setDraft(draft);
   };
 
   private readonly initChatContent = async () => {
@@ -328,16 +323,6 @@ export class AIChatContent extends SignalWatcher(
     }
   }
 
-  public reset() {
-    this.updateContext(DEFAULT_CHAT_CONTEXT_VALUE);
-    this.closePreviewPanel(true);
-  }
-
-  public reloadSession() {
-    this.reset();
-    this.initChatContent().catch(console.error);
-  }
-
   public openPreviewPanel(content?: TemplateResult<1>) {
     this.showPreviewPanel = true;
     if (content) this.previewPanelContent = content;
@@ -356,8 +341,19 @@ export class AIChatContent extends SignalWatcher(
 
   override connectedCallback() {
     super.connectedCallback();
+
     this.initChatContent().catch(console.error);
 
+    this.aiDraftService
+      .getDraft()
+      .then(draft => {
+        this.chatContextValue = {
+          ...this.chatContextValue,
+          ...draft,
+        };
+      })
+      .catch(console.error);
+
     this._disposables.add(
       AIProvider.slots.actions.subscribe(({ event }) => {
         const { status } = this.chatContextValue;
@@ -390,10 +386,7 @@ export class AIChatContent extends SignalWatcher(
   }
 
   override render() {
-    const left = html`${this.chatTitle
-        ? html`<div class="ai-chat-title">${this.chatTitle}</div>`
-        : nothing}
-      <ai-chat-messages
+    const left = html` <ai-chat-messages
         class=${classMap({
           'ai-chat-messages': true,
           'independent-mode': !!this.independentMode,
@@ -418,6 +411,7 @@ export class AIChatContent extends SignalWatcher(
         .independentMode=${this.independentMode}
         .messages=${this.messages}
         .docDisplayService=${this.docDisplayConfig}
+        .onOpenDoc=${this.onOpenDoc}
       ></ai-chat-messages>
       <ai-chat-composer
         style=${styleMap({
@@ -439,6 +433,7 @@ export class AIChatContent extends SignalWatcher(
         .searchMenuConfig=${this.searchMenuConfig}
         .affineWorkspaceDialogService=${this.affineWorkspaceDialogService}
         .notificationService=${this.notificationService}
+        .aiDraftService=${this.aiDraftService}
         .trackOptions=${{
           where: 'chat-panel',
           control: 'chat-send',

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

@@ -1,10 +1,11 @@
+import type { AIDraftService } from '@affine/core/modules/ai-button';
 import type { CopilotChatHistoryFragment } from '@affine/graphql';
 import { SignalWatcher, WithDisposable } from '@blocksuite/affine/global/lit';
 import { unsafeCSSVar, unsafeCSSVarV2 } from '@blocksuite/affine/shared/theme';
 import type { EditorHost } from '@blocksuite/affine/std';
 import { ShadowlessElement } from '@blocksuite/affine/std';
 import { ArrowUpBigIcon, CloseIcon } from '@blocksuite/icons/lit';
-import { css, html, nothing } from 'lit';
+import { css, html, nothing, type PropertyValues } from 'lit';
 import { property, query, state } from 'lit/decorators.js';
 import { repeat } from 'lit/directives/repeat.js';
 import { styleMap } from 'lit/directives/style-map.js';
@@ -351,6 +352,9 @@ export class AIChatInput extends SignalWatcher(
   @property({ attribute: false })
   accessor searchMenuConfig!: SearchMenuConfig;
 
+  @property({ attribute: false })
+  accessor aiDraftService!: AIDraftService;
+
   @property({ attribute: false })
   accessor isRootSession: boolean = true;
 
@@ -379,6 +383,7 @@ export class AIChatInput extends SignalWatcher(
 
   override connectedCallback() {
     super.connectedCallback();
+
     this._disposables.add(
       AIProvider.slots.requestSendWithChat.subscribe(
         (params: AISendParams | null) => {
@@ -399,6 +404,17 @@ export class AIChatInput extends SignalWatcher(
     );
   }
 
+  protected override firstUpdated(changedProperties: PropertyValues): void {
+    super.firstUpdated(changedProperties);
+    this.aiDraftService
+      .getDraft()
+      .then(draft => {
+        this.textarea.value = draft.input;
+        this.isInputEmpty = !this.textarea.value.trim();
+      })
+      .catch(console.error);
+  }
+
   protected override render() {
     const { images, status } = this.chatContextValue;
     const hasImages = images.length > 0;
@@ -506,9 +522,11 @@ export class AIChatInput extends SignalWatcher(
     }
   };
 
-  private readonly _handleInput = () => {
+  private readonly _handleInput = async () => {
     const { textarea } = this;
-    this.isInputEmpty = !textarea.value.trim();
+    const value = textarea.value.trim();
+    this.isInputEmpty = !value;
+
     textarea.style.height = 'auto';
     textarea.style.height = textarea.scrollHeight + 'px';
     let imagesHeight = this.imagePreviewGrid?.scrollHeight ?? 0;
@@ -517,6 +535,10 @@ export class AIChatInput extends SignalWatcher(
       textarea.style.height = '148px';
       textarea.style.overflowY = 'scroll';
     }
+
+    await this.aiDraftService.setDraft({
+      input: value,
+    });
   };
 
   private readonly _handleKeyDown = async (evt: KeyboardEvent) => {
@@ -572,6 +594,9 @@ export class AIChatInput extends SignalWatcher(
     this.textarea.style.height = 'unset';
 
     await this.send(value);
+    await this.aiDraftService.setDraft({
+      input: '',
+    });
   };
 
   private readonly _handleModelChange = (modelId: string) => {

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

@@ -206,6 +206,9 @@ export class AIChatMessages extends WithDisposable(ShadowlessElement) {
   @property({ attribute: false })
   accessor docDisplayService!: DocDisplayConfig;
 
+  @property({ attribute: false })
+  accessor onOpenDoc!: (docId: string, sessionId?: string) => void;
+
   @query('.chat-panel-messages-container')
   accessor messagesContainer: HTMLDivElement | null = null;
 
@@ -333,6 +336,7 @@ export class AIChatMessages extends WithDisposable(ShadowlessElement) {
                     .width=${this.width}
                     .independentMode=${this.independentMode}
                     .docDisplayService=${this.docDisplayService}
+                    .onOpenDoc=${this.onOpenDoc}
                   ></chat-message-assistant>`;
                 } else if (isChatAction(item) && this.host) {
                   return html`<chat-message-action