@dxos/plugin-assistant 0.8.4-main.f5c0578 → 0.8.4-main.fcc0d83b33

package/src/components/ChatThread/ToolBlock.tsx

@@ -1,125 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { type FC, useEffect, useMemo, useRef, useState } from 'react';
-
-import { type AgentStatus, type Tool } from '@dxos/ai';
-import { type ThemedClassName, useTranslation } from '@dxos/react-ui';
-import { NumericTabs, StatusRoll, ToggleContainer } from '@dxos/react-ui-components';
-import { type JsonProps, Json as NativeJson } from '@dxos/react-ui-syntax-highlighter';
-import { type DataType } from '@dxos/schema';
-import { isNonNullable, isNotFalsy } from '@dxos/util';
-
-import { meta } from '../../meta';
-
-export const isToolMessage = (message: DataType.Message) => {
-  return message.blocks.some((block) => block._tag === 'toolCall' || block._tag === 'toolResult');
-};
-
-const getToolName = (tool: Tool) => {
-  return tool.namespace && tool.function ? `${tool.namespace}:${tool.function}` : tool.name.split('_').pop();
-};
-
-export type ToolBlockProps = ThemedClassName<{
-  message: DataType.Message;
-  tools?: Tool[];
-}>;
-
-export const ToolBlock: FC<ToolBlockProps> = ({ classNames, message, tools }) => {
-  const { t } = useTranslation(meta.id);
-  const { blocks = [] } = message;
-
-  const getToolCaption = (tool?: Tool, status?: AgentStatus) => {
-    if (!tool) {
-      return t('calling tool label');
-    }
-
-    return status?.message ?? tool.caption ?? [t('calling label') + getToolName(tool)].join(' ');
-  };
-
-  let request: { tool: Tool | undefined; block: any } | undefined;
-  const toolBlocks = blocks.filter((block) => block._tag === 'toolCall' || block._tag === 'toolResult');
-  const items = toolBlocks
-    .map((block) => {
-      switch (block._tag) {
-        case 'toolCall': {
-          // TODO(burdon): Skip these updates?
-          if (block.pending && request?.block.toolCallId === block.toolCallId) {
-            return null;
-          }
-
-          request = { tool: tools?.find((tool) => tool.name === block.name), block };
-          return {
-            title: getToolCaption(request.tool, request.block.status),
-            block,
-          };
-        }
-
-        case 'toolResult': {
-          if (!request || block.error) {
-            return {
-              title: t('error label'),
-              block,
-            };
-          }
-
-          return {
-            title: getToolCaption(request.tool, request.block.status),
-            block,
-          };
-        }
-
-        default: {
-          request = undefined;
-          return {
-            title: t('error label'),
-            block,
-          };
-        }
-      }
-    })
-    .filter(isNonNullable);
-
-  return <ToolContainer classNames={classNames} items={items} />;
-};
-
-export const ToolContainer: FC<ThemedClassName<{ items: { title: string; block: any }[] }>> = ({
-  classNames,
-  items,
-}) => {
-  const tabsRef = useRef<HTMLDivElement>(null);
-  const [selected, setSelected] = useState(0);
-  const [open, setOpen] = useState(false);
-  useEffect(() => {
-    if (open) {
-      tabsRef.current?.focus();
-    }
-  }, [open]);
-
-  const handleSelect = (index: number) => {
-    if (index === selected) {
-      setOpen(false);
-    } else {
-      setSelected(index);
-    }
-  };
-
-  const title = useMemo(() => {
-    const lines = items.map((item) => item.title).filter(isNotFalsy);
-    return <StatusRoll key='status-roll' lines={lines} duration={1_000} autoAdvance />;
-  }, [items]);
-
-  return (
-    <ToggleContainer classNames={['flex flex-col', classNames]} title={title} open={open} onChangeOpen={setOpen}>
-      <div className='w-full grid grid-cols-[32px_1fr]'>
-        <NumericTabs ref={tabsRef} length={items.length} selected={selected} onSelect={handleSelect} />
-        <Json data={items[selected].block} />
-      </div>
-    </ToggleContainer>
-  );
-};
-
-export const Json = ({ data }: Pick<JsonProps, 'data'>) => (
-  <NativeJson data={data} classNames='!p-1 text-xs bg-transparent' />
-);

package/src/components/ChatThread/reducer.ts

@@ -1,52 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type DataType } from '@dxos/schema';
-
-// TODO(burdon): Move to util?
-type Reducer<R, I> = (acc: R, value: I) => R;
-
-/**
- * Reducer that collapses related message blocks into single messages.
- * For example, combines tool request/response pairs into a single message.
- */
-export const messageReducer: Reducer<{ messages: DataType.Message[]; current?: DataType.Message }, DataType.Message> = (
-  { current, messages },
-  message,
-) => {
-  let i = 0;
-  for (const block of message.blocks) {
-    switch (block._tag) {
-      case 'toolCall':
-      case 'toolResult': {
-        if (current) {
-          current.blocks.push(block);
-        } else {
-          current = {
-            ...message,
-            id: [message.id, i].join('_'),
-            blocks: [block],
-          };
-          messages.push(current);
-        }
-        break;
-      }
-
-      case 'text':
-      default: {
-        current = undefined;
-        messages.push({
-          ...message,
-          id: [message.id, i].join('_'),
-          blocks: [block],
-        });
-        break;
-      }
-    }
-
-    i++;
-  }
-
-  return { current, messages };
-};

package/src/components/PromptSettings.tsx

@@ -1,89 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React from 'react';
-
-import { type Template } from '@dxos/blueprints';
-import { Toolbar, useTranslation } from '@dxos/react-ui';
-
-import { meta } from '../meta';
-
-export const PromptSettings = ({ template }: { template: Template.Template }) => {
-  const { t } = useTranslation(meta.id);
-
-  // const handleKindChange = useCallback(
-  //   (value: string) => {
-  //     const kind = Match.type<string>().pipe(
-  //       Match.withReturnType<Schema.Schema.Type<typeof Template.Template>>(),
-  //       Match.when('always', () => ({ include: 'always' })),
-  //       Match.when('schema-matching', () => ({ include: 'schema-matching', typename: '' })),
-  //       Match.when('automatically', () => ({ include: 'automatically', description: '' })),
-  //       Match.orElse(() => ({ include: 'manual' })),
-  //     )(value);
-  //
-  //     template.kind = kind;
-  //   },
-  //   [template],
-  // );
-
-  // const handleTypenameChange = useCallback(
-  //   debounce((event: ChangeEvent<HTMLInputElement>) => {
-  //     if (template.kind.include === 'schema-matching') {
-  //       template.kind.typename = event.target.value;
-  //     }
-  //   }, 300),
-  //   [template.kind.include],
-  // );
-
-  // const handleDescriptionChange = useCallback(
-  //   debounce((event: ChangeEvent<HTMLInputElement>) => {
-  //     if (template.kind.include === 'automatically') {
-  //       template.kind.description = event.target.value;
-  //     }
-  //   }, 300),
-  //   [template.kind.include],
-  // );
-
-  return (
-    <div className='flex flex-col gap-4'>
-      <h2>{t('prompt rules label')}</h2>
-      <Toolbar.Root>
-        {/* <Select.Root value={template.kind.include} onValueChange={handleKindChange}>
-          <Toolbar.Button asChild>
-            <Select.TriggerButton />
-          </Toolbar.Button>
-          <Select.Portal>
-            <Select.Content>
-              <Select.Viewport>
-                {Template.Kinds.map((kind) => (
-                  <Select.Option key={kind} value={kind}>
-                    {kind}
-                  </Select.Option>
-                ))}
-              </Select.Viewport>
-            </Select.Content>
-          </Select.Portal>
-        </Select.Root> */}
-        {/* {template.kind.include === 'schema-matching' && (
-          <Input.Root>
-            <Input.TextInput
-              placeholder={t('typename placeholder')}
-              defaultValue={template.kind.typename}
-              onChange={handleTypenameChange}
-            />
-          </Input.Root>
-        )} */}
-        {/* {template.kind.include === 'automatically' && (
-          <Input.Root>
-            <Input.TextInput
-              placeholder={t('description placeholder')}
-              defaultValue={template.kind.description}
-              onChange={handleDescriptionChange}
-            />
-          </Input.Root>
-        )} */}
-      </Toolbar.Root>
-    </div>
-  );
-};

package/src/components/SequenceContainer.tsx

@@ -1,133 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type EditorView } from '@codemirror/view';
-import JSON5 from 'json5';
-import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react';
-
-import { ToolRegistry } from '@dxos/ai';
-import { getSpace } from '@dxos/client/echo';
-import { type Sequence, type SequenceDefinition } from '@dxos/conductor';
-import { Key } from '@dxos/echo';
-import { Toolbar, useTranslation } from '@dxos/react-ui';
-import { useSelectionManager } from '@dxos/react-ui-attention';
-import { StackItem } from '@dxos/react-ui-stack';
-
-import { meta } from '../meta';
-
-import { SequenceEditor } from './SequenceEditor';
-
-// TODO(burdon): Move to config.
-export const EXA_API_KEY = '9c7e17ff-0c85-4cd5-827a-8b489f139e03';
-
-// TODO(burdon): Validate.
-const parseSequence = (text: string): SequenceDefinition | undefined => {
-  try {
-    const json = JSON5.parse(text);
-    const { steps } = json;
-    return { steps };
-  } catch (error) {
-    // Ignore.
-  }
-};
-
-export const SequenceContainer = ({ sequence }: { sequence: Sequence }) => {
-  const { t } = useTranslation(meta.id);
-  const selectionManager = useSelectionManager();
-  const [definition, setDefinition] = useState<SequenceDefinition>();
-  useEffect(() => {
-    setDefinition({
-      steps: sequence.steps.map(({ instructions, tools }) => ({ instructions, tools })),
-    });
-  }, [sequence]);
-
-  const editorRef = useRef<EditorView | undefined>(undefined);
-
-  // TODO(burdon): Factor out.
-  const toolRegistry = useMemo(() => {
-    const space = getSpace(sequence);
-    if (!space) {
-      return;
-    }
-
-    // TODO(burdon): How should the queue be created?
-    // eslint-disable-next-line no-unused-vars
-    const _queue = space.queues.create();
-
-    return new ToolRegistry([
-      // createExaTool({ apiKey: EXA_API_KEY }),
-      // createLocalSearchTool(space.db, queue),
-      // createGraphWriterTool({
-      //   db: space.db,
-      //   queue,
-      //   schema: [], // TODO(burdon): Get schema from client/sequence?
-      //   onDone: async (objects) => {
-      //     await queue.append(objects);
-      //   },
-      // }),
-    ]);
-  }, [sequence]);
-
-  const formatAndSave = useCallback((): SequenceDefinition | undefined => {
-    if (!sequence) {
-      return;
-    }
-
-    const text = editorRef.current?.state.doc.toString();
-    if (!text) {
-      return;
-    }
-
-    const definition = parseSequence(text);
-    if (!definition) {
-      return;
-    }
-
-    setDefinition(definition);
-    const formatted = JSON.stringify(definition, null, 2);
-    editorRef.current?.dispatch({
-      changes: { from: 0, to: text.length, insert: formatted },
-    });
-
-    sequence.steps.length = 0;
-    for (const step of definition.steps) {
-      sequence.steps.push({
-        id: Key.ObjectId.random(),
-        instructions: step.instructions,
-        tools: step.tools,
-      });
-    }
-
-    return sequence;
-  }, [sequence]);
-
-  // TODO(burdon): Save raw sequence separately from parsed sequence? (like Script).
-  const handleSave = useCallback(() => formatAndSave(), [formatAndSave]);
-
-  const handleRun = useCallback(async () => {
-    // if (!aiClient?.value || !toolRegistry) {
-    //   return;
-    // }
-    // formatAndSave();
-    // // Get input from selection.
-    // const input = Array.from(getSelectionSet(selectionManager)).map((id) => DXN.fromLocalObjectId(id));
-    // if (!input.length) {
-    //   return;
-    // }
-    // const machine = new SequenceMachine(toolRegistry, sequence).setLogger(new QueueLogger(sequence));
-    // await machine.runToCompletion({ aiClient: aiClient.value, input });
-  }, [sequence, formatAndSave, selectionManager, toolRegistry]);
-
-  return (
-    <StackItem.Content toolbar>
-      <Toolbar.Root>
-        <Toolbar.Button onClick={handleSave}>{t('button save')}</Toolbar.Button>
-        <Toolbar.Button onClick={handleRun}>{t('button run')}</Toolbar.Button>
-      </Toolbar.Root>
-      {definition && <SequenceEditor ref={editorRef} sequence={definition} />}
-    </StackItem.Content>
-  );
-};
-
-export default SequenceContainer;

package/src/components/SequenceEditor/SequenceEditor.stories.tsx

@@ -1,33 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import '@dxos-theme';
-
-import { type Meta, type StoryObj } from '@storybook/react-vite';
-
-import { withLayout, withTheme } from '@dxos/storybook-utils';
-
-import { RESEARCH_SEQUENCE_DEFINITION } from '../../testing';
-import { translations } from '../../translations';
-
-import { SequenceEditor } from './SequenceEditor';
-
-const meta: Meta<typeof SequenceEditor> = {
-  title: 'plugins/plugin-assistant/SequenceEditor',
-  component: SequenceEditor,
-  decorators: [withTheme, withLayout({ fullscreen: true, classNames: 'mli-auto max-is-[50rem] justify-center' })],
-  parameters: {
-    translations,
-  },
-};
-
-export default meta;
-
-type Story = StoryObj<typeof SequenceEditor>;
-
-export const Default: Story = {
-  args: {
-    sequence: RESEARCH_SEQUENCE_DEFINITION,
-  },
-};

package/src/components/SequenceEditor/SequenceEditor.tsx

@@ -1,50 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { forwardRef, useImperativeHandle } from 'react';
-
-import { SequenceDefinition } from '@dxos/conductor';
-import { Type } from '@dxos/echo';
-import { type ThemedClassName, useThemeContext } from '@dxos/react-ui';
-import {
-  type EditorView,
-  createBasicExtensions,
-  createJsonExtensions,
-  createThemeExtensions,
-  useTextEditor,
-} from '@dxos/react-ui-editor';
-import { mx } from '@dxos/react-ui-theme';
-
-export type SequenceEditorProps = ThemedClassName<{
-  sequence: SequenceDefinition;
-}>;
-
-// TODO(burdon): Factor out JsonEditor.
-export const SequenceEditor = forwardRef<EditorView | undefined, SequenceEditorProps>(
-  ({ classNames, sequence }, forwardedRef) => {
-    const { themeMode } = useThemeContext();
-    const { parentRef, view } = useTextEditor({
-      initialValue: JSON.stringify(sequence, null, 2),
-      extensions: [
-        createBasicExtensions({
-          lineNumbers: true,
-          lineWrapping: false,
-          monospace: true,
-          scrollPastEnd: true,
-        }),
-        createThemeExtensions({
-          themeMode,
-          syntaxHighlighting: true,
-        }),
-        createJsonExtensions({
-          schema: Type.toJsonSchema(SequenceDefinition, { strict: true }),
-        }),
-      ],
-    });
-
-    useImperativeHandle(forwardedRef, () => view, [view]);
-
-    return <div ref={parentRef} className={mx('overflow-hidden', classNames)} />;
-  },
-);

package/src/components/SequenceEditor/index.ts

@@ -1,5 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-export * from './SequenceEditor';

package/src/components/Timeline/Timeline.stories.tsx

@@ -1,144 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import '@dxos-theme';
-
-import { type Meta, type StoryObj } from '@storybook/react-vite';
-import React, { useRef, useState } from 'react';
-
-import { LogLevel } from '@dxos/log';
-import { faker } from '@dxos/random';
-import { Button, Toolbar, useInterval } from '@dxos/react-ui';
-import { ScrollContainer, type ScrollController } from '@dxos/react-ui-components';
-import { withLayout, withTheme } from '@dxos/storybook-utils';
-
-import { type Branch, type Commit, IconType, Timeline } from './Timeline';
-
-faker.seed(1);
-
-const meta: Meta<typeof Timeline> = {
-  title: 'plugins/plugin-assistant/Timeline',
-  component: Timeline,
-  decorators: [withTheme, withLayout({ fullscreen: true })],
-};
-
-export default meta;
-
-type Story = StoryObj<typeof meta>;
-
-export const Default: Story = {
-  args: {
-    branches: [{ name: 'main' }, { name: 'feature-a' }, { name: 'feature-b' }, { name: 'feature-c' }],
-    commits: [
-      { id: 'c1', message: faker.lorem.paragraph(), branch: 'main' },
-      { id: 'c2', message: faker.lorem.paragraph(), branch: 'main', parent: 'c1' },
-      { id: 'c3', message: faker.lorem.paragraph(), branch: 'feature-a', parent: 'c2' },
-      { id: 'c4', message: faker.lorem.paragraph(), branch: 'main', parent: 'c2' },
-      { id: 'c5', message: faker.lorem.paragraph(), branch: 'feature-b', parent: 'c2' },
-      { id: 'c6', message: faker.lorem.paragraph(), branch: 'feature-a', parent: 'c3' },
-      { id: 'c7', message: faker.lorem.paragraph(), branch: 'feature-a', parent: 'c6' },
-      { id: 'c8', message: faker.lorem.paragraph(), branch: 'feature-c', parent: 'c6' },
-      { id: 'c9', message: faker.lorem.paragraph(), branch: 'main', parent: 'c4' },
-    ],
-  },
-};
-
-export const Linear: Story = {
-  args: {
-    branches: [{ name: 'main' }],
-    commits: [
-      { id: 'c1', message: faker.lorem.paragraph(), branch: 'main' },
-      { id: 'c2', message: faker.lorem.paragraph(), branch: 'main', parent: 'c1' },
-      { id: 'c3', message: faker.lorem.paragraph(), branch: 'main', parent: 'c2' },
-      { id: 'c4', message: faker.lorem.paragraph(), branch: 'main', parent: 'c3' },
-    ],
-  },
-};
-
-export const Empty: Story = {
-  args: {
-    branches: [],
-    commits: [],
-  },
-};
-
-export const Random: Story = {
-  render: () => {
-    const [branches, setBranches] = useState<Branch[]>([{ name: 'main' }]);
-    const [commits, setCommits] = useState<Commit[]>([
-      {
-        id: faker.string.uuid(),
-        branch: branches[0].name,
-        message: faker.lorem.paragraph(),
-      },
-    ]);
-    const lastCommit = useRef<string | undefined>(commits[0].id);
-    const lastBranch = useRef<string>(branches[0].name);
-
-    const [running, setRunning] = useState(true);
-    useInterval(
-      () => {
-        if (!running) {
-          return;
-        }
-
-        const p = Math.random();
-        if (p < 0.15 && branches.length < 6) {
-          const branch = { name: faker.lorem.word() } satisfies Branch;
-          setBranches((branches) => [...branches, branch]);
-          lastBranch.current = branch.name;
-        } else if (p < 0.4) {
-          const branch = branches[Math.floor(Math.random() * branches.length)];
-          lastBranch.current = branch.name;
-        }
-
-        const commit = {
-          id: faker.string.uuid(),
-          branch: lastBranch.current,
-          icon: faker.helpers.arrayElement([
-            IconType.WARN,
-            IconType.CHECK,
-            IconType.ROCKET,
-            IconType.X,
-            IconType.FLAG,
-            IconType.TIMER,
-            IconType.USER,
-            IconType.USER_INTERACTION,
-            IconType.AGENT,
-          ]),
-          level: faker.helpers.arrayElement([
-            LogLevel.TRACE,
-            LogLevel.DEBUG,
-            LogLevel.VERBOSE,
-            LogLevel.INFO,
-            LogLevel.WARN,
-            LogLevel.ERROR,
-          ]),
-          message: faker.lorem.paragraph(),
-          parent: lastCommit.current,
-        } satisfies Commit;
-        lastCommit.current = commit.id;
-        setCommits((commits) => [...commits, commit]);
-      },
-      500,
-      [running],
-    );
-
-    const scrollerRef = useRef<ScrollController>(null);
-
-    return (
-      <div className='flex flex-col is-full bs-full overflow-hidden'>
-        <Toolbar.Root>
-          <Button onClick={() => setRunning(true)}>Start</Button>
-          <Button onClick={() => setRunning(false)}>Stop</Button>
-          <Button onClick={() => scrollerRef.current?.scrollToTop()}>Top</Button>
-          <Button onClick={() => scrollerRef.current?.scrollToBottom()}>Bottom</Button>
-        </Toolbar.Root>
-        <ScrollContainer ref={scrollerRef} classNames='border border-separator'>
-          <Timeline branches={branches} commits={commits} />
-        </ScrollContainer>
-      </div>
-    );
-  },
-};