langchain 0.0.4 → 0.0.5

package/dist/esm/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): {

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

@@ -4,26 +4,6 @@ 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;
 import { LLMChain } from "../../chains";
 import { Agent, staticImplements, } from "../index";
@@ -31,6 +11,11 @@ import { PromptTemplate } from "../../prompt";
 import { PREFIX, SUFFIX, formatInstructions } from "./prompt";
 import { deserializeHelper } from "../helpers";
 const FINAL_ANSWER_ACTION = "Final Answer:";
+/**
+ * Agent for the MRKL chain.
+ * @augments Agent
+ * @augments StaticAgent
+ */
 let ZeroShotAgent = ZeroShotAgent_1 = class ZeroShotAgent extends Agent {
     constructor(input) {
         super(input);
@@ -52,6 +37,15 @@ let ZeroShotAgent = ZeroShotAgent_1 = class ZeroShotAgent extends 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 = PREFIX, suffix = SUFFIX, inputVariables = ["input", "agent_scratchpad"], } = args !== null && args !== void 0 ? args : {};
         const toolStrings = tools
@@ -88,15 +82,13 @@ let ZeroShotAgent = ZeroShotAgent_1 = class ZeroShotAgent extends 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 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 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/esm/agents/mrkl/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../agents/mrkl/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EACL,KAAK,EAIL,gBAAgB,GAEjB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE/C,MAAM,mBAAmB,GAAG,eAAe,CAAC;AAuBrC,IAAM,aAAa,qBAAnB,MAAM,aAAc,SAAQ,KAAK;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,MAAM,CAAC,YAAY,CAAC,KAAa,EAAE,IAAuB;QACxD,MAAM,EACJ,MAAM,GAAG,MAAM,EACf,MAAM,GAAG,MAAM,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,kBAAkB,CAAC,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,cAAc,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,QAAQ,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,CAAO,WAAW,CACtB,IAAiE;;YAEjE,MAAM,EAAE,GAAG,EAAE,KAAK,KAAc,IAAI,EAAb,IAAI,UAAK,IAAI,EAA9B,gBAAuB,CAAO,CAAC;YACrC,OAAO,iBAAiB,CACtB,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;gBACxC,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,cAAc,EAAE,IAAI,CAAC,eAAe;aACrC,CAAC,EACJ,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,eAAa,CAAC,IAAI,CAAC,CAClC,CAAC;QACJ,CAAC;KAAA;CACF,CAAA;AA1FY,aAAa;IADzB,CAAC,gBAA6B,CAAC;GACnB,aAAa,CA0FzB;SA1FY,aAAa"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../agents/mrkl/index.ts"],"names":[],"mappings":";;;;;;;AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EACL,KAAK,EAIL,gBAAgB,GAEjB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE/C,MAAM,mBAAmB,GAAG,eAAe,CAAC;AAyB5C;;;;GAIG;AAEI,IAAM,aAAa,qBAAnB,MAAM,aAAc,SAAQ,KAAK;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,MAAM,EACf,MAAM,GAAG,MAAM,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,kBAAkB,CAAC,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,cAAc,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,QAAQ,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,iBAAiB,CACtB,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,gBAA6B,CAAC;GACnB,aAAa,CAmGzB;SAnGY,aAAa"}

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

@@ -1,4 +1,3 @@
-/* eslint-disable max-len */
 export const PREFIX = `Answer the following questions as best you can. You have access to the following tools:`;
 export const formatInstructions = (toolNames) => `Use the following format:
 

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

@@ -1 +1 @@
-{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../../../agents/mrkl/prompt.ts"],"names":[],"mappings":"AAAA,4BAA4B;AAC5B,MAAM,CAAC,MAAM,MAAM,GAAG,yFAAyF,CAAC;AAChH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAChC,SAAiB,EACjB,EAAE,CAAC;;;;gDAI2C,SAAS;;;;;8DAKK,CAAC;AAC/D,MAAM,CAAC,MAAM,MAAM,GAAG;;;2BAGK,CAAC"}
+{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../../../agents/mrkl/prompt.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,MAAM,GAAG,yFAAyF,CAAC;AAChH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAChC,SAAiB,EACjB,EAAE,CAAC;;;;gDAI2C,SAAS;;;;;8DAKK,CAAC;AAC/D,MAAM,CAAC,MAAM,MAAM,GAAG;;;2BAGK,CAAC"}

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

@@ -1,29 +1,33 @@
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
 import { test } from "@jest/globals";
 import { OpenAI } from "../../llms/openai";
 import { loadAgent } from "../load";
 import { AgentExecutor } from "../index";
 import { SerpAPI } from "../tools/serpapi";
-test("Run agent from hub", () => __awaiter(void 0, void 0, void 0, function* () {
-    const model = new OpenAI({});
-    const tools = [SerpAPI()];
-    const agent = yield loadAgent("lc://agents/zero-shot-react-description/agent.json", { llm: model, tools });
+import { Calculator } from "../tools/calculator";
+import { initializeAgentExecutor } from "../initialize";
+test("Run agent from hub", async () => {
+    const model = new OpenAI({ temperature: 0 });
+    const tools = [new SerpAPI(), new Calculator()];
+    const agent = await loadAgent("lc://agents/zero-shot-react-description/agent.json", { llm: model, tools });
     const executor = AgentExecutor.fromAgentAndTools({
         agent,
         tools,
         returnIntermediateSteps: true,
     });
-    const res = yield executor.call({
+    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);
+}, 30000);
+test("Run agent locally", async () => {
+    const model = new OpenAI({ temperature: 0 });
+    const tools = [new SerpAPI(), new Calculator()];
+    const executor = await 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/esm/agents/tests/agent.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent.test.js","sourceRoot":"","sources":["../../../../agents/tests/agent.test.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,aAAa,EAAQ,MAAM,UAAU,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAE3C,IAAI,CAAC,oBAAoB,EAAE,GAAS,EAAE;IACpC,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAW,CAAC,OAAO,EAAE,CAAC,CAAC;IAClC,MAAM,KAAK,GAAG,MAAM,SAAS,CAC3B,oDAAoD,EACpD,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,CACtB,CAAC;IACF,MAAM,QAAQ,GAAG,aAAa,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,CAAA,EAAE,KAAK,CAAC,CAAC"}
+{"version":3,"file":"agent.test.js","sourceRoot":"","sources":["../../../../agents/tests/agent.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,aAAa,EAAQ,MAAM,UAAU,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAExD,IAAI,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;IACpC,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAC,WAAW,EAAE,CAAC,EAAC,CAAC,CAAC;IAC3C,MAAM,KAAK,GAAW,CAAC,IAAI,OAAO,EAAE,EAAE,IAAI,UAAU,EAAE,CAAC,CAAC;IACxD,MAAM,KAAK,GAAG,MAAM,SAAS,CAC3B,oDAAoD,EACpD,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,CACtB,CAAC;IACF,MAAM,QAAQ,GAAG,aAAa,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;AAGV,IAAI,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;IACnC,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAC,WAAW,EAAE,CAAC,EAAC,CAAC,CAAC;IAC3C,MAAM,KAAK,GAAG,CAAC,IAAI,OAAO,EAAE,EAAE,IAAI,UAAU,EAAE,CAAC,CAAC;IAEhD,MAAM,QAAQ,GAAG,MAAM,uBAAuB,CAC5C,KAAK,EAAE,KAAK,EAAE,6BAA6B,CAC5C,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;IAE7B,MAAM,KAAK,GAAG,kCAAkC;QAC9C,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/esm/agents/tests/calculator.test.d.ts

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

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

@@ -0,0 +1,28 @@
+import { test, expect } from "@jest/globals";
+import { Calculator } from "../tools/calculator";
+test("Calculator tool, sum", async () => {
+    const calculator = new Calculator();
+    const result = await calculator.call("1 + 1");
+    expect(result).toBe("2");
+});
+test("Calculator tool, product", async () => {
+    const calculator = new Calculator();
+    const result = await calculator.call("2 * 3");
+    expect(result).toBe("6");
+});
+test("Calculator tool, division", async () => {
+    const calculator = new Calculator();
+    const result = await calculator.call("7 /2");
+    expect(result).toBe("3.5");
+});
+test("Calculator tool, exponentiation", async () => {
+    const calculator = new Calculator();
+    const result = await calculator.call("2 ^ 8");
+    expect(result).toBe("256");
+});
+test("Calculator tool, complicated expression", async () => {
+    const calculator = new Calculator();
+    const result = await calculator.call("((2 + 3) * 4) / 2");
+    expect(result).toBe("10");
+});
+//# sourceMappingURL=calculator.test.js.map

package/dist/esm/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,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEjD,IAAI,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;IACtC,MAAM,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC3B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;IAC1C,MAAM,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC3B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;IAC3C,MAAM,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;IACjD,MAAM,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;IACzD,MAAM,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC,CAAC,CAAC"}

package/dist/esm/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/esm/agents/tools/base.js

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

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

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

package/dist/esm/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/esm/agents/tools/calculator.js

@@ -0,0 +1,29 @@
+import { Parser } from "expr-eval";
+import { Tool } from "./base";
+export class Calculator extends 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 Parser.evaluate(input).toString();
+        }
+        catch (error) {
+            return "I don't know how to do that.";
+        }
+    }
+}
+//# sourceMappingURL=calculator.js.map

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

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

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

@@ -1,7 +1,3 @@
 export { SerpAPI } from "./serpapi";
-export interface Tool {
-    call: (arg: string) => Promise<string>;
-    name: string;
-    description: string;
-    returnDirect?: boolean;
-}
+export { Calculator } from "./calculator";
+export { Tool } from "./base";

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

@@ -1,2 +1,4 @@
 export { SerpAPI } from "./serpapi";
+export { Calculator } from "./calculator";
+export { Tool } from "./base";
 //# sourceMappingURL=index.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../agents/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../agents/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC"}

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

@@ -1,3 +1,18 @@
-import { GoogleParameters } from "serpapi";
-import { Tool } from "./index";
-export declare const SerpAPI: (params?: Partial<GoogleParameters>, apiKey?: string) => Tool;
+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/esm/agents/tools/serpapi.js

@@ -1,46 +1,89 @@
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-import { getJson } from "serpapi";
-export const SerpAPI = (params, apiKey) => {
-    const key = apiKey !== null && apiKey !== void 0 ? apiKey : process.env.SERPAPI_API_KEY;
-    return {
-        name: "search",
-        call: (input) => __awaiter(void 0, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g;
-            const res = yield getJson("google", Object.assign(Object.assign({}, params), { api_key: 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";
-        }),
-        description: 
-        // eslint-disable-next-line max-len
-        "a search engine. useful for when you need to answer questions about current events. input should be a search query.",
-    };
-};
+import { Tool } from "./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.
+ */
+export class SerpAPI extends 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";
+    }
+}
 //# sourceMappingURL=serpapi.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"serpapi.js","sourceRoot":"","sources":["../../../../agents/tools/serpapi.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,EAAE,OAAO,EAAoB,MAAM,SAAS,CAAC;AAGpD,MAAM,CAAC,MAAM,OAAO,GAAG,CACrB,MAAkC,EAClC,MAAe,EACT,EAAE;IACR,MAAM,GAAG,GAAG,MAAM,aAAN,MAAM,cAAN,MAAM,GAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAClD,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,CAAO,KAAa,EAAE,EAAE;;YAC5B,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,QAAQ,kCAC7B,MAAM,KACT,OAAO,EAAE,GAAG,EACZ,CAAC,EAAE,KAAK,IACR,CAAC;YAEH,IAAI,GAAG,CAAC,KAAK,EAAE;gBACb,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC;aACzD;YAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,MAAM,EAAE;gBAC1B,OAAO,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC;aAC9B;YAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,OAAO,EAAE;gBAC3B,OAAO,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC;aAC/B;YAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,yBAAyB,EAAE;gBAC7C,OAAO,GAAG,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC;aACpD;YAED,IAAI,MAAA,GAAG,CAAC,cAAc,0CAAE,cAAc,EAAE;gBACtC,OAAO,GAAG,CAAC,cAAc,CAAC,cAAc,CAAC;aAC1C;YAED,IAAI,MAAA,GAAG,CAAC,eAAe,0CAAE,WAAW,EAAE;gBACpC,OAAO,GAAG,CAAC,eAAe,CAAC,WAAW,CAAC;aACxC;YAED,IAAI,MAAA,MAAA,GAAG,CAAC,eAAe,0CAAG,CAAC,CAAC,0CAAE,OAAO,EAAE;gBACrC,OAAO,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;aACvC;YAED,OAAO,6BAA6B,CAAC;QACvC,CAAC,CAAA;QAED,WAAW;QACT,mCAAmC;QACnC,qHAAqH;KACxH,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"file":"serpapi.js","sourceRoot":"","sources":["../../../../agents/tools/serpapi.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;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,MAAM,OAAO,OAAQ,SAAQ,IAAI;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"}

package/dist/esm/chains/base.d.ts

@@ -1,12 +1,42 @@
-import { SerializedLLMChain } from "./index";
+import { LLMChain, StuffDocumentsChain, VectorDBQAChain, ChatVectorDBQAChain } from "./index";
+import { BaseMemory } from "../memory";
 export type ChainValues = Record<string, any>;
-type SerializedBaseChain = SerializedLLMChain;
-export declare abstract class BaseChain {
+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[];
-    static deserialize(data: SerializedBaseChain): Promise<BaseChain>;
+    /**
+     * Load a chain from a json-like object describing it.
+     */
+    static deserialize(data: SerializedBaseChain, values?: LoadValues): Promise<BaseChain>;
 }
 export {};

package/dist/esm/chains/base.js

@@ -1,16 +1,54 @@
-import { LLMChain } from "./index";
+import { LLMChain, StuffDocumentsChain, VectorDBQAChain, ChatVectorDBQAChain } from "./index";
+const chainClasses = [LLMChain, StuffDocumentsChain, VectorDBQAChain, ChatVectorDBQAChain];
+/**
+ * Base interface that all chains must implement.
+ */
 export class BaseChain {
-    call(values) {
+    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 = structuredClone(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
-        return this._call(values);
+        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);
     }
-    static deserialize(data) {
+    /**
+     * Load a chain from a json-like object describing it.
+     */
+    static deserialize(data, values = {}) {
         switch (data._type) {
             case "llm_chain":
                 return LLMChain.deserialize(data);
+            case "stuff_documents_chain":
+                return StuffDocumentsChain.deserialize(data);
+            case "vector_db_qa":
+                return VectorDBQAChain.deserialize(data, values);
             default:
                 throw new Error(`Invalid prompt type in config: ${data._type}`);
         }

package/dist/esm/chains/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../chains/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAsB,MAAM,SAAS,CAAC;AAMvD,MAAM,OAAgB,SAAS;IAO7B,IAAI,CAAC,MAAmB;QACtB,kCAAkC;QAClC,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,MAAqB;QACzB,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM,CAAC,WAAW,CAAC,IAAyB;QAC1C,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,WAAW;gBACd,OAAO,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACpC;gBACE,MAAM,IAAI,KAAK,CACb,kCACG,IAA4B,CAAC,KAChC,EAAE,CACH,CAAC;SACL;IACH,CAAC;CACF"}
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../chains/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,mBAAmB,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAQ9F,MAAM,YAAY,GAAG,CAAC,QAAQ,EAAE,mBAAmB,EAAE,eAAe,EAAE,mBAAmB,CAAC,CAAC;AAU3F;;GAEG;AACH,MAAM,OAAgB,SAAS;IAA/B;QACE;;;;;WAAoB;IAmEtB,CAAC;IAlDC;;;;OAIG;IACH,KAAK,CAAC,IAAI,CAAC,MAAmB;QAC5B,MAAM,UAAU,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;QAC3C,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,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACpC,KAAK,uBAAuB;gBAC1B,OAAO,mBAAmB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAC/C,KAAK,cAAc;gBACjB,OAAO,eAAe,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"}