langchain 1.0.0-alpha.7 → 1.0.0-alpha.8

package/dist/chains/chat_vector_db_chain.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"chat_vector_db_chain.js","names":["fields: ChatVectorDBQAChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun","question: string","chatHistory: string","result","data: SerializedChatVectorDBQAChain","values: LoadValues","llm: BaseLanguageModelInterface","vectorstore: VectorStoreInterface","options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    }"],"sources":["../../src/chains/chat_vector_db_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { loadQAStuffChain } from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\nconst 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.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\n\n/**\n * Interface for the input parameters of the ChatVectorDBQAChain class.\n */\nexport interface ChatVectorDBQAChainInput extends ChainInputs {\n  vectorstore: VectorStoreInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  outputKey?: string;\n  inputKey?: string;\n  k?: number;\n}\n\n/** @deprecated use `ConversationalRetrievalQAChain` instead. */\nexport class ChatVectorDBQAChain\n  extends BaseChain\n  implements ChatVectorDBQAChainInput\n{\n  k = 4;\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  outputKey = \"result\";\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  vectorstore: VectorStoreInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: ChatVectorDBQAChainInput) {\n    super(fields);\n    this.vectorstore = fields.vectorstore;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.k = fields.k ?? this.k;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`chat history key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string = values[this.chatHistoryKey];\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      console.log(\"_call\", values, keys);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.vectorstore.similaritySearch(\n      newQuestion,\n      this.k,\n      undefined,\n      runManager?.getChild(\"vectorstore\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    const result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      return {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"chat-vector-db\" as const;\n  }\n\n  static async deserialize(\n    data: SerializedChatVectorDBQAChain,\n    values: LoadValues\n  ) {\n    if (!(\"vectorstore\" in values)) {\n      throw new Error(\n        `Need to pass in a vectorstore to deserialize VectorDBQAChain`\n      );\n    }\n    const { vectorstore } = values;\n\n    return new ChatVectorDBQAChain({\n      combineDocumentsChain: await BaseChain.deserialize(\n        data.combine_documents_chain\n      ),\n      questionGeneratorChain: await LLMChain.deserialize(\n        data.question_generator\n      ),\n      k: data.k,\n      vectorstore,\n    });\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    return {\n      _type: this._chainType(),\n      combine_documents_chain: this.combineDocumentsChain.serialize(),\n      question_generator: this.questionGeneratorChain.serialize(),\n      k: this.k,\n    };\n  }\n\n  /**\n   * Creates an instance of ChatVectorDBQAChain using a BaseLanguageModel\n   * and other options.\n   * @param llm Instance of BaseLanguageModel used to generate a new question.\n   * @param vectorstore Instance of VectorStore used for vector operations.\n   * @param options (Optional) Additional options for creating the ChatVectorDBQAChain instance.\n   * @returns New instance of ChatVectorDBQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    vectorstore: VectorStoreInterface,\n    options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    } = {}\n  ): ChatVectorDBQAChain {\n    const { questionGeneratorTemplate, qaTemplate, verbose, ...rest } = options;\n    const question_generator_prompt = PromptTemplate.fromTemplate(\n      questionGeneratorTemplate || question_generator_template\n    );\n    const qa_prompt = PromptTemplate.fromTemplate(qaTemplate || qa_template);\n\n    const qaChain = loadQAStuffChain(llm, { prompt: qa_prompt, verbose });\n    const questionGeneratorChain = new LLMChain({\n      prompt: question_generator_prompt,\n      llm,\n      verbose,\n    });\n    const instance = new this({\n      vectorstore,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;;AAaA,MAAM,8BAA8B,CAAC;;;;;oBAKjB,CAAC;AAErB,MAAM,cAAc,CAAC;;;;;eAKN,CAAC;;AAgBhB,IAAa,sBAAb,MAAa,4BACH,UAEV;CACE,IAAI;CAEJ,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,KAAK,cAAe;CAC5C;CAED,YAAY;CAEZ,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,SAAU;CACxB;CAED;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAYA,QAAkC;EAC5C,MAAM,OAAO;EACb,KAAK,cAAc,OAAO;EAC1B,KAAK,wBAAwB,OAAO;EACpC,KAAK,yBAAyB,OAAO;EACrC,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,IAAI,OAAO,KAAK,KAAK;EAC1B,KAAK,wBACH,OAAO,yBAAyB,KAAK;CACxC;;CAGD,MAAM,MACJC,QACAC,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,CAAC,aAAa,EAAE,KAAK,SAAS,WAAW,CAAC;AAE5D,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,CAAC,iBAAiB,EAAE,KAAK,SAAS,WAAW,CAAC;EAEhE,MAAMC,WAAmB,OAAO,KAAK;EACrC,MAAMC,cAAsB,OAAO,KAAK;EACxC,IAAI,cAAc;AAClB,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAMC,WAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;GACf,GACD,YAAY,SAAS,qBAAqB,CAC3C;GACD,MAAM,OAAO,OAAO,KAAKA,SAAO;GAChC,QAAQ,IAAI,SAAS,QAAQ,KAAK;AAClC,OAAI,KAAK,WAAW,GAClB,cAAcA,SAAO,KAAK;OAE1B,OAAM,IAAI,MACR;EAGL;EACD,MAAM,OAAO,MAAM,KAAK,YAAY,iBAClC,aACA,KAAK,GACL,QACA,YAAY,SAAS,cAAc,CACpC;EACD,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;EACf;EACD,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,sBACP,QAAO;GACL,GAAG;GACH,iBAAiB;EAClB;AAEH,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,aAAa,YACXC,MACAC,QACA;AACA,MAAI,EAAE,iBAAiB,QACrB,OAAM,IAAI,MACR,CAAC,4DAA4D,CAAC;EAGlE,MAAM,EAAE,aAAa,GAAG;AAExB,SAAO,IAAI,oBAAoB;GAC7B,uBAAuB,MAAM,UAAU,YACrC,KAAK,wBACN;GACD,wBAAwB,MAAM,SAAS,YACrC,KAAK,mBACN;GACD,GAAG,KAAK;GACR;EACD;CACF;CAED,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,yBAAyB,KAAK,sBAAsB,WAAW;GAC/D,oBAAoB,KAAK,uBAAuB,WAAW;GAC3D,GAAG,KAAK;EACT;CACF;;;;;;;;;CAUD,OAAO,QACLC,KACAC,aACAC,UAQI,CAAE,GACe;EACrB,MAAM,EAAE,2BAA2B,YAAY,QAAS,GAAG,MAAM,GAAG;EACpE,MAAM,4BAA4B,eAAe,aAC/C,6BAA6B,4BAC9B;EACD,MAAM,YAAY,eAAe,aAAa,cAAc,YAAY;EAExE,MAAM,UAAU,iBAAiB,KAAK;GAAE,QAAQ;GAAW;EAAS,EAAC;EACrE,MAAM,yBAAyB,IAAI,SAAS;GAC1C,QAAQ;GACR;GACA;EACD;EACD,MAAM,WAAW,IAAI,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA,GAAG;EACJ;AACD,SAAO;CACR;AACF"}

package/dist/chains/combine_docs_chain.cjs

@@ -1,275 +0,0 @@
-const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
-const require_base = require('./base.cjs');
-const require_llm_chain = require('./llm_chain.cjs');
-const __langchain_core_prompts = require_rolldown_runtime.__toESM(require("@langchain/core/prompts"));
-
-//#region src/chains/combine_docs_chain.ts
-/**
-* Chain that combines documents by stuffing into context.
-* @augments BaseChain
-* @augments StuffDocumentsChainInput
-*/
-var StuffDocumentsChain = class StuffDocumentsChain extends require_base.BaseChain {
-	static lc_name() {
-		return "StuffDocumentsChain";
-	}
-	llmChain;
-	inputKey = "input_documents";
-	documentVariableName = "context";
-	get inputKeys() {
-		return [this.inputKey, ...this.llmChain.inputKeys].filter((key) => key !== this.documentVariableName);
-	}
-	get outputKeys() {
-		return this.llmChain.outputKeys;
-	}
-	constructor(fields) {
-		super(fields);
-		this.llmChain = fields.llmChain;
-		this.documentVariableName = fields.documentVariableName ?? this.documentVariableName;
-		this.inputKey = fields.inputKey ?? this.inputKey;
-	}
-	/** @ignore */
-	_prepInputs(values) {
-		if (!(this.inputKey in values)) throw new Error(`Document key ${this.inputKey} not found.`);
-		const { [this.inputKey]: docs,...rest } = values;
-		const texts = docs.map(({ pageContent }) => pageContent);
-		const text = texts.join("\n\n");
-		return {
-			...rest,
-			[this.documentVariableName]: text
-		};
-	}
-	/** @ignore */
-	async _call(values, runManager) {
-		const result = await this.llmChain.call(this._prepInputs(values), runManager?.getChild("combine_documents"));
-		return result;
-	}
-	_chainType() {
-		return "stuff_documents_chain";
-	}
-	static async deserialize(data) {
-		if (!data.llm_chain) throw new Error("Missing llm_chain");
-		return new StuffDocumentsChain({ llmChain: await require_llm_chain.LLMChain.deserialize(data.llm_chain) });
-	}
-	serialize() {
-		return {
-			_type: this._chainType(),
-			llm_chain: this.llmChain.serialize()
-		};
-	}
-};
-/**
-* Combine documents by mapping a chain over them, then combining results.
-* @augments BaseChain
-* @augments StuffDocumentsChainInput
-*/
-var MapReduceDocumentsChain = class MapReduceDocumentsChain extends require_base.BaseChain {
-	static lc_name() {
-		return "MapReduceDocumentsChain";
-	}
-	llmChain;
-	inputKey = "input_documents";
-	documentVariableName = "context";
-	returnIntermediateSteps = false;
-	get inputKeys() {
-		return [this.inputKey, ...this.combineDocumentChain.inputKeys];
-	}
-	get outputKeys() {
-		return this.combineDocumentChain.outputKeys;
-	}
-	maxTokens = 3e3;
-	maxIterations = 10;
-	ensureMapStep = false;
-	combineDocumentChain;
-	constructor(fields) {
-		super(fields);
-		this.llmChain = fields.llmChain;
-		this.combineDocumentChain = fields.combineDocumentChain;
-		this.documentVariableName = fields.documentVariableName ?? this.documentVariableName;
-		this.ensureMapStep = fields.ensureMapStep ?? this.ensureMapStep;
-		this.inputKey = fields.inputKey ?? this.inputKey;
-		this.maxTokens = fields.maxTokens ?? this.maxTokens;
-		this.maxIterations = fields.maxIterations ?? this.maxIterations;
-		this.returnIntermediateSteps = fields.returnIntermediateSteps ?? false;
-	}
-	/** @ignore */
-	async _call(values, runManager) {
-		if (!(this.inputKey in values)) throw new Error(`Document key ${this.inputKey} not found.`);
-		const { [this.inputKey]: docs,...rest } = values;
-		let currentDocs = docs;
-		let intermediateSteps = [];
-		for (let i = 0; i < this.maxIterations; i += 1) {
-			const inputs = currentDocs.map((d) => ({
-				[this.documentVariableName]: d.pageContent,
-				...rest
-			}));
-			const canSkipMapStep = i !== 0 || !this.ensureMapStep;
-			if (canSkipMapStep) {
-				const formatted = await this.combineDocumentChain.llmChain.prompt.format(this.combineDocumentChain._prepInputs({
-					[this.combineDocumentChain.inputKey]: currentDocs,
-					...rest
-				}));
-				const length = await this.combineDocumentChain.llmChain._getNumTokens(formatted);
-				const withinTokenLimit = length < this.maxTokens;
-				if (withinTokenLimit) break;
-			}
-			const results = await this.llmChain.apply(inputs, runManager ? Array.from({ length: inputs.length }, (_, i$1) => runManager.getChild(`map_${i$1 + 1}`)) : void 0);
-			const { outputKey } = this.llmChain;
-			if (this.returnIntermediateSteps) intermediateSteps = intermediateSteps.concat(results.map((r) => r[outputKey]));
-			currentDocs = results.map((r) => ({
-				pageContent: r[outputKey],
-				metadata: {}
-			}));
-		}
-		const newInputs = {
-			[this.combineDocumentChain.inputKey]: currentDocs,
-			...rest
-		};
-		const result = await this.combineDocumentChain.call(newInputs, runManager?.getChild("combine_documents"));
-		if (this.returnIntermediateSteps) return {
-			...result,
-			intermediateSteps
-		};
-		return result;
-	}
-	_chainType() {
-		return "map_reduce_documents_chain";
-	}
-	static async deserialize(data) {
-		if (!data.llm_chain) throw new Error("Missing llm_chain");
-		if (!data.combine_document_chain) throw new Error("Missing combine_document_chain");
-		return new MapReduceDocumentsChain({
-			llmChain: await require_llm_chain.LLMChain.deserialize(data.llm_chain),
-			combineDocumentChain: await StuffDocumentsChain.deserialize(data.combine_document_chain)
-		});
-	}
-	serialize() {
-		return {
-			_type: this._chainType(),
-			llm_chain: this.llmChain.serialize(),
-			combine_document_chain: this.combineDocumentChain.serialize()
-		};
-	}
-};
-/**
-* Combine documents by doing a first pass and then refining on more documents.
-* @augments BaseChain
-* @augments RefineDocumentsChainInput
-*/
-var RefineDocumentsChain = class RefineDocumentsChain extends require_base.BaseChain {
-	static lc_name() {
-		return "RefineDocumentsChain";
-	}
-	llmChain;
-	inputKey = "input_documents";
-	outputKey = "output_text";
-	documentVariableName = "context";
-	initialResponseName = "existing_answer";
-	refineLLMChain;
-	get defaultDocumentPrompt() {
-		return new __langchain_core_prompts.PromptTemplate({
-			inputVariables: ["page_content"],
-			template: "{page_content}"
-		});
-	}
-	documentPrompt = this.defaultDocumentPrompt;
-	get inputKeys() {
-		return [...new Set([
-			this.inputKey,
-			...this.llmChain.inputKeys,
-			...this.refineLLMChain.inputKeys
-		])].filter((key) => key !== this.documentVariableName && key !== this.initialResponseName);
-	}
-	get outputKeys() {
-		return [this.outputKey];
-	}
-	constructor(fields) {
-		super(fields);
-		this.llmChain = fields.llmChain;
-		this.refineLLMChain = fields.refineLLMChain;
-		this.documentVariableName = fields.documentVariableName ?? this.documentVariableName;
-		this.inputKey = fields.inputKey ?? this.inputKey;
-		this.outputKey = fields.outputKey ?? this.outputKey;
-		this.documentPrompt = fields.documentPrompt ?? this.documentPrompt;
-		this.initialResponseName = fields.initialResponseName ?? this.initialResponseName;
-	}
-	/** @ignore */
-	async _constructInitialInputs(doc, rest) {
-		const baseInfo = {
-			page_content: doc.pageContent,
-			...doc.metadata
-		};
-		const documentInfo = {};
-		this.documentPrompt.inputVariables.forEach((value) => {
-			documentInfo[value] = baseInfo[value];
-		});
-		const baseInputs = { [this.documentVariableName]: await this.documentPrompt.format({ ...documentInfo }) };
-		const inputs = {
-			...baseInputs,
-			...rest
-		};
-		return inputs;
-	}
-	/** @ignore */
-	async _constructRefineInputs(doc, res) {
-		const baseInfo = {
-			page_content: doc.pageContent,
-			...doc.metadata
-		};
-		const documentInfo = {};
-		this.documentPrompt.inputVariables.forEach((value) => {
-			documentInfo[value] = baseInfo[value];
-		});
-		const baseInputs = { [this.documentVariableName]: await this.documentPrompt.format({ ...documentInfo }) };
-		const inputs = {
-			[this.initialResponseName]: res,
-			...baseInputs
-		};
-		return inputs;
-	}
-	/** @ignore */
-	async _call(values, runManager) {
-		if (!(this.inputKey in values)) throw new Error(`Document key ${this.inputKey} not found.`);
-		const { [this.inputKey]: docs,...rest } = values;
-		const currentDocs = docs;
-		const initialInputs = await this._constructInitialInputs(currentDocs[0], rest);
-		let res = await this.llmChain.predict({ ...initialInputs }, runManager?.getChild("answer"));
-		const refineSteps = [res];
-		for (let i = 1; i < currentDocs.length; i += 1) {
-			const refineInputs = await this._constructRefineInputs(currentDocs[i], res);
-			const inputs = {
-				...refineInputs,
-				...rest
-			};
-			res = await this.refineLLMChain.predict({ ...inputs }, runManager?.getChild("refine"));
-			refineSteps.push(res);
-		}
-		return { [this.outputKey]: res };
-	}
-	_chainType() {
-		return "refine_documents_chain";
-	}
-	static async deserialize(data) {
-		const SerializedLLMChain = data.llm_chain;
-		if (!SerializedLLMChain) throw new Error("Missing llm_chain");
-		const SerializedRefineDocumentChain = data.refine_llm_chain;
-		if (!SerializedRefineDocumentChain) throw new Error("Missing refine_llm_chain");
-		return new RefineDocumentsChain({
-			llmChain: await require_llm_chain.LLMChain.deserialize(SerializedLLMChain),
-			refineLLMChain: await require_llm_chain.LLMChain.deserialize(SerializedRefineDocumentChain)
-		});
-	}
-	serialize() {
-		return {
-			_type: this._chainType(),
-			llm_chain: this.llmChain.serialize(),
-			refine_llm_chain: this.refineLLMChain.serialize()
-		};
-	}
-};
-
-//#endregion
-exports.MapReduceDocumentsChain = MapReduceDocumentsChain;
-exports.RefineDocumentsChain = RefineDocumentsChain;
-exports.StuffDocumentsChain = StuffDocumentsChain;
-//# sourceMappingURL=combine_docs_chain.cjs.map

package/dist/chains/combine_docs_chain.cjs.map

@@ -1 +0,0 @@
-{"version":3,"file":"combine_docs_chain.cjs","names":["BaseChain","fields: StuffDocumentsChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun","data: SerializedStuffDocumentsChain","LLMChain","fields: MapReduceDocumentsChainInput","intermediateSteps: string[]","i","data: SerializedMapReduceDocumentsChain","PromptTemplate","fields: RefineDocumentsChainInput","doc: Document","rest: Record<string, unknown>","baseInfo: Record<string, unknown>","documentInfo: Record<string, unknown>","baseInputs: Record<string, unknown>","res: string","data: SerializedRefineDocumentsChain"],"sources":["../../src/chains/combine_docs_chain.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport type {\n  SerializedStuffDocumentsChain,\n  SerializedMapReduceDocumentsChain,\n  SerializedRefineDocumentsChain,\n} from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\n\n/**\n * Interface for the input properties of the StuffDocumentsChain class.\n */\nexport interface StuffDocumentsChainInput extends ChainInputs {\n  /** LLM Wrapper to use after formatting documents */\n  llmChain: LLMChain;\n  inputKey?: string;\n  /** Variable name in the LLM chain to put the documents in */\n  documentVariableName?: string;\n}\n\n/**\n * Chain that combines documents by stuffing into context.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport class StuffDocumentsChain\n  extends BaseChain\n  implements StuffDocumentsChainInput\n{\n  static lc_name() {\n    return \"StuffDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  documentVariableName = \"context\";\n\n  get inputKeys() {\n    return [this.inputKey, ...this.llmChain.inputKeys].filter(\n      (key) => key !== this.documentVariableName\n    );\n  }\n\n  get outputKeys() {\n    return this.llmChain.outputKeys;\n  }\n\n  constructor(fields: StuffDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n  }\n\n  /** @ignore */\n  _prepInputs(values: ChainValues): ChainValues {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n    const texts = (docs as Document[]).map(({ pageContent }) => pageContent);\n    const text = texts.join(\"\\n\\n\");\n    return {\n      ...rest,\n      [this.documentVariableName]: text,\n    };\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const result = await this.llmChain.call(\n      this._prepInputs(values),\n      runManager?.getChild(\"combine_documents\")\n    );\n    return result;\n  }\n\n  _chainType() {\n    return \"stuff_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedStuffDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    return new StuffDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n    });\n  }\n\n  serialize(): SerializedStuffDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the MapReduceDocumentsChain\n * class.\n */\nexport interface MapReduceDocumentsChainInput extends StuffDocumentsChainInput {\n  /** The maximum number of tokens before requiring to do the reduction */\n  maxTokens?: number;\n  /** The maximum number of iterations to run through the map */\n  maxIterations?: number;\n  /** Ensures that the map step is taken regardless of max tokens */\n  ensureMapStep?: boolean;\n  /** Chain to use to combine results of applying llm_chain to documents. */\n  combineDocumentChain: StuffDocumentsChain;\n  /** Return the results of the map steps in the output. */\n  returnIntermediateSteps?: boolean;\n}\n\n/**\n * Combine documents by mapping a chain over them, then combining results.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport class MapReduceDocumentsChain\n  extends BaseChain\n  implements MapReduceDocumentsChainInput\n{\n  static lc_name() {\n    return \"MapReduceDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  documentVariableName = \"context\";\n\n  returnIntermediateSteps = false;\n\n  get inputKeys() {\n    return [this.inputKey, ...this.combineDocumentChain.inputKeys];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentChain.outputKeys;\n  }\n\n  maxTokens = 3000;\n\n  maxIterations = 10;\n\n  ensureMapStep = false;\n\n  combineDocumentChain: StuffDocumentsChain;\n\n  constructor(fields: MapReduceDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.combineDocumentChain = fields.combineDocumentChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.ensureMapStep = fields.ensureMapStep ?? this.ensureMapStep;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.maxTokens = fields.maxTokens ?? this.maxTokens;\n    this.maxIterations = fields.maxIterations ?? this.maxIterations;\n    this.returnIntermediateSteps = fields.returnIntermediateSteps ?? false;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    let currentDocs = docs as Document[];\n    let intermediateSteps: string[] = [];\n\n    // For each iteration, we'll use the `llmChain` to get a new result\n    for (let i = 0; i < this.maxIterations; i += 1) {\n      const inputs = currentDocs.map((d) => ({\n        [this.documentVariableName]: d.pageContent,\n        ...rest,\n      }));\n\n      const canSkipMapStep = i !== 0 || !this.ensureMapStep;\n      if (canSkipMapStep) {\n        // Calculate the total tokens required in the input\n        const formatted =\n          await this.combineDocumentChain.llmChain.prompt.format(\n            this.combineDocumentChain._prepInputs({\n              [this.combineDocumentChain.inputKey]: currentDocs,\n              ...rest,\n            })\n          );\n        const length = await this.combineDocumentChain.llmChain._getNumTokens(\n          formatted\n        );\n\n        const withinTokenLimit = length < this.maxTokens;\n        // If we can skip the map step, and we're within the token limit, we don't\n        // need to run the map step, so just break out of the loop.\n        if (withinTokenLimit) {\n          break;\n        }\n      }\n\n      const results = await this.llmChain.apply(\n        inputs,\n        // If we have a runManager, then we need to create a child for each input\n        // so that we can track the progress of each input.\n        runManager\n          ? Array.from({ length: inputs.length }, (_, i) =>\n              runManager.getChild(`map_${i + 1}`)\n            )\n          : undefined\n      );\n      const { outputKey } = this.llmChain;\n\n      // If the flag is set, then concat that to the intermediate steps\n      if (this.returnIntermediateSteps) {\n        intermediateSteps = intermediateSteps.concat(\n          results.map((r) => r[outputKey])\n        );\n      }\n\n      currentDocs = results.map((r) => ({\n        pageContent: r[outputKey],\n        metadata: {},\n      }));\n    }\n\n    // Now, with the final result of all the inputs from the `llmChain`, we can\n    // run the `combineDocumentChain` over them.\n    const newInputs = {\n      [this.combineDocumentChain.inputKey]: currentDocs,\n      ...rest,\n    };\n    const result = await this.combineDocumentChain.call(\n      newInputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n\n    // Return the intermediate steps results if the flag is set\n    if (this.returnIntermediateSteps) {\n      return { ...result, intermediateSteps };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"map_reduce_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedMapReduceDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    if (!data.combine_document_chain) {\n      throw new Error(\"Missing combine_document_chain\");\n    }\n\n    return new MapReduceDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n      combineDocumentChain: await StuffDocumentsChain.deserialize(\n        data.combine_document_chain\n      ),\n    });\n  }\n\n  serialize(): SerializedMapReduceDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      combine_document_chain: this.combineDocumentChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the RefineDocumentsChain class.\n */\nexport interface RefineDocumentsChainInput extends StuffDocumentsChainInput {\n  refineLLMChain: LLMChain;\n  documentPrompt?: BasePromptTemplate;\n  initialResponseName?: string;\n  documentVariableName?: string;\n  outputKey?: string;\n}\n\n/**\n * Combine documents by doing a first pass and then refining on more documents.\n * @augments BaseChain\n * @augments RefineDocumentsChainInput\n */\nexport class RefineDocumentsChain\n  extends BaseChain\n  implements RefineDocumentsChainInput\n{\n  static lc_name() {\n    return \"RefineDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  outputKey = \"output_text\";\n\n  documentVariableName = \"context\";\n\n  initialResponseName = \"existing_answer\";\n\n  refineLLMChain: LLMChain;\n\n  get defaultDocumentPrompt(): BasePromptTemplate {\n    return new PromptTemplate({\n      inputVariables: [\"page_content\"],\n      template: \"{page_content}\",\n    });\n  }\n\n  documentPrompt = this.defaultDocumentPrompt;\n\n  get inputKeys() {\n    return [\n      ...new Set([\n        this.inputKey,\n        ...this.llmChain.inputKeys,\n        ...this.refineLLMChain.inputKeys,\n      ]),\n    ].filter(\n      (key) =>\n        key !== this.documentVariableName && key !== this.initialResponseName\n    );\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  constructor(fields: RefineDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.refineLLMChain = fields.refineLLMChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.documentPrompt = fields.documentPrompt ?? this.documentPrompt;\n    this.initialResponseName =\n      fields.initialResponseName ?? this.initialResponseName;\n  }\n\n  /** @ignore */\n  async _constructInitialInputs(doc: Document, rest: Record<string, unknown>) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { ...baseInputs, ...rest };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _constructRefineInputs(doc: Document, res: string) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { [this.initialResponseName]: res, ...baseInputs };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    const currentDocs = docs as Document[];\n\n    const initialInputs = await this._constructInitialInputs(\n      currentDocs[0],\n      rest\n    );\n    let res = await this.llmChain.predict(\n      { ...initialInputs },\n      runManager?.getChild(\"answer\")\n    );\n\n    const refineSteps = [res];\n\n    for (let i = 1; i < currentDocs.length; i += 1) {\n      const refineInputs = await this._constructRefineInputs(\n        currentDocs[i],\n        res\n      );\n      const inputs = { ...refineInputs, ...rest };\n      res = await this.refineLLMChain.predict(\n        { ...inputs },\n        runManager?.getChild(\"refine\")\n      );\n      refineSteps.push(res);\n    }\n\n    return { [this.outputKey]: res };\n  }\n\n  _chainType() {\n    return \"refine_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedRefineDocumentsChain) {\n    const SerializedLLMChain = data.llm_chain;\n\n    if (!SerializedLLMChain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    const SerializedRefineDocumentChain = data.refine_llm_chain;\n\n    if (!SerializedRefineDocumentChain) {\n      throw new Error(\"Missing refine_llm_chain\");\n    }\n\n    return new RefineDocumentsChain({\n      llmChain: await LLMChain.deserialize(SerializedLLMChain),\n      refineLLMChain: await LLMChain.deserialize(SerializedRefineDocumentChain),\n    });\n  }\n\n  serialize(): SerializedRefineDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      refine_llm_chain: this.refineLLMChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;;AA4BA,IAAa,sBAAb,MAAa,4BACHA,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA,WAAW;CAEX,uBAAuB;CAEvB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,SAAS,SAAU,EAAC,OACjD,CAAC,QAAQ,QAAQ,KAAK,qBACvB;CACF;CAED,IAAI,aAAa;AACf,SAAO,KAAK,SAAS;CACtB;CAED,YAAYC,QAAkC;EAC5C,MAAM,OAAO;EACb,KAAK,WAAW,OAAO;EACvB,KAAK,uBACH,OAAO,wBAAwB,KAAK;EACtC,KAAK,WAAW,OAAO,YAAY,KAAK;CACzC;;CAGD,YAAYC,QAAkC;AAC5C,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,CAAC,aAAa,EAAE,KAAK,SAAS,WAAW,CAAC;EAE5D,MAAM,EAAE,CAAC,KAAK,WAAW,KAAM,GAAG,MAAM,GAAG;EAC3C,MAAM,QAAS,KAAoB,IAAI,CAAC,EAAE,aAAa,KAAK,YAAY;EACxE,MAAM,OAAO,MAAM,KAAK,OAAO;AAC/B,SAAO;GACL,GAAG;IACF,KAAK,uBAAuB;EAC9B;CACF;;CAGD,MAAM,MACJA,QACAC,YACsB;EACtB,MAAM,SAAS,MAAM,KAAK,SAAS,KACjC,KAAK,YAAY,OAAO,EACxB,YAAY,SAAS,oBAAoB,CAC1C;AACD,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,aAAa,YAAYC,MAAqC;AAC5D,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM;AAGlB,SAAO,IAAI,oBAAoB,EAC7B,UAAU,MAAMC,2BAAS,YAAY,KAAK,UAAU,CACrD;CACF;CAED,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;EACrC;CACF;AACF;;;;;;AAwBD,IAAa,0BAAb,MAAa,gCACHL,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA,WAAW;CAEX,uBAAuB;CAEvB,0BAA0B;CAE1B,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,qBAAqB,SAAU;CAC/D;CAED,IAAI,aAAa;AACf,SAAO,KAAK,qBAAqB;CAClC;CAED,YAAY;CAEZ,gBAAgB;CAEhB,gBAAgB;CAEhB;CAEA,YAAYM,QAAsC;EAChD,MAAM,OAAO;EACb,KAAK,WAAW,OAAO;EACvB,KAAK,uBAAuB,OAAO;EACnC,KAAK,uBACH,OAAO,wBAAwB,KAAK;EACtC,KAAK,gBAAgB,OAAO,iBAAiB,KAAK;EAClD,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,gBAAgB,OAAO,iBAAiB,KAAK;EAClD,KAAK,0BAA0B,OAAO,2BAA2B;CAClE;;CAGD,MAAM,MACJJ,QACAC,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,CAAC,aAAa,EAAE,KAAK,SAAS,WAAW,CAAC;EAE5D,MAAM,EAAE,CAAC,KAAK,WAAW,KAAM,GAAG,MAAM,GAAG;EAE3C,IAAI,cAAc;EAClB,IAAII,oBAA8B,CAAE;AAGpC,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,eAAe,KAAK,GAAG;GAC9C,MAAM,SAAS,YAAY,IAAI,CAAC,OAAO;KACpC,KAAK,uBAAuB,EAAE;IAC/B,GAAG;GACJ,GAAE;GAEH,MAAM,iBAAiB,MAAM,KAAK,CAAC,KAAK;AACxC,OAAI,gBAAgB;IAElB,MAAM,YACJ,MAAM,KAAK,qBAAqB,SAAS,OAAO,OAC9C,KAAK,qBAAqB,YAAY;MACnC,KAAK,qBAAqB,WAAW;KACtC,GAAG;IACJ,EAAC,CACH;IACH,MAAM,SAAS,MAAM,KAAK,qBAAqB,SAAS,cACtD,UACD;IAED,MAAM,mBAAmB,SAAS,KAAK;AAGvC,QAAI,iBACF;GAEH;GAED,MAAM,UAAU,MAAM,KAAK,SAAS,MAClC,QAGA,aACI,MAAM,KAAK,EAAE,QAAQ,OAAO,OAAQ,GAAE,CAAC,GAAGC,QACxC,WAAW,SAAS,CAAC,IAAI,EAAEA,MAAI,GAAG,CAAC,CACpC,GACD,OACL;GACD,MAAM,EAAE,WAAW,GAAG,KAAK;AAG3B,OAAI,KAAK,yBACP,oBAAoB,kBAAkB,OACpC,QAAQ,IAAI,CAAC,MAAM,EAAE,WAAW,CACjC;GAGH,cAAc,QAAQ,IAAI,CAAC,OAAO;IAChC,aAAa,EAAE;IACf,UAAU,CAAE;GACb,GAAE;EACJ;EAID,MAAM,YAAY;IACf,KAAK,qBAAqB,WAAW;GACtC,GAAG;EACJ;EACD,MAAM,SAAS,MAAM,KAAK,qBAAqB,KAC7C,WACA,YAAY,SAAS,oBAAoB,CAC1C;AAGD,MAAI,KAAK,wBACP,QAAO;GAAE,GAAG;GAAQ;EAAmB;AAEzC,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,aAAa,YAAYC,MAAyC;AAChE,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM;AAGlB,MAAI,CAAC,KAAK,uBACR,OAAM,IAAI,MAAM;AAGlB,SAAO,IAAI,wBAAwB;GACjC,UAAU,MAAMJ,2BAAS,YAAY,KAAK,UAAU;GACpD,sBAAsB,MAAM,oBAAoB,YAC9C,KAAK,uBACN;EACF;CACF;CAED,YAA+C;AAC7C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,wBAAwB,KAAK,qBAAqB,WAAW;EAC9D;CACF;AACF;;;;;;AAkBD,IAAa,uBAAb,MAAa,6BACHL,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA,WAAW;CAEX,YAAY;CAEZ,uBAAuB;CAEvB,sBAAsB;CAEtB;CAEA,IAAI,wBAA4C;AAC9C,SAAO,IAAIU,wCAAe;GACxB,gBAAgB,CAAC,cAAe;GAChC,UAAU;EACX;CACF;CAED,iBAAiB,KAAK;CAEtB,IAAI,YAAY;AACd,SAAO,CACL,GAAG,IAAI,IAAI;GACT,KAAK;GACL,GAAG,KAAK,SAAS;GACjB,GAAG,KAAK,eAAe;EACxB,EACF,EAAC,OACA,CAAC,QACC,QAAQ,KAAK,wBAAwB,QAAQ,KAAK,oBACrD;CACF;CAED,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,SAAU;CACxB;CAED,YAAYC,QAAmC;EAC7C,MAAM,OAAO;EACb,KAAK,WAAW,OAAO;EACvB,KAAK,iBAAiB,OAAO;EAC7B,KAAK,uBACH,OAAO,wBAAwB,KAAK;EACtC,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,iBAAiB,OAAO,kBAAkB,KAAK;EACpD,KAAK,sBACH,OAAO,uBAAuB,KAAK;CACtC;;CAGD,MAAM,wBAAwBC,KAAeC,MAA+B;EAC1E,MAAMC,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;EACR;EACD,MAAMC,eAAwC,CAAE;EAChD,KAAK,eAAe,eAAe,QAAQ,CAAC,UAAU;GACpD,aAAa,SAAS,SAAS;EAChC,EAAC;EAEF,MAAMC,aAAsC,GACzC,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,aACJ,EAAC,CACH;EACD,MAAM,SAAS;GAAE,GAAG;GAAY,GAAG;EAAM;AACzC,SAAO;CACR;;CAGD,MAAM,uBAAuBJ,KAAeK,KAAa;EACvD,MAAMH,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;EACR;EACD,MAAMC,eAAwC,CAAE;EAChD,KAAK,eAAe,eAAe,QAAQ,CAAC,UAAU;GACpD,aAAa,SAAS,SAAS;EAChC,EAAC;EACF,MAAMC,aAAsC,GACzC,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,aACJ,EAAC,CACH;EACD,MAAM,SAAS;IAAG,KAAK,sBAAsB;GAAK,GAAG;EAAY;AACjE,SAAO;CACR;;CAGD,MAAM,MACJd,QACAC,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,CAAC,aAAa,EAAE,KAAK,SAAS,WAAW,CAAC;EAE5D,MAAM,EAAE,CAAC,KAAK,WAAW,KAAM,GAAG,MAAM,GAAG;EAE3C,MAAM,cAAc;EAEpB,MAAM,gBAAgB,MAAM,KAAK,wBAC/B,YAAY,IACZ,KACD;EACD,IAAI,MAAM,MAAM,KAAK,SAAS,QAC5B,EAAE,GAAG,cAAe,GACpB,YAAY,SAAS,SAAS,CAC/B;EAED,MAAM,cAAc,CAAC,GAAI;AAEzB,OAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,KAAK,GAAG;GAC9C,MAAM,eAAe,MAAM,KAAK,uBAC9B,YAAY,IACZ,IACD;GACD,MAAM,SAAS;IAAE,GAAG;IAAc,GAAG;GAAM;GAC3C,MAAM,MAAM,KAAK,eAAe,QAC9B,EAAE,GAAG,OAAQ,GACb,YAAY,SAAS,SAAS,CAC/B;GACD,YAAY,KAAK,IAAI;EACtB;AAED,SAAO,GAAG,KAAK,YAAY,IAAK;CACjC;CAED,aAAa;AACX,SAAO;CACR;CAED,aAAa,YAAYe,MAAsC;EAC7D,MAAM,qBAAqB,KAAK;AAEhC,MAAI,CAAC,mBACH,OAAM,IAAI,MAAM;EAGlB,MAAM,gCAAgC,KAAK;AAE3C,MAAI,CAAC,8BACH,OAAM,IAAI,MAAM;AAGlB,SAAO,IAAI,qBAAqB;GAC9B,UAAU,MAAMb,2BAAS,YAAY,mBAAmB;GACxD,gBAAgB,MAAMA,2BAAS,YAAY,8BAA8B;EAC1E;CACF;CAED,YAA4C;AAC1C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,kBAAkB,KAAK,eAAe,WAAW;EAClD;CACF;AACF"}

package/dist/chains/combine_docs_chain.d.cts

@@ -1,126 +0,0 @@
-import { SerializedMapReduceDocumentsChain, SerializedRefineDocumentsChain, SerializedStuffDocumentsChain } from "./serde.cjs";
-import { BaseChain, ChainInputs } from "./base.cjs";
-import { LLMChain } from "./llm_chain.cjs";
-import { BasePromptValueInterface } from "../langchain-core/dist/prompt_values.cjs";
-import { ChainValues } from "@langchain/core/utils/types";
-import { BasePromptTemplate } from "@langchain/core/prompts";
-import { Document } from "@langchain/core/documents";
-import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
-
-//#region src/chains/combine_docs_chain.d.ts
-/**
- * Interface for the input properties of the StuffDocumentsChain class.
- */
-interface StuffDocumentsChainInput extends ChainInputs {
-  /** LLM Wrapper to use after formatting documents */
-  llmChain: LLMChain;
-  inputKey?: string;
-  /** Variable name in the LLM chain to put the documents in */
-  documentVariableName?: string;
-}
-/**
- * Chain that combines documents by stuffing into context.
- * @augments BaseChain
- * @augments StuffDocumentsChainInput
- */
-declare class StuffDocumentsChain extends BaseChain implements StuffDocumentsChainInput {
-  static lc_name(): string;
-  llmChain: LLMChain;
-  inputKey: string;
-  documentVariableName: string;
-  get inputKeys(): string[];
-  get outputKeys(): string[];
-  constructor(fields: StuffDocumentsChainInput);
-  /** @ignore */
-  _prepInputs(values: ChainValues): ChainValues;
-  /** @ignore */
-  _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
-  _chainType(): "stuff_documents_chain";
-  static deserialize(data: SerializedStuffDocumentsChain): Promise<StuffDocumentsChain>;
-  serialize(): SerializedStuffDocumentsChain;
-}
-/**
- * Interface for the input properties of the MapReduceDocumentsChain
- * class.
- */
-interface MapReduceDocumentsChainInput extends StuffDocumentsChainInput {
-  /** The maximum number of tokens before requiring to do the reduction */
-  maxTokens?: number;
-  /** The maximum number of iterations to run through the map */
-  maxIterations?: number;
-  /** Ensures that the map step is taken regardless of max tokens */
-  ensureMapStep?: boolean;
-  /** Chain to use to combine results of applying llm_chain to documents. */
-  combineDocumentChain: StuffDocumentsChain;
-  /** Return the results of the map steps in the output. */
-  returnIntermediateSteps?: boolean;
-}
-/**
- * Combine documents by mapping a chain over them, then combining results.
- * @augments BaseChain
- * @augments StuffDocumentsChainInput
- */
-declare class MapReduceDocumentsChain extends BaseChain implements MapReduceDocumentsChainInput {
-  static lc_name(): string;
-  llmChain: LLMChain;
-  inputKey: string;
-  documentVariableName: string;
-  returnIntermediateSteps: boolean;
-  get inputKeys(): string[];
-  get outputKeys(): string[];
-  maxTokens: number;
-  maxIterations: number;
-  ensureMapStep: boolean;
-  combineDocumentChain: StuffDocumentsChain;
-  constructor(fields: MapReduceDocumentsChainInput);
-  /** @ignore */
-  _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
-  _chainType(): "map_reduce_documents_chain";
-  static deserialize(data: SerializedMapReduceDocumentsChain): Promise<MapReduceDocumentsChain>;
-  serialize(): SerializedMapReduceDocumentsChain;
-}
-/**
- * Interface for the input properties of the RefineDocumentsChain class.
- */
-interface RefineDocumentsChainInput extends StuffDocumentsChainInput {
-  refineLLMChain: LLMChain;
-  documentPrompt?: BasePromptTemplate;
-  initialResponseName?: string;
-  documentVariableName?: string;
-  outputKey?: string;
-}
-/**
- * Combine documents by doing a first pass and then refining on more documents.
- * @augments BaseChain
- * @augments RefineDocumentsChainInput
- */
-declare class RefineDocumentsChain extends BaseChain implements RefineDocumentsChainInput {
-  static lc_name(): string;
-  llmChain: LLMChain;
-  inputKey: string;
-  outputKey: string;
-  documentVariableName: string;
-  initialResponseName: string;
-  refineLLMChain: LLMChain;
-  get defaultDocumentPrompt(): BasePromptTemplate;
-  documentPrompt: BasePromptTemplate<any, BasePromptValueInterface, any>;
-  get inputKeys(): string[];
-  get outputKeys(): string[];
-  constructor(fields: RefineDocumentsChainInput);
-  /** @ignore */
-  _constructInitialInputs(doc: Document, rest: Record<string, unknown>): Promise<{
-    [x: string]: unknown;
-  }>;
-  /** @ignore */
-  _constructRefineInputs(doc: Document, res: string): Promise<{
-    [x: string]: unknown;
-  }>;
-  /** @ignore */
-  _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
-  _chainType(): "refine_documents_chain";
-  static deserialize(data: SerializedRefineDocumentsChain): Promise<RefineDocumentsChain>;
-  serialize(): SerializedRefineDocumentsChain;
-}
-//#endregion
-export { MapReduceDocumentsChain, MapReduceDocumentsChainInput, RefineDocumentsChain, RefineDocumentsChainInput, StuffDocumentsChain, StuffDocumentsChainInput };
-//# sourceMappingURL=combine_docs_chain.d.cts.map

package/dist/chains/combine_docs_chain.d.cts.map

@@ -1 +0,0 @@
-{"version":3,"file":"combine_docs_chain.d.cts","names":["Document","ChainValues","CallbackManagerForChainRun","BasePromptTemplate","SerializedStuffDocumentsChain","SerializedMapReduceDocumentsChain","SerializedRefineDocumentsChain","BaseChain","ChainInputs","LLMChain","StuffDocumentsChainInput","StuffDocumentsChain","Promise","MapReduceDocumentsChainInput","MapReduceDocumentsChain","RefineDocumentsChainInput","RefineDocumentsChain","_________langchain_core_dist_prompt_values_js0","BasePromptValueInterface","Record"],"sources":["../../src/chains/combine_docs_chain.d.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport type { SerializedStuffDocumentsChain, SerializedMapReduceDocumentsChain, SerializedRefineDocumentsChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\n/**\n * Interface for the input properties of the StuffDocumentsChain class.\n */\nexport interface StuffDocumentsChainInput extends ChainInputs {\n    /** LLM Wrapper to use after formatting documents */\n    llmChain: LLMChain;\n    inputKey?: string;\n    /** Variable name in the LLM chain to put the documents in */\n    documentVariableName?: string;\n}\n/**\n * Chain that combines documents by stuffing into context.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport declare class StuffDocumentsChain extends BaseChain implements StuffDocumentsChainInput {\n    static lc_name(): string;\n    llmChain: LLMChain;\n    inputKey: string;\n    documentVariableName: string;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    constructor(fields: StuffDocumentsChainInput);\n    /** @ignore */\n    _prepInputs(values: ChainValues): ChainValues;\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"stuff_documents_chain\";\n    static deserialize(data: SerializedStuffDocumentsChain): Promise<StuffDocumentsChain>;\n    serialize(): SerializedStuffDocumentsChain;\n}\n/**\n * Interface for the input properties of the MapReduceDocumentsChain\n * class.\n */\nexport interface MapReduceDocumentsChainInput extends StuffDocumentsChainInput {\n    /** The maximum number of tokens before requiring to do the reduction */\n    maxTokens?: number;\n    /** The maximum number of iterations to run through the map */\n    maxIterations?: number;\n    /** Ensures that the map step is taken regardless of max tokens */\n    ensureMapStep?: boolean;\n    /** Chain to use to combine results of applying llm_chain to documents. */\n    combineDocumentChain: StuffDocumentsChain;\n    /** Return the results of the map steps in the output. */\n    returnIntermediateSteps?: boolean;\n}\n/**\n * Combine documents by mapping a chain over them, then combining results.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport declare class MapReduceDocumentsChain extends BaseChain implements MapReduceDocumentsChainInput {\n    static lc_name(): string;\n    llmChain: LLMChain;\n    inputKey: string;\n    documentVariableName: string;\n    returnIntermediateSteps: boolean;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    maxTokens: number;\n    maxIterations: number;\n    ensureMapStep: boolean;\n    combineDocumentChain: StuffDocumentsChain;\n    constructor(fields: MapReduceDocumentsChainInput);\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"map_reduce_documents_chain\";\n    static deserialize(data: SerializedMapReduceDocumentsChain): Promise<MapReduceDocumentsChain>;\n    serialize(): SerializedMapReduceDocumentsChain;\n}\n/**\n * Interface for the input properties of the RefineDocumentsChain class.\n */\nexport interface RefineDocumentsChainInput extends StuffDocumentsChainInput {\n    refineLLMChain: LLMChain;\n    documentPrompt?: BasePromptTemplate;\n    initialResponseName?: string;\n    documentVariableName?: string;\n    outputKey?: string;\n}\n/**\n * Combine documents by doing a first pass and then refining on more documents.\n * @augments BaseChain\n * @augments RefineDocumentsChainInput\n */\nexport declare class RefineDocumentsChain extends BaseChain implements RefineDocumentsChainInput {\n    static lc_name(): string;\n    llmChain: LLMChain;\n    inputKey: string;\n    outputKey: string;\n    documentVariableName: string;\n    initialResponseName: string;\n    refineLLMChain: LLMChain;\n    get defaultDocumentPrompt(): BasePromptTemplate;\n    documentPrompt: BasePromptTemplate<any, import(\"../../../langchain-core/dist/prompt_values.js\").BasePromptValueInterface, any>;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    constructor(fields: RefineDocumentsChainInput);\n    /** @ignore */\n    _constructInitialInputs(doc: Document, rest: Record<string, unknown>): Promise<{\n        [x: string]: unknown;\n    }>;\n    /** @ignore */\n    _constructRefineInputs(doc: Document, res: string): Promise<{\n        [x: string]: unknown;\n    }>;\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"refine_documents_chain\";\n    static deserialize(data: SerializedRefineDocumentsChain): Promise<RefineDocumentsChain>;\n    serialize(): SerializedRefineDocumentsChain;\n}\n"],"mappings":";;;;;;;;;;;;;AAUiBU,UAAAA,wBAAAA,SAAiCF,WAAT,CAAA;EAAA;EAAA,QAE3BC,EAAAA,QAAAA;EAAQ,QAF4BD,CAAAA,EAAAA,MAAAA;EAAW;EAYxCG,oBAAAA,CAAAA,EAAmB,MAAA;;;;;;;AAWIT,cAXvBS,mBAAAA,SAA4BJ,SAAAA,YAAqBG,wBAW1BR,CAAAA;EAA0B,OAAWD,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAAW,QAAnBW,EAT3DH,QAS2DG;EAAO,QAEnDR,EAAAA,MAAAA;EAA6B,oBAAWO,EAAAA,MAAAA;EAAmB,IAA3BC,SAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAO,IACnDR,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAA6B,WAdGG,CAAAA,MAAAA,EAOzBG,wBAPyBH;EAAS;EAAoC,WAAA,CAAA,MAAA,EAStEN,WATsE,CAAA,EASxDA,WATwD;EAoB7EY;EAA4B,KAAA,CAAA,MAAA,EAT3BZ,WAS2B,EAAA,UAAA,CAAA,EATDC,0BASC,CAAA,EAT4BU,OAS5B,CAToCX,WASpC,CAAA;EAAA,UAQnBU,CAAAA,CAAAA,EAAAA,uBAAAA;EAAmB,OARSD,WAAAA,CAAAA,IAAAA,EAPzBN,6BAOyBM,CAAAA,EAPOE,OAOPF,CAPeC,mBAOfD,CAAAA;EAAwB,SAAA,CAAA,CAAA,EAN7DN,6BAM6D;AAiB9E;;;;;AAckBH,UA/BDY,4BAAAA,SAAqCH,wBA+BpCT,CAAAA;EAAW;EAAyC,SAAWA,CAAAA,EAAAA,MAAAA;EAAW;EAAZ,aAEnDI,CAAAA,EAAAA,MAAAA;EAAiC;EAAkC,aAA/BO,CAAAA,EAAAA,OAAAA;EAAO;EACtB,oBAjBGL,EAT3BI,mBAS2BJ;EAAS;EAAwC,uBAAA,CAAA,EAAA,OAAA;AAsBtG;;;;;AAA2E;AAYtDS,cAlCAF,uBAAAA,SAAgCP,SAAAA,YAAqBM,4BAkCjC,CAAA;EAAA,OAAA,OAAA,CAAA,CAAA,EAAA,MAAA;EAAA,QAE3BJ,EAlCAA,QAkCAA;EAAQ,QAKFA,EAAAA,MAAAA;EAAQ,oBACKN,EAAAA,MAAAA;EAAkB,uBAAA,EAAA,OAAA;EACyE,IAAxGA,SAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAkB,IAGdY,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAyB,SAEhBf,EAAAA,MAAAA;EAAQ,aAAQmB,EAAAA,MAAAA;EAAM,aAAoBP,EAAAA,OAAAA;EAAO,oBAIlDZ,EAzCNW,mBAyCMX;EAAQ,WAAgBY,CAAAA,MAAAA,EAxChCC,4BAwCgCD;EAAO;EAIlC,KAAeV,CAAAA,MAAAA,EA1C1BD,WA0C0BC,EAAAA,UAAAA,CAAAA,EA1CAA,0BA0CAA,CAAAA,EA1C6BU,OA0C7BV,CA1CqCD,WA0CrCC,CAAAA;EAA0B,UAAWD,CAAAA,CAAAA,EAAAA,4BAAAA;EAAW,OAAnBW,WAAAA,CAAAA,IAAAA,EAxC5CP,iCAwC4CO,CAAAA,EAxCRA,OAwCQA,CAxCAE,uBAwCAF,CAAAA;EAAO,SAEnDN,CAAAA,CAAAA,EAzCZD,iCAyCYC;;;;;AAxB0CS,UAZtDA,yBAAAA,SAAkCL,wBAYoBK,CAAAA;EAAyB,cAAA,EAX5EN,QAW4E;mBAV3EN;;;;;;;;;;cAUAa,oBAAAA,SAA6BT,SAAAA,YAAqBQ;;YAEzDN;;;;;kBAKMA;+BACaN;kBACbA,wBAD+B,wBAAA;;;sBAI3BY;;+BAESf,gBAAgBmB,0BAA0BP;;;;8BAI3CZ,wBAAwBY;;;;gBAItCX,0BAA0BC,6BAA6BU,QAAQX;;2BAEpDK,iCAAiCM,QAAQI;eACrDV"}

package/dist/chains/combine_docs_chain.d.ts

@@ -1,126 +0,0 @@
-import { SerializedMapReduceDocumentsChain, SerializedRefineDocumentsChain, SerializedStuffDocumentsChain } from "./serde.js";
-import { BaseChain, ChainInputs } from "./base.js";
-import { LLMChain } from "./llm_chain.js";
-import { BasePromptValueInterface } from "../libs/langchain-core/dist/prompt_values.js";
-import { Document } from "@langchain/core/documents";
-import { ChainValues } from "@langchain/core/utils/types";
-import { BasePromptTemplate } from "@langchain/core/prompts";
-import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
-
-//#region src/chains/combine_docs_chain.d.ts
-/**
- * Interface for the input properties of the StuffDocumentsChain class.
- */
-interface StuffDocumentsChainInput extends ChainInputs {
-  /** LLM Wrapper to use after formatting documents */
-  llmChain: LLMChain;
-  inputKey?: string;
-  /** Variable name in the LLM chain to put the documents in */
-  documentVariableName?: string;
-}
-/**
- * Chain that combines documents by stuffing into context.
- * @augments BaseChain
- * @augments StuffDocumentsChainInput
- */
-declare class StuffDocumentsChain extends BaseChain implements StuffDocumentsChainInput {
-  static lc_name(): string;
-  llmChain: LLMChain;
-  inputKey: string;
-  documentVariableName: string;
-  get inputKeys(): string[];
-  get outputKeys(): string[];
-  constructor(fields: StuffDocumentsChainInput);
-  /** @ignore */
-  _prepInputs(values: ChainValues): ChainValues;
-  /** @ignore */
-  _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
-  _chainType(): "stuff_documents_chain";
-  static deserialize(data: SerializedStuffDocumentsChain): Promise<StuffDocumentsChain>;
-  serialize(): SerializedStuffDocumentsChain;
-}
-/**
- * Interface for the input properties of the MapReduceDocumentsChain
- * class.
- */
-interface MapReduceDocumentsChainInput extends StuffDocumentsChainInput {
-  /** The maximum number of tokens before requiring to do the reduction */
-  maxTokens?: number;
-  /** The maximum number of iterations to run through the map */
-  maxIterations?: number;
-  /** Ensures that the map step is taken regardless of max tokens */
-  ensureMapStep?: boolean;
-  /** Chain to use to combine results of applying llm_chain to documents. */
-  combineDocumentChain: StuffDocumentsChain;
-  /** Return the results of the map steps in the output. */
-  returnIntermediateSteps?: boolean;
-}
-/**
- * Combine documents by mapping a chain over them, then combining results.
- * @augments BaseChain
- * @augments StuffDocumentsChainInput
- */
-declare class MapReduceDocumentsChain extends BaseChain implements MapReduceDocumentsChainInput {
-  static lc_name(): string;
-  llmChain: LLMChain;
-  inputKey: string;
-  documentVariableName: string;
-  returnIntermediateSteps: boolean;
-  get inputKeys(): string[];
-  get outputKeys(): string[];
-  maxTokens: number;
-  maxIterations: number;
-  ensureMapStep: boolean;
-  combineDocumentChain: StuffDocumentsChain;
-  constructor(fields: MapReduceDocumentsChainInput);
-  /** @ignore */
-  _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
-  _chainType(): "map_reduce_documents_chain";
-  static deserialize(data: SerializedMapReduceDocumentsChain): Promise<MapReduceDocumentsChain>;
-  serialize(): SerializedMapReduceDocumentsChain;
-}
-/**
- * Interface for the input properties of the RefineDocumentsChain class.
- */
-interface RefineDocumentsChainInput extends StuffDocumentsChainInput {
-  refineLLMChain: LLMChain;
-  documentPrompt?: BasePromptTemplate;
-  initialResponseName?: string;
-  documentVariableName?: string;
-  outputKey?: string;
-}
-/**
- * Combine documents by doing a first pass and then refining on more documents.
- * @augments BaseChain
- * @augments RefineDocumentsChainInput
- */
-declare class RefineDocumentsChain extends BaseChain implements RefineDocumentsChainInput {
-  static lc_name(): string;
-  llmChain: LLMChain;
-  inputKey: string;
-  outputKey: string;
-  documentVariableName: string;
-  initialResponseName: string;
-  refineLLMChain: LLMChain;
-  get defaultDocumentPrompt(): BasePromptTemplate;
-  documentPrompt: BasePromptTemplate<any, BasePromptValueInterface, any>;
-  get inputKeys(): string[];
-  get outputKeys(): string[];
-  constructor(fields: RefineDocumentsChainInput);
-  /** @ignore */
-  _constructInitialInputs(doc: Document, rest: Record<string, unknown>): Promise<{
-    [x: string]: unknown;
-  }>;
-  /** @ignore */
-  _constructRefineInputs(doc: Document, res: string): Promise<{
-    [x: string]: unknown;
-  }>;
-  /** @ignore */
-  _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
-  _chainType(): "refine_documents_chain";
-  static deserialize(data: SerializedRefineDocumentsChain): Promise<RefineDocumentsChain>;
-  serialize(): SerializedRefineDocumentsChain;
-}
-//#endregion
-export { MapReduceDocumentsChain, MapReduceDocumentsChainInput, RefineDocumentsChain, RefineDocumentsChainInput, StuffDocumentsChain, StuffDocumentsChainInput };
-//# sourceMappingURL=combine_docs_chain.d.ts.map

package/dist/chains/combine_docs_chain.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"combine_docs_chain.d.ts","names":["Document","ChainValues","CallbackManagerForChainRun","BasePromptTemplate","SerializedStuffDocumentsChain","SerializedMapReduceDocumentsChain","SerializedRefineDocumentsChain","BaseChain","ChainInputs","LLMChain","StuffDocumentsChainInput","StuffDocumentsChain","Promise","MapReduceDocumentsChainInput","MapReduceDocumentsChain","RefineDocumentsChainInput","RefineDocumentsChain","_________langchain_core_dist_prompt_values_js0","BasePromptValueInterface","Record"],"sources":["../../src/chains/combine_docs_chain.d.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport type { SerializedStuffDocumentsChain, SerializedMapReduceDocumentsChain, SerializedRefineDocumentsChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\n/**\n * Interface for the input properties of the StuffDocumentsChain class.\n */\nexport interface StuffDocumentsChainInput extends ChainInputs {\n    /** LLM Wrapper to use after formatting documents */\n    llmChain: LLMChain;\n    inputKey?: string;\n    /** Variable name in the LLM chain to put the documents in */\n    documentVariableName?: string;\n}\n/**\n * Chain that combines documents by stuffing into context.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport declare class StuffDocumentsChain extends BaseChain implements StuffDocumentsChainInput {\n    static lc_name(): string;\n    llmChain: LLMChain;\n    inputKey: string;\n    documentVariableName: string;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    constructor(fields: StuffDocumentsChainInput);\n    /** @ignore */\n    _prepInputs(values: ChainValues): ChainValues;\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"stuff_documents_chain\";\n    static deserialize(data: SerializedStuffDocumentsChain): Promise<StuffDocumentsChain>;\n    serialize(): SerializedStuffDocumentsChain;\n}\n/**\n * Interface for the input properties of the MapReduceDocumentsChain\n * class.\n */\nexport interface MapReduceDocumentsChainInput extends StuffDocumentsChainInput {\n    /** The maximum number of tokens before requiring to do the reduction */\n    maxTokens?: number;\n    /** The maximum number of iterations to run through the map */\n    maxIterations?: number;\n    /** Ensures that the map step is taken regardless of max tokens */\n    ensureMapStep?: boolean;\n    /** Chain to use to combine results of applying llm_chain to documents. */\n    combineDocumentChain: StuffDocumentsChain;\n    /** Return the results of the map steps in the output. */\n    returnIntermediateSteps?: boolean;\n}\n/**\n * Combine documents by mapping a chain over them, then combining results.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport declare class MapReduceDocumentsChain extends BaseChain implements MapReduceDocumentsChainInput {\n    static lc_name(): string;\n    llmChain: LLMChain;\n    inputKey: string;\n    documentVariableName: string;\n    returnIntermediateSteps: boolean;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    maxTokens: number;\n    maxIterations: number;\n    ensureMapStep: boolean;\n    combineDocumentChain: StuffDocumentsChain;\n    constructor(fields: MapReduceDocumentsChainInput);\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"map_reduce_documents_chain\";\n    static deserialize(data: SerializedMapReduceDocumentsChain): Promise<MapReduceDocumentsChain>;\n    serialize(): SerializedMapReduceDocumentsChain;\n}\n/**\n * Interface for the input properties of the RefineDocumentsChain class.\n */\nexport interface RefineDocumentsChainInput extends StuffDocumentsChainInput {\n    refineLLMChain: LLMChain;\n    documentPrompt?: BasePromptTemplate;\n    initialResponseName?: string;\n    documentVariableName?: string;\n    outputKey?: string;\n}\n/**\n * Combine documents by doing a first pass and then refining on more documents.\n * @augments BaseChain\n * @augments RefineDocumentsChainInput\n */\nexport declare class RefineDocumentsChain extends BaseChain implements RefineDocumentsChainInput {\n    static lc_name(): string;\n    llmChain: LLMChain;\n    inputKey: string;\n    outputKey: string;\n    documentVariableName: string;\n    initialResponseName: string;\n    refineLLMChain: LLMChain;\n    get defaultDocumentPrompt(): BasePromptTemplate;\n    documentPrompt: BasePromptTemplate<any, import(\"../../../langchain-core/dist/prompt_values.js\").BasePromptValueInterface, any>;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    constructor(fields: RefineDocumentsChainInput);\n    /** @ignore */\n    _constructInitialInputs(doc: Document, rest: Record<string, unknown>): Promise<{\n        [x: string]: unknown;\n    }>;\n    /** @ignore */\n    _constructRefineInputs(doc: Document, res: string): Promise<{\n        [x: string]: unknown;\n    }>;\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"refine_documents_chain\";\n    static deserialize(data: SerializedRefineDocumentsChain): Promise<RefineDocumentsChain>;\n    serialize(): SerializedRefineDocumentsChain;\n}\n"],"mappings":";;;;;;;;;;;;;AAUiBU,UAAAA,wBAAAA,SAAiCF,WAAT,CAAA;EAAA;EAAA,QAE3BC,EAAAA,QAAAA;EAAQ,QAF4BD,CAAAA,EAAAA,MAAAA;EAAW;EAYxCG,oBAAAA,CAAAA,EAAmB,MAAA;;;;;;;AAWIT,cAXvBS,mBAAAA,SAA4BJ,SAAAA,YAAqBG,wBAW1BR,CAAAA;EAA0B,OAAWD,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAAW,QAAnBW,EAT3DH,QAS2DG;EAAO,QAEnDR,EAAAA,MAAAA;EAA6B,oBAAWO,EAAAA,MAAAA;EAAmB,IAA3BC,SAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAO,IACnDR,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAA6B,WAdGG,CAAAA,MAAAA,EAOzBG,wBAPyBH;EAAS;EAAoC,WAAA,CAAA,MAAA,EAStEN,WATsE,CAAA,EASxDA,WATwD;EAoB7EY;EAA4B,KAAA,CAAA,MAAA,EAT3BZ,WAS2B,EAAA,UAAA,CAAA,EATDC,0BASC,CAAA,EAT4BU,OAS5B,CAToCX,WASpC,CAAA;EAAA,UAQnBU,CAAAA,CAAAA,EAAAA,uBAAAA;EAAmB,OARSD,WAAAA,CAAAA,IAAAA,EAPzBN,6BAOyBM,CAAAA,EAPOE,OAOPF,CAPeC,mBAOfD,CAAAA;EAAwB,SAAA,CAAA,CAAA,EAN7DN,6BAM6D;AAiB9E;;;;;AAckBH,UA/BDY,4BAAAA,SAAqCH,wBA+BpCT,CAAAA;EAAW;EAAyC,SAAWA,CAAAA,EAAAA,MAAAA;EAAW;EAAZ,aAEnDI,CAAAA,EAAAA,MAAAA;EAAiC;EAAkC,aAA/BO,CAAAA,EAAAA,OAAAA;EAAO;EACtB,oBAjBGL,EAT3BI,mBAS2BJ;EAAS;EAAwC,uBAAA,CAAA,EAAA,OAAA;AAsBtG;;;;;AAA2E;AAYtDS,cAlCAF,uBAAAA,SAAgCP,SAAAA,YAAqBM,4BAkCjC,CAAA;EAAA,OAAA,OAAA,CAAA,CAAA,EAAA,MAAA;EAAA,QAE3BJ,EAlCAA,QAkCAA;EAAQ,QAKFA,EAAAA,MAAAA;EAAQ,oBACKN,EAAAA,MAAAA;EAAkB,uBAAA,EAAA,OAAA;EACyE,IAAxGA,SAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAkB,IAGdY,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAyB,SAEhBf,EAAAA,MAAAA;EAAQ,aAAQmB,EAAAA,MAAAA;EAAM,aAAoBP,EAAAA,OAAAA;EAAO,oBAIlDZ,EAzCNW,mBAyCMX;EAAQ,WAAgBY,CAAAA,MAAAA,EAxChCC,4BAwCgCD;EAAO;EAIlC,KAAeV,CAAAA,MAAAA,EA1C1BD,WA0C0BC,EAAAA,UAAAA,CAAAA,EA1CAA,0BA0CAA,CAAAA,EA1C6BU,OA0C7BV,CA1CqCD,WA0CrCC,CAAAA;EAA0B,UAAWD,CAAAA,CAAAA,EAAAA,4BAAAA;EAAW,OAAnBW,WAAAA,CAAAA,IAAAA,EAxC5CP,iCAwC4CO,CAAAA,EAxCRA,OAwCQA,CAxCAE,uBAwCAF,CAAAA;EAAO,SAEnDN,CAAAA,CAAAA,EAzCZD,iCAyCYC;;;;;AAxB0CS,UAZtDA,yBAAAA,SAAkCL,wBAYoBK,CAAAA;EAAyB,cAAA,EAX5EN,QAW4E;mBAV3EN;;;;;;;;;;cAUAa,oBAAAA,SAA6BT,SAAAA,YAAqBQ;;YAEzDN;;;;;kBAKMA;+BACaN;kBACbA,wBAD+B,wBAAA;;;sBAI3BY;;+BAESf,gBAAgBmB,0BAA0BP;;;;8BAI3CZ,wBAAwBY;;;;gBAItCX,0BAA0BC,6BAA6BU,QAAQX;;2BAEpDK,iCAAiCM,QAAQI;eACrDV"}