@dxos/assistant 0.8.4-main.f9ba587 → 0.8.4-main.fd6878d

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dxos/assistant",
-  "version": "0.8.4-main.f9ba587",
+  "version": "0.8.4-main.fd6878d",
   "description": "Natural language interface to DXOS",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
@@ -10,43 +10,67 @@
   "type": "module",
   "exports": {
     ".": {
+      "source": "./src/index.ts",
       "types": "./dist/types/src/index.d.ts",
       "browser": "./dist/lib/browser/index.mjs",
       "node": "./dist/lib/node-esm/index.mjs"
+    },
+    "./extraction": {
+      "source": "./src/extraction/index.ts",
+      "types": "./dist/types/src/extraction/index.d.ts",
+      "browser": "./dist/lib/browser/extraction/index.mjs",
+      "node": "./dist/lib/node-esm/extraction/index.mjs"
     }
   },
   "types": "dist/types/src/index.d.ts",
+  "typesVersions": {
+    "*": {
+      "types": [
+        "dist/types/src/types/index.d.ts"
+      ],
+      "extraction": [
+        "dist/types/src/extraction/index.d.ts"
+      ]
+    }
+  },
   "files": [
     "dist",
     "src"
   ],
   "dependencies": {
+    "@effect/ai": "0.25.2",
+    "@effect/ai-anthropic": "^0.15.1",
+    "@effect/platform": "^0.90.2",
+    "@effect/platform-node": "^0.95.0",
     "@preact/signals-core": "^1.9.0",
     "@xenova/transformers": "^2.17.2",
     "chalk": "^4.1.0",
-    "exa-js": "^1.5.12",
-    "@dxos/ai": "0.8.4-main.f9ba587",
-    "@dxos/artifact": "0.8.4-main.f9ba587",
-    "@dxos/async": "0.8.4-main.f9ba587",
-    "@dxos/context": "0.8.4-main.f9ba587",
-    "@dxos/debug": "0.8.4-main.f9ba587",
-    "@dxos/echo-db": "0.8.4-main.f9ba587",
-    "@dxos/echo-protocol": "0.8.4-main.f9ba587",
-    "@dxos/echo-schema": "0.8.4-main.f9ba587",
-    "@dxos/effect": "0.8.4-main.f9ba587",
-    "@dxos/graph": "0.8.4-main.f9ba587",
-    "@dxos/functions": "0.8.4-main.f9ba587",
-    "@dxos/echo": "0.8.4-main.f9ba587",
-    "@dxos/invariant": "0.8.4-main.f9ba587",
-    "@dxos/live-object": "0.8.4-main.f9ba587",
-    "@dxos/keys": "0.8.4-main.f9ba587",
-    "@dxos/log": "0.8.4-main.f9ba587",
-    "@dxos/schema": "0.8.4-main.f9ba587",
-    "@dxos/node-std": "0.8.4-main.f9ba587",
-    "@dxos/util": "0.8.4-main.f9ba587"
+    "handlebars": "4.7.8",
+    "lodash.defaultsdeep": "^4.6.1",
+    "@dxos/async": "0.8.4-main.fd6878d",
+    "@dxos/context": "0.8.4-main.fd6878d",
+    "@dxos/blueprints": "0.8.4-main.fd6878d",
+    "@dxos/debug": "0.8.4-main.fd6878d",
+    "@dxos/echo-db": "0.8.4-main.fd6878d",
+    "@dxos/echo": "0.8.4-main.fd6878d",
+    "@dxos/echo-protocol": "0.8.4-main.fd6878d",
+    "@dxos/echo-schema": "0.8.4-main.fd6878d",
+    "@dxos/errors": "0.8.4-main.fd6878d",
+    "@dxos/functions": "0.8.4-main.fd6878d",
+    "@dxos/effect": "0.8.4-main.fd6878d",
+    "@dxos/invariant": "0.8.4-main.fd6878d",
+    "@dxos/graph": "0.8.4-main.fd6878d",
+    "@dxos/ai": "0.8.4-main.fd6878d",
+    "@dxos/keys": "0.8.4-main.fd6878d",
+    "@dxos/log": "0.8.4-main.fd6878d",
+    "@dxos/node-std": "0.8.4-main.fd6878d",
+    "@dxos/schema": "0.8.4-main.fd6878d",
+    "@dxos/util": "0.8.4-main.fd6878d",
+    "@dxos/live-object": "0.8.4-main.fd6878d"
   },
   "devDependencies": {
-    "effect": "3.16.13",
+    "@types/lodash.defaultsdeep": "^4.6.6",
+    "effect": "3.17.7",
     "inquirer": "^12.0.0"
   },
   "peerDependencies": {

package/src/{context/binder.ts → conversation/context.ts}

@@ -2,32 +2,74 @@
 // Copyright 2025 DXOS.org
 //
 
-import { computed, type ReadonlySignal } from '@preact/signals-core';
+import { type ReadonlySignal, computed } from '@preact/signals-core';
+import { Schema } from 'effect';
 import { Array, pipe } from 'effect';
 
-import { Obj, type Type, type Ref, type Relation } from '@dxos/echo';
+import { Blueprint } from '@dxos/blueprints';
+import { Obj, type Ref, type Relation, Type } from '@dxos/echo';
 import { type Queue } from '@dxos/echo-db';
 import { ComplexSet } from '@dxos/util';
 
-import { ContextBinding } from './binding';
-import { type Blueprint } from '../blueprint';
+/**
+ * Thread message that binds or unbinds contextual objects to a conversation.
+ */
+// TODO(burdon): Move to @dxos/schema ContentBlock.
+export const ContextBinding = Schema.Struct({
+  blueprints: Schema.Struct({
+    added: Schema.Array(Type.Ref(Blueprint.Blueprint)),
+    removed: Schema.Array(Type.Ref(Blueprint.Blueprint)),
+  }),
+
+  // TODO(burdon): Type.Expando => Type.Obj (or Obj.Any?)
+  objects: Schema.Struct({
+    added: Schema.Array(Type.Ref(Type.Expando)),
+    removed: Schema.Array(Type.Ref(Type.Expando)),
+  }),
+}).pipe(
+  Type.Obj({
+    typename: 'dxos.org/type/ContextBinding',
+    version: '0.1.0',
+  }),
+);
+
+export interface ContextBinding extends Schema.Schema.Type<typeof ContextBinding> {}
+
+export type BindingProps = Partial<{
+  blueprints: Ref.Ref<Blueprint.Blueprint>[];
+  objects: Ref.Ref<Type.Expando>[];
+}>;
+
+export class Bindings {
+  readonly blueprints = new ComplexSet<Ref.Ref<Blueprint.Blueprint>>((ref) => ref.dxn.toString());
+  readonly objects = new ComplexSet<Ref.Ref<Type.Expando>>((ref) => ref.dxn.toString());
+
+  toJSON() {
+    return {
+      blueprints: Array.fromIterable(this.blueprints).map((ref) => ref.dxn.toString()),
+      objects: Array.fromIterable(this.objects).map((ref) => ref.dxn.toString()),
+    };
+  }
+}
 
 /**
  * Manages bindings of blueprints and objects to a conversation.
  */
-export class ContextBinder {
-  constructor(private readonly _queue: Queue) {}
-
+export class AiContextBinder {
   /**
    * Reactive query of all bindings.
    */
   // TODO(burdon): Cache value?
   readonly bindings: ReadonlySignal<Bindings> = computed(() => this._reduce(this._queue.objects));
 
-  // TODO(burdon): load refs?
-  readonly blueprints: ReadonlySignal<Ref.Ref<Blueprint>[]> = computed(() => [...this.bindings.value.blueprints]);
+  readonly blueprints: ReadonlySignal<Ref.Ref<Blueprint.Blueprint>[]> = computed(() => [
+    ...this.bindings.value.blueprints,
+  ]);
+
   readonly objects: ReadonlySignal<Ref.Ref<Type.Expando>[]> = computed(() => [...this.bindings.value.objects]);
 
+  constructor(private readonly _queue: Queue) {}
+
   /**
    * Asynchronous query of all bindings.
    */
@@ -89,13 +131,3 @@ export class ContextBinder {
     );
   }
 }
-
-export type BindingProps = Partial<{
-  blueprints: Ref.Ref<Blueprint>[];
-  objects: Ref.Ref<Type.Expando>[];
-}>;
-
-export class Bindings {
-  blueprints = new ComplexSet<Ref.Ref<Blueprint>>((ref) => ref.dxn.toString());
-  objects = new ComplexSet<Ref.Ref<Type.Expando>>((ref) => ref.dxn.toString());
-}

package/src/conversation/conversation.ts

@@ -2,37 +2,41 @@
 // Copyright 2025 DXOS.org
 //
 
-import { Message, type ExecutableTool } from '@dxos/ai';
-import { type ArtifactDefinition } from '@dxos/artifact';
+import { type AiError, type AiLanguageModel, type AiTool, type AiToolkit } from '@effect/ai';
+import { Effect } from 'effect';
+
+import {
+  type AiInputPreprocessingError,
+  type AiToolNotFoundError,
+  type ToolExecutionService,
+  type ToolResolverService,
+} from '@dxos/ai';
 import { Event } from '@dxos/async';
-import { Obj, Ref } from '@dxos/echo';
+import { Obj } from '@dxos/echo';
 import { type Queue } from '@dxos/echo-db';
-import { AiService, ToolResolverService, type ServiceContainer } from '@dxos/functions';
+import { DatabaseService } from '@dxos/functions';
 import { log } from '@dxos/log';
-import { isNonNullable } from '@dxos/util';
+import { DataType } from '@dxos/schema';
 
-import { ContextBinder, type ContextBinding } from '../context';
-import { AISession, type SessionRunOptions } from '../session';
+import { type AiAssistantError } from '../errors';
+import { AiSession, type GenerationObserver } from '../session';
 
-export interface ConversationRunOptions {
-  systemPrompt?: string;
-  prompt: string;
-
-  tools?: ExecutableTool[];
-  artifacts?: ArtifactDefinition[];
-  extensions?: ToolContextExtensions;
-  generationOptions?: SessionRunOptions['generationOptions'];
+import { AiContextBinder, type ContextBinding } from './context';
 
-  /** @depreacated Should be managed by the conversation. */
-  requiredArtifactIds?: string[];
+export interface AiConversationRunParams<Tools extends AiTool.Any> {
+  prompt: string;
+  system?: string;
+  toolkit?: AiToolkit.AiToolkit<Tools>;
+  observer?: GenerationObserver;
 
-  // TODO(dmaretskyi): Move into conversation.
-  artifactDiffResolver?: SessionRunOptions['artifactDiffResolver'];
+  /**
+   * @deprecated Remove
+   */
+  session?: AiSession;
 }
 
-export type ConversationOptions = {
-  serviceContainer: ServiceContainer;
-  queue: Queue<Message | ContextBinding>;
+export type AiConversationOptions = {
+  queue: Queue<DataType.Message | ContextBinding>;
 };
 
 /**
@@ -40,64 +44,82 @@ export type ConversationOptions = {
  * Context + history + artifacts.
  * Backed by a Queue.
  */
-export class Conversation {
-  private readonly _serviceContainer: ServiceContainer;
-  private readonly _queue: Queue<Message | ContextBinding>;
+export class AiConversation {
+  private readonly _queue: Queue<DataType.Message | ContextBinding>;
 
   /**
-   * Fired when the execution loop begins.
-   * This is called before the first message is sent.
+   * Blueprints bound to the conversation.
    */
-  public readonly onBegin = new Event<AISession>();
+  public readonly _context: AiContextBinder;
 
   /**
-   * Blueprints bound to the conversation.
+   * Fired when the execution loop begins.
+   * This is called before the first message is sent.
+   *
+   * @deprecated Pass in a session instead.
    */
-  public readonly context: ContextBinder;
+  // TODO(burdon): Is this still deprecated?
+  public readonly onBegin = new Event<AiSession>();
 
-  constructor(options: ConversationOptions) {
-    this._serviceContainer = options.serviceContainer;
+  constructor(options: AiConversationOptions) {
     this._queue = options.queue;
-    this.context = new ContextBinder(this._queue);
+    this._context = new AiContextBinder(this._queue);
   }
 
-  async run(options: ConversationRunOptions): Promise<Message[]> {
-    const session = new AISession({ operationModel: 'configured' });
-    this.onBegin.emit(session);
-
-    const history = await this.getHistory();
-    const context = await this.context.query();
-    const blueprints = (await Ref.Array.loadAll([...context.blueprints])).filter(isNonNullable);
-    if (blueprints.length > 1) {
-      log.warn('multiple blueprints are not yet supported');
-    }
-
-    const messages = await session.run({
-      history,
-      prompt: options.prompt,
-      systemPrompt: (options.systemPrompt ?? '') + (blueprints.at(0)?.instructions ?? ''),
-
-      // TODO(dmaretskyi): Artifacts come from the blueprint?
-      artifacts: options.artifacts ?? [],
-      tools: blueprints.at(0)?.tools ?? [],
-      executableTools: options.tools,
-      requiredArtifactIds: options.requiredArtifactIds,
-      client: this._serviceContainer.getService(AiService).client,
-      toolResolver: this._serviceContainer.getService(ToolResolverService).toolResolver,
-      extensions: {
-        serviceContainer: this._serviceContainer,
-        ...options.extensions,
-      },
-      generationOptions: options.generationOptions,
-    });
-
-    await this._queue.append(messages);
-
-    return messages;
+  get context() {
+    return this._context;
   }
 
-  async getHistory(): Promise<Message[]> {
+  async getHistory(): Promise<DataType.Message[]> {
     const queueItems = await this._queue.queryObjects();
-    return queueItems.filter(Obj.instanceOf(Message));
+    return queueItems.filter(Obj.instanceOf(DataType.Message));
   }
+
+  /**
+   * Executes a prompt.
+   * Each invocation creates a new `AiSession`, which handles potential tool calls.
+   */
+  run = <Tools extends AiTool.Any>({
+    // TODO(burdon): Decide whether to pass in or to fully encapsulate.
+    session = new AiSession(),
+    ...params
+  }: AiConversationRunParams<Tools>): Effect.Effect<
+    DataType.Message[],
+    AiAssistantError | AiInputPreprocessingError | AiError.AiError | AiToolNotFoundError,
+    AiLanguageModel.AiLanguageModel | ToolResolverService | ToolExecutionService | AiTool.ToHandler<Tools>
+  > =>
+    Effect.gen(this, function* () {
+      const history = yield* Effect.promise(() => this.getHistory());
+      const context = yield* Effect.promise(() => this.context.query());
+      const blueprints = yield* Effect.forEach(context.blueprints.values(), DatabaseService.load);
+
+      // TODO(burdon): These don't need to be loaded; just need id and typename from context.
+      const objects = yield* Effect.forEach(context.objects.values(), DatabaseService.load);
+
+      log.info('run', {
+        prompt: params.prompt,
+        system: params.system,
+        history: history.length,
+        objects: objects.length,
+        blueprints: blueprints.length,
+        toolkit: params.toolkit,
+      });
+
+      const start = Date.now();
+      this.onBegin.emit(session);
+
+      const messages = yield* session.run({
+        prompt: params.prompt,
+        system: params.system,
+        history,
+        objects,
+        blueprints,
+        toolkit: params.toolkit,
+        observer: params.observer,
+      });
+
+      log.info('result', { messages: messages, duration: Date.now() - start });
+      yield* Effect.promise(() => this._queue.append(messages));
+      return messages;
+    }).pipe(Effect.withSpan('AiConversation.run'));
 }

package/src/conversation/index.ts

@@ -2,4 +2,5 @@
 // Copyright 2025 DXOS.org
 //
 
+export * from './context';
 export * from './conversation';

package/src/errors.ts

@@ -0,0 +1,10 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { BaseError } from '@dxos/errors';
+
+/**
+ * Generic error for AI agent execution.
+ */
+export class AiAssistantError extends BaseError.extend('AI_ASSISTANT_ERROR') {}

package/src/extraction/extraction-llm-function/extraction-llm-function.test.ts

@@ -2,35 +2,24 @@
 // Copyright 2025 DXOS.org
 //
 
-import { describe, test, beforeAll } from 'vitest';
+import { beforeAll, describe, test } from 'vitest';
 
-import { EdgeAiServiceClient } from '@dxos/ai';
-import { AI_SERVICE_ENDPOINT } from '@dxos/ai/testing';
-import { FunctionExecutor, ServiceContainer } from '@dxos/functions';
+import { todo } from '@dxos/debug';
+import { FunctionExecutor } from '@dxos/functions';
 import { log } from '@dxos/log';
 import { createTestData } from '@dxos/schema/testing';
 import { range } from '@dxos/util';
 
-import { extractionAnthropicFn } from './extraction-llm-function';
 import { processTranscriptMessage } from '../extraction';
 
+import { extractionAnthropicFn } from './extraction-llm-function';
+
+// TODO(burdon): Rewrite test.
 describe.skip('LLM EntityExtraction', () => {
   let executor: FunctionExecutor;
 
   beforeAll(async () => {
-    executor = new FunctionExecutor(
-      new ServiceContainer().setServices({
-        ai: {
-          client: new EdgeAiServiceClient({
-            endpoint: AI_SERVICE_ENDPOINT.REMOTE,
-            defaultGenerationOptions: {
-              // model: '@anthropic/claude-sonnet-4-20250514',
-              model: '@anthropic/claude-3-5-haiku-20241022',
-            },
-          }),
-        },
-      }),
-    );
+    executor = new FunctionExecutor(todo());
   });
 
   test('should process a transcript block', async () => {

package/src/extraction/extraction-llm-function/extraction-llm-function.ts

@@ -2,16 +2,20 @@
 // Copyright 2025 DXOS.org
 //
 
-import { Message, ToolRegistry } from '@dxos/ai';
+// ISSUE(burdon): defineFunction
+// @ts-nocheck
+
+import { AiService, ToolRegistry } from '@dxos/ai';
 import { Obj } from '@dxos/echo';
 import { create } from '@dxos/echo-schema';
-import { AiService, defineFunction, type FunctionDefinition } from '@dxos/functions';
-import { DataType } from '@dxos/schema';
+import { type FunctionDefinition, defineFunction } from '@dxos/functions';
+import { type ContentBlock, DataType } from '@dxos/schema';
 
-import PROMPT from './instructions.tpl?raw';
-import { AISession } from '../../session';
+import { AiSession } from '../../session';
 import { ExtractionInput, ExtractionOutput } from '../extraction';
-import { insertReferences, ReferencedQuotes } from '../quotes';
+import { ReferencedQuotes, insertReferences } from '../quotes';
+
+import PROMPT from './instructions.tpl?raw';
 
 export const extractionAnthropicFn: FunctionDefinition<ExtractionInput, ExtractionOutput> = defineFunction({
   description: 'Extract entities from the transcript message and add them to the message.',
@@ -19,8 +23,8 @@ export const extractionAnthropicFn: FunctionDefinition<ExtractionInput, Extracti
   outputSchema: ExtractionOutput,
   handler: async ({ data: { message, objects }, context }) => {
     const startTime = performance.now();
-    const ai = context.getService(AiService);
-    const session = new AISession({ operationModel: 'configured' });
+    const ai = context.getService(AiService.AiService);
+    const session = new AiSession({ operationModel: 'configured' });
     const result = await session.runStructured(ReferencedQuotes, {
       generationOptions: {
         model: '@anthropic/claude-3-5-haiku-20241022',
@@ -28,17 +32,18 @@ export const extractionAnthropicFn: FunctionDefinition<ExtractionInput, Extracti
       client: ai.client,
       systemPrompt: PROMPT,
       history: [
-        Obj.make(Message, {
-          role: 'user',
-          content: [
+        Obj.make(DataType.Message, {
+          created: new Date().toISOString(),
+          sender: { role: 'user' },
+          blocks: [
             {
-              type: 'text',
+              _tag: 'text',
               text: `<context>${JSON.stringify(objects)}</context>`,
-            },
+            } satisfies ContentBlock.Text,
             {
-              type: 'text',
+              _tag: 'text',
               text: `<transcript>${JSON.stringify(message.blocks)}</transcript>`,
-            },
+            } satisfies ContentBlock.Text,
           ],
         }),
       ],
@@ -46,13 +51,13 @@ export const extractionAnthropicFn: FunctionDefinition<ExtractionInput, Extracti
       prompt: '',
       tools: [],
       toolResolver: new ToolRegistry([]),
-    });
+    } as any); // TODO(burdon): Rewrite test.
 
     return {
       message: create(DataType.Message, {
         ...message,
         blocks: message.blocks.map((block, i) =>
-          block.type !== 'transcription'
+          block._tag !== 'transcript'
             ? block
             : {
                 ...block,

package/src/extraction/extraction-ner-function/extraction-ner-function.test.ts

@@ -2,19 +2,20 @@
 // Copyright 2025 DXOS.org
 //
 
-import { describe, test, beforeAll } from 'vitest';
+import { beforeAll, describe, test } from 'vitest';
 
 import { type EchoDatabase } from '@dxos/echo-db';
 import { EchoTestBuilder } from '@dxos/echo-db/testing';
 import { FunctionExecutor, ServiceContainer } from '@dxos/functions';
 import { log } from '@dxos/log';
 import { DataType } from '@dxos/schema';
-import { createTestData, Testing } from '@dxos/schema/testing';
+import { Testing, createTestData } from '@dxos/schema/testing';
 import { range } from '@dxos/util';
 
-import { extractionNerFn } from './extraction-ner-function';
 import { processTranscriptMessage } from '../extraction';
 
+import { extractionNerFn } from './extraction-ner-function';
+
 describe.skip('NER EntityExtraction', () => {
   let builder: EchoTestBuilder;
   let db: EchoDatabase;

package/src/extraction/extraction-ner-function/extraction-ner-function.ts

@@ -2,6 +2,9 @@
 // Copyright 2025 DXOS.org
 //
 
+// ISSUE(burdon): defineFunction
+// @ts-nocheck
+
 import { asyncTimeout } from '@dxos/async';
 import { create } from '@dxos/echo-schema';
 import { DatabaseService, defineFunction } from '@dxos/functions';
@@ -9,7 +12,6 @@ import { invariant } from '@dxos/invariant';
 import { log } from '@dxos/log';
 import { DataType } from '@dxos/schema';
 
-import { extractFullEntities } from './named-entity-recognition';
 import { ExtractionInput, ExtractionOutput } from '../extraction';
 import { findQuotes, insertReferences } from '../quotes';
 
@@ -22,10 +24,11 @@ export const extractionNerFn = defineFunction({
     log.info('input', { message, options });
     const startTime = performance.now();
     const { db } = context.getService(DatabaseService);
+    const { extractFullEntities } = await import('./named-entity-recognition');
 
     const entitiesPromise = Promise.all(
       message.blocks.map(async (block) => {
-        invariant(block.type === 'transcription' || block.type === 'text', 'Block must have text');
+        invariant(block._tag === 'transcript' || block._tag === 'text', 'Block must have text');
         return extractFullEntities(block.text);
       }),
     ).then((entities) => entities.flat());
@@ -36,8 +39,8 @@ export const extractionNerFn = defineFunction({
       db,
     );
     const blocksWithReferences = message.blocks.map((block, i) => {
-      invariant(block.type === 'transcription' || block.type === 'text', 'Block must have text');
-      return { ...block, text: insertReferences(block.text, quoteReferences) };
+      invariant(block._tag === 'transcript' || block._tag === 'text', 'Block must have text');
+      return { ...block, text: insertReferences(block._tag, quoteReferences) };
     });
     const messageWithReferences = create(DataType.Message, {
       ...message,

package/src/extraction/extraction-ner-function/named-entity-recognition.test.ts

@@ -57,7 +57,7 @@ describe('NamedEntityRecognition', () => {
     const { transcriptWoflram, transcriptJosiah } = await createTestData();
 
     const blocks = [...transcriptWoflram, ...transcriptJosiah].flatMap((message) =>
-      message.blocks.flatMap((block) => (block.type === 'transcription' ? [block.text] : [])),
+      message.blocks.flatMap((block) => (block._tag === 'transcript' ? [block.text] : [])),
     );
 
     for (const block of blocks) {
@@ -70,7 +70,7 @@ describe('NamedEntityRecognition', () => {
     const { transcriptMessages } = await createTestData();
 
     const blocks = transcriptMessages.flatMap((message) =>
-      message.blocks.flatMap((block) => (block.type === 'transcription' ? [block.text] : [])),
+      message.blocks.flatMap((block) => (block._tag === 'transcript' ? [block.text] : [])),
     );
 
     for (const block of blocks) {

package/src/extraction/extraction-ner-function/named-entity-recognition.ts

@@ -4,9 +4,9 @@
 
 import {
   type TokenClassificationOutput,
+  type TokenClassificationPipelineType,
   type TokenClassificationSingle,
   pipeline,
-  type TokenClassificationPipelineType,
 } from '@xenova/transformers';
 
 import { log } from '@dxos/log';

package/src/extraction/quotes.test.ts

@@ -2,7 +2,7 @@
 // Copyright 2025 DXOS.org
 //
 
-import { describe, test, expect } from 'vitest';
+import { describe, expect, test } from 'vitest';
 
 import { insertReferences } from './quotes';
 

package/src/extraction/quotes.ts

@@ -4,7 +4,7 @@
 
 import { Schema } from 'effect';
 
-import { Query, Filter } from '@dxos/echo';
+import { Filter, Query } from '@dxos/echo';
 import { type EchoDatabase } from '@dxos/echo-db';
 import { ObjectId } from '@dxos/echo-schema';
 import { DXN } from '@dxos/keys';

package/src/functions/index.ts

@@ -0,0 +1,5 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+export * from './services';