@dxos/plugin-assistant 0.8.4-main.9735255 → 0.8.4-main.9be5663bfe

package/src/capabilities/react-surface/react-surface.tsx

@@ -1,95 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import React from 'react';
-
-import { Capability, Common } from '@dxos/app-framework';
-import { useSettingsState } from '@dxos/app-framework/react';
-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 { Layout } from '@dxos/react-ui-mosaic';
-
-import {
-  AssistantSettings,
-  BlueprintArticle,
-  ChatCompanion,
-  ChatContainer,
-  ChatDialog,
-  PromptArticle,
-} from '../../components';
-import { ASSISTANT_DIALOG, meta } from '../../meta';
-import { Assistant } from '../../types';
-
-export default Capability.makeModule(() =>
-  Effect.succeed(
-    Capability.contributes(Common.Capability.ReactSurface, [
-      Common.createSurface({
-        id: `${meta.id}/plugin-settings`,
-        role: 'article',
-        filter: (data): data is { subject: Common.Capability.Settings } =>
-          Common.Capability.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} />;
-        },
-      }),
-      Common.createSurface({
-        id: `${meta.id}/chat`,
-        role: 'article',
-        filter: (data): data is { subject: Assistant.Chat; variant: undefined } =>
-          Obj.instanceOf(Assistant.Chat, data.subject) && data.variant !== 'assistant-chat',
-        component: ({ data, role, ref }) => <ChatContainer role={role} subject={data.subject} ref={ref} />,
-      }),
-      // TODO(wittjosiah): This is flashing when chat changes.
-      Common.createSurface({
-        id: `${meta.id}/companion-chat`,
-        role: 'article',
-        filter: (data): data is { companionTo: Obj.Unknown; subject: Assistant.Chat | 'assistant-chat' } =>
-          Obj.isObject(data.companionTo) &&
-          (Obj.instanceOf(Assistant.Chat, data.subject) || data.subject === 'assistant-chat'),
-        component: ({ data, role, ref }) => <ChatCompanion role={role} data={data} ref={ref} />,
-      }),
-      Common.createSurface({
-        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 (
-            <Layout.Main role={role}>
-              <InvocationTraceContainer db={space?.db} queueDxn={queueDxn} target={target} detailAxis='block' />
-            </Layout.Main>
-          );
-        },
-      }),
-      Common.createSurface({
-        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} />,
-      }),
-      Common.createSurface({
-        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} />,
-      }),
-      Common.createSurface({
-        id: ASSISTANT_DIALOG,
-        role: 'dialog',
-        filter: (data): data is { props: { chat: Assistant.Chat } } => data.component === ASSISTANT_DIALOG,
-        component: ({ data }) => <ChatDialog {...data.props} />,
-      }),
-    ]),
-  ),
-);

package/src/capabilities/repair/index.ts

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

package/src/capabilities/repair/repair.ts

@@ -1,63 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-
-import { Capability } from '@dxos/app-framework';
-import { Blueprint, Prompt } from '@dxos/blueprints';
-import { Obj, Ref, Type } from '@dxos/echo';
-import { SpaceCapabilities } from '@dxos/plugin-space';
-import { type Space } from '@dxos/react-client/echo';
-import { Collection } from '@dxos/schema';
-
-import { Assistant } from '../../types';
-
-export default Capability.makeModule(() =>
-  Effect.succeed(
-    Capability.contributes(SpaceCapabilities.Repair, async ({ space }) => {
-      await ensureSystemCollections(space);
-    }),
-  ),
-);
-
-/**
- * Ensure the root collection has system collections for AI Chats, Blueprints, and Prompts.
- */
-const ensureSystemCollections = async (space: Space) => {
-  const rootCollection: Collection.Collection = await space.properties[Collection.Collection.typename]?.load();
-  if (!rootCollection) {
-    return;
-  }
-
-  const objects = await Promise.all(rootCollection.objects.map((ref) => ref.load()));
-  const chats = objects.find(
-    (object) => Obj.instanceOf(Collection.Managed, object) && object.key === Assistant.Chat.typename,
-  );
-  if (!chats) {
-    const chatsCollectionRef = Ref.make(Collection.makeManaged({ key: Assistant.Chat.typename }));
-    Obj.change(rootCollection, (c) => {
-      c.objects.push(chatsCollectionRef);
-    });
-  }
-
-  const blueprints = objects.find(
-    (object) => Obj.instanceOf(Collection.Managed, object) && object.key === Blueprint.Blueprint.typename,
-  );
-  if (!blueprints) {
-    const blueprintsCollectionRef = Ref.make(Collection.makeManaged({ key: Blueprint.Blueprint.typename }));
-    Obj.change(rootCollection, (c) => {
-      c.objects.push(blueprintsCollectionRef);
-    });
-  }
-
-  const prompts = objects.find(
-    (object) => Obj.instanceOf(Collection.Managed, object) && object.key === Type.getTypename(Prompt.Prompt),
-  );
-  if (!prompts) {
-    const promptsCollectionRef = Ref.make(Collection.makeManaged({ key: Type.getTypename(Prompt.Prompt) }));
-    Obj.change(rootCollection, (c) => {
-      c.objects.push(promptsCollectionRef);
-    });
-  }
-};

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/capabilities/toolkit/toolkit.ts

@@ -1,26 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Layer from 'effect/Layer';
-
-import { Capability, Common } from '@dxos/app-framework';
-import { GenericToolkit } from '@dxos/assistant';
-import { AssistantToolkit, SystemToolkit, WebSearchToolkit } from '@dxos/assistant-toolkit';
-
-export default Capability.makeModule(() =>
-  Effect.succeed([
-    Capability.contributes(
-      Common.Capability.Toolkit,
-      GenericToolkit.make(AssistantToolkit.AssistantToolkit, AssistantToolkit.layer()),
-    ),
-
-    // TODO(burdon): How to manage dependencies across blueprints.
-    Capability.contributes(
-      Common.Capability.Toolkit,
-      GenericToolkit.make(SystemToolkit.SystemToolkit, SystemToolkit.layer()),
-    ),
-    Capability.contributes(Common.Capability.Toolkit, GenericToolkit.make(WebSearchToolkit, Layer.empty)),
-  ]),
-);

package/src/components/BlueprintArticle.tsx

@@ -1,29 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React from 'react';
-
-import { type SurfaceComponentProps } from '@dxos/app-framework/react';
-import { type Blueprint } from '@dxos/blueprints';
-import { Obj } from '@dxos/echo';
-import { Toolbar } from '@dxos/react-ui';
-import { useAttention } from '@dxos/react-ui-attention';
-import { Layout } from '@dxos/react-ui-mosaic';
-
-import { TemplateEditor } from './TemplateEditor';
-
-export type BlueprintArticleProps = SurfaceComponentProps<Blueprint.Blueprint>;
-
-export const BlueprintArticle = ({ role, subject }: BlueprintArticleProps) => {
-  const { hasAttention } = useAttention(Obj.getDXN(subject).toString());
-
-  return (
-    <Layout.Main role={role} toolbar>
-      <Toolbar.Root disabled={!hasAttention} />
-      <TemplateEditor id={subject.id} template={subject.instructions} classNames='container-max-width' />
-    </Layout.Main>
-  );
-};
-
-export default BlueprintArticle;

package/src/components/ChatContainer.tsx

@@ -1,61 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { forwardRef } from 'react';
-
-import { type SurfaceComponentProps, useAtomCapability } from '@dxos/app-framework/react';
-import { type Space, getSpace } from '@dxos/client/echo';
-import { type Obj } from '@dxos/echo';
-import { Layout } from '@dxos/react-ui-mosaic';
-
-import { useBlueprintRegistry, useChatProcessor, useChatServices, useOnline, usePresets } from '../hooks';
-import { type Assistant, AssistantCapabilities } from '../types';
-
-import { Chat, type ChatRootProps } from './Chat';
-
-export type ChatContainerProps = SurfaceComponentProps<
-  Assistant.Chat | undefined,
-  {
-    space?: Space;
-    companionTo?: Obj.Unknown;
-  } & Pick<ChatRootProps, 'onEvent'>
->;
-
-export const ChatContainer = forwardRef<HTMLDivElement, ChatContainerProps>((props, forwardedRef) => {
-  const { role, subject: chat, space: spaceProp, companionTo, onEvent } = props;
-  const space = spaceProp ?? getSpace(chat);
-  const settings = useAtomCapability(AssistantCapabilities.Settings);
-  const services = useChatServices({ id: space?.id, chat });
-  const [online, setOnline] = useOnline();
-  const { preset, ...chatProps } = usePresets(online);
-  const blueprintRegistry = useBlueprintRegistry();
-  const processor = useChatProcessor({
-    space,
-    chat,
-    preset,
-    services,
-    blueprintRegistry,
-    settings,
-  });
-
-  if (!processor) {
-    return null;
-  }
-
-  return (
-    <Layout.Main toolbar role={role} ref={forwardedRef}>
-      <Chat.Root db={space?.db} chat={chat} processor={processor} onEvent={onEvent}>
-        <Chat.Toolbar companionTo={companionTo} />
-        <Chat.Viewport classNames='container-max-width'>
-          <Chat.Thread />
-          <div role='none' className='p-4'>
-            <Chat.Prompt {...chatProps} outline preset={preset?.id} online={online} onOnlineChange={setOnline} />
-          </div>
-        </Chat.Viewport>
-      </Chat.Root>
-    </Layout.Main>
-  );
-});
-
-export default ChatContainer;

package/src/components/PromptArticle.tsx

@@ -1,50 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { useMemo } from 'react';
-
-import { type SurfaceComponentProps } from '@dxos/app-framework/react';
-import { Agent } from '@dxos/assistant-toolkit';
-import { type Prompt } from '@dxos/blueprints';
-import { Obj } from '@dxos/echo';
-import { invokeFunctionWithTracing, useComputeRuntimeCallback } from '@dxos/plugin-automation';
-import { Toolbar, useTranslation } from '@dxos/react-ui';
-import { useAttention } from '@dxos/react-ui-attention';
-import { Layout } from '@dxos/react-ui-mosaic';
-
-import { meta } from '../meta';
-
-import { TemplateEditor } from './TemplateEditor';
-
-export type PromptArticleProps = SurfaceComponentProps<Prompt.Prompt>;
-
-export const PromptArticle = ({ role, subject }: PromptArticleProps) => {
-  const { t } = useTranslation(meta.id);
-  const db = Obj.getDatabase(subject);
-  const { hasAttention } = useAttention(Obj.getDXN(subject).toString());
-
-  const inputData = useMemo(
-    () =>
-      subject && {
-        prompt: db?.makeRef(Obj.getDXN(subject)),
-        input: {},
-      },
-    [subject, db],
-  );
-
-  const handleRun = useComputeRuntimeCallback(db?.spaceId, () => invokeFunctionWithTracing(Agent.prompt, inputData), [
-    inputData,
-  ]);
-
-  return (
-    <Layout.Main role={role} toolbar>
-      <Toolbar.Root disabled={!hasAttention} onClick={handleRun}>
-        <Toolbar.IconButton iconOnly icon='ph--play--regular' label={t('run prompt label')} onClick={handleRun} />
-      </Toolbar.Root>
-      <TemplateEditor id={subject.id} template={subject.instructions} classNames='container-max-width' />
-    </Layout.Main>
-  );
-};
-
-export default PromptArticle;

package/src/functions/index.ts

@@ -1,7 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export { default as list } from './object-list';
-export { default as load } from './object-load';
-export { default as search } from './object-search';

package/src/functions/object-list.ts

@@ -1,49 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
-
-import { Filter, Obj, Query } from '@dxos/echo';
-import { Database } from '@dxos/echo';
-import { defineFunction } from '@dxos/functions';
-import { trim } from '@dxos/util';
-
-// TODO(burdon): Move to toolkit (i.e., tool not function).
-export default defineFunction({
-  key: 'dxos.org/function/assistant/object-list',
-  name: 'List objects',
-  description:
-    // TODO(wittjosiah): Find a better way to prompt for looking up typenames before querying with them.
-    trim`
-      Lists the objects of the given type.
-      Check the get-schemas tool for available types before calling this function.
-    `,
-  inputSchema: Schema.Struct({
-    typename: Schema.String.annotations({
-      description: 'The typename of the objects to list.',
-    }),
-  }),
-  outputSchema: Schema.Struct({
-    results: Schema.Array(
-      Schema.Struct({
-        dxn: Schema.String.annotations({ description: 'The DXN of the object.' }),
-        label: Schema.optional(Schema.String.annotations({ description: 'The label of the object.' })),
-      }),
-    ),
-  }),
-  handler: Effect.fn(function* ({ data: { typename } }) {
-    // TODO(wittjosiah): Typename query is not working for dynamic schemas.
-    const [schema] = yield* Database.Service.runSchemaQuery({ typename });
-    const filter = schema ? Filter.type(schema) : Filter.typename(typename);
-
-    const objects = yield* Database.Service.runQuery(Query.select(filter));
-    const results = objects.map((object) => ({
-      dxn: Obj.getDXN(object).toString(),
-      label: Obj.getLabel(object),
-    }));
-
-    return { results };
-  }),
-});

package/src/functions/object-load.ts

@@ -1,47 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Function from 'effect/Function';
-import * as Option from 'effect/Option';
-import * as Schema from 'effect/Schema';
-
-import { ArtifactId } from '@dxos/assistant';
-import { Entity } from '@dxos/echo';
-import { Database } from '@dxos/echo';
-import { defineFunction } from '@dxos/functions';
-
-// TODO(burdon): Move to toolkit (i.e., tool not function).
-export default defineFunction({
-  key: 'dxos.org/function/assistant/object-load',
-  name: 'Load object',
-  description: 'Loads the object.',
-  inputSchema: Schema.Struct({
-    // TODO(dmaretskyi): Use Ref.
-    id: ArtifactId.annotations({
-      description: 'The ID of the object.',
-    }),
-    typename: Schema.String.annotations({
-      description: 'The typename of the object.',
-    }),
-  }),
-  outputSchema: Schema.Struct({
-    // TODO(wittjosiah): Type.Entity.Any
-    object: Schema.Any,
-  }),
-  handler: Effect.fn(function* ({ data: { id, typename } }) {
-    const object = yield* Database.Service.resolve(ArtifactId.toDXN(id));
-    return yield* Function.pipe(
-      Option.fromNullable(object),
-      Option.flatMap((object) => (Entity.getTypename(object) === typename ? Option.some(object) : Option.none())),
-      Option.match({
-        onNone: () => Effect.fail('Object not found.'),
-        onSome: (object) => {
-          // log.info('object', { object });
-          return Effect.succeed({ object });
-        },
-      }),
-    );
-  }),
-});

package/src/functions/object-search.ts

@@ -1,46 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
-
-import { Filter, Obj, Query } from '@dxos/echo';
-import { Database } from '@dxos/echo';
-import { defineFunction } from '@dxos/functions';
-import { trim } from '@dxos/util';
-
-// TODO(burdon): Move to toolkit (i.e., tool not function).
-export default defineFunction({
-  key: 'dxos.org/function/assistant/object-search',
-  name: 'Search objects',
-  description:
-    // TODO(wittjosiah): Find a better way to prompt for looking up typenames before querying with them.
-    trim`
-      Searches the objects using full-text search.
-    `,
-  inputSchema: Schema.Struct({
-    query: Schema.String.annotations({
-      description: 'The query to search for.',
-    }),
-  }),
-  outputSchema: Schema.Struct({
-    results: Schema.Array(
-      Schema.Struct({
-        dxn: Schema.String.annotations({ description: 'The DXN of the object.' }),
-        label: Schema.optional(Schema.String.annotations({ description: 'The label of the object.' })),
-      }),
-    ),
-  }),
-  handler: Effect.fn(function* ({ data: { query } }) {
-    const objects = yield* Database.Service.runQuery(
-      Query.select(Filter.text(query, { type: 'full-text' })).options({ allQueuesFromSpaces: true }),
-    );
-    const results = objects.map((object) => ({
-      dxn: Obj.getDXN(object).toString(),
-      label: Obj.getLabel(object),
-    }));
-
-    return { results };
-  }),
-});

package/src/processor/update-name.ts

@@ -1,59 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Exit from 'effect/Exit';
-import * as Fiber from 'effect/Fiber';
-import * as Runtime from 'effect/Runtime';
-
-import { AiService, DEFAULT_EDGE_MODEL, type ModelName } from '@dxos/ai';
-import { type AiConversation, AiSession } from '@dxos/assistant';
-import { Obj } from '@dxos/echo';
-import { throwCause } from '@dxos/effect';
-import { trim } from '@dxos/util';
-
-import { type Assistant } from '../types';
-
-import { type AiChatServices } from './processor';
-
-/**
- * Update the current chat's name.
- */
-// TODO(burdon): Convert this into a plugin tool.
-export const updateName = async (
-  runtime: Runtime.Runtime<AiChatServices>,
-  conversation: AiConversation,
-  chat: Assistant.Chat,
-  model: ModelName = DEFAULT_EDGE_MODEL,
-): Promise<void> => {
-  const history = await conversation.getHistory();
-  const system = trim`
-    It is extremely important that you respond only with the title and nothing else.
-    If you cannot do this effectively respond with "New Chat".
-  `;
-  const prompt = 'Suggest a name for this chat';
-
-  const fiber = Effect.gen(this, function* () {
-    const session = new AiSession();
-    return yield* session.run({ system, prompt, history });
-  }).pipe(
-    Effect.provide(AiService.model(model)),
-    Effect.tap((messages) => {
-      // TODO(burdon): Parse response (should update via tool).
-      const message = messages.find((message) => message.sender.role === 'assistant');
-      const title = message?.blocks.find((block) => block._tag === 'text')?.text;
-      if (title) {
-        Obj.change(chat, (c) => {
-          c.name = title;
-        });
-      }
-    }),
-    Runtime.runFork(runtime), // Run in the background.
-  );
-
-  const response = await fiber.pipe(Fiber.join, Effect.runPromiseExit);
-  if (!Exit.isSuccess(response)) {
-    throwCause(response.cause);
-  }
-};

package/src/types/AssistantAction.ts

@@ -1,75 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import * as Schema from 'effect/Schema';
-
-import { Capability } from '@dxos/app-framework';
-import { SpaceSchema } from '@dxos/client/echo';
-import { Database, Type } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
-import { Collection } from '@dxos/schema';
-
-import { meta } from '../meta';
-
-import { Chat } from './Assistant';
-
-// Operations
-const ASSISTANT_OPERATION = `${meta.id}/operation`;
-
-export namespace AssistantOperation {
-  export const OnCreateSpace = Operation.make({
-    meta: { key: `${ASSISTANT_OPERATION}/on-create-space`, name: 'On Create Space' },
-    services: [Capability.Service],
-    schema: {
-      input: Schema.Struct({
-        space: SpaceSchema,
-        rootCollection: Collection.Collection,
-      }),
-      output: Schema.Void,
-    },
-  });
-
-  export const CreateChat = Operation.make({
-    meta: { key: `${ASSISTANT_OPERATION}/create-chat`, name: 'Create Chat' },
-    services: [Capability.Service],
-    schema: {
-      input: Schema.Struct({
-        db: Database.Database,
-        name: Schema.optional(Schema.String),
-      }),
-      output: Schema.Struct({
-        object: Chat,
-      }),
-    },
-  });
-
-  export const UpdateChatName = Operation.make({
-    meta: { key: `${ASSISTANT_OPERATION}/update-chat-name`, name: 'Update Chat Name' },
-    services: [Capability.Service],
-    schema: {
-      input: Schema.Struct({
-        chat: Chat,
-      }),
-      output: Schema.Void,
-    },
-  });
-
-  export const SetCurrentChat = Operation.make({
-    meta: { key: `${ASSISTANT_OPERATION}/set-current-chat`, name: 'Set Current Chat' },
-    services: [Capability.Service],
-    schema: {
-      input: Schema.Struct({
-        companionTo: Type.Obj,
-        chat: Chat.pipe(Schema.optional),
-      }),
-      output: Schema.Void,
-    },
-  });
-
-  export const BlueprintForm = Schema.Struct({
-    key: Schema.String,
-    name: Schema.String,
-    description: Schema.optional(Schema.String),
-  });
-}