@dxos/assistant 0.8.2-staging.7ac8446 → 0.8.2

package/dist/types/src/cypher/query-executor.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"query-executor.d.ts","sourceRoot":"","sources":["../../../../src/cypher/query-executor.ts"],"names":[],"mappings":"AAQA,OAAO,EAAuB,KAAK,aAAa,EAAqB,MAAM,kBAAkB,CAAC;AAG9F,MAAM,WAAW,UAAU;IACzB,QAAQ,EAAE,CAAC,MAAM,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAC1D,gBAAgB,EAAE,CAAC,MAAM,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,KAAK,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;CAC3E;AAED,MAAM,MAAM,IAAI,GAAG;IACjB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,cAAc,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACpC,MAAM,EAAE,IAAI,CAAC;IACb,MAAM,EAAE,IAAI,CAAC;CACd,CAAC;AAEF,eAAO,MAAM,gBAAgB,eACf,UAAU,QAChB,aAAa,KAClB,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAsFnC,CAAC;AAEF,eAAO,MAAM,YAAY,eAAgB,UAAU,SAAS,MAAM,KAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAGrG,CAAC"}

package/dist/types/src/cypher/query-executor.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=query-executor.test.d.ts.map

package/dist/types/src/cypher/query-executor.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"query-executor.test.d.ts","sourceRoot":"","sources":["../../../../src/cypher/query-executor.test.ts"],"names":[],"mappings":""}

package/dist/types/src/cypher/schema.d.ts

@@ -1,12 +0,0 @@
-import { type JsonSchemaType } from '@dxos/echo-schema';
-/**
- * Formats ECHO schema to be consumed by LLM.
- * Schema is formatted in cypher format.
- * Objects are represented as nodes.
- * References are represented as relationships.
- */
-export declare const formatJsonSchemaForLLM: (schemaTypes: JsonSchemaType[]) => string[];
-export declare const formatNodeLabel: (typename: string) => string;
-export declare const formatInferredRelationshipLabel: (typename: string, property: string) => string;
-export declare const isReference: (schema: JsonSchemaType) => boolean;
-//# sourceMappingURL=schema.d.ts.map

package/dist/types/src/cypher/schema.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../../../src/cypher/schema.ts"],"names":[],"mappings":"AAKA,OAAO,EAA2B,KAAK,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAIjF;;;;;GAKG;AACH,eAAO,MAAM,sBAAsB,gBAAiB,cAAc,EAAE,KAAG,MAAM,EAK5E,CAAC;AAkDF,eAAO,MAAM,eAAe,aAAc,MAAM,WAE/C,CAAC;AAEF,eAAO,MAAM,+BAA+B,aAAc,MAAM,YAAY,MAAM,WAGjF,CAAC;AAEF,eAAO,MAAM,WAAW,WAAY,cAAc,YAMD,CAAC"}

package/dist/types/src/cypher/schema.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=schema.test.d.ts.map

package/dist/types/src/cypher/schema.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"schema.test.d.ts","sourceRoot":"","sources":["../../../../src/cypher/schema.test.ts"],"names":[],"mappings":""}

package/dist/types/src/testing/artifacts.test.d.ts

@@ -1 +0,0 @@
-//# sourceMappingURL=artifacts.test.d.ts.map

package/dist/types/src/testing/artifacts.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"artifacts.test.d.ts","sourceRoot":"","sources":["../../../../src/testing/artifacts.test.ts"],"names":[],"mappings":""}

package/dist/types/src/testing/data.d.ts

@@ -1,2 +0,0 @@
-export declare const DATA_MESSAGES: string[];
-//# sourceMappingURL=data.d.ts.map

package/dist/types/src/testing/data.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"data.d.ts","sourceRoot":"","sources":["../../../../src/testing/data.ts"],"names":[],"mappings":"AAIA,eAAO,MAAM,aAAa,UAQzB,CAAC"}

package/dist/types/src/testing/defs.d.ts

@@ -1,5 +0,0 @@
-export declare const AI_SERVICE_ENDPOINT: {
-    LOCAL: string;
-    REMOTE: string;
-};
-//# sourceMappingURL=defs.d.ts.map

package/dist/types/src/testing/defs.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"defs.d.ts","sourceRoot":"","sources":["../../../../src/testing/defs.ts"],"names":[],"mappings":"AAKA,eAAO,MAAM,mBAAmB;;;CAG/B,CAAC"}

package/dist/types/src/testing/logger.d.ts

@@ -1,12 +0,0 @@
-import type { ImageSource } from '@dxos/artifact';
-import type { ConversationEvent } from '../conversation';
-export declare const createLogger: ({ stream, filter, onImage, }?: {
-    stream?: boolean;
-    filter?: (event: ConversationEvent) => boolean;
-    onImage?: (img: {
-        id: string;
-        description?: string;
-        source: ImageSource;
-    }) => void;
-}) => (event: ConversationEvent) => void;
-//# sourceMappingURL=logger.d.ts.map

package/dist/types/src/testing/logger.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../../../src/testing/logger.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,WAAW,EAAuB,MAAM,gBAAgB,CAAC;AAEvE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAEzD,eAAO,MAAM,YAAY,kCAItB;IACD,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,iBAAiB,KAAK,OAAO,CAAC;IAC/C,OAAO,CAAC,EAAE,CAAC,GAAG,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,WAAW,CAAA;KAAE,KAAK,IAAI,CAAC;CACpF,aAIgB,iBAAiB,SA4GjC,CAAC"}

package/dist/types/src/testing/object-extraction.test.d.ts

@@ -1 +0,0 @@
-//# sourceMappingURL=object-extraction.test.d.ts.map

package/dist/types/src/testing/object-extraction.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"object-extraction.test.d.ts","sourceRoot":"","sources":["../../../../src/testing/object-extraction.test.ts"],"names":[],"mappings":""}

package/dist/types/src/testing/ollama.d.ts

@@ -1,6 +0,0 @@
-import { OllamaClient, type OllamaClientParams } from '../ai-service';
-/**
- * Create a test client with small local model and no temperature for predictable results.
- */
-export declare const createTestOllamaClient: (options?: Pick<OllamaClientParams, "tools">) => OllamaClient;
-//# sourceMappingURL=ollama.d.ts.map

package/dist/types/src/testing/ollama.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../../../src/testing/ollama.ts"],"names":[],"mappings":"AAIA,OAAO,EAAwB,YAAY,EAAE,KAAK,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAE5F;;GAEG;AACH,eAAO,MAAM,sBAAsB,aAAc,IAAI,CAAC,kBAAkB,EAAE,OAAO,CAAC,iBAKjF,CAAC"}

package/dist/types/src/testing/query-promts.d.ts

@@ -1,5 +0,0 @@
-import { Schema as S } from 'effect';
-import { type DataSource } from '../cypher';
-export declare const createCypherTool: (dataSource: DataSource, schemaTypes?: S.Schema.Any[]) => import("@dxos/artifact").Tool;
-export declare const createSystemPrompt: (schemaTypes: S.Schema.Any[]) => string;
-//# sourceMappingURL=query-promts.d.ts.map

package/dist/types/src/testing/query-promts.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"query-promts.d.ts","sourceRoot":"","sources":["../../../../src/testing/query-promts.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,MAAM,IAAI,CAAC,EAAE,MAAM,QAAQ,CAAC;AAMrC,OAAO,EAAwC,KAAK,UAAU,EAAE,MAAM,WAAW,CAAC;AAGlF,eAAO,MAAM,gBAAgB,eAAgB,UAAU,gBAAe,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,kCA0ChF,CAAC;AAEL,eAAO,MAAM,kBAAkB,gBAAiB,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,WAiB7D,CAAC"}

package/dist/types/src/testing/query.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=query.test.d.ts.map

package/dist/types/src/testing/query.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"query.test.d.ts","sourceRoot":"","sources":["../../../../src/testing/query.test.ts"],"names":[],"mappings":""}

package/dist/types/src/testing/test-data.d.ts

@@ -1,29 +0,0 @@
-import { type Schema as S } from 'effect';
-import type { EchoDatabase } from '@dxos/echo-db';
-import { type JsonSchemaType } from '@dxos/echo-schema';
-import { type DataSource, type Node, type Relationship } from '../cypher';
-export declare const seedTestData: (db: EchoDatabase) => void;
-export declare const createTestData: () => MockDataSource;
-export declare class MockDataSource implements DataSource {
-    objects: Record<string, {
-        schema: S.Schema.All;
-        typeDxn: string;
-        id: string;
-        data: unknown;
-    }>;
-    schema: Record<string, {
-        json: JsonSchemaType;
-        instance: S.Schema.All;
-    }>;
-    nodes: Node[];
-    relationships: Relationship[];
-    getNodes({ label }: {
-        label?: string;
-    }): Promise<Node[]>;
-    getRelationships({ label }: {
-        label?: string;
-    }): Promise<Relationship[]>;
-    add<S extends S.Schema.All>(schema: S, data: S.Schema.Type<S>): S.Schema.Type<S>;
-    computeGraph(): void;
-}
-//# sourceMappingURL=test-data.d.ts.map

package/dist/types/src/testing/test-data.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"test-data.d.ts","sourceRoot":"","sources":["../../../../src/testing/test-data.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,KAAK,MAAM,IAAI,CAAC,EAAE,MAAM,QAAQ,CAAC;AAG1C,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,EAAmD,KAAK,cAAc,EAAY,MAAM,mBAAmB,CAAC;AAInH,OAAO,EACL,KAAK,UAAU,EACf,KAAK,IAAI,EACT,KAAK,YAAY,EAGlB,MAAM,WAAW,CAAC;AAGnB,eAAO,MAAM,YAAY,OAAQ,YAAY,SAoH5C,CAAC;AAGF,eAAO,MAAM,cAAc,QAAO,cAuGjC,CAAC;AAEF,qBAAa,cAAe,YAAW,UAAU;IAC/C,OAAO,EAAE,MAAM,CACb,MAAM,EACN;QACE,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;QACrB,OAAO,EAAE,MAAM,CAAC;QAChB,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,OAAO,CAAC;KACf,CACF,CAAM;IAEP,MAAM,EAAE,MAAM,CACZ,MAAM,EACN;QACE,IAAI,EAAE,cAAc,CAAC;QACrB,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;KACxB,CACF,CAAM;IAEP,KAAK,EAAE,IAAI,EAAE,CAAM;IACnB,aAAa,EAAE,YAAY,EAAE,CAAM;IAE7B,QAAQ,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE;IAItC,gBAAgB,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE;IAIpD,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;IAsB7D,YAAY;CA0Eb"}

package/dist/types/src/testing/test-schema.d.ts

@@ -1,50 +0,0 @@
-import { Schema as S } from 'effect';
-export declare const ObjectId: S.SchemaClass<string, string, never>;
-export declare const Contact: S.mutable<S.Struct<{
-    id: S.SchemaClass<string, string, never>;
-    name: S.SchemaClass<string, string, never>;
-    email: S.optional<typeof S.String>;
-}>>;
-export interface Contact extends S.Schema.Type<typeof Contact> {
-}
-export declare const Project: S.mutable<S.Struct<{
-    id: S.SchemaClass<string, string, never>;
-    name: S.SchemaClass<string, string, never>;
-    description: S.optional<typeof S.String>;
-}>>;
-export interface Project extends S.Schema.Type<typeof Project> {
-}
-export declare const Task: S.mutable<S.Struct<{
-    id: S.SchemaClass<string, string, never>;
-    name: S.SchemaClass<string, string, never>;
-    description: S.optional<typeof S.String>;
-    project: import("@dxos/echo-schema").Ref$<{
-        name: string;
-        description?: string | undefined;
-        id: string;
-    }>;
-    assignee: import("@dxos/echo-schema").Ref$<{
-        name: string;
-        id: string;
-        email?: string | undefined;
-    }>;
-}>>;
-export interface Task extends S.Schema.Type<typeof Task> {
-}
-export declare const Org: S.mutable<S.Struct<{
-    id: S.SchemaClass<string, string, never>;
-    name: S.SchemaClass<string, string, never>;
-    projects: S.Array$<import("@dxos/echo-schema").Ref$<{
-        name: string;
-        description?: string | undefined;
-        id: string;
-    }>>;
-    employees: S.Array$<import("@dxos/echo-schema").Ref$<{
-        name: string;
-        id: string;
-        email?: string | undefined;
-    }>>;
-}>>;
-export interface Org extends S.Schema.Type<typeof Org> {
-}
-//# sourceMappingURL=test-schema.d.ts.map

package/dist/types/src/testing/test-schema.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"test-schema.d.ts","sourceRoot":"","sources":["../../../../src/testing/test-schema.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,MAAM,IAAI,CAAC,EAAE,MAAM,QAAQ,CAAC;AAQrC,eAAO,MAAM,QAAQ,sCAAyE,CAAC;AAE/F,eAAO,MAAM,OAAO;;;;GAahB,CAAC;AACL,MAAM,WAAW,OAAQ,SAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,OAAO,CAAC;CAAG;AAEjE,eAAO,MAAM,OAAO;;;;GAahB,CAAC;AACL,MAAM,WAAW,OAAQ,SAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,OAAO,CAAC;CAAG;AAEjE,eAAO,MAAM,IAAI;;;;;;;;;;;;;;GAeb,CAAC;AACL,MAAM,WAAW,IAAK,SAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC;CAAG;AAE3D,eAAO,MAAM,GAAG;;;;;;;;;;;;;GAcZ,CAAC;AACL,MAAM,WAAW,GAAI,SAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,GAAG,CAAC;CAAG"}

package/dist/types/src/util/index.d.ts

@@ -1,2 +0,0 @@
-export * from './trim';
-//# sourceMappingURL=index.d.ts.map

package/dist/types/src/util/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/util/index.ts"],"names":[],"mappings":"AAIA,cAAc,QAAQ,CAAC"}

package/dist/types/src/util/trim.d.ts

@@ -1,5 +0,0 @@
-/**
- * Trims multiline strings.
- */
-export declare const trim: (strings: TemplateStringsArray, ...values: any[]) => string;
-//# sourceMappingURL=trim.d.ts.map

package/dist/types/src/util/trim.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"trim.d.ts","sourceRoot":"","sources":["../../../../src/util/trim.ts"],"names":[],"mappings":"AAIA;;GAEG;AAEH,eAAO,MAAM,IAAI,YAAa,oBAAoB,aAAa,GAAG,EAAE,WAMnE,CAAC"}

package/src/ai-service/client.test.ts

@@ -1,265 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { Schema as S } from 'effect';
-import { test, describe } from 'vitest';
-
-import { defineTool, Message, ToolResult, type Tool } from '@dxos/artifact';
-import { toJsonSchema, ObjectId, createStatic } from '@dxos/echo-schema';
-import { invariant } from '@dxos/invariant';
-import { SpaceId } from '@dxos/keys';
-import { log } from '@dxos/log';
-
-import { DEFAULT_EDGE_MODEL } from './defs';
-import { AIServiceEdgeClient } from './edge-client';
-import { OllamaClient } from './ollama-client';
-import { MixedStreamParser } from './parser';
-import { ToolTypes } from './types';
-import { createTestOllamaClient, AI_SERVICE_ENDPOINT } from '../testing';
-
-// log.config({ filter: 'debug' });
-
-describe.skip('AI Service Client', () => {
-  test('client generation', async () => {
-    const client = new AIServiceEdgeClient({
-      endpoint: AI_SERVICE_ENDPOINT.LOCAL,
-    });
-
-    const spaceId = SpaceId.random();
-    const threadId = ObjectId.random();
-
-    await client.appendMessages([
-      {
-        id: ObjectId.random(),
-        spaceId,
-        threadId,
-        role: 'user',
-        content: [{ type: 'text', text: 'Hello' }],
-      },
-    ]);
-
-    const stream = await client.exec({
-      model: DEFAULT_EDGE_MODEL,
-      spaceId,
-      threadId,
-      systemPrompt: 'You are a poet',
-      tools: [],
-    });
-    for await (const event of stream) {
-      log('event', event);
-    }
-
-    log('full message', {
-      message: await stream.complete(),
-    });
-  });
-
-  test('tool calls', async () => {
-    const client = new AIServiceEdgeClient({
-      endpoint: AI_SERVICE_ENDPOINT.LOCAL,
-    });
-
-    const custodian: Tool = {
-      name: 'custodian',
-      description: 'Custodian can tell you the password if you say the magic word',
-      parameters: toJsonSchema(
-        S.Struct({
-          magicWord: S.String.annotations({ description: 'The magic word. Should be exactly "pretty please"' }),
-        }),
-      ),
-    };
-
-    const spaceId = SpaceId.random();
-    const threadId = ObjectId.random();
-
-    await client.appendMessages([
-      {
-        id: ObjectId.random(),
-        spaceId,
-        threadId,
-        role: 'user',
-        content: [{ type: 'text', text: 'What is the password? Ask the custodian' }],
-      },
-    ]);
-
-    {
-      const stream1 = await client.exec({
-        model: DEFAULT_EDGE_MODEL,
-        spaceId,
-        threadId,
-        systemPrompt: 'You are a helpful assistant.',
-        tools: [custodian],
-      });
-
-      for await (const event of stream1) {
-        log('event', event);
-      }
-
-      // TODO(burdon): !!!
-      await stream1.complete();
-      const messages: Message[] = [];
-
-      const [message1] = messages;
-      log('full message', { message: message1 });
-      await client.appendMessages([message1]);
-
-      const toolUse = message1.content.find(({ type }) => type === 'tool_use')!;
-      invariant(toolUse.type === 'tool_use');
-      await client.appendMessages([
-        {
-          id: ObjectId.random(),
-          spaceId,
-          threadId,
-          role: 'user',
-          content: [{ type: 'tool_result', toolUseId: toolUse.id, content: 'password="The sky is blue"' }],
-        },
-      ]);
-    }
-
-    {
-      const stream2 = await client.exec({
-        model: DEFAULT_EDGE_MODEL,
-        spaceId,
-        threadId,
-        systemPrompt: 'You are a helpful assistant.',
-        tools: [custodian],
-      });
-
-      for await (const event of stream2) {
-        log('event', event);
-      }
-
-      await stream2.complete();
-      const messages: Message[] = [];
-
-      const [message2] = messages;
-      log('full message', { message: message2 });
-    }
-  });
-
-  test.skip('image generation', async () => {
-    const client = new AIServiceEdgeClient({
-      endpoint: AI_SERVICE_ENDPOINT.LOCAL,
-    });
-
-    const spaceId = SpaceId.random();
-    const threadId = ObjectId.random();
-
-    await client.appendMessages([
-      {
-        id: ObjectId.random(),
-        spaceId,
-        threadId,
-        role: 'user',
-        content: [{ type: 'text', text: 'Generate an image of a cat' }],
-      },
-    ]);
-
-    const stream = await client.exec({
-      model: DEFAULT_EDGE_MODEL,
-      spaceId,
-      threadId,
-      tools: [
-        {
-          name: 'text-to-image',
-          type: ToolTypes.TextToImage,
-          // options: {
-          //   model: '@cf/stabilityai/stable-diffusion-xl-base-1.0',
-          // },
-        },
-      ],
-    });
-
-    for await (const event of stream) {
-      log('event', event);
-    }
-
-    log('full message', { message: await stream.complete() });
-  });
-});
-
-describe.skip('Ollama Client', () => {
-  test('basic', async (ctx) => {
-    const isRunning = await OllamaClient.isRunning();
-    if (!isRunning) {
-      ctx.skip();
-    }
-
-    const client = createTestOllamaClient();
-    const parser = new MixedStreamParser();
-
-    const messages = await parser.parse(
-      await client.exec({
-        prompt: createStatic(Message, {
-          role: 'user',
-          content: [{ type: 'text', text: 'Hello, world!' }],
-        }),
-      }),
-    );
-
-    log('messages', { messages });
-  });
-
-  test('tool calls', async (ctx) => {
-    const isRunning = await OllamaClient.isRunning();
-    if (!isRunning) {
-      ctx.skip();
-    }
-
-    const client = createTestOllamaClient({
-      tools: [
-        defineTool('test', {
-          name: 'encrypt',
-          description: 'Encrypt a message',
-          schema: S.Struct({
-            message: S.String.annotations({ description: 'The message to encrypt' }),
-          }),
-          execute: async ({ message }) => ToolResult.Success(message.split('').reverse().join('')),
-        }),
-      ],
-    });
-    const parser = new MixedStreamParser();
-    parser.streamEvent.on((event) => {
-      // log('event', { event });
-    });
-
-    const messages = await parser.parse(
-      await client.exec({
-        prompt: createStatic(Message, {
-          role: 'user',
-          content: [{ type: 'text', text: 'What is the encrypted message for "Hello, world!"' }],
-        }),
-      }),
-    );
-
-    log('messages', { messages });
-  });
-
-  test('text-to-image', async (ctx) => {
-    const isRunning = await OllamaClient.isRunning();
-    if (!isRunning) {
-      ctx.skip();
-    }
-
-    const client = createTestOllamaClient();
-    const parser = new MixedStreamParser();
-
-    const messages = await parser.parse(
-      await client.exec({
-        prompt: createStatic(Message, {
-          role: 'user',
-          content: [{ type: 'text', text: 'Generate an image of a cat' }],
-        }),
-        tools: [
-          {
-            name: 'text-to-image',
-            type: ToolTypes.TextToImage,
-          },
-        ],
-      }),
-    );
-
-    log('messages', { messages });
-  });
-});

package/src/ai-service/defs.ts

@@ -1,39 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-// TODO(burdon): Share with Edge.
-// TODO(burdon): Move to service config retrieved by HTTP?
-
-export const DEFAULT_EDGE_MODEL = '@anthropic/claude-3-5-sonnet-20241022';
-
-export const DEFAULT_EDGE_MODELS = [
-  // AI Gateway.
-  // https://developers.cloudflare.com/ai-gateway
-  '@anthropic/claude-3-5-haiku-20241022',
-  '@anthropic/claude-3-5-sonnet-20241022',
-  '@anthropic/claude-3-7-sonnet-20250219',
-
-  // Workers AI.
-  // https://developers.cloudflare.com/workers-ai/models
-  '@cf/deepseek-ai/deepseek-r1-distill-qwen-32b',
-  '@hf/nousresearch/hermes-2-pro-mistral-7b',
-  '@ollama/llama-3-1-nemotron-70b-instruct',
-  '@ollama/llama-3-1-nemotron-mini-4b-instruct',
-  '@ollama/llama-3-2-3b',
-] as const;
-
-// TODO(burdon): Config.
-export const DEFAULT_OLLAMA_ENDPOINT = 'http://localhost:11434';
-
-export const DEFAULT_OLLAMA_MODEL = 'llama3.2:1b';
-
-/**
- * https://ollama.com/library
- */
-export const DEFAULT_OLLAMA_MODELS = [
-  //
-  'llama3.2:1b',
-  'llama3:70b',
-  'deepseek-r1:latest',
-] as const;

package/src/ai-service/edge-client.ts

@@ -1,67 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { Schema as S } from 'effect';
-
-import { Message } from '@dxos/artifact';
-import { assertArgument, invariant } from '@dxos/invariant';
-import { log } from '@dxos/log';
-
-import type { AIServiceClient, GenerationStream } from './service';
-import { createGenerationStream } from './stream';
-import { type GenerateRequest } from './types';
-
-export type AIServiceEdgeClientOptions = {
-  endpoint: string;
-};
-
-export class AIServiceEdgeClient implements AIServiceClient {
-  private readonly _endpoint: string;
-
-  constructor({ endpoint }: AIServiceEdgeClientOptions) {
-    invariant(endpoint, 'endpoint is required');
-    this._endpoint = endpoint;
-  }
-
-  /**
-   * @deprecated
-   */
-  // TODO(burdon): Remove.
-  async appendMessages(messages: Message[]): Promise<void> {
-    const url = `${this._endpoint}/message`;
-    const res = await fetch(url, {
-      method: 'POST',
-      headers: { 'Content-Type': 'application/json' },
-      body: JSON.stringify({
-        messages: await S.encodePromise(S.Array(Message))(messages),
-      }),
-    });
-
-    if (!res.ok) {
-      throw new Error(`${await res.text()} [${url}]`);
-    }
-  }
-
-  /**
-   * Process request and open message stream.
-   */
-  async exec(request: GenerateRequest): Promise<GenerationStream> {
-    assertArgument(typeof request.model === 'string', 'model is required');
-    log('requesting', { endpoint: this._endpoint });
-    const controller = new AbortController();
-    const response = await fetch(`${this._endpoint}/generate`, {
-      signal: controller.signal,
-      method: 'POST',
-      headers: { 'Content-Type': 'application/json' },
-      body: JSON.stringify({
-        // TODO(dmaretskyi): Errors if tools are not provided.
-        tools: request.tools ?? [],
-        ...request,
-      }),
-    });
-
-    invariant(response.body instanceof ReadableStream);
-    return createGenerationStream(response, controller);
-  }
-}

package/src/ai-service/index.ts

@@ -1,13 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-export * from './defs';
-export * from './edge-client';
-export * from './message-collector';
-export * from './ollama-client';
-export * from './parser';
-export * from './service';
-export * from './stream';
-export * from './transform';
-export * from './types';