@dxos/plugin-automation 0.7.5-labs.e27f9b9 → 0.7.5-labs.f400bbc

package/src/components/ChatContainer/ChatContainer.tsx

@@ -1,117 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { useCallback, useEffect, useMemo, useState } from 'react';
-
-import { Capabilities, useCapabilities, useCapability, useIntentDispatcher } from '@dxos/app-framework';
-import { createSystemPrompt, type Message, type Tool } from '@dxos/artifact';
-import { FunctionType } from '@dxos/functions';
-import { invariant } from '@dxos/invariant';
-import { DXN, QueueSubspaceTags } from '@dxos/keys';
-import { useConfig } from '@dxos/react-client';
-import { Filter, getSpace, useQuery } from '@dxos/react-client/echo';
-import { useEdgeClient, useQueue } from '@dxos/react-edge-client';
-import { StackItem } from '@dxos/react-ui-stack';
-import { isNotNullOrUndefined } from '@dxos/util';
-
-import { AutomationCapabilities } from '../../capabilities';
-import { ChatProcessor } from '../../hooks';
-import { covertFunctionToTool, createToolsFromService } from '../../tools';
-import { MockServiceRegistry, type AIChatType } from '../../types';
-import { Thread } from '../Thread';
-
-export const ChatContainer = ({ chat, role }: { chat: AIChatType; role: string }) => {
-  const config = useConfig();
-  const space = getSpace(chat);
-  const aiClient = useCapability(AutomationCapabilities.AiClient);
-
-  const artifactDefinitions = useCapabilities(Capabilities.ArtifactDefinition);
-  const globalTools = useCapabilities(Capabilities.Tools);
-  const functions = useQuery(space, Filter.schema(FunctionType));
-  const serviceRegistry = useMemo(() => new MockServiceRegistry(), []);
-
-  const [serviceTools, setServiceTools] = useState<Tool[]>([]);
-  useEffect(() => {
-    queueMicrotask(async () => {
-      const services = await serviceRegistry.queryServices({});
-      const tools = await Promise.all(services.map((service) => createToolsFromService(service)));
-      setServiceTools(tools.flat());
-    });
-  }, []);
-
-  const tools = useMemo(
-    () => [
-      ...globalTools.flat(),
-      ...artifactDefinitions.flatMap((definition) => definition.tools),
-      ...functions
-        .map((fn) => covertFunctionToTool(fn, config.values.runtime?.services?.edge?.url ?? '', space?.id))
-        .filter(isNotNullOrUndefined),
-      ...serviceTools,
-    ],
-    [globalTools, artifactDefinitions, functions, serviceTools, space?.id],
-  );
-  const systemPrompt = useMemo(
-    () => createSystemPrompt({ artifacts: artifactDefinitions.map((definition) => definition.instructions) }),
-    [artifactDefinitions],
-  );
-
-  // TODO(burdon): Create hook.
-  // TODO(wittjosiah): Should these be created in the component?
-  // TODO(zan): Combine with ai service client?
-  const { dispatchPromise: dispatch } = useIntentDispatcher();
-  const processor = useMemo(
-    () =>
-      new ChatProcessor(
-        aiClient,
-        tools,
-        {
-          space,
-          dispatch,
-        },
-        {
-          model: '@anthropic/claude-3-5-sonnet-20241022',
-          systemPrompt,
-        },
-      ),
-    [aiClient, tools, space, dispatch, systemPrompt],
-  );
-
-  // TODO(wittjosiah): Remove transformation.
-  const queueDxn = useMemo(
-    () => new DXN(DXN.kind.QUEUE, [QueueSubspaceTags.DATA, space!.id, chat.queue.dxn.parts.at(-1)!]),
-    [chat.queue.dxn],
-  );
-  const edgeClient = useEdgeClient();
-  const messageQueue = useQueue<Message>(edgeClient, queueDxn);
-  const messages = [...(messageQueue?.items ?? []), ...processor.messages.value];
-
-  const handleSubmit = useCallback(
-    async (message: string) => {
-      if (processor.streaming.value) {
-        await processor.cancel();
-      }
-
-      invariant(messageQueue);
-      await processor.request(message, {
-        history: messageQueue.items,
-        onComplete: (messages) => messageQueue.append(messages),
-      });
-    },
-    [processor, messageQueue],
-  );
-
-  const handleStop = useCallback(() => {
-    if (processor.streaming.value) {
-      void processor.cancel();
-    }
-  }, [processor]);
-
-  return (
-    <StackItem.Content toolbar={false} role={role}>
-      <Thread messages={messages} streaming={processor.streaming.value} onSubmit={handleSubmit} onStop={handleStop} />
-    </StackItem.Content>
-  );
-};
-
-export default ChatContainer;

package/src/components/ChatContainer/index.ts

@@ -1,8 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { ChatContainer } from './ChatContainer';
-
-export * from './ChatContainer';
-export default ChatContainer;

package/src/components/MarkdownViewer/MarkdownViewer.stories.tsx

@@ -1,56 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import '@dxos-theme';
-
-import { type StoryObj, type Meta } from '@storybook/react';
-
-import { faker } from '@dxos/random';
-import { withLayout, withTheme } from '@dxos/storybook-utils';
-
-import { MarkdownViewer } from './MarkdownViewer';
-
-faker.seed(0);
-
-const meta: Meta<typeof MarkdownViewer> = {
-  title: 'plugins/plugin-automation/MarkdownViewer',
-  component: MarkdownViewer,
-  decorators: [withTheme, withLayout({ fullscreen: true, classNames: 'justify-center' })],
-};
-
-export default meta;
-
-type Story = StoryObj<typeof MarkdownViewer>;
-
-const content = `
-# Hello, world!
-
-${faker.lorem.paragraphs(1)}
-
-Here's a JSON block:
-
-~~~json
-{
-  "hello": "world"
-}
-~~~
-
-And some code:
-
-~~~ts
-const App = () => {
-  const title = 'Hello, world!';
-  return <div>{title}</div>;
-};
-~~~
-
-${faker.lorem.paragraphs(1)}
-`;
-
-export const Default: Story = {
-  args: {
-    classNames: 'w-[500px] border border-border rounded-md p-4',
-    content,
-  },
-};

package/src/components/MarkdownViewer/MarkdownViewer.tsx

@@ -1,79 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { type FC } from 'react';
-import ReactMarkdown from 'react-markdown';
-
-import { type ThemedClassName } from '@dxos/react-ui';
-import { SyntaxHighlighter } from '@dxos/react-ui-syntax-highlighter';
-import { mx } from '@dxos/react-ui-theme';
-import { omit } from '@dxos/util';
-
-type MarkdownViewerProps = ThemedClassName<{
-  content?: string;
-}>;
-
-/**
- * Transform text into react elements.
- *
- * https://github.com/remarkjs/react-markdown
- * markdown -> remark -> [mdast -> remark plugins] -> [hast -> rehype plugins] -> components -> react elements.
- */
-// TODO(burdon): Styles.
-// TODO(burdon): Factor out (react-ui-syntax-highlighter).
-export const MarkdownViewer: FC<MarkdownViewerProps> = ({ classNames, content = '' }) => {
-  return (
-    <div className={mx('space-y-2', classNames)}>
-      <ReactMarkdown
-        components={{
-          a: ({ node, children, href, ...props }) => (
-            <a
-              href={href}
-              className='text-primary-500 hover:text-primary-500'
-              target='_blank'
-              rel='noopener noreferrer'
-              {...props}
-            >
-              {children}
-            </a>
-          ),
-          ol: ({ node, children, ...props }) => (
-            <ol className='leading-tight list-decimal pl-6' {...omit(props, ['ordered'])}>
-              {children}
-            </ol>
-          ),
-          ul: ({ node, children, ...props }) => (
-            <ul className='leading-tight list-disc pl-6' {...omit(props, ['ordered'])}>
-              {children}
-            </ul>
-          ),
-          li: ({ node, children, ...props }) => (
-            <li className='' {...omit(props, ['ordered'])}>
-              {children}
-            </li>
-          ),
-          blockquote: ({ node, children, ...props }) => (
-            <blockquote className='border-l-4 border-primary-500 pl-4 my-4 text-primary-500' {...props}>
-              {children}
-            </blockquote>
-          ),
-          code: ({ children, className }) => {
-            const [_, language] = /language-(\w+)/.exec(className || '') || [];
-            return (
-              <SyntaxHighlighter PreTag='div' language={language} className='p-0'>
-                {children}
-              </SyntaxHighlighter>
-            );
-          },
-        }}
-      >
-        {content}
-      </ReactMarkdown>
-    </div>
-  );
-};
-
-// const Cursor = () => {
-//   return <span className='animate-[pulse_1s_steps(1)_infinite] text-primary-500'>▊</span>;
-// };

package/src/components/MarkdownViewer/index.ts

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

package/src/components/PromptEditor/PromptEditor.stories.tsx

@@ -1,64 +0,0 @@
-//
-// Copyright 2023 DXOS.org
-//
-
-import '@dxos-theme';
-
-import { type Meta } from '@storybook/react';
-import React, { useState } from 'react';
-
-import { create, makeRef } from '@dxos/live-object';
-import { useClient } from '@dxos/react-client';
-import { withClientProvider } from '@dxos/react-client/testing';
-import { withLayout, withTheme } from '@dxos/storybook-utils';
-
-import { PromptEditor } from './PromptEditor';
-import translations from '../../translations';
-import { ChainPromptType, ChainType } from '../../types';
-
-const template = [
-  '# Comment',
-  '',
-  'You are a machine that is an expert chess player.',
-  'The move history of the current game is: {history}',
-  'If asked to suggest a move explain why it is a good move.',
-  '',
-  '---',
-  '',
-  '{input}',
-].join('\n');
-
-const DefaultStory = () => {
-  const client = useClient();
-  const [chain] = useState(() => {
-    const space = client.spaces.default;
-    return space.db.add(
-      create(ChainType, {
-        prompts: [makeRef(create(ChainPromptType, { command: 'test', template, inputs: [] }))],
-      }),
-    );
-  });
-
-  return (
-    <div role='none' className='flex w-[350px] border border-separator overflow-hidden'>
-      <PromptEditor prompt={chain.prompts![0]!.target!} />
-    </div>
-  );
-};
-
-export const Default = {};
-
-const meta: Meta = {
-  title: 'plugins/plugin-automation/PromptTemplate',
-  render: DefaultStory,
-  decorators: [
-    withClientProvider({ createIdentity: true, createSpace: true, types: [ChainType, ChainPromptType] }),
-    withLayout({ fullscreen: true, classNames: 'flex justify-center m-2' }),
-    withTheme,
-  ],
-  parameters: {
-    translations,
-  },
-};
-
-export default meta;

package/src/components/PromptEditor/PromptEditor.tsx

@@ -1,222 +0,0 @@
-//
-// Copyright 2023 DXOS.org
-//
-
-import { type Schema as S } from '@effect/schema';
-import React, { useEffect } from 'react';
-
-import { createDocAccessor } from '@dxos/react-client/echo';
-import { Input, Select, useThemeContext, useTranslation } from '@dxos/react-ui';
-import {
-  createBasicExtensions,
-  createDataExtensions,
-  createThemeExtensions,
-  useTextEditor,
-} from '@dxos/react-ui-editor';
-import { attentionSurface, groupBorder, mx } from '@dxos/react-ui-theme';
-import { nonNullable } from '@dxos/util';
-
-import { nameRegex, promptExtension } from './prompt-extension';
-import { AUTOMATION_PLUGIN } from '../../meta';
-import { type ChainInput, ChainInputType, type ChainPromptType } from '../../types';
-
-const inputTypes = [
-  {
-    value: ChainInputType.VALUE,
-    label: 'Value',
-  },
-  {
-    value: ChainInputType.PASS_THROUGH,
-    label: 'Pass through',
-  },
-  {
-    value: ChainInputType.RETRIEVER,
-    label: 'Retriever',
-  },
-  // {
-  //   value: ChainInputType.FUNCTION,
-  //   label: 'Function',
-  // },
-  // {
-  //   value: ChainInputType.QUERY,
-  //   label: 'Query',
-  // },
-  {
-    value: ChainInputType.RESOLVER,
-    label: 'Resolver',
-  },
-  {
-    value: ChainInputType.CONTEXT,
-    label: 'Context',
-  },
-  {
-    value: ChainInputType.SCHEMA,
-    label: 'Schema',
-  },
-];
-
-const getInputType = (type: string) => inputTypes.find(({ value }) => String(value) === type)?.value;
-
-const usePromptInputs = (prompt: ChainPromptType) => {
-  useEffect(() => {
-    const text = prompt.template ?? '';
-    if (!prompt.inputs) {
-      prompt.inputs = []; // TODO(burdon): Required?
-    }
-
-    const regex = new RegExp(nameRegex, 'g');
-    const variables = new Set<string>([...text.matchAll(regex)].map((m) => m[1]));
-
-    // Create map of unclaimed inputs.
-    const unclaimed = new Map<string, ChainInput>(
-      prompt.inputs?.filter(nonNullable).map((input) => [input.name, input]),
-    );
-    const missing: string[] = [];
-    Array.from(variables.values()).forEach((name) => {
-      if (unclaimed.has(name)) {
-        unclaimed.delete(name);
-      } else {
-        missing.push(name);
-      }
-    });
-
-    // Match or create new inputs.
-    const values = unclaimed.values();
-    missing.forEach((name) => {
-      const next = values.next().value;
-      if (next) {
-        next.name = name;
-      } else {
-        prompt.inputs?.push({ name });
-      }
-    });
-
-    // Remove unclaimed (deleted) inputs.
-    // TODO(burdon): If user types incorrect name value, it will be deleted. Garbage collect?
-    for (const input of values) {
-      prompt.inputs.splice(prompt.inputs.indexOf(input), 1);
-    }
-  }, [prompt.template]);
-};
-
-export type PromptEditorProps = {
-  prompt: ChainPromptType;
-  commandEditable?: boolean;
-  schema?: S.Schema<any, any, any>;
-};
-
-export const PromptEditor = ({ prompt, commandEditable = true }: PromptEditorProps) => {
-  const { t } = useTranslation(AUTOMATION_PLUGIN);
-  const { themeMode } = useThemeContext();
-
-  const { parentRef } = useTextEditor(
-    () => ({
-      initialValue: prompt.template,
-      extensions: [
-        createDataExtensions({
-          id: prompt.id,
-          text: prompt.template !== undefined ? createDocAccessor(prompt, ['template']) : undefined,
-        }),
-        createBasicExtensions({
-          bracketMatching: false,
-          lineWrapping: true,
-          placeholder: t('template placeholder'),
-        }),
-        createThemeExtensions({
-          themeMode,
-          slots: {
-            content: { className: '!p-3' },
-          },
-        }),
-        promptExtension,
-      ],
-    }),
-    [themeMode, prompt],
-  );
-
-  usePromptInputs(prompt);
-
-  return (
-    <div className={mx('flex flex-col w-full overflow-hidden gap-4', groupBorder)}>
-      {commandEditable && (
-        <div className='flex items-center pl-4'>
-          <span className='text-neutral-500'>/</span>
-          <Input.Root>
-            <Input.TextInput
-              placeholder={t('command placeholder')}
-              classNames={mx('is-full bg-transparent m-2')}
-              value={prompt.command ?? ''}
-              onChange={(event) => {
-                prompt.command = event.target.value.replace(/\w/g, '');
-              }}
-            />
-          </Input.Root>
-        </div>
-      )}
-
-      <div ref={parentRef} className={mx(attentionSurface, 'rounded', 'min-h-[120px]')} />
-
-      {(prompt.inputs?.length ?? 0) > 0 && (
-        <div className='flex flex-col'>
-          {/* TODO(zan): Improve layout with grid */}
-          <table className='w-full table-fixed border-collapse my-2'>
-            <tbody>
-              {prompt.inputs?.filter(nonNullable).map((input) => (
-                <tr key={input.name}>
-                  <td className='w-[160px] p-1 font-mono text-sm whitespace-nowrap truncate'>
-                    <code className='px-2'>{input.name}</code>
-                  </td>
-                  <td className='w-[120px] p-1'>
-                    <Input.Root>
-                      <Select.Root
-                        value={String(input.type)}
-                        onValueChange={(type) => {
-                          input.type = getInputType(type) ?? ChainInputType.VALUE;
-                        }}
-                      >
-                        <Select.TriggerButton placeholder='Type' classNames='is-full' />
-                        <Select.Portal>
-                          <Select.Content>
-                            <Select.Viewport>
-                              {inputTypes.map(({ value, label }) => (
-                                <Select.Option key={value} value={String(value)}>
-                                  {label}
-                                </Select.Option>
-                              ))}
-                            </Select.Viewport>
-                          </Select.Content>
-                        </Select.Portal>
-                      </Select.Root>
-                    </Input.Root>
-                  </td>
-                  <td className='p-1 pr-2'>
-                    {input.type !== undefined &&
-                      [
-                        ChainInputType.VALUE,
-                        ChainInputType.CONTEXT,
-                        ChainInputType.RESOLVER,
-                        ChainInputType.SCHEMA,
-                      ].includes(input.type) && (
-                        <div>
-                          <Input.Root>
-                            <Input.TextInput
-                              placeholder={t('command placeholder')}
-                              classNames={mx('is-full bg-transparent')}
-                              value={input.value ?? ''}
-                              onChange={(event) => {
-                                input.value = event.target.value;
-                              }}
-                            />
-                          </Input.Root>
-                        </div>
-                      )}
-                  </td>
-                </tr>
-              ))}
-            </tbody>
-          </table>
-        </div>
-      )}
-    </div>
-  );
-};

package/src/components/PromptEditor/index.ts

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

package/src/components/PromptEditor/prompt-extension.ts

@@ -1,43 +0,0 @@
-//
-// Copyright 2023 DXOS.org
-//
-
-import { HighlightStyle, StreamLanguage, syntaxHighlighting } from '@codemirror/language';
-import { type Extension } from '@codemirror/state';
-
-import { tags } from '@dxos/react-ui-editor';
-import { mx } from '@dxos/react-ui-theme';
-
-export const nameRegex = /\{([\w-]+)}/;
-
-const parser = StreamLanguage.define({
-  token: (stream) => {
-    if (stream.eatSpace()) {
-      return null;
-    }
-    if (stream.match(/^#.*/)) {
-      return 'lineComment';
-    }
-    if (stream.match(/^-+$/)) {
-      return 'lineComment';
-    }
-    if (stream.match(nameRegex)) {
-      return 'variableName';
-    }
-    stream.next();
-    return null;
-  },
-});
-
-/**
- * https://codemirror.net/examples/styling
- * https://lezer.codemirror.net/docs/ref/#highlight
- */
-const highlightStyles = HighlightStyle.define([
-  {
-    tag: tags.variableName,
-    class: mx('rounded border border-yellow-500 bg-yellow-100 text-black font-mono text-sm'),
-  },
-]);
-
-export const promptExtension: Extension = [parser, syntaxHighlighting(highlightStyles)];

package/src/components/PromptEditor/types.tsx

@@ -1,28 +0,0 @@
-//
-// Copyright 2023 DXOS.org
-//
-
-// TODO(burdon): Factor out (shared with backend). core/chain (@dxos/chain).
-// TODO(burdon): Agent/plan-and-execute (with tools). Create design doc/ontology of sequence/agent.
-
-export type Step = {
-  id: string;
-};
-
-export type Variable = {
-  name: string;
-  type: 'value' | 'function' | 'query' | 'retriever' | 'pass-through';
-};
-
-export type Prompt = Step & {
-  title?: string;
-  description?: string;
-  template: string;
-  variables: Variable[];
-};
-
-export type Sequence = Step & {
-  runnable: Step[];
-};
-
-export type Chain = {};

package/src/components/ServiceRegistry/ServiceRegistry.stories.tsx

@@ -1,48 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import '@dxos-theme';
-
-import { type Meta, type StoryObj } from '@storybook/react';
-import React from 'react';
-
-import { useSpace } from '@dxos/react-client/echo';
-import { withClientProvider } from '@dxos/react-client/testing';
-import { withLayout, withTheme } from '@dxos/storybook-utils';
-
-import { ServiceRegistry } from './ServiceRegistry';
-
-const meta: Meta<typeof ServiceRegistry> = {
-  title: 'plugins/plugin-automation/ServiceRegistry',
-  component: ServiceRegistry,
-  render: ({ space: _ignore, ...args }) => {
-    const space = useSpace();
-    if (!space) {
-      return <div />;
-    }
-
-    return (
-      <div className='h-full w-[300px] overflow-hidden'>
-        <ServiceRegistry space={space} {...args} />
-      </div>
-    );
-  },
-  decorators: [
-    withClientProvider({
-      createIdentity: true,
-      createSpace: true,
-      // types: [ServiceType], // TODO(burdon): Doesn't fit type constraint???
-    }),
-    withLayout({ fullscreen: true, tooltips: true, classNames: 'flex justify-center' }),
-    withTheme,
-  ],
-};
-
-export default meta;
-
-type Story = StoryObj<typeof ServiceRegistry>;
-
-export const Default: Story = {
-  args: {},
-};