langchain 0.0.4 → 0.0.6

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

@@ -0,0 +1,18 @@
+"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/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;QACX,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;KACrC,CAAC;IACF,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/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 prompts_1 = require("../../prompts");
+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 = prompts_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/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,2CAA+C;AAC/C,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,wBAAc,CAAC,YAAY,CACxC,6CAA6C,CAC9C,CAAC;IACF,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/chains/tests/combine_docs_chain.test.js

@@ -0,0 +1,39 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const openai_1 = require("../../llms/openai");
+const prompts_1 = require("../../prompts");
+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 prompts_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/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,2CAA+C;AAC/C,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,wBAAc,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;QACpC,QAAQ;QACR,oBAAoB,EAAE,KAAK;KAC5B,CAAC,CAAC;IACH,MAAM,IAAI,GAAG;QACX,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;KACrC,CAAC;IACF,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,EAC3B,iDAAiD,CAClD,CAAC;IACF,MAAM,IAAI,GAAG;QACX,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;KACrC,CAAC;IACF,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/chains/tests/llm_chain.test.js

@@ -0,0 +1,29 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const openai_1 = require("../../llms/openai");
+const prompts_1 = require("../../prompts");
+const llm_chain_1 = require("../llm_chain");
+const load_1 = require("../load");
+(0, globals_1.test)("Test OpenAI", async () => {
+    const model = new openai_1.OpenAI({});
+    const prompt = new prompts_1.PromptTemplate({
+        template: "Print {foo}",
+        inputVariables: ["foo"],
+    });
+    const chain = new llm_chain_1.LLMChain({ prompt, llm: model });
+    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)("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/chains/tests/llm_chain.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm_chain.test.js","sourceRoot":"","sources":["../../../chains/tests/llm_chain.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,2CAA+C;AAC/C,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,wBAAc,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/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 prompts_1 = require("../../prompts");
+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 prompts_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/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,2CAA+C;AAC/C,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,wBAAc,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/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/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/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/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/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/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/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/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/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/embeddings/index.d.ts

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

package/dist/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/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/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 {};

package/dist/embeddings/openai.js

@@ -0,0 +1,86 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIEmbeddings = void 0;
+const exponential_backoff_1 = require("exponential-backoff");
+const util_1 = require("../util");
+const base_1 = require("./base");
+let Configuration = null;
+let OpenAIApi = null;
+try {
+    // eslint-disable-next-line global-require,import/no-extraneous-dependencies
+    ({ Configuration, OpenAIApi } = require("openai"));
+}
+catch (_a) {
+    // ignore error
+}
+class OpenAIEmbeddings extends base_1.Embeddings {
+    constructor(fields) {
+        var _a, _b, _c;
+        super();
+        Object.defineProperty(this, "modelName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "text-embedding-ada-002"
+        });
+        Object.defineProperty(this, "batchSize", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 20
+        });
+        Object.defineProperty(this, "maxRetries", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 6
+        });
+        Object.defineProperty(this, "client", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        if (Configuration === null || OpenAIApi === null) {
+            throw new Error("Please install openai as a dependency with, e.g. `npm install -S openai`");
+        }
+        this.modelName = (_a = fields === null || fields === void 0 ? void 0 : fields.modelName) !== null && _a !== void 0 ? _a : this.modelName;
+        this.batchSize = (_b = fields === null || fields === void 0 ? void 0 : fields.batchSize) !== null && _b !== void 0 ? _b : this.batchSize;
+        const clientConfig = new Configuration({
+            apiKey: (_c = fields === null || fields === void 0 ? void 0 : fields.openAIApiKey) !== null && _c !== void 0 ? _c : process.env.OPENAI_API_KEY,
+        });
+        this.client = new OpenAIApi(clientConfig);
+    }
+    async embedDocuments(texts) {
+        const subPrompts = (0, util_1.chunkArray)(texts, this.batchSize);
+        const embeddings = [];
+        for (let i = 0; i < subPrompts.length; i += 1) {
+            const input = subPrompts[i];
+            const { data } = await this.embeddingWithRetry({
+                model: this.modelName,
+                input,
+            });
+            for (let j = 0; j < input.length; j += 1) {
+                embeddings.push(data.data[j].embedding);
+            }
+        }
+        return embeddings;
+    }
+    async embedQuery(text) {
+        const { data } = await this.embeddingWithRetry({
+            model: this.modelName,
+            input: text,
+        });
+        return data.data[0].embedding;
+    }
+    embeddingWithRetry(request) {
+        const makeCompletionRequest = () => this.client.createEmbedding(request);
+        return (0, exponential_backoff_1.backOff)(makeCompletionRequest, {
+            startingDelay: 4,
+            maxDelay: 10,
+            numOfAttempts: this.maxRetries,
+        });
+    }
+}
+exports.OpenAIEmbeddings = OpenAIEmbeddings;
+//# sourceMappingURL=openai.js.map

package/dist/embeddings/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../embeddings/openai.ts"],"names":[],"mappings":";;;AAKA,6DAA8C;AAC9C,kCAAqC;AACrC,iCAAoC;AAEpC,IAAI,aAAa,GAAiC,IAAI,CAAC;AACvD,IAAI,SAAS,GAA6B,IAAI,CAAC;AAE/C,IAAI;IACF,4EAA4E;IAC5E,CAAC,EAAE,aAAa,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;CACpD;AAAC,WAAM;IACN,eAAe;CAChB;AAMD,MAAa,gBAAiB,SAAQ,iBAAU;IAS9C,YACE,MAKC;;QAED,KAAK,EAAE,CAAC;QAhBV;;;;mBAAY,wBAAwB;WAAC;QAErC;;;;mBAAY,EAAE;WAAC;QAEf;;;;mBAAa,CAAC;WAAC;QAEf;;;;;WAA2B;QAWzB,IAAI,aAAa,KAAK,IAAI,IAAI,SAAS,KAAK,IAAI,EAAE;YAChD,MAAM,IAAI,KAAK,CACb,0EAA0E,CAC3E,CAAC;SACH;QAED,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;QACrD,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;YACrC,MAAM,EAAE,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,YAAY,mCAAI,OAAO,CAAC,GAAG,CAAC,cAAc;SAC3D,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAe;QAClC,MAAM,UAAU,GAAG,IAAA,iBAAU,EAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAErD,MAAM,UAAU,GAAG,EAAE,CAAC;QAEtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAC7C,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAC5B,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC;gBAC7C,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,KAAK;aACN,CAAC,CAAC;YACH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;gBACxC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;aACzC;SACF;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC;YAC7C,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAChC,CAAC;IAEO,kBAAkB,CAAC,OAA+B;QACxD,MAAM,qBAAqB,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QACzE,OAAO,IAAA,6BAAO,EAAC,qBAAqB,EAAE;YACpC,aAAa,EAAE,CAAC;YAChB,QAAQ,EAAE,EAAE;YACZ,aAAa,EAAE,IAAI,CAAC,UAAU;SAC/B,CAAC,CAAC;IACL,CAAC;CACF;AAnED,4CAmEC"}

package/dist/embeddings/tests/openai.test.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const openai_1 = require("../openai");
+(0, globals_1.test)("Test OpenAIEmbeddings.embedQuery", async () => {
+    const embeddings = new openai_1.OpenAIEmbeddings();
+    const res = await embeddings.embedQuery("Hello world");
+    (0, globals_1.expect)(typeof res[0]).toBe("number");
+});
+(0, globals_1.test)("Test OpenAIEmbeddings.embedDocuments", async () => {
+    const embeddings = new openai_1.OpenAIEmbeddings();
+    const res = await embeddings.embedDocuments(["Hello world", "Bye bye"]);
+    (0, globals_1.expect)(res).toHaveLength(2);
+    (0, globals_1.expect)(typeof res[0][0]).toBe("number");
+    (0, globals_1.expect)(typeof res[1][0]).toBe("number");
+});
+//# sourceMappingURL=openai.test.js.map

package/dist/embeddings/tests/openai.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.test.js","sourceRoot":"","sources":["../../../embeddings/tests/openai.test.ts"],"names":[],"mappings":";;AAAA,2CAA6C;AAC7C,sCAA6C;AAE7C,IAAA,cAAI,EAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,UAAU,GAAG,IAAI,yBAAgB,EAAE,CAAC;IAC1C,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;IACvD,IAAA,gBAAM,EAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;IACtD,MAAM,UAAU,GAAG,IAAI,yBAAgB,EAAE,CAAC;IAC1C,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,cAAc,CAAC,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC,CAAC;IACxE,IAAA,gBAAM,EAAC,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC5B,IAAA,gBAAM,EAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxC,IAAA,gBAAM,EAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC"}

package/dist/{cjs/index.d.ts → index.d.ts}

@@ -1,3 +1,3 @@
-export { PromptTemplate, BasePromptTemplate, FewShotPromptTemplate, } from "./prompt";
+export { PromptTemplate, BasePromptTemplate, FewShotPromptTemplate, } from "./prompts";
 export { LLMChain } from "./chains";
 export { OpenAI } from "./llms";

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

@@ -1,10 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.OpenAI = exports.LLMChain = exports.FewShotPromptTemplate = exports.BasePromptTemplate = exports.PromptTemplate = void 0;
-var prompt_1 = require("./prompt");
-Object.defineProperty(exports, "PromptTemplate", { enumerable: true, get: function () { return prompt_1.PromptTemplate; } });
-Object.defineProperty(exports, "BasePromptTemplate", { enumerable: true, get: function () { return prompt_1.BasePromptTemplate; } });
-Object.defineProperty(exports, "FewShotPromptTemplate", { enumerable: true, get: function () { return prompt_1.FewShotPromptTemplate; } });
+var prompts_1 = require("./prompts");
+Object.defineProperty(exports, "PromptTemplate", { enumerable: true, get: function () { return prompts_1.PromptTemplate; } });
+Object.defineProperty(exports, "BasePromptTemplate", { enumerable: true, get: function () { return prompts_1.BasePromptTemplate; } });
+Object.defineProperty(exports, "FewShotPromptTemplate", { enumerable: true, get: function () { return prompts_1.FewShotPromptTemplate; } });
 var chains_1 = require("./chains");
 Object.defineProperty(exports, "LLMChain", { enumerable: true, get: function () { return chains_1.LLMChain; } });
 var llms_1 = require("./llms");

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":";;;AAAA,qCAImB;AAHjB,yGAAA,cAAc,OAAA;AACd,6GAAA,kBAAkB,OAAA;AAClB,gHAAA,qBAAqB,OAAA;AAGvB,mCAAoC;AAA3B,kGAAA,QAAQ,OAAA;AACjB,+BAAgC;AAAvB,8FAAA,MAAM,OAAA"}