@kyuda/n8n-nodes-databricks 1.0.2

package/dist/nodes/agents/DatabricksAiAgent/src/utils/helpers.js

@@ -0,0 +1,221 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getConnectedTools = void 0;
+exports.getMetadataFiltersValues = getMetadataFiltersValues;
+exports.isBaseChatMemory = isBaseChatMemory;
+exports.isBaseChatMessageHistory = isBaseChatMessageHistory;
+exports.isChatInstance = isChatInstance;
+exports.isToolsInstance = isToolsInstance;
+exports.getPromptInputByType = getPromptInputByType;
+exports.getSessionId = getSessionId;
+exports.logAiEvent = logAiEvent;
+exports.serializeChatHistory = serializeChatHistory;
+exports.escapeSingleCurlyBrackets = escapeSingleCurlyBrackets;
+exports.unwrapNestedOutput = unwrapNestedOutput;
+exports.hasLongSequentialRepeat = hasLongSequentialRepeat;
+const agents_1 = require("@langchain/classic/agents");
+const n8n_workflow_1 = require("n8n-workflow");
+const N8nTool_1 = require("./N8nTool");
+function hasMethods(obj, ...methodNames) {
+    return methodNames.every((methodName) => typeof obj === 'object' &&
+        obj !== null &&
+        methodName in obj &&
+        typeof obj[methodName] === 'function');
+}
+function getMetadataFiltersValues(ctx, itemIndex) {
+    const options = ctx.getNodeParameter('options', itemIndex, {});
+    if (options.metadata) {
+        const { metadataValues: metadata } = options.metadata;
+        if (metadata.length > 0) {
+            return metadata.reduce((acc, { name, value }) => ({ ...acc, [name]: value }), {});
+        }
+    }
+    if (options.searchFilterJson) {
+        return ctx.getNodeParameter('options.searchFilterJson', itemIndex, '', {
+            ensureType: 'object',
+        });
+    }
+    return undefined;
+}
+function isBaseChatMemory(obj) {
+    return hasMethods(obj, 'loadMemoryVariables', 'saveContext');
+}
+function isBaseChatMessageHistory(obj) {
+    return hasMethods(obj, 'getMessages', 'addMessage');
+}
+function isChatInstance(model) {
+    var _a;
+    const namespace = (_a = model === null || model === void 0 ? void 0 : model.lc_namespace) !== null && _a !== void 0 ? _a : [];
+    return namespace.includes('chat_models');
+}
+function isToolsInstance(model) {
+    var _a;
+    const namespace = (_a = model === null || model === void 0 ? void 0 : model.lc_namespace) !== null && _a !== void 0 ? _a : [];
+    return namespace.includes('tools');
+}
+function getPromptInputByType(options) {
+    const { ctx, i, promptTypeKey, inputKey } = options;
+    const promptType = ctx.getNodeParameter(promptTypeKey, i, 'define');
+    let input;
+    if (promptType === 'auto') {
+        input = ctx.evaluateExpression('{{ $json["chatInput"] }}', i);
+    }
+    else {
+        input = ctx.getNodeParameter(inputKey, i);
+    }
+    if (input === undefined) {
+        throw new n8n_workflow_1.NodeOperationError(ctx.getNode(), 'No prompt specified', {
+            description: "Expected to find the prompt in an input field called 'chatInput' (this is what the chat trigger node outputs). To use something else, change the 'Prompt' parameter",
+        });
+    }
+    return input;
+}
+function getSessionId(ctx, itemIndex, selectorKey = 'sessionIdType', autoSelect = 'fromInput', customKey = 'sessionKey') {
+    var _a, _b;
+    let sessionId = '';
+    const selectorType = ctx.getNodeParameter(selectorKey, itemIndex);
+    if (selectorType === autoSelect) {
+        if ('getBodyData' in ctx) {
+            const bodyData = (_a = ctx.getBodyData()) !== null && _a !== void 0 ? _a : {};
+            sessionId = bodyData.sessionId;
+        }
+        else {
+            sessionId = ctx.evaluateExpression('{{ $json.sessionId }}', itemIndex);
+            if (!sessionId || sessionId === undefined) {
+                try {
+                    const chatTrigger = (_b = ctx.getChatTrigger) === null || _b === void 0 ? void 0 : _b.call(ctx);
+                    if (chatTrigger) {
+                        sessionId = ctx.evaluateExpression(`{{ $('${chatTrigger.name}').first().json.sessionId }}`, itemIndex);
+                    }
+                }
+                catch (error) { }
+            }
+        }
+        if (sessionId === '' || sessionId === undefined) {
+            throw new n8n_workflow_1.NodeOperationError(ctx.getNode(), 'No session ID found', {
+                description: "Expected to find the session ID in an input field called 'sessionId' (this is what the chat trigger node outputs). To use something else, change the 'Session ID' parameter",
+                itemIndex,
+            });
+        }
+    }
+    else {
+        sessionId = ctx.getNodeParameter(customKey, itemIndex, '');
+        if (sessionId === '' || sessionId === undefined) {
+            throw new n8n_workflow_1.NodeOperationError(ctx.getNode(), 'Key parameter is empty', {
+                description: "Provide a key to use as session ID in the 'Key' parameter or use the 'Connected Chat Trigger Node' option to use the session ID from your Chat Trigger",
+                itemIndex,
+            });
+        }
+    }
+    return sessionId;
+}
+function logAiEvent(executeFunctions, event, data) {
+    try {
+        executeFunctions.logAiEvent(event, data ? (0, n8n_workflow_1.jsonStringify)(data) : undefined);
+    }
+    catch (error) {
+        executeFunctions.logger.debug(`Error logging AI event: ${event}`);
+    }
+}
+function serializeChatHistory(chatHistory) {
+    return chatHistory
+        .map((chatMessage) => {
+        if (chatMessage._getType() === 'human') {
+            return `Human: ${chatMessage.content}`;
+        }
+        else if (chatMessage._getType() === 'ai') {
+            return `Assistant: ${chatMessage.content}`;
+        }
+        else {
+            return `${chatMessage.content}`;
+        }
+    })
+        .join('\n');
+}
+function escapeSingleCurlyBrackets(text) {
+    if (text === undefined)
+        return undefined;
+    let result = text;
+    result = result
+        .replace(/(?<!{){{{(?!{)/g, '{{{{')
+        .replace(/(?<!})}}}(?!})/g, '}}}}')
+        .replace(/(?<!{){(?!{)/g, '{{')
+        .replace(/(?<!})}(?!})/g, '}}');
+    return result;
+}
+const getConnectedTools = async (ctx, enforceUniqueNames, convertStructuredTool = true, escapeCurlyBrackets = false) => {
+    var _a, _b;
+    const connectedTools = ((_a = (await ctx.getInputConnectionData('ai_tool', 0))) !== null && _a !== void 0 ? _a : []).flatMap((toolOrToolkit) => {
+        if (toolOrToolkit instanceof agents_1.Toolkit || (typeof toolOrToolkit.getTools === 'function')) {
+            return toolOrToolkit.getTools();
+        }
+        return toolOrToolkit;
+    });
+    if (!enforceUniqueNames)
+        return connectedTools;
+    const seenNames = new Set();
+    const finalTools = [];
+    for (const tool of connectedTools) {
+        const { name } = tool;
+        if (seenNames.has(name)) {
+            throw new n8n_workflow_1.NodeOperationError(ctx.getNode(), `You have multiple tools with the same name: '${name}', please rename them to avoid conflicts`);
+        }
+        seenNames.add(name);
+        if (escapeCurlyBrackets) {
+            tool.description = (_b = escapeSingleCurlyBrackets(tool.description)) !== null && _b !== void 0 ? _b : tool.description;
+        }
+        if (convertStructuredTool && tool instanceof N8nTool_1.N8nTool) {
+            finalTools.push(tool.asDynamicTool());
+        }
+        else {
+            finalTools.push(tool);
+        }
+    }
+    return finalTools;
+};
+exports.getConnectedTools = getConnectedTools;
+function unwrapNestedOutput(output) {
+    if ('output' in output &&
+        Object.keys(output).length === 1 &&
+        typeof output.output === 'object' &&
+        output.output !== null &&
+        'output' in output.output &&
+        Object.keys(output.output).length === 1) {
+        return output.output;
+    }
+    return output;
+}
+function hasLongSequentialRepeat(text, threshold = 1000) {
+    try {
+        if (text === null ||
+            typeof text !== 'string' ||
+            text.length === 0 ||
+            threshold <= 0 ||
+            text.length < threshold) {
+            return false;
+        }
+        const iterator = text[Symbol.iterator]();
+        let prev = iterator.next();
+        if (prev.done) {
+            return false;
+        }
+        let count = 1;
+        for (const char of iterator) {
+            if (char === prev.value) {
+                count++;
+                if (count >= threshold) {
+                    return true;
+                }
+            }
+            else {
+                count = 1;
+                prev = { value: char, done: false };
+            }
+        }
+        return false;
+    }
+    catch (error) {
+        return false;
+    }
+}
+//# sourceMappingURL=helpers.js.map

package/dist/nodes/agents/DatabricksAiAgent/src/utils/helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../../../../../nodes/agents/DatabricksAiAgent/src/utils/helpers.ts"],"names":[],"mappings":";;;AA6BA,4DAyBC;AAED,4CAEC;AAED,4DAEC;AAED,wCAIC;AAED,0CAIC;AAED,oDAwBC;AAED,oCAsDC;AAED,gCAUC;AAED,oDAYC;AAED,8DAgBC;AAsDD,gDAaC;AASD,0DAsCC;AArTD,sDAAoD;AAEpD,+CAAiE;AASjE,uCAAoC;AAGpC,SAAS,UAAU,CAAI,GAAY,EAAE,GAAG,WAAmC;IACvE,OAAO,WAAW,CAAC,KAAK,CACpB,CAAC,UAAU,EAAE,EAAE,CACX,OAAO,GAAG,KAAK,QAAQ;QACvB,GAAG,KAAK,IAAI;QACZ,UAAU,IAAI,GAAG;QACjB,OAAQ,GAAwC,CAAC,UAAU,CAAC,KAAK,UAAU,CAClF,CAAC;AACN,CAAC;AAED,SAAgB,wBAAwB,CACpC,GAA6C,EAC7C,SAAiB;IAEjB,MAAM,OAAO,GAAG,GAAG,CAAC,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;IAE/D,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACnB,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,QAK5C,CAAC;QACF,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QACtF,CAAC;IACL,CAAC;IAED,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;QAC3B,OAAO,GAAG,CAAC,gBAAgB,CAAC,0BAA0B,EAAE,SAAS,EAAE,EAAE,EAAE;YACnE,UAAU,EAAE,QAAQ;SACvB,CAA0B,CAAC;IAChC,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAgB,gBAAgB,CAAC,GAAY;IACzC,OAAO,UAAU,CAAiB,GAAG,EAAE,qBAAqB,EAAE,aAAa,CAAC,CAAC;AACjF,CAAC;AAED,SAAgB,wBAAwB,CAAC,GAAY;IACjD,OAAO,UAAU,CAAyB,GAAG,EAAE,aAAa,EAAE,YAAY,CAAC,CAAC;AAChF,CAAC;AAED,SAAgB,cAAc,CAAC,KAAc;;IACzC,MAAM,SAAS,GAAG,MAAC,KAAiB,aAAjB,KAAK,uBAAL,KAAK,CAAc,YAAY,mCAAI,EAAE,CAAC;IAEzD,OAAO,SAAS,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;AAC7C,CAAC;AAED,SAAgB,eAAe,CAAC,KAAc;;IAC1C,MAAM,SAAS,GAAG,MAAC,KAAc,aAAd,KAAK,uBAAL,KAAK,CAAW,YAAY,mCAAI,EAAE,CAAC;IAEtD,OAAO,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AACvC,CAAC;AAED,SAAgB,oBAAoB,CAAC,OAKpC;IACG,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IACpD,MAAM,UAAU,GAAG,GAAG,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,EAAE,QAAQ,CAAW,CAAC;IAE9E,IAAI,KAAK,CAAC;IACV,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;QACxB,KAAK,GAAG,GAAG,CAAC,kBAAkB,CAAC,0BAA0B,EAAE,CAAC,CAAW,CAAC;IAC5E,CAAC;SAAM,CAAC;QACJ,KAAK,GAAG,GAAG,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAW,CAAC;IACxD,CAAC;IAED,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACtB,MAAM,IAAI,iCAAkB,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,qBAAqB,EAAE;YAC/D,WAAW,EACP,qKAAqK;SAC5K,CAAC,CAAC;IACP,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAgB,YAAY,CACxB,GAA6C,EAC7C,SAAiB,EACjB,WAAW,GAAG,eAAe,EAC7B,UAAU,GAAG,WAAW,EACxB,SAAS,GAAG,YAAY;;IAExB,IAAI,SAAS,GAAG,EAAE,CAAC;IACnB,MAAM,YAAY,GAAG,GAAG,CAAC,gBAAgB,CAAC,WAAW,EAAE,SAAS,CAAW,CAAC;IAE5E,IAAI,YAAY,KAAK,UAAU,EAAE,CAAC;QAG9B,IAAI,aAAa,IAAI,GAAG,EAAE,CAAC;YACvB,MAAM,QAAQ,GAAG,MAAA,GAAG,CAAC,WAAW,EAAE,mCAAI,EAAE,CAAC;YACzC,SAAS,GAAG,QAAQ,CAAC,SAAmB,CAAC;QAC7C,CAAC;aAAM,CAAC;YACJ,SAAS,GAAG,GAAG,CAAC,kBAAkB,CAAC,uBAAuB,EAAE,SAAS,CAAW,CAAC;YAGjF,IAAI,CAAC,SAAS,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBACxC,IAAI,CAAC;oBAED,MAAM,WAAW,GAAG,MAAA,GAAG,CAAC,cAAc,mDAAI,CAAC;oBAE3C,IAAI,WAAW,EAAE,CAAC;wBACd,SAAS,GAAG,GAAG,CAAC,kBAAkB,CAC9B,SAAS,WAAW,CAAC,IAAI,8BAA8B,EACvD,SAAS,CACF,CAAC;oBAChB,CAAC;gBACL,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC,CAAC,CAAC;YACvB,CAAC;QACL,CAAC;QAED,IAAI,SAAS,KAAK,EAAE,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC9C,MAAM,IAAI,iCAAkB,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,qBAAqB,EAAE;gBAC/D,WAAW,EACP,6KAA6K;gBACjL,SAAS;aACZ,CAAC,CAAC;QACP,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,SAAS,GAAG,GAAG,CAAC,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE,EAAE,CAAW,CAAC;QACrE,IAAI,SAAS,KAAK,EAAE,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC9C,MAAM,IAAI,iCAAkB,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,wBAAwB,EAAE;gBAClE,WAAW,EACP,wJAAwJ;gBAC5J,SAAS;aACZ,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAgB,UAAU,CACtB,gBAA0D,EAC1D,KAAc,EACd,IAAkB;IAElB,IAAI,CAAC;QACD,gBAAgB,CAAC,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,IAAA,4BAAa,EAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAC/E,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,KAAK,EAAE,CAAC,CAAC;IACtE,CAAC;AACL,CAAC;AAED,SAAgB,oBAAoB,CAAC,WAA0B;IAC3D,OAAO,WAAW;SACb,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE;QACjB,IAAI,WAAW,CAAC,QAAQ,EAAE,KAAK,OAAO,EAAE,CAAC;YACrC,OAAO,UAAU,WAAW,CAAC,OAAO,EAAE,CAAC;QAC3C,CAAC;aAAM,IAAI,WAAW,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC;YACzC,OAAO,cAAc,WAAW,CAAC,OAAO,EAAE,CAAC;QAC/C,CAAC;aAAM,CAAC;YACJ,OAAO,GAAG,WAAW,CAAC,OAAO,EAAE,CAAC;QACpC,CAAC;IACL,CAAC,CAAC;SACD,IAAI,CAAC,IAAI,CAAC,CAAC;AACpB,CAAC;AAED,SAAgB,yBAAyB,CAAC,IAAa;IACnD,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IAEzC,IAAI,MAAM,GAAG,IAAI,CAAC;IAElB,MAAM,GAAG,MAAM;SAEV,OAAO,CAAC,iBAAiB,EAAE,MAAM,CAAC;SAClC,OAAO,CAAC,iBAAiB,EAAE,MAAM,CAAC;SAGlC,OAAO,CAAC,eAAe,EAAE,IAAI,CAAC;SAE9B,OAAO,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;IAEpC,OAAO,MAAM,CAAC;AAClB,CAAC;AAEM,MAAM,iBAAiB,GAAG,KAAK,EAClC,GAAiE,EACjE,kBAA2B,EAC3B,wBAAiC,IAAI,EACrC,sBAA+B,KAAK,EACtC,EAAE;;IACA,MAAM,cAAc,GAAG,CACnB,MAAC,CAAC,MAAM,GAAG,CAAC,sBAAsB,CAAC,SAAS,EAAE,CAAC,CAAC,CAA2B,mCAC3E,EAAE,CACL,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,EAAE;QAExB,IAAI,aAAa,YAAY,gBAAO,IAAI,CAAC,OAAQ,aAAqB,CAAC,QAAQ,KAAK,UAAU,CAAC,EAAE,CAAC;YAC9F,OAAQ,aAAyB,CAAC,QAAQ,EAAY,CAAC;QAC3D,CAAC;QAED,OAAO,aAAa,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,kBAAkB;QAAE,OAAO,cAAc,CAAC;IAE/C,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IAEpC,MAAM,UAAU,GAAW,EAAE,CAAC;IAE9B,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;QAChC,MAAM,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;QACtB,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,iCAAkB,CACxB,GAAG,CAAC,OAAO,EAAE,EACb,gDAAgD,IAAI,0CAA0C,CACjG,CAAC;QACN,CAAC;QACD,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAEpB,IAAI,mBAAmB,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,GAAG,MAAA,yBAAyB,CAAC,IAAI,CAAC,WAAW,CAAC,mCAAI,IAAI,CAAC,WAAW,CAAC;QACvF,CAAC;QAED,IAAI,qBAAqB,IAAI,IAAI,YAAY,iBAAO,EAAE,CAAC;YACnD,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;QAC1C,CAAC;aAAM,CAAC;YACJ,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;IACL,CAAC;IAED,OAAO,UAAU,CAAC;AACtB,CAAC,CAAC;AA9CW,QAAA,iBAAiB,qBA8C5B;AAMF,SAAgB,kBAAkB,CAAC,MAA+B;IAC9D,IACI,QAAQ,IAAI,MAAM;QAClB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC;QAChC,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ;QACjC,MAAM,CAAC,MAAM,KAAK,IAAI;QACtB,QAAQ,IAAI,MAAM,CAAC,MAAM;QACzB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC,EACzC,CAAC;QACC,OAAO,MAAM,CAAC,MAAiC,CAAC;IACpD,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC;AASD,SAAgB,uBAAuB,CAAC,IAAY,EAAE,SAAS,GAAG,IAAI;IAClE,IAAI,CAAC;QAED,IACI,IAAI,KAAK,IAAI;YACb,OAAO,IAAI,KAAK,QAAQ;YACxB,IAAI,CAAC,MAAM,KAAK,CAAC;YACjB,SAAS,IAAI,CAAC;YACd,IAAI,CAAC,MAAM,GAAG,SAAS,EACzB,CAAC;YACC,OAAO,KAAK,CAAC;QACjB,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzC,IAAI,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;QAE3B,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACZ,OAAO,KAAK,CAAC;QACjB,CAAC;QAED,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,IAAI,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC;gBACtB,KAAK,EAAE,CAAC;gBACR,IAAI,KAAK,IAAI,SAAS,EAAE,CAAC;oBACrB,OAAO,IAAI,CAAC;gBAChB,CAAC;YACL,CAAC;iBAAM,CAAC;gBACJ,KAAK,GAAG,CAAC,CAAC;gBACV,IAAI,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;YACxC,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAEb,OAAO,KAAK,CAAC;IACjB,CAAC;AACL,CAAC"}

package/dist/nodes/agents/DatabricksAiAgent/src/utils/options.d.ts

@@ -0,0 +1,2 @@
+import type { INodeProperties } from 'n8n-workflow';
+export declare const commonOptions: INodeProperties[];

package/dist/nodes/agents/DatabricksAiAgent/src/utils/options.js

@@ -0,0 +1,38 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.commonOptions = void 0;
+const prompt_1 = require("./prompt");
+exports.commonOptions = [
+    {
+        displayName: 'System Message',
+        name: 'systemMessage',
+        type: 'string',
+        default: prompt_1.SYSTEM_MESSAGE,
+        description: 'The message that will be sent to the agent before the conversation starts',
+        typeOptions: {
+            rows: 6,
+        },
+    },
+    {
+        displayName: 'Max Iterations',
+        name: 'maxIterations',
+        type: 'number',
+        default: 10,
+        description: 'The maximum number of iterations the agent will run before stopping',
+    },
+    {
+        displayName: 'Return Intermediate Steps',
+        name: 'returnIntermediateSteps',
+        type: 'boolean',
+        default: false,
+        description: 'Whether or not the output should include intermediate steps the agent took',
+    },
+    {
+        displayName: 'Automatically Passthrough Binary Images',
+        name: 'passthroughBinaryImages',
+        type: 'boolean',
+        default: true,
+        description: 'Whether or not binary images should be automatically passed through to the agent as image type messages',
+    },
+];
+//# sourceMappingURL=options.js.map

package/dist/nodes/agents/DatabricksAiAgent/src/utils/options.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"options.js","sourceRoot":"","sources":["../../../../../../nodes/agents/DatabricksAiAgent/src/utils/options.ts"],"names":[],"mappings":";;;AAEA,qCAA0C;AAE7B,QAAA,aAAa,GAAsB;IAC5C;QACI,WAAW,EAAE,gBAAgB;QAC7B,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE,QAAQ;QACd,OAAO,EAAE,uBAAc;QACvB,WAAW,EAAE,2EAA2E;QACxF,WAAW,EAAE;YACT,IAAI,EAAE,CAAC;SACV;KACJ;IACD;QACI,WAAW,EAAE,gBAAgB;QAC7B,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE,QAAQ;QACd,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,qEAAqE;KACrF;IACD;QACI,WAAW,EAAE,2BAA2B;QACxC,IAAI,EAAE,yBAAyB;QAC/B,IAAI,EAAE,SAAS;QACf,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,4EAA4E;KAC5F;IACD;QACI,WAAW,EAAE,yCAAyC;QACtD,IAAI,EAAE,yBAAyB;QAC/B,IAAI,EAAE,SAAS;QACf,OAAO,EAAE,IAAI;QACb,WAAW,EACP,yGAAyG;KAChH;CACJ,CAAC"}

package/dist/nodes/agents/DatabricksAiAgent/src/utils/output_parsers/N8nStructuredOutputParser.d.ts

@@ -0,0 +1,12 @@
+import type { Callbacks } from '@langchain/core/callbacks/manager';
+import { StructuredOutputParser } from '@langchain/classic/output_parsers';
+import type { ISupplyDataFunctions } from 'n8n-workflow';
+import { z } from 'zod';
+export declare class N8nStructuredOutputParser extends StructuredOutputParser<any> {
+    private context;
+    constructor(context: ISupplyDataFunctions, zodSchema: z.ZodSchema<object>);
+    lc_namespace: string[];
+    parse(text: string, _callbacks?: Callbacks, errorMapper?: (error: Error) => Error): Promise<object>;
+    static fromZodJsonSchema(zodSchema: z.ZodSchema<object>, nodeVersion: number, context: ISupplyDataFunctions): Promise<N8nStructuredOutputParser>;
+    getSchema(): any;
+}

package/dist/nodes/agents/DatabricksAiAgent/src/utils/output_parsers/N8nStructuredOutputParser.js

@@ -0,0 +1,106 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.N8nStructuredOutputParser = void 0;
+const output_parsers_1 = require("@langchain/classic/output_parsers");
+const get_1 = __importDefault(require("lodash/get"));
+const n8n_workflow_1 = require("n8n-workflow");
+const zod_1 = require("zod");
+const helpers_1 = require("../helpers");
+const STRUCTURED_OUTPUT_KEY = '__structured__output';
+const STRUCTURED_OUTPUT_OBJECT_KEY = '__structured__output__object';
+const STRUCTURED_OUTPUT_ARRAY_KEY = '__structured__output__array';
+class N8nStructuredOutputParser extends output_parsers_1.StructuredOutputParser {
+    constructor(context, zodSchema) {
+        super(zodSchema);
+        this.context = context;
+        this.lc_namespace = ['langchain', 'output_parsers', 'structured'];
+    }
+    async parse(text, _callbacks, errorMapper) {
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j;
+        const { index } = this.context.addInputData(n8n_workflow_1.NodeConnectionTypes.AiOutputParser, [
+            [{ json: { action: 'parse', text } }],
+        ]);
+        try {
+            const jsonString = text.includes('```') ? text.split(/```(?:json)?/)[1] : text;
+            const json = JSON.parse(jsonString.trim());
+            const parsed = await this.schema.parseAsync(json);
+            let result = ((_c = (_b = (_a = (0, get_1.default)(parsed, [STRUCTURED_OUTPUT_KEY, STRUCTURED_OUTPUT_OBJECT_KEY])) !== null && _a !== void 0 ? _a : (0, get_1.default)(parsed, [STRUCTURED_OUTPUT_KEY, STRUCTURED_OUTPUT_ARRAY_KEY])) !== null && _b !== void 0 ? _b : (0, get_1.default)(parsed, STRUCTURED_OUTPUT_KEY)) !== null && _c !== void 0 ? _c : parsed);
+            result = (0, helpers_1.unwrapNestedOutput)(result);
+            (0, helpers_1.logAiEvent)(this.context, 'ai-output-parsed', { text, response: result });
+            this.context.addOutputData(n8n_workflow_1.NodeConnectionTypes.AiOutputParser, index, [
+                [{ json: { action: 'parse', response: result } }],
+            ]);
+            return result;
+        }
+        catch (e) {
+            const nodeError = new n8n_workflow_1.NodeOperationError(this.context.getNode(), "Model output doesn't fit required format", {
+                description: "To continue the execution when this happens, change the 'On Error' parameter in the root node's settings",
+            });
+            if (e instanceof SyntaxError) {
+                nodeError.context.outputParserFailReason = 'Invalid JSON in model output';
+            }
+            else if ((typeof text === 'string' && text.trim() === '{}') ||
+                (e instanceof zod_1.z.ZodError &&
+                    ((_d = e.issues) === null || _d === void 0 ? void 0 : _d[0]) &&
+                    ((_e = e.issues) === null || _e === void 0 ? void 0 : _e[0].code) === 'invalid_type' &&
+                    ((_g = (_f = e.issues) === null || _f === void 0 ? void 0 : _f[0].path) === null || _g === void 0 ? void 0 : _g[0]) === 'output' &&
+                    ((_h = e.issues) === null || _h === void 0 ? void 0 : _h[0].expected) === 'object' &&
+                    ((_j = e.issues) === null || _j === void 0 ? void 0 : _j[0].received) === 'undefined')) {
+                nodeError.context.outputParserFailReason = 'Model output wrapper is an empty object';
+            }
+            else if (e instanceof zod_1.z.ZodError) {
+                nodeError.context.outputParserFailReason =
+                    'Model output does not match the expected schema';
+            }
+            const errorMessage = e instanceof Error ? e.message : String(e);
+            (0, helpers_1.logAiEvent)(this.context, 'ai-output-parsed', {
+                text,
+                response: errorMessage,
+            });
+            this.context.addOutputData(n8n_workflow_1.NodeConnectionTypes.AiOutputParser, index, nodeError);
+            if (errorMapper && e instanceof Error) {
+                throw errorMapper(e);
+            }
+            throw nodeError;
+        }
+    }
+    static async fromZodJsonSchema(zodSchema, nodeVersion, context) {
+        let returnSchema;
+        if (nodeVersion === 1) {
+            returnSchema = zod_1.z.object({
+                [STRUCTURED_OUTPUT_KEY]: zod_1.z
+                    .object({
+                    [STRUCTURED_OUTPUT_OBJECT_KEY]: zodSchema.optional().nullable(),
+                    [STRUCTURED_OUTPUT_ARRAY_KEY]: zod_1.z.array(zodSchema).optional().nullable(),
+                })
+                    .describe(`Wrapper around the output data. It can only contain ${STRUCTURED_OUTPUT_OBJECT_KEY} or ${STRUCTURED_OUTPUT_ARRAY_KEY} but never both.`)
+                    .refine((data) => {
+                    return (Boolean(data[STRUCTURED_OUTPUT_OBJECT_KEY]) !==
+                        Boolean(data[STRUCTURED_OUTPUT_ARRAY_KEY]));
+                }, {
+                    message: 'One and only one of __structured__output__object and __structured__output__array should be present.',
+                    path: [STRUCTURED_OUTPUT_KEY],
+                }),
+            });
+        }
+        else if (nodeVersion < 1.3) {
+            returnSchema = zod_1.z.object({
+                output: zodSchema.optional().nullable(),
+            });
+        }
+        else {
+            returnSchema = zod_1.z.object({
+                output: zodSchema,
+            });
+        }
+        return new N8nStructuredOutputParser(context, returnSchema);
+    }
+    getSchema() {
+        return this.schema;
+    }
+}
+exports.N8nStructuredOutputParser = N8nStructuredOutputParser;
+//# sourceMappingURL=N8nStructuredOutputParser.js.map

package/dist/nodes/agents/DatabricksAiAgent/src/utils/output_parsers/N8nStructuredOutputParser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"N8nStructuredOutputParser.js","sourceRoot":"","sources":["../../../../../../../nodes/agents/DatabricksAiAgent/src/utils/output_parsers/N8nStructuredOutputParser.ts"],"names":[],"mappings":";;;;;;AACA,sEAA2E;AAC3E,qDAA6B;AAE7B,+CAAuE;AACvE,6BAAwB;AAExB,wCAA4D;AAE5D,MAAM,qBAAqB,GAAG,sBAAsB,CAAC;AACrD,MAAM,4BAA4B,GAAG,8BAA8B,CAAC;AACpE,MAAM,2BAA2B,GAAG,6BAA6B,CAAC;AAKlE,MAAa,yBAA0B,SAAQ,uCAA2B;IACzE,YACS,OAA6B,EACrC,SAA8B;QAE9B,KAAK,CAAC,SAAS,CAAC,CAAC;QAHT,YAAO,GAAP,OAAO,CAAsB;QAMtC,iBAAY,GAAG,CAAC,WAAW,EAAE,gBAAgB,EAAE,YAAY,CAAC,CAAC;IAF7D,CAAC;IAID,KAAK,CAAC,KAAK,CACV,IAAY,EACZ,UAAsB,EACtB,WAAqC;;QAErC,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,kCAAmB,CAAC,cAAc,EAAE;YAC/E,CAAC,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC;SACrC,CAAC,CAAC;QAEH,IAAI,CAAC;YACJ,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC/E,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;YAC3C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAElD,IAAI,MAAM,GAAG,CAAC,MAAA,MAAA,MAAA,IAAA,aAAG,EAAC,MAAM,EAAE,CAAC,qBAAqB,EAAE,4BAA4B,CAAC,CAAC,mCAC/E,IAAA,aAAG,EAAC,MAAM,EAAE,CAAC,qBAAqB,EAAE,2BAA2B,CAAC,CAAC,mCACjE,IAAA,aAAG,EAAC,MAAM,EAAE,qBAAqB,CAAC,mCAClC,MAAM,CAA4B,CAAC;YAGpC,MAAM,GAAG,IAAA,4BAAkB,EAAC,MAAM,CAAC,CAAC;YAEpC,IAAA,oBAAU,EAAC,IAAI,CAAC,OAAO,EAAE,kBAAkB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YAEzE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,kCAAmB,CAAC,cAAc,EAAE,KAAK,EAAE;gBACrE,CAAC,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,CAAC;aACjD,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QACf,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACZ,MAAM,SAAS,GAAG,IAAI,iCAAkB,CACvC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EACtB,0CAA0C,EAC1C;gBACC,WAAW,EACV,0GAA0G;aAC3G,CACD,CAAC;YAGF,IAAI,CAAC,YAAY,WAAW,EAAE,CAAC;gBAC9B,SAAS,CAAC,OAAO,CAAC,sBAAsB,GAAG,8BAA8B,CAAC;YAC3E,CAAC;iBAAM,IACN,CAAC,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,IAAI,CAAC;gBAClD,CAAC,CAAC,YAAY,OAAC,CAAC,QAAQ;qBACvB,MAAA,CAAC,CAAC,MAAM,0CAAG,CAAC,CAAC,CAAA;oBACb,CAAA,MAAA,CAAC,CAAC,MAAM,0CAAG,CAAC,EAAE,IAAI,MAAK,cAAc;oBACrC,CAAA,MAAA,MAAA,CAAC,CAAC,MAAM,0CAAG,CAAC,EAAE,IAAI,0CAAG,CAAC,CAAC,MAAK,QAAQ;oBACpC,CAAA,MAAA,CAAC,CAAC,MAAM,0CAAG,CAAC,EAAE,QAAQ,MAAK,QAAQ;oBACnC,CAAA,MAAA,CAAC,CAAC,MAAM,0CAAG,CAAC,EAAE,QAAQ,MAAK,WAAW,CAAC,EACvC,CAAC;gBACF,SAAS,CAAC,OAAO,CAAC,sBAAsB,GAAG,yCAAyC,CAAC;YACtF,CAAC;iBAAM,IAAI,CAAC,YAAY,OAAC,CAAC,QAAQ,EAAE,CAAC;gBACpC,SAAS,CAAC,OAAO,CAAC,sBAAsB;oBACvC,iDAAiD,CAAC;YACpD,CAAC;YAED,MAAM,YAAY,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAChE,IAAA,oBAAU,EAAC,IAAI,CAAC,OAAO,EAAE,kBAAkB,EAAE;gBAC5C,IAAI;gBACJ,QAAQ,EAAE,YAAY;aACtB,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,kCAAmB,CAAC,cAAc,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;YACjF,IAAI,WAAW,IAAI,CAAC,YAAY,KAAK,EAAE,CAAC;gBACvC,MAAM,WAAW,CAAC,CAAC,CAAC,CAAC;YACtB,CAAC;YAED,MAAM,SAAS,CAAC;QACjB,CAAC;IACF,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAC7B,SAA8B,EAC9B,WAAmB,EACnB,OAA6B;QAE7B,IAAI,YAAqD,CAAC;QAC1D,IAAI,WAAW,KAAK,CAAC,EAAE,CAAC;YACvB,YAAY,GAAG,OAAC,CAAC,MAAM,CAAC;gBACvB,CAAC,qBAAqB,CAAC,EAAE,OAAC;qBACxB,MAAM,CAAC;oBACP,CAAC,4BAA4B,CAAC,EAAE,SAAS,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;oBAC/D,CAAC,2BAA2B,CAAC,EAAE,OAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;iBACvE,CAAC;qBACD,QAAQ,CACR,uDAAuD,4BAA4B,OAAO,2BAA2B,kBAAkB,CACvI;qBACA,MAAM,CACN,CAAC,IAAI,EAAE,EAAE;oBAER,OAAO,CACN,OAAO,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;wBAC3C,OAAO,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC,CAC1C,CAAC;gBACH,CAAC,EACD;oBACC,OAAO,EACN,qGAAqG;oBACtG,IAAI,EAAE,CAAC,qBAAqB,CAAC;iBAC7B,CACD;aACF,CAAC,CAAC;QACJ,CAAC;aAAM,IAAI,WAAW,GAAG,GAAG,EAAE,CAAC;YAC9B,YAAY,GAAG,OAAC,CAAC,MAAM,CAAC;gBACvB,MAAM,EAAE,SAAS,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;aACvC,CAAC,CAAC;QACJ,CAAC;aAAM,CAAC;YACP,YAAY,GAAG,OAAC,CAAC,MAAM,CAAC;gBACvB,MAAM,EAAE,SAAS;aACjB,CAAC,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,yBAAyB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;IAC7D,CAAC;IAED,SAAS;QACR,OAAO,IAAI,CAAC,MAAM,CAAC;IACpB,CAAC;CACD;AAjID,8DAiIC"}

package/dist/nodes/agents/DatabricksAiAgent/src/utils/prompt.d.ts

@@ -0,0 +1 @@
+export declare const SYSTEM_MESSAGE = "You are a helpful assistant";

package/dist/nodes/agents/DatabricksAiAgent/src/utils/prompt.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SYSTEM_MESSAGE = void 0;
+exports.SYSTEM_MESSAGE = 'You are a helpful assistant';
+//# sourceMappingURL=prompt.js.map

package/dist/nodes/agents/DatabricksAiAgent/src/utils/prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../../../../../nodes/agents/DatabricksAiAgent/src/utils/prompt.ts"],"names":[],"mappings":";;;AAAa,QAAA,cAAc,GAAG,6BAA6B,CAAC"}

package/dist/nodes/agents/DatabricksAiAgent/src/utils/retry.d.ts

@@ -0,0 +1,16 @@
+import type { IExecuteFunctions, ISupplyDataFunctions } from 'n8n-workflow';
+export interface RetryOptions {
+    maxRetries?: number;
+    baseDelay?: number;
+    maxDelay?: number;
+    useJitter?: boolean;
+}
+export declare function httpRequestWithRetry<T = any>(ctx: IExecuteFunctions | ISupplyDataFunctions, requestOptions: any, retryOptions?: RetryOptions): Promise<T>;
+export declare function maskCredentials(credentials: {
+    host?: string;
+    token?: string;
+}): {
+    host: string;
+    token: string;
+};
+export declare function maskTokensInText(text: string): string;

package/dist/nodes/agents/DatabricksAiAgent/src/utils/retry.js

@@ -0,0 +1,132 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.httpRequestWithRetry = httpRequestWithRetry;
+exports.maskCredentials = maskCredentials;
+exports.maskTokensInText = maskTokensInText;
+const n8n_workflow_1 = require("n8n-workflow");
+const constants_1 = require("../constants");
+function analyzeError(error) {
+    const errorMsg = error instanceof Error ? error.message.toLowerCase() : String(error).toLowerCase();
+    let statusCode;
+    if (typeof error === 'object' && error !== null) {
+        if ('statusCode' in error) {
+            statusCode = error.statusCode;
+        }
+        else if ('status' in error) {
+            statusCode = error.status;
+        }
+    }
+    if (statusCode && constants_1.NON_RETRYABLE_STATUS_CODES.includes(statusCode)) {
+        return {
+            isRetryable: false,
+            statusCode,
+            message: errorMsg,
+        };
+    }
+    if (statusCode && constants_1.RETRYABLE_STATUS_CODES.includes(statusCode)) {
+        return {
+            isRetryable: true,
+            statusCode,
+            message: errorMsg,
+        };
+    }
+    const nonRetryablePatterns = [
+        'unauthorized',
+        'forbidden',
+        'bad request',
+        'not found',
+        'invalid',
+        'authentication',
+        'permission',
+    ];
+    if (nonRetryablePatterns.some(pattern => errorMsg.includes(pattern))) {
+        return {
+            isRetryable: false,
+            statusCode,
+            message: errorMsg,
+        };
+    }
+    const retryablePatterns = [
+        'timeout',
+        'etimedout',
+        'econnreset',
+        'econnrefused',
+        'network',
+        'socket hang up',
+        'rate limit',
+        'too many requests',
+    ];
+    if (retryablePatterns.some(pattern => errorMsg.includes(pattern))) {
+        return {
+            isRetryable: true,
+            statusCode,
+            message: errorMsg,
+        };
+    }
+    return {
+        isRetryable: true,
+        statusCode,
+        message: errorMsg,
+    };
+}
+function calculateDelay(attempt, options) {
+    const exponentialDelay = options.baseDelay * Math.pow(2, attempt);
+    const cappedDelay = Math.min(exponentialDelay, options.maxDelay);
+    if (options.useJitter) {
+        return Math.floor(cappedDelay * (0.5 + Math.random() * 0.5));
+    }
+    return cappedDelay;
+}
+async function httpRequestWithRetry(ctx, requestOptions, retryOptions = {}) {
+    var _a, _b, _c, _d;
+    const options = {
+        maxRetries: (_a = retryOptions.maxRetries) !== null && _a !== void 0 ? _a : constants_1.MLFLOW_CONSTANTS.DEFAULT_MAX_RETRIES,
+        baseDelay: (_b = retryOptions.baseDelay) !== null && _b !== void 0 ? _b : constants_1.MLFLOW_CONSTANTS.DEFAULT_RETRY_BASE_DELAY_MS,
+        maxDelay: (_c = retryOptions.maxDelay) !== null && _c !== void 0 ? _c : constants_1.MLFLOW_CONSTANTS.MAX_RETRY_DELAY_MS,
+        useJitter: (_d = retryOptions.useJitter) !== null && _d !== void 0 ? _d : true,
+    };
+    let lastError;
+    const url = requestOptions.url || requestOptions.baseURL;
+    for (let attempt = 0; attempt < options.maxRetries; attempt++) {
+        try {
+            const response = await ctx.helpers.httpRequest(requestOptions);
+            return response;
+        }
+        catch (error) {
+            lastError = error instanceof Error ? error : new Error(String(error));
+            const errorInfo = analyzeError(error);
+            if (!errorInfo.isRetryable) {
+                ctx.logger.debug(`Non-retryable error for ${url}: ${errorInfo.message}` +
+                    (errorInfo.statusCode ? ` (status: ${errorInfo.statusCode})` : ''));
+                throw lastError;
+            }
+            if (attempt === options.maxRetries - 1) {
+                break;
+            }
+            const delay = calculateDelay(attempt, options);
+            ctx.logger.debug(`HTTP request to ${url} failed (attempt ${attempt + 1}/${options.maxRetries}): ${errorInfo.message}` +
+                (errorInfo.statusCode ? ` (status: ${errorInfo.statusCode})` : '') +
+                `. Retrying in ${delay}ms...`);
+            await new Promise(resolve => setTimeout(resolve, delay));
+        }
+    }
+    const errorMessage = constants_1.ERROR_MESSAGES.MAX_RETRIES_EXCEEDED(options.maxRetries, (lastError === null || lastError === void 0 ? void 0 : lastError.message) || 'Unknown error');
+    throw new n8n_workflow_1.NodeOperationError(ctx.getNode(), errorMessage);
+}
+function maskCredentials(credentials) {
+    return {
+        host: credentials.host
+            ? credentials.host.replace(/^(https?:\/\/[^.]+)(\.[^.]+\..*)$/, '$1.***')
+            : '***',
+        token: credentials.token && credentials.token.length >= 4
+            ? `${credentials.token.substring(0, 4)}****`
+            : '***',
+    };
+}
+function maskTokensInText(text) {
+    let masked = text.replace(constants_1.TOKEN_PATTERNS.DATABRICKS_TOKEN, `dapi${constants_1.TOKEN_PATTERNS.MASK}`);
+    masked = masked.replace(constants_1.TOKEN_PATTERNS.GENERIC_LONG_TOKEN, constants_1.TOKEN_PATTERNS.MASK);
+    masked = masked.replace(constants_1.TOKEN_PATTERNS.BEARER_TOKEN, `Bearer ${constants_1.TOKEN_PATTERNS.MASK}`);
+    return masked;
+}
+//# sourceMappingURL=retry.js.map

package/dist/nodes/agents/DatabricksAiAgent/src/utils/retry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry.js","sourceRoot":"","sources":["../../../../../../nodes/agents/DatabricksAiAgent/src/utils/retry.ts"],"names":[],"mappings":";;AA4KA,oDA6DC;AAoBD,0CAmBC;AAcD,4CAWC;AAlSD,+CAAkD;AAClD,4CAMsB;AA+BtB,SAAS,YAAY,CAAC,KAAc;IAChC,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAGpG,IAAI,UAA8B,CAAC;IACnC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC9C,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;YACxB,UAAU,GAAG,KAAK,CAAC,UAAoB,CAAC;QAC5C,CAAC;aAAM,IAAI,QAAQ,IAAI,KAAK,EAAE,CAAC;YAC3B,UAAU,GAAG,KAAK,CAAC,MAAgB,CAAC;QACxC,CAAC;IACL,CAAC;IAGD,IAAI,UAAU,IAAI,sCAA0B,CAAC,QAAQ,CAAC,UAAiB,CAAC,EAAE,CAAC;QACvE,OAAO;YACH,WAAW,EAAE,KAAK;YAClB,UAAU;YACV,OAAO,EAAE,QAAQ;SACpB,CAAC;IACN,CAAC;IAGD,IAAI,UAAU,IAAI,kCAAsB,CAAC,QAAQ,CAAC,UAAiB,CAAC,EAAE,CAAC;QACnE,OAAO;YACH,WAAW,EAAE,IAAI;YACjB,UAAU;YACV,OAAO,EAAE,QAAQ;SACpB,CAAC;IACN,CAAC;IAGD,MAAM,oBAAoB,GAAG;QACzB,cAAc;QACd,WAAW;QACX,aAAa;QACb,WAAW;QACX,SAAS;QACT,gBAAgB;QAChB,YAAY;KACf,CAAC;IAEF,IAAI,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;QACnE,OAAO;YACH,WAAW,EAAE,KAAK;YAClB,UAAU;YACV,OAAO,EAAE,QAAQ;SACpB,CAAC;IACN,CAAC;IAGD,MAAM,iBAAiB,GAAG;QACtB,SAAS;QACT,WAAW;QACX,YAAY;QACZ,cAAc;QACd,SAAS;QACT,gBAAgB;QAChB,YAAY;QACZ,mBAAmB;KACtB,CAAC;IAEF,IAAI,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;QAChE,OAAO;YACH,WAAW,EAAE,IAAI;YACjB,UAAU;YACV,OAAO,EAAE,QAAQ;SACpB,CAAC;IACN,CAAC;IAGD,OAAO;QACH,WAAW,EAAE,IAAI;QACjB,UAAU;QACV,OAAO,EAAE,QAAQ;KACpB,CAAC;AACN,CAAC;AASD,SAAS,cAAc,CAAC,OAAe,EAAE,OAA+B;IAEpE,MAAM,gBAAgB,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IAGlE,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,gBAAgB,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IAGjE,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QACpB,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,OAAO,WAAW,CAAC;AACvB,CAAC;AA6BM,KAAK,UAAU,oBAAoB,CACtC,GAA6C,EAC7C,cAAmB,EACnB,eAA6B,EAAE;;IAG/B,MAAM,OAAO,GAA2B;QACpC,UAAU,EAAE,MAAA,YAAY,CAAC,UAAU,mCAAI,4BAAgB,CAAC,mBAAmB;QAC3E,SAAS,EAAE,MAAA,YAAY,CAAC,SAAS,mCAAI,4BAAgB,CAAC,2BAA2B;QACjF,QAAQ,EAAE,MAAA,YAAY,CAAC,QAAQ,mCAAI,4BAAgB,CAAC,kBAAkB;QACtE,SAAS,EAAE,MAAA,YAAY,CAAC,SAAS,mCAAI,IAAI;KAC5C,CAAC;IAEF,IAAI,SAA4B,CAAC;IACjC,MAAM,GAAG,GAAG,cAAc,CAAC,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC;IAEzD,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;QAC5D,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;YAC/D,OAAO,QAAa,CAAC;QACzB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,SAAS,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAGtE,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;YAGtC,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC;gBACzB,GAAG,CAAC,MAAM,CAAC,KAAK,CACZ,2BAA2B,GAAG,KAAK,SAAS,CAAC,OAAO,EAAE;oBACtD,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,aAAa,SAAS,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CACrE,CAAC;gBACF,MAAM,SAAS,CAAC;YACpB,CAAC;YAGD,IAAI,OAAO,KAAK,OAAO,CAAC,UAAU,GAAG,CAAC,EAAE,CAAC;gBACrC,MAAM;YACV,CAAC;YAGD,MAAM,KAAK,GAAG,cAAc,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAE/C,GAAG,CAAC,MAAM,CAAC,KAAK,CACZ,mBAAmB,GAAG,oBAAoB,OAAO,GAAG,CAAC,IAAI,OAAO,CAAC,UAAU,MAAM,SAAS,CAAC,OAAO,EAAE;gBACpG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,aAAa,SAAS,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClE,iBAAiB,KAAK,OAAO,CAChC,CAAC;YAGF,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;QAC7D,CAAC;IACL,CAAC;IAGD,MAAM,YAAY,GAAG,0BAAc,CAAC,oBAAoB,CACpD,OAAO,CAAC,UAAU,EAClB,CAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,OAAO,KAAI,eAAe,CACxC,CAAC;IAEF,MAAM,IAAI,iCAAkB,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,YAAY,CAAC,CAAC;AAC9D,CAAC;AAoBD,SAAgB,eAAe,CAAC,WAG/B;IAIG,OAAO;QACH,IAAI,EAAE,WAAW,CAAC,IAAI;YAClB,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CACtB,mCAAmC,EACnC,QAAQ,CACX;YACD,CAAC,CAAC,KAAK;QAEX,KAAK,EAAE,WAAW,CAAC,KAAK,IAAI,WAAW,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC;YACrD,CAAC,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM;YAC5C,CAAC,CAAC,KAAK;KACd,CAAC;AACN,CAAC;AAcD,SAAgB,gBAAgB,CAAC,IAAY;IAEzC,IAAI,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,0BAAc,CAAC,gBAAgB,EAAE,OAAO,0BAAc,CAAC,IAAI,EAAE,CAAC,CAAC;IAGzF,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,0BAAc,CAAC,kBAAkB,EAAE,0BAAc,CAAC,IAAI,CAAC,CAAC;IAGhF,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,0BAAc,CAAC,YAAY,EAAE,UAAU,0BAAc,CAAC,IAAI,EAAE,CAAC,CAAC;IAEtF,OAAO,MAAM,CAAC;AAClB,CAAC"}

package/dist/nodes/agents/DatabricksAiAgent/src/utils/security.d.ts

@@ -0,0 +1,18 @@
+export interface ValidationResult {
+    isValid: boolean;
+    error?: string;
+    sanitized?: string;
+}
+export declare function validateExperimentName(experimentName: string): ValidationResult;
+export declare function validateDatabricksHost(host: string): ValidationResult;
+export declare function sanitizeExperimentId(experimentId: string): string | null;
+export declare function validateApiResponse(response: unknown, expectedFields: string[]): ValidationResult;
+export declare class RateLimiter {
+    private requests;
+    private readonly maxRequests;
+    private readonly windowMs;
+    constructor(maxRequests?: number, windowMs?: number);
+    isAllowed(): boolean;
+    getRemaining(): number;
+    reset(): void;
+}