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

package/src/hooks/invocation-handler.ts

@@ -1,109 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { sleep } from '@dxos/async';
-import { getObjectCore, ResultFormat } from '@dxos/echo-db';
-import { type AnyObjectData } from '@dxos/echo-schema';
-import { FunctionType } from '@dxos/functions';
-import { type FunctionTrigger } from '@dxos/functions';
-import { invariant } from '@dxos/invariant';
-import { DXN, LOCAL_SPACE_TAG } from '@dxos/keys';
-import { log } from '@dxos/log';
-import { type Client, type Config } from '@dxos/react-client';
-import { type Space } from '@dxos/react-client/echo';
-
-const MAX_RETRIES = 3;
-const RETRY_DELAY = 1_000;
-
-const callFunction = async (funcUrl: string, trigger: any, data: any) => {
-  const body = { event: 'trigger', trigger, data };
-
-  let retryCount = 0;
-  while (retryCount < MAX_RETRIES) {
-    log.info('exec', { funcUrl, body, retryCount });
-    const response = await fetch(funcUrl, {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-      },
-      body: JSON.stringify(body),
-    });
-
-    const data = await response.text();
-    log.info('response', { status: response.status, body: data });
-    if (response.status === 409) {
-      retryCount++;
-      await sleep(RETRY_DELAY * Math.min(retryCount, 2));
-      continue;
-    }
-
-    return { status: response.status, data };
-  }
-  return { status: 500 };
-};
-
-export const invokeFunction = async (client: Client, space: Space, trigger: FunctionTrigger, data: any) => {
-  try {
-    invariant(trigger.spec);
-    invariant(trigger.function);
-
-    const script = await space.db.query({ id: trigger.function }, { format: ResultFormat.Plain }).first();
-    const { objects: functions } = await space.db
-      .query({ __typename: FunctionType.typename }, { format: ResultFormat.Plain })
-      .run();
-    const func = functions.find((fn) => referenceEquals(fn.source, trigger.function!)) as AnyObjectData | undefined;
-    const funcSlug = func?.__meta.keys.find((key) => key.source === USERFUNCTIONS_META_KEY)?.id;
-    if (!funcSlug) {
-      log.warn('function not deployed', { scriptId: script.id, name: script.name });
-      return 404;
-    }
-
-    const funcUrl = getFunctionUrl(client.config, funcSlug, space.id);
-    const triggerData: AnyObjectData = {
-      ...getObjectCore(trigger).toPlainObject(),
-      // TODO: Remove when functions can query by DXN.
-      promptId: trigger.meta?.prompt?.id,
-    };
-    // TODO: Remove when functions can add objects and easily modify collections (push, splice).
-    return (await callFunction(funcUrl, triggerData, data)).status;
-  } catch (err) {
-    return 400;
-  }
-};
-
-const USERFUNCTIONS_META_KEY = 'dxos.org/service/function';
-
-const getFunctionUrl = (config: Config, slug: string, spaceId?: string) => {
-  const baseUrl = new URL('functions/', config.values.runtime?.services?.edge?.url);
-
-  // Leading slashes cause the URL to be treated as an absolute path.
-  const relativeUrl = slug.replace(/^\//, '');
-  const url = new URL(`./${relativeUrl}`, baseUrl.toString());
-  spaceId && url.searchParams.set('spaceId', spaceId);
-  url.protocol = 'https';
-  return url.toString();
-};
-
-// TODO(dmaretskyi): Factor out.
-
-type ReferenceLike = { '/': string } | string;
-
-const referenceEquals = (a: ReferenceLike, b: ReferenceLike): boolean => {
-  const aDXN = toDXN(a);
-  const bDXN = toDXN(b);
-  return aDXN.toString() === bDXN.toString();
-};
-
-const toDXN = (ref: ReferenceLike): DXN => {
-  if (typeof ref === 'string') {
-    if (ref.startsWith('dxn:')) {
-      return DXN.parse(ref);
-    } else {
-      return new DXN(DXN.kind.ECHO, [LOCAL_SPACE_TAG, ref]);
-    }
-  }
-
-  invariant(typeof ref['/'] === 'string');
-  return DXN.parse(ref['/']);
-};

package/src/hooks/processor.test.ts

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

package/src/hooks/processor.ts

@@ -1,201 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { type Signal, batch, computed, signal } from '@preact/signals-core';
-
-import { type PromiseIntentDispatcher } from '@dxos/app-framework';
-import { type Tool, Message, type MessageContentBlock } from '@dxos/artifact';
-import {
-  isToolUse,
-  runTools,
-  type AIServiceClientImpl,
-  type GenerateRequest,
-  type GenerationStream,
-  MixedStreamParser,
-} from '@dxos/assistant';
-import { createStatic } from '@dxos/echo-schema';
-import { invariant } from '@dxos/invariant';
-import { log } from '@dxos/log';
-import { type Space } from '@dxos/react-client/echo';
-
-const defaultOptions: Pick<GenerateRequest, 'model' | 'systemPrompt'> = {
-  model: '@anthropic/claude-3-5-sonnet-20241022',
-  systemPrompt: '',
-};
-
-// TODO(burdon): Factor out.
-declare global {
-  interface ToolContextExtensions {
-    space?: Space;
-    dispatch?: PromiseIntentDispatcher;
-  }
-}
-
-type RequestOptions = {
-  history?: Message[];
-  onComplete?: (messages: Message[]) => void;
-};
-
-/**
- * Handles interactions with the AI service.
- * Maintains a queue of messages and handles streaming responses from the AI service.
- * Executes tools based on AI responses.
- * Supports cancellation of in-progress requests.
- */
-export class ChatProcessor {
-  /** SSE stream parser. */
-  private readonly _parser = new MixedStreamParser();
-
-  /** Current streaming response (iterator). */
-  private _stream: GenerationStream | undefined;
-
-  /** Pending messages (incl. the user request). */
-  private readonly _pending: Signal<Message[]> = signal([]);
-
-  /** Current streaming block (from the AI service). */
-  private readonly _block: Signal<MessageContentBlock | undefined> = signal(undefined);
-
-  /** Prior history from queue. */
-  private _history: Message[] = [];
-
-  /**
-   * Streaming state.
-   * @reactive
-   */
-  public readonly streaming: Signal<boolean> = computed(() => this._block.value !== undefined);
-
-  /**
-   * Array of Messages (incl. the current message being streamed).
-   * @reactive
-   */
-  public readonly messages: Signal<Message[]> = computed(() => {
-    const messages = [...this._pending.value];
-    if (this._block.value) {
-      const current = messages.pop();
-      invariant(current);
-      const { content, ...rest } = current;
-      const message = { ...rest, content: [...content, this._block.value] };
-      messages.push(message);
-    }
-
-    return messages;
-  });
-
-  constructor(
-    private readonly _client: AIServiceClientImpl,
-    private readonly _tools?: Tool[],
-    private readonly _extensions?: ToolContextExtensions,
-    private readonly _options: Pick<GenerateRequest, 'model' | 'systemPrompt'> = defaultOptions,
-  ) {
-    // Message complete.
-    this._parser.message.on((message) => {
-      batch(() => {
-        this._pending.value = [...this._pending.value, message];
-        this._block.value = undefined;
-      });
-    });
-
-    // Streaming update (happens before message complete).
-    this._parser.update.on((block) => {
-      batch(() => {
-        this._block.value = block;
-      });
-    });
-  }
-
-  /**
-   * Make GPT request.
-   */
-  async request(message: string, options: RequestOptions = {}): Promise<Message[]> {
-    batch(() => {
-      this._history = options.history ?? [];
-      this._pending.value = [
-        createStatic(Message, {
-          role: 'user',
-          content: [{ type: 'text', text: message }],
-        }),
-      ];
-      this._block.value = undefined;
-    });
-
-    await this._generate();
-    options.onComplete?.(this._pending.value);
-    return this._reset();
-  }
-
-  /**
-   * Cancel pending requests.
-   * @returns Pending requests (incl. the request message).
-   */
-  async cancel(): Promise<Message[]> {
-    log.info('cancelling...');
-    this._stream?.abort();
-    return this._reset();
-  }
-
-  private async _reset(): Promise<Message[]> {
-    const messages = this._pending.value;
-    batch(() => {
-      this._history = [];
-      this._pending.value = [];
-      this._block.value = undefined;
-    });
-
-    return messages;
-  }
-
-  /**
-   * Generate a response from the AI service.
-   * Iterates over tool requests.
-   */
-  private async _generate() {
-    try {
-      let more = false;
-      do {
-        log.info('requesting...', { history: this._history.length, messages: this._pending.value.length });
-        this._stream = await this._client.generate({
-          ...this._options,
-          // TODO(burdon): Rename messages or separate history/message.
-          history: [...this._history, ...this._pending.value],
-          tools: this._tools,
-        });
-
-        // Wait until complete.
-        await this._parser.parse(this._stream);
-        await this._stream.complete();
-
-        // Add messages.
-        log.info('response', { messages: this._pending.value });
-
-        // Resolve tool use locally.
-        more = false;
-        const message = this._pending.value.at(-1);
-        invariant(message);
-        if (isToolUse(message)) {
-          log.info('tool request...');
-          const response = await runTools({
-            message: this._pending.value.at(-1)!,
-            tools: this._tools ?? [],
-            extensions: this._extensions,
-          });
-
-          log.info('tool response', { response });
-          switch (response.type) {
-            case 'continue': {
-              this._pending.value = [...this._pending.value, response.message];
-              more = true;
-              break;
-            }
-          }
-        }
-      } while (more);
-    } catch (err) {
-      // TODO(burdon): Handle error.
-      log.catch('request failed', { err });
-    } finally {
-      log.info('done');
-      this._stream = undefined;
-    }
-  }
-}

package/src/hooks/useLocalTriggerManager.ts

@@ -1,82 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { useEffect, useState } from 'react';
-
-import { Mutex } from '@dxos/async';
-import { Context } from '@dxos/context';
-import { createSubscriptionTrigger, type TriggerFactory } from '@dxos/functions';
-import { FunctionTrigger } from '@dxos/functions';
-import { invariant } from '@dxos/invariant';
-import { log } from '@dxos/log';
-import { useClient } from '@dxos/react-client';
-import { Filter, type Space, useQuery } from '@dxos/react-client/echo';
-
-import { invokeFunction } from './invocation-handler';
-
-const registerTriggersMutex = new Mutex();
-
-export const useLocalTriggerManager = (space: Space) => {
-  const client = useClient();
-
-  // TODO(burdon): Factor out, creating context for plugin (runs outside of component).
-  const [registry] = useState(new Map<string, Context>());
-  const triggers = useQuery(space, Filter.schema(FunctionTrigger));
-  useEffect(() => {
-    setTimeout(async () => {
-      // Mark-and-sweep removing disabled triggers.
-      await registerTriggersMutex.executeSynchronized(async () => {
-        const deprecated = new Set(Array.from(registry.keys()));
-        log('triggers', {
-          deprecated,
-          all: triggers.map((t) => t.id),
-          enabled: triggers.filter((t) => t.enabled).map((t) => t.id),
-        });
-
-        for (const trigger of triggers) {
-          if (trigger.enabled) {
-            if (registry.has(trigger.id)) {
-              deprecated.delete(trigger.id);
-              continue;
-            }
-            log.info('activating trigger', trigger.id);
-
-            const ctx = new Context();
-            registry.set(trigger.id, ctx);
-            const triggerSpec = trigger.spec;
-            invariant(triggerSpec);
-
-            let triggerFactory: TriggerFactory<any>;
-            if (triggerSpec.type === 'subscription') {
-              triggerFactory = createSubscriptionTrigger;
-            } else {
-              log.info('unsupported trigger', { type: triggerSpec.type });
-              continue;
-            }
-
-            await triggerFactory(ctx, space, trigger.spec, (data: any) => {
-              return invokeFunction(client, space, trigger, data);
-            });
-          }
-        }
-
-        for (const id of deprecated) {
-          const ctx = registry.get(id);
-          if (ctx) {
-            await ctx.dispose();
-            registry.delete(id);
-          }
-        }
-      });
-    });
-  }, [JSON.stringify(triggers)]);
-
-  useEffect(() => {
-    return () => {
-      for (const ctx of registry.values()) {
-        void ctx.dispose();
-      }
-    };
-  }, []);
-};

package/src/hooks/useServiceRegistry.ts

@@ -1,22 +0,0 @@
-//
-// Copyright 2025 DXOS.org
-//
-
-import { useEffect, useMemo, useState } from 'react';
-
-import { type ServiceType, MockServiceRegistry, type ServiceQuery } from '../types';
-
-export const useServiceRegistry = (query: ServiceQuery): ServiceType[] => {
-  const registry = useMemo(() => new MockServiceRegistry(), []);
-  const [services, setServices] = useState<ServiceType[]>([]);
-  useEffect(() => {
-    const t = setTimeout(async () => {
-      const services = await registry.queryServices(query);
-      setServices(services);
-    });
-
-    return () => clearTimeout(t);
-  }, [query, registry]);
-
-  return services;
-};

package/src/presets.ts

@@ -1,248 +0,0 @@
-//
-// Copyright 2024 DXOS.org
-//
-
-import { create } from '@dxos/live-object';
-
-import { ChainInputType, ChainPromptType } from './types';
-
-export const str = (...text: (string | undefined | boolean)[]): string =>
-  text
-    .filter((value) => value !== undefined && value !== false)
-    .flat()
-    .join('\n');
-
-export type Preset = {
-  id: string;
-  title: string;
-  createPrompt: () => ChainPromptType;
-};
-
-export const chainPresets: Preset[] = [
-  {
-    id: 'dxos.org/prompt/translate',
-    title: 'Translate',
-    createPrompt: () =>
-      create(ChainPromptType, {
-        command: 'say',
-        template: str(
-          // prettier-ignore
-          'Translate the following into {language}:',
-          '',
-          '---',
-          '',
-          '{input}',
-        ),
-        inputs: [
-          {
-            type: ChainInputType.VALUE,
-            name: 'language',
-            value: 'japanese',
-          },
-          {
-            type: ChainInputType.PASS_THROUGH,
-            name: 'input',
-          },
-          // TODO(burdon): Message.
-          // {
-          //   type: ChainInputType.CONTEXT,
-          //   name: 'input',
-          //   value: 'object.text',
-          // },
-        ],
-      }),
-  },
-  {
-    id: 'dxos.org/prompt/chess',
-    title: 'Chess',
-    createPrompt: () =>
-      create(ChainPromptType, {
-        command: 'hint',
-        template: str(
-          // prettier-ignore
-          'You are a machine that is an expert chess player.',
-          '',
-          'The move history of the current game is: {history}',
-          '',
-          'Suggest the next move and very briefly explain your strategy in a couple of sentences.',
-        ),
-        inputs: [
-          {
-            type: ChainInputType.CONTEXT,
-            name: 'history',
-            value: 'object.pgn',
-          },
-        ],
-      }),
-  },
-  {
-    id: 'dxos.org/prompt/mermaid',
-    title: 'Mermaid',
-    createPrompt: () =>
-      create(ChainPromptType, {
-        command: 'draw',
-        template: str(
-          // prettier-ignore
-          'Create a simplified mermaid graph representing the text below.',
-          'Do not explain anything.',
-          '',
-          '---',
-          '',
-          '{input}',
-        ),
-        inputs: [
-          {
-            type: ChainInputType.PASS_THROUGH,
-            name: 'input',
-          },
-        ],
-      }),
-  },
-  {
-    id: 'dxos.org/prompt/list',
-    title: 'List',
-    createPrompt: () =>
-      create(ChainPromptType, {
-        command: 'list',
-        template: str(
-          // prettier-ignore
-          'You are a machine that only replies with valid, iterable RFC8259 compliant JSON in your responses.',
-          'Your entire response should be a single array of JSON objects.',
-          '',
-          'Your entire response should be a map where the key is the type and the value is a single array of JSON objects conforming to the following types:',
-          '',
-          '{schema}',
-          '',
-          '---',
-          '',
-          '{question}',
-        ),
-        inputs: [
-          {
-            type: ChainInputType.SCHEMA,
-            name: 'schema',
-            value: 'example.com/type/project',
-          },
-          {
-            type: ChainInputType.PASS_THROUGH,
-            name: 'question',
-          },
-        ],
-      }),
-  },
-  {
-    id: 'dxos.org/prompt/base',
-    title: 'RAG',
-    createPrompt: () =>
-      create(ChainPromptType, {
-        command: 'rag',
-        template: str(
-          // prettier-ignore
-          "Very briefly answer the question based only on the following context and say if you don't know the answer.",
-          // 'answer the question using the following context as well as your training data:',
-          '',
-          '{context}',
-          '',
-          '---',
-          '',
-          'question: {question}',
-        ),
-        inputs: [
-          {
-            type: ChainInputType.RETRIEVER,
-            name: 'context',
-          },
-          {
-            type: ChainInputType.CONTEXT,
-            name: 'question',
-            value: 'object.text',
-          },
-        ],
-      }),
-  },
-  {
-    id: 'dxos.org/prompt/lookup',
-    title: 'Lookup',
-    createPrompt: () =>
-      create(ChainPromptType, {
-        command: 'lookup',
-        template: str(
-          // prettier-ignore
-          'Lookup and very briefly summarize the following topic in one or two sentences:',
-          '',
-          '---',
-          '',
-          '{input}',
-        ),
-        inputs: [
-          {
-            type: ChainInputType.CONTEXT,
-            name: 'input',
-            value: 'text',
-          },
-        ],
-      }),
-  },
-  {
-    id: 'dxos.org/prompt/extract',
-    title: 'Extract',
-    createPrompt: () =>
-      create(ChainPromptType, {
-        command: 'extract',
-        template: str(
-          // prettier-ignore
-          'List all people and companies mentioned in the text below.',
-          '',
-          'You are a machine that only replies with valid, iterable RFC8259 compliant JSON in your responses.',
-          'Your entire response should be a map where the key is the type and the value is a single array of JSON objects conforming to the following types:',
-          '',
-          '{contact}',
-          '{company}',
-          '',
-          '---',
-          '',
-          '{input}',
-        ),
-        inputs: [
-          {
-            type: ChainInputType.SCHEMA,
-            name: 'contact',
-            value: 'example.com/type/contact',
-          },
-          {
-            type: ChainInputType.SCHEMA,
-            name: 'company',
-            value: 'example.com/type/organization',
-          },
-          {
-            type: ChainInputType.CONTEXT,
-            name: 'input',
-            value: 'text',
-          },
-        ],
-      }),
-  },
-  {
-    id: 'dxos.org/prompt/discord',
-    title: 'Summarize',
-    createPrompt: () =>
-      create(ChainPromptType, {
-        command: 'summarize',
-        template: str(
-          // prettier-ignore
-          'Summarize what the team is working on and format it as a markdown table without any explanation.',
-          '',
-          '---',
-          '',
-          '{context}',
-        ),
-        inputs: [
-          {
-            type: ChainInputType.RESOLVER,
-            name: 'context',
-            value: 'discord.messages.recent',
-          },
-        ],
-      }),
-  },
-];