@dxos/plugin-assistant 0.8.4-main.c85a9c8dae → 0.8.4-main.d9fc60f731

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

@@ -0,0 +1,139 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { WidgetType } from '@codemirror/view';
+
+import { Domino } from '@dxos/ui';
+
+import { styles } from './defaults';
+
+/**
+ * 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 ReasoningWidgetBlockIdProps = {
+  id?: string;
+  range?: { from: number; to: number };
+};
+
+/**
+ * AI reasoning widget.
+ */
+export class ReasoningWidget 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,
+  ) {
+    super();
+    this.#pos = pos === undefined || pos === '' ? 'reasoning' : String(pos);
+  }
+
+  override eq(other: this) {
+    return this.text === other.text && this.#pos === other.#pos;
+  }
+
+  override toDOM() {
+    return Domino.of('div')
+      .classNames(styles.padding)
+      .append(
+        Domino.of('div')
+          .classNames('relative overflow-hidden p-px', styles.border)
+          .attributes({ 'data-trail-container': '' })
+          .append(
+            Domino.of('div')
+              .classNames(
+                'relative z-10 bg-base-surface rounded-sm text-sm text-subdued py-1',
+                'grid grid-cols-[24px_1fr] gap-x-0.5 items-start',
+              )
+              .append(
+                Domino.of('div')
+                  .classNames('flex h-5 w-full shrink-0 items-center justify-center self-start')
+                  .append(Domino.svg('ph--brain--regular').classNames('shrink-0 size-4 opacity-70')),
+                Domino.of('div')
+                  .classNames('px-2 max-h-[5lh] overflow-y-auto dx-scrollbar-thin min-w-0')
+                  .text(this.text)
+                  .attributes({ 'data-reasoning-text': '' }),
+              ),
+            Domino.of('div').attributes({ 'data-id': this.#pos }),
+          ),
+      ).root;
+  }
+
+  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 bordered container — setting `textContent` on the wrong
+    // ancestor would remove `[data-id]`.
+    dom.querySelector<HTMLElement>('[data-reasoning-text]')?.replaceChildren(this.text);
+
+    // Recreate trail host if it was removed by a prior timeout (streaming paused > TRAIL_REMOVAL_DELAY_MS).
+    const container = dom.querySelector<HTMLElement>('[data-trail-container]');
+    let trailHost = container?.querySelector<HTMLElement>('[data-id]');
+    if (container && !trailHost) {
+      trailHost = Domino.of('div').attributes({ 'data-id': this.#pos }).root;
+      container.append(trailHost);
+    }
+    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/ReferenceWidget.ts

@@ -0,0 +1,33 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { WidgetType } from '@codemirror/view';
+
+import { Domino } from '@dxos/ui';
+
+import { styles } from './defaults';
+
+export class ReferenceWidget extends WidgetType {
+  constructor(
+    private text: string,
+    private dxn: string,
+  ) {
+    super();
+  }
+
+  override eq(other: this) {
+    return this.dxn === other.dxn;
+  }
+
+  override toDOM() {
+    return Domino.of('div')
+      .classNames(styles.padding)
+      .append(
+        Domino.of('dx-anchor' as any)
+          .classNames('dx-tag--anchor')
+          .attributes({ dxn: this.dxn })
+          .text(this.text),
+      ).root;
+  }
+}

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

@@ -0,0 +1,39 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { WidgetType } from '@codemirror/view';
+
+import { Domino } from '@dxos/ui';
+
+import { styles } from './defaults';
+
+/**
+ * Simple prompt widget.
+ */
+export class SelectWidget extends WidgetType {
+  constructor(private options: string[]) {
+    super();
+  }
+
+  override eq(other: this) {
+    return JSON.stringify(this.options) === JSON.stringify(other.options);
+  }
+
+  /**
+   * NOTE: Container must set var based on user's identity.
+   */
+  override toDOM() {
+    return Domino.of('div')
+      .attributes({ role: 'group' })
+      .classNames(styles.padding, 'flex flex-wrap gap-1')
+      .append(
+        ...this.options.map((option) =>
+          Domino.of('button')
+            .classNames('dx-button inline-block max-w-[100cqi]')
+            .attributes({ 'data-action': 'submit', 'data-value': option, 'data-density': 'md' })
+            .text(option),
+        ),
+      ).root;
+  }
+}

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

@@ -0,0 +1,31 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { WidgetType } from '@codemirror/view';
+
+import { Domino } from '@dxos/ui';
+
+import { styles } from './defaults';
+
+/**
+ * Simple stats widget.
+ */
+export class StatsWidget extends WidgetType {
+  constructor(private text: string) {
+    super();
+  }
+
+  override eq(other: this) {
+    return this.text === other.text;
+  }
+
+  override toDOM() {
+    return Domino.of('div').classNames(styles.padding, 'text-sm text-placeholder').text(this.text).root;
+  }
+
+  override updateDOM(dom: HTMLElement) {
+    dom.textContent = this.text;
+    return true;
+  }
+}

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

@@ -0,0 +1,53 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { WidgetType } from '@codemirror/view';
+
+import { Domino } from '@dxos/ui';
+
+import { styles } from './defaults';
+
+/**
+ * AI status widget.
+ */
+export class StatusWidget extends WidgetType {
+  constructor(private readonly text: string) {
+    super();
+  }
+
+  override eq(other: this) {
+    return this.text === other.text;
+  }
+
+  override toDOM() {
+    return Domino.of('div')
+      .classNames(styles.padding)
+      .append(
+        Domino.of('div')
+          .classNames('relative overflow-hidden rounded-sm')
+          .append(
+            Domino.of('div')
+              .classNames('grid grid-cols-[24px_1fr] gap-x-0.5 gap-y-0 items-start px-0.5 py-0.5 text-placeholder')
+              .append(
+                Domino.of('div')
+                  .classNames('flex h-5 w-full shrink-0 items-center justify-center self-start')
+                  .append(Domino.svg('ph--info--regular').classNames('shrink-0 size-4 opacity-70')),
+                Domino.of('div')
+                  .classNames('relative min-w-0')
+                  .append(
+                    Domino.of('div')
+                      .classNames('relative z-10 rounded-sm text-sm leading-5')
+                      .attributes({ 'data-status-text': '' })
+                      .text(this.text),
+                  ),
+              ),
+          ),
+      ).root;
+  }
+
+  override updateDOM(dom: HTMLElement) {
+    dom.querySelector<HTMLElement>('[data-status-text]')?.replaceChildren(this.text);
+    return true;
+  }
+}

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

@@ -0,0 +1,41 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { WidgetType } from '@codemirror/view';
+
+import { Domino, mx } from '@dxos/ui';
+
+/**
+ * Simple prompt widget.
+ */
+export class SuggestionWidget extends WidgetType {
+  constructor(private text: string) {
+    super();
+  }
+
+  override eq(other: this) {
+    return this.text === other.text;
+  }
+
+  override toDOM() {
+    // 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(
+        Domino.of('button')
+          .attributes({
+            'data-action': 'submit',
+            'data-density': 'md',
+            'data-value': this.text,
+          })
+          .classNames(mx('dx-button gap-2 w-full overflow-hidden'))
+          .append(
+            Domino.of('dx-icon' as any).attributes({
+              icon: 'ph--lightning--regular',
+            }),
+            Domino.of('span').classNames('flex-1 truncate min-w-0').text(this.text),
+          ),
+      ).root;
+  }
+}

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

@@ -0,0 +1,27 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import React from 'react';
+
+import { useTranslation } from '@dxos/react-ui';
+import { TogglePanel } from '@dxos/react-ui-components';
+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={styles.border}>
+      <TogglePanel.Header classNames='text-sm bg-group-surface'>{t('summary.label')}</TogglePanel.Header>
+      <TogglePanel.Content>
+        <div className='p-1 text-sm text-subdued'>{children}</div>
+      </TogglePanel.Content>
+    </TogglePanel.Root>
+  );
+};

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

@@ -0,0 +1,55 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+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 { ToolWidget, type ToolWidgetProps } from './ToolWidget';
+
+const baseWidgetProps = {
+  _tag: 'toolCall',
+  range: { from: 0, to: 1 },
+} satisfies Pick<ToolWidgetProps, '_tag' | 'range'>;
+
+const toolCallAndResult: ContentBlock.Any[] = [
+  {
+    _tag: 'toolCall',
+    toolCallId: 'tc-story-1',
+    name: 'example_tool',
+    input: JSON.stringify({ query: 'status', limit: 10 }),
+    providerExecuted: false,
+  },
+  {
+    _tag: 'toolResult',
+    toolCallId: 'tc-story-1',
+    name: 'example_tool',
+    result: JSON.stringify({ ok: true, rows: [{ id: 1 }, { id: 2 }] }),
+    providerExecuted: false,
+  },
+];
+
+const meta = {
+  title: 'plugins/plugin-assistant/components/ChatWidgets/ToolWidget',
+  component: ToolWidget,
+  decorators: [withTheme(), withLayout({ layout: 'column' })],
+  parameters: {
+    layout: 'fullscreen',
+    translations,
+  },
+} satisfies Meta<typeof ToolWidget>;
+
+export default meta;
+
+type Story = StoryObj<typeof meta>;
+
+export const Default: Story = {
+  args: {
+    ...baseWidgetProps,
+    blocks: toolCallAndResult,
+  },
+};

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

@@ -0,0 +1,171 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import type * as Tool from '@effect/ai/Tool';
+import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react';
+
+import { Icon, useTranslation } from '@dxos/react-ui';
+import { NumericTabs, TextCrawl, TogglePanel, type TogglePanelRootProps } from '@dxos/react-ui-components';
+import { JsonHighlighter } from '@dxos/react-ui-syntax-highlighter';
+import { type ContentBlock, type Message } from '@dxos/types';
+import { type XmlWidgetProps } from '@dxos/ui-editor';
+import { isNonNullable, safeParseJson } from '@dxos/util';
+
+import { meta } from '#meta';
+
+export const isToolMessage = (message: Message.Message) => {
+  return message.blocks.some((block: ContentBlock.Any) => block._tag === 'toolCall' || block._tag === 'toolResult');
+};
+
+export type ToolWidgetProps = XmlWidgetProps<{
+  blocks: ContentBlock.Any[];
+}>;
+
+export const ToolWidget = ({ view, blocks = [] }: ToolWidgetProps) => {
+  const { t } = useTranslation(meta.id);
+
+  const items = useMemo<ToolPanelProps['items']>(() => {
+    let lastToolCall: { tool: Tool.Any | undefined; block: ContentBlock.ToolCall } | undefined;
+    // TODO(burdon): Get from context?
+    const tools: Tool.Any[] = []; //processor.conversation.toolkit?.tools ?? [];
+    return blocks
+      .filter((block) => block._tag === 'toolCall' || block._tag === 'toolResult' || block._tag === 'stats')
+      .map((block) => {
+        switch (block._tag) {
+          case 'toolCall': {
+            if (block.pending && lastToolCall?.block.toolCallId === block.toolCallId) {
+              return null;
+            }
+
+            const tool = tools.find((tool) => tool.name === block.name);
+            lastToolCall = { tool, block };
+            return {
+              title: tool?.description ?? [t('tool-call.label'), block.name].filter(Boolean).join(' '),
+              icon: block.operationIcon,
+              content: {
+                ...block,
+                input: safeParseJson(block.input),
+              },
+            };
+          }
+
+          case 'toolResult': {
+            // TODO(burdon): Parse error type.
+            if (block.error) {
+              return {
+                title: t('tool-error.label'),
+                icon: lastToolCall?.block.operationIcon,
+                content: block,
+              };
+            }
+
+            const title =
+              lastToolCall?.tool?.description ??
+              [t('tool-result.label'), lastToolCall?.block.name].filter(Boolean).join(' ');
+            const icon = lastToolCall?.block.operationIcon;
+            lastToolCall = undefined;
+            return {
+              title,
+              icon,
+              content: {
+                ...block,
+                result: typeof block.result === 'string' ? safeParseJson(block.result) : block.result,
+              },
+            };
+          }
+
+          case 'stats': {
+            if (!lastToolCall) {
+              return null;
+            }
+
+            return {
+              title: t('stats.label'),
+              icon: lastToolCall.block.operationIcon,
+              content: block,
+            };
+          }
+        }
+      })
+      .filter(isNonNullable);
+  }, [blocks, t]);
+
+  const handleChangeOpen = useCallback(() => {
+    setTimeout(() => {
+      // Measure after animation.
+      view?.requestMeasure();
+    }, 1_000);
+  }, [view]);
+
+  // Ignore if empty.
+  if (!items.length) {
+    return null;
+  }
+
+  return <ToolPanel items={items} onChangeOpen={handleChangeOpen} />;
+};
+
+type ToolPanelItem = { title: string; icon?: string; content: any };
+
+type ToolPanelProps = {
+  items: ToolPanelItem[];
+} & Pick<TogglePanelRootProps, 'onChangeOpen'>;
+
+const DEFAULT_TOOL_ICON = 'ph--wrench--regular';
+
+const ToolPanel = ({ items, onChangeOpen }: ToolPanelProps) => {
+  const tabsRef = useRef<HTMLDivElement>(null);
+  const [selected, setSelected] = useState(0);
+  const [open, setOpen] = useState(false);
+
+  // Clamp selected to avoid out-of-bounds after items shrink.
+  useEffect(() => {
+    setSelected((prev) => Math.min(prev, Math.max(0, items.length - 1)));
+  }, [items.length]);
+
+  useEffect(() => {
+    onChangeOpen?.(open);
+    if (open) {
+      tabsRef.current?.focus();
+    }
+  }, [open, onChangeOpen]);
+
+  const handleSelect = useCallback((index: number) => {
+    setSelected(index);
+  }, []);
+
+  // Prefer the icon from the latest tool call so the header reflects what's currently active.
+  // TextCrawl shows multiple titles, but only one icon slot — using the most recent keeps it in sync
+  // with the visible operation while it streams.
+  const headerIcon = items[items.length - 1]?.icon ?? items[0]?.icon ?? DEFAULT_TOOL_ICON;
+
+  return (
+    <TogglePanel.Root open={open} onChangeOpen={setOpen}>
+      <TogglePanel.Header classNames='flex items-center gap-2 text-sm text-placeholder'>
+        <Icon icon={headerIcon} size={4} classNames='shrink-0 opacity-70' />
+        <TextCrawl key='status-roll' lines={items.map((item) => item.title)} autoAdvance greedy />
+      </TogglePanel.Header>
+      <TogglePanel.Content>
+        <TogglePanel.Viewport classNames='grid grid-cols-[32px_1fr]'>
+          <NumericTabs
+            ref={tabsRef}
+            classNames='p-1'
+            length={items.length}
+            selected={selected}
+            onSelect={handleSelect}
+          />
+          <JsonHighlighter
+            data={items[selected]?.content}
+            classNames='p-1 text-xs bg-transparent'
+            replacer={{
+              maxDepth: 3,
+              maxArrayLen: 10,
+              maxStringLen: 128,
+            }}
+          />
+        </TogglePanel.Viewport>
+      </TogglePanel.Content>
+    </TogglePanel.Root>
+  );
+};

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

@@ -0,0 +1,13 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+export * from './FallbackWidget';
+export * from './ReasoningWidget';
+export * from './ReferenceWidget';
+export * from './SelectWidget';
+export * from './StatsWidget';
+export * from './StatusWidget';
+export * from './SummaryWidget';
+export * from './SuggestionWidget';
+export * from './ToolWidget';