langchain 0.0.4 → 0.0.5

package/dist/cjs/chains/load.js

@@ -1,29 +1,37 @@
 "use strict";
-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());
-    });
-};
 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 = (file) => __awaiter(void 0, void 0, void 0, function* () {
+const loadChainFromFile = async (file, values = {}) => {
     const serialized = (0, util_1.parseFileConfig)(file);
-    console.log({ serialized });
-    return _1.BaseChain.deserialize(serialized);
-});
-const loadChain = (uri) => __awaiter(void 0, void 0, void 0, function* () {
-    const hubResult = yield (0, hub_1.loadFromHub)(uri, loadChainFromFile, "chains", new Set(["json", "yaml"]));
+    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);
-});
+    return loadChainFromFile(uri, values);
+};
 exports.loadChain = loadChain;
 //# sourceMappingURL=load.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../chains/load.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,wBAA8B;AAC9B,qCAA0C;AAC1C,kCAA0C;AAE1C,MAAM,iBAAiB,GAAG,CAAO,IAAY,EAAE,EAAE;IAC/C,MAAM,UAAU,GAAG,IAAA,sBAAe,EAAC,IAAI,CAAC,CAAC;IACzC,OAAO,CAAC,GAAG,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;IAC5B,OAAO,YAAS,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;AAC3C,CAAC,CAAA,CAAC;AAEK,MAAM,SAAS,GAAG,CAAO,GAAW,EAAsB,EAAE;IACjE,MAAM,SAAS,GAAG,MAAM,IAAA,iBAAW,EACjC,GAAG,EACH,iBAAiB,EACjB,QAAQ,EACR,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAC1B,CAAC;IACF,IAAI,SAAS,EAAE;QACb,OAAO,SAAS,CAAC;KAClB;IAED,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;AAChC,CAAC,CAAA,CAAC;AAZW,QAAA,SAAS,aAYpB"}
+{"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/cjs/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/cjs/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/cjs/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;AAG7C,MAAM,WAAW,GAAG,CAAC,GAAY,EAAE,MAAM,GAAG,iCAAiB,EAAE,EAAE;IACpE,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/C,MAAM,KAAK,GAAG,IAAI,wCAAmB,CAAC,EAAC,QAAQ,EAAC,CAAC,CAAC;IAClD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAJS,QAAA,WAAW,eAIpB"}

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

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

package/dist/cjs/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 prompt_1 = require("../../prompt");
+exports.DEFAULT_QA_PROMPT = new prompt_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/cjs/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,yCAA8C;AAEjC,QAAA,iBAAiB,GAAG,IAAI,uBAAc,CAAC;IAChD,QAAQ,EAAE,0NAA0N;IACpO,cAAc,EAAE,CAAC,SAAS,EAAE,UAAU,CAAC;CACxC,CAAC,CAAC"}

package/dist/cjs/chains/question_answering/tests/load.test.d.ts

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

package/dist/cjs/chains/question_answering/tests/load.test.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const openai_1 = require("../../../llms/openai");
+const load_1 = require("../load");
+const document_1 = require("../../../document");
+(0, globals_1.test)("Test loadQAChain", async () => {
+    const model = new openai_1.OpenAI({});
+    const chain = (0, load_1.loadQAChain)(model);
+    const docs = [new document_1.Document({ pageContent: 'foo' }), new document_1.Document({ pageContent: 'bar' }), new document_1.Document({ pageContent: 'baz' }),];
+    const res = await chain.call({ input_documents: docs, question: "Whats up" });
+    console.log({ res });
+});
+//# sourceMappingURL=load.test.js.map

package/dist/cjs/chains/question_answering/tests/load.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"load.test.js","sourceRoot":"","sources":["../../../../../chains/question_answering/tests/load.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,iDAA8C;AAC9C,kCAAsC;AACtC,gDAA6C;AAE7C,IAAA,cAAI,EAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;IAClC,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,IAAA,kBAAW,EAAC,KAAK,CAAC,CAAC;IACjC,MAAM,IAAI,GAAG,CAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAI,CAAC;IAC1I,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;IAC9E,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/cjs/chains/tests/chat_vector_db_qa_chain.test.d.ts

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

package/dist/cjs/chains/tests/chat_vector_db_qa_chain.test.js

@@ -0,0 +1,35 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const openai_1 = require("../../llms/openai");
+const prompt_1 = require("../../prompt");
+const llm_chain_1 = require("../llm_chain");
+const combine_docs_chain_1 = require("../combine_docs_chain");
+const chat_vector_db_chain_1 = require("../chat_vector_db_chain");
+const hnswlib_1 = require("../../vectorstores/hnswlib");
+const embeddings_1 = require("../../embeddings");
+(0, globals_1.test)("Test ChatVectorDBQAChain", async () => {
+    const model = new openai_1.OpenAI({});
+    const prompt = prompt_1.PromptTemplate.fromTemplate("Print {question}, and ignore {chat_history}");
+    const vectorStore = await hnswlib_1.HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world", "bye", "hi"], [{ id: 2 }, { id: 1 }, { id: 3 }, { id: 4 }, { id: 5 }], new embeddings_1.OpenAIEmbeddings());
+    const llmChain = new llm_chain_1.LLMChain({ prompt, llm: model });
+    const combineDocsChain = new combine_docs_chain_1.StuffDocumentsChain({
+        llmChain,
+        documentVariableName: "foo",
+    });
+    const chain = new chat_vector_db_chain_1.ChatVectorDBQAChain({
+        combineDocumentsChain: combineDocsChain,
+        vectorstore: vectorStore,
+        questionGeneratorChain: llmChain,
+    });
+    const res = await chain.call({ question: "foo", chat_history: "bar" });
+    console.log({ res });
+});
+(0, globals_1.test)("Test ChatVectorDBQAChain from LLM", async () => {
+    const model = new openai_1.OpenAI({});
+    const vectorStore = await hnswlib_1.HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world", "bye", "hi"], [{ id: 2 }, { id: 1 }, { id: 3 }, { id: 4 }, { id: 5 }], new embeddings_1.OpenAIEmbeddings());
+    const chain = chat_vector_db_chain_1.ChatVectorDBQAChain.fromLLM(model, vectorStore);
+    const res = await chain.call({ question: "foo", chat_history: "bar" });
+    console.log({ res });
+});
+//# sourceMappingURL=chat_vector_db_qa_chain.test.js.map

package/dist/cjs/chains/tests/chat_vector_db_qa_chain.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat_vector_db_qa_chain.test.js","sourceRoot":"","sources":["../../../../chains/tests/chat_vector_db_qa_chain.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,yCAA8C;AAC9C,4CAAwC;AACxC,8DAA4D;AAC5D,kEAA8D;AAC9D,wDAAqD;AACrD,iDAAoD;AAEpD,IAAA,cAAI,EAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;IAC1C,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,uBAAc,CAAC,YAAY,CAAC,6CAA6C,CAAC,CAAC;IAC1F,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACvD,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,MAAM,gBAAgB,GAAG,IAAI,wCAAmB,CAAC;QAC/C,QAAQ;QACR,oBAAoB,EAAE,KAAK;KAC5B,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,0CAAmB,CAAC;QACpC,qBAAqB,EAAE,gBAAgB;QACvC,WAAW,EAAE,WAAW;QACxB,sBAAsB,EAAE,QAAQ;KACjC,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;IACvE,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;IACnD,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACvD,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,MAAM,KAAK,GAAG,0CAAmB,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IAC9D,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;IACvE,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/cjs/chains/tests/combine_docs_chain.test.d.ts

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

package/dist/cjs/chains/tests/combine_docs_chain.test.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const openai_1 = require("../../llms/openai");
+const prompt_1 = require("../../prompt");
+const llm_chain_1 = require("../llm_chain");
+const load_1 = require("../load");
+const combine_docs_chain_1 = require("../combine_docs_chain");
+const document_1 = require("../../document");
+(0, globals_1.test)("Test StuffDocumentsChain", async () => {
+    const model = new openai_1.OpenAI({});
+    const prompt = new prompt_1.PromptTemplate({
+        template: "Print {foo}",
+        inputVariables: ["foo"],
+    });
+    const llmChain = new llm_chain_1.LLMChain({ prompt, llm: model });
+    const chain = new combine_docs_chain_1.StuffDocumentsChain({ llmChain, documentVariableName: "foo" });
+    const docs = [new document_1.Document({ pageContent: 'foo' }), new document_1.Document({ pageContent: 'bar' }), new document_1.Document({ pageContent: 'baz' }),];
+    const res = await chain.call({ input_documents: docs });
+    console.log({ res });
+});
+(0, globals_1.test)("Load chain from hub", async () => {
+    const chain = await (0, load_1.loadChain)("lc://chains/question_answering/stuff/chain.json");
+    const docs = [new document_1.Document({ pageContent: 'foo' }), new document_1.Document({ pageContent: 'bar' }), new document_1.Document({ pageContent: 'baz' }),];
+    const res = await chain.call({ input_documents: docs, question: "what up" });
+    console.log({ res });
+});
+//# sourceMappingURL=combine_docs_chain.test.js.map

package/dist/cjs/chains/tests/combine_docs_chain.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"combine_docs_chain.test.js","sourceRoot":"","sources":["../../../../chains/tests/combine_docs_chain.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,yCAA8C;AAC9C,4CAAwC;AACxC,kCAAoC;AACpC,8DAA4D;AAC5D,6CAA0C;AAE1C,IAAA,cAAI,EAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;IAC1C,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,uBAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,MAAM,KAAK,GAAG,IAAI,wCAAmB,CAAC,EAAE,QAAQ,EAAE,oBAAoB,EAAE,KAAK,EAAC,CAAC,CAAC;IAChF,MAAM,IAAI,GAAG,CAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAI,CAAC;IAC1I,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAS,EAAC,iDAAiD,CAAC,CAAC;IACjF,MAAM,IAAI,GAAG,CAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAI,CAAC;IAC1I,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;IAC7E,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/cjs/chains/tests/llm_chain.test.js

@@ -1,32 +1,29 @@
 "use strict";
-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());
-    });
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 const globals_1 = require("@jest/globals");
 const openai_1 = require("../../llms/openai");
 const prompt_1 = require("../../prompt");
 const llm_chain_1 = require("../llm_chain");
 const load_1 = require("../load");
-(0, globals_1.test)("Test OpenAI", () => __awaiter(void 0, void 0, void 0, function* () {
+(0, globals_1.test)("Test OpenAI", async () => {
     const model = new openai_1.OpenAI({});
     const prompt = new prompt_1.PromptTemplate({
         template: "Print {foo}",
         inputVariables: ["foo"],
     });
     const chain = new llm_chain_1.LLMChain({ prompt, llm: model });
-    const res = yield chain.call({ foo: "my favorite color" });
+    const res = await chain.call({ foo: "my favorite color" });
+    console.log({ res });
+});
+(0, globals_1.test)("Load chain from hub", async () => {
+    const chain = await (0, load_1.loadChain)("lc://chains/hello-world/chain.json");
+    const res = await chain.call({ topic: "my favorite color" });
     console.log({ res });
-}));
-(0, globals_1.test)("Load chain from hub", () => __awaiter(void 0, void 0, void 0, function* () {
-    const chain = yield (0, load_1.loadChain)("lc://chains/hello-world/chain.json");
-    const res = yield chain.call({ topic: "my favorite color" });
+});
+(0, globals_1.test)("Test ConversationChain", async () => {
+    const model = new openai_1.OpenAI({});
+    const chain = new llm_chain_1.ConversationChain({ llm: model });
+    const res = await chain.call({ input: "my favorite color" });
     console.log({ res });
-}));
+});
 //# sourceMappingURL=llm_chain.test.js.map

package/dist/cjs/chains/tests/llm_chain.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"llm_chain.test.js","sourceRoot":"","sources":["../../../../chains/tests/llm_chain.test.ts"],"names":[],"mappings":";;;;;;;;;;;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,yCAA8C;AAC9C,4CAAwC;AACxC,kCAAoC;AAEpC,IAAA,cAAI,EAAC,aAAa,EAAE,GAAS,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,uBAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAA,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,qBAAqB,EAAE,GAAS,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAS,EAAC,oCAAoC,CAAC,CAAC;IACpE,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAA,CAAC,CAAC"}
+{"version":3,"file":"llm_chain.test.js","sourceRoot":"","sources":["../../../../chains/tests/llm_chain.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,yCAA8C;AAC9C,4CAA2D;AAC3D,kCAAoC;AAEpC,IAAA,cAAI,EAAC,aAAa,EAAE,KAAK,IAAI,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,uBAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAS,EAAC,oCAAoC,CAAC,CAAC;IACpE,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;IACxC,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,IAAI,6BAAiB,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACpD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/cjs/chains/tests/vector_db_qa_chain.test.d.ts

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

package/dist/cjs/chains/tests/vector_db_qa_chain.test.js

@@ -0,0 +1,46 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const openai_1 = require("../../llms/openai");
+const prompt_1 = require("../../prompt");
+const llm_chain_1 = require("../llm_chain");
+const load_1 = require("../load");
+const combine_docs_chain_1 = require("../combine_docs_chain");
+const vector_db_qa_1 = require("../vector_db_qa");
+const hnswlib_1 = require("../../vectorstores/hnswlib");
+const embeddings_1 = require("../../embeddings");
+(0, globals_1.test)("Test VectorDBQAChain", async () => {
+    const model = new openai_1.OpenAI({});
+    const prompt = new prompt_1.PromptTemplate({
+        template: "Print {foo}",
+        inputVariables: ["foo"],
+    });
+    const vectorStore = await hnswlib_1.HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world", "bye", "hi"], [{ id: 2 }, { id: 1 }, { id: 3 }, { id: 4 }, { id: 5 }], new embeddings_1.OpenAIEmbeddings());
+    const llmChain = new llm_chain_1.LLMChain({ prompt, llm: model });
+    const combineDocsChain = new combine_docs_chain_1.StuffDocumentsChain({
+        llmChain,
+        documentVariableName: "foo",
+    });
+    const chain = new vector_db_qa_1.VectorDBQAChain({
+        combineDocumentsChain: combineDocsChain,
+        vectorstore: vectorStore,
+    });
+    const res = await chain.call({ query: "What up" });
+    console.log({ res });
+});
+(0, globals_1.test)("Test VectorDBQAChain from LLM", async () => {
+    const model = new openai_1.OpenAI({});
+    const vectorStore = await hnswlib_1.HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world", "bye", "hi"], [{ id: 2 }, { id: 1 }, { id: 3 }, { id: 4 }, { id: 5 }], new embeddings_1.OpenAIEmbeddings());
+    const chain = vector_db_qa_1.VectorDBQAChain.fromLLM(model, vectorStore);
+    const res = await chain.call({ query: "What up" });
+    console.log({ res });
+});
+(0, globals_1.test)("Load chain from hub", async () => {
+    const vectorStore = await hnswlib_1.HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world", "bye", "hi"], [{ id: 2 }, { id: 1 }, { id: 3 }, { id: 4 }, { id: 5 }], new embeddings_1.OpenAIEmbeddings());
+    const chain = await (0, load_1.loadChain)("lc://chains/vector-db-qa/stuff/chain.json", {
+        vectorstore: vectorStore,
+    });
+    const res = await chain.call({ query: "what up" });
+    console.log({ res });
+});
+//# sourceMappingURL=vector_db_qa_chain.test.js.map

package/dist/cjs/chains/tests/vector_db_qa_chain.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector_db_qa_chain.test.js","sourceRoot":"","sources":["../../../../chains/tests/vector_db_qa_chain.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,yCAA8C;AAC9C,4CAAwC;AACxC,kCAAoC;AACpC,8DAA4D;AAC5D,kDAAkD;AAClD,wDAAqD;AACrD,iDAAoD;AAEpD,IAAA,cAAI,EAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;IACtC,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,uBAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACvD,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,MAAM,gBAAgB,GAAG,IAAI,wCAAmB,CAAC;QAC/C,QAAQ;QACR,oBAAoB,EAAE,KAAK;KAC5B,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,8BAAe,CAAC;QAChC,qBAAqB,EAAE,gBAAgB;QACvC,WAAW,EAAE,WAAW;KACzB,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;IAC/C,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACvD,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,MAAM,KAAK,GAAG,8BAAe,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IAC1D,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACvD,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAS,EAAC,2CAA2C,EAAE;QACzE,WAAW,EAAE,WAAW;KACzB,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/cjs/chains/vector_db_qa.d.ts

@@ -0,0 +1,36 @@
+import { BaseChain, ChainValues, SerializedStuffDocumentsChain, StuffDocumentsChain } from "./index";
+import { VectorStore } from "../vectorstores/base";
+import { BaseLLM } from "../llms";
+export type LoadValues = Record<string, any>;
+export interface VectorDBQAChainInput {
+    vectorstore: VectorStore;
+    k: number;
+    combineDocumentsChain: StuffDocumentsChain;
+    outputKey: string;
+    inputKey: string;
+}
+export type SerializedVectorDBQAChain = {
+    _type: "vector_db_qa";
+    k: number;
+    combine_documents_chain: SerializedStuffDocumentsChain;
+    combine_documents_chain_path?: string;
+};
+export declare class VectorDBQAChain extends BaseChain implements VectorDBQAChainInput {
+    k: number;
+    inputKey: string;
+    outputKey: string;
+    vectorstore: VectorStore;
+    combineDocumentsChain: StuffDocumentsChain;
+    constructor(fields: {
+        vectorstore: VectorStore;
+        combineDocumentsChain: StuffDocumentsChain;
+        inputKey?: string;
+        outputKey?: string;
+        k?: number;
+    });
+    _call(values: ChainValues): Promise<ChainValues>;
+    _chainType(): "vector_db_qa";
+    static deserialize(data: SerializedVectorDBQAChain, values: LoadValues): Promise<VectorDBQAChain>;
+    serialize(): SerializedVectorDBQAChain;
+    static fromLLM(llm: BaseLLM, vectorstore: VectorStore): VectorDBQAChain;
+}

package/dist/cjs/chains/vector_db_qa.js

@@ -0,0 +1,85 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VectorDBQAChain = void 0;
+const index_1 = require("./index");
+const util_1 = require("../util");
+class VectorDBQAChain 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: "query"
+        });
+        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
+        });
+        this.vectorstore = fields.vectorstore;
+        this.combineDocumentsChain = fields.combineDocumentsChain;
+        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.`);
+        }
+        const question = values[this.inputKey];
+        const docs = await this.vectorstore.similaritySearch(question, this.k);
+        const inputs = { question, input_documents: docs };
+        const result = await this.combineDocumentsChain.call(inputs);
+        return result;
+    }
+    _chainType() {
+        return "vector_db_qa";
+    }
+    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);
+        return new VectorDBQAChain({
+            combineDocumentsChain: await index_1.StuffDocumentsChain.deserialize(serializedCombineDocumentsChain),
+            k: data.k,
+            vectorstore,
+        });
+    }
+    serialize() {
+        return {
+            _type: this._chainType(),
+            combine_documents_chain: this.combineDocumentsChain.serialize(),
+            k: this.k,
+        };
+    }
+    static fromLLM(llm, vectorstore) {
+        const qaChain = (0, index_1.loadQAChain)(llm);
+        const instance = new this({ vectorstore, combineDocumentsChain: qaChain });
+        return instance;
+    }
+}
+exports.VectorDBQAChain = VectorDBQAChain;
+//# sourceMappingURL=vector_db_qa.js.map

package/dist/cjs/chains/vector_db_qa.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector_db_qa.js","sourceRoot":"","sources":["../../../chains/vector_db_qa.ts"],"names":[],"mappings":";;;AAAA,mCAMiB;AAKjB,kCAAgD;AAmBhD,MAAa,eAAgB,SAAQ,iBAAS;IAW5C,YAAY,MAMX;;QACC,KAAK,EAAE,CAAC;QAjBV;;;;mBAAI,CAAC;WAAC;QAEN;;;;mBAAW,OAAO;WAAC;QAEnB;;;;mBAAY,QAAQ;WAAC;QAErB;;;;;WAAyB;QAEzB;;;;;WAA2C;QAUzC,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,qBAAqB,CAAC;QAC1D,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,MAAM,QAAQ,GAAW,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,EAAE,QAAQ,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,cAAuB,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,IAA+B,EAC/B,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;QAEnC,OAAO,IAAI,eAAe,CAAC;YACzB,qBAAqB,EAAE,MAAM,2BAAmB,CAAC,WAAW,CAC1D,+BAA+B,CAChC;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,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,CAAC,CAAC;QACjC,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,EAAE,WAAW,EAAE,qBAAqB,EAAE,OAAO,EAAE,CAAC,CAAC;QAC3E,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AA9ED,0CA8EC"}

package/dist/cjs/document.d.ts

@@ -0,0 +1,13 @@
+interface DocumentParams {
+    pageContent: string;
+    metadata: Record<string, any>;
+}
+/**
+ * Interface for interacting with a document.
+ */
+export declare class Document implements DocumentParams {
+    pageContent: string;
+    metadata: Record<string, any>;
+    constructor(fields?: Partial<DocumentParams>);
+}
+export {};

package/dist/cjs/document.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Document = void 0;
+/**
+ * Interface for interacting with a document.
+ */
+class Document {
+    constructor(fields) {
+        var _a, _b;
+        Object.defineProperty(this, "pageContent", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        Object.defineProperty(this, "metadata", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.pageContent = (_a = fields === null || fields === void 0 ? void 0 : fields.pageContent) !== null && _a !== void 0 ? _a : this.pageContent;
+        this.metadata = (_b = fields === null || fields === void 0 ? void 0 : fields.metadata) !== null && _b !== void 0 ? _b : {};
+    }
+}
+exports.Document = Document;
+//# sourceMappingURL=document.js.map

package/dist/cjs/document.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"document.js","sourceRoot":"","sources":["../../document.ts"],"names":[],"mappings":";;;AAOA;;GAEG;AACH,MAAa,QAAQ;IAMnB,YAAY,MAAgC;;QAL5C;;;;;WAAoB;QAEpB,8DAA8D;QAC9D;;;;;WAA8B;QAG5B,IAAI,CAAC,WAAW,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,WAAW,mCAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,QAAQ,mCAAI,EAAE,CAAC;IACzC,CAAC;CACF;AAVD,4BAUC"}

package/dist/cjs/embeddings/base.d.ts

@@ -0,0 +1,4 @@
+export declare abstract class Embeddings {
+    abstract embedDocuments(documents: string[]): Promise<number[][]>;
+    abstract embedQuery(document: string): Promise<number[]>;
+}

package/dist/cjs/embeddings/base.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Embeddings = void 0;
+class Embeddings {
+}
+exports.Embeddings = Embeddings;
+//# sourceMappingURL=base.js.map

package/dist/cjs/embeddings/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../embeddings/base.ts"],"names":[],"mappings":";;;AAAA,MAAsB,UAAU;CAI/B;AAJD,gCAIC"}

package/dist/cjs/embeddings/index.d.ts

@@ -0,0 +1 @@
+export { OpenAIEmbeddings } from "./openai";

package/dist/cjs/embeddings/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIEmbeddings = void 0;
+var openai_1 = require("./openai");
+Object.defineProperty(exports, "OpenAIEmbeddings", { enumerable: true, get: function () { return openai_1.OpenAIEmbeddings; } });
+//# sourceMappingURL=index.js.map

package/dist/cjs/embeddings/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../embeddings/index.ts"],"names":[],"mappings":";;;AAAA,mCAA4C;AAAnC,0GAAA,gBAAgB,OAAA"}

package/dist/cjs/embeddings/openai.d.ts

@@ -0,0 +1,20 @@
+import { Embeddings } from "./base";
+interface ModelParams {
+    modelName: string;
+}
+export declare class OpenAIEmbeddings extends Embeddings implements ModelParams {
+    modelName: string;
+    batchSize: number;
+    maxRetries: number;
+    private client;
+    constructor(fields?: Partial<ModelParams> & {
+        verbose?: boolean;
+        batchSize?: number;
+        maxRetries?: number;
+        openAIApiKey?: string;
+    });
+    embedDocuments(texts: string[]): Promise<number[][]>;
+    embedQuery(text: string): Promise<number[]>;
+    private embeddingWithRetry;
+}
+export {};