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

package/src/research/research.ts

@@ -1,90 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Schema } from 'effect';
-
-import { AgentStatusReport, ConsolePrinter, ToolRegistry } from '@dxos/ai';
-import { create } from '@dxos/echo-schema';
-import { AiService, CredentialsService, DatabaseService, defineFunction, TracingService } from '@dxos/functions';
-import { log } from '@dxos/log';
-import { DataTypes } from '@dxos/schema';
-
-import { createExaTool, createMockExaTool } from './exa';
-import { createGraphWriterTool } from './graph';
-// TODO(dmaretskyi): Vite build bug with instruction files with the same filename getting mixed-up
-import PROMPT from './instructions-research.tpl?raw';
-import { AISession } from '../session';
-
-/**
- * Exec external service and return the results as a Subgraph.
- */
-// TODO(burdon): Rename.
-export const researchFn = defineFunction({
-  description: 'Research the web for information',
-  inputSchema: Schema.Struct({
-    query: Schema.String.annotations({
-      description: 'The query to search for.',
-    }),
-
-    // TOOD(burdon): Move to context.
-    mockSearch: Schema.optional(Schema.Boolean).annotations({
-      description: 'Whether to use the mock search tool.',
-      default: false,
-    }),
-  }),
-  outputSchema: Schema.Struct({
-    // result: Schema.String,
-    result: Schema.Any,
-  }),
-  handler: async ({ data: { query, mockSearch }, context }) => {
-    const ai = context.getService(AiService);
-    const credentials = context.getService(CredentialsService);
-    const { db } = context.getService(DatabaseService);
-    // const queues = context.getService(QueueService);
-    const tracing = context.getService(TracingService);
-
-    tracing.write(create(AgentStatusReport, { message: 'Researching...' }));
-
-    const exaCredential = await credentials.getCredential({ service: 'exa.ai' });
-    const searchTool = mockSearch ? createMockExaTool() : createExaTool({ apiKey: exaCredential.apiKey! });
-
-    const printer = new ConsolePrinter();
-    const session = new AISession({ operationModel: 'configured' });
-    session.message.on((message) => printer.printMessage(message));
-    session.userMessage.on((message) => printer.printMessage(message));
-    session.block.on((block) => printer.printContentBlock(block));
-    session.statusReport.on((status) => {
-      log.info('[agent] status', { status });
-      tracing.write(status);
-    });
-    session.streamEvent.on((event) => log('stream', { event }));
-
-    const graphWriteTool = createGraphWriterTool({ db, schema: DataTypes });
-    log.info('graphWriteTool', { schema: graphWriteTool.parameters });
-
-    // TODO(dmaretskyi): Consider adding this pattern as the "Graph" output mode for the session.
-    const result = await session.run({
-      client: ai.client,
-      systemPrompt: PROMPT,
-      artifacts: [],
-      tools: [searchTool.id, graphWriteTool.id],
-      history: [],
-      prompt: query,
-      toolResolver: new ToolRegistry([searchTool, graphWriteTool]),
-    });
-
-    return {
-      result: result as any,
-    };
-
-    // queues.contextQueue!.append(data);
-
-    // return {
-    //   result: `
-    //   The research results are placed in the following objects:
-    //     ${data.map((object, id) => `[obj_${id}][dxn:echo:@:${object.id}]`).join('\n')}
-    //   `,
-    // };
-  },
-});

package/src/session/session.test.ts

@@ -1,275 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Schema } from 'effect';
-import { describe, test } from 'vitest';
-
-import { EdgeAiServiceClient, ConsolePrinter, createTool, ToolResult, ToolRegistry } from '@dxos/ai';
-import { AI_SERVICE_ENDPOINT } from '@dxos/ai/testing';
-import { ArtifactId, defineArtifact } from '@dxos/artifact';
-import { Type, Obj } from '@dxos/echo';
-import { ObjectId } from '@dxos/echo-schema';
-import { DXN } from '@dxos/keys';
-import { log } from '@dxos/log';
-
-import { AISession } from './session';
-
-// Define a calendar event artifact schema.
-const CalendarEventSchema = Schema.Struct({
-  title: Schema.String,
-  startTime: Schema.String,
-  endTime: Schema.String,
-  description: Schema.String,
-}).pipe(
-  Type.Obj({
-    typename: 'example.com/type/CalendarEvent',
-    version: '0.1.0',
-  }),
-);
-
-type CalendarEvent = Schema.Schema.Type<typeof CalendarEventSchema>;
-
-// TODO(burdon): Flaky.
-describe.runIf(process.env.DX_RUN_SLOW_TESTS === '1')('AISession with Ollama', () => {
-  test('tool', async () => {
-    const aiClient = new EdgeAiServiceClient({ endpoint: AI_SERVICE_ENDPOINT.REMOTE });
-    // const aiClient = new OllamaAiServiceClient({
-    //   overrides: { model: 'llama3.1:8b' },
-    // });
-    const session = new AISession({ operationModel: 'configured' });
-
-    const printer = new ConsolePrinter();
-    session.message.on((message) => printer.printMessage(message));
-    session.userMessage.on((message) => printer.printMessage(message));
-    session.block.on((block) => printer.printContentBlock(block));
-
-    // session.update.on((update) => {
-    //   log('update', { update });
-    // });
-
-    const calculatorTool = createTool('test', {
-      name: 'calculator',
-      description: 'Adds to numbers',
-      schema: Schema.Struct({
-        operand1: Schema.Number,
-        operand2: Schema.Number,
-      }),
-      execute: async ({ operand1, operand2 }) => {
-        return ToolResult.Success(operand1 + operand2);
-      },
-    });
-
-    // Test creating an itinerary
-    const response = await session.run({
-      client: aiClient,
-      tools: [calculatorTool.id],
-      artifacts: [],
-      requiredArtifactIds: [],
-      history: [],
-      generationOptions: {
-        model: '@anthropic/claude-3-5-haiku-20241022',
-      },
-      prompt: 'What is 10 + 20?',
-      toolResolver: new ToolRegistry([calculatorTool]),
-    });
-
-    log('result', { response });
-  });
-
-  test('create calendar itinerary', { timeout: 60_000 }, async () => {
-    const aiClient = new EdgeAiServiceClient({ endpoint: AI_SERVICE_ENDPOINT.REMOTE });
-    // const aiClient = new OllamaAiServiceClient({
-    //   overrides: { model: 'llama3.1:8b' },
-    // });
-    const session = new AISession({ operationModel: 'configured' });
-
-    const objects = new Set<string>();
-
-    // Define calendar artifact.
-    const calendarArtifact = defineArtifact({
-      id: 'artifact:dxos.org/plugin/calendar',
-      name: 'Calendar',
-      instructions: 'Use this to create and query calendar events.',
-      schema: CalendarEventSchema,
-      tools: [
-        createTool('calendar', {
-          name: 'query',
-          description: 'Query the calendar for events',
-          schema: Schema.Struct({}),
-          execute: async () => {
-            return ToolResult.Success(CALENDAR_EVENTS);
-          },
-        }),
-      ],
-    });
-
-    const tableArtifact = defineArtifact({
-      id: 'artifact:dxos.org/plugin/table',
-      name: 'Table',
-      instructions: 'Use this to create and manage tables. Each table has a unique id.',
-      schema: Schema.Struct({}),
-      tools: [
-        createTool('table', {
-          name: 'create',
-          description: 'Create a table',
-          schema: Schema.Struct({
-            data: Schema.Array(Schema.Any).annotations({ description: 'Array of data payloads to add as rows' }),
-          }),
-          execute: async ({ data }) => {
-            log('create table', { data });
-            const id = DXN.fromLocalObjectId(ObjectId.random()).toString();
-            objects.add(id);
-            // TODO(dmaretskyi): consider xml for refs instead of @dxn:echo:@:XXXXX
-            return ToolResult.Success(`table @${id}`);
-          },
-        }),
-      ],
-    });
-
-    const mapArtifact = defineArtifact({
-      id: 'artifact:dxos.org/plugin/map',
-      name: 'Map',
-      instructions:
-        'Use this to create and manage maps. Maps source data from tables. Table id is required to create a map.',
-      schema: Schema.Struct({}),
-      tools: [
-        createTool('map', {
-          name: 'create',
-          description: 'Create a map',
-          schema: Schema.Struct({
-            source: ArtifactId.annotations({
-              description: 'The table that will be used as the source of the map',
-            }),
-          }),
-          execute: async ({ source }) => {
-            // TODO(dmaretskyi): Use effect-schema decode instead of manual parsing.
-            const sourceId = ArtifactId.toDXN(source);
-            if (!objects.has(sourceId.toString())) {
-              return ToolResult.Error(`table id=${source} not found`);
-            }
-            log('create map', { sourceId });
-            const id = DXN.fromLocalObjectId(ObjectId.random()).toString();
-            objects.add(id);
-            // TODO(dmaretskyi): consider xml for refs instead of @dxn:echo:@:XXXXX
-            return ToolResult.Success(`map @${id}`);
-          },
-        }),
-      ],
-    });
-
-    const scriptArtifact = defineArtifact({
-      id: 'artifact:dxos.org/plugin/script',
-      name: 'Script',
-      instructions: 'Use this to create and manage scripts',
-      schema: Schema.Struct({}),
-      tools: [],
-    });
-
-    // session.streamEvent.on((event) => {
-    //   printStreamEvent(event);
-    // });
-
-    const printer = new ConsolePrinter();
-    session.message.on((message) => printer.printMessage(message));
-    session.userMessage.on((message) => printer.printMessage(message));
-    session.block.on((block) => printer.printContentBlock(block));
-
-    // session.update.on((update) => {
-    //   log('update', { update });
-    // });
-
-    // Test creating an itinerary
-    const response = await session.run({
-      client: aiClient,
-      tools: [],
-      artifacts: [calendarArtifact, tableArtifact, mapArtifact, scriptArtifact],
-      requiredArtifactIds: [calendarArtifact.id, tableArtifact.id, mapArtifact.id, scriptArtifact.id],
-      history: [],
-      generationOptions: {
-        model: '@anthropic/claude-3-5-haiku-20241022',
-      },
-      prompt: 'create a table and map for a travel itinerary based on events in my calendar',
-      toolResolver: new ToolRegistry([]),
-    });
-
-    log('result', {
-      objects,
-      finalMessage: response.at(-1),
-    });
-  });
-});
-
-// Travel to rome, florence, livorno, siena, madrid for conferences
-const CALENDAR_EVENTS: CalendarEvent[] = [
-  Obj.make(CalendarEventSchema, {
-    title: 'Exploring Ancient Ruins in Rome',
-    startTime: '2024-01-01T10:00:00Z',
-    endTime: '2024-01-01T11:00:00Z',
-    description: 'Tech conference at the historic Colosseum with networking opportunities',
-  }),
-  Obj.make(CalendarEventSchema, {
-    title: 'Renaissance Tech Summit in Florence',
-    startTime: '2024-01-01T11:00:00Z',
-    endTime: '2024-01-01T12:00:00Z',
-    description: 'Discussing AI innovations surrounded by Renaissance art',
-  }),
-  Obj.make(CalendarEventSchema, {
-    title: 'Travel to Livorno',
-    startTime: '2024-01-01T12:00:00Z',
-    endTime: '2024-01-01T13:00:00Z',
-    description: 'Travel to Livorno',
-  }),
-  Obj.make(CalendarEventSchema, {
-    title: 'Travel to Siena',
-    startTime: '2024-01-01T13:00:00Z',
-    endTime: '2024-01-01T14:00:00Z',
-    description: 'Travel to Siena',
-  }),
-  Obj.make(CalendarEventSchema, {
-    title: 'Travel to Madrid',
-    startTime: '2024-01-01T14:00:00Z',
-    endTime: '2024-01-01T15:00:00Z',
-    description: 'Travel to Madrid',
-  }),
-];
-
-// const printStreamEvent = (event: GenerationStreamEvent) => {
-//   switch (event.type) {
-//     case 'message_start': {
-//       process.stdout.write(`${event.message.role.toUpperCase()}\n\n`);
-//       for (const content of event.message.content) {
-//         printContentBlock(content);
-//       }
-//       break;
-//     }
-//     case 'content_block_start': {
-//       printContentBlock(event.content);
-//       break;
-//     }
-//     case 'content_block_delta': {
-//       switch (event.delta.type) {
-//         case 'text_delta': {
-//           process.stdout.write(event.delta.text);
-//           break;
-//         }
-//         case 'input_json_delta': {
-//           process.stdout.write(event.delta.partial_json);
-//           break;
-//         }
-//       }
-//       break;
-//     }
-//     case 'content_block_stop': {
-//       process.stdout.write('\n');
-//       break;
-//     }
-//     case 'message_delta': {
-//       break;
-//     }
-//     case 'message_stop': {
-//       process.stdout.write('\n\n');
-//       break;
-//     }
-//   }
-// };