@dxos/plugin-assistant 0.8.4-main.7996785055 → 0.8.4-main.8baae0fced

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

@@ -1,230 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Capabilities, Capability } from '@dxos/app-framework';
-import { AppCapabilities, companionSegment, getPersonalSpace, LayoutOperation } from '@dxos/app-toolkit';
-import { Chat } from '@dxos/assistant-toolkit';
-import { Blueprint, Prompt } from '@dxos/blueprints';
-import { Sequence } from '@dxos/conductor';
-import { DXN, 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 { DECK_COMPANION_TYPE, 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/operations';
-import { Query } from '@dxos/react-client/echo';
-
-import { ASSISTANT_DIALOG, meta } from '../../meta';
-import { AssistantCapabilities } from '../../types';
-import { AssistantOperation } from '../../operations';
-
-/** Match ECHO objects that are NOT chats. */
-const whenNonChatObject = NodeMatcher.whenAll(
-  NodeMatcher.whenEchoObject,
-  NodeMatcher.whenNot(NodeMatcher.whenEchoTypeMatches(Chat.Chat)),
-);
-
-export default Capability.makeModule(
-  Effect.fnUntraced(function* () {
-    const capabilities = yield* Capability.Service;
-
-    const extensions = yield* Effect.all([
-      GraphBuilder.createTypeExtension({
-        id: `${meta.id}.root`,
-        type: Chat.Chat,
-        actions: (chat) => {
-          return Effect.succeed([
-            {
-              id: AssistantOperation.UpdateChatName.meta.key,
-              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: `${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(Capabilities.OperationInvoker);
-                const space = getActiveSpace(capabilities) ?? getPersonalSpace(client);
-                if (!space) {
-                  return;
-                }
-                const chat = yield* Effect.tryPromise(() => getOrCreateChat(operationInvoker.invokePromise, space.db));
-                if (!chat) {
-                  return;
-                }
-
-                yield* Operation.invoke(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',
-                },
-              },
-            },
-            {
-              id: `${meta.id}.reset-blueprints`,
-              data: Effect.fnUntraced(function* () {
-                const capabilities = yield* Capability.Service;
-                const client = yield* Capability.get(ClientCapabilities.Client);
-                const space = getActiveSpace(capabilities) ?? getPersonalSpace(client);
-                if (!space) {
-                  return;
-                }
-                const blueprints = yield* Effect.tryPromise(() =>
-                  space.db.query(Query.type(Blueprint.Blueprint)).run(),
-                );
-                for (const blueprint of blueprints) {
-                  space.db.remove(blueprint);
-                }
-                yield* Database.flush();
-              }),
-              properties: {
-                label: ['reset blueprints label', { ns: meta.id }],
-                icon: 'ph--arrow-clockwise--regular',
-              },
-            },
-          ]),
-      }),
-
-      // Don't show assistant companion when a chat is already the primary object.
-      GraphBuilder.createExtension({
-        id: `${meta.id}.companion-chat`,
-        match: whenNonChatObject,
-        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: 'assistant-chat',
-                  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;
-
-            // Return the resolved chat object, or fall back to 'assistant-chat' string if it can't be resolved.
-            // This ensures the companion remains visible even during transient states.
-            return [
-              {
-                id: 'assistant-chat',
-                type: PLANK_COMPANION_TYPE,
-                data: Obj.isObject(currentChat) ? currentChat : 'assistant-chat',
-                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: () =>
-          Effect.succeed([
-            {
-              id: 'invocations',
-              type: PLANK_COMPANION_TYPE,
-              data: 'invocations',
-              properties: {
-                label: ['invocations label', { ns: meta.id }],
-                icon: 'ph--clock-countdown--regular',
-                disposition: 'hidden',
-              },
-            },
-          ]),
-      }),
-
-      GraphBuilder.createExtension({
-        id: `${meta.id}.trace`,
-        match: NodeMatcher.whenRoot,
-        connector: () =>
-          Effect.succeed([
-            {
-              id: companionSegment('trace'),
-              type: DECK_COMPANION_TYPE,
-              data: 'trace' as const,
-              properties: {
-                label: ['trace label', { ns: meta.id }],
-                icon: 'ph--line-segments--regular',
-                disposition: 'hidden',
-                position: 'fallback',
-              },
-            },
-          ]),
-      }),
-    ]);
-
-    return Capability.contributes(AppCapabilities.AppGraphBuilder, extensions);
-  }),
-);
-
-// TODO(burdon): Factor out.
-const getOrCreateChat = async (
-  invokePromise: OperationInvoker.OperationInvoker['invokePromise'],
-  db: Database.Database,
-): Promise<Chat.Chat | undefined> => {
-  // TODO(wittjosiah): This should be possible with a single query.
-  const allChats = await db.query(Query.type(Chat.Chat)).run();
-  const relatedChats = await db.query(Query.type(Chat.Chat).sourceOf(Chat.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(Chat.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/index.ts

@@ -1,10 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Capability } from '@dxos/app-framework';
-import type { OperationHandlerSet } from '@dxos/operation';
-
-export { type OperationHandlerSet };
-
-export const BlueprintDefinition = Capability.lazy('BlueprintDefinition', () => import('./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/markdown/index.ts

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

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

@@ -1,11 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Capability } from '@dxos/app-framework';
-import { OperationHandlerSet } from '@dxos/operation';
-
-export const OperationHandler = Capability.lazy<OperationHandlerSet.OperationHandlerSet>(
-  'OperationHandler',
-  () => import('./operation-handler'),
-);

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,158 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import React from 'react';
-
-import { Capabilities, Capability } from '@dxos/app-framework';
-import { Surface, useSettingsState } from '@dxos/app-framework/ui';
-import { AppCapabilities } from '@dxos/app-toolkit';
-import { Chat, Project } from '@dxos/assistant-toolkit';
-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 { Panel } from '@dxos/react-ui';
-import { useActiveSpace } from '@dxos/plugin-space';
-
-import {
-  AssistantSettings,
-  BlueprintArticle,
-  ChatCompanion,
-  ChatContainer,
-  ChatDialog,
-  ProjectArticle,
-  ProjectSettings,
-  PromptArticle,
-  PromptList,
-  TracePanel,
-  TriggerStatus,
-} from '../../containers';
-import { ASSISTANT_DIALOG, meta } from '../../meta';
-import { type Assistant } from '../../types';
-
-export default Capability.makeModule(() =>
-  Effect.succeed(
-    Capability.contributes(Capabilities.ReactSurface, [
-      Surface.create({
-        id: `${meta.id}.plugin-settings`,
-        role: 'article',
-        filter: (data): data is { subject: AppCapabilities.Settings } =>
-          AppCapabilities.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} />;
-        },
-      }),
-      Surface.create({
-        id: `${meta.id}.chat`,
-        role: 'article',
-        filter: (data): data is { attendableId: string; subject: Chat.Chat; variant: undefined } =>
-          typeof data.attendableId === 'string' &&
-          Obj.instanceOf(Chat.Chat, data.subject) &&
-          data.variant !== 'assistant-chat',
-        component: ({ data, role, ref }) => (
-          <ChatContainer role={role} subject={data.subject} attendableId={data.attendableId} ref={ref} />
-        ),
-      }),
-      Surface.create({
-        id: `${meta.id}.project`,
-        role: 'article',
-        filter: (data): data is { subject: Project.Project } => Obj.instanceOf(Project.Project, data.subject),
-        component: ({ data, role }) => <ProjectArticle role={role} subject={data.subject} />,
-      }),
-      Surface.create({
-        id: `${meta.id}.project.companion.settings`,
-        role: 'object-settings',
-        filter: (data): data is { subject: Project.Project } => Obj.instanceOf(Project.Project, data.subject),
-        component: ({ data }) => <ProjectSettings subject={data.subject} />,
-      }),
-      // TODO(wittjosiah): This is flashing when chat changes.
-      Surface.create({
-        id: `${meta.id}.companion-chat`,
-        role: 'article',
-        filter: (
-          data,
-        ): data is { subject: Chat.Chat | 'assistant-chat'; attendableId: string; companionTo: Obj.Unknown } =>
-          typeof data.attendableId === 'string' &&
-          Obj.isObject(data.companionTo) &&
-          (Obj.instanceOf(Chat.Chat, data.subject) || data.subject === 'assistant-chat'),
-        component: ({ data, role, ref }) => (
-          <ChatCompanion
-            role={role}
-            data={data}
-            // TODO(burdon): See comment in ChatCompanion.tsx
-            // data={data.subject}
-            // companionTo={data.companionTo}
-            // attendableId={data.attendableId}
-            ref={ref}
-          />
-        ),
-      }),
-      Surface.create({
-        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 (
-            <Panel.Root role={role} className='dx-document'>
-              <Panel.Content asChild>
-                <InvocationTraceContainer db={space?.db} queueDxn={queueDxn} target={target} detailAxis='block' />
-              </Panel.Content>
-            </Panel.Root>
-          );
-        },
-      }),
-      Surface.create({
-        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} />,
-      }),
-      Surface.create({
-        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} />,
-      }),
-      Surface.create({
-        id: ASSISTANT_DIALOG,
-        role: 'dialog',
-        filter: (data): data is { props: { chat: Chat.Chat } } => data.component === ASSISTANT_DIALOG,
-        component: ({ data }) => <ChatDialog {...data.props} />,
-      }),
-      Surface.create({
-        id: `${meta.id}.trace`,
-        role: 'deck-companion--trace',
-        filter: (data): data is { subject: 'trace' } => data.subject === 'trace',
-        component: () => {
-          const space = useActiveSpace();
-          if (!space) {
-            return null;
-          }
-
-          return <TracePanel space={space} />;
-        },
-      }),
-      Surface.create({
-        id: `${meta.id}.status`,
-        role: 'status-indicator',
-        component: () => <TriggerStatus />,
-      }),
-      Surface.create({
-        id: `${meta.id}.prompts`,
-        role: 'prompts',
-        filter: (data): data is { subject: Obj.Unknown } => Obj.isObject(data.subject),
-        component: ({ data }) => <PromptList subject={data.subject} />,
-      }),
-    ]),
-  ),
-);

package/src/capabilities/settings/index.ts

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

package/src/capabilities/state/index.ts

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

package/src/capabilities/toolkit/index.ts

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

package/src/components/ChatProgress/ChatProgress.tsx

@@ -1,56 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react';
-
-import { addEventListener } from '@dxos/async';
-import { type Queue } from '@dxos/echo-db';
-import { ProgressBar, type ProgressBarProps, TextCrawl, useExecutionGraph } from '@dxos/react-ui-components';
-
-export type ChatProgressProps = {
-  traceQueue?: Queue;
-};
-
-/**
- * Displays execution progress from the space's invocation trace queue.
- */
-export const ChatProgress = ({ traceQueue }: ChatProgressProps) => {
-  const { commits } = useExecutionGraph(traceQueue, true);
-  const nodes = useMemo(() => commits.map((commit) => ({ id: commit.id, message: commit.message })), [commits]);
-  const lines = useMemo(() => commits.map((commit) => commit.message), [commits]);
-
-  const [index, setIndex] = useState<number | undefined>(undefined);
-  const handleSelect = useCallback<NonNullable<ProgressBarProps['onSelect']>>(({ index }) => {
-    setIndex((current) => {
-      return current === index ? undefined : index;
-    });
-  }, []);
-
-  const ref = useRef<HTMLDivElement>(null);
-  useEffect(() => {
-    if (!ref.current) {
-      return;
-    }
-
-    return addEventListener(ref.current, 'keydown', (event) => {
-      switch (event.key) {
-        case 'ArrowLeft': {
-          setIndex((current) => (current === undefined || current <= 0 ? 0 : current - 1));
-          break;
-        }
-        case 'ArrowRight': {
-          setIndex((current) => (current === undefined || current >= nodes.length ? nodes.length - 1 : current + 1));
-          break;
-        }
-      }
-    });
-  }, [nodes.length]);
-
-  return (
-    <div role='none' tabIndex={0} ref={ref} className='flex flex-col outline-hidden'>
-      <ProgressBar nodes={nodes} index={index} onSelect={handleSelect} />
-      <TextCrawl classNames='ps-4 text-sm text-description' lines={lines} autoAdvance />
-    </div>
-  );
-};

package/src/components/ChatProgress/index.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './ChatProgress';