@dxos/plugin-assistant 0.8.4-main.c4373fc → 0.8.4-main.c85a9c8dae

package/src/stories/components/TasksContainer.tsx

@@ -1,47 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React from 'react';
-
-import { Filter, Obj } from '@dxos/echo';
-import { Markdown } from '@dxos/plugin-markdown/types';
-import { createDocAccessor, useQuery } from '@dxos/react-client/echo';
-import { Toolbar, useThemeContext } from '@dxos/react-ui';
-import {
-  Editor,
-  createBasicExtensions,
-  createDataExtensions,
-  createMarkdownExtensions,
-  createThemeExtensions,
-  outliner,
-} from '@dxos/react-ui-editor';
-
-import { type ComponentProps } from './types';
-
-export const TasksContainer = ({ space }: ComponentProps) => {
-  const { themeMode } = useThemeContext();
-  const [document] = useQuery(space, Filter.type(Markdown.Document));
-  if (!document?.content.target) {
-    return null;
-  }
-
-  return (
-    <>
-      <Toolbar.Root classNames='border-b border-subduedSeparator'>
-        <h2>{Obj.getLabel(document)}</h2>
-      </Toolbar.Root>
-      <Editor
-        id={document.id}
-        classNames='h-full p-2 overflow-hidden'
-        extensions={[
-          createDataExtensions({ id: document.id, text: createDocAccessor(document.content.target, ['content']) }),
-          createBasicExtensions({ readOnly: false }),
-          createMarkdownExtensions(),
-          createThemeExtensions({ themeMode }),
-          outliner(),
-        ]}
-      />
-    </>
-  );
-};

package/src/stories/components/TokenManagerContainer.tsx

@@ -1,14 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { useMemo } from 'react';
-
-import { Surface } from '@dxos/app-framework';
-import { meta } from '@dxos/plugin-token-manager';
-
-export const TokenManagerContainer = () => {
-  const data = useMemo(() => ({ subject: `${meta.id}/space-settings` }), []);
-
-  return <Surface role='article' data={data} limit={1} />;
-};

package/src/stories/components/TriggersContainer.tsx

@@ -1,12 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React from 'react';
-
-import { Surface } from '@dxos/app-framework';
-import { meta as automationMeta } from '@dxos/plugin-automation';
-
-export const TriggersContainer = () => {
-  return <Surface role='article' data={{ subject: `${automationMeta.id}/space-settings-automation` }} />;
-};

package/src/stories/components/index.ts

@@ -1,22 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './types';
-
-export * from './BlueprintContainer';
-export * from './ChatContainer';
-export * from './ChessContainer';
-export * from './CommentsContainer';
-export * from './ExecutionGraphContainer';
-export * from './GraphContainer';
-export * from './InvocationsContainer';
-export * from './MessageContainer';
-export * from './ProjectContainer';
-export * from './PromptContainer';
-export * from './ResearchInputStack';
-export * from './ResearchOutputStack';
-export * from './ScriptContainer';
-export * from './TasksContainer';
-export * from './TokenManagerContainer';
-export * from './TriggersContainer';

package/src/stories/components/types.ts

@@ -1,11 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type Space } from '@dxos/client/echo';
-
-export type ComponentProps = {
-  space: Space;
-  debug?: boolean;
-  onEvent?: (event: string) => void;
-};

package/src/stories/testing/data.ts

@@ -1,135 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type Live, type Space } from '@dxos/client/echo';
-import { Obj, Relation, type Type } from '@dxos/echo';
-import { invariant } from '@dxos/invariant';
-import { DataType } from '@dxos/schema';
-
-export const testTypes: Type.Obj.Any[] = [
-  DataType.Organization,
-  DataType.Person,
-  DataType.Employer,
-  DataType.HasConnection,
-];
-
-export const organizations: (Type.Properties<DataType.Organization> & { id: string })[] = [
-  { id: 'dxos', name: 'DXOS', website: 'https://dxos.org' },
-  { id: 'socket_supply', name: 'Socket Supply', website: 'https://socketsupply.com' },
-  { id: 'ink_and_switch', name: 'Ink & Switch', website: 'https://inkandswitch.com' },
-  { id: 'effectful', name: 'Effectful', website: 'https://effect.co' },
-  { id: 'blueyard', name: 'Blue Yard', website: 'https://blueyard.com' },
-  { id: 'backed', name: 'Backed', website: 'https://backed.vc' },
-  { id: 'protocol_labs', name: 'Protocol Labs', website: 'https://protocol.ai' },
-  { id: 'newlab', name: 'Newlab', website: 'https://newlab.com' },
-  { id: 'google', name: 'Google', website: 'https://google.com' },
-  { id: 'microsoft', name: 'Microsoft', website: 'https://microsoft.com' },
-  { id: 'openai', name: 'OpenAI', website: 'https://openai.com' },
-  { id: 'anthropic', name: 'Anthropic', website: 'https://anthropic.com' },
-  { id: 'amazon', name: 'Amazon', website: 'https://amazon.com' },
-  { id: 'deshaw', name: 'D. E. Shaw & Co.', website: 'https://deshaw.com' },
-];
-
-const people: (Type.Properties<DataType.Person> & { id: string })[] = [
-  { id: 'rich_burdon', fullName: 'Rich Burdon' },
-  { id: 'josiah_witt', fullName: 'Josiah Witt' },
-  { id: 'dima_dmaretskyi', fullName: 'Dima Maretskyi' },
-  { id: 'chad_fowler', fullName: 'Chad Fowler' },
-  { id: 'ciaran_oleary', fullName: "Ciarán O'Leary" },
-  { id: 'jason_whitmire', fullName: 'Jason Whitmire' },
-  { id: 'alex_brunicki', fullName: 'Alex Brunicki' },
-  { id: 'andre_de_haes', fullName: 'Andre de Haes' },
-  { id: 'scott_cohen', fullName: 'Scott Cohen' },
-  { id: 'juan_benet', fullName: 'Juan Benet' },
-  { id: 'satya_nadella', fullName: 'Satya Nadella' },
-  { id: 'kevin_scott', fullName: 'Kevin Scott' },
-  { id: 'jeff_bezos', fullName: 'Jeff Bezos' },
-];
-
-const testObjects: Record<string, any[]> = {
-  [DataType.Organization.typename]: organizations,
-  [DataType.Person.typename]: people,
-};
-
-const testRelationships: Record<
-  string,
-  ({
-    source: string;
-    target: string;
-  } & Record<string, any>)[]
-> = {
-  [DataType.Employer.typename]: [
-    // prettier-ignore
-    { source: 'rich_burdon', target: 'dxos' },
-    { source: 'rich_burdon', target: 'google', active: false }, // TODO(burdon): Should not contribute to force.
-    { source: 'rich_burdon', target: 'deshaw', active: false },
-    { source: 'josiah_witt', target: 'dxos' },
-    { source: 'dima_dmaretskyi', target: 'dxos' },
-    { source: 'chad_fowler', target: 'blueyard' },
-    { source: 'chad_fowler', target: 'microsoft', active: false },
-    { source: 'ciaran_oleary', target: 'blueyard' },
-    { source: 'jason_whitmire', target: 'blueyard' },
-    { source: 'juan_benet', target: 'protocol_labs' },
-    { source: 'alex_brunicki', target: 'backed' },
-    { source: 'andre_de_haes', target: 'backed' },
-    { source: 'scott_cohen', target: 'newlab' },
-    { source: 'satya_nadella', target: 'microsoft' },
-    { source: 'kevin_scott', target: 'microsoft' },
-    { source: 'kevin_scott', target: 'google', active: false },
-    { source: 'jeff_bezos', target: 'amazon' },
-    { source: 'jeff_bezos', target: 'deshaw', active: false },
-  ],
-
-  // TODO(burdon): Limit graph view to selected relationship types.
-  [DataType.HasConnection.typename]: [
-    // prettier-ignore
-    { kind: 'partner', source: 'dxos', target: 'ink_and_switch' },
-    { kind: 'partner', source: 'dxos', target: 'effectful' },
-    { kind: 'partner', source: 'dxos', target: 'socket_supply' },
-
-    // prettier-ignore
-    { kind: 'investor', source: 'blueyard', target: 'dxos' },
-    { kind: 'investor', source: 'blueyard', target: 'protocol_labs' },
-    { kind: 'investor', source: 'protocol_labs', target: 'dxos' },
-    { kind: 'investor', source: 'backed', target: 'dxos' },
-    { kind: 'investor', source: 'newlab', target: 'dxos' },
-    { kind: 'investor', source: 'microsoft', target: 'openai' },
-    { kind: 'investor', source: 'google', target: 'anthropic' },
-    { kind: 'investor', source: 'amazon', target: 'anthropic' },
-  ],
-};
-
-export const addTestData = async (space: Space): Promise<void> => {
-  const objectMap = new Map<string, Live<any>>();
-
-  for (const [typename, objects] of Object.entries(testObjects)) {
-    const schema = space.db.graph.schemaRegistry.getSchema(typename);
-    invariant(schema, `Schema not found: ${typename}`);
-    for (const { id, ...data } of objects) {
-      const object = space.db.add(Obj.make(schema, data));
-      objectMap.set(id, object);
-    }
-  }
-
-  for (const [typename, relationships] of Object.entries(testRelationships)) {
-    const schema = space.db.graph.schemaRegistry.getSchema(typename);
-    invariant(schema, `Schema not found: ${typename}`);
-
-    for (const { source, target, ...data } of relationships) {
-      const sourceObject = objectMap.get(source);
-      const targetObject = objectMap.get(target);
-      invariant(sourceObject, `Source object not found: ${source}`);
-      invariant(targetObject, `Target object not found: ${target}`);
-
-      space.db.add(
-        Relation.make(schema, {
-          // TODO(burdon): Test source/target types match.
-          [Relation.Source]: sourceObject,
-          [Relation.Target]: targetObject,
-          ...data,
-        }),
-      );
-    }
-  }
-};

package/src/stories/testing/index.ts

@@ -1,6 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './testing';
-export * from './schema';

package/src/stories/testing/schema.ts

@@ -1,17 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Schema from 'effect/Schema';
-
-import { Type } from '@dxos/echo';
-import { Queue } from '@dxos/echo-db';
-
-export const ResearchInputQueue = Schema.Struct({
-  queue: Type.Ref(Queue),
-}).pipe(
-  Type.Obj({
-    typename: 'dxos.org/type/ResearchInputQueue',
-    version: '0.1.0',
-  }),
-);

package/src/stories/testing/testing.tsx

@@ -1,291 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import * as Tool from '@effect/ai/Tool';
-import * as Toolkit from '@effect/ai/Toolkit';
-import * as Console from 'effect/Console';
-import * as Schema from 'effect/Schema';
-
-import { SERVICES_CONFIG } from '@dxos/ai/testing';
-import {
-  Capabilities,
-  Events,
-  IntentPlugin,
-  LayoutAction,
-  type Plugin,
-  type PluginContext,
-  SettingsPlugin,
-  allOf,
-  contributes,
-  createIntent,
-  createResolver,
-  defineModule,
-  definePlugin,
-} from '@dxos/app-framework';
-import { withPluginManager } from '@dxos/app-framework/testing';
-import { AiContextBinder, ArtifactId } from '@dxos/assistant';
-import {
-  DESIGN_BLUEPRINT,
-  PLANNING_BLUEPRINT,
-  agent,
-  createResearchNote,
-  readDocument,
-  readTasks,
-  research,
-  updateDocument,
-  updateTasks,
-} from '@dxos/assistant-testing';
-import { Blueprint, Prompt } from '@dxos/blueprints';
-import { type Space } from '@dxos/client/echo';
-import { Obj, Ref } from '@dxos/echo';
-import { FunctionTrigger, FunctionType, exampleFunctions } from '@dxos/functions';
-import { log } from '@dxos/log';
-import { AttentionPlugin } from '@dxos/plugin-attention';
-import { AutomationPlugin } from '@dxos/plugin-automation';
-import { ClientCapabilities, ClientEvents, ClientPlugin } from '@dxos/plugin-client';
-import { type ClientPluginOptions } from '@dxos/plugin-client/types';
-import { DeckAction } from '@dxos/plugin-deck/types';
-import { GraphPlugin } from '@dxos/plugin-graph';
-import { Markdown } from '@dxos/plugin-markdown/types';
-import { PreviewPlugin } from '@dxos/plugin-preview';
-import { SpacePlugin } from '@dxos/plugin-space';
-import { StorybookLayoutPlugin } from '@dxos/plugin-storybook-layout';
-import { ThemePlugin } from '@dxos/plugin-theme';
-import { type Client, Config } from '@dxos/react-client';
-import { defaultTx } from '@dxos/react-ui-theme';
-import { DataType } from '@dxos/schema';
-import { trim } from '@dxos/util';
-
-import { AssistantPlugin } from '../../AssistantPlugin';
-import { Assistant, AssistantAction } from '../../types';
-
-// TODO(burdon): Factor out.
-export const config = {
-  remote: new Config({
-    runtime: {
-      services: SERVICES_CONFIG.REMOTE,
-    },
-  }),
-  persistent: new Config({
-    runtime: {
-      client: {
-        storage: {
-          persistent: true,
-        },
-      },
-      services: SERVICES_CONFIG.REMOTE,
-    },
-  }),
-  local: new Config({
-    runtime: {
-      services: SERVICES_CONFIG.LOCAL,
-    },
-  }),
-};
-
-class TestingToolkit extends Toolkit.make(
-  Tool.make('open-item', {
-    description: trim`
-      Opens an item in the application.
-    `,
-    parameters: {
-      id: ArtifactId,
-    },
-    success: Schema.Any,
-    failure: Schema.Never,
-  }),
-) {
-  static layer = (_context: PluginContext) =>
-    TestingToolkit.toLayer({
-      'open-item': ({ id }) => Console.log('Called open-item', { id }),
-    });
-}
-
-type DecoratorsProps = Omit<ClientPluginOptions, 'onClientInitialized' | 'onSpacesReady'> &
-  Pick<StoryPluginOptions, 'onChatCreated'> & {
-    plugins?: Plugin[];
-    accessTokens?: DataType.AccessToken[];
-    onInit?: (props: { client: Client; space: Space }) => Promise<void>;
-  };
-
-/**
- * Create storybook decorators.
- */
-export const getDecorators = ({
-  types = [],
-  plugins = [],
-  accessTokens = [],
-  onInit,
-  onChatCreated,
-  ...props
-}: DecoratorsProps) => [
-  withPluginManager({
-    plugins: [
-      // System plugins.
-      AttentionPlugin(),
-      AutomationPlugin(),
-      GraphPlugin(),
-      IntentPlugin(),
-      SettingsPlugin(),
-      SpacePlugin({}),
-      ClientPlugin({
-        types: [
-          Markdown.Document,
-          Assistant.Chat,
-          Blueprint.Blueprint,
-          Prompt.Prompt,
-          DataType.AccessToken,
-          FunctionTrigger,
-          FunctionType,
-          ...types,
-        ],
-        onClientInitialized: async ({ client }) => {
-          log('onClientInitialized', { identity: client.halo.identity.get()?.did });
-          // Abort if already initialized.
-          if (client.halo.identity.get()) {
-            return;
-          }
-
-          await client.halo.createIdentity();
-          await client.spaces.waitUntilReady();
-
-          const space = client.spaces.default;
-          // TODO(burdon): Should not require this.
-          //  ERROR: invariant violation: Database was not initialized with root object.
-          // TODO(burdon): onSpacesReady is never called.
-          await space.waitUntilReady();
-
-          // Add tokens.
-          for (const accessToken of accessTokens) {
-            space.db.add(Obj.clone(accessToken));
-          }
-
-          await space.db.flush({ indexes: true });
-          await onInit?.({ client, space });
-          await space.db.flush({ indexes: true });
-        },
-        ...props,
-      }),
-
-      // Cards
-      ThemePlugin({ tx: defaultTx }),
-      StorybookLayoutPlugin({}),
-      PreviewPlugin(),
-
-      // User plugins.
-      AssistantPlugin(),
-      StoryPlugin({ onChatCreated }),
-
-      // Test-specific.
-      ...plugins,
-    ],
-  }),
-];
-
-/**
- * Creates access tokens from environment variables.
- * @param tokens - Record of token sources mapped to their VITE_ prefixed environment variable values
- * @returns Array of AccessToken objects for non-empty token values
- * @example
- * ```tsx
- * const tokens = accessTokensFromEnv({
- *   'exa.ai': process.env.VITE_EXA_API_KEY,
- *   'linear.app': process.env.VITE_LINEAR_API_KEY
- * });
- * ```
- * @note All environment variables should use the VITE_ prefix for proper Vite bundling
- */
-
-export const accessTokensFromEnv = (tokens: Record<string, string | undefined>) => {
-  return Object.entries(tokens)
-    .filter(([, token]) => !!token)
-    .map(([source, token]) => Obj.make(DataType.AccessToken, { source, token: token! }));
-};
-
-type StoryPluginOptions = {
-  onChatCreated?: (props: { space: Space; chat: Assistant.Chat; binder: AiContextBinder }) => Promise<void>;
-};
-
-const StoryPlugin = definePlugin<StoryPluginOptions>(
-  {
-    id: 'example.com/plugin/testing',
-    name: 'Testing',
-  },
-  ({ onChatCreated }) => [
-    defineModule({
-      id: 'example.com/plugin/testing/module/testing',
-      activatesOn: Events.SetupArtifactDefinition,
-      activate: () => [
-        // TODO(burdon): Move into assistnat?
-        contributes(Capabilities.BlueprintDefinition, DESIGN_BLUEPRINT),
-        contributes(Capabilities.BlueprintDefinition, PLANNING_BLUEPRINT),
-        contributes(Capabilities.Functions, [agent]),
-        contributes(Capabilities.Functions, [readDocument, updateDocument]),
-        contributes(Capabilities.Functions, [readTasks, updateTasks]),
-        contributes(Capabilities.Functions, [research, createResearchNote]),
-        contributes(Capabilities.Functions, [exampleFunctions.reply]),
-      ],
-    }),
-    defineModule({
-      id: 'example.com/plugin/testing/module/toolkit',
-      activatesOn: Events.Startup,
-      activate: (context) => [
-        contributes(Capabilities.Toolkit, TestingToolkit),
-        contributes(Capabilities.ToolkitHandler, TestingToolkit.layer(context)),
-      ],
-    }),
-    defineModule({
-      id: 'example.com/plugin/testing/module/setup',
-      activatesOn: allOf(Events.DispatcherReady, ClientEvents.SpacesReady),
-      activate: async (context) => {
-        const client = context.getCapability(ClientCapabilities.Client);
-        const space = client.spaces.default;
-        const { dispatchPromise: dispatch } = context.getCapability(Capabilities.IntentDispatcher);
-
-        // Ensure workspace is set.
-        await dispatch(createIntent(LayoutAction.SwitchWorkspace, { part: 'workspace', subject: space.id }));
-
-        // Create initial chat.
-        await dispatch(createIntent(AssistantAction.CreateChat, { space }));
-
-        return [];
-      },
-    }),
-    defineModule({
-      id: 'example.com/plugin/testing/module/intent-resolver',
-      activatesOn: Events.SetupIntentResolver,
-      activate: () => [
-        contributes(Capabilities.IntentResolver, [
-          createResolver({
-            intent: DeckAction.ChangeCompanion,
-            resolve: () => ({}),
-          }),
-          createResolver({
-            intent: AssistantAction.CreateChat,
-            position: 'hoist',
-            resolve: async ({ space, name }) => {
-              const queue = space.queues.create();
-              const traceQueue = space.queues.create();
-              const chat = Obj.make(Assistant.Chat, {
-                name,
-                queue: Ref.fromDXN(queue.dxn),
-                traceQueue: Ref.fromDXN(traceQueue.dxn),
-              });
-              const binder = new AiContextBinder(queue);
-
-              // Story-specific behaviour to allow chat creation to be extended.
-              space.db.add(chat);
-              await space.db.flush({ indexes: true });
-              await onChatCreated?.({ space, chat, binder });
-
-              return {
-                data: { object: chat },
-              };
-            },
-          }),
-        ]),
-      ],
-    }),
-  ],
-);