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

package/src/stories/testing/testing.tsx

@@ -0,0 +1,254 @@
+//
+// 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/index.ts

@@ -2,6 +2,7 @@
 // Copyright 2024 DXOS.org
 //
 
-export * from './test-sequence';
+export * from './test-data';
 export * from './test-functions';
+export * from './test-sequence';
 export * from './test-services';

package/src/testing/test-data.ts

@@ -0,0 +1,109 @@
+//
+// 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

@@ -2,10 +2,15 @@
 // Copyright 2024 DXOS.org
 //
 
-import { DEFAULT_EDGE_MODEL, type GenerateRequest } from '@dxos/ai';
+import { DEFAULT_EDGE_MODEL } from '@dxos/ai';
 import { createSystemPrompt } from '@dxos/assistant';
 
-export const createProcessorOptions = (artifacts: string[]): Pick<GenerateRequest, 'model' | 'systemPrompt'> => ({
+import { type AiChatProcessorOptions } from '../processor';
+
+/**
+ * @deprecated
+ */
+export const createProcessorOptions = (): Pick<AiChatProcessorOptions, 'model' | 'system'> => ({
   model: DEFAULT_EDGE_MODEL,
-  systemPrompt: createSystemPrompt({ artifacts }),
+  system: createSystemPrompt({}),
 });

package/src/testing/test-sequence.ts

@@ -2,33 +2,8 @@
 // Copyright 2025 DXOS.org
 //
 
-import { ToolId, ToolRegistry } from '@dxos/ai';
-// import { createExaTool, createGraphWriterTool, createLocalSearchTool } from '@dxos/assistant';
-import { type Space } from '@dxos/client/echo';
-import { SequenceParser, type SequenceDefinition } from '@dxos/conductor';
-import { type DXN } from '@dxos/keys';
-import { isNonNullable } from '@dxos/util';
-
-// TODO(dmaretskyi): make db available through services (same as function executor).
-// TODO(burdon): Can tools implement "aspects" so that variants can be used rather than an explicit reference?
-export const createToolRegistry = (space: Space, queueDxn?: DXN): ToolRegistry => {
-  return new ToolRegistry(
-    [
-      // createExaTool({ apiKey: EXA_API_KEY }),
-      // createLocalSearchTool(space.db, queueDxn && space.queues.get(queueDxn)),
-      // queueDxn &&
-      //   createGraphWriterTool({
-      //     db: space.db,
-      //     queue: space.queues.get(queueDxn),
-      //     schema: DataTypes,
-      //     onDone: async (objects) => {
-      //       const queue = space.queues.get(queueDxn);
-      //       await queue.append(objects);
-      //     },
-      //   }),
-    ].filter(isNonNullable),
-  );
-};
+import { ToolId } from '@dxos/ai';
+import { type SequenceDefinition, SequenceParser } from '@dxos/conductor';
 
 export const RESEARCH_SEQUENCE_DEFINITION: SequenceDefinition = {
   steps: [

package/src/translations.ts

@@ -2,13 +2,14 @@
 // Copyright 2023 DXOS.org
 //
 
-import { Blueprint, Template } from '@dxos/blueprints';
+import { Blueprint } from '@dxos/blueprints';
 import { Sequence } from '@dxos/conductor';
 import { type Resource } from '@dxos/react-ui';
 
 import { meta } from './meta';
 import { Assistant } from './types';
 
+// TODO(burdon): Standardize translation names.
 export const translations = [
   {
     'en-US': {
@@ -22,15 +23,6 @@ export const translations = [
         'rename object label': 'Rename blueprint',
         'delete object label': 'Delete blueprint',
       },
-      [Template.Template.typename]: {
-        'typename label': 'Template',
-        'typename label_zero': 'Templates',
-        'typename label_one': 'Template',
-        'typename label_other': 'Templates',
-        'object name placeholder': 'New template',
-        'rename object label': 'Rename template',
-        'delete object label': 'Delete template',
-      },
       // TODO(burdon): From conductor.
       [Sequence.typename]: {
         'typename label': 'Sequence',
@@ -42,19 +34,18 @@ export const translations = [
         'delete object label': 'Delete sequence',
       },
       [Assistant.Chat.typename]: {
-        'typename label': 'Assistant',
-        'typename label_zero': 'Assistants',
-        'typename label_one': 'Assistant',
-        'typename label_other': 'Assistants',
-        'object name placeholder': 'New assistant',
-        'rename object label': 'Rename assistant',
-        'delete object label': 'Delete assistant',
+        'typename label': 'AI Chat',
+        'object name placeholder': 'New AI Chat',
+        'rename object label': 'Rename AI Chat',
+        'delete object label': 'Delete AI Chat',
       },
+      // TODO(burdon): Reconcile with react-ui-chat.
       [meta.id]: {
         'templates label': 'Templates',
         'open ambient chat label': 'Open Assistant',
         'assistant chat label': 'Assistant',
         'plugin name': 'Assistant',
+        'settings title': 'Assistant settings',
         'object placeholder': 'New prompt',
         'create object label': 'Create prompt',
         'create trigger label': 'Create trigger',
@@ -67,24 +58,43 @@ export const translations = [
         'description placeholder': 'Enter description of when this template should be used',
         'select preset template placeholder': 'Select preset',
         'service registry label': 'Service Registry',
+        'type filter placeholder': 'Type',
+        'any type filter label': 'Any',
 
         'sequence logs label': 'Logs',
 
         'assistant dialog title': 'Assistant',
         'open assistant label': 'Open assistant',
 
-        'no tools': 'No tools are configured.',
+        'no tools': 'No tools are configured',
+        'no results': 'No results',
 
+        'button retry': 'Retry',
+        'button suggestion': 'Assistant suggestion',
+        'button select option': 'Select option',
+        'button cancel': 'Cancel',
         'button save': 'Save',
         'button run': 'Run',
+        'button new thread': 'New Chat',
+        'chat history label': 'Chat History',
+
+        'toolkit label': 'Toolkit',
+        'summary label': 'Summary',
+
+        'calling tool label': 'Calling tool...',
+        'calling label': 'Calling',
+        'error label': 'Error',
 
         'search placeholder': 'Search...',
         'prompt placeholder': 'Ask or search for anything...',
-        'context objects placeholder': 'Add to context...',
-        'button add blueprint': 'Configure blueprints',
+        'button context objects': 'Add to context',
+        'button context settings': 'Chat settings',
         'button microphone': 'Click to speak',
         'button cancel processing': 'Stop processing',
-        'button scroll down': 'Scroll down',
+        'blueprints in context title': 'Blueprints',
+        'objects in context title': 'Content',
+        'remove object in context label': 'Remove document',
+        'chat model title': 'Model',
 
         'settings default label': 'Default',
         'settings custom prompts label': 'Use custom prompts',

package/src/types/Assistant.ts

@@ -0,0 +1,61 @@
+//
+// Copyright 2024 DXOS.org
+//
+
+import { Schema } from 'effect';
+
+import { Queue } from '@dxos/client/echo';
+import { Type } from '@dxos/echo';
+import { LabelAnnotation } from '@dxos/echo-schema';
+
+import { LLM_PROVIDERS } from './defs';
+
+/**
+ * AI chat.
+ */
+export const Chat = Schema.Struct({
+  id: Type.ObjectId,
+  name: Schema.optional(Schema.String),
+  queue: Type.Ref(Queue),
+  // TODO(dmaretskyi): Eventually this and the message queue will be the same.
+  traceQueue: Schema.optional(Type.Ref(Queue)),
+}).pipe(
+  Type.Obj({
+    typename: 'dxos.org/type/assistant/Chat',
+    version: '0.2.0',
+  }),
+  LabelAnnotation.set(['name']),
+);
+
+export interface Chat extends Schema.Schema.Type<typeof Chat> {}
+
+/**
+ * Relation between a Chat and companion objects (e.g., artifacts).
+ */
+export const CompanionTo = Schema.Struct({
+  id: Type.ObjectId,
+}).pipe(
+  Type.Relation({
+    typename: 'dxos.org/relation/assistant/CompanionTo',
+    version: '0.1.0',
+    source: Chat,
+    target: Type.Expando,
+  }),
+);
+
+export interface CompanionTo extends Schema.Schema.Type<typeof CompanionTo> {}
+
+/**
+ * Plugin settings.
+ */
+export const Settings = Schema.mutable(
+  Schema.Struct({
+    llmProvider: Schema.optional(Schema.Literal(...LLM_PROVIDERS)),
+    edgeModel: Schema.optional(Schema.String),
+    ollamaModel: Schema.optional(Schema.String),
+    lmstudioModel: Schema.optional(Schema.String),
+    customPrompts: Schema.optional(Schema.Boolean),
+  }),
+);
+
+export type Settings = Schema.Schema.Type<typeof Settings>;

package/src/types/AssistantAction.ts

@@ -0,0 +1,62 @@
+//
+// Copyright 2024 DXOS.org
+//
+
+import { Schema } from 'effect';
+
+import { Blueprint } from '@dxos/blueprints';
+import { EchoObjectSchema, SpaceSchema } from '@dxos/client/echo';
+import { Sequence } from '@dxos/conductor';
+import { DataType } from '@dxos/schema';
+
+import { meta } from '../meta';
+
+import { Chat } from './Assistant';
+
+export class OnSpaceCreated extends Schema.TaggedClass<OnSpaceCreated>()(`${meta.id}/on-space-created`, {
+  input: Schema.Struct({
+    space: SpaceSchema,
+    rootCollection: DataType.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 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 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,
+  }),
+  output: Schema.Void,
+}) {}

package/src/types/defs.ts

@@ -0,0 +1,5 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+export const LLM_PROVIDERS = ['edge', 'ollama', 'lmstudio'] as const;

package/src/types/index.ts

@@ -2,5 +2,8 @@
 // Copyright 2024 DXOS.org
 //
 
-export * from './schema';
+export * as Assistant from './Assistant';
+export * as AssistantAction from './AssistantAction';
+
+export * from './defs';
 export * from './service';