langchain 0.0.4 → 0.0.6

package/dist/chains/chat_vector_db_chain.js

@@ -0,0 +1,146 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatVectorDBQAChain = void 0;
+const index_1 = require("./index");
+const prompts_1 = require("../prompts");
+const util_1 = require("../util");
+const question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.
+
+Chat History:
+{chat_history}
+Follow Up Input: {question}
+Standalone question:`;
+const question_generator_prompt = prompts_1.PromptTemplate.fromTemplate(question_generator_template);
+const qa_template = `Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.
+
+{context}
+
+Question: {question}
+Helpful Answer:`;
+const qa_prompt = prompts_1.PromptTemplate.fromTemplate(qa_template);
+class ChatVectorDBQAChain extends index_1.BaseChain {
+    constructor(fields) {
+        var _a, _b, _c;
+        super();
+        Object.defineProperty(this, "k", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 4
+        });
+        Object.defineProperty(this, "inputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "question"
+        });
+        Object.defineProperty(this, "chatHistoryKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "chat_history"
+        });
+        Object.defineProperty(this, "outputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "result"
+        });
+        Object.defineProperty(this, "vectorstore", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "combineDocumentsChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "questionGeneratorChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.vectorstore = fields.vectorstore;
+        this.combineDocumentsChain = fields.combineDocumentsChain;
+        this.questionGeneratorChain = fields.questionGeneratorChain;
+        this.inputKey = (_a = fields.inputKey) !== null && _a !== void 0 ? _a : this.inputKey;
+        this.outputKey = (_b = fields.outputKey) !== null && _b !== void 0 ? _b : this.outputKey;
+        this.k = (_c = fields.k) !== null && _c !== void 0 ? _c : this.k;
+    }
+    async _call(values) {
+        if (!(this.inputKey in values)) {
+            throw new Error(`Question key ${this.inputKey} not found.`);
+        }
+        if (!(this.chatHistoryKey in values)) {
+            throw new Error(`chat history key ${this.inputKey} not found.`);
+        }
+        const question = values[this.inputKey];
+        const chatHistory = values[this.chatHistoryKey];
+        let newQuestion = question;
+        if (chatHistory.length > 0) {
+            const result = await this.questionGeneratorChain.call({
+                question,
+                chat_history: chatHistory,
+            });
+            const keys = Object.keys(result);
+            if (keys.length === 1) {
+                newQuestion = result[keys[0]];
+            }
+            else {
+                throw new Error("Return from llm chain has multiple values, only single values supported.");
+            }
+        }
+        const docs = await this.vectorstore.similaritySearch(newQuestion, this.k);
+        const inputs = {
+            question,
+            input_documents: docs,
+            chat_history: chatHistory,
+        };
+        const result = await this.combineDocumentsChain.call(inputs);
+        return result;
+    }
+    _chainType() {
+        return "chat-vector-db";
+    }
+    static async deserialize(data, values) {
+        if (!("vectorstore" in values)) {
+            throw new Error(`Need to pass in a vectorstore to deserialize VectorDBQAChain`);
+        }
+        const { vectorstore } = values;
+        const serializedCombineDocumentsChain = (0, util_1.resolveConfigFromFile)("combine_documents_chain", data);
+        const serializedQuestionGeneratorChain = (0, util_1.resolveConfigFromFile)("question_generator", data);
+        return new ChatVectorDBQAChain({
+            combineDocumentsChain: await index_1.StuffDocumentsChain.deserialize(serializedCombineDocumentsChain),
+            questionGeneratorChain: await index_1.LLMChain.deserialize(serializedQuestionGeneratorChain),
+            k: data.k,
+            vectorstore,
+        });
+    }
+    serialize() {
+        return {
+            _type: this._chainType(),
+            combine_documents_chain: this.combineDocumentsChain.serialize(),
+            question_generator: this.questionGeneratorChain.serialize(),
+            k: this.k,
+        };
+    }
+    static fromLLM(llm, vectorstore) {
+        const qaChain = (0, index_1.loadQAChain)(llm, qa_prompt);
+        const questionGeneratorChain = new index_1.LLMChain({
+            prompt: question_generator_prompt,
+            llm,
+        });
+        const instance = new this({
+            vectorstore,
+            combineDocumentsChain: qaChain,
+            questionGeneratorChain,
+        });
+        return instance;
+    }
+}
+exports.ChatVectorDBQAChain = ChatVectorDBQAChain;
+//# sourceMappingURL=chat_vector_db_chain.js.map

package/dist/chains/chat_vector_db_chain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat_vector_db_chain.js","sourceRoot":"","sources":["../../chains/chat_vector_db_chain.ts"],"names":[],"mappings":";;;AAAA,mCAQiB;AAEjB,wCAA4C;AAK5C,kCAAgD;AAIhD,MAAM,2BAA2B,GAAG;;;;;qBAKf,CAAC;AACtB,MAAM,yBAAyB,GAAG,wBAAc,CAAC,YAAY,CAC3D,2BAA2B,CAC5B,CAAC;AAEF,MAAM,WAAW,GAAG;;;;;gBAKJ,CAAC;AACjB,MAAM,SAAS,GAAG,wBAAc,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;AAmB3D,MAAa,mBACX,SAAQ,iBAAS;IAiBjB,YAAY,MAOX;;QACC,KAAK,EAAE,CAAC;QAtBV;;;;mBAAI,CAAC;WAAC;QAEN;;;;mBAAW,UAAU;WAAC;QAEtB;;;;mBAAiB,cAAc;WAAC;QAEhC;;;;mBAAY,QAAQ;WAAC;QAErB;;;;;WAAyB;QAEzB;;;;;WAA2C;QAE3C;;;;;WAAiC;QAW/B,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,qBAAqB,CAAC;QAC1D,IAAI,CAAC,sBAAsB,GAAG,MAAM,CAAC,sBAAsB,CAAC;QAC5D,IAAI,CAAC,QAAQ,GAAG,MAAA,MAAM,CAAC,QAAQ,mCAAI,IAAI,CAAC,QAAQ,CAAC;QACjD,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,CAAC,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;QACpD,IAAI,CAAC,CAAC,GAAG,MAAA,MAAM,CAAC,CAAC,mCAAI,IAAI,CAAC,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CAAC,gBAAgB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SAC7D;QACD,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,IAAI,MAAM,CAAC,EAAE;YACpC,MAAM,IAAI,KAAK,CAAC,oBAAoB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SACjE;QACD,MAAM,QAAQ,GAAW,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,WAAW,GAAW,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACxD,IAAI,WAAW,GAAG,QAAQ,CAAC;QAC3B,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE;YAC1B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC;gBACpD,QAAQ;gBACR,YAAY,EAAE,WAAW;aAC1B,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE;gBACrB,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;aAC/B;iBAAM;gBACL,MAAM,IAAI,KAAK,CACb,0EAA0E,CAC3E,CAAC;aACH;SACF;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1E,MAAM,MAAM,GAAG;YACb,QAAQ;YACR,eAAe,EAAE,IAAI;YACrB,YAAY,EAAE,WAAW;SAC1B,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,gBAAyB,CAAC;IACnC,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,IAAmC,EACnC,MAAkB;QAElB,IAAI,CAAC,CAAC,aAAa,IAAI,MAAM,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CACb,8DAA8D,CAC/D,CAAC;SACH;QACD,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAC/B,MAAM,+BAA+B,GAAG,IAAA,4BAAqB,EAG3D,yBAAyB,EAAE,IAAI,CAAC,CAAC;QACnC,MAAM,gCAAgC,GAAG,IAAA,4BAAqB,EAG5D,oBAAoB,EAAE,IAAI,CAAC,CAAC;QAE9B,OAAO,IAAI,mBAAmB,CAAC;YAC7B,qBAAqB,EAAE,MAAM,2BAAmB,CAAC,WAAW,CAC1D,+BAA+B,CAChC;YACD,sBAAsB,EAAE,MAAM,gBAAQ,CAAC,WAAW,CAChD,gCAAgC,CACjC;YACD,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,uBAAuB,EAAE,IAAI,CAAC,qBAAqB,CAAC,SAAS,EAAE;YAC/D,kBAAkB,EAAE,IAAI,CAAC,sBAAsB,CAAC,SAAS,EAAE;YAC3D,CAAC,EAAE,IAAI,CAAC,CAAC;SACV,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,WAAwB;QACnD,MAAM,OAAO,GAAG,IAAA,mBAAW,EAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC5C,MAAM,sBAAsB,GAAG,IAAI,gBAAQ,CAAC;YAC1C,MAAM,EAAE,yBAAyB;YACjC,GAAG;SACJ,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC;YACxB,WAAW;YACX,qBAAqB,EAAE,OAAO;YAC9B,sBAAsB;SACvB,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AA9HD,kDA8HC"}

package/dist/chains/combine_docs_chain.d.ts

@@ -0,0 +1,35 @@
+import { BaseChain, ChainValues, LLMChain, SerializedLLMChain } from "./index";
+export interface StuffDocumentsChainInput {
+    /** LLM Wrapper to use after formatting documents */
+    llmChain: LLMChain;
+    inputKey: string;
+    outputKey: string;
+    /** Variable name in the LLM chain to put the documents in */
+    documentVariableName: string;
+}
+export type SerializedStuffDocumentsChain = {
+    _type: "stuff_documents_chain";
+    llm_chain?: SerializedLLMChain;
+    llm_chain_path?: string;
+};
+/**
+ * Chain that combines documents by stuffing into context.
+ * @augments BaseChain
+ * @augments StuffDocumentsChainInput
+ */
+export declare class StuffDocumentsChain extends BaseChain implements StuffDocumentsChainInput {
+    llmChain: LLMChain;
+    inputKey: string;
+    outputKey: string;
+    documentVariableName: string;
+    constructor(fields: {
+        llmChain: LLMChain;
+        inputKey?: string;
+        outputKey?: string;
+        documentVariableName?: string;
+    });
+    _call(values: ChainValues): Promise<ChainValues>;
+    _chainType(): "stuff_documents_chain";
+    static deserialize(data: SerializedStuffDocumentsChain): Promise<StuffDocumentsChain>;
+    serialize(): SerializedStuffDocumentsChain;
+}

package/dist/chains/combine_docs_chain.js

@@ -0,0 +1,75 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StuffDocumentsChain = void 0;
+const index_1 = require("./index");
+const util_1 = require("../util");
+/**
+ * Chain that combines documents by stuffing into context.
+ * @augments BaseChain
+ * @augments StuffDocumentsChainInput
+ */
+class StuffDocumentsChain extends index_1.BaseChain {
+    constructor(fields) {
+        var _a, _b, _c;
+        super();
+        Object.defineProperty(this, "llmChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "inputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "input_documents"
+        });
+        Object.defineProperty(this, "outputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "output_text"
+        });
+        Object.defineProperty(this, "documentVariableName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "context"
+        });
+        this.llmChain = fields.llmChain;
+        this.documentVariableName =
+            (_a = fields.documentVariableName) !== null && _a !== void 0 ? _a : this.documentVariableName;
+        this.inputKey = (_b = fields.inputKey) !== null && _b !== void 0 ? _b : this.inputKey;
+        this.outputKey = (_c = fields.outputKey) !== null && _c !== void 0 ? _c : this.outputKey;
+    }
+    async _call(values) {
+        if (!(this.inputKey in values)) {
+            throw new Error(`Document key ${this.inputKey} not found.`);
+        }
+        const { [this.inputKey]: docs, ...rest } = values;
+        const texts = docs.map(({ pageContent }) => pageContent);
+        const text = texts.join("\n\n");
+        const result = await this.llmChain.call({
+            ...rest,
+            [this.documentVariableName]: text,
+        });
+        return result;
+    }
+    _chainType() {
+        return "stuff_documents_chain";
+    }
+    static async deserialize(data) {
+        const SerializedLLMChain = (0, util_1.resolveConfigFromFile)("llm_chain", data);
+        return new StuffDocumentsChain({
+            llmChain: await index_1.LLMChain.deserialize(SerializedLLMChain),
+        });
+    }
+    serialize() {
+        return {
+            _type: this._chainType(),
+            llm_chain: this.llmChain.serialize(),
+        };
+    }
+}
+exports.StuffDocumentsChain = StuffDocumentsChain;
+//# sourceMappingURL=combine_docs_chain.js.map

package/dist/chains/combine_docs_chain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"combine_docs_chain.js","sourceRoot":"","sources":["../../chains/combine_docs_chain.ts"],"names":[],"mappings":";;;AAAA,mCAA+E;AAI/E,kCAAgD;AAiBhD;;;;GAIG;AACH,MAAa,mBACX,SAAQ,iBAAS;IAWjB,YAAY,MAKX;;QACC,KAAK,EAAE,CAAC;QAdV;;;;;WAAmB;QAEnB;;;;mBAAW,iBAAiB;WAAC;QAE7B;;;;mBAAY,aAAa;WAAC;QAE1B;;;;mBAAuB,SAAS;WAAC;QAS/B,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,oBAAoB;YACvB,MAAA,MAAM,CAAC,oBAAoB,mCAAI,IAAI,CAAC,oBAAoB,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,MAAA,MAAM,CAAC,QAAQ,mCAAI,IAAI,CAAC,QAAQ,CAAC;QACjD,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,CAAC,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CAAC,gBAAgB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SAC7D;QACD,MAAM,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QAClD,MAAM,KAAK,GAAI,IAAmB,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,IAAI,CAAC,oBAAoB,CAAC,EAAE,IAAI;SAClC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,uBAAgC,CAAC;IAC1C,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAmC;QAC1D,MAAM,kBAAkB,GAAG,IAAA,4BAAqB,EAG9C,WAAW,EAAE,IAAI,CAAC,CAAC;QAErB,OAAO,IAAI,mBAAmB,CAAC;YAC7B,QAAQ,EAAE,MAAM,gBAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC;SACzD,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE;SACrC,CAAC;IACJ,CAAC;CACF;AA7DD,kDA6DC"}

package/dist/chains/conversation.d.ts

@@ -0,0 +1,12 @@
+import { LLMChain } from "./llm_chain";
+import { BaseLLM } from "../llms";
+import { BasePromptTemplate } from "../prompts";
+import { BaseMemory } from "../memory";
+export declare class ConversationChain extends LLMChain {
+    constructor(fields: {
+        llm: BaseLLM;
+        prompt?: BasePromptTemplate;
+        outputKey?: string;
+        memory?: BaseMemory;
+    });
+}

package/dist/chains/conversation.js

@@ -0,0 +1,29 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ConversationChain = void 0;
+const llm_chain_1 = require("./llm_chain");
+const prompts_1 = require("../prompts");
+const memory_1 = require("../memory");
+const defaultTemplate = `The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.
+
+Current conversation:
+{history}
+Human: {input}
+AI:`;
+const defaultPrompt = new prompts_1.PromptTemplate({
+    template: defaultTemplate,
+    inputVariables: ["history", "input"],
+});
+class ConversationChain extends llm_chain_1.LLMChain {
+    constructor(fields) {
+        var _a, _b, _c;
+        super({
+            prompt: (_a = fields.prompt) !== null && _a !== void 0 ? _a : defaultPrompt,
+            llm: fields.llm,
+            outputKey: (_b = fields.outputKey) !== null && _b !== void 0 ? _b : "response",
+        });
+        this.memory = (_c = fields.memory) !== null && _c !== void 0 ? _c : new memory_1.BufferMemory();
+    }
+}
+exports.ConversationChain = ConversationChain;
+//# sourceMappingURL=conversation.js.map

package/dist/chains/conversation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversation.js","sourceRoot":"","sources":["../../chains/conversation.ts"],"names":[],"mappings":";;;AAAA,2CAAuC;AAEvC,wCAAgE;AAEhE,sCAAqD;AAErD,MAAM,eAAe,GAAG;;;;;IAKpB,CAAC;AAEL,MAAM,aAAa,GAAG,IAAI,wBAAc,CAAC;IACvC,QAAQ,EAAE,eAAe;IACzB,cAAc,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC;CACrC,CAAC,CAAC;AAEH,MAAa,iBAAkB,SAAQ,oBAAQ;IAC7C,YAAY,MAKX;;QACC,KAAK,CAAC;YACJ,MAAM,EAAE,MAAA,MAAM,CAAC,MAAM,mCAAI,aAAa;YACtC,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,SAAS,EAAE,MAAA,MAAM,CAAC,SAAS,mCAAI,UAAU;SAC1C,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAA,MAAM,CAAC,MAAM,mCAAI,IAAI,qBAAY,EAAE,CAAC;IACpD,CAAC;CACF;AAdD,8CAcC"}

package/dist/chains/index.d.ts

@@ -0,0 +1,7 @@
+export { BaseChain, ChainValues } from "./base";
+export { SerializedLLMChain, LLMChain, ConversationChain } from "./llm_chain";
+export { SerializedStuffDocumentsChain, StuffDocumentsChain, } from "./combine_docs_chain";
+export { ChatVectorDBQAChain, SerializedChatVectorDBQAChain, } from "./chat_vector_db_chain";
+export { VectorDBQAChain, SerializedVectorDBQAChain } from "./vector_db_qa";
+export { loadChain } from "./load";
+export { loadQAChain } from "./question_answering/load";

package/dist/chains/index.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loadQAChain = exports.loadChain = exports.VectorDBQAChain = exports.ChatVectorDBQAChain = exports.StuffDocumentsChain = exports.ConversationChain = exports.LLMChain = exports.BaseChain = void 0;
+var base_1 = require("./base");
+Object.defineProperty(exports, "BaseChain", { enumerable: true, get: function () { return base_1.BaseChain; } });
+var llm_chain_1 = require("./llm_chain");
+Object.defineProperty(exports, "LLMChain", { enumerable: true, get: function () { return llm_chain_1.LLMChain; } });
+Object.defineProperty(exports, "ConversationChain", { enumerable: true, get: function () { return llm_chain_1.ConversationChain; } });
+var combine_docs_chain_1 = require("./combine_docs_chain");
+Object.defineProperty(exports, "StuffDocumentsChain", { enumerable: true, get: function () { return combine_docs_chain_1.StuffDocumentsChain; } });
+var chat_vector_db_chain_1 = require("./chat_vector_db_chain");
+Object.defineProperty(exports, "ChatVectorDBQAChain", { enumerable: true, get: function () { return chat_vector_db_chain_1.ChatVectorDBQAChain; } });
+var vector_db_qa_1 = require("./vector_db_qa");
+Object.defineProperty(exports, "VectorDBQAChain", { enumerable: true, get: function () { return vector_db_qa_1.VectorDBQAChain; } });
+var load_1 = require("./load");
+Object.defineProperty(exports, "loadChain", { enumerable: true, get: function () { return load_1.loadChain; } });
+var load_2 = require("./question_answering/load");
+Object.defineProperty(exports, "loadQAChain", { enumerable: true, get: function () { return load_2.loadQAChain; } });
+//# sourceMappingURL=index.js.map

package/dist/chains/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../chains/index.ts"],"names":[],"mappings":";;;AAAA,+BAAgD;AAAvC,iGAAA,SAAS,OAAA;AAClB,yCAA8E;AAAjD,qGAAA,QAAQ,OAAA;AAAE,8GAAA,iBAAiB,OAAA;AACxD,2DAG8B;AAD5B,yHAAA,mBAAmB,OAAA;AAErB,+DAGgC;AAF9B,2HAAA,mBAAmB,OAAA;AAGrB,+CAA4E;AAAnE,+GAAA,eAAe,OAAA;AACxB,+BAAmC;AAA1B,iGAAA,SAAS,OAAA;AAClB,kDAAwD;AAA/C,mGAAA,WAAW,OAAA"}

package/dist/{cjs/chains → chains}/llm_chain.d.ts

@@ -1,9 +1,13 @@
 import { BaseChain, ChainValues } from "./index";
 import { BaseLLM, SerializedLLM } from "../llms";
-import { BasePromptTemplate, SerializedBasePromptTemplate } from "../prompt";
+import { BaseMemory } from "../memory";
+import { BasePromptTemplate, SerializedBasePromptTemplate } from "../prompts";
 export interface LLMChainInput {
+    /** Prompt object to use */
     prompt: BasePromptTemplate;
+    /** LLM Wrapper to use */
     llm: BaseLLM;
+    /** @ignore */
     outputKey: string;
 }
 export type SerializedLLMChain = {
@@ -13,6 +17,18 @@ export type SerializedLLMChain = {
     prompt?: SerializedBasePromptTemplate;
     prompt_path?: string;
 };
+/**
+ * Chain to run queries against LLMs.
+ * @augments BaseChain
+ * @augments LLMChainInput
+ *
+ * @example
+ * ```ts
+ * import { LLMChain, OpenAI, PromptTemplate } from "langchain";
+ * const prompt = PromptTemplate.fromTemplate("Tell me a {adjective} joke");
+ * const llm = LLMChain({ llm: new OpenAI(), prompt });
+ * ```
+ */
 export declare class LLMChain extends BaseChain implements LLMChainInput {
     prompt: BasePromptTemplate;
     llm: BaseLLM;
@@ -23,8 +39,27 @@ export declare class LLMChain extends BaseChain implements LLMChainInput {
         outputKey?: string;
     });
     _call(values: ChainValues): Promise<ChainValues>;
+    /**
+     * Format prompt with values and pass to LLM
+     *
+     * @param values - keys to pass to prompt template
+     * @returns Completion from LLM.
+     *
+     * @example
+     * ```ts
+     * llm.predict({ adjective: "funny" })
+     * ```
+     */
     predict(values: ChainValues): Promise<string>;
     _chainType(): "llm_chain";
     static deserialize(data: SerializedLLMChain): Promise<LLMChain>;
     serialize(): SerializedLLMChain;
 }
+export declare class ConversationChain extends LLMChain {
+    constructor(fields: {
+        llm: BaseLLM;
+        prompt?: BasePromptTemplate;
+        outputKey?: string;
+        memory?: BaseMemory;
+    });
+}

package/dist/chains/llm_chain.js

@@ -0,0 +1,115 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ConversationChain = exports.LLMChain = void 0;
+const index_1 = require("./index");
+const llms_1 = require("../llms");
+const memory_1 = require("../memory");
+const prompts_1 = require("../prompts");
+const util_1 = require("../util");
+/**
+ * Chain to run queries against LLMs.
+ * @augments BaseChain
+ * @augments LLMChainInput
+ *
+ * @example
+ * ```ts
+ * import { LLMChain, OpenAI, PromptTemplate } from "langchain";
+ * const prompt = PromptTemplate.fromTemplate("Tell me a {adjective} joke");
+ * const llm = LLMChain({ llm: new OpenAI(), prompt });
+ * ```
+ */
+class LLMChain extends index_1.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;
+    }
+    async _call(values) {
+        let stop;
+        if ("stop" in values && Array.isArray(values.stop)) {
+            stop = values.stop;
+        }
+        const formattedString = this.prompt.format(values);
+        const llmResult = await this.llm.call(formattedString, stop);
+        const result = { [this.outputKey]: llmResult };
+        return result;
+    }
+    /**
+     * Format prompt with values and pass to LLM
+     *
+     * @param values - keys to pass to prompt template
+     * @returns Completion from LLM.
+     *
+     * @example
+     * ```ts
+     * llm.predict({ adjective: "funny" })
+     * ```
+     */
+    async predict(values) {
+        const output = await this.call(values);
+        return output[this.outputKey];
+    }
+    _chainType() {
+        return "llm_chain";
+    }
+    static async deserialize(data) {
+        const serializedLLM = (0, util_1.resolveConfigFromFile)("llm", data);
+        const serializedPrompt = (0, util_1.resolveConfigFromFile)("prompt", data);
+        return new LLMChain({
+            llm: await llms_1.BaseLLM.deserialize(serializedLLM),
+            prompt: await prompts_1.BasePromptTemplate.deserialize(serializedPrompt),
+        });
+    }
+    serialize() {
+        return {
+            _type: this._chainType(),
+            llm: this.llm.serialize(),
+            prompt: this.prompt.serialize(),
+        };
+    }
+}
+exports.LLMChain = LLMChain;
+// eslint-disable-next-line max-len
+const defaultTemplate = `The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.
+
+Current conversation:
+{history}
+Human: {input}
+AI:`;
+const defaultPrompt = new prompts_1.PromptTemplate({
+    template: defaultTemplate,
+    inputVariables: ["history", "input"],
+});
+class ConversationChain extends LLMChain {
+    constructor(fields) {
+        var _a, _b, _c;
+        super({
+            prompt: (_a = fields.prompt) !== null && _a !== void 0 ? _a : defaultPrompt,
+            llm: fields.llm,
+            outputKey: (_b = fields.outputKey) !== null && _b !== void 0 ? _b : "response",
+        });
+        this.memory = (_c = fields.memory) !== null && _c !== void 0 ? _c : new memory_1.BufferMemory();
+    }
+}
+exports.ConversationChain = ConversationChain;
+//# sourceMappingURL=llm_chain.js.map

package/dist/chains/llm_chain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm_chain.js","sourceRoot":"","sources":["../../chains/llm_chain.ts"],"names":[],"mappings":";;;AAAA,mCAAiD;AAEjD,kCAAiD;AAEjD,sCAAqD;AACrD,wCAIoB;AAEpB,kCAAgD;AAoBhD;;;;;;;;;;;GAWG;AACH,MAAa,QAAS,SAAQ,iBAAS;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;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,IAAI,CAAC;QACT,IAAI,MAAM,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAClD,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;SACpB;QACD,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC;QAC/C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,OAAO,CAAC,MAAmB;QAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvC,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC;IAED,UAAU;QACR,OAAO,WAAoB,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAwB;QAC/C,MAAM,aAAa,GAAG,IAAA,4BAAqB,EACzC,KAAK,EACL,IAAI,CACL,CAAC;QACF,MAAM,gBAAgB,GAAG,IAAA,4BAAqB,EAG5C,QAAQ,EAAE,IAAI,CAAC,CAAC;QAElB,OAAO,IAAI,QAAQ,CAAC;YAClB,GAAG,EAAE,MAAM,cAAO,CAAC,WAAW,CAAC,aAAa,CAAC;YAC7C,MAAM,EAAE,MAAM,4BAAkB,CAAC,WAAW,CAAC,gBAAgB,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;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;AAxED,4BAwEC;AAED,mCAAmC;AACnC,MAAM,eAAe,GAAG;;;;;IAKpB,CAAC;AAEL,MAAM,aAAa,GAAG,IAAI,wBAAc,CAAC;IACvC,QAAQ,EAAE,eAAe;IACzB,cAAc,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC;CACrC,CAAC,CAAC;AAEH,MAAa,iBAAkB,SAAQ,QAAQ;IAC7C,YAAY,MAKX;;QACC,KAAK,CAAC;YACJ,MAAM,EAAE,MAAA,MAAM,CAAC,MAAM,mCAAI,aAAa;YACtC,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,SAAS,EAAE,MAAA,MAAM,CAAC,SAAS,mCAAI,UAAU;SAC1C,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAA,MAAM,CAAC,MAAM,mCAAI,IAAI,qBAAY,EAAE,CAAC;IACpD,CAAC;CACF;AAdD,8CAcC"}

package/dist/chains/load.d.ts

@@ -0,0 +1,21 @@
+import { BaseChain } from ".";
+export type LoadValues = Record<string, any>;
+/**
+ * Load a chain from {@link https://github.com/hwchase17/langchain-hub | LangchainHub} or local filesystem.
+ *
+ * @example
+ * Loading from LangchainHub:
+ * ```ts
+ * import { loadChain } from "langchain/chains";
+ * const chain = await loadChain("lc://chains/hello-world/chain.json");
+ * const res = await chain.call({ topic: "my favorite color" });
+ * ```
+ *
+ * @example
+ * Loading from local filesystem:
+ * ```ts
+ * import { loadChain } from "langchain/chains";
+ * const chain = await loadChain("/path/to/chain.json");
+ * ```
+ */
+export declare const loadChain: (uri: string, values?: LoadValues) => Promise<BaseChain>;

package/dist/chains/load.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loadChain = void 0;
+const _1 = require(".");
+const hub_1 = require("../util/hub");
+const util_1 = require("../util");
+const loadChainFromFile = async (file, values = {}) => {
+    const serialized = (0, util_1.parseFileConfig)(file);
+    return _1.BaseChain.deserialize(serialized, values);
+};
+/**
+ * Load a chain from {@link https://github.com/hwchase17/langchain-hub | LangchainHub} or local filesystem.
+ *
+ * @example
+ * Loading from LangchainHub:
+ * ```ts
+ * import { loadChain } from "langchain/chains";
+ * const chain = await loadChain("lc://chains/hello-world/chain.json");
+ * const res = await chain.call({ topic: "my favorite color" });
+ * ```
+ *
+ * @example
+ * Loading from local filesystem:
+ * ```ts
+ * import { loadChain } from "langchain/chains";
+ * const chain = await loadChain("/path/to/chain.json");
+ * ```
+ */
+const loadChain = async (uri, values = {}) => {
+    const hubResult = await (0, hub_1.loadFromHub)(uri, (uri) => loadChainFromFile(uri, values), "chains", new Set(["json", "yaml"]), values);
+    if (hubResult) {
+        return hubResult;
+    }
+    return loadChainFromFile(uri, values);
+};
+exports.loadChain = loadChain;
+//# sourceMappingURL=load.js.map

package/dist/chains/load.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"load.js","sourceRoot":"","sources":["../../chains/load.ts"],"names":[],"mappings":";;;AAAA,wBAA8B;AAC9B,qCAA0C;AAC1C,kCAA0C;AAK1C,MAAM,iBAAiB,GAAG,KAAK,EAAE,IAAY,EAAE,SAAqB,EAAE,EAAE,EAAE;IACxE,MAAM,UAAU,GAAG,IAAA,sBAAe,EAAC,IAAI,CAAC,CAAC;IACzC,OAAO,YAAS,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;AACnD,CAAC,CAAC;AAEF;;;;;;;;;;;;;;;;;GAiBG;AACI,MAAM,SAAS,GAAG,KAAK,EAC5B,GAAW,EACX,SAAqB,EAAE,EACH,EAAE;IACtB,MAAM,SAAS,GAAG,MAAM,IAAA,iBAAW,EACjC,GAAG,EACH,CAAC,GAAG,EAAE,EAAE,CAAC,iBAAiB,CAAC,GAAG,EAAE,MAAM,CAAC,EACvC,QAAQ,EACR,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,EACzB,MAAM,CACP,CAAC;IACF,IAAI,SAAS,EAAE;QACb,OAAO,SAAS,CAAC;KAClB;IAED,OAAO,iBAAiB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;AACxC,CAAC,CAAC;AAhBW,QAAA,SAAS,aAgBpB"}

package/dist/chains/question_answering/load.d.ts

@@ -0,0 +1,3 @@
+import { BaseLLM } from "../../llms";
+import { StuffDocumentsChain } from "../combine_docs_chain";
+export declare const loadQAChain: (llm: BaseLLM, prompt?: import("../..").PromptTemplate) => StuffDocumentsChain;

package/dist/chains/question_answering/load.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loadQAChain = void 0;
+const llm_chain_1 = require("../llm_chain");
+const combine_docs_chain_1 = require("../combine_docs_chain");
+const stuff_prompts_1 = require("./stuff_prompts");
+const loadQAChain = (llm, prompt = stuff_prompts_1.DEFAULT_QA_PROMPT) => {
+    const llmChain = new llm_chain_1.LLMChain({ prompt, llm });
+    const chain = new combine_docs_chain_1.StuffDocumentsChain({ llmChain });
+    return chain;
+};
+exports.loadQAChain = loadQAChain;
+//# sourceMappingURL=load.js.map

package/dist/chains/question_answering/load.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../chains/question_answering/load.ts"],"names":[],"mappings":";;;AACA,4CAAwC;AACxC,8DAA4D;AAC5D,mDAAoD;AAE7C,MAAM,WAAW,GAAG,CAAC,GAAY,EAAE,MAAM,GAAG,iCAAiB,EAAE,EAAE;IACtE,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/C,MAAM,KAAK,GAAG,IAAI,wCAAmB,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IACpD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAJW,QAAA,WAAW,eAItB"}

package/dist/chains/question_answering/stuff_prompts.d.ts

@@ -0,0 +1,2 @@
+import { PromptTemplate } from "../../prompts";
+export declare const DEFAULT_QA_PROMPT: PromptTemplate;

package/dist/chains/question_answering/stuff_prompts.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_QA_PROMPT = void 0;
+/* eslint-disable */
+const prompts_1 = require("../../prompts");
+exports.DEFAULT_QA_PROMPT = new prompts_1.PromptTemplate({
+    template: "Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:",
+    inputVariables: ["context", "question"],
+});
+//# sourceMappingURL=stuff_prompts.js.map

package/dist/chains/question_answering/stuff_prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stuff_prompts.js","sourceRoot":"","sources":["../../../chains/question_answering/stuff_prompts.ts"],"names":[],"mappings":";;;AAAA,oBAAoB;AACpB,2CAA+C;AAElC,QAAA,iBAAiB,GAAG,IAAI,wBAAc,CAAC;IAClD,QAAQ,EACN,0NAA0N;IAC5N,cAAc,EAAE,CAAC,SAAS,EAAE,UAAU,CAAC;CACxC,CAAC,CAAC"}