@cogitator-ai/workflows 0.1.0

package/dist/manager/scheduler.js

@@ -0,0 +1,355 @@
+/**
+ * Workflow Scheduler
+ *
+ * Features:
+ * - Priority queue for pending runs
+ * - Cron-based scheduling
+ * - Delayed execution
+ * - Concurrency control
+ */
+import { nanoid } from 'nanoid';
+import { getNextCronOccurrence, validateCronExpression, } from '../timers/cron-parser';
+/**
+ * Priority queue implementation using a heap
+ */
+export class PriorityQueue {
+    items = [];
+    /**
+     * Add item to queue (sorted by scheduledFor then priority)
+     */
+    enqueue(item) {
+        this.items.push(item);
+        this.bubbleUp(this.items.length - 1);
+    }
+    /**
+     * Remove and return highest priority item
+     */
+    dequeue() {
+        if (this.items.length === 0)
+            return undefined;
+        if (this.items.length === 1)
+            return this.items.pop();
+        const result = this.items[0];
+        this.items[0] = this.items.pop();
+        this.bubbleDown(0);
+        return result;
+    }
+    /**
+     * Peek at highest priority item without removing
+     */
+    peek() {
+        return this.items[0];
+    }
+    /**
+     * Remove item by runId
+     */
+    remove(runId) {
+        const index = this.items.findIndex((item) => item.runId === runId);
+        if (index === -1)
+            return false;
+        if (index === this.items.length - 1) {
+            this.items.pop();
+        }
+        else {
+            this.items[index] = this.items.pop();
+            this.bubbleDown(index);
+            this.bubbleUp(index);
+        }
+        return true;
+    }
+    /**
+     * Get items ready to execute (scheduledFor <= now)
+     */
+    getReady(now = Date.now()) {
+        const ready = [];
+        while (this.items.length > 0) {
+            const top = this.items[0];
+            if (top.scheduledFor > now)
+                break;
+            ready.push(this.dequeue());
+        }
+        return ready;
+    }
+    /**
+     * Get queue size
+     */
+    size() {
+        return this.items.length;
+    }
+    /**
+     * Clear the queue
+     */
+    clear() {
+        this.items = [];
+    }
+    /**
+     * Get all items (for inspection)
+     */
+    getAll() {
+        return [...this.items];
+    }
+    bubbleUp(index) {
+        while (index > 0) {
+            const parentIndex = Math.floor((index - 1) / 2);
+            if (this.compare(this.items[index], this.items[parentIndex]) >= 0)
+                break;
+            [this.items[index], this.items[parentIndex]] = [
+                this.items[parentIndex],
+                this.items[index],
+            ];
+            index = parentIndex;
+        }
+    }
+    bubbleDown(index) {
+        while (true) {
+            const leftChild = 2 * index + 1;
+            const rightChild = 2 * index + 2;
+            let smallest = index;
+            if (leftChild < this.items.length &&
+                this.compare(this.items[leftChild], this.items[smallest]) < 0) {
+                smallest = leftChild;
+            }
+            if (rightChild < this.items.length &&
+                this.compare(this.items[rightChild], this.items[smallest]) < 0) {
+                smallest = rightChild;
+            }
+            if (smallest === index)
+                break;
+            [this.items[index], this.items[smallest]] = [
+                this.items[smallest],
+                this.items[index],
+            ];
+            index = smallest;
+        }
+    }
+    compare(a, b) {
+        if (a.scheduledFor !== b.scheduledFor) {
+            return a.scheduledFor - b.scheduledFor;
+        }
+        return a.priority - b.priority;
+    }
+}
+/**
+ * Job scheduler for workflow runs
+ */
+export class JobScheduler {
+    queue = new PriorityQueue();
+    cronJobs = new Map();
+    runStore;
+    maxConcurrency;
+    pollInterval;
+    onRunReady;
+    runningCount = 0;
+    pollTimer;
+    disposed = false;
+    constructor(config) {
+        this.runStore = config.runStore;
+        this.maxConcurrency = config.maxConcurrency ?? Infinity;
+        this.pollInterval = config.pollInterval ?? 1000;
+        this.onRunReady = config.onRunReady;
+    }
+    /**
+     * Start the scheduler
+     */
+    start() {
+        if (this.pollTimer)
+            return;
+        this.pollTimer = setInterval(() => {
+            this.tick();
+        }, this.pollInterval);
+        this.tick();
+    }
+    /**
+     * Stop the scheduler
+     */
+    stop() {
+        if (this.pollTimer) {
+            clearInterval(this.pollTimer);
+            this.pollTimer = undefined;
+        }
+    }
+    /**
+     * Schedule a workflow run
+     */
+    async scheduleRun(workflow, options = {}) {
+        const runId = nanoid();
+        const now = Date.now();
+        let scheduledFor = now;
+        if (options.at) {
+            scheduledFor = options.at;
+        }
+        else if (options.cron) {
+            validateCronExpression(options.cron);
+            scheduledFor = getNextCronOccurrence(options.cron, {
+                currentDate: new Date(),
+                timezone: options.timezone,
+            }).getTime();
+        }
+        const run = {
+            id: runId,
+            workflowName: workflow.name,
+            status: scheduledFor > now ? 'scheduled' : 'pending',
+            state: options.input ?? {},
+            input: options.input,
+            currentNodes: [],
+            completedNodes: [],
+            failedNodes: [],
+            scheduledFor: scheduledFor > now ? scheduledFor : undefined,
+            priority: options.priority ?? 0,
+            tags: options.tags ?? [],
+            triggerId: options.triggerId,
+            traceId: options.traceContext?.traceId,
+            metadata: {
+                baggage: options.baggage,
+                timeout: options.timeout,
+                maxRetries: options.maxRetries,
+            },
+        };
+        await this.runStore.save(run);
+        this.queue.enqueue({
+            runId,
+            workflowName: workflow.name,
+            priority: options.priority ?? 0,
+            scheduledFor,
+        });
+        return runId;
+    }
+    /**
+     * Register a cron-based recurring job
+     */
+    registerCronJob(workflow, expression, options = {}) {
+        validateCronExpression(expression);
+        const jobId = options.id ?? nanoid();
+        const nextRun = getNextCronOccurrence(expression, {
+            currentDate: new Date(),
+            timezone: options.timezone,
+        }).getTime();
+        const job = {
+            id: jobId,
+            workflowName: workflow.name,
+            workflow: workflow,
+            expression,
+            timezone: options.timezone,
+            options: options.jobOptions,
+            nextRun,
+            enabled: true,
+        };
+        this.cronJobs.set(jobId, job);
+        return jobId;
+    }
+    /**
+     * Unregister a cron job
+     */
+    unregisterCronJob(jobId) {
+        return this.cronJobs.delete(jobId);
+    }
+    /**
+     * Enable/disable a cron job
+     */
+    setCronJobEnabled(jobId, enabled) {
+        const job = this.cronJobs.get(jobId);
+        if (!job)
+            return false;
+        job.enabled = enabled;
+        if (enabled) {
+            job.nextRun = getNextCronOccurrence(job.expression, {
+                currentDate: new Date(),
+                timezone: job.timezone,
+            }).getTime();
+        }
+        return true;
+    }
+    /**
+     * Cancel a scheduled run
+     */
+    async cancelRun(runId, reason) {
+        const run = await this.runStore.get(runId);
+        if (!run)
+            return false;
+        if (run.status !== 'scheduled' && run.status !== 'pending') {
+            return false;
+        }
+        this.queue.remove(runId);
+        await this.runStore.update(runId, {
+            status: 'cancelled',
+            completedAt: Date.now(),
+            error: reason ? { name: 'CancelError', message: reason } : undefined,
+        });
+        return true;
+    }
+    /**
+     * Mark run as started
+     */
+    runStarted(_runId) {
+        this.runningCount++;
+    }
+    /**
+     * Mark run as completed
+     */
+    runCompleted(_runId) {
+        this.runningCount = Math.max(0, this.runningCount - 1);
+    }
+    /**
+     * Get queue size
+     */
+    getQueueSize() {
+        return this.queue.size();
+    }
+    /**
+     * Get running count
+     */
+    getRunningCount() {
+        return this.runningCount;
+    }
+    /**
+     * Get all cron jobs
+     */
+    getCronJobs() {
+        return Array.from(this.cronJobs.values());
+    }
+    /**
+     * Dispose the scheduler
+     */
+    dispose() {
+        this.disposed = true;
+        this.stop();
+        this.queue.clear();
+        this.cronJobs.clear();
+    }
+    tick() {
+        if (this.disposed)
+            return;
+        const now = Date.now();
+        for (const job of this.cronJobs.values()) {
+            if (!job.enabled)
+                continue;
+            if (job.nextRun <= now) {
+                this.scheduleRun(job.workflow, {
+                    ...job.options,
+                    triggerId: `cron:${job.id}`,
+                });
+                job.nextRun = getNextCronOccurrence(job.expression, {
+                    currentDate: new Date(now + 1000),
+                    timezone: job.timezone,
+                }).getTime();
+            }
+        }
+        if (this.runningCount >= this.maxConcurrency)
+            return;
+        const ready = this.queue.getReady(now);
+        for (const item of ready) {
+            if (this.runningCount >= this.maxConcurrency) {
+                this.queue.enqueue(item);
+                break;
+            }
+            this.onRunReady?.(item.runId);
+        }
+    }
+}
+/**
+ * Create a job scheduler
+ */
+export function createJobScheduler(config) {
+    return new JobScheduler(config);
+}
+//# sourceMappingURL=scheduler.js.map

package/dist/manager/scheduler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"scheduler.js","sourceRoot":"","sources":["../../src/manager/scheduler.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAQhC,OAAO,EACL,qBAAqB,EACrB,sBAAsB,GACvB,MAAM,uBAAuB,CAAC;AAsB/B;;GAEG;AACH,MAAM,OAAO,aAAa;IAChB,KAAK,GAAgB,EAAE,CAAC;IAEhC;;OAEG;IACH,OAAO,CAAC,IAAe;QACrB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAC9C,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QAErD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,EAAG,CAAC;QAClC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACnB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,IAAI;QACF,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAa;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;QACnE,IAAI,KAAK,KAAK,CAAC,CAAC;YAAE,OAAO,KAAK,CAAC;QAE/B,IAAI,KAAK,KAAK,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QACnB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,EAAG,CAAC;YACtC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACvB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,MAAc,IAAI,CAAC,GAAG,EAAE;QAC/B,MAAM,KAAK,GAAgB,EAAE,CAAC;QAE9B,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAC1B,IAAI,GAAG,CAAC,YAAY,GAAG,GAAG;gBAAE,MAAM;YAClC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAG,CAAC,CAAC;QAC9B,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,IAAI;QACF,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;IAClB,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IAEO,QAAQ,CAAC,KAAa;QAC5B,OAAO,KAAK,GAAG,CAAC,EAAE,CAAC;YACjB,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAChD,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC;gBAAE,MAAM;YAEzE,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,GAAG;gBAC7C,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC;gBACvB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;aAClB,CAAC;YACF,KAAK,GAAG,WAAW,CAAC;QACtB,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,KAAa;QAC9B,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,SAAS,GAAG,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC;YAChC,MAAM,UAAU,GAAG,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC;YACjC,IAAI,QAAQ,GAAG,KAAK,CAAC;YAErB,IACE,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM;gBAC7B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,EAC7D,CAAC;gBACD,QAAQ,GAAG,SAAS,CAAC;YACvB,CAAC;YAED,IACE,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM;gBAC9B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,EAC9D,CAAC;gBACD,QAAQ,GAAG,UAAU,CAAC;YACxB,CAAC;YAED,IAAI,QAAQ,KAAK,KAAK;gBAAE,MAAM;YAE9B,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,GAAG;gBAC1C,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;gBACpB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;aAClB,CAAC;YACF,KAAK,GAAG,QAAQ,CAAC;QACnB,CAAC;IACH,CAAC;IAEO,OAAO,CAAC,CAAY,EAAE,CAAY;QACxC,IAAI,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,YAAY,EAAE,CAAC;YACtC,OAAO,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,YAAY,CAAC;QACzC,CAAC;QACD,OAAO,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC;IACjC,CAAC;CACF;AAgBD;;GAEG;AACH,MAAM,OAAO,YAAY;IACf,KAAK,GAAG,IAAI,aAAa,EAAE,CAAC;IAC5B,QAAQ,GAAG,IAAI,GAAG,EAAmB,CAAC;IACtC,QAAQ,CAAW;IACnB,cAAc,CAAS;IACvB,YAAY,CAAS;IACrB,UAAU,CAA2B;IACrC,YAAY,GAAG,CAAC,CAAC;IACjB,SAAS,CAAkC;IAC3C,QAAQ,GAAG,KAAK,CAAC;IAEzB,YAAY,MAAuB;QACjC,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,QAAQ,CAAC;QACxD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,IAAI,CAAC;QAChD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAE3B,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE;YAChC,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,CAAC,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QAEtB,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAED;;OAEG;IACH,IAAI;QACF,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC9B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CACf,QAAqB,EACrB,UAA2B,EAAE;QAE7B,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,IAAI,YAAY,GAAG,GAAG,CAAC;QACvB,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;YACf,YAAY,GAAG,OAAO,CAAC,EAAE,CAAC;QAC5B,CAAC;aAAM,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YACxB,sBAAsB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACrC,YAAY,GAAG,qBAAqB,CAAC,OAAO,CAAC,IAAI,EAAE;gBACjD,WAAW,EAAE,IAAI,IAAI,EAAE;gBACvB,QAAQ,EAAE,OAAO,CAAC,QAAQ;aAC3B,CAAC,CAAC,OAAO,EAAE,CAAC;QACf,CAAC;QAED,MAAM,GAAG,GAAgB;YACvB,EAAE,EAAE,KAAK;YACT,YAAY,EAAE,QAAQ,CAAC,IAAI;YAC3B,MAAM,EAAE,YAAY,GAAG,GAAG,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;YACpD,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,EAAE;YAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,YAAY,EAAE,EAAE;YAChB,cAAc,EAAE,EAAE;YAClB,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,YAAY,GAAG,GAAG,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS;YAC3D,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,CAAC;YAC/B,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;YACxB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,OAAO,EAAE,OAAO,CAAC,YAAY,EAAE,OAAO;YACtC,QAAQ,EAAE;gBACR,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,UAAU,EAAE,OAAO,CAAC,UAAU;aAC/B;SACF,CAAC;QAEF,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE9B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;YACjB,KAAK;YACL,YAAY,EAAE,QAAQ,CAAC,IAAI;YAC3B,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,CAAC;YAC/B,YAAY;SACb,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,eAAe,CACb,QAAqB,EACrB,UAAkB,EAClB,UAII,EAAE;QAEN,sBAAsB,CAAC,UAAU,CAAC,CAAC;QAEnC,MAAM,KAAK,GAAG,OAAO,CAAC,EAAE,IAAI,MAAM,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,qBAAqB,CAAC,UAAU,EAAE;YAChD,WAAW,EAAE,IAAI,IAAI,EAAE;YACvB,QAAQ,EAAE,OAAO,CAAC,QAAQ;SAC3B,CAAC,CAAC,OAAO,EAAE,CAAC;QAEb,MAAM,GAAG,GAAY;YACnB,EAAE,EAAE,KAAK;YACT,YAAY,EAAE,QAAQ,CAAC,IAAI;YAC3B,QAAQ,EAAE,QAA8C;YACxD,UAAU;YACV,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,OAAO,EAAE,OAAO,CAAC,UAAU;YAC3B,OAAO;YACP,OAAO,EAAE,IAAI;SACd,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAE9B,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,KAAa;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,KAAa,EAAE,OAAgB;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,GAAG;YAAE,OAAO,KAAK,CAAC;QAEvB,GAAG,CAAC,OAAO,GAAG,OAAO,CAAC;QACtB,IAAI,OAAO,EAAE,CAAC;YACZ,GAAG,CAAC,OAAO,GAAG,qBAAqB,CAAC,GAAG,CAAC,UAAU,EAAE;gBAClD,WAAW,EAAE,IAAI,IAAI,EAAE;gBACvB,QAAQ,EAAE,GAAG,CAAC,QAAQ;aACvB,CAAC,CAAC,OAAO,EAAE,CAAC;QACf,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,KAAa,EAAE,MAAe;QAC5C,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,GAAG;YAAE,OAAO,KAAK,CAAC;QAEvB,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC3D,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAEzB,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;YAChC,MAAM,EAAE,WAAW;YACnB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;YACvB,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS;SACrE,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,MAAc;QACvB,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,MAAc;QACzB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;IACzD,CAAC;IAED;;OAEG;IACH,YAAY;QACV,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,eAAe;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;IAEO,IAAI;QACV,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QAE1B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,GAAG,CAAC,OAAO;gBAAE,SAAS;YAE3B,IAAI,GAAG,CAAC,OAAO,IAAI,GAAG,EAAE,CAAC;gBACvB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE;oBAC7B,GAAG,GAAG,CAAC,OAAO;oBACd,SAAS,EAAE,QAAQ,GAAG,CAAC,EAAE,EAAE;iBAC5B,CAAC,CAAC;gBAEH,GAAG,CAAC,OAAO,GAAG,qBAAqB,CAAC,GAAG,CAAC,UAAU,EAAE;oBAClD,WAAW,EAAE,IAAI,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC;oBACjC,QAAQ,EAAE,GAAG,CAAC,QAAQ;iBACvB,CAAC,CAAC,OAAO,EAAE,CAAC;YACf,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO;QAErD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QAEvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBAC7C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBACzB,MAAM;YACR,CAAC;YAED,IAAI,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,MAAuB;IACxD,OAAO,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;AAClC,CAAC"}

package/dist/manager/workflow-manager.d.ts

@@ -0,0 +1,114 @@
+/**
+ * Workflow Manager
+ *
+ * Features:
+ * - Schedule workflows for later execution
+ * - Execute workflows immediately
+ * - Cancel, pause, resume, retry runs
+ * - Replay from specific nodes
+ * - Query run status and history
+ * - Cleanup old runs
+ */
+import type { Workflow, WorkflowState, WorkflowResult, WorkflowRun, WorkflowRunFilters, WorkflowRunStats, WorkflowManager as IWorkflowManager, ScheduleOptions, WorkflowExecuteOptionsV2, RunStore, CheckpointStore } from '@cogitator-ai/types';
+import type { Cogitator } from '@cogitator-ai/core';
+/**
+ * Workflow manager configuration
+ */
+export interface WorkflowManagerConfig {
+    cogitator: Cogitator;
+    runStore?: RunStore;
+    checkpointStore?: CheckpointStore;
+    maxConcurrency?: number;
+    defaultTimeout?: number;
+    onRunStateChange?: (run: WorkflowRun) => void;
+}
+/**
+ * Workflow manager implementation
+ */
+export declare class DefaultWorkflowManager implements IWorkflowManager {
+    private cogitator;
+    private runStore;
+    private checkpointStore?;
+    private scheduler;
+    private executor;
+    private workflows;
+    private activeRuns;
+    private stateChangeCallbacks;
+    constructor(config: WorkflowManagerConfig);
+    /**
+     * Start the manager (begins processing scheduled runs)
+     */
+    start(): void;
+    /**
+     * Stop the manager
+     */
+    stop(): void;
+    /**
+     * Register a workflow for scheduling
+     */
+    registerWorkflow<S extends WorkflowState>(workflow: Workflow<S>): void;
+    /**
+     * Schedule a workflow for later execution
+     */
+    schedule<S extends WorkflowState>(workflow: Workflow<S>, options?: ScheduleOptions): Promise<string>;
+    /**
+     * Execute a workflow immediately
+     */
+    execute<S extends WorkflowState>(workflow: Workflow<S>, input?: Partial<S>, options?: WorkflowExecuteOptionsV2): Promise<WorkflowResult<S>>;
+    /**
+     * Cancel a run
+     */
+    cancel(runId: string, reason?: string): Promise<void>;
+    /**
+     * Get run status
+     */
+    getStatus(runId: string): Promise<WorkflowRun | null>;
+    /**
+     * List runs with filters
+     */
+    listRuns(filters?: WorkflowRunFilters): Promise<WorkflowRun[]>;
+    /**
+     * Get run statistics
+     */
+    getStats(workflowName?: string): Promise<WorkflowRunStats>;
+    /**
+     * Pause a running workflow
+     */
+    pause(runId: string): Promise<void>;
+    /**
+     * Resume a paused workflow
+     */
+    resume(runId: string): Promise<void>;
+    /**
+     * Retry a failed run
+     */
+    retry(runId: string): Promise<string>;
+    /**
+     * Replay a workflow from a specific node
+     */
+    replay<S extends WorkflowState>(workflow: Workflow<S>, runId: string, fromNode: string): Promise<WorkflowResult<S>>;
+    /**
+     * Get count of active runs
+     */
+    getActiveCount(): Promise<number>;
+    /**
+     * Subscribe to run state changes
+     */
+    onRunStateChange(callback: (run: WorkflowRun) => void): () => void;
+    /**
+     * Cleanup old runs
+     */
+    cleanup(olderThan: number): Promise<number>;
+    /**
+     * Dispose the manager
+     */
+    dispose(): void;
+    private handleRunReady;
+    private updateRunNodes;
+    private notifyStateChange;
+}
+/**
+ * Create a workflow manager
+ */
+export declare function createWorkflowManager(config: WorkflowManagerConfig): DefaultWorkflowManager;
+//# sourceMappingURL=workflow-manager.d.ts.map

package/dist/manager/workflow-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-manager.d.ts","sourceRoot":"","sources":["../../src/manager/workflow-manager.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAGH,OAAO,KAAK,EACV,QAAQ,EACR,aAAa,EACb,cAAc,EACd,WAAW,EACX,kBAAkB,EAClB,gBAAgB,EAChB,eAAe,IAAI,gBAAgB,EACnC,eAAe,EACf,wBAAwB,EACxB,QAAQ,EACR,eAAe,EAChB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAKpD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,SAAS,CAAC;IACrB,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,gBAAgB,CAAC,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,IAAI,CAAC;CAC/C;AAED;;GAEG;AACH,qBAAa,sBAAuB,YAAW,gBAAgB;IAC7D,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,QAAQ,CAAW;IAC3B,OAAO,CAAC,eAAe,CAAC,CAAkB;IAC1C,OAAO,CAAC,SAAS,CAAe;IAChC,OAAO,CAAC,QAAQ,CAAmB;IACnC,OAAO,CAAC,SAAS,CAA8C;IAC/D,OAAO,CAAC,UAAU,CAA4C;IAC9D,OAAO,CAAC,oBAAoB,CAAyC;gBAEzD,MAAM,EAAE,qBAAqB;IAkBzC;;OAEG;IACH,KAAK,IAAI,IAAI;IAIb;;OAEG;IACH,IAAI,IAAI,IAAI;IAIZ;;OAEG;IACH,gBAAgB,CAAC,CAAC,SAAS,aAAa,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,GAAG,IAAI;IAItE;;OAEG;IACG,QAAQ,CAAC,CAAC,SAAS,aAAa,EACpC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,EACrB,OAAO,CAAC,EAAE,eAAe,GACxB,OAAO,CAAC,MAAM,CAAC;IAKlB;;OAEG;IACG,OAAO,CAAC,CAAC,SAAS,aAAa,EACnC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,EACrB,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,EAClB,OAAO,CAAC,EAAE,wBAAwB,GACjC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;IAsG7B;;OAEG;IACG,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA4B3D;;OAEG;IACG,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAI3D;;OAEG;IACG,QAAQ,CAAC,OAAO,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAIpE;;OAEG;IACG,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAIhE;;OAEG;IACG,KAAK,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBzC;;OAEG;IACG,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiB1C;;OAEG;IACG,KAAK,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IA8C3C;;OAEG;IACG,MAAM,CAAC,CAAC,SAAS,aAAa,EAClC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,EACrB,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;IAuD7B;;OAEG;IACG,cAAc,IAAI,OAAO,CAAC,MAAM,CAAC;IAMvC;;OAEG;IACH,gBAAgB,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,IAAI,GAAG,MAAM,IAAI;IAOlE;;OAEG;IACG,OAAO,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAIjD;;OAEG;IACH,OAAO,IAAI,IAAI;YAQD,cAAc;YA4Dd,cAAc;IA2B5B,OAAO,CAAC,iBAAiB;CAQ1B;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CACnC,MAAM,EAAE,qBAAqB,GAC5B,sBAAsB,CAExB"}