@langchain/langgraph 1.2.2 → 1.2.4

package/dist/pregel/index.js

@@ -1,4 +1,4 @@
-import { CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, CONFIG_KEY_CHECKPOINTER, CONFIG_KEY_CHECKPOINT_NS, CONFIG_KEY_DURABILITY, CONFIG_KEY_NODE_FINISHED, CONFIG_KEY_READ, CONFIG_KEY_SEND, CONFIG_KEY_STREAM, CONFIG_KEY_TASK_ID, COPY, END, ERROR, INPUT, INTERRUPT, NULL_TASK_ID, PUSH, TASKS, isInterrupted } from "../constants.js";
+import { CONFIG_KEY_CHECKPOINTER, CONFIG_KEY_CHECKPOINT_NS, CONFIG_KEY_NODE_FINISHED, CONFIG_KEY_READ, CONFIG_KEY_SEND, CONFIG_KEY_STREAM, INPUT, INTERRUPT, NULL_TASK_ID, PUSH, TASKS, isInterrupted } from "../constants.js";
 import { GraphRecursionError, GraphValueError, InvalidUpdateError } from "../errors.js";
 import { createCheckpoint, emptyChannels, getOnlyChannels } from "../channels/base.js";
 import { ensureLangGraphConfig, getConfig, recastCheckpointNamespace } from "./utils/config.js";
@@ -19,7 +19,6 @@ import { Topic } from "../channels/topic.js";
 import { interrupt } from "../interrupt.js";
 import { SCHEDULED, compareChannelVersions, copyCheckpoint, emptyCheckpoint, uuid5 } from "@langchain/langgraph-checkpoint";
 import { Runnable, RunnableSequence, _coerceToRunnable, getCallbackManagerForConfig, mergeConfigs, patchConfig } from "@langchain/core/runnables";
-
 //#region src/pregel/index.ts
 /**
 * Utility class for working with channels in the Pregel system.
@@ -260,7 +259,7 @@ var Pregel = class extends PartialRunnable {
 		if (streamMode != null && !Array.isArray(streamMode)) streamMode = [streamMode];
 		this.nodes = fields.nodes;
 		this.channels = fields.channels;
-		if (TASKS in this.channels && "lc_graph_name" in this.channels[TASKS] && this.channels[TASKS].lc_graph_name !== "Topic") throw new Error(`Channel '${TASKS}' is reserved and cannot be used in the graph.`);
+		if ("__pregel_tasks" in this.channels && "lc_graph_name" in this.channels["__pregel_tasks"] && this.channels["__pregel_tasks"].lc_graph_name !== "Topic") throw new Error(`Channel '${TASKS}' is reserved and cannot be used in the graph.`);
 		else this.channels[TASKS] = new Topic({ accumulate: false });
 		this.autoValidate = fields.autoValidate ?? this.autoValidate;
 		this.streamMode = streamMode ?? this.streamMode;
@@ -392,7 +391,7 @@ var Pregel = class extends PartialRunnable {
 					if (recurse) {
 						let newNamespace = namespace;
 						if (namespace !== void 0) newNamespace = namespace.slice(name.length + 1);
-						for (const [subgraphName, subgraph] of graph.getSubgraphs(newNamespace, recurse)) yield [`${name}${CHECKPOINT_NAMESPACE_SEPARATOR}${subgraphName}`, subgraph];
+						for (const [subgraphName, subgraph] of graph.getSubgraphs(newNamespace, recurse)) yield [`${name}|${subgraphName}`, subgraph];
 					}
 				}
 			}
@@ -446,8 +445,8 @@ var Pregel = class extends PartialRunnable {
 		for (const task of nextTasks) {
 			const matchingSubgraph = subgraphs.find(([name]) => name === task.name);
 			if (!matchingSubgraph) continue;
-			let taskNs = `${String(task.name)}${CHECKPOINT_NAMESPACE_END}${task.id}`;
-			if (parentNamespace) taskNs = `${parentNamespace}${CHECKPOINT_NAMESPACE_SEPARATOR}${taskNs}`;
+			let taskNs = `${String(task.name)}:${task.id}`;
+			if (parentNamespace) taskNs = `${parentNamespace}|${taskNs}`;
 			if (subgraphCheckpointer === void 0) {
 				const config = { configurable: {
 					thread_id: saved.config.configurable?.thread_id,
@@ -468,8 +467,8 @@ var Pregel = class extends PartialRunnable {
 			const nextTaskById = Object.fromEntries(nextTasks.map((task) => [task.id, task]));
 			for (const [taskId, channel, value] of saved.pendingWrites) {
 				if ([
-					ERROR,
-					INTERRUPT,
+					"__error__",
+					"__interrupt__",
 					SCHEDULED
 				].includes(channel)) continue;
 				if (!(taskId in nextTaskById)) continue;
@@ -504,13 +503,12 @@ var Pregel = class extends PartialRunnable {
 	* @throws {GraphValueError} If no checkpointer is configured
 	*/
 	async getState(config, options) {
-		const checkpointer = config.configurable?.[CONFIG_KEY_CHECKPOINTER] ?? this.checkpointer;
+		const checkpointer = config.configurable?.["__pregel_checkpointer"] ?? this.checkpointer;
 		if (!checkpointer) throw new GraphValueError("No checkpointer set", { lc_error_code: "MISSING_CHECKPOINTER" });
 		const checkpointNamespace = config.configurable?.checkpoint_ns ?? "";
-		if (checkpointNamespace !== "" && config.configurable?.[CONFIG_KEY_CHECKPOINTER] === void 0) {
+		if (checkpointNamespace !== "" && config.configurable?.["__pregel_checkpointer"] === void 0) {
 			const recastNamespace = recastCheckpointNamespace(checkpointNamespace);
 			for await (const [name, subgraph] of this.getSubgraphsAsync(recastNamespace, true)) if (name === recastNamespace) return await subgraph.getState(patchConfigurable(config, { [CONFIG_KEY_CHECKPOINTER]: checkpointer }), { subgraphs: options?.subgraphs });
-			throw new Error(`Subgraph with namespace "${recastNamespace}" not found.`);
 		}
 		const mergedConfig = mergeConfigs(this.config, config);
 		const saved = await checkpointer.getTuple(config);
@@ -535,16 +533,15 @@ var Pregel = class extends PartialRunnable {
 	* @throws {Error} If no checkpointer is configured
 	*/
 	async *getStateHistory(config, options) {
-		const checkpointer = config.configurable?.[CONFIG_KEY_CHECKPOINTER] ?? this.checkpointer;
+		const checkpointer = config.configurable?.["__pregel_checkpointer"] ?? this.checkpointer;
 		if (!checkpointer) throw new GraphValueError("No checkpointer set", { lc_error_code: "MISSING_CHECKPOINTER" });
 		const checkpointNamespace = config.configurable?.checkpoint_ns ?? "";
-		if (checkpointNamespace !== "" && config.configurable?.[CONFIG_KEY_CHECKPOINTER] === void 0) {
+		if (checkpointNamespace !== "" && config.configurable?.["__pregel_checkpointer"] === void 0) {
 			const recastNamespace = recastCheckpointNamespace(checkpointNamespace);
 			for await (const [name, pregel] of this.getSubgraphsAsync(recastNamespace, true)) if (name === recastNamespace) {
 				yield* pregel.getStateHistory(patchConfigurable(config, { [CONFIG_KEY_CHECKPOINTER]: checkpointer }), options);
 				return;
 			}
-			throw new Error(`Subgraph with namespace "${recastNamespace}" not found.`);
 		}
 		const mergedConfig = mergeConfigs(this.config, config, { configurable: { checkpoint_ns: checkpointNamespace } });
 		for await (const checkpointTuple of checkpointer.list(mergedConfig, options)) yield this._prepareStateSnapshot({
@@ -569,12 +566,12 @@ var Pregel = class extends PartialRunnable {
 	* @throws {InvalidUpdateError} If the update cannot be attributed to a node or an update can be only applied in sequence.
 	*/
 	async bulkUpdateState(startConfig, supersteps) {
-		const checkpointer = startConfig.configurable?.[CONFIG_KEY_CHECKPOINTER] ?? this.checkpointer;
+		const checkpointer = startConfig.configurable?.["__pregel_checkpointer"] ?? this.checkpointer;
 		if (!checkpointer) throw new GraphValueError("No checkpointer set", { lc_error_code: "MISSING_CHECKPOINTER" });
 		if (supersteps.length === 0) throw new Error("No supersteps provided");
 		if (supersteps.some((s) => s.updates.length === 0)) throw new Error("No updates provided");
 		const checkpointNamespace = startConfig.configurable?.checkpoint_ns ?? "";
-		if (checkpointNamespace !== "" && startConfig.configurable?.[CONFIG_KEY_CHECKPOINTER] === void 0) {
+		if (checkpointNamespace !== "" && startConfig.configurable?.["__pregel_checkpointer"] === void 0) {
 			const recastNamespace = recastCheckpointNamespace(checkpointNamespace);
 			for await (const [, pregel] of this.getSubgraphsAsync(recastNamespace, true)) return await pregel.bulkUpdateState(patchConfigurable(startConfig, { [CONFIG_KEY_CHECKPOINTER]: checkpointer }), supersteps);
 			throw new Error(`Subgraph "${recastNamespace}" not found`);
@@ -604,7 +601,7 @@ var Pregel = class extends PartialRunnable {
 				}, {}), saved ? saved.metadata : void 0);
 			}
 			const channels = emptyChannels(this.channels, checkpoint);
-			if (values === null && asNode === END) {
+			if (values === null && asNode === "__end__") {
 				if (updates.length > 1) throw new InvalidUpdateError(`Cannot apply multiple updates when clearing state`);
 				if (saved) {
 					const nextTasks = _prepareNextTasks(checkpoint, saved.pendingWrites || [], this.nodes, channels, saved.config, true, {
@@ -620,8 +617,8 @@ var Pregel = class extends PartialRunnable {
 					}], checkpointer.getNextVersion.bind(checkpointer), this.triggerToNodes);
 					for (const [taskId, k, v] of saved.pendingWrites || []) {
 						if ([
-							ERROR,
-							INTERRUPT,
+							"__error__",
+							"__interrupt__",
 							SCHEDULED
 						].includes(k)) continue;
 						if (!(taskId in nextTasks)) continue;
@@ -636,7 +633,7 @@ var Pregel = class extends PartialRunnable {
 					parents: saved?.metadata?.parents ?? {}
 				}, getNewChannelVersions(checkpointPreviousVersions, checkpoint.channel_versions)), saved ? saved.metadata : void 0);
 			}
-			if (asNode === COPY) {
+			if (asNode === "__copy__") {
 				if (updates.length > 1) throw new InvalidUpdateError(`Cannot copy checkpoint with multiple updates`);
 				if (saved == null) throw new InvalidUpdateError(`Cannot copy a non-existent checkpoint`);
 				const isCopyWithUpdates = (values) => {
@@ -673,7 +670,7 @@ var Pregel = class extends PartialRunnable {
 				}
 				return patchCheckpointMap(nextConfig, saved.metadata);
 			}
-			if (asNode === INPUT) {
+			if (asNode === "__input__") {
 				if (updates.length > 1) throw new InvalidUpdateError(`Cannot apply multiple updates when updating as input`);
 				const inputWrites = await gatherIterator(mapInput(this.inputChannels, values));
 				if (inputWrites.length === 0) throw new InvalidUpdateError(`Received no input writes for ${JSON.stringify(this.inputChannels, null, 2)}`);
@@ -705,8 +702,8 @@ var Pregel = class extends PartialRunnable {
 				}], void 0, this.triggerToNodes);
 				for (const [tid, k, v] of saved.pendingWrites) {
 					if ([
-						ERROR,
-						INTERRUPT,
+						"__error__",
+						"__interrupt__",
 						SCHEDULED
 					].includes(k) || nextTasks[tid] === void 0) continue;
 					nextTasks[tid].writes.push([k, v]);
@@ -761,7 +758,7 @@ var Pregel = class extends PartialRunnable {
 					proc: writers.length > 1 ? RunnableSequence.from(writers, { omitSequenceTags: true }) : writers[0],
 					writes: [],
 					triggers: [INTERRUPT],
-					id: taskId ?? uuid5(INTERRUPT, checkpoint.id),
+					id: taskId ?? uuid5("__interrupt__", checkpoint.id),
 					writers: []
 				});
 			}
@@ -855,13 +852,13 @@ var Pregel = class extends PartialRunnable {
 			defaultStreamMode = Array.isArray(streamMode) ? streamMode : [streamMode];
 			streamModeSingle = typeof streamMode === "string";
 		} else {
-			if (config.configurable?.[CONFIG_KEY_TASK_ID] !== void 0) defaultStreamMode = ["values"];
+			if (config.configurable?.["__pregel_task_id"] !== void 0) defaultStreamMode = ["values"];
 			else defaultStreamMode = this.streamMode;
 			streamModeSingle = true;
 		}
 		let defaultCheckpointer;
 		if (this.checkpointer === false) defaultCheckpointer = void 0;
-		else if (config !== void 0 && config.configurable?.[CONFIG_KEY_CHECKPOINTER] !== void 0) defaultCheckpointer = config.configurable[CONFIG_KEY_CHECKPOINTER];
+		else if (config !== void 0 && config.configurable?.["__pregel_checkpointer"] !== void 0) defaultCheckpointer = config.configurable[CONFIG_KEY_CHECKPOINTER];
 		else if (this.checkpointer === true) throw new Error("checkpointer: true cannot be used for root graphs.");
 		else defaultCheckpointer = this.checkpointer;
 		const defaultStore = config.store ?? this.store;
@@ -872,7 +869,7 @@ var Pregel = class extends PartialRunnable {
 			if (config.checkpointDuring === false) return "exit";
 			return "async";
 		})();
-		const defaultDurability = config.durability ?? checkpointDuringDurability ?? config?.configurable?.[CONFIG_KEY_DURABILITY] ?? "async";
+		const defaultDurability = config.durability ?? checkpointDuringDurability ?? config?.configurable?.["__pregel_durability"] ?? "async";
 		return [
 			defaultDebug,
 			defaultStreamMode,
@@ -966,8 +963,8 @@ var Pregel = class extends PartialRunnable {
 		const stream = new IterableReadableWritableStream({ modes: new Set(streamMode) });
 		if (this.checkpointer === true) {
 			config.configurable ??= {};
-			const ns = config.configurable[CONFIG_KEY_CHECKPOINT_NS] ?? "";
-			config.configurable[CONFIG_KEY_CHECKPOINT_NS] = ns.split(CHECKPOINT_NAMESPACE_SEPARATOR).map((part) => part.split(CHECKPOINT_NAMESPACE_END)[0]).join(CHECKPOINT_NAMESPACE_SEPARATOR);
+			const ns = config.configurable["checkpoint_ns"] ?? "";
+			config.configurable[CONFIG_KEY_CHECKPOINT_NS] = ns.split("|").map((part) => part.split(":")[0]).join("|");
 		}
 		if (streamMode.includes("messages")) {
 			const messageStreamer = new StreamMessagesHandler((chunk) => stream.push(chunk));
@@ -993,7 +990,7 @@ var Pregel = class extends PartialRunnable {
 		}
 		config.writer ??= (chunk) => {
 			if (!streamMode.includes("custom")) return;
-			const ns = (getConfig()?.configurable?.[CONFIG_KEY_CHECKPOINT_NS])?.split(CHECKPOINT_NAMESPACE_SEPARATOR).slice(0, -1);
+			const ns = (getConfig()?.configurable?.[CONFIG_KEY_CHECKPOINT_NS])?.split("|").slice(0, -1);
 			stream.push([
 				ns ?? [],
 				"custom",
@@ -1168,7 +1165,7 @@ var Pregel = class extends PartialRunnable {
 		await this.cache?.clear([]);
 	}
 };
-
 //#endregion
 export { Channel, Pregel };
+
 //# sourceMappingURL=index.js.map