@dxos/assistant-toolkit 0.8.4-main.3eb6e50203 → 0.8.4-main.3fbcb4aa9b

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

@@ -1,33 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { ToolId } from '@dxos/ai';
-import { Blueprint } from '@dxos/blueprints';
-import { 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 = Blueprint.make({
-  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)),
-});

package/src/blueprints/markdown/index.ts

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { blueprint } from './markdown-blueprint';
-
-export default blueprint;

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

@@ -1,24 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Blueprint } from '@dxos/blueprints';
-import { Ref } from '@dxos/echo';
-import { Text } from '@dxos/schema';
-import { trim } from '@dxos/util';
-
-import { Document } from '../../functions';
-
-const instructions = trim`
-  You read, write & create markdown documents.
-`;
-
-export const blueprint = Blueprint.make({
-  key: 'dxos.org/blueprint/markdown',
-  name: 'Markdown',
-  description: 'Work with markdown documents.',
-  instructions: {
-    source: Ref.make(Text.make(instructions)),
-  },
-  tools: Blueprint.toolDefinitions({ functions: [Document.read, Document.update, Document.create] }),
-});

package/src/blueprints/planning/planning-blueprint.test.ts

@@ -1,120 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { describe, it } from '@effect/vitest';
-import * as Effect from 'effect/Effect';
-import * as Layer from 'effect/Layer';
-
-import { AiService } from '@dxos/ai';
-import { AiServiceTestingPreset } from '@dxos/ai/testing';
-import { AiConversation, type ContextBinding, GenericToolkit, ToolExecutionServices } from '@dxos/assistant';
-import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { Database } from '@dxos/echo';
-import { acquireReleaseResource } from '@dxos/effect';
-import { TestHelpers } from '@dxos/effect/testing';
-import { QueueService, TracingService } from '@dxos/functions';
-import { FunctionImplementationResolver } from '@dxos/functions-runtime';
-import { FunctionInvocationServiceLayerTestMocked, TestDatabaseLayer } from '@dxos/functions-runtime/testing';
-import { log } from '@dxos/log';
-import { Markdown } from '@dxos/plugin-markdown/types';
-import { Text } from '@dxos/schema';
-import { type Message } from '@dxos/types';
-import { trim } from '@dxos/util';
-
-import { Tasks } from '../../functions';
-import { type TestStep, runSteps } from '../testing';
-
-import { blueprint } from './planning-blueprint';
-
-describe('Planning Blueprint', { timeout: 120_000 }, () => {
-  it.scoped(
-    'planning blueprint',
-    Effect.fn(
-      function* ({ expect }) {
-        const queue = yield* QueueService.createQueue<Message.Message | ContextBinding>();
-        const conversation = yield* acquireReleaseResource(() => new AiConversation({ queue }));
-
-        yield* Database.add(blueprint);
-        yield* Effect.promise(() =>
-          conversation.context.bind({
-            blueprints: [Ref.make(blueprint)],
-          }),
-        );
-
-        const artifact = yield* Database.add(Markdown.make());
-        let prevContent = artifact.content;
-        const matchList =
-          ({ includes = [], excludes = [] }: { includes: RegExp[]; excludes?: RegExp[] }) =>
-          async () => {
-            const { content } = await artifact.content.load();
-            log.info('spec', { doc: artifact.content.target?.content });
-            expect(content).not.toBe(prevContent);
-            const text = content.toLowerCase();
-            for (const include of includes) {
-              expect(text.match(include), `does not include: ${include}`).toBeTruthy();
-            }
-            for (const exclude of excludes) {
-              expect(text.match(exclude), `includes: ${exclude}`).toBeFalsy();
-            }
-            prevContent = artifact.content;
-          };
-
-        const system = 'You are a helpful assistant.';
-        const steps: TestStep[] = [
-          {
-            system,
-            prompt: trim`
-              I'm building a shelf.
-              Maintain a shopping list here: ${Obj.getDXN(artifact)}
-              I need a hammer, nails, and a saw.
-            `,
-            test: matchList({
-              includes: [/- \[.+hammer/, /- \[.+nails/, /- \[.+saw/],
-            }),
-          },
-          {
-            system,
-            prompt: trim`
-              I will need a board too.
-            `,
-            test: matchList({
-              includes: [/- \[.+board/],
-            }),
-          },
-          {
-            system,
-            prompt: trim`
-              Actually I'm going to use screws and a screwdriver.
-            `,
-            test: matchList({
-              includes: [/- \[.+screwdriver/, /- \[.+screws/],
-              excludes: [/- \[.+hammer/, /- \[.+nails/],
-            }),
-          },
-        ];
-
-        yield* runSteps(conversation, steps);
-      },
-      Effect.provide(
-        Layer.mergeAll(
-          TestDatabaseLayer({ types: [Text.Text, Markdown.Document, Blueprint.Blueprint] }),
-          ToolExecutionServices,
-          AiService.model('@anthropic/claude-3-5-sonnet-20241022'),
-        ).pipe(
-          Layer.provideMerge(
-            FunctionInvocationServiceLayerTestMocked({ functions: [Tasks.read, Tasks.update] }).pipe(
-              Layer.provideMerge(TracingService.layerNoop),
-            ),
-          ),
-          Layer.provideMerge(FunctionImplementationResolver.layerTest({ functions: [Tasks.read, Tasks.update] })),
-          Layer.provideMerge(TestDatabaseLayer({ types: [Text.Text, Markdown.Document, Blueprint.Blueprint] })),
-          Layer.provideMerge(Layer.mergeAll(GenericToolkit.providerEmpty, AiServiceTestingPreset('direct'))),
-        ),
-      ),
-      TestHelpers.provideTestContext,
-      TestHelpers.taggedTest('llm'),
-    ),
-  );
-});

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

@@ -1,96 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { ToolId } from '@dxos/ai';
-import { Blueprint } from '@dxos/blueprints';
-import { Ref } from '@dxos/echo';
-import { Text } from '@dxos/schema';
-import { trim } from '@dxos/util';
-
-import { Tasks } from '../../functions';
-
-/**
- * Agent prompt instructions for managing hierarchical task lists.
- */
-const instructions = trim`
-  You are a task management agent that maintains hierarchical task lists where each line is a task.
-
-  ## Document Format
-  You will receive task lists with line numbers prefixed like:
-  
-  ${'```'}
-  1. - [ ] First main task
-  2.   - [ ] Subtask 1: Research phase
-  3.     - [x] Literature review
-  4.     - [ ] Stakeholder interviews
-  5.   - [ ] Subtask 2: Implementation
-  6.     - [ ] Setup infrastructure
-  7.     - [ ] Write core functionality
-  8. - [ ] Another main task
-  ${'```'}
-
-  ## Task Hierarchy
-  - 0 spaces: Top-level tasks
-  - 2 spaces: First-level subtasks
-  - 4 spaces: Second-level subtasks
-  - 6 spaces: Third-level subtasks (and so on)
-
-  ## Available Operations
-  You can modify the task list using these operations:
-
-  1. **insertTask(lineNumber, text, completed?, indent?)** - Insert a new task
-  2. **deleteTask(lineNumber)** - Delete a task
-  3. **updateTaskText(lineNumber, text)** - Change task description
-  4. **toggleTaskCompletion(lineNumber, completed?)** - Mark task complete/incomplete
-  5. **setTaskIndent(lineNumber, indent)** - Change task hierarchy level
-
-  ## Examples
-
-  ### Example 1: Adding a subtask
-  **User:** "Add a subtask 'Code review' under the task on line 1"
-  **Response:** \`insertTask(2, "Code review", false, 2)\`
-
-  ### Example 2: Marking a task complete
-  **User:** "Mark the task on line 3 as complete"
-  **Response:** \`toggleTaskCompletion(3, true)\`
-
-  ### Example 3: Updating task text
-  **User:** "Change the task on line 5 to 'Backend implementation'"
-  **Response:** \`updateTaskText(5, "Backend implementation")\`
-
-  ### Example 4: Creating a task hierarchy
-  **User:** "Add a main task 'Testing phase' with two subtasks"
-  **Response:**
-  \`\`\`
-  insertTask(999, "Testing phase", false, 0)
-  insertTask(999, "Unit tests", false, 2)
-  insertTask(999, "Integration tests", false, 2)
-  \`\`\`
-
-  ### Example 5: Reorganizing hierarchy
-  **User:** "Move the task on line 4 to be a main task (top level)"
-  **Response:** \`setTaskIndent(4, 0)\`
-
-  ### Example 6: Adding nested subtasks
-  **User:** "Add a sub-subtask 'Write test cases' under line 6"
-  **Response:** \`insertTask(7, "Write test cases", false, 4)\`
-
-  ## Guidelines
-  - Always reference line numbers from the numbered document you receive
-  - Use line number 999 to append to the end of the document
-  - Maintain logical hierarchy (subtasks should be indented under their parent)
-  - Use appropriate indentation levels (0, 2, 4, 6, etc. spaces)
-  - When creating subtasks, consider the parent task's completion status
-  - Be precise with task descriptions and hierarchy levels
-`;
-
-export const blueprint = Blueprint.make({
-  key: 'dxos.org/blueprint/planning',
-  name: 'Planning',
-  description: 'Plans and tracks complex tasks with artifact management.',
-  instructions: {
-    source: Ref.make(Text.make(instructions)),
-  },
-  tools: [Tasks.read, Tasks.update].map((fn) => ToolId.make(fn.key)),
-});

package/src/blueprints/research/index.ts

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { blueprint } from './research-blueprint';
-
-export default blueprint;

package/src/blueprints/research/research-blueprint.test.ts

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { describe } from '@effect/vitest';
-
-describe.runIf(process.env.DX_RUN_SLOW_TESTS === '1')('Research Blueprint', { timeout: 120_000 }, () => {});

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

@@ -1,51 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { ToolId } from '@dxos/ai';
-import { Blueprint } from '@dxos/blueprints';
-import { Ref } from '@dxos/echo';
-import { Text } from '@dxos/schema';
-import { trim } from '@dxos/util';
-
-import { Research } from '../../functions';
-
-/**
- * Agent prompt instructions for managing hierarchical task lists.
- */
-const instructions = trim`
-  {{! Research }}
-
-  You are an analyst that does research tasks using tools that scrape the web and create structured data.
-  Structured data extraction is an experimental feature -- only enable it if the user explicitly requests it in the prompt.
-  Prefer updating existing notes instead of creating new ones.
-
-  <strcutured_mode>
-    When you are done, reply with the created objects.
-    Do not print the data, instead reply with inline references to the created objects.
-    Those will be later substituted with the pills representing the created objects.
-    Print the rest of the created objects as block references after the main note.
-
-    <example>
-      Based on my research, Google was founded by @dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK464FSCKVQJAB2H662M and @dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK46K31DDW62PBW9H2ZQ
-
-      <object><dxn>dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK464FSCKVQJAB2H662M</dxn></object>
-      <object><dxn>dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK46K31DDW62PBW9H2ZQ</dxn></object>
-      <object><dxn>dxn:queue:data:B6INSIBY3CBEF4M5VZRYBCMAHQMPYK5AJ:01K24XMVHSZHS97SG1VTVQDM5Z:01K24XPK46K31DDW62PBW92333</dxn></object>
-    </example>
-  </structured_mode>
-
-  <unstructured_mode>
-    Reply normally with the text mode of the result of your research.
-  </unstructured_mode>
-`;
-
-export const blueprint = Blueprint.make({
-  key: 'dxos.org/blueprint/research',
-  name: 'Research',
-  description: 'Researches the web and creates structured data.',
-  instructions: {
-    source: Ref.make(Text.make(instructions)),
-  },
-  tools: [Research.create, Research.research].map((fn) => ToolId.make(fn.key)),
-});

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

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

package/src/chat/Chat.ts

@@ -1,45 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import * as Schema from 'effect/Schema';
-
-import { Obj, Type } from '@dxos/echo';
-import { FormInputAnnotation, LabelAnnotation } from '@dxos/echo/internal';
-import { Queue } from '@dxos/echo-db';
-
-/**
- * AI chat.
- */
-export const Chat = Schema.Struct({
-  name: Schema.String.pipe(Schema.optional),
-  queue: Type.Ref(Queue).pipe(FormInputAnnotation.set(false)),
-  // TODO(dmaretskyi): Eventually this and the message queue will be the same.
-  traceQueue: Type.Ref(Queue).pipe(FormInputAnnotation.set(false), Schema.optional),
-}).pipe(
-  Type.object({
-    typename: 'dxos.org/type/assistant/Chat',
-    version: '0.2.0',
-  }),
-  LabelAnnotation.set(['name']),
-);
-
-export interface Chat extends Schema.Schema.Type<typeof Chat> {}
-
-export const make = (props: Obj.MakeProps<typeof Chat>) => Obj.make(Chat, props);
-
-/**
- * Relation between a Chat and companion objects (e.g., artifacts).
- */
-export const CompanionTo = Schema.Struct({
-  id: Obj.ID,
-}).pipe(
-  Type.relation({
-    typename: 'dxos.org/relation/assistant/CompanionTo',
-    version: '0.1.0',
-    source: Chat,
-    target: Type.Obj,
-  }),
-);
-
-export interface CompanionTo extends Schema.Schema.Type<typeof CompanionTo> {}

package/src/chat/index.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-export * as Chat from './Chat';

package/src/experimental/feed.test.ts

@@ -1,106 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-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 { GenericToolkit, ToolExecutionServices } from '@dxos/assistant';
-import { TestHelpers } from '@dxos/effect/testing';
-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'), ToolExecutionServices).pipe(
-  Layer.provideMerge(GenericToolkit.providerEmpty),
-  Layer.provideMerge(
-    Layer.mergeAll(
-      AiServiceTestingPreset('direct'),
-      TestDatabaseLayer({
-        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/fetch-messages.test.ts

@@ -1,54 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-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 { GenericToolkit, ToolExecutionServices } from '@dxos/assistant';
-import { TestHelpers } from '@dxos/effect/testing';
-import { CredentialsService, FunctionInvocationService, TracingService } from '@dxos/functions';
-import { FunctionInvocationServiceLayerTestMocked, TestDatabaseLayer } from '@dxos/functions-runtime/testing';
-
-import { default as fetchMessages } from './fetch-messages';
-
-const TestLayer = Layer.mergeAll(AiService.model('@anthropic/claude-opus-4-0'), ToolExecutionServices).pipe(
-  Layer.provideMerge(
-    Layer.mergeAll(
-      GenericToolkit.providerEmpty,
-      AiServiceTestingPreset('direct'),
-      TestDatabaseLayer({}),
-      CredentialsService.layerConfig([{ service: 'discord.com', apiKey: Config.redacted('DISCORD_TOKEN') }]),
-      FetchHttpClient.layer,
-      FunctionInvocationServiceLayerTestMocked({ functions: [fetchMessages] }).pipe(
-        Layer.provideMerge(TracingService.layerNoop),
-      ),
-    ),
-  ),
-);
-
-const DXOS_SERVER_ID = '837138313172353095';
-
-describe('Feed', { timeout: 600_000 }, () => {
-  it.effect(
-    'fetch discord messages',
-    Effect.fnUntraced(
-      function* (_) {
-        const messages = yield* FunctionInvocationService.invokeFunction(fetchMessages, {
-          serverId: DXOS_SERVER_ID,
-          // channelId: '1404487604761526423',
-          last: '7d',
-        });
-        console.log(messages);
-      },
-      Effect.provide(TestLayer),
-      TestHelpers.provideTestContext,
-      TestHelpers.taggedTest('sync'),
-    ),
-  );
-});