@dxos/assistant-toolkit 0.8.4-main.66e292d → 0.8.4-main.6fa680abb7

package/dist/types/src/toolkits/SystemToolkit.d.ts

@@ -1,67 +0,0 @@
-import * as Tool from '@effect/ai/Tool';
-import * as Toolkit from '@effect/ai/Toolkit';
-import type * as Layer from 'effect/Layer';
-import * as Schema from 'effect/Schema';
-import { DatabaseService } from '@dxos/functions';
-export declare const SystemToolkit: Toolkit.Toolkit<{
-    readonly "schema-list": Tool.Tool<"schema-list", {
-        readonly parameters: Schema.Struct<{
-            limit: typeof Schema.Number;
-        }>;
-        readonly success: typeof Schema.Any;
-        readonly failure: typeof Schema.Never;
-    }, DatabaseService>;
-    readonly "schema-add": Tool.Tool<"schema-add", {
-        readonly parameters: Schema.Struct<{
-            name: typeof Schema.String;
-            typename: typeof Schema.String;
-            jsonSchema: typeof Schema.Any;
-        }>;
-        readonly success: typeof Schema.Any;
-        readonly failure: typeof Schema.Never;
-    }, DatabaseService>;
-    readonly "object-create": Tool.Tool<"object-create", {
-        readonly parameters: Schema.Struct<{
-            typename: typeof Schema.String;
-            data: typeof Schema.Any;
-        }>;
-        readonly success: typeof Schema.Any;
-        readonly failure: typeof Schema.Never;
-    }, DatabaseService>;
-    readonly "object-remove": Tool.Tool<"object-remove", {
-        readonly parameters: Schema.Struct<{
-            id: Schema.Schema<string, string, never>;
-        }>;
-        readonly success: typeof Schema.Any;
-        readonly failure: typeof Schema.Never;
-    }, DatabaseService>;
-    readonly "relation-create": Tool.Tool<"relation-create", {
-        readonly parameters: Schema.Struct<{
-            typename: typeof Schema.String;
-            source: Schema.Schema<string, string, never>;
-            target: Schema.Schema<string, string, never>;
-            data: Schema.SchemaClass<any, any, never>;
-        }>;
-        readonly success: typeof Schema.Any;
-        readonly failure: typeof Schema.Never;
-    }, DatabaseService>;
-    readonly "tag-add": Tool.Tool<"tag-add", {
-        readonly parameters: Schema.Struct<{
-            tagId: Schema.Schema<string, string, never>;
-            objectId: Schema.Schema<string, string, never>;
-        }>;
-        readonly success: typeof Schema.Any;
-        readonly failure: typeof Schema.Never;
-    }, DatabaseService>;
-    readonly "tag-remove": Tool.Tool<"tag-remove", {
-        readonly parameters: Schema.Struct<{
-            tagId: Schema.Schema<string, string, never>;
-            objectId: Schema.Schema<string, string, never>;
-        }>;
-        readonly success: typeof Schema.Any;
-        readonly failure: typeof Schema.Never;
-    }, DatabaseService>;
-}>;
-export declare const tools: ("schema-list" | "schema-add" | "object-create" | "object-remove" | "relation-create" | "tag-add" | "tag-remove")[];
-export declare const layer: () => Layer.Layer<Tool.Handler<any>, never, never>;
-//# sourceMappingURL=SystemToolkit.d.ts.map

package/dist/types/src/toolkits/SystemToolkit.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"SystemToolkit.d.ts","sourceRoot":"","sources":["../../../../src/toolkits/SystemToolkit.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,IAAI,MAAM,iBAAiB,CAAC;AACxC,OAAO,KAAK,OAAO,MAAM,oBAAoB,CAAC;AAE9C,OAAO,KAAK,KAAK,KAAK,MAAM,cAAc,CAAC;AAE3C,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AAIxC,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAGlD,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAoIzB,CAAC;AAEF,eAAO,MAAM,KAAK,qHAAmC,CAAC;AAEtD,eAAO,MAAM,KAAK,QAAO,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,KAAK,CA8EhE,CAAC"}

package/src/blueprints/discord/discord-blueprint.ts

@@ -1,34 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { ToolId } from '@dxos/ai';
-import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { Text } from '@dxos/schema';
-import { trim } from '@dxos/util';
-
-import { Discord } from '../../functions';
-
-/**
- * Agent prompt instructions for managing hierarchical task lists.
- */
-const instructions = trim`
-  You are able to fetch messages from Discord servers.
-
-  Known servers:
-
-  DXOS serverId: 837138313172353095
-`;
-
-export const blueprint = Obj.make(Blueprint.Blueprint, {
-  key: 'dxos.org/blueprint/discord',
-  name: 'Discord',
-  description: 'Discord integration.',
-  instructions: {
-    source: Ref.make(Text.make(instructions)),
-  },
-  tools: [ToolId.make(Discord.fetch.key)],
-});
-
-export default blueprint;

package/src/blueprints/linear/linear-blueprint.ts

@@ -1,35 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { ToolId } from '@dxos/ai';
-import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { Text } from '@dxos/schema';
-import { trim } from '@dxos/util';
-
-import { Linear } from '../../functions';
-
-/**
- * Agent prompt instructions for managing hierarchical task lists.
- */
-const instructions = trim`
-  You are able to sync Linear workspaces.
-  Sometimes sync does not complete in one go and you need to call the function again.
-
-  Known workspaces:
-
-  DXOS teamId: 1127c63a-6f77-4725-9229-50f6cd47321c
-`;
-
-export const blueprint = Obj.make(Blueprint.Blueprint, {
-  key: 'dxos.org/blueprint/linear',
-  name: 'Linear',
-  description: 'Syncs Linear workspaces.',
-  instructions: {
-    source: Ref.make(Text.make(instructions)),
-  },
-  tools: [Linear.sync].map((tool) => ToolId.make(tool.key)),
-});
-
-export default blueprint;

package/src/blueprints/websearch/websearch-blueprint.ts

@@ -1,20 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { ToolId } from '@dxos/ai';
-import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { Text } from '@dxos/schema';
-
-const blueprint: Blueprint.Blueprint = Obj.make(Blueprint.Blueprint, {
-  key: 'dxos.org/blueprint/web-search',
-  name: 'Web Search',
-  description: 'Search the web.',
-  instructions: {
-    source: Ref.make(Text.make()),
-  },
-  tools: [ToolId.make('AnthropicWebSearch')],
-});
-
-export default blueprint;

package/src/experimental/feed.test.ts

@@ -1,111 +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 * as Redacted from 'effect/Redacted';
-
-import { AiService } from '@dxos/ai';
-import { AiServiceTestingPreset, EXA_API_KEY } from '@dxos/ai/testing';
-import { makeToolExecutionServiceFromFunctions, makeToolResolverFromFunctions } from '@dxos/assistant';
-import { TestHelpers } from '@dxos/effect';
-import { CredentialsService, FunctionInvocationService } from '@dxos/functions';
-import { TracingServiceExt } from '@dxos/functions-runtime';
-import {
-  FunctionInvocationServiceLayerTestMocked,
-  TestDatabaseLayer,
-  testStoragePath,
-} from '@dxos/functions-runtime/testing';
-
-import { Discord, Linear } from '../functions';
-
-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: [],
-        storagePath: testStoragePath({ name: 'feed-test' }),
-      }),
-      CredentialsService.layerConfig([
-        { service: 'exa.ai', apiKey: Config.succeed(Redacted.make(EXA_API_KEY)) },
-        { service: 'discord.com', apiKey: Config.redacted('DISCORD_TOKEN') },
-        { service: 'linear.app', apiKey: Config.redacted('LINEAR_API_KEY') },
-      ]),
-      FunctionInvocationServiceLayerTestMocked({ functions: [Linear.sync, Discord.fetch] }).pipe(
-        Layer.provideMerge(TracingServiceExt.layerLogInfo()),
-      ),
-      FetchHttpClient.layer,
-    ),
-  ),
-);
-
-describe('Feed', { timeout: 600_000 }, () => {
-  it.effect(
-    'fetch discord messages',
-    Effect.fnUntraced(
-      function* (_) {
-        // const messages = yield* LocalFunctionExecutionService.invokeFunction(fetchDiscordMessages, {
-        //   serverId: '837138313172353095',
-        //   // channelId: '1404487604761526423',
-        //   after: Date.now() / 1000 - 128 * 3600,
-        // });
-        // for (const message of messages) {
-        //   console.log(message.sender.name, message.blocks.find((block) => block._tag === 'text')?.text);
-        // }
-        // console.log(`Fetched ${messages.length} messages`);
-
-        // const result = yield* AiSession.run({
-        //   history: [
-        //     Obj.make(Message.Message, {
-        //       created: new Date().toISOString(),
-        //       sender: { role: 'user' },
-        //       blocks: messages
-        //         .map(
-        //           (message) =>
-        //             ({
-        //               _tag: 'text',
-        //               text: message.sender.name + ': ' + message.blocks.find((block) => block._tag === 'text')?.text,
-        //             }) as const,
-        //         )
-        //         .filter((block) => block._tag === 'text' && block.text.trim().length > 0),
-        //     }),
-        //   ],
-        //   prompt: 'Summarize the messages.',
-        //   system: 'Summarize the messages.',
-        // }).pipe(Effect.provide(AiService.model('@anthropic/claude-3-5-haiku-latest')));
-        // console.log(result);
-
-        const linearIssues = yield* FunctionInvocationService.invokeFunction(Linear.sync, {
-          team: '1127c63a-6f77-4725-9229-50f6cd47321c',
-        });
-        console.log(linearIssues);
-
-        // const result = yield* AiSession.run({
-        //   history: [
-        //     Obj.make(Message.Message, {
-        //       created: new Date().toISOString(),
-        //       sender: { role: 'user' },
-        //       blocks: [{ _tag: 'text', text: JSON.stringify(linearIssues) }],
-        //     }),
-        //   ],
-        //   prompt: 'Summarize the whats new.',
-        //   system: 'Summarize the whats new. Reference specific people by name.',
-        // }).pipe(Effect.provide(AiService.model('@anthropic/claude-sonnet-4-0')));
-        // console.log(result.at(-1)?.blocks.find((block) => block._tag === 'text')?.text);
-      },
-      Effect.provide(TestLayer),
-      TestHelpers.taggedTest('llm'),
-    ),
-  );
-});

package/src/functions/discord/index.ts

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

package/src/functions/document/index.ts

@@ -1,12 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { default as read$ } from './read';
-import { default as update$ } from './update';
-
-/** @deprecated */
-export namespace Document {
-  export const read = read$;
-  export const update = update$;
-}

package/src/functions/document/read.ts

@@ -1,29 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
-
-import { ArtifactId } from '@dxos/assistant';
-import { DatabaseService, defineFunction } from '@dxos/functions';
-import { Markdown } from '@dxos/plugin-markdown/types';
-
-export default defineFunction({
-  key: 'dxos.org/function/markdown/read',
-  name: 'Read markdown document',
-  description: 'Read markdown document.',
-  inputSchema: Schema.Struct({
-    id: ArtifactId.annotations({
-      description: 'The ID of the document to read.',
-    }),
-  }),
-  outputSchema: Schema.Struct({
-    content: Schema.String,
-  }),
-  handler: Effect.fn(function* ({ data: { id } }) {
-    const doc = yield* DatabaseService.resolve(ArtifactId.toDXN(id), Markdown.Document);
-    const { content } = yield* DatabaseService.load(doc.content);
-    return { content };
-  }),
-});

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/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,49 +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 { SystemTypeAnnotation } from '@dxos/echo/internal';
-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',
-  }),
-  SystemTypeAnnotation.set(true),
-);
-
-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);
-  }),
-);