@langchain/langgraph 0.4.8 → 1.0.0-alpha.0

package/dist/pregel/debug.cjs

@@ -1,245 +1,175 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.wrap = void 0;
-exports.printCheckpoint = printCheckpoint;
-exports._readChannels = _readChannels;
-exports.mapDebugTasks = mapDebugTasks;
-exports.mapDebugTaskResults = mapDebugTaskResults;
-exports.mapDebugCheckpoint = mapDebugCheckpoint;
-exports.tasksWithWrites = tasksWithWrites;
-exports.printStepCheckpoint = printStepCheckpoint;
-exports.printStepTasks = printStepTasks;
-exports.printStepWrites = printStepWrites;
-const constants_js_1 = require("../constants.cjs");
-const errors_js_1 = require("../errors.cjs");
-const io_js_1 = require("./io.cjs");
-const subgraph_js_1 = require("./utils/subgraph.cjs");
+const require_constants = require('../constants.cjs');
+const require_io = require('./io.cjs');
+const require_subgraph = require('./utils/subgraph.cjs');
+
+//#region src/pregel/debug.ts
 const COLORS_MAP = {
-    blue: {
-        start: "\x1b[34m",
-        end: "\x1b[0m",
-    },
-    green: {
-        start: "\x1b[32m",
-        end: "\x1b[0m",
-    },
-    yellow: {
-        start: "\x1b[33;1m",
-        end: "\x1b[0m",
-    },
+	blue: {
+		start: "\x1B[34m",
+		end: "\x1B[0m"
+	},
+	green: {
+		start: "\x1B[32m",
+		end: "\x1B[0m"
+	},
+	yellow: {
+		start: "\x1B[33;1m",
+		end: "\x1B[0m"
+	}
 };
 /**
- * Wrap some text in a color for printing to the console.
- */
+* Wrap some text in a color for printing to the console.
+*/
 const wrap = (color, text) => `${color.start}${text}${color.end}`;
-exports.wrap = wrap;
-function printCheckpoint(step, channels) {
-    console.log([
-        `${(0, exports.wrap)(COLORS_MAP.blue, "[langgraph/checkpoint]")}`,
-        `Finishing step ${step}. Channel values:\n`,
-        `\n${JSON.stringify(Object.fromEntries(_readChannels(channels)), null, 2)}`,
-    ].join(""));
-}
-function* _readChannels(channels
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-) {
-    for (const [name, channel] of Object.entries(channels)) {
-        try {
-            yield [name, channel.get()];
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        }
-        catch (error) {
-            if (error.name === errors_js_1.EmptyChannelError.unminifiable_name) {
-                // Skip the channel if it's empty
-                continue;
-            }
-            else {
-                throw error; // Re-throw the error if it's not an EmptyChannelError
-            }
-        }
-    }
-}
 function* mapDebugTasks(tasks) {
-    for (const { id, name, input, config, triggers, writes } of tasks) {
-        if (config?.tags?.includes(constants_js_1.TAG_HIDDEN))
-            continue;
-        const interrupts = writes
-            .filter(([writeId, n]) => {
-            return writeId === id && n === constants_js_1.INTERRUPT;
-        })
-            .map(([, v]) => {
-            return v;
-        });
-        yield { id, name, input, triggers, interrupts };
-    }
+	for (const { id, name, input, config, triggers, writes } of tasks) {
+		if (config?.tags?.includes(require_constants.TAG_HIDDEN)) continue;
+		const interrupts = writes.filter(([writeId, n]) => {
+			return writeId === id && n === require_constants.INTERRUPT;
+		}).map(([, v]) => {
+			return v;
+		});
+		yield {
+			id,
+			name,
+			input,
+			triggers,
+			interrupts
+		};
+	}
 }
 function isMultipleChannelWrite(value) {
-    if (typeof value !== "object" || value === null)
-        return false;
-    return "$writes" in value && Array.isArray(value.$writes);
+	if (typeof value !== "object" || value === null) return false;
+	return "$writes" in value && Array.isArray(value.$writes);
 }
 function mapTaskResultWrites(writes) {
-    const result = {};
-    for (const [channel, value] of writes) {
-        const strChannel = String(channel);
-        if (strChannel in result) {
-            const channelWrites = isMultipleChannelWrite(result[strChannel])
-                ? result[strChannel].$writes
-                : [result[strChannel]];
-            channelWrites.push(value);
-            result[strChannel] = { $writes: channelWrites };
-        }
-        else {
-            result[strChannel] = value;
-        }
-    }
-    return result;
+	const result = {};
+	for (const [channel, value] of writes) {
+		const strChannel = String(channel);
+		if (strChannel in result) {
+			const channelWrites = isMultipleChannelWrite(result[strChannel]) ? result[strChannel].$writes : [result[strChannel]];
+			channelWrites.push(value);
+			result[strChannel] = { $writes: channelWrites };
+		} else result[strChannel] = value;
+	}
+	return result;
 }
 function* mapDebugTaskResults(tasks, streamChannels) {
-    for (const [{ id, name, config }, writes] of tasks) {
-        if (config?.tags?.includes(constants_js_1.TAG_HIDDEN))
-            continue;
-        yield {
-            id,
-            name,
-            result: mapTaskResultWrites(writes.filter(([channel]) => {
-                return Array.isArray(streamChannels)
-                    ? streamChannels.includes(channel)
-                    : channel === streamChannels;
-            })),
-            interrupts: writes.filter((w) => w[0] === constants_js_1.INTERRUPT).map((w) => w[1]),
-        };
-    }
+	for (const [{ id, name, config }, writes] of tasks) {
+		if (config?.tags?.includes(require_constants.TAG_HIDDEN)) continue;
+		yield {
+			id,
+			name,
+			result: mapTaskResultWrites(writes.filter(([channel]) => {
+				return Array.isArray(streamChannels) ? streamChannels.includes(channel) : channel === streamChannels;
+			})),
+			interrupts: writes.filter((w) => w[0] === require_constants.INTERRUPT).map((w) => w[1])
+		};
+	}
 }
 function* mapDebugCheckpoint(config, channels, streamChannels, metadata, tasks, pendingWrites, parentConfig, outputKeys) {
-    function formatConfig(config) {
-        // make sure the config is consistent with Python
-        const pyConfig = {};
-        if (config.callbacks != null)
-            pyConfig.callbacks = config.callbacks;
-        if (config.configurable != null)
-            pyConfig.configurable = config.configurable;
-        if (config.maxConcurrency != null)
-            pyConfig.max_concurrency = config.maxConcurrency;
-        if (config.metadata != null)
-            pyConfig.metadata = config.metadata;
-        if (config.recursionLimit != null)
-            pyConfig.recursion_limit = config.recursionLimit;
-        if (config.runId != null)
-            pyConfig.run_id = config.runId;
-        if (config.runName != null)
-            pyConfig.run_name = config.runName;
-        if (config.tags != null)
-            pyConfig.tags = config.tags;
-        return pyConfig;
-    }
-    const parentNs = config.configurable?.checkpoint_ns;
-    const taskStates = {};
-    for (const task of tasks) {
-        const candidates = task.subgraphs?.length ? task.subgraphs : [task.proc];
-        if (!candidates.find(subgraph_js_1.findSubgraphPregel))
-            continue;
-        let taskNs = `${task.name}:${task.id}`;
-        if (parentNs)
-            taskNs = `${parentNs}|${taskNs}`;
-        taskStates[task.id] = {
-            configurable: {
-                thread_id: config.configurable?.thread_id,
-                checkpoint_ns: taskNs,
-            },
-        };
-    }
-    yield {
-        config: formatConfig(config),
-        values: (0, io_js_1.readChannels)(channels, streamChannels),
-        metadata,
-        next: tasks.map((task) => task.name),
-        tasks: tasksWithWrites(tasks, pendingWrites, taskStates, outputKeys),
-        parentConfig: parentConfig ? formatConfig(parentConfig) : undefined,
-    };
+	function formatConfig(config$1) {
+		const pyConfig = {};
+		if (config$1.callbacks != null) pyConfig.callbacks = config$1.callbacks;
+		if (config$1.configurable != null) pyConfig.configurable = config$1.configurable;
+		if (config$1.maxConcurrency != null) pyConfig.max_concurrency = config$1.maxConcurrency;
+		if (config$1.metadata != null) pyConfig.metadata = config$1.metadata;
+		if (config$1.recursionLimit != null) pyConfig.recursion_limit = config$1.recursionLimit;
+		if (config$1.runId != null) pyConfig.run_id = config$1.runId;
+		if (config$1.runName != null) pyConfig.run_name = config$1.runName;
+		if (config$1.tags != null) pyConfig.tags = config$1.tags;
+		return pyConfig;
+	}
+	const parentNs = config.configurable?.checkpoint_ns;
+	const taskStates = {};
+	for (const task of tasks) {
+		const candidates = task.subgraphs?.length ? task.subgraphs : [task.proc];
+		if (!candidates.find(require_subgraph.findSubgraphPregel)) continue;
+		let taskNs = `${task.name}:${task.id}`;
+		if (parentNs) taskNs = `${parentNs}|${taskNs}`;
+		taskStates[task.id] = { configurable: {
+			thread_id: config.configurable?.thread_id,
+			checkpoint_ns: taskNs
+		} };
+	}
+	yield {
+		config: formatConfig(config),
+		values: require_io.readChannels(channels, streamChannels),
+		metadata,
+		next: tasks.map((task) => task.name),
+		tasks: tasksWithWrites(tasks, pendingWrites, taskStates, outputKeys),
+		parentConfig: parentConfig ? formatConfig(parentConfig) : void 0
+	};
 }
 function tasksWithWrites(tasks, pendingWrites, states, outputKeys) {
-    return tasks.map((task) => {
-        const error = pendingWrites.find(([id, n]) => id === task.id && n === constants_js_1.ERROR)?.[2];
-        const interrupts = pendingWrites
-            .filter(([id, n]) => id === task.id && n === constants_js_1.INTERRUPT)
-            .map(([, , v]) => v);
-        const result = (() => {
-            if (error || interrupts.length || !pendingWrites.length)
-                return undefined;
-            const idx = pendingWrites.findIndex(([tid, n]) => tid === task.id && n === constants_js_1.RETURN);
-            if (idx >= 0)
-                return pendingWrites[idx][2];
-            if (typeof outputKeys === "string") {
-                return pendingWrites.find(([tid, n]) => tid === task.id && n === outputKeys)?.[2];
-            }
-            if (Array.isArray(outputKeys)) {
-                const results = pendingWrites
-                    .filter(([tid, n]) => tid === task.id && outputKeys.includes(n))
-                    .map(([, n, v]) => [n, v]);
-                if (!results.length)
-                    return undefined;
-                return mapTaskResultWrites(results);
-            }
-            return undefined;
-        })();
-        if (error) {
-            return {
-                id: task.id,
-                name: task.name,
-                path: task.path,
-                error,
-                interrupts,
-                result,
-            };
-        }
-        const taskState = states?.[task.id];
-        return {
-            id: task.id,
-            name: task.name,
-            path: task.path,
-            interrupts,
-            ...(taskState !== undefined ? { state: taskState } : {}),
-            result,
-        };
-    });
+	return tasks.map((task) => {
+		const error = pendingWrites.find(([id, n]) => id === task.id && n === require_constants.ERROR)?.[2];
+		const interrupts = pendingWrites.filter(([id, n]) => id === task.id && n === require_constants.INTERRUPT).map(([, , v]) => v);
+		const result = (() => {
+			if (error || interrupts.length || !pendingWrites.length) return void 0;
+			const idx = pendingWrites.findIndex(([tid, n]) => tid === task.id && n === require_constants.RETURN);
+			if (idx >= 0) return pendingWrites[idx][2];
+			if (typeof outputKeys === "string") return pendingWrites.find(([tid, n]) => tid === task.id && n === outputKeys)?.[2];
+			if (Array.isArray(outputKeys)) {
+				const results = pendingWrites.filter(([tid, n]) => tid === task.id && outputKeys.includes(n)).map(([, n, v]) => [n, v]);
+				if (!results.length) return void 0;
+				return mapTaskResultWrites(results);
+			}
+			return void 0;
+		})();
+		if (error) return {
+			id: task.id,
+			name: task.name,
+			path: task.path,
+			error,
+			interrupts,
+			result
+		};
+		const taskState = states?.[task.id];
+		return {
+			id: task.id,
+			name: task.name,
+			path: task.path,
+			interrupts,
+			...taskState !== void 0 ? { state: taskState } : {},
+			result
+		};
+	});
 }
 function printStepCheckpoint(step, channels, whitelist) {
-    console.log([
-        `${(0, exports.wrap)(COLORS_MAP.blue, `[${step}:checkpoint]`)}`,
-        `\x1b[1m State at the end of step ${step}:\x1b[0m\n`,
-        JSON.stringify((0, io_js_1.readChannels)(channels, whitelist), null, 2),
-    ].join(""));
+	console.log([
+		`${wrap(COLORS_MAP.blue, `[${step}:checkpoint]`)}`,
+		`\x1b[1m State at the end of step ${step}:\x1b[0m\n`,
+		JSON.stringify(require_io.readChannels(channels, whitelist), null, 2)
+	].join(""));
 }
 function printStepTasks(step, nextTasks) {
-    const nTasks = nextTasks.length;
-    console.log([
-        `${(0, exports.wrap)(COLORS_MAP.blue, `[${step}:tasks]`)}`,
-        `\x1b[1m Starting step ${step} with ${nTasks} task${nTasks === 1 ? "" : "s"}:\x1b[0m\n`,
-        nextTasks
-            .map((task) => `- ${(0, exports.wrap)(COLORS_MAP.green, String(task.name))} -> ${JSON.stringify(task.input, null, 2)}`)
-            .join("\n"),
-    ].join(""));
+	const nTasks = nextTasks.length;
+	console.log([
+		`${wrap(COLORS_MAP.blue, `[${step}:tasks]`)}`,
+		`\x1b[1m Starting step ${step} with ${nTasks} task${nTasks === 1 ? "" : "s"}:\x1b[0m\n`,
+		nextTasks.map((task) => `- ${wrap(COLORS_MAP.green, String(task.name))} -> ${JSON.stringify(task.input, null, 2)}`).join("\n")
+	].join(""));
 }
 function printStepWrites(step, writes, whitelist) {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const byChannel = {};
-    for (const [channel, value] of writes) {
-        if (whitelist.includes(channel)) {
-            if (!byChannel[channel]) {
-                byChannel[channel] = [];
-            }
-            byChannel[channel].push(value);
-        }
-    }
-    console.log([
-        `${(0, exports.wrap)(COLORS_MAP.blue, `[${step}:writes]`)}`,
-        `\x1b[1m Finished step ${step} with writes to ${Object.keys(byChannel).length} channel${Object.keys(byChannel).length !== 1 ? "s" : ""}:\x1b[0m\n`,
-        Object.entries(byChannel)
-            .map(([name, vals]) => `- ${(0, exports.wrap)(COLORS_MAP.yellow, name)} -> ${vals
-            .map((v) => JSON.stringify(v))
-            .join(", ")}`)
-            .join("\n"),
-    ].join(""));
+	const byChannel = {};
+	for (const [channel, value] of writes) if (whitelist.includes(channel)) {
+		if (!byChannel[channel]) byChannel[channel] = [];
+		byChannel[channel].push(value);
+	}
+	console.log([
+		`${wrap(COLORS_MAP.blue, `[${step}:writes]`)}`,
+		`\x1b[1m Finished step ${step} with writes to ${Object.keys(byChannel).length} channel${Object.keys(byChannel).length !== 1 ? "s" : ""}:\x1b[0m\n`,
+		Object.entries(byChannel).map(([name, vals]) => `- ${wrap(COLORS_MAP.yellow, name)} -> ${vals.map((v) => JSON.stringify(v)).join(", ")}`).join("\n")
+	].join(""));
 }
-//# sourceMappingURL=debug.js.map
+
+//#endregion
+exports.mapDebugCheckpoint = mapDebugCheckpoint;
+exports.mapDebugTaskResults = mapDebugTaskResults;
+exports.mapDebugTasks = mapDebugTasks;
+exports.printStepCheckpoint = printStepCheckpoint;
+exports.printStepTasks = printStepTasks;
+exports.printStepWrites = printStepWrites;
+exports.tasksWithWrites = tasksWithWrites;
+//# sourceMappingURL=debug.cjs.map

package/dist/pregel/debug.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"debug.cjs","names":["COLORS_MAP: ConsoleColorMap","TAG_HIDDEN","INTERRUPT","result: Record<string, unknown>","pyConfig: Partial<\n      Record<CamelToSnake<keyof RunnableConfig>, unknown>\n    >","config","taskStates: Record<string, RunnableConfig | StateSnapshot>","findSubgraphPregel","readChannels","ERROR","RETURN","byChannel: Record<string, any[]>"],"sources":["../../src/pregel/debug.ts"],"sourcesContent":["import { RunnableConfig } from \"@langchain/core/runnables\";\nimport {\n  CheckpointMetadata,\n  CheckpointPendingWrite,\n  PendingWrite,\n} from \"@langchain/langgraph-checkpoint\";\nimport { BaseChannel } from \"../channels/base.js\";\nimport {\n  ERROR,\n  Interrupt,\n  INTERRUPT,\n  RETURN,\n  TAG_HIDDEN,\n} from \"../constants.js\";\nimport { EmptyChannelError } from \"../errors.js\";\nimport {\n  PregelExecutableTask,\n  PregelTaskDescription,\n  StateSnapshot,\n} from \"./types.js\";\nimport { readChannels } from \"./io.js\";\nimport { findSubgraphPregel } from \"./utils/subgraph.js\";\n\ntype ConsoleColors = {\n  start: string;\n  end: string;\n};\n\ntype ConsoleColorMap = {\n  [key: string]: ConsoleColors;\n};\n\nconst COLORS_MAP: ConsoleColorMap = {\n  blue: {\n    start: \"\\x1b[34m\",\n    end: \"\\x1b[0m\",\n  },\n  green: {\n    start: \"\\x1b[32m\",\n    end: \"\\x1b[0m\",\n  },\n  yellow: {\n    start: \"\\x1b[33;1m\",\n    end: \"\\x1b[0m\",\n  },\n};\n\n/**\n * Wrap some text in a color for printing to the console.\n */\nexport const wrap = (color: ConsoleColors, text: string): string =>\n  `${color.start}${text}${color.end}`;\n\nexport function printCheckpoint<Value>(\n  step: number,\n  channels: Record<string, BaseChannel<Value>>\n) {\n  console.log(\n    [\n      `${wrap(COLORS_MAP.blue, \"[langgraph/checkpoint]\")}`,\n      `Finishing step ${step}. Channel values:\\n`,\n      `\\n${JSON.stringify(\n        Object.fromEntries(_readChannels<Value>(channels)),\n        null,\n        2\n      )}`,\n    ].join(\"\")\n  );\n}\n\nexport function* _readChannels<Value>(\n  channels: Record<string, BaseChannel<Value>>\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n): IterableIterator<[string, any]> {\n  for (const [name, channel] of Object.entries(channels)) {\n    try {\n      yield [name, channel.get()];\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    } catch (error: any) {\n      if (error.name === EmptyChannelError.unminifiable_name) {\n        // Skip the channel if it's empty\n        continue;\n      } else {\n        throw error; // Re-throw the error if it's not an EmptyChannelError\n      }\n    }\n  }\n}\n\nexport function* mapDebugTasks<N extends PropertyKey, C extends PropertyKey>(\n  tasks: readonly PregelExecutableTask<N, C>[]\n) {\n  for (const { id, name, input, config, triggers, writes } of tasks) {\n    if (config?.tags?.includes(TAG_HIDDEN)) continue;\n\n    const interrupts = writes\n      .filter(([writeId, n]) => {\n        return writeId === id && n === INTERRUPT;\n      })\n      .map(([, v]) => {\n        return v;\n      });\n    yield { id, name, input, triggers, interrupts };\n  }\n}\n\nfunction isMultipleChannelWrite(\n  value: unknown\n): value is { $writes: unknown[] } {\n  if (typeof value !== \"object\" || value === null) return false;\n  return \"$writes\" in value && Array.isArray(value.$writes);\n}\n\nfunction mapTaskResultWrites(writes: PendingWrite<unknown>[]) {\n  const result: Record<string, unknown> = {};\n\n  for (const [channel, value] of writes) {\n    const strChannel = String(channel);\n\n    if (strChannel in result) {\n      const channelWrites = isMultipleChannelWrite(result[strChannel])\n        ? result[strChannel].$writes\n        : [result[strChannel]];\n\n      channelWrites.push(value);\n      result[strChannel] = { $writes: channelWrites };\n    } else {\n      result[strChannel] = value;\n    }\n  }\n  return result;\n}\n\nexport function* mapDebugTaskResults<\n  N extends PropertyKey,\n  C extends PropertyKey\n>(\n  tasks: readonly [PregelExecutableTask<N, C>, PendingWrite<C>[]][],\n  streamChannels: PropertyKey | Array<PropertyKey>\n) {\n  for (const [{ id, name, config }, writes] of tasks) {\n    if (config?.tags?.includes(TAG_HIDDEN)) continue;\n    yield {\n      id,\n      name,\n      result: mapTaskResultWrites(\n        writes.filter(([channel]) => {\n          return Array.isArray(streamChannels)\n            ? streamChannels.includes(channel)\n            : channel === streamChannels;\n        })\n      ),\n      interrupts: writes.filter((w) => w[0] === INTERRUPT).map((w) => w[1]),\n    };\n  }\n}\n\ntype ChannelKey = string | number | symbol;\n\nexport function* mapDebugCheckpoint<\n  N extends PropertyKey,\n  C extends PropertyKey\n>(\n  config: RunnableConfig,\n  channels: Record<string, BaseChannel>,\n  streamChannels: string | string[],\n  metadata: CheckpointMetadata,\n  tasks: readonly PregelExecutableTask<N, C>[],\n  pendingWrites: CheckpointPendingWrite[],\n  parentConfig: RunnableConfig | undefined,\n  outputKeys: ChannelKey | ChannelKey[]\n) {\n  function formatConfig(config: RunnableConfig) {\n    // https://stackoverflow.com/a/78298178\n    type CamelToSnake<\n      T extends string,\n      A extends string = \"\"\n    > = T extends `${infer F}${infer R}`\n      ? CamelToSnake<\n          R,\n          `${A}${F extends Lowercase<F> ? F : `_${Lowercase<F>}`}`\n        >\n      : A;\n\n    // make sure the config is consistent with Python\n    const pyConfig: Partial<\n      Record<CamelToSnake<keyof RunnableConfig>, unknown>\n    > = {};\n\n    if (config.callbacks != null) pyConfig.callbacks = config.callbacks;\n    if (config.configurable != null)\n      pyConfig.configurable = config.configurable;\n    if (config.maxConcurrency != null)\n      pyConfig.max_concurrency = config.maxConcurrency;\n\n    if (config.metadata != null) pyConfig.metadata = config.metadata;\n    if (config.recursionLimit != null)\n      pyConfig.recursion_limit = config.recursionLimit;\n    if (config.runId != null) pyConfig.run_id = config.runId;\n    if (config.runName != null) pyConfig.run_name = config.runName;\n    if (config.tags != null) pyConfig.tags = config.tags;\n\n    return pyConfig;\n  }\n\n  const parentNs = config.configurable?.checkpoint_ns;\n  const taskStates: Record<string, RunnableConfig | StateSnapshot> = {};\n\n  for (const task of tasks) {\n    const candidates = task.subgraphs?.length ? task.subgraphs : [task.proc];\n    if (!candidates.find(findSubgraphPregel)) continue;\n\n    let taskNs = `${task.name as string}:${task.id}`;\n    if (parentNs) taskNs = `${parentNs}|${taskNs}`;\n\n    taskStates[task.id] = {\n      configurable: {\n        thread_id: config.configurable?.thread_id,\n        checkpoint_ns: taskNs,\n      },\n    };\n  }\n\n  yield {\n    config: formatConfig(config),\n    values: readChannels(channels, streamChannels),\n    metadata,\n    next: tasks.map((task) => task.name),\n    tasks: tasksWithWrites(tasks, pendingWrites, taskStates, outputKeys),\n    parentConfig: parentConfig ? formatConfig(parentConfig) : undefined,\n  };\n}\n\nexport function tasksWithWrites<N extends PropertyKey, C extends PropertyKey>(\n  tasks: PregelTaskDescription[] | readonly PregelExecutableTask<N, C>[],\n  pendingWrites: CheckpointPendingWrite[],\n  states: Record<string, RunnableConfig | StateSnapshot> | undefined,\n  outputKeys: ChannelKey[] | ChannelKey\n): PregelTaskDescription[] {\n  return tasks.map((task): PregelTaskDescription => {\n    const error = pendingWrites.find(\n      ([id, n]) => id === task.id && n === ERROR\n    )?.[2];\n\n    const interrupts = pendingWrites\n      .filter(([id, n]) => id === task.id && n === INTERRUPT)\n      .map(([, , v]) => v) as Interrupt[];\n\n    const result = (() => {\n      if (error || interrupts.length || !pendingWrites.length) return undefined;\n\n      const idx = pendingWrites.findIndex(\n        ([tid, n]) => tid === task.id && n === RETURN\n      );\n\n      if (idx >= 0) return pendingWrites[idx][2];\n\n      if (typeof outputKeys === \"string\") {\n        return pendingWrites.find(\n          ([tid, n]) => tid === task.id && n === outputKeys\n        )?.[2];\n      }\n\n      if (Array.isArray(outputKeys)) {\n        const results = pendingWrites\n          .filter(([tid, n]) => tid === task.id && outputKeys.includes(n))\n          .map(([, n, v]) => [n, v] as PendingWrite<C>);\n\n        if (!results.length) return undefined;\n        return mapTaskResultWrites(results);\n      }\n\n      return undefined;\n    })();\n\n    if (error) {\n      return {\n        id: task.id,\n        name: task.name as string,\n        path: task.path,\n        error,\n        interrupts,\n        result,\n      };\n    }\n\n    const taskState = states?.[task.id];\n    return {\n      id: task.id,\n      name: task.name as string,\n      path: task.path,\n      interrupts,\n      ...(taskState !== undefined ? { state: taskState } : {}),\n      result,\n    };\n  });\n}\n\nexport function printStepCheckpoint(\n  step: number,\n  channels: Record<string, BaseChannel<unknown>>,\n  whitelist: string[]\n): void {\n  console.log(\n    [\n      `${wrap(COLORS_MAP.blue, `[${step}:checkpoint]`)}`,\n      `\\x1b[1m State at the end of step ${step}:\\x1b[0m\\n`,\n      JSON.stringify(readChannels(channels, whitelist), null, 2),\n    ].join(\"\")\n  );\n}\n\nexport function printStepTasks<N extends PropertyKey, C extends PropertyKey>(\n  step: number,\n  nextTasks: readonly PregelExecutableTask<N, C>[]\n): void {\n  const nTasks = nextTasks.length;\n  console.log(\n    [\n      `${wrap(COLORS_MAP.blue, `[${step}:tasks]`)}`,\n      `\\x1b[1m Starting step ${step} with ${nTasks} task${\n        nTasks === 1 ? \"\" : \"s\"\n      }:\\x1b[0m\\n`,\n      nextTasks\n        .map(\n          (task) =>\n            `- ${wrap(COLORS_MAP.green, String(task.name))} -> ${JSON.stringify(\n              task.input,\n              null,\n              2\n            )}`\n        )\n        .join(\"\\n\"),\n    ].join(\"\")\n  );\n}\n\nexport function printStepWrites(\n  step: number,\n  writes: PendingWrite[],\n  whitelist: string[]\n): void {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  const byChannel: Record<string, any[]> = {};\n\n  for (const [channel, value] of writes) {\n    if (whitelist.includes(channel)) {\n      if (!byChannel[channel]) {\n        byChannel[channel] = [];\n      }\n      byChannel[channel].push(value);\n    }\n  }\n\n  console.log(\n    [\n      `${wrap(COLORS_MAP.blue, `[${step}:writes]`)}`,\n      `\\x1b[1m Finished step ${step} with writes to ${\n        Object.keys(byChannel).length\n      } channel${Object.keys(byChannel).length !== 1 ? \"s\" : \"\"}:\\x1b[0m\\n`,\n      Object.entries(byChannel)\n        .map(\n          ([name, vals]) =>\n            `- ${wrap(COLORS_MAP.yellow, name)} -> ${vals\n              .map((v) => JSON.stringify(v))\n              .join(\", \")}`\n        )\n        .join(\"\\n\"),\n    ].join(\"\")\n  );\n}\n"],"mappings":";;;;;AAgCA,MAAMA,aAA8B;CAClC,MAAM;EACJ,OAAO;EACP,KAAK;;CAEP,OAAO;EACL,OAAO;EACP,KAAK;;CAEP,QAAQ;EACN,OAAO;EACP,KAAK;;;;;;AAOT,MAAa,QAAQ,OAAsB,SACzC,GAAG,MAAM,QAAQ,OAAO,MAAM;AAsChC,UAAiB,cACf,OACA;AACA,MAAK,MAAM,EAAE,IAAI,MAAM,OAAO,QAAQ,UAAU,YAAY,OAAO;AACjE,MAAI,QAAQ,MAAM,SAASC,8BAAa;EAExC,MAAM,aAAa,OAChB,QAAQ,CAAC,SAAS,OAAO;AACxB,UAAO,YAAY,MAAM,MAAMC;KAEhC,KAAK,GAAG,OAAO;AACd,UAAO;;AAEX,QAAM;GAAE;GAAI;GAAM;GAAO;GAAU;;;;AAIvC,SAAS,uBACP,OACiC;AACjC,KAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,QAAO,aAAa,SAAS,MAAM,QAAQ,MAAM;;AAGnD,SAAS,oBAAoB,QAAiC;CAC5D,MAAMC,SAAkC;AAExC,MAAK,MAAM,CAAC,SAAS,UAAU,QAAQ;EACrC,MAAM,aAAa,OAAO;AAE1B,MAAI,cAAc,QAAQ;GACxB,MAAM,gBAAgB,uBAAuB,OAAO,eAChD,OAAO,YAAY,UACnB,CAAC,OAAO;AAEZ,iBAAc,KAAK;AACnB,UAAO,cAAc,EAAE,SAAS;QAEhC,QAAO,cAAc;;AAGzB,QAAO;;AAGT,UAAiB,oBAIf,OACA,gBACA;AACA,MAAK,MAAM,CAAC,EAAE,IAAI,MAAM,UAAU,WAAW,OAAO;AAClD,MAAI,QAAQ,MAAM,SAASF,8BAAa;AACxC,QAAM;GACJ;GACA;GACA,QAAQ,oBACN,OAAO,QAAQ,CAAC,aAAa;AAC3B,WAAO,MAAM,QAAQ,kBACjB,eAAe,SAAS,WACxB,YAAY;;GAGpB,YAAY,OAAO,QAAQ,MAAM,EAAE,OAAOC,6BAAW,KAAK,MAAM,EAAE;;;;AAOxE,UAAiB,mBAIf,QACA,UACA,gBACA,UACA,OACA,eACA,cACA,YACA;CACA,SAAS,aAAa,UAAwB;EAa5C,MAAME,WAEF;AAEJ,MAAIC,SAAO,aAAa,KAAM,UAAS,YAAYA,SAAO;AAC1D,MAAIA,SAAO,gBAAgB,KACzB,UAAS,eAAeA,SAAO;AACjC,MAAIA,SAAO,kBAAkB,KAC3B,UAAS,kBAAkBA,SAAO;AAEpC,MAAIA,SAAO,YAAY,KAAM,UAAS,WAAWA,SAAO;AACxD,MAAIA,SAAO,kBAAkB,KAC3B,UAAS,kBAAkBA,SAAO;AACpC,MAAIA,SAAO,SAAS,KAAM,UAAS,SAASA,SAAO;AACnD,MAAIA,SAAO,WAAW,KAAM,UAAS,WAAWA,SAAO;AACvD,MAAIA,SAAO,QAAQ,KAAM,UAAS,OAAOA,SAAO;AAEhD,SAAO;;CAGT,MAAM,WAAW,OAAO,cAAc;CACtC,MAAMC,aAA6D;AAEnE,MAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,aAAa,KAAK,WAAW,SAAS,KAAK,YAAY,CAAC,KAAK;AACnE,MAAI,CAAC,WAAW,KAAKC,qCAAqB;EAE1C,IAAI,SAAS,GAAG,KAAK,KAAe,GAAG,KAAK;AAC5C,MAAI,SAAU,UAAS,GAAG,SAAS,GAAG;AAEtC,aAAW,KAAK,MAAM,EACpB,cAAc;GACZ,WAAW,OAAO,cAAc;GAChC,eAAe;;;AAKrB,OAAM;EACJ,QAAQ,aAAa;EACrB,QAAQC,wBAAa,UAAU;EAC/B;EACA,MAAM,MAAM,KAAK,SAAS,KAAK;EAC/B,OAAO,gBAAgB,OAAO,eAAe,YAAY;EACzD,cAAc,eAAe,aAAa,gBAAgB;;;AAI9D,SAAgB,gBACd,OACA,eACA,QACA,YACyB;AACzB,QAAO,MAAM,KAAK,SAAgC;EAChD,MAAM,QAAQ,cAAc,MACzB,CAAC,IAAI,OAAO,OAAO,KAAK,MAAM,MAAMC,2BACnC;EAEJ,MAAM,aAAa,cAChB,QAAQ,CAAC,IAAI,OAAO,OAAO,KAAK,MAAM,MAAMP,6BAC5C,KAAK,KAAK,OAAO;EAEpB,MAAM,gBAAgB;AACpB,OAAI,SAAS,WAAW,UAAU,CAAC,cAAc,OAAQ,QAAO;GAEhE,MAAM,MAAM,cAAc,WACvB,CAAC,KAAK,OAAO,QAAQ,KAAK,MAAM,MAAMQ;AAGzC,OAAI,OAAO,EAAG,QAAO,cAAc,KAAK;AAExC,OAAI,OAAO,eAAe,SACxB,QAAO,cAAc,MAClB,CAAC,KAAK,OAAO,QAAQ,KAAK,MAAM,MAAM,cACrC;AAGN,OAAI,MAAM,QAAQ,aAAa;IAC7B,MAAM,UAAU,cACb,QAAQ,CAAC,KAAK,OAAO,QAAQ,KAAK,MAAM,WAAW,SAAS,IAC5D,KAAK,GAAG,GAAG,OAAO,CAAC,GAAG;AAEzB,QAAI,CAAC,QAAQ,OAAQ,QAAO;AAC5B,WAAO,oBAAoB;;AAG7B,UAAO;;AAGT,MAAI,MACF,QAAO;GACL,IAAI,KAAK;GACT,MAAM,KAAK;GACX,MAAM,KAAK;GACX;GACA;GACA;;EAIJ,MAAM,YAAY,SAAS,KAAK;AAChC,SAAO;GACL,IAAI,KAAK;GACT,MAAM,KAAK;GACX,MAAM,KAAK;GACX;GACA,GAAI,cAAc,SAAY,EAAE,OAAO,cAAc;GACrD;;;;AAKN,SAAgB,oBACd,MACA,UACA,WACM;AACN,SAAQ,IACN;EACE,GAAG,KAAK,WAAW,MAAM,IAAI,KAAK;EAClC,oCAAoC,KAAK;EACzC,KAAK,UAAUF,wBAAa,UAAU,YAAY,MAAM;GACxD,KAAK;;AAIX,SAAgB,eACd,MACA,WACM;CACN,MAAM,SAAS,UAAU;AACzB,SAAQ,IACN;EACE,GAAG,KAAK,WAAW,MAAM,IAAI,KAAK;EAClC,yBAAyB,KAAK,QAAQ,OAAO,OAC3C,WAAW,IAAI,KAAK,IACrB;EACD,UACG,KACE,SACC,KAAK,KAAK,WAAW,OAAO,OAAO,KAAK,OAAO,MAAM,KAAK,UACxD,KAAK,OACL,MACA,MAGL,KAAK;GACR,KAAK;;AAIX,SAAgB,gBACd,MACA,QACA,WACM;CAEN,MAAMG,YAAmC;AAEzC,MAAK,MAAM,CAAC,SAAS,UAAU,OAC7B,KAAI,UAAU,SAAS,UAAU;AAC/B,MAAI,CAAC,UAAU,SACb,WAAU,WAAW;AAEvB,YAAU,SAAS,KAAK;;AAI5B,SAAQ,IACN;EACE,GAAG,KAAK,WAAW,MAAM,IAAI,KAAK;EAClC,yBAAyB,KAAK,kBAC5B,OAAO,KAAK,WAAW,OACxB,UAAU,OAAO,KAAK,WAAW,WAAW,IAAI,MAAM,GAAG;EAC1D,OAAO,QAAQ,WACZ,KACE,CAAC,MAAM,UACN,KAAK,KAAK,WAAW,QAAQ,MAAM,MAAM,KACtC,KAAK,MAAM,KAAK,UAAU,IAC1B,KAAK,SAEX,KAAK;GACR,KAAK"}