@dxos/plugin-assistant 0.8.4-main.5ea62a8 → 0.8.4-main.ae835ea

package/src/stories/testing/testing.tsx

@@ -1,254 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { AiTool, AiToolkit } from '@effect/ai';
-import { Console, Schema } from 'effect';
-
-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,
-  LINEAR_BLUEPRINT,
-  PLANNING_BLUEPRINT,
-  RESEARCH_BLUEPRINT,
-  readDocument,
-  readTasks,
-  remoteServiceEndpoints,
-  research,
-  syncLinearIssues,
-  updateDocument,
-  updateTasks,
-} from '@dxos/assistant-testing';
-import { Blueprint } from '@dxos/blueprints';
-import { type Space } from '@dxos/client/echo';
-import { Obj, Ref } from '@dxos/echo';
-import { log } from '@dxos/log';
-import { AttentionPlugin } from '@dxos/plugin-attention';
-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 { Config } from '@dxos/react-client';
-import { defaultTx } from '@dxos/react-ui-theme';
-import { DataType } from '@dxos/schema';
-import { withLayout } from '@dxos/storybook-utils';
-import { trim } from '@dxos/util';
-
-import { AssistantPlugin } from '../../AssistantPlugin';
-import { Assistant } from '../../types';
-
-// TODO(burdon): Factor out.
-export const config = {
-  remote: new Config({
-    runtime: {
-      services: {
-        ai: {
-          // TODO(burdon): Normalize props ('url'?)
-          server: remoteServiceEndpoints.ai,
-        },
-        edge: {
-          url: remoteServiceEndpoints.edge,
-        },
-      },
-    },
-  }),
-  persistent: new Config({
-    runtime: {
-      client: {
-        storage: {
-          persistent: true,
-        },
-      },
-      services: {
-        ai: {
-          server: remoteServiceEndpoints.ai,
-        },
-        edge: {
-          url: remoteServiceEndpoints.edge,
-        },
-      },
-    },
-  }),
-};
-
-class TestingToolkit extends AiToolkit.make(
-  AiTool.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'> & {
-  plugins?: Plugin[];
-  accessTokens?: DataType.AccessToken[];
-  onInit?: (props: { space: Space; chat: Assistant.Chat; binder: AiContextBinder }) => Promise<void>;
-};
-
-/**
- * Create storybook decorators.
- */
-export const getDecorators = ({ types = [], plugins = [], accessTokens = [], onInit, ...props }: DecoratorsProps) => [
-  withPluginManager({
-    plugins: [
-      // System plugins.
-      AttentionPlugin(),
-      GraphPlugin(),
-      IntentPlugin(),
-      SettingsPlugin(),
-      SpacePlugin(),
-      ClientPlugin({
-        types: [Markdown.Document, Assistant.Chat, Blueprint.Blueprint, DataType.AccessToken, ...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));
-          }
-
-          // Create chat and queue.
-          const chat = space.db.add(
-            Obj.make(Assistant.Chat, {
-              queue: Ref.fromDXN(space.queues.create().dxn),
-              traceQueue: Ref.fromDXN(space.queues.create().dxn),
-            }),
-          );
-
-          await space.db.flush({ indexes: true });
-          await onInit?.({ space, chat, binder: new AiContextBinder(await chat.queue.load()) });
-          await space.db.flush({ indexes: true });
-        },
-        ...props,
-      }),
-
-      // Cards
-      ThemePlugin({ tx: defaultTx }),
-      StorybookLayoutPlugin(),
-      PreviewPlugin(),
-
-      // User plugins.
-      AssistantPlugin(),
-
-      // Custom.
-      definePlugin({ id: 'example.com/plugin/testing', name: 'Testing' }, [
-        defineModule({
-          id: 'example.com/plugin/testing/module/testing',
-          activatesOn: Events.SetupArtifactDefinition,
-          activate: () => [
-            contributes(Capabilities.BlueprintDefinition, DESIGN_BLUEPRINT),
-            contributes(Capabilities.BlueprintDefinition, PLANNING_BLUEPRINT),
-            contributes(Capabilities.BlueprintDefinition, RESEARCH_BLUEPRINT),
-            contributes(Capabilities.BlueprintDefinition, LINEAR_BLUEPRINT),
-            contributes(Capabilities.Functions, [readDocument, updateDocument]),
-            contributes(Capabilities.Functions, [readTasks, updateTasks]),
-            contributes(Capabilities.Functions, [research]),
-            contributes(Capabilities.Functions, [syncLinearIssues]),
-          ],
-        }),
-        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/set-workspace',
-          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);
-            await dispatch(createIntent(LayoutAction.SwitchWorkspace, { part: 'workspace', subject: space.id }));
-            return [];
-          },
-        }),
-        defineModule({
-          id: 'example.com/plugin/testing/module/intent-resolver',
-          activatesOn: Events.SetupIntentResolver,
-          activate: () => [
-            contributes(Capabilities.IntentResolver, [
-              createResolver({
-                intent: DeckAction.ChangeCompanion,
-                resolve: () => ({}),
-              }),
-            ]),
-          ],
-        }),
-      ]),
-
-      // Test-specific.
-      ...plugins,
-    ],
-  }),
-  withLayout({
-    fullscreen: true,
-    classNames: 'justify-center bg-deckSurface',
-  }),
-];
-
-/**
- * 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! }));
-};

package/src/testing/test-data.ts

@@ -1,109 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Obj } from '@dxos/echo';
-import { DataType } from '@dxos/schema';
-
-export const createTestTranscription = (): DataType.Message[] => {
-  const timeInterval = 1000;
-  const transcription: DataType.Message[] = [
-    {
-      text: 'Hey everyone, glad we could all connect today. I was thinking we could discuss where AI might be heading in the next decade.',
-      sender: 'Mykola',
-    },
-    {
-      text: 'Yeah, that sounds great. Personally, I think AI will get deeply integrated into daily life—like assistants that don’t just answer questions but anticipate our needs.',
-      sender: 'Rich',
-    },
-    {
-      text: 'That’s true, but there are risks too. If AI predicts too much, we might lose the sense of making decisions ourselves.',
-      sender: 'Dima',
-    },
-    {
-      text: 'Good point. I think the balance will come from better human-AI collaboration rather than pure automation.',
-      sender: 'Dima',
-    },
-    {
-      text: 'Right, like AI as a co-pilot instead of a replacement. I see that happening a lot in creative fields already.',
-      sender: 'Mykola',
-    },
-    {
-      text: 'Exactly. And in science, too—AI models can suggest hypotheses humans wouldn’t think of, speeding up discoveries.',
-      sender: 'Rich',
-    },
-    {
-      text: 'So it looks like the future is less about AI replacing us, and more about AI amplifying what humans can do.',
-      sender: 'Dima',
-    },
-    {
-      text: 'I think we should research alignment of AI with human values.',
-      sender: 'Mykola',
-    },
-    {
-      text: 'I agree. We should also research the risks of AI generating content and polluting knowledge bases.',
-      sender: 'Rich',
-    },
-  ].map((message, index, array) => {
-    const created = new Date(Date.now() - (array.length - index) * timeInterval);
-    return Obj.make(DataType.Message, {
-      created: created.toISOString(),
-      blocks: [{ _tag: 'transcript', started: created.toISOString(), text: message.text }],
-      sender: { identityDid: message.sender },
-    });
-  });
-
-  return transcription;
-};
-
-// TODO(wittjosiah): Find way to use data generator to generate substantive messages that could be summarized.
-export const createTestMailbox = (): DataType.Message[] => {
-  const timeInterval = 1000;
-  const messages: DataType.Message[] = [
-    {
-      text: 'Subject: Project Kickoff\n\nHi team,\n\nWe are excited to announce the kickoff of the Apollo project. Please review the attached roadmap and be prepared for our first meeting on Monday.\n\nBest,\nAlice',
-      sender: 'alice.johnson@acme-corp.com',
-    },
-    {
-      text: 'Subject: Sales Update\n\nDear all,\n\nQ2 sales numbers have exceeded expectations. Congratulations to everyone for their hard work!\n\nRegards,\nBob',
-      sender: 'bob.smith@salesforce.biz',
-    },
-    {
-      text: 'Subject: Invoice Reminder\n\nHello,\n\nThis is a friendly reminder that invoice #12345 is due next week. Please let us know if you have any questions.\n\nThank you,\nCarol',
-      sender: 'carol.finance@invoicesolutions.com',
-    },
-    {
-      text: 'Subject: Meeting Rescheduled\n\nHi team,\n\nTomorrow’s product design meeting has been moved to Friday at 2pm. Please update your calendars.\n\nThanks,\nDavid',
-      sender: 'david.lee@productdesign.io',
-    },
-    {
-      text: 'Subject: Welcome Aboard!\n\nWelcome to the company, Emily! We are thrilled to have you join the marketing department.\n\nBest,\nFiona',
-      sender: 'fiona.hr@enterpriseco.com',
-    },
-    {
-      text: 'Subject: IT Maintenance\n\nDear colleagues,\n\nScheduled IT maintenance will occur this Saturday from 1am to 5am. Please save your work accordingly.\n\nSincerely,\nGreg',
-      sender: 'greg.it@supporthub.net',
-    },
-    {
-      text: 'Subject: Lunch & Learn\n\nJoin us for a Lunch & Learn session on cloud security this Thursday in the main conference room.\n\nSee you there,\nHelen',
-      sender: 'helen.training@acme-corp.com',
-    },
-    {
-      text: 'Subject: Client Feedback\n\nHi,\n\nOur client, InnovateX, sent positive feedback on the recent deployment. Great job, everyone!\n\nCheers,\nIvan',
-      sender: 'ivan.account@consultingpartners.org',
-    },
-    {
-      text: 'Subject: Policy Update\n\nPlease review the updated remote work policy attached to this email. Let us know if you have any questions.\n\nBest regards,\nJulia',
-      sender: 'julia.admin@acme-corp.com',
-    },
-  ].map((message, index, array) => {
-    const created = new Date(Date.now() - (array.length - index) * timeInterval);
-    return Obj.make(DataType.Message, {
-      created: created.toISOString(),
-      blocks: [{ _tag: 'text', text: message.text }],
-      sender: { email: message.sender },
-    });
-  });
-
-  return messages;
-};

package/src/testing/test-functions.ts

@@ -1,16 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { DEFAULT_EDGE_MODEL } from '@dxos/ai';
-import { createSystemPrompt } from '@dxos/assistant';
-
-import { type AiChatProcessorOptions } from '../processor';
-
-/**
- * @deprecated
- */
-export const createProcessorOptions = (): Pick<AiChatProcessorOptions, 'model' | 'system'> => ({
-  model: DEFAULT_EDGE_MODEL,
-  system: createSystemPrompt({}),
-});