langchain 0.0.63 → 0.0.65

package/dist/agents/agent.cjs

@@ -100,9 +100,8 @@ class LLMSingleActionAgent extends BaseSingleActionAgent {
             intermediate_steps: steps,
             stop: this.stop,
             ...inputs,
-            callbackManager,
-        });
-        return this.outputParser.parse(output[this.llmChain.outputKey]);
+        }, callbackManager);
+        return this.outputParser.parse(output[this.llmChain.outputKey], callbackManager);
     }
 }
 exports.LLMSingleActionAgent = LLMSingleActionAgent;
@@ -203,7 +202,7 @@ class Agent extends BaseSingleActionAgent {
             newInputs.stop = this._stop();
         }
         const output = await this.llmChain.predict(newInputs, callbackManager);
-        return this.outputParser.parse(output);
+        return this.outputParser.parse(output, callbackManager);
     }
     /**
      * Decide what to do given some input.

package/dist/agents/agent.js

@@ -94,9 +94,8 @@ export class LLMSingleActionAgent extends BaseSingleActionAgent {
             intermediate_steps: steps,
             stop: this.stop,
             ...inputs,
-            callbackManager,
-        });
-        return this.outputParser.parse(output[this.llmChain.outputKey]);
+        }, callbackManager);
+        return this.outputParser.parse(output[this.llmChain.outputKey], callbackManager);
     }
 }
 /**
@@ -196,7 +195,7 @@ export class Agent extends BaseSingleActionAgent {
             newInputs.stop = this._stop();
         }
         const output = await this.llmChain.predict(newInputs, callbackManager);
-        return this.outputParser.parse(output);
+        return this.outputParser.parse(output, callbackManager);
     }
     /**
      * Decide what to do given some input.

package/dist/agents/chat_convo/prompt.cjs

@@ -31,7 +31,7 @@ Use this if you want to respond directly to the human. Markdown code snippet for
 \`\`\`json
 {{{{
     "action": "Final Answer",
-    "action_input": string \\ You should put what you want to return to use here
+    "action_input": string \\ You should put what you want to return to use here and make sure to use valid json newline characters.
 }}}}
 \`\`\``;
 exports.DEFAULT_SUFFIX = `TOOLS

package/dist/agents/chat_convo/prompt.d.ts

@@ -1,5 +1,5 @@
 export declare const DEFAULT_PREFIX = "Assistant is a large language model trained by OpenAI.\n\nAssistant is designed to be able to assist with a wide range of tasks, from answering simple questions to providing in-depth explanations and discussions on a wide range of topics. As a language model, Assistant is able to generate human-like text based on the input it receives, allowing it to engage in natural-sounding conversations and provide responses that are coherent and relevant to the topic at hand.\n\nAssistant is constantly learning and improving, and its capabilities are constantly evolving. It is able to process and understand large amounts of text, and can use this knowledge to provide accurate and informative responses to a wide range of questions. Additionally, Assistant is able to generate its own text based on the input it receives, allowing it to engage in discussions and provide explanations and descriptions on a wide range of topics.\n\nOverall, Assistant is a powerful system that can help with a wide range of tasks and provide valuable insights and information on a wide range of topics. Whether you need help with a specific question or just want to have a conversation about a particular topic, Assistant is here to assist.";
 export declare const PREFIX_END = " However, above all else, all responses must adhere to the format of RESPONSE FORMAT INSTRUCTIONS.";
-export declare const FORMAT_INSTRUCTIONS = "RESPONSE FORMAT INSTRUCTIONS\n----------------------------\n\nWhen responding to me please, please output a response in one of two formats:\n\n**Option 1:**\nUse this if you want the human to use a tool.\nMarkdown code snippet formatted in the following schema:\n\n```json\n{{{{\n    \"action\": string \\ The action to take. Must be one of {tool_names}\n    \"action_input\": string \\ The input to the action\n}}}}\n```\n\n**Option #2:**\nUse this if you want to respond directly to the human. Markdown code snippet formatted in the following schema:\n\n```json\n{{{{\n    \"action\": \"Final Answer\",\n    \"action_input\": string \\ You should put what you want to return to use here\n}}}}\n```";
+export declare const FORMAT_INSTRUCTIONS = "RESPONSE FORMAT INSTRUCTIONS\n----------------------------\n\nWhen responding to me please, please output a response in one of two formats:\n\n**Option 1:**\nUse this if you want the human to use a tool.\nMarkdown code snippet formatted in the following schema:\n\n```json\n{{{{\n    \"action\": string \\ The action to take. Must be one of {tool_names}\n    \"action_input\": string \\ The input to the action\n}}}}\n```\n\n**Option #2:**\nUse this if you want to respond directly to the human. Markdown code snippet formatted in the following schema:\n\n```json\n{{{{\n    \"action\": \"Final Answer\",\n    \"action_input\": string \\ You should put what you want to return to use here and make sure to use valid json newline characters.\n}}}}\n```";
 export declare const DEFAULT_SUFFIX = "TOOLS\n------\nAssistant can ask the user to use tools to look up information that may be helpful in answering the users original question. The tools the human can use are:\n\n{{tools}}\n\n{format_instructions}\n\nUSER'S INPUT\n--------------------\nHere is the user's input (remember to respond with a markdown code snippet of a json blob with a single action, and NOTHING else):\n\n{{{{input}}}}";
 export declare const TEMPLATE_TOOL_RESPONSE = "TOOL RESPONSE:\n---------------------\n{observation}\n\nUSER'S INPUT\n--------------------\n\nOkay, so what is the response to my original question? If using information from tools, you must say it explicitly - I have forgotten all TOOL RESPONSES! Remember to respond with a markdown code snippet of a json blob with a single action, and NOTHING else.";

package/dist/agents/chat_convo/prompt.js

@@ -28,7 +28,7 @@ Use this if you want to respond directly to the human. Markdown code snippet for
 \`\`\`json
 {{{{
     "action": "Final Answer",
-    "action_input": string \\ You should put what you want to return to use here
+    "action_input": string \\ You should put what you want to return to use here and make sure to use valid json newline characters.
 }}}}
 \`\`\``;
 export const DEFAULT_SUFFIX = `TOOLS

package/dist/agents/initialize.cjs

@@ -7,7 +7,7 @@ const index_js_2 = require("./chat_convo/index.cjs");
 const executor_js_1 = require("./executor.cjs");
 const index_js_3 = require("./mrkl/index.cjs");
 /**
- * @deprecated use initializeExecutorWithOptions instead
+ * @deprecated use initializeAgentExecutorWithOptions instead
  */
 const initializeAgentExecutor = async (tools, llm, _agentType, _verbose, _callbackManager) => {
     const agentType = _agentType ?? "zero-shot-react-description";

package/dist/agents/initialize.d.ts

@@ -7,7 +7,7 @@ import { AgentExecutor, AgentExecutorInput } from "./executor.js";
 import { ZeroShotAgent } from "./mrkl/index.js";
 type AgentType = "zero-shot-react-description" | "chat-zero-shot-react-description" | "chat-conversational-react-description";
 /**
- * @deprecated use initializeExecutorWithOptions instead
+ * @deprecated use initializeAgentExecutorWithOptions instead
  */
 export declare const initializeAgentExecutor: (tools: Tool[], llm: BaseLanguageModel, _agentType?: AgentType, _verbose?: boolean, _callbackManager?: CallbackManager) => Promise<AgentExecutor>;
 /**

package/dist/agents/initialize.js

@@ -4,7 +4,7 @@ import { ChatConversationalAgent } from "./chat_convo/index.js";
 import { AgentExecutor } from "./executor.js";
 import { ZeroShotAgent } from "./mrkl/index.js";
 /**
- * @deprecated use initializeExecutorWithOptions instead
+ * @deprecated use initializeAgentExecutorWithOptions instead
  */
 export const initializeAgentExecutor = async (tools, llm, _agentType, _verbose, _callbackManager) => {
     const agentType = _agentType ?? "zero-shot-react-description";

package/dist/agents/types.cjs

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AgentActionOutputParser = void 0;
-const index_js_1 = require("../schema/index.cjs");
-class AgentActionOutputParser extends index_js_1.BaseOutputParser {
+const output_parser_js_1 = require("../schema/output_parser.cjs");
+class AgentActionOutputParser extends output_parser_js_1.BaseOutputParser {
 }
 exports.AgentActionOutputParser = AgentActionOutputParser;

package/dist/agents/types.d.ts

@@ -1,6 +1,7 @@
 import { LLMChain } from "../chains/llm_chain.js";
 import { SerializedLLMChain } from "../chains/serde.js";
-import { AgentAction, AgentFinish, BaseOutputParser } from "../schema/index.js";
+import { AgentAction, AgentFinish } from "../schema/index.js";
+import { BaseOutputParser } from "../schema/output_parser.js";
 export interface AgentInput {
     llmChain: LLMChain;
     outputParser: AgentActionOutputParser;

package/dist/agents/types.js

@@ -1,3 +1,3 @@
-import { BaseOutputParser } from "../schema/index.js";
+import { BaseOutputParser } from "../schema/output_parser.js";
 export class AgentActionOutputParser extends BaseOutputParser {
 }

package/dist/base_language/index.d.ts

@@ -31,17 +31,20 @@ export interface BaseLanguageModelParams extends AsyncCallerParams, BaseLangChai
      */
     callbackManager?: CallbackManager;
 }
+export interface BaseLanguageModelCallOptions {
+}
 /**
  * Base class for language models.
  */
 export declare abstract class BaseLanguageModel extends BaseLangChain implements BaseLanguageModelParams {
+    CallOptions: BaseLanguageModelCallOptions;
     /**
      * The async caller should be used by subclasses to make any async calls,
      * which will thus benefit from the concurrency and retry logic.
      */
     caller: AsyncCaller;
     constructor(params: BaseLanguageModelParams);
-    abstract generatePrompt(promptValues: BasePromptValue[], stop?: string[], callbacks?: Callbacks): Promise<LLMResult>;
+    abstract generatePrompt(promptValues: BasePromptValue[], stop?: string[] | this["CallOptions"], callbacks?: Callbacks): Promise<LLMResult>;
     abstract _modelType(): string;
     abstract _llmType(): string;
     private _encoding?;

package/dist/callbacks/manager.cjs

@@ -319,7 +319,11 @@ class CallbackManager extends BaseCallbackManager {
             }
             if (tracingEnabled &&
                 !callbackManager.handlers.some((handler) => handler.name === "langchain_tracer")) {
-                callbackManager.addHandler(await (0, initialize_js_1.getTracingCallbackHandler)(), true);
+                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.js

@@ -312,7 +312,11 @@ export class CallbackManager extends BaseCallbackManager {
             }
             if (tracingEnabled &&
                 !callbackManager.handlers.some((handler) => handler.name === "langchain_tracer")) {
-                callbackManager.addHandler(await getTracingCallbackHandler(), true);
+                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/analyze_documents_chain.cjs

@@ -10,7 +10,7 @@ const text_splitter_js_1 = require("../text_splitter.cjs");
  */
 class AnalyzeDocumentChain extends base_js_1.BaseChain {
     constructor(fields) {
-        super(undefined, fields.verbose, fields.callbackManager);
+        super(fields);
         Object.defineProperty(this, "inputKey", {
             enumerable: true,
             configurable: true,
@@ -41,7 +41,7 @@ class AnalyzeDocumentChain extends base_js_1.BaseChain {
         return this.combineDocumentsChain.outputKeys;
     }
     /** @ignore */
-    async _call(values) {
+    async _call(values, runManager) {
         if (!(this.inputKey in values)) {
             throw new Error(`Document key ${this.inputKey} not found.`);
         }
@@ -49,7 +49,7 @@ class AnalyzeDocumentChain extends base_js_1.BaseChain {
         const currentDoc = doc;
         const currentDocs = await this.textSplitter.createDocuments([currentDoc]);
         const newInputs = { input_documents: currentDocs, ...rest };
-        const result = await this.combineDocumentsChain.call(newInputs);
+        const result = await this.combineDocumentsChain.call(newInputs, runManager?.getChild());
         return result;
     }
     _chainType() {

package/dist/chains/analyze_documents_chain.d.ts

@@ -2,6 +2,7 @@ import { BaseChain, ChainInputs } from "./base.js";
 import { TextSplitter } from "../text_splitter.js";
 import { ChainValues } from "../schema/index.js";
 import { SerializedAnalyzeDocumentChain } from "./serde.js";
+import { CallbackManagerForChainRun } from "../callbacks/manager.js";
 export type LoadValues = Record<string, any>;
 export interface AnalyzeDocumentChainInput extends Omit<ChainInputs, "memory"> {
     combineDocumentsChain: BaseChain;
@@ -21,7 +22,7 @@ export declare class AnalyzeDocumentChain extends BaseChain implements AnalyzeDo
     get inputKeys(): string[];
     get outputKeys(): string[];
     /** @ignore */
-    _call(values: ChainValues): Promise<ChainValues>;
+    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
     _chainType(): "analyze_document_chain";
     static deserialize(data: SerializedAnalyzeDocumentChain, values: LoadValues): Promise<AnalyzeDocumentChain>;
     serialize(): SerializedAnalyzeDocumentChain;

package/dist/chains/analyze_documents_chain.js

@@ -7,7 +7,7 @@ import { RecursiveCharacterTextSplitter, } from "../text_splitter.js";
  */
 export class AnalyzeDocumentChain extends BaseChain {
     constructor(fields) {
-        super(undefined, fields.verbose, fields.callbackManager);
+        super(fields);
         Object.defineProperty(this, "inputKey", {
             enumerable: true,
             configurable: true,
@@ -38,7 +38,7 @@ export class AnalyzeDocumentChain extends BaseChain {
         return this.combineDocumentsChain.outputKeys;
     }
     /** @ignore */
-    async _call(values) {
+    async _call(values, runManager) {
         if (!(this.inputKey in values)) {
             throw new Error(`Document key ${this.inputKey} not found.`);
         }
@@ -46,7 +46,7 @@ export class AnalyzeDocumentChain extends BaseChain {
         const currentDoc = doc;
         const currentDocs = await this.textSplitter.createDocuments([currentDoc]);
         const newInputs = { input_documents: currentDocs, ...rest };
-        const result = await this.combineDocumentsChain.call(newInputs);
+        const result = await this.combineDocumentsChain.call(newInputs, runManager?.getChild());
         return result;
     }
     _chainType() {

package/dist/chains/base.cjs

@@ -8,15 +8,30 @@ const index_js_2 = require("../base_language/index.cjs");
  * Base interface that all chains must implement.
  */
 class BaseChain extends index_js_2.BaseLangChain {
-    constructor(memory, verbose, callbacks) {
-        super({ verbose, callbacks });
-        Object.defineProperty(this, "memory", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.memory = memory;
+    constructor(fields, 
+    /** @deprecated */
+    verbose, 
+    /** @deprecated */
+    callbacks) {
+        if (arguments.length === 1 &&
+            typeof fields === "object" &&
+            !("saveContext" in fields)) {
+            // fields is not a BaseMemory
+            const { memory, callbackManager, ...rest } = fields;
+            super({ ...rest, callbacks: callbackManager ?? rest.callbacks });
+            this.memory = memory;
+        }
+        else {
+            // fields is a BaseMemory
+            super({ verbose, callbacks });
+            this.memory = fields;
+        }
+    }
+    /**
+     * Return a json-like object representing this chain.
+     */
+    serialize() {
+        throw new Error("Method not implemented.");
     }
     async run(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -36,7 +51,7 @@ class BaseChain extends index_js_2.BaseLangChain {
     /**
      * Run the core logic of this chain and add to output if desired.
      *
-     * Wraps {@link _call} and handles memory.
+     * Wraps _call and handles memory.
      */
     async call(values, callbacks) {
         const fullValues = { ...values };
@@ -82,8 +97,12 @@ class BaseChain extends index_js_2.BaseLangChain {
                 const { LLMChain } = await import("./llm_chain.js");
                 return LLMChain.deserialize(data);
             }
+            case "sequential_chain": {
+                const { SequentialChain } = await import("./sequential_chain.js");
+                return SequentialChain.deserialize(data);
+            }
             case "simple_sequential_chain": {
-                const { SimpleSequentialChain } = await import("./simple_sequential_chain.js");
+                const { SimpleSequentialChain } = await import("./sequential_chain.js");
                 return SimpleSequentialChain.deserialize(data);
             }
             case "stuff_documents_chain": {

package/dist/chains/base.d.ts

@@ -16,7 +16,11 @@ export interface ChainInputs extends BaseLangChainParams {
  */
 export declare abstract class BaseChain extends BaseLangChain implements ChainInputs {
     memory?: BaseMemory;
-    constructor(memory?: BaseMemory, verbose?: boolean, callbacks?: Callbacks);
+    constructor(fields?: BaseMemory | ChainInputs, 
+    /** @deprecated */
+    verbose?: boolean, 
+    /** @deprecated */
+    callbacks?: Callbacks);
     /**
      * Run the core logic of this chain and return the output
      */
@@ -28,14 +32,14 @@ export declare abstract class BaseChain extends BaseLangChain implements ChainIn
     /**
      * Return a json-like object representing this chain.
      */
-    abstract serialize(): SerializedBaseChain;
+    serialize(): SerializedBaseChain;
     abstract get inputKeys(): string[];
     abstract get outputKeys(): string[];
     run(input: any, callbacks?: Callbacks): Promise<string>;
     /**
      * Run the core logic of this chain and add to output if desired.
      *
-     * Wraps {@link _call} and handles memory.
+     * Wraps _call and handles memory.
      */
     call(values: ChainValues, callbacks?: Callbacks): Promise<ChainValues>;
     /**

package/dist/chains/base.js

@@ -5,15 +5,30 @@ import { BaseLangChain } from "../base_language/index.js";
  * Base interface that all chains must implement.
  */
 export class BaseChain extends BaseLangChain {
-    constructor(memory, verbose, callbacks) {
-        super({ verbose, callbacks });
-        Object.defineProperty(this, "memory", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.memory = memory;
+    constructor(fields, 
+    /** @deprecated */
+    verbose, 
+    /** @deprecated */
+    callbacks) {
+        if (arguments.length === 1 &&
+            typeof fields === "object" &&
+            !("saveContext" in fields)) {
+            // fields is not a BaseMemory
+            const { memory, callbackManager, ...rest } = fields;
+            super({ ...rest, callbacks: callbackManager ?? rest.callbacks });
+            this.memory = memory;
+        }
+        else {
+            // fields is a BaseMemory
+            super({ verbose, callbacks });
+            this.memory = fields;
+        }
+    }
+    /**
+     * Return a json-like object representing this chain.
+     */
+    serialize() {
+        throw new Error("Method not implemented.");
     }
     async run(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -33,7 +48,7 @@ export class BaseChain extends BaseLangChain {
     /**
      * Run the core logic of this chain and add to output if desired.
      *
-     * Wraps {@link _call} and handles memory.
+     * Wraps _call and handles memory.
      */
     async call(values, callbacks) {
         const fullValues = { ...values };
@@ -79,8 +94,12 @@ export class BaseChain extends BaseLangChain {
                 const { LLMChain } = await import("./llm_chain.js");
                 return LLMChain.deserialize(data);
             }
+            case "sequential_chain": {
+                const { SequentialChain } = await import("./sequential_chain.js");
+                return SequentialChain.deserialize(data);
+            }
             case "simple_sequential_chain": {
-                const { SimpleSequentialChain } = await import("./simple_sequential_chain.js");
+                const { SimpleSequentialChain } = await import("./sequential_chain.js");
                 return SimpleSequentialChain.deserialize(data);
             }
             case "stuff_documents_chain": {

package/dist/chains/chat_vector_db_chain.cjs

@@ -25,7 +25,7 @@ class ChatVectorDBQAChain extends base_js_1.BaseChain {
         return [this.outputKey];
     }
     constructor(fields) {
-        super();
+        super(fields);
         Object.defineProperty(this, "k", {
             enumerable: true,
             configurable: true,
@@ -84,7 +84,7 @@ class ChatVectorDBQAChain extends base_js_1.BaseChain {
             fields.returnSourceDocuments ?? this.returnSourceDocuments;
     }
     /** @ignore */
-    async _call(values) {
+    async _call(values, runManager) {
         if (!(this.inputKey in values)) {
             throw new Error(`Question key ${this.inputKey} not found.`);
         }
@@ -98,7 +98,7 @@ class ChatVectorDBQAChain extends base_js_1.BaseChain {
             const result = await this.questionGeneratorChain.call({
                 question,
                 chat_history: chatHistory,
-            });
+            }, runManager?.getChild());
             const keys = Object.keys(result);
             console.log("_call", values, keys);
             if (keys.length === 1) {
@@ -114,7 +114,7 @@ class ChatVectorDBQAChain extends base_js_1.BaseChain {
             input_documents: docs,
             chat_history: chatHistory,
         };
-        const result = await this.combineDocumentsChain.call(inputs);
+        const result = await this.combineDocumentsChain.call(inputs, runManager?.getChild());
         if (this.returnSourceDocuments) {
             return {
                 ...result,

package/dist/chains/chat_vector_db_chain.d.ts

@@ -2,16 +2,18 @@ import { BaseLanguageModel } from "../base_language/index.js";
 import { VectorStore } from "../vectorstores/base.js";
 import { SerializedChatVectorDBQAChain } from "./serde.js";
 import { ChainValues } from "../schema/index.js";
-import { BaseChain } from "./base.js";
+import { BaseChain, ChainInputs } from "./base.js";
 import { LLMChain } from "./llm_chain.js";
+import { CallbackManagerForChainRun } from "../callbacks/manager.js";
 export type LoadValues = Record<string, any>;
-export interface ChatVectorDBQAChainInput {
+export interface ChatVectorDBQAChainInput extends ChainInputs {
     vectorstore: VectorStore;
-    k: number;
     combineDocumentsChain: BaseChain;
     questionGeneratorChain: LLMChain;
-    outputKey: string;
-    inputKey: string;
+    returnSourceDocuments?: boolean;
+    outputKey?: string;
+    inputKey?: string;
+    k?: number;
 }
 export declare class ChatVectorDBQAChain extends BaseChain implements ChatVectorDBQAChainInput {
     k: number;
@@ -24,17 +26,9 @@ export declare class ChatVectorDBQAChain extends BaseChain implements ChatVector
     combineDocumentsChain: BaseChain;
     questionGeneratorChain: LLMChain;
     returnSourceDocuments: boolean;
-    constructor(fields: {
-        vectorstore: VectorStore;
-        combineDocumentsChain: BaseChain;
-        questionGeneratorChain: LLMChain;
-        inputKey?: string;
-        outputKey?: string;
-        k?: number;
-        returnSourceDocuments?: boolean;
-    });
+    constructor(fields: ChatVectorDBQAChainInput);
     /** @ignore */
-    _call(values: ChainValues): Promise<ChainValues>;
+    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
     _chainType(): "chat-vector-db";
     static deserialize(data: SerializedChatVectorDBQAChain, values: LoadValues): Promise<ChatVectorDBQAChain>;
     serialize(): SerializedChatVectorDBQAChain;

package/dist/chains/chat_vector_db_chain.js

@@ -22,7 +22,7 @@ export class ChatVectorDBQAChain extends BaseChain {
         return [this.outputKey];
     }
     constructor(fields) {
-        super();
+        super(fields);
         Object.defineProperty(this, "k", {
             enumerable: true,
             configurable: true,
@@ -81,7 +81,7 @@ export class ChatVectorDBQAChain extends BaseChain {
             fields.returnSourceDocuments ?? this.returnSourceDocuments;
     }
     /** @ignore */
-    async _call(values) {
+    async _call(values, runManager) {
         if (!(this.inputKey in values)) {
             throw new Error(`Question key ${this.inputKey} not found.`);
         }
@@ -95,7 +95,7 @@ export class ChatVectorDBQAChain extends BaseChain {
             const result = await this.questionGeneratorChain.call({
                 question,
                 chat_history: chatHistory,
-            });
+            }, runManager?.getChild());
             const keys = Object.keys(result);
             console.log("_call", values, keys);
             if (keys.length === 1) {
@@ -111,7 +111,7 @@ export class ChatVectorDBQAChain extends BaseChain {
             input_documents: docs,
             chat_history: chatHistory,
         };
-        const result = await this.combineDocumentsChain.call(inputs);
+        const result = await this.combineDocumentsChain.call(inputs, runManager?.getChild());
         if (this.returnSourceDocuments) {
             return {
                 ...result,

package/dist/chains/combine_docs_chain.cjs

@@ -17,7 +17,7 @@ class StuffDocumentsChain extends base_js_1.BaseChain {
         return this.llmChain.outputKeys;
     }
     constructor(fields) {
-        super();
+        super(fields);
         Object.defineProperty(this, "llmChain", {
             enumerable: true,
             configurable: true,
@@ -42,7 +42,7 @@ class StuffDocumentsChain extends base_js_1.BaseChain {
         this.inputKey = fields.inputKey ?? this.inputKey;
     }
     /** @ignore */
-    async _call(values) {
+    async _call(values, runManager) {
         if (!(this.inputKey in values)) {
             throw new Error(`Document key ${this.inputKey} not found.`);
         }
@@ -52,7 +52,7 @@ class StuffDocumentsChain extends base_js_1.BaseChain {
         const result = await this.llmChain.call({
             ...rest,
             [this.documentVariableName]: text,
-        });
+        }, runManager?.getChild());
         return result;
     }
     _chainType() {
@@ -87,7 +87,7 @@ class MapReduceDocumentsChain extends base_js_1.BaseChain {
         return this.combineDocumentChain.outputKeys;
     }
     constructor(fields) {
-        super();
+        super(fields);
         Object.defineProperty(this, "llmChain", {
             enumerable: true,
             configurable: true,
@@ -140,7 +140,7 @@ class MapReduceDocumentsChain extends base_js_1.BaseChain {
         this.maxIterations = fields.maxIterations ?? this.maxIterations;
     }
     /** @ignore */
-    async _call(values) {
+    async _call(values, runManager) {
         if (!(this.inputKey in values)) {
             throw new Error(`Document key ${this.inputKey} not found.`);
         }
@@ -161,14 +161,14 @@ class MapReduceDocumentsChain extends base_js_1.BaseChain {
             if (canSkipMapStep && withinTokenLimit) {
                 break;
             }
-            const results = await this.llmChain.apply(inputs);
+            const results = await this.llmChain.apply(inputs, runManager ? [runManager.getChild()] : undefined);
             const { outputKey } = this.llmChain;
             currentDocs = results.map((r) => ({
                 pageContent: r[outputKey],
             }));
         }
         const newInputs = { input_documents: currentDocs, ...rest };
-        const result = await this.combineDocumentChain.call(newInputs);
+        const result = await this.combineDocumentChain.call(newInputs, runManager?.getChild());
         return result;
     }
     _chainType() {
@@ -214,7 +214,7 @@ class RefineDocumentsChain extends base_js_1.BaseChain {
         return [this.outputKey];
     }
     constructor(fields) {
-        super();
+        super(fields);
         Object.defineProperty(this, "llmChain", {
             enumerable: true,
             configurable: true,
@@ -304,19 +304,19 @@ class RefineDocumentsChain extends base_js_1.BaseChain {
         return inputs;
     }
     /** @ignore */
-    async _call(values) {
+    async _call(values, runManager) {
         if (!(this.inputKey in values)) {
             throw new Error(`Document key ${this.inputKey} not found.`);
         }
         const { [this.inputKey]: docs, ...rest } = values;
         const currentDocs = docs;
         const initialInputs = await this._constructInitialInputs(currentDocs[0], rest);
-        let res = await this.llmChain.predict({ ...initialInputs });
+        let res = await this.llmChain.predict({ ...initialInputs }, runManager?.getChild());
         const refineSteps = [res];
         for (let i = 1; i < currentDocs.length; i += 1) {
             const refineInputs = await this._constructRefineInputs(currentDocs[i], res);
             const inputs = { ...refineInputs, ...rest };
-            res = await this.refineLLMChain.predict({ ...inputs });
+            res = await this.refineLLMChain.predict({ ...inputs }, runManager?.getChild());
             refineSteps.push(res);
         }
         return { [this.outputKey]: res };