langchain 1.0.0-alpha.7 → 1.0.0-alpha.9

package/dist/retrievers/multi_query.cjs

@@ -1,150 +0,0 @@
-const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
-const require_llm_chain = require('../chains/llm_chain.cjs');
-const __langchain_core_prompts = require_rolldown_runtime.__toESM(require("@langchain/core/prompts"));
-const __langchain_core_output_parsers = require_rolldown_runtime.__toESM(require("@langchain/core/output_parsers"));
-const __langchain_core_retrievers = require_rolldown_runtime.__toESM(require("@langchain/core/retrievers"));
-
-//#region src/retrievers/multi_query.ts
-var multi_query_exports = {};
-require_rolldown_runtime.__export(multi_query_exports, { MultiQueryRetriever: () => MultiQueryRetriever });
-var LineListOutputParser = class extends __langchain_core_output_parsers.BaseOutputParser {
-	static lc_name() {
-		return "LineListOutputParser";
-	}
-	lc_namespace = [
-		"langchain",
-		"retrievers",
-		"multiquery"
-	];
-	async parse(text) {
-		const startKeyIndex = text.indexOf("<questions>");
-		const endKeyIndex = text.indexOf("</questions>");
-		const questionsStartIndex = startKeyIndex === -1 ? 0 : startKeyIndex + 11;
-		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.");
-	}
-};
-const DEFAULT_QUERY_PROMPT = /* @__PURE__ */ new __langchain_core_prompts.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}`
-});
-/**
-* @example
-* ```typescript
-* const retriever = new MultiQueryRetriever.fromLLM({
-*   llm: new ChatAnthropic({}),
-*   retriever: new MemoryVectorStore().asRetriever(),
-*   verbose: true,
-* });
-* const retrievedDocs = await retriever.getRelevantDocuments(
-*   "What are mitochondria made of?",
-* );
-* ```
-*/
-var MultiQueryRetriever = class extends __langchain_core_retrievers.BaseRetriever {
-	static lc_name() {
-		return "MultiQueryRetriever";
-	}
-	lc_namespace = [
-		"langchain",
-		"retrievers",
-		"multiquery"
-	];
-	retriever;
-	llmChain;
-	queryCount = 3;
-	parserKey = "lines";
-	documentCompressor;
-	documentCompressorFilteringFn;
-	constructor(fields) {
-		super(fields);
-		this.retriever = fields.retriever;
-		this.llmChain = fields.llmChain;
-		this.queryCount = fields.queryCount ?? this.queryCount;
-		this.parserKey = fields.parserKey ?? this.parserKey;
-		this.documentCompressor = fields.documentCompressor;
-		this.documentCompressorFilteringFn = fields.documentCompressorFilteringFn;
-	}
-	static fromLLM(fields) {
-		const { retriever, llm, prompt = DEFAULT_QUERY_PROMPT, queryCount, parserKey,...rest } = fields;
-		const outputParser = new LineListOutputParser();
-		const llmChain = new require_llm_chain.LLMChain({
-			llm,
-			prompt,
-			outputParser
-		});
-		return new this({
-			retriever,
-			llmChain,
-			queryCount,
-			parserKey,
-			...rest
-		});
-	}
-	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;
-	}
-	async _retrieveDocuments(queries, runManager) {
-		const documents = [];
-		await Promise.all(queries.map(async (query) => {
-			const docs = await this.retriever.getRelevantDocuments(query, runManager?.getChild());
-			documents.push(...docs);
-		}));
-		return documents;
-	}
-	_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);
-		let outputDocs = uniqueDocuments;
-		if (this.documentCompressor && uniqueDocuments.length) {
-			outputDocs = await this.documentCompressor.compressDocuments(uniqueDocuments, question, runManager?.getChild());
-			if (this.documentCompressorFilteringFn) outputDocs = this.documentCompressorFilteringFn(outputDocs);
-		}
-		return outputDocs;
-	}
-};
-
-//#endregion
-exports.MultiQueryRetriever = MultiQueryRetriever;
-Object.defineProperty(exports, 'multi_query_exports', {
-  enumerable: true,
-  get: function () {
-    return multi_query_exports;
-  }
-});
-//# sourceMappingURL=multi_query.cjs.map

package/dist/retrievers/multi_query.cjs.map

@@ -1 +0,0 @@
-{"version":3,"file":"multi_query.cjs","names":["BaseOutputParser","text: string","PromptTemplate","BaseRetriever","fields: MultiQueryRetrieverInput","fields: Omit<MultiQueryRetrieverInput, \"llmChain\"> & {\n      llm: BaseLanguageModelInterface;\n      prompt?: BasePromptTemplate;\n    }","LLMChain","question: string","runManager?: CallbackManagerForRetrieverRun","queries: string[]","documents: Document[]","uniqueDocumentsDict: { [key: string]: Document }"],"sources":["../../src/retrievers/multi_query.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport {\n  BaseRetriever,\n  type BaseRetrieverInput,\n  type BaseRetrieverInterface,\n} from \"@langchain/core/retrievers\";\nimport { Document } from \"@langchain/core/documents\";\nimport { BaseOutputParser } from \"@langchain/core/output_parsers\";\nimport { PromptTemplate, BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForRetrieverRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../chains/llm_chain.js\";\nimport type { BaseDocumentCompressor } from \"./document_compressors/index.js\";\n\ninterface LineList {\n  lines: string[];\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type MultiDocs = Document<Record<string, any>>[];\n\nclass LineListOutputParser extends BaseOutputParser<LineList> {\n  static lc_name() {\n    return \"LineListOutputParser\";\n  }\n\n  lc_namespace = [\"langchain\", \"retrievers\", \"multiquery\"];\n\n  async parse(text: string): Promise<LineList> {\n    const startKeyIndex = text.indexOf(\"<questions>\");\n    const endKeyIndex = text.indexOf(\"</questions>\");\n    const questionsStartIndex =\n      startKeyIndex === -1 ? 0 : startKeyIndex + \"<questions>\".length;\n    const questionsEndIndex = endKeyIndex === -1 ? text.length : endKeyIndex;\n    const lines = text\n      .slice(questionsStartIndex, questionsEndIndex)\n      .trim()\n      .split(\"\\n\")\n      .filter((line) => line.trim() !== \"\");\n    return { lines };\n  }\n\n  getFormatInstructions(): string {\n    throw new Error(\"Not implemented.\");\n  }\n}\n\n// Create template\nconst DEFAULT_QUERY_PROMPT = /* #__PURE__ */ new PromptTemplate({\n  inputVariables: [\"question\", \"queryCount\"],\n  template: `You are an AI language model assistant. Your task is\nto generate {queryCount} different versions of the given user\nquestion to retrieve relevant documents from a vector database.\nBy generating multiple perspectives on the user question,\nyour goal is to help the user overcome some of the limitations\nof distance-based similarity search.\n\nProvide these alternative questions separated by newlines between XML tags. For example:\n\n<questions>\nQuestion 1\nQuestion 2\nQuestion 3\n</questions>\n\nOriginal question: {question}`,\n});\n\nexport interface MultiQueryRetrieverInput extends BaseRetrieverInput {\n  retriever: BaseRetrieverInterface;\n  /** @deprecated Pass a custom prompt into `.fromLLM` instead. */\n  llmChain: LLMChain<LineList>;\n  queryCount?: number;\n  parserKey?: string;\n  documentCompressor?: BaseDocumentCompressor | undefined;\n  documentCompressorFilteringFn?: (docs: MultiDocs) => MultiDocs;\n}\n\n/**\n * @example\n * ```typescript\n * const retriever = new MultiQueryRetriever.fromLLM({\n *   llm: new ChatAnthropic({}),\n *   retriever: new MemoryVectorStore().asRetriever(),\n *   verbose: true,\n * });\n * const retrievedDocs = await retriever.getRelevantDocuments(\n *   \"What are mitochondria made of?\",\n * );\n * ```\n */\nexport class MultiQueryRetriever extends BaseRetriever {\n  static lc_name() {\n    return \"MultiQueryRetriever\";\n  }\n\n  lc_namespace = [\"langchain\", \"retrievers\", \"multiquery\"];\n\n  private retriever: BaseRetrieverInterface;\n\n  private llmChain: LLMChain<LineList>;\n\n  private queryCount = 3;\n\n  private parserKey = \"lines\";\n\n  documentCompressor: BaseDocumentCompressor | undefined;\n\n  documentCompressorFilteringFn?: MultiQueryRetrieverInput[\"documentCompressorFilteringFn\"];\n\n  constructor(fields: MultiQueryRetrieverInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.llmChain = fields.llmChain;\n    this.queryCount = fields.queryCount ?? this.queryCount;\n    this.parserKey = fields.parserKey ?? this.parserKey;\n    this.documentCompressor = fields.documentCompressor;\n    this.documentCompressorFilteringFn = fields.documentCompressorFilteringFn;\n  }\n\n  static fromLLM(\n    fields: Omit<MultiQueryRetrieverInput, \"llmChain\"> & {\n      llm: BaseLanguageModelInterface;\n      prompt?: BasePromptTemplate;\n    }\n  ): MultiQueryRetriever {\n    const {\n      retriever,\n      llm,\n      prompt = DEFAULT_QUERY_PROMPT,\n      queryCount,\n      parserKey,\n      ...rest\n    } = fields;\n    const outputParser = new LineListOutputParser();\n    const llmChain = new LLMChain({ llm, prompt, outputParser });\n    return new this({ retriever, llmChain, queryCount, parserKey, ...rest });\n  }\n\n  // Generate the different queries for each retrieval, using our llmChain\n  private async _generateQueries(\n    question: string,\n    runManager?: CallbackManagerForRetrieverRun\n  ): Promise<string[]> {\n    const response = await this.llmChain.call(\n      { question, queryCount: this.queryCount },\n      runManager?.getChild()\n    );\n    const lines = response.text[this.parserKey] || [];\n    if (this.verbose) {\n      console.log(`Generated queries: ${lines}`);\n    }\n    return lines;\n  }\n\n  // Retrieve documents using the original retriever\n  private async _retrieveDocuments(\n    queries: string[],\n    runManager?: CallbackManagerForRetrieverRun\n  ): Promise<Document[]> {\n    const documents: Document[] = [];\n    await Promise.all(\n      queries.map(async (query) => {\n        const docs = await this.retriever.getRelevantDocuments(\n          query,\n          runManager?.getChild()\n        );\n        documents.push(...docs);\n      })\n    );\n    return documents;\n  }\n\n  // Deduplicate the documents that were returned in multiple retrievals\n  private _uniqueUnion(documents: Document[]): Document[] {\n    const uniqueDocumentsDict: { [key: string]: Document } = {};\n\n    for (const doc of documents) {\n      const key = `${doc.pageContent}:${JSON.stringify(\n        Object.entries(doc.metadata).sort()\n      )}`;\n      uniqueDocumentsDict[key] = doc;\n    }\n\n    const uniqueDocuments = Object.values(uniqueDocumentsDict);\n    return uniqueDocuments;\n  }\n\n  async _getRelevantDocuments(\n    question: string,\n    runManager?: CallbackManagerForRetrieverRun\n  ): Promise<Document[]> {\n    const queries = await this._generateQueries(question, runManager);\n    const documents = await this._retrieveDocuments(queries, runManager);\n    const uniqueDocuments = this._uniqueUnion(documents);\n\n    let outputDocs = uniqueDocuments;\n    if (this.documentCompressor && uniqueDocuments.length) {\n      outputDocs = await this.documentCompressor.compressDocuments(\n        uniqueDocuments,\n        question,\n        runManager?.getChild()\n      );\n      if (this.documentCompressorFilteringFn) {\n        outputDocs = this.documentCompressorFilteringFn(outputDocs);\n      }\n    }\n\n    return outputDocs;\n  }\n}\n"],"mappings":";;;;;;;;;AAoBA,IAAM,uBAAN,cAAmCA,iDAA2B;CAC5D,OAAO,UAAU;AACf,SAAO;CACR;CAED,eAAe;EAAC;EAAa;EAAc;CAAa;CAExD,MAAM,MAAMC,MAAiC;EAC3C,MAAM,gBAAgB,KAAK,QAAQ,cAAc;EACjD,MAAM,cAAc,KAAK,QAAQ,eAAe;EAChD,MAAM,sBACJ,kBAAkB,KAAK,IAAI,gBAAgB;EAC7C,MAAM,oBAAoB,gBAAgB,KAAK,KAAK,SAAS;EAC7D,MAAM,QAAQ,KACX,MAAM,qBAAqB,kBAAkB,CAC7C,MAAM,CACN,MAAM,KAAK,CACX,OAAO,CAAC,SAAS,KAAK,MAAM,KAAK,GAAG;AACvC,SAAO,EAAE,MAAO;CACjB;CAED,wBAAgC;AAC9B,QAAM,IAAI,MAAM;CACjB;AACF;AAGD,MAAM,uCAAuC,IAAIC,wCAAe;CAC9D,gBAAgB,CAAC,YAAY,YAAa;CAC1C,UAAU,CAAC;;;;;;;;;;;;;;;6BAegB,CAAC;AAC7B;;;;;;;;;;;;;;AAyBD,IAAa,sBAAb,cAAyCC,0CAAc;CACrD,OAAO,UAAU;AACf,SAAO;CACR;CAED,eAAe;EAAC;EAAa;EAAc;CAAa;CAExD,AAAQ;CAER,AAAQ;CAER,AAAQ,aAAa;CAErB,AAAQ,YAAY;CAEpB;CAEA;CAEA,YAAYC,QAAkC;EAC5C,MAAM,OAAO;EACb,KAAK,YAAY,OAAO;EACxB,KAAK,WAAW,OAAO;EACvB,KAAK,aAAa,OAAO,cAAc,KAAK;EAC5C,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,qBAAqB,OAAO;EACjC,KAAK,gCAAgC,OAAO;CAC7C;CAED,OAAO,QACLC,QAIqB;EACrB,MAAM,EACJ,WACA,KACA,SAAS,sBACT,YACA,UACA,GAAG,MACJ,GAAG;EACJ,MAAM,eAAe,IAAI;EACzB,MAAM,WAAW,IAAIC,2BAAS;GAAE;GAAK;GAAQ;EAAc;AAC3D,SAAO,IAAI,KAAK;GAAE;GAAW;GAAU;GAAY;GAAW,GAAG;EAAM;CACxE;CAGD,MAAc,iBACZC,UACAC,YACmB;EACnB,MAAM,WAAW,MAAM,KAAK,SAAS,KACnC;GAAE;GAAU,YAAY,KAAK;EAAY,GACzC,YAAY,UAAU,CACvB;EACD,MAAM,QAAQ,SAAS,KAAK,KAAK,cAAc,CAAE;AACjD,MAAI,KAAK,SACP,QAAQ,IAAI,CAAC,mBAAmB,EAAE,OAAO,CAAC;AAE5C,SAAO;CACR;CAGD,MAAc,mBACZC,SACAD,YACqB;EACrB,MAAME,YAAwB,CAAE;EAChC,MAAM,QAAQ,IACZ,QAAQ,IAAI,OAAO,UAAU;GAC3B,MAAM,OAAO,MAAM,KAAK,UAAU,qBAChC,OACA,YAAY,UAAU,CACvB;GACD,UAAU,KAAK,GAAG,KAAK;EACxB,EAAC,CACH;AACD,SAAO;CACR;CAGD,AAAQ,aAAaA,WAAmC;EACtD,MAAMC,sBAAmD,CAAE;AAE3D,OAAK,MAAM,OAAO,WAAW;GAC3B,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,KAAK,UACrC,OAAO,QAAQ,IAAI,SAAS,CAAC,MAAM,CACpC,EAAE;GACH,oBAAoB,OAAO;EAC5B;EAED,MAAM,kBAAkB,OAAO,OAAO,oBAAoB;AAC1D,SAAO;CACR;CAED,MAAM,sBACJJ,UACAC,YACqB;EACrB,MAAM,UAAU,MAAM,KAAK,iBAAiB,UAAU,WAAW;EACjE,MAAM,YAAY,MAAM,KAAK,mBAAmB,SAAS,WAAW;EACpE,MAAM,kBAAkB,KAAK,aAAa,UAAU;EAEpD,IAAI,aAAa;AACjB,MAAI,KAAK,sBAAsB,gBAAgB,QAAQ;GACrD,aAAa,MAAM,KAAK,mBAAmB,kBACzC,iBACA,UACA,YAAY,UAAU,CACvB;AACD,OAAI,KAAK,+BACP,aAAa,KAAK,8BAA8B,WAAW;EAE9D;AAED,SAAO;CACR;AACF"}

package/dist/retrievers/multi_query.d.cts

@@ -1,59 +0,0 @@
-import { LLMChain } from "../chains/llm_chain.cjs";
-import { BaseDocumentCompressor } from "./document_compressors/index.cjs";
-import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
-import { BasePromptTemplate } from "@langchain/core/prompts";
-import { Document } from "@langchain/core/documents";
-import { CallbackManagerForRetrieverRun } from "@langchain/core/callbacks/manager";
-import { BaseRetriever, BaseRetrieverInput, BaseRetrieverInterface } from "@langchain/core/retrievers";
-
-//#region src/retrievers/multi_query.d.ts
-interface LineList {
-  lines: string[];
-}
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-type MultiDocs = Document<Record<string, any>>[];
-interface MultiQueryRetrieverInput extends BaseRetrieverInput {
-  retriever: BaseRetrieverInterface;
-  /** @deprecated Pass a custom prompt into `.fromLLM` instead. */
-  llmChain: LLMChain<LineList>;
-  queryCount?: number;
-  parserKey?: string;
-  documentCompressor?: BaseDocumentCompressor | undefined;
-  documentCompressorFilteringFn?: (docs: MultiDocs) => MultiDocs;
-}
-/**
- * @example
- * ```typescript
- * const retriever = new MultiQueryRetriever.fromLLM({
- *   llm: new ChatAnthropic({}),
- *   retriever: new MemoryVectorStore().asRetriever(),
- *   verbose: true,
- * });
- * const retrievedDocs = await retriever.getRelevantDocuments(
- *   "What are mitochondria made of?",
- * );
- * ```
- */
-declare class MultiQueryRetriever extends BaseRetriever {
-  static lc_name(): string;
-  lc_namespace: string[];
-  private retriever;
-  private llmChain;
-  private queryCount;
-  private parserKey;
-  documentCompressor: BaseDocumentCompressor | undefined;
-  documentCompressorFilteringFn?: MultiQueryRetrieverInput["documentCompressorFilteringFn"];
-  constructor(fields: MultiQueryRetrieverInput);
-  static fromLLM(fields: Omit<MultiQueryRetrieverInput, "llmChain"> & {
-    llm: BaseLanguageModelInterface;
-    prompt?: BasePromptTemplate;
-  }): MultiQueryRetriever;
-  private _generateQueries;
-  private _retrieveDocuments;
-  // Deduplicate the documents that were returned in multiple retrievals
-  private _uniqueUnion;
-  _getRelevantDocuments(question: string, runManager?: CallbackManagerForRetrieverRun): Promise<Document[]>;
-}
-//#endregion
-export { MultiDocs, MultiQueryRetriever, MultiQueryRetrieverInput };
-//# sourceMappingURL=multi_query.d.cts.map

package/dist/retrievers/multi_query.d.cts.map

@@ -1 +0,0 @@
-{"version":3,"file":"multi_query.d.cts","names":["BaseLanguageModelInterface","BaseRetriever","BaseRetrieverInput","BaseRetrieverInterface","Document","BasePromptTemplate","CallbackManagerForRetrieverRun","LLMChain","BaseDocumentCompressor","LineList","MultiDocs","Record","MultiQueryRetrieverInput","MultiQueryRetriever","Omit","Promise"],"sources":["../../src/retrievers/multi_query.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { BaseRetriever, type BaseRetrieverInput, type BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { Document } from \"@langchain/core/documents\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForRetrieverRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../chains/llm_chain.js\";\nimport type { BaseDocumentCompressor } from \"./document_compressors/index.js\";\ninterface LineList {\n    lines: string[];\n}\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type MultiDocs = Document<Record<string, any>>[];\nexport interface MultiQueryRetrieverInput extends BaseRetrieverInput {\n    retriever: BaseRetrieverInterface;\n    /** @deprecated Pass a custom prompt into `.fromLLM` instead. */\n    llmChain: LLMChain<LineList>;\n    queryCount?: number;\n    parserKey?: string;\n    documentCompressor?: BaseDocumentCompressor | undefined;\n    documentCompressorFilteringFn?: (docs: MultiDocs) => MultiDocs;\n}\n/**\n * @example\n * ```typescript\n * const retriever = new MultiQueryRetriever.fromLLM({\n *   llm: new ChatAnthropic({}),\n *   retriever: new MemoryVectorStore().asRetriever(),\n *   verbose: true,\n * });\n * const retrievedDocs = await retriever.getRelevantDocuments(\n *   \"What are mitochondria made of?\",\n * );\n * ```\n */\nexport declare class MultiQueryRetriever extends BaseRetriever {\n    static lc_name(): string;\n    lc_namespace: string[];\n    private retriever;\n    private llmChain;\n    private queryCount;\n    private parserKey;\n    documentCompressor: BaseDocumentCompressor | undefined;\n    documentCompressorFilteringFn?: MultiQueryRetrieverInput[\"documentCompressorFilteringFn\"];\n    constructor(fields: MultiQueryRetrieverInput);\n    static fromLLM(fields: Omit<MultiQueryRetrieverInput, \"llmChain\"> & {\n        llm: BaseLanguageModelInterface;\n        prompt?: BasePromptTemplate;\n    }): MultiQueryRetriever;\n    private _generateQueries;\n    private _retrieveDocuments;\n    // Deduplicate the documents that were returned in multiple retrievals\n    private _uniqueUnion;\n    _getRelevantDocuments(question: string, runManager?: CallbackManagerForRetrieverRun): Promise<Document[]>;\n}\nexport {};\n"],"mappings":";;;;;;;;;UAOUS,QAAAA;;AADoE;AAK9E;AAAqB,KAATC,SAAAA,GAAYN,QAAH,CAAYO,MAAZ,CAAA,MAAA,EAAA,GAAA,CAAA,CAAA,EAAA;AAAYA,UAChBC,wBAAAA,SAAiCV,kBADjBS,CAAAA;EAAM,SAAfP,EAETD,sBAFSC;EAAQ;EACfQ,QAAAA,EAGHL,QAHGK,CAGMH,QAHkB,CAAA;EAAA,UAAA,CAAA,EAAA,MAAA;EAAA,SAC1BN,CAAAA,EAAAA,MAAAA;EAAsB,kBAEdM,CAAAA,EAGED,sBAHFC,GAAAA,SAAAA;EAAQ,6BAAjBF,CAAAA,EAAAA,CAAAA,IAAAA,EAI6BG,SAJ7BH,EAAAA,GAI2CG,SAJ3CH;;;;;AAHsD;AAsBpE;;;;;;;;;AAaQM,cAbaA,mBAAAA,SAA4BZ,aAAAA,CAazCY;EAAmB,OAK8BP,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAA8B,YAAWF,EAAAA,MAAAA,EAAAA;EAAQ,QAAhBW,SAAAA;EAAO,QAlBhDd,QAAAA;EAAa,QAAA,UAAA;;sBAOtCO;kCACYI;sBACZA;yBACGE,KAAKF;SACnBZ;aACIK;MACTQ;;;;;uDAKiDP,iCAAiCS,QAAQX"}

package/dist/retrievers/multi_query.d.ts

@@ -1,59 +0,0 @@
-import { LLMChain } from "../chains/llm_chain.js";
-import { BaseDocumentCompressor } from "./document_compressors/index.js";
-import { Document } from "@langchain/core/documents";
-import { BasePromptTemplate } from "@langchain/core/prompts";
-import { CallbackManagerForRetrieverRun } from "@langchain/core/callbacks/manager";
-import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
-import { BaseRetriever, BaseRetrieverInput, BaseRetrieverInterface } from "@langchain/core/retrievers";
-
-//#region src/retrievers/multi_query.d.ts
-interface LineList {
-  lines: string[];
-}
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-type MultiDocs = Document<Record<string, any>>[];
-interface MultiQueryRetrieverInput extends BaseRetrieverInput {
-  retriever: BaseRetrieverInterface;
-  /** @deprecated Pass a custom prompt into `.fromLLM` instead. */
-  llmChain: LLMChain<LineList>;
-  queryCount?: number;
-  parserKey?: string;
-  documentCompressor?: BaseDocumentCompressor | undefined;
-  documentCompressorFilteringFn?: (docs: MultiDocs) => MultiDocs;
-}
-/**
- * @example
- * ```typescript
- * const retriever = new MultiQueryRetriever.fromLLM({
- *   llm: new ChatAnthropic({}),
- *   retriever: new MemoryVectorStore().asRetriever(),
- *   verbose: true,
- * });
- * const retrievedDocs = await retriever.getRelevantDocuments(
- *   "What are mitochondria made of?",
- * );
- * ```
- */
-declare class MultiQueryRetriever extends BaseRetriever {
-  static lc_name(): string;
-  lc_namespace: string[];
-  private retriever;
-  private llmChain;
-  private queryCount;
-  private parserKey;
-  documentCompressor: BaseDocumentCompressor | undefined;
-  documentCompressorFilteringFn?: MultiQueryRetrieverInput["documentCompressorFilteringFn"];
-  constructor(fields: MultiQueryRetrieverInput);
-  static fromLLM(fields: Omit<MultiQueryRetrieverInput, "llmChain"> & {
-    llm: BaseLanguageModelInterface;
-    prompt?: BasePromptTemplate;
-  }): MultiQueryRetriever;
-  private _generateQueries;
-  private _retrieveDocuments;
-  // Deduplicate the documents that were returned in multiple retrievals
-  private _uniqueUnion;
-  _getRelevantDocuments(question: string, runManager?: CallbackManagerForRetrieverRun): Promise<Document[]>;
-}
-//#endregion
-export { MultiDocs, MultiQueryRetriever, MultiQueryRetrieverInput };
-//# sourceMappingURL=multi_query.d.ts.map

package/dist/retrievers/multi_query.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"multi_query.d.ts","names":["BaseLanguageModelInterface","BaseRetriever","BaseRetrieverInput","BaseRetrieverInterface","Document","BasePromptTemplate","CallbackManagerForRetrieverRun","LLMChain","BaseDocumentCompressor","LineList","MultiDocs","Record","MultiQueryRetrieverInput","MultiQueryRetriever","Omit","Promise"],"sources":["../../src/retrievers/multi_query.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { BaseRetriever, type BaseRetrieverInput, type BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { Document } from \"@langchain/core/documents\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForRetrieverRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../chains/llm_chain.js\";\nimport type { BaseDocumentCompressor } from \"./document_compressors/index.js\";\ninterface LineList {\n    lines: string[];\n}\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type MultiDocs = Document<Record<string, any>>[];\nexport interface MultiQueryRetrieverInput extends BaseRetrieverInput {\n    retriever: BaseRetrieverInterface;\n    /** @deprecated Pass a custom prompt into `.fromLLM` instead. */\n    llmChain: LLMChain<LineList>;\n    queryCount?: number;\n    parserKey?: string;\n    documentCompressor?: BaseDocumentCompressor | undefined;\n    documentCompressorFilteringFn?: (docs: MultiDocs) => MultiDocs;\n}\n/**\n * @example\n * ```typescript\n * const retriever = new MultiQueryRetriever.fromLLM({\n *   llm: new ChatAnthropic({}),\n *   retriever: new MemoryVectorStore().asRetriever(),\n *   verbose: true,\n * });\n * const retrievedDocs = await retriever.getRelevantDocuments(\n *   \"What are mitochondria made of?\",\n * );\n * ```\n */\nexport declare class MultiQueryRetriever extends BaseRetriever {\n    static lc_name(): string;\n    lc_namespace: string[];\n    private retriever;\n    private llmChain;\n    private queryCount;\n    private parserKey;\n    documentCompressor: BaseDocumentCompressor | undefined;\n    documentCompressorFilteringFn?: MultiQueryRetrieverInput[\"documentCompressorFilteringFn\"];\n    constructor(fields: MultiQueryRetrieverInput);\n    static fromLLM(fields: Omit<MultiQueryRetrieverInput, \"llmChain\"> & {\n        llm: BaseLanguageModelInterface;\n        prompt?: BasePromptTemplate;\n    }): MultiQueryRetriever;\n    private _generateQueries;\n    private _retrieveDocuments;\n    // Deduplicate the documents that were returned in multiple retrievals\n    private _uniqueUnion;\n    _getRelevantDocuments(question: string, runManager?: CallbackManagerForRetrieverRun): Promise<Document[]>;\n}\nexport {};\n"],"mappings":";;;;;;;;;UAOUS,QAAAA;;AADoE;AAK9E;AAAqB,KAATC,SAAAA,GAAYN,QAAH,CAAYO,MAAZ,CAAA,MAAA,EAAA,GAAA,CAAA,CAAA,EAAA;AAAYA,UAChBC,wBAAAA,SAAiCV,kBADjBS,CAAAA;EAAM,SAAfP,EAETD,sBAFSC;EAAQ;EACfQ,QAAAA,EAGHL,QAHGK,CAGMH,QAHkB,CAAA;EAAA,UAAA,CAAA,EAAA,MAAA;EAAA,SAC1BN,CAAAA,EAAAA,MAAAA;EAAsB,kBAEdM,CAAAA,EAGED,sBAHFC,GAAAA,SAAAA;EAAQ,6BAAjBF,CAAAA,EAAAA,CAAAA,IAAAA,EAI6BG,SAJ7BH,EAAAA,GAI2CG,SAJ3CH;;;;;AAHsD;AAsBpE;;;;;;;;;AAaQM,cAbaA,mBAAAA,SAA4BZ,aAAAA,CAazCY;EAAmB,OAK8BP,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAA8B,YAAWF,EAAAA,MAAAA,EAAAA;EAAQ,QAAhBW,SAAAA;EAAO,QAlBhDd,QAAAA;EAAa,QAAA,UAAA;;sBAOtCO;kCACYI;sBACZA;yBACGE,KAAKF;SACnBZ;aACIK;MACTQ;;;;;uDAKiDP,iCAAiCS,QAAQX"}

package/dist/retrievers/multi_query.js

@@ -1,144 +0,0 @@
-import { __export } from "../_virtual/rolldown_runtime.js";
-import { LLMChain } from "../chains/llm_chain.js";
-import { PromptTemplate } from "@langchain/core/prompts";
-import { BaseOutputParser } from "@langchain/core/output_parsers";
-import { BaseRetriever } from "@langchain/core/retrievers";
-
-//#region src/retrievers/multi_query.ts
-var multi_query_exports = {};
-__export(multi_query_exports, { MultiQueryRetriever: () => MultiQueryRetriever });
-var LineListOutputParser = class extends BaseOutputParser {
-	static lc_name() {
-		return "LineListOutputParser";
-	}
-	lc_namespace = [
-		"langchain",
-		"retrievers",
-		"multiquery"
-	];
-	async parse(text) {
-		const startKeyIndex = text.indexOf("<questions>");
-		const endKeyIndex = text.indexOf("</questions>");
-		const questionsStartIndex = startKeyIndex === -1 ? 0 : startKeyIndex + 11;
-		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.");
-	}
-};
-const DEFAULT_QUERY_PROMPT = /* @__PURE__ */ new 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}`
-});
-/**
-* @example
-* ```typescript
-* const retriever = new MultiQueryRetriever.fromLLM({
-*   llm: new ChatAnthropic({}),
-*   retriever: new MemoryVectorStore().asRetriever(),
-*   verbose: true,
-* });
-* const retrievedDocs = await retriever.getRelevantDocuments(
-*   "What are mitochondria made of?",
-* );
-* ```
-*/
-var MultiQueryRetriever = class extends BaseRetriever {
-	static lc_name() {
-		return "MultiQueryRetriever";
-	}
-	lc_namespace = [
-		"langchain",
-		"retrievers",
-		"multiquery"
-	];
-	retriever;
-	llmChain;
-	queryCount = 3;
-	parserKey = "lines";
-	documentCompressor;
-	documentCompressorFilteringFn;
-	constructor(fields) {
-		super(fields);
-		this.retriever = fields.retriever;
-		this.llmChain = fields.llmChain;
-		this.queryCount = fields.queryCount ?? this.queryCount;
-		this.parserKey = fields.parserKey ?? this.parserKey;
-		this.documentCompressor = fields.documentCompressor;
-		this.documentCompressorFilteringFn = fields.documentCompressorFilteringFn;
-	}
-	static fromLLM(fields) {
-		const { retriever, llm, prompt = DEFAULT_QUERY_PROMPT, queryCount, parserKey,...rest } = fields;
-		const outputParser = new LineListOutputParser();
-		const llmChain = new LLMChain({
-			llm,
-			prompt,
-			outputParser
-		});
-		return new this({
-			retriever,
-			llmChain,
-			queryCount,
-			parserKey,
-			...rest
-		});
-	}
-	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;
-	}
-	async _retrieveDocuments(queries, runManager) {
-		const documents = [];
-		await Promise.all(queries.map(async (query) => {
-			const docs = await this.retriever.getRelevantDocuments(query, runManager?.getChild());
-			documents.push(...docs);
-		}));
-		return documents;
-	}
-	_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);
-		let outputDocs = uniqueDocuments;
-		if (this.documentCompressor && uniqueDocuments.length) {
-			outputDocs = await this.documentCompressor.compressDocuments(uniqueDocuments, question, runManager?.getChild());
-			if (this.documentCompressorFilteringFn) outputDocs = this.documentCompressorFilteringFn(outputDocs);
-		}
-		return outputDocs;
-	}
-};
-
-//#endregion
-export { MultiQueryRetriever, multi_query_exports };
-//# sourceMappingURL=multi_query.js.map

package/dist/retrievers/multi_query.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"multi_query.js","names":["text: string","fields: MultiQueryRetrieverInput","fields: Omit<MultiQueryRetrieverInput, \"llmChain\"> & {\n      llm: BaseLanguageModelInterface;\n      prompt?: BasePromptTemplate;\n    }","question: string","runManager?: CallbackManagerForRetrieverRun","queries: string[]","documents: Document[]","uniqueDocumentsDict: { [key: string]: Document }"],"sources":["../../src/retrievers/multi_query.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport {\n  BaseRetriever,\n  type BaseRetrieverInput,\n  type BaseRetrieverInterface,\n} from \"@langchain/core/retrievers\";\nimport { Document } from \"@langchain/core/documents\";\nimport { BaseOutputParser } from \"@langchain/core/output_parsers\";\nimport { PromptTemplate, BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForRetrieverRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../chains/llm_chain.js\";\nimport type { BaseDocumentCompressor } from \"./document_compressors/index.js\";\n\ninterface LineList {\n  lines: string[];\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type MultiDocs = Document<Record<string, any>>[];\n\nclass LineListOutputParser extends BaseOutputParser<LineList> {\n  static lc_name() {\n    return \"LineListOutputParser\";\n  }\n\n  lc_namespace = [\"langchain\", \"retrievers\", \"multiquery\"];\n\n  async parse(text: string): Promise<LineList> {\n    const startKeyIndex = text.indexOf(\"<questions>\");\n    const endKeyIndex = text.indexOf(\"</questions>\");\n    const questionsStartIndex =\n      startKeyIndex === -1 ? 0 : startKeyIndex + \"<questions>\".length;\n    const questionsEndIndex = endKeyIndex === -1 ? text.length : endKeyIndex;\n    const lines = text\n      .slice(questionsStartIndex, questionsEndIndex)\n      .trim()\n      .split(\"\\n\")\n      .filter((line) => line.trim() !== \"\");\n    return { lines };\n  }\n\n  getFormatInstructions(): string {\n    throw new Error(\"Not implemented.\");\n  }\n}\n\n// Create template\nconst DEFAULT_QUERY_PROMPT = /* #__PURE__ */ new PromptTemplate({\n  inputVariables: [\"question\", \"queryCount\"],\n  template: `You are an AI language model assistant. Your task is\nto generate {queryCount} different versions of the given user\nquestion to retrieve relevant documents from a vector database.\nBy generating multiple perspectives on the user question,\nyour goal is to help the user overcome some of the limitations\nof distance-based similarity search.\n\nProvide these alternative questions separated by newlines between XML tags. For example:\n\n<questions>\nQuestion 1\nQuestion 2\nQuestion 3\n</questions>\n\nOriginal question: {question}`,\n});\n\nexport interface MultiQueryRetrieverInput extends BaseRetrieverInput {\n  retriever: BaseRetrieverInterface;\n  /** @deprecated Pass a custom prompt into `.fromLLM` instead. */\n  llmChain: LLMChain<LineList>;\n  queryCount?: number;\n  parserKey?: string;\n  documentCompressor?: BaseDocumentCompressor | undefined;\n  documentCompressorFilteringFn?: (docs: MultiDocs) => MultiDocs;\n}\n\n/**\n * @example\n * ```typescript\n * const retriever = new MultiQueryRetriever.fromLLM({\n *   llm: new ChatAnthropic({}),\n *   retriever: new MemoryVectorStore().asRetriever(),\n *   verbose: true,\n * });\n * const retrievedDocs = await retriever.getRelevantDocuments(\n *   \"What are mitochondria made of?\",\n * );\n * ```\n */\nexport class MultiQueryRetriever extends BaseRetriever {\n  static lc_name() {\n    return \"MultiQueryRetriever\";\n  }\n\n  lc_namespace = [\"langchain\", \"retrievers\", \"multiquery\"];\n\n  private retriever: BaseRetrieverInterface;\n\n  private llmChain: LLMChain<LineList>;\n\n  private queryCount = 3;\n\n  private parserKey = \"lines\";\n\n  documentCompressor: BaseDocumentCompressor | undefined;\n\n  documentCompressorFilteringFn?: MultiQueryRetrieverInput[\"documentCompressorFilteringFn\"];\n\n  constructor(fields: MultiQueryRetrieverInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.llmChain = fields.llmChain;\n    this.queryCount = fields.queryCount ?? this.queryCount;\n    this.parserKey = fields.parserKey ?? this.parserKey;\n    this.documentCompressor = fields.documentCompressor;\n    this.documentCompressorFilteringFn = fields.documentCompressorFilteringFn;\n  }\n\n  static fromLLM(\n    fields: Omit<MultiQueryRetrieverInput, \"llmChain\"> & {\n      llm: BaseLanguageModelInterface;\n      prompt?: BasePromptTemplate;\n    }\n  ): MultiQueryRetriever {\n    const {\n      retriever,\n      llm,\n      prompt = DEFAULT_QUERY_PROMPT,\n      queryCount,\n      parserKey,\n      ...rest\n    } = fields;\n    const outputParser = new LineListOutputParser();\n    const llmChain = new LLMChain({ llm, prompt, outputParser });\n    return new this({ retriever, llmChain, queryCount, parserKey, ...rest });\n  }\n\n  // Generate the different queries for each retrieval, using our llmChain\n  private async _generateQueries(\n    question: string,\n    runManager?: CallbackManagerForRetrieverRun\n  ): Promise<string[]> {\n    const response = await this.llmChain.call(\n      { question, queryCount: this.queryCount },\n      runManager?.getChild()\n    );\n    const lines = response.text[this.parserKey] || [];\n    if (this.verbose) {\n      console.log(`Generated queries: ${lines}`);\n    }\n    return lines;\n  }\n\n  // Retrieve documents using the original retriever\n  private async _retrieveDocuments(\n    queries: string[],\n    runManager?: CallbackManagerForRetrieverRun\n  ): Promise<Document[]> {\n    const documents: Document[] = [];\n    await Promise.all(\n      queries.map(async (query) => {\n        const docs = await this.retriever.getRelevantDocuments(\n          query,\n          runManager?.getChild()\n        );\n        documents.push(...docs);\n      })\n    );\n    return documents;\n  }\n\n  // Deduplicate the documents that were returned in multiple retrievals\n  private _uniqueUnion(documents: Document[]): Document[] {\n    const uniqueDocumentsDict: { [key: string]: Document } = {};\n\n    for (const doc of documents) {\n      const key = `${doc.pageContent}:${JSON.stringify(\n        Object.entries(doc.metadata).sort()\n      )}`;\n      uniqueDocumentsDict[key] = doc;\n    }\n\n    const uniqueDocuments = Object.values(uniqueDocumentsDict);\n    return uniqueDocuments;\n  }\n\n  async _getRelevantDocuments(\n    question: string,\n    runManager?: CallbackManagerForRetrieverRun\n  ): Promise<Document[]> {\n    const queries = await this._generateQueries(question, runManager);\n    const documents = await this._retrieveDocuments(queries, runManager);\n    const uniqueDocuments = this._uniqueUnion(documents);\n\n    let outputDocs = uniqueDocuments;\n    if (this.documentCompressor && uniqueDocuments.length) {\n      outputDocs = await this.documentCompressor.compressDocuments(\n        uniqueDocuments,\n        question,\n        runManager?.getChild()\n      );\n      if (this.documentCompressorFilteringFn) {\n        outputDocs = this.documentCompressorFilteringFn(outputDocs);\n      }\n    }\n\n    return outputDocs;\n  }\n}\n"],"mappings":";;;;;;;;;AAoBA,IAAM,uBAAN,cAAmC,iBAA2B;CAC5D,OAAO,UAAU;AACf,SAAO;CACR;CAED,eAAe;EAAC;EAAa;EAAc;CAAa;CAExD,MAAM,MAAMA,MAAiC;EAC3C,MAAM,gBAAgB,KAAK,QAAQ,cAAc;EACjD,MAAM,cAAc,KAAK,QAAQ,eAAe;EAChD,MAAM,sBACJ,kBAAkB,KAAK,IAAI,gBAAgB;EAC7C,MAAM,oBAAoB,gBAAgB,KAAK,KAAK,SAAS;EAC7D,MAAM,QAAQ,KACX,MAAM,qBAAqB,kBAAkB,CAC7C,MAAM,CACN,MAAM,KAAK,CACX,OAAO,CAAC,SAAS,KAAK,MAAM,KAAK,GAAG;AACvC,SAAO,EAAE,MAAO;CACjB;CAED,wBAAgC;AAC9B,QAAM,IAAI,MAAM;CACjB;AACF;AAGD,MAAM,uCAAuC,IAAI,eAAe;CAC9D,gBAAgB,CAAC,YAAY,YAAa;CAC1C,UAAU,CAAC;;;;;;;;;;;;;;;6BAegB,CAAC;AAC7B;;;;;;;;;;;;;;AAyBD,IAAa,sBAAb,cAAyC,cAAc;CACrD,OAAO,UAAU;AACf,SAAO;CACR;CAED,eAAe;EAAC;EAAa;EAAc;CAAa;CAExD,AAAQ;CAER,AAAQ;CAER,AAAQ,aAAa;CAErB,AAAQ,YAAY;CAEpB;CAEA;CAEA,YAAYC,QAAkC;EAC5C,MAAM,OAAO;EACb,KAAK,YAAY,OAAO;EACxB,KAAK,WAAW,OAAO;EACvB,KAAK,aAAa,OAAO,cAAc,KAAK;EAC5C,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,qBAAqB,OAAO;EACjC,KAAK,gCAAgC,OAAO;CAC7C;CAED,OAAO,QACLC,QAIqB;EACrB,MAAM,EACJ,WACA,KACA,SAAS,sBACT,YACA,UACA,GAAG,MACJ,GAAG;EACJ,MAAM,eAAe,IAAI;EACzB,MAAM,WAAW,IAAI,SAAS;GAAE;GAAK;GAAQ;EAAc;AAC3D,SAAO,IAAI,KAAK;GAAE;GAAW;GAAU;GAAY;GAAW,GAAG;EAAM;CACxE;CAGD,MAAc,iBACZC,UACAC,YACmB;EACnB,MAAM,WAAW,MAAM,KAAK,SAAS,KACnC;GAAE;GAAU,YAAY,KAAK;EAAY,GACzC,YAAY,UAAU,CACvB;EACD,MAAM,QAAQ,SAAS,KAAK,KAAK,cAAc,CAAE;AACjD,MAAI,KAAK,SACP,QAAQ,IAAI,CAAC,mBAAmB,EAAE,OAAO,CAAC;AAE5C,SAAO;CACR;CAGD,MAAc,mBACZC,SACAD,YACqB;EACrB,MAAME,YAAwB,CAAE;EAChC,MAAM,QAAQ,IACZ,QAAQ,IAAI,OAAO,UAAU;GAC3B,MAAM,OAAO,MAAM,KAAK,UAAU,qBAChC,OACA,YAAY,UAAU,CACvB;GACD,UAAU,KAAK,GAAG,KAAK;EACxB,EAAC,CACH;AACD,SAAO;CACR;CAGD,AAAQ,aAAaA,WAAmC;EACtD,MAAMC,sBAAmD,CAAE;AAE3D,OAAK,MAAM,OAAO,WAAW;GAC3B,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,KAAK,UACrC,OAAO,QAAQ,IAAI,SAAS,CAAC,MAAM,CACpC,EAAE;GACH,oBAAoB,OAAO;EAC5B;EAED,MAAM,kBAAkB,OAAO,OAAO,oBAAoB;AAC1D,SAAO;CACR;CAED,MAAM,sBACJJ,UACAC,YACqB;EACrB,MAAM,UAAU,MAAM,KAAK,iBAAiB,UAAU,WAAW;EACjE,MAAM,YAAY,MAAM,KAAK,mBAAmB,SAAS,WAAW;EACpE,MAAM,kBAAkB,KAAK,aAAa,UAAU;EAEpD,IAAI,aAAa;AACjB,MAAI,KAAK,sBAAsB,gBAAgB,QAAQ;GACrD,aAAa,MAAM,KAAK,mBAAmB,kBACzC,iBACA,UACA,YAAY,UAAU,CACvB;AACD,OAAI,KAAK,+BACP,aAAa,KAAK,8BAA8B,WAAW;EAE9D;AAED,SAAO;CACR;AACF"}

package/dist/retrievers/multi_vector.cjs

@@ -1,67 +0,0 @@
-const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
-const require_storage_encoder_backed = require('../storage/encoder_backed.cjs');
-const __langchain_core_retrievers = require_rolldown_runtime.__toESM(require("@langchain/core/retrievers"));
-
-//#region src/retrievers/multi_vector.ts
-var multi_vector_exports = {};
-require_rolldown_runtime.__export(multi_vector_exports, { MultiVectorRetriever: () => MultiVectorRetriever });
-/**
-* A retriever that retrieves documents from a vector store and a document
-* store. It uses the vector store to find relevant documents based on a
-* query, and then retrieves the full documents from the document store.
-* @example
-* ```typescript
-* const retriever = new MultiVectorRetriever({
-*   vectorstore: new FaissStore(),
-*   byteStore: new InMemoryStore<Unit8Array>(),
-*   idKey: "doc_id",
-*   childK: 20,
-*   parentK: 5,
-* });
-*
-* const retrieverResult = await retriever.getRelevantDocuments("justice breyer");
-* console.log(retrieverResult[0].pageContent.length);
-* ```
-*/
-var MultiVectorRetriever = class extends __langchain_core_retrievers.BaseRetriever {
-	static lc_name() {
-		return "MultiVectorRetriever";
-	}
-	lc_namespace = [
-		"langchain",
-		"retrievers",
-		"multi_vector"
-	];
-	vectorstore;
-	docstore;
-	idKey;
-	childK;
-	parentK;
-	constructor(args) {
-		super(args);
-		this.vectorstore = args.vectorstore;
-		if (args.byteStore) this.docstore = require_storage_encoder_backed.createDocumentStoreFromByteStore(args.byteStore);
-		else if (args.docstore) this.docstore = args.docstore;
-		else throw new Error("byteStore and docstore are undefined. Please provide at least one.");
-		this.idKey = args.idKey ?? "doc_id";
-		this.childK = args.childK;
-		this.parentK = args.parentK;
-	}
-	async _getRelevantDocuments(query) {
-		const subDocs = await this.vectorstore.similaritySearch(query, this.childK);
-		const ids = [];
-		for (const doc of subDocs) if (doc.metadata[this.idKey] && !ids.includes(doc.metadata[this.idKey])) ids.push(doc.metadata[this.idKey]);
-		const docs = await this.docstore.mget(ids);
-		return docs.filter((doc) => doc !== void 0).slice(0, this.parentK);
-	}
-};
-
-//#endregion
-exports.MultiVectorRetriever = MultiVectorRetriever;
-Object.defineProperty(exports, 'multi_vector_exports', {
-  enumerable: true,
-  get: function () {
-    return multi_vector_exports;
-  }
-});
-//# sourceMappingURL=multi_vector.cjs.map

package/dist/retrievers/multi_vector.cjs.map

@@ -1 +0,0 @@
-{"version":3,"file":"multi_vector.cjs","names":["BaseRetriever","args: MultiVectorRetrieverInput","createDocumentStoreFromByteStore","query: string","ids: string[]"],"sources":["../../src/retrievers/multi_vector.ts"],"sourcesContent":["import {\n  BaseRetriever,\n  type BaseRetrieverInput,\n} from \"@langchain/core/retrievers\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { Document } from \"@langchain/core/documents\";\nimport { BaseStore, type BaseStoreInterface } from \"@langchain/core/stores\";\nimport { createDocumentStoreFromByteStore } from \"../storage/encoder_backed.js\";\n\n/**\n * Arguments for the MultiVectorRetriever class.\n */\nexport interface MultiVectorRetrieverInput extends BaseRetrieverInput {\n  vectorstore: VectorStoreInterface;\n  /** @deprecated Prefer `byteStore`. */\n  docstore?: BaseStoreInterface<string, Document>;\n  byteStore?: BaseStore<string, Uint8Array>;\n  idKey?: string;\n  childK?: number;\n  parentK?: number;\n}\n\n/**\n * A retriever that retrieves documents from a vector store and a document\n * store. It uses the vector store to find relevant documents based on a\n * query, and then retrieves the full documents from the document store.\n * @example\n * ```typescript\n * const retriever = new MultiVectorRetriever({\n *   vectorstore: new FaissStore(),\n *   byteStore: new InMemoryStore<Unit8Array>(),\n *   idKey: \"doc_id\",\n *   childK: 20,\n *   parentK: 5,\n * });\n *\n * const retrieverResult = await retriever.getRelevantDocuments(\"justice breyer\");\n * console.log(retrieverResult[0].pageContent.length);\n * ```\n */\nexport class MultiVectorRetriever extends BaseRetriever {\n  static lc_name() {\n    return \"MultiVectorRetriever\";\n  }\n\n  lc_namespace = [\"langchain\", \"retrievers\", \"multi_vector\"];\n\n  public vectorstore: VectorStoreInterface;\n\n  public docstore: BaseStoreInterface<string, Document>;\n\n  protected idKey: string;\n\n  protected childK?: number;\n\n  protected parentK?: number;\n\n  constructor(args: MultiVectorRetrieverInput) {\n    super(args);\n    this.vectorstore = args.vectorstore;\n    if (args.byteStore) {\n      this.docstore = createDocumentStoreFromByteStore(args.byteStore);\n    } else if (args.docstore) {\n      this.docstore = args.docstore;\n    } else {\n      throw new Error(\n        \"byteStore and docstore are undefined. Please provide at least one.\"\n      );\n    }\n    this.idKey = args.idKey ?? \"doc_id\";\n    this.childK = args.childK;\n    this.parentK = args.parentK;\n  }\n\n  async _getRelevantDocuments(query: string): Promise<Document[]> {\n    const subDocs = await this.vectorstore.similaritySearch(query, this.childK);\n    const ids: string[] = [];\n    for (const doc of subDocs) {\n      if (doc.metadata[this.idKey] && !ids.includes(doc.metadata[this.idKey])) {\n        ids.push(doc.metadata[this.idKey]);\n      }\n    }\n    const docs = await this.docstore.mget(ids);\n    return docs\n      .filter((doc) => doc !== undefined)\n      .slice(0, this.parentK) as Document[];\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AAwCA,IAAa,uBAAb,cAA0CA,0CAAc;CACtD,OAAO,UAAU;AACf,SAAO;CACR;CAED,eAAe;EAAC;EAAa;EAAc;CAAe;CAE1D,AAAO;CAEP,AAAO;CAEP,AAAU;CAEV,AAAU;CAEV,AAAU;CAEV,YAAYC,MAAiC;EAC3C,MAAM,KAAK;EACX,KAAK,cAAc,KAAK;AACxB,MAAI,KAAK,WACP,KAAK,WAAWC,gEAAiC,KAAK,UAAU;WACvD,KAAK,UACd,KAAK,WAAW,KAAK;MAErB,OAAM,IAAI,MACR;EAGJ,KAAK,QAAQ,KAAK,SAAS;EAC3B,KAAK,SAAS,KAAK;EACnB,KAAK,UAAU,KAAK;CACrB;CAED,MAAM,sBAAsBC,OAAoC;EAC9D,MAAM,UAAU,MAAM,KAAK,YAAY,iBAAiB,OAAO,KAAK,OAAO;EAC3E,MAAMC,MAAgB,CAAE;AACxB,OAAK,MAAM,OAAO,QAChB,KAAI,IAAI,SAAS,KAAK,UAAU,CAAC,IAAI,SAAS,IAAI,SAAS,KAAK,OAAO,EACrE,IAAI,KAAK,IAAI,SAAS,KAAK,OAAO;EAGtC,MAAM,OAAO,MAAM,KAAK,SAAS,KAAK,IAAI;AAC1C,SAAO,KACJ,OAAO,CAAC,QAAQ,QAAQ,OAAU,CAClC,MAAM,GAAG,KAAK,QAAQ;CAC1B;AACF"}

package/dist/retrievers/multi_vector.d.cts

@@ -1,51 +0,0 @@
-import { Document } from "@langchain/core/documents";
-import { VectorStoreInterface } from "@langchain/core/vectorstores";
-import { BaseRetriever, BaseRetrieverInput } from "@langchain/core/retrievers";
-import { BaseStore, BaseStoreInterface } from "@langchain/core/stores";
-
-//#region src/retrievers/multi_vector.d.ts
-
-/**
- * Arguments for the MultiVectorRetriever class.
- */
-interface MultiVectorRetrieverInput extends BaseRetrieverInput {
-  vectorstore: VectorStoreInterface;
-  /** @deprecated Prefer `byteStore`. */
-  docstore?: BaseStoreInterface<string, Document>;
-  byteStore?: BaseStore<string, Uint8Array>;
-  idKey?: string;
-  childK?: number;
-  parentK?: number;
-}
-/**
- * A retriever that retrieves documents from a vector store and a document
- * store. It uses the vector store to find relevant documents based on a
- * query, and then retrieves the full documents from the document store.
- * @example
- * ```typescript
- * const retriever = new MultiVectorRetriever({
- *   vectorstore: new FaissStore(),
- *   byteStore: new InMemoryStore<Unit8Array>(),
- *   idKey: "doc_id",
- *   childK: 20,
- *   parentK: 5,
- * });
- *
- * const retrieverResult = await retriever.getRelevantDocuments("justice breyer");
- * console.log(retrieverResult[0].pageContent.length);
- * ```
- */
-declare class MultiVectorRetriever extends BaseRetriever {
-  static lc_name(): string;
-  lc_namespace: string[];
-  vectorstore: VectorStoreInterface;
-  docstore: BaseStoreInterface<string, Document>;
-  protected idKey: string;
-  protected childK?: number;
-  protected parentK?: number;
-  constructor(args: MultiVectorRetrieverInput);
-  _getRelevantDocuments(query: string): Promise<Document[]>;
-}
-//#endregion
-export { MultiVectorRetriever, MultiVectorRetrieverInput };
-//# sourceMappingURL=multi_vector.d.cts.map

package/dist/retrievers/multi_vector.d.cts.map

@@ -1 +0,0 @@
-{"version":3,"file":"multi_vector.d.cts","names":["BaseRetriever","BaseRetrieverInput","VectorStoreInterface","Document","BaseStore","BaseStoreInterface","MultiVectorRetrieverInput","Uint8Array","MultiVectorRetriever","Promise"],"sources":["../../src/retrievers/multi_vector.d.ts"],"sourcesContent":["import { BaseRetriever, type BaseRetrieverInput } from \"@langchain/core/retrievers\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { Document } from \"@langchain/core/documents\";\nimport { BaseStore, type BaseStoreInterface } from \"@langchain/core/stores\";\n/**\n * Arguments for the MultiVectorRetriever class.\n */\nexport interface MultiVectorRetrieverInput extends BaseRetrieverInput {\n    vectorstore: VectorStoreInterface;\n    /** @deprecated Prefer `byteStore`. */\n    docstore?: BaseStoreInterface<string, Document>;\n    byteStore?: BaseStore<string, Uint8Array>;\n    idKey?: string;\n    childK?: number;\n    parentK?: number;\n}\n/**\n * A retriever that retrieves documents from a vector store and a document\n * store. It uses the vector store to find relevant documents based on a\n * query, and then retrieves the full documents from the document store.\n * @example\n * ```typescript\n * const retriever = new MultiVectorRetriever({\n *   vectorstore: new FaissStore(),\n *   byteStore: new InMemoryStore<Unit8Array>(),\n *   idKey: \"doc_id\",\n *   childK: 20,\n *   parentK: 5,\n * });\n *\n * const retrieverResult = await retriever.getRelevantDocuments(\"justice breyer\");\n * console.log(retrieverResult[0].pageContent.length);\n * ```\n */\nexport declare class MultiVectorRetriever extends BaseRetriever {\n    static lc_name(): string;\n    lc_namespace: string[];\n    vectorstore: VectorStoreInterface;\n    docstore: BaseStoreInterface<string, Document>;\n    protected idKey: string;\n    protected childK?: number;\n    protected parentK?: number;\n    constructor(args: MultiVectorRetrieverInput);\n    _getRelevantDocuments(query: string): Promise<Document[]>;\n}\n"],"mappings":";;;;;;;;;AAOA;AAA0C,UAAzBM,yBAAAA,SAAkCL,kBAAT,CAAA;EAAA,WACzBC,EAAAA,oBAAAA;EAAoB;EAEa,QAAnCG,CAAAA,EAAAA,kBAAAA,CAAAA,MAAAA,EAA2BF,QAA3BE,CAAAA;EAAkB,SACCE,CAAAA,EAAlBH,SAAkBG,CAAAA,MAAAA,EAAAA,UAAAA,CAAAA;EAAU,KAA5BH,CAAAA,EAAAA,MAAAA;EAAS,MAJ0BH,CAAAA,EAAAA,MAAAA;EAAkB,OAAA,CAAA,EAAA,MAAA;AA2BrE;;;;;;;;;AAA+D;;;;;;;;;;cAA1CO,oBAAAA,SAA6BR,aAAAA;;;eAGjCE;YACHG,2BAA2BF;;;;oBAInBG;wCACoBG,QAAQN"}