@dxos/assistant-toolkit 0.8.4-main.bc674ce → 0.8.4-main.c85a9c8dae

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

@@ -0,0 +1,58 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { describe, it } from '@effect/vitest';
+import * as Effect from 'effect/Effect';
+
+import { AssistantTestLayer } from '@dxos/assistant/testing';
+import { Obj, Query } from '@dxos/echo';
+import { Database } from '@dxos/echo';
+import { TestHelpers } from '@dxos/effect/testing';
+import { FunctionInvocationService } from '@dxos/functions';
+import { Person, Pipeline, Task } from '@dxos/types';
+
+import { LINEAR_ID_KEY, default as fetchLinearIssues } from './sync-issues';
+
+const TestLayer = AssistantTestLayer({
+  functions: [fetchLinearIssues],
+  types: [Task.Task, Person.Person, Pipeline.Pipeline],
+  credentials: [{ service: 'linear.app', apiKey: process.env.LINEAR_API_KEY }],
+  tracing: 'pretty',
+});
+
+describe.skip('Linear', { timeout: 600_000 }, () => {
+  it.effect(
+    'sync',
+    Effect.fnUntraced(
+      function* (_) {
+        yield* Database.flush({ indexes: true });
+
+        yield* FunctionInvocationService.invokeFunction(fetchLinearIssues, {
+          team: '1127c63a-6f77-4725-9229-50f6cd47321c',
+        });
+
+        const persons = yield* Database.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 projects = yield* Database.runQuery(Query.type(Pipeline.Pipeline));
+        console.log('projects', {
+          count: projects.length,
+          projects: projects.map((_) => `(${_.id}) ${Obj.getLabel(_)} [${Obj.getKeys(_, LINEAR_ID_KEY)[0]?.id}]`),
+        });
+        const tasks = yield* Database.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* Database.flush({ indexes: true });
+      },
+      Effect.provide(TestLayer),
+      TestHelpers.taggedTest('sync'),
+      TestHelpers.provideTestContext,
+    ),
+  );
+});

package/src/{functions/linear → blueprints/linear/functions}/sync-issues.ts

@@ -13,10 +13,10 @@ import { Filter, Obj, Query, Ref, type Type } from '@dxos/echo';
 import { Database } from '@dxos/echo';
 import { CredentialsService, defineFunction, withAuthorization } from '@dxos/functions';
 import { log } from '@dxos/log';
-import { Person, Project, Task } from '@dxos/types';
+import { Person, Pipeline, Task } from '@dxos/types';
 
-import { syncObjects } from '../../sync';
-import { graphqlRequestBody } from '../../util';
+import { syncObjects } from '../../../sync';
+import { graphqlRequestBody } from '../../../util';
 
 const queryIssues = `
 query Issues($teamId: String!, $after: DateTimeOrDuration!) {
@@ -115,9 +115,9 @@ export default defineFunction({
 
 const getLatestUpdateTimestamp: (
   teamId: string,
-  dataType: Type.Obj.Any,
+  dataType: Type.AnyObj,
 ) => Effect.Effect<string, never, Database.Service> = Effect.fnUntraced(function* (teamId, dataType) {
-  const existingTasks = yield* Database.Service.runQuery(
+  const existingTasks = yield* Database.runQuery(
     Query.type(dataType).select(Filter.foreignKeys(dataType, [{ source: LINEAR_TEAM_ID_KEY, id: teamId }])),
   );
   return Function.pipe(
@@ -172,7 +172,7 @@ const mapLinearIssue = (issue: LinearIssue, { teamId }: { teamId: string }): Tas
     project: !issue.project
       ? undefined
       : Ref.make(
-          Project.make({
+          Pipeline.make({
             [Obj.Meta]: {
               keys: [
                 {

package/src/blueprints/linear/index.ts

@@ -2,6 +2,4 @@
 // Copyright 2025 DXOS.org
 //
 
-import { blueprint } from './linear-blueprint';
-
-export default blueprint;
+export { default as LinearBlueprint } from './blueprint';

package/src/blueprints/markdown/blueprint.ts

@@ -0,0 +1,38 @@
+//
+// 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 { MarkdownFunctions } from './functions';
+
+const BLUEPRINT_KEY = 'dxos.org/blueprint/markdown';
+
+const instructions = trim`
+  You read, write & create markdown documents.
+`;
+
+const functions = Object.values(MarkdownFunctions);
+
+const make = () =>
+  Blueprint.make({
+    key: BLUEPRINT_KEY,
+    name: 'Markdown',
+    description: 'Work with markdown documents.',
+    instructions: {
+      source: Ref.make(Text.make(instructions)),
+    },
+    tools: Blueprint.toolDefinitions({ functions }),
+  });
+
+const blueprint: AppCapabilities.BlueprintDefinition = {
+  key: BLUEPRINT_KEY,
+  functions,
+  make,
+};
+
+export default blueprint;

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

@@ -0,0 +1,29 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+import * as Schema from 'effect/Schema';
+
+import { Database, Ref } from '@dxos/echo';
+import { defineFunction } from '@dxos/functions';
+import { Markdown } from '@dxos/plugin-markdown/types';
+
+export default defineFunction({
+  key: 'dxos.org/function/markdown/create',
+  name: 'Create markdown document',
+  description: 'Creates a new markdown document.',
+  inputSchema: Schema.Struct({
+    name: Schema.String.annotations({
+      description: 'Name of the document.',
+    }),
+    content: Schema.String.annotations({
+      description: 'Content of the document.',
+    }),
+  }),
+  outputSchema: Schema.Void,
+  handler: Effect.fn(function* ({ data: { name, content } }) {
+    const doc = yield* Database.add(Markdown.make({ name, content }));
+    return { document: Ref.make(doc) };
+  }),
+});

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

@@ -0,0 +1,13 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { default as Create } from './create';
+import { default as Read } from './read';
+import { default as Update } from './update';
+
+export const MarkdownFunctions = {
+  Create,
+  Read,
+  Update,
+};

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

@@ -0,0 +1,32 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+import * as Schema from 'effect/Schema';
+
+import { Database, Ref } from '@dxos/echo';
+import { 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. Note that result is a snapshot in time, and might have changed since the document was last read.',
+  inputSchema: Schema.Struct({
+    document: Ref.Ref(Markdown.Document).annotations({
+      description: 'The document to read.',
+    }),
+  }),
+  outputSchema: Schema.Struct({
+    content: Schema.String,
+  }),
+  handler: Effect.fn(function* ({ data: { document } }) {
+    const { content } = yield* document.pipe(
+      Database.load,
+      Effect.flatMap((doc) => doc.content.pipe(Database.load)),
+    );
+    return { content };
+  }),
+});

package/src/{functions/document → blueprints/markdown/functions}/update.ts

@@ -5,8 +5,7 @@
 import * as Effect from 'effect/Effect';
 import * as Schema from 'effect/Schema';
 
-import { ArtifactId } from '@dxos/assistant';
-import { Database, Obj } from '@dxos/echo';
+import { Database, Obj, Ref } from '@dxos/echo';
 import { defineFunction } from '@dxos/functions';
 import { Markdown } from '@dxos/plugin-markdown/types';
 
@@ -15,7 +14,7 @@ export default defineFunction({
   name: 'Update markdown',
   description: 'Updates the entire contents of the markdown document.',
   inputSchema: Schema.Struct({
-    id: ArtifactId.annotations({
+    doc: Ref.Ref(Markdown.Document).annotations({
       description: 'The ID of the document to write.',
     }),
     content: Schema.String.annotations({
@@ -23,9 +22,9 @@ export default defineFunction({
     }),
   }),
   outputSchema: Schema.Void,
-  handler: Effect.fn(function* ({ data: { id, content } }) {
-    const doc = yield* Database.Service.resolve(ArtifactId.toDXN(id), Markdown.Document);
-    const text = yield* Database.Service.load(doc.content);
+  handler: Effect.fn(function* ({ data: { doc, content } }) {
+    const document = yield* Database.load(doc);
+    const text = yield* Database.load(document.content);
     Obj.change(text, (t) => {
       t.content = content;
     });

package/src/blueprints/markdown/index.ts

@@ -0,0 +1,6 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+export { default as MarkdownBlueprint } from './blueprint';
+export { MarkdownFunctions } from './functions';

package/src/blueprints/planning/blueprint.ts

@@ -0,0 +1,28 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import { type AppCapabilities } from '@dxos/app-toolkit';
+import { Blueprint } from '@dxos/blueprints';
+
+import { PlanningFunctions } from './functions';
+
+const BLUEPRINT_KEY = 'dxos.org/blueprint/planning';
+
+const functions = Object.values(PlanningFunctions);
+
+const make = () =>
+  Blueprint.make({
+    key: BLUEPRINT_KEY,
+    name: 'Planning',
+    description: 'Plans and tracks complex tasks with artifact management.',
+    tools: Blueprint.toolDefinitions({ functions: [PlanningFunctions.UpdateTasks] }),
+  });
+
+const blueprint: AppCapabilities.BlueprintDefinition = {
+  key: BLUEPRINT_KEY,
+  functions,
+  make,
+};
+
+export default blueprint;

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

@@ -0,0 +1,9 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import UpdateTasks from './update-tasks';
+
+export const PlanningFunctions = {
+  UpdateTasks,
+};

package/src/blueprints/planning/functions/update-tasks.ts

@@ -0,0 +1,168 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+import * as Schema from 'effect/Schema';
+
+import { AiContextService } from '@dxos/assistant';
+import { Database, Obj } from '@dxos/echo';
+import { defineFunction } from '@dxos/functions';
+import { trim } from '@dxos/util';
+
+import { Plan, Project } from '../../../types';
+
+const INSTRUCTIONS = trim`
+TASK MANAGEMENT TOOL - USAGE GUIDELINES
+
+This tool maintains an organized task list during work sessions to track progress, break down objectives, and ensure thoroughness. After creating initial tasks, update them silently without announcing changes to the user.
+
+=== CORE USAGE PRINCIPLES ===
+
+Create and manage tasks for: multi-step objectives requiring 3+ distinct actions, complex projects needing careful sequencing, user requests for task organization, multiple deliverables provided together, new instructions (capture as tasks immediately with new IDs), completed work (mark done and add follow-ups), and active work (mark as 'in-progress', limit one at a time).
+
+Skip task management for: single straightforward actions, simple requests achievable in 1-2 steps, informational queries, quick lookups or clarifications, and avoid creating verification tasks unless requested.
+
+=== TOOL SPECIFICATION ===
+
+update-tasks requires an array of task objects. Each task object contains:
+- id (string, required): unique identifier like "task_1" or "research_sources"
+- title (string, optional): update to clarify or refine task description
+- status (string, optional): 'todo' | 'in-progress' | 'done'
+
+Task status meanings: 'todo' means not yet started, 'in-progress' means currently being worked on, 'done' means completed successfully.
+
+=== OPERATIONAL GUIDELINES ===
+
+Update tasks in real-time as work progresses. Mark tasks 'done' immediately upon completion. Maintain only ONE task with 'in-progress' status at a time. Complete current tasks before starting new ones. Use specific, actionable task titles. Break complex work into manageable logical pieces. Batch task updates with other actions when possible for efficiency.
+
+=== USAGE EXAMPLES ===
+
+<example type="research_project">
+<user_message>I need to research sustainable packaging options and write a report comparing costs and environmental impact.</user_message>
+<assistant_action>
+Creates tasks:
+1. {id: "research_materials", title: "Research sustainable packaging materials", status: "in-progress"}
+2. {id: "compare_costs", title: "Compile cost comparison data", status: "todo"}
+3. {id: "assess_impact", title: "Evaluate environmental impact metrics", status: "todo"}
+4. {id: "draft_report", title: "Write comparative analysis report", status: "todo"}
+
+Begins research work immediately in same response.
+</assistant_action>
+<reasoning>Multi-phase project requiring systematic tracking across research, analysis, and writing stages.</reasoning>
+</example>
+
+<example type="event_planning">
+<user_message>Help me plan a team retreat - venue, agenda, catering, and send invitations.</user_message>
+<assistant_action>
+Creates tasks:
+1. {id: "venue_options", title: "Research and recommend venues", status: "in-progress"}
+2. {id: "draft_agenda", title: "Create retreat agenda", status: "todo"}
+3. {id: "arrange_catering", title: "Coordinate catering options", status: "todo"}
+4. {id: "send_invites", title: "Draft and send invitations", status: "todo"}
+</assistant_action>
+<reasoning>Multiple distinct deliverables with dependencies requiring organized tracking.</reasoning>
+</example>
+
+<example type="document_analysis">
+<user_message>Review this 30-page contract for key terms, risks, and negotiation points.</user_message>
+<assistant_action>
+Creates tasks:
+1. {id: "extract_key_terms", title: "Identify key contractual terms", status: "in-progress"}
+2. {id: "flag_risks", title: "Flag potential risks and liabilities", status: "todo"}
+3. {id: "negotiation_points", title: "Compile negotiation recommendations", status: "todo"}
+4. {id: "summary_doc", title: "Prepare executive summary", status: "todo"}
+</assistant_action>
+<reasoning>Substantial analysis work requiring methodical breakdown and systematic review.</reasoning>
+</example>
+
+<example type="multiple_deliverables">
+<user_message>I need: 1) competitor analysis for three companies, 2) SWOT analysis, 3) market positioning recommendations.</user_message>
+<assistant_action>
+Creates tasks:
+1. {id: "competitor_research", title: "Research three competitor companies", status: "in-progress"}
+2. {id: "swot_analysis", title: "Develop SWOT analysis", status: "todo"}
+3. {id: "positioning_recs", title: "Create market positioning recommendations", status: "todo"}
+</assistant_action>
+<reasoning>User provided numbered list of distinct deliverables requiring separate effort.</reasoning>
+</example>
+
+<example type="skip_simple_question">
+<user_message>What's the difference between renewable and sustainable energy?</user_message>
+<assistant_action>Provides explanation directly without creating tasks.</assistant_action>
+<reasoning>Informational request with no actionable work to complete or track.</reasoning>
+</example>
+
+<example type="skip_quick_lookup">
+<user_message>Find the population of Tokyo.</user_message>
+<assistant_action>Searches and provides answer without task tracking.</assistant_action>
+<reasoning>Single straightforward lookup completable immediately.</reasoning>
+</example>
+
+<example type="skip_trivial_task">
+<user_message>Summarize this 2-paragraph email.</user_message>
+<assistant_action>Provides summary directly without creating tasks.</assistant_action>
+<reasoning>Single simple action requiring no breakdown or progress tracking.</reasoning>
+</example>
+
+<example type="skip_single_action">
+<user_message>Check if my flight is on time.</user_message>
+<assistant_action>Performs lookup and reports status without task management.</assistant_action>
+<reasoning>One-step action with immediate completion, no organizational benefit from tasks.</reasoning>
+</example>
+
+=== BEST PRACTICES ===
+
+For task creation: use descriptive unique IDs reflecting the work, start first task as 'in-progress', batch initial creation with beginning work. For progress tracking: update status immediately upon completion, keep only one 'in-progress' task unless parallel work is natural, add follow-up tasks as they emerge. For task breakdown: aim for reasonably-scoped tasks, group related small actions into logical units, split tasks requiring different approaches.
+
+When uncertain whether to use task management, err on the side of creating tasks. Proactive organization demonstrates thoroughness and ensures comprehensive work completion.
+`;
+
+const TaskProps = Schema.Struct({
+  id: Plan.TaskId,
+  title: Schema.String,
+  status: Schema.Literal('todo', 'in-progress', 'done'),
+});
+
+export default defineFunction({
+  key: 'dxos.org/function/planning/update-tasks',
+  name: 'Update tasks',
+  description: INSTRUCTIONS,
+  inputSchema: Schema.Struct({
+    tasks: Schema.Array(TaskProps),
+  }),
+  handler: Effect.fn(function* ({ data: { tasks: newTasks } }) {
+    const project = yield* Project.getFromChatContext;
+    const plan = yield* Database.load(project.plan);
+
+    Obj.change(plan, (plan) => {
+      for (const task of newTasks) {
+        const existingTask = plan.tasks.find((t) => t.id === task.id);
+        if (existingTask) {
+          existingTask.title = task.title;
+          existingTask.status = task.status;
+        } else {
+          plan.tasks.push({
+            id: task.id,
+            title: task.title,
+            status: task.status,
+          });
+        }
+      }
+    });
+
+    // console.log('\n====== TASKS ======\n');
+    // for (const task of tasks) {
+    //   console.log(`- **${task.status?.toLocaleUpperCase()}**: ${task.title ?? 'No title'} (id: ${task.id})`);
+    // }
+    // console.log('\n====== END TASKS ======\n');
+
+    return trim`
+      Tasks updated. Don't forget to mark tasks as done when you're done with them or update their status to 'in-progress' when you start working on them.
+      Current plan:
+      <plan>
+        ${Plan.formatPlan(plan)}
+      </plan>
+    `;
+  }, AiContextService.fixFunctionHandlerType),
+});

package/src/blueprints/planning/index.ts

@@ -1,7 +1,5 @@
 //
-// Copyright 2025 DXOS.org
+// Copyright 2026 DXOS.org
 //
 
-import { blueprint } from './planning-blueprint';
-
-export default blueprint;
+export { default as PlanningBlueprint } from './blueprint';

package/src/blueprints/{planning/planning-blueprint.test.ts → planning-old/blueprint.test.ts}

@@ -7,13 +7,9 @@ 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,
-  type ContextBinding,
-  makeToolExecutionServiceFromFunctions,
-  makeToolResolverFromFunctions,
-} from '@dxos/assistant';
+import { AiConversation, type ContextBinding, ToolExecutionServices } from '@dxos/assistant';
 import { Blueprint } from '@dxos/blueprints';
 import { Obj, Ref } from '@dxos/echo';
 import { Database } from '@dxos/echo';
@@ -28,12 +24,13 @@ import { Text } from '@dxos/schema';
 import { type Message } from '@dxos/types';
 import { trim } from '@dxos/util';
 
-import { Tasks } from '../../functions';
-import { type TestStep, runSteps, testToolkit } from '../testing';
+import { type TestStep, runSteps } from '../testing';
 
-import { blueprint } from './planning-blueprint';
+import PlanningOldBlueprint from './blueprint';
+import { TaskFunctions } from './functions';
 
 describe('Planning Blueprint', { timeout: 120_000 }, () => {
+  const blueprint = PlanningOldBlueprint.make();
   it.scoped(
     'planning blueprint',
     Effect.fn(
@@ -41,14 +38,14 @@ describe('Planning Blueprint', { timeout: 120_000 }, () => {
         const queue = yield* QueueService.createQueue<Message.Message | ContextBinding>();
         const conversation = yield* acquireReleaseResource(() => new AiConversation({ queue }));
 
-        yield* Database.Service.add(blueprint);
+        yield* Database.add(blueprint);
         yield* Effect.promise(() =>
           conversation.context.bind({
             blueprints: [Ref.make(blueprint)],
           }),
         );
 
-        const artifact = yield* Database.Service.add(Markdown.make());
+        const artifact = yield* Database.add(Markdown.make());
         let prevContent = artifact.content;
         const matchList =
           ({ includes = [], excludes = [] }: { includes: RegExp[]; excludes?: RegExp[] }) =>
@@ -105,18 +102,17 @@ describe('Planning Blueprint', { timeout: 120_000 }, () => {
       Effect.provide(
         Layer.mergeAll(
           TestDatabaseLayer({ types: [Text.Text, Markdown.Document, Blueprint.Blueprint] }),
-          makeToolResolverFromFunctions([Tasks.read, Tasks.update], testToolkit),
-          makeToolExecutionServiceFromFunctions(testToolkit, testToolkit.toLayer({}) as any),
+          ToolExecutionServices,
           AiService.model('@anthropic/claude-3-5-sonnet-20241022'),
         ).pipe(
           Layer.provideMerge(
-            FunctionInvocationServiceLayerTestMocked({ functions: [Tasks.read, Tasks.update] }).pipe(
+            FunctionInvocationServiceLayerTestMocked({ functions: Object.values(TaskFunctions) }).pipe(
               Layer.provideMerge(TracingService.layerNoop),
             ),
           ),
-          Layer.provideMerge(FunctionImplementationResolver.layerTest({ functions: [Tasks.read, Tasks.update] })),
+          Layer.provideMerge(FunctionImplementationResolver.layerTest({ functions: Object.values(TaskFunctions) })),
           Layer.provideMerge(TestDatabaseLayer({ types: [Text.Text, Markdown.Document, Blueprint.Blueprint] })),
-          Layer.provideMerge(AiServiceTestingPreset('direct')),
+          Layer.provideMerge(Layer.mergeAll(GenericToolkit.providerEmpty, AiServiceTestingPreset('direct'))),
         ),
       ),
       TestHelpers.provideTestContext,

package/src/blueprints/{planning/planning-blueprint.ts → planning-old/blueprint.ts}

@@ -3,12 +3,17 @@
 //
 
 import { ToolId } from '@dxos/ai';
+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 { Tasks } from '../../functions';
+import { TaskFunctions } from './functions';
+
+const BLUEPRINT_KEY = 'dxos.org/blueprint/planning';
+
+const functions = Object.values(TaskFunctions);
 
 /**
  * Agent prompt instructions for managing hierarchical task lists.
@@ -85,12 +90,21 @@ const instructions = trim`
   - 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)),
-});
+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: functions.map((fn) => ToolId.make(fn.key)),
+  });
+
+const blueprint: AppCapabilities.BlueprintDefinition = {
+  key: BLUEPRINT_KEY,
+  functions,
+  make,
+};
+
+export default blueprint;

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

@@ -0,0 +1,11 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import Read from './read';
+import Update from './update';
+
+export const TaskFunctions = {
+  Read,
+  Update,
+};

package/src/{functions/tasks → blueprints/planning-old/functions}/read.ts

@@ -5,8 +5,7 @@
 import * as Effect from 'effect/Effect';
 import * as Schema from 'effect/Schema';
 
-import { ArtifactId } from '@dxos/assistant';
-import { Database } from '@dxos/echo';
+import { Database, Ref } from '@dxos/echo';
 import { defineFunction } from '@dxos/functions';
 import { Markdown } from '@dxos/plugin-markdown/types';
 
@@ -15,18 +14,18 @@ export default defineFunction({
   name: 'Read',
   description: 'Read markdown tasks.',
   inputSchema: Schema.Struct({
-    id: ArtifactId.annotations({
+    doc: Ref.Ref(Markdown.Document).annotations({
       description: 'The ID of the document to read.',
     }),
   }),
   outputSchema: Schema.Struct({
     content: Schema.String,
   }),
-  handler: Effect.fn(function* ({ data: { id } }) {
-    const doc = yield* Database.Service.resolve(ArtifactId.toDXN(id), Markdown.Document);
+  handler: Effect.fn(function* ({ data: { doc } }) {
+    const document = yield* Database.load(doc);
 
     // Return content with line numbers prefixed.
-    const { content } = yield* Database.Service.load(doc.content);
+    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');