@dxos/plugin-assistant 0.8.3 → 0.8.4-main.f9ba587

package/src/components/Prompt/autocomplete.ts

@@ -1,218 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { Prec, type Extension } from '@codemirror/state';
-import {
-  EditorView,
-  Decoration,
-  ViewPlugin,
-  keymap,
-  type DecorationSet,
-  type ViewUpdate,
-  WidgetType,
-} from '@codemirror/view';
-
-export type AutocompleteOptions = {
-  /**
-   * Callback triggered when Enter is pressed.
-   * @param text The current text in the editor
-   * @returns true if the editor should reset the document.
-   */
-  onSubmit?: (text: string) => boolean | void;
-
-  /**
-   * Function that returns a list of suggestions based on the current text.
-   * @param text The current text before the cursor
-   * @returns Array of suggestion strings
-   */
-  onSuggest?: (text: string) => string[];
-
-  /**
-   * ESC pressed.
-   */
-  onCancel?: () => void;
-};
-
-/**
- * Creates an autocomplete extension that shows inline suggestions.
- * Pressing Tab will complete the suggestion.
- */
-// TODO(burdon): Move to react-ui-editor.
-export const autocompleteExtension = ({ onSubmit, onSuggest, onCancel }: AutocompleteOptions): Extension => {
-  const suggest = ViewPlugin.fromClass(
-    class {
-      _decorations: DecorationSet;
-      _currentSuggestion: string | null = null;
-
-      constructor(view: EditorView) {
-        this._decorations = this.computeDecorations(view);
-      }
-
-      update(update: ViewUpdate) {
-        if (update.docChanged || update.selectionSet) {
-          this._decorations = this.computeDecorations(update.view);
-        }
-      }
-
-      private computeDecorations(view: EditorView): DecorationSet {
-        const text = view.state.doc.toString();
-        const suggestions = onSuggest?.(text) ?? [];
-        if (!suggestions.length) {
-          this._currentSuggestion = null;
-          return Decoration.none;
-        }
-
-        // Get the first suggestion.
-        this._currentSuggestion = suggestions[0];
-        const suffix = this._currentSuggestion.slice(text.length);
-        if (!suffix) {
-          return Decoration.none;
-        }
-
-        // Always show ghost text at the end of the document.
-        return Decoration.set([
-          Decoration.widget({
-            widget: new InlineSuggestionWidget(suffix),
-            side: 1,
-          }).range(view.state.doc.length),
-        ]);
-      }
-
-      completeSuggestion(view: EditorView): boolean {
-        if (!this._currentSuggestion) {
-          return false;
-        }
-
-        const text = view.state.doc.toString();
-        const suffix = this._currentSuggestion.slice(text.length);
-        if (!suffix) {
-          return false;
-        }
-
-        view.dispatch({
-          changes: {
-            from: view.state.doc.length,
-            insert: suffix,
-          },
-          selection: {
-            anchor: view.state.doc.length + suffix.length,
-          },
-        });
-
-        return true;
-      }
-    },
-    {
-      decorations: (v) => v._decorations,
-    },
-  );
-
-  return [
-    suggest,
-    EditorView.theme({
-      '.cm-inline-suggestion': {
-        opacity: 0.4,
-      },
-    }),
-
-    Prec.highest(
-      keymap.of([
-        {
-          key: 'Tab',
-          preventDefault: true,
-          run: (view) => {
-            const plugin = view.plugin(suggest);
-            return plugin?.completeSuggestion(view) ?? false;
-          },
-        },
-        {
-          key: 'ArrowRight',
-          preventDefault: true,
-          run: (view) => {
-            // Only complete if cursor is at the end
-            if (view.state.selection.main.head !== view.state.doc.length) {
-              return false;
-            }
-
-            const plugin = view.plugin(suggest);
-            return plugin?.completeSuggestion(view) ?? false;
-          },
-        },
-        {
-          key: 'Enter',
-          preventDefault: true,
-          run: (view) => {
-            const text = view.state.doc.toString().trim();
-            if (text.length > 0 && onSubmit) {
-              const reset = onSubmit(text);
-
-              // Clear the document after calling onEnter.
-              if (reset) {
-                view.dispatch({
-                  changes: {
-                    from: 0,
-                    to: view.state.doc.length,
-                    insert: '',
-                  },
-                });
-              }
-            }
-
-            return true;
-          },
-        },
-        {
-          key: 'Shift-Enter',
-          preventDefault: true,
-          run: (view) => {
-            view.dispatch({
-              changes: {
-                from: view.state.selection.main.head,
-                insert: '\n',
-              },
-              selection: {
-                anchor: view.state.selection.main.head + 1,
-                head: view.state.selection.main.head + 1,
-              },
-            });
-            return true;
-          },
-        },
-        {
-          key: 'Escape',
-          preventDefault: true,
-          run: (view) => {
-            // Clear the entire document.
-            view.dispatch({
-              changes: {
-                from: 0,
-                to: view.state.doc.length,
-                insert: '',
-              },
-            });
-            onCancel?.();
-            return true;
-          },
-        },
-      ]),
-    ),
-  ];
-};
-
-class InlineSuggestionWidget extends WidgetType {
-  constructor(private suffix: string) {
-    super();
-  }
-
-  override toDOM(): HTMLSpanElement {
-    const span = document.createElement('span');
-    span.textContent = this.suffix;
-    span.className = 'cm-inline-suggestion';
-    return span;
-  }
-
-  override eq(other: InlineSuggestionWidget): boolean {
-    return other.suffix === this.suffix;
-  }
-}

package/src/components/Prompt/index.ts

@@ -1,6 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './Prompt';
-export * from './PromptBar';

package/src/components/Prompt/references.ts

@@ -1,180 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { autocompletion, completionKeymap, type CompletionResult } from '@codemirror/autocomplete';
-import { type Extension, RangeSet } from '@codemirror/state';
-import {
-  Decoration,
-  EditorView,
-  keymap,
-  ViewPlugin,
-  WidgetType,
-  type DecorationSet,
-  type ViewUpdate,
-} from '@codemirror/view';
-
-import { Mutex } from '@dxos/async';
-
-export type ReferenceData = {
-  uri: string;
-  label: string;
-  // TODO(dmaretskyi): Consider adding details renderer for when you hover over the reference.
-};
-
-export interface ReferencesProvider {
-  getReferences({ query }: { query: string }): Promise<ReferenceData[]>;
-
-  resolveReference({ uri }: { uri: string }): Promise<ReferenceData | null>;
-}
-
-export type PromptReferencesOptions = {
-  provider: ReferencesProvider;
-  /**
-   * Will prevent the autocomplete from closing when the user blurs the editor.
-   * @default false
-   */
-  debug?: boolean;
-  /**
-   * @default '@'
-   */
-  triggerCharacter?: string;
-};
-
-/**
- * Include references into text.
- */
-export const promptReferences = ({
-  provider,
-  debug = false,
-  triggerCharacter = '@',
-}: PromptReferencesOptions): Extension => {
-  if (triggerCharacter.length !== 1) {
-    throw new Error('triggerCharacter must be a single character');
-  }
-
-  const decorationField = ViewPlugin.fromClass(
-    class ReferenceView {
-      private _mutex = new Mutex();
-
-      decorations: DecorationSet = Decoration.set([]);
-
-      constructor(view: EditorView) {
-        queueMicrotask(async () => {
-          const guard = await this._mutex.acquire();
-          try {
-            this.decorations = await this._computeDecorations(view);
-          } finally {
-            guard.release();
-          }
-        });
-      }
-
-      update(update: ViewUpdate) {
-        if (update.docChanged) {
-          queueMicrotask(async () => {
-            const guard = await this._mutex.acquire();
-            try {
-              this.decorations = await this._computeDecorations(update.view);
-            } finally {
-              guard.release();
-            }
-          });
-        }
-      }
-
-      private async _computeDecorations(view: EditorView): Promise<DecorationSet> {
-        const text = view.state.doc.toString();
-        const references = text.matchAll(new RegExp(`${triggerCharacter}[a-zA-Z0-9@:]+\\s`, 'g'));
-
-        const decorations = [];
-        for (const match of references) {
-          const reference = match[0];
-          const uri = reference.slice(1, -1);
-          const data = await provider.resolveReference({ uri });
-          if (data) {
-            decorations.push(
-              Decoration.replace({
-                widget: new ReferenceWidget(data),
-              }).range(match.index!, match.index! + reference.length),
-            );
-          }
-        }
-
-        return Decoration.set(decorations);
-      }
-    },
-    {
-      decorations: (v) => v.decorations,
-      provide: (plugin) => [
-        EditorView.atomicRanges.of(
-          (view): DecorationSet => view.plugin(decorationField)?.decorations ?? RangeSet.empty,
-        ),
-      ],
-    },
-  );
-
-  return [
-    decorationField,
-
-    EditorView.theme({
-      '.cm-reference-pill': {
-        borderRadius: '0.25rem',
-        borderWidth: '1px',
-        marginRight: '0.25rem',
-        marginLeft: '0.25rem',
-      },
-    }),
-
-    autocompletion({
-      activateOnTyping: true,
-      override: [
-        async (context): Promise<CompletionResult | null> => {
-          const match = context.matchBefore(new RegExp(`${triggerCharacter}[a-zA-Z0-9]+`));
-
-          if (!match || match?.to === match?.from) {
-            return null;
-          }
-
-          const query = match.text.slice(1);
-          const references = await provider.getReferences({ query });
-
-          return {
-            from: match.from,
-            filter: false,
-            options: references.map((reference) => ({
-              label: reference.label,
-              apply: `${triggerCharacter}${reference.uri} `,
-            })),
-          };
-        },
-      ],
-      closeOnBlur: !debug,
-      tooltipClass: () => 'shadow rounded',
-      aboveCursor: true,
-    }),
-
-    keymap.of(completionKeymap),
-  ];
-};
-
-class ReferenceWidget extends WidgetType {
-  constructor(private data: ReferenceData) {
-    super();
-  }
-
-  override toDOM(): HTMLSpanElement {
-    const span = document.createElement('span');
-    span.textContent = `@ ${this.data.label}`;
-    span.className = 'cm-reference-pill';
-    return span;
-  }
-
-  override eq(other: ReferenceWidget): boolean {
-    return other.data.uri === this.data.uri;
-  }
-
-  override ignoreEvent(): boolean {
-    return true;
-  }
-}

package/src/components/Thread/Thread.stories.tsx

@@ -1,201 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import '@dxos-theme';
-
-import { type StoryObj, type Meta } from '@storybook/react';
-import React, { useCallback, useEffect, useState } from 'react';
-
-import { Message } from '@dxos/ai';
-import { IntentPlugin } from '@dxos/app-framework';
-import { withPluginManager } from '@dxos/app-framework/testing';
-import { Obj } from '@dxos/echo';
-import { faker } from '@dxos/random';
-import { withClientProvider } from '@dxos/react-client/testing';
-import { withLayout, withSignals, withTheme } from '@dxos/storybook-utils';
-
-import { Thread, type ThreadProps } from './Thread';
-import translations from '../../translations';
-
-faker.seed(1);
-
-const DefaultStory = ({ messages: _messages, ...props }: ThreadProps) => {
-  const [processing, setProcessing] = useState(false);
-  const [messages, setMessages] = useState<Message[]>(_messages ?? []);
-  useEffect(() => {
-    setMessages(_messages ?? []);
-  }, [_messages]);
-
-  const handleSubmit = useCallback(
-    (text: string) => {
-      const request: Message = Obj.make(Message, { role: 'user', content: [{ type: 'text', text }] });
-      const response: Message = Obj.make(Message, {
-        role: 'assistant',
-        content: [{ type: 'text', disposition: 'cot', pending: true, text: faker.lorem.paragraphs(1) }],
-      });
-      setMessages([...messages, request, response]);
-      setProcessing(true);
-      setTimeout(() => {
-        response.content[0].pending = false;
-        setMessages([
-          ...messages,
-          request,
-          response,
-          Obj.make(Message, {
-            role: 'assistant',
-            content: [{ type: 'text', text: faker.lorem.paragraphs(1) }],
-          }),
-        ]);
-        setProcessing(false);
-      }, 3_000);
-    },
-    [messages],
-  );
-
-  return (
-    <div className='flex w-[30rem] bg-white dark:bg-black'>
-      <Thread
-        {...props}
-        messages={messages}
-        processing={processing}
-        onSubmit={handleSubmit}
-        onCancel={() => setProcessing(false)}
-      />
-    </div>
-  );
-};
-
-const meta: Meta<ThreadProps> = {
-  title: 'plugins/plugin-assistant/Thread',
-  render: DefaultStory,
-  component: Thread,
-  decorators: [
-    withSignals,
-    withClientProvider({ createIdentity: true, createSpace: true }),
-    withPluginManager({ plugins: [IntentPlugin()] }),
-    withTheme,
-    withLayout({ fullscreen: true, classNames: 'flex justify-center' }),
-  ],
-  parameters: {
-    translations,
-  },
-};
-
-export default meta;
-
-type Story = StoryObj<ThreadProps>;
-
-const TEST_MESSAGES: Message[] = [
-  Obj.make(Message, {
-    role: 'user',
-    content: [
-      {
-        type: 'text',
-        text: faker.lorem.sentence(5),
-      },
-    ],
-  }),
-  Obj.make(Message, {
-    role: 'assistant',
-    content: [
-      {
-        type: 'text',
-        disposition: 'cot',
-        text: Array.from({ length: faker.number.int({ min: 3, max: 5 }) })
-          .map((_, idx) => `${idx + 1}. ${faker.lorem.paragraph()}`)
-          .join('\n'),
-      },
-      {
-        type: 'text',
-        text: Array.from({ length: faker.number.int({ min: 2, max: 5 }) })
-          .map(() => faker.lorem.paragraphs())
-          .join('\n\n'),
-      },
-      {
-        type: 'tool_use',
-        id: '1234',
-        name: 'search',
-        input: {},
-      },
-    ],
-  }),
-  Obj.make(Message, {
-    role: 'user',
-    content: [
-      {
-        type: 'tool_result',
-        toolUseId: '1234',
-        content: 'This is a tool result.',
-      },
-    ],
-  }),
-  Obj.make(Message, {
-    role: 'assistant',
-    content: [
-      {
-        type: 'tool_use',
-        id: '4567',
-        name: 'create',
-        input: {},
-      },
-    ],
-  }),
-  Obj.make(Message, {
-    role: 'user',
-    content: [
-      {
-        type: 'tool_result',
-        toolUseId: '4567',
-        content: 'This is a tool result.',
-      },
-    ],
-  }),
-  Obj.make(Message, {
-    role: 'assistant',
-    content: [
-      {
-        type: 'text',
-        text: faker.lorem.paragraphs(1),
-      },
-    ],
-  }),
-];
-
-export const Default: Story = {
-  args: {
-    messages: TEST_MESSAGES,
-  },
-};
-
-export const Input: Story = {
-  args: {
-    processing: true,
-  },
-};
-
-export const Collapse: Story = {
-  args: {
-    collapse: true,
-    messages: TEST_MESSAGES,
-  },
-};
-
-export const Incremental: Story = {
-  render: () => {
-    const [messages, setMessages] = useState<Message[]>([]);
-    useEffect(() => {
-      let i = 0;
-      const interval = setInterval(() => {
-        setMessages((messages) => [...messages, TEST_MESSAGES[i++]]);
-        if (i >= TEST_MESSAGES.length) {
-          clearInterval(interval);
-        }
-      }, 2_000);
-
-      return () => clearInterval(interval);
-    }, []);
-
-    return <DefaultStory messages={messages} collapse />;
-  },
-};