@dxos/plugin-transcription 0.8.3 → 0.8.4-main.c1de068

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dxos/plugin-transcription",
-  "version": "0.8.3",
+  "version": "0.8.4-main.c1de068",
   "description": "Voice transcription.",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
@@ -40,77 +40,77 @@
     "extendable-media-recorder": "^9.2.19",
     "extendable-media-recorder-wav-encoder": "^7.0.123",
     "main-thread-scheduling": "^14.1.1",
-    "react-resize-detector": "^12.0.2",
+    "react-resize-detector": "^11.0.1",
     "react-router-dom": "^6.4.0",
     "react-use": "^17.5.1",
     "wavefile": "^11.0.0",
-    "@dxos/ai": "0.8.3",
-    "@dxos/app-framework": "0.8.3",
-    "@dxos/artifact": "0.8.3",
-    "@dxos/assistant": "0.8.3",
-    "@dxos/async": "0.8.3",
-    "@dxos/client": "0.8.3",
-    "@dxos/codec-protobuf": "0.8.3",
-    "@dxos/context": "0.8.3",
-    "@dxos/debug": "0.8.3",
-    "@dxos/display-name": "0.8.3",
-    "@dxos/echo": "0.8.3",
-    "@dxos/echo-db": "0.8.3",
-    "@dxos/echo-schema": "0.8.3",
-    "@dxos/invariant": "0.8.3",
-    "@dxos/functions": "0.8.3",
-    "@dxos/live-object": "0.8.3",
-    "@dxos/keys": "0.8.3",
-    "@dxos/local-storage": "0.8.3",
-    "@dxos/log": "0.8.3",
-    "@dxos/plugin-client": "0.8.3",
-    "@dxos/plugin-deck": "0.8.3",
-    "@dxos/plugin-graph": "0.8.3",
-    "@dxos/plugin-markdown": "0.8.3",
-    "@dxos/plugin-thread": "0.8.3",
-    "@dxos/protocols": "0.8.3",
-    "@dxos/react-client": "0.8.3",
-    "@dxos/react-edge-client": "0.8.3",
-    "@dxos/plugin-space": "0.8.3",
-    "@dxos/react-ui-attention": "0.8.3",
-    "@dxos/react-ui-editor": "0.8.3",
-    "@dxos/react-ui-components": "0.8.3",
-    "@dxos/react-ui-stack": "0.8.3",
-    "@dxos/react-ui-grid": "0.8.3",
-    "@dxos/schema": "0.8.3",
-    "@dxos/tracing": "0.8.3",
-    "@dxos/react-ui-sfx": "0.8.3",
-    "@dxos/react-ui-form": "0.8.3",
-    "@dxos/util": "0.8.3"
+    "@dxos/ai": "0.8.4-main.c1de068",
+    "@dxos/app-framework": "0.8.4-main.c1de068",
+    "@dxos/assistant": "0.8.4-main.c1de068",
+    "@dxos/blueprints": "0.8.4-main.c1de068",
+    "@dxos/async": "0.8.4-main.c1de068",
+    "@dxos/client": "0.8.4-main.c1de068",
+    "@dxos/codec-protobuf": "0.8.4-main.c1de068",
+    "@dxos/context": "0.8.4-main.c1de068",
+    "@dxos/debug": "0.8.4-main.c1de068",
+    "@dxos/display-name": "0.8.4-main.c1de068",
+    "@dxos/echo": "0.8.4-main.c1de068",
+    "@dxos/echo-db": "0.8.4-main.c1de068",
+    "@dxos/echo-schema": "0.8.4-main.c1de068",
+    "@dxos/functions": "0.8.4-main.c1de068",
+    "@dxos/invariant": "0.8.4-main.c1de068",
+    "@dxos/keys": "0.8.4-main.c1de068",
+    "@dxos/live-object": "0.8.4-main.c1de068",
+    "@dxos/local-storage": "0.8.4-main.c1de068",
+    "@dxos/log": "0.8.4-main.c1de068",
+    "@dxos/plugin-client": "0.8.4-main.c1de068",
+    "@dxos/plugin-deck": "0.8.4-main.c1de068",
+    "@dxos/plugin-graph": "0.8.4-main.c1de068",
+    "@dxos/plugin-markdown": "0.8.4-main.c1de068",
+    "@dxos/plugin-space": "0.8.4-main.c1de068",
+    "@dxos/plugin-thread": "0.8.4-main.c1de068",
+    "@dxos/protocols": "0.8.4-main.c1de068",
+    "@dxos/react-edge-client": "0.8.4-main.c1de068",
+    "@dxos/react-ui-attention": "0.8.4-main.c1de068",
+    "@dxos/react-ui-components": "0.8.4-main.c1de068",
+    "@dxos/react-ui-form": "0.8.4-main.c1de068",
+    "@dxos/react-ui-editor": "0.8.4-main.c1de068",
+    "@dxos/react-ui-sfx": "0.8.4-main.c1de068",
+    "@dxos/react-ui-grid": "0.8.4-main.c1de068",
+    "@dxos/react-client": "0.8.4-main.c1de068",
+    "@dxos/react-ui-stack": "0.8.4-main.c1de068",
+    "@dxos/schema": "0.8.4-main.c1de068",
+    "@dxos/util": "0.8.4-main.c1de068",
+    "@dxos/tracing": "0.8.4-main.c1de068"
   },
   "devDependencies": {
-    "@effect-rx/rx-react": "^0.34.1",
-    "@effect/platform": "0.80.12",
+    "@effect-rx/rx-react": "0.38.0",
+    "@effect/platform": "0.89.0",
     "@types/dom-speech-recognition": "^0.0.6",
     "@types/react": "~18.2.0",
-    "effect": "3.14.21",
+    "effect": "3.17.0",
     "react": "~18.2.0",
     "react-dom": "~18.2.0",
     "tailwindcss": "~3.4.1",
-    "@dxos/plugin-preview": "0.8.3",
-    "@dxos/react-ui": "0.8.3",
-    "@dxos/random": "0.8.3",
-    "@dxos/plugin-storybook-layout": "0.8.3",
-    "@dxos/react-ui-form": "0.8.3",
-    "@dxos/react-ui-syntax-highlighter": "0.8.3",
-    "@dxos/plugin-theme": "0.8.3",
-    "@dxos/react-ui-theme": "0.8.3",
-    "@dxos/storybook-utils": "0.8.3",
-    "@dxos/test-utils": "0.8.3"
+    "@dxos/plugin-preview": "0.8.4-main.c1de068",
+    "@dxos/plugin-theme": "0.8.4-main.c1de068",
+    "@dxos/plugin-storybook-layout": "0.8.4-main.c1de068",
+    "@dxos/random": "0.8.4-main.c1de068",
+    "@dxos/react-ui-form": "0.8.4-main.c1de068",
+    "@dxos/react-ui": "0.8.4-main.c1de068",
+    "@dxos/react-ui-syntax-highlighter": "0.8.4-main.c1de068",
+    "@dxos/react-ui-theme": "0.8.4-main.c1de068",
+    "@dxos/storybook-utils": "0.8.4-main.c1de068",
+    "@dxos/test-utils": "0.8.4-main.c1de068"
   },
-  "optionalDependencies": {
+  "peerDependencies": {
     "@effect-rx/rx-react": "^0.34.1",
-    "@effect/platform": "0.80.12",
+    "@effect/platform": "^0.80.12",
     "effect": "^3.13.3",
     "react": "~18.2.0",
     "react-dom": "~18.2.0",
-    "@dxos/react-ui-theme": "0.8.3",
-    "@dxos/react-ui": "0.8.3"
+    "@dxos/react-ui": "0.8.4-main.c1de068",
+    "@dxos/react-ui-theme": "0.8.4-main.c1de068"
   },
   "publishConfig": {
     "access": "public"

package/src/TranscriptionPlugin.tsx

@@ -11,7 +11,7 @@ import { DataType } from '@dxos/schema';
 import { IntentResolver, ReactSurface, Transcriber } from './capabilities';
 import { renderMarkdown } from './components';
 import { meta } from './meta';
-import translations from './translations';
+import { translations } from './translations';
 import { TranscriptType } from './types';
 
 export const TranscriptionPlugin = () =>

package/src/capabilities/capabilities.ts

@@ -16,9 +16,9 @@ import {
 export namespace TranscriptionCapabilities {
   export type GetTranscriberProps = {
     audioStreamTrack: MediaStreamTrack;
-    onSegments: TranscriberParams['onSegments'];
-    transcriberConfig?: Partial<TranscriberParams['config']>;
     recorderConfig?: Partial<MediaStreamRecorderParams['config']>;
+    transcriberConfig?: Partial<TranscriberParams['config']>;
+    onSegments: TranscriberParams['onSegments'];
   };
   export type GetTranscriber = (props: GetTranscriberProps) => Transcriber;
   export const Transcriber = defineCapability<GetTranscriber>(`${TRANSCRIPTION_PLUGIN}/capability/transcriber`);

package/src/components/Transcript/Transcript.stories.tsx

@@ -4,7 +4,7 @@
 
 import '@dxos-theme';
 
-import { type Meta, type StoryObj } from '@storybook/react';
+import { type Meta, type StoryObj } from '@storybook/react-vite';
 import React, { type FC, useEffect, useMemo, useState } from 'react';
 
 import { IntentPlugin, SettingsPlugin } from '@dxos/app-framework';
@@ -33,7 +33,7 @@ import {
   useTestTranscriptionQueue,
   useTestTranscriptionQueueWithEntityExtraction,
 } from '../../testing';
-import translations from '../../translations';
+import { translations } from '../../translations';
 
 faker.seed(1);
 

package/src/components/Transcript/Transcript.tsx

@@ -30,7 +30,7 @@ export const renderMarkdown =
   (message: DataType.Message, index: number, debug = false): string[] => {
     if (message.sender.role === 'assistant') {
       // Start/stop block.
-      return [message.blocks.find((block) => block.type === 'transcription')?.text ?? '', ''];
+      return [message.blocks.find((block) => block._tag === 'transcript')?.text ?? '', ''];
     }
 
     // TODO(burdon): Use link/reference markup for users (with popover).
@@ -42,7 +42,7 @@ export const renderMarkdown =
       message.sender.name ??
       message.sender.email ??
       message.sender.identityDid;
-    const blocks = message.blocks.filter((block) => block.type === 'transcription');
+    const blocks = message.blocks.filter((block) => block._tag === 'transcript');
     return [
       `###### ${name}` + (debug ? ` [${index}]:${message.id}` : ''),
       blocks.map((block) => block.text.trim()).join(' '),

package/src/components/Transcript/transcript-extension.ts

@@ -38,7 +38,7 @@ export const transcript = ({ model, started, renderButton }: TranscriptOptions):
           let line = view.state.doc.lineAt(from);
           while (line.from <= to) {
             const block = model.getChunkAtLine(line.number)?.blocks[0];
-            const timestamp = block?.type === 'transcription' && block.started;
+            const timestamp = block?._tag === 'transcript' && block.started;
             if (timestamp) {
               builder.add(line.from, line.from, new TimestampMarker(line.number, new Date(timestamp), start));
             }
@@ -218,7 +218,7 @@ const getStartTime = (started?: Date, message?: DataType.Message): Date | undefi
     return started;
   }
 
-  if (message?.blocks[0]?.type === 'transcription' && message.blocks[0].started) {
+  if (message?.blocks[0]?._tag === 'transcript' && message.blocks[0].started) {
     return new Date(message.blocks[0].started);
   }
 

package/src/components/TranscriptContainer.tsx

@@ -2,7 +2,7 @@
 // Copyright 2024 DXOS.org
 //
 
-import React, { type FC } from 'react';
+import React from 'react';
 
 import { fullyQualifiedId, getSpace } from '@dxos/client/echo';
 import { useMembers, useQueue } from '@dxos/react-client/echo';
@@ -13,7 +13,12 @@ import { Transcript, renderMarkdown } from './Transcript';
 import { useQueueModelAdapter } from '../hooks';
 import { type TranscriptType } from '../types';
 
-export const TranscriptionContainer: FC<{ role: string; transcript: TranscriptType }> = ({ role, transcript }) => {
+export type TranscriptionContainerProps = {
+  role: string;
+  transcript: TranscriptType;
+};
+
+export const TranscriptionContainer = ({ transcript }: TranscriptionContainerProps) => {
   const attendableId = fullyQualifiedId(transcript);
   const space = getSpace(transcript);
   const members = useMembers(space?.key).map((member) => member.identity);
@@ -21,7 +26,7 @@ export const TranscriptionContainer: FC<{ role: string; transcript: TranscriptTy
   const model = useQueueModelAdapter(renderMarkdown(members), queue);
 
   return (
-    <StackItem.Content role={role} classNames='container-max-width pli-2'>
+    <StackItem.Content classNames='container-max-width pli-2'>
       <Transcript attendableId={attendableId} space={space} model={model} transcript={transcript} />
     </StackItem.Content>
   );

package/src/components/stories/FileTranscription.stories.tsx

@@ -4,11 +4,9 @@
 
 import '@dxos-theme';
 
-import { type Meta, type StoryObj } from '@storybook/react';
+import { type Meta, type StoryObj } from '@storybook/react-vite';
 import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react';
 
-import { EdgeAiServiceClient } from '@dxos/ai';
-import { AI_SERVICE_ENDPOINT } from '@dxos/ai/testing';
 import { Events, IntentPlugin, SettingsPlugin } from '@dxos/app-framework';
 import { withPluginManager } from '@dxos/app-framework/testing';
 import { scheduleTask } from '@dxos/async';
@@ -88,8 +86,13 @@ const AudioFile = ({
   const model = useQueueModelAdapter(renderMarkdown([]), queue);
   const handleSegments = useCallback<TranscriberParams['onSegments']>(
     async (blocks) => {
-      const message = Obj.make(DataType.Message, { sender: actor, created: new Date().toISOString(), blocks });
-      void queue?.append([message]);
+      void queue?.append([
+        Obj.make(DataType.Message, {
+          created: new Date().toISOString(),
+          sender: actor,
+          blocks,
+        }),
+      ]);
     },
     [queue],
   );
@@ -108,14 +111,14 @@ const AudioFile = ({
     }
     const executor = new FunctionExecutor(
       new ServiceContainer().setServices({
-        ai: {
-          client: new EdgeAiServiceClient({
-            endpoint: AI_SERVICE_ENDPOINT.REMOTE,
-            defaultGenerationOptions: {
-              model: '@anthropic/claude-3-5-sonnet-20241022',
-            },
-          }),
-        },
+        // ai: {
+        //   client: new Edge AiServiceClient({
+        //     endpoint: AI_SERVICE_ENDPOINT.REMOTE,
+        //     defaultGenerationOptions: {
+        //       model: '@anthropic/claude-3-5-sonnet-20241022',
+        //     },
+        //   }),
+        // },
       }),
     );
 

package/src/components/stories/KeyWordRecognition.stories.tsx

@@ -4,7 +4,7 @@
 
 import '@dxos-theme';
 
-import { type StoryObj, type Meta } from '@storybook/react';
+import { type StoryObj, type Meta } from '@storybook/react-vite';
 import React, { useEffect, useMemo, useState } from 'react';
 
 import { log } from '@dxos/log';

package/src/components/stories/MicrophoneTranscription.stories.tsx

@@ -4,19 +4,17 @@
 
 import '@dxos-theme';
 
-import { type Meta, type StoryObj } from '@storybook/react';
+import { type Meta, type StoryObj } from '@storybook/react-vite';
 import React, { useCallback, useEffect, useMemo, useState } from 'react';
 
-import { EdgeAiServiceClient } from '@dxos/ai';
-import { AI_SERVICE_ENDPOINT } from '@dxos/ai/testing';
 import { Events, IntentPlugin, SettingsPlugin } from '@dxos/app-framework';
 import { withPluginManager } from '@dxos/app-framework/testing';
 import {
-  extractionAnthropicFn,
   type ExtractionFunction,
+  extractionAnthropicFn,
   extractionNerFn,
-  processTranscriptMessage,
   getNer,
+  processTranscriptMessage,
 } from '@dxos/assistant';
 import { Filter, Obj, type Type } from '@dxos/echo';
 import { MemoryQueue } from '@dxos/echo-db';
@@ -117,11 +115,15 @@ const DefaultStory = ({
         .then((result) => result.objects);
     }
     if (entityExtraction !== 'none') {
-      const AiService = new EdgeAiServiceClient({
-        endpoint: AI_SERVICE_ENDPOINT.REMOTE,
-      });
       executor = new FunctionExecutor(
-        new ServiceContainer().setServices({ ai: { client: AiService }, database: { db: space!.db } }),
+        new ServiceContainer().setServices({
+          // ai: {
+          //   client: new Edge AiServiceClient({
+          //     endpoint: AI_SERVICE_ENDPOINT.REMOTE,
+          //   }),
+          // },
+          // database: { db: space!.db },
+        }),
       );
     }
 

package/src/hooks/useTranscriber.ts

@@ -14,20 +14,20 @@ import { type Transcriber } from '../transcriber';
  */
 export const useTranscriber = ({
   audioStreamTrack,
-  onSegments,
-  transcriberConfig,
   recorderConfig,
+  transcriberConfig,
+  onSegments,
 }: Partial<TranscriptionCapabilities.GetTranscriberProps>) => {
   const [getTranscriber] = useCapabilities(TranscriptionCapabilities.Transcriber);
 
   // Initialize audio transcription.
   const transcriber = useMemo<Transcriber | undefined>(() => {
-    if (!onSegments || !audioStreamTrack || !getTranscriber) {
+    if (!getTranscriber || !audioStreamTrack || !onSegments) {
       return undefined;
     }
 
-    return getTranscriber({ audioStreamTrack, onSegments, transcriberConfig, recorderConfig });
-  }, [audioStreamTrack, onSegments, getTranscriber, transcriberConfig, recorderConfig]);
+    return getTranscriber({ audioStreamTrack, recorderConfig, transcriberConfig, onSegments });
+  }, [getTranscriber, audioStreamTrack, recorderConfig, transcriberConfig, onSegments]);
 
   useEffect(() => {
     return () => {

package/src/model/model.test.ts

@@ -18,7 +18,7 @@ const blockToMarkdown: ChunkRenderer<DataType.Message> = (
 ): string[] => {
   return [
     `###### ${message.sender.name}`,
-    ...message.blocks.filter((block) => block.type === 'transcription').map((block) => block.text),
+    ...message.blocks.filter((block) => block._tag === 'transcript').map((block) => block.text),
     '',
   ];
 };
@@ -37,7 +37,7 @@ describe('SerializationModel', () => {
       sender: { name: 'Alice' },
       blocks: [
         {
-          type: 'transcription',
+          _tag: 'transcript',
           started: createDate(),
           text: 'Hello world!',
         },
@@ -51,7 +51,7 @@ describe('SerializationModel', () => {
     }
 
     // Update message.
-    message.blocks.push({ type: 'transcription', started: createDate(), text: 'Hello again!' });
+    message.blocks.push({ _tag: 'transcript', started: createDate(), text: 'Hello again!' });
     model.updateChunk(message);
     {
       const text = model.doc.toString();
@@ -74,7 +74,7 @@ describe('SerializationModel', () => {
         sender: { name: 'Alice' },
         blocks: [
           {
-            type: 'transcription',
+            _tag: 'transcript',
             started: createDate(),
             text: 'Hello world!',
           },
@@ -92,7 +92,7 @@ describe('SerializationModel', () => {
         sender: { name: 'Bob' },
         blocks: [
           {
-            type: 'transcription',
+            _tag: 'transcript',
             started: createDate(),
             text: 'Hello world!',
           },
@@ -119,7 +119,7 @@ describe('SerializationModel', () => {
         sender: { name: 'Alice' },
         blocks: [
           {
-            type: 'transcription',
+            _tag: 'transcript',
             started: createDate(),
             text: 'Hello world!',
           },
@@ -131,7 +131,7 @@ describe('SerializationModel', () => {
       expect(view.state.doc.toString()).to.eq('###### Alice\nHello world!\n\n');
 
       // Update message (add block).
-      message.blocks.push({ type: 'transcription', started: createDate(), text: 'Hello again!' });
+      message.blocks.push({ _tag: 'transcript', started: createDate(), text: 'Hello again!' });
       model.updateChunk(message);
       model.sync(adapter);
       expect(view.state.doc.toString()).to.eq('###### Alice\nHello world!\nHello again!\n\n');
@@ -150,7 +150,7 @@ describe('SerializationModel', () => {
       const message = Obj.make(DataType.Message, {
         created: createDate(),
         sender: { name: 'Bob' },
-        blocks: [{ type: 'transcription', started: createDate(), text: 'Hello again!' }],
+        blocks: [{ _tag: 'transcript', started: createDate(), text: 'Hello again!' }],
       });
       model.appendChunk(message);
       model.sync(adapter);

package/src/segments-normalization/normalization.test.ts

@@ -5,8 +5,6 @@
 import { effect } from '@preact/signals-core';
 import { describe, test } from 'vitest';
 
-import { EdgeAiServiceClient, OllamaAiServiceClient } from '@dxos/ai';
-import { AI_SERVICE_ENDPOINT } from '@dxos/ai/testing';
 import { scheduleTaskInterval } from '@dxos/async';
 import { Context } from '@dxos/context';
 import { Obj } from '@dxos/echo';
@@ -53,31 +51,31 @@ const messages: MessageWithRangeId[] = [
   Obj.make(DataType.Message, {
     created: new Date(Date.now() + 1000 * index).toISOString(),
     sender,
-    blocks: [{ type: 'transcription', started: new Date(Date.now() + 1000 * index).toISOString(), text: string }],
-    rangeId: [],
-  } as any),
+    blocks: [{ _tag: 'transcript', started: new Date(Date.now() + 1000 * index).toISOString(), text: string }],
+  }),
 );
 
-const REMOTE_AI = true;
+// eslint-disable-next-line no-unused-vars
+const _REMOTE_AI = true;
 
 describe.skip('SentenceNormalization', () => {
   const getExecutor = () => {
     return new FunctionExecutor(
       new ServiceContainer().setServices({
-        ai: {
-          client: REMOTE_AI
-            ? new EdgeAiServiceClient({
-                endpoint: AI_SERVICE_ENDPOINT.REMOTE,
-                defaultGenerationOptions: {
-                  model: '@anthropic/claude-3-5-sonnet-20241022',
-                },
-              })
-            : new OllamaAiServiceClient({
-                overrides: {
-                  model: 'llama3.1:8b',
-                },
-              }),
-        },
+        // ai: {
+        //   client: REMOTE_AI
+        //     ? new Edge AiServiceClient({
+        //         endpoint: AI_SERVICE_ENDPOINT.REMOTE,
+        //         defaultGenerationOptions: {
+        //           model: '@anthropic/claude-3-5-sonnet-20241022',
+        //         },
+        //       })
+        //     : new Ollama AiServiceClient({
+        //         overrides: {
+        //           model: 'llama3.1:8b',
+        //         },
+        //       }),
+        // },
       }),
     );
   };

package/src/segments-normalization/normalization.ts

@@ -2,12 +2,15 @@
 // Copyright 2025 DXOS.org
 //
 
+// TODO(burdon): Needs refactoring.
+// @ts-nocheck
+
 import { Schema } from 'effect';
 
-import { DEFAULT_EDGE_MODEL, Message } from '@dxos/ai';
-import { AISession } from '@dxos/assistant';
+import { DEFAULT_EDGE_MODEL, AiService } from '@dxos/ai';
+import { AiSession } from '@dxos/assistant';
 import { Obj } from '@dxos/echo';
-import { AiService, defineFunction } from '@dxos/functions';
+import { defineFunction } from '@dxos/functions';
 import { ObjectId } from '@dxos/keys';
 import { log } from '@dxos/log';
 import { DataType } from '@dxos/schema';
@@ -80,21 +83,27 @@ export const sentenceNormalization = defineFunction<NormalizationInput, Normaliz
   handler: async ({ data: { messages }, context }) => {
     log.info('input', { messages });
     const ai = context.getService(AiService);
-    const session = new AISession({ operationModel: 'configured' });
+    const session = new AiSession({ operationModel: 'configured' });
 
-    const response = await session.runStructured(NormalizationOutput, {
-      generationOptions: { model: DEFAULT_EDGE_MODEL },
+    // TODO(dmaretskyi): This got broken after effect-ai transition.
+    const response = session.runStructured(NormalizationOutput, {
+      generationOptions: {
+        model: DEFAULT_EDGE_MODEL,
+      },
       client: ai.client,
       tools: [],
       artifacts: [],
       history: [
-        Obj.make(Message, {
-          role: 'user',
-          content: messages.map((message) => ({ type: 'text', text: JSON.stringify(message) }) as const),
+        Obj.make(DataType.Message, {
+          created: new Date().toISOString(),
+          sender: {
+            role: 'user',
+          },
+          blocks: messages.map((message) => ({ _tag: 'text', text: JSON.stringify(message) }) as const),
         }),
       ],
       prompt,
-    });
+    }) as any;
 
     response.sentences.forEach((sentence) => {
       sentence.id = ObjectId.random();

package/src/testing/testing.ts

@@ -5,13 +5,11 @@
 import { Schema } from 'effect';
 import { useEffect, useMemo, useState } from 'react';
 
-import { EdgeAiServiceClient } from '@dxos/ai';
-import { AI_SERVICE_ENDPOINT } from '@dxos/ai/testing';
 import { extractionAnthropicFn, processTranscriptMessage } from '@dxos/assistant';
 import { scheduleTaskInterval } from '@dxos/async';
 import { Filter, type Queue } from '@dxos/client/echo';
 import { Context } from '@dxos/context';
-import { type Key, Obj, Ref, Type } from '@dxos/echo';
+import { Obj, Ref, Type, type Key } from '@dxos/echo';
 import { createQueueDXN } from '@dxos/echo-schema';
 import { FunctionExecutor, ServiceContainer } from '@dxos/functions';
 import { IdentityDid } from '@dxos/keys';
@@ -69,7 +67,7 @@ export class MessageBuilder extends AbstractMessageBuilder {
     });
   }
 
-  createBlock(): DataType.MessageBlock.Transcription {
+  createBlock(): DataType.MessageBlock.Transcript {
     let text = faker.lorem.paragraph();
     if (this._space) {
       const label = faker.commerce.productName();
@@ -81,7 +79,7 @@ export class MessageBuilder extends AbstractMessageBuilder {
     }
 
     return {
-      type: 'transcription',
+      _tag: 'transcript',
       started: this.next().toISOString(),
       text,
     };
@@ -95,11 +93,13 @@ export class MessageBuilder extends AbstractMessageBuilder {
 
 // TODO(burdon): Reconcile with BlockBuilder.
 class EntityExtractionMessageBuilder extends AbstractMessageBuilder {
-  AiService = new EdgeAiServiceClient({
-    endpoint: AI_SERVICE_ENDPOINT.REMOTE,
-  });
-
-  executor = new FunctionExecutor(new ServiceContainer().setServices({ ai: { client: this.AiService } }));
+  executor = new FunctionExecutor(
+    new ServiceContainer().setServices({
+      // ai: {
+      //   client: this.AiService,
+      // },
+    }),
+  );
 
   space: Space | undefined;
   currentMessage: number = 0;

package/src/transcriber/index.ts

@@ -2,7 +2,7 @@
 // Copyright 2025 DXOS.org
 //
 
-export * from './audio-recorder';
+export type * from './audio-recorder';
 export * from './media-stream-recorder';
 export * from './transcriber';
 export * from './transcription-manager';