@langchain/langgraph 0.3.11 → 0.4.0

package/dist/pregel/algo.cjs

@@ -41,8 +41,7 @@ function shouldInterrupt(checkpoint, interruptNodes, tasks) {
         : interruptNodes.includes(task.name));
     return anyChannelUpdated && anyTriggeredNodeInInterruptNodes;
 }
-function _localRead(step, checkpoint, channels, managed, task, select, fresh = false) {
-    let managedKeys = [];
+function _localRead(checkpoint, channels, task, select, fresh = false) {
     let updated = new Set();
     if (!Array.isArray(select)) {
         for (const [c] of task.writes) {
@@ -54,8 +53,6 @@ function _localRead(step, checkpoint, channels, managed, task, select, fresh = f
         updated = updated || new Set();
     }
     else {
-        managedKeys = select.filter((k) => managed.get(k));
-        select = select.filter((k) => !managed.get(k));
         updated = new Set(select.filter((c) => task.writes.some(([key, _]) => key === c)));
     }
     let values;
@@ -69,20 +66,11 @@ function _localRead(step, checkpoint, channels, managed, task, select, fresh = f
     else {
         values = (0, io_js_1.readChannels)(channels, select);
     }
-    if (managedKeys.length > 0) {
-        for (const k of managedKeys) {
-            const managedValue = managed.get(k);
-            if (managedValue) {
-                const resultOfManagedCall = managedValue.call(step);
-                values[k] = resultOfManagedCall;
-            }
-        }
-    }
     return values;
 }
-function _localWrite(step, 
+function _localWrite(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-commit, processes, managed, 
+commit, processes, 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 writes) {
     for (const [chan, value] of writes) {
@@ -93,8 +81,6 @@ writes) {
             if (!(value.node in processes)) {
                 throw new errors_js_1.InvalidUpdateError(`Invalid node name "${value.node}" in Send packet`);
             }
-            // replace any runtime values with placeholders
-            managed.replaceRuntimeValues(step, value.args);
         }
     }
     commit(writes);
@@ -152,47 +138,32 @@ getNextVersion, triggerToNodes) {
     for (const chan of channelsToConsume) {
         if (chan in onlyChannels && onlyChannels[chan].consume()) {
             if (getNextVersion !== undefined) {
-                checkpoint.channel_versions[chan] = getNextVersion(maxVersion, onlyChannels[chan]);
+                checkpoint.channel_versions[chan] = getNextVersion(maxVersion);
             }
         }
     }
-    // Clear pending sends
-    if (checkpoint.pending_sends?.length && bumpStep) {
-        checkpoint.pending_sends = [];
-    }
     // Group writes by channel
-    const pendingWriteValuesByChannel = {};
-    const pendingWritesByManaged = {};
+    const pendingWritesByChannel = {};
     for (const task of tasks) {
         for (const [chan, val] of task.writes) {
             if (IGNORE.has(chan)) {
                 // do nothing
             }
-            else if (chan === constants_js_1.TASKS) {
-                // TODO: remove branch in 1.0
-                checkpoint.pending_sends.push({
-                    node: val.node,
-                    args: val.args,
-                });
-            }
             else if (chan in onlyChannels) {
-                pendingWriteValuesByChannel[chan] ??= [];
-                pendingWriteValuesByChannel[chan].push(val);
-            }
-            else {
-                pendingWritesByManaged[chan] ??= [];
-                pendingWritesByManaged[chan].push(val);
+                pendingWritesByChannel[chan] ??= [];
+                pendingWritesByChannel[chan].push(val);
             }
         }
     }
-    // find the highest version of all channels
+    // Find the highest version of all channels
+    // TODO: figure out why we need to do this twice (Python only does this once)
     maxVersion = undefined;
     if (Object.keys(checkpoint.channel_versions).length > 0) {
         maxVersion = (0, langgraph_checkpoint_1.maxChannelVersion)(...Object.values(checkpoint.channel_versions));
     }
     const updatedChannels = new Set();
     // Apply writes to channels
-    for (const [chan, vals] of Object.entries(pendingWriteValuesByChannel)) {
+    for (const [chan, vals] of Object.entries(pendingWritesByChannel)) {
         if (chan in onlyChannels) {
             let updated;
             try {
@@ -210,7 +181,7 @@ getNextVersion, triggerToNodes) {
                 }
             }
             if (updated && getNextVersion !== undefined) {
-                checkpoint.channel_versions[chan] = getNextVersion(maxVersion, onlyChannels[chan]);
+                checkpoint.channel_versions[chan] = getNextVersion(maxVersion);
                 // unavailable channels can't trigger tasks, so don't add them
                 if (onlyChannels[chan].isAvailable()) {
                     updatedChannels.add(chan);
@@ -224,7 +195,7 @@ getNextVersion, triggerToNodes) {
             if (onlyChannels[chan].isAvailable() && !updatedChannels.has(chan)) {
                 const updated = onlyChannels[chan].update([]);
                 if (updated && getNextVersion !== undefined) {
-                    checkpoint.channel_versions[chan] = getNextVersion(maxVersion, onlyChannels[chan]);
+                    checkpoint.channel_versions[chan] = getNextVersion(maxVersion);
                     // unavailable channels can't trigger tasks, so don't add them
                     if (onlyChannels[chan].isAvailable()) {
                         updatedChannels.add(chan);
@@ -235,11 +206,10 @@ getNextVersion, triggerToNodes) {
     }
     // If this is (tentatively) the last superstep, notify all channels of finish
     if (bumpStep &&
-        checkpoint.pending_sends.length === 0 &&
         !Object.keys(triggerToNodes ?? {}).some((channel) => updatedChannels.has(channel))) {
         for (const chan of Object.keys(onlyChannels)) {
             if (onlyChannels[chan].finish() && getNextVersion !== undefined) {
-                checkpoint.channel_versions[chan] = getNextVersion(maxVersion, onlyChannels[chan]);
+                checkpoint.channel_versions[chan] = getNextVersion(maxVersion);
                 // unavailable channels can't trigger tasks, so don't add them
                 if (onlyChannels[chan].isAvailable()) {
                     updatedChannels.add(chan);
@@ -247,27 +217,29 @@ getNextVersion, triggerToNodes) {
             }
         }
     }
-    // Return managed values writes to be applied externally
-    return pendingWritesByManaged;
 }
 /**
  * Prepare the set of tasks that will make up the next Pregel step.
  * This is the union of all PUSH tasks (Sends) and PULL tasks (nodes triggered
  * by edges).
  */
-function _prepareNextTasks(checkpoint, pendingWrites, processes, channels, managed, config, forExecution, extra) {
+function _prepareNextTasks(checkpoint, pendingWrites, processes, channels, config, forExecution, extra) {
     const tasks = {};
-    // Consume pending packets
-    for (let i = 0; i < checkpoint.pending_sends.length; i += 1) {
-        const task = _prepareSingleTask([constants_js_1.PUSH, i], checkpoint, pendingWrites, processes, channels, managed, config, forExecution, extra);
-        if (task !== undefined) {
-            tasks[task.id] = task;
+    // Consume pending tasks
+    const tasksChannel = channels[constants_js_1.TASKS];
+    if (tasksChannel?.isAvailable()) {
+        const len = tasksChannel.get().length;
+        for (let i = 0; i < len; i += 1) {
+            const task = _prepareSingleTask([constants_js_1.PUSH, i], checkpoint, pendingWrites, processes, channels, config, forExecution, extra);
+            if (task !== undefined) {
+                tasks[task.id] = task;
+            }
         }
     }
     // Check if any processes should be run in next step
     // If so, prepare the values to be passed to them
     for (const name of Object.keys(processes)) {
-        const task = _prepareSingleTask([constants_js_1.PULL, name], checkpoint, pendingWrites, processes, channels, managed, config, forExecution, extra);
+        const task = _prepareSingleTask([constants_js_1.PULL, name], checkpoint, pendingWrites, processes, channels, config, forExecution, extra);
         if (task !== undefined) {
             tasks[task.id] = task;
         }
@@ -278,7 +250,7 @@ function _prepareNextTasks(checkpoint, pendingWrites, processes, channels, manag
  * Prepares a single task for the next Pregel step, given a task path, which
  * uniquely identifies a PUSH or PULL task within the graph.
  */
-function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, channels, managed, config, forExecution, extra) {
+function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, channels, config, forExecution, extra) {
     const { step, checkpointer, manager } = extra;
     const configurable = config.configurable ?? {};
     const parentNamespace = configurable.checkpoint_ns ?? "";
@@ -324,8 +296,8 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
                     configurable: {
                         [constants_js_1.CONFIG_KEY_TASK_ID]: id,
                         // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                        [constants_js_1.CONFIG_KEY_SEND]: (writes_) => _localWrite(step, (items) => writes.push(...items), processes, managed, writes_),
-                        [constants_js_1.CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(step, checkpoint, channels, managed, {
+                        [constants_js_1.CONFIG_KEY_SEND]: (writes_) => _localWrite((items) => writes.push(...items), processes, writes_),
+                        [constants_js_1.CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(checkpoint, channels, {
                             name: call.name,
                             writes: writes,
                             triggers,
@@ -376,13 +348,16 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
         const index = typeof taskPath[1] === "number"
             ? taskPath[1]
             : parseInt(taskPath[1], 10);
-        if (index >= checkpoint.pending_sends.length) {
+        if (!channels[constants_js_1.TASKS]?.isAvailable()) {
             return undefined;
         }
-        const packet = (0, constants_js_1._isSendInterface)(checkpoint.pending_sends[index]) &&
-            !(0, constants_js_1._isSend)(checkpoint.pending_sends[index])
-            ? new constants_js_1.Send(checkpoint.pending_sends[index].node, checkpoint.pending_sends[index].args)
-            : checkpoint.pending_sends[index];
+        const sends = channels[constants_js_1.TASKS].get();
+        if (index < 0 || index >= sends.length) {
+            return undefined;
+        }
+        const packet = (0, constants_js_1._isSendInterface)(sends[index]) && !(0, constants_js_1._isSend)(sends[index])
+            ? new constants_js_1.Send(sends[index].node, sends[index].args)
+            : sends[index];
         if (!(0, constants_js_1._isSendInterface)(packet)) {
             console.warn(`Ignoring invalid packet ${JSON.stringify(packet)} in pending sends.`);
             return undefined;
@@ -414,7 +389,6 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
             const proc = processes[packet.node];
             const node = proc.getNode();
             if (node !== undefined) {
-                managed.replaceRuntimePlaceholders(step, packet.args);
                 if (proc.metadata !== undefined) {
                     metadata = { ...metadata, ...proc.metadata };
                 }
@@ -435,8 +409,8 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
                         configurable: {
                             [constants_js_1.CONFIG_KEY_TASK_ID]: taskId,
                             // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                            [constants_js_1.CONFIG_KEY_SEND]: (writes_) => _localWrite(step, (items) => writes.push(...items), processes, managed, writes_),
-                            [constants_js_1.CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(step, checkpoint, channels, managed, {
+                            [constants_js_1.CONFIG_KEY_SEND]: (writes_) => _localWrite((items) => writes.push(...items), processes, writes_),
+                            [constants_js_1.CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(checkpoint, channels, {
                                 name: packet.node,
                                 writes: writes,
                                 triggers,
@@ -528,7 +502,7 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
             .sort();
         // If any of the channels read by this process were updated
         if (triggers.length > 0) {
-            const val = _procInput(step, proc, managed, channels, forExecution);
+            const val = _procInput(proc, channels, forExecution);
             if (val === undefined) {
                 return undefined;
             }
@@ -573,10 +547,10 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
                             configurable: {
                                 [constants_js_1.CONFIG_KEY_TASK_ID]: taskId,
                                 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                                [constants_js_1.CONFIG_KEY_SEND]: (writes_) => _localWrite(step, (items) => {
+                                [constants_js_1.CONFIG_KEY_SEND]: (writes_) => _localWrite((items) => {
                                     writes.push(...items);
-                                }, processes, managed, writes_),
-                                [constants_js_1.CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(step, checkpoint, channels, managed, {
+                                }, processes, writes_),
+                                [constants_js_1.CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(checkpoint, channels, {
                                     name,
                                     writes: writes,
                                     triggers,
@@ -633,7 +607,7 @@ function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, chan
  *
  * @internal
  */
-function _procInput(step, proc, managed, channels, forExecution) {
+function _procInput(proc, channels, forExecution) {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     let val;
     if (typeof proc.channels === "object" && !Array.isArray(proc.channels)) {
@@ -667,9 +641,6 @@ function _procInput(step, proc, managed, channels, forExecution) {
                     }
                 }
             }
-            else {
-                val[k] = managed.get(k)?.call(step);
-            }
         }
     }
     else if (Array.isArray(proc.channels)) {

package/dist/pregel/algo.d.ts

@@ -1,10 +1,9 @@
 import { RunnableConfig } from "@langchain/core/runnables";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
-import { All, BaseCheckpointSaver, Checkpoint, ReadonlyCheckpoint, type PendingWrite, type PendingWriteValue, BaseStore, CheckpointPendingWrite } from "@langchain/langgraph-checkpoint";
+import { All, BaseCheckpointSaver, Checkpoint, ReadonlyCheckpoint, type PendingWrite, BaseStore, CheckpointPendingWrite } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { PregelNode } from "./read.js";
 import { PregelExecutableTask, PregelTaskDescription, SimpleTaskPath, TaskPath } from "./types.js";
-import { ManagedValueMapping } from "../managed/base.js";
 import { IterableReadableWritableStream } from "./stream.js";
 /**
  * Construct a type with a set of properties K of type T
@@ -20,9 +19,9 @@ export type WritesProtocol<C = string> = {
 };
 export declare const increment: (current?: number) => number;
 export declare function shouldInterrupt<N extends PropertyKey, C extends PropertyKey>(checkpoint: Checkpoint, interruptNodes: All | N[], tasks: PregelExecutableTask<N, C>[]): boolean;
-export declare function _localRead<Cc extends Record<string, BaseChannel>>(step: number, checkpoint: ReadonlyCheckpoint, channels: Cc, managed: ManagedValueMapping, task: WritesProtocol<keyof Cc>, select: Array<keyof Cc> | keyof Cc, fresh?: boolean): Record<string, unknown> | unknown;
-export declare function _localWrite(step: number, commit: (writes: [string, any][]) => any, processes: Record<string, PregelNode>, managed: ManagedValueMapping, writes: [string, any][]): void;
-export declare function _applyWrites<Cc extends Record<string, BaseChannel>>(checkpoint: Checkpoint, channels: Cc, tasks: WritesProtocol<keyof Cc>[], getNextVersion: ((version: any, channel: BaseChannel) => any) | undefined, triggerToNodes: Record<string, string[]> | undefined): Record<string, PendingWriteValue[]>;
+export declare function _localRead<Cc extends Record<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, channels: Cc, task: WritesProtocol<keyof Cc>, select: Array<keyof Cc> | keyof Cc, fresh?: boolean): Record<string, unknown> | unknown;
+export declare function _localWrite(commit: (writes: [string, any][]) => any, processes: Record<string, PregelNode>, writes: [string, any][]): void;
+export declare function _applyWrites<Cc extends Record<string, BaseChannel>>(checkpoint: Checkpoint, channels: Cc, tasks: WritesProtocol<keyof Cc>[], getNextVersion: ((version: any) => any) | undefined, triggerToNodes: Record<string, string[]> | undefined): void;
 export type NextTaskExtraFields = {
     step: number;
     isResuming?: boolean;
@@ -37,8 +36,8 @@ export type NextTaskExtraFieldsWithStore = NextTaskExtraFields & {
 export type NextTaskExtraFieldsWithoutStore = NextTaskExtraFields & {
     store?: never;
 };
-export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, pendingWrites: [string, string, unknown][] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFieldsWithoutStore): Record<string, PregelTaskDescription>;
-export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, pendingWrites: [string, string, unknown][] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFieldsWithStore): Record<string, PregelExecutableTask<keyof Nn, keyof Cc>>;
-export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: SimpleTaskPath, checkpoint: ReadonlyCheckpoint, pendingWrites: CheckpointPendingWrite[] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFields): PregelTaskDescription | undefined;
-export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: TaskPath, checkpoint: ReadonlyCheckpoint, pendingWrites: CheckpointPendingWrite[] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFields): PregelExecutableTask<keyof Nn, keyof Cc> | undefined;
-export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: TaskPath, checkpoint: ReadonlyCheckpoint, pendingWrites: CheckpointPendingWrite[] | undefined, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: boolean, extra: NextTaskExtraFieldsWithStore): PregelTaskDescription | PregelExecutableTask<keyof Nn, keyof Cc> | undefined;
+export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, pendingWrites: [string, string, unknown][] | undefined, processes: Nn, channels: Cc, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFieldsWithoutStore): Record<string, PregelTaskDescription>;
+export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, pendingWrites: [string, string, unknown][] | undefined, processes: Nn, channels: Cc, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFieldsWithStore): Record<string, PregelExecutableTask<keyof Nn, keyof Cc>>;
+export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: SimpleTaskPath, checkpoint: ReadonlyCheckpoint, pendingWrites: CheckpointPendingWrite[] | undefined, processes: Nn, channels: Cc, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFields): PregelTaskDescription | undefined;
+export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: TaskPath, checkpoint: ReadonlyCheckpoint, pendingWrites: CheckpointPendingWrite[] | undefined, processes: Nn, channels: Cc, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFields): PregelExecutableTask<keyof Nn, keyof Cc> | undefined;
+export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: TaskPath, checkpoint: ReadonlyCheckpoint, pendingWrites: CheckpointPendingWrite[] | undefined, processes: Nn, channels: Cc, config: RunnableConfig, forExecution: boolean, extra: NextTaskExtraFieldsWithStore): PregelTaskDescription | PregelExecutableTask<keyof Nn, keyof Cc> | undefined;

package/dist/pregel/algo.js

@@ -31,8 +31,7 @@ export function shouldInterrupt(checkpoint, interruptNodes, tasks) {
         : interruptNodes.includes(task.name));
     return anyChannelUpdated && anyTriggeredNodeInInterruptNodes;
 }
-export function _localRead(step, checkpoint, channels, managed, task, select, fresh = false) {
-    let managedKeys = [];
+export function _localRead(checkpoint, channels, task, select, fresh = false) {
     let updated = new Set();
     if (!Array.isArray(select)) {
         for (const [c] of task.writes) {
@@ -44,8 +43,6 @@ export function _localRead(step, checkpoint, channels, managed, task, select, fr
         updated = updated || new Set();
     }
     else {
-        managedKeys = select.filter((k) => managed.get(k));
-        select = select.filter((k) => !managed.get(k));
         updated = new Set(select.filter((c) => task.writes.some(([key, _]) => key === c)));
     }
     let values;
@@ -59,20 +56,11 @@ export function _localRead(step, checkpoint, channels, managed, task, select, fr
     else {
         values = readChannels(channels, select);
     }
-    if (managedKeys.length > 0) {
-        for (const k of managedKeys) {
-            const managedValue = managed.get(k);
-            if (managedValue) {
-                const resultOfManagedCall = managedValue.call(step);
-                values[k] = resultOfManagedCall;
-            }
-        }
-    }
     return values;
 }
-export function _localWrite(step, 
+export function _localWrite(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-commit, processes, managed, 
+commit, processes, 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 writes) {
     for (const [chan, value] of writes) {
@@ -83,8 +71,6 @@ writes) {
             if (!(value.node in processes)) {
                 throw new InvalidUpdateError(`Invalid node name "${value.node}" in Send packet`);
             }
-            // replace any runtime values with placeholders
-            managed.replaceRuntimeValues(step, value.args);
         }
     }
     commit(writes);
@@ -142,47 +128,32 @@ getNextVersion, triggerToNodes) {
     for (const chan of channelsToConsume) {
         if (chan in onlyChannels && onlyChannels[chan].consume()) {
             if (getNextVersion !== undefined) {
-                checkpoint.channel_versions[chan] = getNextVersion(maxVersion, onlyChannels[chan]);
+                checkpoint.channel_versions[chan] = getNextVersion(maxVersion);
             }
         }
     }
-    // Clear pending sends
-    if (checkpoint.pending_sends?.length && bumpStep) {
-        checkpoint.pending_sends = [];
-    }
     // Group writes by channel
-    const pendingWriteValuesByChannel = {};
-    const pendingWritesByManaged = {};
+    const pendingWritesByChannel = {};
     for (const task of tasks) {
         for (const [chan, val] of task.writes) {
             if (IGNORE.has(chan)) {
                 // do nothing
             }
-            else if (chan === TASKS) {
-                // TODO: remove branch in 1.0
-                checkpoint.pending_sends.push({
-                    node: val.node,
-                    args: val.args,
-                });
-            }
             else if (chan in onlyChannels) {
-                pendingWriteValuesByChannel[chan] ??= [];
-                pendingWriteValuesByChannel[chan].push(val);
-            }
-            else {
-                pendingWritesByManaged[chan] ??= [];
-                pendingWritesByManaged[chan].push(val);
+                pendingWritesByChannel[chan] ??= [];
+                pendingWritesByChannel[chan].push(val);
             }
         }
     }
-    // find the highest version of all channels
+    // Find the highest version of all channels
+    // TODO: figure out why we need to do this twice (Python only does this once)
     maxVersion = undefined;
     if (Object.keys(checkpoint.channel_versions).length > 0) {
         maxVersion = maxChannelVersion(...Object.values(checkpoint.channel_versions));
     }
     const updatedChannels = new Set();
     // Apply writes to channels
-    for (const [chan, vals] of Object.entries(pendingWriteValuesByChannel)) {
+    for (const [chan, vals] of Object.entries(pendingWritesByChannel)) {
         if (chan in onlyChannels) {
             let updated;
             try {
@@ -200,7 +171,7 @@ getNextVersion, triggerToNodes) {
                 }
             }
             if (updated && getNextVersion !== undefined) {
-                checkpoint.channel_versions[chan] = getNextVersion(maxVersion, onlyChannels[chan]);
+                checkpoint.channel_versions[chan] = getNextVersion(maxVersion);
                 // unavailable channels can't trigger tasks, so don't add them
                 if (onlyChannels[chan].isAvailable()) {
                     updatedChannels.add(chan);
@@ -214,7 +185,7 @@ getNextVersion, triggerToNodes) {
             if (onlyChannels[chan].isAvailable() && !updatedChannels.has(chan)) {
                 const updated = onlyChannels[chan].update([]);
                 if (updated && getNextVersion !== undefined) {
-                    checkpoint.channel_versions[chan] = getNextVersion(maxVersion, onlyChannels[chan]);
+                    checkpoint.channel_versions[chan] = getNextVersion(maxVersion);
                     // unavailable channels can't trigger tasks, so don't add them
                     if (onlyChannels[chan].isAvailable()) {
                         updatedChannels.add(chan);
@@ -225,11 +196,10 @@ getNextVersion, triggerToNodes) {
     }
     // If this is (tentatively) the last superstep, notify all channels of finish
     if (bumpStep &&
-        checkpoint.pending_sends.length === 0 &&
         !Object.keys(triggerToNodes ?? {}).some((channel) => updatedChannels.has(channel))) {
         for (const chan of Object.keys(onlyChannels)) {
             if (onlyChannels[chan].finish() && getNextVersion !== undefined) {
-                checkpoint.channel_versions[chan] = getNextVersion(maxVersion, onlyChannels[chan]);
+                checkpoint.channel_versions[chan] = getNextVersion(maxVersion);
                 // unavailable channels can't trigger tasks, so don't add them
                 if (onlyChannels[chan].isAvailable()) {
                     updatedChannels.add(chan);
@@ -237,27 +207,29 @@ getNextVersion, triggerToNodes) {
             }
         }
     }
-    // Return managed values writes to be applied externally
-    return pendingWritesByManaged;
 }
 /**
  * Prepare the set of tasks that will make up the next Pregel step.
  * This is the union of all PUSH tasks (Sends) and PULL tasks (nodes triggered
  * by edges).
  */
-export function _prepareNextTasks(checkpoint, pendingWrites, processes, channels, managed, config, forExecution, extra) {
+export function _prepareNextTasks(checkpoint, pendingWrites, processes, channels, config, forExecution, extra) {
     const tasks = {};
-    // Consume pending packets
-    for (let i = 0; i < checkpoint.pending_sends.length; i += 1) {
-        const task = _prepareSingleTask([PUSH, i], checkpoint, pendingWrites, processes, channels, managed, config, forExecution, extra);
-        if (task !== undefined) {
-            tasks[task.id] = task;
+    // Consume pending tasks
+    const tasksChannel = channels[TASKS];
+    if (tasksChannel?.isAvailable()) {
+        const len = tasksChannel.get().length;
+        for (let i = 0; i < len; i += 1) {
+            const task = _prepareSingleTask([PUSH, i], checkpoint, pendingWrites, processes, channels, config, forExecution, extra);
+            if (task !== undefined) {
+                tasks[task.id] = task;
+            }
         }
     }
     // Check if any processes should be run in next step
     // If so, prepare the values to be passed to them
     for (const name of Object.keys(processes)) {
-        const task = _prepareSingleTask([PULL, name], checkpoint, pendingWrites, processes, channels, managed, config, forExecution, extra);
+        const task = _prepareSingleTask([PULL, name], checkpoint, pendingWrites, processes, channels, config, forExecution, extra);
         if (task !== undefined) {
             tasks[task.id] = task;
         }
@@ -268,7 +240,7 @@ export function _prepareNextTasks(checkpoint, pendingWrites, processes, channels
  * Prepares a single task for the next Pregel step, given a task path, which
  * uniquely identifies a PUSH or PULL task within the graph.
  */
-export function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, channels, managed, config, forExecution, extra) {
+export function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processes, channels, config, forExecution, extra) {
     const { step, checkpointer, manager } = extra;
     const configurable = config.configurable ?? {};
     const parentNamespace = configurable.checkpoint_ns ?? "";
@@ -314,8 +286,8 @@ export function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processe
                     configurable: {
                         [CONFIG_KEY_TASK_ID]: id,
                         // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                        [CONFIG_KEY_SEND]: (writes_) => _localWrite(step, (items) => writes.push(...items), processes, managed, writes_),
-                        [CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(step, checkpoint, channels, managed, {
+                        [CONFIG_KEY_SEND]: (writes_) => _localWrite((items) => writes.push(...items), processes, writes_),
+                        [CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(checkpoint, channels, {
                             name: call.name,
                             writes: writes,
                             triggers,
@@ -366,13 +338,16 @@ export function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processe
         const index = typeof taskPath[1] === "number"
             ? taskPath[1]
             : parseInt(taskPath[1], 10);
-        if (index >= checkpoint.pending_sends.length) {
+        if (!channels[TASKS]?.isAvailable()) {
             return undefined;
         }
-        const packet = _isSendInterface(checkpoint.pending_sends[index]) &&
-            !_isSend(checkpoint.pending_sends[index])
-            ? new Send(checkpoint.pending_sends[index].node, checkpoint.pending_sends[index].args)
-            : checkpoint.pending_sends[index];
+        const sends = channels[TASKS].get();
+        if (index < 0 || index >= sends.length) {
+            return undefined;
+        }
+        const packet = _isSendInterface(sends[index]) && !_isSend(sends[index])
+            ? new Send(sends[index].node, sends[index].args)
+            : sends[index];
         if (!_isSendInterface(packet)) {
             console.warn(`Ignoring invalid packet ${JSON.stringify(packet)} in pending sends.`);
             return undefined;
@@ -404,7 +379,6 @@ export function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processe
             const proc = processes[packet.node];
             const node = proc.getNode();
             if (node !== undefined) {
-                managed.replaceRuntimePlaceholders(step, packet.args);
                 if (proc.metadata !== undefined) {
                     metadata = { ...metadata, ...proc.metadata };
                 }
@@ -425,8 +399,8 @@ export function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processe
                         configurable: {
                             [CONFIG_KEY_TASK_ID]: taskId,
                             // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                            [CONFIG_KEY_SEND]: (writes_) => _localWrite(step, (items) => writes.push(...items), processes, managed, writes_),
-                            [CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(step, checkpoint, channels, managed, {
+                            [CONFIG_KEY_SEND]: (writes_) => _localWrite((items) => writes.push(...items), processes, writes_),
+                            [CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(checkpoint, channels, {
                                 name: packet.node,
                                 writes: writes,
                                 triggers,
@@ -518,7 +492,7 @@ export function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processe
             .sort();
         // If any of the channels read by this process were updated
         if (triggers.length > 0) {
-            const val = _procInput(step, proc, managed, channels, forExecution);
+            const val = _procInput(proc, channels, forExecution);
             if (val === undefined) {
                 return undefined;
             }
@@ -563,10 +537,10 @@ export function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processe
                             configurable: {
                                 [CONFIG_KEY_TASK_ID]: taskId,
                                 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                                [CONFIG_KEY_SEND]: (writes_) => _localWrite(step, (items) => {
+                                [CONFIG_KEY_SEND]: (writes_) => _localWrite((items) => {
                                     writes.push(...items);
-                                }, processes, managed, writes_),
-                                [CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(step, checkpoint, channels, managed, {
+                                }, processes, writes_),
+                                [CONFIG_KEY_READ]: (select_, fresh_ = false) => _localRead(checkpoint, channels, {
                                     name,
                                     writes: writes,
                                     triggers,
@@ -623,7 +597,7 @@ export function _prepareSingleTask(taskPath, checkpoint, pendingWrites, processe
  *
  * @internal
  */
-function _procInput(step, proc, managed, channels, forExecution) {
+function _procInput(proc, channels, forExecution) {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     let val;
     if (typeof proc.channels === "object" && !Array.isArray(proc.channels)) {
@@ -657,9 +631,6 @@ function _procInput(step, proc, managed, channels, forExecution) {
                     }
                 }
             }
-            else {
-                val[k] = managed.get(k)?.call(step);
-            }
         }
     }
     else if (Array.isArray(proc.channels)) {