langchain 0.0.52 → 0.0.53

package/dist/agents/agent.cjs

@@ -47,9 +47,15 @@ class BaseAgent {
 }
 exports.BaseAgent = BaseAgent;
 class BaseSingleActionAgent extends BaseAgent {
+    _agentActionType() {
+        return "single";
+    }
 }
 exports.BaseSingleActionAgent = BaseSingleActionAgent;
 class BaseMultiActionAgent extends BaseAgent {
+    _agentActionType() {
+        return "multi";
+    }
 }
 exports.BaseMultiActionAgent = BaseMultiActionAgent;
 class LLMSingleActionAgent extends BaseSingleActionAgent {

package/dist/agents/agent.d.ts

@@ -12,6 +12,10 @@ export declare abstract class BaseAgent {
      * Return the string type key uniquely identifying this class of agent.
      */
     _agentType(): string;
+    /**
+     * Return the string type key uniquely identifying multi or single action agents.
+     */
+    abstract _agentActionType(): string;
     /**
      * Return response when agent has been stopped due to max iterations
      */
@@ -22,6 +26,7 @@ export declare abstract class BaseAgent {
     prepareForOutput(_returnValues: AgentFinish["returnValues"], _steps: AgentStep[]): Promise<AgentFinish["returnValues"]>;
 }
 export declare abstract class BaseSingleActionAgent extends BaseAgent {
+    _agentActionType(): string;
     /**
      * Decide what to do, given some input.
      *
@@ -33,6 +38,7 @@ export declare abstract class BaseSingleActionAgent extends BaseAgent {
     abstract plan(steps: AgentStep[], inputs: ChainValues): Promise<AgentAction | AgentFinish>;
 }
 export declare abstract class BaseMultiActionAgent extends BaseAgent {
+    _agentActionType(): string;
     /**
      * Decide what to do, given some input.
      *

package/dist/agents/agent.js

@@ -43,8 +43,14 @@ export class BaseAgent {
     }
 }
 export class BaseSingleActionAgent extends BaseAgent {
+    _agentActionType() {
+        return "single";
+    }
 }
 export class BaseMultiActionAgent extends BaseAgent {
+    _agentActionType() {
+        return "multi";
+    }
 }
 export class LLMSingleActionAgent extends BaseSingleActionAgent {
     constructor(input) {

package/dist/agents/agent_toolkits/zapier/zapier.cjs

@@ -5,7 +5,7 @@ const base_js_1 = require("../base.cjs");
 const zapier_js_1 = require("../../../tools/zapier.cjs");
 class ZapierToolKit extends base_js_1.Toolkit {
     constructor() {
-        super();
+        super(...arguments);
         Object.defineProperty(this, "tools", {
             enumerable: true,
             configurable: true,

package/dist/agents/agent_toolkits/zapier/zapier.d.ts

@@ -3,6 +3,5 @@ import { Tool } from "../../../tools/base.js";
 import { ZapierNLAWrapper } from "../../../tools/zapier.js";
 export declare class ZapierToolKit extends Toolkit {
     tools: Tool[];
-    protected constructor();
     static fromZapierNLAWrapper(zapierNLAWrapper: ZapierNLAWrapper): Promise<ZapierToolKit>;
 }

package/dist/agents/agent_toolkits/zapier/zapier.js

@@ -2,7 +2,7 @@ import { Toolkit } from "../base.js";
 import { ZapierNLARunAction } from "../../../tools/zapier.js";
 export class ZapierToolKit extends Toolkit {
     constructor() {
-        super();
+        super(...arguments);
         Object.defineProperty(this, "tools", {
             enumerable: true,
             configurable: true,

package/dist/agents/executor.cjs

@@ -2,7 +2,6 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AgentExecutor = void 0;
 const base_js_1 = require("../chains/base.cjs");
-const agent_js_1 = require("./agent.cjs");
 /**
  * A chain managing an agent using tools.
  * @augments BaseChain
@@ -45,8 +44,7 @@ class AgentExecutor extends base_js_1.BaseChain {
         });
         this.agent = input.agent;
         this.tools = input.tools;
-        // eslint-disable-next-line no-instanceof/no-instanceof
-        if (this.agent instanceof agent_js_1.BaseMultiActionAgent) {
+        if (this.agent._agentActionType() === "multi") {
             for (const tool of this.tools) {
                 if (tool.returnDirect) {
                     throw new Error(`Tool with return direct ${tool.name} not supported for multi-action agent.`);

package/dist/agents/executor.js

@@ -1,5 +1,4 @@
 import { BaseChain } from "../chains/base.js";
-import { BaseMultiActionAgent } from "./agent.js";
 /**
  * A chain managing an agent using tools.
  * @augments BaseChain
@@ -42,8 +41,7 @@ export class AgentExecutor extends BaseChain {
         });
         this.agent = input.agent;
         this.tools = input.tools;
-        // eslint-disable-next-line no-instanceof/no-instanceof
-        if (this.agent instanceof BaseMultiActionAgent) {
+        if (this.agent._agentActionType() === "multi") {
             for (const tool of this.tools) {
                 if (tool.returnDirect) {
                     throw new Error(`Tool with return direct ${tool.name} not supported for multi-action agent.`);

package/dist/agents/types.d.ts

@@ -12,7 +12,6 @@ export type StoppingMethod = "force" | "generate";
 export type SerializedAgentT<TType extends string = string, FromLLMInput extends Record<string, unknown> = Record<string, unknown>, ConstructorInput extends AgentInput = AgentInput> = {
     _type: TType;
     llm_chain?: SerializedLLMChain;
-    llm_chain_path?: string;
 } & (({
     load_from_llm_and_tools: true;
 } & FromLLMInput) | ({

package/dist/base_language/index.d.ts

@@ -27,8 +27,8 @@ export declare abstract class BaseLanguageModel implements BaseLanguageModelPara
      * The async caller should be used by subclasses to make any async calls,
      * which will thus benefit from the concurrency and retry logic.
      */
-    protected caller: AsyncCaller;
-    protected constructor(params: BaseLanguageModelParams);
+    caller: AsyncCaller;
+    constructor(params: BaseLanguageModelParams);
     abstract generatePrompt(promptValues: BasePromptValue[], stop?: string[]): Promise<LLMResult>;
     abstract _modelType(): string;
     abstract _llmType(): string;

package/dist/chains/combine_docs_chain.cjs

@@ -1,8 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.MapReduceDocumentsChain = exports.StuffDocumentsChain = void 0;
+exports.RefineDocumentsChain = exports.MapReduceDocumentsChain = exports.StuffDocumentsChain = void 0;
 const base_js_1 = require("./base.cjs");
 const llm_chain_js_1 = require("./llm_chain.cjs");
+const prompt_js_1 = require("../prompts/prompt.cjs");
 /**
  * Chain that combines documents by stuffing into context.
  * @augments BaseChain
@@ -77,7 +78,7 @@ class StuffDocumentsChain extends base_js_1.BaseChain {
 }
 exports.StuffDocumentsChain = StuffDocumentsChain;
 /**
- * Chain that combines documents by stuffing into context.
+ * Combine documents by mapping a chain over them, then combining results.
  * @augments BaseChain
  * @augments StuffDocumentsChainInput
  */
@@ -200,3 +201,148 @@ class MapReduceDocumentsChain extends base_js_1.BaseChain {
     }
 }
 exports.MapReduceDocumentsChain = MapReduceDocumentsChain;
+/**
+ * Combine documents by doing a first pass and then refining on more documents.
+ * @augments BaseChain
+ * @augments RefineDocumentsChainInput
+ */
+class RefineDocumentsChain extends base_js_1.BaseChain {
+    get defaultDocumentPrompt() {
+        return new prompt_js_1.PromptTemplate({
+            inputVariables: ["page_content"],
+            template: "{page_content}",
+        });
+    }
+    get inputKeys() {
+        return [this.inputKey, ...this.refineLLMChain.inputKeys];
+    }
+    constructor(fields) {
+        super();
+        Object.defineProperty(this, "llmChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "inputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "input_documents"
+        });
+        Object.defineProperty(this, "outputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "output_text"
+        });
+        Object.defineProperty(this, "documentVariableName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "context"
+        });
+        Object.defineProperty(this, "initialResponseName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "existing_answer"
+        });
+        Object.defineProperty(this, "refineLLMChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "documentPrompt", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: this.defaultDocumentPrompt
+        });
+        this.llmChain = fields.llmChain;
+        this.refineLLMChain = fields.refineLLMChain;
+        this.documentVariableName =
+            fields.documentVariableName ?? this.documentVariableName;
+        this.inputKey = fields.inputKey ?? this.inputKey;
+        this.documentPrompt = fields.documentPrompt ?? this.documentPrompt;
+        this.initialResponseName =
+            fields.initialResponseName ?? this.initialResponseName;
+    }
+    _constructInitialInputs(doc, rest) {
+        const baseInfo = {
+            page_content: doc.pageContent,
+            ...doc.metadata,
+        };
+        const documentInfo = {};
+        this.documentPrompt.inputVariables.forEach((value) => {
+            documentInfo[value] = baseInfo[value];
+        });
+        const baseInputs = {
+            [this.documentVariableName]: this.documentPrompt.format({
+                ...documentInfo,
+            }),
+        };
+        const inputs = { ...baseInputs, ...rest };
+        return inputs;
+    }
+    _constructRefineInputs(doc, res) {
+        const baseInfo = {
+            page_content: doc.pageContent,
+            ...doc.metadata,
+        };
+        const documentInfo = {};
+        this.documentPrompt.inputVariables.forEach((value) => {
+            documentInfo[value] = baseInfo[value];
+        });
+        const baseInputs = {
+            [this.documentVariableName]: this.documentPrompt.format({
+                ...documentInfo,
+            }),
+        };
+        const inputs = { [this.initialResponseName]: res, ...baseInputs };
+        return inputs;
+    }
+    async _call(values) {
+        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 = this._constructInitialInputs(currentDocs[0], rest);
+        let res = await this.llmChain.predict({ ...initialInputs });
+        const refineSteps = [res];
+        for (let i = 1; i < currentDocs.length; i += 1) {
+            const refineInputs = this._constructRefineInputs(currentDocs[i], res);
+            const inputs = { ...refineInputs, ...rest };
+            res = await this.refineLLMChain.predict({ ...inputs });
+            refineSteps.push(res);
+        }
+        return { [this.outputKey]: res };
+    }
+    _chainType() {
+        return "refine_documents_chain";
+    }
+    static async deserialize(data) {
+        const SerializedLLMChain = data.llm_chain;
+        if (!SerializedLLMChain) {
+            throw new Error("Missing llm_chain");
+        }
+        const SerializedRefineDocumentChain = data.refine_llm_chain;
+        if (!SerializedRefineDocumentChain) {
+            throw new Error("Missing refine_llm_chain");
+        }
+        return new RefineDocumentsChain({
+            llmChain: await llm_chain_js_1.LLMChain.deserialize(SerializedLLMChain),
+            refineLLMChain: await llm_chain_js_1.LLMChain.deserialize(SerializedRefineDocumentChain),
+        });
+    }
+    serialize() {
+        return {
+            _type: this._chainType(),
+            llm_chain: this.llmChain.serialize(),
+            refine_llm_chain: this.refineLLMChain.serialize(),
+        };
+    }
+}
+exports.RefineDocumentsChain = RefineDocumentsChain;

package/dist/chains/combine_docs_chain.d.ts

@@ -1,7 +1,9 @@
-import type { SerializedStuffDocumentsChain, SerializedMapReduceDocumentsChain } from "./serde.js";
+import type { SerializedStuffDocumentsChain, SerializedMapReduceDocumentsChain, SerializedRefineDocumentsChain } from "./serde.js";
 import { BaseChain } from "./base.js";
 import { LLMChain } from "./llm_chain.js";
+import { Document } from "../document.js";
 import { ChainValues } from "../schema/index.js";
+import { BasePromptTemplate } from "../prompts/base.js";
 export interface StuffDocumentsChainInput {
     /** LLM Wrapper to use after formatting documents */
     llmChain: LLMChain;
@@ -38,7 +40,7 @@ export interface MapReduceDocumentsChainInput extends StuffDocumentsChainInput {
     combineDocumentsChain: BaseChain;
 }
 /**
- * Chain that combines documents by stuffing into context.
+ * Combine documents by mapping a chain over them, then combining results.
  * @augments BaseChain
  * @augments StuffDocumentsChainInput
  */
@@ -67,3 +69,42 @@ export declare class MapReduceDocumentsChain extends BaseChain implements StuffD
     static deserialize(data: SerializedMapReduceDocumentsChain): Promise<MapReduceDocumentsChain>;
     serialize(): SerializedMapReduceDocumentsChain;
 }
+export interface RefineDocumentsChainInput extends StuffDocumentsChainInput {
+    refineLLMChain: LLMChain;
+    documentPrompt: BasePromptTemplate;
+}
+/**
+ * Combine documents by doing a first pass and then refining on more documents.
+ * @augments BaseChain
+ * @augments RefineDocumentsChainInput
+ */
+export declare class RefineDocumentsChain extends BaseChain implements RefineDocumentsChainInput {
+    llmChain: LLMChain;
+    inputKey: string;
+    outputKey: string;
+    documentVariableName: string;
+    initialResponseName: string;
+    refineLLMChain: LLMChain;
+    get defaultDocumentPrompt(): BasePromptTemplate;
+    documentPrompt: BasePromptTemplate;
+    get inputKeys(): string[];
+    constructor(fields: {
+        llmChain: LLMChain;
+        refineLLMChain: LLMChain;
+        inputKey?: string;
+        outputKey?: string;
+        documentVariableName?: string;
+        documentPrompt?: BasePromptTemplate;
+        initialResponseName?: string;
+    });
+    _constructInitialInputs(doc: Document, rest: Record<string, unknown>): {
+        [x: string]: unknown;
+    };
+    _constructRefineInputs(doc: Document, res: string): {
+        [x: string]: unknown;
+    };
+    _call(values: ChainValues): Promise<ChainValues>;
+    _chainType(): "refine_documents_chain";
+    static deserialize(data: SerializedRefineDocumentsChain): Promise<RefineDocumentsChain>;
+    serialize(): SerializedRefineDocumentsChain;
+}

package/dist/chains/combine_docs_chain.js

@@ -1,5 +1,6 @@
 import { BaseChain } from "./base.js";
 import { LLMChain } from "./llm_chain.js";
+import { PromptTemplate } from "../prompts/prompt.js";
 /**
  * Chain that combines documents by stuffing into context.
  * @augments BaseChain
@@ -73,7 +74,7 @@ export class StuffDocumentsChain extends BaseChain {
     }
 }
 /**
- * Chain that combines documents by stuffing into context.
+ * Combine documents by mapping a chain over them, then combining results.
  * @augments BaseChain
  * @augments StuffDocumentsChainInput
  */
@@ -195,3 +196,147 @@ export class MapReduceDocumentsChain extends BaseChain {
         };
     }
 }
+/**
+ * Combine documents by doing a first pass and then refining on more documents.
+ * @augments BaseChain
+ * @augments RefineDocumentsChainInput
+ */
+export class RefineDocumentsChain extends BaseChain {
+    get defaultDocumentPrompt() {
+        return new PromptTemplate({
+            inputVariables: ["page_content"],
+            template: "{page_content}",
+        });
+    }
+    get inputKeys() {
+        return [this.inputKey, ...this.refineLLMChain.inputKeys];
+    }
+    constructor(fields) {
+        super();
+        Object.defineProperty(this, "llmChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "inputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "input_documents"
+        });
+        Object.defineProperty(this, "outputKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "output_text"
+        });
+        Object.defineProperty(this, "documentVariableName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "context"
+        });
+        Object.defineProperty(this, "initialResponseName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "existing_answer"
+        });
+        Object.defineProperty(this, "refineLLMChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "documentPrompt", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: this.defaultDocumentPrompt
+        });
+        this.llmChain = fields.llmChain;
+        this.refineLLMChain = fields.refineLLMChain;
+        this.documentVariableName =
+            fields.documentVariableName ?? this.documentVariableName;
+        this.inputKey = fields.inputKey ?? this.inputKey;
+        this.documentPrompt = fields.documentPrompt ?? this.documentPrompt;
+        this.initialResponseName =
+            fields.initialResponseName ?? this.initialResponseName;
+    }
+    _constructInitialInputs(doc, rest) {
+        const baseInfo = {
+            page_content: doc.pageContent,
+            ...doc.metadata,
+        };
+        const documentInfo = {};
+        this.documentPrompt.inputVariables.forEach((value) => {
+            documentInfo[value] = baseInfo[value];
+        });
+        const baseInputs = {
+            [this.documentVariableName]: this.documentPrompt.format({
+                ...documentInfo,
+            }),
+        };
+        const inputs = { ...baseInputs, ...rest };
+        return inputs;
+    }
+    _constructRefineInputs(doc, res) {
+        const baseInfo = {
+            page_content: doc.pageContent,
+            ...doc.metadata,
+        };
+        const documentInfo = {};
+        this.documentPrompt.inputVariables.forEach((value) => {
+            documentInfo[value] = baseInfo[value];
+        });
+        const baseInputs = {
+            [this.documentVariableName]: this.documentPrompt.format({
+                ...documentInfo,
+            }),
+        };
+        const inputs = { [this.initialResponseName]: res, ...baseInputs };
+        return inputs;
+    }
+    async _call(values) {
+        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 = this._constructInitialInputs(currentDocs[0], rest);
+        let res = await this.llmChain.predict({ ...initialInputs });
+        const refineSteps = [res];
+        for (let i = 1; i < currentDocs.length; i += 1) {
+            const refineInputs = this._constructRefineInputs(currentDocs[i], res);
+            const inputs = { ...refineInputs, ...rest };
+            res = await this.refineLLMChain.predict({ ...inputs });
+            refineSteps.push(res);
+        }
+        return { [this.outputKey]: res };
+    }
+    _chainType() {
+        return "refine_documents_chain";
+    }
+    static async deserialize(data) {
+        const SerializedLLMChain = data.llm_chain;
+        if (!SerializedLLMChain) {
+            throw new Error("Missing llm_chain");
+        }
+        const SerializedRefineDocumentChain = data.refine_llm_chain;
+        if (!SerializedRefineDocumentChain) {
+            throw new Error("Missing refine_llm_chain");
+        }
+        return new RefineDocumentsChain({
+            llmChain: await LLMChain.deserialize(SerializedLLMChain),
+            refineLLMChain: await LLMChain.deserialize(SerializedRefineDocumentChain),
+        });
+    }
+    serialize() {
+        return {
+            _type: this._chainType(),
+            llm_chain: this.llmChain.serialize(),
+            refine_llm_chain: this.refineLLMChain.serialize(),
+        };
+    }
+}

package/dist/chains/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.RetrievalQAChain = exports.ConversationalRetrievalQAChain = exports.SqlDatabaseChain = exports.loadSummarizationChain = exports.loadQAMapReduceChain = exports.loadQAStuffChain = exports.loadQAChain = exports.VectorDBQAChain = exports.AnalyzeDocumentChain = exports.ChatVectorDBQAChain = exports.MapReduceDocumentsChain = exports.StuffDocumentsChain = 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.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");
@@ -9,6 +9,7 @@ Object.defineProperty(exports, "ConversationChain", { enumerable: true, get: fun
 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; } });
+Object.defineProperty(exports, "RefineDocumentsChain", { enumerable: true, get: function () { return combine_docs_chain_js_1.RefineDocumentsChain; } });
 var chat_vector_db_chain_js_1 = require("./chat_vector_db_chain.cjs");
 Object.defineProperty(exports, "ChatVectorDBQAChain", { enumerable: true, get: function () { return chat_vector_db_chain_js_1.ChatVectorDBQAChain; } });
 var analyze_documents_chain_js_1 = require("./analyze_documents_chain.cjs");
@@ -19,6 +20,7 @@ var load_js_1 = require("./question_answering/load.cjs");
 Object.defineProperty(exports, "loadQAChain", { enumerable: true, get: function () { return load_js_1.loadQAChain; } });
 Object.defineProperty(exports, "loadQAStuffChain", { enumerable: true, get: function () { return load_js_1.loadQAStuffChain; } });
 Object.defineProperty(exports, "loadQAMapReduceChain", { enumerable: true, get: function () { return load_js_1.loadQAMapReduceChain; } });
+Object.defineProperty(exports, "loadQARefineChain", { enumerable: true, get: function () { return load_js_1.loadQARefineChain; } });
 var load_js_2 = require("./summarization/load.cjs");
 Object.defineProperty(exports, "loadSummarizationChain", { enumerable: true, get: function () { return load_js_2.loadSummarizationChain; } });
 var sql_db_chain_js_1 = require("./sql_db/sql_db_chain.cjs");

package/dist/chains/index.d.ts

@@ -1,12 +1,12 @@
 export { BaseChain, ChainInputs } from "./base.js";
 export { LLMChain, ConversationChain } from "./llm_chain.js";
-export { StuffDocumentsChain, MapReduceDocumentsChain, } from "./combine_docs_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";
 export { VectorDBQAChain } from "./vector_db_qa.js";
-export { loadQAChain, loadQAStuffChain, loadQAMapReduceChain, } from "./question_answering/load.js";
+export { loadQAChain, loadQAStuffChain, loadQAMapReduceChain, loadQARefineChain, } from "./question_answering/load.js";
 export { loadSummarizationChain } from "./summarization/load.js";
 export { SqlDatabaseChain } from "./sql_db/sql_db_chain.js";
 export { ConversationalRetrievalQAChain } from "./conversational_retrieval_chain.js";
 export { RetrievalQAChain } from "./retrieval_qa.js";
-export { SerializedLLMChain, SerializedSqlDatabaseChain, SerializedAnalyzeDocumentChain, SerializedBaseChain, SerializedChatVectorDBQAChain, SerializedMapReduceDocumentsChain, SerializedStuffDocumentsChain, SerializedVectorDBQAChain, } from "./serde.js";
+export { SerializedLLMChain, SerializedSqlDatabaseChain, SerializedAnalyzeDocumentChain, SerializedBaseChain, SerializedChatVectorDBQAChain, SerializedMapReduceDocumentsChain, SerializedStuffDocumentsChain, SerializedVectorDBQAChain, SerializedRefineDocumentsChain, } from "./serde.js";

package/dist/chains/index.js

@@ -1,10 +1,10 @@
 export { BaseChain } from "./base.js";
 export { LLMChain, ConversationChain } from "./llm_chain.js";
-export { StuffDocumentsChain, MapReduceDocumentsChain, } from "./combine_docs_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";
 export { VectorDBQAChain } from "./vector_db_qa.js";
-export { loadQAChain, loadQAStuffChain, loadQAMapReduceChain, } from "./question_answering/load.js";
+export { loadQAChain, loadQAStuffChain, loadQAMapReduceChain, loadQARefineChain, } from "./question_answering/load.js";
 export { loadSummarizationChain } from "./summarization/load.js";
 export { SqlDatabaseChain } from "./sql_db/sql_db_chain.js";
 export { ConversationalRetrievalQAChain } from "./conversational_retrieval_chain.js";

package/dist/chains/question_answering/load.cjs

@@ -1,10 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.loadQAMapReduceChain = exports.loadQAStuffChain = exports.loadQAChain = void 0;
+exports.loadQARefineChain = exports.loadQAMapReduceChain = exports.loadQAStuffChain = exports.loadQAChain = void 0;
 const llm_chain_js_1 = require("../llm_chain.cjs");
 const combine_docs_chain_js_1 = require("../combine_docs_chain.cjs");
 const stuff_prompts_js_1 = require("./stuff_prompts.cjs");
 const map_reduce_prompts_js_1 = require("./map_reduce_prompts.cjs");
+const refine_prompts_js_1 = require("./refine_prompts.cjs");
 const loadQAChain = (llm, params = {}) => {
     const { prompt = stuff_prompts_js_1.DEFAULT_QA_PROMPT, combineMapPrompt = map_reduce_prompts_js_1.DEFAULT_COMBINE_QA_PROMPT, combinePrompt = map_reduce_prompts_js_1.COMBINE_PROMPT, type = "stuff", } = params;
     if (type === "stuff") {
@@ -25,6 +26,16 @@ const loadQAChain = (llm, params = {}) => {
         });
         return chain;
     }
+    if (type === "refine") {
+        const { questionPrompt = refine_prompts_js_1.QUESTION_PROMPT_SELECTOR.getPrompt(llm), refinePrompt = refine_prompts_js_1.REFINE_PROMPT_SELECTOR.getPrompt(llm), } = params;
+        const llmChain = new llm_chain_js_1.LLMChain({ prompt: questionPrompt, llm });
+        const refineLLMChain = new llm_chain_js_1.LLMChain({ prompt: refinePrompt, llm });
+        const chain = new combine_docs_chain_js_1.RefineDocumentsChain({
+            llmChain,
+            refineLLMChain,
+        });
+        return chain;
+    }
     throw new Error(`Invalid _type: ${type}`);
 };
 exports.loadQAChain = loadQAChain;
@@ -50,3 +61,14 @@ const loadQAMapReduceChain = (llm, params = {}) => {
     return chain;
 };
 exports.loadQAMapReduceChain = loadQAMapReduceChain;
+const loadQARefineChain = (llm, params = {}) => {
+    const { questionPrompt = refine_prompts_js_1.QUESTION_PROMPT_SELECTOR.getPrompt(llm), refinePrompt = refine_prompts_js_1.REFINE_PROMPT_SELECTOR.getPrompt(llm), } = params;
+    const llmChain = new llm_chain_js_1.LLMChain({ prompt: questionPrompt, llm });
+    const refineLLMChain = new llm_chain_js_1.LLMChain({ prompt: refinePrompt, llm });
+    const chain = new combine_docs_chain_js_1.RefineDocumentsChain({
+        llmChain,
+        refineLLMChain,
+    });
+    return chain;
+};
+exports.loadQARefineChain = loadQARefineChain;

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

@@ -1,13 +1,15 @@
 import { BasePromptTemplate } from "../../prompts/base.js";
-import { StuffDocumentsChain, MapReduceDocumentsChain } from "../combine_docs_chain.js";
+import { StuffDocumentsChain, MapReduceDocumentsChain, RefineDocumentsChain } from "../combine_docs_chain.js";
 import { BaseLanguageModel } from "../../base_language/index.js";
 interface qaChainParams {
     prompt?: BasePromptTemplate;
     combineMapPrompt?: BasePromptTemplate;
     combinePrompt?: BasePromptTemplate;
+    questionPrompt?: BasePromptTemplate;
+    refinePrompt?: BasePromptTemplate;
     type?: string;
 }
-export declare const loadQAChain: (llm: BaseLanguageModel, params?: qaChainParams) => StuffDocumentsChain | MapReduceDocumentsChain;
+export declare const loadQAChain: (llm: BaseLanguageModel, params?: qaChainParams) => StuffDocumentsChain | MapReduceDocumentsChain | RefineDocumentsChain;
 interface StuffQAChainParams {
     prompt?: BasePromptTemplate;
 }
@@ -17,4 +19,9 @@ interface MapReduceQAChainParams {
     combinePrompt?: BasePromptTemplate;
 }
 export declare const loadQAMapReduceChain: (llm: BaseLanguageModel, params?: MapReduceQAChainParams) => MapReduceDocumentsChain;
+interface RefineQAChainParams {
+    questionPrompt?: BasePromptTemplate;
+    refinePrompt?: BasePromptTemplate;
+}
+export declare const loadQARefineChain: (llm: BaseLanguageModel, params?: RefineQAChainParams) => RefineDocumentsChain;
 export {};