@dxos/assistant-toolkit 0.8.4-main.9be5663bfe → 0.8.4-main.abd8ff62ef

package/src/blueprints/design/blueprint.test.ts

@@ -1,97 +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, ConsolePrinter } from '@dxos/ai';
-import { GenericToolkit } from '@dxos/ai';
-import { AiServiceTestingPreset } from '@dxos/ai/testing';
-import { AiConversation, GenerationObserver, ToolExecutionServices } from '@dxos/assistant';
-import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { Database, Feed } from '@dxos/echo';
-import { acquireReleaseResource } from '@dxos/effect';
-import { TestHelpers } from '@dxos/effect/testing';
-import { Trace, TracingService } from '@dxos/functions';
-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 { trim } from '@dxos/util';
-
-import { MarkdownHandlers } from '../markdown';
-import DesignBlueprint from './blueprint';
-
-describe('Design Blueprint', { timeout: 120_000 }, () => {
-  it.scoped(
-    'design blueprint',
-    Effect.fn(
-      function* ({ expect }) {
-        const observer = GenerationObserver.fromPrinter(new ConsolePrinter());
-        const feed = Feed.make();
-        yield* Database.add(feed);
-        const runtime = yield* Effect.runtime<Feed.FeedService>();
-        const conversation = yield* acquireReleaseResource(() => new AiConversation({ feed, runtime }));
-
-        const blueprint = DesignBlueprint.make();
-        yield* Database.add(blueprint);
-        yield* Effect.promise(() =>
-          conversation.context.bind({
-            blueprints: [Ref.make(blueprint)],
-          }),
-        );
-
-        const artifact = yield* Database.add(Markdown.make({ content: 'Hello, world!' }));
-        let prevContent = artifact.content;
-
-        {
-          const prompt = trim`
-            I want to design a new feature for our product.
-
-            We need to add a user profile system with the following requirements:
-            1. Users should be able to create and edit their profiles
-            2. Profile should include basic info like name, bio, avatar
-            3. Users can control privacy settings for their profile
-            4. Profile should show user's activity history
-            5. Need to consider data storage and security implications
-
-            Let's capture the key design decisions in our spec in ${Obj.getDXN(artifact)}
-          `;
-
-          yield* conversation.createRequest({ prompt, observer });
-          log.info('spec', { doc: artifact });
-          expect(artifact.content).not.toBe(prevContent);
-          prevContent = artifact.content;
-        }
-
-        {
-          const prompt = trim`
-            I want this to be built on top of Durable Objects and SQLite database.
-            Adjust the spec to reflect this.
-          `;
-
-          yield* conversation.createRequest({ observer, prompt });
-          expect(artifact.content).not.toBe(prevContent);
-          prevContent = artifact.content;
-        }
-      },
-      Effect.provide(
-        Layer.mergeAll(ToolExecutionServices, AiService.model('@anthropic/claude-3-5-sonnet-20241022')).pipe(
-          Layer.provideMerge(TestDatabaseLayer({ types: [Text.Text, Markdown.Document, Blueprint.Blueprint] })),
-          Layer.provideMerge(Layer.mergeAll(GenericToolkit.providerEmpty, AiServiceTestingPreset('direct'))),
-          Layer.provideMerge(
-            FunctionInvocationServiceLayerTestMocked({
-              functions: MarkdownHandlers,
-            }).pipe(Layer.provideMerge(TracingService.layerNoop)),
-          ),
-          Layer.provideMerge(Trace.writerLayerNoop),
-        ),
-      ),
-      TestHelpers.provideTestContext,
-      TestHelpers.taggedTest('llm'),
-    ),
-  );
-});

package/src/blueprints/design/blueprint.ts

@@ -1,41 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type AppCapabilities } from '@dxos/app-toolkit';
-import { Blueprint } from '@dxos/blueprints';
-import { Ref } from '@dxos/echo';
-import { Text } from '@dxos/schema';
-import { trim } from '@dxos/util';
-
-import { MarkdownCreate, MarkdownRead, MarkdownUpdate } from '../markdown';
-
-const BLUEPRINT_KEY = 'org.dxos.blueprint.design';
-
-const instructions = trim`
-  You manage a design spec based on the conversation.
-  The design spec is a markdown document that is used to record the tasks.
-  The design spec document follows a hierarchical structure, with nested markdown bulleted sections.
-  Use the appropriate tools to read and write the design spec document.
-  Maintain the document so that it can convey all relevant points from the conversation.
-  When replying to the user, be terse with your comments about design doc handling.
-  Do not announce when you read or write the design spec document.
-`;
-
-const make = () =>
-  Blueprint.make({
-    key: BLUEPRINT_KEY,
-    name: 'Design Spec',
-    description: 'Preserve the conversation in the design spec.',
-    instructions: {
-      source: Ref.make(Text.make(instructions)),
-    },
-    tools: Blueprint.toolDefinitions({ operations: [MarkdownCreate, MarkdownRead, MarkdownUpdate] }),
-  });
-
-const blueprint: AppCapabilities.BlueprintDefinition = {
-  key: BLUEPRINT_KEY,
-  make,
-};
-
-export default blueprint;

package/src/blueprints/design/index.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export { default as DesignBlueprint } from './blueprint';

package/src/blueprints/markdown/blueprint.ts

@@ -1,36 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type AppCapabilities } from '@dxos/app-toolkit';
-import { Blueprint } from '@dxos/blueprints';
-import { Ref } from '@dxos/echo';
-import { Text } from '@dxos/schema';
-import { trim } from '@dxos/util';
-
-import { Create, Read, Update } from './functions';
-
-const BLUEPRINT_KEY = 'org.dxos.blueprint.markdown';
-
-const instructions = trim`
-  You read, write & create markdown documents.
-`;
-
-const make = () =>
-  Blueprint.make({
-    key: BLUEPRINT_KEY,
-    name: 'Markdown',
-    description: 'Work with markdown documents.',
-    agentCanEnable: true,
-    instructions: {
-      source: Ref.make(Text.make(instructions)),
-    },
-    tools: Blueprint.toolDefinitions({ operations: [Create, Read, Update] }),
-  });
-
-const blueprint: AppCapabilities.BlueprintDefinition = {
-  key: BLUEPRINT_KEY,
-  make,
-};
-
-export default blueprint;

package/src/blueprints/markdown/functions/create.ts

@@ -1,20 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Database, Ref } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
-import { Markdown } from '@dxos/plugin-markdown/types';
-
-import { Create } from './definitions';
-
-export default Create.pipe(
-  Operation.withHandler(
-    Effect.fn(function* ({ name, content }) {
-      const doc = yield* Database.add(Markdown.make({ name, content }));
-      return { document: Ref.make(doc) };
-    }),
-  ),
-);

package/src/blueprints/markdown/functions/definitions.ts

@@ -1,80 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Schema from 'effect/Schema';
-
-import { Database, Ref } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
-import { Markdown } from '@dxos/plugin-markdown/types';
-
-export const Create = Operation.make({
-  meta: {
-    key: 'org.dxos.function.markdown.create',
-    name: 'Create markdown document',
-    description: 'Creates a new markdown document.',
-  },
-  input: Schema.Struct({
-    name: Schema.String.annotations({
-      description: 'Name of the document.',
-    }),
-    content: Schema.String.annotations({
-      description: 'Content of the document.',
-    }),
-  }),
-  output: Schema.Void,
-  services: [Database.Service],
-});
-
-export const Read = Operation.make({
-  meta: {
-    key: 'org.dxos.function.markdown.read',
-    name: 'Read markdown document',
-    description:
-      'Read markdown document. Note that result is a snapshot in time, and might have changed since the document was last read.',
-  },
-  input: Schema.Struct({
-    document: Ref.Ref(Markdown.Document).annotations({
-      description: 'The document to read.',
-    }),
-  }),
-  output: Schema.Struct({
-    content: Schema.String,
-  }),
-  services: [Database.Service],
-});
-
-export const Open = Operation.make({
-  meta: {
-    key: 'org.dxos.function.markdown.open',
-    name: 'Open',
-    description: 'Opens and reads the contents of a markdown document.',
-  },
-  input: Schema.Struct({
-    doc: Ref.Ref(Markdown.Document).annotations({
-      description: 'The ID of the markdown document.',
-    }),
-  }),
-  output: Schema.Struct({
-    content: Schema.String,
-  }),
-  services: [Database.Service],
-});
-
-export const Update = Operation.make({
-  meta: {
-    key: 'org.dxos.function.markdown.update',
-    name: 'Update markdown',
-    description: 'Updates the entire contents of the markdown document.',
-  },
-  input: Schema.Struct({
-    doc: Ref.Ref(Markdown.Document).annotations({
-      description: 'The ID of the document to write.',
-    }),
-    content: Schema.String.annotations({
-      description: 'New content to write to the document.',
-    }),
-  }),
-  output: Schema.Void,
-  services: [Database.Service],
-});

package/src/blueprints/markdown/functions/index.ts

@@ -1,13 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { OperationHandlerSet } from '@dxos/operation';
-
-export * from './definitions';
-
-export const MarkdownHandlers = OperationHandlerSet.lazy(
-  () => import('./create'),
-  () => import('./read'),
-  () => import('./update'),
-);

package/src/blueprints/markdown/functions/read.ts

@@ -1,22 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Database } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
-
-import { Read } from './definitions';
-
-export default Read.pipe(
-  Operation.withHandler(
-    Effect.fn(function* ({ document }) {
-      const { content } = yield* document.pipe(
-        Database.load,
-        Effect.flatMap((doc) => doc.content.pipe(Database.load)),
-      );
-      return { content };
-    }),
-  ),
-);

package/src/blueprints/markdown/functions/update.ts

@@ -1,22 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Database, Obj } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
-
-import { Update } from './definitions';
-
-export default Update.pipe(
-  Operation.withHandler(
-    Effect.fn(function* ({ doc, content }) {
-      const document = yield* Database.load(doc);
-      const text = yield* Database.load(document.content);
-      Obj.change(text, (text) => {
-        text.content = content;
-      });
-    }),
-  ),
-);

package/src/blueprints/markdown/index.ts

@@ -1,11 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export { default as MarkdownBlueprint } from './blueprint';
-export {
-  Create as MarkdownCreate,
-  Read as MarkdownRead,
-  Update as MarkdownUpdate,
-  MarkdownHandlers,
-} from './functions';

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

@@ -1,122 +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 { GenericToolkit } from '@dxos/ai';
-import { AiServiceTestingPreset } from '@dxos/ai/testing';
-import { AiConversation, ToolExecutionServices } from '@dxos/assistant';
-import { Blueprint } from '@dxos/blueprints';
-import { Obj, Ref } from '@dxos/echo';
-import { Database, Feed } from '@dxos/echo';
-import { acquireReleaseResource } from '@dxos/effect';
-import { TestHelpers } from '@dxos/effect/testing';
-import { Trace, 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 { trim } from '@dxos/util';
-
-import { type TestStep, runSteps } from '../testing';
-import PlanningOldBlueprint from './blueprint';
-import { TaskHandlers } from './functions';
-
-describe('Planning Blueprint', { timeout: 120_000 }, () => {
-  const blueprint = PlanningOldBlueprint.make();
-  it.scoped(
-    'planning blueprint',
-    Effect.fn(
-      function* ({ expect }) {
-        const feed = Feed.make();
-        yield* Database.add(feed);
-        const runtime = yield* Effect.runtime<Feed.FeedService>();
-        const conversation = yield* acquireReleaseResource(() => new AiConversation({ feed, runtime }));
-
-        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: TaskHandlers }).pipe(
-              Layer.provideMerge(Layer.mergeAll(TracingService.layerNoop, Trace.writerLayerNoop)),
-            ),
-          ),
-          Layer.provideMerge(FunctionImplementationResolver.layerTest({ functions: TaskHandlers })),
-          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-old/blueprint.ts

@@ -1,106 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type AppCapabilities } from '@dxos/app-toolkit';
-import { Blueprint } from '@dxos/blueprints';
-import { Ref } from '@dxos/echo';
-import { Text } from '@dxos/schema';
-import { trim } from '@dxos/util';
-
-import { ReadTasks, UpdateTasks } from './functions';
-
-const BLUEPRINT_KEY = 'org.dxos.blueprint.planning';
-
-/**
- * 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
-`;
-
-const make = () =>
-  Blueprint.make({
-    key: BLUEPRINT_KEY,
-    name: 'Planning',
-    description: 'Plans and tracks complex tasks with artifact management.',
-    instructions: {
-      source: Ref.make(Text.make(instructions)),
-    },
-    tools: Blueprint.toolDefinitions({ operations: [ReadTasks, UpdateTasks] }),
-  });
-
-const blueprint: AppCapabilities.BlueprintDefinition = {
-  key: BLUEPRINT_KEY,
-  make,
-};
-
-export default blueprint;

package/src/blueprints/planning-old/functions/definitions.ts

@@ -1,53 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Schema from 'effect/Schema';
-
-import { Database, Ref } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
-import { Markdown } from '@dxos/plugin-markdown/types';
-
-export const ReadTasks = Operation.make({
-  meta: {
-    key: 'org.dxos.function.markdown.read-tasks',
-    name: 'Read',
-    description: 'Read markdown tasks.',
-  },
-  input: Schema.Struct({
-    doc: Ref.Ref(Markdown.Document).annotations({
-      description: 'The ID of the document to read.',
-    }),
-  }),
-  output: Schema.Struct({
-    content: Schema.String,
-  }),
-  services: [Database.Service],
-});
-
-export const UpdateTasks = Operation.make({
-  meta: {
-    key: 'org.dxos.function.markdown.update-tasks',
-    name: 'Update markdown',
-    description: 'Creates and updates tasks in markdown documents.',
-  },
-  input: Schema.Struct({
-    doc: Ref.Ref(Markdown.Document).annotations({
-      description: 'The ID of the document to update.',
-    }),
-    operations: Schema.optional(
-      Schema.Array(
-        Schema.Any.annotations({
-          description: 'Task operations to apply.',
-        }),
-      ),
-    ),
-  }),
-  output: Schema.Struct({
-    content: Schema.String,
-    numberedContent: Schema.String.annotations({
-      description: 'Content with line numbers for agent reference.',
-    }),
-  }),
-  services: [Database.Service],
-});

package/src/blueprints/planning-old/functions/index.ts

@@ -1,12 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { OperationHandlerSet } from '@dxos/operation';
-
-export * from './definitions';
-
-export const TaskHandlers = OperationHandlerSet.lazy(
-  () => import('./read'),
-  () => import('./update'),
-);

package/src/blueprints/planning-old/functions/read.ts

@@ -1,24 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Database } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
-
-import { ReadTasks } from './definitions';
-
-export default ReadTasks.pipe(
-  Operation.withHandler(
-    Effect.fn(function* ({ doc }) {
-      const document = yield* Database.load(doc);
-
-      const { content } = yield* Database.load(document.content);
-      const lines = content.split('\n');
-      const len = String(lines.length).length;
-      const numbered = lines.map((line, i) => `${String(i + 1).padStart(len, ' ')}. ${line}`).join('\n');
-      return { content: numbered };
-    }),
-  ),
-);