langchain 0.0.63 → 0.0.64

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/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/chains/analyze_documents_chain.cjs

@@ -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

@@ -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

@@ -82,8 +82,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.js

@@ -79,8 +79,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

@@ -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

@@ -4,6 +4,7 @@ import { SerializedChatVectorDBQAChain } from "./serde.js";
 import { ChainValues } from "../schema/index.js";
 import { BaseChain } from "./base.js";
 import { LLMChain } from "./llm_chain.js";
+import { CallbackManagerForChainRun } from "../callbacks/manager.js";
 export type LoadValues = Record<string, any>;
 export interface ChatVectorDBQAChainInput {
     vectorstore: VectorStore;
@@ -34,7 +35,7 @@ export declare class ChatVectorDBQAChain extends BaseChain implements ChatVector
         returnSourceDocuments?: boolean;
     });
     /** @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

@@ -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

@@ -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() {
@@ -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() {
@@ -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 };

package/dist/chains/combine_docs_chain.d.ts

@@ -4,6 +4,7 @@ import { LLMChain } from "./llm_chain.js";
 import { Document } from "../document.js";
 import { ChainValues } from "../schema/index.js";
 import { BasePromptTemplate } from "../prompts/base.js";
+import { CallbackManagerForChainRun } from "../callbacks/manager.js";
 export interface StuffDocumentsChainInput {
     /** LLM Wrapper to use after formatting documents */
     llmChain: LLMChain;
@@ -24,7 +25,7 @@ export declare class StuffDocumentsChain extends BaseChain implements StuffDocum
     get outputKeys(): string[];
     constructor(fields: StuffDocumentsChainInput);
     /** @ignore */
-    _call(values: ChainValues): Promise<ChainValues>;
+    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
     _chainType(): "stuff_documents_chain";
     static deserialize(data: SerializedStuffDocumentsChain): Promise<StuffDocumentsChain>;
     serialize(): SerializedStuffDocumentsChain;
@@ -52,7 +53,7 @@ export declare class MapReduceDocumentsChain extends BaseChain implements MapRed
     combineDocumentChain: BaseChain;
     constructor(fields: MapReduceDocumentsChainInput);
     /** @ignore */
-    _call(values: ChainValues): Promise<ChainValues>;
+    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
     _chainType(): "map_reduce_documents_chain";
     static deserialize(data: SerializedMapReduceDocumentsChain): Promise<MapReduceDocumentsChain>;
     serialize(): SerializedMapReduceDocumentsChain;
@@ -90,7 +91,7 @@ export declare class RefineDocumentsChain extends BaseChain implements RefineDoc
         [x: string]: unknown;
     }>;
     /** @ignore */
-    _call(values: ChainValues): Promise<ChainValues>;
+    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
     _chainType(): "refine_documents_chain";
     static deserialize(data: SerializedRefineDocumentsChain): Promise<RefineDocumentsChain>;
     serialize(): SerializedRefineDocumentsChain;

package/dist/chains/combine_docs_chain.js

@@ -39,7 +39,7 @@ export class StuffDocumentsChain extends 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.`);
         }
@@ -49,7 +49,7 @@ export class StuffDocumentsChain extends BaseChain {
         const result = await this.llmChain.call({
             ...rest,
             [this.documentVariableName]: text,
-        });
+        }, runManager?.getChild());
         return result;
     }
     _chainType() {
@@ -136,7 +136,7 @@ export class MapReduceDocumentsChain extends 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.`);
         }
@@ -157,14 +157,14 @@ export class MapReduceDocumentsChain extends 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() {
@@ -299,19 +299,19 @@ export class RefineDocumentsChain extends 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 };

package/dist/chains/conversational_retrieval_chain.cjs

@@ -70,7 +70,7 @@ class ConversationalRetrievalQAChain 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.`);
         }
@@ -84,7 +84,7 @@ class ConversationalRetrievalQAChain extends base_js_1.BaseChain {
             const result = await this.questionGeneratorChain.call({
                 question,
                 chat_history: chatHistory,
-            });
+            }, runManager?.getChild());
             const keys = Object.keys(result);
             if (keys.length === 1) {
                 newQuestion = result[keys[0]];
@@ -99,7 +99,7 @@ class ConversationalRetrievalQAChain 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/conversational_retrieval_chain.d.ts

@@ -3,6 +3,7 @@ import { SerializedChatVectorDBQAChain } from "./serde.js";
 import { ChainValues, BaseRetriever } from "../schema/index.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 ConversationalRetrievalQAChainInput extends Omit<ChainInputs, "memory"> {
     retriever: BaseRetriever;
@@ -22,7 +23,7 @@ export declare class ConversationalRetrievalQAChain extends BaseChain implements
     returnSourceDocuments: boolean;
     constructor(fields: ConversationalRetrievalQAChainInput);
     /** @ignore */
-    _call(values: ChainValues): Promise<ChainValues>;
+    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
     _chainType(): string;
     static deserialize(_data: SerializedChatVectorDBQAChain, _values: LoadValues): Promise<ConversationalRetrievalQAChain>;
     serialize(): SerializedChatVectorDBQAChain;

package/dist/chains/conversational_retrieval_chain.js

@@ -67,7 +67,7 @@ export class ConversationalRetrievalQAChain 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.`);
         }
@@ -81,7 +81,7 @@ export class ConversationalRetrievalQAChain extends BaseChain {
             const result = await this.questionGeneratorChain.call({
                 question,
                 chat_history: chatHistory,
-            });
+            }, runManager?.getChild());
             const keys = Object.keys(result);
             if (keys.length === 1) {
                 newQuestion = result[keys[0]];
@@ -96,7 +96,7 @@ export class ConversationalRetrievalQAChain 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/index.cjs

@@ -1,14 +1,15 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-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.ConversationChain = exports.LLMChain = exports.BaseChain = void 0;
+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");
 Object.defineProperty(exports, "LLMChain", { enumerable: true, get: function () { return llm_chain_js_1.LLMChain; } });
 var conversation_js_1 = require("./conversation.cjs");
 Object.defineProperty(exports, "ConversationChain", { enumerable: true, get: function () { return conversation_js_1.ConversationChain; } });
-var simple_sequential_chain_js_1 = require("./simple_sequential_chain.cjs");
-Object.defineProperty(exports, "SimpleSequentialChain", { enumerable: true, get: function () { return simple_sequential_chain_js_1.SimpleSequentialChain; } });
+var sequential_chain_js_1 = require("./sequential_chain.cjs");
+Object.defineProperty(exports, "SequentialChain", { enumerable: true, get: function () { return sequential_chain_js_1.SequentialChain; } });
+Object.defineProperty(exports, "SimpleSequentialChain", { enumerable: true, get: function () { return sequential_chain_js_1.SimpleSequentialChain; } });
 var combine_docs_chain_js_1 = require("./combine_docs_chain.cjs");
 Object.defineProperty(exports, "StuffDocumentsChain", { enumerable: true, get: function () { return combine_docs_chain_js_1.StuffDocumentsChain; } });
 Object.defineProperty(exports, "MapReduceDocumentsChain", { enumerable: true, get: function () { return combine_docs_chain_js_1.MapReduceDocumentsChain; } });

package/dist/chains/index.d.ts

@@ -1,7 +1,7 @@
 export { BaseChain, ChainInputs } from "./base.js";
 export { LLMChain, LLMChainInput } from "./llm_chain.js";
 export { ConversationChain } from "./conversation.js";
-export { SimpleSequentialChain, SimpleSequentialChainInput, } from "./simple_sequential_chain.js";
+export { SequentialChain, SequentialChainInput, SimpleSequentialChain, SimpleSequentialChainInput, } from "./sequential_chain.js";
 export { StuffDocumentsChain, StuffDocumentsChainInput, MapReduceDocumentsChain, MapReduceDocumentsChainInput, RefineDocumentsChain, RefineDocumentsChainInput, } from "./combine_docs_chain.js";
 export { ChatVectorDBQAChain, ChatVectorDBQAChainInput, } from "./chat_vector_db_chain.js";
 export { AnalyzeDocumentChain, AnalyzeDocumentChainInput, } from "./analyze_documents_chain.js";
@@ -11,4 +11,4 @@ export { loadSummarizationChain } from "./summarization/load.js";
 export { SqlDatabaseChain, SqlDatabaseChainInput, } from "./sql_db/sql_db_chain.js";
 export { ConversationalRetrievalQAChain, ConversationalRetrievalQAChainInput, } from "./conversational_retrieval_chain.js";
 export { RetrievalQAChain, RetrievalQAChainInput } from "./retrieval_qa.js";
-export { SerializedLLMChain, SerializedSimpleSequentialChain, SerializedSqlDatabaseChain, SerializedAnalyzeDocumentChain, SerializedBaseChain, SerializedChatVectorDBQAChain, SerializedMapReduceDocumentsChain, SerializedStuffDocumentsChain, SerializedVectorDBQAChain, SerializedRefineDocumentsChain, } from "./serde.js";
+export { SerializedLLMChain, SerializedSequentialChain, SerializedSimpleSequentialChain, SerializedSqlDatabaseChain, SerializedAnalyzeDocumentChain, SerializedBaseChain, SerializedChatVectorDBQAChain, SerializedMapReduceDocumentsChain, SerializedStuffDocumentsChain, SerializedVectorDBQAChain, SerializedRefineDocumentsChain, } from "./serde.js";

package/dist/chains/index.js

@@ -1,7 +1,7 @@
 export { BaseChain } from "./base.js";
 export { LLMChain } from "./llm_chain.js";
 export { ConversationChain } from "./conversation.js";
-export { SimpleSequentialChain, } from "./simple_sequential_chain.js";
+export { SequentialChain, SimpleSequentialChain, } from "./sequential_chain.js";
 export { StuffDocumentsChain, MapReduceDocumentsChain, RefineDocumentsChain, } from "./combine_docs_chain.js";
 export { ChatVectorDBQAChain, } from "./chat_vector_db_chain.js";
 export { AnalyzeDocumentChain, } from "./analyze_documents_chain.js";

package/dist/chains/retrieval_qa.cjs

@@ -43,14 +43,14 @@ class RetrievalQAChain 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.`);
         }
         const question = values[this.inputKey];
         const docs = await this.retriever.getRelevantDocuments(question);
         const inputs = { question, input_documents: docs };
-        const result = await this.combineDocumentsChain.call(inputs);
+        const result = await this.combineDocumentsChain.call(inputs, runManager?.getChild());
         if (this.returnSourceDocuments) {
             return {
                 ...result,

package/dist/chains/retrieval_qa.d.ts

@@ -2,6 +2,7 @@ import { BaseChain, ChainInputs } from "./base.js";
 import { BaseLLM } from "../llms/base.js";
 import { SerializedVectorDBQAChain } from "./serde.js";
 import { ChainValues, BaseRetriever } from "../schema/index.js";
+import { CallbackManagerForChainRun } from "../callbacks/manager.js";
 export type LoadValues = Record<string, any>;
 export interface RetrievalQAChainInput extends Omit<ChainInputs, "memory"> {
     retriever: BaseRetriever;
@@ -18,7 +19,7 @@ export declare class RetrievalQAChain extends BaseChain implements RetrievalQACh
     returnSourceDocuments: boolean;
     constructor(fields: RetrievalQAChainInput);
     /** @ignore */
-    _call(values: ChainValues): Promise<ChainValues>;
+    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;
     _chainType(): "retrieval_qa";
     static deserialize(_data: SerializedVectorDBQAChain, _values: LoadValues): Promise<RetrievalQAChain>;
     serialize(): SerializedVectorDBQAChain;

package/dist/chains/retrieval_qa.js

@@ -40,14 +40,14 @@ export class RetrievalQAChain 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.`);
         }
         const question = values[this.inputKey];
         const docs = await this.retriever.getRelevantDocuments(question);
         const inputs = { question, input_documents: docs };
-        const result = await this.combineDocumentsChain.call(inputs);
+        const result = await this.combineDocumentsChain.call(inputs, runManager?.getChild());
         if (this.returnSourceDocuments) {
             return {
                 ...result,