@langchain/classic 1.0.40 → 1.0.41

package/dist/chains/retrieval.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"retrieval.cjs","names":["RunnableSequence","RunnablePassthrough"],"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    // oxlint-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  // oxlint-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,0BAAyBA,0BAAAA,iBAAiB,KAAK,EAC5C,UAAU,MAAM,OACjB,UACD,CAAC;KAGF,0BAAyB;AAgB3B,QAduBA,0BAAAA,iBAAiB,KAGrC,CACDC,0BAAAA,oBAAoB,OAAO;EACzB,SAAS,uBAAuB,WAAW,EACzC,SAAS,sBACV,CAAC;EACF,eAAe,UAAU,MAAM,gBAAgB,EAAE;EAClD,CAAC,EACFA,0BAAAA,oBAAoB,OAAO,EACzB,QAAQ,kBACT,CAAC,CACH,CAAC,CAAC,WAAW,EAAE,SAAS,mBAAmB,CACvB"}
+{"version":3,"file":"retrieval.cjs","names":["RunnableSequence","RunnablePassthrough"],"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    // oxlint-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  // oxlint-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;CAChE,OAAO,CAAC,CAAC,KAAK,OAAQ,EAA6B,WAAW;AAChE;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4BA,eAAsB,qBAAgC,EACpD,WACA,oBAQA;CACA,IAAI;CACJ,IAAI,gBAAgB,SAAS,GAC3B,yBAAyBA,0BAAAA,iBAAiB,KAAK,EAC5C,UAAU,MAAM,OACjB,SACF,CAAC;MAGD,yBAAyB;CAgB3B,OAduBA,0BAAAA,iBAAiB,KAGrC,CACDC,0BAAAA,oBAAoB,OAAO;EACzB,SAAS,uBAAuB,WAAW,EACzC,SAAS,qBACX,CAAC;EACD,eAAe,UAAU,MAAM,gBAAgB,CAAC;CAClD,CAAC,GACDA,0BAAAA,oBAAoB,OAAO,EACzB,QAAQ,iBACV,CAAC,CACH,CAAC,CAAC,CAAC,WAAW,EAAE,SAAS,kBAAkB,CACvB;AACtB"}

package/dist/chains/retrieval.d.cts

@@ -2,7 +2,6 @@ import { BaseMessage } from "@langchain/core/messages";
 import { Runnable, RunnableInterface } from "@langchain/core/runnables";
 import { BaseRetrieverInterface } from "@langchain/core/retrievers";
 import { Document, DocumentInterface } from "@langchain/core/documents";
-
 //#region src/chains/retrieval.d.ts
 /**
  * Parameters for the createRetrievalChain method.
@@ -54,10 +53,7 @@ type CreateRetrievalChainParams<RunOutput> = {
  * const response = await chain.invoke({ input: "..." });
  * ```
  */
-declare function createRetrievalChain<RunOutput>({
-  retriever,
-  combineDocsChain
-}: CreateRetrievalChainParams<RunOutput>): Promise<Runnable<{
+declare function createRetrievalChain<RunOutput>({ retriever, combineDocsChain }: CreateRetrievalChainParams<RunOutput>): Promise<Runnable<{
   input: string;
   chat_history?: BaseMessage[] | string;
 } & {

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

@@ -1 +1 @@
-{"version":3,"file":"retrieval.d.cts","names":[],"sources":["../../src/chains/retrieval.ts"],"mappings":";;;;;;;;AAaA;KAAY,0BAAA;EAA0B;;;;;;;;;;EAWpC,SAAA,EACI,sBAAA,GAEA,iBAAA,CAAkB,MAAA,eAAqB,iBAAA;EAH3C;;;;;;;EAYA,gBAAA,EAAkB,iBAAA,CAAkB,MAAA,eAAqB,SAAA;AAAA;;;AAiC3D;;;;;;;;;;;;;;;;;;;;;;;;iBAAsB,oBAAA,WAAA,CAAA;EACpB,SAAA;EACA;AAAA,GACC,0BAAA,CAA2B,SAAA,IAAa,OAAA,CACzC,QAAA;EACI,KAAA;EAAe,YAAA,GAAe,WAAA;AAAA;EAAA,CAC7B,GAAA;AAAA;EAED,OAAA,EAAS,QAAA;EAAY,MAAA,EAAQ,SAAA;AAAA;EAAA,CAAiB,GAAA;AAAA"}
+{"version":3,"file":"retrieval.d.cts","names":[],"sources":["../../src/chains/retrieval.ts"],"mappings":";;;;;;;;KAaY,2BAA2B;;;;;;;;;;;EAWrC,WACI,yBAEA,kBAAkB,qBAAqB;;;;;;;;EAS3C,kBAAkB,kBAAkB,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAiCrC,qBAAqB,aACzC,WACA,oBACC,2BAA2B,aAAa,QACzC;EACI;EAAe,eAAe;;GAC7B;;EAED,SAAS;EAAY,QAAQ;;GAAiB"}

package/dist/chains/retrieval.d.ts

@@ -2,7 +2,6 @@ import { BaseMessage } from "@langchain/core/messages";
 import { Runnable, RunnableInterface } from "@langchain/core/runnables";
 import { Document, DocumentInterface } from "@langchain/core/documents";
 import { BaseRetrieverInterface } from "@langchain/core/retrievers";
-
 //#region src/chains/retrieval.d.ts
 /**
  * Parameters for the createRetrievalChain method.
@@ -54,10 +53,7 @@ type CreateRetrievalChainParams<RunOutput> = {
  * const response = await chain.invoke({ input: "..." });
  * ```
  */
-declare function createRetrievalChain<RunOutput>({
-  retriever,
-  combineDocsChain
-}: CreateRetrievalChainParams<RunOutput>): Promise<Runnable<{
+declare function createRetrievalChain<RunOutput>({ retriever, combineDocsChain }: CreateRetrievalChainParams<RunOutput>): Promise<Runnable<{
   input: string;
   chat_history?: BaseMessage[] | string;
 } & {

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

@@ -1 +1 @@
-{"version":3,"file":"retrieval.d.ts","names":[],"sources":["../../src/chains/retrieval.ts"],"mappings":";;;;;;;;AAaA;KAAY,0BAAA;EAA0B;;;;;;;;;;EAWpC,SAAA,EACI,sBAAA,GAEA,iBAAA,CAAkB,MAAA,eAAqB,iBAAA;EAH3C;;;;;;;EAYA,gBAAA,EAAkB,iBAAA,CAAkB,MAAA,eAAqB,SAAA;AAAA;;;AAiC3D;;;;;;;;;;;;;;;;;;;;;;;;iBAAsB,oBAAA,WAAA,CAAA;EACpB,SAAA;EACA;AAAA,GACC,0BAAA,CAA2B,SAAA,IAAa,OAAA,CACzC,QAAA;EACI,KAAA;EAAe,YAAA,GAAe,WAAA;AAAA;EAAA,CAC7B,GAAA;AAAA;EAED,OAAA,EAAS,QAAA;EAAY,MAAA,EAAQ,SAAA;AAAA;EAAA,CAAiB,GAAA;AAAA"}
+{"version":3,"file":"retrieval.d.ts","names":[],"sources":["../../src/chains/retrieval.ts"],"mappings":";;;;;;;;KAaY,2BAA2B;;;;;;;;;;;EAWrC,WACI,yBAEA,kBAAkB,qBAAqB;;;;;;;;EAS3C,kBAAkB,kBAAkB,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAiCrC,qBAAqB,aACzC,WACA,oBACC,2BAA2B,aAAa,QACzC;EACI;EAAe,eAAe;;GAC7B;;EAED,SAAS;EAAY,QAAQ;;GAAiB"}

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    // oxlint-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  // oxlint-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,CACvB"}
+{"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    // oxlint-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  // oxlint-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;CAChE,OAAO,CAAC,CAAC,KAAK,OAAQ,EAA6B,WAAW;AAChE;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4BA,eAAsB,qBAAgC,EACpD,WACA,oBAQA;CACA,IAAI;CACJ,IAAI,gBAAgB,SAAS,GAC3B,yBAAyB,iBAAiB,KAAK,EAC5C,UAAU,MAAM,OACjB,SACF,CAAC;MAGD,yBAAyB;CAgB3B,OAduB,iBAAiB,KAGrC,CACD,oBAAoB,OAAO;EACzB,SAAS,uBAAuB,WAAW,EACzC,SAAS,qBACX,CAAC;EACD,eAAe,UAAU,MAAM,gBAAgB,CAAC;CAClD,CAAC,GACD,oBAAoB,OAAO,EACzB,QAAQ,iBACV,CAAC,CACH,CAAC,CAAC,CAAC,WAAW,EAAE,SAAS,kBAAkB,CACvB;AACtB"}

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// oxlint-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"}
+{"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// oxlint-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;EACf,OAAO;CACT;CAEA,WAAW;CAEX,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,QAAQ;CACvB;CAEA,IAAI,aAAa;EACf,OAAO,KAAK,sBAAsB,WAAW,OAC3C,KAAK,wBAAwB,CAAC,iBAAiB,IAAI,CAAC,CACtD;CACF;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAA+B;EACzC,MAAM,MAAM;EACZ,KAAK,YAAY,OAAO;EACxB,KAAK,wBAAwB,OAAO;EACpC,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,wBACH,OAAO,yBAAyB,KAAK;CACzC;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,iBAAiB,KAAK,SAAS,aAAa;EAE9D,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,OAAO,MAAM,KAAK,UAAU,OAChC,UACA,YAAY,SAAS,WAAW,CAClC;EACA,MAAM,SAAS;GAAE;GAAU,iBAAiB;GAAM,GAAG;EAAO;EAC5D,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,mBAAmB,CAC1C;EACA,IAAI,KAAK,uBACP,OAAO;GACL,GAAG;GACH,iBAAiB;EACnB;EAEF,OAAO;CACT;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YACX,OACA,SAC2B;EAC3B,MAAM,IAAI,MAAM,iBAAiB;CACnC;CAEA,YAAuC;EACrC,MAAM,IAAI,MAAM,iBAAiB;CACnC;;;;;;;;;CAUA,OAAO,QACL,KACA,WACA,SAOkB;EAClB,MAAM,UAAUC,aAAAA,iBAAiB,KAAK,EACpC,QAAQ,SAAS,OACnB,CAAC;EACD,OAAO,IAAI,KAAK;GACd,GAAG;GACH;GACA,uBAAuB;EACzB,CAAC;CACH;AACF"}

package/dist/chains/retrieval_qa.d.cts

@@ -5,7 +5,6 @@ import { ChainValues } from "@langchain/core/utils/types";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { BaseRetrieverInterface } from "@langchain/core/retrievers";
-
 //#region src/chains/retrieval_qa.d.ts
 type LoadValues = Record<string, any>;
 /**
@@ -74,5 +73,5 @@ declare class RetrievalQAChain extends BaseChain implements RetrievalQAChainInpu
   static fromLLM(llm: BaseLanguageModelInterface, retriever: BaseRetrieverInterface, options?: Partial<Omit<RetrievalQAChainInput, "retriever" | "combineDocumentsChain" | "index">> & StuffQAChainParams): RetrievalQAChain;
 }
 //#endregion
-export { RetrievalQAChain, RetrievalQAChainInput };
+export { LoadValues, RetrievalQAChain, RetrievalQAChainInput };
 //# sourceMappingURL=retrieval_qa.d.cts.map

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

@@ -1 +1 @@
-{"version":3,"file":"retrieval_qa.d.cts","names":[],"sources":["../../src/chains/retrieval_qa.ts"],"mappings":";;;;;;;;;KAYY,UAAA,GAAa,MAAA;;AAAzB;;UAKiB,qBAAA,SAA8B,IAAA,CAAK,WAAA;EAClD,SAAA,EAAW,sBAAA;EACX,qBAAA,EAAuB,SAAA;EACvB,QAAA;EACA,qBAAA;AAAA;;;;;;;;;;;;;;;;;AAmCF;;;;;;;;;;;;;;;;cAAa,gBAAA,SACH,SAAA,YACG,qBAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,QAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAMJ,SAAA,EAAW,sBAAA;EAEX,qBAAA,EAAuB,SAAA;EAEvB,qBAAA;EAEA,WAAA,CAAY,MAAA,EAAQ,qBAAA;EAxBY;EAkC1B,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAuBX,UAAA,CAAA;EAAA,OAIa,WAAA,CACX,KAAA,EAAO,yBAAA,EACP,OAAA,EAAS,UAAA,GACR,OAAA,CAAQ,gBAAA;EAIX,SAAA,CAAA,GAAa,yBAAA;EA/DT;;;;;;;;EAAA,OA2EG,OAAA,CACL,GAAA,EAAK,0BAAA,EACL,SAAA,EAAW,sBAAA,EACX,OAAA,GAAU,OAAA,CACR,IAAA,CACE,qBAAA,sDAIF,kBAAA,GACD,gBAAA;AAAA"}
+{"version":3,"file":"retrieval_qa.d.cts","names":[],"sources":["../../src/chains/retrieval_qa.ts"],"mappings":";;;;;;;;KAYY,aAAa;;;;UAKR,8BAA8B,KAAK;EAClD,WAAW;EACX,uBAAuB;EACvB;EACA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAmCW,yBACH,qBACG;SAEJ;EAIP;MAEI;MAIA;EAMJ,WAAW;EAEX,uBAAuB;EAEvB;EAEA,YAAY,QAAQ;;EAUd,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAuBX;SAIa,YACX,OAAO,2BACP,SAAS,aACR,QAAQ;EAIX,aAAa;;;;;;;;;SAYN,QACL,KAAK,4BACL,WAAW,wBACX,UAAU,QACR,KACE,2EAIF,qBACD"}

package/dist/chains/retrieval_qa.d.ts

@@ -5,7 +5,6 @@ import { BaseLanguageModelInterface } from "@langchain/core/language_models/base
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { ChainValues } from "@langchain/core/utils/types";
 import { BaseRetrieverInterface } from "@langchain/core/retrievers";
-
 //#region src/chains/retrieval_qa.d.ts
 type LoadValues = Record<string, any>;
 /**
@@ -74,5 +73,5 @@ declare class RetrievalQAChain extends BaseChain implements RetrievalQAChainInpu
   static fromLLM(llm: BaseLanguageModelInterface, retriever: BaseRetrieverInterface, options?: Partial<Omit<RetrievalQAChainInput, "retriever" | "combineDocumentsChain" | "index">> & StuffQAChainParams): RetrievalQAChain;
 }
 //#endregion
-export { RetrievalQAChain, RetrievalQAChainInput };
+export { LoadValues, RetrievalQAChain, RetrievalQAChainInput };
 //# sourceMappingURL=retrieval_qa.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"retrieval_qa.d.ts","names":[],"sources":["../../src/chains/retrieval_qa.ts"],"mappings":";;;;;;;;;KAYY,UAAA,GAAa,MAAA;;AAAzB;;UAKiB,qBAAA,SAA8B,IAAA,CAAK,WAAA;EAClD,SAAA,EAAW,sBAAA;EACX,qBAAA,EAAuB,SAAA;EACvB,QAAA;EACA,qBAAA;AAAA;;;;;;;;;;;;;;;;;AAmCF;;;;;;;;;;;;;;;;cAAa,gBAAA,SACH,SAAA,YACG,qBAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,QAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAMJ,SAAA,EAAW,sBAAA;EAEX,qBAAA,EAAuB,SAAA;EAEvB,qBAAA;EAEA,WAAA,CAAY,MAAA,EAAQ,qBAAA;EAxBY;EAkC1B,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAuBX,UAAA,CAAA;EAAA,OAIa,WAAA,CACX,KAAA,EAAO,yBAAA,EACP,OAAA,EAAS,UAAA,GACR,OAAA,CAAQ,gBAAA;EAIX,SAAA,CAAA,GAAa,yBAAA;EA/DT;;;;;;;;EAAA,OA2EG,OAAA,CACL,GAAA,EAAK,0BAAA,EACL,SAAA,EAAW,sBAAA,EACX,OAAA,GAAU,OAAA,CACR,IAAA,CACE,qBAAA,sDAIF,kBAAA,GACD,gBAAA;AAAA"}
+{"version":3,"file":"retrieval_qa.d.ts","names":[],"sources":["../../src/chains/retrieval_qa.ts"],"mappings":";;;;;;;;KAYY,aAAa;;;;UAKR,8BAA8B,KAAK;EAClD,WAAW;EACX,uBAAuB;EACvB;EACA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAmCW,yBACH,qBACG;SAEJ;EAIP;MAEI;MAIA;EAMJ,WAAW;EAEX,uBAAuB;EAEvB;EAEA,YAAY,QAAQ;;EAUd,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAuBX;SAIa,YACX,OAAO,2BACP,SAAS,aACR,QAAQ;EAIX,aAAa;;;;;;;;;SAYN,QACL,KAAK,4BACL,WAAW,wBACX,UAAU,QACR,KACE,2EAIF,qBACD"}

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// oxlint-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// oxlint-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;EACf,OAAO;CACT;CAEA,WAAW;CAEX,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,QAAQ;CACvB;CAEA,IAAI,aAAa;EACf,OAAO,KAAK,sBAAsB,WAAW,OAC3C,KAAK,wBAAwB,CAAC,iBAAiB,IAAI,CAAC,CACtD;CACF;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAA+B;EACzC,MAAM,MAAM;EACZ,KAAK,YAAY,OAAO;EACxB,KAAK,wBAAwB,OAAO;EACpC,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,wBACH,OAAO,yBAAyB,KAAK;CACzC;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,iBAAiB,KAAK,SAAS,aAAa;EAE9D,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,OAAO,MAAM,KAAK,UAAU,OAChC,UACA,YAAY,SAAS,WAAW,CAClC;EACA,MAAM,SAAS;GAAE;GAAU,iBAAiB;GAAM,GAAG;EAAO;EAC5D,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,mBAAmB,CAC1C;EACA,IAAI,KAAK,uBACP,OAAO;GACL,GAAG;GACH,iBAAiB;EACnB;EAEF,OAAO;CACT;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YACX,OACA,SAC2B;EAC3B,MAAM,IAAI,MAAM,iBAAiB;CACnC;CAEA,YAAuC;EACrC,MAAM,IAAI,MAAM,iBAAiB;CACnC;;;;;;;;;CAUA,OAAO,QACL,KACA,WACA,SAOkB;EAClB,MAAM,UAAU,iBAAiB,KAAK,EACpC,QAAQ,SAAS,OACnB,CAAC;EACD,OAAO,IAAI,KAAK;GACd,GAAG;GACH;GACA,uBAAuB;EACzB,CAAC;CACH;AACF"}

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,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"}
+{"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;EACvC,MAAM,MAAM;EACZ,KAAK,WAAW,OAAO;CACzB;CAEA,IAAI,YAAsB;EACxB,OAAO,KAAK,SAAS;CACvB;CAEA,MAAM,MACJ,QACA,YAC6B;EAC7B,OAAO,KAAK,SAAS,QAAQ,QAAQ,YAAY,SAAS,CAAC;CAC7D;CAEA,aAAqB;EACnB,OAAO;CACT;;;;;;;;;;;CAYA,OAAO,QACL,KACA,QACA,SACA;EACA,MAAM,WAAW,IAAIC,kBAAAA,SAA6B;GAAE;GAAK;EAAO,CAAC;EACjE,OAAO,IAAI,eAAe;GAAE,GAAG;GAAS;EAAS,CAAC;CACpD;AACF"}

package/dist/chains/router/llm_router.d.cts

@@ -5,7 +5,6 @@ import { ChainValues } from "@langchain/core/utils/types";
 import { BasePromptTemplate } from "@langchain/core/prompts";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
-
 //#region src/chains/router/llm_router.d.ts
 /**
  * A type that represents the output schema of a router chain. It defines

package/dist/chains/router/llm_router.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"llm_router.d.cts","names":[],"sources":["../../../src/chains/router/llm_router.ts"],"mappings":";;;;;;;;;;;AAYA;;KAAY,kBAAA;EACV,WAAA;EACA,WAAA;IAAA,CAAgB,GAAA;EAAA;AAAA;;AAOlB;;;UAAiB,mBAAA,SAA4B,WAAA;EAC3C,QAAA,EAAU,QAAA,CAAS,kBAAA;AAAA;;;;;;;cASR,cAAA,SAAuB,WAAA,YAAuB,mBAAA;EACzD,QAAA,EAAU,QAAA,CAAS,kBAAA;EAEnB,WAAA,CAAY,MAAA,EAAQ,mBAAA;EAAA,IAKhB,SAAA,CAAA;EAIE,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,eACZ,OAAA,CAAQ,kBAAA;EAIX,UAAA,CAAA;EAlBU;;;;;;;;;;EAAA,OAgCH,OAAA,CACL,GAAA,EAAK,0BAAA,EACL,MAAA,EAAQ,kBAAA,EACR,OAAA,GAAU,IAAA,CAAK,mBAAA,gBAAgC,cAAA;AAAA"}
+{"version":3,"file":"llm_router.d.cts","names":[],"sources":["../../../src/chains/router/llm_router.ts"],"mappings":";;;;;;;;;;;;KAYY;EACV;EACA;KAAgB;;;;;;;UAOD,4BAA4B;EAC3C,UAAU,SAAS;;;;;;;;cASR,uBAAuB,uBAAuB;EACzD,UAAU,SAAS;EAEnB,YAAY,QAAQ;MAKhB;EAIE,MACJ,QAAQ,aACR,aAAa,yCACZ,QAAQ;EAIX;;;;;;;;;;;SAcO,QACL,KAAK,4BACL,QAAQ,oBACR,UAAU,KAAK,mCAAgC"}

package/dist/chains/router/llm_router.d.ts

@@ -5,7 +5,6 @@ import { BasePromptTemplate } from "@langchain/core/prompts";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { ChainValues } from "@langchain/core/utils/types";
-
 //#region src/chains/router/llm_router.d.ts
 /**
  * A type that represents the output schema of a router chain. It defines

package/dist/chains/router/llm_router.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"llm_router.d.ts","names":[],"sources":["../../../src/chains/router/llm_router.ts"],"mappings":";;;;;;;;;;;AAYA;;KAAY,kBAAA;EACV,WAAA;EACA,WAAA;IAAA,CAAgB,GAAA;EAAA;AAAA;;AAOlB;;;UAAiB,mBAAA,SAA4B,WAAA;EAC3C,QAAA,EAAU,QAAA,CAAS,kBAAA;AAAA;;;;;;;cASR,cAAA,SAAuB,WAAA,YAAuB,mBAAA;EACzD,QAAA,EAAU,QAAA,CAAS,kBAAA;EAEnB,WAAA,CAAY,MAAA,EAAQ,mBAAA;EAAA,IAKhB,SAAA,CAAA;EAIE,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,eACZ,OAAA,CAAQ,kBAAA;EAIX,UAAA,CAAA;EAlBU;;;;;;;;;;EAAA,OAgCH,OAAA,CACL,GAAA,EAAK,0BAAA,EACL,MAAA,EAAQ,kBAAA,EACR,OAAA,GAAU,IAAA,CAAK,mBAAA,gBAAgC,cAAA;AAAA"}
+{"version":3,"file":"llm_router.d.ts","names":[],"sources":["../../../src/chains/router/llm_router.ts"],"mappings":";;;;;;;;;;;;KAYY;EACV;EACA;KAAgB;;;;;;;UAOD,4BAA4B;EAC3C,UAAU,SAAS;;;;;;;;cASR,uBAAuB,uBAAuB;EACzD,UAAU,SAAS;EAEnB,YAAY,QAAQ;MAKhB;EAIE,MACJ,QAAQ,aACR,aAAa,yCACZ,QAAQ;EAIX;;;;;;;;;;;SAcO,QACL,KAAK,4BACL,QAAQ,oBACR,UAAU,KAAK,mCAAgC"}

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;EACvC,MAAM,MAAM;EACZ,KAAK,WAAW,OAAO;CACzB;CAEA,IAAI,YAAsB;EACxB,OAAO,KAAK,SAAS;CACvB;CAEA,MAAM,MACJ,QACA,YAC6B;EAC7B,OAAO,KAAK,SAAS,QAAQ,QAAQ,YAAY,SAAS,CAAC;CAC7D;CAEA,aAAqB;EACnB,OAAO;CACT;;;;;;;;;;;CAYA,OAAO,QACL,KACA,QACA,SACA;EACA,MAAM,WAAW,IAAI,SAA6B;GAAE;GAAK;EAAO,CAAC;EACjE,OAAO,IAAI,eAAe;GAAE,GAAG;GAAS;EAAS,CAAC;CACpD;AACF"}

package/dist/chains/router/multi_prompt.cjs

@@ -1,4 +1,3 @@
-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");

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,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,CAEuE,CAAC;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,CAIuB;GACxB,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"}
+{"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;EACA,OAAO,iBAAiB,kBAAkB,KAAK;GAC7C;GACA;GACA;GACA;GACA,qBAAqB;EACvB,CAAC;CACH;;;;;;;;;;;;;;;CAgBA,OAAO,kBACL,KACA,EACE,aACA,oBACA,iBACA,cACA,cACA,uBACA,uBAUgB;EAClB,MAAM,eAAeC,cAAAA,WAAW,aAAa,kBAAkB,CAAC,CAAC,KAC9D,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,MAChC;EAgBA,MAAM,eAAe,IAAIC,eAAAA,mBAdYC,OAAAA,EAAE,OAAO;GAC5C,aAAaA,OAAAA,EACV,OAAO,CAAC,CACR,SAAS,CAAC,CACV,SAAS,6DAA2D;GACvE,aAAaA,OAAAA,EACV,OAAO,EACN,OAAOA,OAAAA,EACJ,OAAO,CAAC,CACR,SAAS,sDAAsD,EACpE,CAAC,CAAC,CACD,SAAS,mCAAmC;EACjD,CAEuE,CAAC;EAExE,MAAM,kBAAkB,aAAa,KAAK,IAAI;EAU9C,MAAM,eAAe,IAAIC,wBAAAA,eAAe;GACtC,WAAA,GAAA,wBAAA,mBAAA,CATAC,4BAAAA,wCACE,aAAa,sBAAsB,EAAE,oBAAoB,EAAE,CAAC,CAC9D,GACA,EACE,cAAc,gBAChB,CAIuB;GACvB,gBAAgB,CAAC,OAAO;GACxB;EACF,CAAC;EAED,MAAM,cAAcC,mBAAAA,eAAe,QAAQ,KAAK,YAAY;EAC5D,MAAM,oBAAoBL,cAAAA,WACxB,aACA,eACF,CAAC,CAAC,QACC,KAAK,CAAC,MAAM,cAAc;GACzB,IAAI;GACJ,IAAI,OAAO,aAAa,UACtB,WAAW;QACN,IAAI,OAAO,aAAa,UAC7B,WAAW,IAAIG,wBAAAA,eAAe;IAClB;IACV,gBAAgB,CAAC,OAAO;GAC1B,CAAC;QAED,MAAM,IAAI,MAAM,yBAAyB;GAE3C,IAAI,QAAkB,IAAIG,kBAAAA,SAAS;IACjC,GAAG;IACH;IACA,QAAQ;GACV,CAAC;GACD,OAAO;EACT,GACA,CAAC,CACH;EAEA,MAAM,YAAY,IAAIC,qBAAAA,kBAAkB;GACtC,GAAG;GACH;GACA,WAAW;EACb,CAAC;EAED,OAAO,IAAI,iBAAiB;GAC1B,GAAG;GACH;GACA;GACA,cAAc,gBAAgB;EAChC,CAAC;CACH;CAEA,aAAqB;EACnB,OAAO;CACT;AACF"}

package/dist/chains/router/multi_prompt.d.cts

@@ -3,7 +3,6 @@ import { LLMChainInput } from "../llm_chain.cjs";
 import { MultiRouteChain, MultiRouteChainInput } from "./multi_route.cjs";
 import { PromptTemplate } from "@langchain/core/prompts";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
-
 //#region src/chains/router/multi_prompt.d.ts
 /**
  * A class that represents a multi-prompt chain in the LangChain
@@ -51,15 +50,7 @@ declare class MultiPromptChain extends MultiRouteChain {
    * @param multiRouteChainOpts Optional parameters for the MultiRouteChainInput, excluding 'defaultChain'.
    * @returns An instance of MultiPromptChain.
    */
-  static fromLLMAndPrompts(llm: BaseLanguageModelInterface, {
-    promptNames,
-    promptDescriptions,
-    promptTemplates,
-    defaultChain,
-    llmChainOpts,
-    conversationChainOpts,
-    multiRouteChainOpts
-  }: {
+  static fromLLMAndPrompts(llm: BaseLanguageModelInterface, { promptNames, promptDescriptions, promptTemplates, defaultChain, llmChainOpts, conversationChainOpts, multiRouteChainOpts }: {
     promptNames: string[];
     promptDescriptions: string[];
     promptTemplates: string[] | PromptTemplate[];

package/dist/chains/router/multi_prompt.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"multi_prompt.d.cts","names":[],"sources":["../../../src/chains/router/multi_prompt.ts"],"mappings":";;;;;;;;;AAuCA;;;;;;;;;;;;;;;;;;;;;;;;;cAAa,gBAAA,SAAyB,eAAA;EAAA;;;EAAA,OAI7B,WAAA,CACL,GAAA,EAAK,0BAAA,EACL,WAAA,YACA,kBAAA,YACA,eAAA,aAA4B,cAAA,IAC5B,YAAA,GAAe,SAAA,EACf,OAAA,GAAU,IAAA,CAAK,oBAAA,oBAAqC,gBAAA;EAN/C;;;;;;;;;;;;;;EAAA,OA+BA,iBAAA,CACL,GAAA,EAAK,0BAAA;IAEH,WAAA;IACA,kBAAA;IACA,eAAA;IACA,YAAA;IACA,YAAA;IACA,qBAAA;IACA;EAAA;IAEA,WAAA;IACA,kBAAA;IACA,eAAA,aAA4B,cAAA;IAC5B,YAAA,GAAe,SAAA;IACf,YAAA,GAAe,IAAA,CAAK,aAAA;IACpB,qBAAA,GAAwB,IAAA,CAAK,aAAA;IAC7B,mBAAA,GAAsB,IAAA,CAAK,oBAAA;EAAA,IAE5B,gBAAA;EA8EH,UAAA,CAAA;AAAA"}
+{"version":3,"file":"multi_prompt.d.cts","names":[],"sources":["../../../src/chains/router/multi_prompt.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAuCa,yBAAyB;;;;SAI7B,YACL,KAAK,4BACL,uBACA,8BACA,4BAA4B,kBAC5B,eAAe,WACf,UAAU,KAAK,wCAAqC;;;;;;;;;;;;;;;SAyB/C,kBACL,KAAK,8BAEH,aACA,oBACA,iBACA,cACA,cACA,uBACA;IAEA;IACA;IACA,4BAA4B;IAC5B,eAAe;IACf,eAAe,KAAK;IACpB,wBAAwB,KAAK;IAC7B,sBAAsB,KAAK;MAE5B;EA8EH"}

package/dist/chains/router/multi_prompt.d.ts

@@ -3,7 +3,6 @@ import { LLMChainInput } from "../llm_chain.js";
 import { MultiRouteChain, MultiRouteChainInput } from "./multi_route.js";
 import { PromptTemplate } from "@langchain/core/prompts";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
-
 //#region src/chains/router/multi_prompt.d.ts
 /**
  * A class that represents a multi-prompt chain in the LangChain
@@ -51,15 +50,7 @@ declare class MultiPromptChain extends MultiRouteChain {
    * @param multiRouteChainOpts Optional parameters for the MultiRouteChainInput, excluding 'defaultChain'.
    * @returns An instance of MultiPromptChain.
    */
-  static fromLLMAndPrompts(llm: BaseLanguageModelInterface, {
-    promptNames,
-    promptDescriptions,
-    promptTemplates,
-    defaultChain,
-    llmChainOpts,
-    conversationChainOpts,
-    multiRouteChainOpts
-  }: {
+  static fromLLMAndPrompts(llm: BaseLanguageModelInterface, { promptNames, promptDescriptions, promptTemplates, defaultChain, llmChainOpts, conversationChainOpts, multiRouteChainOpts }: {
     promptNames: string[];
     promptDescriptions: string[];
     promptTemplates: string[] | PromptTemplate[];

package/dist/chains/router/multi_prompt.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"multi_prompt.d.ts","names":[],"sources":["../../../src/chains/router/multi_prompt.ts"],"mappings":";;;;;;;;;AAuCA;;;;;;;;;;;;;;;;;;;;;;;;;cAAa,gBAAA,SAAyB,eAAA;EAAA;;;EAAA,OAI7B,WAAA,CACL,GAAA,EAAK,0BAAA,EACL,WAAA,YACA,kBAAA,YACA,eAAA,aAA4B,cAAA,IAC5B,YAAA,GAAe,SAAA,EACf,OAAA,GAAU,IAAA,CAAK,oBAAA,oBAAqC,gBAAA;EAN/C;;;;;;;;;;;;;;EAAA,OA+BA,iBAAA,CACL,GAAA,EAAK,0BAAA;IAEH,WAAA;IACA,kBAAA;IACA,eAAA;IACA,YAAA;IACA,YAAA;IACA,qBAAA;IACA;EAAA;IAEA,WAAA;IACA,kBAAA;IACA,eAAA,aAA4B,cAAA;IAC5B,YAAA,GAAe,SAAA;IACf,YAAA,GAAe,IAAA,CAAK,aAAA;IACpB,qBAAA,GAAwB,IAAA,CAAK,aAAA;IAC7B,mBAAA,GAAsB,IAAA,CAAK,oBAAA;EAAA,IAE5B,gBAAA;EA8EH,UAAA,CAAA;AAAA"}
+{"version":3,"file":"multi_prompt.d.ts","names":[],"sources":["../../../src/chains/router/multi_prompt.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAuCa,yBAAyB;;;;SAI7B,YACL,KAAK,4BACL,uBACA,8BACA,4BAA4B,kBAC5B,eAAe,WACf,UAAU,KAAK,wCAAqC;;;;;;;;;;;;;;;SAyB/C,kBACL,KAAK,8BAEH,aACA,oBACA,iBACA,cACA,cACA,uBACA;IAEA;IACA;IACA,4BAA4B;IAC5B,eAAe;IACf,eAAe,KAAK;IACpB,wBAAwB,KAAK;IAC7B,sBAAsB,KAAK;MAE5B;EA8EH"}

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,CAEuE,CAAC;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,CAIuB;GACxB,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;EACA,OAAO,iBAAiB,kBAAkB,KAAK;GAC7C;GACA;GACA;GACA;GACA,qBAAqB;EACvB,CAAC;CACH;;;;;;;;;;;;;;;CAgBA,OAAO,kBACL,KACA,EACE,aACA,oBACA,iBACA,cACA,cACA,uBACA,uBAUgB;EAClB,MAAM,eAAe,WAAW,aAAa,kBAAkB,CAAC,CAAC,KAC9D,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,MAChC;EAgBA,MAAM,eAAe,IAAI,mBAdY,EAAE,OAAO;GAC5C,aAAa,EACV,OAAO,CAAC,CACR,SAAS,CAAC,CACV,SAAS,6DAA2D;GACvE,aAAa,EACV,OAAO,EACN,OAAO,EACJ,OAAO,CAAC,CACR,SAAS,sDAAsD,EACpE,CAAC,CAAC,CACD,SAAS,mCAAmC;EACjD,CAEuE,CAAC;EAExE,MAAM,kBAAkB,aAAa,KAAK,IAAI;EAU9C,MAAM,eAAe,IAAI,eAAe;GACtC,UAVqB,mBACrB,wCACE,aAAa,sBAAsB,EAAE,oBAAoB,EAAE,CAAC,CAC9D,GACA,EACE,cAAc,gBAChB,CAIuB;GACvB,gBAAgB,CAAC,OAAO;GACxB;EACF,CAAC;EAED,MAAM,cAAc,eAAe,QAAQ,KAAK,YAAY;EAC5D,MAAM,oBAAoB,WACxB,aACA,eACF,CAAC,CAAC,QACC,KAAK,CAAC,MAAM,cAAc;GACzB,IAAI;GACJ,IAAI,OAAO,aAAa,UACtB,WAAW;QACN,IAAI,OAAO,aAAa,UAC7B,WAAW,IAAI,eAAe;IAClB;IACV,gBAAgB,CAAC,OAAO;GAC1B,CAAC;QAED,MAAM,IAAI,MAAM,yBAAyB;GAE3C,IAAI,QAAkB,IAAI,SAAS;IACjC,GAAG;IACH;IACA,QAAQ;GACV,CAAC;GACD,OAAO;EACT,GACA,CAAC,CACH;EAEA,MAAM,YAAY,IAAI,kBAAkB;GACtC,GAAG;GACH;GACA,WAAW;EACb,CAAC;EAED,OAAO,IAAI,iBAAiB;GAC1B,GAAG;GACH;GACA;GACA,cAAc,gBAAgB;EAChC,CAAC;CACH;CAEA,aAAqB;EACnB,OAAO;CACT;AACF"}

package/dist/chains/router/multi_retrieval_qa.cjs

@@ -1,4 +1,3 @@
-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");

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,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,CAI6B,CAAC;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,CAGuB;GACxB,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,QAAA,IANiBJ,wBAAAA,eAAe;IAChC,UAFqBK,qBAAAA,iBAAiB,QAAQ,SAAS,QAE/B;IACxB,gBAAgB,CAAC,WAAW,QAAQ;IACrC,CAGO;GACN,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;EACzB,OAAO,CAAC,QAAQ;CAClB;;;;CAKA,OAAO,eACL,KACA,gBACA,uBACA,YACA,kBACA,UACA,SACA;EACA,OAAO,sBAAsB,qBAAqB,KAAK;GACrD;GACA;GACA;GACA;GACA;GACA,yBAAyB;EAC3B,CAAC;CACH;;;;;;;;;;;;;;;;;CAkBA,OAAO,qBACL,KACA,EACE,gBACA,uBACA,YACA,kBACA,UACA,yBACA,wBAcqB;EACvB,MAAM,EAAE,kBAAkB,eAAe,iBAAiB,YAAY,CAAC;EACvE,IAAI,iBAAiB,CAAC,kBACpB,MAAM,IAAI,MACR,0GACF;EAEF,MAAM,eAAeC,cAAAA,WACnB,gBACA,qBACF,CAAC,CAAC,KAAK,CAAC,MAAM,UAAU,GAAG,KAAK,IAAI,MAAM;EAgB1C,MAAM,eAAe,IAAIC,eAAAA,mBAdYC,OAAAA,EAAE,OAAO;GAC5C,aAAaA,OAAAA,EACV,OAAO,CAAC,CACR,SAAS,CAAC,CACV,SAAS,6DAA2D;GACvE,aAAaA,OAAAA,EACV,OAAO,EACN,OAAOA,OAAAA,EACJ,OAAO,CAAC,CACR,SAAS,sDAAsD,EACpE,CAAC,CAAC,CACD,SAAS,mCAAmC;EACjD,CAI6B,CAAC;EAE9B,MAAM,kBAAkB,aAAa,KAAK,IAAI;EAS9C,MAAM,eAAe,IAAIC,wBAAAA,eAAe;GACtC,WAAA,GAAA,wBAAA,mBAAA,CARAC,+BAAAA,2CACE,aAAa,sBAAsB,EAAE,oBAAoB,EAAE,CAAC,CAC9D,GACA,EACE,cAAc,gBAChB,CAGuB;GACvB,gBAAgB,CAAC,OAAO;GACxB;EACF,CAAC;EAED,MAAM,cAAcC,mBAAAA,eAAe,QAAQ,KAAK,YAAY;EAE5D,MAAM,oBAAoBL,cAAAA,WAExB,gBAAgB,YAHF,oBAAoB,WAAW,UAAU,IAAI,CAGxB,CAAC,CAAC,QACpC,KAAK,CAAC,MAAM,WAAW,YAAY;GAClC,MAAM,MAEF,wBAAwB,CAAC;GAC7B,IAAI,QACF,IAAI,SAAS;GAEf,IAAI,QAAQM,qBAAAA,iBAAiB,QAAQ,KAAK,WAAW,GAAG;GACxD,OAAO;EACT,GACA,CAAC,CACH;EAEA,IAAI;EACJ,IAAI,cACF,gBAAgB;OACX,IAAI,kBACT,gBAAgBA,qBAAAA,iBAAiB,QAAQ,KAAK,kBAAkB;GAC9D,GAAG;GACH,QAAQ;EACV,CAAC;OAOD,gBAAgB,IAAIC,qBAAAA,kBAAkB;GACpC;GACA,QAAA,IANiBJ,wBAAAA,eAAe;IAChC,UAFqBK,qBAAAA,iBAAiB,QAAQ,SAAS,OAEhC;IACvB,gBAAgB,CAAC,WAAW,OAAO;GACrC,CAGO;GACL,WAAW;EACb,CAAC;EAGH,OAAO,IAAI,sBAAsB;GAC/B,GAAG;GACH;GACA;GACA,cAAc;EAChB,CAAC;CACH;CAEA,aAAqB;EACnB,OAAO;CACT;AACF"}

package/dist/chains/router/multi_retrieval_qa.d.cts

@@ -4,7 +4,6 @@ import { MultiRouteChain, MultiRouteChainInput } from "./multi_route.cjs";
 import { PromptTemplate } from "@langchain/core/prompts";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { BaseRetrieverInterface } from "@langchain/core/retrievers";
-
 //#region src/chains/router/multi_retrieval_qa.d.ts
 /**
  * A type that represents the default values for the MultiRetrievalQAChain
@@ -73,15 +72,7 @@ declare class MultiRetrievalQAChain extends MultiRouteChain {
    * @param retrievalQAChainOpts Additional optional parameters for the retrieval QA chain.
    * @returns A new instance of MultiRetrievalQAChain.
    */
-  static fromLLMAndRetrievers(llm: BaseLanguageModelInterface, {
-    retrieverNames,
-    retrieverDescriptions,
-    retrievers,
-    retrieverPrompts,
-    defaults,
-    multiRetrievalChainOpts,
-    retrievalQAChainOpts
-  }: {
+  static fromLLMAndRetrievers(llm: BaseLanguageModelInterface, { retrieverNames, retrieverDescriptions, retrievers, retrieverPrompts, defaults, multiRetrievalChainOpts, retrievalQAChainOpts }: {
     retrieverNames: string[];
     retrieverDescriptions: string[];
     retrievers: BaseRetrieverInterface[];
@@ -95,5 +86,5 @@ declare class MultiRetrievalQAChain extends MultiRouteChain {
   _chainType(): string;
 }
 //#endregion
-export { MultiRetrievalQAChain };
+export { MultiRetrievalDefaults, MultiRetrievalQAChain };
 //# sourceMappingURL=multi_retrieval_qa.d.cts.map

package/dist/chains/router/multi_retrieval_qa.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"multi_retrieval_qa.d.cts","names":[],"sources":["../../../src/chains/router/multi_retrieval_qa.ts"],"mappings":";;;;;;;;;;AAwBA;;;KAAY,sBAAA;EACV,gBAAA,GAAmB,sBAAA;EACnB,aAAA,GAAgB,cAAA;EAChB,YAAA,GAAe,SAAA;AAAA;;;;;;;;;AAsCjB;;;;;;;;;;;;;;;;;;;;;;;;;;;cAAa,qBAAA,SAA8B,eAAA;EAAA,IACrC,UAAA,CAAA;EADoD;;;EAAA,OAQjD,cAAA,CACL,GAAA,EAAK,0BAAA,EACL,cAAA,YACA,qBAAA,YACA,UAAA,EAAY,sBAAA,IACZ,gBAAA,GAAmB,cAAA,IACnB,QAAA,GAAW,sBAAA,EACX,OAAA,GAAU,IAAA,CAAK,oBAAA,oBAAqC,qBAAA;EAP/C;;;;;;;;;;;;;;;;EAAA,OAmCA,oBAAA,CACL,GAAA,EAAK,0BAAA;IAEH,cAAA;IACA,qBAAA;IACA,UAAA;IACA,gBAAA;IACA,QAAA;IACA,uBAAA;IACA;EAAA;IAEA,cAAA;IACA,qBAAA;IACA,UAAA,EAAY,sBAAA;IACZ,gBAAA,GAAmB,cAAA;IACnB,QAAA,GAAW,sBAAA;IACX,uBAAA,GAA0B,IAAA,CAAK,oBAAA;IAC/B,oBAAA,GAAuB,OAAA,CACrB,IAAA,CAAK,qBAAA;MAEL,MAAA,GAAS,cAAA;IAAA;EAAA,IAGZ,qBAAA;EA4FH,UAAA,CAAA;AAAA"}
+{"version":3,"file":"multi_retrieval_qa.d.cts","names":[],"sources":["../../../src/chains/router/multi_retrieval_qa.ts"],"mappings":";;;;;;;;;;;;KAwBY;EACV,mBAAmB;EACnB,gBAAgB;EAChB,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAsCJ,8BAA8B;MACrC;;;;SAOG,eACL,KAAK,4BACL,0BACA,iCACA,YAAY,0BACZ,mBAAmB,kBACnB,WAAW,wBACX,UAAU,KAAK,wCAAqC;;;;;;;;;;;;;;;;;SA4B/C,qBACL,KAAK,8BAEH,gBACA,uBACA,YACA,kBACA,UACA,yBACA;IAEA;IACA;IACA,YAAY;IACZ,mBAAmB;IACnB,WAAW;IACX,0BAA0B,KAAK;IAC/B,uBAAuB,QACrB,KAAK;MAEL,SAAS;;MAGZ;EA4FH"}