langchain 0.0.4 → 0.0.6

package/README.md

@@ -0,0 +1,33 @@
+# 🦜️🔗 LangChain.js
+
+⚡ Building applications with LLMs through composability ⚡
+
+**Production Support:** As you move your LangChains into production, we'd love to offer more comprehensive support.
+Please fill out [this form](https://forms.gle/57d8AmXBYp8PP8tZA) and we'll set up a dedicated support Slack channel.
+
+## Quick Install
+
+`yarn add langchain`
+
+```typescript
+import { OpenAI } from 'langchain/llms';
+```
+
+## 🤔 What is this?
+
+Large language models (LLMs) are emerging as a transformative technology, enabling
+developers to build applications that they previously could not.
+But using these LLMs in isolation is often not enough to
+create a truly powerful app - the real power comes when you can combine them with other sources of computation or knowledge.
+
+This library is aimed at assisting in the development of those types of applications.
+
+## Relationship with Python LangChain
+
+This is built to integrate as seamlessly as possible with the [LangChain Python package](https://github.com/hwchase17/langchain). Specifically, this means all objects (prompts, LLMs, chains, etc) are designed in a way where they can be serialized and shared between languages.
+
+The [LangChainHub](https://github.com/hwchase17/langchain-hub) is a central place for the serialized versions of these prompts, chains, and agents.
+
+## 📖 Documentation
+
+For full documentation of prompts, chains, agents and more, please see [here](https://hwchase17.github.io/langchainjs/docs/overview).

package/agents.d.ts

@@ -0,0 +1 @@
+export * from './dist/agents'

package/agents.js

@@ -0,0 +1 @@
+module.exports = require('./dist/agents')

package/chains.d.ts

@@ -0,0 +1 @@
+export * from './dist/chains'

package/chains.js

@@ -0,0 +1 @@
+module.exports = require('./dist/chains')

package/dist/agents/agent.d.ts

@@ -0,0 +1,96 @@
+import { ChainValues } from "../chains";
+import { SerializedZeroShotAgent, AgentAction, AgentFinish, AgentStep, StoppingMethod, Tool } from "./index";
+import { BaseLLM } from "../llms";
+import { LLMChain } from "../chains/llm_chain";
+import { BasePromptTemplate } from "../prompts";
+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;
+}
+export declare const staticImplements: <T>(_: T) => void;
+type SerializedAgent = SerializedZeroShotAgent;
+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[];
+    }): Promise<Agent>;
+}
+export {};

package/dist/agents/agent.js

@@ -0,0 +1,151 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Agent = exports.staticImplements = void 0;
+const index_1 = require("./index");
+class ParseError extends Error {
+    constructor(msg, output) {
+        super(msg);
+        Object.defineProperty(this, "output", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.output = output;
+    }
+}
+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", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "allowedTools", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: undefined
+        });
+        Object.defineProperty(this, "returnValues", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["output"]
+        });
+        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 +
+            [
+                action.log,
+                `${this.observationPrefix()}${observation}`,
+                this.llmPrefix(),
+            ].join("\n"), "");
+    }
+    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);
+    }
+    /**
+     * 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;
+                }
+                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}`);
+    }
+    /**
+     * 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;
+//# sourceMappingURL=agent.js.map

package/dist/agents/agent.js.map

@@ -0,0 +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;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/{esm/agents → 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/agents/executor.js

@@ -0,0 +1,99 @@
+"use strict";
+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;
+        super();
+        Object.defineProperty(this, "agent", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "tools", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "returnIntermediateSteps", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        Object.defineProperty(this, "maxIterations", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 15
+        });
+        Object.defineProperty(this, "earlyStoppingMethod", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "force"
+        });
+        this.agent = input.agent;
+        this.tools = input.tools;
+        this.returnIntermediateSteps =
+            (_a = input.returnIntermediateSteps) !== null && _a !== void 0 ? _a : this.returnIntermediateSteps;
+        this.maxIterations = (_b = input.maxIterations) !== null && _b !== void 0 ? _b : this.maxIterations;
+        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;
+    }
+    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 };
+            }
+            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";
+    }
+    serialize() {
+        throw new Error("Cannot serialize an AgentExecutor");
+    }
+}
+exports.AgentExecutor = AgentExecutor;
+//# sourceMappingURL=executor.js.map

package/dist/agents/executor.js.map

@@ -0,0 +1 @@
+{"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/agents/helpers.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.deserializeHelper = void 0;
+const chains_1 = require("../chains");
+const util_1 = require("../util");
+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.");
+        }
+        if (!tools) {
+            throw new Error("Loading from llm and tools, tools must be provided.");
+        }
+        return fromLLMAndTools(llm, tools, data);
+    }
+    const serializedLLMChain = (0, util_1.resolveConfigFromFile)("llm_chain", data);
+    const llmChain = await chains_1.LLMChain.deserialize(serializedLLMChain);
+    return fromConstructor({ ...data, llmChain });
+};
+exports.deserializeHelper = deserializeHelper;
+//# sourceMappingURL=helpers.js.map

package/dist/agents/helpers.js.map

@@ -0,0 +1 @@
+{"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 → 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 → 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/agents/index.js.map

@@ -0,0 +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;AACtB,iCAA+B;AAAtB,6FAAA,IAAI,OAAA;AACb,2CAAuD;AAA9C,qHAAA,uBAAuB,OAAA;AAEhC,+BAAmC;AAA1B,iGAAA,SAAS,OAAA"}

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

@@ -0,0 +1,19 @@
+"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":
+            return executor_1.AgentExecutor.fromAgentAndTools({
+                agent: mrkl_1.ZeroShotAgent.fromLLMAndTools(llm, tools),
+                tools,
+                returnIntermediateSteps: true,
+            });
+        default:
+            throw new Error("Unknown agent type");
+    }
+};
+exports.initializeAgentExecutor = initializeAgentExecutor;
+//# sourceMappingURL=initialize.js.map

package/dist/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,EAC1C,KAAa,EACb,GAAY,EACZ,SAAS,GAAG,6BAA6B,EACjB,EAAE;IAC1B,QAAQ,SAAS,EAAE;QACjB,KAAK,6BAA6B;YAChC,OAAO,wBAAa,CAAC,iBAAiB,CAAC;gBACrC,KAAK,EAAE,oBAAa,CAAC,eAAe,CAAC,GAAG,EAAE,KAAK,CAAC;gBAChD,KAAK;gBACL,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;QACL;YACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;KACzC;AACH,CAAC,CAAC;AAfW,QAAA,uBAAuB,2BAelC"}

package/dist/agents/load.js

@@ -0,0 +1,19 @@
+"use strict";
+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 = async (file, llmAndTools) => {
+    const serialized = (0, util_1.parseFileConfig)(file);
+    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/agents/load.js.map

@@ -0,0 +1 @@
+{"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/{esm/agents → agents}/mrkl/index.d.ts

@@ -1,6 +1,6 @@
 import { BaseLLM } from "../../llms";
 import { Agent, Tool, AgentInput, SerializedAgentT } from "../index";
-import { PromptTemplate } from "../../prompt";
+import { PromptTemplate } from "../../prompts";
 type SerializedFromLLMAndTools = {
     suffix?: string;
     prefix?: string;
@@ -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): {