@n8n/n8n-nodes-langchain 0.4.3 → 0.6.0

package/dist/nodes/chains/ChainSummarization/V1/ChainSummarizationV1.node.js

@@ -0,0 +1,216 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChainSummarizationV1 = void 0;
+const chains_1 = require("langchain/chains");
+const prompts_1 = require("langchain/prompts");
+const N8nJsonLoader_1 = require("../../../../utils/N8nJsonLoader");
+const N8nBinaryLoader_1 = require("../../../../utils/N8nBinaryLoader");
+const sharedFields_1 = require("../../../../utils/sharedFields");
+const prompt_1 = require("../prompt");
+class ChainSummarizationV1 {
+    constructor(baseDescription) {
+        this.description = {
+            ...baseDescription,
+            version: 1,
+            defaults: {
+                name: 'Summarization Chain',
+                color: '#909298',
+            },
+            inputs: [
+                "main",
+                {
+                    displayName: 'Model',
+                    maxConnections: 1,
+                    type: "ai_languageModel",
+                    required: true,
+                },
+                {
+                    displayName: 'Document',
+                    maxConnections: 1,
+                    type: "ai_document",
+                    required: true,
+                },
+            ],
+            outputs: ["main"],
+            credentials: [],
+            properties: [
+                (0, sharedFields_1.getTemplateNoticeField)(1951),
+                {
+                    displayName: 'Type',
+                    name: 'type',
+                    type: 'options',
+                    description: 'The type of summarization to run',
+                    default: 'map_reduce',
+                    options: [
+                        {
+                            name: 'Map Reduce (Recommended)',
+                            value: 'map_reduce',
+                            description: 'Summarize each document (or chunk) individually, then summarize those summaries',
+                        },
+                        {
+                            name: 'Refine',
+                            value: 'refine',
+                            description: 'Summarize the first document (or chunk). Then update that summary based on the next document (or chunk), and repeat.',
+                        },
+                        {
+                            name: 'Stuff',
+                            value: 'stuff',
+                            description: 'Pass all documents (or chunks) at once. Ideal for small datasets.',
+                        },
+                    ],
+                },
+                {
+                    displayName: 'Options',
+                    name: 'options',
+                    type: 'collection',
+                    default: {},
+                    placeholder: 'Add Option',
+                    options: [
+                        {
+                            displayName: 'Final Prompt to Combine',
+                            name: 'combineMapPrompt',
+                            type: 'string',
+                            hint: 'The prompt to combine individual summaries',
+                            displayOptions: {
+                                show: {
+                                    '/type': ['map_reduce'],
+                                },
+                            },
+                            default: prompt_1.DEFAULT_PROMPT_TEMPLATE,
+                            typeOptions: {
+                                rows: 6,
+                            },
+                        },
+                        {
+                            displayName: 'Individual Summary Prompt',
+                            name: 'prompt',
+                            type: 'string',
+                            default: prompt_1.DEFAULT_PROMPT_TEMPLATE,
+                            hint: 'The prompt to summarize an individual document (or chunk)',
+                            displayOptions: {
+                                show: {
+                                    '/type': ['map_reduce'],
+                                },
+                            },
+                            typeOptions: {
+                                rows: 6,
+                            },
+                        },
+                        {
+                            displayName: 'Prompt',
+                            name: 'prompt',
+                            type: 'string',
+                            default: prompt_1.DEFAULT_PROMPT_TEMPLATE,
+                            displayOptions: {
+                                show: {
+                                    '/type': ['stuff'],
+                                },
+                            },
+                            typeOptions: {
+                                rows: 6,
+                            },
+                        },
+                        {
+                            displayName: 'Subsequent (Refine) Prompt',
+                            name: 'refinePrompt',
+                            type: 'string',
+                            displayOptions: {
+                                show: {
+                                    '/type': ['refine'],
+                                },
+                            },
+                            default: prompt_1.REFINE_PROMPT_TEMPLATE,
+                            hint: 'The prompt to refine the summary based on the next document (or chunk)',
+                            typeOptions: {
+                                rows: 6,
+                            },
+                        },
+                        {
+                            displayName: 'Initial Prompt',
+                            name: 'refineQuestionPrompt',
+                            type: 'string',
+                            displayOptions: {
+                                show: {
+                                    '/type': ['refine'],
+                                },
+                            },
+                            default: prompt_1.DEFAULT_PROMPT_TEMPLATE,
+                            hint: 'The prompt for the first document (or chunk)',
+                            typeOptions: {
+                                rows: 6,
+                            },
+                        },
+                    ],
+                },
+            ],
+        };
+    }
+    async execute() {
+        this.logger.verbose('Executing Vector Store QA Chain');
+        const type = this.getNodeParameter('type', 0);
+        const model = (await this.getInputConnectionData("ai_languageModel", 0));
+        const documentInput = (await this.getInputConnectionData("ai_document", 0));
+        const options = this.getNodeParameter('options', 0, {});
+        const chainArgs = {
+            type,
+        };
+        if (type === 'map_reduce') {
+            const mapReduceArgs = chainArgs;
+            if (options.combineMapPrompt) {
+                mapReduceArgs.combineMapPrompt = new prompts_1.PromptTemplate({
+                    template: options.combineMapPrompt,
+                    inputVariables: ['text'],
+                });
+            }
+            if (options.prompt) {
+                mapReduceArgs.combinePrompt = new prompts_1.PromptTemplate({
+                    template: options.prompt,
+                    inputVariables: ['text'],
+                });
+            }
+        }
+        if (type === 'stuff') {
+            const stuffArgs = chainArgs;
+            if (options.prompt) {
+                stuffArgs.prompt = new prompts_1.PromptTemplate({
+                    template: options.prompt,
+                    inputVariables: ['text'],
+                });
+            }
+        }
+        if (type === 'refine') {
+            const refineArgs = chainArgs;
+            if (options.refinePrompt) {
+                refineArgs.refinePrompt = new prompts_1.PromptTemplate({
+                    template: options.refinePrompt,
+                    inputVariables: ['existing_answer', 'text'],
+                });
+            }
+            if (options.refineQuestionPrompt) {
+                refineArgs.questionPrompt = new prompts_1.PromptTemplate({
+                    template: options.refineQuestionPrompt,
+                    inputVariables: ['text'],
+                });
+            }
+        }
+        const chain = (0, chains_1.loadSummarizationChain)(model, chainArgs);
+        const items = this.getInputData();
+        const returnData = [];
+        for (let itemIndex = 0; itemIndex < items.length; itemIndex++) {
+            let processedDocuments;
+            if (documentInput instanceof N8nJsonLoader_1.N8nJsonLoader || documentInput instanceof N8nBinaryLoader_1.N8nBinaryLoader) {
+                processedDocuments = await documentInput.processItem(items[itemIndex], itemIndex);
+            }
+            else {
+                processedDocuments = documentInput;
+            }
+            const response = await chain.call({
+                input_documents: processedDocuments,
+            });
+            returnData.push({ json: { response } });
+        }
+        return this.prepareOutputData(returnData);
+    }
+}
+exports.ChainSummarizationV1 = ChainSummarizationV1;
+//# sourceMappingURL=ChainSummarizationV1.node.js.map

package/dist/nodes/chains/ChainSummarization/V1/ChainSummarizationV1.node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainSummarizationV1.node.js","sourceRoot":"","sources":["../../../../../nodes/chains/ChainSummarization/V1/ChainSummarizationV1.node.ts"],"names":[],"mappings":";;;AAUA,6CAA0D;AAG1D,+CAAmD;AACnD,mEAAgE;AAChE,uEAAoE;AACpE,iEAAwE;AACxE,sCAA4E;AAE5E,MAAa,oBAAoB;IAGhC,YAAY,eAAyC;QACpD,IAAI,CAAC,WAAW,GAAG;YAClB,GAAG,eAAe;YAClB,OAAO,EAAE,CAAC;YACV,QAAQ,EAAE;gBACT,IAAI,EAAE,qBAAqB;gBAC3B,KAAK,EAAE,SAAS;aAChB;YAED,MAAM,EAAE;;gBAEP;oBACC,WAAW,EAAE,OAAO;oBACpB,cAAc,EAAE,CAAC;oBACjB,IAAI,oBAAoC;oBACxC,QAAQ,EAAE,IAAI;iBACd;gBACD;oBACC,WAAW,EAAE,UAAU;oBACvB,cAAc,EAAE,CAAC;oBACjB,IAAI,eAA+B;oBACnC,QAAQ,EAAE,IAAI;iBACd;aACD;YACD,OAAO,EAAE,QAAyB;YAClC,WAAW,EAAE,EAAE;YACf,UAAU,EAAE;gBACX,IAAA,qCAAsB,EAAC,IAAI,CAAC;gBAC5B;oBACC,WAAW,EAAE,MAAM;oBACnB,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,kCAAkC;oBAC/C,OAAO,EAAE,YAAY;oBACrB,OAAO,EAAE;wBACR;4BACC,IAAI,EAAE,0BAA0B;4BAChC,KAAK,EAAE,YAAY;4BACnB,WAAW,EACV,iFAAiF;yBAClF;wBACD;4BACC,IAAI,EAAE,QAAQ;4BACd,KAAK,EAAE,QAAQ;4BACf,WAAW,EACV,sHAAsH;yBACvH;wBACD;4BACC,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,OAAO;4BACd,WAAW,EAAE,mEAAmE;yBAChF;qBACD;iBACD;gBACD;oBACC,WAAW,EAAE,SAAS;oBACtB,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,YAAY;oBAClB,OAAO,EAAE,EAAE;oBACX,WAAW,EAAE,YAAY;oBACzB,OAAO,EAAE;wBACR;4BACC,WAAW,EAAE,yBAAyB;4BACtC,IAAI,EAAE,kBAAkB;4BACxB,IAAI,EAAE,QAAQ;4BACd,IAAI,EAAE,4CAA4C;4BAClD,cAAc,EAAE;gCACf,IAAI,EAAE;oCACL,OAAO,EAAE,CAAC,YAAY,CAAC;iCACvB;6BACD;4BACD,OAAO,EAAE,gCAAuB;4BAChC,WAAW,EAAE;gCACZ,IAAI,EAAE,CAAC;6BACP;yBACD;wBACD;4BACC,WAAW,EAAE,2BAA2B;4BACxC,IAAI,EAAE,QAAQ;4BACd,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,gCAAuB;4BAChC,IAAI,EAAE,2DAA2D;4BACjE,cAAc,EAAE;gCACf,IAAI,EAAE;oCACL,OAAO,EAAE,CAAC,YAAY,CAAC;iCACvB;6BACD;4BACD,WAAW,EAAE;gCACZ,IAAI,EAAE,CAAC;6BACP;yBACD;wBACD;4BACC,WAAW,EAAE,QAAQ;4BACrB,IAAI,EAAE,QAAQ;4BACd,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,gCAAuB;4BAChC,cAAc,EAAE;gCACf,IAAI,EAAE;oCACL,OAAO,EAAE,CAAC,OAAO,CAAC;iCAClB;6BACD;4BACD,WAAW,EAAE;gCACZ,IAAI,EAAE,CAAC;6BACP;yBACD;wBACD;4BACC,WAAW,EAAE,4BAA4B;4BACzC,IAAI,EAAE,cAAc;4BACpB,IAAI,EAAE,QAAQ;4BACd,cAAc,EAAE;gCACf,IAAI,EAAE;oCACL,OAAO,EAAE,CAAC,QAAQ,CAAC;iCACnB;6BACD;4BACD,OAAO,EAAE,+BAAsB;4BAC/B,IAAI,EAAE,wEAAwE;4BAC9E,WAAW,EAAE;gCACZ,IAAI,EAAE,CAAC;6BACP;yBACD;wBACD;4BACC,WAAW,EAAE,gBAAgB;4BAC7B,IAAI,EAAE,sBAAsB;4BAC5B,IAAI,EAAE,QAAQ;4BACd,cAAc,EAAE;gCACf,IAAI,EAAE;oCACL,OAAO,EAAE,CAAC,QAAQ,CAAC;iCACnB;6BACD;4BACD,OAAO,EAAE,gCAAuB;4BAChC,IAAI,EAAE,8CAA8C;4BACpD,WAAW,EAAE;gCACZ,IAAI,EAAE,CAAC;6BACP;yBACD;qBACD;iBACD;aACD;SACD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO;QACZ,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,iCAAiC,CAAC,CAAC;QACvD,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAsC,CAAC;QAEnF,MAAM,KAAK,GAAG,CAAC,MAAM,IAAI,CAAC,sBAAsB,qBAE/C,CAAC,CACD,CAAsB,CAAC;QAExB,MAAM,aAAa,GAAG,CAAC,MAAM,IAAI,CAAC,sBAAsB,gBAAgC,CAAC,CAAC,CAE/C,CAAC;QAE5C,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,CAKrD,CAAC;QAEF,MAAM,SAAS,GAA6B;YAC3C,IAAI;SACJ,CAAC;QAGF,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YAC3B,MAAM,aAAa,GAAG,SAErB,CAAC;YACF,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;gBAC9B,aAAa,CAAC,gBAAgB,GAAG,IAAI,wBAAc,CAAC;oBACnD,QAAQ,EAAE,OAAO,CAAC,gBAAgB;oBAClC,cAAc,EAAE,CAAC,MAAM,CAAC;iBACxB,CAAC,CAAC;YACJ,CAAC;YACD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACpB,aAAa,CAAC,aAAa,GAAG,IAAI,wBAAc,CAAC;oBAChD,QAAQ,EAAE,OAAO,CAAC,MAAM;oBACxB,cAAc,EAAE,CAAC,MAAM,CAAC;iBACxB,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QAGD,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACtB,MAAM,SAAS,GAAG,SAEjB,CAAC;YACF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACpB,SAAS,CAAC,MAAM,GAAG,IAAI,wBAAc,CAAC;oBACrC,QAAQ,EAAE,OAAO,CAAC,MAAM;oBACxB,cAAc,EAAE,CAAC,MAAM,CAAC;iBACxB,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QAGD,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;YACvB,MAAM,UAAU,GAAG,SAElB,CAAC;YAEF,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;gBAC1B,UAAU,CAAC,YAAY,GAAG,IAAI,wBAAc,CAAC;oBAC5C,QAAQ,EAAE,OAAO,CAAC,YAAY;oBAC9B,cAAc,EAAE,CAAC,iBAAiB,EAAE,MAAM,CAAC;iBAC3C,CAAC,CAAC;YACJ,CAAC;YAED,IAAI,OAAO,CAAC,oBAAoB,EAAE,CAAC;gBAClC,UAAU,CAAC,cAAc,GAAG,IAAI,wBAAc,CAAC;oBAC9C,QAAQ,EAAE,OAAO,CAAC,oBAAoB;oBACtC,cAAc,EAAE,CAAC,MAAM,CAAC;iBACxB,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QAED,MAAM,KAAK,GAAG,IAAA,+BAAsB,EAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAEvD,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAClC,MAAM,UAAU,GAAyB,EAAE,CAAC;QAE5C,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;YAC/D,IAAI,kBAA8B,CAAC;YACnC,IAAI,aAAa,YAAY,6BAAa,IAAI,aAAa,YAAY,iCAAe,EAAE,CAAC;gBACxF,kBAAkB,GAAG,MAAM,aAAa,CAAC,WAAW,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC,CAAC;YACnF,CAAC;iBAAM,CAAC;gBACP,kBAAkB,GAAG,aAAa,CAAC;YACpC,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC;gBACjC,eAAe,EAAE,kBAAkB;aACnC,CAAC,CAAC;YAEH,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC;QAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC;CACD;AAnPD,oDAmPC"}

package/dist/nodes/chains/ChainSummarization/V2/ChainSummarizationV2.node.d.ts

@@ -0,0 +1,6 @@
+import type { INodeTypeBaseDescription, IExecuteFunctions, INodeExecutionData, INodeType, INodeTypeDescription } from 'n8n-workflow';
+export declare class ChainSummarizationV2 implements INodeType {
+    description: INodeTypeDescription;
+    constructor(baseDescription: INodeTypeBaseDescription);
+    execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]>;
+}

package/dist/nodes/chains/ChainSummarization/V2/ChainSummarizationV2.node.js

@@ -0,0 +1,347 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChainSummarizationV2 = void 0;
+const chains_1 = require("langchain/chains");
+const text_splitter_1 = require("langchain/text_splitter");
+const N8nJsonLoader_1 = require("../../../../utils/N8nJsonLoader");
+const N8nBinaryLoader_1 = require("../../../../utils/N8nBinaryLoader");
+const sharedFields_1 = require("../../../../utils/sharedFields");
+const prompt_1 = require("../prompt");
+const helpers_1 = require("../helpers");
+function getInputs(parameters) {
+    const chunkingMode = parameters === null || parameters === void 0 ? void 0 : parameters.chunkingMode;
+    const operationMode = parameters === null || parameters === void 0 ? void 0 : parameters.operationMode;
+    const inputs = [
+        { displayName: '', type: "main" },
+        {
+            displayName: 'Model',
+            maxConnections: 1,
+            type: "ai_languageModel",
+            required: true,
+        },
+    ];
+    if (operationMode === 'documentLoader') {
+        inputs.push({
+            displayName: 'Document',
+            type: "ai_document",
+            required: true,
+            maxConnections: 1,
+        });
+        return inputs;
+    }
+    if (chunkingMode === 'advanced') {
+        inputs.push({
+            displayName: 'Text Splitter',
+            type: "ai_textSplitter",
+            required: false,
+            maxConnections: 1,
+        });
+        return inputs;
+    }
+    return inputs;
+}
+class ChainSummarizationV2 {
+    constructor(baseDescription) {
+        this.description = {
+            ...baseDescription,
+            version: [2],
+            defaults: {
+                name: 'Summarization Chain',
+                color: '#909298',
+            },
+            inputs: `={{ ((parameter) => { ${getInputs.toString()}; return getInputs(parameter) })($parameter) }}`,
+            outputs: ["main"],
+            credentials: [],
+            properties: [
+                (0, sharedFields_1.getTemplateNoticeField)(1951),
+                {
+                    displayName: 'Data to Summarize',
+                    name: 'operationMode',
+                    noDataExpression: true,
+                    type: 'options',
+                    description: 'How to pass data into the summarization chain',
+                    default: 'nodeInputJson',
+                    options: [
+                        {
+                            name: 'Use Node Input (JSON)',
+                            value: 'nodeInputJson',
+                            description: 'Summarize the JSON data coming into this node from the previous one',
+                        },
+                        {
+                            name: 'Use Node Input (Binary)',
+                            value: 'nodeInputBinary',
+                            description: 'Summarize the binary data coming into this node from the previous one',
+                        },
+                        {
+                            name: 'Use Document Loader',
+                            value: 'documentLoader',
+                            description: 'Use a loader sub-node with more configuration options',
+                        },
+                    ],
+                },
+                {
+                    displayName: 'Chunking Strategy',
+                    name: 'chunkingMode',
+                    noDataExpression: true,
+                    type: 'options',
+                    description: 'Chunk splitting strategy',
+                    default: 'simple',
+                    options: [
+                        {
+                            name: 'Simple (Define Below)',
+                            value: 'simple',
+                        },
+                        {
+                            name: 'Advanced',
+                            value: 'advanced',
+                            description: 'Use a splitter sub-node with more configuration options',
+                        },
+                    ],
+                    displayOptions: {
+                        show: {
+                            '/operationMode': ['nodeInputJson', 'nodeInputBinary'],
+                        },
+                    },
+                },
+                {
+                    displayName: 'Characters Per Chunk',
+                    name: 'chunkSize',
+                    description: 'Controls the max size (in terms of number of characters) of the final document chunk',
+                    type: 'number',
+                    default: 1000,
+                    displayOptions: {
+                        show: {
+                            '/chunkingMode': ['simple'],
+                        },
+                    },
+                },
+                {
+                    displayName: 'Chunk Overlap (Characters)',
+                    name: 'chunkOverlap',
+                    type: 'number',
+                    description: 'Specifies how much characters overlap there should be between chunks',
+                    default: 200,
+                    displayOptions: {
+                        show: {
+                            '/chunkingMode': ['simple'],
+                        },
+                    },
+                },
+                {
+                    displayName: 'Options',
+                    name: 'options',
+                    type: 'collection',
+                    default: {},
+                    placeholder: 'Add Option',
+                    options: [
+                        {
+                            displayName: 'Input Data Field Name',
+                            name: 'binaryDataKey',
+                            type: 'string',
+                            default: 'data',
+                            description: 'The name of the field in the agent or chain’s input that contains the binary file to be processed',
+                            displayOptions: {
+                                show: {
+                                    '/operationMode': ['nodeInputBinary'],
+                                },
+                            },
+                        },
+                        {
+                            displayName: 'Summarization Method and Prompts',
+                            name: 'summarizationMethodAndPrompts',
+                            type: 'fixedCollection',
+                            default: {
+                                values: {
+                                    summarizationMethod: 'map_reduce',
+                                    prompt: prompt_1.DEFAULT_PROMPT_TEMPLATE,
+                                    combineMapPrompt: prompt_1.DEFAULT_PROMPT_TEMPLATE,
+                                },
+                            },
+                            placeholder: 'Add Option',
+                            typeOptions: {},
+                            options: [
+                                {
+                                    name: 'values',
+                                    displayName: 'Values',
+                                    values: [
+                                        {
+                                            displayName: 'Summarization Method',
+                                            name: 'summarizationMethod',
+                                            type: 'options',
+                                            description: 'The type of summarization to run',
+                                            default: 'map_reduce',
+                                            options: [
+                                                {
+                                                    name: 'Map Reduce (Recommended)',
+                                                    value: 'map_reduce',
+                                                    description: 'Summarize each document (or chunk) individually, then summarize those summaries',
+                                                },
+                                                {
+                                                    name: 'Refine',
+                                                    value: 'refine',
+                                                    description: 'Summarize the first document (or chunk). Then update that summary based on the next document (or chunk), and repeat.',
+                                                },
+                                                {
+                                                    name: 'Stuff',
+                                                    value: 'stuff',
+                                                    description: 'Pass all documents (or chunks) at once. Ideal for small datasets.',
+                                                },
+                                            ],
+                                        },
+                                        {
+                                            displayName: 'Final Prompt to Combine',
+                                            name: 'combineMapPrompt',
+                                            type: 'string',
+                                            hint: 'The prompt to combine individual summaries',
+                                            displayOptions: {
+                                                hide: {
+                                                    '/options.summarizationMethodAndPrompts.values.summarizationMethod': [
+                                                        'stuff',
+                                                        'refine',
+                                                    ],
+                                                },
+                                            },
+                                            default: prompt_1.DEFAULT_PROMPT_TEMPLATE,
+                                            typeOptions: {
+                                                rows: 9,
+                                            },
+                                        },
+                                        {
+                                            displayName: 'Individual Summary Prompt',
+                                            name: 'prompt',
+                                            type: 'string',
+                                            default: prompt_1.DEFAULT_PROMPT_TEMPLATE,
+                                            hint: 'The prompt to summarize an individual document (or chunk)',
+                                            displayOptions: {
+                                                hide: {
+                                                    '/options.summarizationMethodAndPrompts.values.summarizationMethod': [
+                                                        'stuff',
+                                                        'refine',
+                                                    ],
+                                                },
+                                            },
+                                            typeOptions: {
+                                                rows: 9,
+                                            },
+                                        },
+                                        {
+                                            displayName: 'Prompt',
+                                            name: 'prompt',
+                                            type: 'string',
+                                            default: prompt_1.DEFAULT_PROMPT_TEMPLATE,
+                                            displayOptions: {
+                                                hide: {
+                                                    '/options.summarizationMethodAndPrompts.values.summarizationMethod': [
+                                                        'refine',
+                                                        'map_reduce',
+                                                    ],
+                                                },
+                                            },
+                                            typeOptions: {
+                                                rows: 9,
+                                            },
+                                        },
+                                        {
+                                            displayName: 'Subsequent (Refine) Prompt',
+                                            name: 'refinePrompt',
+                                            type: 'string',
+                                            displayOptions: {
+                                                hide: {
+                                                    '/options.summarizationMethodAndPrompts.values.summarizationMethod': [
+                                                        'stuff',
+                                                        'map_reduce',
+                                                    ],
+                                                },
+                                            },
+                                            default: prompt_1.REFINE_PROMPT_TEMPLATE,
+                                            hint: 'The prompt to refine the summary based on the next document (or chunk)',
+                                            typeOptions: {
+                                                rows: 9,
+                                            },
+                                        },
+                                        {
+                                            displayName: 'Initial Prompt',
+                                            name: 'refineQuestionPrompt',
+                                            type: 'string',
+                                            displayOptions: {
+                                                hide: {
+                                                    '/options.summarizationMethodAndPrompts.values.summarizationMethod': [
+                                                        'stuff',
+                                                        'map_reduce',
+                                                    ],
+                                                },
+                                            },
+                                            default: prompt_1.DEFAULT_PROMPT_TEMPLATE,
+                                            hint: 'The prompt for the first document (or chunk)',
+                                            typeOptions: {
+                                                rows: 9,
+                                            },
+                                        },
+                                    ],
+                                },
+                            ],
+                        },
+                    ],
+                },
+            ],
+        };
+    }
+    async execute() {
+        var _a;
+        this.logger.verbose('Executing Summarization Chain V2');
+        const operationMode = this.getNodeParameter('operationMode', 0, 'nodeInputJson');
+        const chunkingMode = this.getNodeParameter('chunkingMode', 0, 'simple');
+        const model = (await this.getInputConnectionData("ai_languageModel", 0));
+        const items = this.getInputData();
+        const returnData = [];
+        for (let itemIndex = 0; itemIndex < items.length; itemIndex++) {
+            const summarizationMethodAndPrompts = this.getNodeParameter('options.summarizationMethodAndPrompts.values', itemIndex, {});
+            const chainArgs = (0, helpers_1.getChainPromptsArgs)((_a = summarizationMethodAndPrompts.summarizationMethod) !== null && _a !== void 0 ? _a : 'map_reduce', summarizationMethodAndPrompts);
+            const chain = (0, chains_1.loadSummarizationChain)(model, chainArgs);
+            const item = items[itemIndex];
+            let processedDocuments;
+            if (operationMode === 'documentLoader') {
+                const documentInput = (await this.getInputConnectionData("ai_document", 0));
+                const isN8nLoader = documentInput instanceof N8nJsonLoader_1.N8nJsonLoader || documentInput instanceof N8nBinaryLoader_1.N8nBinaryLoader;
+                processedDocuments = isN8nLoader
+                    ? await documentInput.processItem(item, itemIndex)
+                    : documentInput;
+                const response = await chain.call({
+                    input_documents: processedDocuments,
+                });
+                returnData.push({ json: { response } });
+            }
+            if (['nodeInputJson', 'nodeInputBinary'].includes(operationMode)) {
+                let textSplitter;
+                switch (chunkingMode) {
+                    case 'simple':
+                        const chunkSize = this.getNodeParameter('chunkSize', itemIndex, 1000);
+                        const chunkOverlap = this.getNodeParameter('chunkOverlap', itemIndex, 200);
+                        textSplitter = new text_splitter_1.RecursiveCharacterTextSplitter({ chunkOverlap, chunkSize });
+                        break;
+                    case 'advanced':
+                        textSplitter = (await this.getInputConnectionData("ai_textSplitter", 0));
+                        break;
+                    default:
+                        break;
+                }
+                let processor;
+                if (operationMode === 'nodeInputBinary') {
+                    const binaryDataKey = this.getNodeParameter('options.binaryDataKey', itemIndex, 'data');
+                    processor = new N8nBinaryLoader_1.N8nBinaryLoader(this, 'options.', binaryDataKey, textSplitter);
+                }
+                else {
+                    processor = new N8nJsonLoader_1.N8nJsonLoader(this, 'options.', textSplitter);
+                }
+                const processedItem = await processor.processItem(item, itemIndex);
+                const response = await chain.call({
+                    input_documents: processedItem,
+                });
+                returnData.push({ json: { response } });
+            }
+        }
+        return this.prepareOutputData(returnData);
+    }
+}
+exports.ChainSummarizationV2 = ChainSummarizationV2;
+//# sourceMappingURL=ChainSummarizationV2.node.js.map

package/dist/nodes/chains/ChainSummarization/V2/ChainSummarizationV2.node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChainSummarizationV2.node.js","sourceRoot":"","sources":["../../../../../nodes/chains/ChainSummarization/V2/ChainSummarizationV2.node.ts"],"names":[],"mappings":";;;AAUA,6CAA0D;AAI1D,2DAAyE;AACzE,mEAAgE;AAChE,uEAAoE;AACpE,iEAAwE;AACxE,sCAA4E;AAC5E,wCAAiD;AAEjD,SAAS,SAAS,CAAC,UAAuB;IACzC,MAAM,YAAY,GAAG,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,YAAY,CAAC;IAC9C,MAAM,aAAa,GAAG,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,aAAa,CAAC;IAChD,MAAM,MAAM,GAAG;QACd,EAAE,WAAW,EAAE,EAAE,EAAE,IAAI,QAAyB,EAAE;QAClD;YACC,WAAW,EAAE,OAAO;YACpB,cAAc,EAAE,CAAC;YACjB,IAAI,oBAAoC;YACxC,QAAQ,EAAE,IAAI;SACd;KACD,CAAC;IAEF,IAAI,aAAa,KAAK,gBAAgB,EAAE,CAAC;QACxC,MAAM,CAAC,IAAI,CAAC;YACX,WAAW,EAAE,UAAU;YACvB,IAAI,eAA+B;YACnC,QAAQ,EAAE,IAAI;YACd,cAAc,EAAE,CAAC;SACjB,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IACf,CAAC;IAED,IAAI,YAAY,KAAK,UAAU,EAAE,CAAC;QACjC,MAAM,CAAC,IAAI,CAAC;YACX,WAAW,EAAE,eAAe;YAC5B,IAAI,mBAAmC;YACvC,QAAQ,EAAE,KAAK;YACf,cAAc,EAAE,CAAC;SACjB,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IACf,CAAC;IACD,OAAO,MAAM,CAAC;AACf,CAAC;AAED,MAAa,oBAAoB;IAGhC,YAAY,eAAyC;QACpD,IAAI,CAAC,WAAW,GAAG;YAClB,GAAG,eAAe;YAClB,OAAO,EAAE,CAAC,CAAC,CAAC;YACZ,QAAQ,EAAE;gBACT,IAAI,EAAE,qBAAqB;gBAC3B,KAAK,EAAE,SAAS;aAChB;YAED,MAAM,EAAE,yBAAyB,SAAS,CAAC,QAAQ,EAAE,iDAAiD;YACtG,OAAO,EAAE,QAAyB;YAClC,WAAW,EAAE,EAAE;YACf,UAAU,EAAE;gBACX,IAAA,qCAAsB,EAAC,IAAI,CAAC;gBAC5B;oBACC,WAAW,EAAE,mBAAmB;oBAChC,IAAI,EAAE,eAAe;oBACrB,gBAAgB,EAAE,IAAI;oBACtB,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,+CAA+C;oBAC5D,OAAO,EAAE,eAAe;oBACxB,OAAO,EAAE;wBACR;4BACC,IAAI,EAAE,uBAAuB;4BAC7B,KAAK,EAAE,eAAe;4BACtB,WAAW,EAAE,qEAAqE;yBAClF;wBACD;4BACC,IAAI,EAAE,yBAAyB;4BAC/B,KAAK,EAAE,iBAAiB;4BACxB,WAAW,EAAE,uEAAuE;yBACpF;wBACD;4BACC,IAAI,EAAE,qBAAqB;4BAC3B,KAAK,EAAE,gBAAgB;4BACvB,WAAW,EAAE,uDAAuD;yBACpE;qBACD;iBACD;gBACD;oBACC,WAAW,EAAE,mBAAmB;oBAChC,IAAI,EAAE,cAAc;oBACpB,gBAAgB,EAAE,IAAI;oBACtB,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,0BAA0B;oBACvC,OAAO,EAAE,QAAQ;oBACjB,OAAO,EAAE;wBACR;4BACC,IAAI,EAAE,uBAAuB;4BAC7B,KAAK,EAAE,QAAQ;yBACf;wBACD;4BACC,IAAI,EAAE,UAAU;4BAChB,KAAK,EAAE,UAAU;4BACjB,WAAW,EAAE,yDAAyD;yBACtE;qBACD;oBACD,cAAc,EAAE;wBACf,IAAI,EAAE;4BACL,gBAAgB,EAAE,CAAC,eAAe,EAAE,iBAAiB,CAAC;yBACtD;qBACD;iBACD;gBACD;oBACC,WAAW,EAAE,sBAAsB;oBACnC,IAAI,EAAE,WAAW;oBACjB,WAAW,EACV,sFAAsF;oBACvF,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,IAAI;oBACb,cAAc,EAAE;wBACf,IAAI,EAAE;4BACL,eAAe,EAAE,CAAC,QAAQ,CAAC;yBAC3B;qBACD;iBACD;gBACD;oBACC,WAAW,EAAE,4BAA4B;oBACzC,IAAI,EAAE,cAAc;oBACpB,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,sEAAsE;oBACnF,OAAO,EAAE,GAAG;oBACZ,cAAc,EAAE;wBACf,IAAI,EAAE;4BACL,eAAe,EAAE,CAAC,QAAQ,CAAC;yBAC3B;qBACD;iBACD;gBACD;oBACC,WAAW,EAAE,SAAS;oBACtB,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,YAAY;oBAClB,OAAO,EAAE,EAAE;oBACX,WAAW,EAAE,YAAY;oBACzB,OAAO,EAAE;wBACR;4BACC,WAAW,EAAE,uBAAuB;4BACpC,IAAI,EAAE,eAAe;4BACrB,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,MAAM;4BACf,WAAW,EACV,mGAAmG;4BACpG,cAAc,EAAE;gCACf,IAAI,EAAE;oCACL,gBAAgB,EAAE,CAAC,iBAAiB,CAAC;iCACrC;6BACD;yBACD;wBACD;4BACC,WAAW,EAAE,kCAAkC;4BAC/C,IAAI,EAAE,+BAA+B;4BACrC,IAAI,EAAE,iBAAiB;4BACvB,OAAO,EAAE;gCACR,MAAM,EAAE;oCACP,mBAAmB,EAAE,YAAY;oCACjC,MAAM,EAAE,gCAAuB;oCAC/B,gBAAgB,EAAE,gCAAuB;iCACzC;6BACD;4BACD,WAAW,EAAE,YAAY;4BACzB,WAAW,EAAE,EAAE;4BACf,OAAO,EAAE;gCACR;oCACC,IAAI,EAAE,QAAQ;oCACd,WAAW,EAAE,QAAQ;oCACrB,MAAM,EAAE;wCACP;4CACC,WAAW,EAAE,sBAAsB;4CACnC,IAAI,EAAE,qBAAqB;4CAC3B,IAAI,EAAE,SAAS;4CACf,WAAW,EAAE,kCAAkC;4CAC/C,OAAO,EAAE,YAAY;4CACrB,OAAO,EAAE;gDACR;oDACC,IAAI,EAAE,0BAA0B;oDAChC,KAAK,EAAE,YAAY;oDACnB,WAAW,EACV,iFAAiF;iDAClF;gDACD;oDACC,IAAI,EAAE,QAAQ;oDACd,KAAK,EAAE,QAAQ;oDACf,WAAW,EACV,sHAAsH;iDACvH;gDACD;oDACC,IAAI,EAAE,OAAO;oDACb,KAAK,EAAE,OAAO;oDACd,WAAW,EACV,mEAAmE;iDACpE;6CACD;yCACD;wCACD;4CACC,WAAW,EAAE,yBAAyB;4CACtC,IAAI,EAAE,kBAAkB;4CACxB,IAAI,EAAE,QAAQ;4CACd,IAAI,EAAE,4CAA4C;4CAClD,cAAc,EAAE;gDACf,IAAI,EAAE;oDACL,mEAAmE,EAAE;wDACpE,OAAO;wDACP,QAAQ;qDACR;iDACD;6CACD;4CACD,OAAO,EAAE,gCAAuB;4CAChC,WAAW,EAAE;gDACZ,IAAI,EAAE,CAAC;6CACP;yCACD;wCACD;4CACC,WAAW,EAAE,2BAA2B;4CACxC,IAAI,EAAE,QAAQ;4CACd,IAAI,EAAE,QAAQ;4CACd,OAAO,EAAE,gCAAuB;4CAChC,IAAI,EAAE,2DAA2D;4CACjE,cAAc,EAAE;gDACf,IAAI,EAAE;oDACL,mEAAmE,EAAE;wDACpE,OAAO;wDACP,QAAQ;qDACR;iDACD;6CACD;4CACD,WAAW,EAAE;gDACZ,IAAI,EAAE,CAAC;6CACP;yCACD;wCACD;4CACC,WAAW,EAAE,QAAQ;4CACrB,IAAI,EAAE,QAAQ;4CACd,IAAI,EAAE,QAAQ;4CACd,OAAO,EAAE,gCAAuB;4CAChC,cAAc,EAAE;gDACf,IAAI,EAAE;oDACL,mEAAmE,EAAE;wDACpE,QAAQ;wDACR,YAAY;qDACZ;iDACD;6CACD;4CACD,WAAW,EAAE;gDACZ,IAAI,EAAE,CAAC;6CACP;yCACD;wCACD;4CACC,WAAW,EAAE,4BAA4B;4CACzC,IAAI,EAAE,cAAc;4CACpB,IAAI,EAAE,QAAQ;4CACd,cAAc,EAAE;gDACf,IAAI,EAAE;oDACL,mEAAmE,EAAE;wDACpE,OAAO;wDACP,YAAY;qDACZ;iDACD;6CACD;4CACD,OAAO,EAAE,+BAAsB;4CAC/B,IAAI,EAAE,wEAAwE;4CAC9E,WAAW,EAAE;gDACZ,IAAI,EAAE,CAAC;6CACP;yCACD;wCACD;4CACC,WAAW,EAAE,gBAAgB;4CAC7B,IAAI,EAAE,sBAAsB;4CAC5B,IAAI,EAAE,QAAQ;4CACd,cAAc,EAAE;gDACf,IAAI,EAAE;oDACL,mEAAmE,EAAE;wDACpE,OAAO;wDACP,YAAY;qDACZ;iDACD;6CACD;4CACD,OAAO,EAAE,gCAAuB;4CAChC,IAAI,EAAE,8CAA8C;4CACpD,WAAW,EAAE;gDACZ,IAAI,EAAE,CAAC;6CACP;yCACD;qCACD;iCACD;6BACD;yBACD;qBACD;iBACD;aACD;SACD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO;;QACZ,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC;QACxD,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE,CAAC,EAAE,eAAe,CAG5D,CAAC;QACpB,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC,EAAE,QAAQ,CAEzD,CAAC;QAEd,MAAM,KAAK,GAAG,CAAC,MAAM,IAAI,CAAC,sBAAsB,qBAE/C,CAAC,CACD,CAAsB,CAAC;QAExB,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAClC,MAAM,UAAU,GAAyB,EAAE,CAAC;QAE5C,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;YAC/D,MAAM,6BAA6B,GAAG,IAAI,CAAC,gBAAgB,CAC1D,8CAA8C,EAC9C,SAAS,EACT,EAAE,CAOF,CAAC;YAEF,MAAM,SAAS,GAAG,IAAA,6BAAmB,EACpC,MAAA,6BAA6B,CAAC,mBAAmB,mCAAI,YAAY,EACjE,6BAA6B,CAC7B,CAAC;YAEF,MAAM,KAAK,GAAG,IAAA,+BAAsB,EAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YACvD,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC;YAE9B,IAAI,kBAA8B,CAAC;YAGnC,IAAI,aAAa,KAAK,gBAAgB,EAAE,CAAC;gBACxC,MAAM,aAAa,GAAG,CAAC,MAAM,IAAI,CAAC,sBAAsB,gBAEvD,CAAC,CACD,CAA6D,CAAC;gBAE/D,MAAM,WAAW,GAChB,aAAa,YAAY,6BAAa,IAAI,aAAa,YAAY,iCAAe,CAAC;gBAEpF,kBAAkB,GAAG,WAAW;oBAC/B,CAAC,CAAC,MAAM,aAAa,CAAC,WAAW,CAAC,IAAI,EAAE,SAAS,CAAC;oBAClD,CAAC,CAAC,aAAa,CAAC;gBAEjB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC;oBACjC,eAAe,EAAE,kBAAkB;iBACnC,CAAC,CAAC;gBAEH,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;YACzC,CAAC;YAGD,IAAI,CAAC,eAAe,EAAE,iBAAiB,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;gBAClE,IAAI,YAAsC,CAAC;gBAE3C,QAAQ,YAAY,EAAE,CAAC;oBAEtB,KAAK,QAAQ;wBACZ,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE,SAAS,EAAE,IAAI,CAAW,CAAC;wBAChF,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,SAAS,EAAE,GAAG,CAAW,CAAC;wBAErF,YAAY,GAAG,IAAI,8CAA8B,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,CAAC;wBAC/E,MAAM;oBAGP,KAAK,UAAU;wBACd,YAAY,GAAG,CAAC,MAAM,IAAI,CAAC,sBAAsB,oBAEhD,CAAC,CACD,CAA6B,CAAC;wBAC/B,MAAM;oBACP;wBACC,MAAM;gBACR,CAAC;gBAED,IAAI,SAA0C,CAAC;gBAC/C,IAAI,aAAa,KAAK,iBAAiB,EAAE,CAAC;oBACzC,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAC1C,uBAAuB,EACvB,SAAS,EACT,MAAM,CACI,CAAC;oBACZ,SAAS,GAAG,IAAI,iCAAe,CAAC,IAAI,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,CAAC,CAAC;gBAChF,CAAC;qBAAM,CAAC;oBACP,SAAS,GAAG,IAAI,6BAAa,CAAC,IAAI,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC;gBAC/D,CAAC;gBAED,MAAM,aAAa,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;gBACnE,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC;oBACjC,eAAe,EAAE,aAAa;iBAC9B,CAAC,CAAC;gBACH,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;YACzC,CAAC;QACF,CAAC;QAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC;CACD;AA3WD,oDA2WC"}

package/dist/nodes/chains/ChainSummarization/helpers.d.ts

@@ -0,0 +1,9 @@
+import type { SummarizationChainParams } from 'langchain/chains';
+interface ChainTypeOptions {
+    combineMapPrompt?: string;
+    prompt?: string;
+    refinePrompt?: string;
+    refineQuestionPrompt?: string;
+}
+export declare function getChainPromptsArgs(type: 'stuff' | 'map_reduce' | 'refine', options: ChainTypeOptions): SummarizationChainParams;
+export {};

package/dist/nodes/chains/ChainSummarization/helpers.js

@@ -0,0 +1,51 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getChainPromptsArgs = void 0;
+const prompts_1 = require("langchain/prompts");
+function getChainPromptsArgs(type, options) {
+    const chainArgs = {
+        type,
+    };
+    if (type === 'map_reduce') {
+        const mapReduceArgs = chainArgs;
+        if (options.combineMapPrompt) {
+            mapReduceArgs.combineMapPrompt = new prompts_1.PromptTemplate({
+                template: options.combineMapPrompt,
+                inputVariables: ['text'],
+            });
+        }
+        if (options.prompt) {
+            mapReduceArgs.combinePrompt = new prompts_1.PromptTemplate({
+                template: options.prompt,
+                inputVariables: ['text'],
+            });
+        }
+    }
+    if (type === 'stuff') {
+        const stuffArgs = chainArgs;
+        if (options.prompt) {
+            stuffArgs.prompt = new prompts_1.PromptTemplate({
+                template: options.prompt,
+                inputVariables: ['text'],
+            });
+        }
+    }
+    if (type === 'refine') {
+        const refineArgs = chainArgs;
+        if (options.refinePrompt) {
+            refineArgs.refinePrompt = new prompts_1.PromptTemplate({
+                template: options.refinePrompt,
+                inputVariables: ['existing_answer', 'text'],
+            });
+        }
+        if (options.refineQuestionPrompt) {
+            refineArgs.questionPrompt = new prompts_1.PromptTemplate({
+                template: options.refineQuestionPrompt,
+                inputVariables: ['text'],
+            });
+        }
+    }
+    return chainArgs;
+}
+exports.getChainPromptsArgs = getChainPromptsArgs;
+//# sourceMappingURL=helpers.js.map