@langchain/classic 1.0.39 → 1.0.41

package/dist/chains/chat_vector_db_chain.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"chat_vector_db_chain.cjs","names":["BaseChain","LLMChain","PromptTemplate","loadQAStuffChain"],"sources":["../../src/chains/chat_vector_db_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { loadQAStuffChain } from \"./question_answering/load.js\";\n\n// oxlint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\nconst qa_template = `Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\n\n/**\n * Interface for the input parameters of the ChatVectorDBQAChain class.\n */\nexport interface ChatVectorDBQAChainInput extends ChainInputs {\n  vectorstore: VectorStoreInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  outputKey?: string;\n  inputKey?: string;\n  k?: number;\n}\n\n/** @deprecated use `ConversationalRetrievalQAChain` instead. */\nexport class ChatVectorDBQAChain\n  extends BaseChain\n  implements ChatVectorDBQAChainInput\n{\n  k = 4;\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  outputKey = \"result\";\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  vectorstore: VectorStoreInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: ChatVectorDBQAChainInput) {\n    super(fields);\n    this.vectorstore = fields.vectorstore;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.k = fields.k ?? this.k;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`chat history key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string = values[this.chatHistoryKey];\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      console.log(\"_call\", values, keys);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.vectorstore.similaritySearch(\n      newQuestion,\n      this.k,\n      undefined,\n      runManager?.getChild(\"vectorstore\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    const result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      return {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"chat-vector-db\" as const;\n  }\n\n  static async deserialize(\n    data: SerializedChatVectorDBQAChain,\n    values: LoadValues\n  ) {\n    if (!(\"vectorstore\" in values)) {\n      throw new Error(\n        `Need to pass in a vectorstore to deserialize VectorDBQAChain`\n      );\n    }\n    const { vectorstore } = values;\n\n    return new ChatVectorDBQAChain({\n      combineDocumentsChain: await BaseChain.deserialize(\n        data.combine_documents_chain\n      ),\n      questionGeneratorChain: await LLMChain.deserialize(\n        data.question_generator\n      ),\n      k: data.k,\n      vectorstore,\n    });\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    return {\n      _type: this._chainType(),\n      combine_documents_chain: this.combineDocumentsChain.serialize(),\n      question_generator: this.questionGeneratorChain.serialize(),\n      k: this.k,\n    };\n  }\n\n  /**\n   * Creates an instance of ChatVectorDBQAChain using a BaseLanguageModel\n   * and other options.\n   * @param llm Instance of BaseLanguageModel used to generate a new question.\n   * @param vectorstore Instance of VectorStore used for vector operations.\n   * @param options (Optional) Additional options for creating the ChatVectorDBQAChain instance.\n   * @returns New instance of ChatVectorDBQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    vectorstore: VectorStoreInterface,\n    options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    } = {}\n  ): ChatVectorDBQAChain {\n    const { questionGeneratorTemplate, qaTemplate, verbose, ...rest } = options;\n    const question_generator_prompt = PromptTemplate.fromTemplate(\n      questionGeneratorTemplate || question_generator_template\n    );\n    const qa_prompt = PromptTemplate.fromTemplate(qaTemplate || qa_template);\n\n    const qaChain = loadQAStuffChain(llm, { prompt: qa_prompt, verbose });\n    const questionGeneratorChain = new LLMChain({\n      prompt: question_generator_prompt,\n      llm,\n      verbose,\n    });\n    const instance = new this({\n      vectorstore,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;;AAaA,MAAM,8BAA8B;;;;;;AAOpC,MAAM,cAAc;;;;;;;AAqBpB,IAAa,sBAAb,MAAa,4BACHA,aAAAA,UAEV;CACE,IAAI;CAEJ,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,KAAK,eAAe;;CAG7C,YAAY;CAEZ,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,cAAc,OAAO;AAC1B,OAAK,wBAAwB,OAAO;AACpC,OAAK,yBAAyB,OAAO;AACrC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,IAAI,OAAO,KAAK,KAAK;AAC1B,OAAK,wBACH,OAAO,yBAAyB,KAAK;;;CAIzC,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;AAE7D,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,oBAAoB,KAAK,SAAS,aAAa;EAEjE,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,cAAsB,OAAO,KAAK;EACxC,IAAI,cAAc;AAClB,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,SAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;IACf,EACD,YAAY,SAAS,qBAAqB,CAC3C;GACD,MAAM,OAAO,OAAO,KAAK,OAAO;AAChC,WAAQ,IAAI,SAAS,QAAQ,KAAK;AAClC,OAAI,KAAK,WAAW,EAClB,eAAc,OAAO,KAAK;OAE1B,OAAM,IAAI,MACR,2EACD;;EAGL,MAAM,OAAO,MAAM,KAAK,YAAY,iBAClC,aACA,KAAK,GACL,KAAA,GACA,YAAY,SAAS,cAAc,CACpC;EACD,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;GACf;EACD,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,sBACP,QAAO;GACL,GAAG;GACH,iBAAiB;GAClB;AAEH,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YACX,MACA,QACA;AACA,MAAI,EAAE,iBAAiB,QACrB,OAAM,IAAI,MACR,+DACD;EAEH,MAAM,EAAE,gBAAgB;AAExB,SAAO,IAAI,oBAAoB;GAC7B,uBAAuB,MAAMA,aAAAA,UAAU,YACrC,KAAK,wBACN;GACD,wBAAwB,MAAMC,kBAAAA,SAAS,YACrC,KAAK,mBACN;GACD,GAAG,KAAK;GACR;GACD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,yBAAyB,KAAK,sBAAsB,WAAW;GAC/D,oBAAoB,KAAK,uBAAuB,WAAW;GAC3D,GAAG,KAAK;GACT;;;;;;;;;;CAWH,OAAO,QACL,KACA,aACA,UAQI,EAAE,EACe;EACrB,MAAM,EAAE,2BAA2B,YAAY,SAAS,GAAG,SAAS;EACpE,MAAM,4BAA4BC,wBAAAA,eAAe,aAC/C,6BAA6B,4BAC9B;EAGD,MAAM,UAAUC,aAAAA,iBAAiB,KAAK;GAAE,QAFtBD,wBAAAA,eAAe,aAAa,cAAc,YAEH;GAAE;GAAS,CAAC;EACrE,MAAM,yBAAyB,IAAID,kBAAAA,SAAS;GAC1C,QAAQ;GACR;GACA;GACD,CAAC;AAOF,SAAO,IANc,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA,GAAG;GACJ,CACc"}
+{"version":3,"file":"chat_vector_db_chain.cjs","names":["BaseChain","LLMChain","PromptTemplate","loadQAStuffChain"],"sources":["../../src/chains/chat_vector_db_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { loadQAStuffChain } from \"./question_answering/load.js\";\n\n// oxlint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\nconst qa_template = `Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\n\n/**\n * Interface for the input parameters of the ChatVectorDBQAChain class.\n */\nexport interface ChatVectorDBQAChainInput extends ChainInputs {\n  vectorstore: VectorStoreInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  outputKey?: string;\n  inputKey?: string;\n  k?: number;\n}\n\n/** @deprecated use `ConversationalRetrievalQAChain` instead. */\nexport class ChatVectorDBQAChain\n  extends BaseChain\n  implements ChatVectorDBQAChainInput\n{\n  k = 4;\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  outputKey = \"result\";\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  vectorstore: VectorStoreInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: ChatVectorDBQAChainInput) {\n    super(fields);\n    this.vectorstore = fields.vectorstore;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.k = fields.k ?? this.k;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`chat history key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string = values[this.chatHistoryKey];\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      console.log(\"_call\", values, keys);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.vectorstore.similaritySearch(\n      newQuestion,\n      this.k,\n      undefined,\n      runManager?.getChild(\"vectorstore\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    const result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      return {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"chat-vector-db\" as const;\n  }\n\n  static async deserialize(\n    data: SerializedChatVectorDBQAChain,\n    values: LoadValues\n  ) {\n    if (!(\"vectorstore\" in values)) {\n      throw new Error(\n        `Need to pass in a vectorstore to deserialize VectorDBQAChain`\n      );\n    }\n    const { vectorstore } = values;\n\n    return new ChatVectorDBQAChain({\n      combineDocumentsChain: await BaseChain.deserialize(\n        data.combine_documents_chain\n      ),\n      questionGeneratorChain: await LLMChain.deserialize(\n        data.question_generator\n      ),\n      k: data.k,\n      vectorstore,\n    });\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    return {\n      _type: this._chainType(),\n      combine_documents_chain: this.combineDocumentsChain.serialize(),\n      question_generator: this.questionGeneratorChain.serialize(),\n      k: this.k,\n    };\n  }\n\n  /**\n   * Creates an instance of ChatVectorDBQAChain using a BaseLanguageModel\n   * and other options.\n   * @param llm Instance of BaseLanguageModel used to generate a new question.\n   * @param vectorstore Instance of VectorStore used for vector operations.\n   * @param options (Optional) Additional options for creating the ChatVectorDBQAChain instance.\n   * @returns New instance of ChatVectorDBQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    vectorstore: VectorStoreInterface,\n    options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    } = {}\n  ): ChatVectorDBQAChain {\n    const { questionGeneratorTemplate, qaTemplate, verbose, ...rest } = options;\n    const question_generator_prompt = PromptTemplate.fromTemplate(\n      questionGeneratorTemplate || question_generator_template\n    );\n    const qa_prompt = PromptTemplate.fromTemplate(qaTemplate || qa_template);\n\n    const qaChain = loadQAStuffChain(llm, { prompt: qa_prompt, verbose });\n    const questionGeneratorChain = new LLMChain({\n      prompt: question_generator_prompt,\n      llm,\n      verbose,\n    });\n    const instance = new this({\n      vectorstore,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;AAaA,MAAM,8BAA8B;;;;;;AAOpC,MAAM,cAAc;;;;;;;AAqBpB,IAAa,sBAAb,MAAa,4BACHA,aAAAA,UAEV;CACE,IAAI;CAEJ,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,UAAU,KAAK,cAAc;CAC5C;CAEA,YAAY;CAEZ,IAAI,aAAa;EACf,OAAO,CAAC,KAAK,SAAS;CACxB;CAEA;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAAkC;EAC5C,MAAM,MAAM;EACZ,KAAK,cAAc,OAAO;EAC1B,KAAK,wBAAwB,OAAO;EACpC,KAAK,yBAAyB,OAAO;EACrC,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,IAAI,OAAO,KAAK,KAAK;EAC1B,KAAK,wBACH,OAAO,yBAAyB,KAAK;CACzC;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,YAAY;EAE5D,IAAI,EAAE,KAAK,kBAAkB,SAC3B,MAAM,IAAI,MAAM,oBAAoB,KAAK,SAAS,YAAY;EAEhE,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,cAAsB,OAAO,KAAK;EACxC,IAAI,cAAc;EAClB,IAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,SAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;GAChB,GACA,YAAY,SAAS,oBAAoB,CAC3C;GACA,MAAM,OAAO,OAAO,KAAK,MAAM;GAC/B,QAAQ,IAAI,SAAS,QAAQ,IAAI;GACjC,IAAI,KAAK,WAAW,GAClB,cAAc,OAAO,KAAK;QAE1B,MAAM,IAAI,MACR,0EACF;EAEJ;EACA,MAAM,OAAO,MAAM,KAAK,YAAY,iBAClC,aACA,KAAK,GACL,KAAA,GACA,YAAY,SAAS,aAAa,CACpC;EACA,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;EAChB;EACA,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,mBAAmB,CAC1C;EACA,IAAI,KAAK,uBACP,OAAO;GACL,GAAG;GACH,iBAAiB;EACnB;EAEF,OAAO;CACT;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YACX,MACA,QACA;EACA,IAAI,EAAE,iBAAiB,SACrB,MAAM,IAAI,MACR,8DACF;EAEF,MAAM,EAAE,gBAAgB;EAExB,OAAO,IAAI,oBAAoB;GAC7B,uBAAuB,MAAMA,aAAAA,UAAU,YACrC,KAAK,uBACP;GACA,wBAAwB,MAAMC,kBAAAA,SAAS,YACrC,KAAK,kBACP;GACA,GAAG,KAAK;GACR;EACF,CAAC;CACH;CAEA,YAA2C;EACzC,OAAO;GACL,OAAO,KAAK,WAAW;GACvB,yBAAyB,KAAK,sBAAsB,UAAU;GAC9D,oBAAoB,KAAK,uBAAuB,UAAU;GAC1D,GAAG,KAAK;EACV;CACF;;;;;;;;;CAUA,OAAO,QACL,KACA,aACA,UAQI,CAAC,GACgB;EACrB,MAAM,EAAE,2BAA2B,YAAY,SAAS,GAAG,SAAS;EACpE,MAAM,4BAA4BC,wBAAAA,eAAe,aAC/C,6BAA6B,2BAC/B;EAGA,MAAM,UAAUC,aAAAA,iBAAiB,KAAK;GAAE,QAFtBD,wBAAAA,eAAe,aAAa,cAAc,WAEJ;GAAG;EAAQ,CAAC;EACpE,MAAM,yBAAyB,IAAID,kBAAAA,SAAS;GAC1C,QAAQ;GACR;GACA;EACF,CAAC;EAOD,OAAO,IANc,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA,GAAG;EACL,CACc;CAChB;AACF"}

package/dist/chains/chat_vector_db_chain.d.cts

@@ -5,7 +5,6 @@ import { ChainValues } from "@langchain/core/utils/types";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { VectorStoreInterface } from "@langchain/core/vectorstores";
-
 //#region src/chains/chat_vector_db_chain.d.ts
 type LoadValues = Record<string, any>;
 /**
@@ -57,5 +56,5 @@ declare class ChatVectorDBQAChain extends BaseChain implements ChatVectorDBQACha
   }): ChatVectorDBQAChain;
 }
 //#endregion
-export { ChatVectorDBQAChain, ChatVectorDBQAChainInput };
+export { ChatVectorDBQAChain, ChatVectorDBQAChainInput, LoadValues };
 //# sourceMappingURL=chat_vector_db_chain.d.cts.map

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

@@ -1 +1 @@
-{"version":3,"file":"chat_vector_db_chain.d.cts","names":[],"sources":["../../src/chains/chat_vector_db_chain.ts"],"mappings":";;;;;;;;;KAWY,UAAA,GAAa,MAAA;;AAAzB;;UAmBiB,wBAAA,SAAiC,WAAA;EAChD,WAAA,EAAa,oBAAA;EACb,qBAAA,EAAuB,SAAA;EACvB,sBAAA,EAAwB,QAAA;EACxB,qBAAA;EACA,SAAA;EACA,QAAA;EACA,CAAA;AAAA;;cAIW,mBAAA,SACH,SAAA,YACG,wBAAA;EAEX,CAAA;EAEA,QAAA;EAEA,cAAA;EAAA,IAEI,SAAA,CAAA;EAIJ,SAAA;EAAA,IAEI,UAAA,CAAA;EAIJ,WAAA,EAAa,oBAAA;EAEb,qBAAA,EAAuB,SAAA;EAEvB,sBAAA,EAAwB,QAAA;EAExB,qBAAA;EAEA,WAAA,CAAY,MAAA,EAAQ,wBAAA;EAhCpB;EA6CM,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAoDX,UAAA,CAAA;EAAA,OAIa,WAAA,CACX,IAAA,EAAM,6BAAA,EACN,MAAA,EAAQ,UAAA,GAAU,OAAA,CAAA,mBAAA;EAqBpB,SAAA,CAAA,GAAa,6BAAA;;;;;;;;;SAiBN,OAAA,CACL,GAAA,EAAK,0BAAA,EACL,WAAA,EAAa,oBAAA,EACb,OAAA;IACE,QAAA;IACA,SAAA;IACA,CAAA;IACA,qBAAA;IACA,yBAAA;IACA,UAAA;IACA,OAAA;EAAA,IAED,mBAAA;AAAA"}
+{"version":3,"file":"chat_vector_db_chain.d.cts","names":[],"sources":["../../src/chains/chat_vector_db_chain.ts"],"mappings":";;;;;;;;KAWY,aAAa;;;;UAmBR,iCAAiC;EAChD,aAAa;EACb,uBAAuB;EACvB,wBAAwB;EACxB;EACA;EACA;EACA;;;cAIW,4BACH,qBACG;EAEX;EAEA;EAEA;MAEI;EAIJ;MAEI;EAIJ,aAAa;EAEb,uBAAuB;EAEvB,wBAAwB;EAExB;EAEA,YAAY,QAAQ;;EAad,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAoDX;SAIa,YACX,MAAM,+BACN,QAAQ,aAAU,QAAA;EAqBpB,aAAa;;;;;;;;;SAiBN,QACL,KAAK,4BACL,aAAa,sBACb;IACE;IACA;IACA;IACA;IACA;IACA;IACA;MAED"}

package/dist/chains/chat_vector_db_chain.d.ts

@@ -5,7 +5,6 @@ import { BaseLanguageModelInterface } from "@langchain/core/language_models/base
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { ChainValues } from "@langchain/core/utils/types";
 import { VectorStoreInterface } from "@langchain/core/vectorstores";
-
 //#region src/chains/chat_vector_db_chain.d.ts
 type LoadValues = Record<string, any>;
 /**
@@ -57,5 +56,5 @@ declare class ChatVectorDBQAChain extends BaseChain implements ChatVectorDBQACha
   }): ChatVectorDBQAChain;
 }
 //#endregion
-export { ChatVectorDBQAChain, ChatVectorDBQAChainInput };
+export { ChatVectorDBQAChain, ChatVectorDBQAChainInput, LoadValues };
 //# sourceMappingURL=chat_vector_db_chain.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"chat_vector_db_chain.d.ts","names":[],"sources":["../../src/chains/chat_vector_db_chain.ts"],"mappings":";;;;;;;;;KAWY,UAAA,GAAa,MAAA;;AAAzB;;UAmBiB,wBAAA,SAAiC,WAAA;EAChD,WAAA,EAAa,oBAAA;EACb,qBAAA,EAAuB,SAAA;EACvB,sBAAA,EAAwB,QAAA;EACxB,qBAAA;EACA,SAAA;EACA,QAAA;EACA,CAAA;AAAA;;cAIW,mBAAA,SACH,SAAA,YACG,wBAAA;EAEX,CAAA;EAEA,QAAA;EAEA,cAAA;EAAA,IAEI,SAAA,CAAA;EAIJ,SAAA;EAAA,IAEI,UAAA,CAAA;EAIJ,WAAA,EAAa,oBAAA;EAEb,qBAAA,EAAuB,SAAA;EAEvB,sBAAA,EAAwB,QAAA;EAExB,qBAAA;EAEA,WAAA,CAAY,MAAA,EAAQ,wBAAA;EAhCpB;EA6CM,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAoDX,UAAA,CAAA;EAAA,OAIa,WAAA,CACX,IAAA,EAAM,6BAAA,EACN,MAAA,EAAQ,UAAA,GAAU,OAAA,CAAA,mBAAA;EAqBpB,SAAA,CAAA,GAAa,6BAAA;;;;;;;;;SAiBN,OAAA,CACL,GAAA,EAAK,0BAAA,EACL,WAAA,EAAa,oBAAA,EACb,OAAA;IACE,QAAA;IACA,SAAA;IACA,CAAA;IACA,qBAAA;IACA,yBAAA;IACA,UAAA;IACA,OAAA;EAAA,IAED,mBAAA;AAAA"}
+{"version":3,"file":"chat_vector_db_chain.d.ts","names":[],"sources":["../../src/chains/chat_vector_db_chain.ts"],"mappings":";;;;;;;;KAWY,aAAa;;;;UAmBR,iCAAiC;EAChD,aAAa;EACb,uBAAuB;EACvB,wBAAwB;EACxB;EACA;EACA;EACA;;;cAIW,4BACH,qBACG;EAEX;EAEA;EAEA;MAEI;EAIJ;MAEI;EAIJ,aAAa;EAEb,uBAAuB;EAEvB,wBAAwB;EAExB;EAEA,YAAY,QAAQ;;EAad,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAoDX;SAIa,YACX,MAAM,+BACN,QAAQ,aAAU,QAAA;EAqBpB,aAAa;;;;;;;;;SAiBN,QACL,KAAK,4BACL,aAAa,sBACb;IACE;IACA;IACA;IACA;IACA;IACA;IACA;MAED"}

package/dist/chains/chat_vector_db_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat_vector_db_chain.js","names":[],"sources":["../../src/chains/chat_vector_db_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { loadQAStuffChain } from \"./question_answering/load.js\";\n\n// oxlint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\nconst qa_template = `Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\n\n/**\n * Interface for the input parameters of the ChatVectorDBQAChain class.\n */\nexport interface ChatVectorDBQAChainInput extends ChainInputs {\n  vectorstore: VectorStoreInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  outputKey?: string;\n  inputKey?: string;\n  k?: number;\n}\n\n/** @deprecated use `ConversationalRetrievalQAChain` instead. */\nexport class ChatVectorDBQAChain\n  extends BaseChain\n  implements ChatVectorDBQAChainInput\n{\n  k = 4;\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  outputKey = \"result\";\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  vectorstore: VectorStoreInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: ChatVectorDBQAChainInput) {\n    super(fields);\n    this.vectorstore = fields.vectorstore;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.k = fields.k ?? this.k;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`chat history key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string = values[this.chatHistoryKey];\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      console.log(\"_call\", values, keys);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.vectorstore.similaritySearch(\n      newQuestion,\n      this.k,\n      undefined,\n      runManager?.getChild(\"vectorstore\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    const result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      return {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"chat-vector-db\" as const;\n  }\n\n  static async deserialize(\n    data: SerializedChatVectorDBQAChain,\n    values: LoadValues\n  ) {\n    if (!(\"vectorstore\" in values)) {\n      throw new Error(\n        `Need to pass in a vectorstore to deserialize VectorDBQAChain`\n      );\n    }\n    const { vectorstore } = values;\n\n    return new ChatVectorDBQAChain({\n      combineDocumentsChain: await BaseChain.deserialize(\n        data.combine_documents_chain\n      ),\n      questionGeneratorChain: await LLMChain.deserialize(\n        data.question_generator\n      ),\n      k: data.k,\n      vectorstore,\n    });\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    return {\n      _type: this._chainType(),\n      combine_documents_chain: this.combineDocumentsChain.serialize(),\n      question_generator: this.questionGeneratorChain.serialize(),\n      k: this.k,\n    };\n  }\n\n  /**\n   * Creates an instance of ChatVectorDBQAChain using a BaseLanguageModel\n   * and other options.\n   * @param llm Instance of BaseLanguageModel used to generate a new question.\n   * @param vectorstore Instance of VectorStore used for vector operations.\n   * @param options (Optional) Additional options for creating the ChatVectorDBQAChain instance.\n   * @returns New instance of ChatVectorDBQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    vectorstore: VectorStoreInterface,\n    options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    } = {}\n  ): ChatVectorDBQAChain {\n    const { questionGeneratorTemplate, qaTemplate, verbose, ...rest } = options;\n    const question_generator_prompt = PromptTemplate.fromTemplate(\n      questionGeneratorTemplate || question_generator_template\n    );\n    const qa_prompt = PromptTemplate.fromTemplate(qaTemplate || qa_template);\n\n    const qaChain = loadQAStuffChain(llm, { prompt: qa_prompt, verbose });\n    const questionGeneratorChain = new LLMChain({\n      prompt: question_generator_prompt,\n      llm,\n      verbose,\n    });\n    const instance = new this({\n      vectorstore,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;AAaA,MAAM,8BAA8B;;;;;;AAOpC,MAAM,cAAc;;;;;;;AAqBpB,IAAa,sBAAb,MAAa,4BACH,UAEV;CACE,IAAI;CAEJ,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,KAAK,eAAe;;CAG7C,YAAY;CAEZ,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,cAAc,OAAO;AAC1B,OAAK,wBAAwB,OAAO;AACpC,OAAK,yBAAyB,OAAO;AACrC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,IAAI,OAAO,KAAK,KAAK;AAC1B,OAAK,wBACH,OAAO,yBAAyB,KAAK;;;CAIzC,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;AAE7D,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,oBAAoB,KAAK,SAAS,aAAa;EAEjE,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,cAAsB,OAAO,KAAK;EACxC,IAAI,cAAc;AAClB,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,SAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;IACf,EACD,YAAY,SAAS,qBAAqB,CAC3C;GACD,MAAM,OAAO,OAAO,KAAK,OAAO;AAChC,WAAQ,IAAI,SAAS,QAAQ,KAAK;AAClC,OAAI,KAAK,WAAW,EAClB,eAAc,OAAO,KAAK;OAE1B,OAAM,IAAI,MACR,2EACD;;EAGL,MAAM,OAAO,MAAM,KAAK,YAAY,iBAClC,aACA,KAAK,GACL,KAAA,GACA,YAAY,SAAS,cAAc,CACpC;EACD,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;GACf;EACD,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,sBACP,QAAO;GACL,GAAG;GACH,iBAAiB;GAClB;AAEH,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YACX,MACA,QACA;AACA,MAAI,EAAE,iBAAiB,QACrB,OAAM,IAAI,MACR,+DACD;EAEH,MAAM,EAAE,gBAAgB;AAExB,SAAO,IAAI,oBAAoB;GAC7B,uBAAuB,MAAM,UAAU,YACrC,KAAK,wBACN;GACD,wBAAwB,MAAM,SAAS,YACrC,KAAK,mBACN;GACD,GAAG,KAAK;GACR;GACD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,yBAAyB,KAAK,sBAAsB,WAAW;GAC/D,oBAAoB,KAAK,uBAAuB,WAAW;GAC3D,GAAG,KAAK;GACT;;;;;;;;;;CAWH,OAAO,QACL,KACA,aACA,UAQI,EAAE,EACe;EACrB,MAAM,EAAE,2BAA2B,YAAY,SAAS,GAAG,SAAS;EACpE,MAAM,4BAA4B,eAAe,aAC/C,6BAA6B,4BAC9B;EAGD,MAAM,UAAU,iBAAiB,KAAK;GAAE,QAFtB,eAAe,aAAa,cAAc,YAEH;GAAE;GAAS,CAAC;EACrE,MAAM,yBAAyB,IAAI,SAAS;GAC1C,QAAQ;GACR;GACA;GACD,CAAC;AAOF,SAAO,IANc,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA,GAAG;GACJ,CACc"}
+{"version":3,"file":"chat_vector_db_chain.js","names":[],"sources":["../../src/chains/chat_vector_db_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { loadQAStuffChain } from \"./question_answering/load.js\";\n\n// oxlint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\nconst qa_template = `Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\n\n/**\n * Interface for the input parameters of the ChatVectorDBQAChain class.\n */\nexport interface ChatVectorDBQAChainInput extends ChainInputs {\n  vectorstore: VectorStoreInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  outputKey?: string;\n  inputKey?: string;\n  k?: number;\n}\n\n/** @deprecated use `ConversationalRetrievalQAChain` instead. */\nexport class ChatVectorDBQAChain\n  extends BaseChain\n  implements ChatVectorDBQAChainInput\n{\n  k = 4;\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  outputKey = \"result\";\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  vectorstore: VectorStoreInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: ChatVectorDBQAChainInput) {\n    super(fields);\n    this.vectorstore = fields.vectorstore;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.k = fields.k ?? this.k;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`chat history key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string = values[this.chatHistoryKey];\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      console.log(\"_call\", values, keys);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.vectorstore.similaritySearch(\n      newQuestion,\n      this.k,\n      undefined,\n      runManager?.getChild(\"vectorstore\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    const result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      return {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"chat-vector-db\" as const;\n  }\n\n  static async deserialize(\n    data: SerializedChatVectorDBQAChain,\n    values: LoadValues\n  ) {\n    if (!(\"vectorstore\" in values)) {\n      throw new Error(\n        `Need to pass in a vectorstore to deserialize VectorDBQAChain`\n      );\n    }\n    const { vectorstore } = values;\n\n    return new ChatVectorDBQAChain({\n      combineDocumentsChain: await BaseChain.deserialize(\n        data.combine_documents_chain\n      ),\n      questionGeneratorChain: await LLMChain.deserialize(\n        data.question_generator\n      ),\n      k: data.k,\n      vectorstore,\n    });\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    return {\n      _type: this._chainType(),\n      combine_documents_chain: this.combineDocumentsChain.serialize(),\n      question_generator: this.questionGeneratorChain.serialize(),\n      k: this.k,\n    };\n  }\n\n  /**\n   * Creates an instance of ChatVectorDBQAChain using a BaseLanguageModel\n   * and other options.\n   * @param llm Instance of BaseLanguageModel used to generate a new question.\n   * @param vectorstore Instance of VectorStore used for vector operations.\n   * @param options (Optional) Additional options for creating the ChatVectorDBQAChain instance.\n   * @returns New instance of ChatVectorDBQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    vectorstore: VectorStoreInterface,\n    options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    } = {}\n  ): ChatVectorDBQAChain {\n    const { questionGeneratorTemplate, qaTemplate, verbose, ...rest } = options;\n    const question_generator_prompt = PromptTemplate.fromTemplate(\n      questionGeneratorTemplate || question_generator_template\n    );\n    const qa_prompt = PromptTemplate.fromTemplate(qaTemplate || qa_template);\n\n    const qaChain = loadQAStuffChain(llm, { prompt: qa_prompt, verbose });\n    const questionGeneratorChain = new LLMChain({\n      prompt: question_generator_prompt,\n      llm,\n      verbose,\n    });\n    const instance = new this({\n      vectorstore,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;AAaA,MAAM,8BAA8B;;;;;;AAOpC,MAAM,cAAc;;;;;;;AAqBpB,IAAa,sBAAb,MAAa,4BACH,UAEV;CACE,IAAI;CAEJ,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,UAAU,KAAK,cAAc;CAC5C;CAEA,YAAY;CAEZ,IAAI,aAAa;EACf,OAAO,CAAC,KAAK,SAAS;CACxB;CAEA;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAAkC;EAC5C,MAAM,MAAM;EACZ,KAAK,cAAc,OAAO;EAC1B,KAAK,wBAAwB,OAAO;EACpC,KAAK,yBAAyB,OAAO;EACrC,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,IAAI,OAAO,KAAK,KAAK;EAC1B,KAAK,wBACH,OAAO,yBAAyB,KAAK;CACzC;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,YAAY;EAE5D,IAAI,EAAE,KAAK,kBAAkB,SAC3B,MAAM,IAAI,MAAM,oBAAoB,KAAK,SAAS,YAAY;EAEhE,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,cAAsB,OAAO,KAAK;EACxC,IAAI,cAAc;EAClB,IAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,SAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;GAChB,GACA,YAAY,SAAS,oBAAoB,CAC3C;GACA,MAAM,OAAO,OAAO,KAAK,MAAM;GAC/B,QAAQ,IAAI,SAAS,QAAQ,IAAI;GACjC,IAAI,KAAK,WAAW,GAClB,cAAc,OAAO,KAAK;QAE1B,MAAM,IAAI,MACR,0EACF;EAEJ;EACA,MAAM,OAAO,MAAM,KAAK,YAAY,iBAClC,aACA,KAAK,GACL,KAAA,GACA,YAAY,SAAS,aAAa,CACpC;EACA,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;EAChB;EACA,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,mBAAmB,CAC1C;EACA,IAAI,KAAK,uBACP,OAAO;GACL,GAAG;GACH,iBAAiB;EACnB;EAEF,OAAO;CACT;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YACX,MACA,QACA;EACA,IAAI,EAAE,iBAAiB,SACrB,MAAM,IAAI,MACR,8DACF;EAEF,MAAM,EAAE,gBAAgB;EAExB,OAAO,IAAI,oBAAoB;GAC7B,uBAAuB,MAAM,UAAU,YACrC,KAAK,uBACP;GACA,wBAAwB,MAAM,SAAS,YACrC,KAAK,kBACP;GACA,GAAG,KAAK;GACR;EACF,CAAC;CACH;CAEA,YAA2C;EACzC,OAAO;GACL,OAAO,KAAK,WAAW;GACvB,yBAAyB,KAAK,sBAAsB,UAAU;GAC9D,oBAAoB,KAAK,uBAAuB,UAAU;GAC1D,GAAG,KAAK;EACV;CACF;;;;;;;;;CAUA,OAAO,QACL,KACA,aACA,UAQI,CAAC,GACgB;EACrB,MAAM,EAAE,2BAA2B,YAAY,SAAS,GAAG,SAAS;EACpE,MAAM,4BAA4B,eAAe,aAC/C,6BAA6B,2BAC/B;EAGA,MAAM,UAAU,iBAAiB,KAAK;GAAE,QAFtB,eAAe,aAAa,cAAc,WAEJ;GAAG;EAAQ,CAAC;EACpE,MAAM,yBAAyB,IAAI,SAAS;GAC1C,QAAQ;GACR;GACA;EACF,CAAC;EAOD,OAAO,IANc,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA,GAAG;EACL,CACc;CAChB;AACF"}

package/dist/chains/combine_docs_chain.cjs

@@ -1,4 +1,3 @@
-require("../_virtual/_rolldown/runtime.cjs");
 const require_base = require("./base.cjs");
 const require_llm_chain = require("./llm_chain.cjs");
 let _langchain_core_prompts = require("@langchain/core/prompts");
@@ -168,7 +167,7 @@ var RefineDocumentsChain = class RefineDocumentsChain extends require_base.BaseC
 	}
 	documentPrompt = this.defaultDocumentPrompt;
 	get inputKeys() {
-		return [...new Set([
+		return [.../* @__PURE__ */ new Set([
 			this.inputKey,
 			...this.llmChain.inputKeys,
 			...this.refineLLMChain.inputKeys

package/dist/chains/combine_docs_chain.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"combine_docs_chain.cjs","names":["BaseChain","LLMChain","PromptTemplate"],"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 =\n          await this.combineDocumentChain.llmChain._getNumTokens(formatted);\n\n        const withinTokenLimit = length < this.maxTokens;\n        // If we can skip the map step, and we're within the token limit, we don't\n        // need to run the map step, so just break out of the loop.\n        if (withinTokenLimit) {\n          break;\n        }\n      }\n\n      const results = await this.llmChain.apply(\n        inputs,\n        // If we have a runManager, then we need to create a child for each input\n        // so that we can track the progress of each input.\n        runManager\n          ? Array.from({ length: inputs.length }, (_, i) =>\n              runManager.getChild(`map_${i + 1}`)\n            )\n          : undefined\n      );\n      const { outputKey } = this.llmChain;\n\n      // If the flag is set, then concat that to the intermediate steps\n      if (this.returnIntermediateSteps) {\n        intermediateSteps = intermediateSteps.concat(\n          results.map((r) => r[outputKey])\n        );\n      }\n\n      currentDocs = results.map((r) => ({\n        pageContent: r[outputKey],\n        metadata: {},\n      }));\n    }\n\n    // Now, with the final result of all the inputs from the `llmChain`, we can\n    // run the `combineDocumentChain` over them.\n    const newInputs = {\n      [this.combineDocumentChain.inputKey]: currentDocs,\n      ...rest,\n    };\n    const result = await this.combineDocumentChain.call(\n      newInputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n\n    // Return the intermediate steps results if the flag is set\n    if (this.returnIntermediateSteps) {\n      return { ...result, intermediateSteps };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"map_reduce_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedMapReduceDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    if (!data.combine_document_chain) {\n      throw new Error(\"Missing combine_document_chain\");\n    }\n\n    return new MapReduceDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n      combineDocumentChain: await StuffDocumentsChain.deserialize(\n        data.combine_document_chain\n      ),\n    });\n  }\n\n  serialize(): SerializedMapReduceDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      combine_document_chain: this.combineDocumentChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the RefineDocumentsChain class.\n */\nexport interface RefineDocumentsChainInput extends StuffDocumentsChainInput {\n  refineLLMChain: LLMChain;\n  documentPrompt?: BasePromptTemplate;\n  initialResponseName?: string;\n  documentVariableName?: string;\n  outputKey?: string;\n}\n\n/**\n * Combine documents by doing a first pass and then refining on more documents.\n * @augments BaseChain\n * @augments RefineDocumentsChainInput\n */\nexport class RefineDocumentsChain\n  extends BaseChain\n  implements RefineDocumentsChainInput\n{\n  static lc_name() {\n    return \"RefineDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  outputKey = \"output_text\";\n\n  documentVariableName = \"context\";\n\n  initialResponseName = \"existing_answer\";\n\n  refineLLMChain: LLMChain;\n\n  get defaultDocumentPrompt(): BasePromptTemplate {\n    return new PromptTemplate({\n      inputVariables: [\"page_content\"],\n      template: \"{page_content}\",\n    });\n  }\n\n  documentPrompt = this.defaultDocumentPrompt;\n\n  get inputKeys() {\n    return [\n      ...new Set([\n        this.inputKey,\n        ...this.llmChain.inputKeys,\n        ...this.refineLLMChain.inputKeys,\n      ]),\n    ].filter(\n      (key) =>\n        key !== this.documentVariableName && key !== this.initialResponseName\n    );\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  constructor(fields: RefineDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.refineLLMChain = fields.refineLLMChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.documentPrompt = fields.documentPrompt ?? this.documentPrompt;\n    this.initialResponseName =\n      fields.initialResponseName ?? this.initialResponseName;\n  }\n\n  /** @ignore */\n  async _constructInitialInputs(doc: Document, rest: Record<string, unknown>) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { ...baseInputs, ...rest };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _constructRefineInputs(doc: Document, res: string) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { [this.initialResponseName]: res, ...baseInputs };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    const currentDocs = docs as Document[];\n\n    const initialInputs = await this._constructInitialInputs(\n      currentDocs[0],\n      rest\n    );\n    let res = await this.llmChain.predict(\n      { ...initialInputs },\n      runManager?.getChild(\"answer\")\n    );\n\n    const refineSteps = [res];\n\n    for (let i = 1; i < currentDocs.length; i += 1) {\n      const refineInputs = await this._constructRefineInputs(\n        currentDocs[i],\n        res\n      );\n      const inputs = { ...refineInputs, ...rest };\n      res = await this.refineLLMChain.predict(\n        { ...inputs },\n        runManager?.getChild(\"refine\")\n      );\n      refineSteps.push(res);\n    }\n\n    return { [this.outputKey]: res };\n  }\n\n  _chainType() {\n    return \"refine_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedRefineDocumentsChain) {\n    const SerializedLLMChain = data.llm_chain;\n\n    if (!SerializedLLMChain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    const SerializedRefineDocumentChain = data.refine_llm_chain;\n\n    if (!SerializedRefineDocumentChain) {\n      throw new Error(\"Missing refine_llm_chain\");\n    }\n\n    return new RefineDocumentsChain({\n      llmChain: await LLMChain.deserialize(SerializedLLMChain),\n      refineLLMChain: await LLMChain.deserialize(SerializedRefineDocumentChain),\n    });\n  }\n\n  serialize(): SerializedRefineDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      refine_llm_chain: this.refineLLMChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;AA4BA,IAAa,sBAAb,MAAa,4BACHA,aAAAA,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,SAAS,UAAU,CAAC,QAChD,QAAQ,QAAQ,KAAK,qBACvB;;CAGH,IAAI,aAAa;AACf,SAAO,KAAK,SAAS;;CAGvB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;;;CAI1C,YAAY,QAAkC;AAC5C,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,OADS,KAAoB,KAAK,EAAE,kBAAkB,YAC1C,CAAC,KAAK,OAAO;AAC/B,SAAO;GACL,GAAG;IACF,KAAK,uBAAuB;GAC9B;;;CAIH,MAAM,MACJ,QACA,YACsB;AAKtB,SAAO,MAJc,KAAK,SAAS,KACjC,KAAK,YAAY,OAAO,EACxB,YAAY,SAAS,oBAAoB,CAC1C;;CAIH,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAqC;AAC5D,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,SAAO,IAAI,oBAAoB,EAC7B,UAAU,MAAMC,kBAAAA,SAAS,YAAY,KAAK,UAAU,EACrD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACrC;;;;;;;;AA0BL,IAAa,0BAAb,MAAa,gCACHD,aAAAA,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,0BAA0B;CAE1B,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,qBAAqB,UAAU;;CAGhE,IAAI,aAAa;AACf,SAAO,KAAK,qBAAqB;;CAGnC,YAAY;CAEZ,gBAAgB;CAEhB,gBAAgB;CAEhB;CAEA,YAAY,QAAsC;AAChD,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBAAuB,OAAO;AACnC,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,0BAA0B,OAAO,2BAA2B;;;CAInE,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,IAAI,cAAc;EAClB,IAAI,oBAA8B,EAAE;AAGpC,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,eAAe,KAAK,GAAG;GAC9C,MAAM,SAAS,YAAY,KAAK,OAAO;KACpC,KAAK,uBAAuB,EAAE;IAC/B,GAAG;IACJ,EAAE;AAGH,OADuB,MAAM,KAAK,CAAC,KAAK,eACpB;IAElB,MAAM,YACJ,MAAM,KAAK,qBAAqB,SAAS,OAAO,OAC9C,KAAK,qBAAqB,YAAY;MACnC,KAAK,qBAAqB,WAAW;KACtC,GAAG;KACJ,CAAC,CACH;AAOH,QAHyB,MAFjB,KAAK,qBAAqB,SAAS,cAAc,UAAU,GAEjC,KAAK,UAIrC;;GAIJ,MAAM,UAAU,MAAM,KAAK,SAAS,MAClC,QAGA,aACI,MAAM,KAAK,EAAE,QAAQ,OAAO,QAAQ,GAAG,GAAG,MACxC,WAAW,SAAS,OAAO,IAAI,IAAI,CACpC,GACD,KAAA,EACL;GACD,MAAM,EAAE,cAAc,KAAK;AAG3B,OAAI,KAAK,wBACP,qBAAoB,kBAAkB,OACpC,QAAQ,KAAK,MAAM,EAAE,WAAW,CACjC;AAGH,iBAAc,QAAQ,KAAK,OAAO;IAChC,aAAa,EAAE;IACf,UAAU,EAAE;IACb,EAAE;;EAKL,MAAM,YAAY;IACf,KAAK,qBAAqB,WAAW;GACtC,GAAG;GACJ;EACD,MAAM,SAAS,MAAM,KAAK,qBAAqB,KAC7C,WACA,YAAY,SAAS,oBAAoB,CAC1C;AAGD,MAAI,KAAK,wBACP,QAAO;GAAE,GAAG;GAAQ;GAAmB;AAEzC,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAyC;AAChE,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,MAAI,CAAC,KAAK,uBACR,OAAM,IAAI,MAAM,iCAAiC;AAGnD,SAAO,IAAI,wBAAwB;GACjC,UAAU,MAAMC,kBAAAA,SAAS,YAAY,KAAK,UAAU;GACpD,sBAAsB,MAAM,oBAAoB,YAC9C,KAAK,uBACN;GACF,CAAC;;CAGJ,YAA+C;AAC7C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,wBAAwB,KAAK,qBAAqB,WAAW;GAC9D;;;;;;;;AAoBL,IAAa,uBAAb,MAAa,6BACHD,aAAAA,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,YAAY;CAEZ,uBAAuB;CAEvB,sBAAsB;CAEtB;CAEA,IAAI,wBAA4C;AAC9C,SAAO,IAAIE,wBAAAA,eAAe;GACxB,gBAAgB,CAAC,eAAe;GAChC,UAAU;GACX,CAAC;;CAGJ,iBAAiB,KAAK;CAEtB,IAAI,YAAY;AACd,SAAO,CACL,GAAG,IAAI,IAAI;GACT,KAAK;GACL,GAAG,KAAK,SAAS;GACjB,GAAG,KAAK,eAAe;GACxB,CAAC,CACH,CAAC,QACC,QACC,QAAQ,KAAK,wBAAwB,QAAQ,KAAK,oBACrD;;CAGH,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB,YAAY,QAAmC;AAC7C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,iBAAiB,OAAO;AAC7B,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,iBAAiB,OAAO,kBAAkB,KAAK;AACpD,OAAK,sBACH,OAAO,uBAAuB,KAAK;;;CAIvC,MAAM,wBAAwB,KAAe,MAA+B;EAC1E,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;AAQF,SAAO;IALJ,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC;GAE4B,GAAG;GACtB;;;CAIf,MAAM,uBAAuB,KAAe,KAAa;EACvD,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;EACF,MAAM,aAAsC,GACzC,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC,EACH;AAED,SAAO;IADW,KAAK,sBAAsB;GAAK,GAAG;GACxC;;;CAIf,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,cAAc;EAEpB,MAAM,gBAAgB,MAAM,KAAK,wBAC/B,YAAY,IACZ,KACD;EACD,IAAI,MAAM,MAAM,KAAK,SAAS,QAC5B,EAAE,GAAG,eAAe,EACpB,YAAY,SAAS,SAAS,CAC/B;EAED,MAAM,cAAc,CAAC,IAAI;AAEzB,OAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,KAAK,GAAG;GAK9C,MAAM,SAAS;IAAE,GAAG,MAJO,KAAK,uBAC9B,YAAY,IACZ,IACD;IACiC,GAAG;IAAM;AAC3C,SAAM,MAAM,KAAK,eAAe,QAC9B,EAAE,GAAG,QAAQ,EACb,YAAY,SAAS,SAAS,CAC/B;AACD,eAAY,KAAK,IAAI;;AAGvB,SAAO,GAAG,KAAK,YAAY,KAAK;;CAGlC,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAsC;EAC7D,MAAM,qBAAqB,KAAK;AAEhC,MAAI,CAAC,mBACH,OAAM,IAAI,MAAM,oBAAoB;EAGtC,MAAM,gCAAgC,KAAK;AAE3C,MAAI,CAAC,8BACH,OAAM,IAAI,MAAM,2BAA2B;AAG7C,SAAO,IAAI,qBAAqB;GAC9B,UAAU,MAAMD,kBAAAA,SAAS,YAAY,mBAAmB;GACxD,gBAAgB,MAAMA,kBAAAA,SAAS,YAAY,8BAA8B;GAC1E,CAAC;;CAGJ,YAA4C;AAC1C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,kBAAkB,KAAK,eAAe,WAAW;GAClD"}
+{"version":3,"file":"combine_docs_chain.cjs","names":["BaseChain","LLMChain","PromptTemplate"],"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 =\n          await this.combineDocumentChain.llmChain._getNumTokens(formatted);\n\n        const withinTokenLimit = length < this.maxTokens;\n        // If we can skip the map step, and we're within the token limit, we don't\n        // need to run the map step, so just break out of the loop.\n        if (withinTokenLimit) {\n          break;\n        }\n      }\n\n      const results = await this.llmChain.apply(\n        inputs,\n        // If we have a runManager, then we need to create a child for each input\n        // so that we can track the progress of each input.\n        runManager\n          ? Array.from({ length: inputs.length }, (_, i) =>\n              runManager.getChild(`map_${i + 1}`)\n            )\n          : undefined\n      );\n      const { outputKey } = this.llmChain;\n\n      // If the flag is set, then concat that to the intermediate steps\n      if (this.returnIntermediateSteps) {\n        intermediateSteps = intermediateSteps.concat(\n          results.map((r) => r[outputKey])\n        );\n      }\n\n      currentDocs = results.map((r) => ({\n        pageContent: r[outputKey],\n        metadata: {},\n      }));\n    }\n\n    // Now, with the final result of all the inputs from the `llmChain`, we can\n    // run the `combineDocumentChain` over them.\n    const newInputs = {\n      [this.combineDocumentChain.inputKey]: currentDocs,\n      ...rest,\n    };\n    const result = await this.combineDocumentChain.call(\n      newInputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n\n    // Return the intermediate steps results if the flag is set\n    if (this.returnIntermediateSteps) {\n      return { ...result, intermediateSteps };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"map_reduce_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedMapReduceDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    if (!data.combine_document_chain) {\n      throw new Error(\"Missing combine_document_chain\");\n    }\n\n    return new MapReduceDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n      combineDocumentChain: await StuffDocumentsChain.deserialize(\n        data.combine_document_chain\n      ),\n    });\n  }\n\n  serialize(): SerializedMapReduceDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      combine_document_chain: this.combineDocumentChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the RefineDocumentsChain class.\n */\nexport interface RefineDocumentsChainInput extends StuffDocumentsChainInput {\n  refineLLMChain: LLMChain;\n  documentPrompt?: BasePromptTemplate;\n  initialResponseName?: string;\n  documentVariableName?: string;\n  outputKey?: string;\n}\n\n/**\n * Combine documents by doing a first pass and then refining on more documents.\n * @augments BaseChain\n * @augments RefineDocumentsChainInput\n */\nexport class RefineDocumentsChain\n  extends BaseChain\n  implements RefineDocumentsChainInput\n{\n  static lc_name() {\n    return \"RefineDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  outputKey = \"output_text\";\n\n  documentVariableName = \"context\";\n\n  initialResponseName = \"existing_answer\";\n\n  refineLLMChain: LLMChain;\n\n  get defaultDocumentPrompt(): BasePromptTemplate {\n    return new PromptTemplate({\n      inputVariables: [\"page_content\"],\n      template: \"{page_content}\",\n    });\n  }\n\n  documentPrompt = this.defaultDocumentPrompt;\n\n  get inputKeys() {\n    return [\n      ...new Set([\n        this.inputKey,\n        ...this.llmChain.inputKeys,\n        ...this.refineLLMChain.inputKeys,\n      ]),\n    ].filter(\n      (key) =>\n        key !== this.documentVariableName && key !== this.initialResponseName\n    );\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  constructor(fields: RefineDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.refineLLMChain = fields.refineLLMChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.documentPrompt = fields.documentPrompt ?? this.documentPrompt;\n    this.initialResponseName =\n      fields.initialResponseName ?? this.initialResponseName;\n  }\n\n  /** @ignore */\n  async _constructInitialInputs(doc: Document, rest: Record<string, unknown>) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { ...baseInputs, ...rest };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _constructRefineInputs(doc: Document, res: string) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { [this.initialResponseName]: res, ...baseInputs };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    const currentDocs = docs as Document[];\n\n    const initialInputs = await this._constructInitialInputs(\n      currentDocs[0],\n      rest\n    );\n    let res = await this.llmChain.predict(\n      { ...initialInputs },\n      runManager?.getChild(\"answer\")\n    );\n\n    const refineSteps = [res];\n\n    for (let i = 1; i < currentDocs.length; i += 1) {\n      const refineInputs = await this._constructRefineInputs(\n        currentDocs[i],\n        res\n      );\n      const inputs = { ...refineInputs, ...rest };\n      res = await this.refineLLMChain.predict(\n        { ...inputs },\n        runManager?.getChild(\"refine\")\n      );\n      refineSteps.push(res);\n    }\n\n    return { [this.outputKey]: res };\n  }\n\n  _chainType() {\n    return \"refine_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedRefineDocumentsChain) {\n    const SerializedLLMChain = data.llm_chain;\n\n    if (!SerializedLLMChain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    const SerializedRefineDocumentChain = data.refine_llm_chain;\n\n    if (!SerializedRefineDocumentChain) {\n      throw new Error(\"Missing refine_llm_chain\");\n    }\n\n    return new RefineDocumentsChain({\n      llmChain: await LLMChain.deserialize(SerializedLLMChain),\n      refineLLMChain: await LLMChain.deserialize(SerializedRefineDocumentChain),\n    });\n  }\n\n  serialize(): SerializedRefineDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      refine_llm_chain: this.refineLLMChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;AA4BA,IAAa,sBAAb,MAAa,4BACHA,aAAAA,UAEV;CACE,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA,WAAW;CAEX,uBAAuB;CAEvB,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,UAAU,GAAG,KAAK,SAAS,SAAS,CAAC,CAAC,QAChD,QAAQ,QAAQ,KAAK,oBACxB;CACF;CAEA,IAAI,aAAa;EACf,OAAO,KAAK,SAAS;CACvB;CAEA,YAAY,QAAkC;EAC5C,MAAM,MAAM;EACZ,KAAK,WAAW,OAAO;EACvB,KAAK,uBACH,OAAO,wBAAwB,KAAK;EACtC,KAAK,WAAW,OAAO,YAAY,KAAK;CAC1C;;CAGA,YAAY,QAAkC;EAC5C,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,YAAY;EAE5D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,OADS,KAAoB,KAAK,EAAE,kBAAkB,WAC3C,CAAC,CAAC,KAAK,MAAM;EAC9B,OAAO;GACL,GAAG;IACF,KAAK,uBAAuB;EAC/B;CACF;;CAGA,MAAM,MACJ,QACA,YACsB;EAKtB,OAAO,MAJc,KAAK,SAAS,KACjC,KAAK,YAAY,MAAM,GACvB,YAAY,SAAS,mBAAmB,CAC1C;CAEF;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YAAY,MAAqC;EAC5D,IAAI,CAAC,KAAK,WACR,MAAM,IAAI,MAAM,mBAAmB;EAGrC,OAAO,IAAI,oBAAoB,EAC7B,UAAU,MAAMC,kBAAAA,SAAS,YAAY,KAAK,SAAS,EACrD,CAAC;CACH;CAEA,YAA2C;EACzC,OAAO;GACL,OAAO,KAAK,WAAW;GACvB,WAAW,KAAK,SAAS,UAAU;EACrC;CACF;AACF;;;;;;AAwBA,IAAa,0BAAb,MAAa,gCACHD,aAAAA,UAEV;CACE,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA,WAAW;CAEX,uBAAuB;CAEvB,0BAA0B;CAE1B,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,UAAU,GAAG,KAAK,qBAAqB,SAAS;CAC/D;CAEA,IAAI,aAAa;EACf,OAAO,KAAK,qBAAqB;CACnC;CAEA,YAAY;CAEZ,gBAAgB;CAEhB,gBAAgB;CAEhB;CAEA,YAAY,QAAsC;EAChD,MAAM,MAAM;EACZ,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;CACnE;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,YAAY;EAE5D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,IAAI,cAAc;EAClB,IAAI,oBAA8B,CAAC;EAGnC,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,eAAe,KAAK,GAAG;GAC9C,MAAM,SAAS,YAAY,KAAK,OAAO;KACpC,KAAK,uBAAuB,EAAE;IAC/B,GAAG;GACL,EAAE;GAGF,IADuB,MAAM,KAAK,CAAC,KAAK,eACpB;IAElB,MAAM,YACJ,MAAM,KAAK,qBAAqB,SAAS,OAAO,OAC9C,KAAK,qBAAqB,YAAY;MACnC,KAAK,qBAAqB,WAAW;KACtC,GAAG;IACL,CAAC,CACH;IAOF,IAHyB,MAFjB,KAAK,qBAAqB,SAAS,cAAc,SAAS,IAEhC,KAAK,WAIrC;GAEJ;GAEA,MAAM,UAAU,MAAM,KAAK,SAAS,MAClC,QAGA,aACI,MAAM,KAAK,EAAE,QAAQ,OAAO,OAAO,IAAI,GAAG,MACxC,WAAW,SAAS,OAAO,IAAI,GAAG,CACpC,IACA,KAAA,CACN;GACA,MAAM,EAAE,cAAc,KAAK;GAG3B,IAAI,KAAK,yBACP,oBAAoB,kBAAkB,OACpC,QAAQ,KAAK,MAAM,EAAE,UAAU,CACjC;GAGF,cAAc,QAAQ,KAAK,OAAO;IAChC,aAAa,EAAE;IACf,UAAU,CAAC;GACb,EAAE;EACJ;EAIA,MAAM,YAAY;IACf,KAAK,qBAAqB,WAAW;GACtC,GAAG;EACL;EACA,MAAM,SAAS,MAAM,KAAK,qBAAqB,KAC7C,WACA,YAAY,SAAS,mBAAmB,CAC1C;EAGA,IAAI,KAAK,yBACP,OAAO;GAAE,GAAG;GAAQ;EAAkB;EAExC,OAAO;CACT;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YAAY,MAAyC;EAChE,IAAI,CAAC,KAAK,WACR,MAAM,IAAI,MAAM,mBAAmB;EAGrC,IAAI,CAAC,KAAK,wBACR,MAAM,IAAI,MAAM,gCAAgC;EAGlD,OAAO,IAAI,wBAAwB;GACjC,UAAU,MAAMC,kBAAAA,SAAS,YAAY,KAAK,SAAS;GACnD,sBAAsB,MAAM,oBAAoB,YAC9C,KAAK,sBACP;EACF,CAAC;CACH;CAEA,YAA+C;EAC7C,OAAO;GACL,OAAO,KAAK,WAAW;GACvB,WAAW,KAAK,SAAS,UAAU;GACnC,wBAAwB,KAAK,qBAAqB,UAAU;EAC9D;CACF;AACF;;;;;;AAkBA,IAAa,uBAAb,MAAa,6BACHD,aAAAA,UAEV;CACE,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA,WAAW;CAEX,YAAY;CAEZ,uBAAuB;CAEvB,sBAAsB;CAEtB;CAEA,IAAI,wBAA4C;EAC9C,OAAO,IAAIE,wBAAAA,eAAe;GACxB,gBAAgB,CAAC,cAAc;GAC/B,UAAU;EACZ,CAAC;CACH;CAEA,iBAAiB,KAAK;CAEtB,IAAI,YAAY;EACd,OAAO,CACL,mBAAG,IAAI,IAAI;GACT,KAAK;GACL,GAAG,KAAK,SAAS;GACjB,GAAG,KAAK,eAAe;EACzB,CAAC,CACH,CAAC,CAAC,QACC,QACC,QAAQ,KAAK,wBAAwB,QAAQ,KAAK,mBACtD;CACF;CAEA,IAAI,aAAa;EACf,OAAO,CAAC,KAAK,SAAS;CACxB;CAEA,YAAY,QAAmC;EAC7C,MAAM,MAAM;EACZ,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;CACvC;;CAGA,MAAM,wBAAwB,KAAe,MAA+B;EAC1E,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;EACT;EACA,MAAM,eAAwC,CAAC;EAC/C,KAAK,eAAe,eAAe,SAAS,UAAU;GACpD,aAAa,SAAS,SAAS;EACjC,CAAC;EAQD,OAAO;IALJ,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,aACL,CAAC;GAE6B,GAAG;EACvB;CACd;;CAGA,MAAM,uBAAuB,KAAe,KAAa;EACvD,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;EACT;EACA,MAAM,eAAwC,CAAC;EAC/C,KAAK,eAAe,eAAe,SAAS,UAAU;GACpD,aAAa,SAAS,SAAS;EACjC,CAAC;EACD,MAAM,aAAsC,GACzC,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,aACL,CAAC,EACH;EAEA,OAAO;IADW,KAAK,sBAAsB;GAAK,GAAG;EACzC;CACd;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,YAAY;EAE5D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,cAAc;EAEpB,MAAM,gBAAgB,MAAM,KAAK,wBAC/B,YAAY,IACZ,IACF;EACA,IAAI,MAAM,MAAM,KAAK,SAAS,QAC5B,EAAE,GAAG,cAAc,GACnB,YAAY,SAAS,QAAQ,CAC/B;EAEA,MAAM,cAAc,CAAC,GAAG;EAExB,KAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,KAAK,GAAG;GAK9C,MAAM,SAAS;IAAE,GAAG,MAJO,KAAK,uBAC9B,YAAY,IACZ,GACF;IACkC,GAAG;GAAK;GAC1C,MAAM,MAAM,KAAK,eAAe,QAC9B,EAAE,GAAG,OAAO,GACZ,YAAY,SAAS,QAAQ,CAC/B;GACA,YAAY,KAAK,GAAG;EACtB;EAEA,OAAO,GAAG,KAAK,YAAY,IAAI;CACjC;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YAAY,MAAsC;EAC7D,MAAM,qBAAqB,KAAK;EAEhC,IAAI,CAAC,oBACH,MAAM,IAAI,MAAM,mBAAmB;EAGrC,MAAM,gCAAgC,KAAK;EAE3C,IAAI,CAAC,+BACH,MAAM,IAAI,MAAM,0BAA0B;EAG5C,OAAO,IAAI,qBAAqB;GAC9B,UAAU,MAAMD,kBAAAA,SAAS,YAAY,kBAAkB;GACvD,gBAAgB,MAAMA,kBAAAA,SAAS,YAAY,6BAA6B;EAC1E,CAAC;CACH;CAEA,YAA4C;EAC1C,OAAO;GACL,OAAO,KAAK,WAAW;GACvB,WAAW,KAAK,SAAS,UAAU;GACnC,kBAAkB,KAAK,eAAe,UAAU;EAClD;CACF;AACF"}

package/dist/chains/combine_docs_chain.d.cts

@@ -3,10 +3,8 @@ import { BaseChain, ChainInputs } from "./base.cjs";
 import { LLMChain } from "./llm_chain.cjs";
 import { ChainValues } from "@langchain/core/utils/types";
 import { BasePromptTemplate } from "@langchain/core/prompts";
-import * as _$_langchain_core_prompt_values0 from "@langchain/core/prompt_values";
 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.
@@ -103,7 +101,7 @@ declare class RefineDocumentsChain extends BaseChain implements RefineDocumentsC
   initialResponseName: string;
   refineLLMChain: LLMChain;
   get defaultDocumentPrompt(): BasePromptTemplate;
-  documentPrompt: BasePromptTemplate<any, _$_langchain_core_prompt_values0.BasePromptValueInterface, any>;
+  documentPrompt: BasePromptTemplate<any, import("@langchain/core/prompt_values").BasePromptValueInterface, any>;
   get inputKeys(): string[];
   get outputKeys(): string[];
   constructor(fields: RefineDocumentsChainInput);

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

@@ -1 +1 @@
-{"version":3,"file":"combine_docs_chain.d.cts","names":[],"sources":["../../src/chains/combine_docs_chain.ts"],"mappings":";;;;;;;;;;;;;UAeiB,wBAAA,SAAiC,WAAA;EAAR;EAExC,QAAA,EAAU,QAAA;EACV,QAAA;EAHgD;EAKhD,oBAAA;AAAA;;;;;AAQF;cAAa,mBAAA,SACH,SAAA,YACG,wBAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,QAAA,EAAU,QAAA;EAEV,QAAA;EAEA,oBAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAMA,UAAA,CAAA;EAIJ,WAAA,CAAY,MAAA,EAAQ,wBAAA;EA0BT;EAjBX,WAAA,CAAY,MAAA,EAAQ,WAAA,GAAc,WAAA;EA6BH;EAfzB,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAQX,UAAA,CAAA;EAAA,OAIa,WAAA,CAAY,IAAA,EAAM,6BAAA,GAA6B,OAAA,CAAA,mBAAA;EAU5D,SAAA,CAAA,GAAa,6BAAA;AAAA;;;;;UAYE,4BAAA,SAAqC,wBAAA;EA5EpD;EA8EA,SAAA;EA5EA;EA8EA,aAAA;EA1EI;EA4EJ,aAAA;EAlEA;EAoEA,oBAAA,EAAsB,mBAAA;EApEV;EAsEZ,uBAAA;AAAA;;;;;;cAQW,uBAAA,SACH,SAAA,YACG,4BAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,QAAA,EAAU,QAAA;EAEV,QAAA;EAEA,oBAAA;EAEA,uBAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAIJ,SAAA;EAEA,aAAA;EAEA,aAAA;EAEA,oBAAA,EAAsB,mBAAA;EAEtB,WAAA,CAAY,MAAA,EAAQ,4BAAA;EA9DsB;EA4EpC,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAgFX,UAAA,CAAA;EAAA,OAIa,WAAA,CAAY,IAAA,EAAM,iCAAA,GAAiC,OAAA,CAAA,uBAAA;EAiBhE,SAAA,CAAA,GAAa,iCAAA;AAAA;;;;UAYE,yBAAA,SAAkC,wBAAA;EACjD,cAAA,EAAgB,QAAA;EAChB,cAAA,GAAiB,kBAAA;EACjB,mBAAA;EACA,oBAAA;EACA,SAAA;AAAA;;;;;;cAQW,oBAAA,SACH,SAAA,YACG,yBAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,QAAA,EAAU,QAAA;EAEV,QAAA;EAEA,SAAA;EAEA,oBAAA;EAEA,mBAAA;EAEA,cAAA,EAAgB,QAAA;EAAA,IAEZ,qBAAA,CAAA,GAAyB,kBAAA;EAO7B,cAAA,EAAc,kBAAA,MAPiC,gCAAA,CAOjC,wBAAA;EAAA,IAEV,SAAA,CAAA;EAAA,IAaA,UAAA,CAAA;EAIJ,WAAA,CAAY,MAAA,EAAQ,yBAAA;EAzNb;EAuOD,uBAAA,CAAwB,GAAA,EAAK,QAAA,EAAU,IAAA,EAAM,MAAA,oBAAuB,OAAA;IAAA;;EA/N1E;EAmPM,sBAAA,CAAuB,GAAA,EAAK,QAAA,EAAU,GAAA,WAAW,OAAA;IAAA;;EAvOvD;EA0PM,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAmCX,UAAA,CAAA;EAAA,OAIa,WAAA,CAAY,IAAA,EAAM,8BAAA,GAA8B,OAAA,CAAA,oBAAA;EAmB7D,SAAA,CAAA,GAAa,8BAAA;AAAA"}
+{"version":3,"file":"combine_docs_chain.d.cts","names":[],"sources":["../../src/chains/combine_docs_chain.ts"],"mappings":";;;;;;;;;;;UAeiB,iCAAiC;;EAEhD,UAAU;EACV;;EAEA;;;;;;;cAQW,4BACH,qBACG;SAEJ;EAIP,UAAU;EAEV;EAEA;MAEI;MAMA;EAIJ,YAAY,QAAQ;;EASpB,YAAY,QAAQ,cAAc;;EAc5B,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAQX;SAIa,YAAY,MAAM,gCAA6B,QAAA;EAU5D,aAAa;;;;;;UAYE,qCAAqC;;EAEpD;;EAEA;;EAEA;;EAEA,sBAAsB;;EAEtB;;;;;;;cAQW,gCACH,qBACG;SAEJ;EAIP,UAAU;EAEV;EAEA;EAEA;MAEI;MAIA;EAIJ;EAEA;EAEA;EAEA,sBAAsB;EAEtB,YAAY,QAAQ;;EAcd,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAgFX;SAIa,YAAY,MAAM,oCAAiC,QAAA;EAiBhE,aAAa;;;;;UAYE,kCAAkC;EACjD,gBAAgB;EAChB,iBAAiB;EACjB;EACA;EACA;;;;;;;cAQW,6BACH,qBACG;SAEJ;EAIP,UAAU;EAEV;EAEA;EAEA;EAEA;EAEA,gBAAgB;MAEZ,yBAAyB;EAO7B,gBAAc,gEAAA;MAEV;MAaA;EAIJ,YAAY,QAAQ;;EAcd,wBAAwB,KAAK,UAAU,MAAM,0BAAuB;;;;EAoBpE,uBAAuB,KAAK,UAAU,cAAW;;;;EAmBjD,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAmCX;SAIa,YAAY,MAAM,iCAA8B,QAAA;EAmB7D,aAAa"}

package/dist/chains/combine_docs_chain.d.ts

@@ -5,8 +5,6 @@ import { BasePromptTemplate } from "@langchain/core/prompts";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { ChainValues } from "@langchain/core/utils/types";
 import { Document } from "@langchain/core/documents";
-import * as _$_langchain_core_prompt_values0 from "@langchain/core/prompt_values";
-
 //#region src/chains/combine_docs_chain.d.ts
 /**
  * Interface for the input properties of the StuffDocumentsChain class.
@@ -103,7 +101,7 @@ declare class RefineDocumentsChain extends BaseChain implements RefineDocumentsC
   initialResponseName: string;
   refineLLMChain: LLMChain;
   get defaultDocumentPrompt(): BasePromptTemplate;
-  documentPrompt: BasePromptTemplate<any, _$_langchain_core_prompt_values0.BasePromptValueInterface, any>;
+  documentPrompt: BasePromptTemplate<any, import("@langchain/core/prompt_values").BasePromptValueInterface, any>;
   get inputKeys(): string[];
   get outputKeys(): string[];
   constructor(fields: RefineDocumentsChainInput);

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

@@ -1 +1 @@
-{"version":3,"file":"combine_docs_chain.d.ts","names":[],"sources":["../../src/chains/combine_docs_chain.ts"],"mappings":";;;;;;;;;;;;;UAeiB,wBAAA,SAAiC,WAAA;EAAR;EAExC,QAAA,EAAU,QAAA;EACV,QAAA;EAHgD;EAKhD,oBAAA;AAAA;;;;;AAQF;cAAa,mBAAA,SACH,SAAA,YACG,wBAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,QAAA,EAAU,QAAA;EAEV,QAAA;EAEA,oBAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAMA,UAAA,CAAA;EAIJ,WAAA,CAAY,MAAA,EAAQ,wBAAA;EA0BT;EAjBX,WAAA,CAAY,MAAA,EAAQ,WAAA,GAAc,WAAA;EA6BH;EAfzB,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAQX,UAAA,CAAA;EAAA,OAIa,WAAA,CAAY,IAAA,EAAM,6BAAA,GAA6B,OAAA,CAAA,mBAAA;EAU5D,SAAA,CAAA,GAAa,6BAAA;AAAA;;;;;UAYE,4BAAA,SAAqC,wBAAA;EA5EpD;EA8EA,SAAA;EA5EA;EA8EA,aAAA;EA1EI;EA4EJ,aAAA;EAlEA;EAoEA,oBAAA,EAAsB,mBAAA;EApEV;EAsEZ,uBAAA;AAAA;;;;;;cAQW,uBAAA,SACH,SAAA,YACG,4BAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,QAAA,EAAU,QAAA;EAEV,QAAA;EAEA,oBAAA;EAEA,uBAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAIJ,SAAA;EAEA,aAAA;EAEA,aAAA;EAEA,oBAAA,EAAsB,mBAAA;EAEtB,WAAA,CAAY,MAAA,EAAQ,4BAAA;EA9DsB;EA4EpC,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAgFX,UAAA,CAAA;EAAA,OAIa,WAAA,CAAY,IAAA,EAAM,iCAAA,GAAiC,OAAA,CAAA,uBAAA;EAiBhE,SAAA,CAAA,GAAa,iCAAA;AAAA;;;;UAYE,yBAAA,SAAkC,wBAAA;EACjD,cAAA,EAAgB,QAAA;EAChB,cAAA,GAAiB,kBAAA;EACjB,mBAAA;EACA,oBAAA;EACA,SAAA;AAAA;;;;;;cAQW,oBAAA,SACH,SAAA,YACG,yBAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,QAAA,EAAU,QAAA;EAEV,QAAA;EAEA,SAAA;EAEA,oBAAA;EAEA,mBAAA;EAEA,cAAA,EAAgB,QAAA;EAAA,IAEZ,qBAAA,CAAA,GAAyB,kBAAA;EAO7B,cAAA,EAAc,kBAAA,MAPiC,gCAAA,CAOjC,wBAAA;EAAA,IAEV,SAAA,CAAA;EAAA,IAaA,UAAA,CAAA;EAIJ,WAAA,CAAY,MAAA,EAAQ,yBAAA;EAzNb;EAuOD,uBAAA,CAAwB,GAAA,EAAK,QAAA,EAAU,IAAA,EAAM,MAAA,oBAAuB,OAAA;IAAA;;EA/N1E;EAmPM,sBAAA,CAAuB,GAAA,EAAK,QAAA,EAAU,GAAA,WAAW,OAAA;IAAA;;EAvOvD;EA0PM,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAmCX,UAAA,CAAA;EAAA,OAIa,WAAA,CAAY,IAAA,EAAM,8BAAA,GAA8B,OAAA,CAAA,oBAAA;EAmB7D,SAAA,CAAA,GAAa,8BAAA;AAAA"}
+{"version":3,"file":"combine_docs_chain.d.ts","names":[],"sources":["../../src/chains/combine_docs_chain.ts"],"mappings":";;;;;;;;;;;UAeiB,iCAAiC;;EAEhD,UAAU;EACV;;EAEA;;;;;;;cAQW,4BACH,qBACG;SAEJ;EAIP,UAAU;EAEV;EAEA;MAEI;MAMA;EAIJ,YAAY,QAAQ;;EASpB,YAAY,QAAQ,cAAc;;EAc5B,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAQX;SAIa,YAAY,MAAM,gCAA6B,QAAA;EAU5D,aAAa;;;;;;UAYE,qCAAqC;;EAEpD;;EAEA;;EAEA;;EAEA,sBAAsB;;EAEtB;;;;;;;cAQW,gCACH,qBACG;SAEJ;EAIP,UAAU;EAEV;EAEA;EAEA;MAEI;MAIA;EAIJ;EAEA;EAEA;EAEA,sBAAsB;EAEtB,YAAY,QAAQ;;EAcd,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAgFX;SAIa,YAAY,MAAM,oCAAiC,QAAA;EAiBhE,aAAa;;;;;UAYE,kCAAkC;EACjD,gBAAgB;EAChB,iBAAiB;EACjB;EACA;EACA;;;;;;;cAQW,6BACH,qBACG;SAEJ;EAIP,UAAU;EAEV;EAEA;EAEA;EAEA;EAEA,gBAAgB;MAEZ,yBAAyB;EAO7B,gBAAc,gEAAA;MAEV;MAaA;EAIJ,YAAY,QAAQ;;EAcd,wBAAwB,KAAK,UAAU,MAAM,0BAAuB;;;;EAoBpE,uBAAuB,KAAK,UAAU,cAAW;;;;EAmBjD,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAmCX;SAIa,YAAY,MAAM,iCAA8B,QAAA;EAmB7D,aAAa"}

package/dist/chains/combine_docs_chain.js

@@ -167,7 +167,7 @@ var RefineDocumentsChain = class RefineDocumentsChain extends BaseChain {
 	}
 	documentPrompt = this.defaultDocumentPrompt;
 	get inputKeys() {
-		return [...new Set([
+		return [.../* @__PURE__ */ new Set([
 			this.inputKey,
 			...this.llmChain.inputKeys,
 			...this.refineLLMChain.inputKeys

package/dist/chains/combine_docs_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"combine_docs_chain.js","names":[],"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 =\n          await this.combineDocumentChain.llmChain._getNumTokens(formatted);\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;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,SAAS,UAAU,CAAC,QAChD,QAAQ,QAAQ,KAAK,qBACvB;;CAGH,IAAI,aAAa;AACf,SAAO,KAAK,SAAS;;CAGvB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;;;CAI1C,YAAY,QAAkC;AAC5C,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,OADS,KAAoB,KAAK,EAAE,kBAAkB,YAC1C,CAAC,KAAK,OAAO;AAC/B,SAAO;GACL,GAAG;IACF,KAAK,uBAAuB;GAC9B;;;CAIH,MAAM,MACJ,QACA,YACsB;AAKtB,SAAO,MAJc,KAAK,SAAS,KACjC,KAAK,YAAY,OAAO,EACxB,YAAY,SAAS,oBAAoB,CAC1C;;CAIH,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAqC;AAC5D,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,SAAO,IAAI,oBAAoB,EAC7B,UAAU,MAAM,SAAS,YAAY,KAAK,UAAU,EACrD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACrC;;;;;;;;AA0BL,IAAa,0BAAb,MAAa,gCACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,0BAA0B;CAE1B,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,qBAAqB,UAAU;;CAGhE,IAAI,aAAa;AACf,SAAO,KAAK,qBAAqB;;CAGnC,YAAY;CAEZ,gBAAgB;CAEhB,gBAAgB;CAEhB;CAEA,YAAY,QAAsC;AAChD,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBAAuB,OAAO;AACnC,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,0BAA0B,OAAO,2BAA2B;;;CAInE,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,IAAI,cAAc;EAClB,IAAI,oBAA8B,EAAE;AAGpC,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,eAAe,KAAK,GAAG;GAC9C,MAAM,SAAS,YAAY,KAAK,OAAO;KACpC,KAAK,uBAAuB,EAAE;IAC/B,GAAG;IACJ,EAAE;AAGH,OADuB,MAAM,KAAK,CAAC,KAAK,eACpB;IAElB,MAAM,YACJ,MAAM,KAAK,qBAAqB,SAAS,OAAO,OAC9C,KAAK,qBAAqB,YAAY;MACnC,KAAK,qBAAqB,WAAW;KACtC,GAAG;KACJ,CAAC,CACH;AAOH,QAHyB,MAFjB,KAAK,qBAAqB,SAAS,cAAc,UAAU,GAEjC,KAAK,UAIrC;;GAIJ,MAAM,UAAU,MAAM,KAAK,SAAS,MAClC,QAGA,aACI,MAAM,KAAK,EAAE,QAAQ,OAAO,QAAQ,GAAG,GAAG,MACxC,WAAW,SAAS,OAAO,IAAI,IAAI,CACpC,GACD,KAAA,EACL;GACD,MAAM,EAAE,cAAc,KAAK;AAG3B,OAAI,KAAK,wBACP,qBAAoB,kBAAkB,OACpC,QAAQ,KAAK,MAAM,EAAE,WAAW,CACjC;AAGH,iBAAc,QAAQ,KAAK,OAAO;IAChC,aAAa,EAAE;IACf,UAAU,EAAE;IACb,EAAE;;EAKL,MAAM,YAAY;IACf,KAAK,qBAAqB,WAAW;GACtC,GAAG;GACJ;EACD,MAAM,SAAS,MAAM,KAAK,qBAAqB,KAC7C,WACA,YAAY,SAAS,oBAAoB,CAC1C;AAGD,MAAI,KAAK,wBACP,QAAO;GAAE,GAAG;GAAQ;GAAmB;AAEzC,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAyC;AAChE,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,MAAI,CAAC,KAAK,uBACR,OAAM,IAAI,MAAM,iCAAiC;AAGnD,SAAO,IAAI,wBAAwB;GACjC,UAAU,MAAM,SAAS,YAAY,KAAK,UAAU;GACpD,sBAAsB,MAAM,oBAAoB,YAC9C,KAAK,uBACN;GACF,CAAC;;CAGJ,YAA+C;AAC7C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,wBAAwB,KAAK,qBAAqB,WAAW;GAC9D;;;;;;;;AAoBL,IAAa,uBAAb,MAAa,6BACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,YAAY;CAEZ,uBAAuB;CAEvB,sBAAsB;CAEtB;CAEA,IAAI,wBAA4C;AAC9C,SAAO,IAAI,eAAe;GACxB,gBAAgB,CAAC,eAAe;GAChC,UAAU;GACX,CAAC;;CAGJ,iBAAiB,KAAK;CAEtB,IAAI,YAAY;AACd,SAAO,CACL,GAAG,IAAI,IAAI;GACT,KAAK;GACL,GAAG,KAAK,SAAS;GACjB,GAAG,KAAK,eAAe;GACxB,CAAC,CACH,CAAC,QACC,QACC,QAAQ,KAAK,wBAAwB,QAAQ,KAAK,oBACrD;;CAGH,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB,YAAY,QAAmC;AAC7C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,iBAAiB,OAAO;AAC7B,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,iBAAiB,OAAO,kBAAkB,KAAK;AACpD,OAAK,sBACH,OAAO,uBAAuB,KAAK;;;CAIvC,MAAM,wBAAwB,KAAe,MAA+B;EAC1E,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;AAQF,SAAO;IALJ,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC;GAE4B,GAAG;GACtB;;;CAIf,MAAM,uBAAuB,KAAe,KAAa;EACvD,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;EACF,MAAM,aAAsC,GACzC,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC,EACH;AAED,SAAO;IADW,KAAK,sBAAsB;GAAK,GAAG;GACxC;;;CAIf,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,cAAc;EAEpB,MAAM,gBAAgB,MAAM,KAAK,wBAC/B,YAAY,IACZ,KACD;EACD,IAAI,MAAM,MAAM,KAAK,SAAS,QAC5B,EAAE,GAAG,eAAe,EACpB,YAAY,SAAS,SAAS,CAC/B;EAED,MAAM,cAAc,CAAC,IAAI;AAEzB,OAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,KAAK,GAAG;GAK9C,MAAM,SAAS;IAAE,GAAG,MAJO,KAAK,uBAC9B,YAAY,IACZ,IACD;IACiC,GAAG;IAAM;AAC3C,SAAM,MAAM,KAAK,eAAe,QAC9B,EAAE,GAAG,QAAQ,EACb,YAAY,SAAS,SAAS,CAC/B;AACD,eAAY,KAAK,IAAI;;AAGvB,SAAO,GAAG,KAAK,YAAY,KAAK;;CAGlC,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAsC;EAC7D,MAAM,qBAAqB,KAAK;AAEhC,MAAI,CAAC,mBACH,OAAM,IAAI,MAAM,oBAAoB;EAGtC,MAAM,gCAAgC,KAAK;AAE3C,MAAI,CAAC,8BACH,OAAM,IAAI,MAAM,2BAA2B;AAG7C,SAAO,IAAI,qBAAqB;GAC9B,UAAU,MAAM,SAAS,YAAY,mBAAmB;GACxD,gBAAgB,MAAM,SAAS,YAAY,8BAA8B;GAC1E,CAAC;;CAGJ,YAA4C;AAC1C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,kBAAkB,KAAK,eAAe,WAAW;GAClD"}
+{"version":3,"file":"combine_docs_chain.js","names":[],"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 =\n          await this.combineDocumentChain.llmChain._getNumTokens(formatted);\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;EACf,OAAO;CACT;CAEA;CAEA,WAAW;CAEX,uBAAuB;CAEvB,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,UAAU,GAAG,KAAK,SAAS,SAAS,CAAC,CAAC,QAChD,QAAQ,QAAQ,KAAK,oBACxB;CACF;CAEA,IAAI,aAAa;EACf,OAAO,KAAK,SAAS;CACvB;CAEA,YAAY,QAAkC;EAC5C,MAAM,MAAM;EACZ,KAAK,WAAW,OAAO;EACvB,KAAK,uBACH,OAAO,wBAAwB,KAAK;EACtC,KAAK,WAAW,OAAO,YAAY,KAAK;CAC1C;;CAGA,YAAY,QAAkC;EAC5C,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,YAAY;EAE5D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,OADS,KAAoB,KAAK,EAAE,kBAAkB,WAC3C,CAAC,CAAC,KAAK,MAAM;EAC9B,OAAO;GACL,GAAG;IACF,KAAK,uBAAuB;EAC/B;CACF;;CAGA,MAAM,MACJ,QACA,YACsB;EAKtB,OAAO,MAJc,KAAK,SAAS,KACjC,KAAK,YAAY,MAAM,GACvB,YAAY,SAAS,mBAAmB,CAC1C;CAEF;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YAAY,MAAqC;EAC5D,IAAI,CAAC,KAAK,WACR,MAAM,IAAI,MAAM,mBAAmB;EAGrC,OAAO,IAAI,oBAAoB,EAC7B,UAAU,MAAM,SAAS,YAAY,KAAK,SAAS,EACrD,CAAC;CACH;CAEA,YAA2C;EACzC,OAAO;GACL,OAAO,KAAK,WAAW;GACvB,WAAW,KAAK,SAAS,UAAU;EACrC;CACF;AACF;;;;;;AAwBA,IAAa,0BAAb,MAAa,gCACH,UAEV;CACE,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA,WAAW;CAEX,uBAAuB;CAEvB,0BAA0B;CAE1B,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,UAAU,GAAG,KAAK,qBAAqB,SAAS;CAC/D;CAEA,IAAI,aAAa;EACf,OAAO,KAAK,qBAAqB;CACnC;CAEA,YAAY;CAEZ,gBAAgB;CAEhB,gBAAgB;CAEhB;CAEA,YAAY,QAAsC;EAChD,MAAM,MAAM;EACZ,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;CACnE;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,YAAY;EAE5D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,IAAI,cAAc;EAClB,IAAI,oBAA8B,CAAC;EAGnC,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,eAAe,KAAK,GAAG;GAC9C,MAAM,SAAS,YAAY,KAAK,OAAO;KACpC,KAAK,uBAAuB,EAAE;IAC/B,GAAG;GACL,EAAE;GAGF,IADuB,MAAM,KAAK,CAAC,KAAK,eACpB;IAElB,MAAM,YACJ,MAAM,KAAK,qBAAqB,SAAS,OAAO,OAC9C,KAAK,qBAAqB,YAAY;MACnC,KAAK,qBAAqB,WAAW;KACtC,GAAG;IACL,CAAC,CACH;IAOF,IAHyB,MAFjB,KAAK,qBAAqB,SAAS,cAAc,SAAS,IAEhC,KAAK,WAIrC;GAEJ;GAEA,MAAM,UAAU,MAAM,KAAK,SAAS,MAClC,QAGA,aACI,MAAM,KAAK,EAAE,QAAQ,OAAO,OAAO,IAAI,GAAG,MACxC,WAAW,SAAS,OAAO,IAAI,GAAG,CACpC,IACA,KAAA,CACN;GACA,MAAM,EAAE,cAAc,KAAK;GAG3B,IAAI,KAAK,yBACP,oBAAoB,kBAAkB,OACpC,QAAQ,KAAK,MAAM,EAAE,UAAU,CACjC;GAGF,cAAc,QAAQ,KAAK,OAAO;IAChC,aAAa,EAAE;IACf,UAAU,CAAC;GACb,EAAE;EACJ;EAIA,MAAM,YAAY;IACf,KAAK,qBAAqB,WAAW;GACtC,GAAG;EACL;EACA,MAAM,SAAS,MAAM,KAAK,qBAAqB,KAC7C,WACA,YAAY,SAAS,mBAAmB,CAC1C;EAGA,IAAI,KAAK,yBACP,OAAO;GAAE,GAAG;GAAQ;EAAkB;EAExC,OAAO;CACT;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YAAY,MAAyC;EAChE,IAAI,CAAC,KAAK,WACR,MAAM,IAAI,MAAM,mBAAmB;EAGrC,IAAI,CAAC,KAAK,wBACR,MAAM,IAAI,MAAM,gCAAgC;EAGlD,OAAO,IAAI,wBAAwB;GACjC,UAAU,MAAM,SAAS,YAAY,KAAK,SAAS;GACnD,sBAAsB,MAAM,oBAAoB,YAC9C,KAAK,sBACP;EACF,CAAC;CACH;CAEA,YAA+C;EAC7C,OAAO;GACL,OAAO,KAAK,WAAW;GACvB,WAAW,KAAK,SAAS,UAAU;GACnC,wBAAwB,KAAK,qBAAqB,UAAU;EAC9D;CACF;AACF;;;;;;AAkBA,IAAa,uBAAb,MAAa,6BACH,UAEV;CACE,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA,WAAW;CAEX,YAAY;CAEZ,uBAAuB;CAEvB,sBAAsB;CAEtB;CAEA,IAAI,wBAA4C;EAC9C,OAAO,IAAI,eAAe;GACxB,gBAAgB,CAAC,cAAc;GAC/B,UAAU;EACZ,CAAC;CACH;CAEA,iBAAiB,KAAK;CAEtB,IAAI,YAAY;EACd,OAAO,CACL,mBAAG,IAAI,IAAI;GACT,KAAK;GACL,GAAG,KAAK,SAAS;GACjB,GAAG,KAAK,eAAe;EACzB,CAAC,CACH,CAAC,CAAC,QACC,QACC,QAAQ,KAAK,wBAAwB,QAAQ,KAAK,mBACtD;CACF;CAEA,IAAI,aAAa;EACf,OAAO,CAAC,KAAK,SAAS;CACxB;CAEA,YAAY,QAAmC;EAC7C,MAAM,MAAM;EACZ,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;CACvC;;CAGA,MAAM,wBAAwB,KAAe,MAA+B;EAC1E,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;EACT;EACA,MAAM,eAAwC,CAAC;EAC/C,KAAK,eAAe,eAAe,SAAS,UAAU;GACpD,aAAa,SAAS,SAAS;EACjC,CAAC;EAQD,OAAO;IALJ,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,aACL,CAAC;GAE6B,GAAG;EACvB;CACd;;CAGA,MAAM,uBAAuB,KAAe,KAAa;EACvD,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;EACT;EACA,MAAM,eAAwC,CAAC;EAC/C,KAAK,eAAe,eAAe,SAAS,UAAU;GACpD,aAAa,SAAS,SAAS;EACjC,CAAC;EACD,MAAM,aAAsC,GACzC,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,aACL,CAAC,EACH;EAEA,OAAO;IADW,KAAK,sBAAsB;GAAK,GAAG;EACzC;CACd;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,YAAY;EAE5D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,cAAc;EAEpB,MAAM,gBAAgB,MAAM,KAAK,wBAC/B,YAAY,IACZ,IACF;EACA,IAAI,MAAM,MAAM,KAAK,SAAS,QAC5B,EAAE,GAAG,cAAc,GACnB,YAAY,SAAS,QAAQ,CAC/B;EAEA,MAAM,cAAc,CAAC,GAAG;EAExB,KAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,KAAK,GAAG;GAK9C,MAAM,SAAS;IAAE,GAAG,MAJO,KAAK,uBAC9B,YAAY,IACZ,GACF;IACkC,GAAG;GAAK;GAC1C,MAAM,MAAM,KAAK,eAAe,QAC9B,EAAE,GAAG,OAAO,GACZ,YAAY,SAAS,QAAQ,CAC/B;GACA,YAAY,KAAK,GAAG;EACtB;EAEA,OAAO,GAAG,KAAK,YAAY,IAAI;CACjC;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YAAY,MAAsC;EAC7D,MAAM,qBAAqB,KAAK;EAEhC,IAAI,CAAC,oBACH,MAAM,IAAI,MAAM,mBAAmB;EAGrC,MAAM,gCAAgC,KAAK;EAE3C,IAAI,CAAC,+BACH,MAAM,IAAI,MAAM,0BAA0B;EAG5C,OAAO,IAAI,qBAAqB;GAC9B,UAAU,MAAM,SAAS,YAAY,kBAAkB;GACvD,gBAAgB,MAAM,SAAS,YAAY,6BAA6B;EAC1E,CAAC;CACH;CAEA,YAA4C;EAC1C,OAAO;GACL,OAAO,KAAK,WAAW;GACvB,WAAW,KAAK,SAAS,UAAU;GACnC,kBAAkB,KAAK,eAAe,UAAU;EAClD;CACF;AACF"}

package/dist/chains/combine_documents/base.cjs

@@ -1,8 +1,7 @@
-require("../../_virtual/_rolldown/runtime.cjs");
 let _langchain_core_prompts = require("@langchain/core/prompts");
 //#region src/chains/combine_documents/base.ts
 const DOCUMENTS_KEY = "context";
-const DEFAULT_DOCUMENT_PROMPT = /* @__PURE__ */ _langchain_core_prompts.PromptTemplate.fromTemplate("{page_content}");
+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 "";
 	return (await Promise.all(documents.map((document) => documentPrompt.withConfig({ runName: "document_formatter" }).invoke({

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

@@ -1 +1 @@
-{"version":3,"file":"base.cjs","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":";;;AAMA,MAAa,gBAAgB;AAG7B,MAAa,0BACK,wCAAA,eAAe,aAAa,iBAAiB;AAE/D,eAAsB,gBAAgB,EACpC,gBACA,mBACA,WACA,UAMC;AACD,KAAI,aAAa,QAAQ,UAAU,WAAW,EAC5C,QAAO;AAYT,SAAO,MAVqB,QAAQ,IAClC,UAAU,KAAK,aACb,eACG,WAAW,EAAE,SAAS,sBAAsB,CAAC,CAC7C,OACC;EAAE,GAAG,SAAS;EAAU,cAAc,SAAS;EAAa,EAC5D,OACD,CACJ,CACF,EACoB,KAAK,kBAAkB"}
+{"version":3,"file":"base.cjs","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":";;AAMA,MAAa,gBAAgB;AAG7B,MAAa,0BACK,wCAAA,eAAe,aAAa,gBAAgB;AAE9D,eAAsB,gBAAgB,EACpC,gBACA,mBACA,WACA,UAMC;CACD,IAAI,aAAa,QAAQ,UAAU,WAAW,GAC5C,OAAO;CAYT,QAAO,MAVqB,QAAQ,IAClC,UAAU,KAAK,aACb,eACG,WAAW,EAAE,SAAS,qBAAqB,CAAC,CAAC,CAC7C,OACC;EAAE,GAAG,SAAS;EAAU,cAAc,SAAS;CAAY,GAC3D,MACF,CACJ,CACF,EAAA,CACqB,KAAK,iBAAiB;AAC7C"}

package/dist/chains/combine_documents/base.js

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

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

@@ -1 +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":";;AAMA,MAAa,gBAAgB;AAG7B,MAAa,0BACK,+BAAe,aAAa,iBAAiB;AAE/D,eAAsB,gBAAgB,EACpC,gBACA,mBACA,WACA,UAMC;AACD,KAAI,aAAa,QAAQ,UAAU,WAAW,EAC5C,QAAO;AAYT,SAAO,MAVqB,QAAQ,IAClC,UAAU,KAAK,aACb,eACG,WAAW,EAAE,SAAS,sBAAsB,CAAC,CAC7C,OACC;EAAE,GAAG,SAAS;EAAU,cAAc,SAAS;EAAa,EAC5D,OACD,CACJ,CACF,EACoB,KAAK,kBAAkB"}
+{"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":";;AAMA,MAAa,gBAAgB;AAG7B,MAAa,0BACK,+BAAe,aAAa,gBAAgB;AAE9D,eAAsB,gBAAgB,EACpC,gBACA,mBACA,WACA,UAMC;CACD,IAAI,aAAa,QAAQ,UAAU,WAAW,GAC5C,OAAO;CAYT,QAAO,MAVqB,QAAQ,IAClC,UAAU,KAAK,aACb,eACG,WAAW,EAAE,SAAS,qBAAqB,CAAC,CAAC,CAC7C,OACC;EAAE,GAAG,SAAS;EAAU,cAAc,SAAS;CAAY,GAC3D,MACF,CACJ,CACF,EAAA,CACqB,KAAK,iBAAiB;AAC7C"}

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

@@ -1 +1 @@
-{"version":3,"file":"reduce.cjs","names":[],"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  // oxlint-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,gBACd,MAEA,YACA,UACc;CACd,MAAM,mBAAiC,EAAE;CACzC,IAAI,gBAA4B,EAAE;AAClC,MAAK,MAAM,OAAO,MAAM;AACtB,gBAAc,KAAK,IAAI;AAEvB,MADkB,WAAW,cAChB,GAAG,UAAU;AACxB,OAAI,cAAc,WAAW,EAC3B,OAAM,IAAI,MACR,+EACD;AAEH,oBAAiB,KAAK,cAAc,MAAM,GAAG,GAAG,CAAC;AACjD,mBAAgB,cAAc,MAAM,GAAG;;;AAG3C,kBAAiB,KAAK,cAAc;AACpC,QAAO;;;;;;;;;;;;;;;;AAiBT,eAAsB,aACpB,MACA,qBACmB;AAEnB,QAAO;EAAE,aAAa,MADD,oBAAoB,KAAK;EAChB,UAAU,qBAAqB,KAAK;EAAE;;AAGtE,SAAS,qBAAqB,MAAwC;CACpE,MAAM,mBAA2C,EAAE;AACnD,MAAK,MAAM,OAAO,KAAK,GAAG,SACxB,KAAI,OAAO,KAAK,GAAG,SACjB,kBAAiB,OAAO,OAAO,KAAK,GAAG,SAAS,KAAK;AAGzD,MAAK,MAAM,OAAO,KAAK,MAAM,EAAE,CAC7B,MAAK,MAAM,OAAO,IAAI,SACpB,KAAI,OAAO,iBACT,kBAAiB,QAAQ,KAAK,IAAI,SAAS;KAE3C,kBAAiB,OAAO,OAAO,IAAI,SAAS,KAAK;AAIvD,QAAO"}
+{"version":3,"file":"reduce.cjs","names":[],"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  // oxlint-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,gBACd,MAEA,YACA,UACc;CACd,MAAM,mBAAiC,CAAC;CACxC,IAAI,gBAA4B,CAAC;CACjC,KAAK,MAAM,OAAO,MAAM;EACtB,cAAc,KAAK,GAAG;EAEtB,IADkB,WAAW,aACjB,IAAI,UAAU;GACxB,IAAI,cAAc,WAAW,GAC3B,MAAM,IAAI,MACR,8EACF;GAEF,iBAAiB,KAAK,cAAc,MAAM,GAAG,EAAE,CAAC;GAChD,gBAAgB,cAAc,MAAM,EAAE;EACxC;CACF;CACA,iBAAiB,KAAK,aAAa;CACnC,OAAO;AACT;;;;;;;;;;;;;;;AAgBA,eAAsB,aACpB,MACA,qBACmB;CAEnB,OAAO;EAAE,aAAa,MADD,oBAAoB,IAAI;EACf,UAAU,qBAAqB,IAAI;CAAE;AACrE;AAEA,SAAS,qBAAqB,MAAwC;CACpE,MAAM,mBAA2C,CAAC;CAClD,KAAK,MAAM,OAAO,KAAK,EAAE,CAAC,UACxB,IAAI,OAAO,KAAK,EAAE,CAAC,UACjB,iBAAiB,OAAO,OAAO,KAAK,EAAE,CAAC,SAAS,IAAI;CAGxD,KAAK,MAAM,OAAO,KAAK,MAAM,CAAC,GAC5B,KAAK,MAAM,OAAO,IAAI,UACpB,IAAI,OAAO,kBACT,iBAAiB,QAAQ,KAAK,IAAI,SAAS;MAE3C,iBAAiB,OAAO,OAAO,IAAI,SAAS,IAAI;CAItD,OAAO;AACT"}

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

@@ -1,5 +1,4 @@
 import { Document } from "@langchain/core/documents";
-
 //#region src/chains/combine_documents/reduce.d.ts
 /**
  * Splits a list of documents into sublists based on a maximum token limit.

package/dist/chains/combine_documents/reduce.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"reduce.d.cts","names":[],"sources":["../../../src/chains/combine_documents/reduce.ts"],"mappings":";;;;;AAaA;;;;;;;;;iBAAgB,eAAA,CACd,IAAA,EAAM,QAAA,IAEN,UAAA,MAAgB,IAAA,iBAChB,QAAA,WACC,QAAA;;;AAkCH;;;;;;;;;;;;iBAAsB,YAAA,CACpB,IAAA,EAAM,QAAA,IACN,mBAAA,GAAsB,IAAA,EAAM,QAAA,OAAe,OAAA,WAC1C,OAAA,CAAQ,QAAA"}
+{"version":3,"file":"reduce.d.cts","names":[],"sources":["../../../src/chains/combine_documents/reduce.ts"],"mappings":";;;;;;;;;;;;;iBAagB,gBACd,MAAM,YAEN,gBAAgB,qBAChB,mBACC;;;;;;;;;;;;;;;iBAkCmB,aACpB,MAAM,YACN,sBAAsB,MAAM,eAAe,kBAC1C,QAAQ"}

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

@@ -1,5 +1,4 @@
 import { Document } from "@langchain/core/documents";
-
 //#region src/chains/combine_documents/reduce.d.ts
 /**
  * Splits a list of documents into sublists based on a maximum token limit.

package/dist/chains/combine_documents/reduce.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"reduce.d.ts","names":[],"sources":["../../../src/chains/combine_documents/reduce.ts"],"mappings":";;;;;AAaA;;;;;;;;;iBAAgB,eAAA,CACd,IAAA,EAAM,QAAA,IAEN,UAAA,MAAgB,IAAA,iBAChB,QAAA,WACC,QAAA;;;AAkCH;;;;;;;;;;;;iBAAsB,YAAA,CACpB,IAAA,EAAM,QAAA,IACN,mBAAA,GAAsB,IAAA,EAAM,QAAA,OAAe,OAAA,WAC1C,OAAA,CAAQ,QAAA"}
+{"version":3,"file":"reduce.d.ts","names":[],"sources":["../../../src/chains/combine_documents/reduce.ts"],"mappings":";;;;;;;;;;;;;iBAagB,gBACd,MAAM,YAEN,gBAAgB,qBAChB,mBACC;;;;;;;;;;;;;;;iBAkCmB,aACpB,MAAM,YACN,sBAAsB,MAAM,eAAe,kBAC1C,QAAQ"}

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

@@ -1 +1 @@
-{"version":3,"file":"reduce.js","names":[],"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  // oxlint-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,gBACd,MAEA,YACA,UACc;CACd,MAAM,mBAAiC,EAAE;CACzC,IAAI,gBAA4B,EAAE;AAClC,MAAK,MAAM,OAAO,MAAM;AACtB,gBAAc,KAAK,IAAI;AAEvB,MADkB,WAAW,cAChB,GAAG,UAAU;AACxB,OAAI,cAAc,WAAW,EAC3B,OAAM,IAAI,MACR,+EACD;AAEH,oBAAiB,KAAK,cAAc,MAAM,GAAG,GAAG,CAAC;AACjD,mBAAgB,cAAc,MAAM,GAAG;;;AAG3C,kBAAiB,KAAK,cAAc;AACpC,QAAO;;;;;;;;;;;;;;;;AAiBT,eAAsB,aACpB,MACA,qBACmB;AAEnB,QAAO;EAAE,aAAa,MADD,oBAAoB,KAAK;EAChB,UAAU,qBAAqB,KAAK;EAAE;;AAGtE,SAAS,qBAAqB,MAAwC;CACpE,MAAM,mBAA2C,EAAE;AACnD,MAAK,MAAM,OAAO,KAAK,GAAG,SACxB,KAAI,OAAO,KAAK,GAAG,SACjB,kBAAiB,OAAO,OAAO,KAAK,GAAG,SAAS,KAAK;AAGzD,MAAK,MAAM,OAAO,KAAK,MAAM,EAAE,CAC7B,MAAK,MAAM,OAAO,IAAI,SACpB,KAAI,OAAO,iBACT,kBAAiB,QAAQ,KAAK,IAAI,SAAS;KAE3C,kBAAiB,OAAO,OAAO,IAAI,SAAS,KAAK;AAIvD,QAAO"}
+{"version":3,"file":"reduce.js","names":[],"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  // oxlint-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,gBACd,MAEA,YACA,UACc;CACd,MAAM,mBAAiC,CAAC;CACxC,IAAI,gBAA4B,CAAC;CACjC,KAAK,MAAM,OAAO,MAAM;EACtB,cAAc,KAAK,GAAG;EAEtB,IADkB,WAAW,aACjB,IAAI,UAAU;GACxB,IAAI,cAAc,WAAW,GAC3B,MAAM,IAAI,MACR,8EACF;GAEF,iBAAiB,KAAK,cAAc,MAAM,GAAG,EAAE,CAAC;GAChD,gBAAgB,cAAc,MAAM,EAAE;EACxC;CACF;CACA,iBAAiB,KAAK,aAAa;CACnC,OAAO;AACT;;;;;;;;;;;;;;;AAgBA,eAAsB,aACpB,MACA,qBACmB;CAEnB,OAAO;EAAE,aAAa,MADD,oBAAoB,IAAI;EACf,UAAU,qBAAqB,IAAI;CAAE;AACrE;AAEA,SAAS,qBAAqB,MAAwC;CACpE,MAAM,mBAA2C,CAAC;CAClD,KAAK,MAAM,OAAO,KAAK,EAAE,CAAC,UACxB,IAAI,OAAO,KAAK,EAAE,CAAC,UACjB,iBAAiB,OAAO,OAAO,KAAK,EAAE,CAAC,SAAS,IAAI;CAGxD,KAAK,MAAM,OAAO,KAAK,MAAM,CAAC,GAC5B,KAAK,MAAM,OAAO,IAAI,UACpB,IAAI,OAAO,kBACT,iBAAiB,QAAQ,KAAK,IAAI,SAAS;MAE3C,iBAAiB,OAAO,OAAO,IAAI,SAAS,IAAI;CAItD,OAAO;AACT"}

package/dist/chains/combine_documents/stuff.cjs

@@ -1,4 +1,3 @@
-require("../../_virtual/_rolldown/runtime.cjs");
 const require_base = require("./base.cjs");
 let _langchain_core_runnables = require("@langchain/core/runnables");
 let _langchain_core_output_parsers = require("@langchain/core/output_parsers");