langchain 0.0.4 → 0.0.5

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

@@ -0,0 +1,9 @@
+import { LLMChain } from "../llm_chain";
+import { StuffDocumentsChain } from "../combine_docs_chain";
+import { DEFAULT_QA_PROMPT } from "./stuff_prompts";
+export const loadQAChain = (llm, prompt = DEFAULT_QA_PROMPT) => {
+    const llmChain = new LLMChain({ prompt, llm });
+    const chain = new StuffDocumentsChain({ llmChain });
+    return chain;
+};
+//# sourceMappingURL=load.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../../chains/question_answering/load.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAGpD,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,GAAY,EAAE,MAAM,GAAG,iBAAiB,EAAE,EAAE;IACpE,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/C,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,EAAC,QAAQ,EAAC,CAAC,CAAC;IAClD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC"}

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

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

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

@@ -0,0 +1,7 @@
+/* eslint-disable */
+import { PromptTemplate } from "../../prompt";
+export const DEFAULT_QA_PROMPT = new 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/esm/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,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE9C,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,cAAc,CAAC;IAChD,QAAQ,EAAE,0NAA0N;IACpO,cAAc,EAAE,CAAC,SAAS,EAAE,UAAU,CAAC;CACxC,CAAC,CAAC"}

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

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

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

@@ -0,0 +1,12 @@
+import { test } from "@jest/globals";
+import { OpenAI } from "../../../llms/openai";
+import { loadQAChain } from "../load";
+import { Document } from "../../../document";
+test("Test loadQAChain", async () => {
+    const model = new OpenAI({});
+    const chain = loadQAChain(model);
+    const docs = [new Document({ pageContent: 'foo' }), new Document({ pageContent: 'bar' }), new Document({ pageContent: 'baz' }),];
+    const res = await chain.call({ input_documents: docs, question: "Whats up" });
+    console.log({ res });
+});
+//# sourceMappingURL=load.test.js.map

package/dist/esm/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,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACtC,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,IAAI,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;IAClC,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;IACjC,MAAM,IAAI,GAAG,CAAK,IAAI,QAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,QAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,QAAQ,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/esm/chains/tests/chat_vector_db_qa_chain.test.d.ts

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

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

@@ -0,0 +1,33 @@
+import { test } from "@jest/globals";
+import { OpenAI } from "../../llms/openai";
+import { PromptTemplate } from "../../prompt";
+import { LLMChain } from "../llm_chain";
+import { StuffDocumentsChain } from "../combine_docs_chain";
+import { ChatVectorDBQAChain } from "../chat_vector_db_chain";
+import { HNSWLib } from "../../vectorstores/hnswlib";
+import { OpenAIEmbeddings } from "../../embeddings";
+test("Test ChatVectorDBQAChain", async () => {
+    const model = new OpenAI({});
+    const prompt = PromptTemplate.fromTemplate("Print {question}, and ignore {chat_history}");
+    const vectorStore = await HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world", "bye", "hi"], [{ id: 2 }, { id: 1 }, { id: 3 }, { id: 4 }, { id: 5 }], new OpenAIEmbeddings());
+    const llmChain = new LLMChain({ prompt, llm: model });
+    const combineDocsChain = new StuffDocumentsChain({
+        llmChain,
+        documentVariableName: "foo",
+    });
+    const chain = new ChatVectorDBQAChain({
+        combineDocumentsChain: combineDocsChain,
+        vectorstore: vectorStore,
+        questionGeneratorChain: llmChain,
+    });
+    const res = await chain.call({ question: "foo", chat_history: "bar" });
+    console.log({ res });
+});
+test("Test ChatVectorDBQAChain from LLM", async () => {
+    const model = new OpenAI({});
+    const vectorStore = await HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world", "bye", "hi"], [{ id: 2 }, { id: 1 }, { id: 3 }, { id: 4 }, { id: 5 }], new OpenAIEmbeddings());
+    const chain = 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/esm/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,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAEpD,IAAI,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;IAC1C,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,cAAc,CAAC,YAAY,CAAC,6CAA6C,CAAC,CAAC;IAC1F,MAAM,WAAW,GAAG,MAAM,OAAO,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,gBAAgB,EAAE,CACvB,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,MAAM,gBAAgB,GAAG,IAAI,mBAAmB,CAAC;QAC/C,QAAQ;QACR,oBAAoB,EAAE,KAAK;KAC5B,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,mBAAmB,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,IAAI,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;IACnD,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,WAAW,GAAG,MAAM,OAAO,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,gBAAgB,EAAE,CACvB,CAAC;IACF,MAAM,KAAK,GAAG,mBAAmB,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/esm/chains/tests/combine_docs_chain.test.d.ts

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

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

@@ -0,0 +1,26 @@
+import { test } from "@jest/globals";
+import { OpenAI } from "../../llms/openai";
+import { PromptTemplate } from "../../prompt";
+import { LLMChain } from "../llm_chain";
+import { loadChain } from "../load";
+import { StuffDocumentsChain } from "../combine_docs_chain";
+import { Document } from "../../document";
+test("Test StuffDocumentsChain", async () => {
+    const model = new OpenAI({});
+    const prompt = new PromptTemplate({
+        template: "Print {foo}",
+        inputVariables: ["foo"],
+    });
+    const llmChain = new LLMChain({ prompt, llm: model });
+    const chain = new StuffDocumentsChain({ llmChain, documentVariableName: "foo" });
+    const docs = [new Document({ pageContent: 'foo' }), new Document({ pageContent: 'bar' }), new Document({ pageContent: 'baz' }),];
+    const res = await chain.call({ input_documents: docs });
+    console.log({ res });
+});
+test("Load chain from hub", async () => {
+    const chain = await loadChain("lc://chains/question_answering/stuff/chain.json");
+    const docs = [new Document({ pageContent: 'foo' }), new Document({ pageContent: 'bar' }), new 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/esm/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,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,IAAI,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;IAC1C,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,EAAE,QAAQ,EAAE,oBAAoB,EAAE,KAAK,EAAC,CAAC,CAAC;IAChF,MAAM,IAAI,GAAG,CAAK,IAAI,QAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,QAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,QAAQ,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,IAAI,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,iDAAiD,CAAC,CAAC;IACjF,MAAM,IAAI,GAAG,CAAK,IAAI,QAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,QAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,QAAQ,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/esm/chains/tests/llm_chain.test.js

@@ -1,30 +1,27 @@
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
 import { test } from "@jest/globals";
 import { OpenAI } from "../../llms/openai";
 import { PromptTemplate } from "../../prompt";
-import { LLMChain } from "../llm_chain";
+import { LLMChain, ConversationChain } from "../llm_chain";
 import { loadChain } from "../load";
-test("Test OpenAI", () => __awaiter(void 0, void 0, void 0, function* () {
+test("Test OpenAI", async () => {
     const model = new OpenAI({});
     const prompt = new PromptTemplate({
         template: "Print {foo}",
         inputVariables: ["foo"],
     });
     const chain = new 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 });
+});
+test("Load chain from hub", async () => {
+    const chain = await loadChain("lc://chains/hello-world/chain.json");
+    const res = await chain.call({ topic: "my favorite color" });
     console.log({ res });
-}));
-test("Load chain from hub", () => __awaiter(void 0, void 0, void 0, function* () {
-    const chain = yield loadChain("lc://chains/hello-world/chain.json");
-    const res = yield chain.call({ topic: "my favorite color" });
+});
+test("Test ConversationChain", async () => {
+    const model = new OpenAI({});
+    const chain = new ConversationChain({ llm: model });
+    const res = await chain.call({ input: "my favorite color" });
     console.log({ res });
-}));
+});
 //# sourceMappingURL=llm_chain.test.js.map

package/dist/esm/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,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAEpC,IAAI,CAAC,aAAa,EAAE,GAAS,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,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,IAAI,CAAC,qBAAqB,EAAE,GAAS,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,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,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAC3D,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAEpC,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,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,IAAI,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,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,IAAI,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;IACxC,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,IAAI,iBAAiB,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/esm/chains/tests/vector_db_qa_chain.test.d.ts

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

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

@@ -0,0 +1,44 @@
+import { test } from "@jest/globals";
+import { OpenAI } from "../../llms/openai";
+import { PromptTemplate } from "../../prompt";
+import { LLMChain } from "../llm_chain";
+import { loadChain } from "../load";
+import { StuffDocumentsChain } from "../combine_docs_chain";
+import { VectorDBQAChain } from "../vector_db_qa";
+import { HNSWLib } from "../../vectorstores/hnswlib";
+import { OpenAIEmbeddings } from "../../embeddings";
+test("Test VectorDBQAChain", async () => {
+    const model = new OpenAI({});
+    const prompt = new PromptTemplate({
+        template: "Print {foo}",
+        inputVariables: ["foo"],
+    });
+    const vectorStore = await HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world", "bye", "hi"], [{ id: 2 }, { id: 1 }, { id: 3 }, { id: 4 }, { id: 5 }], new OpenAIEmbeddings());
+    const llmChain = new LLMChain({ prompt, llm: model });
+    const combineDocsChain = new StuffDocumentsChain({
+        llmChain,
+        documentVariableName: "foo",
+    });
+    const chain = new VectorDBQAChain({
+        combineDocumentsChain: combineDocsChain,
+        vectorstore: vectorStore,
+    });
+    const res = await chain.call({ query: "What up" });
+    console.log({ res });
+});
+test("Test VectorDBQAChain from LLM", async () => {
+    const model = new OpenAI({});
+    const vectorStore = await HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world", "bye", "hi"], [{ id: 2 }, { id: 1 }, { id: 3 }, { id: 4 }, { id: 5 }], new OpenAIEmbeddings());
+    const chain = VectorDBQAChain.fromLLM(model, vectorStore);
+    const res = await chain.call({ query: "What up" });
+    console.log({ res });
+});
+test("Load chain from hub", async () => {
+    const vectorStore = await HNSWLib.fromTexts(["Hello world", "Bye bye", "hello nice world", "bye", "hi"], [{ id: 2 }, { id: 1 }, { id: 3 }, { id: 4 }, { id: 5 }], new OpenAIEmbeddings());
+    const chain = await 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/esm/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,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAEpD,IAAI,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;IACtC,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,MAAM,OAAO,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,gBAAgB,EAAE,CACvB,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,MAAM,gBAAgB,GAAG,IAAI,mBAAmB,CAAC;QAC/C,QAAQ;QACR,oBAAoB,EAAE,KAAK;KAC5B,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,eAAe,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,IAAI,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;IAC/C,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,WAAW,GAAG,MAAM,OAAO,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,gBAAgB,EAAE,CACvB,CAAC;IACF,MAAM,KAAK,GAAG,eAAe,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,IAAI,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,WAAW,GAAG,MAAM,OAAO,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,gBAAgB,EAAE,CACvB,CAAC;IACF,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,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/esm/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/esm/chains/vector_db_qa.js

@@ -0,0 +1,81 @@
+import { BaseChain, StuffDocumentsChain, loadQAChain, } from "./index";
+import { resolveConfigFromFile } from "../util";
+export class VectorDBQAChain extends 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 = resolveConfigFromFile("combine_documents_chain", data);
+        return new VectorDBQAChain({
+            combineDocumentsChain: await 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 = loadQAChain(llm);
+        const instance = new this({ vectorstore, combineDocumentsChain: qaChain });
+        return instance;
+    }
+}
+//# sourceMappingURL=vector_db_qa.js.map

package/dist/esm/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,OAAO,EACL,SAAS,EAGT,mBAAmB,EACnB,WAAW,GACZ,MAAM,SAAS,CAAC;AAKjB,OAAO,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;AAmBhD,MAAM,OAAO,eAAgB,SAAQ,SAAS;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,qBAAqB,CAG3D,yBAAyB,EAAE,IAAI,CAAC,CAAC;QAEnC,OAAO,IAAI,eAAe,CAAC;YACzB,qBAAqB,EAAE,MAAM,mBAAmB,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,WAAW,CAAC,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"}

package/dist/esm/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/esm/document.js

@@ -0,0 +1,24 @@
+/**
+ * Interface for interacting with a document.
+ */
+export 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 : {};
+    }
+}
+//# sourceMappingURL=document.js.map

package/dist/esm/document.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"document.js","sourceRoot":"","sources":["../../document.ts"],"names":[],"mappings":"AAOA;;GAEG;AACH,MAAM,OAAO,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"}

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

@@ -0,0 +1,3 @@
+export class Embeddings {
+}
+//# sourceMappingURL=base.js.map

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

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

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

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

package/dist/esm/embeddings/index.js

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

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

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

package/dist/esm/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/esm/embeddings/openai.js

@@ -0,0 +1,82 @@
+import { backOff } from "exponential-backoff";
+import { chunkArray } from "../util";
+import { Embeddings } from "./base";
+let Configuration = null;
+let OpenAIApi = null;
+try {
+    // eslint-disable-next-line global-require
+    ({ Configuration, OpenAIApi } = require("openai"));
+}
+catch (_a) {
+    // ignore error
+}
+export class OpenAIEmbeddings extends 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 = 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 backOff(makeCompletionRequest, {
+            startingDelay: 4,
+            maxDelay: 10,
+            numOfAttempts: this.maxRetries,
+        });
+    }
+}
+//# sourceMappingURL=openai.js.map

package/dist/esm/embeddings/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../embeddings/openai.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAEpC,IAAI,aAAa,GAAiC,IAAI,CAAC;AACvD,IAAI,SAAS,GAA6B,IAAI,CAAC;AAE/C,IAAI;IACF,0CAA0C;IAC1C,CAAC,EAAE,aAAa,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;CACpD;AAAC,WAAM;IACN,eAAe;CAChB;AAMD,MAAM,OAAO,gBAAiB,SAAQ,UAAU;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,UAAU,CAAC,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,OAAO,CAAC,qBAAqB,EAAE;YACpC,aAAa,EAAE,CAAC;YAChB,QAAQ,EAAE,EAAE;YACZ,aAAa,EAAE,IAAI,CAAC,UAAU;SAC/B,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/esm/embeddings/tests/openai.test.d.ts

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

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

@@ -0,0 +1,15 @@
+import { test, expect } from "@jest/globals";
+import { OpenAIEmbeddings } from "../openai";
+test("Test OpenAIEmbeddings.embedQuery", async () => {
+    const embeddings = new OpenAIEmbeddings();
+    const res = await embeddings.embedQuery("Hello world");
+    expect(typeof res[0]).toBe("number");
+});
+test("Test OpenAIEmbeddings.embedDocuments", async () => {
+    const embeddings = new OpenAIEmbeddings();
+    const res = await embeddings.embedDocuments(["Hello world", "Bye bye"]);
+    expect(res).toHaveLength(2);
+    expect(typeof res[0][0]).toBe("number");
+    expect(typeof res[1][0]).toBe("number");
+});
+//# sourceMappingURL=openai.test.js.map

package/dist/esm/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,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAE7C,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,UAAU,GAAG,IAAI,gBAAgB,EAAE,CAAC;IAC1C,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;IACvD,MAAM,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;IACtD,MAAM,UAAU,GAAG,IAAI,gBAAgB,EAAE,CAAC;IAC1C,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,cAAc,CAAC,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC,CAAC;IACxE,MAAM,CAAC,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC5B,MAAM,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxC,MAAM,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC"}