@dxos/plugin-assistant 0.8.4-main.72ec0f3 → 0.8.4-main.74a063c4e0

package/src/components/ChatThread/reducers.ts

@@ -1,151 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type ContentBlock, type Message } from '@dxos/types';
-
-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: Message.Message[];
-    current?: Message.Message;
-    toolBlock?: boolean;
-    assistantMessages?: Message.Message[];
-  },
-  Message.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: Message.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,
-  );
-};

package/src/components/PromptArticle.tsx

@@ -1,51 +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 { getSpace } from '@dxos/react-client/echo';
-import { Toolbar, useTranslation } from '@dxos/react-ui';
-import { useAttention } from '@dxos/react-ui-attention';
-import { StackItem } from '@dxos/react-ui-stack';
-
-import { meta } from '../meta';
-
-import { TemplateEditor } from './TemplateEditor';
-
-export type PromptArticleProps = SurfaceComponentProps<Prompt.Prompt>;
-
-export const PromptArticle = ({ subject }: PromptArticleProps) => {
-  const { t } = useTranslation(meta.id);
-  const space = getSpace(subject);
-  const { hasAttention } = useAttention(Obj.getDXN(subject).toString());
-
-  const inputData = useMemo(
-    () =>
-      subject && {
-        prompt: space?.db.ref(Obj.getDXN(subject)),
-        input: {},
-      },
-    [subject, space],
-  );
-
-  const handleRun = useComputeRuntimeCallback(space, () => invokeFunctionWithTracing(Agent.prompt, inputData), [
-    inputData,
-  ]);
-
-  return (
-    <StackItem.Content 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' />
-    </StackItem.Content>
-  );
-};
-
-export default PromptArticle;

package/src/events.ts

@@ -1,11 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { defineEvent } from '@dxos/app-framework';
-
-import { meta } from './meta';
-
-export namespace AssistantEvents {
-  export const SetupAiServiceProviders = defineEvent(`${meta.id}/event/setup-ai-service-providers`);
-}

package/src/functions/analysis.ts

@@ -1,22 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
-
-import { defineFunction } from '@dxos/functions';
-import { log } from '@dxos/log';
-
-export default defineFunction({
-  key: 'dxos.org/function/assistant-analysis',
-  name: 'Assistant analysis',
-  description: 'Enter analysis analysis mode.',
-  inputSchema: Schema.Struct({
-    enable: Schema.Boolean.annotations({ description: 'Enable or disable analysis mode.' }),
-  }),
-  outputSchema: Schema.Void,
-  handler: Effect.fn(function* ({ context, data: { enable } }) {
-    log.info('analysis mode', { enable });
-  }),
-});

package/src/functions/index.ts

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

package/src/functions/object-list.ts

@@ -1,48 +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 { DatabaseService, 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* DatabaseService.runSchemaQuery({ typename });
-    const filter = schema ? Filter.type(schema) : Filter.typename(typename);
-
-    const { objects } = yield* DatabaseService.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,45 +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 { Obj } from '@dxos/echo';
-import { DatabaseService, 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({
-    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.Obj.Any | Type.Relation.Any
-    object: Schema.Any,
-  }),
-  handler: Effect.fn(function* ({ data: { id, typename } }) {
-    const object = yield* DatabaseService.resolve(ArtifactId.toDXN(id));
-    return yield* Function.pipe(
-      Option.fromNullable(object),
-      Option.flatMap((object) => (Obj.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/hooks/useItemTypes.ts

@@ -1,37 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Option from 'effect/Option';
-import { useEffect, useState } from 'react';
-
-import { type Type } from '@dxos/echo';
-import { type Space } from '@dxos/react-client/echo';
-import { ItemAnnotation } from '@dxos/schema';
-
-// TODO(burdon): Pass in filter.
-export const useItemTypes = (space?: Space): Type.Obj.Any[] => {
-  const [types, setTypes] = useState<Type.Obj.Any[]>([]);
-  useEffect(() => {
-    if (!space) {
-      return;
-    }
-
-    return space.db.schemaRegistry.query().subscribe(
-      (query) => {
-        const types = Array.from(
-          new Set(
-            [...space.db.graph.schemaRegistry.schemas, ...query.results].filter((type) =>
-              Option.isSome(ItemAnnotation.get(type)),
-            ),
-          ),
-        );
-
-        setTypes(types);
-      },
-      { fire: true },
-    );
-  }, [space]);
-
-  return types;
-};

package/src/processor/update-name.ts

@@ -1,56 +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 { 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) {
-        chat.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,79 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import * as Schema from 'effect/Schema';
-
-import { Blueprint, Prompt } from '@dxos/blueprints';
-import { EchoObjectSchema, SpaceSchema } from '@dxos/client/echo';
-import { Sequence } from '@dxos/conductor';
-import { Collection } from '@dxos/schema';
-
-import { meta } from '../meta';
-
-import { Chat } from './Assistant';
-
-// TODO(burdon): Name?
-export class onCreateSpace extends Schema.TaggedClass<onCreateSpace>()(`${meta.id}/on-space-created`, {
-  input: Schema.Struct({
-    space: SpaceSchema,
-    rootCollection: Collection.Collection,
-  }),
-  output: Schema.Void,
-}) {}
-
-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: Chat,
-  }),
-}) {}
-
-export class UpdateChatName extends Schema.TaggedClass<UpdateChatName>()(`${meta.id}/action/update-name`, {
-  input: Schema.Struct({
-    chat: Chat,
-  }),
-  output: Schema.Void,
-}) {}
-
-export const BlueprintForm = Schema.Struct({
-  key: Schema.String,
-  name: Schema.String,
-  description: Schema.optional(Schema.String),
-});
-
-export class CreateBlueprint extends Schema.TaggedClass<CreateBlueprint>()(`${meta.id}/action/create-blueprint`, {
-  input: BlueprintForm,
-  output: Schema.Struct({
-    object: Blueprint.Blueprint,
-  }),
-}) {}
-
-export class CreatePrompt extends Schema.TaggedClass<CreatePrompt>()(`${meta.id}/action/create-prompt`, {
-  input: Schema.Struct({
-    name: Schema.optional(Schema.String),
-  }),
-  output: Schema.Struct({
-    object: Prompt.Prompt,
-  }),
-}) {}
-
-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,
-  }),
-}) {}
-
-export class SetCurrentChat extends Schema.TaggedClass<SetCurrentChat>()(`${meta.id}/action/set-current-chat`, {
-  input: Schema.Struct({
-    companionTo: EchoObjectSchema,
-    chat: Chat.pipe(Schema.optional),
-  }),
-  output: Schema.Void,
-}) {}