@langchain/classic 1.0.23 → 1.0.24-dev-1773698445534

package/dist/chains/retrieval.js

@@ -1,6 +1,5 @@
 import { __exportAll } from "../_virtual/_rolldown/runtime.js";
 import { RunnablePassthrough, RunnableSequence } from "@langchain/core/runnables";
-
 //#region src/chains/retrieval.ts
 var retrieval_exports = /* @__PURE__ */ __exportAll({ createRetrievalChain: () => createRetrievalChain });
 function isBaseRetriever(x) {
@@ -41,7 +40,7 @@ async function createRetrievalChain({ retriever, combineDocsChain }) {
 		chat_history: (input) => input.chat_history ?? []
 	}), RunnablePassthrough.assign({ answer: combineDocsChain })]).withConfig({ runName: "retrieval_chain" });
 }
-
 //#endregion
 export { createRetrievalChain, retrieval_exports };
+
 //# sourceMappingURL=retrieval.js.map

package/dist/chains/retrieval.js.map

@@ -1 +1 @@
-{"version":3,"file":"retrieval.js","names":[],"sources":["../../src/chains/retrieval.ts"],"sourcesContent":["import type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport {\n  type Runnable,\n  RunnableSequence,\n  type RunnableInterface,\n  RunnablePassthrough,\n} from \"@langchain/core/runnables\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\nimport type { DocumentInterface, Document } from \"@langchain/core/documents\";\n\n/**\n * Parameters for the createRetrievalChain method.\n */\nexport type CreateRetrievalChainParams<RunOutput> = {\n  /**\n   * Retriever-like object that returns list of documents. Should\n   * either be a subclass of BaseRetriever or a Runnable that returns\n   * a list of documents. If a subclass of BaseRetriever, then it\n   * is expected that an `input` key be passed in - this is what\n   * is will be used to pass into the retriever. If this is NOT a\n   * subclass of BaseRetriever, then all the inputs will be passed\n   * into this runnable, meaning that runnable should take a object\n   * as input.\n   */\n  retriever:\n    | BaseRetrieverInterface\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    | RunnableInterface<Record<string, any>, DocumentInterface[]>;\n  /**\n   * Runnable that takes inputs and produces a string output.\n   * The inputs to this will be any original inputs to this chain, a new\n   * context key with the retrieved documents, and chat_history (if not present\n   * in the inputs) with a value of `[]` (to easily enable conversational\n   * retrieval).\n   */\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  combineDocsChain: RunnableInterface<Record<string, any>, RunOutput>;\n};\n\nfunction isBaseRetriever(x: unknown): x is BaseRetrieverInterface {\n  return !!x && typeof (x as BaseRetrieverInterface).invoke === \"function\";\n}\n\n/**\n * Create a retrieval chain that retrieves documents and then passes them on.\n * @param {CreateRetrievalChainParams} params A params object\n *     containing a retriever and a combineDocsChain.\n * @returns An LCEL Runnable which returns a an object\n *     containing at least `context` and `answer` keys.\n * @example\n * ```typescript\n * // pnpm add langchain @langchain/openai\n *\n * import { ChatOpenAI } from \"@langchain/openai\";\n * import { pull } from \"langchain/hub\";\n * import { createRetrievalChain } from \"@langchain/classic/chains/retrieval\";\n * import { createStuffDocumentsChain } from \"@langchain/classic/chains/combine_documents\";\n *\n * const retrievalQAChatPrompt = await pull(\"langchain-ai/retrieval-qa-chat\");\n * const llm = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const retriever = ...\n * const combineDocsChain = await createStuffDocumentsChain(...);\n * const retrievalChain = await createRetrievalChain({\n *   retriever,\n *   combineDocsChain,\n * });\n * const response = await chain.invoke({ input: \"...\" });\n * ```\n */\nexport async function createRetrievalChain<RunOutput>({\n  retriever,\n  combineDocsChain,\n}: CreateRetrievalChainParams<RunOutput>): Promise<\n  Runnable<\n    { input: string; chat_history?: BaseMessage[] | string } & {\n      [key: string]: unknown;\n    },\n    { context: Document[]; answer: RunOutput } & { [key: string]: unknown }\n  >\n> {\n  let retrieveDocumentsChain: Runnable<{ input: string }, DocumentInterface[]>;\n  if (isBaseRetriever(retriever)) {\n    retrieveDocumentsChain = RunnableSequence.from([\n      (input) => input.input,\n      retriever,\n    ]);\n  } else {\n    // TODO: Fix typing by adding withConfig to core RunnableInterface\n    retrieveDocumentsChain = retriever as Runnable;\n  }\n  const retrievalChain = RunnableSequence.from<{\n    input: string;\n    chat_history?: BaseMessage[] | string;\n  }>([\n    RunnablePassthrough.assign({\n      context: retrieveDocumentsChain.withConfig({\n        runName: \"retrieve_documents\",\n      }),\n      chat_history: (input) => input.chat_history ?? [],\n    }),\n    RunnablePassthrough.assign({\n      answer: combineDocsChain,\n    }),\n  ]).withConfig({ runName: \"retrieval_chain\" });\n  return retrievalChain;\n}\n"],"mappings":";;;;;AAuCA,SAAS,gBAAgB,GAAyC;AAChE,QAAO,CAAC,CAAC,KAAK,OAAQ,EAA6B,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA6BhE,eAAsB,qBAAgC,EACpD,WACA,oBAQA;CACA,IAAI;AACJ,KAAI,gBAAgB,UAAU,CAC5B,0BAAyB,iBAAiB,KAAK,EAC5C,UAAU,MAAM,OACjB,UACD,CAAC;KAGF,0BAAyB;AAgB3B,QAduB,iBAAiB,KAGrC,CACD,oBAAoB,OAAO;EACzB,SAAS,uBAAuB,WAAW,EACzC,SAAS,sBACV,CAAC;EACF,eAAe,UAAU,MAAM,gBAAgB,EAAE;EAClD,CAAC,EACF,oBAAoB,OAAO,EACzB,QAAQ,kBACT,CAAC,CACH,CAAC,CAAC,WAAW,EAAE,SAAS,mBAAmB,CAAC"}
+{"version":3,"file":"retrieval.js","names":[],"sources":["../../src/chains/retrieval.ts"],"sourcesContent":["import type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport {\n  type Runnable,\n  RunnableSequence,\n  type RunnableInterface,\n  RunnablePassthrough,\n} from \"@langchain/core/runnables\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\nimport type { DocumentInterface, Document } from \"@langchain/core/documents\";\n\n/**\n * Parameters for the createRetrievalChain method.\n */\nexport type CreateRetrievalChainParams<RunOutput> = {\n  /**\n   * Retriever-like object that returns list of documents. Should\n   * either be a subclass of BaseRetriever or a Runnable that returns\n   * a list of documents. If a subclass of BaseRetriever, then it\n   * is expected that an `input` key be passed in - this is what\n   * is will be used to pass into the retriever. If this is NOT a\n   * subclass of BaseRetriever, then all the inputs will be passed\n   * into this runnable, meaning that runnable should take a object\n   * as input.\n   */\n  retriever:\n    | BaseRetrieverInterface\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    | RunnableInterface<Record<string, any>, DocumentInterface[]>;\n  /**\n   * Runnable that takes inputs and produces a string output.\n   * The inputs to this will be any original inputs to this chain, a new\n   * context key with the retrieved documents, and chat_history (if not present\n   * in the inputs) with a value of `[]` (to easily enable conversational\n   * retrieval).\n   */\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  combineDocsChain: RunnableInterface<Record<string, any>, RunOutput>;\n};\n\nfunction isBaseRetriever(x: unknown): x is BaseRetrieverInterface {\n  return !!x && typeof (x as BaseRetrieverInterface).invoke === \"function\";\n}\n\n/**\n * Create a retrieval chain that retrieves documents and then passes them on.\n * @param {CreateRetrievalChainParams} params A params object\n *     containing a retriever and a combineDocsChain.\n * @returns An LCEL Runnable which returns a an object\n *     containing at least `context` and `answer` keys.\n * @example\n * ```typescript\n * // pnpm add langchain @langchain/openai\n *\n * import { ChatOpenAI } from \"@langchain/openai\";\n * import { pull } from \"langchain/hub\";\n * import { createRetrievalChain } from \"@langchain/classic/chains/retrieval\";\n * import { createStuffDocumentsChain } from \"@langchain/classic/chains/combine_documents\";\n *\n * const retrievalQAChatPrompt = await pull(\"langchain-ai/retrieval-qa-chat\");\n * const llm = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const retriever = ...\n * const combineDocsChain = await createStuffDocumentsChain(...);\n * const retrievalChain = await createRetrievalChain({\n *   retriever,\n *   combineDocsChain,\n * });\n * const response = await chain.invoke({ input: \"...\" });\n * ```\n */\nexport async function createRetrievalChain<RunOutput>({\n  retriever,\n  combineDocsChain,\n}: CreateRetrievalChainParams<RunOutput>): Promise<\n  Runnable<\n    { input: string; chat_history?: BaseMessage[] | string } & {\n      [key: string]: unknown;\n    },\n    { context: Document[]; answer: RunOutput } & { [key: string]: unknown }\n  >\n> {\n  let retrieveDocumentsChain: Runnable<{ input: string }, DocumentInterface[]>;\n  if (isBaseRetriever(retriever)) {\n    retrieveDocumentsChain = RunnableSequence.from([\n      (input) => input.input,\n      retriever,\n    ]);\n  } else {\n    // TODO: Fix typing by adding withConfig to core RunnableInterface\n    retrieveDocumentsChain = retriever as Runnable;\n  }\n  const retrievalChain = RunnableSequence.from<{\n    input: string;\n    chat_history?: BaseMessage[] | string;\n  }>([\n    RunnablePassthrough.assign({\n      context: retrieveDocumentsChain.withConfig({\n        runName: \"retrieve_documents\",\n      }),\n      chat_history: (input) => input.chat_history ?? [],\n    }),\n    RunnablePassthrough.assign({\n      answer: combineDocsChain,\n    }),\n  ]).withConfig({ runName: \"retrieval_chain\" });\n  return retrievalChain;\n}\n"],"mappings":";;;;AAuCA,SAAS,gBAAgB,GAAyC;AAChE,QAAO,CAAC,CAAC,KAAK,OAAQ,EAA6B,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA6BhE,eAAsB,qBAAgC,EACpD,WACA,oBAQA;CACA,IAAI;AACJ,KAAI,gBAAgB,UAAU,CAC5B,0BAAyB,iBAAiB,KAAK,EAC5C,UAAU,MAAM,OACjB,UACD,CAAC;KAGF,0BAAyB;AAgB3B,QAduB,iBAAiB,KAGrC,CACD,oBAAoB,OAAO;EACzB,SAAS,uBAAuB,WAAW,EACzC,SAAS,sBACV,CAAC;EACF,eAAe,UAAU,MAAM,gBAAgB,EAAE;EAClD,CAAC,EACF,oBAAoB,OAAO,EACzB,QAAQ,kBACT,CAAC,CACH,CAAC,CAAC,WAAW,EAAE,SAAS,mBAAmB,CAAC"}

package/dist/chains/retrieval_qa.cjs

@@ -1,6 +1,5 @@
-const require_base = require('./base.cjs');
-const require_load = require('./question_answering/load.cjs');
-
+const require_base = require("./base.cjs");
+const require_load = require("./question_answering/load.cjs");
 //#region src/chains/retrieval_qa.ts
 /**
 * Class representing a chain for performing question-answering tasks with
@@ -98,7 +97,7 @@ var RetrievalQAChain = class extends require_base.BaseChain {
 		});
 	}
 };
-
 //#endregion
 exports.RetrievalQAChain = RetrievalQAChain;
+
 //# sourceMappingURL=retrieval_qa.cjs.map

package/dist/chains/retrieval_qa.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"retrieval_qa.cjs","names":["BaseChain","loadQAStuffChain"],"sources":["../../src/chains/retrieval_qa.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { SerializedVectorDBQAChain } from \"./serde.js\";\nimport {\n  StuffQAChainParams,\n  loadQAStuffChain,\n} from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\n/**\n * Interface for the input parameters of the RetrievalQAChain class.\n */\nexport interface RetrievalQAChainInput extends Omit<ChainInputs, \"memory\"> {\n  retriever: BaseRetrieverInterface;\n  combineDocumentsChain: BaseChain;\n  inputKey?: string;\n  returnSourceDocuments?: boolean;\n}\n\n/**\n * Class representing a chain for performing question-answering tasks with\n * a retrieval component.\n * @example\n * ```typescript\n * import { createStuffDocumentsChain } from \"@langchain/classic/chains/combine_documents\";\n * import { ChatPromptTemplate } from \"@langchain/core/prompts\";\n * import { createRetrievalChain } from \"@langchain/classic/chains/retrieval\";\n * import { MemoryVectorStore } from \"@langchain/classic/vectorstores/memory\";\n *\n * const documents = [...your documents here];\n * const embeddings = ...your embeddings model;\n * const llm = ...your LLM model;\n *\n * const vectorstore = await MemoryVectorStore.fromDocuments(\n *   documents,\n *   embeddings\n * );\n * const prompt = ChatPromptTemplate.fromTemplate(`Answer the user's question: {input} based on the following context {context}`);\n *\n * const combineDocsChain = await createStuffDocumentsChain({\n *   llm,\n *   prompt,\n * });\n * const retriever = vectorstore.asRetriever();\n *\n * const retrievalChain = await createRetrievalChain({\n *   combineDocsChain,\n *   retriever,\n * });\n * ```\n */\nexport class RetrievalQAChain\n  extends BaseChain\n  implements RetrievalQAChainInput\n{\n  static lc_name() {\n    return \"RetrievalQAChain\";\n  }\n\n  inputKey = \"query\";\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentsChain.outputKeys.concat(\n      this.returnSourceDocuments ? [\"sourceDocuments\"] : []\n    );\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: RetrievalQAChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\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    const question: string = values[this.inputKey];\n    const docs = await this.retriever.invoke(\n      question,\n      runManager?.getChild(\"retriever\")\n    );\n    const inputs = { question, input_documents: docs, ...values };\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 \"retrieval_qa\" as const;\n  }\n\n  static async deserialize(\n    _data: SerializedVectorDBQAChain,\n    _values: LoadValues\n  ): Promise<RetrievalQAChain> {\n    throw new Error(\"Not implemented\");\n  }\n\n  serialize(): SerializedVectorDBQAChain {\n    throw new Error(\"Not implemented\");\n  }\n\n  /**\n   * Creates a new instance of RetrievalQAChain using a BaseLanguageModel\n   * and a BaseRetriever.\n   * @param llm The BaseLanguageModel used to generate a new question.\n   * @param retriever The BaseRetriever used to retrieve relevant documents.\n   * @param options Optional parameters for the RetrievalQAChain.\n   * @returns A new instance of RetrievalQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<\n        RetrievalQAChainInput,\n        \"retriever\" | \"combineDocumentsChain\" | \"index\"\n      >\n    > &\n      StuffQAChainParams\n  ): RetrievalQAChain {\n    const qaChain = loadQAStuffChain(llm, {\n      prompt: options?.prompt,\n    });\n    return new this({\n      ...options,\n      retriever,\n      combineDocumentsChain: qaChain,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDA,IAAa,mBAAb,cACUA,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT,WAAW;CAEX,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,SAAS;;CAGxB,IAAI,aAAa;AACf,SAAO,KAAK,sBAAsB,WAAW,OAC3C,KAAK,wBAAwB,CAAC,kBAAkB,GAAG,EAAE,CACtD;;CAGH;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAA+B;AACzC,QAAM,OAAO;AACb,OAAK,YAAY,OAAO;AACxB,OAAK,wBAAwB,OAAO;AACpC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,wBACH,OAAO,yBAAyB,KAAK;;;CAIzC,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,iBAAiB,KAAK,SAAS,cAAc;EAE/D,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,OAAO,MAAM,KAAK,UAAU,OAChC,UACA,YAAY,SAAS,YAAY,CAClC;EACD,MAAM,SAAS;GAAE;GAAU,iBAAiB;GAAM,GAAG;GAAQ;EAC7D,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,OACA,SAC2B;AAC3B,QAAM,IAAI,MAAM,kBAAkB;;CAGpC,YAAuC;AACrC,QAAM,IAAI,MAAM,kBAAkB;;;;;;;;;;CAWpC,OAAO,QACL,KACA,WACA,SAOkB;EAClB,MAAM,UAAUC,8BAAiB,KAAK,EACpC,QAAQ,SAAS,QAClB,CAAC;AACF,SAAO,IAAI,KAAK;GACd,GAAG;GACH;GACA,uBAAuB;GACxB,CAAC"}
+{"version":3,"file":"retrieval_qa.cjs","names":["BaseChain","loadQAStuffChain"],"sources":["../../src/chains/retrieval_qa.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { SerializedVectorDBQAChain } from \"./serde.js\";\nimport {\n  StuffQAChainParams,\n  loadQAStuffChain,\n} from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\n/**\n * Interface for the input parameters of the RetrievalQAChain class.\n */\nexport interface RetrievalQAChainInput extends Omit<ChainInputs, \"memory\"> {\n  retriever: BaseRetrieverInterface;\n  combineDocumentsChain: BaseChain;\n  inputKey?: string;\n  returnSourceDocuments?: boolean;\n}\n\n/**\n * Class representing a chain for performing question-answering tasks with\n * a retrieval component.\n * @example\n * ```typescript\n * import { createStuffDocumentsChain } from \"@langchain/classic/chains/combine_documents\";\n * import { ChatPromptTemplate } from \"@langchain/core/prompts\";\n * import { createRetrievalChain } from \"@langchain/classic/chains/retrieval\";\n * import { MemoryVectorStore } from \"@langchain/classic/vectorstores/memory\";\n *\n * const documents = [...your documents here];\n * const embeddings = ...your embeddings model;\n * const llm = ...your LLM model;\n *\n * const vectorstore = await MemoryVectorStore.fromDocuments(\n *   documents,\n *   embeddings\n * );\n * const prompt = ChatPromptTemplate.fromTemplate(`Answer the user's question: {input} based on the following context {context}`);\n *\n * const combineDocsChain = await createStuffDocumentsChain({\n *   llm,\n *   prompt,\n * });\n * const retriever = vectorstore.asRetriever();\n *\n * const retrievalChain = await createRetrievalChain({\n *   combineDocsChain,\n *   retriever,\n * });\n * ```\n */\nexport class RetrievalQAChain\n  extends BaseChain\n  implements RetrievalQAChainInput\n{\n  static lc_name() {\n    return \"RetrievalQAChain\";\n  }\n\n  inputKey = \"query\";\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentsChain.outputKeys.concat(\n      this.returnSourceDocuments ? [\"sourceDocuments\"] : []\n    );\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: RetrievalQAChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\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    const question: string = values[this.inputKey];\n    const docs = await this.retriever.invoke(\n      question,\n      runManager?.getChild(\"retriever\")\n    );\n    const inputs = { question, input_documents: docs, ...values };\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 \"retrieval_qa\" as const;\n  }\n\n  static async deserialize(\n    _data: SerializedVectorDBQAChain,\n    _values: LoadValues\n  ): Promise<RetrievalQAChain> {\n    throw new Error(\"Not implemented\");\n  }\n\n  serialize(): SerializedVectorDBQAChain {\n    throw new Error(\"Not implemented\");\n  }\n\n  /**\n   * Creates a new instance of RetrievalQAChain using a BaseLanguageModel\n   * and a BaseRetriever.\n   * @param llm The BaseLanguageModel used to generate a new question.\n   * @param retriever The BaseRetriever used to retrieve relevant documents.\n   * @param options Optional parameters for the RetrievalQAChain.\n   * @returns A new instance of RetrievalQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<\n        RetrievalQAChainInput,\n        \"retriever\" | \"combineDocumentsChain\" | \"index\"\n      >\n    > &\n      StuffQAChainParams\n  ): RetrievalQAChain {\n    const qaChain = loadQAStuffChain(llm, {\n      prompt: options?.prompt,\n    });\n    return new this({\n      ...options,\n      retriever,\n      combineDocumentsChain: qaChain,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDA,IAAa,mBAAb,cACUA,aAAAA,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT,WAAW;CAEX,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,SAAS;;CAGxB,IAAI,aAAa;AACf,SAAO,KAAK,sBAAsB,WAAW,OAC3C,KAAK,wBAAwB,CAAC,kBAAkB,GAAG,EAAE,CACtD;;CAGH;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAA+B;AACzC,QAAM,OAAO;AACb,OAAK,YAAY,OAAO;AACxB,OAAK,wBAAwB,OAAO;AACpC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,wBACH,OAAO,yBAAyB,KAAK;;;CAIzC,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,iBAAiB,KAAK,SAAS,cAAc;EAE/D,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,OAAO,MAAM,KAAK,UAAU,OAChC,UACA,YAAY,SAAS,YAAY,CAClC;EACD,MAAM,SAAS;GAAE;GAAU,iBAAiB;GAAM,GAAG;GAAQ;EAC7D,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,OACA,SAC2B;AAC3B,QAAM,IAAI,MAAM,kBAAkB;;CAGpC,YAAuC;AACrC,QAAM,IAAI,MAAM,kBAAkB;;;;;;;;;;CAWpC,OAAO,QACL,KACA,WACA,SAOkB;EAClB,MAAM,UAAUC,aAAAA,iBAAiB,KAAK,EACpC,QAAQ,SAAS,QAClB,CAAC;AACF,SAAO,IAAI,KAAK;GACd,GAAG;GACH;GACA,uBAAuB;GACxB,CAAC"}

package/dist/chains/retrieval_qa.js

@@ -1,6 +1,5 @@
 import { BaseChain } from "./base.js";
 import { loadQAStuffChain } from "./question_answering/load.js";
-
 //#region src/chains/retrieval_qa.ts
 /**
 * Class representing a chain for performing question-answering tasks with
@@ -98,7 +97,7 @@ var RetrievalQAChain = class extends BaseChain {
 		});
 	}
 };
-
 //#endregion
 export { RetrievalQAChain };
+
 //# sourceMappingURL=retrieval_qa.js.map

package/dist/chains/retrieval_qa.js.map

@@ -1 +1 @@
-{"version":3,"file":"retrieval_qa.js","names":[],"sources":["../../src/chains/retrieval_qa.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { SerializedVectorDBQAChain } from \"./serde.js\";\nimport {\n  StuffQAChainParams,\n  loadQAStuffChain,\n} from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\n/**\n * Interface for the input parameters of the RetrievalQAChain class.\n */\nexport interface RetrievalQAChainInput extends Omit<ChainInputs, \"memory\"> {\n  retriever: BaseRetrieverInterface;\n  combineDocumentsChain: BaseChain;\n  inputKey?: string;\n  returnSourceDocuments?: boolean;\n}\n\n/**\n * Class representing a chain for performing question-answering tasks with\n * a retrieval component.\n * @example\n * ```typescript\n * import { createStuffDocumentsChain } from \"@langchain/classic/chains/combine_documents\";\n * import { ChatPromptTemplate } from \"@langchain/core/prompts\";\n * import { createRetrievalChain } from \"@langchain/classic/chains/retrieval\";\n * import { MemoryVectorStore } from \"@langchain/classic/vectorstores/memory\";\n *\n * const documents = [...your documents here];\n * const embeddings = ...your embeddings model;\n * const llm = ...your LLM model;\n *\n * const vectorstore = await MemoryVectorStore.fromDocuments(\n *   documents,\n *   embeddings\n * );\n * const prompt = ChatPromptTemplate.fromTemplate(`Answer the user's question: {input} based on the following context {context}`);\n *\n * const combineDocsChain = await createStuffDocumentsChain({\n *   llm,\n *   prompt,\n * });\n * const retriever = vectorstore.asRetriever();\n *\n * const retrievalChain = await createRetrievalChain({\n *   combineDocsChain,\n *   retriever,\n * });\n * ```\n */\nexport class RetrievalQAChain\n  extends BaseChain\n  implements RetrievalQAChainInput\n{\n  static lc_name() {\n    return \"RetrievalQAChain\";\n  }\n\n  inputKey = \"query\";\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentsChain.outputKeys.concat(\n      this.returnSourceDocuments ? [\"sourceDocuments\"] : []\n    );\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: RetrievalQAChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\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    const question: string = values[this.inputKey];\n    const docs = await this.retriever.invoke(\n      question,\n      runManager?.getChild(\"retriever\")\n    );\n    const inputs = { question, input_documents: docs, ...values };\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 \"retrieval_qa\" as const;\n  }\n\n  static async deserialize(\n    _data: SerializedVectorDBQAChain,\n    _values: LoadValues\n  ): Promise<RetrievalQAChain> {\n    throw new Error(\"Not implemented\");\n  }\n\n  serialize(): SerializedVectorDBQAChain {\n    throw new Error(\"Not implemented\");\n  }\n\n  /**\n   * Creates a new instance of RetrievalQAChain using a BaseLanguageModel\n   * and a BaseRetriever.\n   * @param llm The BaseLanguageModel used to generate a new question.\n   * @param retriever The BaseRetriever used to retrieve relevant documents.\n   * @param options Optional parameters for the RetrievalQAChain.\n   * @returns A new instance of RetrievalQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<\n        RetrievalQAChainInput,\n        \"retriever\" | \"combineDocumentsChain\" | \"index\"\n      >\n    > &\n      StuffQAChainParams\n  ): RetrievalQAChain {\n    const qaChain = loadQAStuffChain(llm, {\n      prompt: options?.prompt,\n    });\n    return new this({\n      ...options,\n      retriever,\n      combineDocumentsChain: qaChain,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDA,IAAa,mBAAb,cACU,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT,WAAW;CAEX,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,SAAS;;CAGxB,IAAI,aAAa;AACf,SAAO,KAAK,sBAAsB,WAAW,OAC3C,KAAK,wBAAwB,CAAC,kBAAkB,GAAG,EAAE,CACtD;;CAGH;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAA+B;AACzC,QAAM,OAAO;AACb,OAAK,YAAY,OAAO;AACxB,OAAK,wBAAwB,OAAO;AACpC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,wBACH,OAAO,yBAAyB,KAAK;;;CAIzC,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,iBAAiB,KAAK,SAAS,cAAc;EAE/D,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,OAAO,MAAM,KAAK,UAAU,OAChC,UACA,YAAY,SAAS,YAAY,CAClC;EACD,MAAM,SAAS;GAAE;GAAU,iBAAiB;GAAM,GAAG;GAAQ;EAC7D,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,OACA,SAC2B;AAC3B,QAAM,IAAI,MAAM,kBAAkB;;CAGpC,YAAuC;AACrC,QAAM,IAAI,MAAM,kBAAkB;;;;;;;;;;CAWpC,OAAO,QACL,KACA,WACA,SAOkB;EAClB,MAAM,UAAU,iBAAiB,KAAK,EACpC,QAAQ,SAAS,QAClB,CAAC;AACF,SAAO,IAAI,KAAK;GACd,GAAG;GACH;GACA,uBAAuB;GACxB,CAAC"}
+{"version":3,"file":"retrieval_qa.js","names":[],"sources":["../../src/chains/retrieval_qa.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { SerializedVectorDBQAChain } from \"./serde.js\";\nimport {\n  StuffQAChainParams,\n  loadQAStuffChain,\n} from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\n/**\n * Interface for the input parameters of the RetrievalQAChain class.\n */\nexport interface RetrievalQAChainInput extends Omit<ChainInputs, \"memory\"> {\n  retriever: BaseRetrieverInterface;\n  combineDocumentsChain: BaseChain;\n  inputKey?: string;\n  returnSourceDocuments?: boolean;\n}\n\n/**\n * Class representing a chain for performing question-answering tasks with\n * a retrieval component.\n * @example\n * ```typescript\n * import { createStuffDocumentsChain } from \"@langchain/classic/chains/combine_documents\";\n * import { ChatPromptTemplate } from \"@langchain/core/prompts\";\n * import { createRetrievalChain } from \"@langchain/classic/chains/retrieval\";\n * import { MemoryVectorStore } from \"@langchain/classic/vectorstores/memory\";\n *\n * const documents = [...your documents here];\n * const embeddings = ...your embeddings model;\n * const llm = ...your LLM model;\n *\n * const vectorstore = await MemoryVectorStore.fromDocuments(\n *   documents,\n *   embeddings\n * );\n * const prompt = ChatPromptTemplate.fromTemplate(`Answer the user's question: {input} based on the following context {context}`);\n *\n * const combineDocsChain = await createStuffDocumentsChain({\n *   llm,\n *   prompt,\n * });\n * const retriever = vectorstore.asRetriever();\n *\n * const retrievalChain = await createRetrievalChain({\n *   combineDocsChain,\n *   retriever,\n * });\n * ```\n */\nexport class RetrievalQAChain\n  extends BaseChain\n  implements RetrievalQAChainInput\n{\n  static lc_name() {\n    return \"RetrievalQAChain\";\n  }\n\n  inputKey = \"query\";\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentsChain.outputKeys.concat(\n      this.returnSourceDocuments ? [\"sourceDocuments\"] : []\n    );\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: RetrievalQAChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\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    const question: string = values[this.inputKey];\n    const docs = await this.retriever.invoke(\n      question,\n      runManager?.getChild(\"retriever\")\n    );\n    const inputs = { question, input_documents: docs, ...values };\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 \"retrieval_qa\" as const;\n  }\n\n  static async deserialize(\n    _data: SerializedVectorDBQAChain,\n    _values: LoadValues\n  ): Promise<RetrievalQAChain> {\n    throw new Error(\"Not implemented\");\n  }\n\n  serialize(): SerializedVectorDBQAChain {\n    throw new Error(\"Not implemented\");\n  }\n\n  /**\n   * Creates a new instance of RetrievalQAChain using a BaseLanguageModel\n   * and a BaseRetriever.\n   * @param llm The BaseLanguageModel used to generate a new question.\n   * @param retriever The BaseRetriever used to retrieve relevant documents.\n   * @param options Optional parameters for the RetrievalQAChain.\n   * @returns A new instance of RetrievalQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<\n        RetrievalQAChainInput,\n        \"retriever\" | \"combineDocumentsChain\" | \"index\"\n      >\n    > &\n      StuffQAChainParams\n  ): RetrievalQAChain {\n    const qaChain = loadQAStuffChain(llm, {\n      prompt: options?.prompt,\n    });\n    return new this({\n      ...options,\n      retriever,\n      combineDocumentsChain: qaChain,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDA,IAAa,mBAAb,cACU,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT,WAAW;CAEX,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,SAAS;;CAGxB,IAAI,aAAa;AACf,SAAO,KAAK,sBAAsB,WAAW,OAC3C,KAAK,wBAAwB,CAAC,kBAAkB,GAAG,EAAE,CACtD;;CAGH;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAA+B;AACzC,QAAM,OAAO;AACb,OAAK,YAAY,OAAO;AACxB,OAAK,wBAAwB,OAAO;AACpC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,wBACH,OAAO,yBAAyB,KAAK;;;CAIzC,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,iBAAiB,KAAK,SAAS,cAAc;EAE/D,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,OAAO,MAAM,KAAK,UAAU,OAChC,UACA,YAAY,SAAS,YAAY,CAClC;EACD,MAAM,SAAS;GAAE;GAAU,iBAAiB;GAAM,GAAG;GAAQ;EAC7D,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,OACA,SAC2B;AAC3B,QAAM,IAAI,MAAM,kBAAkB;;CAGpC,YAAuC;AACrC,QAAM,IAAI,MAAM,kBAAkB;;;;;;;;;;CAWpC,OAAO,QACL,KACA,WACA,SAOkB;EAClB,MAAM,UAAU,iBAAiB,KAAK,EACpC,QAAQ,SAAS,QAClB,CAAC;AACF,SAAO,IAAI,KAAK;GACd,GAAG;GACH;GACA,uBAAuB;GACxB,CAAC"}

package/dist/chains/router/llm_router.cjs

@@ -1,6 +1,5 @@
-const require_llm_chain = require('../llm_chain.cjs');
-const require_multi_route = require('./multi_route.cjs');
-
+const require_llm_chain = require("../llm_chain.cjs");
+const require_multi_route = require("./multi_route.cjs");
 //#region src/chains/router/llm_router.ts
 /**
 * A class that represents an LLM router chain in the LangChain framework.
@@ -44,7 +43,7 @@ var LLMRouterChain = class LLMRouterChain extends require_multi_route.RouterChai
 		});
 	}
 };
-
 //#endregion
 exports.LLMRouterChain = LLMRouterChain;
+
 //# sourceMappingURL=llm_router.cjs.map

package/dist/chains/router/llm_router.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"llm_router.cjs","names":["RouterChain","LLMChain"],"sources":["../../../src/chains/router/llm_router.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { RouterChain } from \"./multi_route.js\";\nimport { ChainInputs } from \"../../chains/base.js\";\n\n/**\n * A type that represents the output schema of a router chain. It defines\n * the structure of the output data returned by the router chain.\n */\nexport type RouterOutputSchema = {\n  destination: string;\n  next_inputs: { [key: string]: string };\n};\n\n/**\n * An interface that extends the default ChainInputs interface and adds an\n * additional \"llmChain\" property.\n */\nexport interface LLMRouterChainInput extends ChainInputs {\n  llmChain: LLMChain<RouterOutputSchema>;\n}\n\n/**\n * A class that represents an LLM router chain in the LangChain framework.\n * It extends the RouterChain class and implements the LLMRouterChainInput\n * interface. It provides additional functionality specific to LLMs and\n * routing based on LLM predictions.\n */\nexport class LLMRouterChain extends RouterChain implements LLMRouterChainInput {\n  llmChain: LLMChain<RouterOutputSchema>;\n\n  constructor(fields: LLMRouterChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n  }\n\n  get inputKeys(): string[] {\n    return this.llmChain.inputKeys;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun | undefined\n  ): Promise<RouterOutputSchema> {\n    return this.llmChain.predict(values, runManager?.getChild());\n  }\n\n  _chainType(): string {\n    return \"llm_router_chain\";\n  }\n\n  /**\n   * A static method that creates an instance of LLMRouterChain from a\n   * BaseLanguageModel and a BasePromptTemplate. It takes in an optional\n   * options object and returns an instance of LLMRouterChain with the\n   * specified LLMChain.\n   * @param llm A BaseLanguageModel instance.\n   * @param prompt A BasePromptTemplate instance.\n   * @param options Optional LLMRouterChainInput object, excluding \"llmChain\".\n   * @returns An instance of LLMRouterChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    prompt: BasePromptTemplate,\n    options?: Omit<LLMRouterChainInput, \"llmChain\">\n  ) {\n    const llmChain = new LLMChain<RouterOutputSchema>({ llm, prompt });\n    return new LLMRouterChain({ ...options, llmChain });\n  }\n}\n"],"mappings":";;;;;;;;;;AA+BA,IAAa,iBAAb,MAAa,uBAAuBA,gCAA2C;CAC7E;CAEA,YAAY,QAA6B;AACvC,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;;CAGzB,IAAI,YAAsB;AACxB,SAAO,KAAK,SAAS;;CAGvB,MAAM,MACJ,QACA,YAC6B;AAC7B,SAAO,KAAK,SAAS,QAAQ,QAAQ,YAAY,UAAU,CAAC;;CAG9D,aAAqB;AACnB,SAAO;;;;;;;;;;;;CAaT,OAAO,QACL,KACA,QACA,SACA;EACA,MAAM,WAAW,IAAIC,2BAA6B;GAAE;GAAK;GAAQ,CAAC;AAClE,SAAO,IAAI,eAAe;GAAE,GAAG;GAAS;GAAU,CAAC"}
+{"version":3,"file":"llm_router.cjs","names":["RouterChain","LLMChain"],"sources":["../../../src/chains/router/llm_router.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { RouterChain } from \"./multi_route.js\";\nimport { ChainInputs } from \"../../chains/base.js\";\n\n/**\n * A type that represents the output schema of a router chain. It defines\n * the structure of the output data returned by the router chain.\n */\nexport type RouterOutputSchema = {\n  destination: string;\n  next_inputs: { [key: string]: string };\n};\n\n/**\n * An interface that extends the default ChainInputs interface and adds an\n * additional \"llmChain\" property.\n */\nexport interface LLMRouterChainInput extends ChainInputs {\n  llmChain: LLMChain<RouterOutputSchema>;\n}\n\n/**\n * A class that represents an LLM router chain in the LangChain framework.\n * It extends the RouterChain class and implements the LLMRouterChainInput\n * interface. It provides additional functionality specific to LLMs and\n * routing based on LLM predictions.\n */\nexport class LLMRouterChain extends RouterChain implements LLMRouterChainInput {\n  llmChain: LLMChain<RouterOutputSchema>;\n\n  constructor(fields: LLMRouterChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n  }\n\n  get inputKeys(): string[] {\n    return this.llmChain.inputKeys;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun | undefined\n  ): Promise<RouterOutputSchema> {\n    return this.llmChain.predict(values, runManager?.getChild());\n  }\n\n  _chainType(): string {\n    return \"llm_router_chain\";\n  }\n\n  /**\n   * A static method that creates an instance of LLMRouterChain from a\n   * BaseLanguageModel and a BasePromptTemplate. It takes in an optional\n   * options object and returns an instance of LLMRouterChain with the\n   * specified LLMChain.\n   * @param llm A BaseLanguageModel instance.\n   * @param prompt A BasePromptTemplate instance.\n   * @param options Optional LLMRouterChainInput object, excluding \"llmChain\".\n   * @returns An instance of LLMRouterChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    prompt: BasePromptTemplate,\n    options?: Omit<LLMRouterChainInput, \"llmChain\">\n  ) {\n    const llmChain = new LLMChain<RouterOutputSchema>({ llm, prompt });\n    return new LLMRouterChain({ ...options, llmChain });\n  }\n}\n"],"mappings":";;;;;;;;;AA+BA,IAAa,iBAAb,MAAa,uBAAuBA,oBAAAA,YAA2C;CAC7E;CAEA,YAAY,QAA6B;AACvC,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;;CAGzB,IAAI,YAAsB;AACxB,SAAO,KAAK,SAAS;;CAGvB,MAAM,MACJ,QACA,YAC6B;AAC7B,SAAO,KAAK,SAAS,QAAQ,QAAQ,YAAY,UAAU,CAAC;;CAG9D,aAAqB;AACnB,SAAO;;;;;;;;;;;;CAaT,OAAO,QACL,KACA,QACA,SACA;EACA,MAAM,WAAW,IAAIC,kBAAAA,SAA6B;GAAE;GAAK;GAAQ,CAAC;AAClE,SAAO,IAAI,eAAe;GAAE,GAAG;GAAS;GAAU,CAAC"}

package/dist/chains/router/llm_router.js

@@ -1,6 +1,5 @@
 import { LLMChain } from "../llm_chain.js";
 import { RouterChain } from "./multi_route.js";
-
 //#region src/chains/router/llm_router.ts
 /**
 * A class that represents an LLM router chain in the LangChain framework.
@@ -44,7 +43,7 @@ var LLMRouterChain = class LLMRouterChain extends RouterChain {
 		});
 	}
 };
-
 //#endregion
 export { LLMRouterChain };
+
 //# sourceMappingURL=llm_router.js.map

package/dist/chains/router/llm_router.js.map

@@ -1 +1 @@
-{"version":3,"file":"llm_router.js","names":[],"sources":["../../../src/chains/router/llm_router.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { RouterChain } from \"./multi_route.js\";\nimport { ChainInputs } from \"../../chains/base.js\";\n\n/**\n * A type that represents the output schema of a router chain. It defines\n * the structure of the output data returned by the router chain.\n */\nexport type RouterOutputSchema = {\n  destination: string;\n  next_inputs: { [key: string]: string };\n};\n\n/**\n * An interface that extends the default ChainInputs interface and adds an\n * additional \"llmChain\" property.\n */\nexport interface LLMRouterChainInput extends ChainInputs {\n  llmChain: LLMChain<RouterOutputSchema>;\n}\n\n/**\n * A class that represents an LLM router chain in the LangChain framework.\n * It extends the RouterChain class and implements the LLMRouterChainInput\n * interface. It provides additional functionality specific to LLMs and\n * routing based on LLM predictions.\n */\nexport class LLMRouterChain extends RouterChain implements LLMRouterChainInput {\n  llmChain: LLMChain<RouterOutputSchema>;\n\n  constructor(fields: LLMRouterChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n  }\n\n  get inputKeys(): string[] {\n    return this.llmChain.inputKeys;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun | undefined\n  ): Promise<RouterOutputSchema> {\n    return this.llmChain.predict(values, runManager?.getChild());\n  }\n\n  _chainType(): string {\n    return \"llm_router_chain\";\n  }\n\n  /**\n   * A static method that creates an instance of LLMRouterChain from a\n   * BaseLanguageModel and a BasePromptTemplate. It takes in an optional\n   * options object and returns an instance of LLMRouterChain with the\n   * specified LLMChain.\n   * @param llm A BaseLanguageModel instance.\n   * @param prompt A BasePromptTemplate instance.\n   * @param options Optional LLMRouterChainInput object, excluding \"llmChain\".\n   * @returns An instance of LLMRouterChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    prompt: BasePromptTemplate,\n    options?: Omit<LLMRouterChainInput, \"llmChain\">\n  ) {\n    const llmChain = new LLMChain<RouterOutputSchema>({ llm, prompt });\n    return new LLMRouterChain({ ...options, llmChain });\n  }\n}\n"],"mappings":";;;;;;;;;;AA+BA,IAAa,iBAAb,MAAa,uBAAuB,YAA2C;CAC7E;CAEA,YAAY,QAA6B;AACvC,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;;CAGzB,IAAI,YAAsB;AACxB,SAAO,KAAK,SAAS;;CAGvB,MAAM,MACJ,QACA,YAC6B;AAC7B,SAAO,KAAK,SAAS,QAAQ,QAAQ,YAAY,UAAU,CAAC;;CAG9D,aAAqB;AACnB,SAAO;;;;;;;;;;;;CAaT,OAAO,QACL,KACA,QACA,SACA;EACA,MAAM,WAAW,IAAI,SAA6B;GAAE;GAAK;GAAQ,CAAC;AAClE,SAAO,IAAI,eAAe;GAAE,GAAG;GAAS;GAAU,CAAC"}
+{"version":3,"file":"llm_router.js","names":[],"sources":["../../../src/chains/router/llm_router.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { RouterChain } from \"./multi_route.js\";\nimport { ChainInputs } from \"../../chains/base.js\";\n\n/**\n * A type that represents the output schema of a router chain. It defines\n * the structure of the output data returned by the router chain.\n */\nexport type RouterOutputSchema = {\n  destination: string;\n  next_inputs: { [key: string]: string };\n};\n\n/**\n * An interface that extends the default ChainInputs interface and adds an\n * additional \"llmChain\" property.\n */\nexport interface LLMRouterChainInput extends ChainInputs {\n  llmChain: LLMChain<RouterOutputSchema>;\n}\n\n/**\n * A class that represents an LLM router chain in the LangChain framework.\n * It extends the RouterChain class and implements the LLMRouterChainInput\n * interface. It provides additional functionality specific to LLMs and\n * routing based on LLM predictions.\n */\nexport class LLMRouterChain extends RouterChain implements LLMRouterChainInput {\n  llmChain: LLMChain<RouterOutputSchema>;\n\n  constructor(fields: LLMRouterChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n  }\n\n  get inputKeys(): string[] {\n    return this.llmChain.inputKeys;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun | undefined\n  ): Promise<RouterOutputSchema> {\n    return this.llmChain.predict(values, runManager?.getChild());\n  }\n\n  _chainType(): string {\n    return \"llm_router_chain\";\n  }\n\n  /**\n   * A static method that creates an instance of LLMRouterChain from a\n   * BaseLanguageModel and a BasePromptTemplate. It takes in an optional\n   * options object and returns an instance of LLMRouterChain with the\n   * specified LLMChain.\n   * @param llm A BaseLanguageModel instance.\n   * @param prompt A BasePromptTemplate instance.\n   * @param options Optional LLMRouterChainInput object, excluding \"llmChain\".\n   * @returns An instance of LLMRouterChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    prompt: BasePromptTemplate,\n    options?: Omit<LLMRouterChainInput, \"llmChain\">\n  ) {\n    const llmChain = new LLMChain<RouterOutputSchema>({ llm, prompt });\n    return new LLMRouterChain({ ...options, llmChain });\n  }\n}\n"],"mappings":";;;;;;;;;AA+BA,IAAa,iBAAb,MAAa,uBAAuB,YAA2C;CAC7E;CAEA,YAAY,QAA6B;AACvC,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;;CAGzB,IAAI,YAAsB;AACxB,SAAO,KAAK,SAAS;;CAGvB,MAAM,MACJ,QACA,YAC6B;AAC7B,SAAO,KAAK,SAAS,QAAQ,QAAQ,YAAY,UAAU,CAAC;;CAG9D,aAAqB;AACnB,SAAO;;;;;;;;;;;;CAaT,OAAO,QACL,KACA,QACA,SACA;EACA,MAAM,WAAW,IAAI,SAA6B;GAAE;GAAK;GAAQ,CAAC;AAClE,SAAO,IAAI,eAAe;GAAE,GAAG;GAAS;GAAU,CAAC"}

package/dist/chains/router/multi_prompt.cjs

@@ -1,14 +1,13 @@
-const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
-const require_llm_chain = require('../llm_chain.cjs');
-const require_conversation = require('../conversation.cjs');
-const require_multi_route = require('./multi_route.cjs');
-const require_llm_router = require('./llm_router.cjs');
-const require_multi_prompt_prompt = require('./multi_prompt_prompt.cjs');
-const require_utils = require('./utils.cjs');
-const require_router = require('../../output_parsers/router.cjs');
+require("../../_virtual/_rolldown/runtime.cjs");
+const require_llm_chain = require("../llm_chain.cjs");
+const require_conversation = require("../conversation.cjs");
+const require_multi_route = require("./multi_route.cjs");
+const require_llm_router = require("./llm_router.cjs");
+const require_multi_prompt_prompt = require("./multi_prompt_prompt.cjs");
+const require_utils = require("./utils.cjs");
+const require_router = require("../../output_parsers/router.cjs");
 let _langchain_core_prompts = require("@langchain/core/prompts");
 let zod_v3 = require("zod/v3");
-
 //#region src/chains/router/multi_prompt.ts
 /**
 * A class that represents a multi-prompt chain in the LangChain
@@ -108,7 +107,7 @@ var MultiPromptChain = class MultiPromptChain extends require_multi_route.MultiR
 		return "multi_prompt_chain";
 	}
 };
-
 //#endregion
 exports.MultiPromptChain = MultiPromptChain;
+
 //# sourceMappingURL=multi_prompt.cjs.map

package/dist/chains/router/multi_prompt.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"multi_prompt.cjs","names":["MultiRouteChain","zipEntries","RouterOutputParser","z","PromptTemplate","STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE","LLMRouterChain","LLMChain","ConversationChain"],"sources":["../../../src/chains/router/multi_prompt.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { z } from \"zod/v3\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE } from \"./multi_prompt_prompt.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain, LLMChainInput } from \"../../chains/llm_chain.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport { ConversationChain } from \"../../chains/conversation.js\";\nimport { zipEntries } from \"./utils.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A class that represents a multi-prompt chain in the LangChain\n * framework. It extends the MultiRouteChain class and provides additional\n * functionality specific to multi-prompt chains.\n * @example\n * ```typescript\n * const multiPromptChain = MultiPromptChain.fromLLMAndPrompts(\n *   new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n *   {\n *     promptNames: [\"physics\", \"math\", \"history\"],\n *     promptDescriptions: [\n *       \"Good for answering questions about physics\",\n *       \"Good for answering math questions\",\n *       \"Good for answering questions about history\",\n *     ],\n *     promptTemplates: [\n *       `You are a very smart physics professor. Here is a question:\\n{input}\\n`,\n *       `You are a very good mathematician. Here is a question:\\n{input}\\n`,\n *       `You are a very smart history professor. Here is a question:\\n{input}\\n`,\n *     ],\n *   }\n * );\n * const result = await multiPromptChain.call({\n *   input: \"What is the speed of light?\",\n * });\n * ```\n */\nexport class MultiPromptChain extends MultiRouteChain {\n  /**\n   * @deprecated Use `fromLLMAndPrompts` instead\n   */\n  static fromPrompts(\n    llm: BaseLanguageModelInterface,\n    promptNames: string[],\n    promptDescriptions: string[],\n    promptTemplates: string[] | PromptTemplate[],\n    defaultChain?: BaseChain,\n    options?: Omit<MultiRouteChainInput, \"defaultChain\">\n  ) {\n    return MultiPromptChain.fromLLMAndPrompts(llm, {\n      promptNames,\n      promptDescriptions,\n      promptTemplates,\n      defaultChain,\n      multiRouteChainOpts: options,\n    });\n  }\n\n  /**\n   * A static method that creates an instance of MultiPromptChain from a\n   * BaseLanguageModel and a set of prompts. It takes in optional parameters\n   * for the default chain and additional options.\n   * @param llm A BaseLanguageModel instance.\n   * @param promptNames An array of prompt names.\n   * @param promptDescriptions An array of prompt descriptions.\n   * @param promptTemplates An array of prompt templates.\n   * @param defaultChain An optional BaseChain instance to be used as the default chain.\n   * @param llmChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'prompt'.\n   * @param conversationChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'outputKey'.\n   * @param multiRouteChainOpts Optional parameters for the MultiRouteChainInput, excluding 'defaultChain'.\n   * @returns An instance of MultiPromptChain.\n   */\n  static fromLLMAndPrompts(\n    llm: BaseLanguageModelInterface,\n    {\n      promptNames,\n      promptDescriptions,\n      promptTemplates,\n      defaultChain,\n      llmChainOpts,\n      conversationChainOpts,\n      multiRouteChainOpts,\n    }: {\n      promptNames: string[];\n      promptDescriptions: string[];\n      promptTemplates: string[] | PromptTemplate[];\n      defaultChain?: BaseChain;\n      llmChainOpts?: Omit<LLMChainInput, \"llm\" | \"prompt\">;\n      conversationChainOpts?: Omit<LLMChainInput, \"llm\" | \"outputKey\">;\n      multiRouteChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n    }\n  ): MultiPromptChain {\n    const destinations = zipEntries(promptNames, promptDescriptions).map(\n      ([name, desc]) => `${name}: ${desc}`\n    );\n\n    const structuredOutputParserSchema = z.object({\n      destination: z\n        .string()\n        .optional()\n        .describe('name of the question answering system to use or \"DEFAULT\"'),\n      next_inputs: z\n        .object({\n          input: z\n            .string()\n            .describe(\"a potentially modified version of the original input\"),\n        })\n        .describe(\"input to be fed to the next model\"),\n    });\n\n    const outputParser = new RouterOutputParser(structuredOutputParserSchema);\n\n    const destinationsStr = destinations.join(\"\\n\");\n    const routerTemplate = interpolateFString(\n      STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE(\n        outputParser.getFormatInstructions({ interpolationDepth: 4 })\n      ),\n      {\n        destinations: destinationsStr,\n      }\n    );\n\n    const routerPrompt = new PromptTemplate({\n      template: routerTemplate,\n      inputVariables: [\"input\"],\n      outputParser,\n    });\n\n    const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n    const destinationChains = zipEntries<[string, string | PromptTemplate]>(\n      promptNames,\n      promptTemplates\n    ).reduce(\n      (acc, [name, template]) => {\n        let myPrompt: string | PromptTemplate;\n        if (typeof template === \"object\") {\n          myPrompt = template;\n        } else if (typeof template === \"string\") {\n          myPrompt = new PromptTemplate({\n            template: template as string,\n            inputVariables: [\"input\"],\n          });\n        } else {\n          throw new Error(\"Invalid prompt template\");\n        }\n        acc[name as string] = new LLMChain({\n          ...llmChainOpts,\n          llm,\n          prompt: myPrompt,\n        });\n        return acc;\n      },\n      {} as { [name: string]: LLMChain }\n    );\n\n    const convChain = new ConversationChain({\n      ...conversationChainOpts,\n      llm,\n      outputKey: \"text\",\n    });\n\n    return new MultiPromptChain({\n      ...multiRouteChainOpts,\n      routerChain,\n      destinationChains,\n      defaultChain: defaultChain ?? convChain,\n    });\n  }\n\n  _chainType(): string {\n    return \"multi_prompt_chain\";\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuCA,IAAa,mBAAb,MAAa,yBAAyBA,oCAAgB;;;;CAIpD,OAAO,YACL,KACA,aACA,oBACA,iBACA,cACA,SACA;AACA,SAAO,iBAAiB,kBAAkB,KAAK;GAC7C;GACA;GACA;GACA;GACA,qBAAqB;GACtB,CAAC;;;;;;;;;;;;;;;;CAiBJ,OAAO,kBACL,KACA,EACE,aACA,oBACA,iBACA,cACA,cACA,uBACA,uBAUgB;EAClB,MAAM,eAAeC,yBAAW,aAAa,mBAAmB,CAAC,KAC9D,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,OAC/B;EAgBD,MAAM,eAAe,IAAIC,kCAdYC,SAAE,OAAO;GAC5C,aAAaA,SACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAaA,SACV,OAAO,EACN,OAAOA,SACJ,QAAQ,CACR,SAAS,uDAAuD,EACpE,CAAC,CACD,SAAS,oCAAoC;GACjD,CAAC,CAEuE;EAEzE,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAU/C,MAAM,eAAe,IAAIC,uCAAe;GACtC,0DATAC,oEACE,aAAa,sBAAsB,EAAE,oBAAoB,GAAG,CAAC,CAC9D,EACD,EACE,cAAc,iBACf,CACF;GAIC,gBAAgB,CAAC,QAAQ;GACzB;GACD,CAAC;EAEF,MAAM,cAAcC,kCAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,oBAAoBL,yBACxB,aACA,gBACD,CAAC,QACC,KAAK,CAAC,MAAM,cAAc;GACzB,IAAI;AACJ,OAAI,OAAO,aAAa,SACtB,YAAW;YACF,OAAO,aAAa,SAC7B,YAAW,IAAIG,uCAAe;IAClB;IACV,gBAAgB,CAAC,QAAQ;IAC1B,CAAC;OAEF,OAAM,IAAI,MAAM,0BAA0B;AAE5C,OAAI,QAAkB,IAAIG,2BAAS;IACjC,GAAG;IACH;IACA,QAAQ;IACT,CAAC;AACF,UAAO;KAET,EAAE,CACH;EAED,MAAM,YAAY,IAAIC,uCAAkB;GACtC,GAAG;GACH;GACA,WAAW;GACZ,CAAC;AAEF,SAAO,IAAI,iBAAiB;GAC1B,GAAG;GACH;GACA;GACA,cAAc,gBAAgB;GAC/B,CAAC;;CAGJ,aAAqB;AACnB,SAAO"}
+{"version":3,"file":"multi_prompt.cjs","names":["MultiRouteChain","zipEntries","RouterOutputParser","z","PromptTemplate","STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE","LLMRouterChain","LLMChain","ConversationChain"],"sources":["../../../src/chains/router/multi_prompt.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { z } from \"zod/v3\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE } from \"./multi_prompt_prompt.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain, LLMChainInput } from \"../../chains/llm_chain.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport { ConversationChain } from \"../../chains/conversation.js\";\nimport { zipEntries } from \"./utils.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A class that represents a multi-prompt chain in the LangChain\n * framework. It extends the MultiRouteChain class and provides additional\n * functionality specific to multi-prompt chains.\n * @example\n * ```typescript\n * const multiPromptChain = MultiPromptChain.fromLLMAndPrompts(\n *   new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n *   {\n *     promptNames: [\"physics\", \"math\", \"history\"],\n *     promptDescriptions: [\n *       \"Good for answering questions about physics\",\n *       \"Good for answering math questions\",\n *       \"Good for answering questions about history\",\n *     ],\n *     promptTemplates: [\n *       `You are a very smart physics professor. Here is a question:\\n{input}\\n`,\n *       `You are a very good mathematician. Here is a question:\\n{input}\\n`,\n *       `You are a very smart history professor. Here is a question:\\n{input}\\n`,\n *     ],\n *   }\n * );\n * const result = await multiPromptChain.call({\n *   input: \"What is the speed of light?\",\n * });\n * ```\n */\nexport class MultiPromptChain extends MultiRouteChain {\n  /**\n   * @deprecated Use `fromLLMAndPrompts` instead\n   */\n  static fromPrompts(\n    llm: BaseLanguageModelInterface,\n    promptNames: string[],\n    promptDescriptions: string[],\n    promptTemplates: string[] | PromptTemplate[],\n    defaultChain?: BaseChain,\n    options?: Omit<MultiRouteChainInput, \"defaultChain\">\n  ) {\n    return MultiPromptChain.fromLLMAndPrompts(llm, {\n      promptNames,\n      promptDescriptions,\n      promptTemplates,\n      defaultChain,\n      multiRouteChainOpts: options,\n    });\n  }\n\n  /**\n   * A static method that creates an instance of MultiPromptChain from a\n   * BaseLanguageModel and a set of prompts. It takes in optional parameters\n   * for the default chain and additional options.\n   * @param llm A BaseLanguageModel instance.\n   * @param promptNames An array of prompt names.\n   * @param promptDescriptions An array of prompt descriptions.\n   * @param promptTemplates An array of prompt templates.\n   * @param defaultChain An optional BaseChain instance to be used as the default chain.\n   * @param llmChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'prompt'.\n   * @param conversationChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'outputKey'.\n   * @param multiRouteChainOpts Optional parameters for the MultiRouteChainInput, excluding 'defaultChain'.\n   * @returns An instance of MultiPromptChain.\n   */\n  static fromLLMAndPrompts(\n    llm: BaseLanguageModelInterface,\n    {\n      promptNames,\n      promptDescriptions,\n      promptTemplates,\n      defaultChain,\n      llmChainOpts,\n      conversationChainOpts,\n      multiRouteChainOpts,\n    }: {\n      promptNames: string[];\n      promptDescriptions: string[];\n      promptTemplates: string[] | PromptTemplate[];\n      defaultChain?: BaseChain;\n      llmChainOpts?: Omit<LLMChainInput, \"llm\" | \"prompt\">;\n      conversationChainOpts?: Omit<LLMChainInput, \"llm\" | \"outputKey\">;\n      multiRouteChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n    }\n  ): MultiPromptChain {\n    const destinations = zipEntries(promptNames, promptDescriptions).map(\n      ([name, desc]) => `${name}: ${desc}`\n    );\n\n    const structuredOutputParserSchema = z.object({\n      destination: z\n        .string()\n        .optional()\n        .describe('name of the question answering system to use or \"DEFAULT\"'),\n      next_inputs: z\n        .object({\n          input: z\n            .string()\n            .describe(\"a potentially modified version of the original input\"),\n        })\n        .describe(\"input to be fed to the next model\"),\n    });\n\n    const outputParser = new RouterOutputParser(structuredOutputParserSchema);\n\n    const destinationsStr = destinations.join(\"\\n\");\n    const routerTemplate = interpolateFString(\n      STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE(\n        outputParser.getFormatInstructions({ interpolationDepth: 4 })\n      ),\n      {\n        destinations: destinationsStr,\n      }\n    );\n\n    const routerPrompt = new PromptTemplate({\n      template: routerTemplate,\n      inputVariables: [\"input\"],\n      outputParser,\n    });\n\n    const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n    const destinationChains = zipEntries<[string, string | PromptTemplate]>(\n      promptNames,\n      promptTemplates\n    ).reduce(\n      (acc, [name, template]) => {\n        let myPrompt: string | PromptTemplate;\n        if (typeof template === \"object\") {\n          myPrompt = template;\n        } else if (typeof template === \"string\") {\n          myPrompt = new PromptTemplate({\n            template: template as string,\n            inputVariables: [\"input\"],\n          });\n        } else {\n          throw new Error(\"Invalid prompt template\");\n        }\n        acc[name as string] = new LLMChain({\n          ...llmChainOpts,\n          llm,\n          prompt: myPrompt,\n        });\n        return acc;\n      },\n      {} as { [name: string]: LLMChain }\n    );\n\n    const convChain = new ConversationChain({\n      ...conversationChainOpts,\n      llm,\n      outputKey: \"text\",\n    });\n\n    return new MultiPromptChain({\n      ...multiRouteChainOpts,\n      routerChain,\n      destinationChains,\n      defaultChain: defaultChain ?? convChain,\n    });\n  }\n\n  _chainType(): string {\n    return \"multi_prompt_chain\";\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuCA,IAAa,mBAAb,MAAa,yBAAyBA,oBAAAA,gBAAgB;;;;CAIpD,OAAO,YACL,KACA,aACA,oBACA,iBACA,cACA,SACA;AACA,SAAO,iBAAiB,kBAAkB,KAAK;GAC7C;GACA;GACA;GACA;GACA,qBAAqB;GACtB,CAAC;;;;;;;;;;;;;;;;CAiBJ,OAAO,kBACL,KACA,EACE,aACA,oBACA,iBACA,cACA,cACA,uBACA,uBAUgB;EAClB,MAAM,eAAeC,cAAAA,WAAW,aAAa,mBAAmB,CAAC,KAC9D,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,OAC/B;EAgBD,MAAM,eAAe,IAAIC,eAAAA,mBAdYC,OAAAA,EAAE,OAAO;GAC5C,aAAaA,OAAAA,EACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAaA,OAAAA,EACV,OAAO,EACN,OAAOA,OAAAA,EACJ,QAAQ,CACR,SAAS,uDAAuD,EACpE,CAAC,CACD,SAAS,oCAAoC;GACjD,CAAC,CAEuE;EAEzE,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAU/C,MAAM,eAAe,IAAIC,wBAAAA,eAAe;GACtC,WAAA,GAAA,wBAAA,oBATAC,4BAAAA,wCACE,aAAa,sBAAsB,EAAE,oBAAoB,GAAG,CAAC,CAC9D,EACD,EACE,cAAc,iBACf,CACF;GAIC,gBAAgB,CAAC,QAAQ;GACzB;GACD,CAAC;EAEF,MAAM,cAAcC,mBAAAA,eAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,oBAAoBL,cAAAA,WACxB,aACA,gBACD,CAAC,QACC,KAAK,CAAC,MAAM,cAAc;GACzB,IAAI;AACJ,OAAI,OAAO,aAAa,SACtB,YAAW;YACF,OAAO,aAAa,SAC7B,YAAW,IAAIG,wBAAAA,eAAe;IAClB;IACV,gBAAgB,CAAC,QAAQ;IAC1B,CAAC;OAEF,OAAM,IAAI,MAAM,0BAA0B;AAE5C,OAAI,QAAkB,IAAIG,kBAAAA,SAAS;IACjC,GAAG;IACH;IACA,QAAQ;IACT,CAAC;AACF,UAAO;KAET,EAAE,CACH;EAED,MAAM,YAAY,IAAIC,qBAAAA,kBAAkB;GACtC,GAAG;GACH;GACA,WAAW;GACZ,CAAC;AAEF,SAAO,IAAI,iBAAiB;GAC1B,GAAG;GACH;GACA;GACA,cAAc,gBAAgB;GAC/B,CAAC;;CAGJ,aAAqB;AACnB,SAAO"}

package/dist/chains/router/multi_prompt.js

@@ -7,7 +7,6 @@ import { zipEntries } from "./utils.js";
 import { RouterOutputParser } from "../../output_parsers/router.js";
 import { PromptTemplate, interpolateFString } from "@langchain/core/prompts";
 import { z } from "zod/v3";
-
 //#region src/chains/router/multi_prompt.ts
 /**
 * A class that represents a multi-prompt chain in the LangChain
@@ -107,7 +106,7 @@ var MultiPromptChain = class MultiPromptChain extends MultiRouteChain {
 		return "multi_prompt_chain";
 	}
 };
-
 //#endregion
 export { MultiPromptChain };
+
 //# sourceMappingURL=multi_prompt.js.map

package/dist/chains/router/multi_prompt.js.map

@@ -1 +1 @@
-{"version":3,"file":"multi_prompt.js","names":[],"sources":["../../../src/chains/router/multi_prompt.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { z } from \"zod/v3\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE } from \"./multi_prompt_prompt.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain, LLMChainInput } from \"../../chains/llm_chain.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport { ConversationChain } from \"../../chains/conversation.js\";\nimport { zipEntries } from \"./utils.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A class that represents a multi-prompt chain in the LangChain\n * framework. It extends the MultiRouteChain class and provides additional\n * functionality specific to multi-prompt chains.\n * @example\n * ```typescript\n * const multiPromptChain = MultiPromptChain.fromLLMAndPrompts(\n *   new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n *   {\n *     promptNames: [\"physics\", \"math\", \"history\"],\n *     promptDescriptions: [\n *       \"Good for answering questions about physics\",\n *       \"Good for answering math questions\",\n *       \"Good for answering questions about history\",\n *     ],\n *     promptTemplates: [\n *       `You are a very smart physics professor. Here is a question:\\n{input}\\n`,\n *       `You are a very good mathematician. Here is a question:\\n{input}\\n`,\n *       `You are a very smart history professor. Here is a question:\\n{input}\\n`,\n *     ],\n *   }\n * );\n * const result = await multiPromptChain.call({\n *   input: \"What is the speed of light?\",\n * });\n * ```\n */\nexport class MultiPromptChain extends MultiRouteChain {\n  /**\n   * @deprecated Use `fromLLMAndPrompts` instead\n   */\n  static fromPrompts(\n    llm: BaseLanguageModelInterface,\n    promptNames: string[],\n    promptDescriptions: string[],\n    promptTemplates: string[] | PromptTemplate[],\n    defaultChain?: BaseChain,\n    options?: Omit<MultiRouteChainInput, \"defaultChain\">\n  ) {\n    return MultiPromptChain.fromLLMAndPrompts(llm, {\n      promptNames,\n      promptDescriptions,\n      promptTemplates,\n      defaultChain,\n      multiRouteChainOpts: options,\n    });\n  }\n\n  /**\n   * A static method that creates an instance of MultiPromptChain from a\n   * BaseLanguageModel and a set of prompts. It takes in optional parameters\n   * for the default chain and additional options.\n   * @param llm A BaseLanguageModel instance.\n   * @param promptNames An array of prompt names.\n   * @param promptDescriptions An array of prompt descriptions.\n   * @param promptTemplates An array of prompt templates.\n   * @param defaultChain An optional BaseChain instance to be used as the default chain.\n   * @param llmChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'prompt'.\n   * @param conversationChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'outputKey'.\n   * @param multiRouteChainOpts Optional parameters for the MultiRouteChainInput, excluding 'defaultChain'.\n   * @returns An instance of MultiPromptChain.\n   */\n  static fromLLMAndPrompts(\n    llm: BaseLanguageModelInterface,\n    {\n      promptNames,\n      promptDescriptions,\n      promptTemplates,\n      defaultChain,\n      llmChainOpts,\n      conversationChainOpts,\n      multiRouteChainOpts,\n    }: {\n      promptNames: string[];\n      promptDescriptions: string[];\n      promptTemplates: string[] | PromptTemplate[];\n      defaultChain?: BaseChain;\n      llmChainOpts?: Omit<LLMChainInput, \"llm\" | \"prompt\">;\n      conversationChainOpts?: Omit<LLMChainInput, \"llm\" | \"outputKey\">;\n      multiRouteChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n    }\n  ): MultiPromptChain {\n    const destinations = zipEntries(promptNames, promptDescriptions).map(\n      ([name, desc]) => `${name}: ${desc}`\n    );\n\n    const structuredOutputParserSchema = z.object({\n      destination: z\n        .string()\n        .optional()\n        .describe('name of the question answering system to use or \"DEFAULT\"'),\n      next_inputs: z\n        .object({\n          input: z\n            .string()\n            .describe(\"a potentially modified version of the original input\"),\n        })\n        .describe(\"input to be fed to the next model\"),\n    });\n\n    const outputParser = new RouterOutputParser(structuredOutputParserSchema);\n\n    const destinationsStr = destinations.join(\"\\n\");\n    const routerTemplate = interpolateFString(\n      STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE(\n        outputParser.getFormatInstructions({ interpolationDepth: 4 })\n      ),\n      {\n        destinations: destinationsStr,\n      }\n    );\n\n    const routerPrompt = new PromptTemplate({\n      template: routerTemplate,\n      inputVariables: [\"input\"],\n      outputParser,\n    });\n\n    const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n    const destinationChains = zipEntries<[string, string | PromptTemplate]>(\n      promptNames,\n      promptTemplates\n    ).reduce(\n      (acc, [name, template]) => {\n        let myPrompt: string | PromptTemplate;\n        if (typeof template === \"object\") {\n          myPrompt = template;\n        } else if (typeof template === \"string\") {\n          myPrompt = new PromptTemplate({\n            template: template as string,\n            inputVariables: [\"input\"],\n          });\n        } else {\n          throw new Error(\"Invalid prompt template\");\n        }\n        acc[name as string] = new LLMChain({\n          ...llmChainOpts,\n          llm,\n          prompt: myPrompt,\n        });\n        return acc;\n      },\n      {} as { [name: string]: LLMChain }\n    );\n\n    const convChain = new ConversationChain({\n      ...conversationChainOpts,\n      llm,\n      outputKey: \"text\",\n    });\n\n    return new MultiPromptChain({\n      ...multiRouteChainOpts,\n      routerChain,\n      destinationChains,\n      defaultChain: defaultChain ?? convChain,\n    });\n  }\n\n  _chainType(): string {\n    return \"multi_prompt_chain\";\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuCA,IAAa,mBAAb,MAAa,yBAAyB,gBAAgB;;;;CAIpD,OAAO,YACL,KACA,aACA,oBACA,iBACA,cACA,SACA;AACA,SAAO,iBAAiB,kBAAkB,KAAK;GAC7C;GACA;GACA;GACA;GACA,qBAAqB;GACtB,CAAC;;;;;;;;;;;;;;;;CAiBJ,OAAO,kBACL,KACA,EACE,aACA,oBACA,iBACA,cACA,cACA,uBACA,uBAUgB;EAClB,MAAM,eAAe,WAAW,aAAa,mBAAmB,CAAC,KAC9D,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,OAC/B;EAgBD,MAAM,eAAe,IAAI,mBAdY,EAAE,OAAO;GAC5C,aAAa,EACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAa,EACV,OAAO,EACN,OAAO,EACJ,QAAQ,CACR,SAAS,uDAAuD,EACpE,CAAC,CACD,SAAS,oCAAoC;GACjD,CAAC,CAEuE;EAEzE,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAU/C,MAAM,eAAe,IAAI,eAAe;GACtC,UAVqB,mBACrB,wCACE,aAAa,sBAAsB,EAAE,oBAAoB,GAAG,CAAC,CAC9D,EACD,EACE,cAAc,iBACf,CACF;GAIC,gBAAgB,CAAC,QAAQ;GACzB;GACD,CAAC;EAEF,MAAM,cAAc,eAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,oBAAoB,WACxB,aACA,gBACD,CAAC,QACC,KAAK,CAAC,MAAM,cAAc;GACzB,IAAI;AACJ,OAAI,OAAO,aAAa,SACtB,YAAW;YACF,OAAO,aAAa,SAC7B,YAAW,IAAI,eAAe;IAClB;IACV,gBAAgB,CAAC,QAAQ;IAC1B,CAAC;OAEF,OAAM,IAAI,MAAM,0BAA0B;AAE5C,OAAI,QAAkB,IAAI,SAAS;IACjC,GAAG;IACH;IACA,QAAQ;IACT,CAAC;AACF,UAAO;KAET,EAAE,CACH;EAED,MAAM,YAAY,IAAI,kBAAkB;GACtC,GAAG;GACH;GACA,WAAW;GACZ,CAAC;AAEF,SAAO,IAAI,iBAAiB;GAC1B,GAAG;GACH;GACA;GACA,cAAc,gBAAgB;GAC/B,CAAC;;CAGJ,aAAqB;AACnB,SAAO"}
+{"version":3,"file":"multi_prompt.js","names":[],"sources":["../../../src/chains/router/multi_prompt.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { z } from \"zod/v3\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE } from \"./multi_prompt_prompt.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain, LLMChainInput } from \"../../chains/llm_chain.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport { ConversationChain } from \"../../chains/conversation.js\";\nimport { zipEntries } from \"./utils.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A class that represents a multi-prompt chain in the LangChain\n * framework. It extends the MultiRouteChain class and provides additional\n * functionality specific to multi-prompt chains.\n * @example\n * ```typescript\n * const multiPromptChain = MultiPromptChain.fromLLMAndPrompts(\n *   new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n *   {\n *     promptNames: [\"physics\", \"math\", \"history\"],\n *     promptDescriptions: [\n *       \"Good for answering questions about physics\",\n *       \"Good for answering math questions\",\n *       \"Good for answering questions about history\",\n *     ],\n *     promptTemplates: [\n *       `You are a very smart physics professor. Here is a question:\\n{input}\\n`,\n *       `You are a very good mathematician. Here is a question:\\n{input}\\n`,\n *       `You are a very smart history professor. Here is a question:\\n{input}\\n`,\n *     ],\n *   }\n * );\n * const result = await multiPromptChain.call({\n *   input: \"What is the speed of light?\",\n * });\n * ```\n */\nexport class MultiPromptChain extends MultiRouteChain {\n  /**\n   * @deprecated Use `fromLLMAndPrompts` instead\n   */\n  static fromPrompts(\n    llm: BaseLanguageModelInterface,\n    promptNames: string[],\n    promptDescriptions: string[],\n    promptTemplates: string[] | PromptTemplate[],\n    defaultChain?: BaseChain,\n    options?: Omit<MultiRouteChainInput, \"defaultChain\">\n  ) {\n    return MultiPromptChain.fromLLMAndPrompts(llm, {\n      promptNames,\n      promptDescriptions,\n      promptTemplates,\n      defaultChain,\n      multiRouteChainOpts: options,\n    });\n  }\n\n  /**\n   * A static method that creates an instance of MultiPromptChain from a\n   * BaseLanguageModel and a set of prompts. It takes in optional parameters\n   * for the default chain and additional options.\n   * @param llm A BaseLanguageModel instance.\n   * @param promptNames An array of prompt names.\n   * @param promptDescriptions An array of prompt descriptions.\n   * @param promptTemplates An array of prompt templates.\n   * @param defaultChain An optional BaseChain instance to be used as the default chain.\n   * @param llmChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'prompt'.\n   * @param conversationChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'outputKey'.\n   * @param multiRouteChainOpts Optional parameters for the MultiRouteChainInput, excluding 'defaultChain'.\n   * @returns An instance of MultiPromptChain.\n   */\n  static fromLLMAndPrompts(\n    llm: BaseLanguageModelInterface,\n    {\n      promptNames,\n      promptDescriptions,\n      promptTemplates,\n      defaultChain,\n      llmChainOpts,\n      conversationChainOpts,\n      multiRouteChainOpts,\n    }: {\n      promptNames: string[];\n      promptDescriptions: string[];\n      promptTemplates: string[] | PromptTemplate[];\n      defaultChain?: BaseChain;\n      llmChainOpts?: Omit<LLMChainInput, \"llm\" | \"prompt\">;\n      conversationChainOpts?: Omit<LLMChainInput, \"llm\" | \"outputKey\">;\n      multiRouteChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n    }\n  ): MultiPromptChain {\n    const destinations = zipEntries(promptNames, promptDescriptions).map(\n      ([name, desc]) => `${name}: ${desc}`\n    );\n\n    const structuredOutputParserSchema = z.object({\n      destination: z\n        .string()\n        .optional()\n        .describe('name of the question answering system to use or \"DEFAULT\"'),\n      next_inputs: z\n        .object({\n          input: z\n            .string()\n            .describe(\"a potentially modified version of the original input\"),\n        })\n        .describe(\"input to be fed to the next model\"),\n    });\n\n    const outputParser = new RouterOutputParser(structuredOutputParserSchema);\n\n    const destinationsStr = destinations.join(\"\\n\");\n    const routerTemplate = interpolateFString(\n      STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE(\n        outputParser.getFormatInstructions({ interpolationDepth: 4 })\n      ),\n      {\n        destinations: destinationsStr,\n      }\n    );\n\n    const routerPrompt = new PromptTemplate({\n      template: routerTemplate,\n      inputVariables: [\"input\"],\n      outputParser,\n    });\n\n    const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n    const destinationChains = zipEntries<[string, string | PromptTemplate]>(\n      promptNames,\n      promptTemplates\n    ).reduce(\n      (acc, [name, template]) => {\n        let myPrompt: string | PromptTemplate;\n        if (typeof template === \"object\") {\n          myPrompt = template;\n        } else if (typeof template === \"string\") {\n          myPrompt = new PromptTemplate({\n            template: template as string,\n            inputVariables: [\"input\"],\n          });\n        } else {\n          throw new Error(\"Invalid prompt template\");\n        }\n        acc[name as string] = new LLMChain({\n          ...llmChainOpts,\n          llm,\n          prompt: myPrompt,\n        });\n        return acc;\n      },\n      {} as { [name: string]: LLMChain }\n    );\n\n    const convChain = new ConversationChain({\n      ...conversationChainOpts,\n      llm,\n      outputKey: \"text\",\n    });\n\n    return new MultiPromptChain({\n      ...multiRouteChainOpts,\n      routerChain,\n      destinationChains,\n      defaultChain: defaultChain ?? convChain,\n    });\n  }\n\n  _chainType(): string {\n    return \"multi_prompt_chain\";\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuCA,IAAa,mBAAb,MAAa,yBAAyB,gBAAgB;;;;CAIpD,OAAO,YACL,KACA,aACA,oBACA,iBACA,cACA,SACA;AACA,SAAO,iBAAiB,kBAAkB,KAAK;GAC7C;GACA;GACA;GACA;GACA,qBAAqB;GACtB,CAAC;;;;;;;;;;;;;;;;CAiBJ,OAAO,kBACL,KACA,EACE,aACA,oBACA,iBACA,cACA,cACA,uBACA,uBAUgB;EAClB,MAAM,eAAe,WAAW,aAAa,mBAAmB,CAAC,KAC9D,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,OAC/B;EAgBD,MAAM,eAAe,IAAI,mBAdY,EAAE,OAAO;GAC5C,aAAa,EACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAa,EACV,OAAO,EACN,OAAO,EACJ,QAAQ,CACR,SAAS,uDAAuD,EACpE,CAAC,CACD,SAAS,oCAAoC;GACjD,CAAC,CAEuE;EAEzE,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAU/C,MAAM,eAAe,IAAI,eAAe;GACtC,UAVqB,mBACrB,wCACE,aAAa,sBAAsB,EAAE,oBAAoB,GAAG,CAAC,CAC9D,EACD,EACE,cAAc,iBACf,CACF;GAIC,gBAAgB,CAAC,QAAQ;GACzB;GACD,CAAC;EAEF,MAAM,cAAc,eAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,oBAAoB,WACxB,aACA,gBACD,CAAC,QACC,KAAK,CAAC,MAAM,cAAc;GACzB,IAAI;AACJ,OAAI,OAAO,aAAa,SACtB,YAAW;YACF,OAAO,aAAa,SAC7B,YAAW,IAAI,eAAe;IAClB;IACV,gBAAgB,CAAC,QAAQ;IAC1B,CAAC;OAEF,OAAM,IAAI,MAAM,0BAA0B;AAE5C,OAAI,QAAkB,IAAI,SAAS;IACjC,GAAG;IACH;IACA,QAAQ;IACT,CAAC;AACF,UAAO;KAET,EAAE,CACH;EAED,MAAM,YAAY,IAAI,kBAAkB;GACtC,GAAG;GACH;GACA,WAAW;GACZ,CAAC;AAEF,SAAO,IAAI,iBAAiB;GAC1B,GAAG;GACH;GACA;GACA,cAAc,gBAAgB;GAC/B,CAAC;;CAGJ,aAAqB;AACnB,SAAO"}

package/dist/chains/router/multi_prompt_prompt.cjs

@@ -1,4 +1,3 @@
-
 //#region src/chains/router/multi_prompt_prompt.ts
 const STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE = (formatting) => `Given a raw text input to a language model, select the model prompt best suited for the input. You will be given the names of the available prompts and a description of what the prompt is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response from the language model.
 
@@ -16,7 +15,7 @@ REMEMBER: "next_inputs.input" can just be the original input if you don't think
 
 << OUTPUT >>
 `;
-
 //#endregion
 exports.STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE = STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE;
+
 //# sourceMappingURL=multi_prompt_prompt.cjs.map

package/dist/chains/router/multi_prompt_prompt.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"multi_prompt_prompt.cjs","names":[],"sources":["../../../src/chains/router/multi_prompt_prompt.ts"],"sourcesContent":["export const MULTI_PROMPT_ROUTER_TEMPLATE = `Given a raw text input to a language model, select the model prompt best suited for the input. You will be given the names of the available prompts and a description of what the prompt is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response from the language model.\n\n<< FORMATTING >>\nReturn a markdown code snippet with a JSON object formatted to look like:\n\\`\\`\\`json\n{{{{\n    \"destination\": string \\\\ name of the prompt to use or \"DEFAULT\"\n    \"next_inputs\": string \\\\ a potentially modified version of the original input\n}}}}\n\\`\\`\\`\n\nREMEMBER: \"destination\" MUST be one of the candidate prompt names specified below OR it can be \"DEFAULT\" if the input is not well suited for any of the candidate prompts.\nREMEMBER: \"next_inputs\" can just be the original input if you don't think any modifications are needed.\n\n<< CANDIDATE PROMPTS >>\n{destinations}\n\n<< INPUT >>\n{{input}}\n\n<< OUTPUT >>\n`;\n\nexport const STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE = (\n  formatting: string\n) => `Given a raw text input to a language model, select the model prompt best suited for the input. You will be given the names of the available prompts and a description of what the prompt is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response from the language model.\n\n<< FORMATTING >>\n${formatting}\n\nREMEMBER: \"destination\" MUST be one of the candidate prompt names specified below OR it can be \"DEFAULT\" if the input is not well suited for any of the candidate prompts.\nREMEMBER: \"next_inputs.input\" can just be the original input if you don't think any modifications are needed.\n\n<< CANDIDATE PROMPTS >>\n{destinations}\n\n<< INPUT >>\n{{input}}\n\n<< OUTPUT >>\n`;\n"],"mappings":";;AAuBA,MAAa,2CACX,eACG;;;EAGH,WAAW"}
+{"version":3,"file":"multi_prompt_prompt.cjs","names":[],"sources":["../../../src/chains/router/multi_prompt_prompt.ts"],"sourcesContent":["export const MULTI_PROMPT_ROUTER_TEMPLATE = `Given a raw text input to a language model, select the model prompt best suited for the input. You will be given the names of the available prompts and a description of what the prompt is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response from the language model.\n\n<< FORMATTING >>\nReturn a markdown code snippet with a JSON object formatted to look like:\n\\`\\`\\`json\n{{{{\n    \"destination\": string \\\\ name of the prompt to use or \"DEFAULT\"\n    \"next_inputs\": string \\\\ a potentially modified version of the original input\n}}}}\n\\`\\`\\`\n\nREMEMBER: \"destination\" MUST be one of the candidate prompt names specified below OR it can be \"DEFAULT\" if the input is not well suited for any of the candidate prompts.\nREMEMBER: \"next_inputs\" can just be the original input if you don't think any modifications are needed.\n\n<< CANDIDATE PROMPTS >>\n{destinations}\n\n<< INPUT >>\n{{input}}\n\n<< OUTPUT >>\n`;\n\nexport const STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE = (\n  formatting: string\n) => `Given a raw text input to a language model, select the model prompt best suited for the input. You will be given the names of the available prompts and a description of what the prompt is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response from the language model.\n\n<< FORMATTING >>\n${formatting}\n\nREMEMBER: \"destination\" MUST be one of the candidate prompt names specified below OR it can be \"DEFAULT\" if the input is not well suited for any of the candidate prompts.\nREMEMBER: \"next_inputs.input\" can just be the original input if you don't think any modifications are needed.\n\n<< CANDIDATE PROMPTS >>\n{destinations}\n\n<< INPUT >>\n{{input}}\n\n<< OUTPUT >>\n`;\n"],"mappings":";AAuBA,MAAa,2CACX,eACG;;;EAGH,WAAW"}

package/dist/chains/router/multi_prompt_prompt.js

@@ -15,7 +15,7 @@ REMEMBER: "next_inputs.input" can just be the original input if you don't think
 
 << OUTPUT >>
 `;
-
 //#endregion
 export { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE };
+
 //# sourceMappingURL=multi_prompt_prompt.js.map

package/dist/chains/router/multi_retrieval_prompt.cjs

@@ -1,4 +1,3 @@
-
 //#region src/chains/router/multi_retrieval_prompt.ts
 const STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE = (formatting) => `Given a query to a question answering system, select the system best suited for the input. You will be given the names of the available systems and a description of what questions the system is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response.
 
@@ -16,7 +15,7 @@ REMEMBER: "next_inputs.query" can just be the original input if you don't think
 
 << OUTPUT >>
 `;
-
 //#endregion
 exports.STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE = STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE;
+
 //# sourceMappingURL=multi_retrieval_prompt.cjs.map

package/dist/chains/router/multi_retrieval_prompt.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"multi_retrieval_prompt.cjs","names":[],"sources":["../../../src/chains/router/multi_retrieval_prompt.ts"],"sourcesContent":["export const MULTI_RETRIEVAL_ROUTER_TEMPLATE = `Given a query to a question answering system, select the system best suited for the input. You will be given the names of the available systems and a description of what questions the system is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response.\n\n<< FORMATTING >>\nReturn a markdown code snippet with a JSON object formatted to look like:\n\\`\\`\\`json\n{{{{\n    \"destination\": string \\\\ name of the question answering system to use or \"DEFAULT\"\n    \"next_inputs\": string \\\\ a potentially modified version of the original input\n}}}}\n\\`\\`\\`\n\nREMEMBER: \"destination\" MUST be one of the candidate prompt names specified below OR it can be \"DEFAULT\" if the input is not well suited for any of the candidate prompts.\nREMEMBER: \"next_inputs\" can just be the original input if you don't think any modifications are needed.\n\n<< CANDIDATE PROMPTS >>\n{destinations}\n\n<< INPUT >>\n{{input}}\n\n<< OUTPUT >>\n`;\n\nexport const STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE = (\n  formatting: string\n) => `Given a query to a question answering system, select the system best suited for the input. You will be given the names of the available systems and a description of what questions the system is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response.\n\n<< FORMATTING >>\n${formatting}\n\nREMEMBER: \"destination\" MUST be one of the candidate prompt names specified below OR it can be \"DEFAULT\" if the input is not well suited for any of the candidate prompts.\nREMEMBER: \"next_inputs.query\" can just be the original input if you don't think any modifications are needed.\n\n<< CANDIDATE PROMPTS >>\n{destinations}\n\n<< INPUT >>\n{{input}}\n\n<< OUTPUT >>\n`;\n"],"mappings":";;AAuBA,MAAa,8CACX,eACG;;;EAGH,WAAW"}
+{"version":3,"file":"multi_retrieval_prompt.cjs","names":[],"sources":["../../../src/chains/router/multi_retrieval_prompt.ts"],"sourcesContent":["export const MULTI_RETRIEVAL_ROUTER_TEMPLATE = `Given a query to a question answering system, select the system best suited for the input. You will be given the names of the available systems and a description of what questions the system is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response.\n\n<< FORMATTING >>\nReturn a markdown code snippet with a JSON object formatted to look like:\n\\`\\`\\`json\n{{{{\n    \"destination\": string \\\\ name of the question answering system to use or \"DEFAULT\"\n    \"next_inputs\": string \\\\ a potentially modified version of the original input\n}}}}\n\\`\\`\\`\n\nREMEMBER: \"destination\" MUST be one of the candidate prompt names specified below OR it can be \"DEFAULT\" if the input is not well suited for any of the candidate prompts.\nREMEMBER: \"next_inputs\" can just be the original input if you don't think any modifications are needed.\n\n<< CANDIDATE PROMPTS >>\n{destinations}\n\n<< INPUT >>\n{{input}}\n\n<< OUTPUT >>\n`;\n\nexport const STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE = (\n  formatting: string\n) => `Given a query to a question answering system, select the system best suited for the input. You will be given the names of the available systems and a description of what questions the system is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response.\n\n<< FORMATTING >>\n${formatting}\n\nREMEMBER: \"destination\" MUST be one of the candidate prompt names specified below OR it can be \"DEFAULT\" if the input is not well suited for any of the candidate prompts.\nREMEMBER: \"next_inputs.query\" can just be the original input if you don't think any modifications are needed.\n\n<< CANDIDATE PROMPTS >>\n{destinations}\n\n<< INPUT >>\n{{input}}\n\n<< OUTPUT >>\n`;\n"],"mappings":";AAuBA,MAAa,8CACX,eACG;;;EAGH,WAAW"}

package/dist/chains/router/multi_retrieval_prompt.js

@@ -15,7 +15,7 @@ REMEMBER: "next_inputs.query" can just be the original input if you don't think
 
 << OUTPUT >>
 `;
-
 //#endregion
 export { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE };
+
 //# sourceMappingURL=multi_retrieval_prompt.js.map

package/dist/chains/router/multi_retrieval_qa.cjs

@@ -1,14 +1,13 @@
-const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
-const require_conversation = require('../conversation.cjs');
-const require_retrieval_qa = require('../retrieval_qa.cjs');
-const require_multi_route = require('./multi_route.cjs');
-const require_llm_router = require('./llm_router.cjs');
-const require_utils = require('./utils.cjs');
-const require_router = require('../../output_parsers/router.cjs');
-const require_multi_retrieval_prompt = require('./multi_retrieval_prompt.cjs');
+require("../../_virtual/_rolldown/runtime.cjs");
+const require_conversation = require("../conversation.cjs");
+const require_retrieval_qa = require("../retrieval_qa.cjs");
+const require_multi_route = require("./multi_route.cjs");
+const require_llm_router = require("./llm_router.cjs");
+const require_utils = require("./utils.cjs");
+const require_router = require("../../output_parsers/router.cjs");
+const require_multi_retrieval_prompt = require("./multi_retrieval_prompt.cjs");
 let _langchain_core_prompts = require("@langchain/core/prompts");
 let zod_v3 = require("zod/v3");
-
 //#region src/chains/router/multi_retrieval_qa.ts
 /**
 * A class that represents a multi-retrieval question answering chain in
@@ -124,7 +123,7 @@ var MultiRetrievalQAChain = class MultiRetrievalQAChain extends require_multi_ro
 		return "multi_retrieval_qa_chain";
 	}
 };
-
 //#endregion
 exports.MultiRetrievalQAChain = MultiRetrievalQAChain;
+
 //# sourceMappingURL=multi_retrieval_qa.cjs.map

package/dist/chains/router/multi_retrieval_qa.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"multi_retrieval_qa.cjs","names":["MultiRouteChain","zipEntries","RouterOutputParser","z","PromptTemplate","STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE","LLMRouterChain","RetrievalQAChain","ConversationChain","DEFAULT_TEMPLATE"],"sources":["../../../src/chains/router/multi_retrieval_qa.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport {\n  ConversationChain,\n  DEFAULT_TEMPLATE,\n} from \"../../chains/conversation.js\";\nimport { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE } from \"./multi_retrieval_prompt.js\";\nimport { zipEntries } from \"./utils.js\";\nimport {\n  RetrievalQAChain,\n  RetrievalQAChainInput,\n} from \"../../chains/retrieval_qa.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A type that represents the default values for the MultiRetrievalQAChain\n * class. It includes optional properties for the default retriever,\n * default prompt, and default chain.\n */\nexport type MultiRetrievalDefaults = {\n  defaultRetriever?: BaseRetrieverInterface;\n  defaultPrompt?: PromptTemplate;\n  defaultChain?: BaseChain;\n};\n\n/**\n * A class that represents a multi-retrieval question answering chain in\n * the LangChain framework. It extends the MultiRouteChain class and\n * provides additional functionality specific to multi-retrieval QA\n * chains.\n * @example\n * ```typescript\n * const multiRetrievalQAChain = MultiRetrievalQAChain.fromLLMAndRetrievers(\n *   new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n *   {\n *     retrieverNames: [\"aqua teen\", \"mst3k\", \"animaniacs\"],\n *     retrieverDescriptions: [\n *       \"Good for answering questions about Aqua Teen Hunger Force theme song\",\n *       \"Good for answering questions about Mystery Science Theater 3000 theme song\",\n *       \"Good for answering questions about Animaniacs theme song\",\n *     ],\n *     retrievers: [\n *       new MemoryVectorStore().asRetriever(3),\n *       new MemoryVectorStore().asRetriever(3),\n *       new MemoryVectorStore().asRetriever(3),\n *     ],\n *     retrievalQAChainOpts: {\n *       returnSourceDocuments: true,\n *     },\n *   },\n * );\n *\n * const result = await multiRetrievalQAChain.call({\n *   input:\n *     \"In the Aqua Teen Hunger Force theme song, who calls himself the mike rula?\",\n * });\n *\n * console.log(result.sourceDocuments, result.text);\n * ```\n */\nexport class MultiRetrievalQAChain extends MultiRouteChain {\n  get outputKeys(): string[] {\n    return [\"result\"];\n  }\n\n  /**\n   * @deprecated Use `fromRetrieversAndPrompts` instead\n   */\n  static fromRetrievers(\n    llm: BaseLanguageModelInterface,\n    retrieverNames: string[],\n    retrieverDescriptions: string[],\n    retrievers: BaseRetrieverInterface[],\n    retrieverPrompts?: PromptTemplate[],\n    defaults?: MultiRetrievalDefaults,\n    options?: Omit<MultiRouteChainInput, \"defaultChain\">\n  ) {\n    return MultiRetrievalQAChain.fromLLMAndRetrievers(llm, {\n      retrieverNames,\n      retrieverDescriptions,\n      retrievers,\n      retrieverPrompts,\n      defaults,\n      multiRetrievalChainOpts: options,\n    });\n  }\n\n  /**\n   * A static method that creates an instance of MultiRetrievalQAChain from\n   * a BaseLanguageModel and a set of retrievers. It takes in optional\n   * parameters for the retriever names, descriptions, prompts, defaults,\n   * and additional options. It is an alternative method to fromRetrievers\n   * and provides more flexibility in configuring the underlying chains.\n   * @param llm A BaseLanguageModel instance.\n   * @param retrieverNames An array of retriever names.\n   * @param retrieverDescriptions An array of retriever descriptions.\n   * @param retrievers An array of BaseRetrieverInterface instances.\n   * @param retrieverPrompts An optional array of PromptTemplate instances for the retrievers.\n   * @param defaults An optional MultiRetrievalDefaults instance.\n   * @param multiRetrievalChainOpts Additional optional parameters for the multi-retrieval chain.\n   * @param retrievalQAChainOpts Additional optional parameters for the retrieval QA chain.\n   * @returns A new instance of MultiRetrievalQAChain.\n   */\n  static fromLLMAndRetrievers(\n    llm: BaseLanguageModelInterface,\n    {\n      retrieverNames,\n      retrieverDescriptions,\n      retrievers,\n      retrieverPrompts,\n      defaults,\n      multiRetrievalChainOpts,\n      retrievalQAChainOpts,\n    }: {\n      retrieverNames: string[];\n      retrieverDescriptions: string[];\n      retrievers: BaseRetrieverInterface[];\n      retrieverPrompts?: PromptTemplate[];\n      defaults?: MultiRetrievalDefaults;\n      multiRetrievalChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n      retrievalQAChainOpts?: Partial<\n        Omit<RetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\">\n      > & {\n        prompt?: PromptTemplate;\n      };\n    }\n  ): MultiRetrievalQAChain {\n    const { defaultRetriever, defaultPrompt, defaultChain } = defaults ?? {};\n    if (defaultPrompt && !defaultRetriever) {\n      throw new Error(\n        \"`default_retriever` must be specified if `default_prompt` is \\nprovided. Received only `default_prompt`.\"\n      );\n    }\n    const destinations = zipEntries<[string, string]>(\n      retrieverNames,\n      retrieverDescriptions\n    ).map(([name, desc]) => `${name}: ${desc}`);\n\n    const structuredOutputParserSchema = z.object({\n      destination: z\n        .string()\n        .optional()\n        .describe('name of the question answering system to use or \"DEFAULT\"'),\n      next_inputs: z\n        .object({\n          query: z\n            .string()\n            .describe(\"a potentially modified version of the original input\"),\n        })\n        .describe(\"input to be fed to the next model\"),\n    });\n\n    const outputParser = new RouterOutputParser<\n      typeof structuredOutputParserSchema\n    >(structuredOutputParserSchema);\n\n    const destinationsStr = destinations.join(\"\\n\");\n    const routerTemplate = interpolateFString(\n      STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE(\n        outputParser.getFormatInstructions({ interpolationDepth: 4 })\n      ),\n      {\n        destinations: destinationsStr,\n      }\n    );\n    const routerPrompt = new PromptTemplate({\n      template: routerTemplate,\n      inputVariables: [\"input\"],\n      outputParser,\n    });\n\n    const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n    const prompts = retrieverPrompts ?? retrievers.map(() => null);\n    const destinationChains = zipEntries<\n      [string, BaseRetrieverInterface, PromptTemplate | null]\n    >(retrieverNames, retrievers, prompts).reduce(\n      (acc, [name, retriever, prompt]) => {\n        const opt: Partial<RetrievalQAChainInput> & {\n          prompt?: PromptTemplate;\n        } = retrievalQAChainOpts ?? {};\n        if (prompt) {\n          opt.prompt = prompt;\n        }\n        acc[name] = RetrievalQAChain.fromLLM(llm, retriever, opt);\n        return acc;\n      },\n      {} as { [name: string]: RetrievalQAChain }\n    );\n\n    let _defaultChain;\n    if (defaultChain) {\n      _defaultChain = defaultChain;\n    } else if (defaultRetriever) {\n      _defaultChain = RetrievalQAChain.fromLLM(llm, defaultRetriever, {\n        ...retrievalQAChainOpts,\n        prompt: defaultPrompt,\n      });\n    } else {\n      const promptTemplate = DEFAULT_TEMPLATE.replace(\"input\", \"query\");\n      const prompt = new PromptTemplate({\n        template: promptTemplate,\n        inputVariables: [\"history\", \"query\"],\n      });\n      _defaultChain = new ConversationChain({\n        llm,\n        prompt,\n        outputKey: \"result\",\n      });\n    }\n\n    return new MultiRetrievalQAChain({\n      ...multiRetrievalChainOpts,\n      routerChain,\n      destinationChains,\n      defaultChain: _defaultChain,\n    });\n  }\n\n  _chainType(): string {\n    return \"multi_retrieval_qa_chain\";\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiEA,IAAa,wBAAb,MAAa,8BAA8BA,oCAAgB;CACzD,IAAI,aAAuB;AACzB,SAAO,CAAC,SAAS;;;;;CAMnB,OAAO,eACL,KACA,gBACA,uBACA,YACA,kBACA,UACA,SACA;AACA,SAAO,sBAAsB,qBAAqB,KAAK;GACrD;GACA;GACA;GACA;GACA;GACA,yBAAyB;GAC1B,CAAC;;;;;;;;;;;;;;;;;;CAmBJ,OAAO,qBACL,KACA,EACE,gBACA,uBACA,YACA,kBACA,UACA,yBACA,wBAcqB;EACvB,MAAM,EAAE,kBAAkB,eAAe,iBAAiB,YAAY,EAAE;AACxE,MAAI,iBAAiB,CAAC,iBACpB,OAAM,IAAI,MACR,2GACD;EAEH,MAAM,eAAeC,yBACnB,gBACA,sBACD,CAAC,KAAK,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,OAAO;EAgB3C,MAAM,eAAe,IAAIC,kCAdYC,SAAE,OAAO;GAC5C,aAAaA,SACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAaA,SACV,OAAO,EACN,OAAOA,SACJ,QAAQ,CACR,SAAS,uDAAuD,EACpE,CAAC,CACD,SAAS,oCAAoC;GACjD,CAAC,CAI6B;EAE/B,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAS/C,MAAM,eAAe,IAAIC,uCAAe;GACtC,0DARAC,0EACE,aAAa,sBAAsB,EAAE,oBAAoB,GAAG,CAAC,CAC9D,EACD,EACE,cAAc,iBACf,CACF;GAGC,gBAAgB,CAAC,QAAQ;GACzB;GACD,CAAC;EAEF,MAAM,cAAcC,kCAAe,QAAQ,KAAK,aAAa;EAE7D,MAAM,oBAAoBL,yBAExB,gBAAgB,YAHF,oBAAoB,WAAW,UAAU,KAAK,CAGxB,CAAC,QACpC,KAAK,CAAC,MAAM,WAAW,YAAY;GAClC,MAAM,MAEF,wBAAwB,EAAE;AAC9B,OAAI,OACF,KAAI,SAAS;AAEf,OAAI,QAAQM,sCAAiB,QAAQ,KAAK,WAAW,IAAI;AACzD,UAAO;KAET,EAAE,CACH;EAED,IAAI;AACJ,MAAI,aACF,iBAAgB;WACP,iBACT,iBAAgBA,sCAAiB,QAAQ,KAAK,kBAAkB;GAC9D,GAAG;GACH,QAAQ;GACT,CAAC;MAOF,iBAAgB,IAAIC,uCAAkB;GACpC;GACA,QANa,IAAIJ,uCAAe;IAChC,UAFqBK,sCAAiB,QAAQ,SAAS,QAAQ;IAG/D,gBAAgB,CAAC,WAAW,QAAQ;IACrC,CAAC;GAIA,WAAW;GACZ,CAAC;AAGJ,SAAO,IAAI,sBAAsB;GAC/B,GAAG;GACH;GACA;GACA,cAAc;GACf,CAAC;;CAGJ,aAAqB;AACnB,SAAO"}
+{"version":3,"file":"multi_retrieval_qa.cjs","names":["MultiRouteChain","zipEntries","RouterOutputParser","z","PromptTemplate","STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE","LLMRouterChain","RetrievalQAChain","ConversationChain","DEFAULT_TEMPLATE"],"sources":["../../../src/chains/router/multi_retrieval_qa.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport {\n  ConversationChain,\n  DEFAULT_TEMPLATE,\n} from \"../../chains/conversation.js\";\nimport { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE } from \"./multi_retrieval_prompt.js\";\nimport { zipEntries } from \"./utils.js\";\nimport {\n  RetrievalQAChain,\n  RetrievalQAChainInput,\n} from \"../../chains/retrieval_qa.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A type that represents the default values for the MultiRetrievalQAChain\n * class. It includes optional properties for the default retriever,\n * default prompt, and default chain.\n */\nexport type MultiRetrievalDefaults = {\n  defaultRetriever?: BaseRetrieverInterface;\n  defaultPrompt?: PromptTemplate;\n  defaultChain?: BaseChain;\n};\n\n/**\n * A class that represents a multi-retrieval question answering chain in\n * the LangChain framework. It extends the MultiRouteChain class and\n * provides additional functionality specific to multi-retrieval QA\n * chains.\n * @example\n * ```typescript\n * const multiRetrievalQAChain = MultiRetrievalQAChain.fromLLMAndRetrievers(\n *   new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n *   {\n *     retrieverNames: [\"aqua teen\", \"mst3k\", \"animaniacs\"],\n *     retrieverDescriptions: [\n *       \"Good for answering questions about Aqua Teen Hunger Force theme song\",\n *       \"Good for answering questions about Mystery Science Theater 3000 theme song\",\n *       \"Good for answering questions about Animaniacs theme song\",\n *     ],\n *     retrievers: [\n *       new MemoryVectorStore().asRetriever(3),\n *       new MemoryVectorStore().asRetriever(3),\n *       new MemoryVectorStore().asRetriever(3),\n *     ],\n *     retrievalQAChainOpts: {\n *       returnSourceDocuments: true,\n *     },\n *   },\n * );\n *\n * const result = await multiRetrievalQAChain.call({\n *   input:\n *     \"In the Aqua Teen Hunger Force theme song, who calls himself the mike rula?\",\n * });\n *\n * console.log(result.sourceDocuments, result.text);\n * ```\n */\nexport class MultiRetrievalQAChain extends MultiRouteChain {\n  get outputKeys(): string[] {\n    return [\"result\"];\n  }\n\n  /**\n   * @deprecated Use `fromRetrieversAndPrompts` instead\n   */\n  static fromRetrievers(\n    llm: BaseLanguageModelInterface,\n    retrieverNames: string[],\n    retrieverDescriptions: string[],\n    retrievers: BaseRetrieverInterface[],\n    retrieverPrompts?: PromptTemplate[],\n    defaults?: MultiRetrievalDefaults,\n    options?: Omit<MultiRouteChainInput, \"defaultChain\">\n  ) {\n    return MultiRetrievalQAChain.fromLLMAndRetrievers(llm, {\n      retrieverNames,\n      retrieverDescriptions,\n      retrievers,\n      retrieverPrompts,\n      defaults,\n      multiRetrievalChainOpts: options,\n    });\n  }\n\n  /**\n   * A static method that creates an instance of MultiRetrievalQAChain from\n   * a BaseLanguageModel and a set of retrievers. It takes in optional\n   * parameters for the retriever names, descriptions, prompts, defaults,\n   * and additional options. It is an alternative method to fromRetrievers\n   * and provides more flexibility in configuring the underlying chains.\n   * @param llm A BaseLanguageModel instance.\n   * @param retrieverNames An array of retriever names.\n   * @param retrieverDescriptions An array of retriever descriptions.\n   * @param retrievers An array of BaseRetrieverInterface instances.\n   * @param retrieverPrompts An optional array of PromptTemplate instances for the retrievers.\n   * @param defaults An optional MultiRetrievalDefaults instance.\n   * @param multiRetrievalChainOpts Additional optional parameters for the multi-retrieval chain.\n   * @param retrievalQAChainOpts Additional optional parameters for the retrieval QA chain.\n   * @returns A new instance of MultiRetrievalQAChain.\n   */\n  static fromLLMAndRetrievers(\n    llm: BaseLanguageModelInterface,\n    {\n      retrieverNames,\n      retrieverDescriptions,\n      retrievers,\n      retrieverPrompts,\n      defaults,\n      multiRetrievalChainOpts,\n      retrievalQAChainOpts,\n    }: {\n      retrieverNames: string[];\n      retrieverDescriptions: string[];\n      retrievers: BaseRetrieverInterface[];\n      retrieverPrompts?: PromptTemplate[];\n      defaults?: MultiRetrievalDefaults;\n      multiRetrievalChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n      retrievalQAChainOpts?: Partial<\n        Omit<RetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\">\n      > & {\n        prompt?: PromptTemplate;\n      };\n    }\n  ): MultiRetrievalQAChain {\n    const { defaultRetriever, defaultPrompt, defaultChain } = defaults ?? {};\n    if (defaultPrompt && !defaultRetriever) {\n      throw new Error(\n        \"`default_retriever` must be specified if `default_prompt` is \\nprovided. Received only `default_prompt`.\"\n      );\n    }\n    const destinations = zipEntries<[string, string]>(\n      retrieverNames,\n      retrieverDescriptions\n    ).map(([name, desc]) => `${name}: ${desc}`);\n\n    const structuredOutputParserSchema = z.object({\n      destination: z\n        .string()\n        .optional()\n        .describe('name of the question answering system to use or \"DEFAULT\"'),\n      next_inputs: z\n        .object({\n          query: z\n            .string()\n            .describe(\"a potentially modified version of the original input\"),\n        })\n        .describe(\"input to be fed to the next model\"),\n    });\n\n    const outputParser = new RouterOutputParser<\n      typeof structuredOutputParserSchema\n    >(structuredOutputParserSchema);\n\n    const destinationsStr = destinations.join(\"\\n\");\n    const routerTemplate = interpolateFString(\n      STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE(\n        outputParser.getFormatInstructions({ interpolationDepth: 4 })\n      ),\n      {\n        destinations: destinationsStr,\n      }\n    );\n    const routerPrompt = new PromptTemplate({\n      template: routerTemplate,\n      inputVariables: [\"input\"],\n      outputParser,\n    });\n\n    const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n    const prompts = retrieverPrompts ?? retrievers.map(() => null);\n    const destinationChains = zipEntries<\n      [string, BaseRetrieverInterface, PromptTemplate | null]\n    >(retrieverNames, retrievers, prompts).reduce(\n      (acc, [name, retriever, prompt]) => {\n        const opt: Partial<RetrievalQAChainInput> & {\n          prompt?: PromptTemplate;\n        } = retrievalQAChainOpts ?? {};\n        if (prompt) {\n          opt.prompt = prompt;\n        }\n        acc[name] = RetrievalQAChain.fromLLM(llm, retriever, opt);\n        return acc;\n      },\n      {} as { [name: string]: RetrievalQAChain }\n    );\n\n    let _defaultChain;\n    if (defaultChain) {\n      _defaultChain = defaultChain;\n    } else if (defaultRetriever) {\n      _defaultChain = RetrievalQAChain.fromLLM(llm, defaultRetriever, {\n        ...retrievalQAChainOpts,\n        prompt: defaultPrompt,\n      });\n    } else {\n      const promptTemplate = DEFAULT_TEMPLATE.replace(\"input\", \"query\");\n      const prompt = new PromptTemplate({\n        template: promptTemplate,\n        inputVariables: [\"history\", \"query\"],\n      });\n      _defaultChain = new ConversationChain({\n        llm,\n        prompt,\n        outputKey: \"result\",\n      });\n    }\n\n    return new MultiRetrievalQAChain({\n      ...multiRetrievalChainOpts,\n      routerChain,\n      destinationChains,\n      defaultChain: _defaultChain,\n    });\n  }\n\n  _chainType(): string {\n    return \"multi_retrieval_qa_chain\";\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiEA,IAAa,wBAAb,MAAa,8BAA8BA,oBAAAA,gBAAgB;CACzD,IAAI,aAAuB;AACzB,SAAO,CAAC,SAAS;;;;;CAMnB,OAAO,eACL,KACA,gBACA,uBACA,YACA,kBACA,UACA,SACA;AACA,SAAO,sBAAsB,qBAAqB,KAAK;GACrD;GACA;GACA;GACA;GACA;GACA,yBAAyB;GAC1B,CAAC;;;;;;;;;;;;;;;;;;CAmBJ,OAAO,qBACL,KACA,EACE,gBACA,uBACA,YACA,kBACA,UACA,yBACA,wBAcqB;EACvB,MAAM,EAAE,kBAAkB,eAAe,iBAAiB,YAAY,EAAE;AACxE,MAAI,iBAAiB,CAAC,iBACpB,OAAM,IAAI,MACR,2GACD;EAEH,MAAM,eAAeC,cAAAA,WACnB,gBACA,sBACD,CAAC,KAAK,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,OAAO;EAgB3C,MAAM,eAAe,IAAIC,eAAAA,mBAdYC,OAAAA,EAAE,OAAO;GAC5C,aAAaA,OAAAA,EACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAaA,OAAAA,EACV,OAAO,EACN,OAAOA,OAAAA,EACJ,QAAQ,CACR,SAAS,uDAAuD,EACpE,CAAC,CACD,SAAS,oCAAoC;GACjD,CAAC,CAI6B;EAE/B,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAS/C,MAAM,eAAe,IAAIC,wBAAAA,eAAe;GACtC,WAAA,GAAA,wBAAA,oBARAC,+BAAAA,2CACE,aAAa,sBAAsB,EAAE,oBAAoB,GAAG,CAAC,CAC9D,EACD,EACE,cAAc,iBACf,CACF;GAGC,gBAAgB,CAAC,QAAQ;GACzB;GACD,CAAC;EAEF,MAAM,cAAcC,mBAAAA,eAAe,QAAQ,KAAK,aAAa;EAE7D,MAAM,oBAAoBL,cAAAA,WAExB,gBAAgB,YAHF,oBAAoB,WAAW,UAAU,KAAK,CAGxB,CAAC,QACpC,KAAK,CAAC,MAAM,WAAW,YAAY;GAClC,MAAM,MAEF,wBAAwB,EAAE;AAC9B,OAAI,OACF,KAAI,SAAS;AAEf,OAAI,QAAQM,qBAAAA,iBAAiB,QAAQ,KAAK,WAAW,IAAI;AACzD,UAAO;KAET,EAAE,CACH;EAED,IAAI;AACJ,MAAI,aACF,iBAAgB;WACP,iBACT,iBAAgBA,qBAAAA,iBAAiB,QAAQ,KAAK,kBAAkB;GAC9D,GAAG;GACH,QAAQ;GACT,CAAC;MAOF,iBAAgB,IAAIC,qBAAAA,kBAAkB;GACpC;GACA,QANa,IAAIJ,wBAAAA,eAAe;IAChC,UAFqBK,qBAAAA,iBAAiB,QAAQ,SAAS,QAAQ;IAG/D,gBAAgB,CAAC,WAAW,QAAQ;IACrC,CAAC;GAIA,WAAW;GACZ,CAAC;AAGJ,SAAO,IAAI,sBAAsB;GAC/B,GAAG;GACH;GACA;GACA,cAAc;GACf,CAAC;;CAGJ,aAAqB;AACnB,SAAO"}

package/dist/chains/router/multi_retrieval_qa.js

@@ -7,7 +7,6 @@ import { RouterOutputParser } from "../../output_parsers/router.js";
 import { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE } from "./multi_retrieval_prompt.js";
 import { PromptTemplate, interpolateFString } from "@langchain/core/prompts";
 import { z } from "zod/v3";
-
 //#region src/chains/router/multi_retrieval_qa.ts
 /**
 * A class that represents a multi-retrieval question answering chain in
@@ -123,7 +122,7 @@ var MultiRetrievalQAChain = class MultiRetrievalQAChain extends MultiRouteChain
 		return "multi_retrieval_qa_chain";
 	}
 };
-
 //#endregion
 export { MultiRetrievalQAChain };
+
 //# sourceMappingURL=multi_retrieval_qa.js.map

package/dist/chains/router/multi_retrieval_qa.js.map

@@ -1 +1 @@
-{"version":3,"file":"multi_retrieval_qa.js","names":[],"sources":["../../../src/chains/router/multi_retrieval_qa.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport {\n  ConversationChain,\n  DEFAULT_TEMPLATE,\n} from \"../../chains/conversation.js\";\nimport { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE } from \"./multi_retrieval_prompt.js\";\nimport { zipEntries } from \"./utils.js\";\nimport {\n  RetrievalQAChain,\n  RetrievalQAChainInput,\n} from \"../../chains/retrieval_qa.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A type that represents the default values for the MultiRetrievalQAChain\n * class. It includes optional properties for the default retriever,\n * default prompt, and default chain.\n */\nexport type MultiRetrievalDefaults = {\n  defaultRetriever?: BaseRetrieverInterface;\n  defaultPrompt?: PromptTemplate;\n  defaultChain?: BaseChain;\n};\n\n/**\n * A class that represents a multi-retrieval question answering chain in\n * the LangChain framework. It extends the MultiRouteChain class and\n * provides additional functionality specific to multi-retrieval QA\n * chains.\n * @example\n * ```typescript\n * const multiRetrievalQAChain = MultiRetrievalQAChain.fromLLMAndRetrievers(\n *   new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n *   {\n *     retrieverNames: [\"aqua teen\", \"mst3k\", \"animaniacs\"],\n *     retrieverDescriptions: [\n *       \"Good for answering questions about Aqua Teen Hunger Force theme song\",\n *       \"Good for answering questions about Mystery Science Theater 3000 theme song\",\n *       \"Good for answering questions about Animaniacs theme song\",\n *     ],\n *     retrievers: [\n *       new MemoryVectorStore().asRetriever(3),\n *       new MemoryVectorStore().asRetriever(3),\n *       new MemoryVectorStore().asRetriever(3),\n *     ],\n *     retrievalQAChainOpts: {\n *       returnSourceDocuments: true,\n *     },\n *   },\n * );\n *\n * const result = await multiRetrievalQAChain.call({\n *   input:\n *     \"In the Aqua Teen Hunger Force theme song, who calls himself the mike rula?\",\n * });\n *\n * console.log(result.sourceDocuments, result.text);\n * ```\n */\nexport class MultiRetrievalQAChain extends MultiRouteChain {\n  get outputKeys(): string[] {\n    return [\"result\"];\n  }\n\n  /**\n   * @deprecated Use `fromRetrieversAndPrompts` instead\n   */\n  static fromRetrievers(\n    llm: BaseLanguageModelInterface,\n    retrieverNames: string[],\n    retrieverDescriptions: string[],\n    retrievers: BaseRetrieverInterface[],\n    retrieverPrompts?: PromptTemplate[],\n    defaults?: MultiRetrievalDefaults,\n    options?: Omit<MultiRouteChainInput, \"defaultChain\">\n  ) {\n    return MultiRetrievalQAChain.fromLLMAndRetrievers(llm, {\n      retrieverNames,\n      retrieverDescriptions,\n      retrievers,\n      retrieverPrompts,\n      defaults,\n      multiRetrievalChainOpts: options,\n    });\n  }\n\n  /**\n   * A static method that creates an instance of MultiRetrievalQAChain from\n   * a BaseLanguageModel and a set of retrievers. It takes in optional\n   * parameters for the retriever names, descriptions, prompts, defaults,\n   * and additional options. It is an alternative method to fromRetrievers\n   * and provides more flexibility in configuring the underlying chains.\n   * @param llm A BaseLanguageModel instance.\n   * @param retrieverNames An array of retriever names.\n   * @param retrieverDescriptions An array of retriever descriptions.\n   * @param retrievers An array of BaseRetrieverInterface instances.\n   * @param retrieverPrompts An optional array of PromptTemplate instances for the retrievers.\n   * @param defaults An optional MultiRetrievalDefaults instance.\n   * @param multiRetrievalChainOpts Additional optional parameters for the multi-retrieval chain.\n   * @param retrievalQAChainOpts Additional optional parameters for the retrieval QA chain.\n   * @returns A new instance of MultiRetrievalQAChain.\n   */\n  static fromLLMAndRetrievers(\n    llm: BaseLanguageModelInterface,\n    {\n      retrieverNames,\n      retrieverDescriptions,\n      retrievers,\n      retrieverPrompts,\n      defaults,\n      multiRetrievalChainOpts,\n      retrievalQAChainOpts,\n    }: {\n      retrieverNames: string[];\n      retrieverDescriptions: string[];\n      retrievers: BaseRetrieverInterface[];\n      retrieverPrompts?: PromptTemplate[];\n      defaults?: MultiRetrievalDefaults;\n      multiRetrievalChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n      retrievalQAChainOpts?: Partial<\n        Omit<RetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\">\n      > & {\n        prompt?: PromptTemplate;\n      };\n    }\n  ): MultiRetrievalQAChain {\n    const { defaultRetriever, defaultPrompt, defaultChain } = defaults ?? {};\n    if (defaultPrompt && !defaultRetriever) {\n      throw new Error(\n        \"`default_retriever` must be specified if `default_prompt` is \\nprovided. Received only `default_prompt`.\"\n      );\n    }\n    const destinations = zipEntries<[string, string]>(\n      retrieverNames,\n      retrieverDescriptions\n    ).map(([name, desc]) => `${name}: ${desc}`);\n\n    const structuredOutputParserSchema = z.object({\n      destination: z\n        .string()\n        .optional()\n        .describe('name of the question answering system to use or \"DEFAULT\"'),\n      next_inputs: z\n        .object({\n          query: z\n            .string()\n            .describe(\"a potentially modified version of the original input\"),\n        })\n        .describe(\"input to be fed to the next model\"),\n    });\n\n    const outputParser = new RouterOutputParser<\n      typeof structuredOutputParserSchema\n    >(structuredOutputParserSchema);\n\n    const destinationsStr = destinations.join(\"\\n\");\n    const routerTemplate = interpolateFString(\n      STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE(\n        outputParser.getFormatInstructions({ interpolationDepth: 4 })\n      ),\n      {\n        destinations: destinationsStr,\n      }\n    );\n    const routerPrompt = new PromptTemplate({\n      template: routerTemplate,\n      inputVariables: [\"input\"],\n      outputParser,\n    });\n\n    const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n    const prompts = retrieverPrompts ?? retrievers.map(() => null);\n    const destinationChains = zipEntries<\n      [string, BaseRetrieverInterface, PromptTemplate | null]\n    >(retrieverNames, retrievers, prompts).reduce(\n      (acc, [name, retriever, prompt]) => {\n        const opt: Partial<RetrievalQAChainInput> & {\n          prompt?: PromptTemplate;\n        } = retrievalQAChainOpts ?? {};\n        if (prompt) {\n          opt.prompt = prompt;\n        }\n        acc[name] = RetrievalQAChain.fromLLM(llm, retriever, opt);\n        return acc;\n      },\n      {} as { [name: string]: RetrievalQAChain }\n    );\n\n    let _defaultChain;\n    if (defaultChain) {\n      _defaultChain = defaultChain;\n    } else if (defaultRetriever) {\n      _defaultChain = RetrievalQAChain.fromLLM(llm, defaultRetriever, {\n        ...retrievalQAChainOpts,\n        prompt: defaultPrompt,\n      });\n    } else {\n      const promptTemplate = DEFAULT_TEMPLATE.replace(\"input\", \"query\");\n      const prompt = new PromptTemplate({\n        template: promptTemplate,\n        inputVariables: [\"history\", \"query\"],\n      });\n      _defaultChain = new ConversationChain({\n        llm,\n        prompt,\n        outputKey: \"result\",\n      });\n    }\n\n    return new MultiRetrievalQAChain({\n      ...multiRetrievalChainOpts,\n      routerChain,\n      destinationChains,\n      defaultChain: _defaultChain,\n    });\n  }\n\n  _chainType(): string {\n    return \"multi_retrieval_qa_chain\";\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiEA,IAAa,wBAAb,MAAa,8BAA8B,gBAAgB;CACzD,IAAI,aAAuB;AACzB,SAAO,CAAC,SAAS;;;;;CAMnB,OAAO,eACL,KACA,gBACA,uBACA,YACA,kBACA,UACA,SACA;AACA,SAAO,sBAAsB,qBAAqB,KAAK;GACrD;GACA;GACA;GACA;GACA;GACA,yBAAyB;GAC1B,CAAC;;;;;;;;;;;;;;;;;;CAmBJ,OAAO,qBACL,KACA,EACE,gBACA,uBACA,YACA,kBACA,UACA,yBACA,wBAcqB;EACvB,MAAM,EAAE,kBAAkB,eAAe,iBAAiB,YAAY,EAAE;AACxE,MAAI,iBAAiB,CAAC,iBACpB,OAAM,IAAI,MACR,2GACD;EAEH,MAAM,eAAe,WACnB,gBACA,sBACD,CAAC,KAAK,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,OAAO;EAgB3C,MAAM,eAAe,IAAI,mBAdY,EAAE,OAAO;GAC5C,aAAa,EACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAa,EACV,OAAO,EACN,OAAO,EACJ,QAAQ,CACR,SAAS,uDAAuD,EACpE,CAAC,CACD,SAAS,oCAAoC;GACjD,CAAC,CAI6B;EAE/B,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAS/C,MAAM,eAAe,IAAI,eAAe;GACtC,UATqB,mBACrB,2CACE,aAAa,sBAAsB,EAAE,oBAAoB,GAAG,CAAC,CAC9D,EACD,EACE,cAAc,iBACf,CACF;GAGC,gBAAgB,CAAC,QAAQ;GACzB;GACD,CAAC;EAEF,MAAM,cAAc,eAAe,QAAQ,KAAK,aAAa;EAE7D,MAAM,oBAAoB,WAExB,gBAAgB,YAHF,oBAAoB,WAAW,UAAU,KAAK,CAGxB,CAAC,QACpC,KAAK,CAAC,MAAM,WAAW,YAAY;GAClC,MAAM,MAEF,wBAAwB,EAAE;AAC9B,OAAI,OACF,KAAI,SAAS;AAEf,OAAI,QAAQ,iBAAiB,QAAQ,KAAK,WAAW,IAAI;AACzD,UAAO;KAET,EAAE,CACH;EAED,IAAI;AACJ,MAAI,aACF,iBAAgB;WACP,iBACT,iBAAgB,iBAAiB,QAAQ,KAAK,kBAAkB;GAC9D,GAAG;GACH,QAAQ;GACT,CAAC;MAOF,iBAAgB,IAAI,kBAAkB;GACpC;GACA,QANa,IAAI,eAAe;IAChC,UAFqB,iBAAiB,QAAQ,SAAS,QAAQ;IAG/D,gBAAgB,CAAC,WAAW,QAAQ;IACrC,CAAC;GAIA,WAAW;GACZ,CAAC;AAGJ,SAAO,IAAI,sBAAsB;GAC/B,GAAG;GACH;GACA;GACA,cAAc;GACf,CAAC;;CAGJ,aAAqB;AACnB,SAAO"}
+{"version":3,"file":"multi_retrieval_qa.js","names":[],"sources":["../../../src/chains/router/multi_retrieval_qa.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport {\n  ConversationChain,\n  DEFAULT_TEMPLATE,\n} from \"../../chains/conversation.js\";\nimport { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE } from \"./multi_retrieval_prompt.js\";\nimport { zipEntries } from \"./utils.js\";\nimport {\n  RetrievalQAChain,\n  RetrievalQAChainInput,\n} from \"../../chains/retrieval_qa.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A type that represents the default values for the MultiRetrievalQAChain\n * class. It includes optional properties for the default retriever,\n * default prompt, and default chain.\n */\nexport type MultiRetrievalDefaults = {\n  defaultRetriever?: BaseRetrieverInterface;\n  defaultPrompt?: PromptTemplate;\n  defaultChain?: BaseChain;\n};\n\n/**\n * A class that represents a multi-retrieval question answering chain in\n * the LangChain framework. It extends the MultiRouteChain class and\n * provides additional functionality specific to multi-retrieval QA\n * chains.\n * @example\n * ```typescript\n * const multiRetrievalQAChain = MultiRetrievalQAChain.fromLLMAndRetrievers(\n *   new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n *   {\n *     retrieverNames: [\"aqua teen\", \"mst3k\", \"animaniacs\"],\n *     retrieverDescriptions: [\n *       \"Good for answering questions about Aqua Teen Hunger Force theme song\",\n *       \"Good for answering questions about Mystery Science Theater 3000 theme song\",\n *       \"Good for answering questions about Animaniacs theme song\",\n *     ],\n *     retrievers: [\n *       new MemoryVectorStore().asRetriever(3),\n *       new MemoryVectorStore().asRetriever(3),\n *       new MemoryVectorStore().asRetriever(3),\n *     ],\n *     retrievalQAChainOpts: {\n *       returnSourceDocuments: true,\n *     },\n *   },\n * );\n *\n * const result = await multiRetrievalQAChain.call({\n *   input:\n *     \"In the Aqua Teen Hunger Force theme song, who calls himself the mike rula?\",\n * });\n *\n * console.log(result.sourceDocuments, result.text);\n * ```\n */\nexport class MultiRetrievalQAChain extends MultiRouteChain {\n  get outputKeys(): string[] {\n    return [\"result\"];\n  }\n\n  /**\n   * @deprecated Use `fromRetrieversAndPrompts` instead\n   */\n  static fromRetrievers(\n    llm: BaseLanguageModelInterface,\n    retrieverNames: string[],\n    retrieverDescriptions: string[],\n    retrievers: BaseRetrieverInterface[],\n    retrieverPrompts?: PromptTemplate[],\n    defaults?: MultiRetrievalDefaults,\n    options?: Omit<MultiRouteChainInput, \"defaultChain\">\n  ) {\n    return MultiRetrievalQAChain.fromLLMAndRetrievers(llm, {\n      retrieverNames,\n      retrieverDescriptions,\n      retrievers,\n      retrieverPrompts,\n      defaults,\n      multiRetrievalChainOpts: options,\n    });\n  }\n\n  /**\n   * A static method that creates an instance of MultiRetrievalQAChain from\n   * a BaseLanguageModel and a set of retrievers. It takes in optional\n   * parameters for the retriever names, descriptions, prompts, defaults,\n   * and additional options. It is an alternative method to fromRetrievers\n   * and provides more flexibility in configuring the underlying chains.\n   * @param llm A BaseLanguageModel instance.\n   * @param retrieverNames An array of retriever names.\n   * @param retrieverDescriptions An array of retriever descriptions.\n   * @param retrievers An array of BaseRetrieverInterface instances.\n   * @param retrieverPrompts An optional array of PromptTemplate instances for the retrievers.\n   * @param defaults An optional MultiRetrievalDefaults instance.\n   * @param multiRetrievalChainOpts Additional optional parameters for the multi-retrieval chain.\n   * @param retrievalQAChainOpts Additional optional parameters for the retrieval QA chain.\n   * @returns A new instance of MultiRetrievalQAChain.\n   */\n  static fromLLMAndRetrievers(\n    llm: BaseLanguageModelInterface,\n    {\n      retrieverNames,\n      retrieverDescriptions,\n      retrievers,\n      retrieverPrompts,\n      defaults,\n      multiRetrievalChainOpts,\n      retrievalQAChainOpts,\n    }: {\n      retrieverNames: string[];\n      retrieverDescriptions: string[];\n      retrievers: BaseRetrieverInterface[];\n      retrieverPrompts?: PromptTemplate[];\n      defaults?: MultiRetrievalDefaults;\n      multiRetrievalChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n      retrievalQAChainOpts?: Partial<\n        Omit<RetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\">\n      > & {\n        prompt?: PromptTemplate;\n      };\n    }\n  ): MultiRetrievalQAChain {\n    const { defaultRetriever, defaultPrompt, defaultChain } = defaults ?? {};\n    if (defaultPrompt && !defaultRetriever) {\n      throw new Error(\n        \"`default_retriever` must be specified if `default_prompt` is \\nprovided. Received only `default_prompt`.\"\n      );\n    }\n    const destinations = zipEntries<[string, string]>(\n      retrieverNames,\n      retrieverDescriptions\n    ).map(([name, desc]) => `${name}: ${desc}`);\n\n    const structuredOutputParserSchema = z.object({\n      destination: z\n        .string()\n        .optional()\n        .describe('name of the question answering system to use or \"DEFAULT\"'),\n      next_inputs: z\n        .object({\n          query: z\n            .string()\n            .describe(\"a potentially modified version of the original input\"),\n        })\n        .describe(\"input to be fed to the next model\"),\n    });\n\n    const outputParser = new RouterOutputParser<\n      typeof structuredOutputParserSchema\n    >(structuredOutputParserSchema);\n\n    const destinationsStr = destinations.join(\"\\n\");\n    const routerTemplate = interpolateFString(\n      STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE(\n        outputParser.getFormatInstructions({ interpolationDepth: 4 })\n      ),\n      {\n        destinations: destinationsStr,\n      }\n    );\n    const routerPrompt = new PromptTemplate({\n      template: routerTemplate,\n      inputVariables: [\"input\"],\n      outputParser,\n    });\n\n    const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n    const prompts = retrieverPrompts ?? retrievers.map(() => null);\n    const destinationChains = zipEntries<\n      [string, BaseRetrieverInterface, PromptTemplate | null]\n    >(retrieverNames, retrievers, prompts).reduce(\n      (acc, [name, retriever, prompt]) => {\n        const opt: Partial<RetrievalQAChainInput> & {\n          prompt?: PromptTemplate;\n        } = retrievalQAChainOpts ?? {};\n        if (prompt) {\n          opt.prompt = prompt;\n        }\n        acc[name] = RetrievalQAChain.fromLLM(llm, retriever, opt);\n        return acc;\n      },\n      {} as { [name: string]: RetrievalQAChain }\n    );\n\n    let _defaultChain;\n    if (defaultChain) {\n      _defaultChain = defaultChain;\n    } else if (defaultRetriever) {\n      _defaultChain = RetrievalQAChain.fromLLM(llm, defaultRetriever, {\n        ...retrievalQAChainOpts,\n        prompt: defaultPrompt,\n      });\n    } else {\n      const promptTemplate = DEFAULT_TEMPLATE.replace(\"input\", \"query\");\n      const prompt = new PromptTemplate({\n        template: promptTemplate,\n        inputVariables: [\"history\", \"query\"],\n      });\n      _defaultChain = new ConversationChain({\n        llm,\n        prompt,\n        outputKey: \"result\",\n      });\n    }\n\n    return new MultiRetrievalQAChain({\n      ...multiRetrievalChainOpts,\n      routerChain,\n      destinationChains,\n      defaultChain: _defaultChain,\n    });\n  }\n\n  _chainType(): string {\n    return \"multi_retrieval_qa_chain\";\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiEA,IAAa,wBAAb,MAAa,8BAA8B,gBAAgB;CACzD,IAAI,aAAuB;AACzB,SAAO,CAAC,SAAS;;;;;CAMnB,OAAO,eACL,KACA,gBACA,uBACA,YACA,kBACA,UACA,SACA;AACA,SAAO,sBAAsB,qBAAqB,KAAK;GACrD;GACA;GACA;GACA;GACA;GACA,yBAAyB;GAC1B,CAAC;;;;;;;;;;;;;;;;;;CAmBJ,OAAO,qBACL,KACA,EACE,gBACA,uBACA,YACA,kBACA,UACA,yBACA,wBAcqB;EACvB,MAAM,EAAE,kBAAkB,eAAe,iBAAiB,YAAY,EAAE;AACxE,MAAI,iBAAiB,CAAC,iBACpB,OAAM,IAAI,MACR,2GACD;EAEH,MAAM,eAAe,WACnB,gBACA,sBACD,CAAC,KAAK,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,OAAO;EAgB3C,MAAM,eAAe,IAAI,mBAdY,EAAE,OAAO;GAC5C,aAAa,EACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAa,EACV,OAAO,EACN,OAAO,EACJ,QAAQ,CACR,SAAS,uDAAuD,EACpE,CAAC,CACD,SAAS,oCAAoC;GACjD,CAAC,CAI6B;EAE/B,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAS/C,MAAM,eAAe,IAAI,eAAe;GACtC,UATqB,mBACrB,2CACE,aAAa,sBAAsB,EAAE,oBAAoB,GAAG,CAAC,CAC9D,EACD,EACE,cAAc,iBACf,CACF;GAGC,gBAAgB,CAAC,QAAQ;GACzB;GACD,CAAC;EAEF,MAAM,cAAc,eAAe,QAAQ,KAAK,aAAa;EAE7D,MAAM,oBAAoB,WAExB,gBAAgB,YAHF,oBAAoB,WAAW,UAAU,KAAK,CAGxB,CAAC,QACpC,KAAK,CAAC,MAAM,WAAW,YAAY;GAClC,MAAM,MAEF,wBAAwB,EAAE;AAC9B,OAAI,OACF,KAAI,SAAS;AAEf,OAAI,QAAQ,iBAAiB,QAAQ,KAAK,WAAW,IAAI;AACzD,UAAO;KAET,EAAE,CACH;EAED,IAAI;AACJ,MAAI,aACF,iBAAgB;WACP,iBACT,iBAAgB,iBAAiB,QAAQ,KAAK,kBAAkB;GAC9D,GAAG;GACH,QAAQ;GACT,CAAC;MAOF,iBAAgB,IAAI,kBAAkB;GACpC;GACA,QANa,IAAI,eAAe;IAChC,UAFqB,iBAAiB,QAAQ,SAAS,QAAQ;IAG/D,gBAAgB,CAAC,WAAW,QAAQ;IACrC,CAAC;GAIA,WAAW;GACZ,CAAC;AAGJ,SAAO,IAAI,sBAAsB;GAC/B,GAAG;GACH;GACA;GACA,cAAc;GACf,CAAC;;CAGJ,aAAqB;AACnB,SAAO"}