@dxos/plugin-assistant 0.8.4-main.422d1c7879 → 0.8.4-main.4f23b4e393

package/src/components/ChatThread/widgets/SelectWidget.ts

@@ -6,6 +6,8 @@ import { WidgetType } from '@codemirror/view';
 
 import { Domino } from '@dxos/ui';
 
+import { styles } from './defaults';
+
 /**
  * Simple prompt widget.
  */
@@ -24,7 +26,7 @@ export class SelectWidget extends WidgetType {
   override toDOM() {
     return Domino.of('div')
       .attributes({ role: 'group' })
-      .classNames('flex flex-wrap pt-2 pb-2 gap-1')
+      .classNames(styles.padding, 'flex flex-wrap gap-1')
       .append(
         ...this.options.map((option) =>
           Domino.of('button')

package/src/components/ChatThread/widgets/StatsWidget.ts

@@ -6,6 +6,8 @@ import { WidgetType } from '@codemirror/view';
 
 import { Domino } from '@dxos/ui';
 
+import { styles } from './defaults';
+
 /**
  * Simple stats widget.
  */
@@ -19,7 +21,7 @@ export class StatsWidget extends WidgetType {
   }
 
   override toDOM() {
-    return Domino.of('div').classNames('pt-2 pb-4 text-sm text-placeholder').text(this.text).root;
+    return Domino.of('div').classNames(styles.padding, 'text-sm text-placeholder').text(this.text).root;
   }
 
   override updateDOM(dom: HTMLElement) {

package/src/components/ChatThread/widgets/StatusWidget.ts

@@ -6,38 +6,23 @@ import { WidgetType } from '@codemirror/view';
 
 import { Domino } from '@dxos/ui';
 
-/**
- * Props used to derive a stable key for a reasoning block across CodeMirror widget rebuilds.
- * CodeMirror does not assign persistent IDs to widgets; streaming re-runs the factory each tick.
- */
-export type StatusWidgetBlockIdProps = {
-  id?: string;
-  range?: { from: number; to: number };
-};
+import { styles } from './defaults';
 
 /**
- * AI reasoning widget.
+ * AI status widget.
  */
 export class StatusWidget extends WidgetType {
-  readonly #pos: string;
-  /** The timer id last registered for this instance (may differ from the map if superseded). */
-  #ownedTimerId: ReturnType<typeof setTimeout> | null = null;
-
-  constructor(
-    private readonly text: string,
-    pos: string | number | undefined,
-  ) {
+  constructor(private readonly text: string) {
     super();
-    this.#pos = pos === undefined || pos === '' ? 'status' : String(pos);
   }
 
   override eq(other: this) {
-    return this.text === other.text && this.#pos === other.#pos;
+    return this.text === other.text;
   }
 
   override toDOM() {
     return Domino.of('div')
-      .classNames('py-0.5')
+      .classNames(styles.padding)
       .append(
         Domino.of('div')
           .classNames('relative overflow-hidden rounded-sm')
@@ -55,7 +40,6 @@ export class StatusWidget extends WidgetType {
                       .classNames('relative z-10 rounded-sm text-sm leading-5')
                       .attributes({ 'data-status-text': '' })
                       .text(this.text),
-                    Domino.of('div').attributes({ 'data-id': this.#pos }),
                   ),
               ),
           ),
@@ -63,69 +47,7 @@ export class StatusWidget extends WidgetType {
   }
 
   override updateDOM(dom: HTMLElement) {
-    // Update only the text leaf; `dom` may be wrapped in an outer shell, and the trail host is a
-    // sibling of the text node inside the wrapper — setting `textContent` on the wrong
-    // ancestor would remove `[data-id]`.
     dom.querySelector<HTMLElement>('[data-status-text]')?.replaceChildren(this.text);
-
-    const trailHost = dom.querySelector<HTMLElement>('[data-id]');
-    if (trailHost?.childElementCount === 0) {
-      trailHost.append(...createTrailLayers());
-    }
-
-    this.#scheduleTrailRemoval(dom);
     return true;
   }
-
-  override destroy(_dom: HTMLElement) {
-    this.#clearOwnedTrailTimer();
-  }
-
-  #scheduleTrailRemoval(dom: HTMLElement) {
-    const previous = trailRemovalTimers.get(this.#pos);
-    if (previous !== undefined) {
-      clearTimeout(previous);
-    }
-
-    const timerId = setTimeout(() => {
-      if (trailRemovalTimers.get(this.#pos) !== timerId) {
-        return;
-      }
-      trailRemovalTimers.delete(this.#pos);
-      this.#ownedTimerId = null;
-      dom.querySelector<HTMLElement>('[data-id]')?.remove();
-    }, TRAIL_REMOVAL_DELAY_MS);
-
-    trailRemovalTimers.set(this.#pos, timerId);
-    this.#ownedTimerId = timerId;
-  }
-
-  #clearOwnedTrailTimer() {
-    const active = trailRemovalTimers.get(this.#pos);
-    if (active !== undefined && active === this.#ownedTimerId) {
-      clearTimeout(active);
-      trailRemovalTimers.delete(this.#pos);
-    }
-    this.#ownedTimerId = null;
-  }
 }
-
-const trail = [
-  'absolute z-0 aspect-[2/1] w-16',
-  'bg-[radial-gradient(at_100%_50%,_theme(colors.green.700),_transparent_80%)]',
-  '[offset-anchor:100%_50%] [offset-path:border-box]',
-];
-
-/** Delay after the last content update before hiding the trail. */
-const TRAIL_REMOVAL_DELAY_MS = 1_000;
-
-/**
- * One pending removal timer per logical block. Streaming creates a new widget instance each tick;
- * `destroy` may run after the next instance has already scheduled, so we must not clear a newer timer.
- */
-const trailRemovalTimers = new Map<string, ReturnType<typeof setTimeout>>();
-
-const createTrailLayers = (): [HTMLElement, HTMLElement] => [
-  Domino.of('div').classNames(...trail, 'animate-trail').root,
-  Domino.of('div').classNames(...trail, 'animate-trail-offset').root,
-];

package/src/components/ChatThread/widgets/SuggestionWidget.ts

@@ -19,7 +19,7 @@ export class SuggestionWidget extends WidgetType {
   }
 
   override toDOM() {
-    // NOTE: Container must have `dx-size-container` to support cqi.
+    // NOTE: Container must have `dx-inline-size-container` (or `dx-size-container`) to support cqi.
     return Domino.of('span')
       .classNames(mx('inline-flex max-w-[calc(100cqi-8px)] my-1 pe-2 overflow-hidden'))
       .append(

package/src/components/ChatThread/widgets/SummaryWidget.tsx

@@ -11,17 +11,16 @@ import { type XmlWidgetProps } from '@dxos/ui-editor';
 import { meta } from '#meta';
 
 import { MessageThreadContext } from '../sync';
+import { styles } from './defaults';
 
 export const SummaryWidget = ({ children }: XmlWidgetProps<MessageThreadContext>) => {
   const { t } = useTranslation(meta.id);
 
   return (
-    <TogglePanel.Root classNames='border border-separator rounded-sm'>
+    <TogglePanel.Root classNames={styles.border}>
       <TogglePanel.Header classNames='text-sm bg-group-surface'>{t('summary.label')}</TogglePanel.Header>
       <TogglePanel.Content>
-        <div role='none' className='p-1 text-sm text-description'>
-          {children}
-        </div>
+        <div className='p-1 text-sm text-subdued'>{children}</div>
       </TogglePanel.Content>
     </TogglePanel.Root>
   );

package/src/components/ChatThread/widgets/ToolWidget.stories.tsx

@@ -7,7 +7,8 @@ import { type Meta, type StoryObj } from '@storybook/react-vite';
 import { withLayout, withTheme } from '@dxos/react-ui/testing';
 import { type ContentBlock } from '@dxos/types';
 
-import { translations } from '../../../translations';
+import { translations } from '#translations';
+
 import { ToolWidget, type ToolWidgetProps } from './ToolWidget';
 
 const baseWidgetProps = {

package/src/components/ChatThread/widgets/defaults.ts

@@ -0,0 +1,8 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+export const styles = {
+  padding: 'my-4',
+  border: 'border border-subdued-separator rounded-sm',
+};

package/src/components/ChatThread/widgets/index.ts

@@ -3,12 +3,11 @@
 //
 
 export * from './FallbackWidget';
-export * from './PromptWidget';
 export * from './ReasoningWidget';
 export * from './ReferenceWidget';
 export * from './SelectWidget';
 export * from './StatsWidget';
+export * from './StatusWidget';
 export * from './SummaryWidget';
 export * from './SuggestionWidget';
 export * from './ToolWidget';
-export * from './StatusWidget';

package/src/components/ProcessTree/ProcessTree.stories.tsx

@@ -6,24 +6,14 @@ import { type Meta, type StoryObj } from '@storybook/react-vite';
 import * as Option from 'effect/Option';
 
 import { Process } from '@dxos/functions-runtime';
+import { log } from '@dxos/log';
 import { withLayout, withTheme } from '@dxos/react-ui/testing';
 
-import { ProcessTree } from './ProcessTree';
+import { makeProcess } from '#testing';
 
-const makeProcess = (
-  overrides: Partial<Process.Info> & Pick<Process.Info, 'pid' | 'state'> & { name: string },
-): Process.Info => ({
-  parentPid: null,
-  key: `test.process.${overrides.name}`,
-  params: { name: overrides.name, target: null },
-  error: null,
-  startedAt: Date.now() - 10_000,
-  completedAt: Option.none(),
-  metrics: { wallTime: 0, inputCount: 0, outputCount: 0 },
-  ...overrides,
-});
+import { ProcessTree } from './ProcessTree';
 
-const seedProcesses: Process.Info[] = [
+const processes: Process.Info[] = [
   makeProcess({
     pid: Process.ID.make('97793611-815e-4a67-bc04-60fa67b2c987'),
     name: 'Trigger watcher',
@@ -73,7 +63,7 @@ const seedProcesses: Process.Info[] = [
 const meta: Meta<typeof ProcessTree> = {
   title: 'plugins/plugin-assistant/components/ProcessTree',
   component: ProcessTree,
-  decorators: [withLayout({ layout: 'column', classNames: 'w-(--dx-complementary-sidebar-size)' }), withTheme()],
+  decorators: [withTheme(), withLayout({ layout: 'column', classNames: 'w-(--dx-complementary-sidebar-size)' })],
 } satisfies Meta<typeof ProcessTree>;
 
 export default meta;
@@ -82,28 +72,12 @@ type Story = StoryObj<typeof meta>;
 
 export const Default: Story = {
   args: {
-    processes: seedProcesses,
-  },
-};
-
-export const AllRunning: Story = {
-  args: {
-    processes: [
-      makeProcess({
-        pid: Process.ID.make('1344bac8-745b-44a6-9af6-dc19509a7844'),
-        name: 'Agent alpha',
-        state: Process.State.RUNNING,
-      }),
-      makeProcess({
-        pid: Process.ID.make('b7c9787a-6705-47dc-a197-d37f1adfa338'),
-        name: 'Agent beta',
-        state: Process.State.RUNNING,
-      }),
-      makeProcess({
-        pid: Process.ID.make('43493533-a94c-4072-b18f-c0e0fd806bf5'),
-        name: 'Agent gamma',
-        state: Process.State.RUNNING,
-      }),
-    ],
+    processes: processes,
+    onProcessSelect: (process) => {
+      log.info('select', process);
+    },
+    onProcessTerminate: (process) => {
+      log.info('terminate', process);
+    },
   },
 };

package/src/components/ProcessTree/ProcessTree.tsx

@@ -3,41 +3,47 @@
 //
 
 import * as Match from 'effect/Match';
+import * as Option from 'effect/Option';
 import React from 'react';
 
 import { Process } from '@dxos/functions-runtime';
-import { Icon, ScrollArea, Treegrid } from '@dxos/react-ui';
+import { Icon, IconButton, ScrollArea, Tooltip, Treegrid } from '@dxos/react-ui';
 import { composable, composableProps, mx } from '@dxos/ui-theme';
+import { Unit } from '@dxos/util';
 
 export type ProcessTreeProps = {
+  // TODO(burdon): Atom.
   processes: readonly Process.Info[];
   onProcessSelect?: (process: Process.Info) => void;
+  onProcessTerminate?: (process: Process.Info) => void;
 };
 
 export const ProcessTree = composable<HTMLDivElement, ProcessTreeProps>(
-  ({ processes, onProcessSelect, ...props }, forwardedRef) => {
-    const filteredProcesses: Process.Info[] = [
+  ({ processes, onProcessSelect, onProcessTerminate, ...props }, forwardedRef) => {
+    const sortedProcesses = [
       ...processes.filter((process) => [Process.State.RUNNING, Process.State.HYBERNATING].includes(process.state)),
       ...processes.filter((process) => [Process.State.IDLE].includes(process.state)).slice(0, 3),
-      ...processes
-        .filter((process) =>
-          [Process.State.SUCCEEDED, Process.State.FAILED, Process.State.TERMINATED].includes(process.state),
-        )
-        .slice(0, 3),
-    ];
+      ...processes.filter((process) =>
+        [Process.State.SUCCEEDED, Process.State.FAILED, Process.State.TERMINATED].includes(process.state),
+      ),
+    ].sort((a, b) => {
+      const aCompletedAt = Option.getOrElse(a.completedAt, () => Infinity);
+      const bCompletedAt = Option.getOrElse(b.completedAt, () => Infinity);
+      return bCompletedAt - aCompletedAt;
+    });
 
     return (
       <ScrollArea.Root {...composableProps(props, { classNames: 'dx-expander' })} thin ref={forwardedRef}>
         <ScrollArea.Viewport>
           <Treegrid.Root gridTemplateColumns='1fr'>
-            {filteredProcesses
+            {sortedProcesses
               .filter((process) => process.parentPid === null)
               .map((process) => {
-                const activeChildren = filteredProcesses.filter(
-                  (candidate) =>
-                    candidate.parentPid?.toString() === process.pid.toString() &&
-                    candidate.state === Process.State.RUNNING,
-                );
+                // const activeChildren = filteredProcesses.filter(
+                //   (candidate) =>
+                //     candidate.parentPid?.toString() === process.pid.toString() &&
+                //     candidate.state === Process.State.RUNNING,
+                // );
                 return (
                   <Treegrid.Row
                     key={process.pid.toString()}
@@ -46,34 +52,61 @@ export const ProcessTree = composable<HTMLDivElement, ProcessTreeProps>(
                   >
                     <Treegrid.Cell
                       indent
-                      classNames='flex items-center p-1 gap-2 min-w-0'
+                      classNames={mx(
+                        'grid grid-cols-[min-content_1fr_min-content_min-content] items-center gap-1 min-w-0',
+                        onProcessSelect && 'dx-hover',
+                      )}
                       onClick={() => onProcessSelect?.(process)}
                     >
-                      <Icon
-                        size={4}
-                        classNames={mx(
-                          process.state === Process.State.RUNNING && 'animate-spin',
-                          process.state === Process.State.FAILED && 'text-error-text',
-                          process.state === Process.State.SUCCEEDED && 'text-success-text',
-                        )}
-                        icon={Match.value(process.state).pipe(
-                          Match.when(Process.State.RUNNING, () => 'ph--spinner-gap--regular'),
-                          Match.when(Process.State.SUCCEEDED, () => 'ph--check-circle--regular'),
-                          Match.when(Process.State.FAILED, () => 'ph--warning--regular'),
-                          Match.when(Process.State.HYBERNATING, () => 'ph--spinner--regular'),
-                          Match.when(Process.State.IDLE, () => 'ph--hourglass--regular'),
-                          Match.when(Process.State.TERMINATING, () => 'ph--x-circle--regular'),
-                          Match.when(Process.State.TERMINATED, () => 'ph--x-circle--regular'),
-                          Match.orElse(() => 'ph--spinner-gap--regular'),
-                        )}
-                      />
-                      <div role='none' className='flex items-center gap-2 text-xs overflow-hidden'>
+                      <Tooltip.Trigger className='p-1' content={process.state.toString()}>
+                        <Icon
+                          size={4}
+                          synchronized
+                          classNames={mx(
+                            process.state === Process.State.RUNNING && 'animate-spin',
+                            process.state === Process.State.FAILED && 'text-error-text',
+                            process.state === Process.State.SUCCEEDED && 'text-success-text',
+                          )}
+                          icon={Match.value(process.state).pipe(
+                            Match.when(Process.State.RUNNING, () => 'ph--spinner-gap--regular'),
+                            Match.when(Process.State.SUCCEEDED, () => 'ph--check-circle--regular'),
+                            Match.when(Process.State.FAILED, () => 'ph--warning--regular'),
+                            Match.when(Process.State.HYBERNATING, () => 'ph--spinner--regular'),
+                            Match.when(Process.State.IDLE, () => 'ph--hourglass--regular'),
+                            Match.when(Process.State.TERMINATING, () => 'ph--x-circle--regular'),
+                            Match.when(Process.State.TERMINATED, () => 'ph--x-circle--regular'),
+                            Match.orElse(() => 'ph--spinner-gap--regular'),
+                          )}
+                        />
+                      </Tooltip.Trigger>
+                      <div className='flex items-center gap-2 text-xs overflow-hidden'>
                         {/* TODO(burdon): Name is too long (and not informative). */}
-                        <span className='truncate text-description'>{process.params.name}</span>
+                        <span className='truncate text-description select-none'>
+                          {process.params.name ?? process.pid.toString()}
+                        </span>
                         {/* {activeChildren.length > 0 && (
                           <span className='text-xs text-description ml-1'>{activeChildren[0].params.name}</span>
                         )} */}
                       </div>
+                      {([Process.State.FAILED, Process.State.SUCCEEDED].includes(process.state) && (
+                        <div className='text-xs text-description tabular-nums'>
+                          {Unit.Millisecond(process.metrics.wallTime).toString()}
+                        </div>
+                      )) || <div />}
+                      {onProcessTerminate && (
+                        <IconButton
+                          classNames='min-h-0 p-1'
+                          icon='ph--x--regular'
+                          iconOnly
+                          variant='ghost'
+                          size={4}
+                          label='Actions'
+                          onClick={(event) => {
+                            event.stopPropagation();
+                            onProcessTerminate?.(process);
+                          }}
+                        />
+                      )}
                     </Treegrid.Cell>
                   </Treegrid.Row>
                 );

package/src/components/TaskList/TaskList.stories.tsx

@@ -0,0 +1,44 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import { type Meta, type StoryObj } from '@storybook/react-vite';
+import React from 'react';
+
+import { Plan } from '@dxos/assistant-toolkit';
+import { withLayout, withTheme } from '@dxos/react-ui/testing';
+
+import { TaskList } from './TaskList';
+
+type DefaultStoryProps = {
+  tasks?: Omit<Plan.Task, 'id'>[];
+};
+
+const DefaultStory = ({ tasks = [] }: DefaultStoryProps) => {
+  const plan = React.useMemo(() => Plan.makePlan({ tasks }), [tasks]);
+  return <TaskList plan={plan} />;
+};
+
+const meta = {
+  title: 'plugins/plugin-assistant/components/TaskList',
+  render: (args) => <DefaultStory {...args} />,
+  decorators: [withTheme(), withLayout({ layout: 'column' })],
+} satisfies Meta<typeof DefaultStory>;
+
+export default meta;
+
+type Story = StoryObj<typeof meta>;
+
+export const Default: Story = {
+  args: {
+    tasks: [
+      { title: 'Crack the eggs', status: 'done' },
+      { title: 'Whisk with salt and pepper', status: 'done' },
+      { title: 'Heat the pan with butter', status: 'in-progress' },
+      { title: 'Pour and stir continuously', status: 'todo' },
+      { title: 'Plate and serve', status: 'todo' },
+    ],
+  },
+};
+
+export const Empty: Story = {};

package/src/components/TaskList/TaskList.tsx

@@ -0,0 +1,45 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import React from 'react';
+
+import { Plan } from '@dxos/assistant-toolkit';
+import { useObject } from '@dxos/react-client/echo';
+import { Icon, Tag, type ThemedClassName } from '@dxos/react-ui';
+import { Row, RowList } from '@dxos/react-ui-list';
+import { composable, composableProps } from '@dxos/ui-theme';
+
+export type TaskListProps = ThemedClassName<{
+  plan: Plan.Plan;
+}>;
+
+export const TaskList = composable<HTMLDivElement, TaskListProps>(({ plan, ...props }, forwardedRef) => {
+  const [snapshot] = useObject(plan);
+  const tasks = snapshot?.tasks ?? [];
+
+  return (
+    <RowList.Root>
+      <RowList.Viewport {...composableProps(props, { classNames: 'dx-container' })} ref={forwardedRef}>
+        <RowList.Content aria-label='Tasks'>
+          {tasks.map((task) => (
+            <Row key={task.id} id={task.id} classNames='py-0'>
+              <div className='flex items-center gap-2'>
+                <Icon
+                  icon={task.status === 'done' ? 'ph--check--regular' : 'ph--circle--regular'}
+                  classNames={task.status === 'done' ? 'text-success-text' : undefined}
+                  size={4}
+                />
+                <span className='sr-only'>
+                  {task.status === 'done' ? 'done' : task.status === 'in-progress' ? 'in progress' : 'to do'}
+                </span>
+                <span className='truncate flex-1'>{task.title}</span>
+                {task.status === 'in-progress' && <Tag palette='info'>pending</Tag>}
+              </div>
+            </Row>
+          ))}
+        </RowList.Content>
+      </RowList.Viewport>
+    </RowList.Root>
+  );
+});

package/src/{cli → components/TaskList}/index.ts

@@ -2,4 +2,4 @@
 // Copyright 2026 DXOS.org
 //
 
-export * from './plugin';
+export * from './TaskList';

package/src/components/TemplateEditor/TemplateEditor.stories.tsx

@@ -6,13 +6,14 @@ import { type Meta, type StoryObj } from '@storybook/react-vite';
 import React, { useState } from 'react';
 
 import { createSystemPrompt } from '@dxos/assistant';
-import { Blueprint, Template } from '@dxos/blueprints';
+import { Blueprint, Template } from '@dxos/compute';
 import { useClient } from '@dxos/react-client';
 import { withClientProvider } from '@dxos/react-client/testing';
 import { withLayout, withTheme } from '@dxos/react-ui/testing';
 import { trim } from '@dxos/util';
 
-import { translations } from '../../translations';
+import { translations } from '#translations';
+
 import { TemplateEditor, type TemplateEditorProps } from './TemplateEditor';
 
 const FENCE = '```';
@@ -47,7 +48,7 @@ const DefaultStory = ({ source }: TemplateEditorProps & { source: string }) => {
     const space = client.spaces.get()[0];
     return space.db.add(
       Blueprint.make({
-        key: 'example.com/blueprint/test',
+        key: 'com.example.blueprint.test',
         name: 'Test',
         instructions: Template.make({ source }),
       }),

package/src/components/TemplateEditor/TemplateEditor.tsx

@@ -6,7 +6,7 @@ import { defaultHighlightStyle, syntaxHighlighting } from '@codemirror/language'
 import { composeRefs } from '@radix-ui/react-compose-refs';
 import React from 'react';
 
-import { type Template } from '@dxos/blueprints';
+import { type Template } from '@dxos/compute';
 import { createDocAccessor } from '@dxos/echo-db';
 import { useThemeContext, useTranslation } from '@dxos/react-ui';
 import { useTextEditor } from '@dxos/react-ui-editor';
@@ -35,17 +35,17 @@ export const TemplateEditor = composable<HTMLDivElement, TemplateEditorProps>(
     const { t } = useTranslation(meta.id);
     const { themeMode } = useThemeContext();
     const { parentRef } = useTextEditor(() => {
-      const text = template.source?.target;
-      if (!text) {
+      const target = template.source?.target;
+      if (!target) {
         return {};
       }
 
       return {
-        initialValue: text.content ?? '',
+        initialValue: target.content ?? '',
         extensions: [
           createDataExtensions({
             id,
-            text: createDocAccessor(text, ['content']),
+            text: createDocAccessor(target, ['content']),
           }),
           createBasicExtensions({
             bracketMatching: false,

package/src/components/TemplateEditor/TemplateForm.stories.tsx

@@ -5,7 +5,7 @@
 import { type Meta } from '@storybook/react-vite';
 import React, { useCallback, useState } from 'react';
 
-import { Blueprint, Template } from '@dxos/blueprints';
+import { Blueprint, Template } from '@dxos/compute';
 import { Obj } from '@dxos/echo';
 import { invariant } from '@dxos/invariant';
 import { useClient } from '@dxos/react-client';
@@ -13,7 +13,8 @@ import { withClientProvider } from '@dxos/react-client/testing';
 import { withLayout, withTheme } from '@dxos/react-ui/testing';
 import { trim } from '@dxos/util';
 
-import { translations } from '../../translations';
+import { translations } from '#translations';
+
 import { type TemplateChangeCallback, TemplateForm } from './TemplateForm';
 
 const TEMPLATE = trim`
@@ -33,7 +34,7 @@ const DefaultStory = () => {
     invariant(space, 'TemplateForm story requires at least one space');
     return space.db.add(
       Blueprint.make({
-        key: 'example.com/blueprint/test',
+        key: 'com.example.blueprint.test',
         name: 'Test',
         instructions: Template.make({ source: TEMPLATE }),
       }),
@@ -42,16 +43,12 @@ const DefaultStory = () => {
 
   const handleChange: TemplateChangeCallback = useCallback(
     (mutate) => {
-      Obj.change(blueprint, (blueprint) => mutate(blueprint.instructions));
+      Obj.update(blueprint, (blueprint) => mutate(blueprint.instructions));
     },
     [blueprint],
   );
 
-  return (
-    <div role='none' className='flex w-[40rem] border border-separator overflow-hidden'>
-      <TemplateForm id={blueprint.id} template={blueprint.instructions} onChange={handleChange} />
-    </div>
-  );
+  return <TemplateForm id={blueprint.id} template={blueprint.instructions} onChange={handleChange} />;
 };
 
 const meta = {