@cogitator-ai/workflows 0.1.0

package/dist/manager/workflow-manager.js

@@ -0,0 +1,460 @@
+/**
+ * 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 { nanoid } from 'nanoid';
+import { WorkflowExecutor } from '../executor';
+import { createJobScheduler } from './scheduler';
+import { InMemoryRunStore } from './run-store';
+/**
+ * Workflow manager implementation
+ */
+export class DefaultWorkflowManager {
+    cogitator;
+    runStore;
+    checkpointStore;
+    scheduler;
+    executor;
+    workflows = new Map();
+    activeRuns = new Map();
+    stateChangeCallbacks = new Set();
+    constructor(config) {
+        this.cogitator = config.cogitator;
+        this.runStore = config.runStore ?? new InMemoryRunStore();
+        this.checkpointStore = config.checkpointStore;
+        this.executor = new WorkflowExecutor(this.cogitator, this.checkpointStore);
+        this.scheduler = createJobScheduler({
+            runStore: this.runStore,
+            maxConcurrency: config.maxConcurrency,
+            onRunReady: (runId) => this.handleRunReady(runId),
+        });
+        if (config.onRunStateChange) {
+            this.stateChangeCallbacks.add(config.onRunStateChange);
+        }
+    }
+    /**
+     * Start the manager (begins processing scheduled runs)
+     */
+    start() {
+        this.scheduler.start();
+    }
+    /**
+     * Stop the manager
+     */
+    stop() {
+        this.scheduler.stop();
+    }
+    /**
+     * Register a workflow for scheduling
+     */
+    registerWorkflow(workflow) {
+        this.workflows.set(workflow.name, workflow);
+    }
+    /**
+     * Schedule a workflow for later execution
+     */
+    async schedule(workflow, options) {
+        this.registerWorkflow(workflow);
+        return this.scheduler.scheduleRun(workflow, options);
+    }
+    /**
+     * Execute a workflow immediately
+     */
+    async execute(workflow, input, options) {
+        this.registerWorkflow(workflow);
+        const runId = nanoid();
+        const now = Date.now();
+        const run = {
+            id: runId,
+            workflowName: workflow.name,
+            status: 'running',
+            state: input ?? {},
+            input,
+            currentNodes: [],
+            completedNodes: [],
+            failedNodes: [],
+            startedAt: now,
+            priority: options?.priority ?? 0,
+            tags: options?.tags ?? [],
+            triggerId: options?.triggerId,
+            parentRunId: options?.parentRunId,
+            traceId: options?.parentTraceContext?.traceId,
+            metadata: options?.metadata,
+        };
+        await this.runStore.save(run);
+        this.notifyStateChange(run);
+        const abortController = new AbortController();
+        this.activeRuns.set(runId, { abort: () => abortController.abort() });
+        this.scheduler.runStarted(runId);
+        try {
+            const result = await this.executor.execute(workflow, input, {
+                ...options,
+                onNodeStart: (node) => {
+                    this.updateRunNodes(runId, node, 'start');
+                    options?.onNodeStart?.(node);
+                },
+                onNodeComplete: (node, result, duration) => {
+                    this.updateRunNodes(runId, node, 'complete');
+                    options?.onNodeComplete?.(node, result, duration);
+                },
+                onNodeError: (node, error) => {
+                    this.updateRunNodes(runId, node, 'error');
+                    options?.onNodeError?.(node, error);
+                },
+            });
+            if (result.error) {
+                await this.runStore.update(runId, {
+                    status: 'failed',
+                    state: result.state,
+                    completedAt: Date.now(),
+                    checkpointId: result.checkpointId,
+                    error: {
+                        name: result.error.name,
+                        message: result.error.message,
+                        stack: result.error.stack,
+                    },
+                });
+                const updatedRun = await this.runStore.get(runId);
+                if (updatedRun)
+                    this.notifyStateChange(updatedRun);
+                return result;
+            }
+            await this.runStore.update(runId, {
+                status: 'completed',
+                state: result.state,
+                output: result.state,
+                completedAt: Date.now(),
+                checkpointId: result.checkpointId,
+            });
+            const updatedRun = await this.runStore.get(runId);
+            if (updatedRun)
+                this.notifyStateChange(updatedRun);
+            return result;
+        }
+        catch (error) {
+            const err = error instanceof Error ? error : new Error(String(error));
+            await this.runStore.update(runId, {
+                status: 'failed',
+                completedAt: Date.now(),
+                error: {
+                    name: err.name,
+                    message: err.message,
+                    stack: err.stack,
+                },
+            });
+            const updatedRun = await this.runStore.get(runId);
+            if (updatedRun)
+                this.notifyStateChange(updatedRun);
+            throw error;
+        }
+        finally {
+            this.activeRuns.delete(runId);
+            this.scheduler.runCompleted(runId);
+        }
+    }
+    /**
+     * Cancel a run
+     */
+    async cancel(runId, reason) {
+        const run = await this.runStore.get(runId);
+        if (!run)
+            throw new Error(`Run not found: ${runId}`);
+        if (run.status === 'scheduled' || run.status === 'pending') {
+            await this.scheduler.cancelRun(runId, reason);
+            const updatedRun = await this.runStore.get(runId);
+            if (updatedRun)
+                this.notifyStateChange(updatedRun);
+            return;
+        }
+        if (run.status === 'running' || run.status === 'paused') {
+            const active = this.activeRuns.get(runId);
+            if (active) {
+                active.abort();
+            }
+            await this.runStore.update(runId, {
+                status: 'cancelled',
+                completedAt: Date.now(),
+                error: reason ? { name: 'CancelError', message: reason } : undefined,
+            });
+            const updatedRun = await this.runStore.get(runId);
+            if (updatedRun)
+                this.notifyStateChange(updatedRun);
+        }
+    }
+    /**
+     * Get run status
+     */
+    async getStatus(runId) {
+        return this.runStore.get(runId);
+    }
+    /**
+     * List runs with filters
+     */
+    async listRuns(filters) {
+        return this.runStore.list(filters);
+    }
+    /**
+     * Get run statistics
+     */
+    async getStats(workflowName) {
+        return this.runStore.getStats(workflowName);
+    }
+    /**
+     * Pause a running workflow
+     */
+    async pause(runId) {
+        const run = await this.runStore.get(runId);
+        if (!run)
+            throw new Error(`Run not found: ${runId}`);
+        if (run.status !== 'running') {
+            throw new Error(`Cannot pause run in status: ${run.status}`);
+        }
+        await this.runStore.update(runId, {
+            status: 'paused',
+            pausedAt: Date.now(),
+        });
+        const updatedRun = await this.runStore.get(runId);
+        if (updatedRun)
+            this.notifyStateChange(updatedRun);
+    }
+    /**
+     * Resume a paused workflow
+     */
+    async resume(runId) {
+        const run = await this.runStore.get(runId);
+        if (!run)
+            throw new Error(`Run not found: ${runId}`);
+        if (run.status !== 'paused') {
+            throw new Error(`Cannot resume run in status: ${run.status}`);
+        }
+        await this.runStore.update(runId, {
+            status: 'running',
+            pausedAt: undefined,
+        });
+        const updatedRun = await this.runStore.get(runId);
+        if (updatedRun)
+            this.notifyStateChange(updatedRun);
+    }
+    /**
+     * Retry a failed run
+     */
+    async retry(runId) {
+        const run = await this.runStore.get(runId);
+        if (!run)
+            throw new Error(`Run not found: ${runId}`);
+        if (run.status !== 'failed' && run.status !== 'cancelled') {
+            throw new Error(`Cannot retry run in status: ${run.status}`);
+        }
+        const workflow = this.workflows.get(run.workflowName);
+        if (!workflow) {
+            throw new Error(`Workflow not found: ${run.workflowName}`);
+        }
+        const newRunId = nanoid();
+        const now = Date.now();
+        const newRun = {
+            id: newRunId,
+            workflowName: run.workflowName,
+            status: 'pending',
+            state: (run.input ?? {}),
+            input: run.input,
+            currentNodes: [],
+            completedNodes: [],
+            failedNodes: [],
+            priority: run.priority,
+            tags: [...run.tags, 'retry'],
+            triggerId: run.triggerId,
+            parentRunId: runId,
+            metadata: {
+                ...run.metadata,
+                retriedFrom: runId,
+                retriedAt: now,
+            },
+        };
+        await this.runStore.save(newRun);
+        await this.scheduler.scheduleRun(workflow, {
+            priority: run.priority,
+            tags: newRun.tags,
+        });
+        return newRunId;
+    }
+    /**
+     * Replay a workflow from a specific node
+     */
+    async replay(workflow, runId, fromNode) {
+        const run = await this.runStore.get(runId);
+        if (!run)
+            throw new Error(`Run not found: ${runId}`);
+        if (!run.checkpointId) {
+            throw new Error('Run has no checkpoint to replay from');
+        }
+        const newRunId = nanoid();
+        const now = Date.now();
+        const newRun = {
+            id: newRunId,
+            workflowName: workflow.name,
+            status: 'running',
+            state: run.state,
+            input: run.input,
+            currentNodes: [],
+            completedNodes: run.completedNodes.filter((n) => {
+                return workflow.nodes.has(n) && n !== fromNode;
+            }),
+            failedNodes: [],
+            startedAt: now,
+            priority: run.priority,
+            tags: [...run.tags, 'replay'],
+            parentRunId: runId,
+            metadata: {
+                ...run.metadata,
+                replayedFrom: runId,
+                replayFromNode: fromNode,
+                replayedAt: now,
+            },
+        };
+        await this.runStore.save(newRun);
+        this.notifyStateChange(newRun);
+        const result = await this.executor.execute(workflow, run.state, {
+            checkpoint: !!this.checkpointStore,
+        });
+        await this.runStore.update(newRunId, {
+            status: 'completed',
+            state: result.state,
+            output: result.state,
+            completedAt: Date.now(),
+            checkpointId: result.checkpointId,
+        });
+        const updatedRun = await this.runStore.get(newRunId);
+        if (updatedRun)
+            this.notifyStateChange(updatedRun);
+        return result;
+    }
+    /**
+     * Get count of active runs
+     */
+    async getActiveCount() {
+        return this.runStore.count({
+            status: ['running', 'paused', 'waiting'],
+        });
+    }
+    /**
+     * Subscribe to run state changes
+     */
+    onRunStateChange(callback) {
+        this.stateChangeCallbacks.add(callback);
+        return () => {
+            this.stateChangeCallbacks.delete(callback);
+        };
+    }
+    /**
+     * Cleanup old runs
+     */
+    async cleanup(olderThan) {
+        return this.runStore.cleanup(olderThan);
+    }
+    /**
+     * Dispose the manager
+     */
+    dispose() {
+        this.stop();
+        this.activeRuns.clear();
+        this.workflows.clear();
+        this.stateChangeCallbacks.clear();
+        this.scheduler.dispose();
+    }
+    async handleRunReady(runId) {
+        const run = await this.runStore.get(runId);
+        if (!run)
+            return;
+        const workflow = this.workflows.get(run.workflowName);
+        if (!workflow) {
+            await this.runStore.update(runId, {
+                status: 'failed',
+                completedAt: Date.now(),
+                error: {
+                    name: 'WorkflowNotFoundError',
+                    message: `Workflow not found: ${run.workflowName}`,
+                },
+            });
+            return;
+        }
+        await this.runStore.update(runId, {
+            status: 'running',
+            startedAt: Date.now(),
+        });
+        this.scheduler.runStarted(runId);
+        const active = { abort: () => { } };
+        this.activeRuns.set(runId, active);
+        try {
+            const result = await this.executor.execute(workflow, run.input);
+            await this.runStore.update(runId, {
+                status: 'completed',
+                state: result.state,
+                output: result.state,
+                completedAt: Date.now(),
+                checkpointId: result.checkpointId,
+            });
+        }
+        catch (error) {
+            const err = error instanceof Error ? error : new Error(String(error));
+            await this.runStore.update(runId, {
+                status: 'failed',
+                completedAt: Date.now(),
+                error: {
+                    name: err.name,
+                    message: err.message,
+                    stack: err.stack,
+                },
+            });
+        }
+        finally {
+            this.activeRuns.delete(runId);
+            this.scheduler.runCompleted(runId);
+            const updatedRun = await this.runStore.get(runId);
+            if (updatedRun)
+                this.notifyStateChange(updatedRun);
+        }
+    }
+    async updateRunNodes(runId, nodeId, action) {
+        const run = await this.runStore.get(runId);
+        if (!run)
+            return;
+        const updates = {};
+        switch (action) {
+            case 'start':
+                updates.currentNodes = [...run.currentNodes, nodeId];
+                break;
+            case 'complete':
+                updates.currentNodes = run.currentNodes.filter((n) => n !== nodeId);
+                updates.completedNodes = [...run.completedNodes, nodeId];
+                break;
+            case 'error':
+                updates.currentNodes = run.currentNodes.filter((n) => n !== nodeId);
+                updates.failedNodes = [...run.failedNodes, nodeId];
+                break;
+        }
+        await this.runStore.update(runId, updates);
+    }
+    notifyStateChange(run) {
+        for (const callback of this.stateChangeCallbacks) {
+            try {
+                callback(run);
+            }
+            catch {
+            }
+        }
+    }
+}
+/**
+ * Create a workflow manager
+ */
+export function createWorkflowManager(config) {
+    return new DefaultWorkflowManager(config);
+}
+//# sourceMappingURL=workflow-manager.js.map

package/dist/manager/workflow-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-manager.js","sourceRoot":"","sources":["../../src/manager/workflow-manager.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAehC,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAqB,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAc/C;;GAEG;AACH,MAAM,OAAO,sBAAsB;IACzB,SAAS,CAAY;IACrB,QAAQ,CAAW;IACnB,eAAe,CAAmB;IAClC,SAAS,CAAe;IACxB,QAAQ,CAAmB;IAC3B,SAAS,GAAG,IAAI,GAAG,EAAmC,CAAC;IACvD,UAAU,GAAG,IAAI,GAAG,EAAiC,CAAC;IACtD,oBAAoB,GAAG,IAAI,GAAG,EAA8B,CAAC;IAErE,YAAY,MAA6B;QACvC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAClC,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,IAAI,gBAAgB,EAAE,CAAC;QAC1D,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QAE9C,IAAI,CAAC,QAAQ,GAAG,IAAI,gBAAgB,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAE3E,IAAI,CAAC,SAAS,GAAG,kBAAkB,CAAC;YAClC,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,cAAc,EAAE,MAAM,CAAC,cAAc;YACrC,UAAU,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC;SAClD,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAC5B,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,IAAI;QACF,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,gBAAgB,CAA0B,QAAqB;QAC7D,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,QAA8C,CAAC,CAAC;IACpF,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CACZ,QAAqB,EACrB,OAAyB;QAEzB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAChC,OAAO,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,QAAqB,EACrB,KAAkB,EAClB,OAAkC;QAElC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAEhC,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,MAAM,GAAG,GAAgB;YACvB,EAAE,EAAE,KAAK;YACT,YAAY,EAAE,QAAQ,CAAC,IAAI;YAC3B,MAAM,EAAE,SAAS;YACjB,KAAK,EAAE,KAAK,IAAI,EAAE;YAClB,KAAK;YACL,YAAY,EAAE,EAAE;YAChB,cAAc,EAAE,EAAE;YAClB,WAAW,EAAE,EAAE;YACf,SAAS,EAAE,GAAG;YACd,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,CAAC;YAChC,IAAI,EAAE,OAAO,EAAE,IAAI,IAAI,EAAE;YACzB,SAAS,EAAE,OAAO,EAAE,SAAS;YAC7B,WAAW,EAAE,OAAO,EAAE,WAAW;YACjC,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,OAAO;YAC7C,QAAQ,EAAE,OAAO,EAAE,QAAQ;SAC5B,CAAC;QAEF,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAE5B,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACrE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAEjC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,EAAE;gBAC1D,GAAG,OAAO;gBACV,WAAW,EAAE,CAAC,IAAI,EAAE,EAAE;oBACpB,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;oBAC1C,OAAO,EAAE,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC;gBAC/B,CAAC;gBACD,cAAc,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE;oBACzC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;oBAC7C,OAAO,EAAE,cAAc,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;gBACpD,CAAC;gBACD,WAAW,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;oBAC3B,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;oBAC1C,OAAO,EAAE,WAAW,EAAE,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBACtC,CAAC;aACF,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;oBAChC,MAAM,EAAE,QAAQ;oBAChB,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;oBACvB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI;wBACvB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;wBAC7B,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK;qBAC1B;iBACF,CAAC,CAAC;gBAEH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAClD,IAAI,UAAU;oBAAE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBAEnD,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;gBAChC,MAAM,EAAE,WAAW;gBACnB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,MAAM,EAAE,MAAM,CAAC,KAAK;gBACpB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;gBACvB,YAAY,EAAE,MAAM,CAAC,YAAY;aAClC,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAClD,IAAI,UAAU;gBAAE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEnD,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAEtE,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;gBAChC,MAAM,EAAE,QAAQ;gBAChB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;gBACvB,KAAK,EAAE;oBACL,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,OAAO,EAAE,GAAG,CAAC,OAAO;oBACpB,KAAK,EAAE,GAAG,CAAC,KAAK;iBACjB;aACF,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAClD,IAAI,UAAU;gBAAE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEnD,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9B,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,MAAe;QACzC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,GAAG;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QAErD,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC3D,MAAM,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC9C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAClD,IAAI,UAAU;gBAAE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YACnD,OAAO;QACT,CAAC;QAED,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YACxD,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC1C,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,CAAC;YAED,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;gBAChC,MAAM,EAAE,WAAW;gBACnB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;gBACvB,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS;aACrE,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAClD,IAAI,UAAU;gBAAE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,KAAa;QAC3B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,OAA4B;QACzC,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,YAAqB;QAClC,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,KAAa;QACvB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,GAAG;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QAErD,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,+BAA+B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;YAChC,MAAM,EAAE,QAAQ;YAChB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE;SACrB,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAClD,IAAI,UAAU;YAAE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;IACrD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,KAAa;QACxB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,GAAG;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QAErD,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,gCAAgC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;YAChC,MAAM,EAAE,SAAS;YACjB,QAAQ,EAAE,SAAS;SACpB,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAClD,IAAI,UAAU;YAAE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;IACrD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,KAAa;QACvB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,GAAG;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QAErD,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YAC1D,MAAM,IAAI,KAAK,CAAC,+BAA+B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC;QAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,MAAM,MAAM,GAAgB;YAC1B,EAAE,EAAE,QAAQ;YACZ,YAAY,EAAE,GAAG,CAAC,YAAY;YAC9B,MAAM,EAAE,SAAS;YACjB,KAAK,EAAE,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAkB;YACzC,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,YAAY,EAAE,EAAE;YAChB,cAAc,EAAE,EAAE;YAClB,WAAW,EAAE,EAAE;YACf,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,IAAI,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC;YAC5B,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,WAAW,EAAE,KAAK;YAClB,QAAQ,EAAE;gBACR,GAAG,GAAG,CAAC,QAAQ;gBACf,WAAW,EAAE,KAAK;gBAClB,SAAS,EAAE,GAAG;aACf;SACF,CAAC;QAEF,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEjC,MAAM,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,QAAQ,EAAE;YACzC,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,IAAI,EAAE,MAAM,CAAC,IAAI;SAClB,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CACV,QAAqB,EACrB,KAAa,EACb,QAAgB;QAEhB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,GAAG;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QAErD,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC;QAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,MAAM,MAAM,GAAgB;YAC1B,EAAE,EAAE,QAAQ;YACZ,YAAY,EAAE,QAAQ,CAAC,IAAI;YAC3B,MAAM,EAAE,SAAS;YACjB,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,YAAY,EAAE,EAAE;YAChB,cAAc,EAAE,GAAG,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC9C,OAAO,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,QAAQ,CAAC;YACjD,CAAC,CAAC;YACF,WAAW,EAAE,EAAE;YACf,SAAS,EAAE,GAAG;YACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,IAAI,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC;YAC7B,WAAW,EAAE,KAAK;YAClB,QAAQ,EAAE;gBACR,GAAG,GAAG,CAAC,QAAQ;gBACf,YAAY,EAAE,KAAK;gBACnB,cAAc,EAAE,QAAQ;gBACxB,UAAU,EAAE,GAAG;aAChB;SACF,CAAC;QAEF,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAE/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,KAAmB,EAAE;YAC5E,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe;SACnC,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE;YACnC,MAAM,EAAE,WAAW;YACnB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,MAAM,EAAE,MAAM,CAAC,KAAK;YACpB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;YACvB,YAAY,EAAE,MAAM,CAAC,YAAY;SAClC,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACrD,IAAI,UAAU;YAAE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAEnD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc;QAClB,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YACzB,MAAM,EAAE,CAAC,SAAS,EAAE,QAAQ,EAAE,SAAS,CAAC;SACzC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,QAAoC;QACnD,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxC,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7C,CAAC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,SAAiB;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACxB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACvB,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC;QAClC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,KAAa;QACxC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,GAAG;YAAE,OAAO;QAEjB,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;gBAChC,MAAM,EAAE,QAAQ;gBAChB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;gBACvB,KAAK,EAAE;oBACL,IAAI,EAAE,uBAAuB;oBAC7B,OAAO,EAAE,uBAAuB,GAAG,CAAC,YAAY,EAAE;iBACnD;aACF,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;YAChC,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACjC,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC;QACnC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAEnC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CACxC,QAAQ,EACR,GAAG,CAAC,KAA+B,CACpC,CAAC;YAEF,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;gBAChC,MAAM,EAAE,WAAW;gBACnB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,MAAM,EAAE,MAAM,CAAC,KAAK;gBACpB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;gBACvB,YAAY,EAAE,MAAM,CAAC,YAAY;aAClC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAEtE,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE;gBAChC,MAAM,EAAE,QAAQ;gBAChB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;gBACvB,KAAK,EAAE;oBACL,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,OAAO,EAAE,GAAG,CAAC,OAAO;oBACpB,KAAK,EAAE,GAAG,CAAC,KAAK;iBACjB;aACF,CAAC,CAAC;QACL,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9B,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;YAEnC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAClD,IAAI,UAAU;gBAAE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,KAAa,EACb,MAAc,EACd,MAAsC;QAEtC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,GAAG;YAAE,OAAO;QAEjB,MAAM,OAAO,GAAyB,EAAE,CAAC;QAEzC,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,OAAO;gBACV,OAAO,CAAC,YAAY,GAAG,CAAC,GAAG,GAAG,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;gBACrD,MAAM;YACR,KAAK,UAAU;gBACb,OAAO,CAAC,YAAY,GAAG,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,MAAM,CAAC,CAAC;gBACpE,OAAO,CAAC,cAAc,GAAG,CAAC,GAAG,GAAG,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;gBACzD,MAAM;YACR,KAAK,OAAO;gBACV,OAAO,CAAC,YAAY,GAAG,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,MAAM,CAAC,CAAC;gBACpE,OAAO,CAAC,WAAW,GAAG,CAAC,GAAG,GAAG,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;gBACnD,MAAM;QACV,CAAC;QAED,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC;IAEO,iBAAiB,CAAC,GAAgB;QACxC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YACjD,IAAI,CAAC;gBACH,QAAQ,CAAC,GAAG,CAAC,CAAC;YAChB,CAAC;YAAC,MAAM,CAAC;YACT,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CACnC,MAA6B;IAE7B,OAAO,IAAI,sBAAsB,CAAC,MAAM,CAAC,CAAC;AAC5C,CAAC"}

package/dist/nodes/agent.d.ts

@@ -0,0 +1,24 @@
+/**
+ * AgentNode - Run a Cogitator agent as a workflow node
+ */
+import type { WorkflowNode, WorkflowState } from '@cogitator-ai/types';
+import type { Agent, RunOptions, RunResult } from '@cogitator-ai/core';
+export interface AgentNodeOptions<S = WorkflowState> {
+    /**
+     * Map agent result to state updates
+     */
+    stateMapper?: (result: RunResult) => Partial<S>;
+    /**
+     * Map current state to agent input
+     */
+    inputMapper?: (state: S, input?: unknown) => string;
+    /**
+     * Additional run options for the agent
+     */
+    runOptions?: Partial<RunOptions>;
+}
+/**
+ * Create a workflow node that runs a Cogitator agent
+ */
+export declare function agentNode<S extends WorkflowState = WorkflowState>(agent: Agent, options?: AgentNodeOptions<S>): WorkflowNode<S>;
+//# sourceMappingURL=agent.d.ts.map

package/dist/nodes/agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/nodes/agent.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAc,MAAM,qBAAqB,CAAC;AACnF,OAAO,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAGvE,MAAM,WAAW,gBAAgB,CAAC,CAAC,GAAG,aAAa;IACjD;;OAEG;IACH,WAAW,CAAC,EAAE,CAAC,MAAM,EAAE,SAAS,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;IAEhD;;OAEG;IACH,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,CAAC,EAAE,OAAO,KAAK,MAAM,CAAC;IAEpD;;OAEG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;CAClC;AAED;;GAEG;AACH,wBAAgB,SAAS,CAAC,CAAC,SAAS,aAAa,GAAG,aAAa,EAC/D,KAAK,EAAE,KAAK,EACZ,OAAO,CAAC,EAAE,gBAAgB,CAAC,CAAC,CAAC,GAC5B,YAAY,CAAC,CAAC,CAAC,CA8BjB"}

package/dist/nodes/agent.js

@@ -0,0 +1,37 @@
+/**
+ * AgentNode - Run a Cogitator agent as a workflow node
+ */
+/**
+ * Create a workflow node that runs a Cogitator agent
+ */
+export function agentNode(agent, options) {
+    return {
+        name: agent.name,
+        fn: async (ctx) => {
+            const extCtx = ctx;
+            let input;
+            if (options?.inputMapper) {
+                input = options.inputMapper(ctx.state, ctx.input);
+            }
+            else if (typeof ctx.input === 'string') {
+                input = ctx.input;
+            }
+            else if (ctx.input !== undefined) {
+                input = JSON.stringify(ctx.input);
+            }
+            else {
+                input = JSON.stringify(ctx.state);
+            }
+            const result = await extCtx.cogitator.run(agent, {
+                input,
+                ...options?.runOptions,
+            });
+            const stateUpdate = options?.stateMapper?.(result);
+            return {
+                state: stateUpdate,
+                output: result.output,
+            };
+        },
+    };
+}
+//# sourceMappingURL=agent.js.map

package/dist/nodes/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/nodes/agent.ts"],"names":[],"mappings":"AAAA;;GAEG;AAuBH;;GAEG;AACH,MAAM,UAAU,SAAS,CACvB,KAAY,EACZ,OAA6B;IAE7B,OAAO;QACL,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,EAAE,EAAE,KAAK,EAAE,GAAG,EAA0B,EAAE;YACxC,MAAM,MAAM,GAAG,GAA6B,CAAC;YAE7C,IAAI,KAAa,CAAC;YAClB,IAAI,OAAO,EAAE,WAAW,EAAE,CAAC;gBACzB,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;YACpD,CAAC;iBAAM,IAAI,OAAO,GAAG,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACzC,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;YACpB,CAAC;iBAAM,IAAI,GAAG,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBACnC,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACpC,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE;gBAC/C,KAAK;gBACL,GAAG,OAAO,EAAE,UAAU;aACvB,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,OAAO,EAAE,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC;YAEnD,OAAO;gBACL,KAAK,EAAE,WAAW;gBAClB,MAAM,EAAE,MAAM,CAAC,MAAM;aACtB,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/nodes/base.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Base node interfaces and types
+ */
+import type { WorkflowNode, WorkflowState, NodeConfig, NodeContext, NodeResult } from '@cogitator-ai/types';
+export type { WorkflowNode, NodeConfig, NodeContext, NodeResult };
+/**
+ * Extended context with Cogitator for agent nodes
+ */
+export interface ExtendedNodeContext<S = WorkflowState> extends NodeContext<S> {
+    cogitator: import('@cogitator-ai/core').Cogitator;
+}
+//# sourceMappingURL=base.d.ts.map

package/dist/nodes/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/nodes/base.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EACV,YAAY,EACZ,aAAa,EACb,UAAU,EACV,WAAW,EACX,UAAU,EACX,MAAM,qBAAqB,CAAC;AAE7B,YAAY,EAAE,YAAY,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;AAElE;;GAEG;AACH,MAAM,WAAW,mBAAmB,CAAC,CAAC,GAAG,aAAa,CAAE,SAAQ,WAAW,CAAC,CAAC,CAAC;IAC5E,SAAS,EAAE,OAAO,oBAAoB,EAAE,SAAS,CAAC;CACnD"}

package/dist/nodes/base.js

@@ -0,0 +1,5 @@
+/**
+ * Base node interfaces and types
+ */
+export {};
+//# sourceMappingURL=base.js.map

package/dist/nodes/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/nodes/base.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/dist/nodes/function.d.ts

@@ -0,0 +1,27 @@
+/**
+ * FunctionNode - Run a custom function as a workflow node
+ */
+import type { WorkflowNode, WorkflowState, NodeResult, NodeContext } from '@cogitator-ai/types';
+/**
+ * Simple function that takes state and returns output
+ */
+export type SimpleNodeFn<S, O = unknown> = (state: S, input?: unknown) => Promise<O>;
+/**
+ * Full function that receives context and returns result
+ */
+export type FullNodeFn<S> = (ctx: NodeContext<S>) => Promise<NodeResult<S>>;
+export interface FunctionNodeOptions<S = WorkflowState> {
+    /**
+     * Map function output to state updates
+     */
+    stateMapper?: (output: unknown) => Partial<S>;
+}
+/**
+ * Create a workflow node from a simple async function
+ */
+export declare function functionNode<S extends WorkflowState = WorkflowState, O = unknown>(name: string, fn: SimpleNodeFn<S, O>, options?: FunctionNodeOptions<S>): WorkflowNode<S>;
+/**
+ * Create a workflow node from a full node function
+ */
+export declare function customNode<S extends WorkflowState = WorkflowState>(name: string, fn: FullNodeFn<S>): WorkflowNode<S>;
+//# sourceMappingURL=function.d.ts.map

package/dist/nodes/function.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"function.d.ts","sourceRoot":"","sources":["../../src/nodes/function.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EACV,YAAY,EACZ,aAAa,EACb,UAAU,EACV,WAAW,EACZ,MAAM,qBAAqB,CAAC;AAE7B;;GAEG;AACH,MAAM,MAAM,YAAY,CAAC,CAAC,EAAE,CAAC,GAAG,OAAO,IAAI,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;AAErF;;GAEG;AACH,MAAM,MAAM,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;AAE5E,MAAM,WAAW,mBAAmB,CAAC,CAAC,GAAG,aAAa;IACpD;;OAEG;IACH,WAAW,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;CAC/C;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,CAAC,SAAS,aAAa,GAAG,aAAa,EAAE,CAAC,GAAG,OAAO,EAC/E,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,YAAY,CAAC,CAAC,EAAE,CAAC,CAAC,EACtB,OAAO,CAAC,EAAE,mBAAmB,CAAC,CAAC,CAAC,GAC/B,YAAY,CAAC,CAAC,CAAC,CAajB;AAED;;GAEG;AACH,wBAAgB,UAAU,CAAC,CAAC,SAAS,aAAa,GAAG,aAAa,EAChE,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,UAAU,CAAC,CAAC,CAAC,GAChB,YAAY,CAAC,CAAC,CAAC,CAKjB"}

package/dist/nodes/function.js

@@ -0,0 +1,29 @@
+/**
+ * FunctionNode - Run a custom function as a workflow node
+ */
+/**
+ * Create a workflow node from a simple async function
+ */
+export function functionNode(name, fn, options) {
+    return {
+        name,
+        fn: async (ctx) => {
+            const output = await fn(ctx.state, ctx.input);
+            const stateUpdate = options?.stateMapper?.(output);
+            return {
+                state: stateUpdate,
+                output,
+            };
+        },
+    };
+}
+/**
+ * Create a workflow node from a full node function
+ */
+export function customNode(name, fn) {
+    return {
+        name,
+        fn,
+    };
+}
+//# sourceMappingURL=function.js.map

package/dist/nodes/function.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"function.js","sourceRoot":"","sources":["../../src/nodes/function.ts"],"names":[],"mappings":"AAAA;;GAEG;AA0BH;;GAEG;AACH,MAAM,UAAU,YAAY,CAC1B,IAAY,EACZ,EAAsB,EACtB,OAAgC;IAEhC,OAAO;QACL,IAAI;QACJ,EAAE,EAAE,KAAK,EAAE,GAAG,EAA0B,EAAE;YACxC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,WAAW,GAAG,OAAO,EAAE,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC;YAEnD,OAAO;gBACL,KAAK,EAAE,WAAW;gBAClB,MAAM;aACP,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU,CACxB,IAAY,EACZ,EAAiB;IAEjB,OAAO;QACL,IAAI;QACJ,EAAE;KACH,CAAC;AACJ,CAAC"}

package/dist/nodes/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Pre-built workflow nodes
+ */
+export * from './base';
+export * from './agent';
+export * from './tool';
+export * from './function';
+//# sourceMappingURL=index.d.ts.map

package/dist/nodes/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/nodes/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,cAAc,QAAQ,CAAC;AACvB,cAAc,SAAS,CAAC;AACxB,cAAc,QAAQ,CAAC;AACvB,cAAc,YAAY,CAAC"}

package/dist/nodes/index.js

@@ -0,0 +1,8 @@
+/**
+ * Pre-built workflow nodes
+ */
+export * from './base';
+export * from './agent';
+export * from './tool';
+export * from './function';
+//# sourceMappingURL=index.js.map