@dxos/plugin-assistant 0.8.4-main.74a063c4e0 → 0.8.4-main.765dc60934

package/src/testing/trace-timeline.test.ts

@@ -0,0 +1,250 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { describe, it } from '@effect/vitest';
+import * as Effect from 'effect/Effect';
+
+import {
+  AgentHandlers,
+  AgentPrompt,
+  DatabaseBlueprint,
+  DatabaseHandlers,
+  WebSearchBlueprint,
+  WebSearchHandlers,
+  WebSearchToolkitOpaque,
+} from '@dxos/assistant-toolkit';
+import { Blueprint, Routine, Trace, Trigger, Operation } from '@dxos/compute';
+import { ExampleHandlers, Reply } from '@dxos/compute/testing';
+import { Database, Feed, Filter, Obj, Query, Ref } from '@dxos/echo';
+import { TestHelpers } from '@dxos/effect/testing';
+import { AgentService } from '@dxos/functions-runtime';
+import { FeedTraceSink, TriggerDispatcher } from '@dxos/functions-runtime';
+import { AssistantTestLayerWithTriggers } from '@dxos/functions-runtime/testing';
+import { ObjectId } from '@dxos/keys';
+import { dbg } from '@dxos/log';
+import { renderTimelineAscii } from '@dxos/react-ui-components';
+import { Organization, Person } from '@dxos/types';
+
+import { buildExecutionGraph } from '../containers/TracePanel/execution-graph';
+
+ObjectId.dangerouslyDisableRandomness();
+
+const queryTraceMessages = Effect.gen(function* () {
+  yield* FeedTraceSink.flush();
+  yield* Database.flush();
+  const feed = yield* FeedTraceSink.getOrCreateTraceFeed();
+  return yield* Database.runQuery(Query.select(Filter.type(Trace.Message)).from(feed));
+});
+
+const TestLayer = AssistantTestLayerWithTriggers({
+  types: [Organization.Organization, Person.Person],
+  blueprints: [DatabaseBlueprint.make(), WebSearchBlueprint.make()],
+  operationHandlers: [DatabaseHandlers, AgentHandlers, WebSearchHandlers, ExampleHandlers],
+  toolkits: [WebSearchToolkitOpaque],
+  tracing: 'feed',
+  aiServicePreset: 'edge-remote',
+});
+
+describe('Trace timeline', () => {
+  describe('Agent', () => {
+    it.effect(
+      'create objects via AgentService',
+      Effect.fnUntraced(
+        function* ({ expect }) {
+          const agent = yield* AgentService.createSession({
+            blueprints: [DatabaseBlueprint.make()],
+          });
+          yield* agent.submitPrompt('Create an organization called "Cyberdyne Systems".');
+          yield* agent.waitForCompletion();
+
+          yield* agent.submitPrompt('Create a person named "John Connor".');
+          yield* agent.waitForCompletion();
+
+          const messages = yield* queryTraceMessages;
+          dbg(messages.flatMap(({ meta, events }) => events.map((event) => ({ ...meta, ...event }))));
+          const { commits, branches } = buildExecutionGraph({ traceMessages: messages });
+          dbg(branches);
+          const graph = renderTimelineAscii(commits, branches);
+          expect(`\n${graph}\n`).toMatchInlineSnapshot(`
+            "
+            ●     [atom] Agent processing request...
+            ├──●  [user] Create an organization called "Cyberdyne Systems".
+            │  ●  [function] List schemas - Success
+            │  ●  [function] Create object - Success
+            │  ●  [function] Add to context - Success
+            ◆──╯  [atom] Agent completed request
+            ●  │  [atom] Agent processing request...
+            │  ●  [user] Create a person named "John Connor".
+            │  ●  [function] Create object - Success
+            │  ●  [function] Add to context - Success
+            ◆──╯  [atom] Agent completed request
+            "
+          `);
+        },
+        Effect.provide(TestLayer),
+        TestHelpers.provideTestContext,
+      ),
+      { timeout: 120_000 },
+    );
+
+    it.effect(
+      'invoke prompt and query schema',
+      Effect.fnUntraced(
+        function* ({ expect }) {
+          const agent = yield* AgentService.createSession({
+            blueprints: [DatabaseBlueprint.make()],
+          });
+          yield* Database.add(Obj.make(Organization.Organization, { name: 'Acme Corp' }));
+          yield* Database.add(Obj.make(Organization.Organization, { name: 'Globex Industries' }));
+          yield* agent.submitPrompt('Search for all organizations. How many are there?');
+          yield* agent.waitForCompletion();
+
+          const messages = yield* queryTraceMessages;
+          const { commits, branches } = buildExecutionGraph({ traceMessages: messages });
+          const graph = renderTimelineAscii(commits, branches);
+          expect(`\n${graph}\n`).toMatchInlineSnapshot(`
+              "
+              ●     [atom] Agent processing request...
+              ├──●  [user] Search for all organizations. How many are there?
+              │  ●  [function] List schemas - Success
+              │  ●  [function] Query - Success
+              │  ●  [function] Query - Success
+              ◆──╯  [atom] Agent completed request
+              "
+            `);
+        },
+        Effect.provide(TestLayer),
+        TestHelpers.provideTestContext,
+      ),
+      { timeout: 120_000 },
+    );
+
+    it.effect(
+      'sequential prompts in a session',
+      Effect.fnUntraced(
+        function* ({ expect }) {
+          const agent = yield* AgentService.createSession({
+            blueprints: [DatabaseBlueprint.make()],
+          });
+          yield* agent.submitPrompt('List all available schemas. Tell me what typenames are available.');
+          yield* agent.waitForCompletion();
+
+          yield* agent.submitPrompt('Create an organization called "DXOS" and a person named "Alice".');
+          yield* agent.waitForCompletion();
+
+          yield* agent.submitPrompt('Search for all organizations and persons.');
+          yield* agent.waitForCompletion();
+
+          const messages = yield* queryTraceMessages;
+          const { commits, branches } = buildExecutionGraph({ traceMessages: messages });
+          const graph = renderTimelineAscii(commits, branches);
+          expect(`\n${graph}\n`).toMatchInlineSnapshot(`
+                "
+                ●     [atom] Agent processing request...
+                ├──●  [user] List all available schemas. Tell me what typenames are available.
+                │  ●  [function] List schemas - Success
+                ◆──╯  [atom] Agent completed request
+                ●  │  [atom] Agent processing request...
+                │  ●  [user] Create an organization called "DXOS" and a person named "Alice".
+                │  ●  [function] Create object - Success
+                │  ●  [function] Create object - Success
+                ◆──╯  [atom] Agent completed request
+                ●  │  [atom] Agent processing request...
+                │  ●  [user] Search for all organizations and persons.
+                │  ●  [function] Query - Success
+                │  ●  [function] Query - Success
+                ◆──╯  [atom] Agent completed request
+                "
+              `);
+        },
+        Effect.provide(TestLayer),
+        TestHelpers.provideTestContext,
+      ),
+      { timeout: 180_000 },
+    );
+  });
+
+  describe('Triggers', () => {
+    it.effect(
+      'prompt trigger',
+      Effect.fnUntraced(
+        function* ({ expect }) {
+          const feed = yield* Database.add(Feed.make());
+          yield* Feed.append(feed, [
+            Obj.make(Organization.Organization, {
+              name: 'DXOS',
+            }),
+          ]);
+          const prompt = yield* Database.add(
+            Routine.make({
+              name: 'Research',
+              instructions: 'Research the given topic, or object.',
+              blueprints: [Ref.make(yield* Blueprint.upsert(WebSearchBlueprint.key))],
+            }),
+          );
+          yield* Database.add(
+            Trigger.make({
+              function: Ref.make(Operation.serialize(AgentPrompt)),
+              enabled: true,
+              spec: Trigger.specFeed(feed),
+              input: {
+                prompt: Ref.make(prompt),
+                input: '{{event.item}}',
+              },
+            }),
+          );
+
+          const dispatcher = yield* TriggerDispatcher;
+          yield* dispatcher
+            .invokeScheduledTriggers({ kinds: ['queue'], untilExhausted: true })
+            .pipe(Effect.flatMap(Effect.forEach((result) => result.result)));
+
+          const messages = yield* queryTraceMessages;
+          const { commits, branches } = buildExecutionGraph({ traceMessages: messages });
+          const graph = renderTimelineAscii(commits, branches);
+          expect(`\n${graph}\n`).toMatchInlineSnapshot(`
+                  "
+                  ●     [function] Run Routine
+                  ├──●  [user] Research the given topic, or object.
+                  ●──┤  [function] Run Routine - Success
+                  "
+                `);
+        },
+        Effect.provide(TestLayer),
+        TestHelpers.provideTestContext,
+      ),
+      { timeout: 120_000 },
+    );
+  });
+
+  describe('Operations', () => {
+    it.effect(
+      'operation',
+      Effect.fnUntraced(
+        function* ({ expect }) {
+          yield* Operation.invoke(Reply, { data: 'test-1' });
+          yield* Operation.invoke(Reply, { data: 'test-2' });
+          yield* Operation.invoke(Reply, { data: 'test-3' });
+
+          const messages = yield* queryTraceMessages;
+          const { commits, branches } = buildExecutionGraph({ traceMessages: messages });
+          const graph = renderTimelineAscii(commits, branches);
+          expect(`\n${graph}\n`).toMatchInlineSnapshot(`
+                  "
+                  ●  [function] Reply
+                  ●  [function] Reply - Success
+                  ●  [function] Reply
+                  ●  [function] Reply - Success
+                  ●  [function] Reply
+                  ●  [function] Reply - Success
+                  "
+                `);
+        },
+        Effect.provide(TestLayer),
+        TestHelpers.provideTestContext,
+      ),
+      { timeout: 120_000 },
+    );
+  });
+});

package/src/testing.ts

@@ -0,0 +1,7 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+// Eager re-export of `AssistantPlugin`. See `@dxos/plugin-testing/src/core.ts`
+// for the rationale.
+export * from './AssistantPlugin';

package/src/translations.ts

@@ -2,19 +2,22 @@
 // Copyright 2023 DXOS.org
 //
 
-import { Chat, Project } from '@dxos/assistant-toolkit';
-import { Blueprint, Prompt } from '@dxos/blueprints';
+import { Chat, Agent, McpServer } from '@dxos/assistant-toolkit';
+import { Blueprint, Routine } from '@dxos/compute';
 import { Sequence } from '@dxos/conductor';
 import { Type } from '@dxos/echo';
 import { type Resource } from '@dxos/react-ui';
+import { translations as componentsTranslations } from '@dxos/react-ui-components/translations';
+import { translations as formTranslations } from '@dxos/react-ui-form/translations';
 
 import { meta } from '#meta';
 
 // TODO(burdon): Standardize translation names.
-export const translations = [
+export const translations: Resource[] = [
+  ...componentsTranslations,
+  ...formTranslations,
   {
     'en-US': {
-      // TODO(burdon): From assistant.
       [Blueprint.Blueprint.typename]: {
         'typename.label': 'Blueprint',
         'typename.label_zero': 'Blueprints',
@@ -26,19 +29,18 @@ export const translations = [
         'delete-object.label': 'Delete blueprint',
         'object-deleted.label': 'Blueprint deleted',
       },
-      [Type.getTypename(Prompt.Prompt)]: {
-        'typename.label': 'Prompt',
-        'typename.label_zero': 'Prompts',
-        'typename.label_one': 'Prompt',
-        'typename.label_other': 'Prompts',
-        'object-name.placeholder': 'New prompt',
-        'add-object.label': 'Add prompt',
-        'rename-object.label': 'Rename prompt',
-        'delete-object.label': 'Delete prompt',
-        'object-deleted.label': 'Prompt deleted',
+      [Type.getTypename(Routine.Routine)]: {
+        'typename.label': 'Routine',
+        'typename.label_zero': 'Routines',
+        'typename.label_one': 'Routine',
+        'typename.label_other': 'Routines',
+        'object-name.placeholder': 'New routine',
+        'add-object.label': 'Add routine',
+        'rename-object.label': 'Rename routine',
+        'delete-object.label': 'Delete routine',
+        'object-deleted.label': 'Routine deleted',
       },
-      // TODO(burdon): From conductor.
-      [Sequence.typename]: {
+      [Sequence.Sequence.typename]: {
         'typename.label': 'Sequence',
         'typename.label_zero': 'Sequences',
         'typename.label_one': 'Sequence',
@@ -57,16 +59,22 @@ export const translations = [
         'delete-object.label': 'Delete AI Chat',
         'object-deleted.label': 'AI Chat deleted',
       },
-      [Project.Project.typename]: {
-        'typename.label': 'Project',
-        'typename.label_zero': 'Projects',
-        'typename.label_one': 'Project',
-        'typename.label_other': 'Projects',
-        'object-name.placeholder': 'New project',
-        'add-object.label': 'Add project',
-        'rename-object.label': 'Rename project',
-        'delete-object.label': 'Delete project',
-        'object-deleted.label': 'Project deleted',
+      [McpServer.McpServer.typename]: {
+        'typename.label': 'MCP Server',
+        'typename.label_zero': 'MCP Servers',
+        'typename.label_one': 'MCP Server',
+        'typename.label_other': 'MCP Servers',
+      },
+      [Agent.Agent.typename]: {
+        'typename.label': 'Agent',
+        'typename.label_zero': 'Agents',
+        'typename.label_one': 'Agent',
+        'typename.label_other': 'Agents',
+        'object-name.placeholder': 'New agent',
+        'add-object.label': 'Add agent',
+        'rename-object.label': 'Rename agent',
+        'delete-object.label': 'Delete agent',
+        'object-deleted.label': 'Agent deleted',
       },
       // TODO(burdon): Reconcile with react-ui-chat.
       [meta.id]: {
@@ -90,7 +98,7 @@ export const translations = [
         'type-filter.placeholder': 'Type',
         'any-type-filter.label': 'Any',
         'no-blueprint.message': 'No active blueprints',
-        'tool-call.label': 'Calling tool...',
+        'tool-call.label': 'Calling',
         'tool-result.label': 'Success',
         'tool-error.label': 'Tool call failed',
 
@@ -100,6 +108,7 @@ export const translations = [
         'assistant-dialog.title': 'Assistant',
         'open-assistant.label': 'Open assistant',
         'reset-blueprints.label': 'Reset blueprints',
+        'import-compute-operations.label': 'Import compute operations',
 
         'no-results.message': 'No results',
 
@@ -113,6 +122,7 @@ export const translations = [
         'toolkit.label': 'Toolkit',
         'stats.label': 'Stats',
         'summary.label': 'Summary',
+        'thinking.label': 'Thinking',
 
         'search.placeholder': 'Search...',
         'prompt.placeholder': 'Enter question or command...',
@@ -120,43 +130,58 @@ export const translations = [
         'context-settings.button': 'Chat settings',
         'microphone.button': 'Click to speak',
         'cancel-processing.button': 'Stop processing',
-        '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',
-        'settings.custom-prompts.description': 'Allow the assistant to use custom prompts defined in your spaces.',
-        'settings.llm-provider.label': 'LLM provider',
-        'settings.llm-provider.description': 'Select which language model service to use for AI responses.',
-        'settings.edge-llm-model.label': 'Remote language model',
-        'settings.edge-llm-model.description': 'Choose the remote language model used for AI requests.',
-        'settings.ollama-llm-model.label': 'Ollama language model',
-        'settings.ollama-llm-model.description': 'Choose the locally hosted Ollama model for AI requests.',
-        'settings.default-llm-model.label': 'Default language model',
+        'options.blueprints.title': 'Skills',
+        'options.mcp.title': 'MCP',
+        'options.chat-model.title': 'Models',
+        'remove-object.label': 'Remove object',
+
+        'chat-view.title': 'View',
+        'chat-view.normal.label': 'Normal',
+        'chat-view.summary.label': 'Summary',
+        'chat-view.thinking.label': 'Thinking',
+        'chat-view.debug.label': 'Debug',
+        'mcp-server-add.label': 'Add MCP server',
+        'mcp-server-remove.label': 'Remove MCP server',
+        'mcp-server-name.label': 'Server name',
+        'mcp-server-name.placeholder': 'Name',
+        'mcp-server-url.label': 'Server URL',
+        'mcp-server-url.placeholder': 'https://...',
+        'mcp-server-protocol.label': 'Protocol',
+        'mcp-server-api-key.label': 'API key',
+        'mcp-server-api-key.placeholder': 'API key (optional)',
+        'mcp-server-error.label': 'MCP server unavailable',
+
+        'debug.button': 'Debug',
+        'online-switch.label': 'Online',
+        'run-prompt.label': 'Run prompt',
+        'routine-running.label': 'Running…',
+        'typename.label': 'Typename',
+        'branch-thread.menu': 'Branch chat',
+        'chat-toolbar.title': 'Chat toolbar',
 
         // Trigger status
         'trigger-status-disabled.label': 'Triggers disabled',
         'trigger-status-idle.label': 'Triggers idle',
+        'trigger-status-edge.label': 'Triggers will run on EDGE',
         'trigger-status-running.label': 'Trigger running',
         'trigger-status-error.label': 'Trigger error',
         'trigger-runtime.label': 'Auto trigger execution',
         'trigger-last-invocation.label': 'Last run',
         'trigger-duration.label': 'Duration',
 
-        'debug.button': 'Debug',
-        'online-switch.label': 'Online',
-        'run-prompt.label': 'Run prompt',
-        'typename.label': 'Typename',
-        'branch-thread.menu': 'Branch chat',
-        'chat-toolbar.title': 'Chat toolbar',
+        // AgentArticle.
+        'project-empty-spec.message': 'Open the Properties companion to configure the agent.',
+        'project-empty-spec.description': 'Open the Assistant companion to interact with the agent.',
+        'artifacts.label': 'Artifacts',
+        'inputs.label': 'Inputs',
 
-        // ProjectSettings
+        // AgentProperties.
         'instructions.label': 'Instructions',
+        'instructions.placeholder': 'Enter instructions, goals, and constraints for the assistant.',
         'reset-history.button': 'Reset',
         'subscriptions.label': 'Subscriptions',
       },
     },
   },
-] as const satisfies Resource[];
+];

package/src/types/Assistant.ts

@@ -2,9 +2,10 @@
 // Copyright 2024 DXOS.org
 //
 
-import { Chat as ChatModule } from '@dxos/assistant-toolkit';
+// @import-as-namespace
 
 // Re-export Chat schema for backward compatibility.
+import { Chat as ChatModule } from '@dxos/assistant-toolkit';
 export const Chat = ChatModule.Chat;
 export type Chat = ChatModule.Chat;
 
@@ -12,3 +13,14 @@ export type Chat = ChatModule.Chat;
 import * as SettingsModule from './Settings';
 export const Settings = SettingsModule.Settings;
 export type Settings = SettingsModule.Settings;
+
+export const ChatViews = SettingsModule.ChatViews;
+export const ChatView = SettingsModule.ChatView;
+export type ChatView = SettingsModule.ChatView;
+
+export const ModelProviders = SettingsModule.ModelProviders;
+export const ModelProvider = SettingsModule.ModelProvider;
+export type ModelProvider = SettingsModule.ModelProvider;
+
+export const ModelDefaults = SettingsModule.ModelDefaults;
+export type ModelDefaults = SettingsModule.ModelDefaults;

package/src/types/AssistantCapabilities.ts

@@ -0,0 +1,35 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+// @import-as-namespace
+
+import { type Atom } from '@effect-atom/atom-react';
+import * as Schema from 'effect/Schema';
+
+import { Capability } from '@dxos/app-framework';
+import { type Obj } from '@dxos/echo';
+
+import { meta } from '#meta';
+
+import * as Assistant from './Assistant';
+
+export const Settings = Capability.make<Atom.Writable<Assistant.Settings>>(`${meta.id}.capability.settings`);
+
+export const StateSchema = Schema.mutable(
+  Schema.Struct({
+    /** Map of primary object dxn to current chat dxn. */
+    currentChat: Schema.Record({ key: Schema.String, value: Schema.UndefinedOr(Schema.String) }),
+    /** Map of chat object path to prompt text to auto-submit when the chat opens. */
+    pendingPrompts: Schema.Record({ key: Schema.String, value: Schema.UndefinedOr(Schema.String) }),
+  }),
+);
+
+export type AssistantState = Schema.Schema.Type<typeof StateSchema>;
+
+export const State = Capability.make<Atom.Writable<AssistantState>>(`${meta.id}.capability.state`);
+
+/** Session-scoped cache of transient (not yet persisted) companion chats keyed by companion DXN string. */
+export const CompanionChatCache = Capability.make<Atom.Writable<Record<string, Obj.Unknown | undefined>>>(
+  `${meta.id}.capability.companion-chat-cache`,
+);

package/src/types/AssistantEvents.ts

@@ -0,0 +1,11 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+// @import-as-namespace
+
+import { ActivationEvent } from '@dxos/app-framework';
+
+import { meta } from '#meta';
+
+export const SetupAiServiceProviders = ActivationEvent.make(`${meta.id}.event.setup-ai-service-providers`);

package/src/{operations/definitions.ts → types/AssistantOperation.ts}

@@ -2,15 +2,16 @@
 // Copyright 2025 DXOS.org
 //
 
+// @import-as-namespace
+
 import * as Schema from 'effect/Schema';
 
 import { AiService } from '@dxos/ai';
 import { Capability } from '@dxos/app-framework';
 import { Chat } from '@dxos/assistant-toolkit';
-import { Prompt } from '@dxos/blueprints';
 import { SpaceSchema } from '@dxos/client/echo';
+import { Routine, Operation } from '@dxos/compute';
 import { Collection, Database, DXN, Feed, Obj, Ref } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
 
 import { meta } from '#meta';
 
@@ -68,8 +69,8 @@ export const RunPromptInNewChat = Operation.make({
     objects: Schema.optional(Schema.Array(Obj.Unknown)),
     /** Blueprint keys to look up and bind to the new chat. */
     blueprints: Schema.optional(Schema.Array(Schema.String)),
-    /** Raw instructions or an existing Prompt object reference (e.g. from the prompts surface). */
-    prompt: Schema.Union(Schema.String, Ref.Ref(Prompt.Prompt)),
+    /** Raw instructions or an existing Routine object reference. */
+    prompt: Schema.Union(Schema.String, Ref.Ref(Routine.Routine)),
     /**
      * When true, skips opening the chat: runs the Agent prompt operation against the new chat via the compute runtime (traced).
      */

package/src/types/Settings.ts

@@ -6,16 +6,71 @@
 
 import * as Schema from 'effect/Schema';
 
-import { LLM_PROVIDERS } from './defs';
+import { getLiteralValues } from '@dxos/effect';
 
-export const Settings = Schema.mutable(
+export const ModelProvider = Schema.Union(
+  Schema.Literal('edge').annotations({ title: 'DXOS' }),
+  Schema.Literal('ollama').annotations({ title: 'Ollama' }),
+  Schema.Literal('lmstudio').annotations({ title: 'LM Studio' }),
+);
+export type ModelProvider = Schema.Schema.Type<typeof ModelProvider>;
+export const ModelProviders = getLiteralValues(ModelProvider);
+
+export const ChatView = Schema.Union(
+  Schema.Literal('normal').annotations({ title: 'Normal' }),
+  Schema.Literal('summary').annotations({ title: 'Summary' }),
+  Schema.Literal('thinking').annotations({ title: 'Thinking' }),
+  Schema.Literal('debug').annotations({ title: 'Debug' }),
+);
+export type ChatView = Schema.Schema.Type<typeof ChatView>;
+export const ChatViews = getLiteralValues(ChatView);
+
+export const ModelDefaults = 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),
+    edge: Schema.optional(
+      Schema.String.annotations({
+        title: 'Remote language model',
+        description: 'Choose the remote language model used for AI requests.',
+      }),
+    ),
+    ollama: Schema.optional(
+      Schema.String.annotations({
+        title: 'Ollama language model',
+        description: 'Choose the locally hosted Ollama model for AI requests.',
+      }),
+    ),
+    lmstudio: Schema.optional(
+      Schema.String.annotations({
+        title: 'LM Studio language model',
+        description: 'Choose the locally hosted LM Studio model for AI requests.',
+      }),
+    ),
   }),
 );
+export type ModelDefaults = Schema.Schema.Type<typeof ModelDefaults>;
 
-export type Settings = Schema.Schema.Type<typeof Settings>;
+export const Settings = Schema.mutable(
+  Schema.Struct({
+    customPrompts: Schema.optional(
+      Schema.Boolean.annotations({
+        title: 'Use custom prompts',
+        description: 'Allow the assistant to use custom prompts defined in your spaces.',
+      }),
+    ),
+    chatView: Schema.optional(
+      ChatView.annotations({
+        title: 'Chat view',
+        description:
+          'Controls which message blocks are shown in the chat: normal hides reasoning, thinking shows reasoning, debug shows raw blocks, summary shows only conversational text.',
+      }),
+    ),
+    modelProvider: Schema.optional(
+      ModelProvider.annotations({
+        title: 'LLM provider',
+        description: 'Select which language model service to use for AI responses.',
+      }),
+    ),
+    modelDefaults: Schema.optional(ModelDefaults),
+  }),
+);
+export interface Settings extends Schema.Schema.Type<typeof Settings> {}

package/src/types/index.ts

@@ -2,12 +2,14 @@
 // Copyright 2024 DXOS.org
 //
 
+// TODO(wittjosiah): Remove. This is included in Assistant namespace.
 // Re-export Chat type from assistant-toolkit for public API type declarations.
 export type { Chat as ChatType } from '@dxos/assistant-toolkit';
 
 export * as Assistant from './Assistant';
+export * as AssistantOperation from './AssistantOperation';
+export * as AssistantCapabilities from './AssistantCapabilities';
+export * as AssistantEvents from './AssistantEvents';
 
-export * from './capabilities';
-export * from './defs';
-export * from './events';
+// TODO(wittjosiah): Namespace.
 export * from './service';