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

package/src/execution-graph/execution-graph.ts

@@ -1,211 +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 { LogLevel } from '@dxos/log';
-import { type Commit } from '@dxos/react-ui-components';
-import { ContentBlock, DataType } from '@dxos/schema';
-import { isNotFalsy } from '@dxos/util';
-
-// TODO(burdon): Move to assistant.
-
-// TODO(burdon): Add colors?
-enum IconType {
-  // General status.
-  SUCCESS = 'ph--check-circle--regular',
-  WARNING = 'ph--warning-circle--regular',
-  ERROR = 'ph--x-circle--regular',
-  FLAG = 'ph--flag--regular',
-  ROCKET = 'ph--rocket--regular',
-  TIMER = 'ph--timer--regular',
-
-  // Interactions.
-  USER = 'ph--user--regular',
-  USER_INTERACTION = 'ph--user-sound--regular',
-  AGENT = 'ph--robot--regular',
-  THINKING = '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 = messageToCommit(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,
-          parents:
-            event.parentMessage && event.toolCallId
-              ? [getToolCallId(event.parentMessage, event.toolCallId)]
-              : undefined, // TODO(burdon): Fix.
-        });
-      }
-    }
-  }
-
-  /**
-   * Returns the current state of the graph.
-   */
-  getGraph(lastRequest = false): { branches: string[]; commits: Commit[] } {
-    const idx = lastRequest ? this._commits.findLastIndex((c) => c.tags?.includes('user')) : -1;
-    const commits = idx === -1 ? this._commits : this._commits.slice(idx);
-
-    return {
-      branches: Array.from(this._branchNames),
-      commits: commits,
-    };
-  }
-}
-
-// TODO(burdon): Pass in AiToolProvider.
-const messageToCommit = (message: DataType.Message): Commit[] => {
-  return message.blocks
-    .map((block, idx) => {
-      const branch = getMessageBranch(message);
-      const parent = getParentId(message);
-      const parents = parent ? [parent] : [];
-      switch (block._tag) {
-        case 'text':
-          if (!block.text.trim().length) {
-            return null;
-          }
-          return {
-            id: getGenericBlockId(message.id, idx),
-            branch,
-            parents,
-            ...(message.sender.role === 'user'
-              ? {
-                  icon: IconType.USER,
-                  tags: ['user'],
-                  message: 'Processing request...',
-                }
-              : {
-                  icon: IconType.AGENT,
-                  message: `Response (${block.text.split(' ').length} words)`,
-                }),
-          } satisfies Commit;
-        case 'toolCall':
-          return {
-            id: getToolCallId(message.id, block.toolCallId),
-            branch,
-            parents,
-            icon: IconType.TOOL,
-            level: LogLevel.INFO,
-            // TODO(burdon): Lookup tool name/description?
-            message: `Calling tool (${block.name})`,
-          } satisfies Commit;
-        case 'toolResult':
-          return {
-            id: getToolResultId(message.id, block.toolCallId),
-            branch,
-            parents,
-            icon: block.error ? IconType.ERROR : IconType.SUCCESS,
-            level: block.error ? LogLevel.ERROR : LogLevel.INFO,
-            message: block.error ? 'Tool error: ' + block.error : 'Tool call succeeded',
-          } satisfies Commit;
-        case 'summary':
-          return {
-            id: getGenericBlockId(message.id, idx),
-            branch,
-            parents,
-            icon: IconType.ROCKET,
-            level: LogLevel.INFO,
-            message: ContentBlock.createSummaryMessage(block),
-          } satisfies Commit;
-        case 'status':
-          return {
-            id: getGenericBlockId(message.id, idx),
-            branch,
-            parents,
-            message: block.statusText,
-            level: LogLevel.INFO,
-            icon: IconType.FLAG,
-          } satisfies Commit;
-        case 'reasoning':
-          return {
-            id: getGenericBlockId(message.id, idx),
-            branch,
-            parents,
-            message: block.reasoningText ?? 'Thinking...',
-            icon: IconType.THINKING,
-          } satisfies Commit;
-        case 'reference':
-          return {
-            id: getGenericBlockId(message.id, idx),
-            branch,
-            parents,
-            icon: IconType.LINK,
-            message: stringifyRef(block.reference),
-          } satisfies Commit;
-        default:
-          return null;
-      }
-    })
-    .filter(isNotFalsy);
-};
-
-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 = (options: { parentMessage?: ObjectId; toolCallId?: string }) => {
-  if (options.parentMessage && options.toolCallId) {
-    return `${options.parentMessage}_${options.toolCallId}`;
-  } else if (options.parentMessage) {
-    return options.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 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* ({ context, data: { enable } }) {
-    log.info('analysis mode', { space: context.space?.id, 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,46 +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 } }) {
-    // 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/load.ts

@@ -1,42 +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';
-
-// 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 });
-        },
-      }),
-    );
-  }),
-});

package/src/hooks/useComputeRuntimeCallback.ts

@@ -1,25 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import type { Effect } from 'effect';
-import { useCallback } from 'react';
-
-import { useCapability } from '@dxos/app-framework';
-import type { Space } from '@dxos/client/echo';
-
-import { AssistantCapabilities } from '../capabilities';
-
-/**
- * Create an effectful function that has access to compute services
- */
-export const useComputeRuntimeCallback = <T>(
-  space: Space,
-  fn: () => Effect.Effect<T, any, AssistantCapabilities.ComputeServices>,
-  deps?: React.DependencyList,
-): (() => Promise<T>) => {
-  const computeRuntime = useCapability(AssistantCapabilities.ComputeRuntime);
-  const runtime = computeRuntime.getRuntime(space.id);
-
-  return useCallback(() => runtime.runPromise(fn()), [runtime, ...(deps ?? [])]);
-};

package/src/hooks/useExecutionGraph.ts

@@ -1,20 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { useMemo } from 'react';
-
-import { type Queue } from '@dxos/client/echo';
-import { Obj } from '@dxos/echo';
-import { useQueue } from '@dxos/react-client/echo';
-
-import { ExecutionGraph } from '../execution-graph';
-
-export const useExecutionGraph = (queueRef?: Queue, lastRequest = false) => {
-  const queue = useQueue(queueRef?.dxn);
-  return useMemo(() => {
-    const graph = new ExecutionGraph();
-    graph.addEvents(queue?.objects.filter(Obj.isObject) ?? []);
-    return graph.getGraph(lastRequest);
-  }, [queue?.objects, lastRequest]);
-};

package/src/hooks/useItemTypes.ts

@@ -1,37 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Option } from 'effect';
-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;
-};