ai-workflows 2.1.1 → 2.3.0

package/test/workflow.test.ts

@@ -16,7 +16,7 @@ describe('Workflow - unified $ API', () => {
 
   describe('Workflow()', () => {
     it('should create a workflow with $ context', () => {
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         // Just accessing $ to verify it works
         expect($).toBeDefined()
         expect($.on).toBeDefined()
@@ -36,7 +36,7 @@ describe('Workflow - unified $ API', () => {
     })
 
     it('should capture event handlers registered via $.on', () => {
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.on.Customer.created(() => {})
         $.on.Order.completed(() => {})
       })
@@ -49,7 +49,7 @@ describe('Workflow - unified $ API', () => {
     })
 
     it('should capture schedule handlers registered via $.every', () => {
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.every.hour(() => {})
         $.every.Monday.at9am(() => {})
       })
@@ -58,7 +58,7 @@ describe('Workflow - unified $ API', () => {
     })
 
     it('should capture function source code', () => {
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.on.Test.event(async (data, ctx) => {
           ctx.log('Test event', data)
         })
@@ -73,7 +73,7 @@ describe('Workflow - unified $ API', () => {
     it('should deliver events to registered handlers', async () => {
       const handler = vi.fn()
 
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.on.Customer.created(handler)
       })
 
@@ -82,7 +82,7 @@ describe('Workflow - unified $ API', () => {
 
       expect(handler).toHaveBeenCalledTimes(1)
       expect(handler).toHaveBeenCalledWith(
-        { id: '123', name: 'John' },
+        expect.objectContaining({ id: '123', name: 'John' }),
         expect.objectContaining({
           send: expect.any(Function),
           log: expect.any(Function),
@@ -93,7 +93,7 @@ describe('Workflow - unified $ API', () => {
     it('should allow chained event sending from handlers', async () => {
       const welcomeHandler = vi.fn()
 
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.on.Customer.created(async (customer, $) => {
           await $.send('Email.welcome', { to: customer.email })
         })
@@ -105,13 +105,13 @@ describe('Workflow - unified $ API', () => {
       await workflow.send('Customer.created', { name: 'John', email: 'john@example.com' })
 
       expect(welcomeHandler).toHaveBeenCalledWith(
-        { to: 'john@example.com' },
+        expect.objectContaining({ to: 'john@example.com' }),
         expect.anything()
       )
     })
 
     it('should track events in state history', async () => {
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.on.Test.event(() => {})
       })
 
@@ -129,7 +129,7 @@ describe('Workflow - unified $ API', () => {
     it('should trigger schedule handlers', async () => {
       const handler = vi.fn()
 
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.every.seconds(1)(handler)
       })
 
@@ -147,7 +147,7 @@ describe('Workflow - unified $ API', () => {
     it('should stop schedule handlers on stop', async () => {
       const handler = vi.fn()
 
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.every.seconds(1)(handler)
       })
 
@@ -161,7 +161,7 @@ describe('Workflow - unified $ API', () => {
     })
 
     it('should support $.set and $.get for context data', async () => {
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.on.Test.set(async (data, $) => {
           $.set('value', data.value)
         })
@@ -180,10 +180,7 @@ describe('Workflow - unified $ API', () => {
     })
 
     it('should use initial context from options', () => {
-      const workflow = Workflow(
-        $ => {},
-        { context: { counter: 100 } }
-      )
+      const workflow = Workflow(($) => {}, { context: { counter: 100 } })
 
       expect(workflow.state.context.counter).toBe(100)
     })
@@ -225,8 +222,14 @@ describe('Workflow - unified $ API', () => {
       await $.send('Test.event2', { b: 2 })
 
       expect($.emittedEvents).toHaveLength(2)
-      expect($.emittedEvents[0]).toEqual({ event: 'Test.event1', data: { a: 1 } })
-      expect($.emittedEvents[1]).toEqual({ event: 'Test.event2', data: { b: 2 } })
+      expect($.emittedEvents[0]).toEqual({
+        event: 'Test.event1',
+        data: expect.objectContaining({ a: 1 }),
+      })
+      expect($.emittedEvents[1]).toEqual({
+        event: 'Test.event2',
+        data: expect.objectContaining({ b: 2 }),
+      })
     })
 
     it('should support get/set', () => {
@@ -239,7 +242,7 @@ describe('Workflow - unified $ API', () => {
 
   describe('$.every patterns', () => {
     it('should support $.every.hour', () => {
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.every.hour(() => {})
       })
 
@@ -251,7 +254,7 @@ describe('Workflow - unified $ API', () => {
     })
 
     it('should support $.every.Monday.at9am', () => {
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.every.Monday.at9am(() => {})
       })
 
@@ -263,7 +266,7 @@ describe('Workflow - unified $ API', () => {
     })
 
     it('should support $.every.minutes(30)', () => {
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.every.minutes(30)(() => {})
       })
 
@@ -275,7 +278,7 @@ describe('Workflow - unified $ API', () => {
     })
 
     it('should support $.every("natural language")', () => {
-      const workflow = Workflow($ => {
+      const workflow = Workflow(($) => {
         $.every('first Monday of the month', () => {})
       })
 

package/tsconfig.json

@@ -2,7 +2,9 @@
   "extends": "../../tsconfig.base.json",
   "compilerOptions": {
     "rootDir": "src",
-    "outDir": "dist"
+    "outDir": "dist",
+    "lib": ["ES2022", "ESNext.Disposable"],
+    "types": ["@cloudflare/workers-types", "node"]
   },
   "include": ["src/**/*"],
   "exclude": ["node_modules", "dist", "**/*.test.ts", "**/*.spec.ts"]

package/vitest.config.ts

@@ -2,7 +2,44 @@ import { defineConfig } from 'vitest/config'
 
 export default defineConfig({
   test: {
+    // CRITICAL: Limit concurrency to prevent resource exhaustion
+    maxConcurrency: 1,
+    maxWorkers: 1,
+    minWorkers: 1,
+    fileParallelism: false,
+
     globals: true,
-    include: ['test/**/*.test.ts'],
+    environment: 'node',
+    include: ['test/**/*.test.ts', 'src/**/*.test.ts'],
+    testTimeout: 30000,
+    hookTimeout: 15000,
+
+    // Run tests sequentially with better memory isolation
+    pool: 'forks',
+    poolOptions: {
+      forks: {
+        singleFork: true,
+      },
+    },
+
+    // Coverage configuration
+    coverage: {
+      provider: 'v8',
+      reporter: ['text', 'json', 'html'],
+      include: ['src/**/*.ts'],
+      exclude: ['**/*.test.ts', '**/__tests__/**', '**/node_modules/**'],
+      thresholds: {
+        statements: 65,
+        branches: 60,
+        functions: 60,
+        lines: 65,
+      },
+    },
+
+    // Alias for cloudflare:workers mock (shared across monorepo)
+    alias: {
+      'cloudflare:workers': new URL('../config/mocks/cloudflare-workers.ts', import.meta.url)
+        .pathname,
+    },
   },
 })

package/vitest.workers.config.ts

@@ -0,0 +1,44 @@
+import { defineWorkersConfig } from '@cloudflare/vitest-pool-workers/config'
+
+export default defineWorkersConfig({
+  test: {
+    // CRITICAL: Limit concurrency to prevent resource exhaustion
+    maxConcurrency: 1,
+    maxWorkers: 1,
+    minWorkers: 1,
+    fileParallelism: false,
+
+    poolOptions: {
+      workers: {
+        wrangler: { configPath: './wrangler.jsonc' },
+        // Disable isolated storage for Workflows tests
+        // Workflows require non-isolated mode due to how miniflare handles them
+        isolatedStorage: false,
+        singleWorker: true,
+        miniflare: {
+          compatibilityDate: '2025-01-20',
+          compatibilityFlags: ['nodejs_compat_v2'],
+        },
+      },
+    },
+
+    // Only include worker-specific tests
+    include: ['test/worker/**/*.test.ts'],
+    testTimeout: 60000,
+    hookTimeout: 30000,
+
+    // Coverage configuration
+    coverage: {
+      provider: 'v8',
+      reporter: ['text', 'json', 'html'],
+      include: ['src/**/*.ts'],
+      exclude: ['**/*.test.ts', '**/__tests__/**', '**/node_modules/**'],
+      thresholds: {
+        statements: 65,
+        branches: 60,
+        functions: 60,
+        lines: 65,
+      },
+    },
+  },
+})

package/wrangler.jsonc

@@ -0,0 +1,22 @@
+{
+  // Wrangler configuration for ai-workflows worker tests
+  // Used by @cloudflare/vitest-pool-workers for testing
+  //
+  // Configures Cloudflare Workflows binding for testing DurableStep
+  // with real Workflows runtime semantics.
+
+  "name": "ai-workflows",
+  "main": "src/worker.ts",
+  "compatibility_date": "2025-01-20",
+  "compatibility_flags": ["nodejs_compat_v2"],
+
+  // Cloudflare Workflows binding
+  // This binds the TestWorkflow class to WORKFLOW for testing DurableStep
+  "workflows": [
+    {
+      "name": "test-workflow",
+      "binding": "WORKFLOW",
+      "class_name": "TestWorkflow"
+    }
+  ]
+}

package/.turbo/turbo-test.log

@@ -1,7 +0,0 @@
-
-> ai-workflows@2.0.3 test /Users/nathanclevenger/projects/primitives.org.ai/packages/ai-workflows
-> vitest "--run"
-
-
- RUN  v2.1.9 /Users/nathanclevenger/projects/primitives.org.ai/packages/ai-workflows
-

package/src/context.js

@@ -1,83 +0,0 @@
-/**
- * Workflow context implementation
- */
-/**
- * Create a workflow context
- */
-export function createWorkflowContext(eventBus) {
-    const workflowState = {
-        context: {},
-        history: [],
-    };
-    const addHistory = (entry) => {
-        workflowState.history.push({
-            ...entry,
-            timestamp: Date.now(),
-        });
-    };
-    // Create no-op proxies for on/every (these are used in send context, not workflow setup)
-    const noOpOnProxy = new Proxy({}, {
-        get() {
-            return new Proxy({}, {
-                get() {
-                    return () => { };
-                }
-            });
-        }
-    });
-    const noOpEveryProxy = new Proxy(function () { }, {
-        get() {
-            return () => () => { };
-        },
-        apply() { }
-    });
-    return {
-        async send(event, data) {
-            addHistory({ type: 'event', name: event, data });
-            await eventBus.emit(event, data);
-        },
-        async do(_event, _data) {
-            throw new Error('$.do not available in this context');
-        },
-        async try(_event, _data) {
-            throw new Error('$.try not available in this context');
-        },
-        on: noOpOnProxy,
-        every: noOpEveryProxy,
-        state: workflowState.context,
-        getState() {
-            // Return a deep copy to prevent mutation
-            return {
-                current: workflowState.current,
-                context: { ...workflowState.context },
-                history: [...workflowState.history],
-            };
-        },
-        set(key, value) {
-            workflowState.context[key] = value;
-        },
-        get(key) {
-            return workflowState.context[key];
-        },
-        log(message, data) {
-            addHistory({ type: 'action', name: 'log', data: { message, data } });
-            console.log(`[workflow] ${message}`, data ?? '');
-        },
-    };
-}
-/**
- * Create an isolated workflow context (not connected to event bus)
- * Useful for testing or standalone execution
- */
-export function createIsolatedContext() {
-    const emittedEvents = [];
-    const ctx = createWorkflowContext({
-        async emit(event, data) {
-            emittedEvents.push({ event, data });
-        },
-    });
-    return {
-        ...ctx,
-        getEmittedEvents: () => [...emittedEvents],
-    };
-}

package/src/every.js

@@ -1,267 +0,0 @@
-/**
- * Schedule registration using natural language
- *
- * Usage:
- *   every.hour($ => { ... })
- *   every.Thursday.at8am($ => { ... })
- *   every.weekday.at9am($ => { ... })
- *   every('hour during business hours', $ => { ... })
- *   every('first Monday of the month at 9am', $ => { ... })
- */
-/**
- * Registry of schedule handlers
- */
-const scheduleRegistry = [];
-/**
- * Get all registered schedule handlers
- */
-export function getScheduleHandlers() {
-    return [...scheduleRegistry];
-}
-/**
- * Clear all registered schedule handlers
- */
-export function clearScheduleHandlers() {
-    scheduleRegistry.length = 0;
-}
-/**
- * Register a schedule handler directly
- */
-export function registerScheduleHandler(interval, handler) {
-    scheduleRegistry.push({
-        interval,
-        handler,
-        source: handler.toString(),
-    });
-}
-/**
- * Well-known cron patterns for common schedules
- */
-const KNOWN_PATTERNS = {
-    // Time units
-    'second': '* * * * * *',
-    'minute': '* * * * *',
-    'hour': '0 * * * *',
-    'day': '0 0 * * *',
-    'week': '0 0 * * 0',
-    'month': '0 0 1 * *',
-    'year': '0 0 1 1 *',
-    // Days of week
-    'Monday': '0 0 * * 1',
-    'Tuesday': '0 0 * * 2',
-    'Wednesday': '0 0 * * 3',
-    'Thursday': '0 0 * * 4',
-    'Friday': '0 0 * * 5',
-    'Saturday': '0 0 * * 6',
-    'Sunday': '0 0 * * 0',
-    // Common patterns
-    'weekday': '0 0 * * 1-5',
-    'weekend': '0 0 * * 0,6',
-    'midnight': '0 0 * * *',
-    'noon': '0 12 * * *',
-};
-/**
- * Time suffixes for day-based schedules
- */
-const TIME_PATTERNS = {
-    'at6am': { hour: 6, minute: 0 },
-    'at7am': { hour: 7, minute: 0 },
-    'at8am': { hour: 8, minute: 0 },
-    'at9am': { hour: 9, minute: 0 },
-    'at10am': { hour: 10, minute: 0 },
-    'at11am': { hour: 11, minute: 0 },
-    'at12pm': { hour: 12, minute: 0 },
-    'atnoon': { hour: 12, minute: 0 },
-    'at1pm': { hour: 13, minute: 0 },
-    'at2pm': { hour: 14, minute: 0 },
-    'at3pm': { hour: 15, minute: 0 },
-    'at4pm': { hour: 16, minute: 0 },
-    'at5pm': { hour: 17, minute: 0 },
-    'at6pm': { hour: 18, minute: 0 },
-    'at7pm': { hour: 19, minute: 0 },
-    'at8pm': { hour: 20, minute: 0 },
-    'at9pm': { hour: 21, minute: 0 },
-    'atmidnight': { hour: 0, minute: 0 },
-};
-/**
- * Parse a known pattern or return null
- */
-function parseKnownPattern(pattern) {
-    return KNOWN_PATTERNS[pattern] ?? null;
-}
-/**
- * Combine a day pattern with a time pattern
- */
-function combineWithTime(baseCron, time) {
-    const parts = baseCron.split(' ');
-    parts[0] = String(time.minute);
-    parts[1] = String(time.hour);
-    return parts.join(' ');
-}
-/**
- * AI-powered cron conversion (placeholder - will use ai-functions)
- */
-let cronConverter = null;
-/**
- * Set the AI cron converter function
- */
-export function setCronConverter(converter) {
-    cronConverter = converter;
-}
-/**
- * Convert natural language to cron expression
- */
-export async function toCron(description) {
-    // First check known patterns
-    const known = parseKnownPattern(description);
-    if (known)
-        return known;
-    // If we have an AI converter, use it
-    if (cronConverter) {
-        return cronConverter(description);
-    }
-    // Otherwise, assume it's already a cron expression or throw
-    if (/^[\d\*\-\/\,\s]+$/.test(description)) {
-        return description;
-    }
-    throw new Error(`Unknown schedule pattern: "${description}". ` +
-        `Set up AI conversion with setCronConverter() for natural language schedules.`);
-}
-/**
- * Create the `every` proxy
- */
-function createEveryProxy() {
-    const handler = {
-        get(_target, prop) {
-            // Check if it's a known pattern
-            const pattern = KNOWN_PATTERNS[prop];
-            if (pattern) {
-                // Return an object that can either be called directly or have time accessors
-                const result = (handlerFn) => {
-                    registerScheduleHandler({ type: 'cron', expression: pattern, natural: prop }, handlerFn);
-                };
-                // Add time accessors
-                return new Proxy(result, {
-                    get(_t, timeKey) {
-                        const time = TIME_PATTERNS[timeKey];
-                        if (time) {
-                            const cron = combineWithTime(pattern, time);
-                            return (handlerFn) => {
-                                registerScheduleHandler({ type: 'cron', expression: cron, natural: `${prop}.${timeKey}` }, handlerFn);
-                            };
-                        }
-                        return undefined;
-                    },
-                    apply(_t, _thisArg, args) {
-                        registerScheduleHandler({ type: 'cron', expression: pattern, natural: prop }, args[0]);
-                    }
-                });
-            }
-            // Check for plural time units (e.g., seconds(5), minutes(30))
-            const pluralUnits = {
-                seconds: 'second',
-                minutes: 'minute',
-                hours: 'hour',
-                days: 'day',
-                weeks: 'week',
-            };
-            if (pluralUnits[prop]) {
-                return (value) => (handlerFn) => {
-                    registerScheduleHandler({ type: pluralUnits[prop], value, natural: `${value} ${prop}` }, handlerFn);
-                };
-            }
-            return undefined;
-        },
-        apply(_target, _thisArg, args) {
-            // Called as every('natural language description', handler)
-            const [description, handler] = args;
-            if (typeof description === 'string' && typeof handler === 'function') {
-                // Register with natural language - will be converted to cron at runtime
-                registerScheduleHandler({ type: 'natural', description }, handler);
-            }
-        }
-    };
-    return new Proxy(function () { }, handler);
-}
-/**
- * The `every` function/object for registering scheduled handlers
- *
- * @example
- * ```ts
- * import { every } from 'ai-workflows'
- *
- * // Simple intervals
- * every.hour($ => $.log('Hourly task'))
- * every.day($ => $.log('Daily task'))
- *
- * // Day + time combinations
- * every.Monday.at9am($ => $.log('Monday morning standup'))
- * every.weekday.at8am($ => $.log('Workday start'))
- * every.Friday.at5pm($ => $.log('End of week report'))
- *
- * // Plural intervals with values
- * every.minutes(30)($ => $.log('Every 30 minutes'))
- * every.hours(4)($ => $.log('Every 4 hours'))
- *
- * // Natural language (requires AI converter)
- * every('hour during business hours', $ => { ... })
- * every('first Monday of the month at 9am', $ => { ... })
- * every('every 15 minutes between 9am and 5pm on weekdays', $ => { ... })
- * ```
- */
-export const every = createEveryProxy();
-/**
- * Convert interval to milliseconds (for simulation/testing)
- */
-export function intervalToMs(interval) {
-    switch (interval.type) {
-        case 'second':
-            return (interval.value ?? 1) * 1000;
-        case 'minute':
-            return (interval.value ?? 1) * 60 * 1000;
-        case 'hour':
-            return (interval.value ?? 1) * 60 * 60 * 1000;
-        case 'day':
-            return (interval.value ?? 1) * 24 * 60 * 60 * 1000;
-        case 'week':
-            return (interval.value ?? 1) * 7 * 24 * 60 * 60 * 1000;
-        case 'cron':
-        case 'natural':
-            // Cron/natural expressions need special handling
-            return 0;
-    }
-}
-/**
- * Format interval for display
- */
-export function formatInterval(interval) {
-    if ('natural' in interval && interval.natural) {
-        return interval.natural;
-    }
-    switch (interval.type) {
-        case 'second':
-            return interval.value && interval.value > 1
-                ? `every ${interval.value} seconds`
-                : 'every second';
-        case 'minute':
-            return interval.value && interval.value > 1
-                ? `every ${interval.value} minutes`
-                : 'every minute';
-        case 'hour':
-            return interval.value && interval.value > 1
-                ? `every ${interval.value} hours`
-                : 'every hour';
-        case 'day':
-            return interval.value && interval.value > 1
-                ? `every ${interval.value} days`
-                : 'every day';
-        case 'week':
-            return interval.value && interval.value > 1
-                ? `every ${interval.value} weeks`
-                : 'every week';
-        case 'cron':
-            return `cron: ${interval.expression}`;
-        case 'natural':
-            return interval.description;
-    }
-}