@dxos/assistant-toolkit 0.8.4-main.ae835ea → 0.8.4-main.c351d160a8

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

@@ -0,0 +1,72 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+import * as Schema from 'effect/Schema';
+
+import { AiService } from '@dxos/ai';
+import { AiContextService, AiConversation, type ContextBinding } from '@dxos/assistant';
+import { Database, Obj, Ref } from '@dxos/echo';
+import { type Queue } from '@dxos/echo-db';
+import { acquireReleaseResource } from '@dxos/effect';
+import { TriggerEvent, defineFunction } from '@dxos/functions';
+import { invariant } from '@dxos/invariant';
+import { type Message } from '@dxos/types';
+
+import { Project } from '../../../types';
+
+export default defineFunction({
+  key: 'org.dxos.function.project.agent',
+  name: 'Project Agent',
+  description: 'Agentic worker that drives the project autonomously.',
+  inputSchema: Schema.Struct({
+    project: Schema.suspend(() => Ref.Ref(Project.Project)),
+    prompt: Schema.optional(Schema.String),
+    event: Schema.optional(TriggerEvent.TriggerEvent),
+  }),
+  outputSchema: Schema.Void,
+  services: [AiContextService],
+  handler: Effect.fnUntraced(
+    function* ({ data }) {
+      const project = yield* Database.load(data.project);
+      invariant(Obj.instanceOf(Project.Project, project));
+      invariant(project.chat, 'Project has no chat.');
+
+      const chatQueue = yield* project.chat.pipe(
+        Database.load,
+        Effect.flatMap((chat) => Database.load(chat.queue)),
+      );
+      invariant(chatQueue, 'Project chat queue not found.');
+      const conversation = yield* acquireReleaseResource(
+        () => new AiConversation({ queue: chatQueue as Queue<Message.Message | ContextBinding> }),
+      );
+
+      const iniativesInContext = conversation.context.getObjects().filter(Obj.instanceOf(Project.Project));
+      if (iniativesInContext.length !== 1) {
+        throw new Error('There should be exactly one project in context. Got: ' + iniativesInContext.length);
+      }
+
+      if (!data.prompt && !data.event) {
+        throw new Error('Either prompt or event must be provided.');
+      }
+
+      let input = '';
+      if (data.prompt) {
+        input += `${data.prompt}\n\n`;
+      }
+      if (data.event) {
+        input += `<event>\n${JSON.stringify(data.event, null, 2)}\n</event>\n\n`;
+      }
+
+      yield* conversation
+        .createRequest({
+          prompt: input,
+        })
+        .pipe(Effect.retry({ times: 2 }));
+    },
+    Effect.scoped,
+    Effect.provide(AiService.model('@anthropic/claude-sonnet-4-5')),
+    AiContextService.fixFunctionHandlerType,
+  ) as any,
+});

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

@@ -0,0 +1,63 @@
+//
+// 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 { Plan, Project } from '../../../types';
+
+export default defineFunction({
+  key: 'org.dxos.function.project.get-context',
+  name: 'Get Project Context',
+  description: 'Get the context of an project.',
+  inputSchema: Schema.Struct({}),
+  outputSchema: Schema.Struct({
+    id: Schema.String,
+    name: Schema.String,
+    spec: 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],
+  handler: Effect.fnUntraced(function* ({ data: _data }) {
+    const project = yield* Project.getFromChatContext;
+
+    return {
+      id: project.id,
+      name: project.name,
+      spec: yield* project.spec.pipe(Database.load).pipe(
+        Effect.map((_) => _.content),
+        Effect.catchTag('ObjectNotFoundError', () => Effect.succeed('No spec found.')),
+      ),
+      plan: yield* (
+        project.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(project.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(),
+          };
+        }),
+      ),
+    };
+  }, AiContextService.fixFunctionHandlerType),
+});

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

@@ -0,0 +1,15 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import { default as AddArtifact } from './add-artifact';
+import { default as Agent } from './agent';
+import { default as GetContext } from './get-context';
+import { default as Qualifier } from './qualifier';
+
+export const ProjectFunctions = {
+  AddArtifact,
+  Agent,
+  GetContext,
+  Qualifier,
+};

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

@@ -0,0 +1,96 @@
+//
+// 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 { TriggerEvent, defineFunction } from '@dxos/functions';
+import { invariant } from '@dxos/invariant';
+import { trim } from '@dxos/util';
+
+import { Plan, Project } from '../../../types';
+
+export default defineFunction({
+  key: 'org.dxos.function.project.qualifier',
+  name: 'Project Qualifier',
+  description:
+    'Qualifier that determines if the event is relevant to the project. Puts the data into the input queue of the project.',
+  inputSchema: Schema.Struct({
+    project: Schema.suspend(() => Ref.Ref(Project.Project)),
+    event: TriggerEvent.TriggerEvent,
+  }),
+  outputSchema: Schema.Void,
+  services: [],
+  handler: Effect.fnUntraced(
+    function* ({ data }) {
+      const project = yield* Database.load(data.project);
+      invariant(Obj.instanceOf(Project.Project, project));
+      invariant(project.chat, 'Project has no chat.');
+
+      const { id, name, queue } = project;
+      if (!queue) {
+        throw new Error('Project has no queue.');
+      }
+
+      const plan = yield* Database.load(project.plan);
+      const spec = yield* Database.load(project.spec);
+
+      const {
+        value: { isRelevant },
+      } = yield* 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 project.
+              Respond with true if the event is relevant to the project, false otherwise.
+              If you are not sure, return true.
+              The qualified events will be forwarded to the larger agent that will process them.
+              <project id="${id}" name="${name}">
+                <spec>
+                ${spec.content}
+                </spec>
+                <plan>
+                  ${Plan.formatPlan(plan)}
+                </plan>
+              </project>
+            `,
+          }),
+          Prompt.userMessage({
+            content: [
+              Prompt.makePart('text', {
+                text: trim`
+                  <event>
+                    ${JSON.stringify(data.event, null, 2)}
+                  </event>
+                `,
+              }),
+            ],
+          }),
+        ]),
+      });
+
+      if (isRelevant) {
+        const queueTarget = yield* Database.load(queue);
+        if ('queue' in data.event && data.event.item) {
+          const obj = data.event.item;
+          yield* Effect.promise(() => queueTarget.append([obj]));
+        } else if ('subject' in data.event && Ref.isRef(data.event.subject)) {
+          const obj = yield* Database.load(data.event.subject);
+          yield* Effect.promise(() => queueTarget.append([obj]));
+        } else {
+          throw new Error('Invalid event.');
+        }
+      }
+    },
+    Effect.scoped,
+    Effect.provide(AiService.model('@anthropic/claude-sonnet-4-5')),
+  ),
+});

package/src/blueprints/project/index.ts

@@ -0,0 +1,7 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+export { default as ProjectBlueprint } from './blueprint';
+export { ProjectFunctions } from './functions';
+export { syncProjectTriggers } from './sync-triggers';

package/src/blueprints/project/sync-triggers.ts

@@ -0,0 +1,153 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+import * as Option from 'effect/Option';
+
+import { Database, Feed, Filter, Obj, Ref } from '@dxos/echo';
+import { FunctionDefinition, Trigger } from '@dxos/functions';
+import { FeedAnnotation } from '@dxos/schema';
+
+import { Project } from '../../types';
+
+import { ProjectFunctions } from './functions';
+
+/**
+ * Foreign key {@link PROJECT_TRIGGER_EXTENSION_KEY} => <project id : ObjectId>.
+ */
+const PROJECT_TRIGGER_EXTENSION_KEY = 'org.dxos.extension.ProjectTrigger';
+
+/**
+ * Foreign key {@link PROJECT_TRIGGER_TARGET_EXTENSION_KEY} => <dxn string of subscription target>.
+ */
+const PROJECT_TRIGGER_TARGET_EXTENSION_KEY = 'org.dxos.extension.ProjectTriggerTarget';
+
+/** Checks if an object's schema has the FeedAnnotation. */
+const hasFeedAnnotation = (obj: Obj.Unknown): boolean => {
+  const schema = Obj.getSchema(obj);
+  if (!schema) {
+    return false;
+  }
+  const annotation = FeedAnnotation.get(schema);
+  return Option.isSome(annotation) && annotation.value === true;
+};
+
+/**
+ * Syncs triggers in the database with the project subscriptions.
+ *
+ * @param project - The project whose triggers should be synced.
+ * @returns An Effect that syncs triggers.
+ */
+export const syncProjectTriggers = (project: Project.Project): Effect.Effect<void, never, Database.Service> =>
+  Effect.gen(function* () {
+    const triggers = yield* Database.runQuery(
+      Filter.foreignKeys(Trigger.Trigger, [{ source: PROJECT_TRIGGER_EXTENSION_KEY, id: project.id }]),
+    );
+
+    for (const trigger of triggers) {
+      const target = Obj.getKeys(trigger, PROJECT_TRIGGER_TARGET_EXTENSION_KEY).at(0)?.id;
+
+      const exists = project.subscriptions.find((subscription) => subscription.dxn.toString() === target);
+      if (!exists && !(project.useQualifyingAgent && target === Obj.getDXN(project)?.toString())) {
+        yield* Database.remove(trigger);
+      }
+    }
+
+    for (const subscription of project.subscriptions) {
+      const relevantTrigger = triggers.find((trigger) =>
+        Obj.getKeys(trigger, PROJECT_TRIGGER_TARGET_EXTENSION_KEY).some(
+          (key) => key.id === subscription.dxn.toString(),
+        ),
+      );
+      if (relevantTrigger) {
+        continue;
+      }
+
+      const targetOption = yield* Database.loadOption(subscription);
+      if (Option.isNone(targetOption)) {
+        continue;
+      }
+      const target = targetOption.value;
+
+      let feedObj: Feed.Feed | undefined;
+      if (Obj.instanceOf(Feed.Feed, target)) {
+        feedObj = target;
+      } else if (hasFeedAnnotation(target)) {
+        const feedRef = (target as Obj.Unknown & { feed?: Ref.Ref<Feed.Feed> }).feed;
+        feedObj = feedRef ? Option.getOrUndefined(yield* Database.loadOption(feedRef)) : undefined;
+      }
+
+      const queueDxn = Option.fromNullable(feedObj).pipe(
+        Option.filter(Obj.instanceOf(Feed.Feed)),
+        Option.map(Feed.getQueueDxn),
+        Option.getOrUndefined,
+      );
+      if (!queueDxn) {
+        continue;
+      }
+
+      yield* Database.add(
+        Trigger.make({
+          [Obj.Parent]: project,
+          [Obj.Meta]: {
+            keys: [
+              { source: PROJECT_TRIGGER_EXTENSION_KEY, id: project.id },
+              { source: PROJECT_TRIGGER_TARGET_EXTENSION_KEY, id: subscription.dxn.toString() },
+            ],
+          },
+          enabled: true,
+          spec: {
+            kind: 'queue',
+            queue: queueDxn.toString(),
+          },
+          function: Ref.make(
+            FunctionDefinition.serialize(
+              project.useQualifyingAgent ? ProjectFunctions.Qualifier : ProjectFunctions.Agent,
+            ),
+          ),
+          input: {
+            project: Ref.make(project),
+            event: '{{event}}',
+          },
+          concurrency: project.useQualifyingAgent ? 5 : undefined,
+        }),
+      );
+    }
+
+    if (project.useQualifyingAgent) {
+      const qualifierTrigger = triggers.find((trigger) =>
+        Obj.getKeys(trigger, PROJECT_TRIGGER_TARGET_EXTENSION_KEY).some(
+          (key) => key.id === Obj.getDXN(project)?.toString(),
+        ),
+      );
+      if (!qualifierTrigger && project.queue) {
+        yield* Database.add(
+          Trigger.make({
+            [Obj.Parent]: project,
+            [Obj.Meta]: {
+              keys: [
+                { source: PROJECT_TRIGGER_EXTENSION_KEY, id: project.id },
+                {
+                  source: PROJECT_TRIGGER_TARGET_EXTENSION_KEY,
+                  id: Obj.getDXN(project)?.toString() ?? '',
+                },
+              ],
+            },
+            function: Ref.make(FunctionDefinition.serialize(ProjectFunctions.Agent)),
+            enabled: true,
+            spec: {
+              kind: 'queue',
+              queue: project.queue.dxn.toString(),
+            },
+            input: {
+              project: Ref.make(project),
+              event: '{{event}}',
+            },
+          }),
+        );
+      }
+    }
+
+    yield* Database.flush();
+  });

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

@@ -0,0 +1,45 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import { type AppCapabilities } from '@dxos/app-toolkit';
+import { Blueprint, Template } from '@dxos/blueprints';
+import { trim } from '@dxos/util';
+
+import { ProjectWizardFunctions } from './functions';
+
+const BLUEPRINT_KEY = 'org.dxos.blueprint.project-wizard';
+
+const functions = Object.values(ProjectWizardFunctions);
+
+/**
+ * Creates the Project blueprint. This is a function to avoid circular dependency issues.
+ */
+const make = () =>
+  Blueprint.make({
+    key: BLUEPRINT_KEY,
+    name: 'Project Wizard',
+    description: 'Help the user create a new project.',
+    instructions: Template.make({
+      source: trim`
+        You are a wizard that helps the user create a new project.
+
+        Projects are goal oriented and autonomously driven by the agent.
+        Each project has a spec - the goal of the project.
+        The spec also typically describes what actions to perform in reaction to events (emails).
+        The project has a number of associated artifacts to work with. 
+        Projects can subscribe to emails.
+
+        IMPORTANT: Before attempting to create a project call the [project-rules] tool to get the rules for creating a project.
+      `,
+    }),
+    tools: Blueprint.toolDefinitions({ functions }),
+  });
+
+const blueprint: AppCapabilities.BlueprintDefinition = {
+  key: BLUEPRINT_KEY,
+  functions,
+  make,
+};
+
+export default blueprint;

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

@@ -0,0 +1,57 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+import * as Schema from 'effect/Schema';
+import { Obj, Ref } from '@dxos/echo';
+import { Blueprint } from '@dxos/blueprints';
+
+import { defineFunction } from '@dxos/functions';
+
+import { Project } from '../../../types';
+import { ProjectBlueprint, syncProjectTriggers } from '../../project';
+
+// TODO(dmaretskyi): Remove this with proper function typing.
+const fixEffectType = <A, E, R>(
+  eff: Effect.Effect<A, E, R>,
+): Effect.Effect<A, E, Exclude<R, Blueprint.RegistryService>> => eff as any;
+
+export default defineFunction({
+  key: 'org.dxos.function.project-wizard.create-project',
+  name: 'Create project',
+  description: 'Creates a new project.',
+  inputSchema: Schema.Struct({
+    name: Schema.String.annotations({
+      description: 'The name of the project to create.',
+    }),
+    spec: Schema.String.annotations({
+      description:
+        'The goal of the project. Be specic but not to verbose. The agent will use this as a core objective and set of rules to follow.',
+    }),
+    blueprints: Schema.Array(Schema.String).annotations({
+      description: 'The keys blueprints to use for the project.',
+      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 project. Can be references to mailboxes.',
+    }),
+  }),
+  outputSchema: Project.Project,
+  services: [Blueprint.RegistryService],
+  handler: Effect.fnUntraced(function* ({ data }) {
+    const project = yield* Project.makeInitialized(
+      {
+        name: data.name,
+        spec: data.spec,
+        blueprints: yield* Effect.forEach(data.blueprints, (key) =>
+          Blueprint.upsert(key).pipe(Effect.map(Ref.make), Effect.orDie),
+        ),
+        subscriptions: data.subscriptions,
+      },
+      Obj.clone(ProjectBlueprint.make()),
+    );
+    yield* syncProjectTriggers(project);
+    return project;
+  }, fixEffectType),
+});

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

@@ -0,0 +1,13 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import { default as CreateProject } from './create-project';
+import { default as QueryBlueprints } from './query-blueprints';
+import { default as ProjectRules } from './project-rules';
+
+export const ProjectWizardFunctions = {
+  CreateProject,
+  QueryBlueprints,
+  ProjectRules,
+};

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

@@ -0,0 +1,78 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+import * as Schema from 'effect/Schema';
+
+import { defineFunction } from '@dxos/functions';
+import { trim } from '@dxos/util';
+
+export default defineFunction({
+  key: 'org.dxos.function.project-wizard.project-rules',
+  name: 'Project rules',
+  description: 'Gets the rules for creating a project.',
+  inputSchema: Schema.Struct({}),
+  outputSchema: Schema.String,
+  handler: Effect.fnUntraced(function* ({ data }) {
+    return trim`
+      You can ask the user for qualifying questions about the projects.
+      If projects should actively read incoming emails, query for mailboxes and add a subscription to them.
+      Use [query-blueprints] function to query for available blueprints.
+      Use [create-project] function to create a new project.
+      After creating a project, explicitly remind the user to enable local triggers so the project can be driven autonomously.
+
+      Notable blueprints (query to get their keys):
+
+      - 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 discoraged:
+
+      - Research
+      - Design
+      - Project (not wizard) -- those are project internals, they are auto-added to every new project.
+
+      <example_project>
+        ## 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_project>
+
+      <example_project>
+        ## 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 a project, 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_project>
+
+      <example_project>
+        ## 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.
+        Project should keep a markdown document in artifacts outlining the current state of the comms.
+        Project should propose drafts to follow up on the comms. Only propose sensible drafts, do not propose drafts that are not relevant or actionable.
+      </example_project>
+    `;
+  }),
+});

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

@@ -0,0 +1,27 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { Blueprint } from '@dxos/blueprints';
+import * as Effect from 'effect/Effect';
+import * as Schema from 'effect/Schema';
+
+import { defineFunction } from '@dxos/functions';
+
+// TODO(dmaretskyi): Remove this with proper function typing.
+const fixEffectType = <A, E, R>(
+  eff: Effect.Effect<A, E, R>,
+): Effect.Effect<A, E, Exclude<R, Blueprint.RegistryService>> => eff as any;
+
+export default defineFunction({
+  key: 'org.dxos.function.project-wizard.query-blueprints',
+  name: 'Query blueprints',
+  description: 'Queries the blueprints.',
+  inputSchema: Schema.Struct({}),
+  outputSchema: Schema.Array(Blueprint.Blueprint),
+  services: [Blueprint.RegistryService],
+  handler: Effect.fnUntraced(function* ({ data }) {
+    const registry = yield* Blueprint.RegistryService;
+    return registry.query();
+  }, fixEffectType),
+});

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

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