langchain 0.0.4 → 0.0.6

package/dist/{cjs/agents → agents}/mrkl/index.js

@@ -5,35 +5,20 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
     else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
     return c > 3 && r && Object.defineProperty(target, key, r), r;
 };
-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());
-    });
-};
-var __rest = (this && this.__rest) || function (s, e) {
-    var t = {};
-    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
-        t[p] = s[p];
-    if (s != null && typeof Object.getOwnPropertySymbols === "function")
-        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
-            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
-                t[p[i]] = s[p[i]];
-        }
-    return t;
-};
 var ZeroShotAgent_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ZeroShotAgent = void 0;
 const chains_1 = require("../../chains");
 const index_1 = require("../index");
-const prompt_1 = require("../../prompt");
-const prompt_2 = require("./prompt");
+const prompts_1 = require("../../prompts");
+const prompt_1 = require("./prompt");
 const helpers_1 = require("../helpers");
 const FINAL_ANSWER_ACTION = "Final Answer:";
+/**
+ * Agent for the MRKL chain.
+ * @augments Agent
+ * @augments StaticAgent
+ */
 let ZeroShotAgent = ZeroShotAgent_1 = class ZeroShotAgent extends index_1.Agent {
     constructor(input) {
         super(input);
@@ -55,14 +40,23 @@ let ZeroShotAgent = ZeroShotAgent_1 = class ZeroShotAgent extends index_1.Agent
             throw new Error(msg);
         }
     }
+    /**
+     * 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, args) {
-        const { prefix = prompt_2.PREFIX, suffix = prompt_2.SUFFIX, inputVariables = ["input", "agent_scratchpad"], } = args !== null && args !== void 0 ? args : {};
+        const { prefix = prompt_1.PREFIX, suffix = prompt_1.SUFFIX, inputVariables = ["input", "agent_scratchpad"], } = args !== null && args !== void 0 ? args : {};
         const toolStrings = tools
             .map((tool) => `${tool.name}: ${tool.description}`)
             .join("\n");
-        const instructions = (0, prompt_2.formatInstructions)(toolStrings);
+        const instructions = (0, prompt_1.formatInstructions)(toolStrings);
         const template = [prefix, toolStrings, instructions, suffix].join("\n\n");
-        return new prompt_1.PromptTemplate({
+        return new prompts_1.PromptTemplate({
             template,
             inputVariables,
         });
@@ -91,15 +85,13 @@ let ZeroShotAgent = ZeroShotAgent_1 = class ZeroShotAgent extends index_1.Agent
             input: match[2].trim().replace(/^"+|"+$/g, ""),
         };
     }
-    static deserialize(data) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { llm, tools } = data, rest = __rest(data, ["llm", "tools"]);
-            return (0, helpers_1.deserializeHelper)(llm, tools, rest, (llm, tools, args) => ZeroShotAgent_1.fromLLMAndTools(llm, tools, {
-                prefix: args.prefix,
-                suffix: args.suffix,
-                inputVariables: args.input_variables,
-            }), (args) => new ZeroShotAgent_1(args));
-        });
+    static async deserialize(data) {
+        const { llm, tools, ...rest } = data;
+        return (0, helpers_1.deserializeHelper)(llm, tools, rest, (llm, tools, args) => ZeroShotAgent_1.fromLLMAndTools(llm, tools, {
+            prefix: args.prefix,
+            suffix: args.suffix,
+            inputVariables: args.input_variables,
+        }), (args) => new ZeroShotAgent_1(args));
     }
 };
 ZeroShotAgent = ZeroShotAgent_1 = __decorate([

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../agents/mrkl/index.ts"],"names":[],"mappings":";;;;;;;;;;AACA,yCAAwC;AACxC,oCAOkB;AAClB,2CAA+C;AAC/C,qCAA8D;AAC9D,wCAA+C;AAE/C,MAAM,mBAAmB,GAAG,eAAe,CAAC;AAyB5C;;;;GAIG;AAEI,IAAM,aAAa,qBAAnB,MAAM,aAAc,SAAQ,aAAK;IACtC,YAAY,KAAyB;QACnC,KAAK,CAAC,KAAK,CAAC,CAAC;IACf,CAAC;IAED,UAAU;QACR,OAAO,6BAAsC,CAAC;IAChD,CAAC;IAED,iBAAiB;QACf,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,SAAS;QACP,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,KAAa;QAChC,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5D,IAAI,WAAW,EAAE;YACf,MAAM,GAAG,GACP,cAAc,WAAW,CAAC,IAAI,yBAAyB;gBACvD,kDAAkD,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC;SACtB;IACH,CAAC;IAED;;;;;;;;OAQG;IACH,MAAM,CAAC,YAAY,CAAC,KAAa,EAAE,IAAuB;QACxD,MAAM,EACJ,MAAM,GAAG,eAAM,EACf,MAAM,GAAG,eAAM,EACf,cAAc,GAAG,CAAC,OAAO,EAAE,kBAAkB,CAAC,GAC/C,GAAG,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,EAAE,CAAC;QACf,MAAM,WAAW,GAAG,KAAK;aACtB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;aAClD,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,MAAM,YAAY,GAAG,IAAA,2BAAkB,EAAC,WAAW,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,CAAC,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE1E,OAAO,IAAI,wBAAc,CAAC;YACxB,QAAQ;YACR,cAAc;SACf,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,eAAe,CAAC,GAAY,EAAE,KAAa,EAAE,IAAuB;QACzE,eAAa,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,MAAM,GAAG,eAAa,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,IAAI,iBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;QAC5C,OAAO,IAAI,eAAa,CAAC;YACvB,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SACvC,CAAC,CAAC;IACL,CAAC;IAED,mBAAmB,CAAC,IAAY;QAC9B,IAAI,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE;YACtC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YAC9C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC7C,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC;SACxC;QAED,MAAM,KAAK,GAAG,mCAAmC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,IAAI,CAAC,KAAK,EAAE;YACV,MAAM,IAAI,KAAK,CAAC,+BAA+B,IAAI,EAAE,CAAC,CAAC;SACxD;QAED,OAAO;YACL,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;YACrB,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SAC/C,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,IAAiE;QAEjE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC;QACrC,OAAO,IAAA,2BAAiB,EACtB,GAAG,EACH,KAAK,EACL,IAAI,EACJ,CAAC,GAAY,EAAE,KAAa,EAAE,IAA+B,EAAE,EAAE,CAC/D,eAAa,CAAC,eAAe,CAAC,GAAG,EAAE,KAAK,EAAE;YACxC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,cAAc,EAAE,IAAI,CAAC,eAAe;SACrC,CAAC,EACJ,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,eAAa,CAAC,IAAI,CAAC,CAClC,CAAC;IACJ,CAAC;CACF,CAAA;AAnGY,aAAa;IADzB,CAAC,wBAA6B,CAAC;GACnB,aAAa,CAmGzB;AAnGY,sCAAa"}

package/dist/{cjs/agents → agents}/mrkl/prompt.js

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SUFFIX = exports.formatInstructions = exports.PREFIX = void 0;
-/* eslint-disable max-len */
 exports.PREFIX = `Answer the following questions as best you can. You have access to the following tools:`;
 const formatInstructions = (toolNames) => `Use the following format:
 

package/dist/agents/mrkl/prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../../agents/mrkl/prompt.ts"],"names":[],"mappings":";;;AAAa,QAAA,MAAM,GAAG,yFAAyF,CAAC;AACzG,MAAM,kBAAkB,GAAG,CAChC,SAAiB,EACjB,EAAE,CAAC;;;;gDAI2C,SAAS;;;;;8DAKK,CAAC;AAXlD,QAAA,kBAAkB,sBAWgC;AAClD,QAAA,MAAM,GAAG;;;2BAGK,CAAC"}

package/dist/agents/tests/agent.test.js

@@ -0,0 +1,35 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const openai_1 = require("../../llms/openai");
+const load_1 = require("../load");
+const index_1 = require("../index");
+const serpapi_1 = require("../tools/serpapi");
+const calculator_1 = require("../tools/calculator");
+const initialize_1 = require("../initialize");
+(0, globals_1.test)("Run agent from hub", async () => {
+    const model = new openai_1.OpenAI({ temperature: 0 });
+    const tools = [new serpapi_1.SerpAPI(), new calculator_1.Calculator()];
+    const agent = await (0, load_1.loadAgent)("lc://agents/zero-shot-react-description/agent.json", { llm: model, tools });
+    const executor = index_1.AgentExecutor.fromAgentAndTools({
+        agent,
+        tools,
+        returnIntermediateSteps: true,
+    });
+    const res = await executor.call({
+        input: "Who is Olivia Wilde's boyfriend? What is his current age raised to the 0.23 power?",
+    });
+    console.log(res);
+}, 30000);
+(0, globals_1.test)("Run agent locally", async () => {
+    const model = new openai_1.OpenAI({ temperature: 0 });
+    const tools = [new serpapi_1.SerpAPI(), new calculator_1.Calculator()];
+    const executor = await (0, initialize_1.initializeAgentExecutor)(tools, model, "zero-shot-react-description");
+    console.log("Loaded agent.");
+    const input = "Who is Olivia Wilde's boyfriend?" +
+        " What is his current age raised to the 0.23 power?";
+    console.log(`Executing with input "${input}"...`);
+    const result = await executor.call({ input });
+    console.log(`Got output ${result.output}`);
+}, 30000);
+//# sourceMappingURL=agent.test.js.map

package/dist/agents/tests/agent.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.test.js","sourceRoot":"","sources":["../../../agents/tests/agent.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,kCAAoC;AACpC,oCAA+C;AAC/C,8CAA2C;AAC3C,oDAAiD;AACjD,8CAAwD;AAExD,IAAA,cAAI,EAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;IACpC,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAW,CAAC,IAAI,iBAAO,EAAE,EAAE,IAAI,uBAAU,EAAE,CAAC,CAAC;IACxD,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAS,EAC3B,oDAAoD,EACpD,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,CACtB,CAAC;IACF,MAAM,QAAQ,GAAG,qBAAa,CAAC,iBAAiB,CAAC;QAC/C,KAAK;QACL,KAAK;QACL,uBAAuB,EAAE,IAAI;KAC9B,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;QAC9B,KAAK,EACH,oFAAoF;KACvF,CAAC,CAAC;IACH,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;AACnB,CAAC,EAAE,KAAK,CAAC,CAAC;AAEV,IAAA,cAAI,EAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;IACnC,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAG,CAAC,IAAI,iBAAO,EAAE,EAAE,IAAI,uBAAU,EAAE,CAAC,CAAC;IAEhD,MAAM,QAAQ,GAAG,MAAM,IAAA,oCAAuB,EAC5C,KAAK,EACL,KAAK,EACL,6BAA6B,CAC9B,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;IAE7B,MAAM,KAAK,GACT,kCAAkC;QAClC,oDAAoD,CAAC;IACvD,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,MAAM,CAAC,CAAC;IAElD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAE9C,OAAO,CAAC,GAAG,CAAC,cAAc,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;AAC7C,CAAC,EAAE,KAAK,CAAC,CAAC"}

package/dist/agents/tests/calculator.test.js

@@ -0,0 +1,30 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const calculator_1 = require("../tools/calculator");
+(0, globals_1.test)("Calculator tool, sum", async () => {
+    const calculator = new calculator_1.Calculator();
+    const result = await calculator.call("1 + 1");
+    (0, globals_1.expect)(result).toBe("2");
+});
+(0, globals_1.test)("Calculator tool, product", async () => {
+    const calculator = new calculator_1.Calculator();
+    const result = await calculator.call("2 * 3");
+    (0, globals_1.expect)(result).toBe("6");
+});
+(0, globals_1.test)("Calculator tool, division", async () => {
+    const calculator = new calculator_1.Calculator();
+    const result = await calculator.call("7 /2");
+    (0, globals_1.expect)(result).toBe("3.5");
+});
+(0, globals_1.test)("Calculator tool, exponentiation", async () => {
+    const calculator = new calculator_1.Calculator();
+    const result = await calculator.call("2 ^ 8");
+    (0, globals_1.expect)(result).toBe("256");
+});
+(0, globals_1.test)("Calculator tool, complicated expression", async () => {
+    const calculator = new calculator_1.Calculator();
+    const result = await calculator.call("((2 + 3) * 4) / 2");
+    (0, globals_1.expect)(result).toBe("10");
+});
+//# sourceMappingURL=calculator.test.js.map

package/dist/agents/tests/calculator.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"calculator.test.js","sourceRoot":"","sources":["../../../agents/tests/calculator.test.ts"],"names":[],"mappings":";;AAAA,2CAA6C;AAC7C,oDAAiD;AAEjD,IAAA,cAAI,EAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;IACtC,MAAM,UAAU,GAAG,IAAI,uBAAU,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9C,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC3B,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;IAC1C,MAAM,UAAU,GAAG,IAAI,uBAAU,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9C,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC3B,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;IAC3C,MAAM,UAAU,GAAG,IAAI,uBAAU,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC7C,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;IACjD,MAAM,UAAU,GAAG,IAAI,uBAAU,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9C,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;IACzD,MAAM,UAAU,GAAG,IAAI,uBAAU,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAC1D,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC,CAAC,CAAC"}

package/dist/agents/tools/base.d.ts

@@ -0,0 +1,6 @@
+export declare abstract class Tool {
+    abstract call(arg: string): Promise<string>;
+    abstract name: string;
+    abstract description: string;
+    returnDirect: boolean;
+}

package/dist/agents/tools/base.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Tool = void 0;
+class Tool {
+    constructor() {
+        Object.defineProperty(this, "returnDirect", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+    }
+}
+exports.Tool = Tool;
+//# sourceMappingURL=base.js.map

package/dist/agents/tools/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../agents/tools/base.ts"],"names":[],"mappings":";;;AAAA,MAAsB,IAAI;IAA1B;QAOE;;;;mBAAe,KAAK;WAAC;IACvB,CAAC;CAAA;AARD,oBAQC"}

package/dist/agents/tools/calculator.d.ts

@@ -0,0 +1,6 @@
+import { Tool } from "./base";
+export declare class Calculator extends Tool {
+    name: string;
+    call(input: string): Promise<string>;
+    description: string;
+}

package/dist/agents/tools/calculator.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Calculator = void 0;
+const expr_eval_1 = require("expr-eval");
+const base_1 = require("./base");
+class Calculator extends base_1.Tool {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "name", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "calculator"
+        });
+        Object.defineProperty(this, "description", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: `Useful for getting the result of a math expression. 
+  The input to this tool should be a valid mathmatical expression that can be parsed and evaluated by javascript's expr-eval`
+        });
+    }
+    async call(input) {
+        try {
+            return expr_eval_1.Parser.evaluate(input).toString();
+        }
+        catch (error) {
+            return "I don't know how to do that.";
+        }
+    }
+}
+exports.Calculator = Calculator;
+//# sourceMappingURL=calculator.js.map

package/dist/agents/tools/calculator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"calculator.js","sourceRoot":"","sources":["../../../agents/tools/calculator.ts"],"names":[],"mappings":";;;AAAA,yCAAmC;AAEnC,iCAA8B;AAE9B,MAAa,UAAW,SAAQ,WAAI;IAApC;;QACE;;;;mBAAO,YAAY;WAAC;QAUpB;;;;mBAAc;6HAC6G;WAAC;IAC9H,CAAC;IAVC,KAAK,CAAC,IAAI,CAAC,KAAa;QACtB,IAAI;YACF,OAAO,kBAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;SAC1C;QAAC,OAAO,KAAK,EAAE;YACd,OAAO,8BAA8B,CAAC;SACvC;IACH,CAAC;CAIF;AAbD,gCAaC"}

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

@@ -0,0 +1,3 @@
+export { SerpAPI } from "./serpapi";
+export { Calculator } from "./calculator";
+export { Tool } from "./base";

package/dist/agents/tools/index.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Tool = exports.Calculator = exports.SerpAPI = void 0;
+var serpapi_1 = require("./serpapi");
+Object.defineProperty(exports, "SerpAPI", { enumerable: true, get: function () { return serpapi_1.SerpAPI; } });
+var calculator_1 = require("./calculator");
+Object.defineProperty(exports, "Calculator", { enumerable: true, get: function () { return calculator_1.Calculator; } });
+var base_1 = require("./base");
+Object.defineProperty(exports, "Tool", { enumerable: true, get: function () { return base_1.Tool; } });
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../agents/tools/index.ts"],"names":[],"mappings":";;;AAAA,qCAAoC;AAA3B,kGAAA,OAAO,OAAA;AAChB,2CAA0C;AAAjC,wGAAA,UAAU,OAAA;AACnB,+BAA8B;AAArB,4FAAA,IAAI,OAAA"}

package/dist/{cjs/agents → agents}/tools/llm-math.js.map

@@ -1 +1 @@
-{"version":3,"file":"llm-math.js","sourceRoot":"","sources":["../../../../agents/tools/llm-math.ts"],"names":[],"mappings":""}
+{"version":3,"file":"llm-math.js","sourceRoot":"","sources":["../../../agents/tools/llm-math.ts"],"names":[],"mappings":""}

package/dist/agents/tools/serpapi.d.ts

@@ -0,0 +1,18 @@
+import type { GoogleParameters } from "serpapi";
+import { Tool } from "./base";
+/**
+ * Wrapper around SerpAPI.
+ *
+ * To use, you should have the `serpapi` package installed and the SERPAPI_API_KEY environment variable set.
+ */
+export declare class SerpAPI extends Tool {
+    protected key: string;
+    protected params: Partial<GoogleParameters>;
+    constructor(apiKey?: string | undefined, params?: Partial<GoogleParameters>);
+    name: string;
+    /**
+     * Run query through SerpAPI and parse result
+     */
+    call(input: string): Promise<any>;
+    description: string;
+}

package/dist/agents/tools/serpapi.js

@@ -0,0 +1,93 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SerpAPI = void 0;
+const base_1 = require("./base");
+let getJson = null;
+try {
+    // eslint-disable-next-line global-require,import/no-extraneous-dependencies
+    ({ getJson } = require("serpapi"));
+}
+catch (_a) {
+    // ignore error
+}
+/**
+ * Wrapper around SerpAPI.
+ *
+ * To use, you should have the `serpapi` package installed and the SERPAPI_API_KEY environment variable set.
+ */
+class SerpAPI extends base_1.Tool {
+    constructor(apiKey = process.env.SERPAPI_API_KEY, params = {}) {
+        super();
+        Object.defineProperty(this, "key", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "params", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "name", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "search"
+        });
+        Object.defineProperty(this, "description", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "a search engine. useful for when you need to answer questions about current events. input should be a search query."
+        });
+        // Throw error at construction time.
+        if (getJson === null) {
+            throw new Error("Please install serpapi as a dependency with, e.g. `npm i serpapi`");
+        }
+        if (!apiKey) {
+            throw new Error("SerpAPI API key not set. You can set it as SERPAPI_API_KEY in your .env file, or pass it to SerpAPI.");
+        }
+        this.key = apiKey;
+        this.params = params;
+    }
+    /**
+     * Run query through SerpAPI and parse result
+     */
+    async call(input) {
+        var _a, _b, _c, _d, _e, _f, _g;
+        if (getJson === null) {
+            throw new Error("Please install serpapi as a dependency with, e.g. `npm i serpapi`");
+        }
+        const res = await getJson("google", {
+            ...this.params,
+            api_key: this.key,
+            q: input,
+        });
+        if (res.error) {
+            throw new Error(`Got error from serpAPI: ${res.error}`);
+        }
+        if ((_a = res.answer_box) === null || _a === void 0 ? void 0 : _a.answer) {
+            return res.answer_box.answer;
+        }
+        if ((_b = res.answer_box) === null || _b === void 0 ? void 0 : _b.snippet) {
+            return res.answer_box.snippet;
+        }
+        if ((_c = res.answer_box) === null || _c === void 0 ? void 0 : _c.snippet_highlighted_words) {
+            return res.answer_box.snippet_highlighted_words[0];
+        }
+        if ((_d = res.sports_results) === null || _d === void 0 ? void 0 : _d.game_spotlight) {
+            return res.sports_results.game_spotlight;
+        }
+        if ((_e = res.knowledge_graph) === null || _e === void 0 ? void 0 : _e.description) {
+            return res.knowledge_graph.description;
+        }
+        if ((_g = (_f = res.organic_results) === null || _f === void 0 ? void 0 : _f[0]) === null || _g === void 0 ? void 0 : _g.snippet) {
+            return res.organic_results[0].snippet;
+        }
+        return "No good search result found";
+    }
+}
+exports.SerpAPI = SerpAPI;
+//# sourceMappingURL=serpapi.js.map

package/dist/agents/tools/serpapi.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"serpapi.js","sourceRoot":"","sources":["../../../agents/tools/serpapi.ts"],"names":[],"mappings":";;;AAEA,iCAA8B;AAE9B,IAAI,OAAO,GAA2B,IAAI,CAAC;AAE3C,IAAI;IACF,4EAA4E;IAC5E,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;CACpC;AAAC,WAAM;IACN,eAAe;CAChB;AAED;;;;GAIG;AACH,MAAa,OAAQ,SAAQ,WAAI;IAK/B,YACE,SAA6B,OAAO,CAAC,GAAG,CAAC,eAAe,EACxD,SAAoC,EAAE;QAEtC,KAAK,EAAE,CAAC;QARV;;;;;WAAsB;QAEtB;;;;;WAA4C;QAyB5C;;;;mBAAO,QAAQ;WAAC;QAgDhB;;;;mBACE,qHAAqH;WAAC;QAlEtH,oCAAoC;QACpC,IAAI,OAAO,KAAK,IAAI,EAAE;YACpB,MAAM,IAAI,KAAK,CACb,mEAAmE,CACpE,CAAC;SACH;QAED,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CACb,sGAAsG,CACvG,CAAC;SACH;QAED,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC;QAClB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAID;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,KAAa;;QACtB,IAAI,OAAO,KAAK,IAAI,EAAE;YACpB,MAAM,IAAI,KAAK,CACb,mEAAmE,CACpE,CAAC;SACH;QACD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,QAAQ,EAAE;YAClC,GAAG,IAAI,CAAC,MAAM;YACd,OAAO,EAAE,IAAI,CAAC,GAAG;YACjB,CAAC,EAAE,KAAK;SACT,CAAC,CAAC;QAEH,IAAI,GAAG,CAAC,KAAK,EAAE;YACb,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC;SACzD;QAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,MAAM,EAAE;YAC1B,OAAO,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC;SAC9B;QAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,OAAO,EAAE;YAC3B,OAAO,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC;SAC/B;QAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,yBAAyB,EAAE;YAC7C,OAAO,GAAG,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC;SACpD;QAED,IAAI,MAAA,GAAG,CAAC,cAAc,0CAAE,cAAc,EAAE;YACtC,OAAO,GAAG,CAAC,cAAc,CAAC,cAAc,CAAC;SAC1C;QAED,IAAI,MAAA,GAAG,CAAC,eAAe,0CAAE,WAAW,EAAE;YACpC,OAAO,GAAG,CAAC,eAAe,CAAC,WAAW,CAAC;SACxC;QAED,IAAI,MAAA,MAAA,GAAG,CAAC,eAAe,0CAAG,CAAC,CAAC,0CAAE,OAAO,EAAE;YACrC,OAAO,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;SACvC;QAED,OAAO,6BAA6B,CAAC;IACvC,CAAC;CAIF;AA9ED,0BA8EC"}

package/dist/{cjs/agents → agents}/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../agents/types.ts"],"names":[],"mappings":""}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../agents/types.ts"],"names":[],"mappings":""}

package/dist/cache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache.js","sourceRoot":"","sources":["../cache.ts"],"names":[],"mappings":";;;AAEA,MAAM,MAAM,GAAG,CAAC,MAAc,EAAE,MAAc,EAAU,EAAE,CACxD,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;AAExB,MAAsB,SAAS;CAI9B;AAJD,8BAIC;AAED,MAAa,aAAgC,SAAQ,SAAY;IAG/D;QACE,KAAK,EAAE,CAAC;QAHV;;;;;WAAiC;QAI/B,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;IAClB,CAAC;IAED,MAAM,CAAC,MAAc,EAAE,MAAc;QACnC,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,MAAM,CAAC,MAAc,EAAE,MAAc,EAAE,KAAQ;QAC7C,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,GAAG,KAAK,CAAC;IAC7C,CAAC;CACF;AAfD,sCAeC"}

package/dist/chains/base.d.ts

@@ -0,0 +1,42 @@
+import { LLMChain, StuffDocumentsChain, VectorDBQAChain, ChatVectorDBQAChain } from "./index";
+import { BaseMemory } from "../memory";
+export type ChainValues = Record<string, any>;
+export type LoadValues = Record<string, any>;
+declare const chainClasses: (typeof LLMChain | typeof StuffDocumentsChain | typeof VectorDBQAChain | typeof ChatVectorDBQAChain)[];
+export type SerializedBaseChain = ReturnType<InstanceType<(typeof chainClasses)[number]>["serialize"]>;
+export interface ChainInputs {
+    memory?: BaseMemory;
+}
+/**
+ * Base interface that all chains must implement.
+ */
+export declare abstract class BaseChain implements ChainInputs {
+    memory?: BaseMemory;
+    /**
+     * Run the core logic of this chain and return the output
+     */
+    abstract _call(values: ChainValues): Promise<ChainValues>;
+    /**
+     * Return the string type key uniquely identifying this class of chain.
+     */
+    abstract _chainType(): string;
+    /**
+     * Return a json-like object representing this chain.
+     */
+    abstract serialize(): SerializedBaseChain;
+    /**
+     * Run the core logic of this chain and add to output if desired.
+     *
+     * Wraps {@link _call} and handles memory.
+     */
+    call(values: ChainValues): Promise<ChainValues>;
+    /**
+     * Call the chain on all inputs in the list
+     */
+    apply(inputs: ChainValues[]): ChainValues[];
+    /**
+     * Load a chain from a json-like object describing it.
+     */
+    static deserialize(data: SerializedBaseChain, values?: LoadValues): Promise<BaseChain>;
+}
+export {};

package/dist/chains/base.js

@@ -0,0 +1,67 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseChain = void 0;
+const deepcopy = require("deepcopy");
+const index_1 = require("./index");
+const chainClasses = [
+    index_1.LLMChain,
+    index_1.StuffDocumentsChain,
+    index_1.VectorDBQAChain,
+    index_1.ChatVectorDBQAChain,
+];
+/**
+ * Base interface that all chains must implement.
+ */
+class BaseChain {
+    constructor() {
+        Object.defineProperty(this, "memory", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+    }
+    /**
+     * Run the core logic of this chain and add to output if desired.
+     *
+     * Wraps {@link _call} and handles memory.
+     */
+    async call(values) {
+        const fullValues = deepcopy(values);
+        if (!(this.memory == null)) {
+            const newValues = await this.memory.loadMemoryVariables(values);
+            for (const [key, value] of Object.entries(newValues)) {
+                fullValues[key] = value;
+            }
+        }
+        // TODO(sean) add callback support
+        const outputValues = this._call(fullValues);
+        if (!(this.memory == null)) {
+            this.memory.saveContext(values, outputValues);
+        }
+        return outputValues;
+    }
+    /**
+     * Call the chain on all inputs in the list
+     */
+    apply(inputs) {
+        return inputs.map(this.call);
+    }
+    /**
+     * Load a chain from a json-like object describing it.
+     */
+    static deserialize(data, values = {}) {
+        switch (data._type) {
+            case "llm_chain":
+                return index_1.LLMChain.deserialize(data);
+            case "stuff_documents_chain":
+                return index_1.StuffDocumentsChain.deserialize(data);
+            case "vector_db_qa":
+                return index_1.VectorDBQAChain.deserialize(data, values);
+            default:
+                throw new Error(`Invalid prompt type in config: ${data._type}`);
+        }
+    }
+}
+exports.BaseChain = BaseChain;
+//# sourceMappingURL=base.js.map

package/dist/chains/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../chains/base.ts"],"names":[],"mappings":";;;AAAA,qCAAsC;AACtC,mCAKiB;AAQjB,MAAM,YAAY,GAAG;IACnB,gBAAQ;IACR,2BAAmB;IACnB,uBAAe;IACf,2BAAmB;CACpB,CAAC;AAUF;;GAEG;AACH,MAAsB,SAAS;IAA/B;QACE;;;;;WAAoB;IAmEtB,CAAC;IAlDC;;;;OAIG;IACH,KAAK,CAAC,IAAI,CAAC,MAAmB;QAC5B,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,EAAE;YAC1B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAChE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;gBACpD,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;aACzB;SACF;QACD,kCAAkC;QAClC,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,EAAE;YAC1B,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;SAC/C;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAqB;QACzB,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,WAAW,CAChB,IAAyB,EACzB,SAAqB,EAAE;QAEvB,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,WAAW;gBACd,OAAO,gBAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACpC,KAAK,uBAAuB;gBAC1B,OAAO,2BAAmB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAC/C,KAAK,cAAc;gBACjB,OAAO,uBAAe,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACnD;gBACE,MAAM,IAAI,KAAK,CACb,kCACG,IAA4B,CAAC,KAChC,EAAE,CACH,CAAC;SACL;IACH,CAAC;CACF;AApED,8BAoEC"}

package/dist/chains/chat_vector_db_chain.d.ts

@@ -0,0 +1,41 @@
+import { BaseChain, ChainValues, SerializedStuffDocumentsChain, StuffDocumentsChain, SerializedLLMChain, LLMChain } from "./index";
+import { VectorStore } from "../vectorstores/base";
+import { BaseLLM } from "../llms";
+export type LoadValues = Record<string, any>;
+export interface ChatVectorDBQAChainInput {
+    vectorstore: VectorStore;
+    k: number;
+    combineDocumentsChain: StuffDocumentsChain;
+    questionGeneratorChain: LLMChain;
+    outputKey: string;
+    inputKey: string;
+}
+export type SerializedChatVectorDBQAChain = {
+    _type: "chat-vector-db";
+    k: number;
+    combine_documents_chain: SerializedStuffDocumentsChain;
+    combine_documents_chain_path?: string;
+    question_generator: SerializedLLMChain;
+};
+export declare class ChatVectorDBQAChain extends BaseChain implements ChatVectorDBQAChainInput {
+    k: number;
+    inputKey: string;
+    chatHistoryKey: string;
+    outputKey: string;
+    vectorstore: VectorStore;
+    combineDocumentsChain: StuffDocumentsChain;
+    questionGeneratorChain: LLMChain;
+    constructor(fields: {
+        vectorstore: VectorStore;
+        combineDocumentsChain: StuffDocumentsChain;
+        questionGeneratorChain: LLMChain;
+        inputKey?: string;
+        outputKey?: string;
+        k?: number;
+    });
+    _call(values: ChainValues): Promise<ChainValues>;
+    _chainType(): "chat-vector-db";
+    static deserialize(data: SerializedChatVectorDBQAChain, values: LoadValues): Promise<ChatVectorDBQAChain>;
+    serialize(): SerializedChatVectorDBQAChain;
+    static fromLLM(llm: BaseLLM, vectorstore: VectorStore): ChatVectorDBQAChain;
+}