@dxos/functions-testing 0.8.4-main.8360d9e660 → 0.8.4-main.8baae0fced

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dxos/functions-testing",
-  "version": "0.8.4-main.8360d9e660",
+  "version": "0.8.4-main.8baae0fced",
   "description": "Functions tests",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
@@ -8,7 +8,7 @@
     "type": "git",
     "url": "https://github.com/dxos/dxos"
   },
-  "license": "MIT",
+  "license": "FSL-1.1-Apache-2.0",
   "author": "info@dxos.org",
   "sideEffects": true,
   "type": "module",
@@ -25,29 +25,28 @@
     }
   },
   "types": "dist/types/src/index.d.ts",
-  "typesVersions": {
-    "*": {
-      "testing": [
-        "dist/types/src/testing/index.d.ts"
-      ]
-    }
-  },
   "files": [
     "dist",
     "src"
   ],
   "dependencies": {
     "@effect/platform": "0.94.4",
-    "@dxos/async": "0.8.4-main.8360d9e660",
-    "@dxos/client": "0.8.4-main.8360d9e660",
-    "@dxos/config": "0.8.4-main.8360d9e660",
-    "@dxos/echo": "0.8.4-main.8360d9e660",
-    "@dxos/functions-runtime-cloudflare": "0.8.4-main.8360d9e660",
-    "@dxos/functions": "0.8.4-main.8360d9e660",
-    "@dxos/log": "0.8.4-main.8360d9e660",
-    "@dxos/protocols": "0.8.4-main.8360d9e660",
-    "@dxos/invariant": "0.8.4-main.8360d9e660",
-    "@dxos/functions-runtime": "0.8.4-main.8360d9e660"
+    "@dxos/assistant-toolkit": "0.8.4-main.8baae0fced",
+    "@dxos/compute": "0.8.4-main.8baae0fced",
+    "@dxos/client": "0.8.4-main.8baae0fced",
+    "@dxos/config": "0.8.4-main.8baae0fced",
+    "@dxos/context": "0.8.4-main.8baae0fced",
+    "@dxos/echo": "0.8.4-main.8baae0fced",
+    "@dxos/functions": "0.8.4-main.8baae0fced",
+    "@dxos/async": "0.8.4-main.8baae0fced",
+    "@dxos/functions-runtime": "0.8.4-main.8baae0fced",
+    "@dxos/invariant": "0.8.4-main.8baae0fced",
+    "@dxos/log": "0.8.4-main.8baae0fced",
+    "@dxos/operation": "0.8.4-main.8baae0fced",
+    "@dxos/protocols": "0.8.4-main.8baae0fced",
+    "@dxos/functions-runtime-cloudflare": "0.8.4-main.8baae0fced",
+    "@dxos/schema": "0.8.4-main.8baae0fced",
+    "@dxos/util": "0.8.4-main.8baae0fced"
   },
   "publishConfig": {
     "access": "public"

package/src/bundle.test.ts

@@ -3,7 +3,6 @@
 //
 
 import { join } from 'node:path';
-
 import { describe, it } from 'vitest';
 
 import { bundleFunction } from '@dxos/functions-runtime/native';

package/src/cpu-limit.test.ts

@@ -4,17 +4,18 @@
 
 import { describe, test } from 'vitest';
 
+import { Trigger } from '@dxos/compute';
 import { configPreset } from '@dxos/config';
+import { Context } from '@dxos/context';
 import { Obj, Ref } from '@dxos/echo';
-import { Trigger } from '@dxos/functions';
 import { FunctionRuntimeKind } from '@dxos/protocols';
 
 import { deployFunction, observeInvocations, setup, sync } from './testing';
 
 const FIB_FUNCTION_PATH = new URL('./functions/fib.ts', import.meta.url).pathname;
 
-// Test suite to explore who CPU time limits influence function execution
-describe.runIf(process.env.DX_TEST_TAGS?.includes('functions-e2e'))('CPU limit', () => {
+// Test suite to explore who CPU time limits influence function execution.
+describe('CPU limit', { tags: ['functions-e2e'] }, () => {
   const config = configPreset({ edge: 'dev' });
 
   test('invoke directly', { timeout: 120_000 }, async ({ expect }) => {
@@ -26,6 +27,7 @@ describe.runIf(process.env.DX_TEST_TAGS?.includes('functions-e2e'))('CPU limit',
       FunctionRuntimeKind.enums.WORKERS_FOR_PLATFORMS,
     );
     const result = await functionsServiceClient.invoke(
+      Context.default(),
       func,
       {
         iterations: 100,
@@ -49,12 +51,12 @@ describe.runIf(process.env.DX_TEST_TAGS?.includes('functions-e2e'))('CPU limit',
       Obj.make(Trigger.Trigger, {
         enabled: true,
         function: Ref.make(func),
-        spec: { kind: 'timer', cron: '* */30 * * * *' },
+        spec: Trigger.specTimer('* */30 * * * *'),
         input: { iterations: 100 },
       }),
     );
     await sync(space);
-    const result = await functionsServiceClient.forceRunCronTrigger(space.id, trigger.id);
+    const result = await functionsServiceClient.forceRunCronTrigger(Context.default(), space.id, trigger.id);
     console.log(result);
   });
 
@@ -70,27 +72,27 @@ describe.runIf(process.env.DX_TEST_TAGS?.includes('functions-e2e'))('CPU limit',
       Obj.make(Trigger.Trigger, {
         enabled: true,
         function: Ref.make(func),
-        spec: { kind: 'timer', cron: '* */30 * * * *' },
+        spec: Trigger.specTimer('* */30 * * * *'),
         input: { iterations: 1_000_000_000 },
       }),
     );
     await sync(space);
     {
-      const result = await functionsServiceClient.forceRunCronTrigger(space.id, trigger.id);
+      const result = await functionsServiceClient.forceRunCronTrigger(Context.default(), space.id, trigger.id);
       console.log(result);
     }
 
     {
-      const result = await functionsServiceClient.forceRunCronTrigger(space.id, trigger.id);
+      const result = await functionsServiceClient.forceRunCronTrigger(Context.default(), space.id, trigger.id);
       console.log(result);
     }
 
     {
-      Obj.change(trigger, (t) => {
-        t.input!.iterations = 100;
+      Obj.update(trigger, (trigger) => {
+        trigger.input!.iterations = 100;
       });
       await sync(space);
-      const result = await functionsServiceClient.forceRunCronTrigger(space.id, trigger.id);
+      const result = await functionsServiceClient.forceRunCronTrigger(Context.default(), space.id, trigger.id);
       console.log(result);
     }
   });
@@ -107,7 +109,7 @@ describe.runIf(process.env.DX_TEST_TAGS?.includes('functions-e2e'))('CPU limit',
       Obj.make(Trigger.Trigger, {
         enabled: true,
         function: Ref.make(func),
-        spec: { kind: 'timer', cron: '* * * * * *' },
+        spec: Trigger.specTimer('* * * * * *'),
         input: { iterations: 1_000_000 },
       }),
     );
@@ -115,7 +117,7 @@ describe.runIf(process.env.DX_TEST_TAGS?.includes('functions-e2e'))('CPU limit',
     await observeInvocations(space, 100);
   });
 
-  test.only('break CPU limit through natural exection', { timeout: 520_000 }, async ({ expect }) => {
+  test('break CPU limit through natural exection', { timeout: 520_000 }, async ({ expect }) => {
     const { client, space, functionsServiceClient } = await setup(config);
     const func = await deployFunction(
       space,
@@ -127,21 +129,21 @@ describe.runIf(process.env.DX_TEST_TAGS?.includes('functions-e2e'))('CPU limit',
       Obj.make(Trigger.Trigger, {
         enabled: true,
         function: Ref.make(func),
-        spec: { kind: 'timer', cron: '* * * * * *' },
+        spec: Trigger.specTimer('* * * * * *'),
         input: { iterations: 100 },
       }),
     );
     await sync(space);
     await observeInvocations(space, 5);
 
-    Obj.change(trigger, (t) => {
-      t.input!.iterations = 1_000_000_000;
+    Obj.update(trigger, (trigger) => {
+      trigger.input!.iterations = 1_000_000_000;
     });
     await sync(space);
     await observeInvocations(space, 10);
 
-    Obj.change(trigger, (t) => {
-      t.input!.iterations = 100;
+    Obj.update(trigger, (trigger) => {
+      trigger.input!.iterations = 100;
     });
     await sync(space);
     await observeInvocations(space, 1_000);

package/src/edge-routine.test.ts

@@ -0,0 +1,110 @@
+//
+// Copyright 2026 DXOS.org
+//
+
+import * as Schema from 'effect/Schema';
+import { describe, test } from 'vitest';
+
+import { AgentPrompt, DatabaseBlueprint, Chat } from '@dxos/assistant-toolkit';
+import { Client } from '@dxos/client';
+import { Blueprint, Operation, Routine, Trigger } from '@dxos/compute';
+import { configPreset } from '@dxos/config';
+import { Context } from '@dxos/context';
+import { Feed, Obj, Ref } from '@dxos/echo';
+import { TestSchema } from '@dxos/echo/testing';
+import { dbg, log } from '@dxos/log';
+import { ErrorCodec } from '@dxos/protocols';
+import { EdgeReplicationSetting } from '@dxos/protocols/proto/dxos/echo/metadata';
+import { Text } from '@dxos/schema';
+import { trim } from '@dxos/util';
+
+import { sync } from './testing';
+
+/**
+ * Cron trigger on EDGE runs {@link AgentPrompt} for a {@link Routine} that uses the Database
+ * blueprint Query tool against replicated ECHO documents.
+ *
+ * Prereq: local EDGE (`configPreset({ edge: 'local' })` → `http://localhost:8787`) with LLM
+ * available (API key and/or memo replay under `conversationsCache/`).
+ */
+describe('Edge routine', { tags: ['functions-e2e'] }, () => {
+  const config = configPreset({ edge: 'local' });
+
+  test('timer trigger runs routine with database blueprint on edge', { timeout: 420_000 }, async ({ expect }) => {
+    await using client = await new Client({
+      config,
+      types: [
+        Operation.PersistentOperation,
+        Trigger.Trigger,
+        Routine.Routine,
+        Blueprint.Blueprint,
+        Feed.Feed,
+        Text.Text,
+        Chat.Chat,
+        TestSchema.Organization,
+      ],
+    }).initialize();
+
+    await client.halo.createIdentity();
+    const space = await client.spaces.create();
+    await space.waitUntilReady();
+
+    await space.internal.setEdgeReplicationPreference(EdgeReplicationSetting.ENABLED);
+
+    Obj.update(space.properties, (properties) => {
+      properties.computeEnvironment = 'edge';
+    });
+
+    space.db.add(Obj.make(TestSchema.Organization, { name: 'Acme Corp' }));
+    space.db.add(Obj.make(TestSchema.Organization, { name: 'Globex Industries' }));
+    space.db.add(Obj.make(TestSchema.Organization, { name: 'Initech' }));
+
+    const databaseBlueprint = space.db.add(DatabaseBlueprint.make());
+
+    const routine = space.db.add(
+      Routine.make({
+        name: 'edge-e2e-count-orgs-db-blueprint',
+        instructions: trim`
+              You have access to the Database blueprint tools.
+              Use the Query tool exactly once with typename "${TestSchema.Organization.typename}" and no other arguments.
+              Then call completeJob with the output object { "count": <number of rows returned by Query> }.
+              If you are unable to query -- fail.
+              Do not list schemas first.
+            `,
+        input: Schema.Struct({}),
+        output: Schema.Struct({
+          count: Schema.Number,
+        }),
+        blueprints: [Ref.make(databaseBlueprint)],
+        context: [],
+      }),
+    );
+    const fn = Operation.serialize(AgentPrompt);
+    dbg(Obj.toJSON(fn));
+
+    const trigger = space.db.add(
+      Obj.make(Trigger.Trigger, {
+        enabled: true,
+        function: Ref.make(fn),
+        spec: Trigger.specTimer('* * * * * *'),
+        input: {
+          prompt: Ref.make(routine),
+          input: {},
+          model: '@anthropic/claude-haiku-4-5',
+        },
+      }),
+    );
+
+    await sync(space);
+
+    log('trigger created and synced');
+    log.break();
+
+    const runResult: any = await client.edge.http.forceRunCronTrigger(Context.default(), space.id, trigger.id);
+    if (runResult._kind === 'error') {
+      throw ErrorCodec.decode(runResult.error);
+    }
+    log('trigger ran', { runResult });
+    expect(runResult.result.count).toBe(3);
+  });
+});

package/src/functions/definitions.ts

@@ -0,0 +1,61 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Schema from 'effect/Schema';
+
+import { Operation } from '@dxos/compute';
+import { Database } from '@dxos/echo';
+
+export const Fibonacci = Operation.make({
+  meta: {
+    key: 'org.example.function.fib',
+    name: 'Fibonacci',
+    description: 'Function that calculates a Fibonacci number',
+  },
+  input: Schema.Struct({
+    iterations: Schema.optional(Schema.Number).annotations({
+      description: 'Number of iterations',
+      default: 100_000,
+    }),
+  }),
+  output: Schema.Struct({
+    result: Schema.String,
+  }),
+});
+
+export const Reply = Operation.make({
+  meta: {
+    key: 'org.example.function.reply',
+    name: 'Reply',
+    description: 'Function that echoes the input',
+  },
+  input: Schema.Any,
+  output: Schema.Any,
+});
+
+export const Sleep = Operation.make({
+  meta: {
+    key: 'org.example.function.sleep',
+    name: 'Sleep',
+    description: 'Function that sleeps for a given amount of time',
+  },
+  input: Schema.Struct({
+    duration: Schema.optional(Schema.Number).annotations({
+      description: 'Milliseconds to sleep',
+      default: 100_000,
+    }),
+  }),
+  output: Schema.Void,
+});
+
+export const QueryDb = Operation.make({
+  meta: {
+    key: 'org.example.function.query',
+    name: 'Query',
+    description: 'Queries the database',
+  },
+  input: Schema.Any,
+  output: Schema.Any,
+  services: [Database.Service],
+});

package/src/functions/fib.ts

@@ -3,30 +3,21 @@
 //
 
 import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
 
-import { defineFunction } from '@dxos/functions';
+import { Operation } from '@dxos/compute';
 
-export default defineFunction({
-  key: 'example.org/function/fib',
-  name: 'Fibonacci',
-  description: 'Function that calculates a Fibonacci number',
-  inputSchema: Schema.Struct({
-    iterations: Schema.optional(Schema.Number).annotations({
-      description: 'Number of iterations',
-      default: 100_000,
+import { Fibonacci } from './definitions';
+
+export default Fibonacci.pipe(
+  Operation.withHandler(
+    Effect.fn(function* ({ iterations = 100_000 }) {
+      let a = 0n;
+      let b = 1n;
+      for (let i = 0; i < iterations; i++) {
+        a += b;
+        b = a - b;
+      }
+      return { result: a.toString() };
     }),
-  }),
-  outputSchema: Schema.Struct({
-    result: Schema.String,
-  }),
-  handler: Effect.fn(function* ({ data: { iterations = 100_000 } }) {
-    let a = 0n;
-    let b = 1n;
-    for (let i = 0; i < iterations; i++) {
-      a += b;
-      b = a - b;
-    }
-    return { result: a.toString() };
-  }),
-});
+  ),
+);

package/src/functions/forex-effect.ts

@@ -10,35 +10,39 @@ import * as Effect from 'effect/Effect';
 import * as Schedule from 'effect/Schedule';
 import * as Schema from 'effect/Schema';
 
-import { defineFunction } from '@dxos/functions';
+import { Operation } from '@dxos/compute';
 
-export default defineFunction({
-  key: 'org.dxos.script.forex-effect',
-  name: 'Forex Effect',
-  description: 'Returns the exchange rate between two currencies.',
-
-  inputSchema: Schema.Struct({
+const ForexEffect = Operation.make({
+  meta: {
+    key: 'org.dxos.script.forex-effect',
+    name: 'Forex Effect',
+    description: 'Returns the exchange rate between two currencies.',
+  },
+  input: Schema.Struct({
     from: Schema.String.annotations({ description: 'The source currency' }),
     to: Schema.String.annotations({ description: 'The target currency' }),
   }),
+  output: Schema.String.annotations({ description: 'The exchange rate between the two currencies' }),
+});
 
-  outputSchema: Schema.String.annotations({ description: 'The exchange rate between the two currencies' }),
-
-  handler: Effect.fnUntraced(function* ({ data: { from, to } }: any) {
-    const res = yield* HttpClientRequest.get(`https://free.ratesdb.com/v1/rates?from=${from}&to=${to}`).pipe(
-      HttpClient.execute,
-      Effect.flatMap(
-        HttpClientResponse.schemaBodyJson(
-          Schema.Struct({
-            data: Schema.Struct({ rates: Schema.Record({ key: Schema.String, value: Schema.Number }) }),
-          }),
+export default ForexEffect.pipe(
+  Operation.withHandler(
+    Effect.fnUntraced(function* ({ from, to }) {
+      const res = yield* HttpClientRequest.get(`https://free.ratesdb.com/v1/rates?from=${from}&to=${to}`).pipe(
+        HttpClient.execute,
+        Effect.flatMap(
+          HttpClientResponse.schemaBodyJson(
+            Schema.Struct({
+              data: Schema.Struct({ rates: Schema.Record({ key: Schema.String, value: Schema.Number }) }),
+            }),
+          ),
         ),
-      ),
-      Effect.timeout('1 second'),
-      Effect.retry(Schedule.exponential(1_000).pipe(Schedule.compose(Schedule.recurs(3)))),
-      Effect.scoped,
-    );
+        Effect.timeout('1 second'),
+        Effect.retry(Schedule.exponential(1_000).pipe(Schedule.compose(Schedule.recurs(3)))),
+        Effect.scoped,
+      );
 
-    return res.data.rates[to].toString();
-  }, Effect.provide(FetchHttpClient.layer)),
-});
+      return res.data.rates[to].toString();
+    }, Effect.provide(FetchHttpClient.layer)),
+  ),
+);

package/src/functions/query.ts

@@ -3,19 +3,17 @@
 //
 
 import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
 
-import { Database, Filter, Query } from '@dxos/echo';
-import { defineFunction } from '@dxos/functions';
+import { Operation } from '@dxos/compute';
+import { Database, Filter, Query as EchoQuery } from '@dxos/echo';
 
-export default defineFunction({
-  key: 'example.org/function/query',
-  name: 'Query',
-  description: 'Queries the database',
-  inputSchema: Schema.Any,
-  outputSchema: Schema.Any,
-  handler: Effect.fn(function* ({ data }) {
-    const results = yield* Database.runQuery(Query.select(Filter.everything()));
-    return { count: results.length };
-  }),
-});
+import { QueryDb } from './definitions';
+
+export default QueryDb.pipe(
+  Operation.withHandler(
+    Effect.fn(function* (data) {
+      const results = yield* Database.runQuery(EchoQuery.select(Filter.everything()));
+      return { count: results.length };
+    }),
+  ),
+);

package/src/functions/reply.ts

@@ -4,18 +4,16 @@
 
 import * as Console from 'effect/Console';
 import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
 
-import { defineFunction } from '@dxos/functions';
+import { Operation } from '@dxos/compute';
 
-export default defineFunction({
-  key: 'example.org/function/reply',
-  name: 'Reply',
-  description: 'Function that echoes the input',
-  inputSchema: Schema.Any,
-  outputSchema: Schema.Any,
-  handler: Effect.fn(function* ({ data }) {
-    yield* Console.log('reply', { data });
-    return data;
-  }),
-});
+import { Reply } from './definitions';
+
+export default Reply.pipe(
+  Operation.withHandler(
+    Effect.fn(function* (data) {
+      yield* Console.log('reply', { data });
+      return data;
+    }),
+  ),
+);

package/src/functions/sleep.ts

@@ -3,22 +3,15 @@
 //
 
 import * as Effect from 'effect/Effect';
-import * as Schema from 'effect/Schema';
 
-import { defineFunction } from '@dxos/functions';
+import { Operation } from '@dxos/compute';
 
-export default defineFunction({
-  key: 'example.org/function/sleep',
-  name: 'Sleep',
-  description: 'Function that sleeps for a given amount of time',
-  inputSchema: Schema.Struct({
-    duration: Schema.optional(Schema.Number).annotations({
-      description: 'Milliseconds to sleep',
-      default: 100_000,
+import { Sleep } from './definitions';
+
+export default Sleep.pipe(
+  Operation.withHandler(
+    Effect.fn(function* ({ duration = 100_000 }) {
+      yield* Effect.sleep(duration);
     }),
-  }),
-  outputSchema: Schema.Void,
-  handler: Effect.fn(function* ({ data: { duration = 100_000 } }) {
-    yield* Effect.sleep(duration);
-  }),
-});
+  ),
+);

package/src/testing/util.ts

@@ -7,9 +7,9 @@ import fs from 'node:fs';
 import { sleep } from '@dxos/async';
 import { Client, type Config } from '@dxos/client';
 import { type Space } from '@dxos/client/echo';
-import { Filter, Obj, Query } from '@dxos/echo';
-import { Function } from '@dxos/functions';
-import { Trigger } from '@dxos/functions';
+import { Operation, Trigger } from '@dxos/compute';
+import { Context } from '@dxos/context';
+import { Feed, Obj } from '@dxos/echo';
 import { InvocationTraceEndEvent, InvocationTraceStartEvent } from '@dxos/functions-runtime';
 import { FunctionsServiceClient } from '@dxos/functions-runtime/edge';
 import { bundleFunction } from '@dxos/functions-runtime/native';
@@ -27,7 +27,7 @@ export const writeBundle = (path: string, bundle: BundleResult) => {
 export const setup = async (config: Config) => {
   const client = await new Client({
     config,
-    types: [Function.Function, Trigger.Trigger],
+    types: [Operation.PersistentOperation, Trigger.Trigger],
   }).initialize();
   await client.halo.createIdentity();
 
@@ -53,12 +53,12 @@ export const deployFunction = async (
   functionsServiceClient: FunctionsServiceClient,
   entryPoint: string,
   runtime: FunctionRuntimeKind,
-) => {
+): Promise<Operation.PersistentOperation> => {
   const artifact = await bundleFunction({
     entryPoint,
     verbose: true,
   });
-  const func = await functionsServiceClient.deploy({
+  const func = await functionsServiceClient.deploy(Context.default(), {
     version: '0.0.1',
     ownerPublicKey: space.key,
     entryPoint: artifact.entryPoint,
@@ -83,8 +83,9 @@ export const observeInvocations = async (space: Space, maxCount: number | null)
   let count = 0;
   while (true) {
     try {
-      const invocations =
-        (await space.properties.invocationTraceQueue?.target!.query(Query.select(Filter.everything())).run()) ?? [];
+      const traceFeed = space.properties.invocationTraceFeed?.target;
+      const traceFeedDXN = traceFeed ? Feed.getQueueDxn(traceFeed) : undefined;
+      const invocations = traceFeedDXN ? ((await space.queues.get(traceFeedDXN).queryObjects()) ?? []) : [];
 
       for (const invocation of invocations) {
         if (Obj.instanceOf(InvocationTraceStartEvent, invocation)) {