@dxos/plugin-assistant 0.8.4-main.c4373fc → 0.8.4-main.c85a9c8dae

package/package.json

@@ -1,9 +1,13 @@
 {
   "name": "@dxos/plugin-assistant",
-  "version": "0.8.4-main.c4373fc",
+  "version": "0.8.4-main.c85a9c8dae",
   "description": "Assistant plugin",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/dxos/dxos"
+  },
   "license": "MIT",
   "author": "DXOS.org",
   "sideEffects": true,
@@ -15,6 +19,13 @@
       "browser": "./dist/lib/browser/index.mjs",
       "node": "./dist/lib/node-esm/index.mjs"
     },
+    "./blueprints": {
+      "source": "./src/blueprints/index.ts",
+      "types": "./dist/types/src/blueprints/index.d.ts",
+      "browser": "./dist/lib/browser/blueprints/index.mjs",
+      "node": "./dist/lib/node-esm/blueprints/index.mjs",
+      "default": "./src/blueprints/index.ts"
+    },
     "./types": {
       "source": "./src/types/index.ts",
       "types": "./dist/types/src/types/index.d.ts",
@@ -25,6 +36,9 @@
   "types": "dist/types/src/index.d.ts",
   "typesVersions": {
     "*": {
+      "blueprints": [
+        "dist/types/src/blueprints/index.d.ts"
+      ],
       "types": [
         "dist/types/src/types/index.d.ts"
       ]
@@ -37,132 +51,126 @@
   ],
   "dependencies": {
     "@codemirror/autocomplete": "^6.19.0",
+    "@codemirror/lang-html": "^6.4.11",
+    "@codemirror/lang-markdown": "6.3.4",
+    "@codemirror/lang-xml": "^6.1.0",
+    "@codemirror/language": "^6.11.3",
+    "@codemirror/lint": "6.8.5",
     "@codemirror/state": "^6.5.2",
     "@codemirror/view": "^6.38.4",
-    "@effect/ai": "0.29.1",
-    "@effect/ai-anthropic": "^0.19.2",
-    "@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.12.1",
-    "@preact/signals-react": "^3.3.0",
+    "@effect-atom/atom": "^0.5.1",
+    "@effect-atom/atom-react": "^0.5.0",
+    "@effect/ai": "0.33.2",
+    "@effect/ai-anthropic": "0.23.0",
+    "@effect/ai-openai": "0.37.2",
+    "@effect/experimental": "0.58.0",
+    "@effect/platform-browser": "0.74.0",
     "@radix-ui/react-context": "1.1.1",
-    "effect": "3.18.3",
+    "effect": "3.19.16",
     "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.c4373fc",
-    "@dxos/app-framework": "0.8.4-main.c4373fc",
-    "@dxos/assistant": "0.8.4-main.c4373fc",
-    "@dxos/async": "0.8.4-main.c4373fc",
-    "@dxos/assistant-testing": "0.8.4-main.c4373fc",
-    "@dxos/blueprints": "0.8.4-main.c4373fc",
-    "@dxos/client": "0.8.4-main.c4373fc",
-    "@dxos/compute": "0.8.4-main.c4373fc",
-    "@dxos/context": "0.8.4-main.c4373fc",
-    "@dxos/config": "0.8.4-main.c4373fc",
-    "@dxos/conductor": "0.8.4-main.c4373fc",
-    "@dxos/devtools": "0.8.4-main.c4373fc",
-    "@dxos/echo-db": "0.8.4-main.c4373fc",
-    "@dxos/echo-protocol": "0.8.4-main.c4373fc",
-    "@dxos/echo-query": "0.8.4-main.c4373fc",
-    "@dxos/edge-client": "0.8.4-main.c4373fc",
-    "@dxos/debug": "0.8.4-main.c4373fc",
-    "@dxos/echo": "0.8.4-main.c4373fc",
-    "@dxos/effect": "0.8.4-main.c4373fc",
-    "@dxos/functions": "0.8.4-main.c4373fc",
-    "@dxos/graph": "0.8.4-main.c4373fc",
-    "@dxos/invariant": "0.8.4-main.c4373fc",
-    "@dxos/keys": "0.8.4-main.c4373fc",
-    "@dxos/live-object": "0.8.4-main.c4373fc",
-    "@dxos/lit-ui": "0.8.4-main.c4373fc",
-    "@dxos/local-storage": "0.8.4-main.c4373fc",
-    "@dxos/plugin-attention": "0.8.4-main.c4373fc",
-    "@dxos/plugin-automation": "0.8.4-main.c4373fc",
-    "@dxos/log": "0.8.4-main.c4373fc",
-    "@dxos/plugin-client": "0.8.4-main.c4373fc",
-    "@dxos/plugin-deck": "0.8.4-main.c4373fc",
-    "@dxos/plugin-board": "0.8.4-main.c4373fc",
-    "@dxos/plugin-explorer": "0.8.4-main.c4373fc",
-    "@dxos/plugin-graph": "0.8.4-main.c4373fc",
-    "@dxos/plugin-markdown": "0.8.4-main.c4373fc",
-    "@dxos/plugin-preview": "0.8.4-main.c4373fc",
-    "@dxos/plugin-script": "0.8.4-main.c4373fc",
-    "@dxos/plugin-sketch": "0.8.4-main.c4373fc",
-    "@dxos/plugin-stack": "0.8.4-main.c4373fc",
-    "@dxos/plugin-transcription": "0.8.4-main.c4373fc",
-    "@dxos/plugin-wnfs": "0.8.4-main.c4373fc",
-    "@dxos/protocols": "0.8.4-main.c4373fc",
-    "@dxos/react-client": "0.8.4-main.c4373fc",
-    "@dxos/react-edge-client": "0.8.4-main.c4373fc",
-    "@dxos/plugin-space": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-attention": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-chat": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-components": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-canvas-compute": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-editor": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-form": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-graph": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-canvas-editor": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-list": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-menu": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-searchlist": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-stack": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-syntax-highlighter": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-sfx": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-tabs": "0.8.4-main.c4373fc",
-    "@dxos/util": "0.8.4-main.c4373fc",
-    "@dxos/schema": "0.8.4-main.c4373fc"
+    "@dxos/ai": "0.8.4-main.c85a9c8dae",
+    "@dxos/app-toolkit": "0.8.4-main.c85a9c8dae",
+    "@dxos/assistant": "0.8.4-main.c85a9c8dae",
+    "@dxos/assistant-toolkit": "0.8.4-main.c85a9c8dae",
+    "@dxos/async": "0.8.4-main.c85a9c8dae",
+    "@dxos/blueprints": "0.8.4-main.c85a9c8dae",
+    "@dxos/client": "0.8.4-main.c85a9c8dae",
+    "@dxos/compute": "0.8.4-main.c85a9c8dae",
+    "@dxos/conductor": "0.8.4-main.c85a9c8dae",
+    "@dxos/debug": "0.8.4-main.c85a9c8dae",
+    "@dxos/devtools": "0.8.4-main.c85a9c8dae",
+    "@dxos/context": "0.8.4-main.c85a9c8dae",
+    "@dxos/echo": "0.8.4-main.c85a9c8dae",
+    "@dxos/echo-atom": "0.8.4-main.c85a9c8dae",
+    "@dxos/echo-db": "0.8.4-main.c85a9c8dae",
+    "@dxos/echo-protocol": "0.8.4-main.c85a9c8dae",
+    "@dxos/echo-query": "0.8.4-main.c85a9c8dae",
+    "@dxos/config": "0.8.4-main.c85a9c8dae",
+    "@dxos/edge-client": "0.8.4-main.c85a9c8dae",
+    "@dxos/effect": "0.8.4-main.c85a9c8dae",
+    "@dxos/functions": "0.8.4-main.c85a9c8dae",
+    "@dxos/functions-runtime": "0.8.4-main.c85a9c8dae",
+    "@dxos/graph": "0.8.4-main.c85a9c8dae",
+    "@dxos/keys": "0.8.4-main.c85a9c8dae",
+    "@dxos/app-framework": "0.8.4-main.c85a9c8dae",
+    "@dxos/lit-ui": "0.8.4-main.c85a9c8dae",
+    "@dxos/invariant": "0.8.4-main.c85a9c8dae",
+    "@dxos/operation": "0.8.4-main.c85a9c8dae",
+    "@dxos/log": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-attention": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-automation": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-client": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-deck": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-graph": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-space": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-status-bar": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-map": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-markdown": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-table": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-transcription": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-edge-client": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-attention": "0.8.4-main.c85a9c8dae",
+    "@dxos/protocols": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-canvas-compute": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-client": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-canvas-editor": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-chat": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-components": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-editor": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-form": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-list": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-graph": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-markdown": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-mosaic": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-searchlist": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-sfx": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-menu": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-syntax-highlighter": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-tabs": "0.8.4-main.c85a9c8dae",
+    "@dxos/schema": "0.8.4-main.c85a9c8dae",
+    "@dxos/types": "0.8.4-main.c85a9c8dae",
+    "@dxos/ui": "0.8.4-main.c85a9c8dae",
+    "@dxos/ui-editor": "0.8.4-main.c85a9c8dae",
+    "@dxos/util": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-stack": "0.8.4-main.c85a9c8dae"
   },
   "devDependencies": {
-    "@effect-rx/rx-react": "0.42.4",
-    "@effect/platform": "0.92.1",
+    "@effect/platform": "0.94.4",
     "@types/lodash.defaultsdeep": "^4.6.6",
-    "@types/react": "~19.2.2",
-    "@types/react-dom": "~19.2.1",
-    "effect": "3.18.3",
+    "@types/react": "~19.2.7",
+    "@types/react-dom": "~19.2.3",
+    "effect": "3.19.16",
     "openapi-types": "^12.1.3",
-    "react": "~19.2.0",
-    "react-dom": "~19.2.0",
-    "vite": "7.1.9",
-    "@dxos/assistant-testing": "0.8.4-main.c4373fc",
-    "@dxos/devtools": "0.8.4-main.c4373fc",
-    "@dxos/echo-db": "0.8.4-main.c4373fc",
-    "@dxos/plugin-board": "0.8.4-main.c4373fc",
-    "@dxos/plugin-attention": "0.8.4-main.c4373fc",
-    "@dxos/plugin-chess": "0.8.4-main.c4373fc",
-    "@dxos/plugin-inbox": "0.8.4-main.c4373fc",
-    "@dxos/plugin-preview": "0.8.4-main.c4373fc",
-    "@dxos/plugin-map": "0.8.4-main.c4373fc",
-    "@dxos/plugin-project": "0.8.4-main.c4373fc",
-    "@dxos/plugin-script": "0.8.4-main.c4373fc",
-    "@dxos/plugin-sketch": "0.8.4-main.c4373fc",
-    "@dxos/plugin-markdown": "0.8.4-main.c4373fc",
-    "@dxos/plugin-stack": "0.8.4-main.c4373fc",
-    "@dxos/plugin-theme": "0.8.4-main.c4373fc",
-    "@dxos/plugin-table": "0.8.4-main.c4373fc",
-    "@dxos/plugin-token-manager": "0.8.4-main.c4373fc",
-    "@dxos/plugin-thread": "0.8.4-main.c4373fc",
-    "@dxos/plugin-wnfs": "0.8.4-main.c4373fc",
-    "@dxos/react-ui": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-menu": "0.8.4-main.c4373fc",
-    "@dxos/random": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-theme": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-table": "0.8.4-main.c4373fc",
-    "@dxos/storybook-utils": "0.8.4-main.c4373fc"
+    "react": "~19.2.3",
+    "react-dom": "~19.2.3",
+    "vite": "^7.1.11",
+    "@dxos/devtools": "0.8.4-main.c85a9c8dae",
+    "@dxos/assistant-toolkit": "0.8.4-main.c85a9c8dae",
+    "@dxos/echo-db": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-chess": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-map": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-testing": "0.8.4-main.c85a9c8dae",
+    "@dxos/plugin-table": "0.8.4-main.c85a9c8dae",
+    "@dxos/random": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-menu": "0.8.4-main.c85a9c8dae",
+    "@dxos/storybook-utils": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui-table": "0.8.4-main.c85a9c8dae",
+    "@dxos/ui-theme": "0.8.4-main.c85a9c8dae"
   },
   "peerDependencies": {
-    "@effect-rx/rx-react": "^0.34.1",
-    "@effect/platform": "^0.80.12",
-    "effect": "^3.13.3",
-    "react": "^19.0.0",
-    "react-dom": "^19.0.0",
-    "@dxos/react-ui": "0.8.4-main.c4373fc",
-    "@dxos/react-ui-theme": "0.8.4-main.c4373fc"
+    "@effect-atom/atom-react": "^0.5.0",
+    "@effect/platform": "0.94.4",
+    "effect": "3.19.16",
+    "react": "~19.2.3",
+    "react-dom": "~19.2.3",
+    "@dxos/ui-theme": "0.8.4-main.c85a9c8dae",
+    "@dxos/react-ui": "0.8.4-main.c85a9c8dae"
   },
   "publishConfig": {
     "access": "public"

package/src/AssistantPlugin.tsx

@@ -2,14 +2,21 @@
 // Copyright 2023 DXOS.org
 //
 
-import { Capabilities, Events, contributes, createIntent, defineModule, definePlugin } from '@dxos/app-framework';
-import { ResearchGraph, ResearchOn } from '@dxos/assistant-testing';
+import * as Effect from 'effect/Effect';
+
+import { ActivationEvents, Capability, Plugin } from '@dxos/app-framework';
+import { AppActivationEvents, AppPlugin } from '@dxos/app-toolkit';
+import { Chat, Plan, Project, ProjectBlueprint, 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 { Obj, Type } from '@dxos/echo';
+import { type SpaceId } from '@dxos/keys';
+import { Operation } from '@dxos/operation';
+import { AutomationCapabilities } from '@dxos/plugin-automation/types';
+import { ClientEvents } from '@dxos/plugin-client/types';
+import { SpaceCapabilities, SpaceEvents } from '@dxos/plugin-space/types';
+import { type CreateObject } from '@dxos/plugin-space/types';
+import { HasSubject } from '@dxos/types';
 
 import {
   AiService,
@@ -17,151 +24,149 @@ import {
   AssistantState,
   BlueprintDefinition,
   EdgeModelResolver,
-  IntentResolver,
   LocalModelResolver,
+  OperationResolver,
   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 { AssistantEvents, AssistantOperation } 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),
+export const AssistantPlugin = Plugin.define(meta).pipe(
+  AppPlugin.addAppGraphModule({ activate: AppGraphBuilder }),
+  AppPlugin.addBlueprintDefinitionModule({ activate: BlueprintDefinition }),
+  AppPlugin.addMetadataModule({
+    metadata: [
+      {
+        id: Type.getTypename(Chat.Chat),
         metadata: {
           icon: 'ph--atom--regular',
+          iconHue: 'sky',
+          createObject: ((props, { db }) =>
+            Operation.invoke(AssistantOperation.CreateChat, { db, name: props?.name }).pipe(
+              Effect.map(({ object }) => object),
+            )) satisfies CreateObject,
         },
-      }),
-      contributes(Capabilities.Metadata, {
+      },
+      {
         id: Type.getTypename(Blueprint.Blueprint),
         metadata: {
           icon: 'ph--blueprint--regular',
+          iconHue: 'sky',
+          inputSchema: AssistantOperation.BlueprintForm,
+          createObject: ((props) => Effect.sync(() => Blueprint.make(props))) satisfies CreateObject,
+        },
+      },
+      {
+        id: Type.getTypename(Prompt.Prompt),
+        metadata: {
+          icon: 'ph--scroll--regular',
+          iconHue: 'sky',
+          createObject: ((props) => Effect.sync(() => Prompt.make(props))) satisfies CreateObject,
         },
-      }),
-      contributes(Capabilities.Metadata, {
+      },
+      {
         id: Type.getTypename(Sequence),
         metadata: {
           icon: 'ph--circuitry--regular',
+          iconHue: 'sky',
+          createObject: ((props) => Effect.sync(() => Obj.make(Sequence, props))) satisfies CreateObject,
+          addToCollectionOnCreate: true,
         },
-      }),
+      },
+      {
+        id: Type.getTypename(Project.Project),
+        metadata: {
+          icon: 'ph--circuitry--regular',
+          iconHue: 'sky',
+          createObject: ((_, { db }) =>
+            Project.makeInitialized(
+              {
+                name: 'New Project',
+                spec: 'Not specified yet',
+              },
+              ProjectBlueprint.make(),
+            ).pipe(withComputeRuntime(db.spaceId))) satisfies CreateObject,
+          addToCollectionOnCreate: true,
+        },
+      },
     ],
   }),
-  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),
-        }),
-      ),
+  AppPlugin.addOperationResolverModule({ activate: OperationResolver }),
+  AppPlugin.addSchemaModule({
+    schema: [
+      Chat.Chat,
+      Chat.CompanionTo,
+      Blueprint.Blueprint,
+      HasSubject.HasSubject,
+      Prompt.Prompt,
+      ResearchGraph.ResearchGraph,
+      Project.Project,
+      Plan.Plan,
+      Sequence,
     ],
   }),
-  defineModule({
-    id: `${meta.id}/module/schema`,
-    activatesOn: ClientEvents.SetupSchema,
-    activate: () =>
-      contributes(ClientCapabilities.Schema, [
-        ServiceType,
-        Assistant.CompanionTo,
-        ResearchGraph,
-        ResearchOn,
-        Prompt.Prompt,
-      ]),
+  AppPlugin.addSettingsModule({ activate: Settings }),
+  AppPlugin.addSurfaceModule({
+    activate: ReactSurface,
+    activatesBefore: [AppActivationEvents.SetupArtifactDefinition],
+  }),
+  AppPlugin.addTranslationsModule({ translations }),
+  Plugin.addModule({
+    // TODO(wittjosiah): Does not integrate with settings store.
+    //   Should this be a different event?
+    //   Should settings store be renamed to be more generic?
+    activatesOn: AppActivationEvents.SetupSettings,
+    activate: AssistantState,
   }),
-  defineModule({
-    id: `${meta.id}/module/on-space-created`,
+  Plugin.addModule({
+    id: 'on-space-created',
     activatesOn: SpaceEvents.SpaceCreated,
     activate: () =>
-      contributes(SpaceCapabilities.onCreateSpace, ({ rootCollection, space }) =>
-        createIntent(AssistantAction.onCreateSpace, { rootCollection, space }),
+      Effect.succeed(
+        Capability.contributes(SpaceCapabilities.OnCreateSpace, (params) =>
+          Operation.invoke(AssistantOperation.OnCreateSpace, params),
+        ),
       ),
   }),
-  defineModule({
-    id: `${meta.id}/module/app-graph-builder`,
-    activatesOn: Events.SetupAppGraph,
-    activate: AppGraphBuilder,
+  Plugin.addModule({
+    activatesOn: ClientEvents.SpacesReady,
+    activate: Repair,
   }),
-  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`,
+  Plugin.addModule({
     activatesOn: AssistantEvents.SetupAiServiceProviders,
     activate: EdgeModelResolver,
   }),
-  defineModule({
-    id: `${meta.id}/module/local-model-resolver`,
+  Plugin.addModule({
     activatesOn: AssistantEvents.SetupAiServiceProviders,
     activate: LocalModelResolver,
   }),
-  defineModule({
-    id: `${meta.id}/module/ai-service`,
+  Plugin.addModule({
     activatesBefore: [AssistantEvents.SetupAiServiceProviders],
     // TODO(dmaretskyi): This should activate lazily when the AI chat is used.
-    activatesOn: Events.Startup,
+    activatesOn: ActivationEvents.Startup,
     activate: AiService,
   }),
-  defineModule({
-    id: `${meta.id}/module/blueprint`,
-    activatesOn: Events.SetupArtifactDefinition,
-    activate: BlueprintDefinition,
-  }),
-  defineModule({
-    id: `${meta.id}/module/toolkit`,
+  Plugin.addModule({
     // TODO(wittjosiah): Use a different event.
-    activatesOn: Events.Startup,
+    activatesOn: ActivationEvents.Startup,
     activate: Toolkit,
   }),
-]);
+  Plugin.make,
+);
+
+// TODO(dmaretskyi): Extract to a helper module.
+const withComputeRuntime =
+  (spaceId: SpaceId) =>
+  <A, E, R>(
+    effect: Effect.Effect<A, E, R>,
+  ): Effect.Effect<A, E, Exclude<R, AutomationCapabilities.ComputeServices> | Capability.Service> =>
+    Effect.gen(function* () {
+      // TODO(dmaretskyi): Capability.get has `Error` in the error channel. We should throw those as defects instead.
+      const provider = yield* Capability.get(AutomationCapabilities.ComputeRuntime).pipe(Effect.orDie);
+      const runtime = yield* provider.getRuntime(spaceId).runtimeEffect;
+      return yield* effect.pipe(Effect.provide(runtime));
+    });