langchain 0.0.4

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

@@ -0,0 +1,33 @@
+import { BaseLLM } from "../../llms";
+import { Agent, Tool, AgentInput, SerializedAgentT } from "../index";
+import { PromptTemplate } from "../../prompt";
+type SerializedFromLLMAndTools = {
+    suffix?: string;
+    prefix?: string;
+    input_variables?: string[];
+};
+export type SerializedZeroShotAgent = SerializedAgentT<"zero-shot-react-description", SerializedFromLLMAndTools, AgentInput>;
+type CreatePromptArgs = {
+    suffix?: string;
+    prefix?: string;
+    inputVariables?: string[];
+};
+type ZeroShotAgentInput = AgentInput;
+export declare class ZeroShotAgent extends Agent {
+    constructor(input: ZeroShotAgentInput);
+    _agentType(): "zero-shot-react-description";
+    observationPrefix(): string;
+    llmPrefix(): string;
+    static validateTools(tools: Tool[]): void;
+    static createPrompt(tools: Tool[], args?: CreatePromptArgs): PromptTemplate;
+    static fromLLMAndTools(llm: BaseLLM, tools: Tool[], args?: CreatePromptArgs): ZeroShotAgent;
+    extractToolAndInput(text: string): {
+        tool: string;
+        input: string;
+    } | null;
+    static deserialize(data: SerializedZeroShotAgent & {
+        llm?: BaseLLM;
+        tools?: Tool[];
+    }): Promise<ZeroShotAgent>;
+}
+export {};

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

@@ -0,0 +1,106 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    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";
+import { PromptTemplate } from "../../prompt";
+import { PREFIX, SUFFIX, formatInstructions } from "./prompt";
+import { deserializeHelper } from "../helpers";
+const FINAL_ANSWER_ACTION = "Final Answer:";
+let ZeroShotAgent = ZeroShotAgent_1 = class ZeroShotAgent extends Agent {
+    constructor(input) {
+        super(input);
+    }
+    _agentType() {
+        return "zero-shot-react-description";
+    }
+    observationPrefix() {
+        return "Observation: ";
+    }
+    llmPrefix() {
+        return "Thought:";
+    }
+    static validateTools(tools) {
+        const invalidTool = tools.find((tool) => !tool.description);
+        if (invalidTool) {
+            const msg = `Got a tool ${invalidTool.name} without a description.` +
+                ` This agent requires descriptions for all tools.`;
+            throw new Error(msg);
+        }
+    }
+    static createPrompt(tools, args) {
+        const { prefix = PREFIX, suffix = SUFFIX, inputVariables = ["input", "agent_scratchpad"], } = args !== null && args !== void 0 ? args : {};
+        const toolStrings = tools
+            .map((tool) => `${tool.name}: ${tool.description}`)
+            .join("\n");
+        const instructions = formatInstructions(toolStrings);
+        const template = [prefix, toolStrings, instructions, suffix].join("\n\n");
+        return new PromptTemplate({
+            template,
+            inputVariables,
+        });
+    }
+    static fromLLMAndTools(llm, tools, args) {
+        ZeroShotAgent_1.validateTools(tools);
+        const prompt = ZeroShotAgent_1.createPrompt(tools, args);
+        const chain = new LLMChain({ prompt, llm });
+        return new ZeroShotAgent_1({
+            llmChain: chain,
+            allowedTools: tools.map((t) => t.name),
+        });
+    }
+    extractToolAndInput(text) {
+        if (text.includes(FINAL_ANSWER_ACTION)) {
+            const parts = text.split(FINAL_ANSWER_ACTION);
+            const input = parts[parts.length - 1].trim();
+            return { tool: "Final Answer", input };
+        }
+        const match = /Action: (.*)\nAction Input: (.*)/s.exec(text);
+        if (!match) {
+            throw new Error(`Could not parse LLM output: ${text}`);
+        }
+        return {
+            tool: match[1].trim(),
+            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));
+        });
+    }
+};
+ZeroShotAgent = ZeroShotAgent_1 = __decorate([
+    (staticImplements)
+], ZeroShotAgent);
+export { ZeroShotAgent };
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +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"}

package/dist/esm/agents/mrkl/prompt.d.ts

@@ -0,0 +1,3 @@
+export declare const PREFIX = "Answer the following questions as best you can. You have access to the following tools:";
+export declare const formatInstructions: (toolNames: string) => string;
+export declare const SUFFIX = "Begin!\n\nQuestion: {input}\nThought:{agent_scratchpad}";

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

@@ -0,0 +1,17 @@
+/* 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:
+
+Question: the input question you must answer
+Thought: you should always think about what to do
+Action: the action to take, should be one of [${toolNames}]
+Action Input: the input to the action
+Observation: the result of the action
+... (this Thought/Action/Action Input/Observation can repeat N times)
+Thought: I now know the final answer
+Final Answer: the final answer to the original input question`;
+export const SUFFIX = `Begin!
+
+Question: {input}
+Thought:{agent_scratchpad}`;
+//# sourceMappingURL=prompt.js.map

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

@@ -0,0 +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"}

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

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

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

@@ -0,0 +1,29 @@
+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 });
+    const executor = AgentExecutor.fromAgentAndTools({
+        agent,
+        tools,
+        returnIntermediateSteps: true,
+    });
+    const res = yield executor.call({
+        input: "Who is Olivia Wilde's boyfriend? What is his current age raised to the 0.23 power?",
+    });
+    console.log(res);
+}), 30000);
+//# sourceMappingURL=agent.test.js.map

package/dist/esm/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,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"}

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

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

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

@@ -0,0 +1,2 @@
+export { SerpAPI } from "./serpapi";
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../agents/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}

package/dist/esm/agents/tools/llm-math.js

@@ -0,0 +1,2 @@
+"use strict";
+//# sourceMappingURL=llm-math.js.map

package/dist/esm/agents/tools/llm-math.js.map

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

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

@@ -0,0 +1,3 @@
+import { GoogleParameters } from "serpapi";
+import { Tool } from "./index";
+export declare const SerpAPI: (params?: Partial<GoogleParameters>, apiKey?: string) => Tool;

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

@@ -0,0 +1,46 @@
+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.",
+    };
+};
+//# sourceMappingURL=serpapi.js.map

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

@@ -0,0 +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"}

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

@@ -0,0 +1,25 @@
+import { SerializedLLMChain } from "../chains";
+import { AgentInput } from "./index";
+export type AgentAction = {
+    tool: string;
+    toolInput: string;
+    log: string;
+};
+export type AgentFinish = {
+    returnValues: Record<string, any>;
+    log: string;
+};
+export type AgentStep = {
+    action: AgentAction;
+    observation: string;
+};
+export type StoppingMethod = "force" | "generate";
+export type SerializedAgentT<TType extends string, FromLLMInput, ConstructorInput extends AgentInput> = {
+    _type: TType;
+    llm_chain?: SerializedLLMChain;
+    llm_chain_path?: string;
+} & (({
+    load_from_llm_and_tools: true;
+} & FromLLMInput) | ({
+    load_from_llm_and_tools?: false;
+} & ConstructorInput));

package/dist/esm/agents/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

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

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

package/dist/esm/cache.d.ts

@@ -0,0 +1,11 @@
+import { Generation } from "./llms";
+export declare abstract class BaseCache<T = Generation[]> {
+    abstract lookup(prompt: string, llmKey: string): T | undefined;
+    abstract update(prompt: string, llmKey: string, value: T): void;
+}
+export declare class InMemoryCache<T = Generation[]> extends BaseCache<T> {
+    private cache;
+    constructor();
+    lookup(prompt: string, llmKey: string): T;
+    update(prompt: string, llmKey: string, value: T): void;
+}

package/dist/esm/cache.js

@@ -0,0 +1,22 @@
+const getKey = (prompt, llmKey) => `${prompt}_${llmKey}`;
+export class BaseCache {
+}
+export class InMemoryCache extends BaseCache {
+    constructor() {
+        super();
+        Object.defineProperty(this, "cache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.cache = {};
+    }
+    lookup(prompt, llmKey) {
+        return this.cache[getKey(prompt, llmKey)];
+    }
+    update(prompt, llmKey, value) {
+        this.cache[getKey(prompt, llmKey)] = value;
+    }
+}
+//# sourceMappingURL=cache.js.map

package/dist/esm/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,MAAM,OAAgB,SAAS;CAI9B;AAED,MAAM,OAAO,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"}

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

@@ -0,0 +1,12 @@
+import { SerializedLLMChain } from "./index";
+export type ChainValues = Record<string, any>;
+type SerializedBaseChain = SerializedLLMChain;
+export declare abstract class BaseChain {
+    abstract _call(values: ChainValues): Promise<ChainValues>;
+    abstract _chainType(): string;
+    abstract serialize(): SerializedBaseChain;
+    call(values: ChainValues): Promise<ChainValues>;
+    apply(inputs: ChainValues[]): ChainValues[];
+    static deserialize(data: SerializedBaseChain): Promise<BaseChain>;
+}
+export {};

package/dist/esm/chains/base.js

@@ -0,0 +1,19 @@
+import { LLMChain } from "./index";
+export class BaseChain {
+    call(values) {
+        // TODO(sean) add callback support
+        return this._call(values);
+    }
+    apply(inputs) {
+        return inputs.map(this.call);
+    }
+    static deserialize(data) {
+        switch (data._type) {
+            case "llm_chain":
+                return LLMChain.deserialize(data);
+            default:
+                throw new Error(`Invalid prompt type in config: ${data._type}`);
+        }
+    }
+}
+//# sourceMappingURL=base.js.map

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

@@ -0,0 +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"}

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

@@ -0,0 +1,3 @@
+export { BaseChain, ChainValues } from "./base";
+export { SerializedLLMChain, LLMChain } from "./llm_chain";
+export { loadChain } from "./load";

package/dist/esm/chains/index.js

@@ -0,0 +1,4 @@
+export { BaseChain } from "./base";
+export { LLMChain } from "./llm_chain";
+export { loadChain } from "./load";
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../chains/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAe,MAAM,QAAQ,CAAC;AAChD,OAAO,EAAsB,QAAQ,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC"}

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

@@ -0,0 +1,30 @@
+import { BaseChain, ChainValues } from "./index";
+import { BaseLLM, SerializedLLM } from "../llms";
+import { BasePromptTemplate, SerializedBasePromptTemplate } from "../prompt";
+export interface LLMChainInput {
+    prompt: BasePromptTemplate;
+    llm: BaseLLM;
+    outputKey: string;
+}
+export type SerializedLLMChain = {
+    _type: "llm_chain";
+    llm?: SerializedLLM;
+    llm_path?: string;
+    prompt?: SerializedBasePromptTemplate;
+    prompt_path?: string;
+};
+export declare class LLMChain extends BaseChain implements LLMChainInput {
+    prompt: BasePromptTemplate;
+    llm: BaseLLM;
+    outputKey: string;
+    constructor(fields: {
+        prompt: BasePromptTemplate;
+        llm: BaseLLM;
+        outputKey?: string;
+    });
+    _call(values: ChainValues): Promise<ChainValues>;
+    predict(values: ChainValues): Promise<string>;
+    _chainType(): "llm_chain";
+    static deserialize(data: SerializedLLMChain): Promise<LLMChain>;
+    serialize(): SerializedLLMChain;
+}

package/dist/esm/chains/llm_chain.js

@@ -0,0 +1,79 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import { BaseChain } from "./index";
+import { BaseLLM } from "../llms";
+import { BasePromptTemplate } from "../prompt";
+import { resolveConfigFromFile } from "../util";
+export class LLMChain extends BaseChain {
+    constructor(fields) {
+        var _a;
+        super();
+        Object.defineProperty(this, "prompt", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "llm", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "outputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "text"
+        });
+        this.prompt = fields.prompt;
+        this.llm = fields.llm;
+        this.outputKey = (_a = fields.outputKey) !== null && _a !== void 0 ? _a : this.outputKey;
+    }
+    _call(values) {
+        return __awaiter(this, void 0, void 0, function* () {
+            let stop;
+            if ("stop" in values && Array.isArray(values.stop)) {
+                stop = values.stop;
+            }
+            const formattedString = this.prompt.format(values);
+            const llmResult = yield this.llm.call(formattedString, stop);
+            const result = { [this.outputKey]: llmResult };
+            return result;
+        });
+    }
+    predict(values) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const output = yield this.call(values);
+            return output[this.outputKey];
+        });
+    }
+    _chainType() {
+        return "llm_chain";
+    }
+    static deserialize(data) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const serializedLLM = resolveConfigFromFile("llm", data);
+            const serializedPrompt = resolveConfigFromFile("prompt", data);
+            return new LLMChain({
+                llm: yield BaseLLM.deserialize(serializedLLM),
+                prompt: yield BasePromptTemplate.deserialize(serializedPrompt),
+            });
+        });
+    }
+    serialize() {
+        return {
+            _type: this._chainType(),
+            llm: this.llm.serialize(),
+            prompt: this.prompt.serialize(),
+        };
+    }
+}
+//# sourceMappingURL=llm_chain.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"llm_chain.js","sourceRoot":"","sources":["../../../chains/llm_chain.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,EAAE,SAAS,EAAe,MAAM,SAAS,CAAC;AAEjD,OAAO,EAAE,OAAO,EAAiB,MAAM,SAAS,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAgC,MAAM,WAAW,CAAC;AAE7E,OAAO,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;AAgBhD,MAAM,OAAO,QAAS,SAAQ,SAAS;IAOrC,YAAY,MAIX;;QACC,KAAK,EAAE,CAAC;QAXV;;;;;WAA2B;QAE3B;;;;;WAAa;QAEb;;;;mBAAY,MAAM;WAAC;QAQjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,CAAC,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;IACtD,CAAC;IAEK,KAAK,CAAC,MAAmB;;YAC7B,IAAI,IAAI,CAAC;YACT,IAAI,MAAM,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;gBAClD,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;aACpB;YACD,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACnD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC;YAC/C,OAAO,MAAM,CAAC;QAChB,CAAC;KAAA;IAEK,OAAO,CAAC,MAAmB;;YAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvC,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChC,CAAC;KAAA;IAED,UAAU;QACR,OAAO,WAAoB,CAAC;IAC9B,CAAC;IAED,MAAM,CAAO,WAAW,CAAC,IAAwB;;YAC/C,MAAM,aAAa,GAAG,qBAAqB,CACzC,KAAK,EACL,IAAI,CACL,CAAC;YACF,MAAM,gBAAgB,GAAG,qBAAqB,CAG5C,QAAQ,EAAE,IAAI,CAAC,CAAC;YAElB,OAAO,IAAI,QAAQ,CAAC;gBAClB,GAAG,EAAE,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC;gBAC7C,MAAM,EAAE,MAAM,kBAAkB,CAAC,WAAW,CAAC,gBAAgB,CAAC;aAC/D,CAAC,CAAC;QACL,CAAC;KAAA;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;SAChC,CAAC;IACJ,CAAC;CACF"}