langchain 0.0.4 → 0.0.5

package/dist/cjs/vectorstores/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HNSWLib = void 0;
+var hnswlib_1 = require("./hnswlib");
+Object.defineProperty(exports, "HNSWLib", { enumerable: true, get: function () { return hnswlib_1.HNSWLib; } });
+//# sourceMappingURL=index.js.map

package/dist/cjs/vectorstores/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../vectorstores/index.ts"],"names":[],"mappings":";;;AAAA,qCAAoC;AAA3B,kGAAA,OAAO,OAAA"}

package/dist/cjs/vectorstores/tests/hnswlib.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/cjs/vectorstores/tests/hnswlib.test.js

@@ -0,0 +1,44 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const promises_1 = __importDefault(require("fs/promises"));
+const path_1 = __importDefault(require("path"));
+const os_1 = __importDefault(require("os"));
+const hnswlib_1 = require("../hnswlib");
+const embeddings_1 = require("../../embeddings");
+(0, globals_1.test)("Test HNSWLib.fromTexts", async () => {
+    var _a;
+    const vectorStore = await hnswlib_1.HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world"], [{ id: 2 }, { id: 1 }, { id: 3 }], new embeddings_1.OpenAIEmbeddings());
+    (0, globals_1.expect)((_a = vectorStore.index) === null || _a === void 0 ? void 0 : _a.getCurrentCount()).toBe(3);
+    const resultOne = await vectorStore.similaritySearch("hello world", 1);
+    const resultOneMetadatas = resultOne.map(({ metadata }) => (metadata));
+    (0, globals_1.expect)(resultOneMetadatas).toEqual([{ id: 2 }]);
+    const resultTwo = await vectorStore.similaritySearch("hello world", 3);
+    const resultTwoMetadatas = resultTwo.map(({ metadata }) => (metadata));
+    (0, globals_1.expect)(resultTwoMetadatas).toEqual([{ id: 2 }, { id: 3 }, { id: 1 }]);
+});
+(0, globals_1.test)("Test HNSWLib.load and HNSWLib.save", async () => {
+    var _a;
+    const vectorStore = await hnswlib_1.HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world"], [{ id: 2 }, { id: 1 }, { id: 3 }], new embeddings_1.OpenAIEmbeddings());
+    (0, globals_1.expect)((_a = vectorStore.index) === null || _a === void 0 ? void 0 : _a.getCurrentCount()).toBe(3);
+    const resultOne = await vectorStore.similaritySearch("hello world", 1);
+    const resultOneMetadatas = resultOne.map(({ metadata }) => (metadata));
+    (0, globals_1.expect)(resultOneMetadatas).toEqual([{ id: 2 }]);
+    const resultTwo = await vectorStore.similaritySearch("hello world", 3);
+    const resultTwoMetadatas = resultTwo.map(({ metadata }) => (metadata));
+    (0, globals_1.expect)(resultTwoMetadatas).toEqual([{ id: 2 }, { id: 3 }, { id: 1 }]);
+    const tempDirectory = await promises_1.default.mkdtemp(path_1.default.join(os_1.default.tmpdir(), "lcjs-"));
+    console.log(tempDirectory);
+    await vectorStore.save(tempDirectory);
+    const loadedVectorStore = await hnswlib_1.HNSWLib.load(tempDirectory, new embeddings_1.OpenAIEmbeddings());
+    const resultThree = await loadedVectorStore.similaritySearch("hello world", 1);
+    const resultThreeMetadatas = resultThree.map(({ metadata }) => (metadata));
+    (0, globals_1.expect)(resultThreeMetadatas).toEqual([{ id: 2 }]);
+    const resultFour = await loadedVectorStore.similaritySearch("hello world", 3);
+    const resultFourMetadatas = resultFour.map(({ metadata }) => (metadata));
+    (0, globals_1.expect)(resultFourMetadatas).toEqual([{ id: 2 }, { id: 3 }, { id: 1 }]);
+});
+//# sourceMappingURL=hnswlib.test.js.map

package/dist/cjs/vectorstores/tests/hnswlib.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hnswlib.test.js","sourceRoot":"","sources":["../../../../vectorstores/tests/hnswlib.test.ts"],"names":[],"mappings":";;;;;AAAA,2CAA6C;AAC7C,2DAA6B;AAC7B,gDAAwB;AACxB,4CAAoB;AAEpB,wCAAqC;AACrC,iDAAoD;AAEpD,IAAA,cAAI,EAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;;IACxC,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,CAAC,EAC9C,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACjC,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,IAAA,gBAAM,EAAC,MAAA,WAAW,CAAC,KAAK,0CAAE,eAAe,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAErD,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,kBAAkB,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACrE,IAAA,gBAAM,EAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEhD,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,kBAAkB,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACrE,IAAA,gBAAM,EAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;AACxE,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;;IACpD,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,CAAC,EAC9C,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACjC,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,IAAA,gBAAM,EAAC,MAAA,WAAW,CAAC,KAAK,0CAAE,eAAe,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAErD,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,kBAAkB,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACrE,IAAA,gBAAM,EAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEhD,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,kBAAkB,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACrE,IAAA,gBAAM,EAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEtE,MAAM,aAAa,GAAG,MAAM,kBAAE,CAAC,OAAO,CAAC,cAAI,CAAC,IAAI,CAAC,YAAE,CAAC,MAAM,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC;IAExE,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAE3B,MAAM,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAEtC,MAAM,iBAAiB,GAAG,MAAM,iBAAO,CAAC,IAAI,CAC1C,aAAa,EACb,IAAI,6BAAgB,EAAE,CACvB,CAAC;IAEF,MAAM,WAAW,GAAG,MAAM,iBAAiB,CAAC,gBAAgB,CAC1D,aAAa,EACb,CAAC,CACF,CAAC;IAEF,MAAM,oBAAoB,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACzE,IAAA,gBAAM,EAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAElD,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IAE9E,MAAM,mBAAmB,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACvE,IAAA,gBAAM,EAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;AACzE,CAAC,CAAC,CAAC"}

package/dist/esm/agents/agent.d.ts

@@ -4,7 +4,16 @@ import { BaseLLM } from "../llms";
 import { LLMChain } from "../chains/llm_chain";
 import { BasePromptTemplate } from "../prompt";
 export interface StaticAgent {
+    /**
+     * Create a prompt for this class
+     *
+     * @param tools - List of tools the agent will have access to, used to format the prompt.
+     * @param fields - Additional fields used to format the prompt.
+     *
+     * @returns A PromptTemplate assembled from the given tools and fields.
+     * */
     createPrompt(tools: Tool[], fields?: Record<string, any>): BasePromptTemplate;
+    /** Construct an agent from an LLM and a list of tools */
     fromLLMAndTools(llm: BaseLLM, tools: Tool[], args?: Record<string, any>): Agent;
     validateTools(_: Tool[]): void;
 }
@@ -14,26 +23,71 @@ export interface AgentInput {
     llmChain: LLMChain;
     allowedTools?: string[];
 }
+/**
+ * Class responsible for calling a language model and deciding an action.
+ *
+ * @remarks This is driven by an LLMChain. The prompt in the LLMChain *must*
+ * include a variable called "agent_scratchpad" where the agent can put its
+ * intermediary work.
+ */
 export declare abstract class Agent {
     llmChain: LLMChain;
     allowedTools?: string[];
     returnValues: string[];
     constructor(input: AgentInput);
+    /**
+     * Extract tool and tool input from LLM output.
+     */
     abstract extractToolAndInput(input: string): {
         tool: string;
         input: string;
     } | null;
+    /**
+     * Prefix to append the observation with.
+     */
     abstract observationPrefix(): string;
+    /**
+     * Prefix to append the LLM call with.
+     */
     abstract llmPrefix(): string;
+    /**
+     * Return the string type key uniquely identifying this class of agent.
+     */
     abstract _agentType(): string;
+    /**
+     * Prepare the agent for a new call, if needed
+     */
     prepareForNewCall(): void;
+    /**
+     * Validate that appropriate tools are passed in
+     */
     static validateTools(_: Tool[]): void;
     _stop(): string[];
+    /**
+     * Name of tool to use to terminate the chain.
+     */
     finishToolName(): string;
+    /**
+     * Construct a scratchpad to let the agent continue its thought process
+     */
     private constructScratchPad;
     private _plan;
+    /**
+     * Decide what to do given some input.
+     *
+     * @param steps - Steps the LLM has taken so far, along with observations from each.
+     * @param inputs - User inputs.
+     *
+     * @returns Action specifying what tool to use.
+     */
     plan(steps: AgentStep[], inputs: ChainValues): Promise<AgentAction | AgentFinish>;
+    /**
+     * Return response when agent has been stopped due to max iterations
+     */
     returnStoppedResponse(earlyStoppingMethod: StoppingMethod, steps: AgentStep[], inputs: ChainValues): Promise<AgentFinish>;
+    /**
+     * Load an agent from a json-like object describing it.
+     */
     static deserialize(data: SerializedAgent & {
         llm?: BaseLLM;
         tools?: Tool[];

package/dist/esm/agents/agent.js

@@ -1,12 +1,3 @@
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
 import { ZeroShotAgent, } from "./index";
 class ParseError extends Error {
     constructor(msg, output) {
@@ -21,6 +12,13 @@ class ParseError extends Error {
     }
 }
 export const staticImplements = (_) => { };
+/**
+ * Class responsible for calling a language model and deciding an action.
+ *
+ * @remarks This is driven by an LLMChain. The prompt in the LLMChain *must*
+ * include a variable called "agent_scratchpad" where the agent can put its
+ * intermediary work.
+ */
 export class Agent {
     constructor(input) {
         Object.defineProperty(this, "llmChain", {
@@ -44,15 +42,27 @@ export class Agent {
         this.llmChain = input.llmChain;
         this.allowedTools = input.allowedTools;
     }
+    /**
+     * Prepare the agent for a new call, if needed
+     */
     prepareForNewCall() { }
+    /**
+     * Validate that appropriate tools are passed in
+     */
     // eslint-disable-next-line no-unused-vars
     static validateTools(_) { }
     _stop() {
         return [`\n${this.observationPrefix()}`];
     }
+    /**
+     * Name of tool to use to terminate the chain.
+     */
     finishToolName() {
         return "Final Answer";
     }
+    /**
+     * Construct a scratchpad to let the agent continue its thought process
+     */
     constructScratchPad(steps) {
         return steps.reduce((thoughts, { action, observation }) => thoughts +
             [
@@ -61,64 +71,76 @@ export class Agent {
                 this.llmPrefix(),
             ].join("\n"), "");
     }
-    _plan(steps, inputs, suffix) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const thoughts = this.constructScratchPad(steps);
-            const newInputs = Object.assign(Object.assign({}, inputs), { agent_scratchpad: suffix ? `${thoughts}${suffix}` : thoughts, stop: this._stop() });
-            const output = yield this.llmChain.predict(newInputs);
-            const parsed = this.extractToolAndInput(output);
-            if (!parsed) {
-                throw new ParseError(`Invalid output: ${output}`, output);
-            }
-            const action = {
-                tool: parsed.tool,
-                toolInput: parsed.input,
-                log: output,
-            };
-            if (action.tool === this.finishToolName()) {
-                return { returnValues: { output: action.toolInput }, log: action.log };
-            }
-            return action;
-        });
+    async _plan(steps, inputs, suffix) {
+        const thoughts = this.constructScratchPad(steps);
+        const newInputs = {
+            ...inputs,
+            agent_scratchpad: suffix ? `${thoughts}${suffix}` : thoughts,
+            stop: this._stop(),
+        };
+        const output = await this.llmChain.predict(newInputs);
+        const parsed = this.extractToolAndInput(output);
+        if (!parsed) {
+            throw new ParseError(`Invalid output: ${output}`, output);
+        }
+        const action = {
+            tool: parsed.tool,
+            toolInput: parsed.input,
+            log: output,
+        };
+        if (action.tool === this.finishToolName()) {
+            return { returnValues: { output: action.toolInput }, log: action.log };
+        }
+        return action;
     }
+    /**
+     * Decide what to do given some input.
+     *
+     * @param steps - Steps the LLM has taken so far, along with observations from each.
+     * @param inputs - User inputs.
+     *
+     * @returns Action specifying what tool to use.
+     */
     plan(steps, inputs) {
         return this._plan(steps, inputs);
     }
-    returnStoppedResponse(earlyStoppingMethod, steps, inputs) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (earlyStoppingMethod === "force") {
-                return {
-                    returnValues: { output: "Agent stopped due to max iterations." },
-                    log: "",
-                };
-            }
-            if (earlyStoppingMethod === "generate") {
-                try {
-                    const action = yield this._plan(steps, inputs, "\n\nI now need to return a final answer based on the previous steps:");
-                    if ("returnValues" in action) {
-                        return action;
-                    }
-                    return { returnValues: { output: action.log }, log: action.log };
+    /**
+     * Return response when agent has been stopped due to max iterations
+     */
+    async returnStoppedResponse(earlyStoppingMethod, steps, inputs) {
+        if (earlyStoppingMethod === "force") {
+            return {
+                returnValues: { output: "Agent stopped due to max iterations." },
+                log: "",
+            };
+        }
+        if (earlyStoppingMethod === "generate") {
+            try {
+                const action = await this._plan(steps, inputs, "\n\nI now need to return a final answer based on the previous steps:");
+                if ("returnValues" in action) {
+                    return action;
                 }
-                catch (err) {
-                    if (!(err instanceof ParseError)) {
-                        throw err;
-                    }
-                    return { returnValues: { output: err.output }, log: err.output };
+                return { returnValues: { output: action.log }, log: action.log };
+            }
+            catch (err) {
+                if (!(err instanceof ParseError)) {
+                    throw err;
                 }
+                return { returnValues: { output: err.output }, log: err.output };
             }
-            throw new Error(`Invalid stopping method: ${earlyStoppingMethod}`);
-        });
+        }
+        throw new Error(`Invalid stopping method: ${earlyStoppingMethod}`);
     }
-    static deserialize(data) {
-        return __awaiter(this, void 0, void 0, function* () {
-            switch (data._type) {
-                case "zero-shot-react-description":
-                    return ZeroShotAgent.deserialize(data);
-                default:
-                    throw new Error("Unknown agent type");
-            }
-        });
+    /**
+     * Load an agent from a json-like object describing it.
+     */
+    static async deserialize(data) {
+        switch (data._type) {
+            case "zero-shot-react-description":
+                return ZeroShotAgent.deserialize(data);
+            default:
+                throw new Error("Unknown agent type");
+        }
     }
 }
 //# sourceMappingURL=agent.js.map

package/dist/esm/agents/agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../agents/agent.ts"],"names":[],"mappings":";;;;;;;;;AACA,OAAO,EACL,aAAa,GAOd,MAAM,SAAS,CAAC;AAKjB,MAAM,UAAW,SAAQ,KAAK;IAG5B,YAAY,GAAW,EAAE,MAAc;QACrC,KAAK,CAAC,GAAG,CAAC,CAAC;QAHb;;;;;WAAe;QAIb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAgBD,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAI,CAAI,EAAE,EAAE,GAAE,CAAC,CAAC;AAShD,MAAM,OAAgB,KAAK;IAOzB,YAAY,KAAiB;QAN7B;;;;;WAAmB;QAEnB;;;;mBAA0B,SAAS;WAAC;QAEpC;;;;mBAAe,CAAC,QAAQ,CAAC;WAAC;QAGxB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;IACzC,CAAC;IAYD,iBAAiB,KAAU,CAAC;IAE5B,0CAA0C;IAC1C,MAAM,CAAC,aAAa,CAAC,CAAS,IAAS,CAAC;IAExC,KAAK;QACH,OAAO,CAAC,KAAK,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,cAAc;QACZ,OAAO,cAAc,CAAC;IACxB,CAAC;IAEO,mBAAmB,CAAC,KAAkB;QAC5C,OAAO,KAAK,CAAC,MAAM,CACjB,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,QAAQ;YACR;gBACE,MAAM,CAAC,GAAG;gBACV,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,WAAW,EAAE;gBAC3C,IAAI,CAAC,SAAS,EAAE;aACjB,CAAC,IAAI,CAAC,IAAI,CAAC,EACd,EAAE,CACH,CAAC;IACJ,CAAC;IAEa,KAAK,CACjB,KAAkB,EAClB,MAAmB,EACnB,MAAe;;YAEf,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;YACjD,MAAM,SAAS,mCACV,MAAM,KACT,gBAAgB,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC,QAAQ,EAC5D,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,GACnB,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACtD,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,CAAC,MAAM,EAAE;gBACX,MAAM,IAAI,UAAU,CAAC,mBAAmB,MAAM,EAAE,EAAE,MAAM,CAAC,CAAC;aAC3D;YACD,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,SAAS,EAAE,MAAM,CAAC,KAAK;gBACvB,GAAG,EAAE,MAAM;aACZ,CAAC;YACF,IAAI,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,cAAc,EAAE,EAAE;gBACzC,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;aACxE;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;KAAA;IAED,IAAI,CACF,KAAkB,EAClB,MAAmB;QAEnB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACnC,CAAC;IAEK,qBAAqB,CACzB,mBAAmC,EACnC,KAAkB,EAClB,MAAmB;;YAEnB,IAAI,mBAAmB,KAAK,OAAO,EAAE;gBACnC,OAAO;oBACL,YAAY,EAAE,EAAE,MAAM,EAAE,sCAAsC,EAAE;oBAChE,GAAG,EAAE,EAAE;iBACR,CAAC;aACH;YAED,IAAI,mBAAmB,KAAK,UAAU,EAAE;gBACtC,IAAI;oBACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B,KAAK,EACL,MAAM,EACN,sEAAsE,CACvE,CAAC;oBACF,IAAI,cAAc,IAAI,MAAM,EAAE;wBAC5B,OAAO,MAAM,CAAC;qBACf;oBAED,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;iBAClE;gBAAC,OAAO,GAAG,EAAE;oBACZ,IAAI,CAAC,CAAC,GAAG,YAAY,UAAU,CAAC,EAAE;wBAChC,MAAM,GAAG,CAAC;qBACX;oBACD,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC;iBAClE;aACF;YAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,mBAAmB,EAAE,CAAC,CAAC;QACrE,CAAC;KAAA;IAED,MAAM,CAAO,WAAW,CACtB,IAAyD;;YAEzD,QAAQ,IAAI,CAAC,KAAK,EAAE;gBAClB,KAAK,6BAA6B;oBAChC,OAAO,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;gBACzC;oBACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;aACzC;QACH,CAAC;KAAA;CACF"}
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../agents/agent.ts"],"names":[],"mappings":"AACA,OAAO,EACL,aAAa,GAOd,MAAM,SAAS,CAAC;AAKjB,MAAM,UAAW,SAAQ,KAAK;IAG5B,YAAY,GAAW,EAAE,MAAc;QACrC,KAAK,CAAC,GAAG,CAAC,CAAC;QAHb;;;;;WAAe;QAIb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAyBD,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAI,CAAI,EAAE,EAAE,GAAE,CAAC,CAAC;AAShD;;;;;;GAMG;AACH,MAAM,OAAgB,KAAK;IAOzB,YAAY,KAAiB;QAN7B;;;;;WAAmB;QAEnB;;;;mBAA0B,SAAS;WAAC;QAEpC;;;;mBAAe,CAAC,QAAQ,CAAC;WAAC;QAGxB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;IACzC,CAAC;IAwBD;;OAEG;IACH,iBAAiB,KAAU,CAAC;IAE5B;;OAEG;IACH,0CAA0C;IAC1C,MAAM,CAAC,aAAa,CAAC,CAAS,IAAS,CAAC;IAExC,KAAK;QACH,OAAO,CAAC,KAAK,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,cAAc,CAAC;IACxB,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,KAAkB;QAC5C,OAAO,KAAK,CAAC,MAAM,CACjB,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,QAAQ;YACR;gBACE,MAAM,CAAC,GAAG;gBACV,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,WAAW,EAAE;gBAC3C,IAAI,CAAC,SAAS,EAAE;aACjB,CAAC,IAAI,CAAC,IAAI,CAAC,EACd,EAAE,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,KAAK,CACjB,KAAkB,EAClB,MAAmB,EACnB,MAAe;QAEf,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,SAAS,GAAgB;YAC7B,GAAG,MAAM;YACT,gBAAgB,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC,QAAQ;YAC5D,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE;SACnB,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,UAAU,CAAC,mBAAmB,MAAM,EAAE,EAAE,MAAM,CAAC,CAAC;SAC3D;QACD,MAAM,MAAM,GAAG;YACb,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,SAAS,EAAE,MAAM,CAAC,KAAK;YACvB,GAAG,EAAE,MAAM;SACZ,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,cAAc,EAAE,EAAE;YACzC,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;SACxE;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;OAOG;IACH,IAAI,CACF,KAAkB,EAClB,MAAmB;QAEnB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,qBAAqB,CACzB,mBAAmC,EACnC,KAAkB,EAClB,MAAmB;QAEnB,IAAI,mBAAmB,KAAK,OAAO,EAAE;YACnC,OAAO;gBACL,YAAY,EAAE,EAAE,MAAM,EAAE,sCAAsC,EAAE;gBAChE,GAAG,EAAE,EAAE;aACR,CAAC;SACH;QAED,IAAI,mBAAmB,KAAK,UAAU,EAAE;YACtC,IAAI;gBACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B,KAAK,EACL,MAAM,EACN,sEAAsE,CACvE,CAAC;gBACF,IAAI,cAAc,IAAI,MAAM,EAAE;oBAC5B,OAAO,MAAM,CAAC;iBACf;gBAED,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;aAClE;YAAC,OAAO,GAAG,EAAE;gBACZ,IAAI,CAAC,CAAC,GAAG,YAAY,UAAU,CAAC,EAAE;oBAChC,MAAM,GAAG,CAAC;iBACX;gBACD,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC;aAClE;SACF;QAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,mBAAmB,EAAE,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,IAAyD;QAEzD,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,6BAA6B;gBAChC,OAAO,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACzC;gBACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;SACzC;IACH,CAAC;CACF"}

package/dist/esm/agents/executor.d.ts

@@ -8,6 +8,10 @@ type AgentExecutorInput = {
     maxIterations?: number;
     earlyStoppingMethod?: StoppingMethod;
 };
+/**
+ * A chain managing an agent using tools.
+ * @augments BaseChain
+ */
 export declare class AgentExecutor extends BaseChain {
     agent: Agent;
     tools: Tool[];
@@ -15,6 +19,7 @@ export declare class AgentExecutor extends BaseChain {
     maxIterations?: number;
     earlyStoppingMethod: StoppingMethod;
     constructor(input: AgentExecutorInput);
+    /** Create from agent and a list of tools. */
     static fromAgentAndTools(fields: {
         agent: Agent;
         tools: Tool[];

package/dist/esm/agents/executor.js

@@ -1,13 +1,8 @@
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
 import { BaseChain } from "../chains";
+/**
+ * A chain managing an agent using tools.
+ * @augments BaseChain
+ */
 export class AgentExecutor extends BaseChain {
     constructor(input) {
         var _a, _b, _c;
@@ -50,46 +45,45 @@ export class AgentExecutor extends BaseChain {
         this.earlyStoppingMethod =
             (_c = input.earlyStoppingMethod) !== null && _c !== void 0 ? _c : this.earlyStoppingMethod;
     }
+    /** Create from agent and a list of tools. */
     static fromAgentAndTools(fields) {
         return new AgentExecutor(fields);
     }
     shouldContinue(iterations) {
         return this.maxIterations === undefined || iterations < this.maxIterations;
     }
-    _call(inputs) {
-        return __awaiter(this, void 0, void 0, function* () {
-            this.agent.prepareForNewCall();
-            const toolsByName = Object.fromEntries(this.tools.map((t) => [t.name, t]));
-            const steps = [];
-            let iterations = 0;
-            const getOutput = (finishStep) => {
-                const { returnValues } = finishStep;
-                if (this.returnIntermediateSteps) {
-                    return Object.assign(Object.assign({}, returnValues), { intermediateSteps: steps });
-                }
-                return returnValues;
-            };
-            while (this.shouldContinue(iterations)) {
-                const action = yield this.agent.plan(steps, inputs);
-                if ("returnValues" in action) {
-                    return getOutput(action);
-                }
-                const tool = toolsByName[action.tool];
-                const observation = tool
-                    ? yield tool.call(action.toolInput)
-                    : `${action.tool} is not a valid tool, try another one.`;
-                steps.push({ action, observation });
-                if (tool === null || tool === void 0 ? void 0 : tool.returnDirect) {
-                    return getOutput({
-                        returnValues: { [this.agent.returnValues[0]]: observation },
-                        log: "",
-                    });
-                }
-                iterations += 1;
+    async _call(inputs) {
+        this.agent.prepareForNewCall();
+        const toolsByName = Object.fromEntries(this.tools.map((t) => [t.name.toLowerCase(), t]));
+        const steps = [];
+        let iterations = 0;
+        const getOutput = (finishStep) => {
+            const { returnValues } = finishStep;
+            if (this.returnIntermediateSteps) {
+                return { ...returnValues, intermediateSteps: steps };
             }
-            const finish = yield this.agent.returnStoppedResponse(this.earlyStoppingMethod, steps, inputs);
-            return getOutput(finish);
-        });
+            return returnValues;
+        };
+        while (this.shouldContinue(iterations)) {
+            const action = await this.agent.plan(steps, inputs);
+            if ("returnValues" in action) {
+                return getOutput(action);
+            }
+            const tool = toolsByName[action.tool.toLowerCase()];
+            const observation = tool
+                ? await tool.call(action.toolInput)
+                : `${action.tool} is not a valid tool, try another one.`;
+            steps.push({ action, observation });
+            if (tool === null || tool === void 0 ? void 0 : tool.returnDirect) {
+                return getOutput({
+                    returnValues: { [this.agent.returnValues[0]]: observation },
+                    log: "",
+                });
+            }
+            iterations += 1;
+        }
+        const finish = await this.agent.returnStoppedResponse(this.earlyStoppingMethod, steps, inputs);
+        return getOutput(finish);
     }
     _chainType() {
         return "agent_executor";

package/dist/esm/agents/executor.js.map

@@ -1 +1 @@
-{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../../agents/executor.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,EAAe,SAAS,EAAE,MAAM,WAAW,CAAC;AAYnD,MAAM,OAAO,aAAc,SAAQ,SAAS;IAW1C,YAAY,KAAyB;;QACnC,KAAK,EAAE,CAAC;QAXV;;;;;WAAa;QAEb;;;;;WAAc;QAEd;;;;mBAA0B,KAAK;WAAC;QAEhC;;;;mBAAyB,EAAE;WAAC;QAE5B;;;;mBAAsC,OAAO;WAAC;QAI5C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QACzB,IAAI,CAAC,uBAAuB;YAC1B,MAAA,KAAK,CAAC,uBAAuB,mCAAI,IAAI,CAAC,uBAAuB,CAAC;QAChE,IAAI,CAAC,aAAa,GAAG,MAAA,KAAK,CAAC,aAAa,mCAAI,IAAI,CAAC,aAAa,CAAC;QAC/D,IAAI,CAAC,mBAAmB;YACtB,MAAA,KAAK,CAAC,mBAAmB,mCAAI,IAAI,CAAC,mBAAmB,CAAC;IAC1D,CAAC;IAED,MAAM,CAAC,iBAAiB,CACtB,MAIuB;QAEvB,OAAO,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAEO,cAAc,CAAC,UAAkB;QACvC,OAAO,IAAI,CAAC,aAAa,KAAK,SAAS,IAAI,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC;IAC7E,CAAC;IAEK,KAAK,CAAC,MAAmB;;YAC7B,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;YAC/B,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3E,MAAM,KAAK,GAAgB,EAAE,CAAC;YAC9B,IAAI,UAAU,GAAG,CAAC,CAAC;YAEnB,MAAM,SAAS,GAAG,CAAC,UAAuB,EAAE,EAAE;gBAC5C,MAAM,EAAE,YAAY,EAAE,GAAG,UAAU,CAAC;gBACpC,IAAI,IAAI,CAAC,uBAAuB,EAAE;oBAChC,uCAAY,YAAY,KAAE,iBAAiB,EAAE,KAAK,IAAG;iBACtD;gBACD,OAAO,YAAY,CAAC;YACtB,CAAC,CAAC;YAEF,OAAO,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE;gBACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBACpD,IAAI,cAAc,IAAI,MAAM,EAAE;oBAC5B,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC;iBAC1B;gBAED,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACtC,MAAM,WAAW,GAAG,IAAI;oBACtB,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;oBACnC,CAAC,CAAC,GAAG,MAAM,CAAC,IAAI,wCAAwC,CAAC;gBAC3D,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;gBACpC,IAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,YAAY,EAAE;oBACtB,OAAO,SAAS,CAAC;wBACf,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE;wBAC3D,GAAG,EAAE,EAAE;qBACR,CAAC,CAAC;iBACJ;gBACD,UAAU,IAAI,CAAC,CAAC;aACjB;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,qBAAqB,CACnD,IAAI,CAAC,mBAAmB,EACxB,KAAK,EACL,MAAM,CACP,CAAC;YAEF,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;KAAA;IAED,UAAU;QACR,OAAO,gBAAyB,CAAC;IACnC,CAAC;IAED,SAAS;QACP,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;CACF"}
+{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../../agents/executor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,SAAS,EAAE,MAAM,WAAW,CAAC;AAYnD;;;GAGG;AACH,MAAM,OAAO,aAAc,SAAQ,SAAS;IAW1C,YAAY,KAAyB;;QACnC,KAAK,EAAE,CAAC;QAXV;;;;;WAAa;QAEb;;;;;WAAc;QAEd;;;;mBAA0B,KAAK;WAAC;QAEhC;;;;mBAAyB,EAAE;WAAC;QAE5B;;;;mBAAsC,OAAO;WAAC;QAI5C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QACzB,IAAI,CAAC,uBAAuB;YAC1B,MAAA,KAAK,CAAC,uBAAuB,mCAAI,IAAI,CAAC,uBAAuB,CAAC;QAChE,IAAI,CAAC,aAAa,GAAG,MAAA,KAAK,CAAC,aAAa,mCAAI,IAAI,CAAC,aAAa,CAAC;QAC/D,IAAI,CAAC,mBAAmB;YACtB,MAAA,KAAK,CAAC,mBAAmB,mCAAI,IAAI,CAAC,mBAAmB,CAAC;IAC1D,CAAC;IAED,6CAA6C;IAC7C,MAAM,CAAC,iBAAiB,CACtB,MAIuB;QAEvB,OAAO,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAEO,cAAc,CAAC,UAAkB;QACvC,OAAO,IAAI,CAAC,aAAa,KAAK,SAAS,IAAI,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;QAC/B,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CACpC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,CACjD,CAAC;QACF,MAAM,KAAK,GAAgB,EAAE,CAAC;QAC9B,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,MAAM,SAAS,GAAG,CAAC,UAAuB,EAAE,EAAE;YAC5C,MAAM,EAAE,YAAY,EAAE,GAAG,UAAU,CAAC;YACpC,IAAI,IAAI,CAAC,uBAAuB,EAAE;gBAChC,OAAO,EAAE,GAAG,YAAY,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;aACtD;YACD,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC;QAEF,OAAO,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE;YACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACpD,IAAI,cAAc,IAAI,MAAM,EAAE;gBAC5B,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC;aAC1B;YAED,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YACpD,MAAM,WAAW,GAAG,IAAI;gBACtB,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;gBACnC,CAAC,CAAC,GAAG,MAAM,CAAC,IAAI,wCAAwC,CAAC;YAC3D,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;YACpC,IAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,YAAY,EAAE;gBACtB,OAAO,SAAS,CAAC;oBACf,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE;oBAC3D,GAAG,EAAE,EAAE;iBACR,CAAC,CAAC;aACJ;YACD,UAAU,IAAI,CAAC,CAAC;SACjB;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,qBAAqB,CACnD,IAAI,CAAC,mBAAmB,EACxB,KAAK,EACL,MAAM,CACP,CAAC;QAEF,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC;IAC3B,CAAC;IAED,UAAU;QACR,OAAO,gBAAyB,CAAC;IACnC,CAAC;IAED,SAAS;QACP,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;CACF"}

package/dist/esm/agents/helpers.js

@@ -1,15 +1,6 @@
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
 import { LLMChain } from "../chains";
 import { resolveConfigFromFile } from "../util";
-export const deserializeHelper = (llm, tools, data, fromLLMAndTools, fromConstructor) => __awaiter(void 0, void 0, void 0, function* () {
+export const deserializeHelper = async (llm, tools, data, fromLLMAndTools, fromConstructor) => {
     if (data.load_from_llm_and_tools) {
         if (!llm) {
             throw new Error("Loading from llm and tools, llm must be provided.");
@@ -20,7 +11,7 @@ export const deserializeHelper = (llm, tools, data, fromLLMAndTools, fromConstru
         return fromLLMAndTools(llm, tools, data);
     }
     const serializedLLMChain = resolveConfigFromFile("llm_chain", data);
-    const llmChain = yield LLMChain.deserialize(serializedLLMChain);
-    return fromConstructor(Object.assign(Object.assign({}, data), { llmChain }));
-});
+    const llmChain = await LLMChain.deserialize(serializedLLMChain);
+    return fromConstructor({ ...data, llmChain });
+};
 //# sourceMappingURL=helpers.js.map

package/dist/esm/agents/helpers.js.map

@@ -1 +1 @@
-{"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../../agents/helpers.ts"],"names":[],"mappings":";;;;;;;;;AAEA,OAAO,EAAsB,QAAQ,EAAE,MAAM,WAAW,CAAC;AACzD,OAAO,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;AAEhD,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAM/B,GAAwB,EACxB,KAAyB,EACzB,IAA+B,EAC/B,eAA4D,EAC5D,eAA+B,EACnB,EAAE;IACd,IAAI,IAAI,CAAC,uBAAuB,EAAE;QAChC,IAAI,CAAC,GAAG,EAAE;YACR,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;SACtE;QAED,IAAI,CAAC,KAAK,EAAE;YACV,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;SACxE;QAED,OAAO,eAAe,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;KAC1C;IAED,MAAM,kBAAkB,GAAG,qBAAqB,CAG9C,WAAW,EAAE,IAAI,CAAC,CAAC;IACrB,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;IAChE,OAAO,eAAe,iCAAM,IAAI,KAAE,QAAQ,IAAG,CAAC;AAChD,CAAC,CAAA,CAAC"}
+{"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../../agents/helpers.ts"],"names":[],"mappings":"AAEA,OAAO,EAAsB,QAAQ,EAAE,MAAM,WAAW,CAAC;AACzD,OAAO,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;AAEhD,MAAM,CAAC,MAAM,iBAAiB,GAAG,KAAK,EAMpC,GAAwB,EACxB,KAAyB,EACzB,IAA+B,EAC/B,eAA4D,EAC5D,eAA+B,EACnB,EAAE;IACd,IAAI,IAAI,CAAC,uBAAuB,EAAE;QAChC,IAAI,CAAC,GAAG,EAAE;YACR,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;SACtE;QAED,IAAI,CAAC,KAAK,EAAE;YACV,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;SACxE;QAED,OAAO,eAAe,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;KAC1C;IAED,MAAM,kBAAkB,GAAG,qBAAqB,CAG9C,WAAW,EAAE,IAAI,CAAC,CAAC;IACrB,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;IAChE,OAAO,eAAe,CAAC,EAAE,GAAG,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;AAChD,CAAC,CAAC"}

package/dist/esm/agents/index.d.ts

@@ -3,4 +3,5 @@ export { Agent, StaticAgent, staticImplements, AgentInput } from "./agent";
 export { AgentExecutor } from "./executor";
 export { ZeroShotAgent, SerializedZeroShotAgent } from "./mrkl";
 export { Tool } from "./tools";
+export { initializeAgentExecutor } from "./initialize";
 export { loadAgent } from "./load";

package/dist/esm/agents/index.js

@@ -1,5 +1,7 @@
 export { Agent, staticImplements } from "./agent";
 export { AgentExecutor } from "./executor";
 export { ZeroShotAgent } from "./mrkl";
+export { Tool } from "./tools";
+export { initializeAgentExecutor } from "./initialize";
 export { loadAgent } from "./load";
 //# sourceMappingURL=index.js.map

package/dist/esm/agents/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../agents/index.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,EAAe,gBAAgB,EAAc,MAAM,SAAS,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,aAAa,EAA2B,MAAM,QAAQ,CAAC;AAGhE,OAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../agents/index.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,EAAe,gBAAgB,EAAc,MAAM,SAAS,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,aAAa,EAA2B,MAAM,QAAQ,CAAC;AAChE,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAC,uBAAuB,EAAC,MAAM,cAAc,CAAC;AAErD,OAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC"}

package/dist/esm/agents/initialize.d.ts

@@ -0,0 +1,4 @@
+import { Tool } from "./tools";
+import { BaseLLM } from "../llms";
+import { AgentExecutor } from "./executor";
+export declare const initializeAgentExecutor: (tools: Tool[], llm: BaseLLM, agentType?: string) => Promise<AgentExecutor>;

package/dist/esm/agents/initialize.js

@@ -0,0 +1,17 @@
+import { AgentExecutor } from "./executor";
+import { ZeroShotAgent } from "./mrkl";
+export const initializeAgentExecutor = async (tools, llm, agentType = "zero-shot-react-description") => {
+    switch (agentType) {
+        case "zero-shot-react-description":
+            const agent = ZeroShotAgent.fromLLMAndTools(llm, tools);
+            const executor = AgentExecutor.fromAgentAndTools({
+                agent,
+                tools,
+                returnIntermediateSteps: true,
+            });
+            return executor;
+        default:
+            throw new Error("Unknown agent type");
+    }
+};
+//# sourceMappingURL=initialize.js.map

package/dist/esm/agents/initialize.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"initialize.js","sourceRoot":"","sources":["../../../agents/initialize.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AAEvC,MAAM,CAAC,MAAM,uBAAuB,GAAG,KAAK,EACxC,KAAa,EACb,GAAY,EACZ,SAAS,GAAE,6BAA6B,EAChB,EAAE;IAC1B,QAAQ,SAAS,EAAE;QACf,KAAK,6BAA6B;YAC9B,MAAM,KAAK,GAAG,aAAa,CAAC,eAAe,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,QAAQ,GAAG,aAAa,CAAC,iBAAiB,CAAC;gBAC7C,KAAK;gBACL,KAAK;gBACL,uBAAuB,EAAE,IAAI;aAC5B,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC;QACxB;YACI,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;KAC7C;AACH,CAAC,CAAC"}

package/dist/esm/agents/load.js

@@ -1,24 +1,15 @@
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
 import { Agent } from ".";
 import { loadFromHub } from "../util/hub";
 import { parseFileConfig } from "../util";
-const loadAgentFromFile = (file, llmAndTools) => __awaiter(void 0, void 0, void 0, function* () {
+const loadAgentFromFile = async (file, llmAndTools) => {
     const serialized = parseFileConfig(file);
-    return Agent.deserialize(Object.assign(Object.assign({}, serialized), llmAndTools));
-});
-export const loadAgent = (uri, llmAndTools) => __awaiter(void 0, void 0, void 0, function* () {
-    const hubResult = yield loadFromHub(uri, (u) => loadAgentFromFile(u, llmAndTools), "agents", new Set(["json", "yaml"]));
+    return Agent.deserialize({ ...serialized, ...llmAndTools });
+};
+export const loadAgent = async (uri, llmAndTools) => {
+    const hubResult = await loadFromHub(uri, (u) => loadAgentFromFile(u, llmAndTools), "agents", new Set(["json", "yaml"]));
     if (hubResult) {
         return hubResult;
     }
     return loadAgentFromFile(uri, llmAndTools);
-});
+};
 //# sourceMappingURL=load.js.map

package/dist/esm/agents/load.js.map

@@ -1 +1 @@
-{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../agents/load.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,EAAE,KAAK,EAAQ,MAAM,GAAG,CAAC;AAEhC,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAE1C,MAAM,iBAAiB,GAAG,CACxB,IAAY,EACZ,WAA+C,EAC/C,EAAE;IACF,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;IACzC,OAAO,KAAK,CAAC,WAAW,iCAAM,UAAU,GAAK,WAAW,EAAG,CAAC;AAC9D,CAAC,CAAA,CAAC;AAEF,MAAM,CAAC,MAAM,SAAS,GAAG,CACvB,GAAW,EACX,WAA+C,EAC/B,EAAE;IAClB,MAAM,SAAS,GAAG,MAAM,WAAW,CACjC,GAAG,EACH,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,WAAW,CAAC,EACxC,QAAQ,EACR,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAC1B,CAAC;IACF,IAAI,SAAS,EAAE;QACb,OAAO,SAAS,CAAC;KAClB;IAED,OAAO,iBAAiB,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;AAC7C,CAAC,CAAA,CAAC"}
+{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../agents/load.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAQ,MAAM,GAAG,CAAC;AAEhC,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAE1C,MAAM,iBAAiB,GAAG,KAAK,EAC7B,IAAY,EACZ,WAA+C,EAC/C,EAAE;IACF,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;IACzC,OAAO,KAAK,CAAC,WAAW,CAAC,EAAE,GAAG,UAAU,EAAE,GAAG,WAAW,EAAE,CAAC,CAAC;AAC9D,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,SAAS,GAAG,KAAK,EAC5B,GAAW,EACX,WAA+C,EAC/B,EAAE;IAClB,MAAM,SAAS,GAAG,MAAM,WAAW,CACjC,GAAG,EACH,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,WAAW,CAAC,EACxC,QAAQ,EACR,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAC1B,CAAC;IACF,IAAI,SAAS,EAAE;QACb,OAAO,SAAS,CAAC;KAClB;IAED,OAAO,iBAAiB,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;AAC7C,CAAC,CAAC"}