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

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 />;
-  },
-};

package/src/components/Thread/Thread.tsx

@@ -1,118 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { type CSSProperties, useCallback, useMemo, useRef } from 'react';
-
-import { type Message } from '@dxos/ai';
-import { type Space } from '@dxos/react-client/echo';
-import { useIdentity } from '@dxos/react-client/halo';
-import { type ThemedClassName } from '@dxos/react-ui';
-import { ScrollContainer, type ScrollController } from '@dxos/react-ui-components';
-import { mx } from '@dxos/react-ui-theme';
-import { keyToFallback } from '@dxos/util';
-
-import { ThreadMessage, type ThreadMessageProps } from './ThreadMessage';
-import { messageReducer } from './reducer';
-import { PromptBar, type PromptBarProps } from '../Prompt';
-import type { ReferenceData, ReferencesProvider } from '../Prompt/references';
-
-export interface ContextProvider {
-  query({ query }: { query: string }): Promise<ReferenceData[]>;
-
-  resolveMetadata({ uri }: { uri: string }): Promise<ReferenceData | null>;
-}
-
-export type ThreadProps = ThemedClassName<{
-  space?: Space;
-  messages?: Message[];
-  collapse?: boolean;
-  transcription?: boolean;
-  onOpenChange?: (open: boolean) => void;
-  contextProvider?: ContextProvider;
-}> &
-  Pick<PromptBarProps, 'processing' | 'error' | 'onSubmit' | 'onSuggest' | 'onCancel'> &
-  Pick<ThreadMessageProps, 'debug' | 'tools' | 'onPrompt' | 'onDelete' | 'onAddToGraph'>;
-
-/**
- * Chat thread component.
- */
-export const Thread = ({
-  classNames,
-  space,
-  messages,
-  collapse = true,
-  transcription,
-  processing,
-  error,
-  onSubmit,
-  onCancel,
-  onOpenChange,
-  contextProvider,
-  ...props
-}: ThreadProps) => {
-  const scroller = useRef<ScrollController>(null);
-
-  const identity = useIdentity();
-  const fallbackValue = keyToFallback(identity!.identityKey);
-  const userHue = identity!.profile?.data?.hue || fallbackValue.hue;
-
-  const handleSubmit = useCallback<NonNullable<PromptBarProps['onSubmit']>>(
-    (value: string) => {
-      onSubmit?.(value);
-      scroller.current?.scrollToBottom();
-      return true;
-    },
-    [onSubmit],
-  );
-
-  // TODO(dmaretskyi): This needs to be a separate type: `id` is not a valid ObjectId, this needs to accommodate messageId for deletion.
-  const { messages: filteredMessages = [] } = useMemo(() => {
-    if (collapse) {
-      return (messages ?? []).reduce<{ messages: Message[]; current?: Message }>(messageReducer, {
-        messages: [],
-      });
-    } else {
-      return { messages: messages ?? [] };
-    }
-  }, [messages, collapse]);
-
-  const references = useMemo<ReferencesProvider | undefined>(() => {
-    if (!contextProvider) {
-      return undefined;
-    }
-
-    return {
-      getReferences: async ({ query }: { query: string }) => contextProvider.query({ query }),
-      resolveReference: async ({ uri }: { uri: string }) => contextProvider.resolveMetadata({ uri }),
-    };
-  }, [contextProvider]);
-
-  return (
-    <div role='none' className={mx('flex flex-col grow overflow-hidden', classNames)}>
-      <ScrollContainer ref={scroller} fade>
-        <div
-          role='none'
-          className={mx(filteredMessages.length > 0 && 'pbs-6 pbe-6')}
-          style={{ '--user-fill': `var(--dx-${userHue}Fill)` } as CSSProperties}
-        >
-          {filteredMessages.map((message) => (
-            <ThreadMessage key={message.id} classNames='px-4 pbe-4' space={space} message={message} {...props} />
-          ))}
-        </div>
-      </ScrollContainer>
-
-      {onSubmit && (
-        <PromptBar
-          microphone={transcription}
-          processing={processing}
-          error={error}
-          onSubmit={handleSubmit}
-          onCancel={onCancel}
-          onOpenChange={onOpenChange}
-          references={references}
-        />
-      )}
-    </div>
-  );
-};