@langchain/langgraph 0.4.9 → 1.0.0-alpha.1

package/dist/pregel/io.js

@@ -1,238 +1,135 @@
-import { _isSend, Command, ERROR, INTERRUPT, NULL_TASK_ID, RESUME, RETURN, TAG_HIDDEN, TASKS, } from "../constants.js";
 import { EmptyChannelError, InvalidUpdateError } from "../errors.js";
+import { Command, ERROR, INTERRUPT, NULL_TASK_ID, RESUME, RETURN, TAG_HIDDEN, TASKS, _isSend } from "../constants.js";
 import { isXXH3 } from "../hash.js";
-export function readChannel(channels, chan, catchErrors = true, returnException = false) {
-    try {
-        return channels[chan].get();
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    }
-    catch (e) {
-        if (e.name === EmptyChannelError.unminifiable_name) {
-            if (returnException) {
-                return e;
-            }
-            else if (catchErrors) {
-                return null;
-            }
-        }
-        throw e;
-    }
+
+//#region src/pregel/io.ts
+function readChannel(channels, chan, catchErrors = true, returnException = false) {
+	try {
+		return channels[chan].get();
+	} catch (e) {
+		if (e.name === EmptyChannelError.unminifiable_name) {
+			if (returnException) return e;
+			else if (catchErrors) return null;
+		}
+		throw e;
+	}
 }
-export function readChannels(channels, select, skipEmpty = true
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-) {
-    if (Array.isArray(select)) {
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        const values = {};
-        for (const k of select) {
-            try {
-                values[k] = readChannel(channels, k, !skipEmpty);
-                // eslint-disable-next-line @typescript-eslint/no-explicit-any
-            }
-            catch (e) {
-                if (e.name === EmptyChannelError.unminifiable_name) {
-                    continue;
-                }
-            }
-        }
-        return values;
-    }
-    else {
-        return readChannel(channels, select);
-    }
+function readChannels(channels, select, skipEmpty = true) {
+	if (Array.isArray(select)) {
+		const values = {};
+		for (const k of select) try {
+			values[k] = readChannel(channels, k, !skipEmpty);
+		} catch (e) {
+			if (e.name === EmptyChannelError.unminifiable_name) continue;
+		}
+		return values;
+	} else return readChannel(channels, select);
 }
 /**
- * Map input chunk to a sequence of pending writes in the form (channel, value).
- */
-export function* mapCommand(cmd, pendingWrites) {
-    if (cmd.graph === Command.PARENT) {
-        throw new InvalidUpdateError("There is no parent graph.");
-    }
-    if (cmd.goto) {
-        let sends;
-        if (Array.isArray(cmd.goto)) {
-            sends = cmd.goto;
-        }
-        else {
-            sends = [cmd.goto];
-        }
-        for (const send of sends) {
-            if (_isSend(send)) {
-                yield [NULL_TASK_ID, TASKS, send];
-            }
-            else if (typeof send === "string") {
-                yield [NULL_TASK_ID, `branch:to:${send}`, "__start__"];
-            }
-            else {
-                throw new Error(`In Command.send, expected Send or string, got ${typeof send}`);
-            }
-        }
-    }
-    if (cmd.resume) {
-        if (typeof cmd.resume === "object" &&
-            Object.keys(cmd.resume).length &&
-            Object.keys(cmd.resume).every(isXXH3)) {
-            for (const [tid, resume] of Object.entries(cmd.resume)) {
-                const existing = pendingWrites
-                    .filter((w) => w[0] === tid && w[1] === RESUME)
-                    .map((w) => w[2])
-                    .slice(0, 1) ?? [];
-                existing.push(resume);
-                yield [tid, RESUME, existing];
-            }
-        }
-        else {
-            yield [NULL_TASK_ID, RESUME, cmd.resume];
-        }
-    }
-    if (cmd.update) {
-        if (typeof cmd.update !== "object" || !cmd.update) {
-            throw new Error("Expected cmd.update to be a dict mapping channel names to update values");
-        }
-        if (Array.isArray(cmd.update)) {
-            for (const [k, v] of cmd.update) {
-                yield [NULL_TASK_ID, k, v];
-            }
-        }
-        else {
-            for (const [k, v] of Object.entries(cmd.update)) {
-                yield [NULL_TASK_ID, k, v];
-            }
-        }
-    }
+* Map input chunk to a sequence of pending writes in the form (channel, value).
+*/
+function* mapCommand(cmd, pendingWrites) {
+	if (cmd.graph === Command.PARENT) throw new InvalidUpdateError("There is no parent graph.");
+	if (cmd.goto) {
+		let sends;
+		if (Array.isArray(cmd.goto)) sends = cmd.goto;
+		else sends = [cmd.goto];
+		for (const send of sends) if (_isSend(send)) yield [
+			NULL_TASK_ID,
+			TASKS,
+			send
+		];
+		else if (typeof send === "string") yield [
+			NULL_TASK_ID,
+			`branch:to:${send}`,
+			"__start__"
+		];
+		else throw new Error(`In Command.send, expected Send or string, got ${typeof send}`);
+	}
+	if (cmd.resume) if (typeof cmd.resume === "object" && Object.keys(cmd.resume).length && Object.keys(cmd.resume).every(isXXH3)) for (const [tid, resume] of Object.entries(cmd.resume)) {
+		const existing = pendingWrites.filter((w) => w[0] === tid && w[1] === RESUME).map((w) => w[2]).slice(0, 1) ?? [];
+		existing.push(resume);
+		yield [
+			tid,
+			RESUME,
+			existing
+		];
+	}
+	else yield [
+		NULL_TASK_ID,
+		RESUME,
+		cmd.resume
+	];
+	if (cmd.update) {
+		if (typeof cmd.update !== "object" || !cmd.update) throw new Error("Expected cmd.update to be a dict mapping channel names to update values");
+		if (Array.isArray(cmd.update)) for (const [k, v] of cmd.update) yield [
+			NULL_TASK_ID,
+			k,
+			v
+		];
+		else for (const [k, v] of Object.entries(cmd.update)) yield [
+			NULL_TASK_ID,
+			k,
+			v
+		];
+	}
 }
 /**
- * Map input chunk to a sequence of pending writes in the form [channel, value].
- */
-export function* mapInput(inputChannels, 
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-chunk
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-) {
-    if (chunk !== undefined && chunk !== null) {
-        if (Array.isArray(inputChannels) &&
-            typeof chunk === "object" &&
-            !Array.isArray(chunk)) {
-            for (const k in chunk) {
-                if (inputChannels.includes(k)) {
-                    yield [k, chunk[k]];
-                }
-            }
-        }
-        else if (Array.isArray(inputChannels)) {
-            throw new Error(`Input chunk must be an object when "inputChannels" is an array`);
-        }
-        else {
-            yield [inputChannels, chunk];
-        }
-    }
+* Map input chunk to a sequence of pending writes in the form [channel, value].
+*/
+function* mapInput(inputChannels, chunk) {
+	if (chunk !== void 0 && chunk !== null) if (Array.isArray(inputChannels) && typeof chunk === "object" && !Array.isArray(chunk)) {
+		for (const k in chunk) if (inputChannels.includes(k)) yield [k, chunk[k]];
+	} else if (Array.isArray(inputChannels)) throw new Error(`Input chunk must be an object when "inputChannels" is an array`);
+	else yield [inputChannels, chunk];
 }
 /**
- * Map pending writes (a sequence of tuples (channel, value)) to output chunk.
- */
-export function* mapOutputValues(outputChannels, pendingWrites, channels
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-) {
-    if (Array.isArray(outputChannels)) {
-        if (pendingWrites === true ||
-            pendingWrites.find(([chan, _]) => outputChannels.includes(chan))) {
-            yield readChannels(channels, outputChannels);
-        }
-    }
-    else {
-        if (pendingWrites === true ||
-            pendingWrites.some(([chan, _]) => chan === outputChannels)) {
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-            yield readChannel(channels, outputChannels);
-        }
-    }
+* Map pending writes (a sequence of tuples (channel, value)) to output chunk.
+*/
+function* mapOutputValues(outputChannels, pendingWrites, channels) {
+	if (Array.isArray(outputChannels)) {
+		if (pendingWrites === true || pendingWrites.find(([chan, _]) => outputChannels.includes(chan))) yield readChannels(channels, outputChannels);
+	} else if (pendingWrites === true || pendingWrites.some(([chan, _]) => chan === outputChannels)) yield readChannel(channels, outputChannels);
 }
 /**
- * Map pending writes (a sequence of tuples (channel, value)) to output chunk.
- * @internal
- *
- * @param outputChannels - The channels to output.
- * @param tasks - The tasks to output.
- * @param cached - Whether the output is cached.
- *
- * @returns A generator that yields the output chunk (if any).
- */
-export function* mapOutputUpdates(outputChannels, tasks, cached) {
-    const outputTasks = tasks.filter(([task, ww]) => {
-        return ((task.config === undefined || !task.config.tags?.includes(TAG_HIDDEN)) &&
-            ww[0][0] !== ERROR &&
-            ww[0][0] !== INTERRUPT);
-    });
-    if (!outputTasks.length) {
-        return;
-    }
-    let updated;
-    if (outputTasks.some(([task]) => task.writes.some(([chan, _]) => chan === RETURN))) {
-        // TODO: probably should assert that RETURN is the only "non-special" channel (starts with "__")
-        updated = outputTasks.flatMap(([task]) => task.writes
-            .filter(([chan, _]) => chan === RETURN)
-            .map(([_, value]) => [task.name, value]));
-    }
-    else if (!Array.isArray(outputChannels)) {
-        // special case where graph state is a single channel (MessageGraph)
-        // probably using this in functional API, too
-        updated = outputTasks.flatMap(([task]) => task.writes
-            .filter(([chan, _]) => chan === outputChannels)
-            .map(([_, value]) => [task.name, value]));
-    }
-    else {
-        updated = outputTasks.flatMap(([task]) => {
-            const { writes } = task;
-            const counts = {};
-            for (const [chan] of writes) {
-                if (outputChannels.includes(chan)) {
-                    counts[chan] = (counts[chan] || 0) + 1;
-                }
-            }
-            if (Object.values(counts).some((count) => count > 1)) {
-                // Multiple writes to the same channel: create separate entries
-                return writes
-                    .filter(([chan]) => outputChannels.includes(chan))
-                    .map(([chan, value]) => [task.name, { [chan]: value }]);
-            }
-            else {
-                // Single write to each channel: create a single combined entry
-                return [
-                    [
-                        task.name,
-                        Object.fromEntries(writes.filter(([chan]) => outputChannels.includes(chan))),
-                    ],
-                ];
-            }
-        });
-    }
-    const grouped = {};
-    for (const [node, value] of updated) {
-        if (!(node in grouped)) {
-            grouped[node] = [];
-        }
-        grouped[node].push(value);
-    }
-    const flattened = {};
-    for (const node in grouped) {
-        if (grouped[node].length === 1) {
-            const [write] = grouped[node];
-            flattened[node] = write;
-        }
-        else {
-            flattened[node] = grouped[node];
-        }
-    }
-    if (cached) {
-        flattened["__metadata__"] = { cached };
-    }
-    yield flattened;
-}
-export function single(iter) {
-    // eslint-disable-next-line no-unreachable-loop
-    for (const value of iter) {
-        return value;
-    }
-    return null;
+* Map pending writes (a sequence of tuples (channel, value)) to output chunk.
+* @internal
+*
+* @param outputChannels - The channels to output.
+* @param tasks - The tasks to output.
+* @param cached - Whether the output is cached.
+*
+* @returns A generator that yields the output chunk (if any).
+*/
+function* mapOutputUpdates(outputChannels, tasks, cached) {
+	const outputTasks = tasks.filter(([task, ww]) => {
+		return (task.config === void 0 || !task.config.tags?.includes(TAG_HIDDEN)) && ww[0][0] !== ERROR && ww[0][0] !== INTERRUPT;
+	});
+	if (!outputTasks.length) return;
+	let updated;
+	if (outputTasks.some(([task]) => task.writes.some(([chan, _]) => chan === RETURN))) updated = outputTasks.flatMap(([task]) => task.writes.filter(([chan, _]) => chan === RETURN).map(([_, value]) => [task.name, value]));
+	else if (!Array.isArray(outputChannels)) updated = outputTasks.flatMap(([task]) => task.writes.filter(([chan, _]) => chan === outputChannels).map(([_, value]) => [task.name, value]));
+	else updated = outputTasks.flatMap(([task]) => {
+		const { writes } = task;
+		const counts = {};
+		for (const [chan] of writes) if (outputChannels.includes(chan)) counts[chan] = (counts[chan] || 0) + 1;
+		if (Object.values(counts).some((count) => count > 1)) return writes.filter(([chan]) => outputChannels.includes(chan)).map(([chan, value]) => [task.name, { [chan]: value }]);
+		else return [[task.name, Object.fromEntries(writes.filter(([chan]) => outputChannels.includes(chan)))]];
+	});
+	const grouped = {};
+	for (const [node, value] of updated) {
+		if (!(node in grouped)) grouped[node] = [];
+		grouped[node].push(value);
+	}
+	const flattened = {};
+	for (const node in grouped) if (grouped[node].length === 1) {
+		const [write] = grouped[node];
+		flattened[node] = write;
+	} else flattened[node] = grouped[node];
+	if (cached) flattened["__metadata__"] = { cached };
+	yield flattened;
 }
+
+//#endregion
+export { mapCommand, mapInput, mapOutputUpdates, mapOutputValues, readChannel, readChannels };
 //# sourceMappingURL=io.js.map

package/dist/pregel/io.js.map

@@ -1 +1 @@
-{"version":3,"file":"io.js","sourceRoot":"","sources":["../../src/pregel/io.ts"],"names":[],"mappings":"AAOA,OAAO,EACL,OAAO,EACP,OAAO,EACP,KAAK,EACL,SAAS,EACT,YAAY,EACZ,MAAM,EACN,MAAM,EACN,UAAU,EACV,KAAK,GACN,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AACrE,OAAO,EAAE,MAAM,EAAE,MAAM,YAAY,CAAC;AAEpC,MAAM,UAAU,WAAW,CACzB,QAAgC,EAChC,IAAO,EACP,cAAuB,IAAI,EAC3B,kBAA2B,KAAK;IAEhC,IAAI,CAAC;QACH,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC;QAC5B,8DAA8D;IAChE,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,IAAI,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAC,iBAAiB,EAAE,CAAC;YACnD,IAAI,eAAe,EAAE,CAAC;gBACpB,OAAO,CAAC,CAAC;YACX,CAAC;iBAAM,IAAI,WAAW,EAAE,CAAC;gBACvB,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,MAAM,CAAC,CAAC;IACV,CAAC;AACH,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,QAAgC,EAChC,MAAoB,EACpB,YAAqB,IAAI;AACzB,8DAA8D;;IAE9D,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,8DAA8D;QAC9D,MAAM,MAAM,GAAG,EAAoB,CAAC;QACpC,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YACvB,IAAI,CAAC;gBACH,MAAM,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;gBACjD,8DAA8D;YAChE,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,IAAI,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAC,iBAAiB,EAAE,CAAC;oBACnD,SAAS;gBACX,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;SAAM,CAAC;QACN,OAAO,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,SAAS,CAAC,CAAC,UAAU,CACzB,GAAY,EACZ,aAAuC;IAEvC,IAAI,GAAG,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,IAAI,kBAAkB,CAAC,2BAA2B,CAAC,CAAC;IAC5D,CAAC;IACD,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;QACb,IAAI,KAAK,CAAC;QACV,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5B,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC;QACnB,CAAC;aAAM,CAAC;YACN,KAAK,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClB,MAAM,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YACpC,CAAC;iBAAM,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACpC,MAAM,CAAC,YAAY,EAAE,aAAa,IAAI,EAAE,EAAE,WAAW,CAAC,CAAC;YACzD,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CACb,iDAAiD,OAAO,IAAI,EAAE,CAC/D,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IACD,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;QACf,IACE,OAAO,GAAG,CAAC,MAAM,KAAK,QAAQ;YAC9B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,MAAM;YAC9B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,EACrC,CAAC;YACD,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBACvD,MAAM,QAAQ,GACZ,aAAa;qBACV,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,MAAM,CAAC;qBAC9C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;qBAChB,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBACvB,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACtB,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,YAAY,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IACD,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;QACf,IAAI,OAAO,GAAG,CAAC,MAAM,KAAK,QAAQ,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;YAClD,MAAM,IAAI,KAAK,CACb,yEAAyE,CAC1E,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9B,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;gBAChC,MAAM,CAAC,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;aAAM,CAAC;YACN,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAChD,MAAM,CAAC,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,SAAS,CAAC,CAAC,QAAQ,CACvB,aAA2B;AAC3B,8DAA8D;AAC9D,KAAW;AACX,8DAA8D;;IAE9D,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC1C,IACE,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC;YAC5B,OAAO,KAAK,KAAK,QAAQ;YACzB,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EACrB,CAAC;YACD,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;gBACtB,IAAI,aAAa,CAAC,QAAQ,CAAC,CAAM,CAAC,EAAE,CAAC;oBACnC,MAAM,CAAC,CAAM,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CACb,gEAAgE,CACjE,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,SAAS,CAAC,CAAC,eAAe,CAC9B,cAA4B,EAC5B,aAAgD,EAChD,QAAgC;AAChC,8DAA8D;;IAE9D,IAAI,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QAClC,IACE,aAAa,KAAK,IAAI;YACtB,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAChE,CAAC;YACD,MAAM,YAAY,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;SAAM,CAAC;QACN,IACE,aAAa,KAAK,IAAI;YACtB,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,cAAc,CAAC,EAC1D,CAAC;YACD,8DAA8D;YAC9D,MAAM,WAAW,CAAC,QAAQ,EAAE,cAAc,CAAQ,CAAC;QACrD,CAAC;IACH,CAAC;AACH,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,SAAS,CAAC,CAAC,gBAAgB,CAC/B,cAA4B,EAC5B,KAAiE,EACjE,MAAgB;IAEhB,MAAM,WAAW,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QAC9C,OAAO,CACL,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;YACtE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,KAAK;YAClB,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,SAAS,CACvB,CAAC;IACJ,CAAC,CAAC,CAAC;IACH,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;QACxB,OAAO;IACT,CAAC;IAED,IAAI,OAAuC,CAAC;IAE5C,IACE,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAC1B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,MAAM,CAAC,CACjD,EACD,CAAC;QACD,gGAAgG;QAChG,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CACvC,IAAI,CAAC,MAAM;aACR,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,MAAM,CAAC;aACtC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAiC,CAAC,CAC3E,CAAC;IACJ,CAAC;SAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QAC1C,oEAAoE;QACpE,6CAA6C;QAC7C,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CACvC,IAAI,CAAC,MAAM;aACR,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,cAAc,CAAC;aAC9C,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAiC,CAAC,CAC3E,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE;YACvC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;YACxB,MAAM,MAAM,GAAsB,EAAuB,CAAC;YAC1D,KAAK,MAAM,CAAC,IAAI,CAAC,IAAI,MAAM,EAAE,CAAC;gBAC5B,IAAI,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBAClC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;YAED,IAAK,MAAM,CAAC,MAAM,CAAC,MAAM,CAAc,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC;gBACnE,+DAA+D;gBAC/D,OAAO,MAAM;qBACV,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;qBACjD,GAAG,CACF,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,EAAE,CAChB,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAiC,CACjE,CAAC;YACN,CAAC;iBAAM,CAAC;gBACN,+DAA+D;gBAC/D,OAAO;oBACL;wBACE,IAAI,CAAC,IAAI;wBACT,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CACzD;qBAC8B;iBAClC,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,MAAM,OAAO,GAAG,EAA0B,CAAC;IAE3C,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,OAAO,EAAE,CAAC;QACpC,IAAI,CAAC,CAAC,IAAI,IAAI,OAAO,CAAC,EAAE,CAAC;YACvB,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;QACrB,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,MAAM,SAAS,GAAG,EAAwB,CAAC;IAC3C,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;QAC3B,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;YAC9B,SAAS,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC;QAC1B,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,IAAI,MAAM,EAAE,CAAC;QACX,SAAS,CAAC,cAAmB,CAAC,GAAG,EAAE,MAAM,EAAE,CAAC;IAC9C,CAAC;IACD,MAAM,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,MAAM,CAAI,IAAyB;IACjD,+CAA+C;IAC/C,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC"}
+{"version":3,"file":"io.js","names":["e: any","updated: [N, Record<string, unknown>][]","counts: Record<C, number>"],"sources":["../../src/pregel/io.ts"],"sourcesContent":["import type {\n  CheckpointPendingWrite,\n  PendingWrite,\n} from \"@langchain/langgraph-checkpoint\";\n\nimport type { BaseChannel } from \"../channels/base.js\";\nimport type { PregelExecutableTask } from \"./types.js\";\nimport {\n  _isSend,\n  Command,\n  ERROR,\n  INTERRUPT,\n  NULL_TASK_ID,\n  RESUME,\n  RETURN,\n  TAG_HIDDEN,\n  TASKS,\n} from \"../constants.js\";\nimport { EmptyChannelError, InvalidUpdateError } from \"../errors.js\";\nimport { isXXH3 } from \"../hash.js\";\n\nexport function readChannel<C extends PropertyKey>(\n  channels: Record<C, BaseChannel>,\n  chan: C,\n  catchErrors: boolean = true,\n  returnException: boolean = false\n): unknown | null {\n  try {\n    return channels[chan].get();\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  } catch (e: any) {\n    if (e.name === EmptyChannelError.unminifiable_name) {\n      if (returnException) {\n        return e;\n      } else if (catchErrors) {\n        return null;\n      }\n    }\n    throw e;\n  }\n}\n\nexport function readChannels<C extends PropertyKey>(\n  channels: Record<C, BaseChannel>,\n  select: C | Array<C>,\n  skipEmpty: boolean = true\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n): Record<string, any> | any {\n  if (Array.isArray(select)) {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const values = {} as Record<C, any>;\n    for (const k of select) {\n      try {\n        values[k] = readChannel(channels, k, !skipEmpty);\n        // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      } catch (e: any) {\n        if (e.name === EmptyChannelError.unminifiable_name) {\n          continue;\n        }\n      }\n    }\n    return values;\n  } else {\n    return readChannel(channels, select);\n  }\n}\n\n/**\n * Map input chunk to a sequence of pending writes in the form (channel, value).\n */\nexport function* mapCommand(\n  cmd: Command,\n  pendingWrites: CheckpointPendingWrite[]\n): Generator<[string, string, unknown]> {\n  if (cmd.graph === Command.PARENT) {\n    throw new InvalidUpdateError(\"There is no parent graph.\");\n  }\n  if (cmd.goto) {\n    let sends;\n    if (Array.isArray(cmd.goto)) {\n      sends = cmd.goto;\n    } else {\n      sends = [cmd.goto];\n    }\n    for (const send of sends) {\n      if (_isSend(send)) {\n        yield [NULL_TASK_ID, TASKS, send];\n      } else if (typeof send === \"string\") {\n        yield [NULL_TASK_ID, `branch:to:${send}`, \"__start__\"];\n      } else {\n        throw new Error(\n          `In Command.send, expected Send or string, got ${typeof send}`\n        );\n      }\n    }\n  }\n  if (cmd.resume) {\n    if (\n      typeof cmd.resume === \"object\" &&\n      Object.keys(cmd.resume).length &&\n      Object.keys(cmd.resume).every(isXXH3)\n    ) {\n      for (const [tid, resume] of Object.entries(cmd.resume)) {\n        const existing =\n          pendingWrites\n            .filter((w) => w[0] === tid && w[1] === RESUME)\n            .map((w) => w[2])\n            .slice(0, 1) ?? [];\n        existing.push(resume);\n        yield [tid, RESUME, existing];\n      }\n    } else {\n      yield [NULL_TASK_ID, RESUME, cmd.resume];\n    }\n  }\n  if (cmd.update) {\n    if (typeof cmd.update !== \"object\" || !cmd.update) {\n      throw new Error(\n        \"Expected cmd.update to be a dict mapping channel names to update values\"\n      );\n    }\n\n    if (Array.isArray(cmd.update)) {\n      for (const [k, v] of cmd.update) {\n        yield [NULL_TASK_ID, k, v];\n      }\n    } else {\n      for (const [k, v] of Object.entries(cmd.update)) {\n        yield [NULL_TASK_ID, k, v];\n      }\n    }\n  }\n}\n\n/**\n * Map input chunk to a sequence of pending writes in the form [channel, value].\n */\nexport function* mapInput<C extends PropertyKey>(\n  inputChannels: C | Array<C>,\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  chunk?: any\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n): Generator<[C, any]> {\n  if (chunk !== undefined && chunk !== null) {\n    if (\n      Array.isArray(inputChannels) &&\n      typeof chunk === \"object\" &&\n      !Array.isArray(chunk)\n    ) {\n      for (const k in chunk) {\n        if (inputChannels.includes(k as C)) {\n          yield [k as C, chunk[k]];\n        }\n      }\n    } else if (Array.isArray(inputChannels)) {\n      throw new Error(\n        `Input chunk must be an object when \"inputChannels\" is an array`\n      );\n    } else {\n      yield [inputChannels, chunk];\n    }\n  }\n}\n\n/**\n * Map pending writes (a sequence of tuples (channel, value)) to output chunk.\n */\nexport function* mapOutputValues<C extends PropertyKey>(\n  outputChannels: C | Array<C>,\n  pendingWrites: readonly PendingWrite<C>[] | true,\n  channels: Record<C, BaseChannel>\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n): Generator<Record<string, any>, any> {\n  if (Array.isArray(outputChannels)) {\n    if (\n      pendingWrites === true ||\n      pendingWrites.find(([chan, _]) => outputChannels.includes(chan))\n    ) {\n      yield readChannels(channels, outputChannels);\n    }\n  } else {\n    if (\n      pendingWrites === true ||\n      pendingWrites.some(([chan, _]) => chan === outputChannels)\n    ) {\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      yield readChannel(channels, outputChannels) as any;\n    }\n  }\n}\n\n/**\n * Map pending writes (a sequence of tuples (channel, value)) to output chunk.\n * @internal\n *\n * @param outputChannels - The channels to output.\n * @param tasks - The tasks to output.\n * @param cached - Whether the output is cached.\n *\n * @returns A generator that yields the output chunk (if any).\n */\nexport function* mapOutputUpdates<N extends PropertyKey, C extends PropertyKey>(\n  outputChannels: C | Array<C>,\n  tasks: readonly [PregelExecutableTask<N, C>, PendingWrite<C>[]][],\n  cached?: boolean\n): Generator<Record<N, Record<string, unknown> | unknown>> {\n  const outputTasks = tasks.filter(([task, ww]) => {\n    return (\n      (task.config === undefined || !task.config.tags?.includes(TAG_HIDDEN)) &&\n      ww[0][0] !== ERROR &&\n      ww[0][0] !== INTERRUPT\n    );\n  });\n  if (!outputTasks.length) {\n    return;\n  }\n\n  let updated: [N, Record<string, unknown>][];\n\n  if (\n    outputTasks.some(([task]) =>\n      task.writes.some(([chan, _]) => chan === RETURN)\n    )\n  ) {\n    // TODO: probably should assert that RETURN is the only \"non-special\" channel (starts with \"__\")\n    updated = outputTasks.flatMap(([task]) =>\n      task.writes\n        .filter(([chan, _]) => chan === RETURN)\n        .map(([_, value]) => [task.name, value] as [N, Record<string, unknown>])\n    );\n  } else if (!Array.isArray(outputChannels)) {\n    // special case where graph state is a single channel (MessageGraph)\n    // probably using this in functional API, too\n    updated = outputTasks.flatMap(([task]) =>\n      task.writes\n        .filter(([chan, _]) => chan === outputChannels)\n        .map(([_, value]) => [task.name, value] as [N, Record<string, unknown>])\n    );\n  } else {\n    updated = outputTasks.flatMap(([task]) => {\n      const { writes } = task;\n      const counts: Record<C, number> = {} as Record<C, number>;\n      for (const [chan] of writes) {\n        if (outputChannels.includes(chan)) {\n          counts[chan] = (counts[chan] || 0) + 1;\n        }\n      }\n\n      if ((Object.values(counts) as number[]).some((count) => count > 1)) {\n        // Multiple writes to the same channel: create separate entries\n        return writes\n          .filter(([chan]) => outputChannels.includes(chan))\n          .map(\n            ([chan, value]) =>\n              [task.name, { [chan]: value }] as [N, Record<string, unknown>]\n          );\n      } else {\n        // Single write to each channel: create a single combined entry\n        return [\n          [\n            task.name,\n            Object.fromEntries(\n              writes.filter(([chan]) => outputChannels.includes(chan))\n            ),\n          ] as [N, Record<string, unknown>],\n        ];\n      }\n    });\n  }\n\n  const grouped = {} as Record<N, unknown[]>;\n\n  for (const [node, value] of updated) {\n    if (!(node in grouped)) {\n      grouped[node] = [];\n    }\n    grouped[node].push(value);\n  }\n\n  const flattened = {} as Record<N, unknown>;\n  for (const node in grouped) {\n    if (grouped[node].length === 1) {\n      const [write] = grouped[node];\n      flattened[node] = write;\n    } else {\n      flattened[node] = grouped[node];\n    }\n  }\n\n  if (cached) {\n    flattened[\"__metadata__\" as N] = { cached };\n  }\n  yield flattened;\n}\n\nexport function single<T>(iter: IterableIterator<T>): T | null {\n  // eslint-disable-next-line no-unreachable-loop\n  for (const value of iter) {\n    return value;\n  }\n  return null;\n}\n"],"mappings":";;;;;AAqBA,SAAgB,YACd,UACA,MACA,cAAuB,MACvB,kBAA2B,OACX;AAChB,KAAI;AACF,SAAO,SAAS,MAAM;UAEfA,GAAQ;AACf,MAAI,EAAE,SAAS,kBAAkB,mBAC/B;OAAI,gBACF,QAAO;YACE,YACT,QAAO;;AAGX,QAAM;;;AAIV,SAAgB,aACd,UACA,QACA,YAAqB,MAEM;AAC3B,KAAI,MAAM,QAAQ,SAAS;EAEzB,MAAM,SAAS;AACf,OAAK,MAAM,KAAK,OACd,KAAI;AACF,UAAO,KAAK,YAAY,UAAU,GAAG,CAAC;WAE/BA,GAAQ;AACf,OAAI,EAAE,SAAS,kBAAkB,kBAC/B;;AAIN,SAAO;OAEP,QAAO,YAAY,UAAU;;;;;AAOjC,UAAiB,WACf,KACA,eACsC;AACtC,KAAI,IAAI,UAAU,QAAQ,OACxB,OAAM,IAAI,mBAAmB;AAE/B,KAAI,IAAI,MAAM;EACZ,IAAI;AACJ,MAAI,MAAM,QAAQ,IAAI,MACpB,SAAQ,IAAI;MAEZ,SAAQ,CAAC,IAAI;AAEf,OAAK,MAAM,QAAQ,MACjB,KAAI,QAAQ,MACV,OAAM;GAAC;GAAc;GAAO;;WACnB,OAAO,SAAS,SACzB,OAAM;GAAC;GAAc,aAAa;GAAQ;;MAE1C,OAAM,IAAI,MACR,iDAAiD,OAAO;;AAKhE,KAAI,IAAI,OACN,KACE,OAAO,IAAI,WAAW,YACtB,OAAO,KAAK,IAAI,QAAQ,UACxB,OAAO,KAAK,IAAI,QAAQ,MAAM,QAE9B,MAAK,MAAM,CAAC,KAAK,WAAW,OAAO,QAAQ,IAAI,SAAS;EACtD,MAAM,WACJ,cACG,QAAQ,MAAM,EAAE,OAAO,OAAO,EAAE,OAAO,QACvC,KAAK,MAAM,EAAE,IACb,MAAM,GAAG,MAAM;AACpB,WAAS,KAAK;AACd,QAAM;GAAC;GAAK;GAAQ;;;KAGtB,OAAM;EAAC;EAAc;EAAQ,IAAI;;AAGrC,KAAI,IAAI,QAAQ;AACd,MAAI,OAAO,IAAI,WAAW,YAAY,CAAC,IAAI,OACzC,OAAM,IAAI,MACR;AAIJ,MAAI,MAAM,QAAQ,IAAI,QACpB,MAAK,MAAM,CAAC,GAAG,MAAM,IAAI,OACvB,OAAM;GAAC;GAAc;GAAG;;MAG1B,MAAK,MAAM,CAAC,GAAG,MAAM,OAAO,QAAQ,IAAI,QACtC,OAAM;GAAC;GAAc;GAAG;;;;;;;AAShC,UAAiB,SACf,eAEA,OAEqB;AACrB,KAAI,UAAU,UAAa,UAAU,KACnC,KACE,MAAM,QAAQ,kBACd,OAAO,UAAU,YACjB,CAAC,MAAM,QAAQ,QAEf;OAAK,MAAM,KAAK,MACd,KAAI,cAAc,SAAS,GACzB,OAAM,CAAC,GAAQ,MAAM;YAGhB,MAAM,QAAQ,eACvB,OAAM,IAAI,MACR;KAGF,OAAM,CAAC,eAAe;;;;;AAQ5B,UAAiB,gBACf,gBACA,eACA,UAEqC;AACrC,KAAI,MAAM,QAAQ,iBAChB;MACE,kBAAkB,QAClB,cAAc,MAAM,CAAC,MAAM,OAAO,eAAe,SAAS,OAE1D,OAAM,aAAa,UAAU;YAI7B,kBAAkB,QAClB,cAAc,MAAM,CAAC,MAAM,OAAO,SAAS,gBAG3C,OAAM,YAAY,UAAU;;;;;;;;;;;;AAelC,UAAiB,iBACf,gBACA,OACA,QACyD;CACzD,MAAM,cAAc,MAAM,QAAQ,CAAC,MAAM,QAAQ;AAC/C,UACG,KAAK,WAAW,UAAa,CAAC,KAAK,OAAO,MAAM,SAAS,gBAC1D,GAAG,GAAG,OAAO,SACb,GAAG,GAAG,OAAO;;AAGjB,KAAI,CAAC,YAAY,OACf;CAGF,IAAIC;AAEJ,KACE,YAAY,MAAM,CAAC,UACjB,KAAK,OAAO,MAAM,CAAC,MAAM,OAAO,SAAS,SAI3C,WAAU,YAAY,SAAS,CAAC,UAC9B,KAAK,OACF,QAAQ,CAAC,MAAM,OAAO,SAAS,QAC/B,KAAK,CAAC,GAAG,WAAW,CAAC,KAAK,MAAM;UAE5B,CAAC,MAAM,QAAQ,gBAGxB,WAAU,YAAY,SAAS,CAAC,UAC9B,KAAK,OACF,QAAQ,CAAC,MAAM,OAAO,SAAS,gBAC/B,KAAK,CAAC,GAAG,WAAW,CAAC,KAAK,MAAM;KAGrC,WAAU,YAAY,SAAS,CAAC,UAAU;EACxC,MAAM,EAAE,WAAW;EACnB,MAAMC,SAA4B;AAClC,OAAK,MAAM,CAAC,SAAS,OACnB,KAAI,eAAe,SAAS,MAC1B,QAAO,SAAS,OAAO,SAAS,KAAK;AAIzC,MAAK,OAAO,OAAO,QAAqB,MAAM,UAAU,QAAQ,GAE9D,QAAO,OACJ,QAAQ,CAAC,UAAU,eAAe,SAAS,OAC3C,KACE,CAAC,MAAM,WACN,CAAC,KAAK,MAAM,GAAG,OAAO;MAI5B,QAAO,CACL,CACE,KAAK,MACL,OAAO,YACL,OAAO,QAAQ,CAAC,UAAU,eAAe,SAAS;;CAQ9D,MAAM,UAAU;AAEhB,MAAK,MAAM,CAAC,MAAM,UAAU,SAAS;AACnC,MAAI,EAAE,QAAQ,SACZ,SAAQ,QAAQ;AAElB,UAAQ,MAAM,KAAK;;CAGrB,MAAM,YAAY;AAClB,MAAK,MAAM,QAAQ,QACjB,KAAI,QAAQ,MAAM,WAAW,GAAG;EAC9B,MAAM,CAAC,SAAS,QAAQ;AACxB,YAAU,QAAQ;OAElB,WAAU,QAAQ,QAAQ;AAI9B,KAAI,OACF,WAAU,kBAAuB,EAAE;AAErC,OAAM"}