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

package/src/components/ChatThread/ChatMessage.tsx

@@ -1,268 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { Fragment, type FC, type PropsWithChildren } from 'react';
-
-import { type Tool } from '@dxos/ai';
-import { Surface } from '@dxos/app-framework';
-import { invariant } from '@dxos/invariant';
-import { type Space } from '@dxos/react-client/echo';
-import { Button, Icon, IconButton, useTranslation, type ThemedClassName } 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 { Json, ToolBlock, isToolMessage } from './ToolBlock';
-import { type ChatProcessor } from '../../hooks';
-import { meta } from '../../meta';
-import { type ChatEvent } from '../Chat';
-import { ToolboxContainer } from '../Toolbox';
-
-export type ChatMessageProps = ThemedClassName<{
-  debug?: boolean;
-  space?: Space;
-  message: DataType.Message;
-  // TODO(burdon): Move to context.
-  processor?: ChatProcessor;
-  tools?: Tool[];
-  onEvent?: (event: ChatEvent) => void;
-}>;
-
-export const ChatMessage = ({ classNames, debug, space, message, processor, tools, onEvent }: ChatMessageProps) => {
-  const { t } = useTranslation(meta.id);
-  const {
-    sender: { role },
-    blocks,
-  } = message;
-
-  // TODO(burdon): Restructure types to make check unnecessary.
-  if (isToolMessage(message)) {
-    return (
-      <MessageContainer classNames={mx(classNames, 'animate-[fadeIn_0.5s]')}>
-        <ToolBlock classNames={panelClasses} message={message} tools={tools} />
-      </MessageContainer>
-    );
-  }
-
-  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: BlockComponent = components[block._tag] ?? components.default!;
-        if (!Component) {
-          return null;
-        }
-
-        return (
-          <Fragment key={idx}>
-            <MessageContainer classNames={classNames} user={block._tag === 'text' && role === 'user'}>
-              <Component space={space} processor={processor} block={block} onEvent={onEvent} />
-            </MessageContainer>
-            {debug && (
-              <div className={mx('flex justify-end text-subdued', marginClasses)}>
-                <pre className='text-xs'>{JSON.stringify({ block: block._tag })}</pre>
-              </div>
-            )}
-          </Fragment>
-        );
-      })}
-      <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={() => onEvent?.({ type: 'delete', id: message.id })}
-        />
-      </div>
-    </>
-  );
-};
-
-type BlockComponentProps = {
-  space?: Space;
-  block: ContentBlock.Any;
-  /** @deprecated Replace with context */
-  processor?: ChatProcessor;
-  onEvent?: (event: ChatEvent) => void;
-};
-
-type BlockComponent = FC<BlockComponentProps>;
-
-const components: Partial<Record<ContentBlock.Any['_tag'] | 'default', BlockComponent>> = {
-  //
-  // Text
-  //
-  ['text' as const]: ({ block }) => {
-    invariant(block._tag === 'text');
-    // const [open, setOpen] = useState(block.disposition === 'cot' && block.pending);
-    const title = block.disposition ? titles[block.disposition] : undefined;
-    if (!title) {
-      return <MarkdownViewer content={block.text} />;
-    }
-
-    // TOOD(burdon): Store last time user opened/closed COT.
-    // Autoclose when streaming ends.
-    // useEffect(() => {
-    //   if (block.disposition === 'cot' && !block.pending) {
-    //     setOpen(false);
-    //   }
-    // }, [block.disposition, block.pending]);
-
-    return (
-      <ToggleContainer
-        // open={open}
-        defaultOpen={systemDispositions.includes(block.disposition ?? '') && block.pending}
-        title={title}
-        icon={
-          block.pending ? (
-            <Icon icon={'ph--circle-notch--regular'} classNames='text-subdued animate-spin' size={4} />
-          ) : undefined
-        }
-      >
-        <MarkdownViewer
-          content={block.text}
-          classNames={['pbe-2', systemDispositions.includes(block.disposition ?? '') && 'text-sm text-subdued']}
-        />
-      </ToggleContainer>
-    );
-  },
-
-  //
-  // JSON
-  //
-  ['json' as const]: ({ space, processor, block, onEvent }) => {
-    invariant(block._tag === 'json');
-
-    switch (block.disposition) {
-      case 'tool_list': {
-        return (
-          <ToggleContainer title={titles[block.disposition]} defaultOpen={true}>
-            <ToolboxContainer space={space} processor={processor} classNames='pbe-2' />
-          </ToggleContainer>
-        );
-      }
-
-      case 'suggest': {
-        const { text = '' }: { text: string } = safeParseJson(block.data ?? '{}') ?? ({} as any);
-        return (
-          <IconButton icon='ph--lightning--regular' label={text} onClick={() => onEvent?.({ type: 'submit', text })} />
-        );
-      }
-
-      case 'select': {
-        const { options = [] }: { options: string[] } = safeParseJson(block.data ?? '{}') ?? ({} as any);
-        return (
-          <div className='flex flex-wrap gap-1'>
-            {options.map((text, idx) => (
-              <Button
-                classNames={'animate-[fadeIn_0.5s] rounded-sm text-sm'}
-                key={idx}
-                onClick={() => onEvent?.({ type: 'submit', text })}
-              >
-                {text}
-              </Button>
-            ))}
-          </div>
-        );
-      }
-
-      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: {
-        const title = block.disposition ? titles[block.disposition] : undefined;
-        return (
-          <ToggleContainer title={title ?? 'JSON'}>
-            <Json data={safeParseJson(block.data ?? block)} />
-          </ToggleContainer>
-        );
-      }
-    }
-  },
-
-  //
-  // Default
-  //
-  default: ({ block }) => {
-    let title = titles[block._tag];
-    if (block._tag === 'toolCall') {
-      title = `Tool [${block.name}]`; // TODO(burdon): Get label from tool.
-    } else {
-      title = block._tag;
-    }
-
-    return (
-      <ToggleContainer title={title ?? 'JSON'}>
-        <Json data={block} />
-      </ToggleContainer>
-    );
-  },
-};
-
-// TODO(burdon): Translations.
-const titles: Record<string, string> = {
-  ['cot' as const]: 'Chain of thought',
-  ['artifact' as const]: 'Artifact',
-  ['tool_use' as const]: 'Tool request',
-  ['tool_result' as const]: 'Tool result',
-  ['tool_list' as const]: 'Tools',
-  ['artifact-update' as const]: 'Artifact(s) changed',
-};
-
-const systemDispositions: string[] = ['cot', 'artifact-update'];
-
-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';
-
-const MessageContainer = ({ 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)} />;
-};

package/src/components/ChatThread/ToolBlock.tsx

@@ -1,121 +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 { log } from '@dxos/log';
-import { type ThemedClassName } 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';
-
-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();
-};
-
-const getToolCaption = (tool: Tool | undefined, status: AgentStatus | undefined) => {
-  if (!tool) {
-    return 'Calling tool...';
-  }
-
-  return status?.message ?? tool.caption ?? `Calling ${getToolName(tool)}...`;
-};
-
-export type ToolBlockProps = ThemedClassName<{
-  message: DataType.Message;
-  tools?: Tool[];
-}>;
-
-export const ToolBlock: FC<ToolBlockProps> = ({ classNames, message, tools }) => {
-  const { blocks = [] } = message;
-
-  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, undefined), // block.status), // TODO(burdon): Get status?
-            block,
-          };
-        }
-
-        case 'toolResult': {
-          if (!request) {
-            log.warn('unexpected message', { block });
-            return { title: 'Error', block };
-          }
-
-          return {
-            title: `${getToolCaption(request.tool, undefined)} (Success)`,
-            block,
-          };
-        }
-
-        default: {
-          request = undefined;
-          return {
-            title: 'Error',
-            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/TemplateContainer.tsx

@@ -1,24 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React from 'react';
-
-import { type Template } from '@dxos/blueprints';
-import { StackItem } from '@dxos/react-ui-stack';
-
-import { TemplateEditor } from './TemplateEditor';
-
-export type TemplateContainerProps = {
-  template: Template.Template;
-};
-
-export const TemplateContainer = ({ template }: TemplateContainerProps) => {
-  return (
-    <StackItem.Content classNames='container-max-width'>
-      <TemplateEditor template={template} />
-    </StackItem.Content>
-  );
-};
-
-export default TemplateContainer;

package/src/hooks/chat-processor.test.ts

@@ -1,16 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { describe, it } from 'vitest';
-
-import { ChatProcessor } from './chat-processor';
-
-describe('ChatProcessor', () => {
-  it('should be instantiable', ({ expect }) => {
-    const services = {} as any; // TODO(burdon): Create mock.
-    const client = {} as any; // TODO(burdon): Create mock.
-    const processor = new ChatProcessor(services, client);
-    expect(processor).toBeDefined();
-  });
-});