@dxos/plugin-assistant 0.8.4-main.fd6878d → 0.8.4-staging.60fe92afc8

package/src/components/ChatThread/ChatMessage.tsx

@@ -1,282 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { type FC, Fragment, type PropsWithChildren } from 'react';
-
-import { type Tool } from '@dxos/ai';
-import { ErrorBoundary, Surface } from '@dxos/app-framework';
-import { Obj } from '@dxos/echo';
-import { invariant } from '@dxos/invariant';
-import { DXN, DXN_ECHO_REGEXP } from '@dxos/keys';
-import { type Space } from '@dxos/react-client/echo';
-import { Button, IconButton, type ThemedClassName, useTranslation } from '@dxos/react-ui';
-import {
-  MarkdownViewer,
-  ToggleContainer as NativeToggleContainer,
-  type ToggleContainerProps,
-} from '@dxos/react-ui-components';
-import { mx } from '@dxos/react-ui-theme';
-import { type ContentBlock, type DataType } from '@dxos/schema';
-import { safeParseJson } from '@dxos/util';
-
-import { meta } from '../../meta';
-import { type ChatEvent } from '../Chat';
-import { Toolbox } from '../Toolbox';
-
-import { ObjectLink } from './Link';
-import { Json, ToolBlock, isToolMessage } from './ToolBlock';
-
-const panelClasses = 'flex flex-col is-full bg-activeSurface rounded-sm';
-const marginClasses = 'pie-4 pis-4';
-const paddingClasses = 'pis-2 pie-2 pbs-0.5 pbe-0.5';
-
-export type ChatMessageProps = ThemedClassName<{
-  debug?: boolean;
-  space?: Space;
-  message: DataType.Message;
-  tools?: Tool[];
-  onEvent?: (event: ChatEvent) => void;
-  onDelete?: () => void;
-}>;
-
-export const ChatMessage = ({ classNames, debug, space, message, tools, onEvent, onDelete }: ChatMessageProps) => {
-  const { t } = useTranslation(meta.id);
-  const {
-    sender: { role },
-    blocks,
-  } = message;
-
-  // TODO(burdon): Consolidate tools upstream?
-  if (isToolMessage(message)) {
-    return (
-      <MessageItem classNames={mx(classNames, 'animate-[fadeIn_0.5s]')}>
-        <ToolBlock classNames={panelClasses} message={message} tools={tools} />
-      </MessageItem>
-    );
-  }
-
-  return (
-    <>
-      {debug && (
-        <div className={mx('flex justify-end text-subdued', marginClasses)}>
-          <pre className='text-xs'>{JSON.stringify({ created: message.created })}</pre>
-        </div>
-      )}
-
-      {blocks.map((block, idx) => {
-        // TODO(burdon): Filter empty messages.
-        if (block._tag === 'text' && block.text.replaceAll(/\s+/g, '').length === 0) {
-          return null;
-        }
-
-        const Component: ContentBlockComponent = components[block._tag] ?? components.default!;
-        if (!Component) {
-          return null;
-        }
-
-        return (
-          <Fragment key={idx}>
-            <MessageItem classNames={classNames} user={block._tag === 'text' && role === 'user'}>
-              <ErrorBoundary data={block}>
-                <Component space={space} block={block} onEvent={onEvent} />
-              </ErrorBoundary>
-            </MessageItem>
-            {debug && (
-              <div className={mx('flex justify-end text-subdued', marginClasses)}>
-                <pre className='text-xs'>{JSON.stringify({ block: block._tag })}</pre>
-              </div>
-            )}
-          </Fragment>
-        );
-      })}
-
-      {onDelete && (
-        <div className={mx('flex justify-end pbs-2 pbe-2 opacity-50 hover:opacity-100', marginClasses)}>
-          <IconButton
-            classNames='animate-[fadeIn_0.5s]'
-            icon='ph--trash--regular'
-            iconOnly
-            label={t('button delete message')}
-            onClick={() => onDelete()}
-          />
-        </div>
-      )}
-    </>
-  );
-};
-
-type ContentBlockProps = {
-  space?: Space;
-  block: ContentBlock.Any;
-  onEvent?: (event: ChatEvent) => void;
-};
-
-type ContentBlockComponent = FC<ContentBlockProps>;
-
-/**
- * Components for rendering content blocks.
- */
-const components: Partial<Record<ContentBlock.Any['_tag'] | 'default', ContentBlockComponent>> = {
-  //
-  // Text
-  //
-  ['text' as const]: ({ space, block }) => {
-    invariant(block._tag === 'text');
-    return (
-      <MarkdownViewer
-        content={preprocessTextContent(block.text)}
-        components={{
-          a: ({ node: { properties }, children, href, ...props }) => {
-            if (space && typeof properties?.href === 'string' && properties?.href?.startsWith('dxn')) {
-              try {
-                // TODO(burdon): Check valid length (since serialized).
-                const dxn = DXN.parse(properties.href);
-                return <ObjectLink space={space} dxn={dxn} />;
-              } catch {}
-            }
-
-            // TODO(burdon): Can we revert to the default handler?
-            return (
-              <a
-                href={href}
-                className='text-primary-500 hover:text-primary-500' // TODO(burdon): Token.
-                target='_blank'
-                rel='noopener noreferrer'
-                {...props}
-              >
-                {children}
-              </a>
-            );
-          },
-        }}
-      />
-    );
-  },
-
-  //
-  // Suggest
-  //
-  ['suggest' as const]: ({ block, onEvent }) => {
-    const { t } = useTranslation(meta.id);
-    invariant(block._tag === 'suggest');
-    return (
-      <IconButton
-        icon='ph--lightning--regular'
-        label={block.text}
-        title={t('button suggest')}
-        onClick={() => onEvent?.({ type: 'submit', text: block.text })}
-      />
-    );
-  },
-
-  //
-  // Select
-  //
-  ['select' as const]: ({ block, onEvent }) => {
-    const { t } = useTranslation(meta.id);
-    invariant(block._tag === 'select');
-    return (
-      <div className='flex flex-wrap gap-1'>
-        {block.options.map((option, idx) => (
-          <Button
-            classNames={'animate-[fadeIn_0.5s] rounded-sm text-sm'}
-            key={idx}
-            onClick={() => onEvent?.({ type: 'submit', text: option })}
-            title={t('button select option')}
-          >
-            {option}
-          </Button>
-        ))}
-      </div>
-    );
-  },
-
-  //
-  // Toolkit
-  //
-  ['toolkit' as const]: ({ block }) => {
-    invariant(block._tag === 'toolkit');
-    return (
-      <ToggleContainer title='Toolbox' classNames={panelClasses} defaultOpen>
-        <Toolbox classNames={marginClasses} />
-      </ToggleContainer>
-    );
-  },
-
-  //
-  // JSON
-  //
-  ['json' as const]: ({ block, onEvent }) => {
-    invariant(block._tag === 'json');
-
-    // TODO(burdon): Disposition is deprecated.
-    switch (block.disposition) {
-      case 'graph': {
-        return (
-          <div className='flex flex-wrap gap-1'>
-            <Surface
-              role='card'
-              data={{ subject: JSON.parse(block.data ?? '{}') }}
-              limit={1}
-              fallback={<div className='font-mono text-xs text-pre'>{block.data}</div>}
-            />
-            {onEvent && (
-              <IconButton
-                icon='ph--plus--regular'
-                label='Add to graph'
-                onClick={() => onEvent?.({ type: 'add', object: JSON.parse(block.data ?? '{}') })}
-              />
-            )}
-          </div>
-        );
-      }
-
-      default: {
-        return (
-          <ToggleContainer title={block.disposition ?? block._tag}>
-            <Json data={safeParseJson(block.data ?? block)} />
-          </ToggleContainer>
-        );
-      }
-    }
-  },
-
-  //
-  // Fallback
-  //
-  default: ({ block }) => {
-    return (
-      <ToggleContainer title={block._tag}>
-        <Json data={block} />
-      </ToggleContainer>
-    );
-  },
-};
-
-/**
- * Wrapper for each message.
- */
-const MessageItem = ({ classNames, children, user }: ThemedClassName<PropsWithChildren<{ user?: boolean }>>) => {
-  if (!children) {
-    return null;
-  }
-
-  return (
-    <div role='list-item' className={mx('flex is-full', user && 'justify-end', marginClasses, classNames)}>
-      <div className={mx(user ? ['rounded-sm', 'bg-[--user-fill] text-accentSurfaceText', paddingClasses] : 'is-full')}>
-        {children}
-      </div>
-    </div>
-  );
-};
-
-const ToggleContainer = (props: ToggleContainerProps) => {
-  return <NativeToggleContainer {...props} classNames={mx(panelClasses, props.classNames)} />;
-};
-
-export const renderObjectLink = (obj: Obj.Any) => `[${Obj.getLabel(obj)}](${Obj.getDXN(obj).toString()})`;
-
-// TODO(burdon): Move to parser.
-const preprocessTextContent = (content: string) =>
-  content.replaceAll(new RegExp(DXN_ECHO_REGEXP, 'g'), (_, dxn) => `<${dxn}>`);

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;