@dxos/plugin-assistant 0.8.4-main.f9ba587 → 0.8.4-main.fd6878d

package/src/capabilities/ai-client.ts

@@ -1,39 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { effect, signal } from '@preact/signals-core';
-
-import { type AiServiceClient, EdgeAiServiceClient, OllamaAiServiceClient } from '@dxos/ai';
-import { Capabilities, contributes, type PluginContext } from '@dxos/app-framework';
-import { ClientCapabilities } from '@dxos/plugin-client';
-
-import { AssistantCapabilities } from './capabilities';
-import { meta } from '../meta';
-import { type Assistant } from '../types';
-
-// TODO(wittjosiah): Factor out.
-const DEFAULT_AI_SERVICE_URL = 'http://localhost:8788';
-
-export default (context: PluginContext) => {
-  const client = context.getCapability(ClientCapabilities.Client);
-  const endpoint = client.config.values.runtime?.services?.ai?.server ?? DEFAULT_AI_SERVICE_URL;
-  const aiClient = signal<AiServiceClient>(new EdgeAiServiceClient({ endpoint }));
-
-  const unsubscribe = effect(() => {
-    const settings = context.getCapability(Capabilities.SettingsStore).getStore<Assistant.Settings>(meta.id)?.value;
-    if (settings?.llmProvider === 'ollama') {
-      aiClient.value = new OllamaAiServiceClient();
-    } else {
-      aiClient.value = new EdgeAiServiceClient({
-        endpoint,
-        defaultGenerationOptions: {
-          // model: '@anthropic/claude-sonnet-4-20250514',
-          model: '@anthropic/claude-3-5-haiku-20241022',
-        },
-      });
-    }
-  });
-
-  return contributes(AssistantCapabilities.AiClient, aiClient, () => unsubscribe());
-};

package/src/components/ChatPrompt/ChatOptionsMenu.tsx

@@ -1,52 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { useMemo } from 'react';
-
-import { type Blueprint, type BlueprintRegistry } from '@dxos/assistant';
-import { DropdownMenu, IconButton, Input, useTranslation } from '@dxos/react-ui';
-
-import { meta } from '../../meta';
-
-export type ChatOptionsMenuProps = {
-  blueprintRegistry?: BlueprintRegistry;
-  blueprints?: Blueprint[];
-  onChange?: (key: string, active: boolean) => void;
-};
-
-export const ChatOptionsMenu = ({ blueprintRegistry, blueprints, onChange }: ChatOptionsMenuProps) => {
-  const { t } = useTranslation(meta.id);
-  const blueprintOptions = useMemo(
-    () => blueprintRegistry?.query().map((blueprint) => ({ key: blueprint.key, label: blueprint.name })),
-    [blueprintRegistry],
-  );
-
-  return (
-    <DropdownMenu.Root>
-      <DropdownMenu.Trigger asChild disabled={!blueprintOptions?.length}>
-        <IconButton icon='ph--plus--regular' variant='ghost' size={5} iconOnly label={t('button add blueprint')} />
-      </DropdownMenu.Trigger>
-      <DropdownMenu.Portal>
-        <DropdownMenu.Content side='left'>
-          <DropdownMenu.Viewport>
-            {blueprintOptions?.map((option) => (
-              <DropdownMenu.Item key={option.key}>
-                <Input.Root>
-                  <Input.Checkbox
-                    checked={!!blueprints?.find((blueprint) => blueprint.key === option.key)}
-                    onCheckedChange={(checked) => onChange?.(option.key, !!checked)}
-                  />
-                  {/* TODO(burdon): Remove need for custom margin. */}
-                  {/* TODO(burdon): Clicking on label doesn't toggle checkbox. */}
-                  <Input.Label classNames='m-0'>{option.label}</Input.Label>
-                </Input.Root>
-              </DropdownMenu.Item>
-            ))}
-          </DropdownMenu.Viewport>
-          <DropdownMenu.Arrow />
-        </DropdownMenu.Content>
-      </DropdownMenu.Portal>
-    </DropdownMenu.Root>
-  );
-};

package/src/components/ChatPrompt/events.ts

@@ -1,12 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export type ChatEvent =
-  | { type: 'submit'; text: string }
-  | { type: 'cancel' }
-  | { type: 'thread-open' }
-  | { type: 'thread-close' }
-  | { type: 'scroll-to-bottom' }
-  | { type: 'record-start' }
-  | { type: 'record-stop' };

package/src/components/TemplateContainer.tsx

@@ -1,25 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React from 'react';
-
-import { StackItem } from '@dxos/react-ui-stack';
-
-import { TemplateEditor } from './TemplateEditor';
-import { type TemplateType } from '../types';
-
-export type TemplateContainerProps = {
-  role: string;
-  template: TemplateType;
-};
-
-export const TemplateContainer = ({ role, template }: TemplateContainerProps) => {
-  return (
-    <StackItem.Content role={role} classNames='container-max-width'>
-      <TemplateEditor template={template} />
-    </StackItem.Content>
-  );
-};
-
-export default TemplateContainer;

package/src/hooks/chat-processor.test.ts

@@ -1,15 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { describe, it } from 'vitest';
-
-import { ChatProcessor } from './chat-processor';
-
-describe('ChatProcessor', () => {
-  it('should be instantiable', ({ expect }) => {
-    const client = {} as any; // TODO(burdon): Create mock.
-    const processor = new ChatProcessor(client);
-    expect(processor).toBeDefined();
-  });
-});

package/src/hooks/useBlueprints.ts

@@ -1,31 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { useCallback, useState } from 'react';
-
-import { type Blueprint, type ContextBinder } from '@dxos/assistant';
-import { Ref } from '@dxos/echo';
-import { log } from '@dxos/log';
-import { useTimeout } from '@dxos/react-ui';
-import { isNonNullable } from '@dxos/util';
-
-export type UpdateCallback = (key: string, active: boolean) => void;
-
-export const useBlueprints = (context: ContextBinder): [Blueprint[], UpdateCallback] => {
-  const [blueprints, setBlueprints] = useState<Blueprint[]>([]);
-  useTimeout(
-    async () => {
-      const blueprints = (await Ref.Array.loadAll(context.blueprints.value ?? [])).filter(isNonNullable);
-      setBlueprints(blueprints);
-    },
-    0,
-    [context],
-  );
-
-  const handleUpdate = useCallback<UpdateCallback>((key: string, active: boolean) => {
-    log.info('update', { key, active });
-  }, []);
-
-  return [blueprints, handleUpdate] as const;
-};

package/src/hooks/useChatProcessor.tsx

@@ -1,142 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { useEffect, useMemo, useState } from 'react';
-
-import { DEFAULT_EDGE_MODEL, DEFAULT_OLLAMA_MODEL, type ExecutableTool } from '@dxos/ai';
-import { Capabilities, useCapabilities, useIntentDispatcher } from '@dxos/app-framework';
-import { type ArtifactDefinition, type AssociatedArtifact, createSystemPrompt } from '@dxos/artifact';
-import { type BlueprintRegistry, Conversation } from '@dxos/assistant';
-import { FunctionType, type ServiceContainer } from '@dxos/functions';
-import { log } from '@dxos/log';
-import { useConfig } from '@dxos/react-client';
-import { Filter, fullyQualifiedId, type Queue, type Space, useQuery } from '@dxos/react-client/echo';
-import { isNonNullable } from '@dxos/util';
-
-import { ChatProcessor, type ChatProcessorOptions } from '../hooks';
-import { convertFunctionToTool, createToolsFromService } from '../tools';
-import { type Assistant, ServiceType } from '../types';
-
-type UseChatProcessorProps = {
-  /** @deprecated Why is this required? */
-  part?: 'deck' | 'dialog';
-  space?: Space;
-  chat?: Assistant.Chat;
-  // TODO(burdon): Reconcile all of below (overlapping concepts). Figure out how to inject vie effect layers.
-  serviceContainer: ServiceContainer;
-  blueprintRegistry?: BlueprintRegistry;
-  settings?: Assistant.Settings;
-  /** @deprecated */
-  instructions?: string;
-  /** @deprecated */
-  artifact?: AssociatedArtifact;
-  /** @deprecated */
-  noPluginArtifacts?: boolean;
-};
-
-/**
- * Configure and create ChatProcessor.
- */
-export const useChatProcessor = ({
-  part = 'deck',
-  space,
-  chat,
-  serviceContainer,
-  blueprintRegistry,
-  settings,
-  instructions,
-  artifact,
-  noPluginArtifacts,
-}: UseChatProcessorProps): ChatProcessor | undefined => {
-  const { dispatchPromise: dispatch } = useIntentDispatcher();
-  const globalTools = useCapabilities(Capabilities.Tools);
-
-  // TODO(burdon): Spec artifacts.
-  let artifacts: readonly ArtifactDefinition[] = useCapabilities(Capabilities.ArtifactDefinition);
-  if (noPluginArtifacts) {
-    artifacts = Stable.array;
-  }
-
-  // Services.
-  const services = useQuery(space, Filter.type(ServiceType));
-  const [serviceTools, setServiceTools] = useState<ExecutableTool[]>([]);
-  useEffect(() => {
-    log('creating service tools...');
-    queueMicrotask(async () => {
-      const tools = await Promise.all(services.map((service) => createToolsFromService(service)));
-      setServiceTools(tools.flat());
-    });
-  }, [services]);
-
-  // Tools and context.
-  const config = useConfig();
-  const functions = useQuery(space, Filter.type(FunctionType));
-  const chatId = useMemo(() => (chat ? fullyQualifiedId(chat) : undefined), [chat]);
-  const [tools, extensions] = useMemo(() => {
-    log('creating tools...');
-    const tools: ExecutableTool[] = [
-      ...globalTools.flat(),
-      ...serviceTools,
-      ...functions
-        .map((fn) => convertFunctionToTool(fn, config.values.runtime?.services?.edge?.url ?? '', space?.id))
-        .filter(isNonNullable),
-    ];
-    const extensions = { part, space, dispatch, pivotId: chatId };
-    return [tools, extensions];
-  }, [dispatch, globalTools, space, chatId, serviceTools, functions]);
-
-  // Prompt.
-  const systemPrompt = useMemo(
-    () =>
-      createSystemPrompt({
-        artifacts: artifacts.map((definition) => `${definition.name}\n${definition.instructions}`),
-        artifact,
-        instructions,
-      }),
-    [artifacts, artifact, instructions],
-  );
-
-  // TODO(burdon): Remove default (let backend decide if not specified).
-  const model: ChatProcessorOptions['model'] =
-    settings?.llmProvider === 'ollama'
-      ? ((settings?.ollamaModel ?? DEFAULT_OLLAMA_MODEL) as ChatProcessorOptions['model'])
-      : ((settings?.edgeModel ?? DEFAULT_EDGE_MODEL) as ChatProcessorOptions['model']);
-
-  const conversation = useMemo(() => {
-    if (!chat?.queue.target) {
-      return;
-    }
-
-    return new Conversation({
-      serviceContainer,
-      queue: chat.queue.target as Queue<any>,
-    });
-  }, [chat?.queue.target, serviceContainer]);
-
-  // Create processor.
-  // TODO(burdon): Updated on each query update above; should just update current processor.
-  const processor = useMemo(() => {
-    if (!conversation) {
-      return undefined;
-    }
-
-    log('creating processor...', { settings });
-    return new ChatProcessor(conversation, {
-      tools,
-      extensions,
-      blueprintRegistry,
-      artifacts,
-      systemPrompt,
-      model,
-    });
-  }, [conversation, tools, blueprintRegistry, artifacts, extensions, systemPrompt, model]);
-
-  return processor;
-};
-
-// TODO(dmaretskyi): Extract.
-export const Stable = Object.freeze({
-  array: [] as readonly never[],
-  object: {} as {},
-});

package/src/hooks/useServiceContainer.ts

@@ -1,42 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { useEffect, useMemo, useState } from 'react';
-
-import { ToolRegistry } from '@dxos/ai';
-import { Capabilities, useCapabilities, useCapability } from '@dxos/app-framework';
-import { AiService, DatabaseService, QueueService, ServiceContainer, ToolResolverService } from '@dxos/functions';
-import { type Space } from '@dxos/react-client/echo';
-
-import { AssistantCapabilities } from '../capabilities';
-
-interface UseServiceContainerProps {
-  space?: Space;
-}
-
-export const useServiceContainer = ({ space }: UseServiceContainerProps) => {
-  const aiClient = useCapability(AssistantCapabilities.AiClient);
-  const tools = useCapabilities(Capabilities.Tools).flat();
-
-  const [toolRegistry] = useState(() => new ToolRegistry([]));
-  useEffect(() => {
-    for (const tool of tools) {
-      if (!toolRegistry.has(tool)) {
-        toolRegistry.register(tool);
-      }
-    }
-  }, [toolRegistry, JSON.stringify(tools.map((tool) => tool.id))]);
-
-  return useMemo(
-    () =>
-      new ServiceContainer().setServices({
-        ai: AiService.make(aiClient.value),
-        database: space ? DatabaseService.make(space.db) : undefined,
-        queues: space ? QueueService.make(space.queues, undefined) : undefined,
-        // eventLogger: consoleLogger,
-        toolResolver: ToolResolverService.make(toolRegistry),
-      }),
-    [space, aiClient],
-  );
-};

package/src/hooks/useTextInputEvents.ts

@@ -1,49 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { useState, useCallback, type KeyboardEventHandler, type ChangeEventHandler } from 'react';
-
-/**
- * Hook that handles text input events.
- */
-// TODO(burdon): Factor out.
-export const useTextInputEvents = ({
-  onEnter,
-  onEscape,
-}: {
-  onEnter?: (text: string) => boolean | void;
-  onEscape?: () => void;
-}) => {
-  const [value, setValue] = useState('');
-
-  const onKeyDown = useCallback<KeyboardEventHandler<HTMLInputElement>>(
-    (ev) => {
-      switch (ev.key) {
-        case 'Escape': {
-          onEscape?.();
-          setValue('');
-          break;
-        }
-
-        case 'Enter': {
-          const value = (ev.target as HTMLInputElement).value.trim();
-          if (value.length > 0) {
-            const reset = onEnter?.(value);
-            if (reset) {
-              setValue('');
-            }
-          }
-          break;
-        }
-      }
-    },
-    [onEnter, onEscape],
-  );
-
-  const onChange = useCallback<ChangeEventHandler<HTMLInputElement>>((ev) => {
-    setValue(ev.target.value);
-  }, []);
-
-  return [{ value, onKeyDown, onChange }, setValue];
-};

package/src/types/schema.ts

@@ -1,86 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { Schema } from 'effect';
-
-import { SpaceSchema, Queue } from '@dxos/client/echo';
-import { Sequence } from '@dxos/conductor';
-import { Type } from '@dxos/echo';
-import { LabelAnnotation } from '@dxos/echo-schema';
-
-import { meta } from '../meta';
-
-export const LLM_PROVIDERS = ['edge', 'ollama', 'lmstudio'] as const;
-
-export namespace Assistant {
-  //
-  // Types
-  //
-
-  export const Chat = Schema.Struct({
-    id: Type.ObjectId,
-    name: Schema.optional(Schema.String),
-    queue: Type.Ref(Queue),
-  }).pipe(
-    Type.Obj({
-      typename: 'dxos.org/type/assistant/Chat',
-      version: '0.2.0',
-    }),
-    LabelAnnotation.set(['name']),
-  );
-
-  export interface Chat extends Schema.Schema.Type<typeof Chat> {}
-
-  export const CompanionTo = Schema.Struct({
-    id: Type.ObjectId,
-  }).pipe(
-    Type.Relation({
-      typename: 'dxos.org/relation/assistant/CompanionTo',
-      version: '0.1.0',
-      source: Chat,
-      target: Type.Expando,
-    }),
-  );
-
-  export interface CompanionTo extends Schema.Schema.Type<typeof CompanionTo> {}
-
-  //
-  // Settings
-  //
-
-  export const Settings = Schema.mutable(
-    Schema.Struct({
-      llmProvider: Schema.optional(Schema.Literal(...LLM_PROVIDERS)),
-      edgeModel: Schema.optional(Schema.String),
-      ollamaModel: Schema.optional(Schema.String),
-      lmstudioModel: Schema.optional(Schema.String),
-      customPrompts: Schema.optional(Schema.Boolean),
-    }),
-  );
-
-  export type Settings = Schema.Schema.Type<typeof Settings>;
-
-  //
-  // Actions
-  //
-
-  export class CreateChat extends Schema.TaggedClass<CreateChat>()(`${meta.id}/action}/create-chat`, {
-    input: Schema.Struct({
-      space: SpaceSchema,
-      name: Schema.optional(Schema.String),
-    }),
-    output: Schema.Struct({
-      object: Assistant.Chat,
-    }),
-  }) {}
-
-  export class CreateSequence extends Schema.TaggedClass<CreateSequence>()(`${meta.id}/action}/create-sequence`, {
-    input: Schema.Struct({
-      name: Schema.optional(Schema.String),
-    }),
-    output: Schema.Struct({
-      object: Sequence,
-    }),
-  }) {}
-}

package/src/types/template.ts

@@ -1,66 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { Schema } from 'effect';
-
-import { Type } from '@dxos/echo';
-import { LabelAnnotation } from '@dxos/echo-schema';
-
-// TODO(burdon): Change to Schema.Literal (and discriminated union).
-export enum TemplateInputType {
-  VALUE = 0,
-  PASS_THROUGH = 1,
-  RETRIEVER = 2,
-  FUNCTION = 3,
-  QUERY = 4,
-  RESOLVER = 5,
-  CONTEXT = 6,
-  SCHEMA = 7,
-}
-
-export const TemplateInputSchema = Schema.mutable(
-  Schema.Struct({
-    name: Schema.String,
-    type: Schema.optional(Schema.Enums(TemplateInputType)),
-    value: Schema.optional(Schema.String),
-  }),
-);
-
-export type TemplateInput = Schema.Schema.Type<typeof TemplateInputSchema>;
-
-export const TemplateKinds = ['always', 'schema-matching', 'automatically', 'manual'] as const;
-export type TemplateKind = (typeof TemplateKinds)[number];
-
-export const TemplateKindSchema = Schema.Union(
-  Schema.Struct({
-    include: Schema.Literal('always'),
-  }),
-  Schema.Struct({
-    include: Schema.Literal('schema-matching'),
-    typename: Schema.String,
-  }),
-  Schema.Struct({
-    include: Schema.Literal('automatically'),
-    description: Schema.String,
-  }),
-  Schema.Struct({
-    include: Schema.Literal('manual'),
-  }),
-);
-
-export type TemplateKindType = Schema.Schema.Type<typeof TemplateKindSchema>;
-export const TemplateType = Schema.Struct({
-  name: Schema.optional(Schema.String),
-  kind: Schema.mutable(TemplateKindSchema),
-  source: Schema.String,
-  inputs: Schema.optional(Schema.mutable(Schema.Array(TemplateInputSchema))),
-  command: Schema.optional(Schema.String),
-}).pipe(
-  LabelAnnotation.set(['name']),
-  Type.Obj({
-    typename: 'dxos.org/type/Template',
-    version: '0.1.0',
-  }),
-);
-export interface TemplateType extends Schema.Schema.Type<typeof TemplateType> {}