@oni.bot/core 1.0.2 → 1.1.0

package/dist/pregel/checkpointing.js

@@ -0,0 +1,60 @@
+// ============================================================
+// src/pregel/checkpointing.ts — Checkpoint persistence helpers
+// ============================================================
+import { applyUpdate } from "./state-helpers.js";
+export async function saveCheckpoint(ctx, threadId, step, state, nextNodes, pendingSends, agentId, metadata, pendingWrites) {
+    const cp = (ctx._perInvocationCheckpointer.get(threadId) ?? ctx.checkpointer);
+    if (!cp)
+        return;
+    const cpSpan = ctx.tracer.startCheckpointSpan("put", { threadId });
+    try {
+        await cp.put({
+            threadId, step, state, agentId, metadata, pendingWrites,
+            nextNodes: nextNodes.map(String),
+            pendingSends: pendingSends,
+            timestamp: Date.now(),
+        });
+    }
+    finally {
+        ctx.tracer.endSpan(cpSpan);
+    }
+}
+export async function getState(checkpointer, threadId) {
+    if (!checkpointer)
+        return null;
+    return (await checkpointer.get(threadId))?.state ?? null;
+}
+export async function updateState(checkpointer, channels, threadId, update) {
+    if (!checkpointer)
+        return;
+    const cp = await checkpointer.get(threadId);
+    if (!cp)
+        return;
+    await checkpointer.put({ ...cp, state: applyUpdate(channels, cp.state, update), timestamp: Date.now() });
+}
+export async function getStateAt(checkpointer, threadId, step) {
+    if (!checkpointer)
+        return null;
+    const history = await checkpointer.list(threadId);
+    return history.find((c) => c.step === step)?.state ?? null;
+}
+export async function getHistory(checkpointer, threadId) {
+    if (!checkpointer)
+        return [];
+    return checkpointer.list(threadId);
+}
+export async function forkFrom(checkpointer, threadId, step, newThreadId) {
+    if (!checkpointer)
+        return;
+    const cp = checkpointer;
+    if (typeof cp.fork === "function") {
+        await cp.fork(threadId, step, newThreadId);
+    }
+    else {
+        const history = await checkpointer.list(threadId);
+        for (const c of history.filter((x) => x.step <= step)) {
+            await checkpointer.put({ ...c, threadId: newThreadId });
+        }
+    }
+}
+//# sourceMappingURL=checkpointing.js.map

package/dist/pregel/checkpointing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"checkpointing.js","sourceRoot":"","sources":["../../src/pregel/checkpointing.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,+DAA+D;AAC/D,+DAA+D;AAK/D,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,GAAqB,EACrB,QAAgB,EAChB,IAAY,EACZ,KAAQ,EACR,SAAqB,EACrB,YAA2B,EAC3B,OAAgB,EAChB,QAAkC,EAClC,aAA0E;IAE1E,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,0BAA0B,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,YAAY,CAA4B,CAAC;IACzG,IAAI,CAAC,EAAE;QAAE,OAAO;IAChB,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,mBAAmB,CAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;IACnE,IAAI,CAAC;QACH,MAAM,EAAE,CAAC,GAAG,CAAC;YACX,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,aAAa;YACvD,SAAS,EAAK,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC;YACnC,YAAY,EAAE,YAAY;YAC1B,SAAS,EAAK,IAAI,CAAC,GAAG,EAAE;SACzB,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC7B,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,YAAuC,EACvC,QAAgB;IAEhB,IAAI,CAAC,YAAY;QAAE,OAAO,IAAI,CAAC;IAC/B,OAAO,CAAC,MAAM,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,IAAI,IAAI,CAAC;AAC3D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,YAAuC,EACvC,QAA0B,EAC1B,QAAgB,EAChB,MAAkB;IAElB,IAAI,CAAC,YAAY;QAAE,OAAO;IAC1B,MAAM,EAAE,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC5C,IAAI,CAAC,EAAE;QAAE,OAAO;IAChB,MAAM,YAAY,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;AAC3G,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,YAAuC,EACvC,QAAgB,EAChB,IAAY;IAEZ,IAAI,CAAC,YAAY;QAAE,OAAO,IAAI,CAAC;IAC/B,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAClD,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,KAAK,IAAI,IAAI,CAAC;AAC7D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,YAAuC,EACvC,QAAgB;IAEhB,IAAI,CAAC,YAAY;QAAE,OAAO,EAAE,CAAC;IAC7B,OAAO,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AACrC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,YAAuC,EACvC,QAAgB,EAChB,IAAY,EACZ,WAAmB;IAEnB,IAAI,CAAC,YAAY;QAAE,OAAO;IAC1B,MAAM,EAAE,GAAG,YAAqC,CAAC;IACjD,IAAI,OAAQ,EAA4B,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAC7D,MAAO,EAA4B,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;IACxE,CAAC;SAAM,CAAC;QACN,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClD,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;YACtD,MAAM,YAAY,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/pregel/execution.d.ts

@@ -0,0 +1,7 @@
+import { type NodeDefinition, type ONIConfig, type NodeReturn } from "../types.js";
+import { CircuitBreaker } from "../circuit-breaker.js";
+import { type StreamWriter } from "../context.js";
+import type { PregelContext } from "./types.js";
+export declare function getCircuitBreaker<S extends Record<string, unknown>>(nodeDef: NodeDefinition<S>, circuitBreakers: Map<string, CircuitBreaker>): CircuitBreaker | null;
+export declare function executeNode<S extends Record<string, unknown>>(ctx: PregelContext<S>, nodeDef: NodeDefinition<S>, state: S, config?: ONIConfig, resumeValue?: unknown, hasResume?: boolean, writer?: StreamWriter | null, step?: number, recursionLimit?: number): Promise<NodeReturn<S>>;
+//# sourceMappingURL=execution.d.ts.map

package/dist/pregel/execution.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"execution.d.ts","sourceRoot":"","sources":["../../src/pregel/execution.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,KAAK,cAAc,EAAE,KAAK,SAAS,EAAoB,KAAK,UAAU,EACvE,MAAM,aAAa,CAAC;AAErB,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD,OAAO,EAAoC,KAAK,YAAY,EAAE,MAAM,eAAe,CAAC;AAOpF,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAMhD,wBAAgB,iBAAiB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACjE,OAAO,EAAE,cAAc,CAAC,CAAC,CAAC,EAC1B,eAAe,EAAE,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,GAC3C,cAAc,GAAG,IAAI,CAWvB;AAED,wBAAsB,WAAW,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACjE,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,EACrB,OAAO,EAAE,cAAc,CAAC,CAAC,CAAC,EAC1B,KAAK,EAAE,CAAC,EACR,MAAM,CAAC,EAAE,SAAS,EAClB,WAAW,CAAC,EAAE,OAAO,EACrB,SAAS,CAAC,EAAE,OAAO,EACnB,MAAM,CAAC,EAAE,YAAY,GAAG,IAAI,EAC5B,IAAI,CAAC,EAAE,MAAM,EACb,cAAc,CAAC,EAAE,MAAM,GACtB,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAoJxB"}

package/dist/pregel/execution.js

@@ -0,0 +1,178 @@
+// ============================================================
+// src/pregel/execution.ts — Node execution with retry/timeout/CB
+// ============================================================
+import { ONIError, NodeTimeoutError, NodeExecutionError, CircuitBreakerOpenError } from "../errors.js";
+import { CircuitBreaker } from "../circuit-breaker.js";
+import { withRetry } from "../retry.js";
+import { _runWithContext } from "../context.js";
+import { NodeInterruptSignal, _installInterruptContext, _clearInterruptContext, } from "../hitl/index.js";
+import { runFilters } from "../guardrails/filters.js";
+const DEFAULT_RECURSION_LIMIT = 25;
+/** Maximum entries in the node result cache — oldest entries evicted FIFO when full. */
+const NODE_CACHE_MAX_SIZE = 256;
+export function getCircuitBreaker(nodeDef, circuitBreakers) {
+    if (!nodeDef.circuitBreaker)
+        return null;
+    let cb = circuitBreakers.get(nodeDef.name);
+    if (!cb) {
+        cb = new CircuitBreaker({
+            threshold: nodeDef.circuitBreaker.threshold,
+            resetAfter: nodeDef.circuitBreaker.resetAfter,
+        }, nodeDef.name);
+        circuitBreakers.set(nodeDef.name, cb);
+    }
+    return cb;
+}
+export async function executeNode(ctx, nodeDef, state, config, resumeValue, hasResume, writer, step, recursionLimit) {
+    // Check cache (compute key once, reuse for both lookup and store)
+    let cacheKey;
+    if (nodeDef.cache) {
+        const policy = typeof nodeDef.cache === "object" ? nodeDef.cache : {};
+        const keyFn = policy.key ?? ((s) => JSON.stringify(s));
+        cacheKey = `${nodeDef.name}::${keyFn(state)}`;
+        const cached = ctx.nodeCache.get(cacheKey);
+        if (cached) {
+            const ttl = policy.ttl ?? Infinity;
+            if (Date.now() - cached.timestamp < ttl) {
+                return cached.result;
+            }
+            ctx.nodeCache.delete(cacheKey);
+        }
+    }
+    const _tid = config?.threadId ?? "unknown";
+    const runCtx = {
+        config: config ?? {},
+        store: ctx.store,
+        writer: writer ?? null,
+        state: state,
+        parentGraph: null,
+        parentUpdates: [],
+        step: step ?? 0,
+        recursionLimit: recursionLimit ?? DEFAULT_RECURSION_LIMIT,
+        toolPermissions: ctx.toolPermissions,
+        _recordUsage: (agentName, modelId, usage) => {
+            if (!ctx.budgetTracker)
+                return;
+            const entries = ctx.budgetTracker.record(agentName, modelId, usage);
+            for (const e of entries)
+                ctx.auditLog?.record(_tid, e);
+        },
+        _emitEvent: (event) => ctx.eventBus.emit(event),
+        _auditRecord: (entry) => ctx.auditLog?.record(_tid, entry),
+    };
+    return _runWithContext(runCtx, async () => {
+        _installInterruptContext({
+            nodeName: nodeDef.name,
+            resumeValues: hasResume ? [resumeValue] : [],
+        });
+        try {
+            // Content filter — input direction
+            if (ctx.contentFilters.length > 0) {
+                const inputStr = JSON.stringify(state);
+                const inputCheck = runFilters(ctx.contentFilters, inputStr, "input");
+                if (!inputCheck.passed) {
+                    const threadId = config?.threadId ?? "unknown";
+                    ctx.eventBus.emit({ type: "filter.blocked", filter: inputCheck.blockedBy, agent: nodeDef.name, direction: "input", reason: inputCheck.reason, timestamp: Date.now() });
+                    ctx.auditLog?.record(threadId, { timestamp: Date.now(), agent: nodeDef.name, action: "filter.blocked", data: { filter: inputCheck.blockedBy, direction: "input", reason: inputCheck.reason } });
+                    throw new Error(`Content blocked by filter "${inputCheck.blockedBy}" on input to node "${nodeDef.name}": ${inputCheck.reason}`);
+                }
+                // Apply redaction if content was rewritten by a redacting filter
+                if (inputCheck.content !== inputStr) {
+                    try {
+                        state = JSON.parse(inputCheck.content);
+                    }
+                    catch { /* leave state unchanged on parse failure */ }
+                }
+            }
+            const run = () => Promise.resolve(nodeDef.fn(state, config));
+            // Core execute call: retry-aware
+            const executeCall = async () => {
+                if (nodeDef.retry)
+                    return withRetry(run, nodeDef.name, nodeDef.retry);
+                return run();
+            };
+            // Wrap with timeout if configured (per-node > global default > none)
+            const timeoutMs = nodeDef.timeout ?? ctx.defaults?.nodeTimeout;
+            const executeWithTimeout = async () => {
+                if (timeoutMs != null && timeoutMs > 0) {
+                    const ac = new AbortController();
+                    const timer = setTimeout(() => ac.abort(), timeoutMs);
+                    try {
+                        return await Promise.race([
+                            executeCall(),
+                            new Promise((_, reject) => {
+                                ac.signal.addEventListener("abort", () => {
+                                    reject(new NodeTimeoutError(nodeDef.name, timeoutMs));
+                                });
+                            }),
+                        ]);
+                    }
+                    finally {
+                        clearTimeout(timer);
+                    }
+                }
+                return executeCall();
+            };
+            // Wrap with circuit breaker if configured
+            const cb = getCircuitBreaker(nodeDef, ctx.circuitBreakers);
+            let result;
+            try {
+                if (cb) {
+                    result = await cb.execute(executeWithTimeout);
+                }
+                else {
+                    result = await executeWithTimeout();
+                }
+            }
+            catch (err) {
+                // Pass through interrupt signals (thrown by interrupt() inside nodes)
+                if (err instanceof NodeInterruptSignal)
+                    throw err;
+                // Circuit breaker open — invoke user fallback with real state + error
+                if (err instanceof CircuitBreakerOpenError && nodeDef.circuitBreaker?.fallback) {
+                    result = nodeDef.circuitBreaker.fallback(state, err);
+                }
+                else {
+                    // Pass through structured ONI errors (NodeExecutionError from retry, NodeTimeoutError, etc.)
+                    if (err instanceof ONIError)
+                        throw err;
+                    // Wrap raw errors and non-Error throws in NodeExecutionError
+                    const cause = err instanceof Error ? err : new Error(String(err));
+                    throw new NodeExecutionError(nodeDef.name, cause);
+                }
+            }
+            // Content filter — output direction
+            if (ctx.contentFilters.length > 0 && result != null) {
+                const outputStr = JSON.stringify(result);
+                const outputCheck = runFilters(ctx.contentFilters, outputStr, "output");
+                if (!outputCheck.passed) {
+                    const threadId = config?.threadId ?? "unknown";
+                    ctx.eventBus.emit({ type: "filter.blocked", filter: outputCheck.blockedBy, agent: nodeDef.name, direction: "output", reason: outputCheck.reason, timestamp: Date.now() });
+                    ctx.auditLog?.record(threadId, { timestamp: Date.now(), agent: nodeDef.name, action: "filter.blocked", data: { filter: outputCheck.blockedBy, direction: "output", reason: outputCheck.reason } });
+                    throw new Error(`Content blocked by filter "${outputCheck.blockedBy}" on output of node "${nodeDef.name}": ${outputCheck.reason}`);
+                }
+                // Apply redaction if content was rewritten by a redacting filter
+                if (outputCheck.content !== outputStr) {
+                    try {
+                        result = JSON.parse(outputCheck.content);
+                    }
+                    catch { /* leave result unchanged on parse failure */ }
+                }
+            }
+            // Store in cache (reuse key computed above); evict oldest entry when full
+            if (nodeDef.cache && cacheKey) {
+                if (ctx.nodeCache.size >= NODE_CACHE_MAX_SIZE) {
+                    const oldest = ctx.nodeCache.keys().next().value;
+                    if (oldest !== undefined)
+                        ctx.nodeCache.delete(oldest);
+                }
+                ctx.nodeCache.set(cacheKey, { result, timestamp: Date.now() });
+            }
+            return result;
+        }
+        finally {
+            _clearInterruptContext();
+        }
+    });
+}
+//# sourceMappingURL=execution.js.map

package/dist/pregel/execution.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"execution.js","sourceRoot":"","sources":["../../src/pregel/execution.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,iEAAiE;AACjE,+DAA+D;AAK/D,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,MAAM,cAAc,CAAC;AACvG,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,eAAe,EAAsC,MAAM,eAAe,CAAC;AACpF,OAAO,EACL,mBAAmB,EAAE,wBAAwB,EAAE,sBAAsB,GACtE,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAKtD,MAAM,uBAAuB,GAAG,EAAE,CAAC;AACnC,wFAAwF;AACxF,MAAM,mBAAmB,GAAG,GAAG,CAAC;AAEhC,MAAM,UAAU,iBAAiB,CAC/B,OAA0B,EAC1B,eAA4C;IAE5C,IAAI,CAAC,OAAO,CAAC,cAAc;QAAE,OAAO,IAAI,CAAC;IACzC,IAAI,EAAE,GAAG,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3C,IAAI,CAAC,EAAE,EAAE,CAAC;QACR,EAAE,GAAG,IAAI,cAAc,CAAC;YACtB,SAAS,EAAE,OAAO,CAAC,cAAc,CAAC,SAAS;YAC3C,UAAU,EAAE,OAAO,CAAC,cAAc,CAAC,UAAU;SAC9C,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;QACjB,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,GAAqB,EACrB,OAA0B,EAC1B,KAAQ,EACR,MAAkB,EAClB,WAAqB,EACrB,SAAmB,EACnB,MAA4B,EAC5B,IAAa,EACb,cAAuB;IAEvB,kEAAkE;IAClE,IAAI,QAA4B,CAAC;IACjC,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QAClB,MAAM,MAAM,GAAgB,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QACnF,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QAChE,QAAQ,GAAG,GAAG,OAAO,CAAC,IAAI,KAAK,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;QAC9C,MAAM,MAAM,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC3C,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,IAAI,QAAQ,CAAC;YACnC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;gBACxC,OAAO,MAAM,CAAC,MAAM,CAAC;YACvB,CAAC;YACD,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,EAAE,QAAQ,IAAI,SAAS,CAAC;IAC3C,MAAM,MAAM,GAAe;QACzB,MAAM,EAAW,MAAM,IAAI,EAAE;QAC7B,KAAK,EAAY,GAAG,CAAC,KAAK;QAC1B,MAAM,EAAW,MAAM,IAAI,IAAI;QAC/B,KAAK,EAAY,KAAK;QACtB,WAAW,EAAM,IAAI;QACrB,aAAa,EAAI,EAAE;QACnB,IAAI,EAAa,IAAI,IAAI,CAAC;QAC1B,cAAc,EAAG,cAAc,IAAI,uBAAuB;QAC1D,eAAe,EAAE,GAAG,CAAC,eAAe;QACpC,YAAY,EAAE,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;YAC1C,IAAI,CAAC,GAAG,CAAC,aAAa;gBAAE,OAAO;YAC/B,MAAM,OAAO,GAAG,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;YACpE,KAAK,MAAM,CAAC,IAAI,OAAO;gBAAE,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACzD,CAAC;QACD,UAAU,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAuB,CAAC;QACjE,YAAY,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,EAAE,KAAmB,CAAC;KACzE,CAAC;IAEF,OAAO,eAAe,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE;QACxC,wBAAwB,CAAC;YACvB,QAAQ,EAAM,OAAO,CAAC,IAAI;YAC1B,YAAY,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE;SAC7C,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,mCAAmC;YACnC,IAAI,GAAG,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;gBACvC,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,cAAc,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACrE,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;oBACvB,MAAM,QAAQ,GAAG,MAAM,EAAE,QAAQ,IAAI,SAAS,CAAC;oBAC/C,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,UAAU,CAAC,SAAU,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,MAAO,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBACzK,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,UAAU,CAAC,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;oBAChM,MAAM,IAAI,KAAK,CACb,8BAA8B,UAAU,CAAC,SAAS,uBAAuB,OAAO,CAAC,IAAI,MAAM,UAAU,CAAC,MAAM,EAAE,CAC/G,CAAC;gBACJ,CAAC;gBACD,iEAAiE;gBACjE,IAAI,UAAU,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACpC,IAAI,CAAC;wBAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAM,CAAC;oBAAC,CAAC;oBAAC,MAAM,CAAC,CAAC,4CAA4C,CAAC,CAAC;gBAC7G,CAAC;YACH,CAAC;YAED,MAAM,GAAG,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;YAE7D,iCAAiC;YACjC,MAAM,WAAW,GAAG,KAAK,IAA4B,EAAE;gBACrD,IAAI,OAAO,CAAC,KAAK;oBAAE,OAAO,SAAS,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;gBACtE,OAAO,GAAG,EAAE,CAAC;YACf,CAAC,CAAC;YAEF,qEAAqE;YACrE,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,IAAI,GAAG,CAAC,QAAQ,EAAE,WAAW,CAAC;YAC/D,MAAM,kBAAkB,GAAG,KAAK,IAA4B,EAAE;gBAC5D,IAAI,SAAS,IAAI,IAAI,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;oBACvC,MAAM,EAAE,GAAG,IAAI,eAAe,EAAE,CAAC;oBACjC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;oBACtD,IAAI,CAAC;wBACH,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC;4BACxB,WAAW,EAAE;4BACb,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gCAC/B,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;oCACvC,MAAM,CAAC,IAAI,gBAAgB,CAAC,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC;gCACxD,CAAC,CAAC,CAAC;4BACL,CAAC,CAAC;yBACH,CAAC,CAAC;oBACL,CAAC;4BAAS,CAAC;wBACT,YAAY,CAAC,KAAK,CAAC,CAAC;oBACtB,CAAC;gBACH,CAAC;gBACD,OAAO,WAAW,EAAE,CAAC;YACvB,CAAC,CAAC;YAEF,0CAA0C;YAC1C,MAAM,EAAE,GAAG,iBAAiB,CAAC,OAAO,EAAE,GAAG,CAAC,eAAe,CAAC,CAAC;YAC3D,IAAI,MAAqB,CAAC;YAC1B,IAAI,CAAC;gBACH,IAAI,EAAE,EAAE,CAAC;oBACP,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;gBAChD,CAAC;qBAAM,CAAC;oBACN,MAAM,GAAG,MAAM,kBAAkB,EAAE,CAAC;gBACtC,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,sEAAsE;gBACtE,IAAI,GAAG,YAAY,mBAAmB;oBAAE,MAAM,GAAG,CAAC;gBAClD,sEAAsE;gBACtE,IAAI,GAAG,YAAY,uBAAuB,IAAI,OAAO,CAAC,cAAc,EAAE,QAAQ,EAAE,CAAC;oBAC/E,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC,KAAK,EAAE,GAAG,CAAkB,CAAC;gBACxE,CAAC;qBAAM,CAAC;oBACN,6FAA6F;oBAC7F,IAAI,GAAG,YAAY,QAAQ;wBAAE,MAAM,GAAG,CAAC;oBACvC,6DAA6D;oBAC7D,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;oBAClE,MAAM,IAAI,kBAAkB,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;YAED,oCAAoC;YACpC,IAAI,GAAG,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;gBACpD,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;gBACzC,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC,cAAc,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;gBACxE,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;oBACxB,MAAM,QAAQ,GAAG,MAAM,EAAE,QAAQ,IAAI,SAAS,CAAC;oBAC/C,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,WAAW,CAAC,SAAU,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,CAAC,MAAO,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC5K,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,WAAW,CAAC,SAAS,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;oBACnM,MAAM,IAAI,KAAK,CACb,8BAA8B,WAAW,CAAC,SAAS,wBAAwB,OAAO,CAAC,IAAI,MAAM,WAAW,CAAC,MAAM,EAAE,CAClH,CAAC;gBACJ,CAAC;gBACD,iEAAiE;gBACjE,IAAI,WAAW,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;oBACtC,IAAI,CAAC;wBAAC,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAkB,CAAC;oBAAC,CAAC;oBAAC,MAAM,CAAC,CAAC,6CAA6C,CAAC,CAAC;gBAC5H,CAAC;YACH,CAAC;YAED,0EAA0E;YAC1E,IAAI,OAAO,CAAC,KAAK,IAAI,QAAQ,EAAE,CAAC;gBAC9B,IAAI,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI,mBAAmB,EAAE,CAAC;oBAC9C,MAAM,MAAM,GAAG,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;oBACjD,IAAI,MAAM,KAAK,SAAS;wBAAE,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACzD,CAAC;gBACD,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACjE,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;gBAAS,CAAC;YACT,sBAAsB,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/pregel/index.d.ts

@@ -0,0 +1,61 @@
+import { type NodeDefinition, type Edge, type ChannelSchema, type ONIConfig, type ONIStreamEvent, type StreamMode, type InterruptConfig, type ONICheckpointer, type ONICheckpoint } from "../types.js";
+import { DeadLetterQueue, type DeadLetter } from "../dlq.js";
+import type { BaseStore } from "../store/index.js";
+import { HITLSessionStore } from "../hitl/index.js";
+import { EventBus } from "../events/bus.js";
+import type { GuardrailsConfig } from "../guardrails/types.js";
+import type { EventListeners } from "../events/types.js";
+import { AuditLog } from "../guardrails/audit.js";
+import { BudgetTracker } from "../guardrails/budget.js";
+import { ONITracer, type TracerLike } from "../telemetry.js";
+import type { CustomStreamEvent, MessageStreamEvent } from "../types.js";
+export declare class ONIPregelRunner<S extends Record<string, unknown>> {
+    private readonly nodes;
+    private readonly edges;
+    private readonly channels;
+    private readonly interruptConfig;
+    private readonly checkpointer;
+    private readonly store;
+    private readonly defaults?;
+    private readonly dlq;
+    private hitlStore;
+    private nodeCache;
+    private circuitBreakers;
+    /** Count of concurrent subgraph invocations active on this runner. >0 means running as subgraph. */
+    _subgraphRef: {
+        count: number;
+    };
+    /** Per-invocation parent updates from Command.PARENT, keyed by parent threadId. */
+    readonly _perInvocationParentUpdates: Map<string, Partial<unknown>[]>;
+    /** Per-invocation checkpointer override for subgraph isolation, keyed by threadId. */
+    readonly _perInvocationCheckpointer: Map<string, unknown>;
+    readonly eventBus: EventBus;
+    readonly auditLog: AuditLog | null;
+    readonly budgetTracker: BudgetTracker | null;
+    private readonly contentFilters;
+    private readonly toolPermissions;
+    readonly tracer: ONITracer;
+    /** Pre-indexed edges by source node — O(1) lookup instead of O(n) filter */
+    private readonly _edgesBySource;
+    /** Pre-computed ephemeral channel keys — avoids iterating all channels */
+    private readonly _ephemeralKeys;
+    constructor(nodes: Map<string, NodeDefinition<S>>, edges: Edge<S>[], channels: ChannelSchema<S>, interruptConfig?: InterruptConfig, checkpointer?: ONICheckpointer<S> | null, store?: BaseStore | null, guardrails?: GuardrailsConfig, listeners?: EventListeners, defaults?: {
+        nodeTimeout?: number;
+    } | undefined, dlq?: DeadLetterQueue | null, tracer?: TracerLike | null);
+    private get _ctx();
+    _stream(input: Partial<S>, config?: ONIConfig, streamMode?: StreamMode | StreamMode[]): AsyncGenerator<ONIStreamEvent<S> | CustomStreamEvent | MessageStreamEvent>;
+    invoke(input: Partial<S>, config?: ONIConfig): Promise<S>;
+    stream(input: Partial<S>, config?: ONIConfig & {
+        streamMode?: StreamMode | StreamMode[];
+    }): AsyncGenerator<ONIStreamEvent<S> | CustomStreamEvent | MessageStreamEvent>;
+    batch(inputs: Partial<S>[], config?: ONIConfig): Promise<S[]>;
+    getState(threadId: string): Promise<S | null>;
+    updateState(threadId: string, update: Partial<S>): Promise<void>;
+    getStateAt(threadId: string, step: number): Promise<S | null>;
+    getHistory(threadId: string): Promise<ONICheckpoint<S>[]>;
+    forkFrom(threadId: string, step: number, newThreadId: string): Promise<void>;
+    getPendingInterrupts(threadId: string): import("../hitl/resume.js").HITLSession<S>[];
+    hitlSessionStore(): HITLSessionStore<S>;
+    getDeadLetters(threadId: string): DeadLetter[];
+}
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/pregel/index.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,KAAK,cAAc,EAAE,KAAK,IAAI,EAAE,KAAK,aAAa,EAClD,KAAK,SAAS,EAAE,KAAK,cAAc,EAAE,KAAK,UAAU,EAAE,KAAK,eAAe,EAC1E,KAAK,eAAe,EAAE,KAAK,aAAa,EAEzC,MAAM,aAAa,CAAC;AAErB,OAAO,EAAE,eAAe,EAAE,KAAK,UAAU,EAAE,MAAM,WAAW,CAAC;AAC7D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,KAAK,EAAE,gBAAgB,EAAiB,MAAM,wBAAwB,CAAC;AAE9E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,SAAS,EAAE,KAAK,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7D,OAAO,KAAK,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AASzE,qBAAa,eAAe,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAwB1D,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,YAAY;IAC7B,OAAO,CAAC,QAAQ,CAAC,KAAK;IAGtB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;IAC1B,OAAO,CAAC,QAAQ,CAAC,GAAG;IAhCtB,OAAO,CAAC,SAAS,CAA6B;IAC9C,OAAO,CAAC,SAAS,CAAmE;IACpF,OAAO,CAAC,eAAe,CAAqC;IAC5D,oGAAoG;IACpG,YAAY;;MAAgB;IAC5B,mFAAmF;IACnF,QAAQ,CAAC,2BAA2B,kCAA8C;IAClF,sFAAsF;IACtF,QAAQ,CAAC,0BAA0B,uBAA8B;IAEjE,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC5B,QAAQ,CAAC,QAAQ,EAAE,QAAQ,GAAG,IAAI,CAAC;IACnC,QAAQ,CAAC,aAAa,EAAE,aAAa,GAAG,IAAI,CAAC;IAC7C,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAkB;IACjD,OAAO,CAAC,QAAQ,CAAC,eAAe,CAA8B;IAC9D,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC;IAE3B,4EAA4E;IAC5E,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAyB;IACxD,0EAA0E;IAC1E,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAc;gBAG1B,KAAK,EAAY,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,EAC/C,KAAK,EAAY,IAAI,CAAC,CAAC,CAAC,EAAE,EAC1B,QAAQ,EAAS,aAAa,CAAC,CAAC,CAAC,EACjC,eAAe,GAAE,eAAoB,EACrC,YAAY,GAAK,eAAe,CAAC,CAAC,CAAC,GAAG,IAAW,EACjD,KAAK,GAAY,SAAS,GAAG,IAAW,EACzD,UAAU,CAAC,EAAuB,gBAAgB,EAClD,SAAS,CAAC,EAAwB,cAAc,EAC/B,QAAQ,CAAC,EAAQ;QAAE,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,YAAA,EACzC,GAAG,GAAc,eAAe,GAAG,IAAW,EAC/D,MAAM,CAAC,EAA2B,UAAU,GAAG,IAAI;IA+BrD,OAAO,KAAK,IAAI,GAyBf;IAMM,OAAO,CACZ,KAAK,EAAO,OAAO,CAAC,CAAC,CAAC,EACtB,MAAM,CAAC,EAAK,SAAS,EACrB,UAAU,GAAE,UAAU,GAAG,UAAU,EAAc,GAChD,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,GAAG,iBAAiB,GAAG,kBAAkB,CAAC;IAQvE,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC;IAQxD,MAAM,CACX,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EACjB,MAAM,CAAC,EAAE,SAAS,GAAG;QAAE,UAAU,CAAC,EAAE,UAAU,GAAG,UAAU,EAAE,CAAA;KAAE,GAC9D,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,GAAG,iBAAiB,GAAG,kBAAkB,CAAC;IAIvE,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAa7D,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAI7C,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAMhE,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAI7D,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;IAIzD,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMlF,oBAAoB,CAAC,QAAQ,EAAE,MAAM;IAIrC,gBAAgB,IAAI,gBAAgB,CAAC,CAAC,CAAC;IAMvC,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,UAAU,EAAE;CAG/C"}

package/dist/pregel/index.js

@@ -0,0 +1,154 @@
+// ============================================================
+// @oni.bot/core — Pregel Execution Engine v3
+// New: interrupt() context management, HITL resume, token streaming
+// ============================================================
+import { HITLSessionStore } from "../hitl/index.js";
+import { EventBus } from "../events/bus.js";
+import { AuditLog } from "../guardrails/audit.js";
+import { BudgetTracker } from "../guardrails/budget.js";
+import { ONITracer } from "../telemetry.js";
+import { streamSupersteps } from "./streaming.js";
+import { getState, updateState, getStateAt, getHistory, forkFrom, } from "./checkpointing.js";
+import { getPendingInterrupts } from "./interrupts.js";
+export class ONIPregelRunner {
+    nodes;
+    edges;
+    channels;
+    interruptConfig;
+    checkpointer;
+    store;
+    defaults;
+    dlq;
+    hitlStore = new HITLSessionStore();
+    nodeCache = new Map();
+    circuitBreakers = new Map();
+    /** Count of concurrent subgraph invocations active on this runner. >0 means running as subgraph. */
+    _subgraphRef = { count: 0 };
+    /** Per-invocation parent updates from Command.PARENT, keyed by parent threadId. */
+    _perInvocationParentUpdates = new Map();
+    /** Per-invocation checkpointer override for subgraph isolation, keyed by threadId. */
+    _perInvocationCheckpointer = new Map();
+    eventBus;
+    auditLog;
+    budgetTracker;
+    contentFilters;
+    toolPermissions;
+    tracer;
+    /** Pre-indexed edges by source node — O(1) lookup instead of O(n) filter */
+    _edgesBySource;
+    /** Pre-computed ephemeral channel keys — avoids iterating all channels */
+    _ephemeralKeys;
+    constructor(nodes, edges, channels, interruptConfig = {}, checkpointer = null, store = null, guardrails, listeners, defaults, dlq = null, tracer) {
+        this.nodes = nodes;
+        this.edges = edges;
+        this.channels = channels;
+        this.interruptConfig = interruptConfig;
+        this.checkpointer = checkpointer;
+        this.store = store;
+        this.defaults = defaults;
+        this.dlq = dlq;
+        this.eventBus = new EventBus(listeners);
+        this.auditLog = guardrails?.audit ? new AuditLog() : null;
+        this.budgetTracker = guardrails?.budget ? new BudgetTracker(guardrails.budget) : null;
+        this.contentFilters = guardrails?.filters ?? [];
+        this.toolPermissions = guardrails?.toolPermissions;
+        this.tracer = new ONITracer(tracer ?? null);
+        // Pre-index edges by source for O(1) lookups in getNextNodes
+        this._edgesBySource = new Map();
+        for (const edge of edges) {
+            const from = edge.from;
+            let list = this._edgesBySource.get(from);
+            if (!list) {
+                list = [];
+                this._edgesBySource.set(from, list);
+            }
+            list.push(edge);
+        }
+        // Pre-compute ephemeral keys to avoid scanning all channels per superstep
+        this._ephemeralKeys = Object.keys(channels).filter((k) => channels[k].ephemeral);
+    }
+    // ----------------------------------------------------------------
+    // Build PregelContext for extracted functions
+    // ----------------------------------------------------------------
+    get _ctx() {
+        return {
+            nodes: this.nodes,
+            edges: this.edges,
+            channels: this.channels,
+            interruptConfig: this.interruptConfig,
+            checkpointer: this.checkpointer,
+            store: this.store,
+            circuitBreakers: this.circuitBreakers,
+            tracer: this.tracer,
+            eventBus: this.eventBus,
+            auditLog: this.auditLog,
+            budgetTracker: this.budgetTracker,
+            contentFilters: this.contentFilters,
+            toolPermissions: this.toolPermissions,
+            dlq: this.dlq,
+            hitlStore: this.hitlStore,
+            defaults: this.defaults,
+            nodeCache: this.nodeCache,
+            _subgraphRef: this._subgraphRef,
+            _perInvocationParentUpdates: this._perInvocationParentUpdates,
+            _perInvocationCheckpointer: this._perInvocationCheckpointer,
+            _edgesBySource: this._edgesBySource,
+            _ephemeralKeys: this._ephemeralKeys,
+        };
+    }
+    // ----------------------------------------------------------------
+    // Core stream generator (delegates to streaming module)
+    // ----------------------------------------------------------------
+    async *_stream(input, config, streamMode = "updates") {
+        yield* streamSupersteps(this._ctx, input, config, streamMode);
+    }
+    // ----------------------------------------------------------------
+    // Public API
+    // ----------------------------------------------------------------
+    async invoke(input, config) {
+        let finalState;
+        for await (const evt of this._stream(input, config, "values")) {
+            if (evt.event === "state_update")
+                finalState = evt.data;
+        }
+        return finalState;
+    }
+    async *stream(input, config) {
+        yield* this._stream(input, config, config?.streamMode ?? "updates");
+    }
+    async batch(inputs, config) {
+        return Promise.all(inputs.map((inp, i) => this.invoke(inp, {
+            ...config,
+            threadId: config?.threadId ? `${config.threadId}-${i}` : undefined,
+        })));
+    }
+    // ---- State ----
+    async getState(threadId) {
+        return getState(this.checkpointer, threadId);
+    }
+    async updateState(threadId, update) {
+        return updateState(this.checkpointer, this.channels, threadId, update);
+    }
+    // ---- Time-travel ----
+    async getStateAt(threadId, step) {
+        return getStateAt(this.checkpointer, threadId, step);
+    }
+    async getHistory(threadId) {
+        return getHistory(this.checkpointer, threadId);
+    }
+    async forkFrom(threadId, step, newThreadId) {
+        return forkFrom(this.checkpointer, threadId, step, newThreadId);
+    }
+    // ---- HITL ----
+    getPendingInterrupts(threadId) {
+        return getPendingInterrupts(this.hitlStore, threadId);
+    }
+    hitlSessionStore() {
+        return this.hitlStore;
+    }
+    // ---- Dead Letter Queue ----
+    getDeadLetters(threadId) {
+        return this.dlq?.getAll(threadId) ?? [];
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/pregel/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/pregel/index.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,6CAA6C;AAC7C,oEAAoE;AACpE,+DAA+D;AAW/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAI5C,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,SAAS,EAAmB,MAAM,iBAAiB,CAAC;AAI7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EACL,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ,GACxD,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAEvD,MAAM,OAAO,eAAe;IAwBP;IACA;IACA;IACA;IACA;IACA;IAGA;IACA;IAhCX,SAAS,GAAG,IAAI,gBAAgB,EAAK,CAAC;IACtC,SAAS,GAAG,IAAI,GAAG,EAAwD,CAAC;IAC5E,eAAe,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC5D,oGAAoG;IACpG,YAAY,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;IAC5B,mFAAmF;IAC1E,2BAA2B,GAAG,IAAI,GAAG,EAAmC,CAAC;IAClF,sFAAsF;IAC7E,0BAA0B,GAAG,IAAI,GAAG,EAAmB,CAAC;IAExD,QAAQ,CAAW;IACnB,QAAQ,CAAkB;IAC1B,aAAa,CAAuB;IAC5B,cAAc,CAAkB;IAChC,eAAe,CAA8B;IACrD,MAAM,CAAY;IAE3B,4EAA4E;IAC3D,cAAc,CAAyB;IACxD,0EAA0E;IACzD,cAAc,CAAc;IAE7C,YACmB,KAA+C,EAC/C,KAA0B,EAC1B,QAAiC,EACjC,kBAAmC,EAAE,EACrC,eAA6C,IAAI,EACjD,QAAoC,IAAI,EACzD,UAAkD,EAClD,SAAgD,EAC/B,QAAyC,EACzC,MAA0C,IAAI,EAC/D,MAAmD;QAVlC,UAAK,GAAL,KAAK,CAA0C;QAC/C,UAAK,GAAL,KAAK,CAAqB;QAC1B,aAAQ,GAAR,QAAQ,CAAyB;QACjC,oBAAe,GAAf,eAAe,CAAsB;QACrC,iBAAY,GAAZ,YAAY,CAAqC;QACjD,UAAK,GAAL,KAAK,CAAmC;QAGxC,aAAQ,GAAR,QAAQ,CAAiC;QACzC,QAAG,GAAH,GAAG,CAA2C;QAG/D,IAAI,CAAC,QAAQ,GAAG,IAAI,QAAQ,CAAC,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ,GAAG,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QAC1D,IAAI,CAAC,aAAa,GAAG,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACtF,IAAI,CAAC,cAAc,GAAG,UAAU,EAAE,OAAO,IAAI,EAAE,CAAC;QAChD,IAAI,CAAC,eAAe,GAAG,UAAU,EAAE,eAAe,CAAC;QACnD,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;QAE5C,6DAA6D;QAC7D,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;QAChC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAc,CAAC;YACjC,IAAI,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACzC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,IAAI,GAAG,EAAE,CAAC;gBACV,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACtC,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClB,CAAC;QAED,0EAA0E;QAC1E,IAAI,CAAC,cAAc,GAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAiB,CAAC,MAAM,CACjE,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAC7B,CAAC;IACJ,CAAC;IAED,mEAAmE;IACnE,8CAA8C;IAC9C,mEAAmE;IAEnE,IAAY,IAAI;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,2BAA2B,EAAE,IAAI,CAAC,2BAA2B;YAC7D,0BAA0B,EAAE,IAAI,CAAC,0BAA0B;YAC3D,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,cAAc,EAAE,IAAI,CAAC,cAAc;SACpC,CAAC;IACJ,CAAC;IAED,mEAAmE;IACnE,wDAAwD;IACxD,mEAAmE;IAEnE,KAAK,CAAC,CAAC,OAAO,CACZ,KAAsB,EACtB,MAAqB,EACrB,aAAwC,SAAS;QAEjD,KAAK,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;IAChE,CAAC;IAED,mEAAmE;IACnE,aAAa;IACb,mEAAmE;IAEnE,KAAK,CAAC,MAAM,CAAC,KAAiB,EAAE,MAAkB;QAChD,IAAI,UAAc,CAAC;QACnB,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC;YAC9D,IAAI,GAAG,CAAC,KAAK,KAAK,cAAc;gBAAE,UAAU,GAAG,GAAG,CAAC,IAAS,CAAC;QAC/D,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,KAAiB,EACjB,MAA+D;QAE/D,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,IAAI,SAAS,CAAC,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAoB,EAAE,MAAkB;QAClD,OAAO,OAAO,CAAC,GAAG,CAChB,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CACpB,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE;YACf,GAAG,MAAM;YACT,QAAQ,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;SACnE,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAED,kBAAkB;IAElB,KAAK,CAAC,QAAQ,CAAC,QAAgB;QAC7B,OAAO,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,MAAkB;QACpD,OAAO,WAAW,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;IACzE,CAAC;IAED,wBAAwB;IAExB,KAAK,CAAC,UAAU,CAAC,QAAgB,EAAE,IAAY;QAC7C,OAAO,UAAU,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAAgB;QAC/B,OAAO,UAAU,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,IAAY,EAAE,WAAmB;QAChE,OAAO,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;IAClE,CAAC;IAED,iBAAiB;IAEjB,oBAAoB,CAAC,QAAgB;QACnC,OAAO,oBAAoB,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IACxD,CAAC;IAED,gBAAgB;QACd,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,8BAA8B;IAE9B,cAAc,CAAC,QAAgB;QAC7B,OAAO,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC1C,CAAC;CACF"}

package/dist/pregel/interrupts.d.ts

@@ -0,0 +1,3 @@
+import { HITLSessionStore } from "../hitl/index.js";
+export declare function getPendingInterrupts<S extends Record<string, unknown>>(hitlStore: HITLSessionStore<S>, threadId: string): import("../hitl/resume.js").HITLSession<S>[];
+//# sourceMappingURL=interrupts.d.ts.map

package/dist/pregel/interrupts.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"interrupts.d.ts","sourceRoot":"","sources":["../../src/pregel/interrupts.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAEpD,wBAAgB,oBAAoB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACpE,SAAS,EAAE,gBAAgB,CAAC,CAAC,CAAC,EAC9B,QAAQ,EAAE,MAAM,gDAGjB"}

package/dist/pregel/interrupts.js

@@ -0,0 +1,7 @@
+// ============================================================
+// src/pregel/interrupts.ts — HITL interrupt helpers
+// ============================================================
+export function getPendingInterrupts(hitlStore, threadId) {
+    return hitlStore.getByThread(threadId);
+}
+//# sourceMappingURL=interrupts.js.map

package/dist/pregel/interrupts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"interrupts.js","sourceRoot":"","sources":["../../src/pregel/interrupts.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,oDAAoD;AACpD,+DAA+D;AAI/D,MAAM,UAAU,oBAAoB,CAClC,SAA8B,EAC9B,QAAgB;IAEhB,OAAO,SAAS,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;AACzC,CAAC"}

package/dist/pregel/state-helpers.d.ts

@@ -0,0 +1,12 @@
+import { type NodeName, type Edge, type ChannelSchema, type ONIConfig, type ONIStreamEvent } from "../types.js";
+import type { PendingSend } from "./types.js";
+export declare function buildInitialState<S extends Record<string, unknown>>(channels: ChannelSchema<S>): S;
+export declare function applyUpdate<S extends Record<string, unknown>>(channels: ChannelSchema<S>, current: S, update: Partial<S>): S;
+export declare function resetEphemeral<S extends Record<string, unknown>>(state: S, ephemeralKeys: (keyof S)[], channels: ChannelSchema<S>): S;
+export declare function getNextNodes<S extends Record<string, unknown>>(fromNode: NodeName, state: S, edgesBySource: Map<string, Edge<S>[]>, config?: ONIConfig): {
+    nodes: NodeName[];
+    sends: PendingSend[];
+};
+export declare function checkDynamicInterrupt<S extends Record<string, unknown>>(node: string, timing: "before" | "after", state: S, config?: ONIConfig): void;
+export declare function evt<S extends Record<string, unknown>>(event: ONIStreamEvent<S>["event"], data: Partial<S> | S, step: number, agentId?: string, node?: string): ONIStreamEvent<S>;
+//# sourceMappingURL=state-helpers.d.ts.map

package/dist/pregel/state-helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"state-helpers.d.ts","sourceRoot":"","sources":["../../src/pregel/state-helpers.ts"],"names":[],"mappings":"AAIA,OAAO,EAEL,KAAK,QAAQ,EAAE,KAAK,IAAI,EAAE,KAAK,aAAa,EAC5C,KAAK,SAAS,EAAE,KAAK,cAAc,EACpC,MAAM,aAAa,CAAC;AAErB,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAE9C,wBAAgB,iBAAiB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACjE,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,GACzB,CAAC,CAMH;AAED,wBAAgB,WAAW,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC3D,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,EAC1B,OAAO,EAAE,CAAC,EACV,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,GACjB,CAAC,CAaH;AAED,wBAAgB,cAAc,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC9D,KAAK,EAAE,CAAC,EACR,aAAa,EAAE,CAAC,MAAM,CAAC,CAAC,EAAE,EAC1B,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,GACzB,CAAC,CAOH;AAED,wBAAgB,YAAY,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC5D,QAAQ,EAAE,QAAQ,EAClB,KAAK,EAAE,CAAC,EACR,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EACrC,MAAM,CAAC,EAAE,SAAS,GACjB;IAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;IAAC,KAAK,EAAE,WAAW,EAAE,CAAA;CAAE,CAkB7C;AAED,wBAAgB,qBAAqB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrE,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,QAAQ,GAAG,OAAO,EAC1B,KAAK,EAAE,CAAC,EACR,MAAM,CAAC,EAAE,SAAS,GACjB,IAAI,CAQN;AAED,wBAAgB,GAAG,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnD,KAAK,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EACjC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,EACpB,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,EAChB,IAAI,CAAC,EAAE,MAAM,GACZ,cAAc,CAAC,CAAC,CAAC,CAEnB"}

package/dist/pregel/state-helpers.js

@@ -0,0 +1,71 @@
+// ============================================================
+// src/pregel/state-helpers.ts — Pure state helper functions
+// ============================================================
+import { Send, } from "../types.js";
+import { ONIInterrupt } from "../errors.js";
+export function buildInitialState(channels) {
+    const state = {};
+    for (const key of Object.keys(channels)) {
+        state[key] = channels[key].default();
+    }
+    return state;
+}
+export function applyUpdate(channels, current, update) {
+    const keys = Object.keys(update);
+    if (keys.length === 0)
+        return current;
+    const next = { ...current };
+    for (const key of keys) {
+        if (update[key] !== undefined) {
+            const ch = channels[key];
+            next[key] = ch
+                ? ch.reducer(current[key], update[key])
+                : update[key];
+        }
+    }
+    return next;
+}
+export function resetEphemeral(state, ephemeralKeys, channels) {
+    if (ephemeralKeys.length === 0)
+        return state;
+    const next = { ...state };
+    for (const key of ephemeralKeys) {
+        next[key] = channels[key].default();
+    }
+    return next;
+}
+export function getNextNodes(fromNode, state, edgesBySource, config) {
+    const outgoing = edgesBySource.get(fromNode) ?? [];
+    const nodes = [];
+    const sends = [];
+    for (const edge of outgoing) {
+        if (edge.type === "static") {
+            nodes.push(edge.to);
+        }
+        else {
+            const result = edge.condition(state, config);
+            const resolved = Array.isArray(result) ? result : [result];
+            for (const r of resolved) {
+                if (r instanceof Send)
+                    sends.push({ node: r.node, args: r.args });
+                else
+                    nodes.push(edge.pathMap?.[r] ?? r);
+            }
+        }
+    }
+    return { nodes, sends };
+}
+export function checkDynamicInterrupt(node, timing, state, config) {
+    const dynamics = config?.dynamicInterrupts;
+    if (!dynamics)
+        return;
+    for (const di of dynamics) {
+        if (di.node === node && di.timing === timing && di.condition(state)) {
+            throw new ONIInterrupt(node, timing, state);
+        }
+    }
+}
+export function evt(event, data, step, agentId, node) {
+    return { event, data, step, timestamp: Date.now(), agentId, node };
+}
+//# sourceMappingURL=state-helpers.js.map

package/dist/pregel/state-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"state-helpers.js","sourceRoot":"","sources":["../../src/pregel/state-helpers.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,4DAA4D;AAC5D,+DAA+D;AAE/D,OAAO,EACL,IAAI,GAGL,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAG5C,MAAM,UAAU,iBAAiB,CAC/B,QAA0B;IAE1B,MAAM,KAAK,GAAG,EAAO,CAAC;IACtB,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAgB,EAAE,CAAC;QACvD,KAAK,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;IACvC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,WAAW,CACzB,QAA0B,EAC1B,OAAU,EACV,MAAkB;IAElB,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAgB,CAAC;IAChD,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IACtC,MAAM,IAAI,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;IAC5B,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS,EAAE,CAAC;YAC9B,MAAM,EAAE,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;YACzB,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE;gBACZ,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,GAAG,CAAe,CAAC;gBACrD,CAAC,CAAE,MAAM,CAAC,GAAG,CAAgB,CAAC;QAClC,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,KAAQ,EACR,aAA0B,EAC1B,QAA0B;IAE1B,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAC7C,MAAM,IAAI,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC;IAC1B,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;QAChC,IAAI,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;IACtC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,QAAkB,EAClB,KAAQ,EACR,aAAqC,EACrC,MAAkB;IAElB,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,QAAkB,CAAC,IAAI,EAAE,CAAC;IAC7D,MAAM,KAAK,GAAe,EAAE,CAAC;IAC7B,MAAM,KAAK,GAAkB,EAAE,CAAC;IAEhC,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;QAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACtB,CAAC;aAAM,CAAC;YACN,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YAC3D,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;gBACzB,IAAI,CAAC,YAAY,IAAI;oBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;;oBAC7D,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAW,CAAC,IAAI,CAAC,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,IAAY,EACZ,MAA0B,EAC1B,KAAQ,EACR,MAAkB;IAElB,MAAM,QAAQ,GAAG,MAAM,EAAE,iBAAsD,CAAC;IAChF,IAAI,CAAC,QAAQ;QAAE,OAAO;IACtB,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;QAC1B,IAAI,EAAE,CAAC,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC,MAAM,KAAK,MAAM,IAAI,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;YACpE,MAAM,IAAI,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,GAAG,CACjB,KAAiC,EACjC,IAAoB,EACpB,IAAY,EACZ,OAAgB,EAChB,IAAa;IAEb,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AACrE,CAAC"}

package/dist/pregel/streaming.d.ts

@@ -0,0 +1,5 @@
+import { type ONIConfig, type ONIStreamEvent, type StreamMode } from "../types.js";
+import type { CustomStreamEvent, MessageStreamEvent } from "../types.js";
+import type { PregelContext } from "./types.js";
+export declare function streamSupersteps<S extends Record<string, unknown>>(ctx: PregelContext<S>, input: Partial<S>, config: ONIConfig | undefined, streamMode?: StreamMode | StreamMode[]): AsyncGenerator<ONIStreamEvent<S> | CustomStreamEvent | MessageStreamEvent>;
+//# sourceMappingURL=streaming.d.ts.map