langchain 0.0.147 → 0.0.149

package/dist/llms/{googlevertexai.cjs → googlevertexai/common.js}

@@ -1,25 +1,23 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.GoogleVertexAI = void 0;
-const base_js_1 = require("./base.cjs");
-const googlevertexai_connection_js_1 = require("../util/googlevertexai-connection.cjs");
+import { BaseLLM } from "../base.js";
 /**
- * Enables calls to the Google Cloud's Vertex AI API to access
- * Large Language Models.
- *
- * To use, you will need to have one of the following authentication
- * methods in place:
- * - You are logged into an account permitted to the Google Cloud project
- *   using Vertex AI.
- * - You are running this on a machine using a service account permitted to
- *   the Google Cloud project using Vertex AI.
- * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
- *   path of a credentials file for a service account permitted to the
- *   Google Cloud project using Vertex AI.
+ * Base class for Google Vertex AI LLMs.
+ * Implemented subclasses must provide a GoogleVertexAILLMConnection
+ * with an appropriate auth client.
  */
-class GoogleVertexAI extends base_js_1.BaseLLM {
+export class BaseGoogleVertexAI extends BaseLLM {
+    get lc_aliases() {
+        return {
+            model: "model_name",
+        };
+    }
     constructor(fields) {
         super(fields ?? {});
+        Object.defineProperty(this, "lc_serializable", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         Object.defineProperty(this, "model", {
             enumerable: true,
             configurable: true,
@@ -68,10 +66,9 @@ class GoogleVertexAI extends base_js_1.BaseLLM {
         this.maxOutputTokens = fields?.maxOutputTokens ?? this.maxOutputTokens;
         this.topP = fields?.topP ?? this.topP;
         this.topK = fields?.topK ?? this.topK;
-        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
     }
     _llmType() {
-        return "googlevertexai";
+        return "vertexai";
     }
     async _generate(prompts, options) {
         const generations = await Promise.all(prompts.map((prompt) => this._generatePrompt(prompt, options)));
@@ -132,4 +129,3 @@ class GoogleVertexAI extends base_js_1.BaseLLM {
         return result?.data?.predictions[0];
     }
 }
-exports.GoogleVertexAI = GoogleVertexAI;

package/dist/llms/googlevertexai/index.cjs

@@ -0,0 +1,34 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GoogleVertexAI = void 0;
+const google_auth_library_1 = require("google-auth-library");
+const googlevertexai_connection_js_1 = require("../../util/googlevertexai-connection.cjs");
+const common_js_1 = require("./common.cjs");
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models.
+ *
+ * To use, you will need to have one of the following authentication
+ * methods in place:
+ * - You are logged into an account permitted to the Google Cloud project
+ *   using Vertex AI.
+ * - You are running this on a machine using a service account permitted to
+ *   the Google Cloud project using Vertex AI.
+ * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
+ *   path of a credentials file for a service account permitted to the
+ *   Google Cloud project using Vertex AI.
+ */
+class GoogleVertexAI extends common_js_1.BaseGoogleVertexAI {
+    static lc_name() {
+        return "VertexAI";
+    }
+    constructor(fields) {
+        super(fields);
+        const client = new google_auth_library_1.GoogleAuth({
+            scopes: "https://www.googleapis.com/auth/cloud-platform",
+            ...fields?.authOptions,
+        });
+        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, client);
+    }
+}
+exports.GoogleVertexAI = GoogleVertexAI;

package/dist/llms/googlevertexai/index.d.ts

@@ -0,0 +1,26 @@
+import { GoogleAuthOptions } from "google-auth-library";
+import { GoogleVertexAIBaseLLMInput } from "../../types/googlevertexai-types.js";
+import { BaseGoogleVertexAI } from "./common.js";
+/**
+ * Interface representing the input to the Google Vertex AI model.
+ */
+export interface GoogleVertexAITextInput extends GoogleVertexAIBaseLLMInput<GoogleAuthOptions> {
+}
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models.
+ *
+ * To use, you will need to have one of the following authentication
+ * methods in place:
+ * - You are logged into an account permitted to the Google Cloud project
+ *   using Vertex AI.
+ * - You are running this on a machine using a service account permitted to
+ *   the Google Cloud project using Vertex AI.
+ * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
+ *   path of a credentials file for a service account permitted to the
+ *   Google Cloud project using Vertex AI.
+ */
+export declare class GoogleVertexAI extends BaseGoogleVertexAI<GoogleAuthOptions> {
+    static lc_name(): string;
+    constructor(fields?: GoogleVertexAITextInput);
+}

package/dist/llms/googlevertexai/index.js

@@ -0,0 +1,30 @@
+import { GoogleAuth } from "google-auth-library";
+import { GoogleVertexAILLMConnection } from "../../util/googlevertexai-connection.js";
+import { BaseGoogleVertexAI } from "./common.js";
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models.
+ *
+ * To use, you will need to have one of the following authentication
+ * methods in place:
+ * - You are logged into an account permitted to the Google Cloud project
+ *   using Vertex AI.
+ * - You are running this on a machine using a service account permitted to
+ *   the Google Cloud project using Vertex AI.
+ * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
+ *   path of a credentials file for a service account permitted to the
+ *   Google Cloud project using Vertex AI.
+ */
+export class GoogleVertexAI extends BaseGoogleVertexAI {
+    static lc_name() {
+        return "VertexAI";
+    }
+    constructor(fields) {
+        super(fields);
+        const client = new GoogleAuth({
+            scopes: "https://www.googleapis.com/auth/cloud-platform",
+            ...fields?.authOptions,
+        });
+        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, client);
+    }
+}

package/dist/llms/googlevertexai/web.cjs

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GoogleVertexAI = void 0;
+const googlevertexai_webauth_js_1 = require("../../util/googlevertexai-webauth.cjs");
+const googlevertexai_connection_js_1 = require("../../util/googlevertexai-connection.cjs");
+const common_js_1 = require("./common.cjs");
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models.
+ *
+ * This entrypoint and class are intended to be used in web environments like Edge
+ * functions where you do not have access to the file system. It supports passing
+ * service account credentials directly as a "GOOGLE_VERTEX_AI_WEB_CREDENTIALS"
+ * environment variable or directly as "authOptions.credentials".
+ */
+class GoogleVertexAI extends common_js_1.BaseGoogleVertexAI {
+    static lc_name() {
+        return "VertexAI";
+    }
+    get lc_secrets() {
+        return {
+            "authOptions.credentials": "GOOGLE_VERTEX_AI_WEB_CREDENTIALS",
+        };
+    }
+    constructor(fields) {
+        super(fields);
+        const client = new googlevertexai_webauth_js_1.WebGoogleAuth(fields?.authOptions);
+        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, client);
+    }
+}
+exports.GoogleVertexAI = GoogleVertexAI;

package/dist/llms/googlevertexai/web.d.ts

@@ -0,0 +1,24 @@
+import { WebGoogleAuthOptions } from "../../util/googlevertexai-webauth.js";
+import { GoogleVertexAIBaseLLMInput } from "../../types/googlevertexai-types.js";
+import { BaseGoogleVertexAI } from "./common.js";
+/**
+ * Interface representing the input to the Google Vertex AI model.
+ */
+export interface GoogleVertexAITextInput extends GoogleVertexAIBaseLLMInput<WebGoogleAuthOptions> {
+}
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models.
+ *
+ * This entrypoint and class are intended to be used in web environments like Edge
+ * functions where you do not have access to the file system. It supports passing
+ * service account credentials directly as a "GOOGLE_VERTEX_AI_WEB_CREDENTIALS"
+ * environment variable or directly as "authOptions.credentials".
+ */
+export declare class GoogleVertexAI extends BaseGoogleVertexAI<WebGoogleAuthOptions> {
+    static lc_name(): string;
+    get lc_secrets(): {
+        [key: string]: string;
+    };
+    constructor(fields?: GoogleVertexAITextInput);
+}

package/dist/llms/googlevertexai/web.js

@@ -0,0 +1,27 @@
+import { WebGoogleAuth, } from "../../util/googlevertexai-webauth.js";
+import { GoogleVertexAILLMConnection } from "../../util/googlevertexai-connection.js";
+import { BaseGoogleVertexAI } from "./common.js";
+/**
+ * Enables calls to the Google Cloud's Vertex AI API to access
+ * Large Language Models.
+ *
+ * This entrypoint and class are intended to be used in web environments like Edge
+ * functions where you do not have access to the file system. It supports passing
+ * service account credentials directly as a "GOOGLE_VERTEX_AI_WEB_CREDENTIALS"
+ * environment variable or directly as "authOptions.credentials".
+ */
+export class GoogleVertexAI extends BaseGoogleVertexAI {
+    static lc_name() {
+        return "VertexAI";
+    }
+    get lc_secrets() {
+        return {
+            "authOptions.credentials": "GOOGLE_VERTEX_AI_WEB_CREDENTIALS",
+        };
+    }
+    constructor(fields) {
+        super(fields);
+        const client = new WebGoogleAuth(fields?.authOptions);
+        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, client);
+    }
+}

package/dist/load/import_constants.cjs

@@ -27,6 +27,7 @@ exports.optionalImportEntrypoints = [
     "langchain/llms/raycast",
     "langchain/llms/replicate",
     "langchain/llms/googlevertexai",
+    "langchain/llms/googlevertexai/web",
     "langchain/llms/googlepalm",
     "langchain/llms/sagemaker_endpoint",
     "langchain/llms/bedrock",
@@ -93,6 +94,7 @@ exports.optionalImportEntrypoints = [
     "langchain/document_transformers/html_to_text",
     "langchain/document_transformers/mozilla_readability",
     "langchain/chat_models/googlevertexai",
+    "langchain/chat_models/googlevertexai/web",
     "langchain/chat_models/googlepalm",
     "langchain/sql_db",
     "langchain/callbacks/handlers/llmonitor",

package/dist/load/import_constants.js

@@ -24,6 +24,7 @@ export const optionalImportEntrypoints = [
     "langchain/llms/raycast",
     "langchain/llms/replicate",
     "langchain/llms/googlevertexai",
+    "langchain/llms/googlevertexai/web",
     "langchain/llms/googlepalm",
     "langchain/llms/sagemaker_endpoint",
     "langchain/llms/bedrock",
@@ -90,6 +91,7 @@ export const optionalImportEntrypoints = [
     "langchain/document_transformers/html_to_text",
     "langchain/document_transformers/mozilla_readability",
     "langchain/chat_models/googlevertexai",
+    "langchain/chat_models/googlevertexai/web",
     "langchain/chat_models/googlepalm",
     "langchain/sql_db",
     "langchain/callbacks/handlers/llmonitor",

package/dist/load/import_map.cjs

@@ -25,7 +25,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__ollama = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__minimax = exports.embeddings__openai = exports.embeddings__ollama = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__openai_functions = exports.chains = exports.tools = exports.base_language = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
-exports.evaluation = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__parent_document = exports.retrievers__multi_vector = void 0;
+exports.evaluation = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = void 0;
 exports.load__serializable = __importStar(require("../load/serializable.cjs"));
 exports.agents = __importStar(require("../agents/index.cjs"));
 exports.agents__toolkits = __importStar(require("../agents/toolkits/index.cjs"));
@@ -76,6 +76,7 @@ exports.retrievers__remote = __importStar(require("../retrievers/remote/index.cj
 exports.retrievers__databerry = __importStar(require("../retrievers/databerry.cjs"));
 exports.retrievers__contextual_compression = __importStar(require("../retrievers/contextual_compression.cjs"));
 exports.retrievers__document_compressors = __importStar(require("../retrievers/document_compressors/index.cjs"));
+exports.retrievers__multi_query = __importStar(require("../retrievers/multi_query.cjs"));
 exports.retrievers__multi_vector = __importStar(require("../retrievers/multi_vector.cjs"));
 exports.retrievers__parent_document = __importStar(require("../retrievers/parent_document.cjs"));
 exports.retrievers__time_weighted = __importStar(require("../retrievers/time_weighted.cjs"));

package/dist/load/import_map.d.ts

@@ -48,6 +48,7 @@ export * as retrievers__remote from "../retrievers/remote/index.js";
 export * as retrievers__databerry from "../retrievers/databerry.js";
 export * as retrievers__contextual_compression from "../retrievers/contextual_compression.js";
 export * as retrievers__document_compressors from "../retrievers/document_compressors/index.js";
+export * as retrievers__multi_query from "../retrievers/multi_query.js";
 export * as retrievers__multi_vector from "../retrievers/multi_vector.js";
 export * as retrievers__parent_document from "../retrievers/parent_document.js";
 export * as retrievers__time_weighted from "../retrievers/time_weighted.js";

package/dist/load/import_map.js

@@ -49,6 +49,7 @@ export * as retrievers__remote from "../retrievers/remote/index.js";
 export * as retrievers__databerry from "../retrievers/databerry.js";
 export * as retrievers__contextual_compression from "../retrievers/contextual_compression.js";
 export * as retrievers__document_compressors from "../retrievers/document_compressors/index.js";
+export * as retrievers__multi_query from "../retrievers/multi_query.js";
 export * as retrievers__multi_vector from "../retrievers/multi_vector.js";
 export * as retrievers__parent_document from "../retrievers/parent_document.js";
 export * as retrievers__time_weighted from "../retrievers/time_weighted.js";

package/dist/load/serializable.cjs

@@ -138,11 +138,30 @@ class Serializable {
         }
         // include all secrets used, even if not in kwargs,
         // will be replaced with sentinel value in replaceSecrets
-        for (const key in secrets) {
-            if (key in this && this[key] !== undefined) {
-                kwargs[key] = this[key] || kwargs[key];
+        Object.keys(secrets).forEach((keyPath) => {
+            // eslint-disable-next-line @typescript-eslint/no-this-alias, @typescript-eslint/no-explicit-any
+            let read = this;
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            let write = kwargs;
+            const [last, ...partsReverse] = keyPath.split(".").reverse();
+            for (const key of partsReverse.reverse()) {
+                if (!(key in read) || read[key] === undefined)
+                    return;
+                if (!(key in write) || write[key] === undefined) {
+                    if (typeof read[key] === "object" && read[key] != null) {
+                        write[key] = {};
+                    }
+                    else if (Array.isArray(read[key])) {
+                        write[key] = [];
+                    }
+                }
+                read = read[key];
+                write = write[key];
             }
-        }
+            if (last in read && read[last] !== undefined) {
+                write[last] = write[last] || read[last];
+            }
+        });
         return {
             lc: 1,
             type: "constructor",

package/dist/load/serializable.js

@@ -134,11 +134,30 @@ export class Serializable {
         }
         // include all secrets used, even if not in kwargs,
         // will be replaced with sentinel value in replaceSecrets
-        for (const key in secrets) {
-            if (key in this && this[key] !== undefined) {
-                kwargs[key] = this[key] || kwargs[key];
+        Object.keys(secrets).forEach((keyPath) => {
+            // eslint-disable-next-line @typescript-eslint/no-this-alias, @typescript-eslint/no-explicit-any
+            let read = this;
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            let write = kwargs;
+            const [last, ...partsReverse] = keyPath.split(".").reverse();
+            for (const key of partsReverse.reverse()) {
+                if (!(key in read) || read[key] === undefined)
+                    return;
+                if (!(key in write) || write[key] === undefined) {
+                    if (typeof read[key] === "object" && read[key] != null) {
+                        write[key] = {};
+                    }
+                    else if (Array.isArray(read[key])) {
+                        write[key] = [];
+                    }
+                }
+                read = read[key];
+                write = write[key];
             }
-        }
+            if (last in read && read[last] !== undefined) {
+                write[last] = write[last] || read[last];
+            }
+        });
         return {
             lc: 1,
             type: "constructor",

package/dist/retrievers/multi_query.cjs

@@ -0,0 +1,140 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MultiQueryRetriever = void 0;
+const llm_chain_js_1 = require("../chains/llm_chain.cjs");
+const prompt_js_1 = require("../prompts/prompt.cjs");
+const output_parser_js_1 = require("../schema/output_parser.cjs");
+const retriever_js_1 = require("../schema/retriever.cjs");
+class LineListOutputParser extends output_parser_js_1.BaseOutputParser {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "retrievers", "multiquery"]
+        });
+    }
+    static lc_name() {
+        return "LineListOutputParser";
+    }
+    async parse(text) {
+        const startKeyIndex = text.indexOf("<questions>");
+        const endKeyIndex = text.indexOf("</questions>");
+        const questionsStartIndex = startKeyIndex === -1 ? 0 : startKeyIndex + "<questions>".length;
+        const questionsEndIndex = endKeyIndex === -1 ? text.length : endKeyIndex;
+        const lines = text
+            .slice(questionsStartIndex, questionsEndIndex)
+            .trim()
+            .split("\n")
+            .filter((line) => line.trim() !== "");
+        return { lines };
+    }
+    getFormatInstructions() {
+        throw new Error("Not implemented.");
+    }
+}
+// Create template
+const DEFAULT_QUERY_PROMPT = /* #__PURE__ */ new prompt_js_1.PromptTemplate({
+    inputVariables: ["question", "queryCount"],
+    template: `You are an AI language model assistant. Your task is
+to generate {queryCount} different versions of the given user
+question to retrieve relevant documents from a vector database.
+By generating multiple perspectives on the user question,
+your goal is to help the user overcome some of the limitations
+of distance-based similarity search.
+
+Provide these alternative questions separated by newlines between XML tags. For example:
+
+<questions>
+Question 1
+Question 2
+Question 3
+</questions>
+
+Original question: {question}`,
+});
+// Export class
+class MultiQueryRetriever extends retriever_js_1.BaseRetriever {
+    static lc_name() {
+        return "MultiQueryRetriever";
+    }
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "retrievers", "multiquery"]
+        });
+        Object.defineProperty(this, "retriever", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "llmChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "queryCount", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 3
+        });
+        Object.defineProperty(this, "parserKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "lines"
+        });
+        this.retriever = fields.retriever;
+        this.llmChain = fields.llmChain;
+        this.queryCount = fields.queryCount ?? this.queryCount;
+        this.parserKey = fields.parserKey ?? this.parserKey;
+    }
+    static fromLLM(fields) {
+        const { retriever, llm, prompt = DEFAULT_QUERY_PROMPT, queryCount, parserKey, ...rest } = fields;
+        const outputParser = new LineListOutputParser();
+        const llmChain = new llm_chain_js_1.LLMChain({ llm, prompt, outputParser });
+        return new this({ retriever, llmChain, queryCount, parserKey, ...rest });
+    }
+    // Generate the different queries for each retrieval, using our llmChain
+    async _generateQueries(question, runManager) {
+        const response = await this.llmChain.call({ question, queryCount: this.queryCount }, runManager?.getChild());
+        const lines = response.text[this.parserKey] || [];
+        if (this.verbose) {
+            console.log(`Generated queries: ${lines}`);
+        }
+        return lines;
+    }
+    // Retrieve documents using the original retriever
+    async _retrieveDocuments(queries, runManager) {
+        const documents = [];
+        for (const query of queries) {
+            const docs = await this.retriever.getRelevantDocuments(query, runManager?.getChild());
+            documents.push(...docs);
+        }
+        return documents;
+    }
+    // Deduplicate the documents that were returned in multiple retrievals
+    _uniqueUnion(documents) {
+        const uniqueDocumentsDict = {};
+        for (const doc of documents) {
+            const key = `${doc.pageContent}:${JSON.stringify(Object.entries(doc.metadata).sort())}`;
+            uniqueDocumentsDict[key] = doc;
+        }
+        const uniqueDocuments = Object.values(uniqueDocumentsDict);
+        return uniqueDocuments;
+    }
+    async _getRelevantDocuments(question, runManager) {
+        const queries = await this._generateQueries(question, runManager);
+        const documents = await this._retrieveDocuments(queries, runManager);
+        const uniqueDocuments = this._uniqueUnion(documents);
+        return uniqueDocuments;
+    }
+}
+exports.MultiQueryRetriever = MultiQueryRetriever;

package/dist/retrievers/multi_query.d.ts

@@ -0,0 +1,33 @@
+import { LLMChain } from "../chains/llm_chain.js";
+import { Document } from "../document.js";
+import { BaseRetriever, BaseRetrieverInput } from "../schema/retriever.js";
+import { CallbackManagerForRetrieverRun } from "../callbacks/index.js";
+import { BaseLanguageModel } from "../base_language/index.js";
+import { BasePromptTemplate } from "../prompts/base.js";
+interface LineList {
+    lines: string[];
+}
+export interface MultiQueryRetrieverInput extends BaseRetrieverInput {
+    retriever: BaseRetriever;
+    llmChain: LLMChain<LineList>;
+    queryCount?: number;
+    parserKey?: string;
+}
+export declare class MultiQueryRetriever extends BaseRetriever {
+    static lc_name(): string;
+    lc_namespace: string[];
+    private retriever;
+    private llmChain;
+    private queryCount;
+    private parserKey;
+    constructor(fields: MultiQueryRetrieverInput);
+    static fromLLM(fields: Omit<MultiQueryRetrieverInput, "llmChain"> & {
+        llm: BaseLanguageModel;
+        prompt?: BasePromptTemplate;
+    }): MultiQueryRetriever;
+    private _generateQueries;
+    private _retrieveDocuments;
+    private _uniqueUnion;
+    _getRelevantDocuments(question: string, runManager?: CallbackManagerForRetrieverRun): Promise<Document[]>;
+}
+export {};