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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dxos/plugin-assistant",
-  "version": "0.8.4-main.5acf9ea",
+  "version": "0.8.4-main.5ea62a8",
   "description": "Assistant plugin",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
@@ -10,16 +10,16 @@
   "type": "module",
   "exports": {
     ".": {
+      "source": "./src/index.ts",
       "types": "./dist/types/src/index.d.ts",
       "browser": "./dist/lib/browser/index.mjs",
-      "node": "./dist/lib/node-esm/index.mjs",
-      "source": "./src/index.ts"
+      "node": "./dist/lib/node-esm/index.mjs"
     },
     "./types": {
+      "source": "./src/types/index.ts",
       "types": "./dist/types/src/types/index.d.ts",
       "browser": "./dist/lib/browser/types/index.mjs",
-      "node": "./dist/lib/node-esm/types/index.mjs",
-      "source": "./src/types/index.ts"
+      "node": "./dist/lib/node-esm/types/index.mjs"
     }
   },
   "types": "dist/types/src/index.d.ts",
@@ -41,100 +41,118 @@
     "@codemirror/search": "^6.5.6",
     "@codemirror/state": "^6.4.1",
     "@codemirror/view": "^6.34.1",
-    "@effect/ai": "0.23.0",
+    "@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",
     "@preact-signals/safe-react": "^0.9.0",
     "@preact/signals-core": "^1.9.0",
     "@preact/signals-react": "^3.2.0",
     "@radix-ui/react-context": "1.1.1",
-    "@xiechao/codemirror-lang-handlebars": "1.0.4",
-    "effect": "3.17.0",
+    "effect": "3.17.7",
     "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.5acf9ea",
-    "@dxos/app-framework": "0.8.4-main.5acf9ea",
-    "@dxos/assistant": "0.8.4-main.5acf9ea",
-    "@dxos/async": "0.8.4-main.5acf9ea",
-    "@dxos/blueprints": "0.8.4-main.5acf9ea",
-    "@dxos/compute": "0.8.4-main.5acf9ea",
-    "@dxos/client": "0.8.4-main.5acf9ea",
-    "@dxos/config": "0.8.4-main.5acf9ea",
-    "@dxos/conductor": "0.8.4-main.5acf9ea",
-    "@dxos/context": "0.8.4-main.5acf9ea",
-    "@dxos/debug": "0.8.4-main.5acf9ea",
-    "@dxos/echo": "0.8.4-main.5acf9ea",
-    "@dxos/devtools": "0.8.4-main.5acf9ea",
-    "@dxos/echo-protocol": "0.8.4-main.5acf9ea",
-    "@dxos/echo-schema": "0.8.4-main.5acf9ea",
-    "@dxos/echo-db": "0.8.4-main.5acf9ea",
-    "@dxos/edge-client": "0.8.4-main.5acf9ea",
-    "@dxos/effect": "0.8.4-main.5acf9ea",
-    "@dxos/functions": "0.8.4-main.5acf9ea",
-    "@dxos/invariant": "0.8.4-main.5acf9ea",
-    "@dxos/graph": "0.8.4-main.5acf9ea",
-    "@dxos/keys": "0.8.4-main.5acf9ea",
-    "@dxos/local-storage": "0.8.4-main.5acf9ea",
-    "@dxos/log": "0.8.4-main.5acf9ea",
-    "@dxos/live-object": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-client": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-deck": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-explorer": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-graph": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-markdown": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-inbox": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-preview": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-script": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-sketch": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-space": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-stack": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-wnfs": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-transcription": "0.8.4-main.5acf9ea",
-    "@dxos/protocols": "0.8.4-main.5acf9ea",
-    "@dxos/react-client": "0.8.4-main.5acf9ea",
-    "@dxos/react-edge-client": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-attention": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-canvas-editor": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-canvas-compute": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-chat": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-components": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-editor": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-form": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-list": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-graph": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-sfx": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-stack": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-tag-picker": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-syntax-highlighter": "0.8.4-main.5acf9ea",
-    "@dxos/schema": "0.8.4-main.5acf9ea",
-    "@dxos/util": "0.8.4-main.5acf9ea"
+    "@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"
   },
   "devDependencies": {
     "@effect-rx/rx-react": "^0.38.0",
-    "@effect/platform": "0.89.0",
+    "@effect/platform": "0.90.2",
     "@types/lodash.defaultsdeep": "^4.6.6",
     "@types/react": "~18.2.0",
     "@types/react-dom": "~18.2.0",
-    "effect": "3.17.0",
+    "effect": "3.17.7",
     "openapi-types": "^12.1.3",
     "react": "~18.2.0",
     "react-dom": "~18.2.0",
-    "vite": "5.4.7",
-    "@dxos/assistant-testing": "0.8.4-main.5acf9ea",
-    "@dxos/echo-db": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-chess": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-inbox": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-map": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-table": "0.8.4-main.5acf9ea",
-    "@dxos/random": "0.8.4-main.5acf9ea",
-    "@dxos/plugin-markdown": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-menu": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-theme": "0.8.4-main.5acf9ea",
-    "@dxos/storybook-utils": "0.8.4-main.5acf9ea"
+    "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"
   },
   "peerDependencies": {
     "@effect-rx/rx-react": "^0.34.1",
@@ -142,8 +160,8 @@
     "effect": "^3.13.3",
     "react": "~18.2.0",
     "react-dom": "~18.2.0",
-    "@dxos/react-ui": "0.8.4-main.5acf9ea",
-    "@dxos/react-ui-theme": "0.8.4-main.5acf9ea"
+    "@dxos/react-ui": "0.8.4-main.5ea62a8",
+    "@dxos/react-ui-theme": "0.8.4-main.5ea62a8"
   },
   "publishConfig": {
     "access": "public"

package/src/AssistantPlugin.tsx

@@ -2,9 +2,6 @@
 // Copyright 2023 DXOS.org
 //
 
-import { Layer } from 'effect';
-
-import { AiServiceTestingPreset } from '@dxos/ai/testing';
 import { Capabilities, Events, contributes, createIntent, defineModule, definePlugin } from '@dxos/app-framework';
 import { Blueprint } from '@dxos/blueprints';
 import { Sequence } from '@dxos/conductor';
@@ -13,11 +10,22 @@ import { ClientCapabilities, ClientEvents } from '@dxos/plugin-client';
 import { SpaceCapabilities, SpaceEvents } from '@dxos/plugin-space';
 import { defineObjectForm } from '@dxos/plugin-space/types';
 
-import { AppGraphBuilder, IntentResolver, ReactSurface, Settings } from './capabilities';
-import { AssistantCapabilities } from './capability-definitions';
+import {
+  AiService,
+  AppGraphBuilder,
+  AssistantState,
+  BlueprintDefinition,
+  EdgeModelResolver,
+  IntentResolver,
+  LocalModelResolver,
+  ReactSurface,
+  Settings,
+  Toolkit,
+} from './capabilities';
+import { AssistantEvents } from './events';
 import { meta } from './meta';
 import { translations } from './translations';
-import { Assistant, ServiceType } from './types';
+import { Assistant, AssistantAction, ServiceType } from './types';
 
 export const AssistantPlugin = () =>
   definePlugin(meta, [
@@ -31,6 +39,14 @@ export const AssistantPlugin = () =>
       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,
@@ -63,22 +79,22 @@ export const AssistantPlugin = () =>
           SpaceCapabilities.ObjectForm,
           defineObjectForm({
             objectSchema: Assistant.Chat,
-            getIntent: (_, options) => createIntent(Assistant.CreateChat, { space: options.space }),
+            getIntent: (_, options) => createIntent(AssistantAction.CreateChat, { space: options.space }),
           }),
         ),
         contributes(
           SpaceCapabilities.ObjectForm,
           defineObjectForm({
             objectSchema: Blueprint.Blueprint,
-            formSchema: Assistant.BlueprintForm,
-            getIntent: (props) => createIntent(Assistant.CreateBlueprint, props),
+            formSchema: AssistantAction.BlueprintForm,
+            getIntent: (props) => createIntent(AssistantAction.CreateBlueprint, props),
           }),
         ),
         contributes(
           SpaceCapabilities.ObjectForm,
           defineObjectForm({
             objectSchema: Sequence,
-            getIntent: () => createIntent(Assistant.CreateSequence),
+            getIntent: () => createIntent(AssistantAction.CreateSequence),
           }),
         ),
       ],
@@ -92,8 +108,8 @@ export const AssistantPlugin = () =>
       id: `${meta.id}/module/on-space-created`,
       activatesOn: SpaceEvents.SpaceCreated,
       activate: () =>
-        contributes(SpaceCapabilities.OnSpaceCreated, ({ space, rootCollection }) =>
-          createIntent(Assistant.OnSpaceCreated, { space, rootCollection }),
+        contributes(SpaceCapabilities.OnSpaceCreated, ({ rootCollection, space }) =>
+          createIntent(AssistantAction.OnSpaceCreated, { rootCollection, space }),
         ),
     }),
     defineModule({
@@ -113,12 +129,32 @@ export const AssistantPlugin = () =>
       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: () => [
-        // TODO(dmaretskyi): Read config from settings.
-        contributes(AssistantCapabilities.AiServiceLayer, AiServiceTestingPreset('edge-remote').pipe(Layer.orDie)),
-      ],
+      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

@@ -0,0 +1,28 @@
+//
+// Copyright 2023 DXOS.org
+//
+
+import { Effect, Layer } from 'effect';
+
+import { AiServiceRouter } from '@dxos/ai';
+import { type PluginContext, contributes } from '@dxos/app-framework';
+
+import { AssistantCapabilities } from './capabilities';
+
+export default (context: PluginContext) => {
+  const resolvers = context.getCapabilities(AssistantCapabilities.AiModelResolver);
+
+  // TODO(dmaretskyi): Extract function to reduce them.
+  const combinedLayer = resolvers.reduce(
+    (acc, resolver) => resolver.pipe(Layer.provide(acc)),
+    AiServiceRouter.AiModelResolver.fromModelMap(Effect.succeed({})), // Empty resolver as fallback.
+  );
+
+  return [
+    // TODO(dmaretskyi): Read config from settings.
+    contributes(
+      AssistantCapabilities.AiServiceLayer,
+      AiServiceRouter.AiModelResolver.buildAiService.pipe(Layer.provide(combinedLayer)),
+    ),
+  ];
+};

package/src/capabilities/app-graph-builder.ts

@@ -7,24 +7,26 @@ import { Option, pipe } from 'effect';
 
 import {
   Capabilities,
-  contributes,
-  createIntent,
   LayoutAction,
-  type PromiseIntentDispatcher,
   type PluginContext,
+  type PromiseIntentDispatcher,
+  contributes,
+  createIntent,
 } from '@dxos/app-framework';
 import { Sequence } from '@dxos/conductor';
 import { Obj } from '@dxos/echo';
 import { invariant } from '@dxos/invariant';
 import { ClientCapabilities } from '@dxos/plugin-client';
-import { PLANK_COMPANION_TYPE, ATTENDABLE_PATH_SEPARATOR } from '@dxos/plugin-deck/types';
-import { createExtension, ROOT_ID } from '@dxos/plugin-graph';
-import { getActiveSpace, rxFromQuery } from '@dxos/plugin-space';
+import { ATTENDABLE_PATH_SEPARATOR, PLANK_COMPANION_TYPE } from '@dxos/plugin-deck/types';
+import { ROOT_ID, createExtension, rxFromSignal } from '@dxos/plugin-graph';
+import { getActiveSpace } from '@dxos/plugin-space';
 import { SpaceAction } from '@dxos/plugin-space/types';
-import { type Space, Filter, Query, type QueryResult, fullyQualifiedId, getSpace } from '@dxos/react-client/echo';
+import { Query, type Space, fullyQualifiedId } from '@dxos/react-client/echo';
 
 import { ASSISTANT_DIALOG, meta } from '../meta';
-import { Assistant } from '../types';
+import { Assistant, AssistantAction } from '../types';
+
+import { AssistantCapabilities } from './capabilities';
 
 export default (context: PluginContext) =>
   contributes(Capabilities.AppGraphBuilder, [
@@ -79,44 +81,36 @@ export default (context: PluginContext) =>
     }),
 
     createExtension({
-      id: `${meta.id}/object-chat-companion`,
+      id: `${meta.id}/companion-chat`,
       connector: (node) => {
-        let query: QueryResult<Assistant.Chat> | undefined;
-        return Rx.make((get) => {
-          const nodeOption = get(node);
-          if (Option.isNone(nodeOption)) {
-            return [];
-          }
-
-          const object = nodeOption.value.data;
-          if (!Obj.isObject(object)) {
-            return [];
-          }
-
-          const space = getSpace(object);
-          if (!space) {
-            return [];
-          }
-
-          if (!query) {
-            query = space.db.query(Query.select(Filter.ids(object.id)).targetOf(Assistant.CompanionTo).source());
-          }
+        return Rx.make((get) =>
+          pipe(
+            get(node),
+            Option.flatMap((node) => (Obj.isObject(node.data) ? Option.some(node.data) : Option.none())),
+            Option.map((object) => {
+              const currentChat = get(
+                rxFromSignal(
+                  () => context.getCapability(AssistantCapabilities.State).currentChat[fullyQualifiedId(object)],
+                ),
+              );
 
-          const chat = get(rxFromQuery(query))[0];
-          return [
-            {
-              id: [fullyQualifiedId(object), 'assistant-chat'].join(ATTENDABLE_PATH_SEPARATOR),
-              type: PLANK_COMPANION_TYPE,
-              data: chat ?? 'assistant-chat',
-              properties: {
-                label: ['assistant chat label', { ns: meta.id }],
-                icon: 'ph--sparkle--regular',
-                position: 'hoist',
-                disposition: 'hidden',
-              },
-            },
-          ];
-        });
+              return [
+                {
+                  id: [fullyQualifiedId(object), 'assistant-chat'].join(ATTENDABLE_PATH_SEPARATOR),
+                  type: PLANK_COMPANION_TYPE,
+                  data: currentChat ?? 'assistant-chat',
+                  properties: {
+                    label: ['assistant chat label', { ns: meta.id }],
+                    icon: 'ph--sparkle--regular',
+                    position: 'hoist',
+                    disposition: 'hidden',
+                  },
+                },
+              ];
+            }),
+            Option.getOrElse(() => []),
+          ),
+        );
       },
     }),
 
@@ -155,14 +149,13 @@ const getOrCreateChat = async (
   const { objects: relatedChats } = await space.db
     .query(Query.type(Assistant.Chat).sourceOf(Assistant.CompanionTo).source())
     .run();
+
   const chats = allChats.filter((chat) => !relatedChats.includes(chat));
-  // console.log('objects', JSON.stringify(objects, null, 2));
   if (chats.length > 0) {
-    // TODO(burdon): Is this the most recent?
-    return chats[chats.length - 1];
+    return chats.at(-1);
   }
 
-  const { data } = await dispatch(createIntent(Assistant.CreateChat, { space }));
+  const { data } = await dispatch(createIntent(AssistantAction.CreateChat, { space }));
   invariant(Obj.instanceOf(Assistant.Chat, data?.object));
   await dispatch(createIntent(SpaceAction.AddObject, { target: space, object: data.object }));
   return data.object;

package/src/capabilities/blueprint-definition.ts

@@ -0,0 +1,42 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { Capabilities, type Capability, contributes } from '@dxos/app-framework';
+import { templates } from '@dxos/assistant';
+import { Blueprint } from '@dxos/blueprints';
+import { type FunctionDefinition } from '@dxos/functions';
+
+import { analysis, list, load } from '../functions';
+
+const functions: FunctionDefinition[] = [analysis, list, load];
+const tools = [
+  'add-to-context',
+  // TODO(wittjosiah): Factor out to an ECHO blueprint.
+  'get-schemas',
+  'add-schema',
+  'create-record',
+  // TODO(wittjosiah): Factor out to a generic app-framework blueprint.
+  'open-item',
+  // TODO(burdon): Anthropic only.
+  //  https://docs.anthropic.com/en/docs/agents-and-tools/tool-use/text-editor-tool#example-str-replace-command
+  //  AI_TOOL_NOT_FOUND: str_replace_based_edit_tool
+  // 'str_replace_based_edit_tool',
+];
+
+export const BLUEPRINT_KEY = 'dxos.org/blueprint/assistant';
+
+export const createBlueprint = (): Blueprint.Blueprint =>
+  Blueprint.make({
+    key: BLUEPRINT_KEY,
+    name: 'Assistant',
+    tools: Blueprint.toolDefinitions({ functions, tools }),
+    instructions: templates.system,
+  });
+
+const blueprint = createBlueprint();
+
+export default (): Capability<any>[] => [
+  contributes(Capabilities.Functions, functions),
+  contributes(Capabilities.BlueprintDefinition, blueprint),
+];

package/src/capabilities/capabilities.ts

@@ -0,0 +1,31 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { type Layer } from 'effect';
+
+import { type AiService, type AiServiceRouter } from '@dxos/ai';
+import { defineCapability } from '@dxos/app-framework';
+import { type DeepReadonly } from '@dxos/util';
+
+import { meta } from '../meta';
+import { type Assistant } from '../types';
+
+export namespace AssistantCapabilities {
+  export type AssistantState = {
+    /** Map of primary object fq id to current chat. */
+    currentChat: Record<string, Assistant.Chat>;
+  };
+  export const State = defineCapability<DeepReadonly<AssistantState>>(`${meta.id}/capability/state`);
+  export const MutableState = defineCapability<AssistantState>(`${meta.id}/capability/state`);
+
+  export type AiServiceLayer = Layer.Layer<AiService.AiService>;
+  export const AiServiceLayer = defineCapability<AiServiceLayer>(`${meta.id}/capability/ai-service-factory`);
+
+  /**
+   * Plugins can contribute them to provide model resolvers.
+   */
+  export const AiModelResolver = defineCapability<Layer.Layer<AiServiceRouter.AiModelResolver>>(
+    `${meta.id}/capability/ai-model-resolver`,
+  );
+}

package/src/capabilities/edge-model-resolver.ts

@@ -0,0 +1,27 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as AnthropicClient from '@effect/ai-anthropic/AnthropicClient';
+import { FetchHttpClient } from '@effect/platform';
+import { Layer } from 'effect';
+
+import { AiServiceRouter } from '@dxos/ai';
+import { type Capability, contributes } from '@dxos/app-framework';
+
+import { AssistantCapabilities } from './capabilities';
+
+export default (): Capability<any>[] => [
+  contributes(
+    AssistantCapabilities.AiModelResolver,
+    AiServiceRouter.AnthropicResolver.pipe(
+      Layer.provide(
+        AnthropicClient.layer({
+          // TODO(dmaretskyi): Read endpoint from config/settings.
+          apiUrl: 'https://ai-service.dxos.workers.dev/provider/anthropic',
+        }),
+      ),
+      Layer.provide(FetchHttpClient.layer),
+    ),
+  ),
+];

package/src/capabilities/index.ts

@@ -4,7 +4,16 @@
 
 import { lazy } from '@dxos/app-framework';
 
+export const AiService = lazy(() => import('./ai-service'));
 export const AppGraphBuilder = lazy(() => import('./app-graph-builder'));
+export const AssistantState = lazy(() => import('./state'));
+export const BlueprintDefinition = lazy(() => import('./blueprint-definition'));
+export const EdgeModelResolver = lazy(() => import('./edge-model-resolver'));
 export const IntentResolver = lazy(() => import('./intent-resolver'));
+export const LocalModelResolver = lazy(() => import('./local-model-resolver'));
 export const ReactSurface = lazy(() => import('./react-surface'));
 export const Settings = lazy(() => import('./settings'));
+export const Toolkit = lazy(() => import('./toolkit'));
+
+export * from './blueprint-definition';
+export * from './capabilities';