"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.ChainLlm = void 0;
const n8n_workflow_1 = require("n8n-workflow");
const N8nOutputParser_1 = require("../../../utils/output_parsers/N8nOutputParser");
const methods_1 = require("./methods");
const processItem_1 = require("./methods/processItem");
const error_handling_1 = require("../../vendors/OpenAi/helpers/error-handling");
class ChainLlm {
    constructor() {
        this.description = {
            displayName: 'Basic LLM Chain',
            name: 'chainLlm',
            icon: 'node:basic-llm-chain',
            iconColor: 'black',
            group: ['transform'],
            version: [1, 1.1, 1.2, 1.3, 1.4, 1.5, 1.6, 1.7, 1.8, 1.9],
            description: 'A simple chain to prompt a large language model',
            defaults: {
                name: 'Basic LLM Chain',
            },
            codex: {
                alias: ['LangChain'],
                categories: ['AI'],
                subcategories: {
                    AI: ['Chains', 'Root Nodes'],
                },
                resources: {
                    primaryDocumentation: [
                        {
                            url: 'https://docs.n8n.io/integrations/builtin/cluster-nodes/root-nodes/n8n-nodes-langchain.chainllm/',
                        },
                    ],
                },
            },
            inputs: `={{ ((parameter) => { ${methods_1.getInputs.toString()}; return getInputs(parameter) })($parameter) }}`,
            outputs: [n8n_workflow_1.NodeConnectionTypes.Main],
            builderHint: {
                inputs: {
                    ai_languageModel: { required: true },
                    ai_outputParser: {
                        required: false,
                        displayOptions: { show: { hasOutputParser: [true] } },
                    },
                },
            },
            credentials: [],
            properties: methods_1.nodeProperties,
        };
    }
    async execute() {
        this.logger.debug('Executing Basic LLM Chain');
        const items = this.getInputData();
        const returnData = [];
        const outputParser = await (0, N8nOutputParser_1.getOptionalOutputParser)(this);
        const shouldUnwrapObjects = this.getNode().typeVersion >= 1.6 || !!outputParser;
        const batchSize = this.getNodeParameter('batching.batchSize', 0, 5);
        const delayBetweenBatches = this.getNodeParameter('batching.delayBetweenBatches', 0, 0);
        if (this.getNode().typeVersion >= 1.7 && batchSize > 1) {
            for (let i = 0; i < items.length; i += batchSize) {
                const batch = items.slice(i, i + batchSize);
                const batchPromises = batch.map(async (_item, batchItemIndex) => {
                    return await (0, processItem_1.processItem)(this, i + batchItemIndex);
                });
                const batchResults = await Promise.allSettled(batchPromises);
                batchResults.forEach((promiseResult, batchItemIndex) => {
                    const itemIndex = i + batchItemIndex;
                    if (promiseResult.status === 'rejected') {
                        const error = promiseResult.reason;
                        if (error instanceof n8n_workflow_1.NodeApiError && (0, error_handling_1.isOpenAiError)(error.cause)) {
                            const openAiErrorCode = error.cause.error?.code;
                            if (openAiErrorCode) {
                                const customMessage = (0, error_handling_1.getCustomErrorMessage)(openAiErrorCode);
                                if (customMessage) {
                                    error.message = customMessage;
                                }
                            }
                        }
                        if (this.continueOnFail()) {
                            returnData.push({
                                json: { error: error.message },
                                pairedItem: { item: itemIndex },
                            });
                            return;
                        }
                        throw new n8n_workflow_1.NodeOperationError(this.getNode(), error);
                    }
                    const responses = promiseResult.value;
                    responses.forEach((response) => {
                        returnData.push({
                            json: (0, methods_1.formatResponse)(response, shouldUnwrapObjects),
                        });
                    });
                });
                if (i + batchSize < items.length && delayBetweenBatches > 0) {
                    await (0, n8n_workflow_1.sleep)(delayBetweenBatches);
                }
            }
        }
        else {
            for (let itemIndex = 0; itemIndex < items.length; itemIndex++) {
                try {
                    const responses = await (0, processItem_1.processItem)(this, itemIndex);
                    responses.forEach((response) => {
                        returnData.push({
                            json: (0, methods_1.formatResponse)(response, shouldUnwrapObjects),
                        });
                    });
                }
                catch (error) {
                    if (error instanceof n8n_workflow_1.NodeApiError && (0, error_handling_1.isOpenAiError)(error.cause)) {
                        const openAiErrorCode = error.cause.error?.code;
                        if (openAiErrorCode) {
                            const customMessage = (0, error_handling_1.getCustomErrorMessage)(openAiErrorCode);
                            if (customMessage) {
                                error.message = customMessage;
                            }
                        }
                    }
                    if (this.continueOnFail()) {
                        returnData.push({ json: { error: error.message }, pairedItem: { item: itemIndex } });
                        continue;
                    }
                    throw error;
                }
            }
        }
        return [returnData];
    }
}
exports.ChainLlm = ChainLlm;
//# sourceMappingURL=ChainLlm.node.js.map