langchain 0.0.71 → 0.0.73

package/dist/agents/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AgentActionOutputParser = exports.ZeroShotAgentOutputParser = exports.ZeroShotAgent = exports.initializeAgentExecutorWithOptions = exports.initializeAgentExecutor = exports.AgentExecutor = exports.ChatConversationalAgentOutputParser = exports.ChatConversationalAgent = exports.ChatAgentOutputParser = exports.ChatAgent = exports.Toolkit = exports.createVectorStoreAgent = exports.createSqlAgent = exports.createOpenApiAgent = exports.createJsonAgent = exports.ZapierToolKit = exports.VectorStoreToolkit = exports.VectorStoreRouterToolkit = exports.SqlToolkit = exports.RequestsToolkit = exports.OpenApiToolkit = exports.JsonToolkit = exports.LLMSingleActionAgent = exports.BaseSingleActionAgent = exports.Agent = void 0;
+exports.AgentActionOutputParser = exports.ZeroShotAgentOutputParser = exports.ZeroShotAgent = exports.initializeAgentExecutorWithOptions = exports.initializeAgentExecutor = exports.AgentExecutor = exports.ChatConversationalAgentOutputParser = exports.ChatConversationalAgent = exports.ChatAgentOutputParser = exports.ChatAgent = exports.Toolkit = exports.createVectorStoreRouterAgent = exports.createVectorStoreAgent = exports.createSqlAgent = exports.createOpenApiAgent = exports.createJsonAgent = exports.ZapierToolKit = exports.VectorStoreToolkit = exports.VectorStoreRouterToolkit = exports.SqlToolkit = exports.RequestsToolkit = exports.OpenApiToolkit = exports.JsonToolkit = exports.LLMSingleActionAgent = exports.BaseSingleActionAgent = exports.Agent = void 0;
 var agent_js_1 = require("./agent.cjs");
 Object.defineProperty(exports, "Agent", { enumerable: true, get: function () { return agent_js_1.Agent; } });
 Object.defineProperty(exports, "BaseSingleActionAgent", { enumerable: true, get: function () { return agent_js_1.BaseSingleActionAgent; } });
@@ -17,6 +17,7 @@ Object.defineProperty(exports, "createJsonAgent", { enumerable: true, get: funct
 Object.defineProperty(exports, "createOpenApiAgent", { enumerable: true, get: function () { return index_js_1.createOpenApiAgent; } });
 Object.defineProperty(exports, "createSqlAgent", { enumerable: true, get: function () { return index_js_1.createSqlAgent; } });
 Object.defineProperty(exports, "createVectorStoreAgent", { enumerable: true, get: function () { return index_js_1.createVectorStoreAgent; } });
+Object.defineProperty(exports, "createVectorStoreRouterAgent", { enumerable: true, get: function () { return index_js_1.createVectorStoreRouterAgent; } });
 var base_js_1 = require("./agent_toolkits/base.cjs");
 Object.defineProperty(exports, "Toolkit", { enumerable: true, get: function () { return base_js_1.Toolkit; } });
 var index_js_2 = require("./chat/index.cjs");

package/dist/agents/index.d.ts

@@ -1,5 +1,5 @@
 export { Agent, AgentArgs, BaseSingleActionAgent, LLMSingleActionAgent, LLMSingleActionAgentInput, OutputParserArgs, } from "./agent.js";
-export { JsonToolkit, OpenApiToolkit, RequestsToolkit, SqlToolkit, VectorStoreInfo, VectorStoreRouterToolkit, VectorStoreToolkit, ZapierToolKit, createJsonAgent, createOpenApiAgent, createSqlAgent, SqlCreatePromptArgs, createVectorStoreAgent, } from "./agent_toolkits/index.js";
+export { JsonToolkit, OpenApiToolkit, RequestsToolkit, SqlToolkit, VectorStoreInfo, VectorStoreRouterToolkit, VectorStoreToolkit, ZapierToolKit, createJsonAgent, createOpenApiAgent, createSqlAgent, SqlCreatePromptArgs, createVectorStoreAgent, createVectorStoreRouterAgent, } from "./agent_toolkits/index.js";
 export { Toolkit } from "./agent_toolkits/base.js";
 export { ChatAgent, ChatAgentInput, ChatCreatePromptArgs, } from "./chat/index.js";
 export { ChatAgentOutputParser } from "./chat/outputParser.js";

package/dist/agents/index.js

@@ -1,5 +1,5 @@
 export { Agent, BaseSingleActionAgent, LLMSingleActionAgent, } from "./agent.js";
-export { JsonToolkit, OpenApiToolkit, RequestsToolkit, SqlToolkit, VectorStoreRouterToolkit, VectorStoreToolkit, ZapierToolKit, createJsonAgent, createOpenApiAgent, createSqlAgent, createVectorStoreAgent, } from "./agent_toolkits/index.js";
+export { JsonToolkit, OpenApiToolkit, RequestsToolkit, SqlToolkit, VectorStoreRouterToolkit, VectorStoreToolkit, ZapierToolKit, createJsonAgent, createOpenApiAgent, createSqlAgent, createVectorStoreAgent, createVectorStoreRouterAgent, } from "./agent_toolkits/index.js";
 export { Toolkit } from "./agent_toolkits/base.js";
 export { ChatAgent, } from "./chat/index.js";
 export { ChatAgentOutputParser } from "./chat/outputParser.js";

package/dist/chains/base.cjs

@@ -71,10 +71,10 @@ class BaseChain extends index_js_2.BaseLangChain {
             await runManager?.handleChainError(e);
             throw e;
         }
-        await runManager?.handleChainEnd(outputValues);
         if (!(this.memory == null)) {
             await this.memory.saveContext(values, outputValues);
         }
+        await runManager?.handleChainEnd(outputValues);
         // add the runManager's currentRunId to the outputValues
         Object.defineProperty(outputValues, index_js_1.RUN_KEY, {
             value: runManager ? { runId: runManager?.runId } : undefined,

package/dist/chains/base.js

@@ -68,10 +68,10 @@ export class BaseChain extends BaseLangChain {
             await runManager?.handleChainError(e);
             throw e;
         }
-        await runManager?.handleChainEnd(outputValues);
         if (!(this.memory == null)) {
             await this.memory.saveContext(values, outputValues);
         }
+        await runManager?.handleChainEnd(outputValues);
         // add the runManager's currentRunId to the outputValues
         Object.defineProperty(outputValues, RUN_KEY, {
             value: runManager ? { runId: runManager?.runId } : undefined,

package/dist/chains/conversation.cjs

@@ -1,10 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ConversationChain = void 0;
+exports.ConversationChain = exports.DEFAULT_TEMPLATE = void 0;
 const llm_chain_js_1 = require("./llm_chain.cjs");
 const prompt_js_1 = require("../prompts/prompt.cjs");
 const buffer_memory_js_1 = require("../memory/buffer_memory.cjs");
-const defaultTemplate = `The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.
+exports.DEFAULT_TEMPLATE = `The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.
 
 Current conversation:
 {history}
@@ -15,7 +15,7 @@ class ConversationChain extends llm_chain_js_1.LLMChain {
         super({
             prompt: prompt ??
                 new prompt_js_1.PromptTemplate({
-                    template: defaultTemplate,
+                    template: exports.DEFAULT_TEMPLATE,
                     inputVariables: ["history", "input"],
                 }),
             outputKey: outputKey ?? "response",

package/dist/chains/conversation.d.ts

@@ -1,5 +1,6 @@
 import { LLMChain, LLMChainInput } from "./llm_chain.js";
 import { Optional } from "../types/type-utils.js";
+export declare const DEFAULT_TEMPLATE = "The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.\n\nCurrent conversation:\n{history}\nHuman: {input}\nAI:";
 export declare class ConversationChain extends LLMChain {
     constructor({ prompt, outputKey, memory, ...rest }: Optional<LLMChainInput, "prompt">);
 }

package/dist/chains/conversation.js

@@ -1,7 +1,7 @@
 import { LLMChain } from "./llm_chain.js";
 import { PromptTemplate } from "../prompts/prompt.js";
 import { BufferMemory } from "../memory/buffer_memory.js";
-const defaultTemplate = `The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.
+export const DEFAULT_TEMPLATE = `The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.
 
 Current conversation:
 {history}
@@ -12,7 +12,7 @@ export class ConversationChain extends LLMChain {
         super({
             prompt: prompt ??
                 new PromptTemplate({
-                    template: defaultTemplate,
+                    template: DEFAULT_TEMPLATE,
                     inputVariables: ["history", "input"],
                 }),
             outputKey: outputKey ?? "response",

package/dist/chains/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.OpenAIModerationChain = exports.PRINCIPLES = exports.ConstitutionalPrinciple = exports.ConstitutionalChain = exports.RetrievalQAChain = exports.ConversationalRetrievalQAChain = exports.SqlDatabaseChain = exports.loadSummarizationChain = exports.loadQARefineChain = exports.loadQAMapReduceChain = exports.loadQAStuffChain = exports.loadQAChain = exports.VectorDBQAChain = exports.AnalyzeDocumentChain = exports.ChatVectorDBQAChain = exports.RefineDocumentsChain = exports.MapReduceDocumentsChain = exports.StuffDocumentsChain = exports.SimpleSequentialChain = exports.SequentialChain = exports.ConversationChain = exports.LLMChain = exports.BaseChain = void 0;
+exports.MultiRetrievalQAChain = exports.MultiPromptChain = exports.LLMRouterChain = exports.RouterChain = exports.MultiRouteChain = exports.OpenAIModerationChain = exports.PRINCIPLES = exports.ConstitutionalPrinciple = exports.ConstitutionalChain = exports.RetrievalQAChain = exports.ConversationalRetrievalQAChain = exports.SqlDatabaseChain = exports.loadSummarizationChain = exports.loadQARefineChain = exports.loadQAMapReduceChain = exports.loadQAStuffChain = exports.loadQAChain = exports.VectorDBQAChain = exports.AnalyzeDocumentChain = exports.ChatVectorDBQAChain = exports.RefineDocumentsChain = exports.MapReduceDocumentsChain = exports.StuffDocumentsChain = exports.SimpleSequentialChain = exports.SequentialChain = exports.ConversationChain = exports.LLMChain = exports.BaseChain = void 0;
 var base_js_1 = require("./base.cjs");
 Object.defineProperty(exports, "BaseChain", { enumerable: true, get: function () { return base_js_1.BaseChain; } });
 var llm_chain_js_1 = require("./llm_chain.cjs");
@@ -40,3 +40,12 @@ Object.defineProperty(exports, "ConstitutionalPrinciple", { enumerable: true, ge
 Object.defineProperty(exports, "PRINCIPLES", { enumerable: true, get: function () { return constitutional_principle_js_1.PRINCIPLES; } });
 var openai_moderation_js_1 = require("./openai_moderation.cjs");
 Object.defineProperty(exports, "OpenAIModerationChain", { enumerable: true, get: function () { return openai_moderation_js_1.OpenAIModerationChain; } });
+var multi_route_js_1 = require("./router/multi_route.cjs");
+Object.defineProperty(exports, "MultiRouteChain", { enumerable: true, get: function () { return multi_route_js_1.MultiRouteChain; } });
+Object.defineProperty(exports, "RouterChain", { enumerable: true, get: function () { return multi_route_js_1.RouterChain; } });
+var llm_router_js_1 = require("./router/llm_router.cjs");
+Object.defineProperty(exports, "LLMRouterChain", { enumerable: true, get: function () { return llm_router_js_1.LLMRouterChain; } });
+var multi_prompt_js_1 = require("./router/multi_prompt.cjs");
+Object.defineProperty(exports, "MultiPromptChain", { enumerable: true, get: function () { return multi_prompt_js_1.MultiPromptChain; } });
+var multi_retrieval_qa_js_1 = require("./router/multi_retrieval_qa.cjs");
+Object.defineProperty(exports, "MultiRetrievalQAChain", { enumerable: true, get: function () { return multi_retrieval_qa_js_1.MultiRetrievalQAChain; } });

package/dist/chains/index.d.ts

@@ -15,3 +15,7 @@ export { ConstitutionalChainInput, ConstitutionalChain, } from "./constitutional
 export { ConstitutionalPrinciple, PRINCIPLES, } from "./constitutional_ai/constitutional_principle.js";
 export { SerializedLLMChain, SerializedSequentialChain, SerializedSimpleSequentialChain, SerializedSqlDatabaseChain, SerializedAnalyzeDocumentChain, SerializedBaseChain, SerializedChatVectorDBQAChain, SerializedMapReduceDocumentsChain, SerializedStuffDocumentsChain, SerializedVectorDBQAChain, SerializedRefineDocumentsChain, } from "./serde.js";
 export { OpenAIModerationChain } from "./openai_moderation.js";
+export { MultiRouteChain, MultiRouteChainInput, RouterChain, } from "./router/multi_route.js";
+export { LLMRouterChain, LLMRouterChainInput, RouterOutputSchema, } from "./router/llm_router.js";
+export { MultiPromptChain } from "./router/multi_prompt.js";
+export { MultiRetrievalQAChain } from "./router/multi_retrieval_qa.js";

package/dist/chains/index.js

@@ -14,3 +14,7 @@ export { RetrievalQAChain } from "./retrieval_qa.js";
 export { ConstitutionalChain, } from "./constitutional_ai/constitutional_chain.js";
 export { ConstitutionalPrinciple, PRINCIPLES, } from "./constitutional_ai/constitutional_principle.js";
 export { OpenAIModerationChain } from "./openai_moderation.js";
+export { MultiRouteChain, RouterChain, } from "./router/multi_route.js";
+export { LLMRouterChain, } from "./router/llm_router.js";
+export { MultiPromptChain } from "./router/multi_prompt.js";
+export { MultiRetrievalQAChain } from "./router/multi_retrieval_qa.js";

package/dist/chains/retrieval_qa.cjs

@@ -69,7 +69,9 @@ class RetrievalQAChain extends base_js_1.BaseChain {
         throw new Error("Not implemented");
     }
     static fromLLM(llm, retriever, options) {
-        const qaChain = (0, load_js_1.loadQAStuffChain)(llm);
+        const qaChain = (0, load_js_1.loadQAStuffChain)(llm, {
+            prompt: options?.prompt,
+        });
         return new this({
             retriever,
             combineDocumentsChain: qaChain,

package/dist/chains/retrieval_qa.d.ts

@@ -2,6 +2,7 @@ import { BaseChain, ChainInputs } from "./base.js";
 import { BaseLanguageModel } from "../base_language/index.js";
 import { SerializedVectorDBQAChain } from "./serde.js";
 import { ChainValues, BaseRetriever } from "../schema/index.js";
+import { StuffQAChainParams } from "./question_answering/load.js";
 import { CallbackManagerForChainRun } from "../callbacks/manager.js";
 export type LoadValues = Record<string, any>;
 export interface RetrievalQAChainInput extends Omit<ChainInputs, "memory"> {
@@ -23,5 +24,5 @@ export declare class RetrievalQAChain extends BaseChain implements RetrievalQACh
     _chainType(): "retrieval_qa";
     static deserialize(_data: SerializedVectorDBQAChain, _values: LoadValues): Promise<RetrievalQAChain>;
     serialize(): SerializedVectorDBQAChain;
-    static fromLLM(llm: BaseLanguageModel, retriever: BaseRetriever, options?: Partial<Omit<RetrievalQAChainInput, "combineDocumentsChain" | "index">>): RetrievalQAChain;
+    static fromLLM(llm: BaseLanguageModel, retriever: BaseRetriever, options?: Partial<Omit<RetrievalQAChainInput, "combineDocumentsChain" | "index">> & StuffQAChainParams): RetrievalQAChain;
 }

package/dist/chains/retrieval_qa.js

@@ -1,5 +1,5 @@
 import { BaseChain } from "./base.js";
-import { loadQAStuffChain } from "./question_answering/load.js";
+import { loadQAStuffChain, } from "./question_answering/load.js";
 export class RetrievalQAChain extends BaseChain {
     get inputKeys() {
         return [this.inputKey];
@@ -66,7 +66,9 @@ export class RetrievalQAChain extends BaseChain {
         throw new Error("Not implemented");
     }
     static fromLLM(llm, retriever, options) {
-        const qaChain = loadQAStuffChain(llm);
+        const qaChain = loadQAStuffChain(llm, {
+            prompt: options?.prompt,
+        });
         return new this({
             retriever,
             combineDocumentsChain: qaChain,

package/dist/chains/router/llm_router.cjs

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LLMRouterChain = void 0;
+const llm_chain_js_1 = require("../../chains/llm_chain.cjs");
+const multi_route_js_1 = require("./multi_route.cjs");
+class LLMRouterChain extends multi_route_js_1.RouterChain {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "llmChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        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";
+    }
+    static fromLLM(llm, prompt, options) {
+        const llmChain = new llm_chain_js_1.LLMChain({ llm, prompt });
+        return new LLMRouterChain({ ...options, llmChain });
+    }
+}
+exports.LLMRouterChain = LLMRouterChain;

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

@@ -0,0 +1,24 @@
+import { BasePromptTemplate } from "../../prompts/base.js";
+import { LLMChain } from "../../chains/llm_chain.js";
+import { RouterChain } from "./multi_route.js";
+import { CallbackManagerForChainRun } from "../../callbacks/manager.js";
+import { ChainValues } from "../../schema/index.js";
+import { BaseLanguageModel } from "../../base_language/index.js";
+import { ChainInputs } from "../../chains/base.js";
+export type RouterOutputSchema = {
+    destination: string;
+    next_inputs: {
+        [key: string]: string;
+    };
+};
+export interface LLMRouterChainInput extends ChainInputs {
+    llmChain: LLMChain<RouterOutputSchema>;
+}
+export 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;
+    static fromLLM(llm: BaseLanguageModel, prompt: BasePromptTemplate, options?: Omit<LLMRouterChainInput, "llmChain">): LLMRouterChain;
+}

package/dist/chains/router/llm_router.js

@@ -0,0 +1,27 @@
+import { LLMChain } from "../../chains/llm_chain.js";
+import { RouterChain } from "./multi_route.js";
+export class LLMRouterChain extends RouterChain {
+    constructor(fields) {
+        super(fields);
+        Object.defineProperty(this, "llmChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        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";
+    }
+    static fromLLM(llm, prompt, options) {
+        const llmChain = new LLMChain({ llm, prompt });
+        return new LLMRouterChain({ ...options, llmChain });
+    }
+}

package/dist/chains/router/multi_prompt.cjs

@@ -0,0 +1,76 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MultiPromptChain = void 0;
+const zod_1 = require("zod");
+const multi_route_js_1 = require("./multi_route.cjs");
+const multi_prompt_prompt_js_1 = require("./multi_prompt_prompt.cjs");
+const template_js_1 = require("../../prompts/template.cjs");
+const llm_chain_js_1 = require("../../chains/llm_chain.cjs");
+const prompt_js_1 = require("../../prompts/prompt.cjs");
+const llm_router_js_1 = require("./llm_router.cjs");
+const conversation_js_1 = require("../../chains/conversation.cjs");
+const utils_js_1 = require("./utils.cjs");
+const router_js_1 = require("../../output_parsers/router.cjs");
+class MultiPromptChain extends multi_route_js_1.MultiRouteChain {
+    static fromPrompts(llm, promptNames, promptDescriptions, promptTemplates, defaultChain, options) {
+        const destinations = (0, utils_js_1.zipEntries)(promptNames, promptDescriptions).map(([name, desc]) => `${name}: ${desc}`);
+        const structuredOutputParserSchema = zod_1.z.object({
+            destination: zod_1.z
+                .string()
+                .optional()
+                .describe('name of the question answering system to use or "DEFAULT"'),
+            next_inputs: zod_1.z
+                .object({
+                input: zod_1.z
+                    .string()
+                    .describe("a potentially modified version of the original input"),
+            })
+                .describe("input to be fed to the next model"),
+        });
+        const outputParser = new router_js_1.RouterOutputParser(structuredOutputParserSchema);
+        const destinationsStr = destinations.join("\n");
+        const routerTemplate = (0, template_js_1.interpolateFString)((0, multi_prompt_prompt_js_1.STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE)(outputParser.getFormatInstructions({ interpolationDepth: 4 })), {
+            destinations: destinationsStr,
+        });
+        const routerPrompt = new prompt_js_1.PromptTemplate({
+            template: routerTemplate,
+            inputVariables: ["input"],
+            outputParser,
+        });
+        const routerChain = llm_router_js_1.LLMRouterChain.fromLLM(llm, routerPrompt);
+        const destinationChains = (0, utils_js_1.zipEntries)(promptNames, promptTemplates).reduce((acc, [name, template]) => {
+            let myPrompt;
+            if (typeof template === "object") {
+                myPrompt = template;
+            }
+            else if (typeof template === "string") {
+                myPrompt = new prompt_js_1.PromptTemplate({
+                    template: template,
+                    inputVariables: ["input"],
+                });
+            }
+            else {
+                throw new Error("Invalid prompt template");
+            }
+            acc[name] = new llm_chain_js_1.LLMChain({
+                llm,
+                prompt: myPrompt,
+            });
+            return acc;
+        }, {});
+        const convChain = new conversation_js_1.ConversationChain({
+            llm,
+            outputKey: "text",
+        });
+        return new MultiPromptChain({
+            routerChain,
+            destinationChains,
+            defaultChain: defaultChain ?? convChain,
+            ...options,
+        });
+    }
+    _chainType() {
+        return "multi_prompt_chain";
+    }
+}
+exports.MultiPromptChain = MultiPromptChain;

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

@@ -0,0 +1,8 @@
+import { BaseLanguageModel } from "../../base_language/index.js";
+import { MultiRouteChain, MultiRouteChainInput } from "./multi_route.js";
+import { BaseChain } from "../../chains/base.js";
+import { PromptTemplate } from "../../prompts/prompt.js";
+export declare class MultiPromptChain extends MultiRouteChain {
+    static fromPrompts(llm: BaseLanguageModel, promptNames: string[], promptDescriptions: string[], promptTemplates: string[] | PromptTemplate[], defaultChain?: BaseChain, options?: Omit<MultiRouteChainInput, "defaultChain">): MultiPromptChain;
+    _chainType(): string;
+}

package/dist/chains/router/multi_prompt.js

@@ -0,0 +1,72 @@
+import { z } from "zod";
+import { MultiRouteChain } from "./multi_route.js";
+import { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE } from "./multi_prompt_prompt.js";
+import { interpolateFString } from "../../prompts/template.js";
+import { LLMChain } from "../../chains/llm_chain.js";
+import { PromptTemplate } from "../../prompts/prompt.js";
+import { LLMRouterChain } from "./llm_router.js";
+import { ConversationChain } from "../../chains/conversation.js";
+import { zipEntries } from "./utils.js";
+import { RouterOutputParser } from "../../output_parsers/router.js";
+export class MultiPromptChain extends MultiRouteChain {
+    static fromPrompts(llm, promptNames, promptDescriptions, promptTemplates, defaultChain, options) {
+        const destinations = zipEntries(promptNames, promptDescriptions).map(([name, desc]) => `${name}: ${desc}`);
+        const structuredOutputParserSchema = z.object({
+            destination: z
+                .string()
+                .optional()
+                .describe('name of the question answering system to use or "DEFAULT"'),
+            next_inputs: z
+                .object({
+                input: z
+                    .string()
+                    .describe("a potentially modified version of the original input"),
+            })
+                .describe("input to be fed to the next model"),
+        });
+        const outputParser = new RouterOutputParser(structuredOutputParserSchema);
+        const destinationsStr = destinations.join("\n");
+        const routerTemplate = interpolateFString(STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE(outputParser.getFormatInstructions({ interpolationDepth: 4 })), {
+            destinations: destinationsStr,
+        });
+        const routerPrompt = new PromptTemplate({
+            template: routerTemplate,
+            inputVariables: ["input"],
+            outputParser,
+        });
+        const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);
+        const destinationChains = zipEntries(promptNames, promptTemplates).reduce((acc, [name, template]) => {
+            let myPrompt;
+            if (typeof template === "object") {
+                myPrompt = template;
+            }
+            else if (typeof template === "string") {
+                myPrompt = new PromptTemplate({
+                    template: template,
+                    inputVariables: ["input"],
+                });
+            }
+            else {
+                throw new Error("Invalid prompt template");
+            }
+            acc[name] = new LLMChain({
+                llm,
+                prompt: myPrompt,
+            });
+            return acc;
+        }, {});
+        const convChain = new ConversationChain({
+            llm,
+            outputKey: "text",
+        });
+        return new MultiPromptChain({
+            routerChain,
+            destinationChains,
+            defaultChain: defaultChain ?? convChain,
+            ...options,
+        });
+    }
+    _chainType() {
+        return "multi_prompt_chain";
+    }
+}

package/dist/chains/router/multi_prompt_prompt.cjs

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE = exports.MULTI_PROMPT_ROUTER_TEMPLATE = void 0;
+exports.MULTI_PROMPT_ROUTER_TEMPLATE = `Given a raw text input to a language model, select the model prompt best suited for the input. You will be given the names of the available prompts and a description of what the prompt is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response from the language model.
+
+<< FORMATTING >>
+Return a markdown code snippet with a JSON object formatted to look like:
+\`\`\`json
+{{{{
+    "destination": string \\ name of the prompt to use or "DEFAULT"
+    "next_inputs": string \\ a potentially modified version of the original input
+}}}}
+\`\`\`
+
+REMEMBER: "destination" MUST be one of the candidate prompt names specified below OR it can be "DEFAULT" if the input is not well suited for any of the candidate prompts.
+REMEMBER: "next_inputs" can just be the original input if you don't think any modifications are needed.
+
+<< CANDIDATE PROMPTS >>
+{destinations}
+
+<< INPUT >>
+{{input}}
+
+<< OUTPUT >>
+`;
+const STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE = (formatting) => `Given a raw text input to a language model, select the model prompt best suited for the input. You will be given the names of the available prompts and a description of what the prompt is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response from the language model.
+
+<< FORMATTING >>
+${formatting}
+
+REMEMBER: "destination" MUST be one of the candidate prompt names specified below OR it can be "DEFAULT" if the input is not well suited for any of the candidate prompts.
+REMEMBER: "next_inputs.input" can just be the original input if you don't think any modifications are needed.
+
+<< CANDIDATE PROMPTS >>
+{destinations}
+
+<< INPUT >>
+{{input}}
+
+<< OUTPUT >>
+`;
+exports.STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE = STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE;

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

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

package/dist/chains/router/multi_prompt_prompt.js

@@ -0,0 +1,38 @@
+export const MULTI_PROMPT_ROUTER_TEMPLATE = `Given a raw text input to a language model, select the model prompt best suited for the input. You will be given the names of the available prompts and a description of what the prompt is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response from the language model.
+
+<< FORMATTING >>
+Return a markdown code snippet with a JSON object formatted to look like:
+\`\`\`json
+{{{{
+    "destination": string \\ name of the prompt to use or "DEFAULT"
+    "next_inputs": string \\ a potentially modified version of the original input
+}}}}
+\`\`\`
+
+REMEMBER: "destination" MUST be one of the candidate prompt names specified below OR it can be "DEFAULT" if the input is not well suited for any of the candidate prompts.
+REMEMBER: "next_inputs" can just be the original input if you don't think any modifications are needed.
+
+<< CANDIDATE PROMPTS >>
+{destinations}
+
+<< INPUT >>
+{{input}}
+
+<< OUTPUT >>
+`;
+export const STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE = (formatting) => `Given a raw text input to a language model, select the model prompt best suited for the input. You will be given the names of the available prompts and a description of what the prompt is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response from the language model.
+
+<< FORMATTING >>
+${formatting}
+
+REMEMBER: "destination" MUST be one of the candidate prompt names specified below OR it can be "DEFAULT" if the input is not well suited for any of the candidate prompts.
+REMEMBER: "next_inputs.input" can just be the original input if you don't think any modifications are needed.
+
+<< CANDIDATE PROMPTS >>
+{destinations}
+
+<< INPUT >>
+{{input}}
+
+<< OUTPUT >>
+`;

package/dist/chains/router/multi_retrieval_prompt.cjs

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE = exports.MULTI_RETRIEVAL_ROUTER_TEMPLATE = void 0;
+exports.MULTI_RETRIEVAL_ROUTER_TEMPLATE = `Given a query to a question answering system, select the system best suited for the input. You will be given the names of the available systems and a description of what questions the system is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response.
+
+<< FORMATTING >>
+Return a markdown code snippet with a JSON object formatted to look like:
+\`\`\`json
+{{{{
+    "destination": string \\ name of the question answering system to use or "DEFAULT"
+    "next_inputs": string \\ a potentially modified version of the original input
+}}}}
+\`\`\`
+
+REMEMBER: "destination" MUST be one of the candidate prompt names specified below OR it can be "DEFAULT" if the input is not well suited for any of the candidate prompts.
+REMEMBER: "next_inputs" can just be the original input if you don't think any modifications are needed.
+
+<< CANDIDATE PROMPTS >>
+{destinations}
+
+<< INPUT >>
+{{input}}
+
+<< OUTPUT >>
+`;
+const STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE = (formatting) => `Given a query to a question answering system, select the system best suited for the input. You will be given the names of the available systems and a description of what questions the system is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response.
+
+<< FORMATTING >>
+${formatting}
+
+REMEMBER: "destination" MUST be one of the candidate prompt names specified below OR it can be "DEFAULT" if the input is not well suited for any of the candidate prompts.
+REMEMBER: "next_inputs.query" can just be the original input if you don't think any modifications are needed.
+
+<< CANDIDATE PROMPTS >>
+{destinations}
+
+<< INPUT >>
+{{input}}
+
+<< OUTPUT >>
+`;
+exports.STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE = STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE;

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

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

package/dist/chains/router/multi_retrieval_prompt.js

@@ -0,0 +1,38 @@
+export const MULTI_RETRIEVAL_ROUTER_TEMPLATE = `Given a query to a question answering system, select the system best suited for the input. You will be given the names of the available systems and a description of what questions the system is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response.
+
+<< FORMATTING >>
+Return a markdown code snippet with a JSON object formatted to look like:
+\`\`\`json
+{{{{
+    "destination": string \\ name of the question answering system to use or "DEFAULT"
+    "next_inputs": string \\ a potentially modified version of the original input
+}}}}
+\`\`\`
+
+REMEMBER: "destination" MUST be one of the candidate prompt names specified below OR it can be "DEFAULT" if the input is not well suited for any of the candidate prompts.
+REMEMBER: "next_inputs" can just be the original input if you don't think any modifications are needed.
+
+<< CANDIDATE PROMPTS >>
+{destinations}
+
+<< INPUT >>
+{{input}}
+
+<< OUTPUT >>
+`;
+export const STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE = (formatting) => `Given a query to a question answering system, select the system best suited for the input. You will be given the names of the available systems and a description of what questions the system is best suited for. You may also revise the original input if you think that revising it will ultimately lead to a better response.
+
+<< FORMATTING >>
+${formatting}
+
+REMEMBER: "destination" MUST be one of the candidate prompt names specified below OR it can be "DEFAULT" if the input is not well suited for any of the candidate prompts.
+REMEMBER: "next_inputs.query" can just be the original input if you don't think any modifications are needed.
+
+<< CANDIDATE PROMPTS >>
+{destinations}
+
+<< INPUT >>
+{{input}}
+
+<< OUTPUT >>
+`;