@dxos/plugin-assistant 0.8.4-main.bc674ce → 0.8.4-main.bcb3aa67d6

package/dist/types/src/types/AssistantAction.d.ts

@@ -1,95 +0,0 @@
-import * as Schema from 'effect/Schema';
-import { Capability } from '@dxos/app-framework';
-import { Database, Type } from '@dxos/echo';
-export declare namespace AssistantOperation {
-    const OnCreateSpace: Omit<{
-        readonly meta: {
-            readonly key: `${string}/on-create-space`;
-            readonly name: "On Create Space";
-        };
-        readonly services: readonly [typeof Capability.Service];
-        readonly schema: {
-            readonly input: Schema.Struct<{
-                space: Schema.Schema<import("@dxos/client-protocol").Space, import("@dxos/client-protocol").Space, never>;
-                rootCollection: Type.Obj<{
-                    readonly name?: string | undefined;
-                    readonly objects: readonly import("@dxos/echo/internal").Ref<any>[];
-                }, Schema.Struct.Fields>;
-            }>;
-            readonly output: typeof Schema.Void;
-        };
-    }, "executionMode"> & import("effect/Pipeable").Pipeable & {
-        readonly executionMode: "sync" | "async";
-    };
-    const CreateChat: Omit<{
-        readonly meta: {
-            readonly key: `${string}/create-chat`;
-            readonly name: "Create Chat";
-        };
-        readonly services: readonly [typeof Capability.Service];
-        readonly schema: {
-            readonly input: Schema.Struct<{
-                db: Schema.Schema<Database.Database, Database.Database, never>;
-                name: Schema.optional<typeof Schema.String>;
-            }>;
-            readonly output: Schema.Struct<{
-                object: Type.Obj<{
-                    readonly name?: string | undefined;
-                    readonly queue: import("@dxos/echo/internal").Ref<import("@dxos/echo-db").Queue<import("@dxos/echo/Entity").Unknown>>;
-                    readonly traceQueue?: import("@dxos/echo/internal").Ref<import("@dxos/echo-db").Queue<import("@dxos/echo/Entity").Unknown>> | undefined;
-                }, Schema.Struct.Fields>;
-            }>;
-        };
-    }, "executionMode"> & import("effect/Pipeable").Pipeable & {
-        readonly executionMode: "sync" | "async";
-    };
-    const UpdateChatName: Omit<{
-        readonly meta: {
-            readonly key: `${string}/update-chat-name`;
-            readonly name: "Update Chat Name";
-        };
-        readonly services: readonly [typeof Capability.Service];
-        readonly schema: {
-            readonly input: Schema.Struct<{
-                chat: Type.Obj<{
-                    readonly name?: string | undefined;
-                    readonly queue: import("@dxos/echo/internal").Ref<import("@dxos/echo-db").Queue<import("@dxos/echo/Entity").Unknown>>;
-                    readonly traceQueue?: import("@dxos/echo/internal").Ref<import("@dxos/echo-db").Queue<import("@dxos/echo/Entity").Unknown>> | undefined;
-                }, Schema.Struct.Fields>;
-            }>;
-            readonly output: typeof Schema.Void;
-        };
-    }, "executionMode"> & import("effect/Pipeable").Pipeable & {
-        readonly executionMode: "sync" | "async";
-    };
-    const SetCurrentChat: Omit<{
-        readonly meta: {
-            readonly key: `${string}/set-current-chat`;
-            readonly name: "Set Current Chat";
-        };
-        readonly services: readonly [typeof Capability.Service];
-        readonly schema: {
-            readonly input: Schema.Struct<{
-                companionTo: Schema.Schema<any, {
-                    id: string;
-                } & import("@dxos/echo/internal").AnyProperties, never> & {
-                    readonly "~@dxos/echo/SchemaKind": import("@dxos/echo/internal").EntityKind.Object;
-                } & import("@dxos/echo/internal").TypeMeta;
-                chat: Schema.optional<Type.Obj<{
-                    readonly name?: string | undefined;
-                    readonly queue: import("@dxos/echo/internal").Ref<import("@dxos/echo-db").Queue<import("@dxos/echo/Entity").Unknown>>;
-                    readonly traceQueue?: import("@dxos/echo/internal").Ref<import("@dxos/echo-db").Queue<import("@dxos/echo/Entity").Unknown>> | undefined;
-                }, Schema.Struct.Fields>>;
-            }>;
-            readonly output: typeof Schema.Void;
-        };
-    }, "executionMode"> & import("effect/Pipeable").Pipeable & {
-        readonly executionMode: "sync" | "async";
-    };
-    const BlueprintForm: Schema.Struct<{
-        key: typeof Schema.String;
-        name: typeof Schema.String;
-        description: Schema.optional<typeof Schema.String>;
-    }>;
-}
-//# sourceMappingURL=AssistantAction.d.ts.map

package/dist/types/src/types/AssistantAction.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"AssistantAction.d.ts","sourceRoot":"","sources":["../../../../src/types/AssistantAction.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AAExC,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEjD,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAW5C,yBAAiB,kBAAkB,CAAC;IAC3B,MAAM,aAAa;;;;;;;;;;;;;;;;;;KAUxB,CAAC;IAEI,MAAM,UAAU;;;;;;;;;;;;;;;;;;;;;KAYrB,CAAC;IAEI,MAAM,cAAc;;;;;;;;;;;;;;;;;;KASzB,CAAC;IAEI,MAAM,cAAc;;;;;;;;;;;;;;;;;;;;;;;KAUzB,CAAC;IAEI,MAAM,aAAa;;;;MAIxB,CAAC;CACJ"}

package/src/blueprints/assistant-blueprint.ts

@@ -1,62 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { templates } from '@dxos/assistant';
-import {
-  Agent,
-  AssistantToolkit,
-  Discord,
-  DiscordBlueprint,
-  EntityExtraction,
-  Linear,
-  LinearBlueprint,
-  Research,
-  ResearchBlueprint,
-  SystemToolkit,
-  WebSearchBlueprint,
-} from '@dxos/assistant-toolkit';
-import { Blueprint } from '@dxos/blueprints';
-import { type FunctionDefinition } from '@dxos/functions';
-
-import { list, load, search } from '../functions';
-
-// TODO(burdon): Document plugin structure (blueprint, functions, toolkit.).
-// TODO(burdon): Test framework for developing functions. Error handling.
-// TODO(burdon): Function naming pattern (noun-verb); fully-qualified?
-
-export const deckTools = ['open-item'];
-
-export const functions$: FunctionDefinition[] = [list, load, search];
-export const functions = [
-  ...functions$,
-  // Factor out.
-  Research.create,
-  Research.research,
-  Agent.prompt,
-  EntityExtraction.extract,
-  Discord.fetch,
-  Linear.sync,
-];
-export const tools = [...AssistantToolkit.tools, ...SystemToolkit.tools, ...deckTools];
-
-export const Key = 'dxos.org/blueprint/assistant';
-
-export const make = (): Blueprint.Blueprint =>
-  Blueprint.make({
-    key: Key,
-    name: 'Assistant',
-    tools: Blueprint.toolDefinitions({ functions, tools }),
-    instructions: templates.system,
-  });
-
-export const blueprint = make();
-
-export const blueprints = [
-  blueprint,
-  // Factor out.
-  ResearchBlueprint,
-  WebSearchBlueprint,
-  DiscordBlueprint,
-  LinearBlueprint,
-];

package/src/capabilities/ai-service/index.ts

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Capability } from '@dxos/app-framework';
-
-export const AiService = Capability.lazy('AiService', () => import('./ai-service'));

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

@@ -1,180 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Capability, Common } from '@dxos/app-framework';
-import { Prompt } from '@dxos/blueprints';
-import { Sequence } from '@dxos/conductor';
-import { DXN, type Database, Obj, type Ref } from '@dxos/echo';
-import { AtomObj } from '@dxos/echo-atom';
-import { invariant } from '@dxos/invariant';
-import { Operation, type OperationInvoker } from '@dxos/operation';
-import { ClientCapabilities } from '@dxos/plugin-client';
-import { ATTENDABLE_PATH_SEPARATOR, PLANK_COMPANION_TYPE } from '@dxos/plugin-deck/types';
-import { GraphBuilder, NodeMatcher } from '@dxos/plugin-graph';
-import { getActiveSpace } from '@dxos/plugin-space';
-import { SpaceOperation } from '@dxos/plugin-space/types';
-import { Query } from '@dxos/react-client/echo';
-
-import { ASSISTANT_DIALOG, meta } from '../../meta';
-import { Assistant, AssistantCapabilities, AssistantOperation } from '../../types';
-
-export default Capability.makeModule(
-  Effect.fnUntraced(function* () {
-    const capabilities = yield* Capability.Service;
-
-    const extensions = yield* Effect.all([
-      GraphBuilder.createTypeExtension({
-        id: `${meta.id}/root`,
-        type: Assistant.Chat,
-        actions: (chat) => {
-          const id = Obj.getDXN(chat).toString();
-          return Effect.succeed([
-            {
-              id: `${AssistantOperation.UpdateChatName.meta.key}/${id}`,
-              data: () => Operation.invoke(AssistantOperation.UpdateChatName, { chat }),
-              properties: {
-                label: ['chat update name label', { ns: meta.id }],
-                icon: 'ph--magic-wand--regular',
-                disposition: 'list-item',
-              },
-            },
-          ]);
-        },
-      }),
-
-      GraphBuilder.createExtension({
-        id: `${meta.id}/assistant`,
-        match: NodeMatcher.whenRoot,
-        actions: () =>
-          Effect.succeed([
-            {
-              id: `${Common.LayoutOperation.UpdateDialog.meta.key}/assistant/open`,
-              data: Effect.fnUntraced(function* () {
-                const capabilities = yield* Capability.Service;
-                const client = yield* Capability.get(ClientCapabilities.Client);
-                const operationInvoker = yield* Capability.get(Common.Capability.OperationInvoker);
-                const space = getActiveSpace(capabilities) ?? client.spaces.default;
-                const chat = yield* Effect.tryPromise(() => getOrCreateChat(operationInvoker.invokePromise, space.db));
-                if (!chat) {
-                  return;
-                }
-
-                yield* Operation.invoke(Common.LayoutOperation.UpdateDialog, {
-                  subject: ASSISTANT_DIALOG,
-                  state: true,
-                  blockAlign: 'end',
-                  props: {
-                    chat,
-                  },
-                });
-              }),
-              properties: {
-                label: ['open assistant label', { ns: meta.id }],
-                icon: 'ph--sparkle--regular',
-                disposition: 'pin-end',
-                position: 'hoist',
-                keyBinding: {
-                  macos: 'shift+meta+k',
-                  windows: 'shift+ctrl+k',
-                },
-              },
-            },
-          ]),
-      }),
-
-      GraphBuilder.createExtension({
-        id: `${meta.id}/companion-chat`,
-        match: NodeMatcher.whenEchoObject,
-        connector: (object, get) =>
-          Effect.gen(function* () {
-            const state = get(yield* Capability.get(AssistantCapabilities.State));
-            const currentChatState = state.currentChat[Obj.getDXN(object).toString()];
-            // If no state, continue to allow chat initialization.
-            if (!currentChatState) {
-              return [
-                {
-                  id: [Obj.getDXN(object).toString(), 'assistant-chat'].join(ATTENDABLE_PATH_SEPARATOR),
-                  type: PLANK_COMPANION_TYPE,
-                  data: 'assistant-chat',
-                  properties: {
-                    label: ['assistant chat label', { ns: meta.id }],
-                    icon: 'ph--sparkle--regular',
-                    position: 'hoist',
-                    disposition: 'hidden',
-                  },
-                },
-              ];
-            }
-
-            const db = Obj.getDatabase(object);
-            const currentChatDxn = DXN.tryParse(currentChatState);
-            const currentChatRef = currentChatDxn ? db?.makeRef(currentChatDxn) : undefined;
-            const currentChat = currentChatRef ? get(AtomObj.make(currentChatRef as Ref.Ref<Obj.Unknown>)) : undefined;
-            if (!Obj.isObject(currentChat)) {
-              return [];
-            }
-
-            return [
-              {
-                id: [Obj.getDXN(object).toString(), 'assistant-chat'].join(ATTENDABLE_PATH_SEPARATOR),
-                type: PLANK_COMPANION_TYPE,
-                data: currentChat,
-                properties: {
-                  label: ['assistant chat label', { ns: meta.id }],
-                  icon: 'ph--sparkle--regular',
-                  position: 'hoist',
-                  disposition: 'hidden',
-                },
-              },
-            ];
-          }),
-      }),
-
-      GraphBuilder.createExtension({
-        id: `${meta.id}/invocations`,
-        match: NodeMatcher.whenAny(
-          NodeMatcher.whenEchoTypeMatches(Sequence),
-          NodeMatcher.whenEchoTypeMatches(Prompt.Prompt),
-        ),
-        connector: (node) =>
-          Effect.succeed([
-            {
-              id: [node.id, 'invocations'].join(ATTENDABLE_PATH_SEPARATOR),
-              type: PLANK_COMPANION_TYPE,
-              data: 'invocations',
-              properties: {
-                label: ['invocations label', { ns: meta.id }],
-                icon: 'ph--clock-countdown--regular',
-                disposition: 'hidden',
-              },
-            },
-          ]),
-      }),
-    ]);
-
-    return Capability.contributes(Common.Capability.AppGraphBuilder, extensions);
-  }),
-);
-
-// TODO(burdon): Factor out.
-const getOrCreateChat = async (
-  invokePromise: OperationInvoker.OperationInvoker['invokePromise'],
-  db: Database.Database,
-): Promise<Assistant.Chat | undefined> => {
-  // TODO(wittjosiah): This should be possible with a single query.
-  const allChats = await db.query(Query.type(Assistant.Chat)).run();
-  const relatedChats = await db.query(Query.type(Assistant.Chat).sourceOf(Assistant.CompanionTo).source()).run();
-
-  const chats = allChats.filter((chat) => !relatedChats.includes(chat));
-  if (chats.length > 0) {
-    return chats.at(-1);
-  }
-
-  const { data } = await invokePromise(AssistantOperation.CreateChat, { db });
-  invariant(Obj.instanceOf(Assistant.Chat, data?.object));
-  await invokePromise(SpaceOperation.AddObject, { target: db, object: data.object });
-  return data.object;
-};

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

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Capability } from '@dxos/app-framework';
-
-export const AppGraphBuilder = Capability.lazy('AppGraphBuilder', () => import('./app-graph-builder'));

package/src/capabilities/blueprint-definition/blueprint-definition.ts

@@ -1,60 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Capability, Common } from '@dxos/app-framework';
-import {
-  Agent,
-  Discord,
-  DiscordBlueprint,
-  EntityExtraction,
-  Linear,
-  LinearBlueprint,
-  Research,
-  ResearchBlueprint,
-  WebSearchBlueprint,
-} from '@dxos/assistant-toolkit';
-import { type Blueprint } from '@dxos/blueprints';
-
-import { AssistantBlueprint } from '../../blueprints';
-
-export const createBlueprint: () => Blueprint.Blueprint = AssistantBlueprint.make;
-export { AssistantBlueprint };
-
-export type BlueprintCapabilities = [
-  Capability.Capability<typeof Common.Capability.Functions>,
-  Capability.Capability<typeof Common.Capability.BlueprintDefinition>,
-  Capability.Capability<typeof Common.Capability.Functions>,
-  Capability.Capability<typeof Common.Capability.BlueprintDefinition>,
-  Capability.Capability<typeof Common.Capability.Functions>,
-  Capability.Capability<typeof Common.Capability.BlueprintDefinition>,
-  Capability.Capability<typeof Common.Capability.Functions>,
-  Capability.Capability<typeof Common.Capability.BlueprintDefinition>,
-  Capability.Capability<typeof Common.Capability.Functions>,
-  Capability.Capability<typeof Common.Capability.BlueprintDefinition>,
-];
-
-export default Capability.makeModule<[], BlueprintCapabilities>(() =>
-  Effect.succeed([
-    Capability.contributes(Common.Capability.Functions, AssistantBlueprint.functions$),
-    Capability.contributes(Common.Capability.BlueprintDefinition, AssistantBlueprint.blueprint),
-
-    // TODO(burdon): Factor out.
-    Capability.contributes(Common.Capability.Functions, [Research.create, Research.research]),
-    Capability.contributes(Common.Capability.BlueprintDefinition, ResearchBlueprint),
-
-    // TODO(burdon): Factor out.
-    Capability.contributes(Common.Capability.Functions, [Agent.prompt, EntityExtraction.extract]),
-    Capability.contributes(Common.Capability.BlueprintDefinition, WebSearchBlueprint),
-
-    // TODO(burdon): Factor out.
-    Capability.contributes(Common.Capability.Functions, [Discord.fetch]),
-    Capability.contributes(Common.Capability.BlueprintDefinition, DiscordBlueprint),
-
-    // TODO(burdon): Factor out.
-    Capability.contributes(Common.Capability.Functions, [Linear.sync]),
-    Capability.contributes(Common.Capability.BlueprintDefinition, LinearBlueprint),
-  ]),
-);

package/src/capabilities/blueprint-definition/index.ts

@@ -1,8 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Capability } from '@dxos/app-framework';
-
-export const BlueprintDefinition = Capability.lazy('BlueprintDefinition', () => import('./blueprint-definition'));
-export { AssistantBlueprint } from './blueprint-definition';

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

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Capability } from '@dxos/app-framework';
-
-export const EdgeModelResolver = Capability.lazy('EdgeModelResolver', () => import('./edge-model-resolver'));

package/src/capabilities/local-model-resolver/index.ts

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Capability } from '@dxos/app-framework';
-
-export const LocalModelResolver = Capability.lazy('LocalModelResolver', () => import('./local-model-resolver'));

package/src/capabilities/operation-resolver/index.ts

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Capability } from '@dxos/app-framework';
-
-export const OperationResolver = Capability.lazy('OperationResolver', () => import('./operation-resolver'));

package/src/capabilities/operation-resolver/operation-resolver.ts

@@ -1,110 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Capability, Common } from '@dxos/app-framework';
-import { AiContextBinder, AiConversation } from '@dxos/assistant';
-import { Agent } from '@dxos/assistant-toolkit';
-import { Blueprint, Prompt } from '@dxos/blueprints';
-import { type Queue } from '@dxos/client/echo';
-import { Filter, Obj, Ref, Type } from '@dxos/echo';
-import { TracingService, serializeFunction } from '@dxos/functions';
-import { invariant } from '@dxos/invariant';
-import { Operation, OperationResolver } from '@dxos/operation';
-import { AutomationCapabilities } from '@dxos/plugin-automation';
-import { ClientCapabilities } from '@dxos/plugin-client';
-import { Collection } from '@dxos/schema';
-import { type Message } from '@dxos/types';
-
-import { type AiChatServices, updateName } from '../../processor';
-import { Assistant, AssistantCapabilities, AssistantOperation } from '../../types';
-import { AssistantBlueprint, createBlueprint } from '../blueprint-definition/blueprint-definition';
-
-export default Capability.makeModule(
-  Effect.fnUntraced(function* () {
-    return Capability.contributes(Common.Capability.OperationResolver, [
-      OperationResolver.make({
-        operation: AssistantOperation.OnCreateSpace,
-        handler: Effect.fnUntraced(function* ({ space, rootCollection }) {
-          const chatCollection = Collection.makeManaged({ key: Assistant.Chat.typename });
-          const blueprintCollection = Collection.makeManaged({ key: Blueprint.Blueprint.typename });
-          const promptCollection = Collection.makeManaged({ key: Type.getTypename(Prompt.Prompt) });
-          Obj.change(rootCollection, (c) => {
-            c.objects.push(Ref.make(chatCollection), Ref.make(blueprintCollection), Ref.make(promptCollection));
-          });
-
-          // TODO(wittjosiah): Remove once function registry is avaiable.
-          space.db.add(serializeFunction(Agent.prompt));
-
-          // Create default chat.
-          const { object: chat } = yield* Operation.invoke(AssistantOperation.CreateChat, { db: space.db });
-          space.db.add(chat);
-        }),
-      }),
-      OperationResolver.make({
-        operation: AssistantOperation.CreateChat,
-        handler: Effect.fnUntraced(function* ({ db, name }) {
-          const registry = yield* Capability.get(Common.Capability.AtomRegistry);
-          const client = yield* Capability.get(ClientCapabilities.Client);
-          const space = client.spaces.get(db.spaceId);
-          invariant(space, 'Space not found');
-          const queue = space.queues.create();
-          const chat = Assistant.make({ name, queue: Ref.fromDXN(queue.dxn) });
-          space.db.add(chat);
-
-          // TODO(wittjosiah): This should be a space-level setting.
-          // TODO(burdon): Clone when activated. Copy-on-write for template.
-          const blueprints = yield* Effect.promise(() => db.query(Filter.type(Blueprint.Blueprint)).run());
-          let defaultBlueprint = blueprints.find((blueprint) => blueprint.key === AssistantBlueprint.Key);
-          if (!defaultBlueprint) {
-            defaultBlueprint = db.add(createBlueprint());
-          }
-
-          const binder = new AiContextBinder({ queue, registry });
-          yield* Effect.promise(() =>
-            binder.use((b: AiContextBinder) => b.bind({ blueprints: [Ref.make(defaultBlueprint!)] })),
-          );
-
-          return { object: chat };
-        }),
-      }),
-      OperationResolver.make({
-        operation: AssistantOperation.UpdateChatName,
-        handler: Effect.fnUntraced(function* ({ chat }) {
-          const registry = yield* Capability.get(Common.Capability.AtomRegistry);
-          const db = Obj.getDatabase(chat);
-          const queue = chat.queue.target as Queue<Message.Message>;
-          if (!db || !queue) {
-            return;
-          }
-
-          const runtimeResolver = yield* Capability.get(AutomationCapabilities.ComputeRuntime);
-          const runtime = yield* Effect.promise(() =>
-            runtimeResolver
-              .getRuntime(db.spaceId)
-              .runPromise(Effect.runtime<AiChatServices>().pipe(Effect.provide(TracingService.layerNoop))),
-          );
-
-          yield* Effect.promise(() =>
-            new AiConversation({ queue, registry }).use(async (conversation) =>
-              updateName(runtime, conversation, chat),
-            ),
-          );
-        }),
-      }),
-      OperationResolver.make({
-        operation: AssistantOperation.SetCurrentChat,
-        handler: Effect.fnUntraced(function* ({ companionTo, chat }) {
-          const companionToId = Obj.getDXN(companionTo).toString();
-          const chatId = chat && Obj.getDXN(chat).toString();
-          yield* Common.Capability.updateAtomValue(AssistantCapabilities.State, (current) => ({
-            ...current,
-            currentChat: { ...current.currentChat, [companionToId]: chatId },
-          }));
-        }),
-      }),
-    ]);
-  }),
-);

package/src/capabilities/react-surface/index.ts

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Capability } from '@dxos/app-framework';
-
-export const ReactSurface = Capability.lazy('ReactSurface', () => import('./react-surface'));

package/src/capabilities/react-surface/react-surface.tsx

@@ -1,95 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import React from 'react';
-
-import { Capability, Common } from '@dxos/app-framework';
-import { useSettingsState } from '@dxos/app-framework/react';
-import { Blueprint, Prompt } from '@dxos/blueprints';
-import { getSpace } from '@dxos/client/echo';
-import { Sequence } from '@dxos/conductor';
-import { InvocationTraceContainer } from '@dxos/devtools';
-import { Obj } from '@dxos/echo';
-import { Layout } from '@dxos/react-ui-mosaic';
-
-import {
-  AssistantSettings,
-  BlueprintArticle,
-  ChatCompanion,
-  ChatContainer,
-  ChatDialog,
-  PromptArticle,
-} from '../../components';
-import { ASSISTANT_DIALOG, meta } from '../../meta';
-import { Assistant } from '../../types';
-
-export default Capability.makeModule(() =>
-  Effect.succeed(
-    Capability.contributes(Common.Capability.ReactSurface, [
-      Common.createSurface({
-        id: `${meta.id}/plugin-settings`,
-        role: 'article',
-        filter: (data): data is { subject: Common.Capability.Settings } =>
-          Common.Capability.isSettings(data.subject) && data.subject.prefix === meta.id,
-        component: ({ data: { subject } }) => {
-          const { settings, updateSettings } = useSettingsState<Assistant.Settings>(subject.atom);
-          return <AssistantSettings settings={settings} onSettingsChange={updateSettings} />;
-        },
-      }),
-      Common.createSurface({
-        id: `${meta.id}/chat`,
-        role: 'article',
-        filter: (data): data is { subject: Assistant.Chat; variant: undefined } =>
-          Obj.instanceOf(Assistant.Chat, data.subject) && data.variant !== 'assistant-chat',
-        component: ({ data, role, ref }) => <ChatContainer role={role} subject={data.subject} ref={ref} />,
-      }),
-      // TODO(wittjosiah): This is flashing when chat changes.
-      Common.createSurface({
-        id: `${meta.id}/companion-chat`,
-        role: 'article',
-        filter: (data): data is { companionTo: Obj.Unknown; subject: Assistant.Chat | 'assistant-chat' } =>
-          Obj.isObject(data.companionTo) &&
-          (Obj.instanceOf(Assistant.Chat, data.subject) || data.subject === 'assistant-chat'),
-        component: ({ data, role, ref }) => <ChatCompanion role={role} data={data} ref={ref} />,
-      }),
-      Common.createSurface({
-        id: `${meta.id}/companion-invocations`,
-        role: 'article',
-        filter: (data): data is { companionTo: Sequence } =>
-          (Obj.instanceOf(Sequence, data.companionTo) || Obj.instanceOf(Prompt.Prompt, data.companionTo)) &&
-          data.subject === 'invocations',
-        component: ({ data, role }) => {
-          const space = getSpace(data.companionTo);
-          const queueDxn = space?.properties.invocationTraceQueue?.dxn;
-          // TODO(wittjosiah): Support invocation filtering for prompts.
-          const target = Obj.instanceOf(Prompt.Prompt, data.companionTo) ? undefined : data.companionTo;
-          return (
-            <Layout.Main role={role}>
-              <InvocationTraceContainer db={space?.db} queueDxn={queueDxn} target={target} detailAxis='block' />
-            </Layout.Main>
-          );
-        },
-      }),
-      Common.createSurface({
-        id: `${meta.id}/blueprint`,
-        role: 'article',
-        filter: (data): data is { subject: Blueprint.Blueprint } => Obj.instanceOf(Blueprint.Blueprint, data.subject),
-        component: ({ data }) => <BlueprintArticle subject={data.subject} />,
-      }),
-      Common.createSurface({
-        id: `${meta.id}/prompt`,
-        role: 'article',
-        filter: (data): data is { subject: Prompt.Prompt } => Obj.instanceOf(Prompt.Prompt, data.subject),
-        component: ({ data }) => <PromptArticle subject={data.subject} />,
-      }),
-      Common.createSurface({
-        id: ASSISTANT_DIALOG,
-        role: 'dialog',
-        filter: (data): data is { props: { chat: Assistant.Chat } } => data.component === ASSISTANT_DIALOG,
-        component: ({ data }) => <ChatDialog {...data.props} />,
-      }),
-    ]),
-  ),
-);