@dxos/plugin-assistant 0.8.4-main.ae835ea → 0.8.4-main.bc2380dfbc

package/src/AssistantPlugin.tsx

@@ -1,171 +0,0 @@
-//
-// Copyright 2023 DXOS.org
-//
-
-import { Capabilities, Events, contributes, createIntent, defineModule, definePlugin } from '@dxos/app-framework';
-import { ResearchGraph } from '@dxos/assistant-toolkit';
-import { Blueprint, Prompt } from '@dxos/blueprints';
-import { Sequence } from '@dxos/conductor';
-import { Type } from '@dxos/echo';
-import { ClientCapabilities, ClientEvents } from '@dxos/plugin-client';
-import { SpaceCapabilities, SpaceEvents } from '@dxos/plugin-space';
-import { defineObjectForm } from '@dxos/plugin-space/types';
-import { DataType } from '@dxos/schema';
-
-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, AssistantAction, ServiceType } from './types';
-
-export const AssistantPlugin = definePlugin(meta, () => [
-  defineModule({
-    id: `${meta.id}/module/translations`,
-    activatesOn: Events.SetupTranslations,
-    activate: () => contributes(Capabilities.Translations, translations),
-  }),
-  defineModule({
-    id: `${meta.id}/module/settings`,
-    activatesOn: Events.SetupSettings,
-    activate: Settings,
-  }),
-  defineModule({
-    id: `${meta.id}/module/state`,
-    // TODO(wittjosiah): Does not integrate with settings store.
-    //   Should this be a different event?
-    //   Should settings store be renamed to be more generic?
-    activatesOn: Events.SetupSettings,
-    activate: AssistantState,
-  }),
-  defineModule({
-    id: `${meta.id}/module/metadata`,
-    activatesOn: Events.SetupMetadata,
-    activate: () => [
-      contributes(Capabilities.Metadata, {
-        id: Type.getTypename(Assistant.Chat),
-        metadata: {
-          icon: 'ph--atom--regular',
-          iconHue: 'sky',
-        },
-      }),
-      contributes(Capabilities.Metadata, {
-        id: Type.getTypename(Blueprint.Blueprint),
-        metadata: {
-          icon: 'ph--blueprint--regular',
-          iconHue: 'sky',
-        },
-      }),
-      contributes(Capabilities.Metadata, {
-        id: Type.getTypename(Sequence),
-        metadata: {
-          icon: 'ph--circuitry--regular',
-          iconHue: 'sky',
-        },
-      }),
-    ],
-  }),
-  defineModule({
-    id: `${meta.id}/module/object-form`,
-    activatesOn: ClientEvents.SetupSchema,
-    activate: () => [
-      contributes(
-        SpaceCapabilities.ObjectForm,
-        defineObjectForm({
-          objectSchema: Assistant.Chat,
-          getIntent: (_, options) => createIntent(AssistantAction.CreateChat, { space: options.space }),
-        }),
-      ),
-      contributes(
-        SpaceCapabilities.ObjectForm,
-        defineObjectForm({
-          objectSchema: Blueprint.Blueprint,
-          formSchema: AssistantAction.BlueprintForm,
-          getIntent: (props) => createIntent(AssistantAction.CreateBlueprint, props),
-        }),
-      ),
-      contributes(
-        SpaceCapabilities.ObjectForm,
-        defineObjectForm({
-          objectSchema: Sequence,
-          getIntent: () => createIntent(AssistantAction.CreateSequence),
-        }),
-      ),
-    ],
-  }),
-  defineModule({
-    id: `${meta.id}/module/schema`,
-    activatesOn: ClientEvents.SetupSchema,
-    activate: () =>
-      contributes(ClientCapabilities.Schema, [
-        ServiceType,
-        Assistant.CompanionTo,
-        ResearchGraph,
-        DataType.HasSubject,
-        Prompt.Prompt,
-      ]),
-  }),
-  defineModule({
-    id: `${meta.id}/module/on-space-created`,
-    activatesOn: SpaceEvents.SpaceCreated,
-    activate: () =>
-      contributes(SpaceCapabilities.onCreateSpace, ({ rootCollection, space }) =>
-        createIntent(AssistantAction.onCreateSpace, { rootCollection, space }),
-      ),
-  }),
-  defineModule({
-    id: `${meta.id}/module/app-graph-builder`,
-    activatesOn: Events.SetupAppGraph,
-    activate: AppGraphBuilder,
-  }),
-  defineModule({
-    id: `${meta.id}/module/intent-resolver`,
-    activatesOn: Events.SetupIntentResolver,
-    activate: IntentResolver,
-  }),
-  defineModule({
-    id: `${meta.id}/module/react-surface`,
-    activatesOn: Events.SetupReactSurface,
-    // TODO(wittjosiah): Should occur before the chat is loaded when surfaces activation is more granular.
-    activatesBefore: [Events.SetupArtifactDefinition],
-    activate: ReactSurface,
-  }),
-  defineModule({
-    id: `${meta.id}/module/edge-model-resolver`,
-    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: 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/capabilities.ts

@@ -1,18 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-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`);
-}

package/src/capabilities/intent-resolver.ts

@@ -1,121 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Capabilities, type PluginContext, contributes, createIntent, createResolver } from '@dxos/app-framework';
-import { AiContextBinder, AiConversation } from '@dxos/assistant';
-import { Blueprint, Template } from '@dxos/blueprints';
-import { fullyQualifiedId } from '@dxos/client/echo';
-import { type Queue } from '@dxos/client/echo';
-import { Sequence } from '@dxos/conductor';
-import { Filter, Key, Obj, Ref } from '@dxos/echo';
-import { TracingService } from '@dxos/functions';
-import { AutomationCapabilities } from '@dxos/plugin-automation';
-import { CollectionAction } from '@dxos/plugin-space/types';
-import { getSpace } from '@dxos/react-client/echo';
-import { type DataType } from '@dxos/schema';
-
-import { type AiChatServices, updateName } from '../processor';
-import { Assistant, AssistantAction } from '../types';
-
-import { ASSISTANT_BLUEPRINT_KEY, createBlueprint } from './blueprint-definition';
-import { AssistantCapabilities } from './capabilities';
-
-export default (context: PluginContext) => [
-  contributes(Capabilities.IntentResolver, [
-    createResolver({
-      intent: AssistantAction.onCreateSpace,
-      resolve: ({ space, rootCollection }) =>
-        Effect.gen(function* () {
-          const { dispatch } = context.getCapability(Capabilities.IntentDispatcher);
-          const { object: chatCollection } = yield* dispatch(
-            createIntent(CollectionAction.CreateQueryCollection, { typename: Assistant.Chat.typename }),
-          );
-          const { object: blueprintCollection } = yield* dispatch(
-            createIntent(CollectionAction.CreateQueryCollection, { typename: Blueprint.Blueprint.typename }),
-          );
-          rootCollection.objects.push(Ref.make(chatCollection), Ref.make(blueprintCollection));
-
-          // Create default chat.
-          const { object: chat } = yield* dispatch(createIntent(AssistantAction.CreateChat, { space }));
-          space.db.add(chat);
-        }),
-    }),
-    createResolver({
-      intent: AssistantAction.CreateChat,
-      resolve: async ({ space, name }) => {
-        const queue = space.queues.create();
-        const chat = Assistant.makeChat({ name, queue });
-        const { objects: blueprints } = await space.db.query(Filter.type(Blueprint.Blueprint)).run();
-        // TODO(wittjosiah): This should be a space-level setting.
-        // TODO(burdon): Clone when activated. Copy-on-write for template.
-        let defaultBlueprint = blueprints.find((blueprint) => blueprint.key === ASSISTANT_BLUEPRINT_KEY);
-        if (!defaultBlueprint) {
-          defaultBlueprint = space.db.add(createBlueprint());
-        }
-
-        const binder = new AiContextBinder(queue);
-        await binder.open();
-        await binder.bind({ blueprints: [Ref.make(defaultBlueprint)] });
-        await binder.close();
-
-        return {
-          data: { object: chat },
-        };
-      },
-    }),
-    createResolver({
-      intent: AssistantAction.UpdateChatName,
-      resolve: async ({ chat }) => {
-        const space = getSpace(chat);
-        const queue = chat.queue.target as Queue<DataType.Message>;
-        if (!space || !queue) {
-          return;
-        }
-
-        const runtimeResolver = context.getCapability(AutomationCapabilities.ComputeRuntime);
-        const runtime = await runtimeResolver.getRuntime(space.id).runPromise(
-          Effect.gen(function* () {
-            return yield* Effect.runtime<AiChatServices>().pipe(Effect.provide(TracingService.layerNoop));
-          }),
-        );
-
-        await new AiConversation(queue).use(async (conversation) => updateName(runtime, conversation, chat));
-      },
-    }),
-    createResolver({
-      intent: AssistantAction.CreateBlueprint,
-      resolve: ({ key, name, description }) => ({
-        data: {
-          object: Blueprint.make({ key, name, description, instructions: Template.make({ source: '' }) }),
-        },
-      }),
-    }),
-    createResolver({
-      intent: AssistantAction.CreateSequence,
-      resolve: ({ name }) => ({
-        data: {
-          object: Obj.make(Sequence, {
-            name,
-            steps: [
-              {
-                id: Key.ObjectId.random(),
-                instructions: 'You are a helpful assistant.',
-              },
-            ],
-          }),
-        },
-      }),
-    }),
-    createResolver({
-      intent: AssistantAction.SetCurrentChat,
-      resolve: ({ companionTo, chat }) =>
-        Effect.gen(function* () {
-          const state = context.getCapability(AssistantCapabilities.MutableState);
-          state.currentChat[fullyQualifiedId(companionTo)] = chat;
-        }),
-    }),
-  ]),
-];

package/src/components/BlueprintContainer.tsx

@@ -1,30 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React from 'react';
-
-import { type Blueprint } from '@dxos/blueprints';
-import { fullyQualifiedId } from '@dxos/react-client/echo';
-import { Toolbar } from '@dxos/react-ui';
-import { useAttention } from '@dxos/react-ui-attention';
-import { StackItem } from '@dxos/react-ui-stack';
-
-import { TemplateEditor } from './TemplateEditor';
-
-export type BlueprintContainerProps = {
-  blueprint: Blueprint.Blueprint;
-};
-
-export const BlueprintContainer = ({ blueprint }: BlueprintContainerProps) => {
-  const { hasAttention } = useAttention(fullyQualifiedId(blueprint));
-
-  return (
-    <StackItem.Content toolbar>
-      <Toolbar.Root disabled={!hasAttention} />
-      <TemplateEditor id={blueprint.id} template={blueprint.instructions} classNames='container-max-width' />
-    </StackItem.Content>
-  );
-};
-
-export default BlueprintContainer;

package/src/components/ChatCompanion.tsx

@@ -1,136 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Array from 'effect/Array';
-import * as Effect from 'effect/Effect';
-import * as Function from 'effect/Function';
-import * as Option from 'effect/Option';
-import React, { useMemo } from 'react';
-
-import { Capabilities, createIntent, useCapabilities, useIntentDispatcher } from '@dxos/app-framework';
-import { Blueprint } from '@dxos/blueprints';
-import { getSpace } from '@dxos/client/echo';
-import { Filter, Obj, Query, Ref } from '@dxos/echo';
-import { DatabaseService } from '@dxos/functions';
-import { SpaceAction } from '@dxos/plugin-space/types';
-import { useQuery } from '@dxos/react-client/echo';
-import { useAsyncEffect } from '@dxos/react-ui';
-
-import { ChatContainer } from '../components';
-import { useBlueprintRegistry, useContextBinder } from '../hooks';
-import { Assistant, AssistantAction } from '../types';
-
-export type ChatCompanionProps = {
-  role?: string;
-  data: { companionTo: Obj.Any; subject: Assistant.Chat | 'assistant-chat' };
-};
-
-export const ChatCompanion = ({ role, data }: ChatCompanionProps) => {
-  const companionTo = data.companionTo;
-  const space = getSpace(companionTo);
-  const chat = data.subject === 'assistant-chat' ? undefined : data.subject;
-  const blueprintRegistry = useBlueprintRegistry();
-  const binder = useContextBinder(chat);
-  const { dispatch } = useIntentDispatcher();
-
-  // Initialize companion chat if it doesn't exist.
-  useAsyncEffect(async () => {
-    if (chat || !space) {
-      return;
-    }
-
-    // TODO(burdon): Garbage collection of queues?
-    // TODO(wittjosiah): Figure out how to prevent companion chats from showing up in the navtree.
-    await Effect.gen(function* () {
-      const { objects } = yield* DatabaseService.runQuery(
-        Query.select(Filter.ids(companionTo.id)).targetOf(Assistant.CompanionTo).source(),
-      );
-
-      // TODO(wittjosiah): This should be the default sort order.
-      let nextChat = objects.toSorted((a, b) => a.id.localeCompare(b.id)).at(-1);
-      if (!nextChat) {
-        const { object } = yield* dispatch(createIntent(AssistantAction.CreateChat, { space }));
-        nextChat = object;
-        yield* dispatch(createIntent(SpaceAction.AddObject, { object: nextChat, target: space, hidden: true }));
-        yield* dispatch(
-          createIntent(SpaceAction.AddRelation, {
-            space,
-            schema: Assistant.CompanionTo,
-            source: nextChat,
-            target: data.companionTo,
-          }),
-        );
-      }
-
-      yield* dispatch(createIntent(AssistantAction.SetCurrentChat, { companionTo, chat: nextChat }));
-    }).pipe(Effect.provide(DatabaseService.layer(space.db)), Effect.runPromise);
-  }, [space, chat, companionTo]);
-
-  const metadata = useCapabilities(Capabilities.Metadata);
-  const blueprintKeys = useMemo(
-    () =>
-      Function.pipe(
-        metadata,
-        Array.findFirst(
-          (capability): capability is { id: string; metadata: { blueprints?: string[] } } =>
-            capability.id === Obj.getTypename(companionTo),
-        ),
-        Option.flatMap((c) => Option.fromNullable(c.metadata.blueprints)),
-        Option.getOrElse(() => [] as string[]),
-      ),
-    [metadata, companionTo],
-  );
-  const existingBlueprints = useQuery(space, Filter.type(Blueprint.Blueprint));
-  const pluginBlueprints = useMemo(
-    () => existingBlueprints.filter((blueprint) => blueprintKeys.includes(blueprint.key)),
-    [existingBlueprints, blueprintKeys],
-  );
-
-  // Initialize related blueprints that are not already in the space.
-  useAsyncEffect(async () => {
-    if (!space) {
-      return;
-    }
-
-    // NOTE: This must be run instead of using the useQuery result to avoid duplicates.
-    const { objects: existingBlueprints } = await space.db.query(Filter.type(Blueprint.Blueprint)).run();
-    for (const key of blueprintKeys) {
-      const existingBlueprint = existingBlueprints.find((blueprint) => blueprint.key === key);
-      if (existingBlueprint) {
-        continue;
-      }
-
-      const blueprint = blueprintRegistry.getByKey(key);
-      if (!blueprint) {
-        continue;
-      }
-
-      space.db.add(Obj.clone(blueprint));
-    }
-  }, [space, blueprintRegistry, blueprintKeys]);
-
-  useAsyncEffect(async () => {
-    if (!binder) {
-      return;
-    }
-
-    if (pluginBlueprints.length > 0) {
-      await binder.bind({ blueprints: pluginBlueprints.map((blueprint) => Ref.make(blueprint)) });
-    }
-
-    if (Obj.instanceOf(Blueprint.Blueprint, companionTo)) {
-      await binder.bind({ blueprints: [Ref.make(companionTo)] });
-    } else {
-      await binder.bind({ objects: [Ref.make(companionTo)] });
-    }
-  }, [binder, companionTo, pluginBlueprints]);
-
-  if (!chat) {
-    return null;
-  }
-
-  return <ChatContainer role={role} chat={chat} companionTo={companionTo} />;
-};
-
-export default ChatCompanion;

package/src/components/ChatContainer.tsx

@@ -1,51 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React from 'react';
-
-import { Capabilities, useCapability } from '@dxos/app-framework';
-import { getSpace } from '@dxos/client/echo';
-import { type Obj } from '@dxos/echo';
-import { StackItem } from '@dxos/react-ui-stack';
-
-import { useBlueprintRegistry, useChatProcessor, useChatServices, useOnline, usePresets } from '../hooks';
-import { meta } from '../meta';
-import { type Assistant } from '../types';
-
-import { Chat } from './Chat';
-import { ChatToolbar } from './ChatToolbar';
-
-export type ChatContainerProps = {
-  chat: Assistant.Chat;
-  companionTo?: Obj.Any;
-  role?: string;
-};
-
-export const ChatContainer = ({ chat, companionTo }: ChatContainerProps) => {
-  const space = getSpace(chat);
-  const settings = useCapability(Capabilities.SettingsStore).getStore<Assistant.Settings>(meta.id)?.value;
-  const services = useChatServices({ space, chat });
-  const [online, setOnline] = useOnline();
-  const { preset, ...chatProps } = usePresets(online);
-  const blueprintRegistry = useBlueprintRegistry();
-  const processor = useChatProcessor({ chat, preset, services, blueprintRegistry, settings });
-
-  if (!processor) {
-    return null;
-  }
-
-  return (
-    <StackItem.Content toolbar>
-      <ChatToolbar chat={chat} companionTo={companionTo} />
-      <Chat.Root classNames='container-max-width' chat={chat} processor={processor}>
-        <Chat.Thread />
-        <div className='p-2'>
-          <Chat.Prompt {...chatProps} outline preset={preset?.id} online={online} onOnlineChange={setOnline} />
-        </div>
-      </Chat.Root>
-    </StackItem.Content>
-  );
-};
-
-export default ChatContainer;

package/src/components/ChatProgress/ChatProgress.tsx

@@ -1,67 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react';
-
-import { addEventListener } from '@dxos/async';
-import { Flex, ProgressBar, type ProgressBarProps, TextCrawl, useExecutionGraph } from '@dxos/react-ui-components';
-
-import { type Assistant } from '../../types';
-
-// TODO(burdon): Reset after each session.
-// TODO(burdon): Show after delay.
-// TODO(burdon): Test with errors.
-// TODO(burdon): Filter out empty trace messages.
-
-// {
-//   "id": "01K3TE3C2VAXETP500QKDHK3C4_block_3",
-//   "branch": "main",
-//   "parents": [],
-//   "icon": "ph--robot--regular",
-//   "message": "\n"
-// }
-
-export type ChatProgressProps = {
-  chat: Assistant.Chat;
-};
-
-export const ChatProgress = ({ chat }: ChatProgressProps) => {
-  const { commits } = useExecutionGraph(chat.traceQueue?.target, 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 (
-    <Flex column tabIndex={0} ref={ref} classNames='outline-none'>
-      <ProgressBar nodes={nodes} index={index} onSelect={handleSelect} />
-      <TextCrawl lines={lines} index={index} autoAdvance classNames='pis-4 text-sm text-subdued' />
-    </Flex>
-  );
-};

package/src/components/ChatProgress/index.ts

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