@dxos/plugin-assistant 0.8.4-main.fcfe5033a5 → 0.8.4-staging.60fe92afc8

package/src/testing/{trace-timeline.test.ts → trace-timeline.node.test.ts}

@@ -5,7 +5,6 @@
 import { describe, it } from '@effect/vitest';
 import * as Effect from 'effect/Effect';
 
-import { AgentService } from '@dxos/assistant';
 import {
   AgentHandlers,
   AgentPrompt,
@@ -15,27 +14,27 @@ import {
   WebSearchHandlers,
   WebSearchToolkitOpaque,
 } from '@dxos/assistant-toolkit';
-import { AssistantTestLayerWithTriggers } from '@dxos/assistant/testing';
-import { Blueprint, Prompt } from '@dxos/blueprints';
+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 { ExampleHandlers, Reply, Trace, Trigger } from '@dxos/functions';
+import { AgentService } from '@dxos/functions-runtime';
 import { FeedTraceSink, TriggerDispatcher } from '@dxos/functions-runtime';
-import { ObjectId } from '@dxos/keys';
+import { AssistantTestLayerWithTriggers } from '@dxos/functions-runtime/testing';
+import { EntityId } from '@dxos/keys';
 import { dbg } from '@dxos/log';
-import { Operation } from '@dxos/operation';
 import { renderTimelineAscii } from '@dxos/react-ui-components';
 import { Organization, Person } from '@dxos/types';
 
-import { buildExecutionGraph } from '../containers/TracePanel/execution-graph';
+import { buildExecutionGraph } from '#execution-graph';
 
-ObjectId.dangerouslyDisableRandomness();
+EntityId.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));
+  return yield* Database.query(Query.select(Filter.type(Trace.Message)).from(feed)).run;
 });
 
 const TestLayer = AssistantTestLayerWithTriggers({
@@ -47,7 +46,8 @@ const TestLayer = AssistantTestLayerWithTriggers({
   aiServicePreset: 'edge-remote',
 });
 
-describe('Trace timeline', () => {
+// TODO(dmaretskyi): Flaky snapshots
+describe.skip('Trace timeline', () => {
   describe('Agent', () => {
     it.effect(
       'create objects via AgentService',
@@ -71,15 +71,13 @@ describe('Trace timeline', () => {
             "
             ●     [atom] Agent processing request...
             ├──●  [user] Create an organization called "Cyberdyne Systems".
-            │  ●  [check-circle] list-schemas - Success
-            │  ●  [check-circle] create-object - Success
-            │  ●  [check-circle] add-to-context - Success
-            ◆──╯  [check-circle] Agent completed request
+            │  ●  [list] List schemas - Success
+            │  ●  [plus] Create object - Success
+            ◆──╯  [atom] Agent completed request
             ●  │  [atom] Agent processing request...
             │  ●  [user] Create a person named "John Connor".
-            │  ●  [check-circle] create-object - Success
-            │  ●  [check-circle] add-to-context - Success
-            ◆──╯  [check-circle] Agent completed request
+            │  ●  [plus] Create object - Success
+            ◆──╯  [atom] Agent completed request
             "
           `);
         },
@@ -108,10 +106,10 @@ describe('Trace timeline', () => {
             "
             ●     [atom] Agent processing request...
             ├──●  [user] Search for all organizations. How many are there?
-            │  ●  [check-circle] list-schemas - Success
-            │  ●  [check-circle] query - Success
-            │  ●  [check-circle] query - Success
-            ◆──╯  [check-circle] Agent completed request
+            │  ●  [list] List schemas - Success
+            │  ●  [magnifying-glass] Query - Success
+            │  ●  [magnifying-glass] Query - Success
+            ◆──╯  [atom] Agent completed request
             "
           `);
         },
@@ -141,23 +139,13 @@ describe('Trace timeline', () => {
           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.
-            │  ●  [check-circle] list-schemas - Success
-            ◆──╯  [check-circle] Agent completed request
-            ●  │  [atom] Agent processing request...
-            │  ●  [user] Create an organization called "DXOS" and a person named "Alice".
-            │  ●  [check-circle] create-object - Success
-            │  ●  [check-circle] create-object - Success
-            ◆──╯  [check-circle] Agent completed request
-            ●  │  [atom] Agent processing request...
-            │  ●  [user] Search for all organizations and persons.
-            │  ●  [check-circle] query - Success
-            │  ●  [check-circle] query - Success
-            ◆──╯  [check-circle] Agent completed request
-            "
-          `);
+              "
+              ●     [atom] Agent processing request...
+              ├──●  [user] Search for all organizations. How many are there?
+              │  ●  [magnifying-glass] Query - Success
+              ◆──╯  [atom] Agent completed request
+              "
+            `);
         },
         Effect.provide(TestLayer),
         TestHelpers.provideTestContext,
@@ -178,7 +166,7 @@ describe('Trace timeline', () => {
             }),
           ]);
           const prompt = yield* Database.add(
-            Prompt.make({
+            Routine.make({
               name: 'Research',
               instructions: 'Research the given topic, or object.',
               blueprints: [Ref.make(yield* Blueprint.upsert(WebSearchBlueprint.key))],
@@ -198,17 +186,24 @@ describe('Trace timeline', () => {
 
           const dispatcher = yield* TriggerDispatcher;
           yield* dispatcher
-            .invokeScheduledTriggers({ kinds: ['queue'], untilExhausted: true })
+            .invokeScheduledTriggers({ kinds: ['feed'], 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(`
-            "
-            ●  [check-circle] Agent
-            "
-          `);
+                "
+                ●     [brain] Run Routine
+                ├──●  [user] Research the given topic, or object.
+                │  ●  [wrench] AnthropicWebSearch - Success
+                │  ●  [wrench] AnthropicWebSearch - Success
+                │  ●  [wrench] AnthropicWebSearch - Success
+                │  ●  [wrench] AnthropicWebSearch - Success
+                │  ●  [wrench] completeJob - Success
+                ◆──╯  [brain] Run Routine - Success
+                "
+              `);
         },
         Effect.provide(TestLayer),
         TestHelpers.provideTestContext,
@@ -230,12 +225,12 @@ describe('Trace timeline', () => {
           const { commits, branches } = buildExecutionGraph({ traceMessages: messages });
           const graph = renderTimelineAscii(commits, branches);
           expect(`\n${graph}\n`).toMatchInlineSnapshot(`
-            "
-            ●  [check-circle] Reply
-            ●  [check-circle] Reply
-            ●  [check-circle] Reply
-            "
-          `);
+                  "
+                  ●  [function] Reply - Success
+                  ●  [function] Reply - Success
+                  ●  [function] Reply - Success
+                  "
+                `);
         },
         Effect.provide(TestLayer),
         TestHelpers.provideTestContext,

package/src/testing.ts

@@ -0,0 +1,9 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+// TODO(dmaretskyi): ./testing dir vs testing.ts module?
+
+// Eager re-export of `AssistantPlugin`. See `@dxos/plugin-testing/src/core.ts`
+// for the rationale.
+export * from './AssistantPlugin';

package/src/translations.ts

@@ -3,12 +3,12 @@
 //
 
 import { Chat, Agent, McpServer } from '@dxos/assistant-toolkit';
-import { Blueprint, Prompt } from '@dxos/blueprints';
+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';
-import { translations as formTranslations } from '@dxos/react-ui-form';
+import { translations as componentsTranslations } from '@dxos/react-ui-components/translations';
+import { translations as formTranslations } from '@dxos/react-ui-form/translations';
 
 import { meta } from '#meta';
 
@@ -18,8 +18,7 @@ export const translations: Resource[] = [
   ...formTranslations,
   {
     'en-US': {
-      // TODO(burdon): From assistant.
-      [Blueprint.Blueprint.typename]: {
+      [Type.getTypename(Blueprint.Blueprint)]: {
         'typename.label': 'Blueprint',
         'typename.label_zero': 'Blueprints',
         'typename.label_one': 'Blueprint',
@@ -30,19 +29,18 @@ export const translations: Resource[] = [
         '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]: {
+      [Type.getTypename(Sequence.Sequence)]: {
         'typename.label': 'Sequence',
         'typename.label_zero': 'Sequences',
         'typename.label_one': 'Sequence',
@@ -53,21 +51,24 @@ export const translations: Resource[] = [
         'delete-object.label': 'Delete sequence',
         'object-deleted.label': 'Sequence deleted',
       },
-      [Chat.Chat.typename]: {
+      [Type.getTypename(Chat.Chat)]: {
         'typename.label': 'AI Chat',
+        'typename.label_zero': 'AI Chats',
+        'typename.label_one': 'AI Chat',
+        'typename.label_other': 'AI Chats',
         'object-name.placeholder': 'New AI Chat',
         'add-object.label': 'Add AI chat',
         'rename-object.label': 'Rename AI Chat',
         'delete-object.label': 'Delete AI Chat',
         'object-deleted.label': 'AI Chat deleted',
       },
-      [McpServer.McpServer.typename]: {
+      [Type.getTypename(McpServer.McpServer)]: {
         'typename.label': 'MCP Server',
         'typename.label_zero': 'MCP Servers',
         'typename.label_one': 'MCP Server',
         'typename.label_other': 'MCP Servers',
       },
-      [Agent.Agent.typename]: {
+      [Type.getTypename(Agent.Agent)]: {
         'typename.label': 'Agent',
         'typename.label_zero': 'Agents',
         'typename.label_one': 'Agent',
@@ -109,7 +110,8 @@ export const translations: Resource[] = [
 
         'assistant-dialog.title': 'Assistant',
         'open-assistant.label': 'Open assistant',
-        'reset-blueprints.label': 'Reset blueprints',
+        'import-compute-operations.label': 'Import compute operations',
+        'toggle-trace-panel-debug.label': 'Toggle trace panel debug view',
 
         'no-results.message': 'No results',
 
@@ -119,6 +121,7 @@ export const translations: Resource[] = [
         'rename-thread.button': 'Rename Chat',
         'chat-history.label': 'Chat History',
         'chat-update-name.label': 'Update AI Chat name',
+        'create-chat.label': 'New AI Chat',
 
         'toolkit.label': 'Toolkit',
         'stats.label': 'Stats',
@@ -131,11 +134,17 @@ export const translations: Resource[] = [
         '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': 'Models',
-        'mcp-servers.title': 'MCP',
+
+        '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',
@@ -145,28 +154,21 @@ export const translations: Resource[] = [
         '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',
+        'ai-service-error.label': 'AI service error',
 
         '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',
 
-        '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',
-
         // 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',
@@ -177,7 +179,7 @@ export const translations: Resource[] = [
         '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',
-        'input-queue.label': 'Inputs',
+        'inputs.label': 'Inputs',
 
         // AgentProperties.
         'instructions.label': 'Instructions',

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,32 +2,23 @@
 // 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 { Collection, Database, DXN, Feed, Obj, Ref } from '@dxos/echo';
-import { Operation } from '@dxos/operation';
+import { Routine, Operation } from '@dxos/compute';
+import { Database, Feed, Obj, Ref, Type } from '@dxos/echo';
+import { DXN } from '@dxos/keys';
 
 import { meta } from '#meta';
 
-const ASSISTANT_OPERATION = `${meta.id}.operation`;
-
-export const OnCreateSpace = Operation.make({
-  meta: { key: `${ASSISTANT_OPERATION}.on-create-space`, name: 'On Create Space' },
-  services: [Capability.Service],
-  input: Schema.Struct({
-    space: SpaceSchema,
-    rootCollection: Collection.Collection,
-  }),
-  output: Schema.Void,
-});
+const makeKey = (name: string) => DXN.make(`${meta.id}.operation.${name}`);
 
 export const CreateChat = Operation.make({
-  meta: { key: `${ASSISTANT_OPERATION}.create-chat`, name: 'Create Chat' },
+  meta: { key: makeKey('createChat'), name: 'Create Chat', icon: 'ph--chat-text--regular' },
   services: [Capability.Service],
   input: Schema.Struct({
     db: Database.Database,
@@ -36,31 +27,45 @@ export const CreateChat = Operation.make({
     addToSpace: Schema.optional(Schema.Boolean),
   }),
   output: Schema.Struct({
-    object: Chat.Chat,
+    object: Type.getSchema(Chat.Chat),
   }),
 });
 
 export const UpdateChatName = Operation.make({
-  meta: { key: `${ASSISTANT_OPERATION}.update-chat-name`, name: 'Update Chat Name' },
+  meta: {
+    key: makeKey('updateChatName'),
+    name: 'Update Chat Name',
+    icon: 'ph--pencil--regular',
+  },
   services: [Database.Service, Feed.FeedService, AiService.AiService],
   input: Schema.Struct({
-    chat: Chat.Chat,
+    chat: Type.getSchema(Chat.Chat),
+    /** Initial user message text; used when the feed has no history yet (e.g. auto-rename on first send). */
+    prompt: Schema.optional(Schema.String),
   }),
   output: Schema.Void,
 });
 
 export const SetCurrentChat = Operation.make({
-  meta: { key: `${ASSISTANT_OPERATION}.set-current-chat`, name: 'Set Current Chat' },
+  meta: {
+    key: makeKey('setCurrentChat'),
+    name: 'Set Current Chat',
+    icon: 'ph--chat-text--regular',
+  },
   services: [Capability.Service],
   input: Schema.Struct({
     companionTo: Obj.Unknown,
-    chat: Chat.Chat.pipe(Schema.optional),
+    chat: Type.getSchema(Chat.Chat).pipe(Schema.optional),
   }),
   output: Schema.Void,
 });
 
 export const RunPromptInNewChat = Operation.make({
-  meta: { key: `${ASSISTANT_OPERATION}.run-prompt-in-new-chat`, name: 'Run Prompt In New Chat' },
+  meta: {
+    key: makeKey('runPromptInNewChat'),
+    name: 'Run Prompt In New Chat',
+    icon: 'ph--chat-text--regular',
+  },
   services: [Capability.Service],
   input: Schema.Struct({
     db: Database.Database,
@@ -68,15 +73,15 @@ 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).
      */
     background: Schema.optional(Schema.Boolean),
   }),
   output: Schema.Struct({
-    object: Chat.Chat,
+    object: Type.getSchema(Chat.Chat),
   }),
 });
 
@@ -88,15 +93,16 @@ const NavigationTargetSchema = Schema.Struct({
 
 export const ResolveNavigationTargets = Operation.make({
   meta: {
-    key: `${ASSISTANT_OPERATION}.resolve-navigation-targets`,
+    key: makeKey('resolveNavigationTargets'),
     name: 'Resolve navigation targets',
     description:
       'Resolve navigation targets within the application. The returned paths can be used with the Open operation. Without a query, returns pages that can be navigated to.',
+    icon: 'ph--compass--regular',
   },
   input: Schema.Struct({
     query: Schema.optional(
       Schema.Struct({
-        dxn: DXN.Schema.pipe(Schema.optional),
+        dxn: Schema.optional(DXN.Schema),
       }),
     ),
   }),
@@ -106,15 +112,32 @@ export const ResolveNavigationTargets = Operation.make({
   services: [Capability.Service],
 });
 
+export const ForkChat = Operation.make({
+  meta: { key: makeKey('forkChat'), name: 'Fork Chat', icon: 'ph--git-branch--regular', skipRegistry: true },
+  services: [Capability.Service, Database.Service],
+  input: Schema.Struct({
+    chat: Type.getSchema(Chat.Chat),
+    /** When provided, the forked chat is set as the companion for this object rather than opened standalone. */
+    companionTo: Schema.optional(Obj.Unknown),
+  }),
+  output: Schema.Struct({
+    object: Type.getSchema(Chat.Chat),
+  }),
+});
+
 export const EnsureCompanionChat = Operation.make({
-  meta: { key: `${ASSISTANT_OPERATION}.ensure-companion-chat`, name: 'Ensure Companion Chat' },
+  meta: {
+    key: makeKey('ensureCompanionChat'),
+    name: 'Ensure Companion Chat',
+    icon: 'ph--chat-text--regular',
+  },
   services: [Capability.Service],
   input: Schema.Struct({
     db: Database.Database,
     companionTo: Obj.Unknown,
   }),
   output: Schema.Struct({
-    chat: Chat.Chat,
+    chat: Type.getSchema(Chat.Chat),
     /** Whether the returned chat was already persisted in the space. */
     persisted: Schema.Boolean,
   }),
@@ -125,3 +148,17 @@ export const BlueprintForm = Schema.Struct({
   name: Schema.String,
   description: Schema.optional(Schema.String),
 });
+
+export const ToggleTracePanelDebug = Operation.make({
+  meta: {
+    key: makeKey('toggleTracePanelDebug'),
+    name: 'Toggle trace panel debug',
+    description: 'Toggle trace panel between commit graph and raw span tree JSON.',
+    icon: 'ph--bug--regular',
+  },
+  services: [Capability.Service],
+  input: Schema.Struct({
+    state: Schema.optional(Schema.Boolean),
+  }),
+  output: Schema.Boolean,
+});