@langchain/langgraph 1.3.6 → 1.4.0

package/dist/pregel/call.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"call.cjs","names":["RunnableSequence","RunnableCallable","ChannelWrite","RETURN","PASSTHROUGH","TAG_HIDDEN","AsyncLocalStorageProviderSingleton","CONFIG_KEY_CALL"],"sources":["../../src/pregel/call.ts"],"sourcesContent":["import {\n  Runnable,\n  RunnableConfig,\n  RunnableSequence,\n} from \"@langchain/core/runnables\";\nimport { AsyncLocalStorageProviderSingleton } from \"@langchain/core/singletons\";\nimport { CONFIG_KEY_CALL, RETURN, TAG_HIDDEN } from \"../constants.js\";\nimport { ChannelWrite, PASSTHROUGH } from \"./write.js\";\nimport { CachePolicy, RetryPolicy } from \"./utils/index.js\";\nimport { RunnableCallable, type RunnableCallableArgs } from \"../utils.js\";\nimport { EntrypointFunc, EntrypointReturnT, TaskFunc } from \"../func/types.js\";\nimport { LangGraphRunnableConfig } from \"./runnable_types.js\";\n\n/**\n * Wraps a user function in a Runnable that writes the returned value to the RETURN channel.\n */\nexport function getRunnableForFunc<ArgsT extends unknown[], OutputT>(\n  name: string,\n  func: TaskFunc<ArgsT, OutputT>\n): Runnable<ArgsT, OutputT, LangGraphRunnableConfig> {\n  const run = new RunnableCallable<ArgsT, OutputT>({\n    func: (input: ArgsT) => func(...input),\n    name,\n    trace: false,\n    recurse: false,\n  } as RunnableCallableArgs);\n\n  return new RunnableSequence<ArgsT, OutputT>({\n    name,\n    first: run,\n    last: new ChannelWrite<OutputT>(\n      [{ channel: RETURN, value: PASSTHROUGH }],\n      [TAG_HIDDEN]\n    ),\n  });\n}\n\nexport function getRunnableForEntrypoint<InputT, OutputT>(\n  name: string,\n  func: EntrypointFunc<InputT, OutputT>\n): Runnable<InputT, EntrypointReturnT<OutputT>, LangGraphRunnableConfig> {\n  const run = new RunnableCallable<InputT, EntrypointReturnT<OutputT>>({\n    func: (input: InputT, config: LangGraphRunnableConfig) => {\n      return func(input, config);\n    },\n    name,\n    trace: false,\n    recurse: false,\n  });\n\n  return run;\n}\n\nexport type CallWrapperOptions<ArgsT extends unknown[], OutputT> = {\n  func: TaskFunc<ArgsT, OutputT>;\n  name: string;\n  retry?: RetryPolicy;\n  cache?: CachePolicy;\n};\n\nexport function call<ArgsT extends unknown[], OutputT>(\n  { func, name, cache, retry }: CallWrapperOptions<ArgsT, OutputT>,\n  ...args: ArgsT\n): Promise<OutputT> {\n  const config =\n    AsyncLocalStorageProviderSingleton.getRunnableConfig() as RunnableConfig;\n  if (typeof config.configurable?.[CONFIG_KEY_CALL] === \"function\") {\n    return config.configurable[CONFIG_KEY_CALL](func, name, args, {\n      retry,\n      cache,\n      callbacks: config.callbacks,\n    });\n  }\n  throw new Error(\n    \"Async local storage not initialized. Please call initializeAsyncLocalStorageSingleton() before using this function.\"\n  );\n}\n"],"mappings":";;;;;;;;;AAgBA,SAAgB,mBACd,MACA,MACmD;AAQnD,QAAO,IAAIA,0BAAAA,iBAAiC;EAC1C;EACA,OATU,IAAIC,cAAAA,iBAAiC;GAC/C,OAAO,UAAiB,KAAK,GAAG,MAAM;GACtC;GACA,OAAO;GACP,SAAS;GACV,CAAyB;EAKxB,MAAM,IAAIC,cAAAA,aACR,CAAC;GAAE,SAASC,kBAAAA;GAAQ,OAAOC,cAAAA;GAAa,CAAC,EACzC,CAACC,kBAAAA,WAAW,CACb;EACF,CAAC;;AAGJ,SAAgB,yBACd,MACA,MACuE;AAUvE,QATY,IAAIJ,cAAAA,iBAAqD;EACnE,OAAO,OAAe,WAAoC;AACxD,UAAO,KAAK,OAAO,OAAO;;EAE5B;EACA,OAAO;EACP,SAAS;EACV,CAAC;;AAYJ,SAAgB,KACd,EAAE,MAAM,MAAM,OAAO,SACrB,GAAG,MACe;CAClB,MAAM,SACJK,2BAAAA,mCAAmC,mBAAmB;AACxD,KAAI,OAAO,OAAO,eAAA,qBAAoC,WACpD,QAAO,OAAO,aAAaC,kBAAAA,iBAAiB,MAAM,MAAM,MAAM;EAC5D;EACA;EACA,WAAW,OAAO;EACnB,CAAC;AAEJ,OAAM,IAAI,MACR,sHACD"}
+{"version":3,"file":"call.cjs","names":["RunnableSequence","RunnableCallable","ChannelWrite","RETURN","PASSTHROUGH","TAG_HIDDEN","AsyncLocalStorageProviderSingleton","CONFIG_KEY_CALL"],"sources":["../../src/pregel/call.ts"],"sourcesContent":["import {\n  Runnable,\n  RunnableConfig,\n  RunnableSequence,\n} from \"@langchain/core/runnables\";\nimport { AsyncLocalStorageProviderSingleton } from \"@langchain/core/singletons\";\nimport { CONFIG_KEY_CALL, RETURN, TAG_HIDDEN } from \"../constants.js\";\nimport { ChannelWrite, PASSTHROUGH } from \"./write.js\";\nimport { CachePolicy, RetryPolicy, TimeoutPolicy } from \"./utils/index.js\";\nimport { RunnableCallable, type RunnableCallableArgs } from \"../utils.js\";\nimport { EntrypointFunc, EntrypointReturnT, TaskFunc } from \"../func/types.js\";\nimport { LangGraphRunnableConfig } from \"./runnable_types.js\";\n\n/**\n * Wraps a user function in a Runnable that writes the returned value to the RETURN channel.\n */\nexport function getRunnableForFunc<ArgsT extends unknown[], OutputT>(\n  name: string,\n  func: TaskFunc<ArgsT, OutputT>\n): Runnable<ArgsT, OutputT, LangGraphRunnableConfig> {\n  const run = new RunnableCallable<ArgsT, OutputT>({\n    func: (input: ArgsT) => func(...input),\n    name,\n    trace: false,\n    recurse: false,\n  } as RunnableCallableArgs);\n\n  return new RunnableSequence<ArgsT, OutputT>({\n    name,\n    first: run,\n    last: new ChannelWrite<OutputT>(\n      [{ channel: RETURN, value: PASSTHROUGH }],\n      [TAG_HIDDEN]\n    ),\n  });\n}\n\nexport function getRunnableForEntrypoint<InputT, OutputT>(\n  name: string,\n  func: EntrypointFunc<InputT, OutputT>\n): Runnable<InputT, EntrypointReturnT<OutputT>, LangGraphRunnableConfig> {\n  const run = new RunnableCallable<InputT, EntrypointReturnT<OutputT>>({\n    func: (input: InputT, config: LangGraphRunnableConfig) => {\n      return func(input, config);\n    },\n    name,\n    trace: false,\n    recurse: false,\n  });\n\n  return run;\n}\n\nexport type CallWrapperOptions<ArgsT extends unknown[], OutputT> = {\n  func: TaskFunc<ArgsT, OutputT>;\n  name: string;\n  retry?: RetryPolicy;\n  cache?: CachePolicy;\n  timeout?: TimeoutPolicy;\n};\n\nexport function call<ArgsT extends unknown[], OutputT>(\n  { func, name, cache, retry, timeout }: CallWrapperOptions<ArgsT, OutputT>,\n  ...args: ArgsT\n): Promise<OutputT> {\n  const config =\n    AsyncLocalStorageProviderSingleton.getRunnableConfig() as RunnableConfig;\n  if (typeof config.configurable?.[CONFIG_KEY_CALL] === \"function\") {\n    return config.configurable[CONFIG_KEY_CALL](func, name, args, {\n      retry,\n      cache,\n      timeout,\n      callbacks: config.callbacks,\n    });\n  }\n  throw new Error(\n    \"Async local storage not initialized. Please call initializeAsyncLocalStorageSingleton() before using this function.\"\n  );\n}\n"],"mappings":";;;;;;;;;AAgBA,SAAgB,mBACd,MACA,MACmD;AAQnD,QAAO,IAAIA,0BAAAA,iBAAiC;EAC1C;EACA,OATU,IAAIC,cAAAA,iBAAiC;GAC/C,OAAO,UAAiB,KAAK,GAAG,MAAM;GACtC;GACA,OAAO;GACP,SAAS;GACV,CAAyB;EAKxB,MAAM,IAAIC,cAAAA,aACR,CAAC;GAAE,SAASC,kBAAAA;GAAQ,OAAOC,cAAAA;GAAa,CAAC,EACzC,CAACC,kBAAAA,WAAW,CACb;EACF,CAAC;;AAGJ,SAAgB,yBACd,MACA,MACuE;AAUvE,QATY,IAAIJ,cAAAA,iBAAqD;EACnE,OAAO,OAAe,WAAoC;AACxD,UAAO,KAAK,OAAO,OAAO;;EAE5B;EACA,OAAO;EACP,SAAS;EACV,CAAC;;AAaJ,SAAgB,KACd,EAAE,MAAM,MAAM,OAAO,OAAO,WAC5B,GAAG,MACe;CAClB,MAAM,SACJK,2BAAAA,mCAAmC,mBAAmB;AACxD,KAAI,OAAO,OAAO,eAAA,qBAAoC,WACpD,QAAO,OAAO,aAAaC,kBAAAA,iBAAiB,MAAM,MAAM,MAAM;EAC5D;EACA;EACA;EACA,WAAW,OAAO;EACnB,CAAC;AAEJ,OAAM,IAAI,MACR,sHACD"}

package/dist/pregel/call.js

@@ -32,11 +32,12 @@ function getRunnableForEntrypoint(name, func) {
 		recurse: false
 	});
 }
-function call({ func, name, cache, retry }, ...args) {
+function call({ func, name, cache, retry, timeout }, ...args) {
 	const config = AsyncLocalStorageProviderSingleton.getRunnableConfig();
 	if (typeof config.configurable?.["__pregel_call"] === "function") return config.configurable[CONFIG_KEY_CALL](func, name, args, {
 		retry,
 		cache,
+		timeout,
 		callbacks: config.callbacks
 	});
 	throw new Error("Async local storage not initialized. Please call initializeAsyncLocalStorageSingleton() before using this function.");

package/dist/pregel/call.js.map

@@ -1 +1 @@
-{"version":3,"file":"call.js","names":[],"sources":["../../src/pregel/call.ts"],"sourcesContent":["import {\n  Runnable,\n  RunnableConfig,\n  RunnableSequence,\n} from \"@langchain/core/runnables\";\nimport { AsyncLocalStorageProviderSingleton } from \"@langchain/core/singletons\";\nimport { CONFIG_KEY_CALL, RETURN, TAG_HIDDEN } from \"../constants.js\";\nimport { ChannelWrite, PASSTHROUGH } from \"./write.js\";\nimport { CachePolicy, RetryPolicy } from \"./utils/index.js\";\nimport { RunnableCallable, type RunnableCallableArgs } from \"../utils.js\";\nimport { EntrypointFunc, EntrypointReturnT, TaskFunc } from \"../func/types.js\";\nimport { LangGraphRunnableConfig } from \"./runnable_types.js\";\n\n/**\n * Wraps a user function in a Runnable that writes the returned value to the RETURN channel.\n */\nexport function getRunnableForFunc<ArgsT extends unknown[], OutputT>(\n  name: string,\n  func: TaskFunc<ArgsT, OutputT>\n): Runnable<ArgsT, OutputT, LangGraphRunnableConfig> {\n  const run = new RunnableCallable<ArgsT, OutputT>({\n    func: (input: ArgsT) => func(...input),\n    name,\n    trace: false,\n    recurse: false,\n  } as RunnableCallableArgs);\n\n  return new RunnableSequence<ArgsT, OutputT>({\n    name,\n    first: run,\n    last: new ChannelWrite<OutputT>(\n      [{ channel: RETURN, value: PASSTHROUGH }],\n      [TAG_HIDDEN]\n    ),\n  });\n}\n\nexport function getRunnableForEntrypoint<InputT, OutputT>(\n  name: string,\n  func: EntrypointFunc<InputT, OutputT>\n): Runnable<InputT, EntrypointReturnT<OutputT>, LangGraphRunnableConfig> {\n  const run = new RunnableCallable<InputT, EntrypointReturnT<OutputT>>({\n    func: (input: InputT, config: LangGraphRunnableConfig) => {\n      return func(input, config);\n    },\n    name,\n    trace: false,\n    recurse: false,\n  });\n\n  return run;\n}\n\nexport type CallWrapperOptions<ArgsT extends unknown[], OutputT> = {\n  func: TaskFunc<ArgsT, OutputT>;\n  name: string;\n  retry?: RetryPolicy;\n  cache?: CachePolicy;\n};\n\nexport function call<ArgsT extends unknown[], OutputT>(\n  { func, name, cache, retry }: CallWrapperOptions<ArgsT, OutputT>,\n  ...args: ArgsT\n): Promise<OutputT> {\n  const config =\n    AsyncLocalStorageProviderSingleton.getRunnableConfig() as RunnableConfig;\n  if (typeof config.configurable?.[CONFIG_KEY_CALL] === \"function\") {\n    return config.configurable[CONFIG_KEY_CALL](func, name, args, {\n      retry,\n      cache,\n      callbacks: config.callbacks,\n    });\n  }\n  throw new Error(\n    \"Async local storage not initialized. Please call initializeAsyncLocalStorageSingleton() before using this function.\"\n  );\n}\n"],"mappings":";;;;;;;;;AAgBA,SAAgB,mBACd,MACA,MACmD;AAQnD,QAAO,IAAI,iBAAiC;EAC1C;EACA,OATU,IAAI,iBAAiC;GAC/C,OAAO,UAAiB,KAAK,GAAG,MAAM;GACtC;GACA,OAAO;GACP,SAAS;GACV,CAAyB;EAKxB,MAAM,IAAI,aACR,CAAC;GAAE,SAAS;GAAQ,OAAO;GAAa,CAAC,EACzC,CAAC,WAAW,CACb;EACF,CAAC;;AAGJ,SAAgB,yBACd,MACA,MACuE;AAUvE,QATY,IAAI,iBAAqD;EACnE,OAAO,OAAe,WAAoC;AACxD,UAAO,KAAK,OAAO,OAAO;;EAE5B;EACA,OAAO;EACP,SAAS;EACV,CAAC;;AAYJ,SAAgB,KACd,EAAE,MAAM,MAAM,OAAO,SACrB,GAAG,MACe;CAClB,MAAM,SACJ,mCAAmC,mBAAmB;AACxD,KAAI,OAAO,OAAO,eAAA,qBAAoC,WACpD,QAAO,OAAO,aAAa,iBAAiB,MAAM,MAAM,MAAM;EAC5D;EACA;EACA,WAAW,OAAO;EACnB,CAAC;AAEJ,OAAM,IAAI,MACR,sHACD"}
+{"version":3,"file":"call.js","names":[],"sources":["../../src/pregel/call.ts"],"sourcesContent":["import {\n  Runnable,\n  RunnableConfig,\n  RunnableSequence,\n} from \"@langchain/core/runnables\";\nimport { AsyncLocalStorageProviderSingleton } from \"@langchain/core/singletons\";\nimport { CONFIG_KEY_CALL, RETURN, TAG_HIDDEN } from \"../constants.js\";\nimport { ChannelWrite, PASSTHROUGH } from \"./write.js\";\nimport { CachePolicy, RetryPolicy, TimeoutPolicy } from \"./utils/index.js\";\nimport { RunnableCallable, type RunnableCallableArgs } from \"../utils.js\";\nimport { EntrypointFunc, EntrypointReturnT, TaskFunc } from \"../func/types.js\";\nimport { LangGraphRunnableConfig } from \"./runnable_types.js\";\n\n/**\n * Wraps a user function in a Runnable that writes the returned value to the RETURN channel.\n */\nexport function getRunnableForFunc<ArgsT extends unknown[], OutputT>(\n  name: string,\n  func: TaskFunc<ArgsT, OutputT>\n): Runnable<ArgsT, OutputT, LangGraphRunnableConfig> {\n  const run = new RunnableCallable<ArgsT, OutputT>({\n    func: (input: ArgsT) => func(...input),\n    name,\n    trace: false,\n    recurse: false,\n  } as RunnableCallableArgs);\n\n  return new RunnableSequence<ArgsT, OutputT>({\n    name,\n    first: run,\n    last: new ChannelWrite<OutputT>(\n      [{ channel: RETURN, value: PASSTHROUGH }],\n      [TAG_HIDDEN]\n    ),\n  });\n}\n\nexport function getRunnableForEntrypoint<InputT, OutputT>(\n  name: string,\n  func: EntrypointFunc<InputT, OutputT>\n): Runnable<InputT, EntrypointReturnT<OutputT>, LangGraphRunnableConfig> {\n  const run = new RunnableCallable<InputT, EntrypointReturnT<OutputT>>({\n    func: (input: InputT, config: LangGraphRunnableConfig) => {\n      return func(input, config);\n    },\n    name,\n    trace: false,\n    recurse: false,\n  });\n\n  return run;\n}\n\nexport type CallWrapperOptions<ArgsT extends unknown[], OutputT> = {\n  func: TaskFunc<ArgsT, OutputT>;\n  name: string;\n  retry?: RetryPolicy;\n  cache?: CachePolicy;\n  timeout?: TimeoutPolicy;\n};\n\nexport function call<ArgsT extends unknown[], OutputT>(\n  { func, name, cache, retry, timeout }: CallWrapperOptions<ArgsT, OutputT>,\n  ...args: ArgsT\n): Promise<OutputT> {\n  const config =\n    AsyncLocalStorageProviderSingleton.getRunnableConfig() as RunnableConfig;\n  if (typeof config.configurable?.[CONFIG_KEY_CALL] === \"function\") {\n    return config.configurable[CONFIG_KEY_CALL](func, name, args, {\n      retry,\n      cache,\n      timeout,\n      callbacks: config.callbacks,\n    });\n  }\n  throw new Error(\n    \"Async local storage not initialized. Please call initializeAsyncLocalStorageSingleton() before using this function.\"\n  );\n}\n"],"mappings":";;;;;;;;;AAgBA,SAAgB,mBACd,MACA,MACmD;AAQnD,QAAO,IAAI,iBAAiC;EAC1C;EACA,OATU,IAAI,iBAAiC;GAC/C,OAAO,UAAiB,KAAK,GAAG,MAAM;GACtC;GACA,OAAO;GACP,SAAS;GACV,CAAyB;EAKxB,MAAM,IAAI,aACR,CAAC;GAAE,SAAS;GAAQ,OAAO;GAAa,CAAC,EACzC,CAAC,WAAW,CACb;EACF,CAAC;;AAGJ,SAAgB,yBACd,MACA,MACuE;AAUvE,QATY,IAAI,iBAAqD;EACnE,OAAO,OAAe,WAAoC;AACxD,UAAO,KAAK,OAAO,OAAO;;EAE5B;EACA,OAAO;EACP,SAAS;EACV,CAAC;;AAaJ,SAAgB,KACd,EAAE,MAAM,MAAM,OAAO,OAAO,WAC5B,GAAG,MACe;CAClB,MAAM,SACJ,mCAAmC,mBAAmB;AACxD,KAAI,OAAO,OAAO,eAAA,qBAAoC,WACpD,QAAO,OAAO,aAAa,iBAAiB,MAAM,MAAM,MAAM;EAC5D;EACA;EACA;EACA,WAAW,OAAO;EACnB,CAAC;AAEJ,OAAM,IAAI,MACR,sHACD"}

package/dist/pregel/index.cjs

@@ -6,6 +6,7 @@ const require_config = require("./utils/config.cjs");
 const require_utils = require("../utils.cjs");
 const require_write = require("./write.cjs");
 const require_read = require("./read.cjs");
+const require_runtime = require("./runtime.cjs");
 const require_io = require("./io.cjs");
 const require_index = require("./utils/index.cjs");
 const require_algo = require("./algo.cjs");
@@ -439,7 +440,10 @@ var Pregel = class extends PartialRunnable {
 			config,
 			tasks: []
 		};
-		const channels = require_base.emptyChannels(this.channels, saved.checkpoint);
+		const channels = await require_base.channelsFromCheckpoint(this.channels, saved.checkpoint, {
+			saver: typeof this.checkpointer === "object" ? this.checkpointer : void 0,
+			config: saved.config ?? config
+		});
 		if (saved.pendingWrites?.length) {
 			const nullWrites = saved.pendingWrites.filter(([taskId, _]) => taskId === require_constants.NULL_TASK_ID).map(([_, channel, value]) => [String(channel), value]);
 			if (nullWrites.length > 0) require_algo._applyWrites(saved.checkpoint, channels, [{
@@ -519,7 +523,7 @@ var Pregel = class extends PartialRunnable {
 		const checkpointer = config.configurable?.["__pregel_checkpointer"] ?? this.checkpointer;
 		if (!checkpointer) throw new require_errors.GraphValueError("No checkpointer set", { lc_error_code: "MISSING_CHECKPOINTER" });
 		const checkpointNamespace = config.configurable?.checkpoint_ns ?? "";
-		if (checkpointNamespace !== "" && config.configurable?.["__pregel_checkpointer"] === void 0) {
+		if (checkpointNamespace !== "" && config.configurable?.["__pregel_read"] === void 0 && config.configurable?.["__pregel_checkpointer"] === void 0) {
 			const recastNamespace = require_config.recastCheckpointNamespace(checkpointNamespace);
 			for await (const [name, subgraph] of this.getSubgraphsAsync(recastNamespace, true)) if (name === recastNamespace) return await subgraph.getState(require_utils.patchConfigurable(config, { [require_constants.CONFIG_KEY_CHECKPOINTER]: checkpointer }), { subgraphs: options?.subgraphs });
 		}
@@ -613,7 +617,10 @@ var Pregel = class extends PartialRunnable {
 					parents: saved?.metadata?.parents ?? {}
 				}, {}), saved ? saved.metadata : void 0);
 			}
-			const channels = require_base.emptyChannels(this.channels, checkpoint);
+			const channels = await require_base.channelsFromCheckpoint(this.channels, checkpoint, {
+				saver: checkpointer,
+				config: saved?.config ?? checkpointConfig
+			});
 			if (values === null && asNode === "__end__") {
 				if (updates.length > 1) throw new require_errors.InvalidUpdateError(`Cannot apply multiple updates when clearing state`);
 				if (saved) {
@@ -1044,6 +1051,7 @@ var Pregel = class extends PartialRunnable {
 		};
 		config.interrupt ??= this.userInterrupt ?? require_interrupt.interrupt;
 		if (config.serverInfo == null) config.serverInfo = _buildServerInfo(config);
+		config.control ??= new require_runtime.RunControl();
 		const callbackManagerOptions = { tracerInheritableMetadata: _getTracingMetadataDefaults(config) };
 		const runManager = await (await _langchain_core_callbacks_manager.CallbackManager._configureSync(config?.callbacks, void 0, config?.tags, void 0, config?.metadata, void 0, callbackManagerOptions))?.handleChainStart(this.toJSON(), require_index._coerceToDict(input, "input"), runId, void 0, void 0, void 0, config?.runName ?? this.getName());
 		const channelSpecs = require_base.getOnlyChannels(this.channels);
@@ -1201,6 +1209,10 @@ var Pregel = class extends PartialRunnable {
 					signal: config.signal
 				});
 			}
+			if (loop.status === "draining") {
+				if (loop.control == null) throw new Error("Draining status requires run control");
+				throw new require_errors.GraphDrained(loop.control.drainReason ?? "shutdown");
+			}
 			if (loop.status === "out_of_steps") throw new require_errors.GraphRecursionError([
 				`Recursion limit of ${config.recursionLimit} reached`,
 				"without hitting a stop condition. You can increase the",