@oni.bot/core 1.0.3 → 1.1.0

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../../src/pregel/streaming.ts"],"names":[],"mappings":"AAQA,OAAO,EAEU,KAAK,SAAS,EAAE,KAAK,cAAc,EAAE,KAAK,UAAU,EAEpE,MAAM,aAAa,CAAC;AAQrB,OAAO,KAAK,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACzE,OAAO,KAAK,EAAE,aAAa,EAAe,MAAM,YAAY,CAAC;AAY7D,wBAAuB,gBAAgB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACvE,GAAG,EAAE,aAAa,CAAC,CAAC,CAAC,EACrB,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EACjB,MAAM,EAAE,SAAS,GAAG,SAAS,EAC7B,UAAU,GAAE,UAAU,GAAG,UAAU,EAAc,GAChD,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,GAAG,iBAAiB,GAAG,kBAAkB,CAAC,CAyd5E"}