@dxos/plugin-assistant 0.8.4-main.f5c0578 → 0.8.4-main.fcfe5033a5

package/src/components/Timeline/Timeline.tsx

@@ -1,202 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { Fragment, useMemo } from 'react';
-
-import { LogLevel } from '@dxos/log';
-import { Icon, type ThemedClassName } from '@dxos/react-ui';
-import { mx } from '@dxos/react-ui-theme';
-import { trim } from '@dxos/util';
-
-const lineHeight = 24;
-const columnWidth = 24;
-const nodeRadius = 5;
-const lineStyle = 'stroke-1';
-
-const colors = [
-  { stroke: 'stroke-orange-500', hover: 'group-hover:fill-orange-500' },
-  { stroke: 'stroke-sky-500', hover: 'group-hover:fill-sky-500' },
-  { stroke: 'stroke-green-500', hover: 'group-hover:fill-green-500' },
-  { stroke: 'stroke-fuchsia-500', hover: 'group-hover:fill-fuchsia-500' },
-  { stroke: 'stroke-cyan-500', hover: 'group-hover:fill-cyan-500' },
-  { stroke: 'stroke-emerald-500', hover: 'group-hover:fill-emerald-500' },
-  { stroke: 'stroke-violet-500', hover: 'group-hover:fill-violet-500' },
-  { stroke: 'stroke-teal-500', hover: 'group-hover:fill-teal-500' },
-];
-
-export enum IconType {
-  // General status.
-  WARN = 'ph--warning-circle--regular',
-  CHECK = 'ph--check-circle--regular',
-  ROCKET = 'ph--rocket--regular',
-  X = 'ph--x-circle--regular',
-  FLAG = 'ph--flag--regular',
-  TIMER = 'ph--timer--regular',
-
-  // Interactions.
-  USER = 'ph--user--regular',
-  USER_INTERACTION = 'ph--user-sound--regular',
-  AGENT = 'ph--robot--regular',
-  THINK = 'ph--brain--regular',
-  LINK = 'ph--link--regular',
-  TOOL = 'ph--wrench--regular',
-}
-
-const levelColors: Record<LogLevel, string> = {
-  [LogLevel.TRACE]: 'text-gray-500',
-  [LogLevel.DEBUG]: 'text-gray-500',
-  [LogLevel.VERBOSE]: 'text-gray-500',
-  [LogLevel.INFO]: 'text-green-500',
-  [LogLevel.WARN]: 'text-orange-500',
-  [LogLevel.ERROR]: 'text-red-500',
-};
-
-export type Commit = {
-  id: string;
-  parent?: string;
-  branch: string;
-  icon?: string;
-  level?: LogLevel;
-  message: string;
-  timestamp?: Date;
-  tags?: string[];
-};
-
-export type Branch = {
-  name: string;
-};
-
-export type Span = {
-  start: number;
-  end: number;
-  parent?: number;
-};
-
-export type TimelineProps = ThemedClassName<{
-  branches: Branch[];
-  commits: Commit[];
-  showIcon?: boolean;
-}>;
-
-// TODO(burdon): Reuse in toolCall messages.
-// TODO(burdon): Key up/down; selected.
-export const Timeline = ({ classNames, branches, commits, showIcon = true }: TimelineProps) => {
-  const spans = useMemo(() => {
-    const spans = new Map<string, Span>();
-    branches.forEach((branch) => {
-      spans.set(branch.name, { start: -1, end: -1 });
-    });
-
-    commits.forEach((commit, index) => {
-      const span = spans.get(commit.branch);
-      if (span) {
-        if (span.start === -1) {
-          const parentIndex = commit.parent ? commits.findIndex((c) => c.id === commit.parent) : 0;
-          span.start = parentIndex;
-          span.parent = commit.parent
-            ? branches.findIndex((branch) => branch.name === commits[parentIndex].branch)
-            : undefined;
-        }
-        span.end = index;
-      }
-    });
-
-    return spans;
-  }, [commits, branches]);
-
-  return (
-    <div className={mx('flex flex-col is-full', classNames)}>
-      {commits.map((commit, index) => {
-        return (
-          <div
-            key={commit.id}
-            className='group flex shrink-0 items-center gap-2 overflow-hidden hover:bg-hoverSurface'
-            style={{ height: `${lineHeight}px` }}
-          >
-            <svg width={branches.length * columnWidth} height={lineHeight} className='shrink-0'>
-              {[...branches].reverse().map((branch, _j) => {
-                const j = branches.length - 1 - _j;
-                const span = spans.get(branch.name);
-                const color = colors[j % colors.length];
-                if (!span) {
-                  return null;
-                }
-
-                return (
-                  <Fragment key={j}>
-                    {/* Upper */}
-                    {index !== 0 && span.start !== -1 && span.start < index && span.end >= index && (
-                      <line
-                        x1={j * columnWidth + columnWidth / 2}
-                        y1={0}
-                        x2={j * columnWidth + columnWidth / 2}
-                        y2={lineHeight / 2}
-                        className={mx(lineStyle, color.stroke)}
-                      />
-                    )}
-                    {/* Lower */}
-                    {span.end !== -1 && (span.start < index || span.start === 0) && index < span.end && (
-                      <line
-                        x1={j * columnWidth + columnWidth / 2}
-                        y1={lineHeight / 2}
-                        x2={j * columnWidth + columnWidth / 2}
-                        y2={lineHeight}
-                        className={mx(lineStyle, color.stroke)}
-                      />
-                    )}
-                    {/* Arc to parent */}
-                    {span.start === index && span.parent !== undefined && span.parent !== -1 && (
-                      <path
-                        d={trim`
-                          M ${0.5 + span.parent * columnWidth + columnWidth / 2} ${lineHeight / 2} 
-                          L ${j * columnWidth + columnWidth / 4} ${lineHeight / 2} 
-                          A ${lineHeight / 4} ${lineHeight / 4} 0 0 1 ${j * columnWidth + columnWidth / 2} ${(lineHeight * 3) / 4} 
-                          L ${j * columnWidth + columnWidth / 2} ${lineHeight}
-                        `}
-                        className={mx(lineStyle, color.stroke)}
-                        fill='none'
-                      />
-                    )}
-                    {branch.name === commit.branch && (
-                      <circle
-                        cx={j * columnWidth + columnWidth / 2}
-                        cy={lineHeight / 2}
-                        r={nodeRadius}
-                        className={mx(lineStyle, color.stroke, color.hover)}
-                      />
-                    )}
-                  </Fragment>
-                );
-              })}
-              {branches.map((branch, j) => {
-                const color = colors[j % colors.length];
-                return (
-                  branch.name === commit.branch && (
-                    <circle
-                      key={j}
-                      cx={j * columnWidth + columnWidth / 2}
-                      cy={lineHeight / 2}
-                      r={nodeRadius}
-                      className={mx(lineStyle, color.stroke, color.hover)}
-                    />
-                  )
-                );
-              })}
-            </svg>
-            {showIcon && (
-              <div className='flex shrink-0 w-6 justify-center'>
-                {commit.icon && (
-                  <Icon icon={commit.icon} classNames={mx(commit.level && levelColors[commit.level])} size={4} />
-                )}
-              </div>
-            )}
-            <div className='text-sm truncate cursor-pointer text-subdued group-hover:text-baseText'>
-              {commit.message}
-            </div>
-          </div>
-        );
-      })}
-    </div>
-  );
-};

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/execution-graph/execution-graph.ts

@@ -1,184 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { AgentStatus } from '@dxos/ai';
-import { Obj, type Ref } from '@dxos/echo';
-import { MESSAGE_PROPERTY_TOOL_CALL_ID } from '@dxos/functions';
-import type { ObjectId } from '@dxos/keys';
-import { DataType } from '@dxos/schema';
-
-import type { Branch, Commit } from '../components';
-
-// TODO(dmaretskyi): Deduplicate. Importing from components causes an error.
-enum IconType {
-  // General status.
-  WARN = 'ph--warning-circle--regular',
-  CHECK = 'ph--check-circle--regular',
-  ROCKET = 'ph--rocket--regular',
-  X = 'ph--x-circle--regular',
-  FLAG = 'ph--flag--regular',
-  TIMER = 'ph--timer--regular',
-
-  // Interactions.
-  USER = 'ph--user--regular',
-  USER_INTERACTION = 'ph--user-sound--regular',
-  AGENT = 'ph--robot--regular',
-  THINK = 'ph--brain--regular',
-  LINK = 'ph--link--regular',
-  TOOL = 'ph--wrench--regular',
-}
-
-/**
- * Models the execution graph based on a stream of events.
- */
-export class ExecutionGraph {
-  // TODO(dmaretskyi): Evolve the internal state to support chats, functions, circuit workflows, etc.
-  private _commits: Commit[] = [];
-  private _branchNames = new Set<string>();
-
-  /**
-   * Adds events to the graph.
-   */
-  addEvents(events: Obj.Any[]) {
-    for (const event of events) {
-      if (Obj.instanceOf(DataType.Message, event)) {
-        const messageCommits = chatMessageToCommit(event);
-        this._commits.push(...messageCommits);
-        messageCommits.map((c) => c.branch).forEach((branch) => this._branchNames.add(branch));
-      } else if (Obj.instanceOf(AgentStatus, event)) {
-        const branch = getBranchName({ parentMessage: event.parentMessage, toolCallId: event.toolCallId });
-        this._branchNames.add(branch);
-        this._commits.push({
-          id: event.id,
-          branch,
-          message: event.message,
-          icon: IconType.FLAG,
-          parent:
-            event.parentMessage && event.toolCallId ? getToolCallId(event.parentMessage, event.toolCallId) : undefined,
-        });
-      }
-    }
-  }
-
-  /**
-   * Returns the current state of the graph.
-   */
-  getGraph(): { commits: Commit[]; branches: Branch[] } {
-    return {
-      commits: this._commits,
-      branches: Array.from(this._branchNames).map((name) => ({ name })),
-    };
-  }
-}
-
-const getToolCallId = (messageId: ObjectId, toolCallId: string) => `${messageId}_toolCall_${toolCallId}`;
-const getToolResultId = (messageId: ObjectId, toolCallId: string) => `${messageId}_toolResult_${toolCallId}`;
-const getGenericBlockId = (messageId: ObjectId, idx: number) => `${messageId}_block_${idx}`;
-
-const getBranchName = (opts: { parentMessage?: ObjectId; toolCallId?: string }) => {
-  if (opts.parentMessage && opts.toolCallId) {
-    return `${opts.parentMessage}_${opts.toolCallId}`;
-  } else if (opts.parentMessage) {
-    return opts.parentMessage;
-  } else {
-    return 'main';
-  }
-};
-
-const getMessageBranch = (message: DataType.Message) => {
-  return getBranchName({
-    parentMessage: message.parentMessage,
-    toolCallId: message.properties?.[MESSAGE_PROPERTY_TOOL_CALL_ID],
-  });
-};
-
-const getParentId = (message: DataType.Message) => {
-  if (message.parentMessage && message.properties?.[MESSAGE_PROPERTY_TOOL_CALL_ID]) {
-    return getToolCallId(message.parentMessage, message.properties[MESSAGE_PROPERTY_TOOL_CALL_ID]);
-  } else {
-    return undefined;
-  }
-};
-
-const chatMessageToCommit = (message: DataType.Message): Commit[] => {
-  return message.blocks.map((block, idx) => {
-    const branch = getMessageBranch(message);
-    const parent = getParentId(message);
-    switch (block._tag) {
-      case 'toolCall':
-        return {
-          id: getToolCallId(message.id, block.toolCallId),
-          branch,
-          parent,
-          icon: IconType.TOOL,
-          message: block.name,
-        };
-      case 'toolResult':
-        return {
-          id: getToolResultId(message.id, block.toolCallId),
-          branch,
-          parent,
-          icon: block.error ? IconType.X : IconType.CHECK,
-          message: block.error ? block.error : block.name,
-        };
-      case 'status':
-        return {
-          id: getGenericBlockId(message.id, idx),
-          branch,
-          parent,
-          message: block.statusText,
-          icon: IconType.FLAG,
-        };
-      case 'reasoning':
-        return {
-          id: getGenericBlockId(message.id, idx),
-          branch,
-          parent,
-          message: block.reasoningText ?? 'Thinking...',
-          icon: IconType.THINK,
-        };
-      case 'text':
-        return {
-          id: getGenericBlockId(message.id, idx),
-          branch,
-          parent,
-          icon: message.sender.role === 'user' ? IconType.USER : IconType.AGENT,
-          message: ellipsisEnd(block.text, 64),
-        };
-      case 'reference':
-        return {
-          id: getGenericBlockId(message.id, idx),
-          branch,
-          parent,
-          icon: IconType.LINK,
-          message: stringifyRef(block.reference),
-        };
-      default:
-        return {
-          id: getGenericBlockId(message.id, idx),
-          branch,
-          parent,
-          message: block._tag,
-        };
-    }
-  });
-};
-
-const ellipsisEnd = (str: string, length: number) => {
-  if (str.length > length) {
-    return str.slice(0, length - 1) + '…';
-  }
-  return str;
-};
-
-const stringifyRef = (ref: Ref.Any) => {
-  if (ref.target) {
-    return stringifyObject(ref.target);
-  }
-  return ref.dxn.asEchoDXN()?.echoId ?? ref.dxn.asQueueDXN()?.objectId ?? '';
-};
-
-const stringifyObject = (obj: Obj.Any) => {
-  return Obj.getLabel(obj) ?? Obj.getTypename(obj) ?? obj.id;
-};

package/src/execution-graph/index.ts

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

package/src/functions/analysis.ts

@@ -1,20 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Effect, Schema } from 'effect';
-
-import { defineFunction } from '@dxos/functions';
-import { log } from '@dxos/log';
-
-export default defineFunction({
-  name: 'dxos.org/function/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* ({ 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 './list';
-export { default as load } from './load';

package/src/functions/list.ts

@@ -1,42 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Effect, Schema } from 'effect';
-
-import { Filter, Obj, Query } from '@dxos/echo';
-import { DatabaseService, defineFunction } from '@dxos/functions';
-import { trim } from '@dxos/util';
-
-// TODO(burdon): Factor out to space plugin.
-export default defineFunction({
-  name: 'dxos.org/function/assistant/list',
-  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 list-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 } }) {
-    const { objects } = yield* DatabaseService.runQuery(Query.select(Filter.typename(typename)));
-    const results = objects.map((object) => ({
-      dxn: Obj.getDXN(object).toString(),
-      label: Obj.getLabel(object),
-    }));
-
-    return { results };
-  }),
-});

package/src/functions/load.ts

@@ -1,43 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Effect, Option, Schema, pipe } from 'effect';
-
-import { ArtifactId } from '@dxos/assistant';
-import { Obj } from '@dxos/echo';
-import { DatabaseService, defineFunction } from '@dxos/functions';
-import { log } from '@dxos/log';
-
-// TODO(burdon): Common error handling.
-// TODO(burdon): Factor out to space plugin.
-export default defineFunction({
-  name: 'dxos.org/function/assistant/load',
-  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* 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 });
-        },
-      }),
-    );
-  }),
-});