@langchain/classic 1.0.0-alpha.1

package/dist/chains/retrieval_qa.d.cts

@@ -0,0 +1,79 @@
+import { SerializedVectorDBQAChain } from "./serde.cjs";
+import { BaseChain, ChainInputs } from "./base.cjs";
+import { StuffQAChainParams } from "./question_answering/load.cjs";
+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
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+type LoadValues = Record<string, any>;
+/**
+ * Interface for the input parameters of the RetrievalQAChain class.
+ */
+interface RetrievalQAChainInput extends Omit<ChainInputs, "memory"> {
+  retriever: BaseRetrieverInterface;
+  combineDocumentsChain: BaseChain;
+  inputKey?: string;
+  returnSourceDocuments?: boolean;
+}
+/**
+ * Class representing a chain for performing question-answering tasks with
+ * a retrieval component.
+ * @example
+ * ```typescript
+ * import { createStuffDocumentsChain } from "@langchain/classic/chains/combine_documents";
+ * import { ChatPromptTemplate } from "@langchain/core/prompts";
+ * import { createRetrievalChain } from "@langchain/classic/chains/retrieval";
+ * import { MemoryVectorStore } from "@langchain/classic/vectorstores/memory";
+ *
+ * const documents = [...your documents here];
+ * const embeddings = ...your embeddings model;
+ * const llm = ...your LLM model;
+ *
+ * const vectorstore = await MemoryVectorStore.fromDocuments(
+ *   documents,
+ *   embeddings
+ * );
+ * const prompt = ChatPromptTemplate.fromTemplate(`Answer the user's question: {input} based on the following context {context}`);
+ *
+ * const combineDocsChain = await createStuffDocumentsChain({
+ *   llm,
+ *   prompt,
+ * });
+ * const retriever = vectorstore.asRetriever();
+ *
+ * const retrievalChain = await createRetrievalChain({
+ *   combineDocsChain,
+ *   retriever,
+ * });
+ * ```
+ */
+declare class RetrievalQAChain extends BaseChain implements RetrievalQAChainInput {
+  static lc_name(): string;
+  inputKey: string;
+  get inputKeys(): string[];
+  get outputKeys(): string[];
+  retriever: BaseRetrieverInterface;
+  combineDocumentsChain: BaseChain;
+  returnSourceDocuments: boolean;
+  constructor(fields: RetrievalQAChainInput);
+  /** @ignore */
+  _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
+  _chainType(): "retrieval_qa";
+  static deserialize(_data: SerializedVectorDBQAChain, _values: LoadValues): Promise<RetrievalQAChain>;
+  serialize(): SerializedVectorDBQAChain;
+  /**
+   * Creates a new instance of RetrievalQAChain using a BaseLanguageModel
+   * and a BaseRetriever.
+   * @param llm The BaseLanguageModel used to generate a new question.
+   * @param retriever The BaseRetriever used to retrieve relevant documents.
+   * @param options Optional parameters for the RetrievalQAChain.
+   * @returns A new instance of RetrievalQAChain.
+   */
+  static fromLLM(llm: BaseLanguageModelInterface, retriever: BaseRetrieverInterface, options?: Partial<Omit<RetrievalQAChainInput, "retriever" | "combineDocumentsChain" | "index">> & StuffQAChainParams): RetrievalQAChain;
+}
+//#endregion
+export { RetrievalQAChain, RetrievalQAChainInput };
+//# sourceMappingURL=retrieval_qa.d.cts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieval_qa.d.cts","names":["BaseLanguageModelInterface","BaseRetrieverInterface","ChainValues","CallbackManagerForChainRun","BaseChain","ChainInputs","SerializedVectorDBQAChain","StuffQAChainParams","LoadValues","Record","RetrievalQAChainInput","Omit","RetrievalQAChain","Promise","Partial"],"sources":["../../src/chains/retrieval_qa.d.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 { StuffQAChainParams } from \"./question_answering/load.js\";\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\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 * 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 declare class RetrievalQAChain extends BaseChain implements RetrievalQAChainInput {\n    static lc_name(): string;\n    inputKey: string;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    retriever: BaseRetrieverInterface;\n    combineDocumentsChain: BaseChain;\n    returnSourceDocuments: boolean;\n    constructor(fields: RetrievalQAChainInput);\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"retrieval_qa\";\n    static deserialize(_data: SerializedVectorDBQAChain, _values: LoadValues): Promise<RetrievalQAChain>;\n    serialize(): SerializedVectorDBQAChain;\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(llm: BaseLanguageModelInterface, retriever: BaseRetrieverInterface, options?: Partial<Omit<RetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\" | \"index\">> & StuffQAChainParams): RetrievalQAChain;\n}\n"],"mappings":";;;;;;;;;;KAQYQ,UAAAA,GAAaC;AAAzB;AAIA;;AAAoDJ,UAAnCK,qBAAAA,SAA8BC,IAAKN,CAAAA,WAAAA,EAAAA,QAAAA,CAAAA,CAAAA;EAAW,SAChDJ,EAAAA,sBAAAA;EAAsB,qBACVG,EAAAA,SAAAA;EAAS,QAFWO,CAAAA,EAAAA,MAAAA;EAAI,qBAAA,CAAA,EAAA,OAAA;AAsCnD;;;;;;;;;;;;;;;;;;;;;;;AAAwF;;;;;;;;;;cAAnEC,gBAAAA,SAAyBR,SAAAA,YAAqBM;;;;;aAKpDT;yBACYG;;sBAEHM;;gBAENR,0BAA0BC,6BAA6BU,QAAQX;;4BAEnDI,oCAAoCE,aAAaK,QAAQD;eACtEN;;;;;;;;;sBASON,uCAAuCC,kCAAkCa,QAAQH,KAAKD,2EAA2EH,qBAAqBK"}

package/dist/chains/retrieval_qa.d.ts

@@ -0,0 +1,79 @@
+import { SerializedVectorDBQAChain } from "./serde.js";
+import { BaseChain, ChainInputs } from "./base.js";
+import { StuffQAChainParams } from "./question_answering/load.js";
+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
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+type LoadValues = Record<string, any>;
+/**
+ * Interface for the input parameters of the RetrievalQAChain class.
+ */
+interface RetrievalQAChainInput extends Omit<ChainInputs, "memory"> {
+  retriever: BaseRetrieverInterface;
+  combineDocumentsChain: BaseChain;
+  inputKey?: string;
+  returnSourceDocuments?: boolean;
+}
+/**
+ * Class representing a chain for performing question-answering tasks with
+ * a retrieval component.
+ * @example
+ * ```typescript
+ * import { createStuffDocumentsChain } from "@langchain/classic/chains/combine_documents";
+ * import { ChatPromptTemplate } from "@langchain/core/prompts";
+ * import { createRetrievalChain } from "@langchain/classic/chains/retrieval";
+ * import { MemoryVectorStore } from "@langchain/classic/vectorstores/memory";
+ *
+ * const documents = [...your documents here];
+ * const embeddings = ...your embeddings model;
+ * const llm = ...your LLM model;
+ *
+ * const vectorstore = await MemoryVectorStore.fromDocuments(
+ *   documents,
+ *   embeddings
+ * );
+ * const prompt = ChatPromptTemplate.fromTemplate(`Answer the user's question: {input} based on the following context {context}`);
+ *
+ * const combineDocsChain = await createStuffDocumentsChain({
+ *   llm,
+ *   prompt,
+ * });
+ * const retriever = vectorstore.asRetriever();
+ *
+ * const retrievalChain = await createRetrievalChain({
+ *   combineDocsChain,
+ *   retriever,
+ * });
+ * ```
+ */
+declare class RetrievalQAChain extends BaseChain implements RetrievalQAChainInput {
+  static lc_name(): string;
+  inputKey: string;
+  get inputKeys(): string[];
+  get outputKeys(): string[];
+  retriever: BaseRetrieverInterface;
+  combineDocumentsChain: BaseChain;
+  returnSourceDocuments: boolean;
+  constructor(fields: RetrievalQAChainInput);
+  /** @ignore */
+  _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
+  _chainType(): "retrieval_qa";
+  static deserialize(_data: SerializedVectorDBQAChain, _values: LoadValues): Promise<RetrievalQAChain>;
+  serialize(): SerializedVectorDBQAChain;
+  /**
+   * Creates a new instance of RetrievalQAChain using a BaseLanguageModel
+   * and a BaseRetriever.
+   * @param llm The BaseLanguageModel used to generate a new question.
+   * @param retriever The BaseRetriever used to retrieve relevant documents.
+   * @param options Optional parameters for the RetrievalQAChain.
+   * @returns A new instance of RetrievalQAChain.
+   */
+  static fromLLM(llm: BaseLanguageModelInterface, retriever: BaseRetrieverInterface, options?: Partial<Omit<RetrievalQAChainInput, "retriever" | "combineDocumentsChain" | "index">> & StuffQAChainParams): RetrievalQAChain;
+}
+//#endregion
+export { RetrievalQAChain, RetrievalQAChainInput };
+//# sourceMappingURL=retrieval_qa.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieval_qa.d.ts","names":["BaseLanguageModelInterface","BaseRetrieverInterface","ChainValues","CallbackManagerForChainRun","BaseChain","ChainInputs","SerializedVectorDBQAChain","StuffQAChainParams","LoadValues","Record","RetrievalQAChainInput","Omit","RetrievalQAChain","Promise","Partial"],"sources":["../../src/chains/retrieval_qa.d.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 { StuffQAChainParams } from \"./question_answering/load.js\";\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\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 * 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 declare class RetrievalQAChain extends BaseChain implements RetrievalQAChainInput {\n    static lc_name(): string;\n    inputKey: string;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    retriever: BaseRetrieverInterface;\n    combineDocumentsChain: BaseChain;\n    returnSourceDocuments: boolean;\n    constructor(fields: RetrievalQAChainInput);\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"retrieval_qa\";\n    static deserialize(_data: SerializedVectorDBQAChain, _values: LoadValues): Promise<RetrievalQAChain>;\n    serialize(): SerializedVectorDBQAChain;\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(llm: BaseLanguageModelInterface, retriever: BaseRetrieverInterface, options?: Partial<Omit<RetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\" | \"index\">> & StuffQAChainParams): RetrievalQAChain;\n}\n"],"mappings":";;;;;;;;;;KAQYQ,UAAAA,GAAaC;AAAzB;AAIA;;AAAoDJ,UAAnCK,qBAAAA,SAA8BC,IAAKN,CAAAA,WAAAA,EAAAA,QAAAA,CAAAA,CAAAA;EAAW,SAChDJ,EAAAA,sBAAAA;EAAsB,qBACVG,EAAAA,SAAAA;EAAS,QAFWO,CAAAA,EAAAA,MAAAA;EAAI,qBAAA,CAAA,EAAA,OAAA;AAsCnD;;;;;;;;;;;;;;;;;;;;;;;AAAwF;;;;;;;;;;cAAnEC,gBAAAA,SAAyBR,SAAAA,YAAqBM;;;;;aAKpDT;yBACYG;;sBAEHM;;gBAENR,0BAA0BC,6BAA6BU,QAAQX;;4BAEnDI,oCAAoCE,aAAaK,QAAQD;eACtEN;;;;;;;;;sBASON,uCAAuCC,kCAAkCa,QAAQH,KAAKD,2EAA2EH,qBAAqBK"}

package/dist/chains/retrieval_qa.js

@@ -0,0 +1,104 @@
+import { BaseChain } from "./base.js";
+import { loadQAStuffChain } from "./question_answering/load.js";
+
+//#region src/chains/retrieval_qa.ts
+/**
+* Class representing a chain for performing question-answering tasks with
+* a retrieval component.
+* @example
+* ```typescript
+* import { createStuffDocumentsChain } from "@langchain/classic/chains/combine_documents";
+* import { ChatPromptTemplate } from "@langchain/core/prompts";
+* import { createRetrievalChain } from "@langchain/classic/chains/retrieval";
+* import { MemoryVectorStore } from "@langchain/classic/vectorstores/memory";
+*
+* const documents = [...your documents here];
+* const embeddings = ...your embeddings model;
+* const llm = ...your LLM model;
+*
+* const vectorstore = await MemoryVectorStore.fromDocuments(
+*   documents,
+*   embeddings
+* );
+* const prompt = ChatPromptTemplate.fromTemplate(`Answer the user's question: {input} based on the following context {context}`);
+*
+* const combineDocsChain = await createStuffDocumentsChain({
+*   llm,
+*   prompt,
+* });
+* const retriever = vectorstore.asRetriever();
+*
+* const retrievalChain = await createRetrievalChain({
+*   combineDocsChain,
+*   retriever,
+* });
+* ```
+*/
+var RetrievalQAChain = class extends BaseChain {
+	static lc_name() {
+		return "RetrievalQAChain";
+	}
+	inputKey = "query";
+	get inputKeys() {
+		return [this.inputKey];
+	}
+	get outputKeys() {
+		return this.combineDocumentsChain.outputKeys.concat(this.returnSourceDocuments ? ["sourceDocuments"] : []);
+	}
+	retriever;
+	combineDocumentsChain;
+	returnSourceDocuments = false;
+	constructor(fields) {
+		super(fields);
+		this.retriever = fields.retriever;
+		this.combineDocumentsChain = fields.combineDocumentsChain;
+		this.inputKey = fields.inputKey ?? this.inputKey;
+		this.returnSourceDocuments = fields.returnSourceDocuments ?? this.returnSourceDocuments;
+	}
+	/** @ignore */
+	async _call(values, runManager) {
+		if (!(this.inputKey in values)) throw new Error(`Question key "${this.inputKey}" not found.`);
+		const question = values[this.inputKey];
+		const docs = await this.retriever.invoke(question, runManager?.getChild("retriever"));
+		const inputs = {
+			question,
+			input_documents: docs,
+			...values
+		};
+		const result = await this.combineDocumentsChain.call(inputs, runManager?.getChild("combine_documents"));
+		if (this.returnSourceDocuments) return {
+			...result,
+			sourceDocuments: docs
+		};
+		return result;
+	}
+	_chainType() {
+		return "retrieval_qa";
+	}
+	static async deserialize(_data, _values) {
+		throw new Error("Not implemented");
+	}
+	serialize() {
+		throw new Error("Not implemented");
+	}
+	/**
+	* Creates a new instance of RetrievalQAChain using a BaseLanguageModel
+	* and a BaseRetriever.
+	* @param llm The BaseLanguageModel used to generate a new question.
+	* @param retriever The BaseRetriever used to retrieve relevant documents.
+	* @param options Optional parameters for the RetrievalQAChain.
+	* @returns A new instance of RetrievalQAChain.
+	*/
+	static fromLLM(llm, retriever, options) {
+		const qaChain = loadQAStuffChain(llm, { prompt: options?.prompt });
+		return new this({
+			...options,
+			retriever,
+			combineDocumentsChain: qaChain
+		});
+	}
+};
+
+//#endregion
+export { RetrievalQAChain };
+//# sourceMappingURL=retrieval_qa.js.map

package/dist/chains/retrieval_qa.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieval_qa.js","names":["fields: RetrievalQAChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun","question: string","_data: SerializedVectorDBQAChain","_values: LoadValues","llm: BaseLanguageModelInterface","retriever: BaseRetrieverInterface","options?: Partial<\n      Omit<\n        RetrievalQAChainInput,\n        \"retriever\" | \"combineDocumentsChain\" | \"index\"\n      >\n    > &\n      StuffQAChainParams"],"sources":["../../src/chains/retrieval_qa.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { SerializedVectorDBQAChain } from \"./serde.js\";\nimport {\n  StuffQAChainParams,\n  loadQAStuffChain,\n} from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\n/**\n * Interface for the input parameters of the RetrievalQAChain class.\n */\nexport interface RetrievalQAChainInput extends Omit<ChainInputs, \"memory\"> {\n  retriever: BaseRetrieverInterface;\n  combineDocumentsChain: BaseChain;\n  inputKey?: string;\n  returnSourceDocuments?: boolean;\n}\n\n/**\n * Class representing a chain for performing question-answering tasks with\n * a retrieval component.\n * @example\n * ```typescript\n * import { createStuffDocumentsChain } from \"@langchain/classic/chains/combine_documents\";\n * import { ChatPromptTemplate } from \"@langchain/core/prompts\";\n * import { createRetrievalChain } from \"@langchain/classic/chains/retrieval\";\n * import { MemoryVectorStore } from \"@langchain/classic/vectorstores/memory\";\n *\n * const documents = [...your documents here];\n * const embeddings = ...your embeddings model;\n * const llm = ...your LLM model;\n *\n * const vectorstore = await MemoryVectorStore.fromDocuments(\n *   documents,\n *   embeddings\n * );\n * const prompt = ChatPromptTemplate.fromTemplate(`Answer the user's question: {input} based on the following context {context}`);\n *\n * const combineDocsChain = await createStuffDocumentsChain({\n *   llm,\n *   prompt,\n * });\n * const retriever = vectorstore.asRetriever();\n *\n * const retrievalChain = await createRetrievalChain({\n *   combineDocsChain,\n *   retriever,\n * });\n * ```\n */\nexport class RetrievalQAChain\n  extends BaseChain\n  implements RetrievalQAChainInput\n{\n  static lc_name() {\n    return \"RetrievalQAChain\";\n  }\n\n  inputKey = \"query\";\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentsChain.outputKeys.concat(\n      this.returnSourceDocuments ? [\"sourceDocuments\"] : []\n    );\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: RetrievalQAChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key \"${this.inputKey}\" not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const docs = await this.retriever.invoke(\n      question,\n      runManager?.getChild(\"retriever\")\n    );\n    const inputs = { question, input_documents: docs, ...values };\n    const result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      return {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"retrieval_qa\" as const;\n  }\n\n  static async deserialize(\n    _data: SerializedVectorDBQAChain,\n    _values: LoadValues\n  ): Promise<RetrievalQAChain> {\n    throw new Error(\"Not implemented\");\n  }\n\n  serialize(): SerializedVectorDBQAChain {\n    throw new Error(\"Not implemented\");\n  }\n\n  /**\n   * Creates a new instance of RetrievalQAChain using a BaseLanguageModel\n   * and a BaseRetriever.\n   * @param llm The BaseLanguageModel used to generate a new question.\n   * @param retriever The BaseRetriever used to retrieve relevant documents.\n   * @param options Optional parameters for the RetrievalQAChain.\n   * @returns A new instance of RetrievalQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<\n        RetrievalQAChainInput,\n        \"retriever\" | \"combineDocumentsChain\" | \"index\"\n      >\n    > &\n      StuffQAChainParams\n  ): RetrievalQAChain {\n    const qaChain = loadQAStuffChain(llm, {\n      prompt: options?.prompt,\n    });\n    return new this({\n      ...options,\n      retriever,\n      combineDocumentsChain: qaChain,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDA,IAAa,mBAAb,cACU,UAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED,WAAW;CAEX,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,QAAS;CACvB;CAED,IAAI,aAAa;AACf,SAAO,KAAK,sBAAsB,WAAW,OAC3C,KAAK,wBAAwB,CAAC,iBAAkB,IAAG,CAAE,EACtD;CACF;CAED;CAEA;CAEA,wBAAwB;CAExB,YAAYA,QAA+B;EACzC,MAAM,OAAO;EACb,KAAK,YAAY,OAAO;EACxB,KAAK,wBAAwB,OAAO;EACpC,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,wBACH,OAAO,yBAAyB,KAAK;CACxC;;CAGD,MAAM,MACJC,QACAC,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,CAAC,cAAc,EAAE,KAAK,SAAS,YAAY,CAAC;EAE9D,MAAMC,WAAmB,OAAO,KAAK;EACrC,MAAM,OAAO,MAAM,KAAK,UAAU,OAChC,UACA,YAAY,SAAS,YAAY,CAClC;EACD,MAAM,SAAS;GAAE;GAAU,iBAAiB;GAAM,GAAG;EAAQ;EAC7D,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,sBACP,QAAO;GACL,GAAG;GACH,iBAAiB;EAClB;AAEH,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,aAAa,YACXC,OACAC,SAC2B;AAC3B,QAAM,IAAI,MAAM;CACjB;CAED,YAAuC;AACrC,QAAM,IAAI,MAAM;CACjB;;;;;;;;;CAUD,OAAO,QACLC,KACAC,WACAC,SAOkB;EAClB,MAAM,UAAU,iBAAiB,KAAK,EACpC,QAAQ,SAAS,OAClB,EAAC;AACF,SAAO,IAAI,KAAK;GACd,GAAG;GACH;GACA,uBAAuB;EACxB;CACF;AACF"}

package/dist/chains/router/llm_router.cjs

@@ -0,0 +1,50 @@
+const require_llm_chain = require('../llm_chain.cjs');
+const require_multi_route = require('./multi_route.cjs');
+
+//#region src/chains/router/llm_router.ts
+/**
+* A class that represents an LLM router chain in the LangChain framework.
+* It extends the RouterChain class and implements the LLMRouterChainInput
+* interface. It provides additional functionality specific to LLMs and
+* routing based on LLM predictions.
+*/
+var LLMRouterChain = class LLMRouterChain extends require_multi_route.RouterChain {
+	llmChain;
+	constructor(fields) {
+		super(fields);
+		this.llmChain = fields.llmChain;
+	}
+	get inputKeys() {
+		return this.llmChain.inputKeys;
+	}
+	async _call(values, runManager) {
+		return this.llmChain.predict(values, runManager?.getChild());
+	}
+	_chainType() {
+		return "llm_router_chain";
+	}
+	/**
+	* A static method that creates an instance of LLMRouterChain from a
+	* BaseLanguageModel and a BasePromptTemplate. It takes in an optional
+	* options object and returns an instance of LLMRouterChain with the
+	* specified LLMChain.
+	* @param llm A BaseLanguageModel instance.
+	* @param prompt A BasePromptTemplate instance.
+	* @param options Optional LLMRouterChainInput object, excluding "llmChain".
+	* @returns An instance of LLMRouterChain.
+	*/
+	static fromLLM(llm, prompt, options) {
+		const llmChain = new require_llm_chain.LLMChain({
+			llm,
+			prompt
+		});
+		return new LLMRouterChain({
+			...options,
+			llmChain
+		});
+	}
+};
+
+//#endregion
+exports.LLMRouterChain = LLMRouterChain;
+//# sourceMappingURL=llm_router.cjs.map

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

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

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

@@ -0,0 +1,54 @@
+import { ChainInputs } from "../base.cjs";
+import { LLMChain } from "../llm_chain.cjs";
+import { RouterChain } from "./multi_route.cjs";
+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
+ * the structure of the output data returned by the router chain.
+ */
+type RouterOutputSchema = {
+  destination: string;
+  next_inputs: {
+    [key: string]: string;
+  };
+};
+/**
+ * An interface that extends the default ChainInputs interface and adds an
+ * additional "llmChain" property.
+ */
+interface LLMRouterChainInput extends ChainInputs {
+  llmChain: LLMChain<RouterOutputSchema>;
+}
+/**
+ * A class that represents an LLM router chain in the LangChain framework.
+ * It extends the RouterChain class and implements the LLMRouterChainInput
+ * interface. It provides additional functionality specific to LLMs and
+ * routing based on LLM predictions.
+ */
+declare class LLMRouterChain extends RouterChain implements LLMRouterChainInput {
+  llmChain: LLMChain<RouterOutputSchema>;
+  constructor(fields: LLMRouterChainInput);
+  get inputKeys(): string[];
+  _call(values: ChainValues, runManager?: CallbackManagerForChainRun | undefined): Promise<RouterOutputSchema>;
+  _chainType(): string;
+  /**
+   * A static method that creates an instance of LLMRouterChain from a
+   * BaseLanguageModel and a BasePromptTemplate. It takes in an optional
+   * options object and returns an instance of LLMRouterChain with the
+   * specified LLMChain.
+   * @param llm A BaseLanguageModel instance.
+   * @param prompt A BasePromptTemplate instance.
+   * @param options Optional LLMRouterChainInput object, excluding "llmChain".
+   * @returns An instance of LLMRouterChain.
+   */
+  static fromLLM(llm: BaseLanguageModelInterface, prompt: BasePromptTemplate, options?: Omit<LLMRouterChainInput, "llmChain">): LLMRouterChain;
+}
+//#endregion
+export { LLMRouterChain, LLMRouterChainInput, RouterOutputSchema };
+//# sourceMappingURL=llm_router.d.cts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"llm_router.d.cts","names":["BaseLanguageModelInterface","ChainValues","BasePromptTemplate","CallbackManagerForChainRun","LLMChain","RouterChain","ChainInputs","RouterOutputSchema","LLMRouterChainInput","LLMRouterChain","Promise","Omit"],"sources":["../../../src/chains/router/llm_router.d.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 * 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: {\n        [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 * 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 declare class LLMRouterChain extends RouterChain implements LLMRouterChainInput {\n    llmChain: LLMChain<RouterOutputSchema>;\n    constructor(fields: LLMRouterChainInput);\n    get inputKeys(): string[];\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun | undefined): Promise<RouterOutputSchema>;\n    _chainType(): string;\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(llm: BaseLanguageModelInterface, prompt: BasePromptTemplate, options?: Omit<LLMRouterChainInput, \"llmChain\">): LLMRouterChain;\n}\n"],"mappings":";;;;;;;;;;;;AAWA;AAUA;AAAoC,KAVxBO,kBAAAA,GAUwB;EAAA,WACbA,EAAAA,MAAAA;EAAkB,WAA3BH,EAAAA;IAD+BE,CAAAA,GAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA;EAAW,CAAA;AASxD,CAAA;;;;;AAIkBL,UAbDO,mBAAAA,SAA4BF,WAa3BL,CAAAA;EAAW,QAAeE,EAZ9BC,QAY8BD,CAZrBI,kBAYqBJ,CAAAA;;;;;;;;AAJAE,cAAvBI,cAAAA,SAAuBJ,WAAAA,YAAuBG,mBAAvBH,CAAAA;EAAW,QAAYG,EACrDJ,QADqDI,CAC5CD,kBAD4CC,CAAAA;EAAmB,WAAA,CAAA,MAAA,EAE9DA,mBAF8D;;gBAIpEP,0BAA0BE,yCAAyCO,QAAQH;;;;;;;;;;;;sBAYrEP,oCAAoCE,8BAA8BS,KAAKH,mCAAmCC"}

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

@@ -0,0 +1,54 @@
+import { ChainInputs } from "../base.js";
+import { LLMChain } from "../llm_chain.js";
+import { RouterChain } from "./multi_route.js";
+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
+ * the structure of the output data returned by the router chain.
+ */
+type RouterOutputSchema = {
+  destination: string;
+  next_inputs: {
+    [key: string]: string;
+  };
+};
+/**
+ * An interface that extends the default ChainInputs interface and adds an
+ * additional "llmChain" property.
+ */
+interface LLMRouterChainInput extends ChainInputs {
+  llmChain: LLMChain<RouterOutputSchema>;
+}
+/**
+ * A class that represents an LLM router chain in the LangChain framework.
+ * It extends the RouterChain class and implements the LLMRouterChainInput
+ * interface. It provides additional functionality specific to LLMs and
+ * routing based on LLM predictions.
+ */
+declare class LLMRouterChain extends RouterChain implements LLMRouterChainInput {
+  llmChain: LLMChain<RouterOutputSchema>;
+  constructor(fields: LLMRouterChainInput);
+  get inputKeys(): string[];
+  _call(values: ChainValues, runManager?: CallbackManagerForChainRun | undefined): Promise<RouterOutputSchema>;
+  _chainType(): string;
+  /**
+   * A static method that creates an instance of LLMRouterChain from a
+   * BaseLanguageModel and a BasePromptTemplate. It takes in an optional
+   * options object and returns an instance of LLMRouterChain with the
+   * specified LLMChain.
+   * @param llm A BaseLanguageModel instance.
+   * @param prompt A BasePromptTemplate instance.
+   * @param options Optional LLMRouterChainInput object, excluding "llmChain".
+   * @returns An instance of LLMRouterChain.
+   */
+  static fromLLM(llm: BaseLanguageModelInterface, prompt: BasePromptTemplate, options?: Omit<LLMRouterChainInput, "llmChain">): LLMRouterChain;
+}
+//#endregion
+export { LLMRouterChain, LLMRouterChainInput, RouterOutputSchema };
+//# sourceMappingURL=llm_router.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"llm_router.d.ts","names":["BaseLanguageModelInterface","ChainValues","BasePromptTemplate","CallbackManagerForChainRun","LLMChain","RouterChain","ChainInputs","RouterOutputSchema","LLMRouterChainInput","LLMRouterChain","Promise","Omit"],"sources":["../../../src/chains/router/llm_router.d.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 * 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: {\n        [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 * 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 declare class LLMRouterChain extends RouterChain implements LLMRouterChainInput {\n    llmChain: LLMChain<RouterOutputSchema>;\n    constructor(fields: LLMRouterChainInput);\n    get inputKeys(): string[];\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun | undefined): Promise<RouterOutputSchema>;\n    _chainType(): string;\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(llm: BaseLanguageModelInterface, prompt: BasePromptTemplate, options?: Omit<LLMRouterChainInput, \"llmChain\">): LLMRouterChain;\n}\n"],"mappings":";;;;;;;;;;;;AAWA;AAUA;AAAoC,KAVxBO,kBAAAA,GAUwB;EAAA,WACbA,EAAAA,MAAAA;EAAkB,WAA3BH,EAAAA;IAD+BE,CAAAA,GAAAA,EAAAA,MAAAA,CAAAA,EAAAA,MAAAA;EAAW,CAAA;AASxD,CAAA;;;;;AAIkBL,UAbDO,mBAAAA,SAA4BF,WAa3BL,CAAAA;EAAW,QAAeE,EAZ9BC,QAY8BD,CAZrBI,kBAYqBJ,CAAAA;;;;;;;;AAJAE,cAAvBI,cAAAA,SAAuBJ,WAAAA,YAAuBG,mBAAvBH,CAAAA;EAAW,QAAYG,EACrDJ,QADqDI,CAC5CD,kBAD4CC,CAAAA;EAAmB,WAAA,CAAA,MAAA,EAE9DA,mBAF8D;;gBAIpEP,0BAA0BE,yCAAyCO,QAAQH;;;;;;;;;;;;sBAYrEP,oCAAoCE,8BAA8BS,KAAKH,mCAAmCC"}

package/dist/chains/router/llm_router.js

@@ -0,0 +1,50 @@
+import { LLMChain } from "../llm_chain.js";
+import { RouterChain } from "./multi_route.js";
+
+//#region src/chains/router/llm_router.ts
+/**
+* A class that represents an LLM router chain in the LangChain framework.
+* It extends the RouterChain class and implements the LLMRouterChainInput
+* interface. It provides additional functionality specific to LLMs and
+* routing based on LLM predictions.
+*/
+var LLMRouterChain = class LLMRouterChain extends RouterChain {
+	llmChain;
+	constructor(fields) {
+		super(fields);
+		this.llmChain = fields.llmChain;
+	}
+	get inputKeys() {
+		return this.llmChain.inputKeys;
+	}
+	async _call(values, runManager) {
+		return this.llmChain.predict(values, runManager?.getChild());
+	}
+	_chainType() {
+		return "llm_router_chain";
+	}
+	/**
+	* A static method that creates an instance of LLMRouterChain from a
+	* BaseLanguageModel and a BasePromptTemplate. It takes in an optional
+	* options object and returns an instance of LLMRouterChain with the
+	* specified LLMChain.
+	* @param llm A BaseLanguageModel instance.
+	* @param prompt A BasePromptTemplate instance.
+	* @param options Optional LLMRouterChainInput object, excluding "llmChain".
+	* @returns An instance of LLMRouterChain.
+	*/
+	static fromLLM(llm, prompt, options) {
+		const llmChain = new LLMChain({
+			llm,
+			prompt
+		});
+		return new LLMRouterChain({
+			...options,
+			llmChain
+		});
+	}
+};
+
+//#endregion
+export { LLMRouterChain };
+//# sourceMappingURL=llm_router.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"llm_router.js","names":["fields: LLMRouterChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun | undefined","llm: BaseLanguageModelInterface","prompt: BasePromptTemplate","options?: Omit<LLMRouterChainInput, \"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,uBAAuB,YAA2C;CAC7E;CAEA,YAAYA,QAA6B;EACvC,MAAM,OAAO;EACb,KAAK,WAAW,OAAO;CACxB;CAED,IAAI,YAAsB;AACxB,SAAO,KAAK,SAAS;CACtB;CAED,MAAM,MACJC,QACAC,YAC6B;AAC7B,SAAO,KAAK,SAAS,QAAQ,QAAQ,YAAY,UAAU,CAAC;CAC7D;CAED,aAAqB;AACnB,SAAO;CACR;;;;;;;;;;;CAYD,OAAO,QACLC,KACAC,QACAC,SACA;EACA,MAAM,WAAW,IAAI,SAA6B;GAAE;GAAK;EAAQ;AACjE,SAAO,IAAI,eAAe;GAAE,GAAG;GAAS;EAAU;CACnD;AACF"}