langchain 0.0.74 → 0.0.76

package/dist/callbacks/manager.cjs

@@ -5,6 +5,7 @@ const uuid_1 = require("uuid");
 const base_js_1 = require("./base.cjs");
 const console_js_1 = require("./handlers/console.cjs");
 const initialize_js_1 = require("./handlers/initialize.cjs");
+const base_js_2 = require("../memory/base.cjs");
 class BaseCallbackManager {
     setHandler(handler) {
         return this.setHandlers([handler]);
@@ -209,11 +210,30 @@ class CallbackManager extends BaseCallbackManager {
         this.inheritableHandlers = [];
         this._parentRunId = parentRunId;
     }
-    async handleLLMStart(llm, prompts, runId = (0, uuid_1.v4)()) {
+    async handleLLMStart(llm, prompts, runId = (0, uuid_1.v4)(), _parentRunId = undefined, extraParams = undefined) {
         await Promise.all(this.handlers.map(async (handler) => {
             if (!handler.ignoreLLM) {
                 try {
-                    await handler.handleLLMStart?.(llm, prompts, runId, this._parentRunId);
+                    await handler.handleLLMStart?.(llm, prompts, runId, this._parentRunId, extraParams);
+                }
+                catch (err) {
+                    console.error(`Error in handler ${handler.constructor.name}, handleLLMStart: ${err}`);
+                }
+            }
+        }));
+        return new CallbackManagerForLLMRun(runId, this.handlers, this.inheritableHandlers, this._parentRunId);
+    }
+    async handleChatModelStart(llm, messages, runId = (0, uuid_1.v4)(), _parentRunId = undefined, extraParams = undefined) {
+        let messageStrings;
+        await Promise.all(this.handlers.map(async (handler) => {
+            if (!handler.ignoreLLM) {
+                try {
+                    if (handler.handleChatModelStart)
+                        await handler.handleChatModelStart?.(llm, messages, runId, this._parentRunId, extraParams);
+                    else if (handler.handleLLMStart) {
+                        messageStrings = messages.map((x) => (0, base_js_2.getBufferString)(x));
+                        await handler.handleLLMStart?.(llm, messageStrings, runId, this._parentRunId, extraParams);
+                    }
                 }
                 catch (err) {
                     console.error(`Error in handler ${handler.constructor.name}, handleLLMStart: ${err}`);
@@ -314,10 +334,15 @@ class CallbackManager extends BaseCallbackManager {
                 ? localHandlers.map(ensureHandler)
                 : localHandlers?.handlers, false);
         }
-        const tracingEnabled = typeof process !== "undefined"
+        const tracingV2Enabled = typeof process !== "undefined"
             ? // eslint-disable-next-line no-process-env
-                process.env?.LANGCHAIN_TRACING !== undefined
+                process.env?.LANGCHAIN_TRACING_V2 !== undefined
             : false;
+        const tracingEnabled = tracingV2Enabled ||
+            (typeof process !== "undefined"
+                ? // eslint-disable-next-line no-process-env
+                    process.env?.LANGCHAIN_TRACING !== undefined
+                : false);
         if (options?.verbose || tracingEnabled) {
             if (!callbackManager) {
                 callbackManager = new CallbackManager();
@@ -329,11 +354,16 @@ class CallbackManager extends BaseCallbackManager {
             }
             if (tracingEnabled &&
                 !callbackManager.handlers.some((handler) => handler.name === "langchain_tracer")) {
-                const session = typeof process !== "undefined"
-                    ? // eslint-disable-next-line no-process-env
-                        process.env?.LANGCHAIN_SESSION
-                    : undefined;
-                callbackManager.addHandler(await (0, initialize_js_1.getTracingCallbackHandler)(session), true);
+                if (tracingV2Enabled) {
+                    callbackManager.addHandler(await (0, initialize_js_1.getTracingV2CallbackHandler)(), true);
+                }
+                else {
+                    const session = typeof process !== "undefined"
+                        ? // eslint-disable-next-line no-process-env
+                            process.env?.LANGCHAIN_SESSION
+                        : undefined;
+                    callbackManager.addHandler(await (0, initialize_js_1.getTracingCallbackHandler)(session), true);
+                }
             }
         }
         return callbackManager;

package/dist/callbacks/manager.d.ts

@@ -1,4 +1,4 @@
-import { AgentAction, AgentFinish, ChainValues, LLMResult } from "../schema/index.js";
+import { AgentAction, AgentFinish, BaseChatMessage, ChainValues, LLMResult } from "../schema/index.js";
 import { BaseCallbackHandler, CallbackHandlerMethods } from "./base.js";
 type BaseCallbackManagerMethods = {
     [K in keyof CallbackHandlerMethods]?: (...args: Parameters<Required<CallbackHandlerMethods>[K]>) => Promise<unknown>;
@@ -47,7 +47,10 @@ export declare class CallbackManager extends BaseCallbackManager implements Base
     constructor(parentRunId?: string);
     handleLLMStart(llm: {
         name: string;
-    }, prompts: string[], runId?: string): Promise<CallbackManagerForLLMRun>;
+    }, prompts: string[], runId?: string, _parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined): Promise<CallbackManagerForLLMRun>;
+    handleChatModelStart(llm: {
+        name: string;
+    }, messages: BaseChatMessage[][], runId?: string, _parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined): Promise<CallbackManagerForLLMRun>;
     handleChainStart(chain: {
         name: string;
     }, inputs: ChainValues, runId?: string): Promise<CallbackManagerForChainRun>;

package/dist/callbacks/manager.js

@@ -1,7 +1,8 @@
 import { v4 as uuidv4 } from "uuid";
 import { BaseCallbackHandler } from "./base.js";
 import { ConsoleCallbackHandler } from "./handlers/console.js";
-import { getTracingCallbackHandler } from "./handlers/initialize.js";
+import { getTracingCallbackHandler, getTracingV2CallbackHandler, } from "./handlers/initialize.js";
+import { getBufferString } from "../memory/base.js";
 export class BaseCallbackManager {
     setHandler(handler) {
         return this.setHandlers([handler]);
@@ -202,11 +203,30 @@ export class CallbackManager extends BaseCallbackManager {
         this.inheritableHandlers = [];
         this._parentRunId = parentRunId;
     }
-    async handleLLMStart(llm, prompts, runId = uuidv4()) {
+    async handleLLMStart(llm, prompts, runId = uuidv4(), _parentRunId = undefined, extraParams = undefined) {
         await Promise.all(this.handlers.map(async (handler) => {
             if (!handler.ignoreLLM) {
                 try {
-                    await handler.handleLLMStart?.(llm, prompts, runId, this._parentRunId);
+                    await handler.handleLLMStart?.(llm, prompts, runId, this._parentRunId, extraParams);
+                }
+                catch (err) {
+                    console.error(`Error in handler ${handler.constructor.name}, handleLLMStart: ${err}`);
+                }
+            }
+        }));
+        return new CallbackManagerForLLMRun(runId, this.handlers, this.inheritableHandlers, this._parentRunId);
+    }
+    async handleChatModelStart(llm, messages, runId = uuidv4(), _parentRunId = undefined, extraParams = undefined) {
+        let messageStrings;
+        await Promise.all(this.handlers.map(async (handler) => {
+            if (!handler.ignoreLLM) {
+                try {
+                    if (handler.handleChatModelStart)
+                        await handler.handleChatModelStart?.(llm, messages, runId, this._parentRunId, extraParams);
+                    else if (handler.handleLLMStart) {
+                        messageStrings = messages.map((x) => getBufferString(x));
+                        await handler.handleLLMStart?.(llm, messageStrings, runId, this._parentRunId, extraParams);
+                    }
                 }
                 catch (err) {
                     console.error(`Error in handler ${handler.constructor.name}, handleLLMStart: ${err}`);
@@ -307,10 +327,15 @@ export class CallbackManager extends BaseCallbackManager {
                 ? localHandlers.map(ensureHandler)
                 : localHandlers?.handlers, false);
         }
-        const tracingEnabled = typeof process !== "undefined"
+        const tracingV2Enabled = typeof process !== "undefined"
             ? // eslint-disable-next-line no-process-env
-                process.env?.LANGCHAIN_TRACING !== undefined
+                process.env?.LANGCHAIN_TRACING_V2 !== undefined
             : false;
+        const tracingEnabled = tracingV2Enabled ||
+            (typeof process !== "undefined"
+                ? // eslint-disable-next-line no-process-env
+                    process.env?.LANGCHAIN_TRACING !== undefined
+                : false);
         if (options?.verbose || tracingEnabled) {
             if (!callbackManager) {
                 callbackManager = new CallbackManager();
@@ -322,11 +347,16 @@ export class CallbackManager extends BaseCallbackManager {
             }
             if (tracingEnabled &&
                 !callbackManager.handlers.some((handler) => handler.name === "langchain_tracer")) {
-                const session = typeof process !== "undefined"
-                    ? // eslint-disable-next-line no-process-env
-                        process.env?.LANGCHAIN_SESSION
-                    : undefined;
-                callbackManager.addHandler(await getTracingCallbackHandler(session), true);
+                if (tracingV2Enabled) {
+                    callbackManager.addHandler(await getTracingV2CallbackHandler(), true);
+                }
+                else {
+                    const session = typeof process !== "undefined"
+                        ? // eslint-disable-next-line no-process-env
+                            process.env?.LANGCHAIN_SESSION
+                        : undefined;
+                    callbackManager.addHandler(await getTracingCallbackHandler(session), true);
+                }
             }
         }
         return callbackManager;

package/dist/chains/llm_chain.cjs

@@ -73,14 +73,26 @@ class LLMChain extends base_js_1.BaseChain {
         }
         return finalCompletion;
     }
+    /**
+     * Run the core logic of this chain and add to output if desired.
+     *
+     * Wraps _call and handles memory.
+     */
+    call(values, callbacks) {
+        return super.call(values, callbacks);
+    }
     /** @ignore */
     async _call(values, runManager) {
-        let stop;
-        if ("stop" in values && Array.isArray(values.stop)) {
-            stop = values.stop;
+        const valuesForPrompt = { ...values };
+        const valuesForLLM = {};
+        for (const key of this.llm.callKeys) {
+            if (key in values) {
+                valuesForLLM[key] = values[key];
+                delete valuesForPrompt[key];
+            }
         }
-        const promptValue = await this.prompt.formatPromptValue(values);
-        const { generations } = await this.llm.generatePrompt([promptValue], stop, runManager?.getChild());
+        const promptValue = await this.prompt.formatPromptValue(valuesForPrompt);
+        const { generations } = await this.llm.generatePrompt([promptValue], valuesForLLM, runManager?.getChild());
         return {
             [this.outputKey]: await this._getFinalOutput(generations[0], promptValue, runManager),
         };

package/dist/chains/llm_chain.d.ts

@@ -5,7 +5,7 @@ import { ChainValues, Generation, BasePromptValue } from "../schema/index.js";
 import { BaseOutputParser } from "../schema/output_parser.js";
 import { SerializedLLMChain } from "./serde.js";
 import { CallbackManager } from "../callbacks/index.js";
-import { CallbackManagerForChainRun } from "../callbacks/manager.js";
+import { CallbackManagerForChainRun, Callbacks } from "../callbacks/manager.js";
 export interface LLMChainInput<T extends string | object = string> extends ChainInputs {
     /** Prompt object to use */
     prompt: BasePromptTemplate;
@@ -39,8 +39,14 @@ export declare class LLMChain<T extends string | object = string> extends BaseCh
     constructor(fields: LLMChainInput<T>);
     /** @ignore */
     _getFinalOutput(generations: Generation[], promptValue: BasePromptValue, runManager?: CallbackManagerForChainRun): Promise<unknown>;
+    /**
+     * Run the core logic of this chain and add to output if desired.
+     *
+     * Wraps _call and handles memory.
+     */
+    call(values: ChainValues & this["llm"]["CallOptions"], callbacks?: Callbacks | undefined): Promise<ChainValues>;
     /** @ignore */
-    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
+    _call(values: ChainValues & this["llm"]["CallOptions"], runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
     /**
      * Format prompt with values and pass to LLM
      *
@@ -53,7 +59,7 @@ export declare class LLMChain<T extends string | object = string> extends BaseCh
      * llm.predict({ adjective: "funny" })
      * ```
      */
-    predict(values: ChainValues, callbackManager?: CallbackManager): Promise<T>;
+    predict(values: ChainValues & this["llm"]["CallOptions"], callbackManager?: CallbackManager): Promise<T>;
     _chainType(): "llm_chain";
     static deserialize(data: SerializedLLMChain): Promise<LLMChain<string>>;
     serialize(): SerializedLLMChain;

package/dist/chains/llm_chain.js

@@ -70,14 +70,26 @@ export class LLMChain extends BaseChain {
         }
         return finalCompletion;
     }
+    /**
+     * Run the core logic of this chain and add to output if desired.
+     *
+     * Wraps _call and handles memory.
+     */
+    call(values, callbacks) {
+        return super.call(values, callbacks);
+    }
     /** @ignore */
     async _call(values, runManager) {
-        let stop;
-        if ("stop" in values && Array.isArray(values.stop)) {
-            stop = values.stop;
+        const valuesForPrompt = { ...values };
+        const valuesForLLM = {};
+        for (const key of this.llm.callKeys) {
+            if (key in values) {
+                valuesForLLM[key] = values[key];
+                delete valuesForPrompt[key];
+            }
         }
-        const promptValue = await this.prompt.formatPromptValue(values);
-        const { generations } = await this.llm.generatePrompt([promptValue], stop, runManager?.getChild());
+        const promptValue = await this.prompt.formatPromptValue(valuesForPrompt);
+        const { generations } = await this.llm.generatePrompt([promptValue], valuesForLLM, runManager?.getChild());
         return {
             [this.outputKey]: await this._getFinalOutput(generations[0], promptValue, runManager),
         };

package/dist/chains/question_answering/map_reduce_prompts.cjs

@@ -4,7 +4,7 @@ exports.COMBINE_PROMPT_SELECTOR = exports.COMBINE_PROMPT = exports.COMBINE_QA_PR
 /* eslint-disable spaced-comment */
 const prompt_js_1 = require("../../prompts/prompt.cjs");
 const chat_js_1 = require("../../prompts/chat.cjs");
-const prompt_selector_js_1 = require("../prompt_selector.cjs");
+const conditional_js_1 = require("../../prompts/selectors/conditional.cjs");
 const qa_template = `Use the following portion of a long document to see if any of the text is relevant to answer the question. 
 Return any relevant text verbatim.
 {context}
@@ -24,8 +24,8 @@ const messages = [
 const CHAT_QA_PROMPT = 
 /*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
 exports.COMBINE_QA_PROMPT_SELECTOR = 
-/*#__PURE__*/ new prompt_selector_js_1.ConditionalPromptSelector(exports.DEFAULT_COMBINE_QA_PROMPT, [
-    [prompt_selector_js_1.isChatModel, CHAT_QA_PROMPT],
+/*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_COMBINE_QA_PROMPT, [
+    [conditional_js_1.isChatModel, CHAT_QA_PROMPT],
 ]);
 const combine_prompt = `Given the following extracted parts of a long document and a question, create a final answer. 
 If you don't know the answer, just say that you don't know. Don't try to make up an answer.
@@ -70,6 +70,6 @@ const combine_messages = [
 const CHAT_COMBINE_PROMPT = 
 /*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromPromptMessages(combine_messages);
 exports.COMBINE_PROMPT_SELECTOR = 
-/*#__PURE__*/ new prompt_selector_js_1.ConditionalPromptSelector(exports.COMBINE_PROMPT, [
-    [prompt_selector_js_1.isChatModel, CHAT_COMBINE_PROMPT],
+/*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.COMBINE_PROMPT, [
+    [conditional_js_1.isChatModel, CHAT_COMBINE_PROMPT],
 ]);

package/dist/chains/question_answering/map_reduce_prompts.d.ts

@@ -1,5 +1,5 @@
 import { PromptTemplate } from "../../prompts/prompt.js";
-import { ConditionalPromptSelector } from "../prompt_selector.js";
+import { ConditionalPromptSelector } from "../../prompts/selectors/conditional.js";
 export declare const DEFAULT_COMBINE_QA_PROMPT: PromptTemplate;
 export declare const COMBINE_QA_PROMPT_SELECTOR: ConditionalPromptSelector;
 export declare const COMBINE_PROMPT: PromptTemplate;

package/dist/chains/question_answering/map_reduce_prompts.js

@@ -1,7 +1,7 @@
 /* eslint-disable spaced-comment */
 import { PromptTemplate } from "../../prompts/prompt.js";
 import { ChatPromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate, } from "../../prompts/chat.js";
-import { ConditionalPromptSelector, isChatModel } from "../prompt_selector.js";
+import { ConditionalPromptSelector, isChatModel, } from "../../prompts/selectors/conditional.js";
 const qa_template = `Use the following portion of a long document to see if any of the text is relevant to answer the question. 
 Return any relevant text verbatim.
 {context}

package/dist/chains/question_answering/refine_prompts.cjs

@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.QUESTION_PROMPT_SELECTOR = exports.CHAT_QUESTION_PROMPT = exports.DEFAULT_TEXT_QA_PROMPT = exports.DEFAULT_TEXT_QA_PROMPT_TMPL = exports.REFINE_PROMPT_SELECTOR = exports.CHAT_REFINE_PROMPT = exports.DEFAULT_REFINE_PROMPT = exports.DEFAULT_REFINE_PROMPT_TMPL = void 0;
 /* eslint-disable spaced-comment */
 const index_js_1 = require("../../prompts/index.cjs");
-const prompt_selector_js_1 = require("../prompt_selector.cjs");
+const conditional_js_1 = require("../../prompts/selectors/conditional.cjs");
 exports.DEFAULT_REFINE_PROMPT_TMPL = `The original question is as follows: {question}
 We have provided an existing answer: {existing_answer}
 We have the opportunity to refine the existing answer
@@ -34,8 +34,8 @@ const messages = [
 exports.CHAT_REFINE_PROMPT = 
 /*#__PURE__*/ index_js_1.ChatPromptTemplate.fromPromptMessages(messages);
 exports.REFINE_PROMPT_SELECTOR = 
-/*#__PURE__*/ new prompt_selector_js_1.ConditionalPromptSelector(exports.DEFAULT_REFINE_PROMPT, [
-    [prompt_selector_js_1.isChatModel, exports.CHAT_REFINE_PROMPT],
+/*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_REFINE_PROMPT, [
+    [conditional_js_1.isChatModel, exports.CHAT_REFINE_PROMPT],
 ]);
 exports.DEFAULT_TEXT_QA_PROMPT_TMPL = `Context information is below. 
 ---------------------
@@ -58,6 +58,6 @@ const chat_messages = [
 exports.CHAT_QUESTION_PROMPT = 
 /*#__PURE__*/ index_js_1.ChatPromptTemplate.fromPromptMessages(chat_messages);
 exports.QUESTION_PROMPT_SELECTOR = 
-/*#__PURE__*/ new prompt_selector_js_1.ConditionalPromptSelector(exports.DEFAULT_TEXT_QA_PROMPT, [
-    [prompt_selector_js_1.isChatModel, exports.CHAT_QUESTION_PROMPT],
+/*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_TEXT_QA_PROMPT, [
+    [conditional_js_1.isChatModel, exports.CHAT_QUESTION_PROMPT],
 ]);

package/dist/chains/question_answering/refine_prompts.d.ts

@@ -1,5 +1,5 @@
 import { PromptTemplate, ChatPromptTemplate } from "../../prompts/index.js";
-import { ConditionalPromptSelector } from "../prompt_selector.js";
+import { ConditionalPromptSelector } from "../../prompts/selectors/conditional.js";
 export declare const DEFAULT_REFINE_PROMPT_TMPL = "The original question is as follows: {question}\nWe have provided an existing answer: {existing_answer}\nWe have the opportunity to refine the existing answer\n(only if needed) with some more context below.\n------------\n{context}\n------------\nGiven the new context, refine the original answer to better answer the question. \nIf the context isn't useful, return the original answer.";
 export declare const DEFAULT_REFINE_PROMPT: PromptTemplate;
 export declare const CHAT_REFINE_PROMPT: ChatPromptTemplate;

package/dist/chains/question_answering/refine_prompts.js

@@ -1,6 +1,6 @@
 /* eslint-disable spaced-comment */
 import { PromptTemplate, ChatPromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate, AIMessagePromptTemplate, } from "../../prompts/index.js";
-import { ConditionalPromptSelector, isChatModel } from "../prompt_selector.js";
+import { ConditionalPromptSelector, isChatModel, } from "../../prompts/selectors/conditional.js";
 export const DEFAULT_REFINE_PROMPT_TMPL = `The original question is as follows: {question}
 We have provided an existing answer: {existing_answer}
 We have the opportunity to refine the existing answer

package/dist/chains/question_answering/stuff_prompts.cjs

@@ -4,7 +4,7 @@ exports.QA_PROMPT_SELECTOR = exports.DEFAULT_QA_PROMPT = void 0;
 /* eslint-disable spaced-comment */
 const prompt_js_1 = require("../../prompts/prompt.cjs");
 const chat_js_1 = require("../../prompts/chat.cjs");
-const prompt_selector_js_1 = require("../prompt_selector.cjs");
+const conditional_js_1 = require("../../prompts/selectors/conditional.cjs");
 exports.DEFAULT_QA_PROMPT = new prompt_js_1.PromptTemplate({
     template: "Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:",
     inputVariables: ["context", "question"],
@@ -19,4 +19,4 @@ const messages = [
 ];
 const CHAT_PROMPT = 
 /*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
-exports.QA_PROMPT_SELECTOR = new prompt_selector_js_1.ConditionalPromptSelector(exports.DEFAULT_QA_PROMPT, [[prompt_selector_js_1.isChatModel, CHAT_PROMPT]]);
+exports.QA_PROMPT_SELECTOR = new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_QA_PROMPT, [[conditional_js_1.isChatModel, CHAT_PROMPT]]);

package/dist/chains/question_answering/stuff_prompts.d.ts

@@ -1,4 +1,4 @@
 import { PromptTemplate } from "../../prompts/prompt.js";
-import { ConditionalPromptSelector } from "../prompt_selector.js";
+import { ConditionalPromptSelector } from "../../prompts/selectors/conditional.js";
 export declare const DEFAULT_QA_PROMPT: PromptTemplate;
 export declare const QA_PROMPT_SELECTOR: ConditionalPromptSelector;

package/dist/chains/question_answering/stuff_prompts.js

@@ -1,7 +1,7 @@
 /* eslint-disable spaced-comment */
 import { PromptTemplate } from "../../prompts/prompt.js";
 import { ChatPromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate, } from "../../prompts/chat.js";
-import { ConditionalPromptSelector, isChatModel } from "../prompt_selector.js";
+import { ConditionalPromptSelector, isChatModel, } from "../../prompts/selectors/conditional.js";
 export const DEFAULT_QA_PROMPT = /*#__PURE__*/ new PromptTemplate({
     template: "Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:",
     inputVariables: ["context", "question"],

package/dist/chains/sequential_chain.cjs

@@ -168,8 +168,8 @@ exports.SequentialChain = SequentialChain;
  * Play Synopsis:
  * {synopsis}
  * Review from a New York Times play critic of the above play:`
- * const reviewPromptTempalte = new PromptTemplate({ template: reviewTemplate, inputVariables: ["synopsis"] });
- * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTempalte });
+ * const reviewPromptTemplate = new PromptTemplate({ template: reviewTemplate, inputVariables: ["synopsis"] });
+ * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTemplate });
  *
  * const overallChain = new SimpleSequentialChain({chains: [synopsisChain, reviewChain], verbose:true})
  * const review = await overallChain.run("Tragedy at sunset on the beach")

package/dist/chains/sequential_chain.d.ts

@@ -65,8 +65,8 @@ export interface SimpleSequentialChainInput extends ChainInputs {
  * Play Synopsis:
  * {synopsis}
  * Review from a New York Times play critic of the above play:`
- * const reviewPromptTempalte = new PromptTemplate({ template: reviewTemplate, inputVariables: ["synopsis"] });
- * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTempalte });
+ * const reviewPromptTemplate = new PromptTemplate({ template: reviewTemplate, inputVariables: ["synopsis"] });
+ * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTemplate });
  *
  * const overallChain = new SimpleSequentialChain({chains: [synopsisChain, reviewChain], verbose:true})
  * const review = await overallChain.run("Tragedy at sunset on the beach")

package/dist/chains/sequential_chain.js

@@ -164,8 +164,8 @@ export class SequentialChain extends BaseChain {
  * Play Synopsis:
  * {synopsis}
  * Review from a New York Times play critic of the above play:`
- * const reviewPromptTempalte = new PromptTemplate({ template: reviewTemplate, inputVariables: ["synopsis"] });
- * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTempalte });
+ * const reviewPromptTemplate = new PromptTemplate({ template: reviewTemplate, inputVariables: ["synopsis"] });
+ * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTemplate });
  *
  * const overallChain = new SimpleSequentialChain({chains: [synopsisChain, reviewChain], verbose:true})
  * const review = await overallChain.run("Tragedy at sunset on the beach")

package/dist/chains/sql_db/sql_db_prompt.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SQL_MYSQL_PROMPT = exports.SQL_SQLITE_PROMPT = exports.SQL_POSTGRES_PROMPT = exports.DEFAULT_SQL_DATABASE_PROMPT = void 0;
+exports.SQL_MSSQL_PROMPT = exports.SQL_MYSQL_PROMPT = exports.SQL_SQLITE_PROMPT = exports.SQL_POSTGRES_PROMPT = exports.DEFAULT_SQL_DATABASE_PROMPT = void 0;
 /* eslint-disable spaced-comment */
 const prompt_js_1 = require("../../prompts/prompt.cjs");
 exports.DEFAULT_SQL_DATABASE_PROMPT = new prompt_js_1.PromptTemplate({
@@ -81,3 +81,22 @@ Only use the following tables:
 Question: {input}`,
     inputVariables: ["dialect", "table_info", "input", "top_k"],
 });
+exports.SQL_MSSQL_PROMPT = new prompt_js_1.PromptTemplate({
+    template: `You are an MS SQL expert. Given an input question, first create a syntactically correct MS SQL query to run, then look at the results of the query and return the answer to the input question.
+Unless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the TOP clause as per MS SQL. You can order the results to return the most informative data in the database.
+Never query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in square brackets ([]) to denote them as delimited identifiers.
+Pay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.
+
+Use the following format:
+
+Question: "Question here"
+SQLQuery: "SQL Query to run"
+SQLResult: "Result of the SQLQuery"
+Answer: "Final answer here"
+
+Only use the following tables:
+{table_info}
+
+Question: {input}`,
+    inputVariables: ["dialect", "table_info", "input", "top_k"],
+});

package/dist/chains/sql_db/sql_db_prompt.d.ts

@@ -3,3 +3,4 @@ export declare const DEFAULT_SQL_DATABASE_PROMPT: PromptTemplate;
 export declare const SQL_POSTGRES_PROMPT: PromptTemplate;
 export declare const SQL_SQLITE_PROMPT: PromptTemplate;
 export declare const SQL_MYSQL_PROMPT: PromptTemplate;
+export declare const SQL_MSSQL_PROMPT: PromptTemplate;

package/dist/chains/sql_db/sql_db_prompt.js

@@ -78,3 +78,22 @@ Only use the following tables:
 Question: {input}`,
     inputVariables: ["dialect", "table_info", "input", "top_k"],
 });
+export const SQL_MSSQL_PROMPT = /*#__PURE__*/ new PromptTemplate({
+    template: `You are an MS SQL expert. Given an input question, first create a syntactically correct MS SQL query to run, then look at the results of the query and return the answer to the input question.
+Unless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the TOP clause as per MS SQL. You can order the results to return the most informative data in the database.
+Never query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in square brackets ([]) to denote them as delimited identifiers.
+Pay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.
+
+Use the following format:
+
+Question: "Question here"
+SQLQuery: "SQL Query to run"
+SQLResult: "Result of the SQLQuery"
+Answer: "Final answer here"
+
+Only use the following tables:
+{table_info}
+
+Question: {input}`,
+    inputVariables: ["dialect", "table_info", "input", "top_k"],
+});

package/dist/chains/summarization/load.cjs

@@ -6,39 +6,48 @@ const combine_docs_chain_js_1 = require("../combine_docs_chain.cjs");
 const stuff_prompts_js_1 = require("./stuff_prompts.cjs");
 const refine_prompts_js_1 = require("./refine_prompts.cjs");
 const loadSummarizationChain = (llm, params = { type: "map_reduce" }) => {
+    const { verbose } = params;
     if (params.type === "stuff") {
         const { prompt = stuff_prompts_js_1.DEFAULT_PROMPT } = params;
-        const llmChain = new llm_chain_js_1.LLMChain({ prompt, llm });
+        const llmChain = new llm_chain_js_1.LLMChain({ prompt, llm, verbose });
         const chain = new combine_docs_chain_js_1.StuffDocumentsChain({
             llmChain,
             documentVariableName: "text",
+            verbose,
         });
         return chain;
     }
     if (params.type === "map_reduce") {
         const { combineMapPrompt = stuff_prompts_js_1.DEFAULT_PROMPT, combinePrompt = stuff_prompts_js_1.DEFAULT_PROMPT, returnIntermediateSteps, } = params;
-        const llmChain = new llm_chain_js_1.LLMChain({ prompt: combineMapPrompt, llm });
-        const combineLLMChain = new llm_chain_js_1.LLMChain({ prompt: combinePrompt, llm });
+        const llmChain = new llm_chain_js_1.LLMChain({ prompt: combineMapPrompt, llm, verbose });
+        const combineLLMChain = new llm_chain_js_1.LLMChain({
+            prompt: combinePrompt,
+            llm,
+            verbose,
+        });
         const combineDocumentChain = new combine_docs_chain_js_1.StuffDocumentsChain({
             llmChain: combineLLMChain,
             documentVariableName: "text",
+            verbose,
         });
         const chain = new combine_docs_chain_js_1.MapReduceDocumentsChain({
             llmChain,
             combineDocumentChain,
             documentVariableName: "text",
             returnIntermediateSteps,
+            verbose,
         });
         return chain;
     }
     if (params.type === "refine") {
         const { refinePrompt = refine_prompts_js_1.REFINE_PROMPT, questionPrompt = stuff_prompts_js_1.DEFAULT_PROMPT } = params;
-        const llmChain = new llm_chain_js_1.LLMChain({ prompt: questionPrompt, llm });
-        const refineLLMChain = new llm_chain_js_1.LLMChain({ prompt: refinePrompt, llm });
+        const llmChain = new llm_chain_js_1.LLMChain({ prompt: questionPrompt, llm, verbose });
+        const refineLLMChain = new llm_chain_js_1.LLMChain({ prompt: refinePrompt, llm, verbose });
         const chain = new combine_docs_chain_js_1.RefineDocumentsChain({
             llmChain,
             refineLLMChain,
             documentVariableName: "text",
+            verbose,
         });
         return chain;
     }

package/dist/chains/summarization/load.d.ts

@@ -1,7 +1,11 @@
 import { BaseLanguageModel } from "../../base_language/index.js";
 import { BasePromptTemplate } from "../../prompts/base.js";
 import { StuffDocumentsChain, MapReduceDocumentsChain, RefineDocumentsChain, MapReduceDocumentsChainInput } from "../combine_docs_chain.js";
-export type SummarizationChainParams = {
+type BaseParams = {
+    verbose?: boolean;
+};
+/** @interface */
+export type SummarizationChainParams = BaseParams & ({
     type?: "stuff";
     prompt?: BasePromptTemplate;
 } | ({
@@ -12,5 +16,6 @@ export type SummarizationChainParams = {
     type?: "refine";
     refinePrompt?: BasePromptTemplate;
     questionPrompt?: BasePromptTemplate;
-};
+});
 export declare const loadSummarizationChain: (llm: BaseLanguageModel, params?: SummarizationChainParams) => StuffDocumentsChain | MapReduceDocumentsChain | RefineDocumentsChain;
+export {};