@dxos/plugin-automation 0.7.5-labs.a279d8c → 0.7.5-labs.ea4b4c2

package/src/components/AssistantPanel/system-instructions.ts

@@ -1,166 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { asyncTimeout } from '@dxos/async';
-import type { ReactiveEchoObject } from '@dxos/echo-db';
-import { getTypename } from '@dxos/echo-schema';
-import { log } from '@dxos/log';
-import { Filter, getSpace, ResultFormat } from '@dxos/react-client/echo';
-
-// TODO(burdon): Move into assistant-protocol.
-export type ThreadContext = {
-  subject?: ReactiveEchoObject<any>;
-};
-
-export const createSystemInstructions = async (context: ThreadContext): Promise<string> => {
-  let instructions = `
-    <instructions>
-      Before replying always think step-by-step on how to proceed.
-      Print your thoughts inside <cot> tags.
-
-      <example>
-        <cot>To answer the question I need to ...</cot>
-      </example>
-    </instructions>
-
-    <current_time>${new Date().toLocaleString()}</current_time>
-  `;
-
-  if (context.subject) {
-    instructions += `
-      <user_attention>
-        The user is currently interacting with an object in Composer application:
-
-        ${await formatContextObject(context.subject)}
-      </user_attention>
-    `;
-  }
-
-  return looseFormatXml(instructions);
-};
-
-const formatContextObject = async (object: ReactiveEchoObject<any>): Promise<string> => {
-  let data;
-  try {
-    data = await asyncTimeout(preprocessContextObject(object), CONTEXT_OBJECT_QUERY_TIMEOUT);
-  } catch (err: any) {
-    log.error('Failed to preprocess context object:', { err });
-    data = object;
-  }
-
-  if (typeof data === 'string') {
-    return data;
-  } else {
-    return `
-      <object>
-        <type>${getTypename(object)}</type>
-        <id>${object.id}</id>
-        ${formatObjectAsXMLTags(data)}
-      </object>
-    `;
-  }
-};
-
-const preprocessContextObject = async (object: ReactiveEchoObject<any>): Promise<Record<string, any> | string> => {
-  const space = getSpace(object);
-  if (!space) {
-    return { ...object };
-  }
-
-  // TODO(dmaretskyi): Serialize based on schema annotations.
-  switch (getTypename(object)) {
-    // TODO(dmaretskyi): Reference types somehow without plugin-automation depending on other plugins.
-    case 'dxos.org/type/Document': {
-      const data = space.db
-        .query({ id: object.id }, { format: ResultFormat.Plain, include: { content: true } })
-        .first() ?? { content: { content: '' } };
-
-      return {
-        ...data,
-        threads: undefined,
-      };
-    }
-
-    case 'dxos.org/type/Table': {
-      // TODO(dmaretskyi): Load references.
-      const schema = object.view ? space?.db.schemaRegistry.getSchema(object.view.query.type) : undefined;
-      const { objects: rows } =
-        (schema &&
-          (await space.db
-            .query(Filter.schema(schema), { format: ResultFormat.Plain, limit: TABLE_ROWS_LIMIT })
-            .run())) ??
-        {};
-
-      // TODO(dmaretskyi): Format table schema.
-      return `
-        <object>
-          <id>${object.id}</id>
-          <type>${getTypename(object)}</type>
-          ${formatObjectAsXMLTags(object)}
-
-          <rows>
-            <!-- Limited to first ${TABLE_ROWS_LIMIT} rows. -->
-            ${rows
-              ?.map(
-                (row: any) => `<row>
-                  ${formatObjectAsXMLTags(row)}
-                </row>`,
-              )
-              .join('\n')}
-          </rows>
-
-      `;
-    }
-
-    default:
-      return { ...object };
-  }
-};
-
-const formatObjectAsXMLTags = (object: any, depth = 1): string => {
-  return Object.entries(object)
-    .filter(([key, value]) => ['string', 'number', 'boolean', 'object'].includes(typeof value))
-    .map(([key, value]) => {
-      if (typeof value === 'object' && value !== null) {
-        if (depth === 0) {
-          return '';
-        } else {
-          return `<${key}>
-            ${formatObjectAsXMLTags(value, depth - 1)}
-          </${key}>`;
-        }
-      }
-
-      return `<${key}>${value}</${key}>`;
-    })
-    .join('\n');
-};
-
-const CONTEXT_OBJECT_QUERY_TIMEOUT = 5_000;
-
-const TABLE_ROWS_LIMIT = 10;
-
-/**
- * Formats XML indentation for instructions so they are easier to read during debugging.
- */
-const looseFormatXml = (xml: string): string => {
-  let currentIndent = 0;
-
-  return xml
-    .split('\n')
-    .map((line) => {
-      if (line.match(RE_CLOSE_TAG_LINE)) {
-        currentIndent--;
-      }
-      const indent = currentIndent;
-      if (line.match(RE_OPEN_TAG_LINE)) {
-        currentIndent++;
-      }
-      return ' '.repeat(indent * 2) + line.trimStart();
-    })
-    .join('\n');
-};
-
-const RE_OPEN_TAG_LINE = /^[ ]*<[a-zA-Z0-9\-_]+>[ ]*$/;
-const RE_CLOSE_TAG_LINE = /^[ ]*<\/[a-zA-Z0-9\-_]+>[ ]*$/;

package/src/components/ChatContainer.tsx

@@ -1,80 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import React, { useCallback, useMemo } from 'react';
-
-import { Capabilities, useCapabilities, useCapability, useIntentDispatcher } from '@dxos/app-framework';
-import { createSystemPrompt, type Message } from '@dxos/artifact';
-import { DXN, QueueSubspaceTags } from '@dxos/keys';
-import { getSpace } from '@dxos/react-client/echo';
-import { useEdgeClient, useQueue } from '@dxos/react-edge-client';
-import { StackItem } from '@dxos/react-ui-stack';
-
-import { Thread } from './Thread';
-import { AutomationCapabilities } from '../capabilities';
-import { ChatProcessor } from '../hooks';
-import { type GptChatType } from '../types';
-
-export const ChatContainer = ({ chat, role }: { chat: GptChatType; role: string }) => {
-  const edgeClient = useEdgeClient();
-  const { dispatchPromise: dispatch } = useIntentDispatcher();
-  const space = getSpace(chat);
-  const aiClient = useCapability(AutomationCapabilities.AiClient);
-  const artifactDefinitions = useCapabilities(Capabilities.ArtifactDefinition);
-  const globalTools = useCapabilities(Capabilities.Tools);
-  const tools = useMemo(
-    () => [...globalTools.flat(), ...artifactDefinitions.flatMap((definition) => definition.tools)],
-    [globalTools, artifactDefinitions],
-  );
-  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 processor = useMemo(
-    () =>
-      new ChatProcessor(
-        aiClient,
-        tools,
-        { space, dispatch },
-        { model: '@anthropic/claude-3-5-sonnet-20241022', systemPrompt },
-      ),
-    [aiClient, tools, space, 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 queue = useQueue<Message>(edgeClient, queueDxn);
-  const messages = useMemo(
-    () => [...queue.items, ...processor.messages.value],
-    [queue.items, processor.messages.value],
-  );
-
-  const handleSubmit = useCallback(
-    async (message: string) => {
-      // TODO(burdon): Button to cancel. Otherwise queue request.
-      if (processor.isStreaming) {
-        await processor.cancel();
-      }
-
-      const messages = await processor.request(message, queue.items);
-      // TODO(burdon): Append on success only? If approved by user? Clinet/server.
-      await queue.append(messages);
-    },
-    [processor, queue],
-  );
-
-  return (
-    <StackItem.Content toolbar={false} role={role}>
-      <Thread messages={messages} streaming={processor.isStreaming.value} onSubmit={handleSubmit} />
-    </StackItem.Content>
-  );
-};
-
-export default ChatContainer;