langchain 0.0.39 → 0.0.40

package/dist/chains/conversational_retrieval_chain.d.ts

@@ -0,0 +1,42 @@
+import { BaseLLM } from "../llms/index.js";
+import { SerializedChatVectorDBQAChain } from "./serde.js";
+import { ChainValues, BaseRetriever } from "../schema/index.js";
+import { BaseChain } from "./base.js";
+import { LLMChain } from "./llm_chain.js";
+export type LoadValues = Record<string, any>;
+export interface ConversationalRetrievalQAChainInput {
+    retriever: BaseRetriever;
+    combineDocumentsChain: BaseChain;
+    questionGeneratorChain: LLMChain;
+    outputKey: string;
+    inputKey: string;
+}
+export declare class ConversationalRetrievalQAChain extends BaseChain implements ConversationalRetrievalQAChainInput {
+    inputKey: string;
+    chatHistoryKey: string;
+    get inputKeys(): string[];
+    outputKey: string;
+    retriever: BaseRetriever;
+    combineDocumentsChain: BaseChain;
+    questionGeneratorChain: LLMChain;
+    returnSourceDocuments: boolean;
+    constructor(fields: {
+        retriever: BaseRetriever;
+        combineDocumentsChain: BaseChain;
+        questionGeneratorChain: LLMChain;
+        inputKey?: string;
+        outputKey?: string;
+        returnSourceDocuments?: boolean;
+    });
+    _call(values: ChainValues): Promise<ChainValues>;
+    _chainType(): string;
+    static deserialize(_data: SerializedChatVectorDBQAChain, _values: LoadValues): Promise<ConversationalRetrievalQAChain>;
+    serialize(): SerializedChatVectorDBQAChain;
+    static fromLLM(llm: BaseLLM, retriever: BaseRetriever, options?: {
+        inputKey?: string;
+        outputKey?: string;
+        returnSourceDocuments?: boolean;
+        questionGeneratorTemplate?: string;
+        qaTemplate?: string;
+    }): ConversationalRetrievalQAChain;
+}

package/dist/chains/conversational_retrieval_chain.js

@@ -0,0 +1,138 @@
+import { PromptTemplate } from "../prompts/index.js";
+import { BaseChain } from "./base.js";
+import { LLMChain } from "./llm_chain.js";
+import { loadQAStuffChain } from "./question_answering/load.js";
+const question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.
+
+Chat History:
+{chat_history}
+Follow Up Input: {question}
+Standalone question:`;
+const qa_template = `Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.
+
+{context}
+
+Question: {question}
+Helpful Answer:`;
+export class ConversationalRetrievalQAChain extends BaseChain {
+    get inputKeys() {
+        return [this.inputKey, this.chatHistoryKey];
+    }
+    constructor(fields) {
+        super();
+        Object.defineProperty(this, "inputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "question"
+        });
+        Object.defineProperty(this, "chatHistoryKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "chat_history"
+        });
+        Object.defineProperty(this, "outputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "result"
+        });
+        Object.defineProperty(this, "retriever", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "combineDocumentsChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "questionGeneratorChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "returnSourceDocuments", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        this.retriever = fields.retriever;
+        this.combineDocumentsChain = fields.combineDocumentsChain;
+        this.questionGeneratorChain = fields.questionGeneratorChain;
+        this.inputKey = fields.inputKey ?? this.inputKey;
+        this.outputKey = fields.outputKey ?? this.outputKey;
+        this.returnSourceDocuments =
+            fields.returnSourceDocuments ?? this.returnSourceDocuments;
+    }
+    async _call(values) {
+        if (!(this.inputKey in values)) {
+            throw new Error(`Question key ${this.inputKey} not found.`);
+        }
+        if (!(this.chatHistoryKey in values)) {
+            throw new Error(`chat history key ${this.inputKey} not found.`);
+        }
+        const question = values[this.inputKey];
+        const chatHistory = values[this.chatHistoryKey];
+        let newQuestion = question;
+        if (chatHistory.length > 0) {
+            const result = await this.questionGeneratorChain.call({
+                question,
+                chat_history: chatHistory,
+            });
+            const keys = Object.keys(result);
+            if (keys.length === 1) {
+                newQuestion = result[keys[0]];
+            }
+            else {
+                throw new Error("Return from llm chain has multiple values, only single values supported.");
+            }
+        }
+        const docs = await this.retriever.getRelevantDocuments(newQuestion);
+        const inputs = {
+            question: newQuestion,
+            input_documents: docs,
+            chat_history: chatHistory,
+        };
+        const result = await this.combineDocumentsChain.call(inputs);
+        if (this.returnSourceDocuments) {
+            return {
+                ...result,
+                sourceDocuments: docs,
+            };
+        }
+        return result;
+    }
+    _chainType() {
+        return "conversational_retrieval_chain";
+    }
+    static async deserialize(_data, _values) {
+        throw new Error("Not implemented.");
+    }
+    serialize() {
+        throw new Error("Not implemented.");
+    }
+    static fromLLM(llm, retriever, options = {}) {
+        const { questionGeneratorTemplate, qaTemplate, ...rest } = options;
+        const question_generator_prompt = PromptTemplate.fromTemplate(questionGeneratorTemplate || question_generator_template);
+        const qa_prompt = PromptTemplate.fromTemplate(qaTemplate || qa_template);
+        const qaChain = loadQAStuffChain(llm, { prompt: qa_prompt });
+        const questionGeneratorChain = new LLMChain({
+            prompt: question_generator_prompt,
+            llm,
+        });
+        const instance = new this({
+            retriever,
+            combineDocumentsChain: qaChain,
+            questionGeneratorChain,
+            ...rest,
+        });
+        return instance;
+    }
+}
+//# sourceMappingURL=conversational_retrieval_chain.js.map

package/dist/chains/conversational_retrieval_chain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversational_retrieval_chain.js","sourceRoot":"","sources":["../../src/chains/conversational_retrieval_chain.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAIrD,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAKhE,MAAM,2BAA2B,GAAG;;;;;qBAKf,CAAC;AAEtB,MAAM,WAAW,GAAG;;;;;gBAKJ,CAAC;AAUjB,MAAM,OAAO,8BACX,SAAQ,SAAS;IAOjB,IAAI,SAAS;QACX,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;IAC9C,CAAC;IAYD,YAAY,MAOX;QACC,KAAK,EAAE,CAAC;QA1BV;;;;mBAAW,UAAU;WAAC;QAEtB;;;;mBAAiB,cAAc;WAAC;QAMhC;;;;mBAAY,QAAQ;WAAC;QAErB;;;;;WAAyB;QAEzB;;;;;WAAiC;QAEjC;;;;;WAAiC;QAEjC;;;;mBAAwB,KAAK;WAAC;QAW5B,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAClC,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,qBAAqB,CAAC;QAC1D,IAAI,CAAC,sBAAsB,GAAG,MAAM,CAAC,sBAAsB,CAAC;QAC5D,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC;QACjD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACpD,IAAI,CAAC,qBAAqB;YACxB,MAAM,CAAC,qBAAqB,IAAI,IAAI,CAAC,qBAAqB,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CAAC,gBAAgB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SAC7D;QACD,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,IAAI,MAAM,CAAC,EAAE;YACpC,MAAM,IAAI,KAAK,CAAC,oBAAoB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SACjE;QACD,MAAM,QAAQ,GAAW,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,WAAW,GAAW,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACxD,IAAI,WAAW,GAAG,QAAQ,CAAC;QAC3B,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE;YAC1B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC;gBACpD,QAAQ;gBACR,YAAY,EAAE,WAAW;aAC1B,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE;gBACrB,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;aAC/B;iBAAM;gBACL,MAAM,IAAI,KAAK,CACb,0EAA0E,CAC3E,CAAC;aACH;SACF;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;QACpE,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,WAAW;YACrB,eAAe,EAAE,IAAI;YACrB,YAAY,EAAE,WAAW;SAC1B,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7D,IAAI,IAAI,CAAC,qBAAqB,EAAE;YAC9B,OAAO;gBACL,GAAG,MAAM;gBACT,eAAe,EAAE,IAAI;aACtB,CAAC;SACH;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,gCAAgC,CAAC;IAC1C,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,KAAoC,EACpC,OAAmB;QAEnB,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;IACtC,CAAC;IAED,SAAS;QACP,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;IACtC,CAAC;IAED,MAAM,CAAC,OAAO,CACZ,GAAY,EACZ,SAAwB,EACxB,UAMI,EAAE;QAEN,MAAM,EAAE,yBAAyB,EAAE,UAAU,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;QACnE,MAAM,yBAAyB,GAAG,cAAc,CAAC,YAAY,CAC3D,yBAAyB,IAAI,2BAA2B,CACzD,CAAC;QACF,MAAM,SAAS,GAAG,cAAc,CAAC,YAAY,CAAC,UAAU,IAAI,WAAW,CAAC,CAAC;QAEzE,MAAM,OAAO,GAAG,gBAAgB,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QAC7D,MAAM,sBAAsB,GAAG,IAAI,QAAQ,CAAC;YAC1C,MAAM,EAAE,yBAAyB;YACjC,GAAG;SACJ,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC;YACxB,SAAS;YACT,qBAAqB,EAAE,OAAO;YAC9B,sBAAsB;YACtB,GAAG,IAAI;SACR,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}

package/dist/chains/index.d.ts

@@ -8,5 +8,6 @@ export { loadChain } from "./load.js";
 export { loadQAChain, loadQAStuffChain, loadQAMapReduceChain, } from "./question_answering/load.js";
 export { loadSummarizationChain } from "./summarization/load.js";
 export { SqlDatabaseChain } from "./sql_db/sql_db_chain.js";
+export { ConversationalRetrievalQAChain } from "./conversational_retrieval_chain.js";
+export { RetrievalQAChain } from "./retrieval_qa.js";
 export { SerializedLLMChain, SerializedSqlDatabaseChain, SerializedAnalyzeDocumentChain, SerializedBaseChain, SerializedChatVectorDBQAChain, SerializedMapReduceDocumentsChain, SerializedStuffDocumentsChain, SerializedVectorDBQAChain, } from "./serde.js";
-export { OutputFixingParser } from "../output_parsers/fix.js";

package/dist/chains/index.js

@@ -8,5 +8,6 @@ export { loadChain } from "./load.js";
 export { loadQAChain, loadQAStuffChain, loadQAMapReduceChain, } from "./question_answering/load.js";
 export { loadSummarizationChain } from "./summarization/load.js";
 export { SqlDatabaseChain } from "./sql_db/sql_db_chain.js";
-export { OutputFixingParser } from "../output_parsers/fix.js";
+export { ConversationalRetrievalQAChain } from "./conversational_retrieval_chain.js";
+export { RetrievalQAChain } from "./retrieval_qa.js";
 //# sourceMappingURL=index.js.map

package/dist/chains/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/chains/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAe,MAAM,WAAW,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAC7D,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAChE,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EACL,WAAW,EACX,gBAAgB,EAChB,oBAAoB,GACrB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAW5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/chains/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAe,MAAM,WAAW,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAC7D,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAChE,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EACL,WAAW,EACX,gBAAgB,EAChB,oBAAoB,GACrB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,8BAA8B,EAAE,MAAM,qCAAqC,CAAC;AACrF,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/chains/retrieval_qa.d.ts

@@ -0,0 +1,32 @@
+import { BaseChain } from "./base.js";
+import { BaseLLM } from "../llms/index.js";
+import { SerializedVectorDBQAChain } from "./serde.js";
+import { ChainValues, BaseRetriever } from "../schema/index.js";
+export type LoadValues = Record<string, any>;
+export interface RetrievalQAChainInput {
+    retriever: BaseRetriever;
+    combineDocumentsChain: BaseChain;
+    outputKey: string;
+    inputKey: string;
+    returnSourceDocuments?: boolean;
+}
+export declare class RetrievalQAChain extends BaseChain implements RetrievalQAChainInput {
+    inputKey: string;
+    get inputKeys(): string[];
+    outputKey: string;
+    retriever: BaseRetriever;
+    combineDocumentsChain: BaseChain;
+    returnSourceDocuments: boolean;
+    constructor(fields: {
+        retriever: BaseRetriever;
+        combineDocumentsChain: BaseChain;
+        inputKey?: string;
+        outputKey?: string;
+        returnSourceDocuments?: boolean;
+    });
+    _call(values: ChainValues): Promise<ChainValues>;
+    _chainType(): "retrieval_qa";
+    static deserialize(_data: SerializedVectorDBQAChain, _values: LoadValues): Promise<RetrievalQAChain>;
+    serialize(): SerializedVectorDBQAChain;
+    static fromLLM(llm: BaseLLM, retriever: BaseRetriever, options?: Partial<Omit<RetrievalQAChainInput, "combineDocumentsChain" | "index">>): RetrievalQAChain;
+}

package/dist/chains/retrieval_qa.js

@@ -0,0 +1,80 @@
+import { BaseChain } from "./base.js";
+import { loadQAStuffChain } from "./question_answering/load.js";
+export class RetrievalQAChain extends BaseChain {
+    get inputKeys() {
+        return [this.inputKey];
+    }
+    constructor(fields) {
+        super();
+        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, "retriever", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "combineDocumentsChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "returnSourceDocuments", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        this.retriever = fields.retriever;
+        this.combineDocumentsChain = fields.combineDocumentsChain;
+        this.inputKey = fields.inputKey ?? this.inputKey;
+        this.outputKey = fields.outputKey ?? this.outputKey;
+        this.returnSourceDocuments =
+            fields.returnSourceDocuments ?? this.returnSourceDocuments;
+    }
+    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.retriever.getRelevantDocuments(question);
+        const inputs = { question, input_documents: docs };
+        const result = await this.combineDocumentsChain.call(inputs);
+        if (this.returnSourceDocuments) {
+            return {
+                ...result,
+                sourceDocuments: docs,
+            };
+        }
+        return result;
+    }
+    _chainType() {
+        return "retrieval_qa";
+    }
+    static async deserialize(_data, _values) {
+        throw new Error("Not implemented");
+    }
+    serialize() {
+        throw new Error("Not implemented");
+    }
+    static fromLLM(llm, retriever, options) {
+        const qaChain = loadQAStuffChain(llm);
+        return new this({
+            retriever,
+            combineDocumentsChain: qaChain,
+            ...options,
+        });
+    }
+}
+//# sourceMappingURL=retrieval_qa.js.map

package/dist/chains/retrieval_qa.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieval_qa.js","sourceRoot":"","sources":["../../src/chains/retrieval_qa.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAItC,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAahE,MAAM,OAAO,gBACX,SAAQ,SAAS;IAKjB,IAAI,SAAS;QACX,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACzB,CAAC;IAUD,YAAY,MAMX;QACC,KAAK,EAAE,CAAC;QArBV;;;;mBAAW,OAAO;WAAC;QAMnB;;;;mBAAY,QAAQ;WAAC;QAErB;;;;;WAAyB;QAEzB;;;;;WAAiC;QAEjC;;;;mBAAwB,KAAK;WAAC;QAU5B,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAClC,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,qBAAqB,CAAC;QAC1D,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC;QACjD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACpD,IAAI,CAAC,qBAAqB;YACxB,MAAM,CAAC,qBAAqB,IAAI,IAAI,CAAC,qBAAqB,CAAC;IAC/D,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,SAAS,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QACjE,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,IAAI,IAAI,CAAC,qBAAqB,EAAE;YAC9B,OAAO;gBACL,GAAG,MAAM;gBACT,eAAe,EAAE,IAAI;aACtB,CAAC;SACH;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,cAAuB,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,KAAgC,EAChC,OAAmB;QAEnB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED,SAAS;QACP,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED,MAAM,CAAC,OAAO,CACZ,GAAY,EACZ,SAAwB,EACxB,OAEC;QAED,MAAM,OAAO,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;QACtC,OAAO,IAAI,IAAI,CAAC;YACd,SAAS;YACT,qBAAqB,EAAE,OAAO;YAC9B,GAAG,OAAO;SACX,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/embeddings/cohere.d.ts

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

package/dist/embeddings/cohere.js

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

package/dist/embeddings/cohere.js.map

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

package/dist/retrievers/chatgpt-plugin.d.ts

@@ -0,0 +1,44 @@
+import { Document } from "../document.js";
+import { BaseRetriever } from "../schema/index.js";
+import { AsyncCaller, AsyncCallerParams } from "../util/async_caller.js";
+interface ChatGPTPluginRetrieverFilter {
+    document_id?: string;
+    source?: string;
+    source_id?: string;
+    author?: string;
+    start_date?: string;
+    end_date?: string;
+}
+type ChatGPTPluginRetrieverAuth = false | {
+    bearer: string;
+};
+interface ChatGPTPluginRetrieverParams extends AsyncCallerParams {
+    /**
+     * The URL of the ChatGPTRetrievalPlugin server
+     */
+    url: string;
+    /**
+     * The authentication method to use, currently implemented is
+     * - false: no authentication
+     * - { bearer: string }: Bearer token authentication
+     */
+    auth: ChatGPTPluginRetrieverAuth;
+    /**
+     * The number of results to request from the ChatGPTRetrievalPlugin server
+     */
+    topK?: number;
+    /**
+     * The filter to use when querying the ChatGPTRetrievalPlugin server
+     */
+    filter?: ChatGPTPluginRetrieverFilter;
+}
+export declare class ChatGPTPluginRetriever extends BaseRetriever implements ChatGPTPluginRetrieverParams {
+    url: string;
+    auth: ChatGPTPluginRetrieverAuth;
+    topK: number;
+    filter?: ChatGPTPluginRetrieverFilter;
+    asyncCaller: AsyncCaller;
+    constructor({ url, auth, topK, filter, ...rest }: ChatGPTPluginRetrieverParams);
+    getRelevantDocuments(query: string): Promise<Document[]>;
+}
+export {};

package/dist/retrievers/chatgpt-plugin.js

@@ -0,0 +1,79 @@
+import { Document } from "../document.js";
+import { BaseRetriever } from "../schema/index.js";
+import { AsyncCaller } from "../util/async_caller.js";
+export class ChatGPTPluginRetriever extends BaseRetriever {
+    constructor({ url, auth, topK = 4, filter, ...rest }) {
+        super();
+        Object.defineProperty(this, "url", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "auth", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "topK", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "filter", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "asyncCaller", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.url = url;
+        this.auth = auth;
+        this.topK = topK;
+        this.asyncCaller = new AsyncCaller(rest);
+    }
+    async getRelevantDocuments(query) {
+        const res = await this.asyncCaller.call(fetch, `${this.url}/query`, {
+            method: "POST",
+            body: JSON.stringify({
+                queries: [
+                    {
+                        query,
+                        top_k: this.topK,
+                        filter: this.filter,
+                    },
+                ],
+            }),
+            headers: {
+                Accept: "application/json",
+                "Content-Type": "application/json",
+                ...(this.auth && this.auth.bearer
+                    ? { Authorization: `Bearer ${this.auth.bearer}` }
+                    : {}),
+            },
+        });
+        if (!res.ok) {
+            throw new Error(`Error calling ChatGPTPluginRetriever: ${res.status}`);
+        }
+        const body = await res.json();
+        const results = body?.results?.[0]?.results;
+        if (!results) {
+            // Note an empty array of results would not fall into this case
+            throw new Error("No results returned from ChatGPTPluginRetriever");
+        }
+        return results.map(
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (result) => new Document({
+            pageContent: result.text,
+            metadata: result.metadata,
+        }));
+    }
+}
+//# sourceMappingURL=chatgpt-plugin.js.map

package/dist/retrievers/chatgpt-plugin.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatgpt-plugin.js","sourceRoot":"","sources":["../../src/retrievers/chatgpt-plugin.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAqB,MAAM,yBAAyB,CAAC;AAqCzE,MAAM,OAAO,sBACX,SAAQ,aAAa;IAarB,YAAY,EACV,GAAG,EACH,IAAI,EACJ,IAAI,GAAG,CAAC,EACR,MAAM,EACN,GAAG,IAAI,EACsB;QAC7B,KAAK,EAAE,CAAC;QAjBV;;;;;WAAY;QAEZ;;;;;WAAiC;QAEjC;;;;;WAAa;QAEb;;;;;WAAsC;QAEtC;;;;;WAAyB;QAWvB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,KAAa;QACtC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,GAAG,QAAQ,EAAE;YAClE,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,OAAO,EAAE;oBACP;wBACE,KAAK;wBACL,KAAK,EAAE,IAAI,CAAC,IAAI;wBAChB,MAAM,EAAE,IAAI,CAAC,MAAM;qBACpB;iBACF;aACF,CAAC;YACF,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;gBAClC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM;oBAC/B,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE;oBACjD,CAAC,CAAC,EAAE,CAAC;aACR;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,yCAAyC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;SACxE;QAED,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;QAE5C,IAAI,CAAC,OAAO,EAAE;YACZ,+DAA+D;YAC/D,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;SACpE;QAED,OAAO,OAAO,CAAC,GAAG;QAChB,8DAA8D;QAC9D,CAAC,MAAW,EAAE,EAAE,CACd,IAAI,QAAQ,CAAC;YACX,WAAW,EAAE,MAAM,CAAC,IAAI;YACxB,QAAQ,EAAE,MAAM,CAAC,QAAQ;SAC1B,CAAC,CACL,CAAC;IACJ,CAAC;CACF"}

package/dist/retrievers/index.d.ts

@@ -0,0 +1 @@
+export { ChatGPTPluginRetriever } from "./chatgpt-plugin.js";

package/dist/retrievers/index.js

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

package/dist/retrievers/index.js.map

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

package/dist/schema/index.d.ts

@@ -1,3 +1,4 @@
+import { Document } from "../document.js";
 export type Example = Record<string, string>;
 export type InputValues = Record<string, any>;
 export type PartialValues = Record<string, string | (() => Promise<string>) | (() => string)>;
@@ -79,7 +80,12 @@ export type AgentStep = {
 };
 export type ChainValues = Record<string, any>;
 /**
- * Class to parse the output of an LLM call.
+ * Base Index class. All indexes should extend this class.
+ */
+export declare abstract class BaseRetriever {
+    abstract getRelevantDocuments(query: string): Promise<Document[]>;
+}
+/** Class to parse the output of an LLM call.
  */
 export declare abstract class BaseOutputParser {
     /**

package/dist/schema/index.js

@@ -46,7 +46,11 @@ export class ChatMessage extends BaseChatMessage {
 export class BasePromptValue {
 }
 /**
- * Class to parse the output of an LLM call.
+ * Base Index class. All indexes should extend this class.
+ */
+export class BaseRetriever {
+}
+/** Class to parse the output of an LLM call.
  */
 export class BaseOutputParser {
     async parseWithPrompt(text, _prompt) {

package/dist/schema/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/schema/index.ts"],"names":[],"mappings":"AA0CA,MAAM,OAAgB,eAAe;IAOnC,YAAY,IAAY;QANxB,+BAA+B;QAC/B;;;;;WAAa;QAMX,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;CACF;AAED,MAAM,OAAO,gBAAiB,SAAQ,eAAe;IACnD,QAAQ;QACN,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AAED,MAAM,OAAO,aAAc,SAAQ,eAAe;IAChD,QAAQ;QACN,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAED,MAAM,OAAO,iBAAkB,SAAQ,eAAe;IACpD,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AAED,MAAM,OAAO,WAAY,SAAQ,eAAe;IAG9C,YAAY,IAAY,EAAE,IAAY;QACpC,KAAK,CAAC,IAAI,CAAC,CAAC;QAHd;;;;;WAAa;QAIX,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,QAAQ;QACN,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AAaD;;GAEG;AACH,MAAM,OAAgB,eAAe;CAIpC;AAqBD;;GAEG;AACH,MAAM,OAAgB,gBAAgB;IASpC,KAAK,CAAC,eAAe,CACnB,IAAY,EACZ,OAAwB;QAExB,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAcD;;OAEG;IACH,KAAK;QACH,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;CACF;AAED,MAAM,OAAO,qBAAsB,SAAQ,KAAK;IAC9C,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;IACjB,CAAC;CACF"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/schema/index.ts"],"names":[],"mappings":"AA4CA,MAAM,OAAgB,eAAe;IAOnC,YAAY,IAAY;QANxB,+BAA+B;QAC/B;;;;;WAAa;QAMX,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;CACF;AAED,MAAM,OAAO,gBAAiB,SAAQ,eAAe;IACnD,QAAQ;QACN,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AAED,MAAM,OAAO,aAAc,SAAQ,eAAe;IAChD,QAAQ;QACN,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAED,MAAM,OAAO,iBAAkB,SAAQ,eAAe;IACpD,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AAED,MAAM,OAAO,WAAY,SAAQ,eAAe;IAG9C,YAAY,IAAY,EAAE,IAAY;QACpC,KAAK,CAAC,IAAI,CAAC,CAAC;QAHd;;;;;WAAa;QAIX,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,QAAQ;QACN,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AAaD;;GAEG;AACH,MAAM,OAAgB,eAAe;CAIpC;AAqBD;;GAEG;AACH,MAAM,OAAgB,aAAa;CAElC;AACD;GACG;AACH,MAAM,OAAgB,gBAAgB;IASpC,KAAK,CAAC,eAAe,CACnB,IAAY,EACZ,OAAwB;QAExB,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAcD;;OAEG;IACH,KAAK;QACH,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;CACF;AAED,MAAM,OAAO,qBAAsB,SAAQ,KAAK;IAC9C,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;IACjB,CAAC;CACF"}

package/dist/vectorstores/base.d.ts

@@ -1,15 +1,26 @@
 import { Embeddings } from "../embeddings/base.js";
 import { Document } from "../document.js";
+import { BaseRetriever } from "../schema/index.js";
+export declare class VectorStoreRetriever extends BaseRetriever {
+    vectorStore: VectorStore;
+    k: number;
+    constructor(fields: {
+        vectorStore: VectorStore;
+        k?: number;
+    });
+    getRelevantDocuments(query: string): Promise<Document[]>;
+}
 export declare abstract class VectorStore {
     embeddings: Embeddings;
     constructor(embeddings: Embeddings, _dbConfig: Record<string, any>);
     abstract addVectors(vectors: number[][], documents: Document[]): Promise<void>;
     abstract addDocuments(documents: Document[]): Promise<void>;
     abstract similaritySearchVectorWithScore(query: number[], k: number, filter?: object): Promise<[Document, number][]>;
-    similaritySearch(query: string, k?: number, filter?: object): Promise<Document[]>;
-    similaritySearchWithScore(query: string, k?: number, filter?: object): Promise<[object, number][]>;
+    similaritySearch(query: string, k?: number, filter?: object | undefined): Promise<Document[]>;
+    similaritySearchWithScore(query: string, k?: number, filter?: object | undefined): Promise<[object, number][]>;
     static fromTexts(_texts: string[], _metadatas: object[], _embeddings: Embeddings, _dbConfig: Record<string, any>): Promise<VectorStore>;
     static fromDocuments(_docs: Document[], _embeddings: Embeddings, _dbConfig: Record<string, any>): Promise<VectorStore>;
+    asRetriever(k?: number): BaseRetriever;
 }
 export declare abstract class SaveableVectorStore extends VectorStore {
     abstract save(directory: string): Promise<void>;

package/dist/vectorstores/base.js

@@ -1,3 +1,27 @@
+import { BaseRetriever } from "../schema/index.js";
+export class VectorStoreRetriever extends BaseRetriever {
+    constructor(fields) {
+        super();
+        Object.defineProperty(this, "vectorStore", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "k", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 4
+        });
+        this.vectorStore = fields.vectorStore;
+        this.k = fields.k ?? this.k;
+    }
+    async getRelevantDocuments(query) {
+        const results = await this.vectorStore.similaritySearch(query, 4);
+        return results;
+    }
+}
 export class VectorStore {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     constructor(embeddings, _dbConfig) {
@@ -9,11 +33,11 @@ export class VectorStore {
         });
         this.embeddings = embeddings;
     }
-    async similaritySearch(query, k = 4, filter = {}) {
+    async similaritySearch(query, k = 4, filter = undefined) {
         const results = await this.similaritySearchVectorWithScore(await this.embeddings.embedQuery(query), k, filter);
         return results.map((result) => result[0]);
     }
-    async similaritySearchWithScore(query, k = 4, filter = {}) {
+    async similaritySearchWithScore(query, k = 4, filter = undefined) {
         return this.similaritySearchVectorWithScore(await this.embeddings.embedQuery(query), k, filter);
     }
     static fromTexts(_texts, _metadatas, _embeddings, 
@@ -26,6 +50,9 @@ export class VectorStore {
     _dbConfig) {
         throw new Error("the Langchain vectorstore implementation you are using forgot to override this, please report a bug");
     }
+    asRetriever(k) {
+        return new VectorStoreRetriever({ vectorStore: this, k });
+    }
 }
 export class SaveableVectorStore extends VectorStore {
     static load(_directory, _embeddings) {

package/dist/vectorstores/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/vectorstores/base.ts"],"names":[],"mappings":"AAGA,MAAM,OAAgB,WAAW;IAG/B,8DAA8D;IAC9D,YAAY,UAAsB,EAAE,SAA8B;QAHlE;;;;;WAAuB;QAIrB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAeD,KAAK,CAAC,gBAAgB,CACpB,KAAa,EACb,CAAC,GAAG,CAAC,EACL,SAAiB,EAAE;QAEnB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,+BAA+B,CACxD,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EACvC,CAAC,EACD,MAAM,CACP,CAAC;QAEF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC7B,KAAa,EACb,CAAC,GAAG,CAAC,EACL,SAAiB,EAAE;QAEnB,OAAO,IAAI,CAAC,+BAA+B,CACzC,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EACvC,CAAC,EACD,MAAM,CACP,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,SAAS,CACd,MAAgB,EAChB,UAAoB,EACpB,WAAuB;IACvB,8DAA8D;IAC9D,SAA8B;QAE9B,MAAM,IAAI,KAAK,CACb,qGAAqG,CACtG,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,aAAa,CAClB,KAAiB,EACjB,WAAuB;IACvB,8DAA8D;IAC9D,SAA8B;QAE9B,MAAM,IAAI,KAAK,CACb,qGAAqG,CACtG,CAAC;IACJ,CAAC;CACF;AAED,MAAM,OAAgB,mBAAoB,SAAQ,WAAW;IAG3D,MAAM,CAAC,IAAI,CACT,UAAkB,EAClB,WAAuB;QAEvB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF"}
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/vectorstores/base.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,MAAM,OAAO,oBAAqB,SAAQ,aAAa;IAKrD,YAAY,MAAgD;QAC1D,KAAK,EAAE,CAAC;QALV;;;;;WAAyB;QAEzB;;;;mBAAI,CAAC;WAAC;QAIJ,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,KAAa;QACtC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAClE,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AAED,MAAM,OAAgB,WAAW;IAG/B,8DAA8D;IAC9D,YAAY,UAAsB,EAAE,SAA8B;QAHlE;;;;;WAAuB;QAIrB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAeD,KAAK,CAAC,gBAAgB,CACpB,KAAa,EACb,CAAC,GAAG,CAAC,EACL,SAA6B,SAAS;QAEtC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,+BAA+B,CACxD,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EACvC,CAAC,EACD,MAAM,CACP,CAAC;QAEF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC7B,KAAa,EACb,CAAC,GAAG,CAAC,EACL,SAA6B,SAAS;QAEtC,OAAO,IAAI,CAAC,+BAA+B,CACzC,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EACvC,CAAC,EACD,MAAM,CACP,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,SAAS,CACd,MAAgB,EAChB,UAAoB,EACpB,WAAuB;IACvB,8DAA8D;IAC9D,SAA8B;QAE9B,MAAM,IAAI,KAAK,CACb,qGAAqG,CACtG,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,aAAa,CAClB,KAAiB,EACjB,WAAuB;IACvB,8DAA8D;IAC9D,SAA8B;QAE9B,MAAM,IAAI,KAAK,CACb,qGAAqG,CACtG,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,CAAU;QACpB,OAAO,IAAI,oBAAoB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC;CACF;AAED,MAAM,OAAgB,mBAAoB,SAAQ,WAAW;IAG3D,MAAM,CAAC,IAAI,CACT,UAAkB,EAClB,WAAuB;QAEvB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF"}

package/dist/vectorstores/tests/pinecone.test.js

@@ -14,7 +14,7 @@ test("PineconeStore with external ids", async () => {
     expect(store).toBeDefined();
     await store.addDocuments([{ pageContent: "hello", metadata: { a: 1 } }], ["id1"]);
     expect(client.upsert).toHaveBeenCalledTimes(1);
-    const results = await store.similaritySearch("hello");
+    const results = await store.similaritySearch("hello", 1);
     expect(results).toHaveLength(0);
 });
 test("PineconeStore with generated ids", async () => {
@@ -29,7 +29,7 @@ test("PineconeStore with generated ids", async () => {
     expect(store).toBeDefined();
     await store.addDocuments([{ pageContent: "hello", metadata: { a: 1 } }]);
     expect(client.upsert).toHaveBeenCalledTimes(1);
-    const results = await store.similaritySearch("hello");
+    const results = await store.similaritySearch("hello", 1);
     expect(results).toHaveLength(0);
 });
 //# sourceMappingURL=pinecone.test.js.map

package/dist/vectorstores/tests/pinecone.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"pinecone.test.js","sourceRoot":"","sources":["../../../src/vectorstores/tests/pinecone.test.ts"],"names":[],"mappings":"AAAA,uDAAuD;AACvD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;IACjD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,MAAa,EAAE,CAAC,CAAC;IAE9E,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CACtB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAC9C,CAAC,KAAK,CAAC,CACR,CAAC;IAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAEtD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,MAAa,EAAE,CAAC,CAAC;IAE9E,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAEtD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC"}
+{"version":3,"file":"pinecone.test.js","sourceRoot":"","sources":["../../../src/vectorstores/tests/pinecone.test.ts"],"names":[],"mappings":"AAAA,uDAAuD;AACvD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;IACjD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,MAAa,EAAE,CAAC,CAAC;IAE9E,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CACtB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAC9C,CAAC,KAAK,CAAC,CACR,CAAC;IAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAO,CAAC,iBAAiB,CAAC;YACtC,OAAO,EAAE,EAAE;SACZ,CAAC;KACH,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;IAExC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,MAAa,EAAE,CAAC,CAAC;IAE9E,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5B,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAE/C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.39",
+  "version": "0.0.40",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "main": "./index.js",
@@ -43,6 +43,8 @@
     "callbacks.d.ts",
     "output_parsers.js",
     "output_parsers.d.ts",
+    "retrievers.js",
+    "retrievers.d.ts",
     "index.js",
     "index.d.ts"
   ],
@@ -292,6 +294,10 @@
     "./output_parsers": {
       "types": "./output_parsers.d.ts",
       "import": "./output_parsers.js"
+    },
+    "./retrievers": {
+      "types": "./retrievers.d.ts",
+      "import": "./retrievers.js"
     }
   }
 }

package/retrievers.d.ts

@@ -0,0 +1 @@
+export * from './dist/retrievers/index.js'

package/retrievers.js

@@ -0,0 +1 @@
+export * from './dist/retrievers/index.js'