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

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"}

package/dist/chains/combine_docs_chain.js

@@ -1,272 +0,0 @@
-import { BaseChain } from "./base.js";
-import { LLMChain } from "./llm_chain.js";
-import { PromptTemplate } from "@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 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 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 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 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 BaseChain {
-	static lc_name() {
-		return "RefineDocumentsChain";
-	}
-	llmChain;
-	inputKey = "input_documents";
-	outputKey = "output_text";
-	documentVariableName = "context";
-	initialResponseName = "existing_answer";
-	refineLLMChain;
-	get defaultDocumentPrompt() {
-		return new 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 LLMChain.deserialize(SerializedLLMChain),
-			refineLLMChain: await LLMChain.deserialize(SerializedRefineDocumentChain)
-		});
-	}
-	serialize() {
-		return {
-			_type: this._chainType(),
-			llm_chain: this.llmChain.serialize(),
-			refine_llm_chain: this.refineLLMChain.serialize()
-		};
-	}
-};
-
-//#endregion
-export { MapReduceDocumentsChain, RefineDocumentsChain, StuffDocumentsChain };
-//# sourceMappingURL=combine_docs_chain.js.map

package/dist/chains/combine_docs_chain.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"combine_docs_chain.js","names":["fields: StuffDocumentsChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun","data: SerializedStuffDocumentsChain","fields: MapReduceDocumentsChainInput","intermediateSteps: string[]","i","data: SerializedMapReduceDocumentsChain","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,4BACH,UAEV;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,YAAYA,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,MAAM,SAAS,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,gCACH,UAEV;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,YAAYC,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,MACJH,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,IAAIG,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,MAAM,SAAS,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,6BACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA,WAAW;CAEX,YAAY;CAEZ,uBAAuB;CAEvB,sBAAsB;CAEtB;CAEA,IAAI,wBAA4C;AAC9C,SAAO,IAAI,eAAe;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,MACJZ,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,YAAYa,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,MAAM,SAAS,YAAY,mBAAmB;GACxD,gBAAgB,MAAM,SAAS,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_documents/base.cjs

@@ -1,22 +0,0 @@
-const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
-const __langchain_core_prompts = require_rolldown_runtime.__toESM(require("@langchain/core/prompts"));
-
-//#region src/chains/combine_documents/base.ts
-const DEFAULT_DOCUMENT_SEPARATOR = "\n\n";
-const DOCUMENTS_KEY = "context";
-const DEFAULT_DOCUMENT_PROMPT = /* @__PURE__ */ __langchain_core_prompts.PromptTemplate.fromTemplate("{page_content}");
-async function formatDocuments({ documentPrompt, documentSeparator, documents, config }) {
-	if (documents == null || documents.length === 0) return "";
-	const formattedDocs = await Promise.all(documents.map((document) => documentPrompt.withConfig({ runName: "document_formatter" }).invoke({
-		...document.metadata,
-		page_content: document.pageContent
-	}, config)));
-	return formattedDocs.join(documentSeparator);
-}
-
-//#endregion
-exports.DEFAULT_DOCUMENT_PROMPT = DEFAULT_DOCUMENT_PROMPT;
-exports.DEFAULT_DOCUMENT_SEPARATOR = DEFAULT_DOCUMENT_SEPARATOR;
-exports.DOCUMENTS_KEY = DOCUMENTS_KEY;
-exports.formatDocuments = formatDocuments;
-//# sourceMappingURL=base.cjs.map

package/dist/chains/combine_documents/base.cjs.map

@@ -1 +0,0 @@
-{"version":3,"file":"base.cjs","names":["PromptTemplate"],"sources":["../../../src/chains/combine_documents/base.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport { RunnableConfig } from \"@langchain/core/runnables\";\n\nexport const DEFAULT_DOCUMENT_SEPARATOR = \"\\n\\n\";\n\nexport const DOCUMENTS_KEY = \"context\";\nexport const INTERMEDIATE_STEPS_KEY = \"intermediate_steps\";\n\nexport const DEFAULT_DOCUMENT_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(\"{page_content}\");\n\nexport async function formatDocuments({\n  documentPrompt,\n  documentSeparator,\n  documents,\n  config,\n}: {\n  documentPrompt: BasePromptTemplate;\n  documentSeparator: string;\n  documents: Document[];\n  config?: RunnableConfig;\n}) {\n  if (documents == null || documents.length === 0) {\n    return \"\";\n  }\n  const formattedDocs = await Promise.all(\n    documents.map((document) =>\n      documentPrompt\n        .withConfig({ runName: \"document_formatter\" })\n        .invoke(\n          { ...document.metadata, page_content: document.pageContent },\n          config\n        )\n    )\n  );\n  return formattedDocs.join(documentSeparator);\n}\n"],"mappings":";;;;AAIA,MAAa,6BAA6B;AAE1C,MAAa,gBAAgB;AAG7B,MAAa,0CACKA,wCAAe,aAAa,iBAAiB;AAE/D,eAAsB,gBAAgB,EACpC,gBACA,mBACA,WACA,QAMD,EAAE;AACD,KAAI,aAAa,QAAQ,UAAU,WAAW,EAC5C,QAAO;CAET,MAAM,gBAAgB,MAAM,QAAQ,IAClC,UAAU,IAAI,CAAC,aACb,eACG,WAAW,EAAE,SAAS,qBAAsB,EAAC,CAC7C,OACC;EAAE,GAAG,SAAS;EAAU,cAAc,SAAS;CAAa,GAC5D,OACD,CACJ,CACF;AACD,QAAO,cAAc,KAAK,kBAAkB;AAC7C"}

package/dist/chains/combine_documents/base.js

@@ -1,18 +0,0 @@
-import { PromptTemplate } from "@langchain/core/prompts";
-
-//#region src/chains/combine_documents/base.ts
-const DEFAULT_DOCUMENT_SEPARATOR = "\n\n";
-const DOCUMENTS_KEY = "context";
-const DEFAULT_DOCUMENT_PROMPT = /* @__PURE__ */ PromptTemplate.fromTemplate("{page_content}");
-async function formatDocuments({ documentPrompt, documentSeparator, documents, config }) {
-	if (documents == null || documents.length === 0) return "";
-	const formattedDocs = await Promise.all(documents.map((document) => documentPrompt.withConfig({ runName: "document_formatter" }).invoke({
-		...document.metadata,
-		page_content: document.pageContent
-	}, config)));
-	return formattedDocs.join(documentSeparator);
-}
-
-//#endregion
-export { DEFAULT_DOCUMENT_PROMPT, DEFAULT_DOCUMENT_SEPARATOR, DOCUMENTS_KEY, formatDocuments };
-//# sourceMappingURL=base.js.map

package/dist/chains/combine_documents/base.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"base.js","names":[],"sources":["../../../src/chains/combine_documents/base.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport { RunnableConfig } from \"@langchain/core/runnables\";\n\nexport const DEFAULT_DOCUMENT_SEPARATOR = \"\\n\\n\";\n\nexport const DOCUMENTS_KEY = \"context\";\nexport const INTERMEDIATE_STEPS_KEY = \"intermediate_steps\";\n\nexport const DEFAULT_DOCUMENT_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(\"{page_content}\");\n\nexport async function formatDocuments({\n  documentPrompt,\n  documentSeparator,\n  documents,\n  config,\n}: {\n  documentPrompt: BasePromptTemplate;\n  documentSeparator: string;\n  documents: Document[];\n  config?: RunnableConfig;\n}) {\n  if (documents == null || documents.length === 0) {\n    return \"\";\n  }\n  const formattedDocs = await Promise.all(\n    documents.map((document) =>\n      documentPrompt\n        .withConfig({ runName: \"document_formatter\" })\n        .invoke(\n          { ...document.metadata, page_content: document.pageContent },\n          config\n        )\n    )\n  );\n  return formattedDocs.join(documentSeparator);\n}\n"],"mappings":";;;AAIA,MAAa,6BAA6B;AAE1C,MAAa,gBAAgB;AAG7B,MAAa,0CACK,eAAe,aAAa,iBAAiB;AAE/D,eAAsB,gBAAgB,EACpC,gBACA,mBACA,WACA,QAMD,EAAE;AACD,KAAI,aAAa,QAAQ,UAAU,WAAW,EAC5C,QAAO;CAET,MAAM,gBAAgB,MAAM,QAAQ,IAClC,UAAU,IAAI,CAAC,aACb,eACG,WAAW,EAAE,SAAS,qBAAsB,EAAC,CAC7C,OACC;EAAE,GAAG,SAAS;EAAU,cAAc,SAAS;CAAa,GAC5D,OACD,CACJ,CACF;AACD,QAAO,cAAc,KAAK,kBAAkB;AAC7C"}

package/dist/chains/combine_documents/index.cjs

@@ -1,16 +0,0 @@
-const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
-const require_stuff = require('./stuff.cjs');
-
-//#region src/chains/combine_documents/index.ts
-var combine_documents_exports = {};
-require_rolldown_runtime.__export(combine_documents_exports, { createStuffDocumentsChain: () => require_stuff.createStuffDocumentsChain });
-
-//#endregion
-Object.defineProperty(exports, 'combine_documents_exports', {
-  enumerable: true,
-  get: function () {
-    return combine_documents_exports;
-  }
-});
-exports.createStuffDocumentsChain = require_stuff.createStuffDocumentsChain;
-//# sourceMappingURL=index.cjs.map

package/dist/chains/combine_documents/index.cjs.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.cjs","names":[],"sources":["../../../src/chains/combine_documents/index.ts"],"sourcesContent":["export { createStuffDocumentsChain } from \"./stuff.js\";\n"],"mappings":""}

package/dist/chains/combine_documents/index.d.cts

@@ -1,2 +0,0 @@
-import { createStuffDocumentsChain } from "./stuff.cjs";
-export { createStuffDocumentsChain };

package/dist/chains/combine_documents/index.d.ts

@@ -1,2 +0,0 @@
-import { createStuffDocumentsChain } from "./stuff.js";
-export { createStuffDocumentsChain };

package/dist/chains/combine_documents/index.js

@@ -1,10 +0,0 @@
-import { __export } from "../../_virtual/rolldown_runtime.js";
-import { createStuffDocumentsChain } from "./stuff.js";
-
-//#region src/chains/combine_documents/index.ts
-var combine_documents_exports = {};
-__export(combine_documents_exports, { createStuffDocumentsChain: () => createStuffDocumentsChain });
-
-//#endregion
-export { combine_documents_exports, createStuffDocumentsChain };
-//# sourceMappingURL=index.js.map

package/dist/chains/combine_documents/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","names":[],"sources":["../../../src/chains/combine_documents/index.ts"],"sourcesContent":["export { createStuffDocumentsChain } from \"./stuff.js\";\n"],"mappings":""}

package/dist/chains/combine_documents/reduce.cjs

@@ -1,73 +0,0 @@
-const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
-
-//#region src/chains/combine_documents/reduce.ts
-var reduce_exports = {};
-require_rolldown_runtime.__export(reduce_exports, {
-	collapseDocs: () => collapseDocs,
-	splitListOfDocs: () => splitListOfDocs
-});
-/**
-* Splits a list of documents into sublists based on a maximum token limit.
-*
-* @param {Document[]} docs - The list of documents to be split.
-* @param {Function} lengthFunc - A function that calculates the number of tokens in a list of documents.
-* @param {number} tokenMax - The maximum number of tokens allowed in a sublist.
-*
-* @returns {Document[][]} - A list of document sublists, each sublist contains documents whose total number of tokens does not exceed the tokenMax.
-*
-* @throws {Error} - Throws an error if a single document has more tokens than the tokenMax.
-*/
-function splitListOfDocs(docs, lengthFunc, tokenMax) {
-	const newResultDocList = [];
-	let subResultDocs = [];
-	for (const doc of docs) {
-		subResultDocs.push(doc);
-		const numTokens = lengthFunc(subResultDocs);
-		if (numTokens > tokenMax) {
-			if (subResultDocs.length === 1) throw new Error("A single document was longer than the context length, we cannot handle this.");
-			newResultDocList.push(subResultDocs.slice(0, -1));
-			subResultDocs = subResultDocs.slice(-1);
-		}
-	}
-	newResultDocList.push(subResultDocs);
-	return newResultDocList;
-}
-/**
-* Collapses a list of documents into a single document.
-*
-* This function takes a list of documents and a function to combine the content of these documents.
-* It combines the content of the documents using the provided function and merges the metadata of all documents.
-* If a metadata key is present in multiple documents, the values are concatenated with a comma separator.
-*
-* @param {Document[]} docs - The list of documents to be collapsed.
-* @param {Function} combineDocumentFunc - A function that combines the content of a list of documents into a single string. This function should return a promise that resolves to the combined string.
-*
-* @returns {Promise<Document>} - A promise that resolves to a single document with combined content and merged metadata.
-*
-* @throws {Error} - Throws an error if the combineDocumentFunc does not return a promise or if the promise does not resolve to a string.
-*/
-async function collapseDocs(docs, combineDocumentFunc) {
-	const result = await combineDocumentFunc(docs);
-	return {
-		pageContent: result,
-		metadata: collapseDocsMetadata(docs)
-	};
-}
-function collapseDocsMetadata(docs) {
-	const combinedMetadata = {};
-	for (const key in docs[0].metadata) if (key in docs[0].metadata) combinedMetadata[key] = String(docs[0].metadata[key]);
-	for (const doc of docs.slice(1)) for (const key in doc.metadata) if (key in combinedMetadata) combinedMetadata[key] += `, ${doc.metadata[key]}`;
-	else combinedMetadata[key] = String(doc.metadata[key]);
-	return combinedMetadata;
-}
-
-//#endregion
-exports.collapseDocs = collapseDocs;
-Object.defineProperty(exports, 'reduce_exports', {
-  enumerable: true,
-  get: function () {
-    return reduce_exports;
-  }
-});
-exports.splitListOfDocs = splitListOfDocs;
-//# sourceMappingURL=reduce.cjs.map

package/dist/chains/combine_documents/reduce.cjs.map

@@ -1 +0,0 @@
-{"version":3,"file":"reduce.cjs","names":["docs: Document[]","lengthFunc: (...args: any[]) => any","tokenMax: number","newResultDocList: Document[][]","subResultDocs: Document[]","combineDocumentFunc: (docs: Document[]) => Promise<string>","combinedMetadata: Record<string, string>"],"sources":["../../../src/chains/combine_documents/reduce.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\n\n/**\n * Splits a list of documents into sublists based on a maximum token limit.\n *\n * @param {Document[]} docs - The list of documents to be split.\n * @param {Function} lengthFunc - A function that calculates the number of tokens in a list of documents.\n * @param {number} tokenMax - The maximum number of tokens allowed in a sublist.\n *\n * @returns {Document[][]} - A list of document sublists, each sublist contains documents whose total number of tokens does not exceed the tokenMax.\n *\n * @throws {Error} - Throws an error if a single document has more tokens than the tokenMax.\n */\nexport function splitListOfDocs(\n  docs: Document[],\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  lengthFunc: (...args: any[]) => any,\n  tokenMax: number\n): Document[][] {\n  const newResultDocList: Document[][] = [];\n  let subResultDocs: Document[] = [];\n  for (const doc of docs) {\n    subResultDocs.push(doc);\n    const numTokens = lengthFunc(subResultDocs);\n    if (numTokens > tokenMax) {\n      if (subResultDocs.length === 1) {\n        throw new Error(\n          \"A single document was longer than the context length, we cannot handle this.\"\n        );\n      }\n      newResultDocList.push(subResultDocs.slice(0, -1));\n      subResultDocs = subResultDocs.slice(-1);\n    }\n  }\n  newResultDocList.push(subResultDocs);\n  return newResultDocList;\n}\n\n/**\n * Collapses a list of documents into a single document.\n *\n * This function takes a list of documents and a function to combine the content of these documents.\n * It combines the content of the documents using the provided function and merges the metadata of all documents.\n * If a metadata key is present in multiple documents, the values are concatenated with a comma separator.\n *\n * @param {Document[]} docs - The list of documents to be collapsed.\n * @param {Function} combineDocumentFunc - A function that combines the content of a list of documents into a single string. This function should return a promise that resolves to the combined string.\n *\n * @returns {Promise<Document>} - A promise that resolves to a single document with combined content and merged metadata.\n *\n * @throws {Error} - Throws an error if the combineDocumentFunc does not return a promise or if the promise does not resolve to a string.\n */\nexport async function collapseDocs(\n  docs: Document[],\n  combineDocumentFunc: (docs: Document[]) => Promise<string>\n): Promise<Document> {\n  const result = await combineDocumentFunc(docs);\n  return { pageContent: result, metadata: collapseDocsMetadata(docs) };\n}\n\nfunction collapseDocsMetadata(docs: Document[]): Document[\"metadata\"] {\n  const combinedMetadata: Record<string, string> = {};\n  for (const key in docs[0].metadata) {\n    if (key in docs[0].metadata) {\n      combinedMetadata[key] = String(docs[0].metadata[key]);\n    }\n  }\n  for (const doc of docs.slice(1)) {\n    for (const key in doc.metadata) {\n      if (key in combinedMetadata) {\n        combinedMetadata[key] += `, ${doc.metadata[key]}`;\n      } else {\n        combinedMetadata[key] = String(doc.metadata[key]);\n      }\n    }\n  }\n  return combinedMetadata;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAaA,SAAgB,gBACdA,MAEAC,YACAC,UACc;CACd,MAAMC,mBAAiC,CAAE;CACzC,IAAIC,gBAA4B,CAAE;AAClC,MAAK,MAAM,OAAO,MAAM;EACtB,cAAc,KAAK,IAAI;EACvB,MAAM,YAAY,WAAW,cAAc;AAC3C,MAAI,YAAY,UAAU;AACxB,OAAI,cAAc,WAAW,EAC3B,OAAM,IAAI,MACR;GAGJ,iBAAiB,KAAK,cAAc,MAAM,GAAG,GAAG,CAAC;GACjD,gBAAgB,cAAc,MAAM,GAAG;EACxC;CACF;CACD,iBAAiB,KAAK,cAAc;AACpC,QAAO;AACR;;;;;;;;;;;;;;;AAgBD,eAAsB,aACpBJ,MACAK,qBACmB;CACnB,MAAM,SAAS,MAAM,oBAAoB,KAAK;AAC9C,QAAO;EAAE,aAAa;EAAQ,UAAU,qBAAqB,KAAK;CAAE;AACrE;AAED,SAAS,qBAAqBL,MAAwC;CACpE,MAAMM,mBAA2C,CAAE;AACnD,MAAK,MAAM,OAAO,KAAK,GAAG,SACxB,KAAI,OAAO,KAAK,GAAG,UACjB,iBAAiB,OAAO,OAAO,KAAK,GAAG,SAAS,KAAK;AAGzD,MAAK,MAAM,OAAO,KAAK,MAAM,EAAE,CAC7B,MAAK,MAAM,OAAO,IAAI,SACpB,KAAI,OAAO,kBACT,iBAAiB,QAAQ,CAAC,EAAE,EAAE,IAAI,SAAS,MAAM;MAEjD,iBAAiB,OAAO,OAAO,IAAI,SAAS,KAAK;AAIvD,QAAO;AACR"}