diff --git a/blocksuite/affine/all/src/extensions/migrating-store.ts b/blocksuite/affine/all/src/extensions/migrating-store.ts
index ea5bd28c9..18ea7f126 100644
--- a/blocksuite/affine/all/src/extensions/migrating-store.ts
+++ b/blocksuite/affine/all/src/extensions/migrating-store.ts
@@ -8,10 +8,7 @@ import {
   type StoreExtensionContext,
   StoreExtensionProvider,
 } from '@blocksuite/affine-ext-loader';
-import {
-  RootBlockSchemaExtension,
-  TranscriptionBlockSchemaExtension,
-} from '@blocksuite/affine-model';
+import { RootBlockSchemaExtension } from '@blocksuite/affine-model';
 import {
   HtmlAdapterFactoryExtension,
   ImageProxyService,
@@ -69,7 +66,6 @@ function getNotionHtmlAdapterExtensions(): ExtensionType[] {
 
 const MigratingStoreExtensions: ExtensionType[] = [
   RootBlockSchemaExtension,
-  TranscriptionBlockSchemaExtension,
 
   BlockSelectionExtension,
   TextSelectionExtension,
diff --git a/blocksuite/affine/all/src/schemas.ts b/blocksuite/affine/all/src/schemas.ts
index 6e5d50686..91dc3e33b 100644
--- a/blocksuite/affine/all/src/schemas.ts
+++ b/blocksuite/affine/all/src/schemas.ts
@@ -25,7 +25,6 @@ import {
   RootBlockSchema,
   SurfaceRefBlockSchema,
   TableBlockSchema,
-  TranscriptionBlockSchema,
 } from '@blocksuite/affine-model';
 import type { BlockSchema } from '@blocksuite/store';
 import type { z } from 'zod';
@@ -57,5 +56,4 @@ export const AffineSchemas: z.infer<typeof BlockSchema>[] = [
   LatexBlockSchema,
   TableBlockSchema,
   CalloutBlockSchema,
-  TranscriptionBlockSchema,
 ];
diff --git a/blocksuite/affine/model/src/blocks/index.ts b/blocksuite/affine/model/src/blocks/index.ts
index 269180f3f..0bc0e23af 100644
--- a/blocksuite/affine/model/src/blocks/index.ts
+++ b/blocksuite/affine/model/src/blocks/index.ts
@@ -15,4 +15,3 @@ export * from './paragraph/index.js';
 export * from './root/index.js';
 export * from './surface-ref/index.js';
 export * from './table';
-export * from './transcription';
diff --git a/blocksuite/affine/model/src/blocks/transcription/index.ts b/blocksuite/affine/model/src/blocks/transcription/index.ts
deleted file mode 100644
index 2875594e6..000000000
--- a/blocksuite/affine/model/src/blocks/transcription/index.ts
+++ /dev/null
@@ -1 +0,0 @@
-export * from './transcription-model.js';
diff --git a/packages/frontend/core/src/blocksuite/ai/blocks/ai-chat-block/ai-transcription-block.ts b/packages/frontend/core/src/blocksuite/ai/blocks/ai-chat-block/ai-transcription-block.ts
index ff4897479..d8e7eb5b4 100644
--- a/packages/frontend/core/src/blocksuite/ai/blocks/ai-chat-block/ai-transcription-block.ts
+++ b/packages/frontend/core/src/blocksuite/ai/blocks/ai-chat-block/ai-transcription-block.ts
@@ -1,4 +1,4 @@
-import type { TranscriptionBlockModel } from '@blocksuite/affine/model';
+import type { TranscriptionBlockModel } from '@affine/core/blocksuite/ai/blocks/transcription-block/model';
 import { BlockComponent, BlockViewExtension } from '@blocksuite/affine/std';
 import type { ExtensionType } from '@blocksuite/affine/store';
 import { css, type PropertyValues } from 'lit';
diff --git a/blocksuite/affine/model/src/blocks/transcription/transcription-model.ts b/packages/frontend/core/src/blocksuite/ai/blocks/transcription-block/model.ts
similarity index 96%
rename from blocksuite/affine/model/src/blocks/transcription/transcription-model.ts
rename to packages/frontend/core/src/blocksuite/ai/blocks/transcription-block/model.ts
index 1e19d8971..d91eb8d23 100644
--- a/blocksuite/affine/model/src/blocks/transcription/transcription-model.ts
+++ b/packages/frontend/core/src/blocksuite/ai/blocks/transcription-block/model.ts
@@ -2,7 +2,7 @@ import {
   BlockModel,
   BlockSchemaExtension,
   defineBlockSchema,
-} from '@blocksuite/store';
+} from '@blocksuite/affine/store';
 
 export const TranscriptionBlockFlavour = 'affine:transcription';
 
diff --git a/packages/frontend/core/src/blocksuite/attachment-viewer/audio/transcription-block.tsx b/packages/frontend/core/src/blocksuite/attachment-viewer/audio/transcription-block.tsx
index 41f534611..3e3ed2f1b 100644
--- a/packages/frontend/core/src/blocksuite/attachment-viewer/audio/transcription-block.tsx
+++ b/packages/frontend/core/src/blocksuite/attachment-viewer/audio/transcription-block.tsx
@@ -1,6 +1,6 @@
 import { createReactComponentFromLit } from '@affine/component';
 import { LitTranscriptionBlock } from '@affine/core/blocksuite/ai/blocks/ai-chat-block/ai-transcription-block';
-import type { TranscriptionBlockModel } from '@blocksuite/affine/model';
+import type { TranscriptionBlockModel } from '@affine/core/blocksuite/ai/blocks/transcription-block/model';
 import { LiveData, useLiveData } from '@toeverything/infra';
 import React, { useCallback, useMemo } from 'react';
 
diff --git a/packages/frontend/core/src/blocksuite/manager/migrating-store.ts b/packages/frontend/core/src/blocksuite/manager/migrating-store.ts
index f193b95db..ea515bc25 100644
--- a/packages/frontend/core/src/blocksuite/manager/migrating-store.ts
+++ b/packages/frontend/core/src/blocksuite/manager/migrating-store.ts
@@ -6,6 +6,7 @@ import {
 import { getInternalStoreExtensions } from '@blocksuite/affine/extensions/store';
 
 import { AIChatBlockSchemaExtension } from '../ai/blocks/ai-chat-block/model';
+import { TranscriptionBlockSchemaExtension } from '../ai/blocks/transcription-block/model';
 
 class MigratingAffineStoreExtension extends StoreExtensionProvider {
   override name = 'affine-store-extensions';
@@ -13,6 +14,7 @@ class MigratingAffineStoreExtension extends StoreExtensionProvider {
   override setup(context: StoreExtensionContext) {
     super.setup(context);
     context.register(AIChatBlockSchemaExtension);
+    context.register(TranscriptionBlockSchemaExtension);
   }
 }
 
diff --git a/packages/frontend/core/src/modules/media/entities/audio-attachment-block.ts b/packages/frontend/core/src/modules/media/entities/audio-attachment-block.ts
index 4369cb109..ed2092766 100644
--- a/packages/frontend/core/src/modules/media/entities/audio-attachment-block.ts
+++ b/packages/frontend/core/src/modules/media/entities/audio-attachment-block.ts
@@ -1,13 +1,13 @@
+import {
+  TranscriptionBlockFlavour,
+  type TranscriptionBlockModel,
+} from '@affine/core/blocksuite/ai/blocks/transcription-block/model';
 import { insertFromMarkdown } from '@affine/core/blocksuite/utils';
 import { encodeAudioBlobToOpusSlices } from '@affine/core/utils/webm-encoding';
 import { DebugLogger } from '@affine/debug';
 import { AiJobStatus } from '@affine/graphql';
 import track from '@affine/track';
-import {
-  type AttachmentBlockModel,
-  TranscriptionBlockFlavour,
-  type TranscriptionBlockModel,
-} from '@blocksuite/affine/model';
+import type { AttachmentBlockModel } from '@blocksuite/affine/model';
 import type { AffineTextAttributes } from '@blocksuite/affine/shared/types';
 import { type DeltaInsert, Text } from '@blocksuite/affine/store';
 import { computed } from '@preact/signals-core';
diff --git a/packages/frontend/core/src/modules/media/entities/audio-transcription-job.ts b/packages/frontend/core/src/modules/media/entities/audio-transcription-job.ts
index c73f3b5bc..b119c971e 100644
--- a/packages/frontend/core/src/modules/media/entities/audio-transcription-job.ts
+++ b/packages/frontend/core/src/modules/media/entities/audio-transcription-job.ts
@@ -1,8 +1,8 @@
 import { shallowEqual } from '@affine/component';
+import type { TranscriptionBlockProps } from '@affine/core/blocksuite/ai/blocks/transcription-block/model';
 import { DebugLogger } from '@affine/debug';
 import { UserFriendlyError } from '@affine/error';
 import { AiJobStatus } from '@affine/graphql';
-import type { TranscriptionBlockProps } from '@blocksuite/affine/model';
 import { Entity, LiveData } from '@toeverything/infra';
 
 import type { DefaultServerService, WorkspaceServerService } from '../../cloud';
diff --git a/packages/frontend/core/src/modules/workspace/global-schema.ts b/packages/frontend/core/src/modules/workspace/global-schema.ts
index 24a332a1f..943d5185f 100644
--- a/packages/frontend/core/src/modules/workspace/global-schema.ts
+++ b/packages/frontend/core/src/modules/workspace/global-schema.ts
@@ -1,4 +1,5 @@
 import { AIChatBlockSchema } from '@affine/core/blocksuite/ai/blocks/ai-chat-block/model';
+import { TranscriptionBlockSchema } from '@affine/core/blocksuite/ai/blocks/transcription-block/model';
 import { AffineSchemas } from '@blocksuite/affine/schemas';
 import { Schema } from '@blocksuite/affine/store';
 
@@ -7,7 +8,11 @@ export function getAFFiNEWorkspaceSchema() {
   if (!_schema) {
     _schema = new Schema();
 
-    _schema.register([...AffineSchemas, AIChatBlockSchema]);
+    _schema.register([
+      ...AffineSchemas,
+      AIChatBlockSchema,
+      TranscriptionBlockSchema,
+    ]);
   }
 
   return _schema;