@langchain/core 1.2.9 → 1.2.11

package/dist/runnables/base.js

@@ -85,8 +85,9 @@ var Runnable = class extends Serializable {
 	}
 	async batch(inputs, options, batchOptions) {
 		const configList = this._getOptionsList(options ?? {}, inputs.length);
+		const maxConcurrency = configList[0]?.maxConcurrency ?? batchOptions?.maxConcurrency;
 		const caller = new AsyncCaller({
-			maxConcurrency: configList[0]?.maxConcurrency ?? batchOptions?.maxConcurrency,
+			maxConcurrency,
 			onFailedAttempt: (e) => {
 				throw e;
 			}
@@ -159,7 +160,8 @@ var Runnable = class extends Serializable {
 		delete config.runId;
 		let output;
 		try {
-			output = await raceWithSignal(func.call(this, input, config, runManager), config.signal);
+			const promise = func.call(this, input, config, runManager);
+			output = await raceWithSignal(promise, config.signal);
 		} catch (e) {
 			await runManager?.handleChainError(e);
 			throw e;
@@ -186,7 +188,8 @@ var Runnable = class extends Serializable {
 		}));
 		let outputs;
 		try {
-			outputs = await raceWithSignal(func.call(this, inputs, optionsList, runManagers, batchOptions), optionsList?.[0]?.signal);
+			const promise = func.call(this, inputs, optionsList, runManagers, batchOptions);
+			outputs = await raceWithSignal(promise, optionsList?.[0]?.signal);
 		} catch (e) {
 			await Promise.all(runManagers.map((runManager) => runManager?.handleChainError(e)));
 			throw e;
@@ -213,12 +216,14 @@ var Runnable = class extends Serializable {
 		const outerThis = this;
 		async function* wrapInputForTracing() {
 			for await (const chunk of inputGenerator) {
-				if (finalInputSupported) if (finalInput === void 0) finalInput = chunk;
-				else try {
-					finalInput = outerThis._concatOutputChunks(finalInput, chunk);
-				} catch {
-					finalInput = void 0;
-					finalInputSupported = false;
+				if (finalInputSupported) {
+					if (finalInput === void 0) finalInput = chunk;
+					else try {
+						finalInput = outerThis._concatOutputChunks(finalInput, chunk);
+					} catch {
+						finalInput = void 0;
+						finalInputSupported = false;
+					}
 				}
 				yield chunk;
 			}
@@ -235,12 +240,14 @@ var Runnable = class extends Serializable {
 			if (streamLogHandler !== void 0 && runManager !== void 0) iterator = streamLogHandler.tapOutputIterable(runManager.runId, iterator);
 			for await (const chunk of iterator) {
 				yield chunk;
-				if (finalOutputSupported) if (finalOutput === void 0) finalOutput = chunk;
-				else try {
-					finalOutput = this._concatOutputChunks(finalOutput, chunk);
-				} catch {
-					finalOutput = void 0;
-					finalOutputSupported = false;
+				if (finalOutputSupported) {
+					if (finalOutput === void 0) finalOutput = chunk;
+					else try {
+						finalOutput = this._concatOutputChunks(finalOutput, chunk);
+					} catch {
+						finalOutput = void 0;
+						finalOutputSupported = false;
+					}
 				}
 			}
 		} catch (e) {
@@ -385,14 +392,15 @@ var Runnable = class extends Serializable {
 		async function consumeRunnableStream() {
 			let signal;
 			try {
-				if (config.signal) if ("any" in AbortSignal) signal = AbortSignal.any([abortController.signal, config.signal]);
-				else {
-					const composed = new AbortController();
-					config.signal.addEventListener("abort", () => composed.abort(), { once: true });
-					abortController.signal.addEventListener("abort", () => composed.abort(), { once: true });
-					signal = composed.signal;
-				}
-				else signal = abortController.signal;
+				if (config.signal) {
+					if ("any" in AbortSignal) signal = AbortSignal.any([abortController.signal, config.signal]);
+					else {
+						const composed = new AbortController();
+						config.signal.addEventListener("abort", () => composed.abort(), { once: true });
+						abortController.signal.addEventListener("abort", () => composed.abort(), { once: true });
+						signal = composed.signal;
+					}
+				} else signal = abortController.signal;
 				const runnableStream = await outerThis.stream(input, {
 					...config,
 					signal
@@ -462,9 +470,10 @@ var Runnable = class extends Serializable {
 				let eventType;
 				let data = {};
 				const logEntry = runLog.state.logs[path];
-				if (logEntry.end_time === void 0) if (logEntry.streamed_output.length > 0) eventType = "stream";
-				else eventType = "start";
-				else eventType = "end";
+				if (logEntry.end_time === void 0) {
+					if (logEntry.streamed_output.length > 0) eventType = "stream";
+					else eventType = "start";
+				} else eventType = "end";
 				if (eventType === "start") {
 					if (logEntry.inputs !== void 0) data.input = logEntry.inputs;
 				} else if (eventType === "end") {
@@ -931,7 +940,10 @@ var RunnableSequence = class RunnableSequence extends Runnable {
 		let finalOutput;
 		try {
 			const initialSteps = [this.first, ...this.middle];
-			for (let i = 0; i < initialSteps.length; i += 1) nextStepInput = await raceWithSignal(initialSteps[i].invoke(nextStepInput, patchConfig(config, { callbacks: runManager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${i + 1}`) })), config.signal);
+			for (let i = 0; i < initialSteps.length; i += 1) {
+				const promise = initialSteps[i].invoke(nextStepInput, patchConfig(config, { callbacks: runManager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${i + 1}`) }));
+				nextStepInput = await raceWithSignal(promise, config.signal);
+			}
 			if (config.signal?.aborted) throw getAbortSignalError(config.signal);
 			finalOutput = await this.last.invoke(nextStepInput, patchConfig(config, { callbacks: runManager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${this.steps.length}`) }));
 		} catch (e) {
@@ -951,10 +963,13 @@ var RunnableSequence = class RunnableSequence extends Runnable {
 		}));
 		let nextStepInputs = inputs;
 		try {
-			for (let i = 0; i < this.steps.length; i += 1) nextStepInputs = await raceWithSignal(this.steps[i].batch(nextStepInputs, runManagers.map((runManager, j) => {
-				const childRunManager = runManager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${i + 1}`);
-				return patchConfig(configList[j], { callbacks: childRunManager });
-			}), batchOptions), configList[0]?.signal);
+			for (let i = 0; i < this.steps.length; i += 1) {
+				const promise = this.steps[i].batch(nextStepInputs, runManagers.map((runManager, j) => {
+					const childRunManager = runManager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${i + 1}`);
+					return patchConfig(configList[j], { callbacks: childRunManager });
+				}), batchOptions);
+				nextStepInputs = await raceWithSignal(promise, configList[0]?.signal);
+			}
 		} catch (e) {
 			await Promise.all(runManagers.map((runManager) => runManager?.handleChainError(e)));
 			throw e;
@@ -986,12 +1001,14 @@ var RunnableSequence = class RunnableSequence extends Runnable {
 			for await (const chunk of finalGenerator) {
 				options?.signal?.throwIfAborted();
 				yield chunk;
-				if (concatSupported) if (finalOutput === void 0) finalOutput = chunk;
-				else try {
-					finalOutput = this._concatOutputChunks(finalOutput, chunk);
-				} catch {
-					finalOutput = void 0;
-					concatSupported = false;
+				if (concatSupported) {
+					if (finalOutput === void 0) finalOutput = chunk;
+					else try {
+						finalOutput = this._concatOutputChunks(finalOutput, chunk);
+					} catch {
+						finalOutput = void 0;
+						concatSupported = false;
+					}
 				}
 			}
 		} catch (e) {
@@ -1111,7 +1128,8 @@ var RunnableMap = class RunnableMap extends Runnable {
 			}))];
 		}));
 		while (tasks.size) {
-			const { key, result, gen } = await raceWithSignal(Promise.race(tasks.values()), options?.signal);
+			const promise = Promise.race(tasks.values());
+			const { key, result, gen } = await raceWithSignal(promise, options?.signal);
 			tasks.delete(key);
 			if (!result.done) {
 				yield { [key]: result.value };
@@ -1154,7 +1172,8 @@ var RunnableTraceable = class RunnableTraceable extends Runnable {
 	async invoke(input, options) {
 		const [config] = this._getOptionsList(options ?? {}, 1);
 		const callbacks = await getCallbackManagerForConfig(config);
-		return raceWithSignal(this.func(patchConfig(config, { callbacks }), input), config?.signal);
+		const promise = this.func(patchConfig(config, { callbacks }), input);
+		return raceWithSignal(promise, config?.signal);
 	}
 	async *_streamIterator(input, options) {
 		const [config] = this._getOptionsList(options ?? {}, 1);
@@ -1482,7 +1501,8 @@ var RunnableWithFallbacks = class extends Runnable {
 			config?.signal?.throwIfAborted();
 			const childConfig = patchConfig(otherConfigFields, { callbacks: runManager?.getChild() });
 			try {
-				stream = consumeAsyncIterableInContext(childConfig, await runnable.stream(input, childConfig));
+				const originalStream = await runnable.stream(input, childConfig);
+				stream = consumeAsyncIterableInContext(childConfig, originalStream);
 				break;
 			} catch (e) {
 				if (firstError === void 0) firstError = e;