@dxos/assistant-toolkit 0.8.4-main.e8ec1fe → 0.8.4-main.fcfe5033a5

package/src/blueprints/project/functions/add-artifact.ts

@@ -0,0 +1,30 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+
+import { Database, Obj } from '@dxos/echo';
+import { Operation } from '@dxos/operation';
+
+import { Agent } from '../../../types';
+import { AddArtifact } from './definitions';
+
+export default AddArtifact.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(function* ({ name, artifact }) {
+      if (!(yield* Database.load(artifact))) {
+        throw new Error('Artifact not found.');
+      }
+
+      const agent = yield* Agent.getFromChatContext;
+
+      Obj.change(agent, (agent) => {
+        agent.artifacts.push({
+          name,
+          data: artifact,
+        });
+      });
+    }) as any,
+  ),
+);

package/src/blueprints/project/functions/agent.ts

@@ -0,0 +1,68 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+import * as Layer from 'effect/Layer';
+
+import { AiService } from '@dxos/ai';
+import { AiSession, ToolExecutionServices } from '@dxos/assistant';
+import { Database, Feed, Obj } from '@dxos/echo';
+import { acquireReleaseResource } from '@dxos/effect';
+import { invariant } from '@dxos/invariant';
+import { Operation } from '@dxos/operation';
+
+import { Agent } from '../../../types';
+import { AgentWorker } from './definitions';
+
+export default AgentWorker.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(function* ({ agent: agentRef, prompt, event }) {
+      const agent = yield* Database.load(agentRef).pipe(
+        Effect.catchTag('ObjectNotFoundError', () => Effect.die(new Error('Unable to load agent object.'))),
+      );
+      invariant(Obj.instanceOf(Agent.Agent, agent));
+      invariant(agent.chat, 'Agent has no chat.');
+
+      const chatFeed = yield* agent.chat.pipe(
+        Database.load,
+        Effect.flatMap((chat) => Database.load(chat.feed)),
+        Effect.catchTag('ObjectNotFoundError', () => Effect.die(new Error('Unable to load agent chat feed object.'))),
+      );
+      invariant(chatFeed, 'Agent chat feed not found.');
+      const runtime = yield* Effect.runtime<Feed.FeedService>();
+      const session = yield* acquireReleaseResource(() => new AiSession({ feed: chatFeed, runtime }));
+
+      const agentsInContext = session.context.getObjects().filter(Obj.instanceOf(Agent.Agent));
+      if (agentsInContext.length !== 1) {
+        throw new Error('There should be exactly one agent in context. Got: ' + agentsInContext.length);
+      }
+
+      if (!prompt && !event) {
+        throw new Error('Either prompt or event must be provided.');
+      }
+
+      let input = '';
+      if (prompt) {
+        input += `${prompt}\n\n`;
+      }
+      if (event) {
+        input += `<event>\n${JSON.stringify(event, null, 2)}\n</event>\n\n`;
+      }
+
+      yield* session
+        .createRequest({
+          prompt: input,
+        })
+        .pipe(
+          Effect.provide(
+            Layer.mergeAll(AiService.model('@anthropic/claude-opus-4-6'), ToolExecutionServices).pipe(
+              Layer.provideMerge(Operation.withInvocationOptions({ conversation: Obj.getDXN(chatFeed).toString() })),
+            ),
+          ),
+          Effect.retry({ times: 2 }),
+        );
+    }, Effect.scoped),
+  ),
+  Operation.opaqueHandler,
+);

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

@@ -0,0 +1,92 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import * as Schema from 'effect/Schema';
+
+import { AiService, OpaqueToolkit } from '@dxos/ai';
+import { AiContextService } from '@dxos/assistant';
+import { Database, Feed, Obj, Ref } from '@dxos/echo';
+import { QueueService, Trace, TriggerEvent } from '@dxos/functions';
+import { Operation, OperationRegistry } from '@dxos/operation';
+
+import { Agent } from '../../../types';
+
+export const AgentWorker = Operation.make({
+  meta: {
+    key: 'org.dxos.function.agent.worker',
+    name: 'Agent Worker',
+    description: 'Agentic worker that drives the agent autonomously.',
+  },
+  input: Schema.Struct({
+    agent: Schema.suspend(() => Ref.Ref(Agent.Agent)),
+    prompt: Schema.optional(Schema.String),
+    event: Schema.optional(TriggerEvent.TriggerEvent),
+  }),
+  output: Schema.Void,
+  services: [
+    AiService.AiService,
+    Database.Service,
+    QueueService,
+    Feed.FeedService,
+    OperationRegistry.Service,
+    Trace.TraceService,
+    OpaqueToolkit.OpaqueToolkitProvider,
+  ],
+});
+
+export const Qualifier = Operation.make({
+  meta: {
+    key: 'org.dxos.function.agent.qualifier',
+    name: 'Agent Qualifier',
+    description:
+      'Qualifier that determines if the event is relevant to the agent. Puts the data into the input queue of the agent.',
+  },
+  input: Schema.Struct({
+    agent: Schema.suspend(() => Ref.Ref(Agent.Agent)),
+    event: TriggerEvent.TriggerEvent,
+  }),
+  output: Schema.Void,
+  services: [AiService.AiService, Database.Service],
+});
+
+export const GetContext = Operation.make({
+  meta: {
+    key: 'org.dxos.function.agent.get-context',
+    name: 'Get Agent Context',
+    description: 'Get the context of an agent.',
+  },
+  input: Schema.Struct({}),
+  output: Schema.Struct({
+    id: Schema.String,
+    name: Schema.String,
+    instructions: Schema.String,
+    plan: Schema.String,
+    artifacts: Schema.Array(
+      Schema.Struct({
+        name: Schema.String,
+        type: Schema.optional(Schema.String),
+        dxn: Schema.optional(Schema.String),
+      }),
+    ),
+  }),
+  services: [AiContextService, Database.Service],
+});
+
+export const AddArtifact = Operation.make({
+  meta: {
+    key: 'org.dxos.function.agent.add-artifact',
+    name: 'Add artifact',
+    description: 'Adds a new artifact.',
+  },
+  input: Schema.Struct({
+    name: Schema.String.annotations({
+      description: 'The name of the artifact to add.',
+    }),
+    artifact: Ref.Ref(Obj.Unknown).annotations({
+      description: 'The artifact to add. Do NOT guess or try to generate the ID.',
+    }),
+  }),
+  output: Schema.Void,
+  services: [AiContextService, Database.Service],
+});

package/src/blueprints/project/functions/get-context.ts

@@ -0,0 +1,46 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+
+import { Database, Obj } from '@dxos/echo';
+import { Operation } from '@dxos/operation';
+
+import { Plan, Agent } from '../../../types';
+import { GetContext } from './definitions';
+
+export default GetContext.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(function* () {
+      const agent = yield* Agent.getFromChatContext;
+
+      return {
+        id: agent.id,
+        name: agent.name,
+        instructions: yield* agent.instructions.pipe(Database.load).pipe(
+          Effect.map((_) => _.content),
+          Effect.catchTag('ObjectNotFoundError', () => Effect.succeed('No instructions found.')),
+        ),
+        plan: yield* (
+          agent.plan?.pipe(Database.load).pipe(
+            Effect.map(Plan.formatPlan),
+            Effect.catchTag('ObjectNotFoundError', () => Effect.succeed('No plan found.')),
+          ) ?? Effect.succeed('No plan found.')
+        ),
+        artifacts: yield* Effect.forEach(agent.artifacts, (artifact) =>
+          Effect.gen(function* () {
+            return {
+              name: artifact.name,
+              type: yield* Database.load(artifact.data).pipe(
+                Effect.map(Obj.getTypename),
+                Effect.catchTag('ObjectNotFoundError', () => Effect.succeed('Artifact not found.')),
+              ),
+              dxn: artifact.data.dxn.toString(),
+            };
+          }),
+        ),
+      };
+    }) as any,
+  ),
+);

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

@@ -0,0 +1,14 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import { OperationHandlerSet } from '@dxos/operation';
+
+export * from './definitions';
+
+export const AgentBlueprintHandlers = OperationHandlerSet.lazy(
+  () => import('./add-artifact'),
+  () => import('./agent'),
+  () => import('./get-context'),
+  () => import('./qualifier'),
+);

package/src/blueprints/project/functions/qualifier.ts

@@ -0,0 +1,90 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import * as LanguageModel from '@effect/ai/LanguageModel';
+import * as Prompt from '@effect/ai/Prompt';
+import * as Effect from 'effect/Effect';
+import * as Schema from 'effect/Schema';
+
+import { AiService } from '@dxos/ai';
+import { Database, Obj, Ref } from '@dxos/echo';
+import { invariant } from '@dxos/invariant';
+import { Operation } from '@dxos/operation';
+import { trim } from '@dxos/util';
+
+import { Plan, Agent } from '../../../types';
+import { Qualifier } from './definitions';
+
+export default Qualifier.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(
+      function* ({ agent: agentRef, event }) {
+        const agent = yield* Database.load(agentRef);
+        invariant(Obj.instanceOf(Agent.Agent, agent));
+        invariant(agent.chat, 'Agent has no chat.');
+
+        const { id, name, queue } = agent;
+        if (!queue) {
+          throw new Error('Agent has no queue.');
+        }
+
+        const plan = yield* Database.load(agent.plan);
+        const instructions = yield* Database.load(agent.instructions);
+
+        const { value } = yield* Effect.scoped(
+          LanguageModel.generateObject({
+            schema: Schema.Struct({
+              isRelevant: Schema.Boolean,
+            }),
+            prompt: Prompt.fromMessages([
+              Prompt.systemMessage({
+                content: trim`
+                  You are a qualifying agent that determines if the event is relevant to the agent.
+                  Respond with true if the event is relevant to the agent, false otherwise.
+                  If you are not sure, return true.
+                  The qualified events will be forwarded to the larger agent that will process them.
+                  <agent id="${id}" name="${name}">
+                    <instructions>
+                    ${instructions.content}
+                    </instructions>
+                    <plan>
+                      ${Plan.formatPlan(plan)}
+                    </plan>
+                  </agent>
+                `,
+              }),
+              Prompt.userMessage({
+                content: [
+                  Prompt.makePart('text', {
+                    text: trim`
+                      <event>
+                        ${JSON.stringify(event, null, 2)}
+                      </event>
+                    `,
+                  }),
+                ],
+              }),
+            ]),
+          }),
+        );
+
+        const { isRelevant } = value as { isRelevant: boolean };
+
+        if (isRelevant) {
+          const queueTarget = yield* Database.load(queue);
+          if ('queue' in event && event.item) {
+            const obj = event.item;
+            yield* Effect.promise(() => queueTarget.append([obj]));
+          } else if ('subject' in event && Ref.isRef(event.subject)) {
+            const obj = yield* Database.load(event.subject);
+            yield* Effect.promise(() => queueTarget.append([obj]));
+          } else {
+            throw new Error('Invalid event.');
+          }
+        }
+      },
+      Effect.provide(AiService.model('@anthropic/claude-sonnet-4-5')),
+    ),
+  ),
+);

package/src/blueprints/project/index.ts

@@ -0,0 +1,6 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+export { default as AgentBlueprint } from './blueprint';
+export { AgentWorker, Qualifier, GetContext, AddArtifact, AgentBlueprintHandlers } from './functions';

package/src/blueprints/project-wizard/blueprint.ts

@@ -0,0 +1,48 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import { type AppCapabilities } from '@dxos/app-toolkit';
+import { Blueprint, Template } from '@dxos/blueprints';
+import { trim } from '@dxos/util';
+
+import { AgentRules, CreateAgent, SyncTriggers } from './functions';
+
+const BLUEPRINT_KEY = 'org.dxos.blueprint.agent-wizard';
+
+/**
+ * Creates the Agent Wizard blueprint. This is a function to avoid circular dependency issues.
+ */
+// TODO(dmaretskyi): Combine with Agent Blueprint.
+const make = () =>
+  Blueprint.make({
+    key: BLUEPRINT_KEY,
+    name: 'Agent Wizard',
+    description: 'Help the user create a new agent.',
+    agentCanEnable: true,
+    instructions: Template.make({
+      source: trim`
+        You are a wizard that helps the user create a new agent.
+
+        Agents are goal oriented and autonomously driven.
+        Each agent has instructions - the goal of the agent.
+        The instructions also typically describe what actions to perform in reaction to events (emails).
+        The agent has a number of associated artifacts to work with.
+        Agents can subscribe to emails.
+
+        The agent itself is an ECHO object and can be edited like any other object using the database blueprint.
+        You can edit the agent's instructions, name, and other properties directly.
+        If you edit the agent's subscriptions array, you MUST call the sync-triggers function afterward to synchronize the triggers.
+
+        IMPORTANT: Before attempting to create an agent call the [agent-rules] tool to get the rules for creating an agent.
+      `,
+    }),
+    tools: Blueprint.toolDefinitions({ operations: [AgentRules, CreateAgent, SyncTriggers] }),
+  });
+
+const blueprint: AppCapabilities.BlueprintDefinition = {
+  key: BLUEPRINT_KEY,
+  make,
+};
+
+export default blueprint;

package/src/blueprints/project-wizard/functions/create-project.ts

@@ -0,0 +1,33 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+
+import { Blueprint } from '@dxos/blueprints';
+import { Obj, Ref } from '@dxos/echo';
+import { Operation } from '@dxos/operation';
+
+import { Agent } from '../../../types';
+import { AgentBlueprint } from '../../project';
+import { CreateAgent, SyncTriggers } from './definitions';
+
+export default CreateAgent.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(function* ({ name, instructions, blueprints, subscriptions }) {
+      const agent = yield* Agent.makeInitialized(
+        {
+          name,
+          instructions,
+          blueprints: yield* Effect.forEach(blueprints, (key) =>
+            Blueprint.upsert(key).pipe(Effect.map(Ref.make), Effect.orDie),
+          ),
+          subscriptions,
+        },
+        Obj.clone(AgentBlueprint.make()),
+      );
+      yield* Operation.invoke(SyncTriggers, { agent: Ref.make(agent) });
+      return agent;
+    }),
+  ),
+);

package/src/blueprints/project-wizard/functions/definitions.ts

@@ -0,0 +1,64 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Schema from 'effect/Schema';
+
+import { Blueprint } from '@dxos/blueprints';
+import { Database, Feed, Obj, Ref } from '@dxos/echo';
+import { QueueService } from '@dxos/functions';
+import { Operation } from '@dxos/operation';
+
+import { Agent } from '../../../types';
+
+export const AgentRules = Operation.make({
+  meta: {
+    key: 'org.dxos.function.agent-wizard.agent-rules',
+    name: 'Agent rules',
+    description: 'Gets the rules for creating an agent.',
+  },
+  input: Schema.Struct({}),
+  output: Schema.String,
+});
+
+export const CreateAgent = Operation.make({
+  meta: {
+    key: 'org.dxos.function.agent-wizard.create-agent',
+    name: 'Create agent',
+    description: 'Creates a new agent.',
+  },
+  input: Schema.Struct({
+    name: Schema.String.annotations({
+      description: 'The name of the agent to create.',
+    }),
+    instructions: Schema.String.annotations({
+      description:
+        'The goal of the agent. Be specific but not too verbose. The agent will use this as a core objective and set of rules to follow.',
+    }),
+    blueprints: Schema.Array(Schema.String).annotations({
+      description: 'The blueprint keys to use for the agent.',
+      examples: [['org.dxos.blueprint.markdown', 'org.dxos.blueprint.database']],
+    }),
+    subscriptions: Schema.Array(Ref.Ref(Obj.Unknown)).annotations({
+      description: 'The objects to subscribe to for the agent. Can be references to mailboxes.',
+    }),
+  }),
+  output: Agent.Agent,
+  services: [Blueprint.RegistryService, Database.Service, QueueService, Feed.FeedService],
+});
+
+export const SyncTriggers = Operation.make({
+  meta: {
+    key: 'org.dxos.function.agent.sync-triggers',
+    name: 'Sync triggers',
+    description:
+      'Synchronizes triggers in the database with the agent subscriptions. Call this after editing the subscriptions array.',
+  },
+  input: Schema.Struct({
+    agent: Ref.Ref(Agent.Agent).annotations({
+      description: 'The agent whose triggers should be synced.',
+    }),
+  }),
+  output: Schema.Void,
+  services: [Database.Service],
+});

package/src/blueprints/project-wizard/functions/index.ts

@@ -0,0 +1,13 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import { OperationHandlerSet } from '@dxos/operation';
+
+export * from './definitions';
+
+export const AgentWizardHandlers = OperationHandlerSet.lazy(
+  () => import('./create-project'),
+  () => import('./project-rules'),
+  () => import('./sync-triggers'),
+);

package/src/blueprints/project-wizard/functions/project-rules.ts

@@ -0,0 +1,77 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+
+import { Operation } from '@dxos/operation';
+import { trim } from '@dxos/util';
+
+import { AgentRules } from './definitions';
+
+export default AgentRules.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(function* () {
+      return trim`
+        You can ask the user for qualifying questions about the agents.
+        If agents should actively read incoming emails, query for mailboxes and add a subscription to them.
+        Use [query-blueprints] from the Blueprint Manager to query for available blueprints and their keys.
+        Use [create-agent] function to create a new agent.
+        After creating an agent, explicitly remind the user to enable local triggers so the agent can be driven autonomously.
+
+        Notable blueprints (query to get their keys):
+
+        - Blueprint Manger - allows agent to self-enable blueprints (always include this one).
+        - Database -- CRUD on objects in the ECHO database.
+        - Markdown -- Create and edit markdown documents.
+        - Websearch -- Search the web for information.
+        - Browser -- Virtual browser via playwright when simple WebSearch is not enough.
+        - Memory -- Memory to store and retrieve information.
+
+        Experimental blueprints that are discouraged:
+
+        - Research
+        - Design
+        - Agent (not wizard) -- those are agent internals, they are auto-added to every new agent.
+
+        <example_agent>
+          ## CRM from your inbox
+
+          Blueprints: database, websearch, browser
+
+          Subscribe to your inboxes.
+          Spec says that on every email we should extract People and Organizations and save them to the database, but first query the database for existing people and organizations to avoid duplicates.
+          Also run a quick research on the web to find more information about the person or organization and save it to the database.
+        </example_agent>
+
+        <example_agent>
+          ## Parcel tracking
+
+          Blueprints: database
+
+          Based on emails from vendors (amazon, fedex, etc.) track the status of the parcels ordered online and keep a table of orders.
+
+          Before creating an agent, use the Database blueprint to create an Order schema:
+            - seller
+            - name
+            - price
+            - status (enum of: pending, shipped, waiting for pickup, delivered, cancelled, returned)
+            - tracking numbers/notes
+            - date of delivery
+            - shipping address
+        </example_agent>
+
+        <example_agent>
+          ## Comms assistant
+
+          Blueprints: database, markdown, inbox
+
+          Helps user maintain comms with external parites via email.
+          Ask user specifically who they want to communicate with and on what topic.
+          Agent should keep a markdown document in artifacts outlining the current state of the comms.
+          Agent should propose drafts to follow up on the comms. Only propose sensible drafts, do not propose drafts that are not relevant or actionable.
+        </example_agent>
+      `;
+    }),
+  ),
+);