@dxos/assistant-toolkit 0.8.4-main.ae835ea → 0.8.4-main.bcb3aa67d6

package/src/functions/research/create-document.ts

@@ -1,69 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
-
-import { Relation } from '@dxos/echo';
-import { DatabaseService, TracingService, defineFunction } from '@dxos/functions';
-import { invariant } from '@dxos/invariant';
-import { DXN, ObjectId } from '@dxos/keys';
-import { log } from '@dxos/log';
-import { Markdown } from '@dxos/plugin-markdown/types';
-import { DataType } from '@dxos/schema';
-import { trim } from '@dxos/util';
-
-export default defineFunction({
-  key: 'dxos.org/function/research/create-document',
-  name: 'Create research document',
-  description: 'Creates a note summarizing the research.',
-  inputSchema: Schema.Struct({
-    name: Schema.String.annotations({
-      description: 'Name of the note.',
-    }),
-
-    content: Schema.String.annotations({
-      description: trim`
-        Content of the note. 
-        Supports (and are prefered) references to research objects using @ syntax and <object> tags (refer to research blueprint instructions).
-      `,
-    }),
-
-    // TODO(dmaretskyi): Use a specialized type for this (e.g., ArtifactId renamed as RefFromLLM).
-    target: Schema.String.annotations({
-      description: trim`
-        Id of the object (organization, contact, etc.) for which the research was performed. 
-        This must be a ulid.
-      `,
-    }),
-  }),
-  outputSchema: Schema.Struct({}), // TODO(burdon): Schema.Void?
-  handler: Effect.fnUntraced(function* ({ data: { target, name, content } }) {
-    log.info('Creating research document', { target, name, content });
-
-    yield* DatabaseService.flush({ indexes: true });
-    yield* TracingService.emitStatus({ message: 'Creating research document...' });
-    invariant(ObjectId.isValid(target));
-
-    const targetObj = yield* DatabaseService.resolve(DXN.fromLocalObjectId(target));
-
-    const doc = yield* DatabaseService.add(
-      Markdown.makeDocument({
-        name,
-        content,
-      }),
-    );
-    yield* DatabaseService.add(
-      Relation.make(DataType.HasSubject, {
-        [Relation.Source]: doc,
-        [Relation.Target]: targetObj as any,
-        completedAt: new Date().toISOString(),
-      }),
-    );
-    yield* DatabaseService.flush({ indexes: true });
-
-    log.info('Created research document', { target, name, content });
-    return {};
-  }),
-});

package/src/functions/research/graph.test.ts

@@ -1,69 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as AnthropicClient from '@effect/ai-anthropic/AnthropicClient';
-import * as FetchHttpClient from '@effect/platform/FetchHttpClient';
-import { describe, it } from '@effect/vitest';
-import * as Config from 'effect/Config';
-import * as Effect from 'effect/Effect';
-import * as Function from 'effect/Function';
-import * as Layer from 'effect/Layer';
-
-import { AiService, ToolExecutionService, ToolResolverService } from '@dxos/ai';
-import * as AiServiceRouter from '@dxos/ai/AiServiceRouter';
-import { tapHttpErrors } from '@dxos/ai/testing';
-import { AiSession } from '@dxos/assistant';
-import { TestHelpers } from '@dxos/effect';
-import { DatabaseService, TracingService } from '@dxos/functions';
-import { log } from '@dxos/log';
-import { DataType } from '@dxos/schema';
-
-import { makeGraphWriterHandler, makeGraphWriterToolkit } from './graph';
-
-// import { type EchoTestBuilder } from '@dxos/echo-db/testing';
-
-const TestLayer = Function.pipe(
-  AiService.model('@anthropic/claude-3-5-sonnet-20241022'),
-  Layer.provideMerge(DatabaseService.notAvailable),
-  Layer.provideMerge(ToolResolverService.layerEmpty),
-  Layer.provideMerge(ToolExecutionService.layerEmpty),
-  Layer.provide(AiServiceRouter.AiServiceRouter),
-  Layer.provide(
-    AnthropicClient.layerConfig({
-      apiKey: Config.redacted('ANTHROPIC_API_KEY'),
-      transformClient: tapHttpErrors,
-    }),
-  ),
-  Layer.provide(FetchHttpClient.layer),
-  Layer.provideMerge(TracingService.layerNoop),
-);
-
-describe('graph', () => {
-  // let builder: EchoTestBuilder;
-  // test('findRelatedSchema', async () => {
-  //   const db = await createEchoDatabase();
-  //   const relatedSchemas = await findRelatedSchema(db, Schema.Struct({}));
-  // });
-
-  const Toolkit = makeGraphWriterToolkit({ schema: [DataType.Project] });
-  const ToolkitLayer = makeGraphWriterHandler(Toolkit);
-
-  it.effect.skip(
-    'calculator',
-    Effect.fn(
-      function* (_) {
-        const session = new AiSession();
-        const toolkit = yield* Toolkit;
-        const response = yield* session.run({
-          toolkit,
-          prompt: 'What is 10 + 20?',
-        });
-
-        log.info('response', { response });
-      },
-      Effect.provide(Layer.mergeAll(TestLayer, ToolkitLayer)),
-      TestHelpers.runIf(process.env.ANTHROPIC_API_KEY),
-    ),
-  );
-});

package/src/functions/research/index.ts

@@ -1,15 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { default as create$ } from './create-document';
-import { default as research$ } from './research';
-
-export * from './graph';
-export * from './research-graph';
-export * from './types';
-
-export namespace Research {
-  export const create = create$;
-  export const research = research$;
-}

package/src/functions/research/research-graph.ts

@@ -1,47 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Layer from 'effect/Layer';
-import * as Schema from 'effect/Schema';
-
-import { Obj, Query, Ref, Type } from '@dxos/echo';
-import { Queue } from '@dxos/echo-db';
-import { ContextQueueService, DatabaseService, QueueService } from '@dxos/functions';
-
-/**
- * Container for a set of ephemeral research results.
- */
-export const ResearchGraph = Schema.Struct({
-  queue: Type.Ref(Queue),
-}).pipe(
-  Type.Obj({
-    typename: 'dxos.org/type/ResearchGraph',
-    version: '0.1.0',
-  }),
-);
-
-export interface ResearchGraph extends Schema.Schema.Type<typeof ResearchGraph> {}
-
-export const queryResearchGraph: () => Effect.Effect<ResearchGraph | undefined, never, DatabaseService> = Effect.fn(
-  'queryResearchGraph',
-)(function* () {
-  const { objects } = yield* DatabaseService.runQuery(Query.type(ResearchGraph));
-  return objects.at(0);
-});
-
-export const createResearchGraph: () => Effect.Effect<ResearchGraph, never, DatabaseService | QueueService> = Effect.fn(
-  'createResearchGraph',
-)(function* () {
-  const queue = yield* QueueService.createQueue();
-  return yield* DatabaseService.add(Obj.make(ResearchGraph, { queue: Ref.fromDXN(queue.dxn) }));
-});
-
-export const contextQueueLayerFromResearchGraph = Layer.unwrapEffect(
-  Effect.gen(function* () {
-    const researchGraph = (yield* queryResearchGraph()) ?? (yield* createResearchGraph());
-    const researchQueue = yield* DatabaseService.load(researchGraph.queue);
-    return ContextQueueService.layer(researchQueue);
-  }),
-);