@dxos/plugin-assistant 0.8.4-main.fcfe5033a5 → 0.8.4-staging.60fe92afc8

package/src/AssistantPlugin.tsx

@@ -1,233 +0,0 @@
-//
-// Copyright 2023 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Option from 'effect/Option';
-
-import { ActivationEvent, ActivationEvents, Capability, Plugin } from '@dxos/app-framework';
-import { AppActivationEvents, AppPlugin } from '@dxos/app-toolkit';
-import { Agent, AgentBlueprint, Chat, McpServer, Memory, Plan, ResearchGraph } from '@dxos/assistant-toolkit';
-import { Blueprint, Prompt } from '@dxos/blueprints';
-import { Sequence } from '@dxos/conductor';
-import { Annotation, Feed, 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 { DeckEvents } from '@dxos/plugin-deck/types';
-import { MarkdownEvents } from '@dxos/plugin-markdown';
-import { SpaceOperation } from '@dxos/plugin-space/operations';
-import { type CreateObject, SpaceCapabilities, SpaceEvents } from '@dxos/plugin-space/types';
-import { HasSubject } from '@dxos/types';
-
-import {
-  AiService,
-  AppGraphBuilder,
-  AssistantState,
-  BlueprintDefinition,
-  CompanionChatProvisioner,
-  EdgeModelResolver,
-  LocalModelResolver,
-  MarkdownExtension,
-  Migrations,
-  OperationHandler,
-  ReactSurface,
-  Settings,
-  Toolkit,
-} from '#capabilities';
-import { meta } from '#meta';
-import { AssistantOperation } from '#operations';
-import { AssistantEvents } from '#types';
-
-import { translations } from './translations';
-
-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: Annotation.IconAnnotation.get(Chat.Chat).pipe(Option.getOrThrow).icon,
-          iconHue: Annotation.IconAnnotation.get(Chat.Chat).pipe(Option.getOrThrow).hue ?? 'white',
-          createObject: ((props, options) =>
-            Effect.gen(function* () {
-              const { object } = yield* Operation.invoke(AssistantOperation.CreateChat, {
-                db: options.db,
-                name: props?.name,
-              });
-              return yield* Operation.invoke(SpaceOperation.AddObject, {
-                object,
-                target: options.target,
-                hidden: true,
-                targetNodeId: options.targetNodeId,
-              });
-            })) satisfies CreateObject,
-        },
-      },
-      {
-        id: Type.getTypename(Blueprint.Blueprint),
-        metadata: {
-          icon: Annotation.IconAnnotation.get(Blueprint.Blueprint).pipe(Option.getOrThrow).icon,
-          iconHue: Annotation.IconAnnotation.get(Blueprint.Blueprint).pipe(Option.getOrThrow).hue ?? 'white',
-          inputSchema: AssistantOperation.BlueprintForm,
-          createObject: ((props, options) =>
-            Effect.gen(function* () {
-              const object = Blueprint.make(props);
-              return yield* Operation.invoke(SpaceOperation.AddObject, {
-                object,
-                target: options.target,
-                hidden: true,
-                targetNodeId: options.targetNodeId,
-              });
-            })) satisfies CreateObject,
-        },
-      },
-      {
-        id: Type.getTypename(Prompt.Prompt),
-        metadata: {
-          icon: Annotation.IconAnnotation.get(Prompt.Prompt).pipe(Option.getOrThrow).icon,
-          iconHue: Annotation.IconAnnotation.get(Prompt.Prompt).pipe(Option.getOrThrow).hue ?? 'white',
-          createObject: ((props, options) =>
-            Effect.gen(function* () {
-              const object = Prompt.make(props);
-              return yield* Operation.invoke(SpaceOperation.AddObject, {
-                object,
-                target: options.target,
-                hidden: true,
-                targetNodeId: options.targetNodeId,
-              });
-            })) satisfies CreateObject,
-        },
-      },
-      {
-        id: Type.getTypename(Sequence),
-        metadata: {
-          icon: Annotation.IconAnnotation.get(Sequence).pipe(Option.getOrThrow).icon,
-          iconHue: Annotation.IconAnnotation.get(Sequence).pipe(Option.getOrThrow).hue ?? 'white',
-          createObject: ((props, options) =>
-            Effect.gen(function* () {
-              const object = Obj.make(Sequence, props);
-              return yield* Operation.invoke(SpaceOperation.AddObject, {
-                object,
-                target: options.target,
-                hidden: true,
-                targetNodeId: options.targetNodeId,
-              });
-            })) satisfies CreateObject,
-        },
-      },
-      {
-        id: Type.getTypename(Agent.Agent),
-        metadata: {
-          icon: Annotation.IconAnnotation.get(Agent.Agent).pipe(Option.getOrThrow).icon,
-          iconHue: Annotation.IconAnnotation.get(Agent.Agent).pipe(Option.getOrThrow).hue ?? 'white',
-          createObject: ((props, options) =>
-            Effect.gen(function* () {
-              const object = yield* Agent.makeInitialized({ name: '', instructions: '' }, AgentBlueprint.make()).pipe(
-                withComputeRuntime(options.db.spaceId),
-              );
-              return yield* Operation.invoke(SpaceOperation.AddObject, {
-                object,
-                target: options.target,
-                hidden: true,
-                targetNodeId: options.targetNodeId,
-              });
-            })) satisfies CreateObject,
-        },
-      },
-    ],
-  }),
-  AppPlugin.addOperationHandlerModule({ activate: OperationHandler }),
-  AppPlugin.addSchemaModule({
-    schema: [
-      Chat.Chat,
-      Chat.CompanionTo,
-      Blueprint.Blueprint,
-      Feed.Feed,
-      HasSubject.HasSubject,
-      Prompt.Prompt,
-      ResearchGraph.ResearchGraph,
-      Agent.Agent,
-      McpServer.McpServer,
-      Plan.Plan,
-      Sequence,
-      Memory.Memory,
-    ],
-  }),
-  AppPlugin.addSettingsModule({ activate: Settings }),
-  AppPlugin.addSurfaceModule({
-    activate: ReactSurface,
-    firesBeforeActivation: [AppActivationEvents.SetupArtifactDefinition],
-  }),
-  AppPlugin.addTranslationsModule({ translations }),
-  Plugin.addModule({
-    id: 'markdown',
-    activatesOn: MarkdownEvents.SetupExtensions,
-    activate: MarkdownExtension,
-  }),
-  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,
-  }),
-  Plugin.addModule({
-    id: 'on-space-created',
-    activatesOn: SpaceEvents.SpaceCreated,
-    activate: () =>
-      Effect.succeed(
-        Capability.contributes(SpaceCapabilities.OnCreateSpace, (params) =>
-          Operation.invoke(AssistantOperation.OnCreateSpace, params),
-        ),
-      ),
-  }),
-  Plugin.addModule({
-    activatesOn: AssistantEvents.SetupAiServiceProviders,
-    activate: EdgeModelResolver,
-  }),
-  Plugin.addModule({
-    activatesOn: AssistantEvents.SetupAiServiceProviders,
-    activate: LocalModelResolver,
-  }),
-  Plugin.addModule({
-    firesBeforeActivation: [AssistantEvents.SetupAiServiceProviders],
-    // TODO(dmaretskyi): This should activate lazily when the AI chat is used.
-    activatesOn: ActivationEvents.Startup,
-    activate: AiService,
-  }),
-  Plugin.addModule({
-    // TODO(wittjosiah): Use a different event.
-    activatesOn: ActivationEvents.Startup,
-    activate: Toolkit,
-  }),
-  Plugin.addModule({
-    activatesOn: ActivationEvent.allOf(
-      ActivationEvents.OperationInvokerReady,
-      AppActivationEvents.AppGraphReady,
-      DeckEvents.StateReady,
-    ),
-    activate: CompanionChatProvisioner,
-  }),
-  Plugin.addModule({
-    activatesOn: ClientEvents.SetupMigration,
-    activate: Migrations,
-  }),
-  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));
-    });

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';

package/src/components/ChatThread/widgets/PromptWidget.ts

@@ -1,28 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { WidgetType } from '@codemirror/view';
-
-import { Domino } from '@dxos/ui';
-
-/**
- * Simple prompt widget.
- */
-export class PromptWidget extends WidgetType {
-  constructor(private text: string) {
-    super();
-  }
-
-  override eq(other: this) {
-    return this.text === other.text;
-  }
-
-  /**
-   * NOTE: Container must set var based on user's identity.
-   */
-  override toDOM() {
-    const inner = Domino.of('div').classNames('px-3 py-1.5 bg-(--user-fill) rounded-xs').text(this.text);
-    return Domino.of('div').classNames('flex justify-end my-2').append(inner).root;
-  }
-}

package/src/containers/AgentProperties/AgentProperties.stories.tsx

@@ -1,100 +0,0 @@
-//
-// Copyright 2026 DXOS.org
-//
-
-import { type Meta, type StoryObj } from '@storybook/react-vite';
-import * as Effect from 'effect/Effect';
-import React from 'react';
-
-import { withPluginManager } from '@dxos/app-framework/testing';
-import { Plan, Agent } from '@dxos/assistant-toolkit';
-import { Obj, Ref } from '@dxos/echo';
-import { log } from '@dxos/log';
-import { AutomationPlugin } from '@dxos/plugin-automation';
-import { ClientPlugin } from '@dxos/plugin-client';
-import { initializeIdentity } from '@dxos/plugin-client/testing';
-import { corePlugins } from '@dxos/plugin-testing';
-import { Filter, useQuery, useSpaces } from '@dxos/react-client/echo';
-import { Button } from '@dxos/react-ui';
-import { ObjectProperties } from '@dxos/react-ui-form';
-import { Loading, withLayout } from '@dxos/react-ui/testing';
-import { Text } from '@dxos/schema';
-import { Organization } from '@dxos/types';
-
-import { AssistantPlugin } from '../../AssistantPlugin';
-import { translations } from '../../translations';
-import { AgentProperties } from './AgentProperties';
-
-type DefaultStoryProps = {};
-
-const DefaultStory = (_: DefaultStoryProps) => {
-  const [space] = useSpaces();
-  const [agent] = useQuery(space?.db, Filter.type(Agent.Agent));
-  if (!agent) {
-    return <Loading />;
-  }
-
-  return (
-    <ObjectProperties object={agent}>
-      <AgentProperties role='object-properties' subject={agent} />
-      <Button
-        classNames='mt-form-gap'
-        onClick={() => {
-          log.info('agent', { agent, instructions: agent.instructions.target });
-        }}
-      >
-        Debug
-      </Button>
-    </ObjectProperties>
-  );
-};
-
-const meta = {
-  title: 'plugins/plugin-assistant/containers/AgentProperties',
-  render: DefaultStory,
-  decorators: [
-    withLayout({ layout: 'column' }),
-    withPluginManager({
-      plugins: [
-        ...corePlugins(),
-        ClientPlugin({
-          types: [Agent.Agent, Plan.Plan, Text.Text, Organization.Organization],
-          onClientInitialized: ({ client }) =>
-            Effect.gen(function* () {
-              yield* initializeIdentity(client);
-              const [space] = client.spaces.get();
-              yield* Effect.promise(() => space.waitUntilReady());
-
-              const organization = space.db.add(
-                Obj.make(Organization.Organization, {
-                  name: 'Acme Corp',
-                  description: 'Sample organization for the story.',
-                }),
-              );
-
-              space.db.add(
-                Obj.make(Agent.Agent, {
-                  instructions: Ref.make(Text.make()),
-                  plan: Ref.make(Plan.makePlan({ tasks: [] })),
-                  artifacts: [{ name: 'Organization', data: Ref.make(organization) }],
-                  subscriptions: [],
-                }),
-              );
-            }),
-        }),
-        AutomationPlugin(),
-        AssistantPlugin(),
-      ],
-    }),
-  ],
-  parameters: {
-    layout: 'fullscreen',
-    translations,
-  },
-} satisfies Meta<DefaultStoryProps>;
-
-export default meta;
-
-type Story = StoryObj<typeof meta>;
-
-export const Default: Story = {};

package/src/containers/ChatContainer/ChatContainer.tsx

@@ -1,95 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { forwardRef, useEffect, useRef } from 'react';
-
-import { Capabilities } from '@dxos/app-framework';
-import { useAtomCapability, useCapability } from '@dxos/app-framework/ui';
-import { type AppSurface } from '@dxos/app-toolkit/ui';
-import { type Space, getSpace } from '@dxos/client/echo';
-import { Feed, type Obj } from '@dxos/echo';
-import { Panel } from '@dxos/react-ui';
-import { getParentId } from '@dxos/react-ui-attention';
-
-import { Chat as ChatComponent, type ChatRootProps } from '#components';
-import { useBlueprintRegistry, useChatProcessor, useChatServices, useOnline, usePresets } from '#hooks';
-import { AssistantCapabilities, type ChatType } from '#types';
-
-export type ChatContainerProps = (
-  | AppSurface.ObjectArticleProps<ChatType.Chat | undefined>
-  | AppSurface.ObjectSectionProps<ChatType.Chat | undefined>
-) & {
-  space?: Space;
-  companionTo?: Obj.Unknown;
-} & Pick<ChatRootProps, 'onEvent'>;
-
-export const ChatContainer = forwardRef<HTMLDivElement, ChatContainerProps>((props, forwardedRef) => {
-  const { role, attendableId, subject: chat, space: spaceProp, companionTo, onEvent } = props;
-  const parentId = attendableId ? getParentId(attendableId) : undefined;
-  const space = spaceProp ?? getSpace(chat);
-  const settings = useAtomCapability(AssistantCapabilities.Settings);
-  const runtime = useChatServices({ id: space?.id });
-  const [online, setOnline] = useOnline();
-  const { preset, ...chatProps } = usePresets(online);
-  const blueprintRegistry = useBlueprintRegistry();
-  const processor = useChatProcessor({
-    space,
-    chat,
-    preset,
-    runtime,
-    blueprintRegistry,
-    settings,
-  });
-
-  const registry = useCapability(Capabilities.AtomRegistry);
-  const stateAtom = useCapability(AssistantCapabilities.State);
-  const pendingSubmitted = useRef(false);
-  useEffect(() => {
-    if (!processor || !attendableId || pendingSubmitted.current) {
-      return;
-    }
-
-    const state = registry.get(stateAtom);
-    const pendingPrompt = state.pendingPrompts[attendableId];
-    if (pendingPrompt) {
-      pendingSubmitted.current = true;
-      registry.update(stateAtom, (current) => {
-        const { [attendableId]: _, ...rest } = current.pendingPrompts;
-        return { ...current, pendingPrompts: rest };
-      });
-      void processor.request({ message: pendingPrompt });
-    }
-  }, [processor, attendableId, registry, stateAtom]);
-
-  const feedTarget = chat?.feed.target;
-  const queue = space && feedTarget ? space.queues.get(Feed.getQueueDxn(feedTarget)!) : undefined;
-
-  if (!processor) {
-    return null;
-  }
-
-  return (
-    <ChatComponent.Root db={space?.db} chat={chat} queue={queue} processor={processor} onEvent={onEvent}>
-      <Panel.Root role={role} ref={forwardedRef}>
-        <Panel.Toolbar className='bg-toolbar-surface'>
-          <ChatComponent.Toolbar classNames='dx-document' attendableId={attendableId} companionTo={companionTo} />
-        </Panel.Toolbar>
-        <Panel.Content>
-          <ChatComponent.Viewport>
-            <ChatComponent.Thread />
-            <div role='none' className='dx-document p-4'>
-              <ChatComponent.Prompt
-                {...chatProps}
-                outline
-                preset={preset?.id}
-                online={online}
-                onOnlineChange={setOnline}
-              />
-            </div>
-          </ChatComponent.Viewport>
-        </Panel.Content>
-      </Panel.Root>
-    </ChatComponent.Root>
-  );
-});

package/src/containers/ChatContainer/index.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export { ChatContainer as default } from './ChatContainer';

package/src/containers/PromptArticle/PromptArticle.tsx

@@ -1,59 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { useMemo } from 'react';
-
-import { type AppSurface } from '@dxos/app-toolkit/ui';
-import { AgentPrompt } from '@dxos/assistant-toolkit';
-import { type Prompt } from '@dxos/blueprints';
-import { Obj } from '@dxos/echo';
-import { invariant } from '@dxos/invariant';
-import { Operation } from '@dxos/operation';
-import { useComputeRuntimeCallback } from '@dxos/plugin-automation/hooks';
-import { Panel, Toolbar, useTranslation } from '@dxos/react-ui';
-import { useAttention } from '@dxos/react-ui-attention';
-
-import { TemplateEditor } from '#components';
-import { meta } from '#meta';
-
-export type PromptArticleProps = AppSurface.ObjectArticleProps<Prompt.Prompt>;
-
-export const PromptArticle = ({ role, attendableId, subject }: PromptArticleProps) => {
-  const { t } = useTranslation(meta.id);
-  const { hasAttention } = useAttention(attendableId);
-  const db = Obj.getDatabase(subject);
-
-  const inputData = useMemo<Operation.Definition.Input<typeof AgentPrompt> | undefined>(
-    () =>
-      subject && db
-        ? {
-            prompt: db.makeRef(Obj.getDXN(subject)),
-            input: {},
-          }
-        : undefined,
-    [subject, db],
-  );
-
-  const handleRun = useComputeRuntimeCallback(
-    db?.spaceId,
-    () => {
-      invariant(inputData);
-      return Operation.invoke(AgentPrompt, inputData);
-    },
-    [inputData],
-  );
-
-  return (
-    <Panel.Root role={role} className='dx-document'>
-      <Panel.Toolbar asChild>
-        <Toolbar.Root disabled={!hasAttention} onClick={handleRun}>
-          <Toolbar.IconButton iconOnly icon='ph--play--regular' label={t('run-prompt.label')} onClick={handleRun} />
-        </Toolbar.Root>
-      </Panel.Toolbar>
-      <Panel.Content asChild>
-        <TemplateEditor id={subject.id} template={subject.instructions} />
-      </Panel.Content>
-    </Panel.Root>
-  );
-};

package/src/containers/PromptArticle/index.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export { PromptArticle as default } from './PromptArticle';

package/src/containers/PromptList/index.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export { PromptList as default } from './PromptList';