@kyuda/n8n-nodes-databricks 1.0.2

package/dist/nodes/llms/LmChatDatabricks/LmChatDatabricks.node.js

@@ -0,0 +1,265 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LmChatDatabricks = void 0;
+const n8n_workflow_1 = require("n8n-workflow");
+const openai_1 = require("@langchain/openai");
+const modelCache = new Map();
+const CACHE_TTL = 1 * 60 * 1000;
+function getCacheKey(host) {
+    return `${host}:models`;
+}
+function isCacheValid(entry) {
+    if (!entry)
+        return false;
+    return Date.now() - entry.timestamp < CACHE_TTL;
+}
+class LmChatDatabricks {
+    constructor() {
+        this.description = {
+            displayName: 'Databricks Chat Model',
+            name: 'lmChatDatabricks',
+            icon: 'file:databricks.svg',
+            group: ['transform'],
+            version: 1,
+            description: 'Use Databricks hosted LLM models',
+            defaults: {
+                name: 'Databricks Chat Model',
+            },
+            codex: {
+                categories: ['AI'],
+                subcategories: {
+                    AI: ['Language Models', 'Root Nodes'],
+                    'Language Models': ['Chat Models (Recommended)'],
+                },
+                resources: {
+                    primaryDocumentation: [
+                        {
+                            url: 'https://docs.databricks.com/machine-learning/model-serving/index.html',
+                        },
+                    ],
+                },
+            },
+            inputs: [],
+            outputs: [n8n_workflow_1.NodeConnectionTypes.AiLanguageModel],
+            outputNames: ['Model'],
+            credentials: [
+                {
+                    name: 'databricks',
+                    required: true,
+                },
+            ],
+            requestDefaults: {
+                baseURL: '={{$credentials.host}}',
+                headers: {
+                    Authorization: '=Bearer {{$credentials.token}}',
+                },
+            },
+            properties: [
+                {
+                    displayName: 'Model',
+                    name: 'model',
+                    type: 'resourceLocator',
+                    default: { mode: 'list', value: '' },
+                    required: true,
+                    description: 'The Databricks model serving endpoint to use for chat',
+                    modes: [
+                        {
+                            displayName: 'From List',
+                            name: 'list',
+                            type: 'list',
+                            typeOptions: {
+                                searchListMethod: 'getModels',
+                                searchable: true,
+                            },
+                        },
+                        {
+                            displayName: 'By Name',
+                            name: 'name',
+                            type: 'string',
+                            placeholder: 'e.g. databricks-meta-llama-3-1-70b-instruct',
+                            validation: [
+                                {
+                                    type: 'regex',
+                                    properties: {
+                                        regex: '^[a-zA-Z0-9_-]+$',
+                                        errorMessage: 'Must be a valid endpoint name (alphanumeric, hyphens, and underscores only)',
+                                    },
+                                },
+                            ],
+                        },
+                        {
+                            displayName: 'By URL',
+                            name: 'url',
+                            type: 'string',
+                            placeholder: 'e.g. https://adb-xxx.cloud.databricks.com/serving-endpoints/my-endpoint',
+                            extractValue: {
+                                type: 'regex',
+                                regex: 'https://[^/]+/serving-endpoints/([a-zA-Z0-9_-]+)',
+                            },
+                        },
+                    ],
+                },
+                {
+                    displayName: 'Options',
+                    name: 'options',
+                    type: 'collection',
+                    placeholder: 'Add Option',
+                    default: {},
+                    options: [
+                        {
+                            displayName: 'Maximum Tokens',
+                            name: 'max_tokens',
+                            type: 'number',
+                            typeOptions: {
+                                maxValue: 32768,
+                            },
+                            default: 512,
+                            routing: {
+                                send: {
+                                    type: 'body',
+                                    property: 'max_tokens',
+                                },
+                            },
+                        },
+                        {
+                            displayName: 'Temperature',
+                            name: 'temperature',
+                            type: 'number',
+                            typeOptions: {
+                                maxValue: 2,
+                                minValue: 0,
+                                numberPrecision: 1,
+                            },
+                            default: 0.7,
+                            routing: {
+                                send: {
+                                    type: 'body',
+                                    property: 'temperature',
+                                },
+                            },
+                        },
+                        {
+                            displayName: 'Top P',
+                            name: 'top_p',
+                            type: 'number',
+                            typeOptions: {
+                                maxValue: 1,
+                                minValue: 0,
+                                numberPrecision: 1,
+                            },
+                            default: 1,
+                            routing: {
+                                send: {
+                                    type: 'body',
+                                    property: 'top_p',
+                                },
+                            },
+                        },
+                        {
+                            displayName: 'Conversation ID',
+                            name: 'conversation_id',
+                            type: 'string',
+                            default: '',
+                            description: 'Unique identifier for tracking conversations across requests (Databricks-specific)',
+                        },
+                        {
+                            displayName: 'Return Trace',
+                            name: 'return_trace',
+                            type: 'boolean',
+                            default: false,
+                            description: 'Whether to return the execution trace for debugging (Databricks-specific)',
+                        },
+                    ],
+                },
+            ],
+        };
+        this.methods = {
+            listSearch: {
+                async getModels(filter) {
+                    var _a;
+                    const credentials = await this.getCredentials('databricks');
+                    const host = credentials.host.replace(/\/$/, '');
+                    const cacheKey = getCacheKey(host);
+                    let endpoints = [];
+                    const cachedEntry = modelCache.get(cacheKey);
+                    if (isCacheValid(cachedEntry)) {
+                        endpoints = cachedEntry.data;
+                    }
+                    else {
+                        const response = await this.helpers.httpRequest({
+                            method: 'GET',
+                            url: `${host}/api/2.0/serving-endpoints`,
+                            headers: {
+                                Authorization: `Bearer ${credentials.token}`,
+                                'Accept': 'application/json',
+                            },
+                            json: true,
+                        });
+                        endpoints = (_a = response.endpoints) !== null && _a !== void 0 ? _a : [];
+                        modelCache.set(cacheKey, {
+                            data: endpoints,
+                            timestamp: Date.now(),
+                        });
+                    }
+                    const allResults = endpoints.map((endpoint) => {
+                        var _a;
+                        const modelNames = (((_a = endpoint.config) === null || _a === void 0 ? void 0 : _a.served_entities) || [])
+                            .map(entity => { var _a, _b; return ((_a = entity.external_model) === null || _a === void 0 ? void 0 : _a.name) || ((_b = entity.foundation_model) === null || _b === void 0 ? void 0 : _b.name); })
+                            .filter(Boolean)
+                            .join(', ');
+                        return {
+                            name: endpoint.name,
+                            value: endpoint.name,
+                            description: modelNames || undefined,
+                        };
+                    }).sort((a, b) => a.name.localeCompare(b.name));
+                    if (filter) {
+                        const filterLower = filter.toLowerCase();
+                        const filteredResults = allResults.filter((result) => result.name.toLowerCase().includes(filterLower));
+                        return { results: filteredResults };
+                    }
+                    return { results: allResults };
+                },
+            },
+        };
+    }
+    async supplyData(itemIndex) {
+        var _a, _b;
+        const credentials = await this.getCredentials('databricks');
+        const modelName = this.getNodeParameter('model', itemIndex, '', { extractValue: true });
+        const options = this.getNodeParameter('options', itemIndex, {});
+        const configuration = {
+            baseURL: `${credentials.host}/serving-endpoints`,
+        };
+        const modelKwargs = {};
+        if (options.responseFormat) {
+            modelKwargs.response_format = { type: options.responseFormat };
+        }
+        const hasDbOptions = options.conversation_id || options.return_trace !== undefined;
+        if (hasDbOptions) {
+            modelKwargs.databricks_options = {};
+            if (options.conversation_id) {
+                modelKwargs.databricks_options.conversation_id = options.conversation_id;
+            }
+            if (options.return_trace !== undefined) {
+                modelKwargs.databricks_options.return_trace = options.return_trace;
+            }
+        }
+        const model = new openai_1.ChatOpenAI({
+            openAIApiKey: `${credentials.token}`,
+            modelName,
+            temperature: options.temperature,
+            maxTokens: options.maxTokens,
+            topP: options.topP,
+            timeout: (_a = options.timeout) !== null && _a !== void 0 ? _a : 60000,
+            maxRetries: (_b = options.maxRetries) !== null && _b !== void 0 ? _b : 2,
+            configuration,
+            modelKwargs: Object.keys(modelKwargs).length > 0 ? modelKwargs : undefined,
+        });
+        return {
+            response: model,
+        };
+    }
+}
+exports.LmChatDatabricks = LmChatDatabricks;
+//# sourceMappingURL=LmChatDatabricks.node.js.map

package/dist/nodes/llms/LmChatDatabricks/LmChatDatabricks.node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"LmChatDatabricks.node.js","sourceRoot":"","sources":["../../../../nodes/llms/LmChatDatabricks/LmChatDatabricks.node.ts"],"names":[],"mappings":";;;AAAA,+CAQsB;AACtB,8CAA+C;AAU/C,MAAM,UAAU,GAA4B,IAAI,GAAG,EAAE,CAAC;AACtD,MAAM,SAAS,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;AAGhC,SAAS,WAAW,CAAC,IAAY;IAC7B,OAAO,GAAG,IAAI,SAAS,CAAC;AAC5B,CAAC;AAGD,SAAS,YAAY,CAAC,KAA6B;IAC/C,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,GAAG,SAAS,CAAC;AACpD,CAAC;AAED,MAAa,gBAAgB;IAA7B;QACI,gBAAW,GAAyB;YAChC,WAAW,EAAE,uBAAuB;YACpC,IAAI,EAAE,kBAAkB;YACxB,IAAI,EAAE,qBAAqB;YAC3B,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,kCAAkC;YAC/C,QAAQ,EAAE;gBACN,IAAI,EAAE,uBAAuB;aAChC;YACD,KAAK,EAAE;gBACZ,UAAU,EAAE,CAAC,IAAI,CAAC;gBAClB,aAAa,EAAE;oBACd,EAAE,EAAE,CAAC,iBAAiB,EAAE,YAAY,CAAC;oBACrC,iBAAiB,EAAE,CAAC,2BAA2B,CAAC;iBAChD;gBACQ,SAAS,EAAE;oBACP,oBAAoB,EAAE;wBAClB;4BACI,GAAG,EAAE,uEAAuE;yBAC/E;qBACJ;iBACJ;aACJ;YACD,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,CAAC,kCAAmB,CAAC,eAAe,CAAC;YAC9C,WAAW,EAAE,CAAC,OAAO,CAAC;YACtB,WAAW,EAAE;gBACT;oBACI,IAAI,EAAE,YAAY;oBAClB,QAAQ,EAAE,IAAI;iBACjB;aACJ;YACD,eAAe,EAAE;gBACb,OAAO,EAAE,wBAAwB;gBACjC,OAAO,EAAE;oBACL,aAAa,EAAE,gCAAgC;iBAClD;aACJ;YACD,UAAU,EAAE;gBACR;oBACI,WAAW,EAAE,OAAO;oBACpB,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,iBAAiB;oBACvB,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;oBACpC,QAAQ,EAAE,IAAI;oBACd,WAAW,EAAE,uDAAuD;oBACpE,KAAK,EAAE;wBACH;4BACI,WAAW,EAAE,WAAW;4BACxB,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,MAAM;4BACZ,WAAW,EAAE;gCACT,gBAAgB,EAAE,WAAW;gCAC7B,UAAU,EAAE,IAAI;6BACnB;yBACJ;wBACD;4BACI,WAAW,EAAE,SAAS;4BACtB,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,6CAA6C;4BAC1D,UAAU,EAAE;gCACR;oCACI,IAAI,EAAE,OAAO;oCACb,UAAU,EAAE;wCACR,KAAK,EAAE,kBAAkB;wCACzB,YAAY,EAAE,6EAA6E;qCAC9F;iCACJ;6BACJ;yBACJ;wBACD;4BACI,WAAW,EAAE,QAAQ;4BACrB,IAAI,EAAE,KAAK;4BACX,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,yEAAyE;4BACtF,YAAY,EAAE;gCACV,IAAI,EAAE,OAAO;gCACb,KAAK,EAAE,kDAAkD;6BAC5D;yBACJ;qBACJ;iBACJ;gBACD;oBACI,WAAW,EAAE,SAAS;oBACtB,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,YAAY;oBAClB,WAAW,EAAE,YAAY;oBACzB,OAAO,EAAE,EAAE;oBACX,OAAO,EAAE;wBACL;4BACI,WAAW,EAAE,gBAAgB;4BAC7B,IAAI,EAAE,YAAY;4BAClB,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE;gCACT,QAAQ,EAAE,KAAK;6BAClB;4BACD,OAAO,EAAE,GAAG;4BACZ,OAAO,EAAE;gCACL,IAAI,EAAE;oCACF,IAAI,EAAE,MAAM;oCACZ,QAAQ,EAAE,YAAY;iCACzB;6BACJ;yBACJ;wBACD;4BACI,WAAW,EAAE,aAAa;4BAC1B,IAAI,EAAE,aAAa;4BACnB,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE;gCACT,QAAQ,EAAE,CAAC;gCACX,QAAQ,EAAE,CAAC;gCACX,eAAe,EAAE,CAAC;6BACrB;4BACD,OAAO,EAAE,GAAG;4BACZ,OAAO,EAAE;gCACL,IAAI,EAAE;oCACF,IAAI,EAAE,MAAM;oCACZ,QAAQ,EAAE,aAAa;iCAC1B;6BACJ;yBACJ;wBACD;4BACI,WAAW,EAAE,OAAO;4BACpB,IAAI,EAAE,OAAO;4BACb,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE;gCACT,QAAQ,EAAE,CAAC;gCACX,QAAQ,EAAE,CAAC;gCACX,eAAe,EAAE,CAAC;6BACrB;4BACD,OAAO,EAAE,CAAC;4BACV,OAAO,EAAE;gCACL,IAAI,EAAE;oCACF,IAAI,EAAE,MAAM;oCACZ,QAAQ,EAAE,OAAO;iCACpB;6BACJ;yBACJ;wBACD;4BACI,WAAW,EAAE,iBAAiB;4BAC9B,IAAI,EAAE,iBAAiB;4BACvB,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,EAAE;4BACX,WAAW,EAAE,oFAAoF;yBACpG;wBACD;4BACI,WAAW,EAAE,cAAc;4BAC3B,IAAI,EAAE,cAAc;4BACpB,IAAI,EAAE,SAAS;4BACf,OAAO,EAAE,KAAK;4BACd,WAAW,EAAE,2EAA2E;yBAC3F;qBACJ;iBACJ;aACJ;SACJ,CAAC;QAEF,YAAO,GAAG;YACN,UAAU,EAAE;gBACR,KAAK,CAAC,SAAS,CAA8B,MAAe;;oBACxD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;oBAC5D,MAAM,IAAI,GAAI,WAAW,CAAC,IAAe,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;oBAC7D,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;oBAGnC,IAAI,SAAS,GAAgJ,EAAE,CAAC;oBAChK,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBAE7C,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;wBAE5B,SAAS,GAAG,WAAY,CAAC,IAAI,CAAC;oBAClC,CAAC;yBAAM,CAAC;wBAEJ,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC;4BAC5C,MAAM,EAAE,KAAK;4BACb,GAAG,EAAE,GAAG,IAAI,4BAA4B;4BACxC,OAAO,EAAE;gCACL,aAAa,EAAE,UAAU,WAAW,CAAC,KAAK,EAAE;gCAC5C,QAAQ,EAAE,kBAAkB;6BAC/B;4BACD,IAAI,EAAE,IAAI;yBACb,CAAgK,CAAC;wBAElK,SAAS,GAAG,MAAA,QAAQ,CAAC,SAAS,mCAAI,EAAE,CAAC;wBAGrC,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE;4BACrB,IAAI,EAAE,SAAS;4BACf,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;yBACxB,CAAC,CAAC;oBACP,CAAC;oBAED,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;;wBAC1C,MAAM,UAAU,GAAG,CAAC,CAAA,MAAA,QAAQ,CAAC,MAAM,0CAAE,eAAe,KAAI,EAAE,CAAC;6BACtD,GAAG,CAAC,MAAM,CAAC,EAAE,eAAC,OAAA,CAAA,MAAA,MAAM,CAAC,cAAc,0CAAE,IAAI,MAAI,MAAA,MAAM,CAAC,gBAAgB,0CAAE,IAAI,CAAA,CAAA,EAAA,CAAC;6BAC3E,MAAM,CAAC,OAAO,CAAC;6BACf,IAAI,CAAC,IAAI,CAAC,CAAC;wBAEhB,OAAO;4BACH,IAAI,EAAE,QAAQ,CAAC,IAAI;4BACnB,KAAK,EAAE,QAAQ,CAAC,IAAI;4BACpB,WAAW,EAAE,UAAU,IAAI,SAAS;yBACvC,CAAC;oBACN,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;oBAGhD,IAAI,MAAM,EAAE,CAAC;wBACT,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;wBACzC,MAAM,eAAe,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CACjD,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAClD,CAAC;wBACF,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;oBACxC,CAAC;oBAED,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;gBACnC,CAAC;aACJ;SACJ,CAAC;IA2DN,CAAC;IAzDG,KAAK,CAAC,UAAU,CAA6B,SAAiB;;QAC1D,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC5D,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,CAAW,CAAC;QAElG,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE,EAAE,CAS7D,CAAC;QAGF,MAAM,aAAa,GAAG;YAClB,OAAO,EAAE,GAAG,WAAW,CAAC,IAAI,oBAAoB;SACnD,CAAC;QAGF,MAAM,WAAW,GAAQ,EAAE,CAAC;QAE5B,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YACzB,WAAW,CAAC,eAAe,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,cAAc,EAAE,CAAC;QACnE,CAAC;QAGD,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC;QAEnF,IAAI,YAAY,EAAE,CAAC;YACf,WAAW,CAAC,kBAAkB,GAAG,EAAE,CAAC;YACpC,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;gBAC1B,WAAW,CAAC,kBAAkB,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;YAC7E,CAAC;YACD,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;gBACrC,WAAW,CAAC,kBAAkB,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;YACvE,CAAC;QACL,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,mBAAU,CAAC;YACzB,YAAY,EAAE,GAAG,WAAW,CAAC,KAAK,EAAE;YACpC,SAAS;YACT,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,OAAO,EAAE,MAAA,OAAO,CAAC,OAAO,mCAAI,KAAK;YACjC,UAAU,EAAE,MAAA,OAAO,CAAC,UAAU,mCAAI,CAAC;YACnC,aAAa;YAEb,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;SAC7E,CAAC,CAAC;QAEH,OAAO;YACH,QAAQ,EAAE,KAAK;SAClB,CAAC;IACN,CAAC;CACJ;AAvRD,4CAuRC"}

package/dist/nodes/llms/LmChatDatabricks/databricks.dark.svg

@@ -0,0 +1,19 @@
+<svg version="1.1" id="Layer_1" xmlns:x="ns_extend;" xmlns:i="ns_ai;" xmlns:graph="ns_graphs;" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px" viewBox="0 0 40.1 42" style="enable-background:new 0 0 40.1 42;" xml:space="preserve">
+ <style type="text/css">
+  .st0{fill:#FF3621;}
+ </style>
+ <metadata>
+  <sfw xmlns="ns_sfw;">
+   <slices>
+   </slices>
+   <sliceSourceBounds bottomLeftOrigin="true" height="42" width="40.1" x="-69.1" y="-10.5">
+   </sliceSourceBounds>
+  </sfw>
+ </metadata>
+ <g>
+  <path class="st0" d="M40.1,31.1v-7.4l-0.8-0.5L20.1,33.7l-18.2-10l0-4.3l18.2,9.9l20.1-10.9v-7.3l-0.8-0.5L20.1,21.2L2.6,11.6
+		L20.1,2l14.1,7.7l1.1-0.6V8.3L20.1,0L0,10.9V12L20.1,23l18.2-10v4.4l-18.2,10L0.8,16.8L0,17.3v7.4l20.1,10.9l18.2-9.9v4.3l-18.2,10
+		L0.8,29.5L0,30v1.1L20.1,42L40.1,31.1z">
+  </path>
+ </g>
+</svg>

package/dist/nodes/llms/LmChatDatabricks/databricks.svg

@@ -0,0 +1,19 @@
+<svg version="1.1" id="Layer_1" xmlns:x="ns_extend;" xmlns:i="ns_ai;" xmlns:graph="ns_graphs;" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px" viewBox="0 0 40.1 42" style="enable-background:new 0 0 40.1 42;" xml:space="preserve">
+ <style type="text/css">
+  .st0{fill:#FF3621;}
+ </style>
+ <metadata>
+  <sfw xmlns="ns_sfw;">
+   <slices>
+   </slices>
+   <sliceSourceBounds bottomLeftOrigin="true" height="42" width="40.1" x="-69.1" y="-10.5">
+   </sliceSourceBounds>
+  </sfw>
+ </metadata>
+ <g>
+  <path class="st0" d="M40.1,31.1v-7.4l-0.8-0.5L20.1,33.7l-18.2-10l0-4.3l18.2,9.9l20.1-10.9v-7.3l-0.8-0.5L20.1,21.2L2.6,11.6
+		L20.1,2l14.1,7.7l1.1-0.6V8.3L20.1,0L0,10.9V12L20.1,23l18.2-10v4.4l-18.2,10L0.8,16.8L0,17.3v7.4l20.1,10.9l18.2-9.9v4.3l-18.2,10
+		L0.8,29.5L0,30v1.1L20.1,42L40.1,31.1z">
+  </path>
+ </g>
+</svg>

package/dist/nodes/llms/N8nLlmTracing.d.ts

@@ -0,0 +1,44 @@
+import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
+import type { SerializedFields } from '@langchain/core/dist/load/map_keys';
+import type { Serialized, SerializedNotImplemented, SerializedSecret } from '@langchain/core/load/serializable';
+import type { BaseMessage } from '@langchain/core/messages';
+import type { LLMResult } from '@langchain/core/outputs';
+import type { IDataObject, ISupplyDataFunctions } from 'n8n-workflow';
+import { NodeError } from 'n8n-workflow';
+type TokensUsageParser = (llmOutput: LLMResult['llmOutput']) => {
+    completionTokens: number;
+    promptTokens: number;
+    totalTokens: number;
+};
+type RunDetail = {
+    index: number;
+    messages: BaseMessage[] | string[] | string;
+    options: SerializedSecret | SerializedNotImplemented | SerializedFields;
+};
+export declare class N8nLlmTracing extends BaseCallbackHandler {
+    private executionFunctions;
+    name: string;
+    awaitHandlers: boolean;
+    connectionType: "ai_languageModel";
+    promptTokensEstimate: number;
+    completionTokensEstimate: number;
+    runsMap: Record<string, RunDetail>;
+    options: {
+        tokensUsageParser: (llmOutput: LLMResult["llmOutput"]) => {
+            completionTokens: number;
+            promptTokens: number;
+            totalTokens: number;
+        };
+        errorDescriptionMapper: (error: NodeError) => string | null | undefined;
+    };
+    constructor(executionFunctions: ISupplyDataFunctions, options?: {
+        tokensUsageParser?: TokensUsageParser;
+        errorDescriptionMapper?: (error: NodeError) => string;
+    });
+    estimateTokensFromGeneration(generations: LLMResult['generations']): Promise<number>;
+    estimateTokensFromStringList(list: string[]): Promise<number>;
+    handleLLMEnd(output: LLMResult, runId: string): Promise<void>;
+    handleLLMStart(llm: Serialized, prompts: string[], runId: string): Promise<void>;
+    handleLLMError(error: IDataObject | Error, runId: string, parentRunId?: string | undefined): Promise<void>;
+}
+export {};

package/dist/nodes/llms/N8nLlmTracing.js

@@ -0,0 +1,140 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.N8nLlmTracing = void 0;
+const base_1 = require("@langchain/core/callbacks/base");
+const base_2 = require("@langchain/core/language_models/base");
+const tiktoken_1 = require("@langchain/core/utils/tiktoken");
+const lodash_1 = require("lodash");
+const n8n_workflow_1 = require("n8n-workflow");
+const helpers_1 = require("../../utils/helpers");
+const TIKTOKEN_ESTIMATE_MODEL = 'gpt-4o';
+class N8nLlmTracing extends base_1.BaseCallbackHandler {
+    constructor(executionFunctions, options) {
+        super();
+        this.executionFunctions = executionFunctions;
+        this.name = 'N8nLlmTracing';
+        this.awaitHandlers = true;
+        this.connectionType = n8n_workflow_1.NodeConnectionTypes.AiLanguageModel;
+        this.promptTokensEstimate = 0;
+        this.completionTokensEstimate = 0;
+        this.runsMap = {};
+        this.options = {
+            tokensUsageParser: (llmOutput) => {
+                var _a, _b, _c, _d;
+                const completionTokens = (_b = (_a = llmOutput === null || llmOutput === void 0 ? void 0 : llmOutput.tokenUsage) === null || _a === void 0 ? void 0 : _a.completionTokens) !== null && _b !== void 0 ? _b : 0;
+                const promptTokens = (_d = (_c = llmOutput === null || llmOutput === void 0 ? void 0 : llmOutput.tokenUsage) === null || _c === void 0 ? void 0 : _c.promptTokens) !== null && _d !== void 0 ? _d : 0;
+                return {
+                    completionTokens,
+                    promptTokens,
+                    totalTokens: completionTokens + promptTokens,
+                };
+            },
+            errorDescriptionMapper: (error) => error.description,
+        };
+        this.options = { ...this.options, ...options };
+    }
+    async estimateTokensFromGeneration(generations) {
+        const messages = generations.flatMap((gen) => gen.map((g) => g.text));
+        return await this.estimateTokensFromStringList(messages);
+    }
+    async estimateTokensFromStringList(list) {
+        const embeddingModel = (0, base_2.getModelNameForTiktoken)(TIKTOKEN_ESTIMATE_MODEL);
+        const encoder = await (0, tiktoken_1.encodingForModel)(embeddingModel);
+        const encodedListLength = await Promise.all(list.map(async (text) => encoder.encode(text).length));
+        return encodedListLength.reduce((acc, curr) => acc + curr, 0);
+    }
+    async handleLLMEnd(output, runId) {
+        var _a;
+        const runDetails = (_a = this.runsMap[runId]) !== null && _a !== void 0 ? _a : { index: Object.keys(this.runsMap).length };
+        output.generations = output.generations.map((gen) => gen.map((g) => (0, lodash_1.pick)(g, ['text', 'generationInfo'])));
+        const tokenUsageEstimate = {
+            completionTokens: 0,
+            promptTokens: 0,
+            totalTokens: 0,
+        };
+        const tokenUsage = this.options.tokensUsageParser(output.llmOutput);
+        if (output.generations.length > 0) {
+            tokenUsageEstimate.completionTokens = await this.estimateTokensFromGeneration(output.generations);
+            tokenUsageEstimate.promptTokens = this.promptTokensEstimate;
+            tokenUsageEstimate.totalTokens =
+                tokenUsageEstimate.completionTokens + this.promptTokensEstimate;
+        }
+        const response = {
+            response: { generations: output.generations },
+        };
+        if (tokenUsage.completionTokens > 0) {
+            response.tokenUsage = tokenUsage;
+        }
+        else {
+            response.tokenUsageEstimate = tokenUsageEstimate;
+        }
+        const parsedMessages = typeof runDetails.messages === 'string'
+            ? runDetails.messages
+            : runDetails.messages.map((message) => {
+                if (typeof message === 'string')
+                    return message;
+                if (typeof (message === null || message === void 0 ? void 0 : message.toJSON) === 'function')
+                    return message.toJSON();
+                return message;
+            });
+        this.executionFunctions.addOutputData(this.connectionType, runDetails.index, [
+            [{ json: { ...response } }],
+        ]);
+        (0, helpers_1.logAiEvent)(this.executionFunctions, 'ai-llm-generated-output', {
+            messages: parsedMessages,
+            options: runDetails.options,
+            response,
+        });
+    }
+    async handleLLMStart(llm, prompts, runId) {
+        const estimatedTokens = await this.estimateTokensFromStringList(prompts);
+        const options = llm.type === 'constructor' ? llm.kwargs : llm;
+        const { index } = this.executionFunctions.addInputData(this.connectionType, [
+            [
+                {
+                    json: {
+                        messages: prompts,
+                        estimatedTokens,
+                        options,
+                    },
+                },
+            ],
+        ]);
+        this.runsMap[runId] = {
+            index,
+            options,
+            messages: prompts,
+        };
+        this.promptTokensEstimate = estimatedTokens;
+    }
+    async handleLLMError(error, runId, parentRunId) {
+        var _a;
+        const runDetails = (_a = this.runsMap[runId]) !== null && _a !== void 0 ? _a : { index: Object.keys(this.runsMap).length };
+        if (typeof error === 'object' && (error === null || error === void 0 ? void 0 : error.hasOwnProperty('headers'))) {
+            const errorWithHeaders = error;
+            Object.keys(errorWithHeaders.headers).forEach((key) => {
+                if (!key.startsWith('x-')) {
+                    delete errorWithHeaders.headers[key];
+                }
+            });
+        }
+        if (error instanceof n8n_workflow_1.NodeError) {
+            if (this.options.errorDescriptionMapper) {
+                error.description = this.options.errorDescriptionMapper(error);
+            }
+            this.executionFunctions.addOutputData(this.connectionType, runDetails.index, error);
+        }
+        else {
+            this.executionFunctions.addOutputData(this.connectionType, runDetails.index, new n8n_workflow_1.NodeOperationError(this.executionFunctions.getNode(), error, {
+                functionality: 'configuration-node',
+            }));
+        }
+        (0, helpers_1.logAiEvent)(this.executionFunctions, 'ai-llm-errored', {
+            error: Object.keys(error).length === 0 ? error.toString() : error,
+            runId,
+            parentRunId,
+        });
+    }
+}
+exports.N8nLlmTracing = N8nLlmTracing;
+//# sourceMappingURL=N8nLlmTracing.js.map

package/dist/nodes/llms/N8nLlmTracing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"N8nLlmTracing.js","sourceRoot":"","sources":["../../../nodes/llms/N8nLlmTracing.ts"],"names":[],"mappings":";;;AAAA,yDAAqE;AAErE,+DAA+E;AAQ/E,6DAAkE;AAClE,mCAA8B;AAE9B,+CAAkF;AAElF,4CAA4C;AAc5C,MAAM,uBAAuB,GAAG,QAAQ,CAAC;AACzC,MAAa,aAAc,SAAQ,0BAAmB;IAoCrD,YACS,kBAAwC,EAChD,OAGC;QAED,KAAK,EAAE,CAAC;QANA,uBAAkB,GAAlB,kBAAkB,CAAsB;QApCjD,SAAI,GAAG,eAAe,CAAC;QAIvB,kBAAa,GAAG,IAAI,CAAC;QAErB,mBAAc,GAAG,kCAAmB,CAAC,eAAe,CAAC;QAErD,yBAAoB,GAAG,CAAC,CAAC;QAEzB,6BAAwB,GAAG,CAAC,CAAC;QAQ7B,YAAO,GAA8B,EAAE,CAAC;QAExC,YAAO,GAAG;YAET,iBAAiB,EAAE,CAAC,SAAiC,EAAE,EAAE;;gBACxD,MAAM,gBAAgB,GAAG,MAAC,MAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,UAAU,0CAAE,gBAA2B,mCAAI,CAAC,CAAC;gBAClF,MAAM,YAAY,GAAG,MAAC,MAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,UAAU,0CAAE,YAAuB,mCAAI,CAAC,CAAC;gBAE1E,OAAO;oBACN,gBAAgB;oBAChB,YAAY;oBACZ,WAAW,EAAE,gBAAgB,GAAG,YAAY;iBAC5C,CAAC;YACH,CAAC;YACD,sBAAsB,EAAE,CAAC,KAAgB,EAAE,EAAE,CAAC,KAAK,CAAC,WAAW;SAC/D,CAAC;QAUD,IAAI,CAAC,OAAO,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,OAAO,EAAE,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,4BAA4B,CAAC,WAAqC;QACvE,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACtE,OAAO,MAAM,IAAI,CAAC,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,4BAA4B,CAAC,IAAc;QAChD,MAAM,cAAc,GAAG,IAAA,8BAAuB,EAAC,uBAAuB,CAAC,CAAC;QACxE,MAAM,OAAO,GAAG,MAAM,IAAA,2BAAgB,EAAC,cAAc,CAAC,CAAC;QAEvD,MAAM,iBAAiB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC1C,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CACrD,CAAC;QAEF,OAAO,iBAAiB,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAiB,EAAE,KAAa;;QAGlD,MAAM,UAAU,GAAG,MAAA,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mCAAI,EAAE,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;QAEtF,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACnD,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,aAAI,EAAC,CAAC,EAAE,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC,CAAC,CACnD,CAAC;QAEF,MAAM,kBAAkB,GAAG;YAC1B,gBAAgB,EAAE,CAAC;YACnB,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,CAAC;SACd,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAEpE,IAAI,MAAM,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,kBAAkB,CAAC,gBAAgB,GAAG,MAAM,IAAI,CAAC,4BAA4B,CAC5E,MAAM,CAAC,WAAW,CAClB,CAAC;YAEF,kBAAkB,CAAC,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC;YAC5D,kBAAkB,CAAC,WAAW;gBAC7B,kBAAkB,CAAC,gBAAgB,GAAG,IAAI,CAAC,oBAAoB,CAAC;QAClE,CAAC;QACD,MAAM,QAAQ,GAIV;YACH,QAAQ,EAAE,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE;SAC7C,CAAC;QAGF,IAAI,UAAU,CAAC,gBAAgB,GAAG,CAAC,EAAE,CAAC;YACrC,QAAQ,CAAC,UAAU,GAAG,UAAU,CAAC;QAClC,CAAC;aAAM,CAAC;YACP,QAAQ,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;QAClD,CAAC;QAED,MAAM,cAAc,GACnB,OAAO,UAAU,CAAC,QAAQ,KAAK,QAAQ;YACtC,CAAC,CAAC,UAAU,CAAC,QAAQ;YACrB,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gBACpC,IAAI,OAAO,OAAO,KAAK,QAAQ;oBAAE,OAAO,OAAO,CAAC;gBAChD,IAAI,OAAO,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,MAAM,CAAA,KAAK,UAAU;oBAAE,OAAO,OAAO,CAAC,MAAM,EAAE,CAAC;gBAEnE,OAAO,OAAO,CAAC;YAChB,CAAC,CAAC,CAAC;QAEN,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC,IAAI,CAAC,cAAc,EAAE,UAAU,CAAC,KAAK,EAAE;YAC5E,CAAC,EAAE,IAAI,EAAE,EAAE,GAAG,QAAQ,EAAE,EAAE,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAA,oBAAU,EAAC,IAAI,CAAC,kBAAkB,EAAE,yBAAyB,EAAE;YAC9D,QAAQ,EAAE,cAAc;YACxB,OAAO,EAAE,UAAU,CAAC,OAAO;YAC3B,QAAQ;SACR,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,GAAe,EAAE,OAAiB,EAAE,KAAa;QACrE,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,4BAA4B,CAAC,OAAO,CAAC,CAAC;QAEzE,MAAM,OAAO,GAAG,GAAG,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC;QAC9D,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,IAAI,CAAC,cAAc,EAAE;YAC3E;gBACC;oBACC,IAAI,EAAE;wBACL,QAAQ,EAAE,OAAO;wBACjB,eAAe;wBACf,OAAO;qBACP;iBACD;aACD;SACD,CAAC,CAAC;QAGH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG;YACrB,KAAK;YACL,OAAO;YACP,QAAQ,EAAE,OAAO;SACjB,CAAC;QACF,IAAI,CAAC,oBAAoB,GAAG,eAAe,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,cAAc,CACnB,KAA0B,EAC1B,KAAa,EACb,WAAgC;;QAEhC,MAAM,UAAU,GAAG,MAAA,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mCAAI,EAAE,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;QAGtF,IAAI,OAAO,KAAK,KAAK,QAAQ,KAAI,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,cAAc,CAAC,SAAS,CAAC,CAAA,EAAE,CAAC;YACnE,MAAM,gBAAgB,GAAG,KAA6C,CAAC;YAEvE,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACrD,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC3B,OAAO,gBAAgB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;gBACtC,CAAC;YACF,CAAC,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,YAAY,wBAAS,EAAE,CAAC;YAChC,IAAI,IAAI,CAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC;gBACzC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;YAChE,CAAC;YAED,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC,IAAI,CAAC,cAAc,EAAE,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACrF,CAAC;aAAM,CAAC;YAEP,IAAI,CAAC,kBAAkB,CAAC,aAAa,CACpC,IAAI,CAAC,cAAc,EACnB,UAAU,CAAC,KAAK,EAChB,IAAI,iCAAkB,CAAC,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,EAAE,KAAmB,EAAE;gBAC9E,aAAa,EAAE,oBAAoB;aACnC,CAAC,CACF,CAAC;QACH,CAAC;QAED,IAAA,oBAAU,EAAC,IAAI,CAAC,kBAAkB,EAAE,gBAAgB,EAAE;YACrD,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,KAAK;YACjE,KAAK;YACL,WAAW;SACX,CAAC,CAAC;IACJ,CAAC;CACD;AA9LD,sCA8LC"}

package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.d.ts

@@ -0,0 +1 @@
+export declare const n8nDefaultFailedAttemptHandler: (error: any) => void;

package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.n8nDefaultFailedAttemptHandler = void 0;
+const STATUS_NO_RETRY = [
+    400,
+    401,
+    402,
+    403,
+    404,
+    405,
+    406,
+    407,
+    409,
+];
+const n8nDefaultFailedAttemptHandler = (error) => {
+    var _a, _b, _c, _d, _e, _f;
+    if (((_b = (_a = error === null || error === void 0 ? void 0 : error.message) === null || _a === void 0 ? void 0 : _a.startsWith) === null || _b === void 0 ? void 0 : _b.call(_a, 'Cancel')) ||
+        ((_d = (_c = error === null || error === void 0 ? void 0 : error.message) === null || _c === void 0 ? void 0 : _c.startsWith) === null || _d === void 0 ? void 0 : _d.call(_c, 'AbortError')) ||
+        (error === null || error === void 0 ? void 0 : error.name) === 'AbortError') {
+        throw error;
+    }
+    if ((error === null || error === void 0 ? void 0 : error.code) === 'ECONNABORTED') {
+        throw error;
+    }
+    const status = (_f = (_e = error === null || error === void 0 ? void 0 : error.response) === null || _e === void 0 ? void 0 : _e.status) !== null && _f !== void 0 ? _f : error === null || error === void 0 ? void 0 : error.status;
+    if (status && STATUS_NO_RETRY.includes(+status)) {
+        throw error;
+    }
+};
+exports.n8nDefaultFailedAttemptHandler = n8nDefaultFailedAttemptHandler;
+//# sourceMappingURL=n8nDefaultFailedAttemptHandler.js.map

package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"n8nDefaultFailedAttemptHandler.js","sourceRoot":"","sources":["../../../nodes/llms/n8nDefaultFailedAttemptHandler.ts"],"names":[],"mappings":";;;AAAA,MAAM,eAAe,GAAG;IACvB,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;CACH,CAAC;AASK,MAAM,8BAA8B,GAAG,CAAC,KAAU,EAAE,EAAE;;IAC5D,IAEC,CAAA,MAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,OAAO,0CAAE,UAAU,mDAAG,QAAQ,CAAC;SACtC,MAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,OAAO,0CAAE,UAAU,mDAAG,YAAY,CAAC,CAAA;QAC1C,CAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,IAAI,MAAK,YAAY,EAC3B,CAAC;QACF,MAAM,KAAK,CAAC;IACb,CAAC;IAGD,IAAI,CAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,IAAI,MAAK,cAAc,EAAE,CAAC;QACpC,MAAM,KAAK,CAAC;IACb,CAAC;IAED,MAAM,MAAM,GAEX,MAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,QAAQ,0CAAE,MAAM,mCAAI,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,MAAM,CAAC;IAC1C,IAAI,MAAM,IAAI,eAAe,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;QACjD,MAAM,KAAK,CAAC;IACb,CAAC;AACF,CAAC,CAAC;AArBW,QAAA,8BAA8B,kCAqBzC"}

package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.test.js

@@ -0,0 +1,53 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const n8nDefaultFailedAttemptHandler_1 = require("./n8nDefaultFailedAttemptHandler");
+class MockHttpError extends Error {
+    constructor(message, code) {
+        super(message);
+        this.response = { status: code };
+    }
+}
+describe('n8nDefaultFailedAttemptHandler', () => {
+    it('should throw error if message starts with "Cancel"', () => {
+        const error = new Error('Cancel operation');
+        expect(() => (0, n8nDefaultFailedAttemptHandler_1.n8nDefaultFailedAttemptHandler)(error)).toThrow(error);
+    });
+    it('should throw error if message starts with "AbortError"', () => {
+        const error = new Error('AbortError occurred');
+        expect(() => (0, n8nDefaultFailedAttemptHandler_1.n8nDefaultFailedAttemptHandler)(error)).toThrow(error);
+    });
+    it('should throw error if name is "AbortError"', () => {
+        class MockAbortError extends Error {
+            constructor() {
+                super('Some error');
+                this.name = 'AbortError';
+            }
+        }
+        const error = new MockAbortError();
+        expect(() => (0, n8nDefaultFailedAttemptHandler_1.n8nDefaultFailedAttemptHandler)(error)).toThrow(error);
+    });
+    it('should throw error if code is "ECONNABORTED"', () => {
+        class MockAbortError extends Error {
+            constructor() {
+                super('Some error');
+                this.code = 'ECONNABORTED';
+            }
+        }
+        const error = new MockAbortError();
+        expect(() => (0, n8nDefaultFailedAttemptHandler_1.n8nDefaultFailedAttemptHandler)(error)).toThrow(error);
+    });
+    it('should throw error if status is in STATUS_NO_RETRY', () => {
+        const error = new MockHttpError('Some error', 400);
+        expect(() => (0, n8nDefaultFailedAttemptHandler_1.n8nDefaultFailedAttemptHandler)(error)).toThrow(error);
+    });
+    it('should not throw error if status is not in STATUS_NO_RETRY', () => {
+        const error = new MockHttpError('Some error', 500);
+        error.response = { status: 500 };
+        expect(() => (0, n8nDefaultFailedAttemptHandler_1.n8nDefaultFailedAttemptHandler)(error)).not.toThrow();
+    });
+    it('should not throw error if no conditions are met', () => {
+        const error = new Error('Some random error');
+        expect(() => (0, n8nDefaultFailedAttemptHandler_1.n8nDefaultFailedAttemptHandler)(error)).not.toThrow();
+    });
+});
+//# sourceMappingURL=n8nDefaultFailedAttemptHandler.test.js.map

package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"n8nDefaultFailedAttemptHandler.test.js","sourceRoot":"","sources":["../../../nodes/llms/n8nDefaultFailedAttemptHandler.test.ts"],"names":[],"mappings":";;AAAA,qFAAkF;AAElF,MAAM,aAAc,SAAQ,KAAK;IAGhC,YAAY,OAAe,EAAE,IAAY;QACxC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,QAAQ,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IAClC,CAAC;CACD;AAED,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC/C,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC7D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QAC5C,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,+DAA8B,EAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QACjE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC/C,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,+DAA8B,EAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACrD,MAAM,cAAe,SAAQ,KAAK;YACjC;gBACC,KAAK,CAAC,YAAY,CAAC,CAAC;gBACpB,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC;YAC1B,CAAC;SACD;QAED,MAAM,KAAK,GAAG,IAAI,cAAc,EAAE,CAAC;QAEnC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,+DAA8B,EAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACvD,MAAM,cAAe,SAAQ,KAAK;YAGjC;gBACC,KAAK,CAAC,YAAY,CAAC,CAAC;gBACpB,IAAI,CAAC,IAAI,GAAG,cAAc,CAAC;YAC5B,CAAC;SACD;QAED,MAAM,KAAK,GAAG,IAAI,cAAc,EAAE,CAAC;QACnC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,+DAA8B,EAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC7D,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;QACnD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,+DAA8B,EAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACrE,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;QACnD,KAAK,CAAC,QAAQ,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;QAEjC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,+DAA8B,EAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QAC1D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QAC7C,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,+DAA8B,EAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/dist/nodes/llms/n8nLlmFailedAttemptHandler.d.ts

@@ -0,0 +1,3 @@
+import type { FailedAttemptHandler } from '@langchain/core/dist/utils/async_caller';
+import type { ISupplyDataFunctions } from 'n8n-workflow';
+export declare const makeN8nLlmFailedAttemptHandler: (ctx: ISupplyDataFunctions, handler?: FailedAttemptHandler) => FailedAttemptHandler;

package/dist/nodes/llms/n8nLlmFailedAttemptHandler.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.makeN8nLlmFailedAttemptHandler = void 0;
+const n8n_workflow_1 = require("n8n-workflow");
+const n8nDefaultFailedAttemptHandler_1 = require("./n8nDefaultFailedAttemptHandler");
+const makeN8nLlmFailedAttemptHandler = (ctx, handler) => {
+    return (error) => {
+        try {
+            handler === null || handler === void 0 ? void 0 : handler(error);
+            (0, n8nDefaultFailedAttemptHandler_1.n8nDefaultFailedAttemptHandler)(error);
+        }
+        catch (e) {
+            const apiError = new n8n_workflow_1.NodeApiError(ctx.getNode(), e, {
+                functionality: 'configuration-node',
+            });
+            throw apiError;
+        }
+        if ((error === null || error === void 0 ? void 0 : error.retriesLeft) > 0) {
+            return;
+        }
+        const apiError = new n8n_workflow_1.NodeApiError(ctx.getNode(), error, {
+            functionality: 'configuration-node',
+        });
+        throw apiError;
+    };
+};
+exports.makeN8nLlmFailedAttemptHandler = makeN8nLlmFailedAttemptHandler;
+//# sourceMappingURL=n8nLlmFailedAttemptHandler.js.map