@dxos/plugin-assistant 0.8.4-main.5ea62a8 → 0.8.4-main.66e292d

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dxos/plugin-assistant",
-  "version": "0.8.4-main.5ea62a8",
+  "version": "0.8.4-main.66e292d",
   "description": "Assistant plugin",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
@@ -36,132 +36,117 @@
     "src"
   ],
   "dependencies": {
-    "@codemirror/autocomplete": "^6.18.1",
-    "@codemirror/language": "^6.10.3",
-    "@codemirror/search": "^6.5.6",
-    "@codemirror/state": "^6.4.1",
-    "@codemirror/view": "^6.34.1",
-    "@effect/ai": "0.25.2",
-    "@effect/ai-anthropic": "^0.16.1",
-    "@effect/ai-openai": "^0.29.1",
-    "@lezer/highlight": "^1.2.1",
-    "@lezer/markdown": "^1.3.1",
+    "@codemirror/autocomplete": "^6.19.0",
+    "@codemirror/lang-html": "^6.4.11",
+    "@codemirror/lang-xml": "^6.1.0",
+    "@codemirror/language": "^6.11.3",
+    "@codemirror/state": "^6.5.2",
+    "@codemirror/view": "^6.38.4",
+    "@effect/ai": "0.29.1",
+    "@effect/ai-anthropic": "0.21.1",
+    "@effect/ai-openai": "^0.32.1",
+    "@effect/experimental": "0.56.0",
+    "@effect/platform-browser": "0.72.0",
     "@preact-signals/safe-react": "^0.9.0",
-    "@preact/signals-core": "^1.9.0",
-    "@preact/signals-react": "^3.2.0",
+    "@preact/signals-core": "^1.12.1",
+    "@preact/signals-react": "^3.3.0",
     "@radix-ui/react-context": "1.1.1",
-    "effect": "3.17.7",
+    "effect": "3.18.3",
     "exa-js": "^1.5.12",
     "json5": "^2.2.3",
     "jsonpointer": "^5.0.1",
     "lodash.defaultsdeep": "^4.6.1",
     "unified": "^11.0.0",
     "use-deep-compare-effect": "^1.8.1",
-    "@dxos/ai": "0.8.4-main.5ea62a8",
-    "@dxos/app-framework": "0.8.4-main.5ea62a8",
-    "@dxos/assistant": "0.8.4-main.5ea62a8",
-    "@dxos/async": "0.8.4-main.5ea62a8",
-    "@dxos/blueprints": "0.8.4-main.5ea62a8",
-    "@dxos/compute": "0.8.4-main.5ea62a8",
-    "@dxos/conductor": "0.8.4-main.5ea62a8",
-    "@dxos/context": "0.8.4-main.5ea62a8",
-    "@dxos/debug": "0.8.4-main.5ea62a8",
-    "@dxos/devtools": "0.8.4-main.5ea62a8",
-    "@dxos/echo": "0.8.4-main.5ea62a8",
-    "@dxos/echo-protocol": "0.8.4-main.5ea62a8",
-    "@dxos/echo-db": "0.8.4-main.5ea62a8",
-    "@dxos/echo-schema": "0.8.4-main.5ea62a8",
-    "@dxos/edge-client": "0.8.4-main.5ea62a8",
-    "@dxos/effect": "0.8.4-main.5ea62a8",
-    "@dxos/functions": "0.8.4-main.5ea62a8",
-    "@dxos/client": "0.8.4-main.5ea62a8",
-    "@dxos/config": "0.8.4-main.5ea62a8",
-    "@dxos/graph": "0.8.4-main.5ea62a8",
-    "@dxos/invariant": "0.8.4-main.5ea62a8",
-    "@dxos/keys": "0.8.4-main.5ea62a8",
-    "@dxos/lit-ui": "0.8.4-main.5ea62a8",
-    "@dxos/live-object": "0.8.4-main.5ea62a8",
-    "@dxos/local-storage": "0.8.4-main.5ea62a8",
-    "@dxos/log": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-board": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-attention": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-client": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-deck": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-explorer": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-preview": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-script": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-stack": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-sketch": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-wnfs": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-space": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-transcription": "0.8.4-main.5ea62a8",
-    "@dxos/react-client": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-graph": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-markdown": "0.8.4-main.5ea62a8",
-    "@dxos/protocols": "0.8.4-main.5ea62a8",
-    "@dxos/react-edge-client": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-attention": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-canvas-compute": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-canvas-editor": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-editor": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-form": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-graph": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-list": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-menu": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-searchlist": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-sfx": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-components": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-stack": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-tabs": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-tag-picker": "0.8.4-main.5ea62a8",
-    "@dxos/schema": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-chat": "0.8.4-main.5ea62a8",
-    "@dxos/util": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-syntax-highlighter": "0.8.4-main.5ea62a8"
+    "@dxos/ai": "0.8.4-main.66e292d",
+    "@dxos/app-framework": "0.8.4-main.66e292d",
+    "@dxos/assistant": "0.8.4-main.66e292d",
+    "@dxos/assistant-toolkit": "0.8.4-main.66e292d",
+    "@dxos/async": "0.8.4-main.66e292d",
+    "@dxos/blueprints": "0.8.4-main.66e292d",
+    "@dxos/client": "0.8.4-main.66e292d",
+    "@dxos/compute": "0.8.4-main.66e292d",
+    "@dxos/conductor": "0.8.4-main.66e292d",
+    "@dxos/config": "0.8.4-main.66e292d",
+    "@dxos/debug": "0.8.4-main.66e292d",
+    "@dxos/context": "0.8.4-main.66e292d",
+    "@dxos/devtools": "0.8.4-main.66e292d",
+    "@dxos/echo": "0.8.4-main.66e292d",
+    "@dxos/echo-db": "0.8.4-main.66e292d",
+    "@dxos/echo-protocol": "0.8.4-main.66e292d",
+    "@dxos/echo-query": "0.8.4-main.66e292d",
+    "@dxos/edge-client": "0.8.4-main.66e292d",
+    "@dxos/functions-runtime": "0.8.4-main.66e292d",
+    "@dxos/functions": "0.8.4-main.66e292d",
+    "@dxos/graph": "0.8.4-main.66e292d",
+    "@dxos/invariant": "0.8.4-main.66e292d",
+    "@dxos/keys": "0.8.4-main.66e292d",
+    "@dxos/lit-ui": "0.8.4-main.66e292d",
+    "@dxos/effect": "0.8.4-main.66e292d",
+    "@dxos/live-object": "0.8.4-main.66e292d",
+    "@dxos/local-storage": "0.8.4-main.66e292d",
+    "@dxos/log": "0.8.4-main.66e292d",
+    "@dxos/plugin-attention": "0.8.4-main.66e292d",
+    "@dxos/plugin-automation": "0.8.4-main.66e292d",
+    "@dxos/plugin-client": "0.8.4-main.66e292d",
+    "@dxos/plugin-deck": "0.8.4-main.66e292d",
+    "@dxos/plugin-graph": "0.8.4-main.66e292d",
+    "@dxos/plugin-space": "0.8.4-main.66e292d",
+    "@dxos/protocols": "0.8.4-main.66e292d",
+    "@dxos/react-client": "0.8.4-main.66e292d",
+    "@dxos/react-ui-attention": "0.8.4-main.66e292d",
+    "@dxos/react-edge-client": "0.8.4-main.66e292d",
+    "@dxos/plugin-transcription": "0.8.4-main.66e292d",
+    "@dxos/react-ui-canvas-compute": "0.8.4-main.66e292d",
+    "@dxos/react-ui-canvas-editor": "0.8.4-main.66e292d",
+    "@dxos/react-ui-chat": "0.8.4-main.66e292d",
+    "@dxos/react-ui-components": "0.8.4-main.66e292d",
+    "@dxos/react-ui-editor": "0.8.4-main.66e292d",
+    "@dxos/react-ui-form": "0.8.4-main.66e292d",
+    "@dxos/react-ui-list": "0.8.4-main.66e292d",
+    "@dxos/react-ui-graph": "0.8.4-main.66e292d",
+    "@dxos/react-ui-menu": "0.8.4-main.66e292d",
+    "@dxos/react-ui-searchlist": "0.8.4-main.66e292d",
+    "@dxos/react-ui-sfx": "0.8.4-main.66e292d",
+    "@dxos/react-ui-stack": "0.8.4-main.66e292d",
+    "@dxos/react-ui-syntax-highlighter": "0.8.4-main.66e292d",
+    "@dxos/react-ui-tabs": "0.8.4-main.66e292d",
+    "@dxos/schema": "0.8.4-main.66e292d",
+    "@dxos/types": "0.8.4-main.66e292d",
+    "@dxos/util": "0.8.4-main.66e292d"
   },
   "devDependencies": {
-    "@effect-rx/rx-react": "^0.38.0",
-    "@effect/platform": "0.90.2",
+    "@effect-atom/atom-react": "^0.3.4",
+    "@effect/platform": "0.92.1",
     "@types/lodash.defaultsdeep": "^4.6.6",
-    "@types/react": "~18.2.0",
-    "@types/react-dom": "~18.2.0",
-    "effect": "3.17.7",
+    "@types/react": "~19.2.2",
+    "@types/react-dom": "~19.2.2",
+    "effect": "3.18.3",
     "openapi-types": "^12.1.3",
-    "react": "~18.2.0",
-    "react-dom": "~18.2.0",
-    "vite": "7.1.1",
-    "@dxos/assistant-testing": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-attention": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-chess": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-board": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-inbox": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-map": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-markdown": "0.8.4-main.5ea62a8",
-    "@dxos/echo-db": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-preview": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-stack": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-sketch": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-table": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-theme": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-thread": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-wnfs": "0.8.4-main.5ea62a8",
-    "@dxos/random": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-token-manager": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-menu": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-table": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-theme": "0.8.4-main.5ea62a8",
-    "@dxos/storybook-utils": "0.8.4-main.5ea62a8",
-    "@dxos/plugin-script": "0.8.4-main.5ea62a8"
+    "react": "~19.2.0",
+    "react-dom": "~19.2.0",
+    "vite": "7.1.9",
+    "@dxos/assistant-toolkit": "0.8.4-main.66e292d",
+    "@dxos/devtools": "0.8.4-main.66e292d",
+    "@dxos/echo-db": "0.8.4-main.66e292d",
+    "@dxos/plugin-chess": "0.8.4-main.66e292d",
+    "@dxos/plugin-map": "0.8.4-main.66e292d",
+    "@dxos/random": "0.8.4-main.66e292d",
+    "@dxos/plugin-table": "0.8.4-main.66e292d",
+    "@dxos/react-ui": "0.8.4-main.66e292d",
+    "@dxos/react-ui-menu": "0.8.4-main.66e292d",
+    "@dxos/react-ui-table": "0.8.4-main.66e292d",
+    "@dxos/react-ui-theme": "0.8.4-main.66e292d",
+    "@dxos/storybook-utils": "0.8.4-main.66e292d"
   },
   "peerDependencies": {
-    "@effect-rx/rx-react": "^0.34.1",
+    "@effect-atom/atom-react": "^0.3.4",
     "@effect/platform": "^0.80.12",
     "effect": "^3.13.3",
-    "react": "~18.2.0",
-    "react-dom": "~18.2.0",
-    "@dxos/react-ui": "0.8.4-main.5ea62a8",
-    "@dxos/react-ui-theme": "0.8.4-main.5ea62a8"
+    "react": "^19.0.0",
+    "react-dom": "^19.0.0",
+    "@dxos/react-ui": "0.8.4-main.66e292d",
+    "@dxos/react-ui-theme": "0.8.4-main.66e292d"
   },
   "publishConfig": {
     "access": "public"

package/src/AssistantPlugin.tsx

@@ -3,12 +3,14 @@
 //
 
 import { Capabilities, Events, contributes, createIntent, defineModule, definePlugin } from '@dxos/app-framework';
-import { Blueprint } from '@dxos/blueprints';
+import { ResearchGraph } from '@dxos/assistant-toolkit';
+import { Blueprint, Prompt } from '@dxos/blueprints';
 import { Sequence } from '@dxos/conductor';
 import { Type } from '@dxos/echo';
 import { ClientCapabilities, ClientEvents } from '@dxos/plugin-client';
 import { SpaceCapabilities, SpaceEvents } from '@dxos/plugin-space';
-import { defineObjectForm } from '@dxos/plugin-space/types';
+import { type CreateObjectIntent } from '@dxos/plugin-space/types';
+import { HasSubject } from '@dxos/types';
 
 import {
   AiService,
@@ -19,142 +21,144 @@ import {
   IntentResolver,
   LocalModelResolver,
   ReactSurface,
+  Repair,
   Settings,
   Toolkit,
 } from './capabilities';
 import { AssistantEvents } from './events';
 import { meta } from './meta';
 import { translations } from './translations';
-import { Assistant, AssistantAction, ServiceType } from './types';
+import { Assistant, AssistantAction } from './types';
 
-export const AssistantPlugin = () =>
-  definePlugin(meta, [
-    defineModule({
-      id: `${meta.id}/module/translations`,
-      activatesOn: Events.SetupTranslations,
-      activate: () => contributes(Capabilities.Translations, translations),
-    }),
-    defineModule({
-      id: `${meta.id}/module/settings`,
-      activatesOn: Events.SetupSettings,
-      activate: Settings,
-    }),
-    defineModule({
-      id: `${meta.id}/module/state`,
-      // TODO(wittjosiah): Does not integrate with settings store.
-      //   Should this be a different event?
-      //   Should settings store be renamed to be more generic?
-      activatesOn: Events.SetupSettings,
-      activate: AssistantState,
-    }),
-    defineModule({
-      id: `${meta.id}/module/metadata`,
-      activatesOn: Events.SetupMetadata,
-      activate: () => [
-        contributes(Capabilities.Metadata, {
-          id: Type.getTypename(Assistant.Chat),
-          metadata: {
-            icon: 'ph--atom--regular',
-          },
-        }),
-        contributes(Capabilities.Metadata, {
-          id: Type.getTypename(Blueprint.Blueprint),
-          metadata: {
-            icon: 'ph--blueprint--regular',
-          },
-        }),
-        contributes(Capabilities.Metadata, {
-          id: Type.getTypename(Sequence),
-          metadata: {
-            icon: 'ph--circuitry--regular',
-          },
-        }),
-      ],
-    }),
-    defineModule({
-      id: `${meta.id}/module/object-form`,
-      activatesOn: ClientEvents.SetupSchema,
-      activate: () => [
-        contributes(
-          SpaceCapabilities.ObjectForm,
-          defineObjectForm({
-            objectSchema: Assistant.Chat,
-            getIntent: (_, options) => createIntent(AssistantAction.CreateChat, { space: options.space }),
-          }),
-        ),
-        contributes(
-          SpaceCapabilities.ObjectForm,
-          defineObjectForm({
-            objectSchema: Blueprint.Blueprint,
-            formSchema: AssistantAction.BlueprintForm,
-            getIntent: (props) => createIntent(AssistantAction.CreateBlueprint, props),
-          }),
-        ),
-        contributes(
-          SpaceCapabilities.ObjectForm,
-          defineObjectForm({
-            objectSchema: Sequence,
-            getIntent: () => createIntent(AssistantAction.CreateSequence),
-          }),
-        ),
-      ],
-    }),
-    defineModule({
-      id: `${meta.id}/module/schema`,
-      activatesOn: ClientEvents.SetupSchema,
-      activate: () => contributes(ClientCapabilities.Schema, [ServiceType, Assistant.CompanionTo]),
-    }),
-    defineModule({
-      id: `${meta.id}/module/on-space-created`,
-      activatesOn: SpaceEvents.SpaceCreated,
-      activate: () =>
-        contributes(SpaceCapabilities.OnSpaceCreated, ({ rootCollection, space }) =>
-          createIntent(AssistantAction.OnSpaceCreated, { rootCollection, space }),
-        ),
-    }),
-    defineModule({
-      id: `${meta.id}/module/app-graph-builder`,
-      activatesOn: Events.SetupAppGraph,
-      activate: AppGraphBuilder,
-    }),
-    defineModule({
-      id: `${meta.id}/module/intent-resolver`,
-      activatesOn: Events.SetupIntentResolver,
-      activate: IntentResolver,
-    }),
-    defineModule({
-      id: `${meta.id}/module/react-surface`,
-      activatesOn: Events.SetupReactSurface,
-      // TODO(wittjosiah): Should occur before the chat is loaded when surfaces activation is more granular.
-      activatesBefore: [Events.SetupArtifactDefinition],
-      activate: ReactSurface,
-    }),
-    defineModule({
-      id: `${meta.id}/module/edge-model-resolver`,
-      activatesOn: AssistantEvents.SetupAiServiceProviders,
-      activate: EdgeModelResolver,
-    }),
-    defineModule({
-      id: `${meta.id}/module/local-model-resolver`,
-      activatesOn: AssistantEvents.SetupAiServiceProviders,
-      activate: LocalModelResolver,
-    }),
-    defineModule({
-      id: `${meta.id}/module/ai-service`,
-      activatesBefore: [AssistantEvents.SetupAiServiceProviders],
-      // TODO(dmaretskyi): This should activate lazily when the AI chat is used.
-      activatesOn: Events.Startup,
-      activate: AiService,
-    }),
-    defineModule({
-      id: `${meta.id}/module/blueprint`,
-      activatesOn: Events.SetupArtifactDefinition,
-      activate: BlueprintDefinition,
-    }),
-    defineModule({
-      id: `${meta.id}/module/toolkit`,
-      // TODO(wittjosiah): Use a different event.
-      activatesOn: Events.Startup,
-      activate: Toolkit,
-    }),
-  ]);
+export const AssistantPlugin = definePlugin(meta, () => [
+  defineModule({
+    id: `${meta.id}/module/translations`,
+    activatesOn: Events.SetupTranslations,
+    activate: () => contributes(Capabilities.Translations, translations),
+  }),
+  defineModule({
+    id: `${meta.id}/module/settings`,
+    activatesOn: Events.SetupSettings,
+    activate: Settings,
+  }),
+  defineModule({
+    id: `${meta.id}/module/state`,
+    // TODO(wittjosiah): Does not integrate with settings store.
+    //   Should this be a different event?
+    //   Should settings store be renamed to be more generic?
+    activatesOn: Events.SetupSettings,
+    activate: AssistantState,
+  }),
+  defineModule({
+    id: `${meta.id}/module/metadata`,
+    activatesOn: Events.SetupMetadata,
+    activate: () => [
+      contributes(Capabilities.Metadata, {
+        id: Type.getTypename(Assistant.Chat),
+        metadata: {
+          icon: 'ph--atom--regular',
+          iconHue: 'sky',
+          createObjectIntent: ((_, options) =>
+            createIntent(AssistantAction.CreateChat, { space: options.space })) satisfies CreateObjectIntent,
+        },
+      }),
+      contributes(Capabilities.Metadata, {
+        id: Type.getTypename(Blueprint.Blueprint),
+        metadata: {
+          icon: 'ph--blueprint--regular',
+          iconHue: 'sky',
+          inputSchema: AssistantAction.BlueprintForm,
+          createObjectIntent: ((props) =>
+            createIntent(AssistantAction.CreateBlueprint, props)) satisfies CreateObjectIntent,
+        },
+      }),
+      contributes(Capabilities.Metadata, {
+        id: Type.getTypename(Prompt.Prompt),
+        metadata: {
+          icon: 'ph--scroll--regular',
+          iconHue: 'sky',
+          createObjectIntent: (() => createIntent(AssistantAction.CreatePrompt)) satisfies CreateObjectIntent,
+        },
+      }),
+      contributes(Capabilities.Metadata, {
+        id: Type.getTypename(Sequence),
+        metadata: {
+          icon: 'ph--circuitry--regular',
+          iconHue: 'sky',
+          createObjectIntent: (() => createIntent(AssistantAction.CreateSequence)) satisfies CreateObjectIntent,
+          addToCollectionOnCreate: true,
+        },
+      }),
+    ],
+  }),
+  defineModule({
+    id: `${meta.id}/module/schema`,
+    activatesOn: ClientEvents.SetupSchema,
+    activate: () =>
+      contributes(ClientCapabilities.Schema, [
+        Assistant.Chat,
+        Assistant.CompanionTo,
+        Blueprint.Blueprint,
+        HasSubject.HasSubject,
+        Prompt.Prompt,
+        ResearchGraph,
+        Sequence,
+      ]),
+  }),
+  defineModule({
+    id: `${meta.id}/module/on-space-created`,
+    activatesOn: SpaceEvents.SpaceCreated,
+    activate: () =>
+      contributes(SpaceCapabilities.OnCreateSpace, (params) => createIntent(AssistantAction.OnCreateSpace, params)),
+  }),
+  defineModule({
+    id: `${meta.id}/module/repair`,
+    activatesOn: ClientEvents.SpacesReady,
+    activate: Repair,
+  }),
+  defineModule({
+    id: `${meta.id}/module/app-graph-builder`,
+    activatesOn: Events.SetupAppGraph,
+    activate: AppGraphBuilder,
+  }),
+  defineModule({
+    id: `${meta.id}/module/intent-resolver`,
+    activatesOn: Events.SetupIntentResolver,
+    activate: IntentResolver,
+  }),
+  defineModule({
+    id: `${meta.id}/module/react-surface`,
+    activatesOn: Events.SetupReactSurface,
+    // TODO(wittjosiah): Should occur before the chat is loaded when surfaces activation is more granular.
+    activatesBefore: [Events.SetupArtifactDefinition],
+    activate: ReactSurface,
+  }),
+  defineModule({
+    id: `${meta.id}/module/edge-model-resolver`,
+    activatesOn: AssistantEvents.SetupAiServiceProviders,
+    activate: EdgeModelResolver,
+  }),
+  defineModule({
+    id: `${meta.id}/module/local-model-resolver`,
+    activatesOn: AssistantEvents.SetupAiServiceProviders,
+    activate: LocalModelResolver,
+  }),
+  defineModule({
+    id: `${meta.id}/module/ai-service`,
+    activatesBefore: [AssistantEvents.SetupAiServiceProviders],
+    // TODO(dmaretskyi): This should activate lazily when the AI chat is used.
+    activatesOn: Events.Startup,
+    activate: AiService,
+  }),
+  defineModule({
+    id: `${meta.id}/module/blueprint`,
+    activatesOn: Events.SetupArtifactDefinition,
+    activate: BlueprintDefinition,
+  }),
+  defineModule({
+    id: `${meta.id}/module/toolkit`,
+    // TODO(wittjosiah): Use a different event.
+    activatesOn: Events.Startup,
+    activate: Toolkit,
+  }),
+]);

package/src/capabilities/ai-service.ts

@@ -2,15 +2,14 @@
 // Copyright 2023 DXOS.org
 //
 
-import { Effect, Layer } from 'effect';
+import * as Effect from 'effect/Effect';
+import * as Layer from 'effect/Layer';
 
-import { AiServiceRouter } from '@dxos/ai';
-import { type PluginContext, contributes } from '@dxos/app-framework';
-
-import { AssistantCapabilities } from './capabilities';
+import * as AiServiceRouter from '@dxos/ai/AiServiceRouter';
+import { Capabilities, type PluginContext, contributes } from '@dxos/app-framework';
 
 export default (context: PluginContext) => {
-  const resolvers = context.getCapabilities(AssistantCapabilities.AiModelResolver);
+  const resolvers = context.getCapabilities(Capabilities.AiModelResolver);
 
   // TODO(dmaretskyi): Extract function to reduce them.
   const combinedLayer = resolvers.reduce(
@@ -21,7 +20,7 @@ export default (context: PluginContext) => {
   return [
     // TODO(dmaretskyi): Read config from settings.
     contributes(
-      AssistantCapabilities.AiServiceLayer,
+      Capabilities.AiServiceLayer,
       AiServiceRouter.AiModelResolver.buildAiService.pipe(Layer.provide(combinedLayer)),
     ),
   ];