@n8n/n8n-nodes-langchain 2.43.2 → 2.44.0

package/dist/nodes/vendors/Databricks/mlflow/span-collector.js

@@ -0,0 +1,378 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MlflowSpanCollector = void 0;
+const base_1 = require("@langchain/core/callbacks/base");
+const messages_1 = require("@langchain/core/messages");
+const is_record_1 = require("@n8n/utils/is-record");
+const sanitize_error_detail_1 = require("@n8n/utils/redaction/sanitize-error-detail");
+const scrub_secrets_1 = require("@n8n/utils/scrub-secrets");
+const node_crypto_1 = require("node:crypto");
+const types_1 = require("./types");
+const MAX_ATTRIBUTE_CHARS = 100_000;
+const MAX_STRING_CHARS = 4_000;
+const MAX_ERROR_CHARS = 2_000;
+const TRACE_ID_BYTES = 16;
+const SPAN_ID_BYTES = 8;
+const UNINTERESTING_INPUT_KEYS = [
+    'formatting_instructions',
+    'agent_scratchpad',
+    'chat_history',
+    'steps',
+];
+const ECHOED_OUTPUT_KEYS = [...UNINTERESTING_INPUT_KEYS, 'input', 'system_message'];
+function randomTraceId() {
+    const raw = (0, node_crypto_1.randomBytes)(TRACE_ID_BYTES);
+    return { traceId: `tr-${raw.toString('hex')}`, spanTraceId: raw.toString('base64') };
+}
+function randomSpanId() {
+    return (0, node_crypto_1.randomBytes)(SPAN_ID_BYTES).toString('base64');
+}
+function toNanos(ms) {
+    return `${ms}000000`;
+}
+function serializedName(serialized, fallback) {
+    const id = serialized?.id;
+    return (Array.isArray(id) ? id.at(-1) : undefined) ?? fallback;
+}
+function stringifySafely(value) {
+    const ancestors = [];
+    const json = JSON.stringify(value, function (_key, raw) {
+        if (raw === null || typeof raw !== 'object')
+            return raw;
+        while (ancestors.length > 0 && ancestors.at(-1) !== this)
+            ancestors.pop();
+        if (ancestors.includes(raw))
+            return '[circular]';
+        ancestors.push(raw);
+        return raw;
+    });
+    return json ?? 'null';
+}
+function capStrings(value, seen) {
+    if (typeof value === 'string') {
+        return value.length > MAX_STRING_CHARS
+            ? `${value.slice(0, MAX_STRING_CHARS)}…[truncated ${value.length} chars]`
+            : value;
+    }
+    if (value === null || typeof value !== 'object')
+        return value;
+    if (seen.has(value))
+        return '[circular]';
+    seen.add(value);
+    try {
+        if (Array.isArray(value))
+            return value.map((entry) => capStrings(entry, seen));
+        return Object.fromEntries(Object.entries(value).map(([key, entry]) => [key, capStrings(entry, seen)]));
+    }
+    finally {
+        seen.delete(value);
+    }
+}
+function toAttributeJson(value) {
+    const full = (0, scrub_secrets_1.scrubSecretsInText)(stringifySafely(value));
+    if (full.length <= MAX_ATTRIBUTE_CHARS)
+        return full;
+    const capped = (0, scrub_secrets_1.scrubSecretsInText)(stringifySafely(capStrings(value, new Set())));
+    if (capped.length <= MAX_ATTRIBUTE_CHARS)
+        return capped;
+    return JSON.stringify({ truncated: true, original_size_chars: full.length });
+}
+function errorMessage(error) {
+    const raw = error instanceof Error ? error.message : String(error);
+    return (0, sanitize_error_detail_1.sanitizeErrorDetail)(raw, MAX_ERROR_CHARS);
+}
+function firstNumber(...candidates) {
+    return candidates.find((c) => typeof c === 'number' && Number.isFinite(c));
+}
+function readUsage(raw) {
+    if (!(0, is_record_1.isRecord)(raw))
+        return undefined;
+    const input = firstNumber(raw.promptTokens, raw.input_tokens, raw.prompt_tokens);
+    const output = firstNumber(raw.completionTokens, raw.output_tokens, raw.completion_tokens);
+    if (input === undefined && output === undefined)
+        return undefined;
+    const inputTokens = input ?? 0;
+    const outputTokens = output ?? 0;
+    return {
+        inputTokens,
+        outputTokens,
+        totalTokens: firstNumber(raw.totalTokens, raw.total_tokens) ?? inputTokens + outputTokens,
+    };
+}
+function extractTokenUsage(output) {
+    const fromLlmOutput = readUsage(output.llmOutput?.tokenUsage) ?? readUsage(output.llmOutput?.usage);
+    if (fromLlmOutput)
+        return fromLlmOutput;
+    for (const generation of output.generations.flat()) {
+        const usage = readUsage(generation.generationInfo?.usage);
+        if (usage)
+            return usage;
+    }
+    return undefined;
+}
+function toOpenAiRole(type) {
+    switch (type) {
+        case 'human':
+            return 'user';
+        case 'ai':
+            return 'assistant';
+        default:
+            return type;
+    }
+}
+function toChatMessage(message) {
+    const converted = {
+        role: toOpenAiRole(message.type),
+        content: message.content,
+    };
+    if ('tool_calls' in message && Array.isArray(message.tool_calls)) {
+        converted.tool_calls = message.tool_calls.map((call) => ({
+            id: call.id,
+            type: 'function',
+            function: { name: call.name, arguments: JSON.stringify(call.args ?? {}) },
+        }));
+    }
+    if ('tool_call_id' in message && typeof message.tool_call_id === 'string') {
+        converted.tool_call_id = message.tool_call_id;
+    }
+    return converted;
+}
+function toChatCompletion(output) {
+    return {
+        choices: output.generations.flat().map((generation) => {
+            const message = (0, is_record_1.isRecord)(generation) ? generation.message : undefined;
+            return {
+                message: messages_1.BaseMessage.isInstance(message)
+                    ? toChatMessage(message)
+                    : { role: 'assistant', content: generation.text },
+                finish_reason: generation.generationInfo?.finish_reason,
+            };
+        }),
+    };
+}
+function withoutKeys(values, drop) {
+    return Object.fromEntries(Object.entries(values).filter(([key]) => !drop.includes(key)));
+}
+class MlflowSpanCollector extends base_1.BaseCallbackHandler {
+    constructor() {
+        super();
+        this.name = 'MlflowSpanCollector';
+        this.awaitHandlers = true;
+        this.spans = new Map();
+        this.parentOf = new Map();
+        this.runOrder = [];
+        this.openRuns = new Set();
+        this.lastActivityMs = 0;
+        const ids = randomTraceId();
+        this.traceId = ids.traceId;
+        this.spanTraceId = ids.spanTraceId;
+    }
+    nearestKeptAncestor(runId) {
+        let cursor = runId;
+        const guard = new Set();
+        while (cursor !== undefined && !guard.has(cursor)) {
+            guard.add(cursor);
+            const span = this.spans.get(cursor);
+            if (span)
+                return span;
+            cursor = this.parentOf.get(cursor);
+        }
+        return undefined;
+    }
+    captureRunMetadata(metadata) {
+        if (!(0, is_record_1.isRecord)(metadata))
+            return;
+        if (typeof metadata.execution_id === 'string')
+            this.executionId ??= metadata.execution_id;
+        if (typeof metadata.node === 'string')
+            this.nodeName ??= metadata.node;
+        if ((0, is_record_1.isRecord)(metadata.workflow) && typeof metadata.workflow.id === 'string') {
+            this.workflowId ??= metadata.workflow.id;
+        }
+    }
+    startSpan(params) {
+        const { runId, parentRunId, name, spanType, inputs } = params;
+        if (this.spans.has(runId))
+            return;
+        const startTimeMs = Date.now();
+        this.startedAtMs ??= startTimeMs;
+        this.lastActivityMs = startTimeMs;
+        const span = {
+            trace_id: this.spanTraceId,
+            span_id: randomSpanId(),
+            parent_span_id: this.nearestKeptAncestor(parentRunId)?.span_id,
+            name,
+            start_time_unix_nano: toNanos(startTimeMs),
+            end_time_unix_nano: toNanos(startTimeMs),
+            status: { code: 'OK' },
+            attributes: {
+                [types_1.MLFLOW_ATTRIBUTE.SpanType]: toAttributeJson(spanType),
+                [types_1.MLFLOW_ATTRIBUTE.TraceRequestId]: toAttributeJson(this.traceId),
+                [types_1.MLFLOW_ATTRIBUTE.SpanInputs]: toAttributeJson(inputs),
+                ...(params.messageFormat && {
+                    [types_1.MLFLOW_ATTRIBUTE.MessageFormat]: types_1.MLFLOW_MESSAGE_FORMAT,
+                }),
+            },
+        };
+        this.spans.set(runId, span);
+        this.runOrder.push(runId);
+        this.openRuns.add(runId);
+    }
+    closeSpan(runId) {
+        const span = this.spans.get(runId);
+        if (!span)
+            return undefined;
+        const now = Date.now();
+        this.lastActivityMs = now;
+        span.end_time_unix_nano = toNanos(now);
+        this.openRuns.delete(runId);
+        return span;
+    }
+    endSpan(runId, outputs, tokenUsage) {
+        const span = this.closeSpan(runId);
+        if (!span)
+            return;
+        span.attributes[types_1.MLFLOW_ATTRIBUTE.SpanOutputs] = toAttributeJson(outputs);
+        if (tokenUsage) {
+            this.addTokenUsage(tokenUsage);
+            span.attributes[types_1.MLFLOW_ATTRIBUTE.TokenUsage] = toAttributeJson({
+                input_tokens: tokenUsage.inputTokens,
+                output_tokens: tokenUsage.outputTokens,
+                total_tokens: tokenUsage.totalTokens,
+            });
+        }
+    }
+    addTokenUsage(usage) {
+        this.tokenUsage = {
+            inputTokens: (this.tokenUsage?.inputTokens ?? 0) + usage.inputTokens,
+            outputTokens: (this.tokenUsage?.outputTokens ?? 0) + usage.outputTokens,
+            totalTokens: (this.tokenUsage?.totalTokens ?? 0) + usage.totalTokens,
+        };
+    }
+    failSpan(runId, error) {
+        const span = this.closeSpan(runId);
+        if (!span)
+            return;
+        span.status = { code: 'ERROR', message: errorMessage(error) };
+    }
+    handleChainStart(_chain, inputs, runId, parentRunId, _tags, metadata, _runType, runName) {
+        this.parentOf.set(runId, parentRunId);
+        this.captureRunMetadata(metadata);
+        if (this.rootRunId !== undefined)
+            return;
+        this.rootRunId = runId;
+        this.startSpan({
+            runId,
+            parentRunId: undefined,
+            name: runName ?? 'Agent',
+            spanType: types_1.MLFLOW_SPAN_TYPE.Agent,
+            inputs: withoutKeys(inputs, UNINTERESTING_INPUT_KEYS),
+        });
+        if (typeof inputs.input === 'string')
+            this.requestPreview = inputs.input;
+    }
+    handleChainEnd(outputs, runId) {
+        if (runId !== this.rootRunId)
+            return;
+        if (typeof outputs.output === 'string')
+            this.responsePreview = outputs.output;
+        this.endSpan(runId, withoutKeys(outputs, ECHOED_OUTPUT_KEYS));
+    }
+    handleChainError(error, runId) {
+        this.failSpan(runId, error);
+    }
+    handleChatModelStart(llm, messages, runId, parentRunId, _extraParams, _tags, metadata, runName) {
+        this.parentOf.set(runId, parentRunId);
+        this.captureRunMetadata(metadata);
+        this.startSpan({
+            runId,
+            parentRunId,
+            name: runName ?? serializedName(llm, 'chat model'),
+            spanType: types_1.MLFLOW_SPAN_TYPE.ChatModel,
+            inputs: { messages: messages.flat().map(toChatMessage) },
+            messageFormat: true,
+        });
+    }
+    handleLLMStart(llm, prompts, runId, parentRunId, _extraParams, _tags, metadata, runName) {
+        this.parentOf.set(runId, parentRunId);
+        this.captureRunMetadata(metadata);
+        this.startSpan({
+            runId,
+            parentRunId,
+            name: runName ?? serializedName(llm, 'model'),
+            spanType: types_1.MLFLOW_SPAN_TYPE.Llm,
+            inputs: prompts,
+        });
+    }
+    handleLLMEnd(output, runId) {
+        this.endSpan(runId, toChatCompletion(output), extractTokenUsage(output));
+    }
+    handleLLMError(error, runId) {
+        this.failSpan(runId, error);
+    }
+    handleToolStart(tool, input, runId, parentRunId, _tags, metadata, runName) {
+        this.parentOf.set(runId, parentRunId);
+        this.captureRunMetadata(metadata);
+        this.startSpan({
+            runId,
+            parentRunId,
+            name: runName ?? serializedName(tool, 'tool'),
+            spanType: types_1.MLFLOW_SPAN_TYPE.Tool,
+            inputs: input,
+        });
+    }
+    handleToolEnd(output, runId) {
+        this.endSpan(runId, output);
+    }
+    handleToolError(error, runId) {
+        this.failSpan(runId, error);
+    }
+    handleRetrieverStart(retriever, query, runId, parentRunId, _tags, metadata, name) {
+        this.parentOf.set(runId, parentRunId);
+        this.captureRunMetadata(metadata);
+        this.startSpan({
+            runId,
+            parentRunId,
+            name: name ?? serializedName(retriever, 'retriever'),
+            spanType: types_1.MLFLOW_SPAN_TYPE.Retriever,
+            inputs: query,
+        });
+    }
+    handleRetrieverEnd(documents, runId) {
+        this.endSpan(runId, documents);
+    }
+    handleRetrieverError(error, runId) {
+        this.failSpan(runId, error);
+    }
+    finish() {
+        const root = this.rootRunId === undefined ? undefined : this.spans.get(this.rootRunId);
+        if (!root || this.startedAtMs === undefined)
+            return undefined;
+        const spans = this.runOrder
+            .map((runId) => this.spans.get(runId))
+            .filter((span) => span !== undefined);
+        const endTimeMs = Math.max(this.lastActivityMs, this.startedAtMs);
+        const endNanos = toNanos(endTimeMs);
+        for (const runId of this.openRuns) {
+            const span = this.spans.get(runId);
+            if (span)
+                span.end_time_unix_nano = endNanos;
+        }
+        return {
+            traceId: this.traceId,
+            rootSpanId: root.span_id,
+            spans,
+            startTimeMs: this.startedAtMs,
+            endTimeMs,
+            state: root.status.code,
+            tokenUsage: this.tokenUsage,
+            executionId: this.executionId,
+            workflowId: this.workflowId,
+            nodeName: this.nodeName,
+            requestPreview: this.requestPreview,
+            responsePreview: this.responsePreview,
+        };
+    }
+}
+exports.MlflowSpanCollector = MlflowSpanCollector;
+//# sourceMappingURL=span-collector.js.map

package/dist/nodes/vendors/Databricks/mlflow/span-collector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"span-collector.js","sourceRoot":"","sources":["../../../../../nodes/vendors/Databricks/mlflow/span-collector.ts"],"names":[],"mappings":";;;AAAA,yDAAqE;AAGrE,uDAAuD;AAGvD,oDAAgD;AAChD,sFAAiF;AACjF,4DAA8D;AAC9D,6CAA0C;AAG1C,mCAAoF;AAGpF,MAAM,mBAAmB,GAAG,OAAO,CAAC;AACpC,MAAM,gBAAgB,GAAG,KAAK,CAAC;AAC/B,MAAM,eAAe,GAAG,KAAK,CAAC;AAE9B,MAAM,cAAc,GAAG,EAAE,CAAC;AAC1B,MAAM,aAAa,GAAG,CAAC,CAAC;AAGxB,MAAM,wBAAwB,GAAG;IAChC,yBAAyB;IACzB,kBAAkB;IAClB,cAAc;IACd,OAAO;CACP,CAAC;AAOF,MAAM,kBAAkB,GAAG,CAAC,GAAG,wBAAwB,EAAE,OAAO,EAAE,gBAAgB,CAAC,CAAC;AAUpF,SAAS,aAAa;IACrB,MAAM,GAAG,GAAG,IAAA,yBAAW,EAAC,cAAc,CAAC,CAAC;IACxC,OAAO,EAAE,OAAO,EAAE,MAAM,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;AACtF,CAAC;AAED,SAAS,YAAY;IACpB,OAAO,IAAA,yBAAW,EAAC,aAAa,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AACtD,CAAC;AAED,SAAS,OAAO,CAAC,EAAU;IAC1B,OAAO,GAAG,EAAE,QAAQ,CAAC;AACtB,CAAC;AAGD,SAAS,cAAc,CAAC,UAAkC,EAAE,QAAgB;IAC3E,MAAM,EAAE,GAAG,UAAU,EAAE,EAAE,CAAC;IAC1B,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,QAAQ,CAAC;AAChE,CAAC;AAOD,SAAS,eAAe,CAAC,KAAc;IACtC,MAAM,SAAS,GAAc,EAAE,CAAC;IAChC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,UAAyB,IAAI,EAAE,GAAY;QAC7E,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ;YAAE,OAAO,GAAG,CAAC;QAExD,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI;YAAE,SAAS,CAAC,GAAG,EAAE,CAAC;QAC1E,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,OAAO,YAAY,CAAC;QACjD,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpB,OAAO,GAAG,CAAC;IACZ,CAAC,CAAC,CAAC;IACH,OAAO,IAAI,IAAI,MAAM,CAAC;AACvB,CAAC;AAGD,SAAS,UAAU,CAAC,KAAc,EAAE,IAAiB;IACpD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,KAAK,CAAC,MAAM,GAAG,gBAAgB;YACrC,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,eAAe,KAAK,CAAC,MAAM,SAAS;YACzE,CAAC,CAAC,KAAK,CAAC;IACV,CAAC;IACD,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC9D,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC;QAAE,OAAO,YAAY,CAAC;IAEzC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAChB,IAAI,CAAC;QACJ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;QAC/E,OAAO,MAAM,CAAC,WAAW,CACxB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,CAC3E,CAAC;IACH,CAAC;YAAS,CAAC;QACV,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACpB,CAAC;AACF,CAAC;AAWD,SAAS,eAAe,CAAC,KAAc;IACtC,MAAM,IAAI,GAAG,IAAA,kCAAkB,EAAC,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;IACxD,IAAI,IAAI,CAAC,MAAM,IAAI,mBAAmB;QAAE,OAAO,IAAI,CAAC;IAEpD,MAAM,MAAM,GAAG,IAAA,kCAAkB,EAAC,eAAe,CAAC,UAAU,CAAC,KAAK,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;IACjF,IAAI,MAAM,CAAC,MAAM,IAAI,mBAAmB;QAAE,OAAO,MAAM,CAAC;IAExD,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IACnC,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACnE,OAAO,IAAA,2CAAmB,EAAC,GAAG,EAAE,eAAe,CAAC,CAAC;AAClD,CAAC;AAED,SAAS,WAAW,CAAC,GAAG,UAAqB;IAC5C,OAAO,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;AACzF,CAAC;AAOD,SAAS,SAAS,CAAC,GAAY;IAC9B,IAAI,CAAC,IAAA,oBAAQ,EAAC,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IAErC,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;IACjF,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,gBAAgB,EAAE,GAAG,CAAC,aAAa,EAAE,GAAG,CAAC,iBAAiB,CAAC,CAAC;IAC3F,IAAI,KAAK,KAAK,SAAS,IAAI,MAAM,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IAElE,MAAM,WAAW,GAAG,KAAK,IAAI,CAAC,CAAC;IAC/B,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,CAAC;IACjC,OAAO;QACN,WAAW;QACX,YAAY;QACZ,WAAW,EAAE,WAAW,CAAC,GAAG,CAAC,WAAW,EAAE,GAAG,CAAC,YAAY,CAAC,IAAI,WAAW,GAAG,YAAY;KACzF,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAiB;IAC3C,MAAM,aAAa,GAClB,SAAS,CAAC,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,IAAI,SAAS,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC/E,IAAI,aAAa;QAAE,OAAO,aAAa,CAAC;IAExC,KAAK,MAAM,UAAU,IAAI,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;QACpD,MAAM,KAAK,GAAG,SAAS,CAAC,UAAU,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;QAC1D,IAAI,KAAK;YAAE,OAAO,KAAK,CAAC;IACzB,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAGD,SAAS,YAAY,CAAC,IAAY;IACjC,QAAQ,IAAI,EAAE,CAAC;QACd,KAAK,OAAO;YACX,OAAO,MAAM,CAAC;QACf,KAAK,IAAI;YACR,OAAO,WAAW,CAAC;QACpB;YACC,OAAO,IAAI,CAAC;IACd,CAAC;AACF,CAAC;AAMD,SAAS,aAAa,CAAC,OAAoB;IAC1C,MAAM,SAAS,GAA4B;QAC1C,IAAI,EAAE,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC;QAChC,OAAO,EAAE,OAAO,CAAC,OAAO;KACxB,CAAC;IAEF,IAAI,YAAY,IAAI,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAClE,SAAS,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACxD,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE;SACzE,CAAC,CAAC,CAAC;IACL,CAAC;IACD,IAAI,cAAc,IAAI,OAAO,IAAI,OAAO,OAAO,CAAC,YAAY,KAAK,QAAQ,EAAE,CAAC;QAC3E,SAAS,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IAC/C,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAGD,SAAS,gBAAgB,CAAC,MAAiB;IAC1C,OAAO;QACN,OAAO,EAAE,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;YAErD,MAAM,OAAO,GAAG,IAAA,oBAAQ,EAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;YACtE,OAAO;gBACN,OAAO,EAAE,sBAAW,CAAC,UAAU,CAAC,OAAO,CAAC;oBACvC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC;oBACxB,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,UAAU,CAAC,IAAI,EAAE;gBAClD,aAAa,EAAE,UAAU,CAAC,cAAc,EAAE,aAAa;aACvD,CAAC;QACH,CAAC,CAAC;KACF,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAAC,MAAmB,EAAE,IAAc;IACvD,OAAO,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;AAC1F,CAAC;AAkBD,yBAAiC,SAAQ,0BAAmB;IAY3D;QACC,KAAK,EAAE,CAAC;QAZT,SAAI,GAAG,qBAAqB,CAAC;QAG7B,kBAAa,GAAG,IAAI,CAAC;QAgBJ,UAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;QAGtC,aAAQ,GAAG,IAAI,GAAG,EAA8B,CAAC;QAEjD,aAAQ,GAAa,EAAE,CAAC;QAGxB,aAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;QAMtC,mBAAc,GAAG,CAAC,CAAC;QApB1B,MAAM,GAAG,GAAG,aAAa,EAAE,CAAC;QAC5B,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;QAC3B,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC,WAAW,CAAC;IACpC,CAAC;IAgCO,mBAAmB,CAAC,KAAyB;QACpD,IAAI,MAAM,GAAG,KAAK,CAAC;QACnB,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;QAChC,OAAO,MAAM,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACnD,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAClB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACpC,IAAI,IAAI;gBAAE,OAAO,IAAI,CAAC;YACtB,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC;QACD,OAAO,SAAS,CAAC;IAClB,CAAC;IAEO,kBAAkB,CAAC,QAAiB;QAC3C,IAAI,CAAC,IAAA,oBAAQ,EAAC,QAAQ,CAAC;YAAE,OAAO;QAChC,IAAI,OAAO,QAAQ,CAAC,YAAY,KAAK,QAAQ;YAAE,IAAI,CAAC,WAAW,KAAK,QAAQ,CAAC,YAAY,CAAC;QAC1F,IAAI,OAAO,QAAQ,CAAC,IAAI,KAAK,QAAQ;YAAE,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,IAAI,CAAC;QACvE,IAAI,IAAA,oBAAQ,EAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,OAAO,QAAQ,CAAC,QAAQ,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;YAC7E,IAAI,CAAC,UAAU,KAAK,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1C,CAAC;IACF,CAAC;IAEO,SAAS,CAAC,MAOjB;QACA,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;QAC9D,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,OAAO;QAElC,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC,WAAW,KAAK,WAAW,CAAC;QACjC,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC;QAElC,MAAM,IAAI,GAAe;YACxB,QAAQ,EAAE,IAAI,CAAC,WAAW;YAC1B,OAAO,EAAE,YAAY,EAAE;YACvB,cAAc,EAAE,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAAE,OAAO;YAC9D,IAAI;YACJ,oBAAoB,EAAE,OAAO,CAAC,WAAW,CAAC;YAC1C,kBAAkB,EAAE,OAAO,CAAC,WAAW,CAAC;YACxC,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;YACtB,UAAU,EAAE;gBACX,CAAC,wBAAgB,CAAC,QAAQ,CAAC,EAAE,eAAe,CAAC,QAAQ,CAAC;gBACtD,CAAC,wBAAgB,CAAC,cAAc,CAAC,EAAE,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC;gBAChE,CAAC,wBAAgB,CAAC,UAAU,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC;gBACtD,GAAG,CAAC,MAAM,CAAC,aAAa,IAAI;oBAC3B,CAAC,wBAAgB,CAAC,aAAa,CAAC,EAAE,6BAAqB;iBACvD,CAAC;aACF;SACD,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAC5B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAGO,SAAS,CAAC,KAAa;QAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI;YAAE,OAAO,SAAS,CAAC;QAE5B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,cAAc,GAAG,GAAG,CAAC;QAC1B,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;QACvC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5B,OAAO,IAAI,CAAC;IACb,CAAC;IAEO,OAAO,CAAC,KAAa,EAAE,OAAgB,EAAE,UAAuB;QACvE,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,IAAI,CAAC,UAAU,CAAC,wBAAgB,CAAC,WAAW,CAAC,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;QACzE,IAAI,UAAU,EAAE,CAAC;YAChB,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAC/B,IAAI,CAAC,UAAU,CAAC,wBAAgB,CAAC,UAAU,CAAC,GAAG,eAAe,CAAC;gBAC9D,YAAY,EAAE,UAAU,CAAC,WAAW;gBACpC,aAAa,EAAE,UAAU,CAAC,YAAY;gBACtC,YAAY,EAAE,UAAU,CAAC,WAAW;aACpC,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAEO,aAAa,CAAC,KAAiB;QACtC,IAAI,CAAC,UAAU,GAAG;YACjB,WAAW,EAAE,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,WAAW;YACpE,YAAY,EAAE,CAAC,IAAI,CAAC,UAAU,EAAE,YAAY,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,YAAY;YACvE,WAAW,EAAE,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,WAAW;SACpE,CAAC;IACH,CAAC;IAEO,QAAQ,CAAC,KAAa,EAAE,KAAc;QAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,IAAI,CAAC,MAAM,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;IAC/D,CAAC;IAaD,gBAAgB,CACf,MAAkB,EAClB,MAAmB,EACnB,KAAa,EACb,WAAoB,EACpB,KAAgB,EAChB,QAAkC,EAClC,QAAiB,EACjB,OAAgB;QAEhB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QACtC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAElC,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS;YAAE,OAAO;QAEzC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,SAAS,CAAC;YACd,KAAK;YACL,WAAW,EAAE,SAAS;YACtB,IAAI,EAAE,OAAO,IAAI,OAAO;YACxB,QAAQ,EAAE,wBAAgB,CAAC,KAAK;YAChC,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,wBAAwB,CAAC;SACrD,CAAC,CAAC;QAEH,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;YAAE,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,KAAK,CAAC;IAC1E,CAAC;IAED,cAAc,CAAC,OAAoB,EAAE,KAAa;QACjD,IAAI,KAAK,KAAK,IAAI,CAAC,SAAS;YAAE,OAAO;QAErC,IAAI,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ;YAAE,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9E,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC,CAAC;IAC/D,CAAC;IAED,gBAAgB,CAAC,KAAY,EAAE,KAAa;QAC3C,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC7B,CAAC;IAED,oBAAoB,CACnB,GAAe,EACf,QAAyB,EACzB,KAAa,EACb,WAAoB,EACpB,YAAsC,EACtC,KAAgB,EAChB,QAAkC,EAClC,OAAgB;QAEhB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QACtC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAClC,IAAI,CAAC,SAAS,CAAC;YACd,KAAK;YACL,WAAW;YACX,IAAI,EAAE,OAAO,IAAI,cAAc,CAAC,GAAG,EAAE,YAAY,CAAC;YAClD,QAAQ,EAAE,wBAAgB,CAAC,SAAS;YACpC,MAAM,EAAE,EAAE,QAAQ,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE;YACxD,aAAa,EAAE,IAAI;SACnB,CAAC,CAAC;IACJ,CAAC;IAED,cAAc,CACb,GAAe,EACf,OAAiB,EACjB,KAAa,EACb,WAAoB,EACpB,YAAsC,EACtC,KAAgB,EAChB,QAAkC,EAClC,OAAgB;QAEhB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QACtC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAClC,IAAI,CAAC,SAAS,CAAC;YACd,KAAK;YACL,WAAW;YACX,IAAI,EAAE,OAAO,IAAI,cAAc,CAAC,GAAG,EAAE,OAAO,CAAC;YAC7C,QAAQ,EAAE,wBAAgB,CAAC,GAAG;YAC9B,MAAM,EAAE,OAAO;SACf,CAAC,CAAC;IACJ,CAAC;IAED,YAAY,CAAC,MAAiB,EAAE,KAAa;QAC5C,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,gBAAgB,CAAC,MAAM,CAAC,EAAE,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC;IAC1E,CAAC;IAED,cAAc,CAAC,KAAY,EAAE,KAAa;QACzC,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC7B,CAAC;IAED,eAAe,CACd,IAAgB,EAChB,KAAa,EACb,KAAa,EACb,WAAoB,EACpB,KAAgB,EAChB,QAAkC,EAClC,OAAgB;QAEhB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QACtC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAClC,IAAI,CAAC,SAAS,CAAC;YACd,KAAK;YACL,WAAW;YACX,IAAI,EAAE,OAAO,IAAI,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC;YAC7C,QAAQ,EAAE,wBAAgB,CAAC,IAAI;YAC/B,MAAM,EAAE,KAAK;SACb,CAAC,CAAC;IACJ,CAAC;IAED,aAAa,CAAC,MAAe,EAAE,KAAa;QAC3C,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC7B,CAAC;IAED,eAAe,CAAC,KAAY,EAAE,KAAa;QAC1C,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC7B,CAAC;IAED,oBAAoB,CACnB,SAAqB,EACrB,KAAa,EACb,KAAa,EACb,WAAoB,EACpB,KAAgB,EAChB,QAAkC,EAClC,IAAa;QAEb,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QACtC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAClC,IAAI,CAAC,SAAS,CAAC;YACd,KAAK;YACL,WAAW;YACX,IAAI,EAAE,IAAI,IAAI,cAAc,CAAC,SAAS,EAAE,WAAW,CAAC;YACpD,QAAQ,EAAE,wBAAgB,CAAC,SAAS;YACpC,MAAM,EAAE,KAAK;SACb,CAAC,CAAC;IACJ,CAAC;IAED,kBAAkB,CAAC,SAA8B,EAAE,KAAa;QAC/D,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAChC,CAAC;IAED,oBAAoB,CAAC,KAAY,EAAE,KAAa;QAC/C,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC7B,CAAC;IAMD,MAAM;QACL,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvF,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QAE9D,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ;aACzB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;aACrC,MAAM,CAAC,CAAC,IAAI,EAAsB,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;QAK3D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAClE,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;QACpC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,IAAI;gBAAE,IAAI,CAAC,kBAAkB,GAAG,QAAQ,CAAC;QAC9C,CAAC;QAED,OAAO;YACN,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,OAAO;YACxB,KAAK;YACL,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS;YAGT,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YACvB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,eAAe,EAAE,IAAI,CAAC,eAAe;SACrC,CAAC;IACH,CAAC;CACD"}

package/dist/nodes/vendors/Databricks/mlflow/trace-writer.d.ts

@@ -0,0 +1,41 @@
+import { type IDataObject } from 'n8n-workflow';
+import type { CollectedTrace } from './types';
+export interface MlflowResponse {
+    status: number;
+    body: unknown;
+}
+export type MlflowRequest = (options: {
+    method: 'GET' | 'POST';
+    path: string;
+    qs?: Record<string, string>;
+    body?: IDataObject;
+}) => Promise<MlflowResponse>;
+export type SignedUpload = (options: {
+    url: string;
+    body: string;
+}) => Promise<void>;
+export interface TraceWriterOptions {
+    request: MlflowRequest;
+    upload: SignedUpload;
+    experimentId: string;
+}
+export declare function defaultExperimentName(workflowId: string): string;
+export declare function ensureExperiment(request: MlflowRequest, experimentName: string): Promise<string>;
+export declare function buildTraceInfo(trace: CollectedTrace, experimentId: string): {
+    trace_id: string;
+    client_request_id: string | undefined;
+    trace_location: {
+        type: string;
+        mlflow_experiment: {
+            experiment_id: string;
+        };
+    };
+    request_preview: string;
+    response_preview: string;
+    request_time: string;
+    execution_duration: string;
+    state: import("./types").MlflowStatusCode;
+    trace_metadata: Record<string, string>;
+    tags: Record<string, string>;
+};
+export declare function writeTrace(options: TraceWriterOptions, trace: CollectedTrace): Promise<void>;

package/dist/nodes/vendors/Databricks/mlflow/trace-writer.js

@@ -0,0 +1,129 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.defaultExperimentName = defaultExperimentName;
+exports.ensureExperiment = ensureExperiment;
+exports.buildTraceInfo = buildTraceInfo;
+exports.writeTrace = writeTrace;
+const is_record_1 = require("@n8n/utils/is-record");
+const sanitize_error_detail_1 = require("@n8n/utils/redaction/sanitize-error-detail");
+const n8n_workflow_1 = require("n8n-workflow");
+const types_1 = require("./types");
+const EXPERIMENT_GET_BY_NAME = '/api/2.0/mlflow/experiments/get-by-name';
+const EXPERIMENT_CREATE = '/api/2.0/mlflow/experiments/create';
+const TRACE_CREATE = '/api/3.0/mlflow/traces';
+const uploadCredentialsPath = (traceId) => `/api/2.0/mlflow/traces/${encodeURIComponent(traceId)}/credentials-for-data-upload`;
+const MAX_PREVIEW_CHARS = 1_000;
+const MAX_FAILURE_DETAIL_CHARS = 500;
+function describeFailure(body) {
+    if (!(0, is_record_1.isRecord)(body))
+        return '';
+    const code = typeof body.error_code === 'string' ? body.error_code : '';
+    const message = typeof body.message === 'string' ? body.message : '';
+    return (0, sanitize_error_detail_1.sanitizeErrorDetail)([code, message].filter(Boolean).join(': '), MAX_FAILURE_DETAIL_CHARS);
+}
+function assertOk(response, what) {
+    if (response.status < 200 || response.status >= 300) {
+        const detail = describeFailure(response.body);
+        const message = `Databricks ${what} failed with HTTP ${response.status}${detail ? ` (${detail})` : ''}`;
+        throw response.status >= 500 ? new n8n_workflow_1.OperationalError(message) : new n8n_workflow_1.UserError(message);
+    }
+    return response.body;
+}
+function defaultExperimentName(workflowId) {
+    return `/Shared/n8n-workflows-${workflowId}`;
+}
+async function ensureExperiment(request, experimentName) {
+    const found = await request({
+        method: 'GET',
+        path: EXPERIMENT_GET_BY_NAME,
+        qs: { experiment_name: experimentName },
+    });
+    if (found.status === 200 && (0, is_record_1.isRecord)(found.body) && (0, is_record_1.isRecord)(found.body.experiment)) {
+        const id = found.body.experiment.experiment_id;
+        if (typeof id === 'string')
+            return id;
+    }
+    if (found.status !== 404)
+        assertOk(found, 'experiment lookup');
+    const createResponse = await request({
+        method: 'POST',
+        path: EXPERIMENT_CREATE,
+        body: { name: experimentName },
+    });
+    if (createResponse.status >= 400 &&
+        (0, is_record_1.isRecord)(createResponse.body) &&
+        createResponse.body.error_code === 'RESOURCE_ALREADY_EXISTS') {
+        const afterRace = await request({
+            method: 'GET',
+            path: EXPERIMENT_GET_BY_NAME,
+            qs: { experiment_name: experimentName },
+        });
+        if (afterRace.status === 200 &&
+            (0, is_record_1.isRecord)(afterRace.body) &&
+            (0, is_record_1.isRecord)(afterRace.body.experiment)) {
+            const id = afterRace.body.experiment.experiment_id;
+            if (typeof id === 'string')
+                return id;
+        }
+    }
+    const created = assertOk(createResponse, 'experiment create');
+    if ((0, is_record_1.isRecord)(created) && typeof created.experiment_id === 'string') {
+        return created.experiment_id;
+    }
+    throw new n8n_workflow_1.UnexpectedError('Databricks did not return an experiment id');
+}
+function preview(text, span, attribute) {
+    return (text ?? span?.attributes[attribute] ?? '').slice(0, MAX_PREVIEW_CHARS);
+}
+function buildTraceInfo(trace, experimentId) {
+    const root = trace.spans.find((span) => span.span_id === trace.rootSpanId);
+    const tags = { 'n8n.source': 'n8n' };
+    if (trace.workflowId)
+        tags['n8n.workflowId'] = trace.workflowId;
+    if (trace.nodeName)
+        tags['n8n.node'] = trace.nodeName;
+    const usage = trace.tokenUsage;
+    const traceMetadata = {};
+    if (usage) {
+        traceMetadata[types_1.MLFLOW_TRACE_METADATA.TokenUsage] = JSON.stringify({
+            input_tokens: usage.inputTokens,
+            output_tokens: usage.outputTokens,
+            total_tokens: usage.totalTokens,
+        });
+    }
+    return {
+        trace_id: trace.traceId,
+        client_request_id: trace.executionId,
+        trace_location: {
+            type: 'MLFLOW_EXPERIMENT',
+            mlflow_experiment: { experiment_id: experimentId },
+        },
+        request_preview: preview(trace.requestPreview, root, types_1.MLFLOW_ATTRIBUTE.SpanInputs),
+        response_preview: preview(trace.responsePreview, root, types_1.MLFLOW_ATTRIBUTE.SpanOutputs),
+        request_time: new Date(trace.startTimeMs).toISOString(),
+        execution_duration: `${(trace.endTimeMs - trace.startTimeMs) / 1000}s`,
+        state: trace.state,
+        trace_metadata: traceMetadata,
+        tags,
+    };
+}
+async function writeTrace(options, trace) {
+    const { request, upload, experimentId } = options;
+    assertOk(await request({
+        method: 'POST',
+        path: TRACE_CREATE,
+        body: { trace: { trace_info: buildTraceInfo(trace, experimentId) } },
+    }), 'trace create');
+    const credentials = assertOk(await request({ method: 'GET', path: uploadCredentialsPath(trace.traceId) }), 'upload credentials');
+    const signedUri = (0, is_record_1.isRecord)(credentials) && (0, is_record_1.isRecord)(credentials.credential_info)
+        ? credentials.credential_info.signed_uri
+        : undefined;
+    if (typeof signedUri !== 'string' || signedUri.length === 0) {
+        throw new n8n_workflow_1.UnexpectedError('Databricks did not return an upload URL for the trace data');
+    }
+    await upload({
+        url: signedUri,
+        body: JSON.stringify({ spans: trace.spans }),
+    });
+}
+//# sourceMappingURL=trace-writer.js.map

package/dist/nodes/vendors/Databricks/mlflow/trace-writer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"trace-writer.js","sourceRoot":"","sources":["../../../../../nodes/vendors/Databricks/mlflow/trace-writer.ts"],"names":[],"mappings":";;;;;;AAAA,oDAAgD;AAChD,sFAAiF;AACjF,+CAA8F;AAG9F,mCAAkE;AAQlE,MAAM,sBAAsB,GAAG,yCAAyC,CAAC;AACzE,MAAM,iBAAiB,GAAG,oCAAoC,CAAC;AAC/D,MAAM,YAAY,GAAG,wBAAwB,CAAC;AAC9C,MAAM,qBAAqB,GAAG,CAAC,OAAe,EAAE,EAAE,CACjD,0BAA0B,kBAAkB,CAAC,OAAO,CAAC,8BAA8B,CAAC;AAErF,MAAM,iBAAiB,GAAG,KAAK,CAAC;AAChC,MAAM,wBAAwB,GAAG,GAAG,CAAC;AAmCrC,SAAS,eAAe,CAAC,IAAa;IACrC,IAAI,CAAC,IAAA,oBAAQ,EAAC,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IAC/B,MAAM,IAAI,GAAG,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;IACxE,MAAM,OAAO,GAAG,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IACrE,OAAO,IAAA,2CAAmB,EAAC,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,wBAAwB,CAAC,CAAC;AAClG,CAAC;AAED,SAAS,QAAQ,CAAC,QAAwB,EAAE,IAAY;IACvD,IAAI,QAAQ,CAAC,MAAM,GAAG,GAAG,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;QACrD,MAAM,MAAM,GAAG,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,cAAc,IAAI,qBAAqB,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,KAAK,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QACxG,MAAM,QAAQ,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,+BAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,wBAAS,CAAC,OAAO,CAAC,CAAC;IACvF,CAAC;IACD,OAAO,QAAQ,CAAC,IAAI,CAAC;AACtB,CAAC;AAKD,+BAAsC,UAAkB;IACvD,OAAO,yBAAyB,UAAU,EAAE,CAAC;AAC9C,CAAC;AAQM,KAAK,2BACX,OAAsB,EACtB,cAAsB;IAEtB,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC;QAC3B,MAAM,EAAE,KAAK;QACb,IAAI,EAAE,sBAAsB;QAC5B,EAAE,EAAE,EAAE,eAAe,EAAE,cAAc,EAAE;KACvC,CAAC,CAAC;IAEH,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,IAAI,IAAA,oBAAQ,EAAC,KAAK,CAAC,IAAI,CAAC,IAAI,IAAA,oBAAQ,EAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;QACrF,MAAM,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC;QAC/C,IAAI,OAAO,EAAE,KAAK,QAAQ;YAAE,OAAO,EAAE,CAAC;IACvC,CAAC;IAGD,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG;QAAE,QAAQ,CAAC,KAAK,EAAE,mBAAmB,CAAC,CAAC;IAE/D,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC;QACpC,MAAM,EAAE,MAAM;QACd,IAAI,EAAE,iBAAiB;QACvB,IAAI,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE;KAC9B,CAAC,CAAC;IAMH,IACC,cAAc,CAAC,MAAM,IAAI,GAAG;QAC5B,IAAA,oBAAQ,EAAC,cAAc,CAAC,IAAI,CAAC;QAC7B,cAAc,CAAC,IAAI,CAAC,UAAU,KAAK,yBAAyB,EAC3D,CAAC;QACF,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC;YAC/B,MAAM,EAAE,KAAK;YACb,IAAI,EAAE,sBAAsB;YAC5B,EAAE,EAAE,EAAE,eAAe,EAAE,cAAc,EAAE;SACvC,CAAC,CAAC;QACH,IACC,SAAS,CAAC,MAAM,KAAK,GAAG;YACxB,IAAA,oBAAQ,EAAC,SAAS,CAAC,IAAI,CAAC;YACxB,IAAA,oBAAQ,EAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,EAClC,CAAC;YACF,MAAM,EAAE,GAAG,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC;YACnD,IAAI,OAAO,EAAE,KAAK,QAAQ;gBAAE,OAAO,EAAE,CAAC;QACvC,CAAC;IACF,CAAC;IAED,MAAM,OAAO,GAAG,QAAQ,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;IAC9D,IAAI,IAAA,oBAAQ,EAAC,OAAO,CAAC,IAAI,OAAO,OAAO,CAAC,aAAa,KAAK,QAAQ,EAAE,CAAC;QACpE,OAAO,OAAO,CAAC,aAAa,CAAC;IAC9B,CAAC;IACD,MAAM,IAAI,8BAAe,CAAC,4CAA4C,CAAC,CAAC;AACzE,CAAC;AAOD,SAAS,OAAO,CACf,IAAwB,EACxB,IAA4B,EAC5B,SAAiB;IAEjB,OAAO,CAAC,IAAI,IAAI,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;AAChF,CAAC;AAOD,wBAA+B,KAAqB,EAAE,YAAoB;IACzE,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,KAAK,KAAK,CAAC,UAAU,CAAC,CAAC;IAC3E,MAAM,IAAI,GAA2B,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;IAC7D,IAAI,KAAK,CAAC,UAAU;QAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,KAAK,CAAC,UAAU,CAAC;IAChE,IAAI,KAAK,CAAC,QAAQ;QAAE,IAAI,CAAC,UAAU,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC;IAGtD,MAAM,KAAK,GAAG,KAAK,CAAC,UAAU,CAAC;IAC/B,MAAM,aAAa,GAA2B,EAAE,CAAC;IACjD,IAAI,KAAK,EAAE,CAAC;QACX,aAAa,CAAC,6BAAqB,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;YAChE,YAAY,EAAE,KAAK,CAAC,WAAW;YAC/B,aAAa,EAAE,KAAK,CAAC,YAAY;YACjC,YAAY,EAAE,KAAK,CAAC,WAAW;SAC/B,CAAC,CAAC;IACJ,CAAC;IAED,OAAO;QACN,QAAQ,EAAE,KAAK,CAAC,OAAO;QAEvB,iBAAiB,EAAE,KAAK,CAAC,WAAW;QACpC,cAAc,EAAE;YACf,IAAI,EAAE,mBAAmB;YACzB,iBAAiB,EAAE,EAAE,aAAa,EAAE,YAAY,EAAE;SAClD;QACD,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,IAAI,EAAE,wBAAgB,CAAC,UAAU,CAAC;QACjF,gBAAgB,EAAE,OAAO,CAAC,KAAK,CAAC,eAAe,EAAE,IAAI,EAAE,wBAAgB,CAAC,WAAW,CAAC;QACpF,YAAY,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE;QACvD,kBAAkB,EAAE,GAAG,CAAC,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC,WAAW,CAAC,GAAG,IAAI,GAAG;QACtE,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,cAAc,EAAE,aAAa;QAC7B,IAAI;KACJ,CAAC;AACH,CAAC;AAaM,KAAK,qBACX,OAA2B,EAC3B,KAAqB;IAErB,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC;IAElD,QAAQ,CACP,MAAM,OAAO,CAAC;QACb,MAAM,EAAE,MAAM;QACd,IAAI,EAAE,YAAY;QAClB,IAAI,EAAE,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,cAAc,CAAC,KAAK,EAAE,YAAY,CAAC,EAAE,EAAE;KACpE,CAAC,EACF,cAAc,CACd,CAAC;IAEF,MAAM,WAAW,GAAG,QAAQ,CAC3B,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,qBAAqB,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,EAC5E,oBAAoB,CACpB,CAAC;IAEF,MAAM,SAAS,GACd,IAAA,oBAAQ,EAAC,WAAW,CAAC,IAAI,IAAA,oBAAQ,EAAC,WAAW,CAAC,eAAe,CAAC;QAC7D,CAAC,CAAC,WAAW,CAAC,eAAe,CAAC,UAAU;QACxC,CAAC,CAAC,SAAS,CAAC;IACd,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7D,MAAM,IAAI,8BAAe,CAAC,4DAA4D,CAAC,CAAC;IACzF,CAAC;IAED,MAAM,MAAM,CAAC;QACZ,GAAG,EAAE,SAAS;QACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;KAC5C,CAAC,CAAC;AACJ,CAAC"}

package/dist/nodes/vendors/Databricks/mlflow/transport.d.ts

@@ -0,0 +1,7 @@
+import { type IExecuteFunctions } from 'n8n-workflow';
+import { type DatabricksOAuth2Credential } from '../../../../utils/databricks/token-provider';
+import type { MlflowRequest, SignedUpload } from './trace-writer';
+export declare function createMlflowTransport(ctx: IExecuteFunctions, credential: DatabricksOAuth2Credential): {
+    request: MlflowRequest;
+    upload: SignedUpload;
+};

package/dist/nodes/vendors/Databricks/mlflow/transport.js

@@ -0,0 +1,72 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createMlflowTransport = createMlflowTransport;
+const ai_utilities_1 = require("@n8n/ai-utilities");
+const constants_1 = require("n8n-nodes-base/dist/nodes/Databricks/constants");
+const n8n_workflow_1 = require("n8n-workflow");
+const constants_2 = require("../../../../utils/databricks/constants");
+const token_provider_1 = require("../../../../utils/databricks/token-provider");
+const REQUEST_TIMEOUT_MS = 60_000;
+function toMlflowResponse(error) {
+    if (error instanceof n8n_workflow_1.NodeApiError && error.httpCode) {
+        return { status: Number(error.httpCode), body: error.context.data ?? {} };
+    }
+    throw error;
+}
+function createMlflowTransport(ctx, credential) {
+    (0, constants_2.assertHttpsHost)(ctx, credential.host);
+    const host = credential.host.replace(/\/$/, '');
+    const egressFilter = ctx.helpers.getSecureEgressFilter();
+    const cancelSignal = ctx.getExecutionCancelSignal();
+    const allowedDomains = (0, n8n_workflow_1.getCredentialAllowedDomains)({
+        node: ctx.getNode(),
+        credentialData: credential,
+        credentialOwnedSurface: true,
+        nodeEndpointUrl: host,
+    });
+    const request = async ({ method, path, qs, body }) => {
+        try {
+            const response = await ctx.helpers.httpRequestWithAuthentication.call(ctx, token_provider_1.DATABRICKS_CREDENTIAL_TYPE, {
+                method,
+                url: `${host}${path}`,
+                qs,
+                body,
+                headers: { 'User-Agent': constants_1.DATABRICKS_PARTNER_USER_AGENT },
+                timeout: REQUEST_TIMEOUT_MS,
+                allowedDomains,
+                returnFullResponse: true,
+            });
+            return { status: response.statusCode, body: response.body };
+        }
+        catch (error) {
+            return toMlflowResponse(error);
+        }
+    };
+    const withDeadline = async (input, init) => {
+        const signals = [AbortSignal.timeout(REQUEST_TIMEOUT_MS)];
+        if (init?.signal)
+            signals.push(init.signal);
+        if (cancelSignal)
+            signals.push(cancelSignal);
+        return await (0, ai_utilities_1.proxyFetch)({
+            input,
+            init: { ...init, signal: AbortSignal.any(signals) },
+            egressFilter,
+        });
+    };
+    const upload = async ({ url, body }) => {
+        if (!URL.canParse(url) || new URL(url).protocol !== 'https:') {
+            throw new n8n_workflow_1.NodeOperationError(ctx.getNode(), 'Databricks upload URL must use https');
+        }
+        const response = await withDeadline(url, {
+            method: 'PUT',
+            body,
+            headers: { 'Content-Type': 'application/json' },
+        });
+        if (!response.ok) {
+            throw new n8n_workflow_1.OperationalError(`Databricks trace data upload failed with HTTP ${response.status}`);
+        }
+    };
+    return { request, upload };
+}
+//# sourceMappingURL=transport.js.map