@langchain/classic 1.0.0-alpha.1

package/dist/chains/combine_docs_chain.d.cts

@@ -0,0 +1,126 @@
+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 { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
+import { Document } from "@langchain/core/documents";
+
+//#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

@@ -0,0 +1 @@
+{"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

@@ -0,0 +1,126 @@
+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 { BasePromptTemplate } from "@langchain/core/prompts";
+import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
+import { ChainValues } from "@langchain/core/utils/types";
+
+//#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

@@ -0,0 +1 @@
+{"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

@@ -0,0 +1,272 @@
+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

@@ -0,0 +1 @@
+{"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

@@ -0,0 +1,22 @@
+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

@@ -0,0 +1 @@
+{"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

@@ -0,0 +1,18 @@
+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

@@ -0,0 +1 @@
+{"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"}