langchain 0.0.197-rc.0 → 0.0.197-rc.1

package/dist/agents/executor.js

@@ -1,8 +1,211 @@
 import { BaseChain } from "../chains/base.js";
 import { RunnableAgent, } from "./agent.js";
+import { CallbackManager, } from "../callbacks/manager.js";
 import { OutputParserException } from "../schema/output_parser.js";
-import { Tool, ToolInputParsingException, } from "../tools/base.js";
+import { ToolInputParsingException, Tool, } from "../tools/base.js";
 import { Runnable } from "../schema/runnable/base.js";
+import { Serializable } from "../load/serializable.js";
+export class AgentExecutorIterator extends Serializable {
+    get finalOutputs() {
+        return this._finalOutputs;
+    }
+    /** Intended to be used as a setter method, needs to be async. */
+    async setFinalOutputs(value) {
+        this._finalOutputs = undefined;
+        if (value) {
+            const preparedOutputs = await this.agentExecutor.prepOutputs(this.inputs, value, true);
+            this._finalOutputs = preparedOutputs;
+        }
+    }
+    get nameToToolMap() {
+        const toolMap = this.agentExecutor.tools.map((tool) => ({
+            [tool.name]: tool,
+        }));
+        return Object.assign({}, ...toolMap);
+    }
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "agents", "executor_iterator"]
+        });
+        Object.defineProperty(this, "agentExecutor", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "inputs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "callbacks", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "tags", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "metadata", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "runName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "_finalOutputs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "runManager", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "intermediateSteps", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
+        Object.defineProperty(this, "iterations", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 0
+        });
+        this.agentExecutor = fields.agentExecutor;
+        this.inputs = fields.inputs;
+        this.tags = fields.tags;
+        this.metadata = fields.metadata;
+        this.runName = fields.runName;
+        this.runManager = fields.runManager;
+    }
+    /**
+     * Reset the iterator to its initial state, clearing intermediate steps,
+     * iterations, and the final output.
+     */
+    reset() {
+        this.intermediateSteps = [];
+        this.iterations = 0;
+        this._finalOutputs = undefined;
+    }
+    updateIterations() {
+        this.iterations += 1;
+    }
+    async *streamIterator() {
+        this.reset();
+        // Loop to handle iteration
+        while (true) {
+            try {
+                if (this.iterations === 0) {
+                    await this.onFirstStep();
+                }
+                const result = await this._callNext();
+                yield result;
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            }
+            catch (e) {
+                if ("message" in e &&
+                    e.message.startsWith("Final outputs already reached: ")) {
+                    if (!this.finalOutputs) {
+                        throw e;
+                    }
+                    return this.finalOutputs;
+                }
+                if (this.runManager) {
+                    await this.runManager.handleChainError(e);
+                }
+                throw e;
+            }
+        }
+    }
+    /**
+     * Perform any necessary setup for the first step
+     * of the asynchronous iterator.
+     */
+    async onFirstStep() {
+        if (this.iterations === 0) {
+            const callbackManager = await CallbackManager.configure(this.callbacks, this.agentExecutor.callbacks, this.tags, this.agentExecutor.tags, this.metadata, this.agentExecutor.metadata, {
+                verbose: this.agentExecutor.verbose,
+            });
+            this.runManager = await callbackManager?.handleChainStart(this.agentExecutor.toJSON(), this.inputs, undefined, undefined, this.tags, this.metadata, this.runName);
+        }
+    }
+    /**
+     * Execute the next step in the chain using the
+     * AgentExecutor's _takeNextStep method.
+     */
+    async _executeNextStep(runManager) {
+        return this.agentExecutor._takeNextStep(this.nameToToolMap, this.inputs, this.intermediateSteps, runManager);
+    }
+    /**
+     * Process the output of the next step,
+     * handling AgentFinish and tool return cases.
+     */
+    async _processNextStepOutput(nextStepOutput, runManager) {
+        if ("returnValues" in nextStepOutput) {
+            const output = await this.agentExecutor._return(nextStepOutput, this.intermediateSteps, runManager);
+            if (this.runManager) {
+                await this.runManager.handleChainEnd(output);
+            }
+            await this.setFinalOutputs(output);
+            return output;
+        }
+        this.intermediateSteps = this.intermediateSteps.concat(nextStepOutput);
+        let output = {};
+        if (Array.isArray(nextStepOutput) && nextStepOutput.length === 1) {
+            const nextStep = nextStepOutput[0];
+            const toolReturn = await this.agentExecutor._getToolReturn(nextStep);
+            if (toolReturn) {
+                output = await this.agentExecutor._return(toolReturn, this.intermediateSteps, runManager);
+                if (this.runManager) {
+                    await this.runManager.handleChainEnd(output);
+                }
+                await this.setFinalOutputs(output);
+            }
+        }
+        output = { intermediateSteps: nextStepOutput };
+        return output;
+    }
+    async _stop() {
+        const output = await this.agentExecutor.agent.returnStoppedResponse(this.agentExecutor.earlyStoppingMethod, this.intermediateSteps, this.inputs);
+        const returnedOutput = await this.agentExecutor._return(output, this.intermediateSteps, this.runManager);
+        await this.setFinalOutputs(returnedOutput);
+        return returnedOutput;
+    }
+    async _callNext() {
+        // final output already reached: stopiteration (final output)
+        if (this.finalOutputs) {
+            throw new Error(`Final outputs already reached: ${JSON.stringify(this.finalOutputs, null, 2)}`);
+        }
+        // timeout/max iterations: stopiteration (stopped response)
+        if (!this.agentExecutor.shouldContinueGetter(this.iterations)) {
+            return this._stop();
+        }
+        const nextStepOutput = await this._executeNextStep(this.runManager);
+        const output = await this._processNextStepOutput(nextStepOutput, this.runManager);
+        this.updateIterations();
+        return output;
+    }
+}
 /**
  * Tool that just returns the query.
  * Used for exception tracking.
@@ -134,6 +337,9 @@ export class AgentExecutor extends BaseChain {
     static fromAgentAndTools(fields) {
         return new AgentExecutor(fields);
     }
+    get shouldContinueGetter() {
+        return this.shouldContinue.bind(this);
+    }
     /**
      * Method that checks if the agent execution should continue based on the
      * number of iterations.
@@ -253,6 +459,149 @@ export class AgentExecutor extends BaseChain {
         const finish = await this.agent.returnStoppedResponse(this.earlyStoppingMethod, steps, inputs);
         return getOutput(finish);
     }
+    async _takeNextStep(nameToolMap, inputs, intermediateSteps, runManager) {
+        let output;
+        try {
+            output = await this.agent.plan(intermediateSteps, inputs, runManager?.getChild());
+        }
+        catch (e) {
+            // eslint-disable-next-line no-instanceof/no-instanceof
+            if (e instanceof OutputParserException) {
+                let observation;
+                let text = e.message;
+                if (this.handleParsingErrors === true) {
+                    if (e.sendToLLM) {
+                        observation = e.observation;
+                        text = e.llmOutput ?? "";
+                    }
+                    else {
+                        observation = "Invalid or incomplete response";
+                    }
+                }
+                else if (typeof this.handleParsingErrors === "string") {
+                    observation = this.handleParsingErrors;
+                }
+                else if (typeof this.handleParsingErrors === "function") {
+                    observation = this.handleParsingErrors(e);
+                }
+                else {
+                    throw e;
+                }
+                output = {
+                    tool: "_Exception",
+                    toolInput: observation,
+                    log: text,
+                };
+            }
+            else {
+                throw e;
+            }
+        }
+        if ("returnValues" in output) {
+            return output;
+        }
+        let actions;
+        if (Array.isArray(output)) {
+            actions = output;
+        }
+        else {
+            actions = [output];
+        }
+        const result = [];
+        for (const agentAction of actions) {
+            let observation = "";
+            if (runManager) {
+                await runManager?.handleAgentAction(agentAction);
+            }
+            if (agentAction.tool in nameToolMap) {
+                const tool = nameToolMap[agentAction.tool];
+                try {
+                    observation = await tool.call(agentAction.toolInput, runManager?.getChild());
+                }
+                catch (e) {
+                    // eslint-disable-next-line no-instanceof/no-instanceof
+                    if (e instanceof ToolInputParsingException) {
+                        if (this.handleParsingErrors === true) {
+                            observation =
+                                "Invalid or incomplete tool input. Please try again.";
+                        }
+                        else if (typeof this.handleParsingErrors === "string") {
+                            observation = this.handleParsingErrors;
+                        }
+                        else if (typeof this.handleParsingErrors === "function") {
+                            observation = this.handleParsingErrors(e);
+                        }
+                        else {
+                            throw e;
+                        }
+                        observation = await new ExceptionTool().call(observation, runManager?.getChild());
+                    }
+                }
+            }
+            else {
+                observation = `${agentAction.tool} is not a valid tool, try another available tool: ${Object.keys(nameToolMap).join(", ")}`;
+            }
+            result.push({
+                action: agentAction,
+                observation,
+            });
+        }
+        return result;
+    }
+    async _return(output, intermediateSteps, runManager) {
+        if (runManager) {
+            await runManager.handleAgentEnd(output);
+        }
+        const finalOutput = output.returnValues;
+        if (this.returnIntermediateSteps) {
+            finalOutput.intermediateSteps = intermediateSteps;
+        }
+        return finalOutput;
+    }
+    async _getToolReturn(nextStepOutput) {
+        const { action, observation } = nextStepOutput;
+        const nameToolMap = Object.fromEntries(this.tools.map((t) => [t.name.toLowerCase(), t]));
+        const [returnValueKey = "output"] = this.agent.returnValues;
+        // Invalid tools won't be in the map, so we return False.
+        if (action.tool in nameToolMap) {
+            if (nameToolMap[action.tool].returnDirect) {
+                return {
+                    returnValues: { [returnValueKey]: observation },
+                    log: "",
+                };
+            }
+        }
+        return null;
+    }
+    _returnStoppedResponse(earlyStoppingMethod) {
+        if (earlyStoppingMethod === "force") {
+            return {
+                returnValues: {
+                    output: "Agent stopped due to iteration limit or time limit.",
+                },
+                log: "",
+            };
+        }
+        throw new Error(`Got unsupported early_stopping_method: ${earlyStoppingMethod}`);
+    }
+    async *_streamIterator(
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    inputs) {
+        const agentExecutorIterator = new AgentExecutorIterator({
+            inputs,
+            agentExecutor: this,
+            metadata: this.metadata,
+            tags: this.tags,
+            callbacks: this.callbacks,
+        });
+        const iterator = agentExecutorIterator.streamIterator();
+        for await (const step of iterator) {
+            if (!step) {
+                continue;
+            }
+            yield step;
+        }
+    }
     _chainType() {
         return "agent_executor";
     }

package/dist/agents/toolkits/conversational_retrieval/token_buffer_memory.d.ts

@@ -33,7 +33,7 @@ export declare class OpenAIAgentTokenBufferMemory extends BaseChatMemory {
      * Retrieves the messages from the chat history.
      * @returns Promise that resolves with the messages from the chat history.
      */
-    getMessages(): Promise<import("langchain-core/messages").BaseMessage[]>;
+    getMessages(): Promise<import("@langchain/core/messages").BaseMessage[]>;
     /**
      * Loads memory variables from the input values.
      * @param _values Input values.

package/dist/base_language/count_tokens.cjs

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getEmbeddingContextSize = exports.getModelContextSize = exports.calculateMaxTokens = void 0;
-var base_1 = require("langchain-core/language_models/base");
+var base_1 = require("@langchain/core/language_models/base");
 Object.defineProperty(exports, "calculateMaxTokens", { enumerable: true, get: function () { return base_1.calculateMaxTokens; } });
 Object.defineProperty(exports, "getModelContextSize", { enumerable: true, get: function () { return base_1.getModelContextSize; } });
 Object.defineProperty(exports, "getEmbeddingContextSize", { enumerable: true, get: function () { return base_1.getEmbeddingContextSize; } });

package/dist/base_language/count_tokens.d.ts

@@ -1 +1 @@
-export { calculateMaxTokens, getModelContextSize, getEmbeddingContextSize, } from "langchain-core/language_models/base";
+export { calculateMaxTokens, getModelContextSize, getEmbeddingContextSize, } from "@langchain/core/language_models/base";

package/dist/base_language/count_tokens.js

@@ -1 +1 @@
-export { calculateMaxTokens, getModelContextSize, getEmbeddingContextSize, } from "langchain-core/language_models/base";
+export { calculateMaxTokens, getModelContextSize, getEmbeddingContextSize, } from "@langchain/core/language_models/base";

package/dist/base_language/index.cjs

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getModelContextSize = exports.calculateMaxTokens = exports.BaseLanguageModel = exports.BaseLangChain = void 0;
-var base_1 = require("langchain-core/language_models/base");
+var base_1 = require("@langchain/core/language_models/base");
 Object.defineProperty(exports, "BaseLangChain", { enumerable: true, get: function () { return base_1.BaseLangChain; } });
 Object.defineProperty(exports, "BaseLanguageModel", { enumerable: true, get: function () { return base_1.BaseLanguageModel; } });
 /*

package/dist/base_language/index.d.ts

@@ -1,2 +1,2 @@
-export { type SerializedLLM, type BaseLangChainParams, BaseLangChain, type BaseLanguageModelParams, type BaseLanguageModelCallOptions, type BaseFunctionCallOptions, type BaseLanguageModelInput, BaseLanguageModel, } from "langchain-core/language_models/base";
+export { type SerializedLLM, type BaseLangChainParams, BaseLangChain, type BaseLanguageModelParams, type BaseLanguageModelCallOptions, type BaseFunctionCallOptions, type BaseLanguageModelInput, BaseLanguageModel, } from "@langchain/core/language_models/base";
 export { calculateMaxTokens, getModelContextSize } from "./count_tokens.js";

package/dist/base_language/index.js

@@ -1,4 +1,4 @@
-export { BaseLangChain, BaseLanguageModel, } from "langchain-core/language_models/base";
+export { BaseLangChain, BaseLanguageModel, } from "@langchain/core/language_models/base";
 /*
  * Export utility functions for token calculations:
  * - calculateMaxTokens: Calculate max tokens for a given model and prompt (the model context size - tokens in prompt).

package/dist/cache/base.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/caches"), exports);
+__exportStar(require("@langchain/core/caches"), exports);

package/dist/cache/base.d.ts

@@ -1 +1 @@
-export * from "langchain-core/caches";
+export * from "@langchain/core/caches";

package/dist/cache/base.js

@@ -1 +1 @@
-export * from "langchain-core/caches";
+export * from "@langchain/core/caches";

package/dist/cache/index.cjs

@@ -1,5 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.InMemoryCache = void 0;
-var caches_1 = require("langchain-core/caches");
+var caches_1 = require("@langchain/core/caches");
 Object.defineProperty(exports, "InMemoryCache", { enumerable: true, get: function () { return caches_1.InMemoryCache; } });

package/dist/cache/index.d.ts

@@ -1 +1 @@
-export { InMemoryCache } from "langchain-core/caches";
+export { InMemoryCache } from "@langchain/core/caches";

package/dist/cache/index.js

@@ -1 +1 @@
-export { InMemoryCache } from "langchain-core/caches";
+export { InMemoryCache } from "@langchain/core/caches";

package/dist/callbacks/base.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/callbacks/base"), exports);
+__exportStar(require("@langchain/core/callbacks/base"), exports);

package/dist/callbacks/base.d.ts

@@ -1 +1 @@
-export * from "langchain-core/callbacks/base";
+export * from "@langchain/core/callbacks/base";

package/dist/callbacks/base.js

@@ -1 +1 @@
-export * from "langchain-core/callbacks/base";
+export * from "@langchain/core/callbacks/base";

package/dist/callbacks/handlers/console.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/tracers/console"), exports);
+__exportStar(require("@langchain/core/tracers/console"), exports);

package/dist/callbacks/handlers/console.d.ts

@@ -1 +1 @@
-export * from "langchain-core/tracers/console";
+export * from "@langchain/core/tracers/console";

package/dist/callbacks/handlers/console.js

@@ -1 +1 @@
-export * from "langchain-core/tracers/console";
+export * from "@langchain/core/tracers/console";

package/dist/callbacks/handlers/initialize.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/tracers/initialize"), exports);
+__exportStar(require("@langchain/core/tracers/initialize"), exports);

package/dist/callbacks/handlers/initialize.d.ts

@@ -1 +1 @@
-export * from "langchain-core/tracers/initialize";
+export * from "@langchain/core/tracers/initialize";

package/dist/callbacks/handlers/initialize.js

@@ -1 +1 @@
-export * from "langchain-core/tracers/initialize";
+export * from "@langchain/core/tracers/initialize";

package/dist/callbacks/handlers/log_stream.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/tracers/log_stream"), exports);
+__exportStar(require("@langchain/core/tracers/log_stream"), exports);

package/dist/callbacks/handlers/log_stream.d.ts

@@ -1 +1 @@
-export * from "langchain-core/tracers/log_stream";
+export * from "@langchain/core/tracers/log_stream";

package/dist/callbacks/handlers/log_stream.js

@@ -1 +1 @@
-export * from "langchain-core/tracers/log_stream";
+export * from "@langchain/core/tracers/log_stream";

package/dist/callbacks/handlers/run_collector.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/tracers/run_collector"), exports);
+__exportStar(require("@langchain/core/tracers/run_collector"), exports);

package/dist/callbacks/handlers/run_collector.d.ts

@@ -1 +1 @@
-export * from "langchain-core/tracers/run_collector";
+export * from "@langchain/core/tracers/run_collector";

package/dist/callbacks/handlers/run_collector.js

@@ -1 +1 @@
-export * from "langchain-core/tracers/run_collector";
+export * from "@langchain/core/tracers/run_collector";

package/dist/callbacks/handlers/tracer.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/tracers/base"), exports);
+__exportStar(require("@langchain/core/tracers/base"), exports);

package/dist/callbacks/handlers/tracer.d.ts

@@ -1 +1 @@
-export * from "langchain-core/tracers/base";
+export * from "@langchain/core/tracers/base";

package/dist/callbacks/handlers/tracer.js

@@ -1 +1 @@
-export * from "langchain-core/tracers/base";
+export * from "@langchain/core/tracers/base";

package/dist/callbacks/handlers/tracer_langchain.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/tracers/tracer_langchain"), exports);
+__exportStar(require("@langchain/core/tracers/tracer_langchain"), exports);

package/dist/callbacks/handlers/tracer_langchain.d.ts

@@ -1 +1 @@
-export * from "langchain-core/tracers/tracer_langchain";
+export * from "@langchain/core/tracers/tracer_langchain";

package/dist/callbacks/handlers/tracer_langchain.js

@@ -1 +1 @@
-export * from "langchain-core/tracers/tracer_langchain";
+export * from "@langchain/core/tracers/tracer_langchain";

package/dist/callbacks/handlers/tracer_langchain_v1.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/tracers/tracer_langchain_v1"), exports);
+__exportStar(require("@langchain/core/tracers/tracer_langchain_v1"), exports);

package/dist/callbacks/handlers/tracer_langchain_v1.d.ts

@@ -1 +1 @@
-export * from "langchain-core/tracers/tracer_langchain_v1";
+export * from "@langchain/core/tracers/tracer_langchain_v1";

package/dist/callbacks/handlers/tracer_langchain_v1.js

@@ -1 +1 @@
-export * from "langchain-core/tracers/tracer_langchain_v1";
+export * from "@langchain/core/tracers/tracer_langchain_v1";

package/dist/callbacks/manager.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/callbacks/manager"), exports);
+__exportStar(require("@langchain/core/callbacks/manager"), exports);

package/dist/callbacks/manager.d.ts

@@ -1 +1 @@
-export * from "langchain-core/callbacks/manager";
+export * from "@langchain/core/callbacks/manager";

package/dist/callbacks/manager.js

@@ -1 +1 @@
-export * from "langchain-core/callbacks/manager";
+export * from "@langchain/core/callbacks/manager";

package/dist/callbacks/promises.cjs

@@ -14,4 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("langchain-core/callbacks/promises"), exports);
+__exportStar(require("@langchain/core/callbacks/promises"), exports);

package/dist/callbacks/promises.d.ts

@@ -1 +1 @@
-export * from "langchain-core/callbacks/promises";
+export * from "@langchain/core/callbacks/promises";

package/dist/callbacks/promises.js

@@ -1 +1 @@
-export * from "langchain-core/callbacks/promises";
+export * from "@langchain/core/callbacks/promises";

package/dist/chains/base.cjs

@@ -50,6 +50,22 @@ class BaseChain extends index_js_2.BaseLangChain {
     async invoke(input, config) {
         return this.call(input, config);
     }
+    _validateOutputs(outputs) {
+        const missingKeys = this.outputKeys.filter((k) => !(k in outputs));
+        if (missingKeys.length) {
+            throw new Error(`Missing output keys: ${missingKeys.join(", ")} from chain ${this._chainType()}`);
+        }
+    }
+    async prepOutputs(inputs, outputs, returnOnlyOutputs = false) {
+        this._validateOutputs(outputs);
+        if (this.memory) {
+            await this.memory.saveContext(inputs, outputs);
+        }
+        if (returnOnlyOutputs) {
+            return outputs;
+        }
+        return { ...inputs, ...outputs };
+    }
     /**
      * Return a json-like object representing this chain.
      */