@dxos/assistant-toolkit 0.8.4-main.72ec0f3 → 0.8.4-main.74a063c4e0

package/src/functions/github/fetch-prs.ts

@@ -1,30 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as HttpClient from '@effect/platform/HttpClient';
-import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
-
-import { defineFunction, withAuthorization } from '@dxos/functions';
-
-export default defineFunction({
-  key: 'dxos.org/function/github/fetch-prs',
-  name: 'Fetch PRs',
-  description: 'Fetches PRs from GitHub.',
-  inputSchema: Schema.Struct({
-    owner: Schema.String.annotations({
-      description: 'GitHub owner.',
-    }),
-    repo: Schema.String.annotations({
-      description: 'GitHub repository.',
-    }),
-  }),
-  handler: Effect.fnUntraced(function* ({ data }) {
-    const client = yield* HttpClient.HttpClient.pipe(Effect.map(withAuthorization({ service: 'github.com' })));
-
-    const response = yield* client.get(`https://api.github.com/repos/${data.owner}/${data.repo}/pulls`);
-    const json: any = yield* response.json;
-    return json;
-  }),
-});

package/src/functions/linear/index.ts

@@ -1,9 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { default as sync$ } from './sync-issues';
-
-export namespace Linear {
-  export const sync = sync$;
-}

package/src/functions/linear/linear.test.ts

@@ -1,83 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Toolkit from '@effect/ai/Toolkit';
-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 Layer from 'effect/Layer';
-
-import { AiService } from '@dxos/ai';
-import { AiServiceTestingPreset } from '@dxos/ai/testing';
-import { makeToolExecutionServiceFromFunctions, makeToolResolverFromFunctions } from '@dxos/assistant';
-import { Obj, Query } from '@dxos/echo';
-import { TestHelpers } from '@dxos/effect';
-import { CredentialsService, DatabaseService, FunctionInvocationService } from '@dxos/functions';
-import { TracingServiceExt } from '@dxos/functions-runtime';
-import {
-  FunctionInvocationServiceLayerTestMocked,
-  TestDatabaseLayer,
-  testStoragePath,
-} from '@dxos/functions-runtime/testing';
-import { Person, Project, Task } from '@dxos/types';
-
-import { LINEAR_ID_KEY, default as fetchLinearIssues } from './sync-issues';
-
-const TestLayer = Layer.mergeAll(
-  AiService.model('@anthropic/claude-opus-4-0'),
-  makeToolResolverFromFunctions([], Toolkit.make()),
-  makeToolExecutionServiceFromFunctions(Toolkit.make() as any, Layer.empty as any),
-).pipe(
-  Layer.provideMerge(
-    Layer.mergeAll(
-      AiServiceTestingPreset('direct'),
-      TestDatabaseLayer({
-        // indexing: { vector: true },
-        types: [Task.Task, Person.Person, Project.Project],
-        storagePath: testStoragePath({ name: 'feed-test-13' }),
-      }),
-      CredentialsService.layerConfig([{ service: 'linear.app', apiKey: Config.redacted('LINEAR_API_KEY') }]),
-      FunctionInvocationServiceLayerTestMocked({ functions: [fetchLinearIssues] }).pipe(
-        Layer.provideMerge(TracingServiceExt.layerLogInfo()),
-      ),
-      FetchHttpClient.layer,
-    ),
-  ),
-);
-
-describe('Linear', { timeout: 600_000 }, () => {
-  it.effect(
-    'sync',
-    Effect.fnUntraced(
-      function* (_) {
-        yield* DatabaseService.flush({ indexes: true });
-
-        yield* FunctionInvocationService.invokeFunction(fetchLinearIssues, {
-          team: '1127c63a-6f77-4725-9229-50f6cd47321c',
-        });
-
-        const { objects: persons } = yield* DatabaseService.runQuery(Query.type(Person.Person));
-        console.log('people', {
-          count: persons.length,
-          people: persons.map((_) => `(${_.id}) ${Obj.getLabel(_)} [${Obj.getKeys(_, LINEAR_ID_KEY)[0]?.id}]`),
-        });
-        const { objects: projects } = yield* DatabaseService.runQuery(Query.type(Project.Project));
-        console.log('projects', {
-          count: projects.length,
-          projects: projects.map((_) => `(${_.id}) ${Obj.getLabel(_)} [${Obj.getKeys(_, LINEAR_ID_KEY)[0]?.id}]`),
-        });
-        const { objects: tasks } = yield* DatabaseService.runQuery(Query.type(Task.Task));
-        console.log('tasks', {
-          count: tasks.length,
-          tasks: tasks.map((_) => `(${_.id}) ${Obj.getLabel(_)} [${Obj.getKeys(_, LINEAR_ID_KEY)[0]?.id}]`),
-        });
-
-        yield* DatabaseService.flush({ indexes: true });
-      },
-      Effect.provide(TestLayer),
-      TestHelpers.taggedTest('sync'),
-    ),
-  );
-});

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

@@ -1,75 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
-
-import { ArtifactId } from '@dxos/assistant';
-import { Obj, Relation } from '@dxos/echo';
-import { DatabaseService, TracingService, defineFunction } from '@dxos/functions';
-import { log } from '@dxos/log';
-import { Markdown } from '@dxos/plugin-markdown/types';
-import { HasSubject } from '@dxos/types';
-import { trim } from '@dxos/util';
-
-export default defineFunction({
-  key: 'dxos.org/function/research/document-create',
-  name: 'Create research document',
-  description: 'Creates a note summarizing the research.',
-  inputSchema: Schema.Struct({
-    subject: ArtifactId.annotations({
-      description: trim`
-        ID of the object (organization, contact, etc.) for which the research was performed. 
-      `,
-    }),
-    name: Schema.String.annotations({
-      description: 'Name of the document.',
-    }),
-    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).
-      `,
-    }),
-  }),
-  outputSchema: Schema.Struct({
-    document: ArtifactId.annotations({
-      description: 'DXN of the created document.',
-    }),
-  }),
-  handler: Effect.fnUntraced(function* ({ data: { subject, name, content } }) {
-    log.info('Creating research document', { subject, name, content });
-
-    // TODO(burdon): Auto flush before and after calling function?
-    yield* DatabaseService.flush({ indexes: true });
-    yield* TracingService.emitStatus({ message: 'Creating research document...' });
-
-    // TODO(burdon): Type check.
-    const target = (yield* DatabaseService.resolve(ArtifactId.toDXN(subject))) as Obj.Any;
-
-    // Create document.
-    const object = yield* DatabaseService.add(
-      Markdown.make({
-        name,
-        content,
-      }),
-    );
-
-    // Create relation.
-    yield* DatabaseService.add(
-      Relation.make(HasSubject.HasSubject, {
-        [Relation.Source]: object,
-        [Relation.Target]: target,
-        completedAt: new Date().toISOString(),
-      }),
-    );
-
-    yield* DatabaseService.flush({ indexes: true });
-    log.info('Created research document', { subject, object });
-
-    return {
-      document: Obj.getDXN(object).toString(),
-    };
-  }),
-});

package/src/functions/research/index.ts

@@ -1,14 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { default as create$ } from './document-create';
-import { default as research$ } from './research';
-
-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);
-  }),
-);