langchain 0.0.4 → 0.0.5

package/dist/cjs/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/cjs/agents/agent.js

@@ -1,13 +1,4 @@
 "use strict";
-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());
-    });
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Agent = exports.staticImplements = void 0;
 const index_1 = require("./index");
@@ -25,6 +16,13 @@ class ParseError extends Error {
 }
 const staticImplements = (_) => { };
 exports.staticImplements = 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.
+ */
 class Agent {
     constructor(input) {
         Object.defineProperty(this, "llmChain", {
@@ -48,15 +46,27 @@ 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 +
             [
@@ -65,64 +75,76 @@ 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 index_1.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 index_1.ZeroShotAgent.deserialize(data);
+            default:
+                throw new Error("Unknown agent type");
+        }
     }
 }
 exports.Agent = Agent;

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

@@ -1 +1 @@
-{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../agents/agent.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,mCAQiB;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;AAgBM,MAAM,gBAAgB,GAAG,CAAI,CAAI,EAAE,EAAE,GAAE,CAAC,CAAC;AAAnC,QAAA,gBAAgB,oBAAmB;AAShD,MAAsB,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,qBAAa,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;gBACzC;oBACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;aACzC;QACH,CAAC;KAAA;CACF;AA/HD,sBA+HC"}
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../agents/agent.ts"],"names":[],"mappings":";;;AACA,mCAQiB;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;AAyBM,MAAM,gBAAgB,GAAG,CAAI,CAAI,EAAE,EAAE,GAAE,CAAC,CAAC;AAAnC,QAAA,gBAAgB,oBAAmB;AAShD;;;;;;GAMG;AACH,MAAsB,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,qBAAa,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACzC;gBACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;SACzC;IACH,CAAC;CACF;AArKD,sBAqKC"}

package/dist/cjs/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/cjs/agents/executor.js

@@ -1,16 +1,11 @@
 "use strict";
-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());
-    });
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AgentExecutor = void 0;
 const chains_1 = require("../chains");
+/**
+ * A chain managing an agent using tools.
+ * @augments BaseChain
+ */
 class AgentExecutor extends chains_1.BaseChain {
     constructor(input) {
         var _a, _b, _c;
@@ -53,46 +48,45 @@ class AgentExecutor extends chains_1.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/cjs/agents/executor.js.map

@@ -1 +1 @@
-{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../../agents/executor.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,sCAAmD;AAYnD,MAAa,aAAc,SAAQ,kBAAS;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;AAtFD,sCAsFC"}
+{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../../agents/executor.ts"],"names":[],"mappings":";;;AAAA,sCAAmD;AAYnD;;;GAGG;AACH,MAAa,aAAc,SAAQ,kBAAS;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;AAzFD,sCAyFC"}

package/dist/cjs/agents/helpers.js

@@ -1,18 +1,9 @@
 "use strict";
-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());
-    });
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.deserializeHelper = void 0;
 const chains_1 = require("../chains");
 const util_1 = require("../util");
-const deserializeHelper = (llm, tools, data, fromLLMAndTools, fromConstructor) => __awaiter(void 0, void 0, void 0, function* () {
+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.");
@@ -23,8 +14,8 @@ const deserializeHelper = (llm, tools, data, fromLLMAndTools, fromConstructor) =
         return fromLLMAndTools(llm, tools, data);
     }
     const serializedLLMChain = (0, util_1.resolveConfigFromFile)("llm_chain", data);
-    const llmChain = yield chains_1.LLMChain.deserialize(serializedLLMChain);
-    return fromConstructor(Object.assign(Object.assign({}, data), { llmChain }));
-});
+    const llmChain = await chains_1.LLMChain.deserialize(serializedLLMChain);
+    return fromConstructor({ ...data, llmChain });
+};
 exports.deserializeHelper = deserializeHelper;
 //# sourceMappingURL=helpers.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../../agents/helpers.ts"],"names":[],"mappings":";;;;;;;;;;;;AAEA,sCAAyD;AACzD,kCAAgD;AAEzC,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,IAAA,4BAAqB,EAG9C,WAAW,EAAE,IAAI,CAAC,CAAC;IACrB,MAAM,QAAQ,GAAG,MAAM,iBAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;IAChE,OAAO,eAAe,iCAAM,IAAI,KAAE,QAAQ,IAAG,CAAC;AAChD,CAAC,CAAA,CAAC;AA9BW,QAAA,iBAAiB,qBA8B5B"}
+{"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../../agents/helpers.ts"],"names":[],"mappings":";;;AAEA,sCAAyD;AACzD,kCAAgD;AAEzC,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,IAAA,4BAAqB,EAG9C,WAAW,EAAE,IAAI,CAAC,CAAC;IACrB,MAAM,QAAQ,GAAG,MAAM,iBAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;IAChE,OAAO,eAAe,CAAC,EAAE,GAAG,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;AAChD,CAAC,CAAC;AA9BW,QAAA,iBAAiB,qBA8B5B"}

package/dist/cjs/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/cjs/agents/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.loadAgent = exports.ZeroShotAgent = exports.AgentExecutor = exports.staticImplements = exports.Agent = void 0;
+exports.loadAgent = exports.initializeAgentExecutor = exports.Tool = exports.ZeroShotAgent = exports.AgentExecutor = exports.staticImplements = exports.Agent = void 0;
 var agent_1 = require("./agent");
 Object.defineProperty(exports, "Agent", { enumerable: true, get: function () { return agent_1.Agent; } });
 Object.defineProperty(exports, "staticImplements", { enumerable: true, get: function () { return agent_1.staticImplements; } });
@@ -8,6 +8,10 @@ var executor_1 = require("./executor");
 Object.defineProperty(exports, "AgentExecutor", { enumerable: true, get: function () { return executor_1.AgentExecutor; } });
 var mrkl_1 = require("./mrkl");
 Object.defineProperty(exports, "ZeroShotAgent", { enumerable: true, get: function () { return mrkl_1.ZeroShotAgent; } });
+var tools_1 = require("./tools");
+Object.defineProperty(exports, "Tool", { enumerable: true, get: function () { return tools_1.Tool; } });
+var initialize_1 = require("./initialize");
+Object.defineProperty(exports, "initializeAgentExecutor", { enumerable: true, get: function () { return initialize_1.initializeAgentExecutor; } });
 var load_1 = require("./load");
 Object.defineProperty(exports, "loadAgent", { enumerable: true, get: function () { return load_1.loadAgent; } });
 //# sourceMappingURL=index.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../agents/index.ts"],"names":[],"mappings":";;;AAOA,iCAA2E;AAAlE,8FAAA,KAAK,OAAA;AAAe,yGAAA,gBAAgB,OAAA;AAC7C,uCAA2C;AAAlC,yGAAA,aAAa,OAAA;AACtB,+BAAgE;AAAvD,qGAAA,aAAa,OAAA;AAGtB,+BAAmC;AAA1B,iGAAA,SAAS,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../agents/index.ts"],"names":[],"mappings":";;;AAOA,iCAA2E;AAAlE,8FAAA,KAAK,OAAA;AAAe,yGAAA,gBAAgB,OAAA;AAC7C,uCAA2C;AAAlC,yGAAA,aAAa,OAAA;AACtB,+BAAgE;AAAvD,qGAAA,aAAa,OAAA;AACtB,iCAA+B;AAAtB,6FAAA,IAAI,OAAA;AACb,2CAAqD;AAA7C,qHAAA,uBAAuB,OAAA;AAE/B,+BAAmC;AAA1B,iGAAA,SAAS,OAAA"}

package/dist/cjs/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/cjs/agents/initialize.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.initializeAgentExecutor = void 0;
+const executor_1 = require("./executor");
+const mrkl_1 = require("./mrkl");
+const initializeAgentExecutor = async (tools, llm, agentType = "zero-shot-react-description") => {
+    switch (agentType) {
+        case "zero-shot-react-description":
+            const agent = mrkl_1.ZeroShotAgent.fromLLMAndTools(llm, tools);
+            const executor = executor_1.AgentExecutor.fromAgentAndTools({
+                agent,
+                tools,
+                returnIntermediateSteps: true,
+            });
+            return executor;
+        default:
+            throw new Error("Unknown agent type");
+    }
+};
+exports.initializeAgentExecutor = initializeAgentExecutor;
+//# sourceMappingURL=initialize.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"initialize.js","sourceRoot":"","sources":["../../../agents/initialize.ts"],"names":[],"mappings":";;;AAEA,yCAA2C;AAC3C,iCAAuC;AAEhC,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,oBAAa,CAAC,eAAe,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,QAAQ,GAAG,wBAAa,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;AAjBS,QAAA,uBAAuB,2BAiBhC"}

package/dist/cjs/agents/load.js

@@ -1,28 +1,19 @@
 "use strict";
-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());
-    });
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.loadAgent = void 0;
 const _1 = require(".");
 const hub_1 = require("../util/hub");
 const util_1 = require("../util");
-const loadAgentFromFile = (file, llmAndTools) => __awaiter(void 0, void 0, void 0, function* () {
+const loadAgentFromFile = async (file, llmAndTools) => {
     const serialized = (0, util_1.parseFileConfig)(file);
-    return _1.Agent.deserialize(Object.assign(Object.assign({}, serialized), llmAndTools));
-});
-const loadAgent = (uri, llmAndTools) => __awaiter(void 0, void 0, void 0, function* () {
-    const hubResult = yield (0, hub_1.loadFromHub)(uri, (u) => loadAgentFromFile(u, llmAndTools), "agents", new Set(["json", "yaml"]));
+    return _1.Agent.deserialize({ ...serialized, ...llmAndTools });
+};
+const loadAgent = async (uri, llmAndTools) => {
+    const hubResult = await (0, hub_1.loadFromHub)(uri, (u) => loadAgentFromFile(u, llmAndTools), "agents", new Set(["json", "yaml"]));
     if (hubResult) {
         return hubResult;
     }
     return loadAgentFromFile(uri, llmAndTools);
-});
+};
 exports.loadAgent = loadAgent;
 //# sourceMappingURL=load.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../agents/load.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,wBAAgC;AAEhC,qCAA0C;AAC1C,kCAA0C;AAE1C,MAAM,iBAAiB,GAAG,CACxB,IAAY,EACZ,WAA+C,EAC/C,EAAE;IACF,MAAM,UAAU,GAAG,IAAA,sBAAe,EAAC,IAAI,CAAC,CAAC;IACzC,OAAO,QAAK,CAAC,WAAW,iCAAM,UAAU,GAAK,WAAW,EAAG,CAAC;AAC9D,CAAC,CAAA,CAAC;AAEK,MAAM,SAAS,GAAG,CACvB,GAAW,EACX,WAA+C,EAC/B,EAAE;IAClB,MAAM,SAAS,GAAG,MAAM,IAAA,iBAAW,EACjC,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;AAfW,QAAA,SAAS,aAepB"}
+{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../agents/load.ts"],"names":[],"mappings":";;;AAAA,wBAAgC;AAEhC,qCAA0C;AAC1C,kCAA0C;AAE1C,MAAM,iBAAiB,GAAG,KAAK,EAC7B,IAAY,EACZ,WAA+C,EAC/C,EAAE;IACF,MAAM,UAAU,GAAG,IAAA,sBAAe,EAAC,IAAI,CAAC,CAAC;IACzC,OAAO,QAAK,CAAC,WAAW,CAAC,EAAE,GAAG,UAAU,EAAE,GAAG,WAAW,EAAE,CAAC,CAAC;AAC9D,CAAC,CAAC;AAEK,MAAM,SAAS,GAAG,KAAK,EAC5B,GAAW,EACX,WAA+C,EAC/B,EAAE;IAClB,MAAM,SAAS,GAAG,MAAM,IAAA,iBAAW,EACjC,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;AAfW,QAAA,SAAS,aAepB"}

package/dist/cjs/agents/mrkl/index.d.ts

@@ -8,17 +8,34 @@ type SerializedFromLLMAndTools = {
 };
 export type SerializedZeroShotAgent = SerializedAgentT<"zero-shot-react-description", SerializedFromLLMAndTools, AgentInput>;
 type CreatePromptArgs = {
+    /** String to put after the list of tools. */
     suffix?: string;
+    /** String to put before the list of tools. */
     prefix?: string;
+    /** List of input variables the final prompt will expect. */
     inputVariables?: string[];
 };
 type ZeroShotAgentInput = AgentInput;
+/**
+ * Agent for the MRKL chain.
+ * @augments Agent
+ * @augments StaticAgent
+ */
 export declare class ZeroShotAgent extends Agent {
     constructor(input: ZeroShotAgentInput);
     _agentType(): "zero-shot-react-description";
     observationPrefix(): string;
     llmPrefix(): string;
     static validateTools(tools: Tool[]): void;
+    /**
+     * Create prompt in the style of the zero shot agent.
+     *
+     * @param tools - List of tools the agent will have access to, used to format the prompt.
+     * @param args - Arguments to create the prompt with.
+     * @param args.suffix - String to put after the list of tools.
+     * @param args.prefix - String to put before the list of tools.
+     * @param args.inputVariables - List of input variables the final prompt will expect.
+     */
     static createPrompt(tools: Tool[], args?: CreatePromptArgs): PromptTemplate;
     static fromLLMAndTools(llm: BaseLLM, tools: Tool[], args?: CreatePromptArgs): ZeroShotAgent;
     extractToolAndInput(text: string): {