@dxos/plugin-assistant 0.8.4-main.dedc0f3 → 0.8.4-main.dfabb4ec29

package/src/capabilities/compute-runtime.ts

@@ -1,124 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type AiTool, AiToolkit } from '@effect/ai';
-import { BrowserKeyValueStore } from '@effect/platform-browser';
-import { Effect, Layer, ManagedRuntime } from 'effect';
-
-import { Capabilities, type PluginContext, contributes } from '@dxos/app-framework';
-import { makeToolExecutionServiceFromFunctions, makeToolResolverFromFunctions } from '@dxos/assistant';
-import { PropertiesType } from '@dxos/client/echo';
-import { Resource } from '@dxos/context';
-import { Query, Ref } from '@dxos/echo';
-import {
-  CredentialsService,
-  DatabaseService,
-  FunctionImplementationResolver,
-  InvocationTracer,
-  LocalFunctionExecutionService,
-  QueueService,
-  RemoteFunctionExecutionService,
-  TriggerDispatcher,
-} from '@dxos/functions';
-import { TriggerStateStore } from '@dxos/functions';
-import { invariant } from '@dxos/invariant';
-import { type SpaceId } from '@dxos/keys';
-import { ClientCapabilities } from '@dxos/plugin-client';
-
-import { AssistantCapabilities } from './capabilities';
-
-export default async (context: PluginContext) => {
-  const provider = await new ComputeRuntimeProviderImpl(context).open();
-  return contributes(AssistantCapabilities.ComputeRuntime, provider, async () => {
-    await provider.close();
-  });
-};
-
-class ComputeRuntimeProviderImpl extends Resource implements AssistantCapabilities.ComputeRuntimeProvider {
-  readonly #runtimes = new Map<SpaceId, ManagedRuntime.ManagedRuntime<AssistantCapabilities.ComputeServices, never>>();
-  readonly #context: PluginContext;
-
-  constructor(context: PluginContext) {
-    super();
-    this.#context = context;
-  }
-
-  protected override async _open() {}
-
-  protected override async _close() {
-    await Promise.all(Array.from(this.#runtimes.values()).map((rt) => rt.dispose()));
-    this.#runtimes.clear();
-  }
-
-  getRuntime(spaceId: SpaceId): ManagedRuntime.ManagedRuntime<AssistantCapabilities.ComputeServices, never> {
-    if (this.#runtimes.has(spaceId)) {
-      return this.#runtimes.get(spaceId)!;
-    }
-
-    const layer = Layer.unwrapEffect(
-      Effect.gen(this, function* () {
-        const client = this.#context.getCapability(ClientCapabilities.Client);
-        const serviceLayer =
-          this.#context.getCapability(AssistantCapabilities.AiServiceLayer) ?? Layer.die('AiService not found');
-
-        // TODO(dmaretskyi): Make those reactive.
-        const functions = this.#context.getCapabilities(Capabilities.Functions);
-        const toolkits = this.#context.getCapabilities(Capabilities.Toolkit);
-        const handlers = this.#context.getCapabilities(Capabilities.ToolkitHandler);
-
-        const allFunctions = functions.flat();
-        // TODO(wittjosiah): Don't cast.
-        const toolkit = AiToolkit.merge(...toolkits) as AiToolkit.Any as AiToolkit.AiToolkit<AiTool.Any>;
-        const handlersLayer = Layer.mergeAll(Layer.empty, ...handlers);
-
-        const space = client.spaces.get(spaceId);
-        invariant(space);
-        yield* Effect.promise(() => space.waitUntilReady());
-
-        return Layer.mergeAll(TriggerDispatcher.layer({ timeControl: 'natural' })).pipe(
-          Layer.provideMerge(
-            Layer.mergeAll(
-              IvocationTracerLive,
-              TriggerStateStore.layerKv.pipe(Layer.provide(BrowserKeyValueStore.layerLocalStorage)),
-              serviceLayer,
-              makeToolResolverFromFunctions(allFunctions, toolkit),
-              makeToolExecutionServiceFromFunctions(allFunctions, toolkit, handlersLayer),
-              CredentialsService.layerFromDatabase(),
-            ),
-          ),
-          Layer.provideMerge(
-            Layer.mergeAll(
-              space ? DatabaseService.layer(space.db) : DatabaseService.notAvailable,
-              space ? QueueService.layer(space.queues) : QueueService.notAvailable,
-              LocalFunctionExecutionService.layerLive,
-              RemoteFunctionExecutionService.mockLayer,
-            ),
-          ),
-          Layer.provideMerge(FunctionImplementationResolver.layerTest({ functions: allFunctions })),
-        );
-      }),
-    );
-
-    const runtime = ManagedRuntime.make(layer);
-    this.#runtimes.set(spaceId, runtime);
-
-    return runtime;
-  }
-}
-
-const IvocationTracerLive = Layer.unwrapEffect(
-  Effect.gen(function* () {
-    const {
-      objects: [properties],
-    } = yield* DatabaseService.runQuery(Query.type(PropertiesType));
-    invariant(properties);
-    if (!properties.invocationTraceQueue) {
-      const queue = yield* QueueService.createQueue({ subspaceTag: 'trace' });
-      properties.invocationTraceQueue = Ref.fromDXN(queue.dxn);
-    }
-    const queue = properties.invocationTraceQueue.target;
-    invariant(queue);
-    return InvocationTracer.layerLive({ invocationTraceQueue: queue });
-  }),
-);

package/src/capabilities/intent-resolver.ts

@@ -1,94 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Effect } from 'effect';
-
-import { Capabilities, type PluginContext, contributes, createIntent, createResolver } from '@dxos/app-framework';
-import { AiContextBinder } from '@dxos/assistant';
-import { Blueprint, Template } from '@dxos/blueprints';
-import { fullyQualifiedId } from '@dxos/client/echo';
-import { Sequence } from '@dxos/conductor';
-import { Filter, Key, Obj, Ref } from '@dxos/echo';
-import { CollectionAction } from '@dxos/plugin-space/types';
-
-import { Assistant, AssistantAction } from '../types';
-
-import { BLUEPRINT_KEY, createBlueprint } from './blueprint-definition';
-import { AssistantCapabilities } from './capabilities';
-
-export default (context: PluginContext) => [
-  contributes(Capabilities.IntentResolver, [
-    createResolver({
-      intent: AssistantAction.OnSpaceCreated,
-      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 = Obj.make(Assistant.Chat, { name, queue: Ref.fromDXN(queue.dxn) });
-        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 === BLUEPRINT_KEY);
-        if (!defaultBlueprint) {
-          defaultBlueprint = space.db.add(createBlueprint());
-        }
-
-        const binder = new AiContextBinder(queue);
-        await binder.bind({ blueprints: [Ref.make(defaultBlueprint)] });
-
-        return {
-          data: { object: 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,24 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React from 'react';
-
-import { type Blueprint } from '@dxos/blueprints';
-import { StackItem } from '@dxos/react-ui-stack';
-
-import { TemplateEditor } from './TemplateEditor';
-
-export type BlueprintContainerProps = {
-  blueprint: Blueprint.Blueprint;
-};
-
-export const BlueprintContainer = ({ blueprint }: BlueprintContainerProps) => {
-  return (
-    <StackItem.Content classNames='container-max-width'>
-      <TemplateEditor id={blueprint.id} template={blueprint.instructions} />
-    </StackItem.Content>
-  );
-};
-
-export default BlueprintContainer;

package/src/components/ChatCompanion.tsx

@@ -1,111 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Array, Effect, Option, pipe } from 'effect';
-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 { 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 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(
-    () =>
-      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 allBlueprints = useQuery(space, Filter.type(Blueprint.Blueprint));
-  const pluginBlueprints = useMemo(
-    () => allBlueprints.filter((blueprint) => blueprintKeys.includes(blueprint.key)),
-    [allBlueprints, blueprintKeys],
-  );
-
-  // TODO(wittjosiah): Occasionally this fails to bind but seems to be an upstream issue.
-  //   It seems like the queue object signal emits as an empty array after previously emitting a non-empty array.
-  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 { Toolbar } from './Toolbar';
-
-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={!!companionTo} classNames='container-max-width'>
-      {!!companionTo && <Toolbar chat={chat} companionTo={companionTo} />}
-      <Chat.Root 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,68 +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 } from '@dxos/react-ui-components';
-
-import { useExecutionGraph } from '../../hooks';
-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';

package/src/components/ChatThread/reducers.ts

@@ -1,151 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type ContentBlock, type DataType } from '@dxos/schema';
-
-type Reducer<R, I> = (acc: R, value: I, idx: number) => R;
-
-/**
- * Reducer that collapses related message blocks into single messages.
- * For example, combines tool request/response pairs into a single message.
- *
- * Typical order:
- * - message 0
- *  - text
- * - message 1
- *  - text
- *  - toolCall
- *  - summary
- * - message 2
- *  - toolResult
- * - message 3
- *  - text
- *  - summary
- *
- * Coverts to:
- * - message 0
- *  - text
- * - message 1
- *  - text
- * - message 2
- *  - toolCall
- *  - toolResult
- *  - summary
- * - message 3
- *  - text
- *  - summary
- */
-export const reduceMessages: Reducer<
-  {
-    messages: DataType.Message[];
-    current?: DataType.Message;
-    toolBlock?: boolean;
-    assistantMessages?: DataType.Message[];
-  },
-  DataType.Message
-> = ({ messages, current, toolBlock, assistantMessages = [] }, message) => {
-  // Treat tool calls as assistant messages.
-  let assistant = message?.sender.role === 'assistant';
-  let i = 0;
-  for (const block of message.blocks) {
-    switch (block._tag) {
-      // TODO(burdon): Preserve call tool title.
-      case 'toolCall': {
-        assistant = true;
-        if (!toolBlock) {
-          current = {
-            ...message,
-            id: [message.id, i++].join('_'),
-            blocks: [block],
-          };
-          messages.push(current);
-          toolBlock = true;
-        } else {
-          current!.blocks.push(block);
-        }
-        break;
-      }
-
-      case 'toolResult': {
-        assistant = true;
-        if (toolBlock) {
-          current!.blocks.push(block);
-        }
-        break;
-      }
-
-      case 'summary': {
-        assistant = true;
-        if (toolBlock) {
-          current!.blocks.push(block);
-        } else {
-          const summary = reduceSummary([...assistantMessages, message]);
-          current!.blocks.push(summary);
-        }
-        break;
-      }
-
-      default: {
-        assistant = message.sender.role === 'assistant';
-        current = {
-          ...message,
-          id: [message.id, i++].join('_'),
-          blocks: [block],
-        };
-        messages.push(current);
-        toolBlock = false;
-        break;
-      }
-    }
-  }
-
-  return {
-    messages,
-    current,
-    toolBlock,
-    assistantMessages: assistant ? [...assistantMessages, message] : undefined,
-  };
-};
-
-/**
- * Accumulate token counts from all summary blocks in pending messages.
- */
-const reduceSummary = (messages: DataType.Message[]): ContentBlock.Summary => {
-  let start: number | undefined;
-  return messages.reduce<ContentBlock.Summary>(
-    (acc, msg) => {
-      const time = new Date(msg.created).getTime();
-      if (!start) {
-        start = time;
-      }
-      msg.blocks.forEach((block) => {
-        switch (block._tag) {
-          case 'toolCall': {
-            acc.toolCalls = (acc.toolCalls ?? 0) + 1;
-            break;
-          }
-          case 'summary': {
-            acc.model = block.model;
-            if (block.usage) {
-              acc.message = block.message;
-              acc.usage = {
-                inputTokens: (acc.usage?.inputTokens ?? 0) + (block.usage.inputTokens ?? 0),
-                outputTokens: (acc.usage?.outputTokens ?? 0) + (block.usage.outputTokens ?? 0),
-                totalTokens: (acc.usage?.totalTokens ?? 0) + (block.usage.totalTokens ?? 0),
-              };
-              acc.duration = time - (start ?? time);
-            }
-            break;
-          }
-        }
-      });
-
-      return acc;
-    },
-    {
-      _tag: 'summary',
-      duration: 0,
-    } satisfies ContentBlock.Summary,
-  );
-};