@kyuda/n8n-nodes-databricks 1.2.2 → 1.3.0

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@kyuda/n8n-nodes-databricks",
-  "version": "1.2.2",
-  "description": "Comprehensive Databricks integration for n8n — SQL, Unity Catalog, Jobs, Clusters, Files, Genie, Vector Search, Model Serving, and LangChain AI nodes",
+  "version": "1.3.0",
+  "description": "Comprehensive Databricks integration for n8n — SQL, Unity Catalog, Jobs, Clusters, Files, Genie, Vector Search, and Model Serving",
   "keywords": [
     "n8n",
     "n8n-node",
@@ -11,9 +11,7 @@
     "machine-learning",
     "vector-search",
     "unity-catalog",
-    "genie",
-    "langchain",
-    "ai-agent"
+    "genie"
   ],
   "license": "MIT",
   "homepage": "https://github.com/kyudahq/n8n",
@@ -27,12 +25,9 @@
     "directory": "@kyuda/n8n-nodes-databricks"
   },
   "main": "dist/index.js",
-  "_moduleAliases": {
-    "@utils": "dist/utils"
-  },
   "scripts": {
     "build": "rm -rf dist/ && tsc -p tsconfig.json && tsc-alias -p tsconfig.json && gulp build:icons",
-    "start": "node -r module-alias/register dist/index.js",
+    "start": "node dist/index.js",
     "dev": "concurrently \"npm:dev:*\"",
     "dev:watch": "tsc -p tsconfig.json --watch && tsc-alias -p tsconfig.json --watch",
     "dev:serve": "NODE_ENV=development ts-node-dev -r tsconfig-paths/register --respawn --inspect --ignore-watch node_modules ./src/index.ts",
@@ -52,40 +47,21 @@
       "dist/credentials/Databricks.credentials.js"
     ],
     "nodes": [
-      "dist/nodes/Databricks/Databricks.node.js",
-      "dist/nodes/llms/LmChatDatabricks/LmChatDatabricks.node.js",
-      "dist/nodes/embeddings/EmbeddingsDatabricks/EmbeddingsDatabricks.node.js",
-      "dist/nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.js",
-      "dist/nodes/agents/DatabricksAiAgent/DatabricksAiAgent.node.js"
+      "dist/nodes/Databricks/Databricks.node.js"
     ]
   },
   "dependencies": {
-    "@langchain/classic": "^1.0.17",
-    "@langchain/community": "^1.1.15",
-    "@langchain/core": "^1.1.24",
-    "@langchain/openai": "^1.2.7",
-    "@langchain/textsplitters": "^1.0.1",
-    "@n8n/json-schema-to-zod": "^1.6.0",
-    "langchain": "^1.2.24",
-    "lodash": "^4.17.23",
-    "log4js": "^6.9.1",
     "mime-types": "^3.0.1",
-    "mlflow-tracing": "0.1.0",
-    "n8n-workflow": "2.3.1",
-    "tmp-promise": "3.0.3",
-    "zod": "^3.25.76"
+    "n8n-workflow": "2.3.1"
   },
   "devDependencies": {
     "@types/jest": "^29.5.14",
-    "@types/json-schema": "^7.0.15",
-    "@types/lodash": "^4.17.16",
     "@types/mime-types": "^3.0.1",
     "@types/node": "^20.19.1",
     "concurrently": "^9.1.2",
     "gulp": "^5.0.0",
     "gulp-cli": "^3.0.0",
     "jest-mock-extended": "^4.0.0",
-    "module-alias": "^2.3.4",
     "ts-node-dev": "^2.0.0",
     "tsc-alias": "^1.8.15",
     "tsconfig-paths": "^4.2.0",

package/dist/nodes/agents/DatabricksAiAgent/CallbackHandler.d.ts

@@ -1,66 +0,0 @@
-import type { AgentAction, AgentFinish } from "@langchain/core/agents";
-import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
-import type { Document } from "@langchain/core/documents";
-import type { Serialized } from "@langchain/core/load/serializable";
-import { BaseMessage, type BaseMessageFields } from "@langchain/core/messages";
-import type { LLMResult } from "@langchain/core/outputs";
-import type { ChainValues } from "@langchain/core/utils/types";
-export type LlmMessage = {
-    role: string;
-    content: BaseMessageFields["content"];
-    additional_kwargs?: BaseMessageFields["additional_kwargs"];
-};
-export type AnonymousLlmMessage = {
-    content: BaseMessageFields["content"];
-    additional_kwargs?: BaseMessageFields["additional_kwargs"];
-};
-type ErrorLike = Error | {
-    message: string;
-    stack?: string;
-    [key: string]: unknown;
-};
-type LLMTokenIndex = import("@langchain/core/dist/callbacks/base").NewTokenIndices;
-type ConstructorParams = {
-    userId?: string;
-    sessionId?: string;
-    tags?: string[];
-    version?: string;
-    traceMetadata?: Record<string, unknown>;
-};
-export declare class CallbackHandler extends BaseCallbackHandler {
-    name: string;
-    private readonly runMap;
-    private readonly maxMapSize;
-    private _lastTraceId;
-    get lastTraceId(): string | null;
-    constructor(params?: ConstructorParams);
-    get logger(): import("log4js/types/log4js").Logger;
-    private logError;
-    private ensureMapSize;
-    cleanup(): void;
-    handleLLMNewToken(token: string, _idx: LLMTokenIndex, runId: string, _parentRunId?: string, _tags?: string[], _fields?: Record<string, unknown>): Promise<void>;
-    handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, runType?: string, name?: string): Promise<void>;
-    handleAgentAction(action: AgentAction, runId: string, parentRunId?: string): Promise<void>;
-    handleAgentEnd?(action: AgentFinish, runId: string, parentRunId?: string): Promise<void>;
-    handleChainError(err: ErrorLike, runId: string, parentRunId?: string | undefined): Promise<void>;
-    handleGenerationStart(llm: Serialized, messages: Record<string, unknown> | Record<string, unknown>[] | string[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
-    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
-    handleChainEnd(outputs: ChainValues, runId: string, parentRunId?: string | undefined): Promise<void>;
-    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
-    handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
-    handleRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string): Promise<void>;
-    handleRetrieverEnd(documents: Document<Record<string, any>>[], runId: string, parentRunId?: string | undefined): Promise<void>;
-    handleRetrieverError(err: ErrorLike, runId: string, parentRunId?: string | undefined): Promise<void>;
-    handleToolEnd(output: string, runId: string, parentRunId?: string | undefined): Promise<void>;
-    handleToolError(err: ErrorLike, runId: string, parentRunId?: string | undefined): Promise<void>;
-    handleLLMEnd(output: LLMResult, runId: string, parentRunId?: string | undefined): Promise<void>;
-    handleLLMError(err: ErrorLike, runId: string, parentRunId?: string | undefined): Promise<void>;
-    private startAndRegisterOtelSpan;
-    private handleOtelSpanEnd;
-    private parseAzureRefusalError;
-    private joinTagsAndMetaData;
-    private extractUsageMetadata;
-    private extractModelNameFromMetadata;
-    private extractChatMessageContent;
-}
-export {};

package/dist/nodes/agents/DatabricksAiAgent/CallbackHandler.js

@@ -1,531 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.CallbackHandler = void 0;
-const base_1 = require("@langchain/core/callbacks/base");
-const messages_1 = require("@langchain/core/messages");
-const log4js_1 = require("log4js");
-const mlflow = __importStar(require("mlflow-tracing"));
-class CallbackHandler extends base_1.BaseCallbackHandler {
-    get lastTraceId() {
-        return this._lastTraceId;
-    }
-    constructor(params) {
-        super();
-        this.name = "MLFlowCallbackHandler";
-        this.runMap = new Map();
-        this.maxMapSize = 1000;
-        this._lastTraceId = null;
-    }
-    get logger() {
-        return (0, log4js_1.getLogger)("Callback-handler");
-    }
-    logError(context, error) {
-        const message = error instanceof Error ? error.message : String(error);
-        this.logger.debug(`${context}: ${message}`);
-    }
-    ensureMapSize() {
-        if (this.runMap.size <= this.maxMapSize)
-            return;
-        this.logger.warn(`runMap size (${this.runMap.size}) exceeded limit (${this.maxMapSize}). ` +
-            'Cleaning up oldest entries. This may indicate spans not being properly closed.');
-        const entries = Array.from(this.runMap.entries());
-        const toDelete = entries.slice(0, entries.length - this.maxMapSize);
-        for (const [key, span] of toDelete) {
-            try {
-                span.end();
-                this.logger.debug(`Force-closed orphaned span: ${key}`);
-            }
-            catch (e) {
-                this.logError(`Failed to close orphaned span ${key}`, e);
-            }
-            this.runMap.delete(key);
-        }
-    }
-    cleanup() {
-        if (this.runMap.size === 0)
-            return;
-        this.logger.debug(`Cleaning up ${this.runMap.size} remaining spans`);
-        for (const [runId, span] of this.runMap.entries()) {
-            try {
-                span.end();
-            }
-            catch (e) {
-                this.logError(`Failed to end span ${runId} during cleanup`, e);
-            }
-        }
-        this.runMap.clear();
-    }
-    async handleLLMNewToken(token, _idx, runId, _parentRunId, _tags, _fields) {
-        this.logger.info(`LLM returning token: ${token}`);
-    }
-    async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name) {
-        var _a, _b;
-        try {
-            this.logger.debug(`Chain start with Id: ${runId}`);
-            const runName = (_b = name !== null && name !== void 0 ? name : (_a = chain.id.at(-1)) === null || _a === void 0 ? void 0 : _a.toString()) !== null && _b !== void 0 ? _b : "Langchain Run";
-            const filter_chains = ["runnablelambda", "runnablemap", "toolcallingagentoutputparser"];
-            const filter_span = runName ? filter_chains.some(sub => runName.toLowerCase().includes(sub)) : false;
-            if (filter_span) {
-                return;
-            }
-            let chat_history = 'chat_history' in inputs ? inputs['chat_history'] : undefined;
-            if (chat_history) {
-                chat_history = chat_history.map((m) => this.extractChatMessageContent(m));
-            }
-            const user_message = { "content": inputs['input'], "role": "user" };
-            const messages = chat_history ? chat_history.concat([user_message]) : [user_message];
-            this.startAndRegisterOtelSpan({
-                type: mlflow.SpanType.AGENT,
-                runName: runName,
-                runId: runId,
-                parentRunId: parentRunId,
-                tags: tags,
-                metadata: metadata,
-                attributes: {
-                    messages: messages,
-                    system_message: inputs['system_message']
-                },
-            });
-        }
-        catch (e) {
-            this.logger.debug(e instanceof Error ? e.message : String(e));
-        }
-    }
-    async handleAgentAction(action, runId, parentRunId) {
-        try {
-            this.logger.debug(`Agent action ${action.tool} with ID: ${runId}`);
-        }
-        catch (e) {
-            this.logger.debug(e instanceof Error ? e.message : String(e));
-        }
-    }
-    async handleAgentEnd(action, runId, parentRunId) {
-        try {
-            this.logger.debug(`Agent finish with ID: ${runId}`);
-        }
-        catch (e) {
-            this.logger.debug(e instanceof Error ? e.message : String(e));
-        }
-    }
-    async handleChainError(err, runId, parentRunId) {
-        try {
-            const errorMessage = err instanceof Error ? err.message : String(err);
-            this.logger.debug(`Chain error: ${errorMessage} with ID: ${runId}`);
-            const azureRefusalError = this.parseAzureRefusalError(err);
-            this.handleOtelSpanEnd({
-                runId,
-                attributes: {
-                    level: "ERROR",
-                    statusMessage: errorMessage + azureRefusalError,
-                },
-            });
-        }
-        catch (e) {
-            this.logError('handleChainError', e);
-        }
-    }
-    async handleGenerationStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
-        var _a, _b;
-        this.logger.debug(`Generation start with ID: ${runId} and parentRunId ${parentRunId}`);
-        const runName = (_b = name !== null && name !== void 0 ? name : (_a = llm.id.at(-1)) === null || _a === void 0 ? void 0 : _a.toString()) !== null && _b !== void 0 ? _b : "Langchain Generation";
-        const modelParameters = {};
-        const invocationParams = extraParams === null || extraParams === void 0 ? void 0 : extraParams["invocation_params"];
-        if (invocationParams) {
-            const paramKeys = [
-                'temperature',
-                'max_tokens',
-                'top_p',
-                'frequency_penalty',
-                'presence_penalty',
-                'request_timeout'
-            ];
-            for (const key of paramKeys) {
-                const value = invocationParams[key];
-                if (value !== undefined && value !== null && typeof value === 'number') {
-                    modelParameters[key] = value;
-                }
-            }
-        }
-        this.startAndRegisterOtelSpan({
-            type: mlflow.SpanType.CHAT_MODEL,
-            runName: runName,
-            runId: runId,
-            parentRunId: parentRunId,
-            metadata: metadata,
-            tags: tags,
-            attributes: {
-                messages: messages,
-                modelParameters: modelParameters
-            },
-        });
-    }
-    async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
-        try {
-            this.logger.debug(`Chat model start with ID: ${runId}`);
-            const prompts = messages.flatMap((message) => message.map((m) => this.extractChatMessageContent(m)));
-            this.handleGenerationStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name);
-        }
-        catch (e) {
-            this.logger.debug(e instanceof Error ? e.message : String(e));
-        }
-    }
-    async handleChainEnd(outputs, runId, parentRunId) {
-        try {
-            this.logger.debug(`Chain end with ID: ${runId}`);
-            let finalOutput = outputs;
-            if (typeof outputs === "object" &&
-                "output" in outputs &&
-                typeof outputs["output"] === "string") {
-                finalOutput = outputs["output"];
-            }
-            this.handleOtelSpanEnd({
-                runId,
-                attributes: {
-                    output: finalOutput,
-                },
-            });
-        }
-        catch (e) {
-            this.logger.debug(e instanceof Error ? e.message : String(e));
-        }
-    }
-    async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name) {
-        try {
-            this.logger.debug(`LLM start with ID: ${runId}`);
-            this.handleGenerationStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name);
-        }
-        catch (e) {
-            this.logger.debug(e instanceof Error ? e.message : String(e));
-        }
-    }
-    async handleToolStart(tool, input, runId, parentRunId, tags, metadata, name) {
-        var _a, _b;
-        try {
-            this.logger.debug(`Tool start with ID: ${runId}`);
-            const tool_name = (_b = name !== null && name !== void 0 ? name : (_a = tool.id.at(-1)) === null || _a === void 0 ? void 0 : _a.toString()) !== null && _b !== void 0 ? _b : "Tool execution";
-            this.startAndRegisterOtelSpan({
-                type: mlflow.SpanType.TOOL,
-                runId: runId,
-                parentRunId: parentRunId,
-                runName: tool_name,
-                attributes: {
-                    tool_name: tool_name,
-                    args: input,
-                },
-                metadata: metadata,
-                tags: tags,
-            });
-        }
-        catch (e) {
-            this.logger.debug(e instanceof Error ? e.message : String(e));
-        }
-    }
-    async handleRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name) {
-        var _a, _b;
-        try {
-            this.logger.debug(`Retriever start with ID: ${runId}`);
-            this.startAndRegisterOtelSpan({
-                type: mlflow.SpanType.RETRIEVER,
-                runId: runId,
-                parentRunId: parentRunId,
-                runName: (_b = name !== null && name !== void 0 ? name : (_a = retriever.id.at(-1)) === null || _a === void 0 ? void 0 : _a.toString()) !== null && _b !== void 0 ? _b : "Retriever",
-                attributes: {
-                    input: query,
-                },
-                tags: tags,
-                metadata: metadata,
-            });
-        }
-        catch (e) {
-            this.logger.debug(e instanceof Error ? e.message : String(e));
-        }
-    }
-    async handleRetrieverEnd(documents, runId, parentRunId) {
-        try {
-            this.logger.debug(`Retriever end with ID: ${runId}`);
-            this.handleOtelSpanEnd({
-                runId,
-                attributes: {
-                    output: documents,
-                },
-            });
-        }
-        catch (e) {
-            this.logger.debug(e instanceof Error ? e.message : String(e));
-        }
-    }
-    async handleRetrieverError(err, runId, parentRunId) {
-        try {
-            const errorMessage = err instanceof Error ? err.message : String(err);
-            this.logger.debug(`Retriever error: ${errorMessage} with ID: ${runId}`);
-            this.handleOtelSpanEnd({
-                runId,
-                attributes: {
-                    level: "ERROR",
-                    statusMessage: errorMessage,
-                },
-            });
-        }
-        catch (e) {
-            this.logError('handleRetrieverError', e);
-        }
-    }
-    async handleToolEnd(output, runId, parentRunId) {
-        try {
-            this.logger.debug(`Tool end with ID: ${runId}`);
-            this.handleOtelSpanEnd({
-                runId,
-                attributes: { output },
-            });
-        }
-        catch (e) {
-            this.logger.debug(e instanceof Error ? e.message : String(e));
-        }
-    }
-    async handleToolError(err, runId, parentRunId) {
-        try {
-            const errorMessage = err instanceof Error ? err.message : String(err);
-            this.logger.debug(`Tool error ${errorMessage} with ID: ${runId}`);
-            this.handleOtelSpanEnd({
-                runId,
-                attributes: {
-                    level: "ERROR",
-                    statusMessage: errorMessage,
-                },
-            });
-        }
-        catch (e) {
-            this.logError('handleToolError', e);
-        }
-    }
-    async handleLLMEnd(output, runId, parentRunId) {
-        var _a, _b, _c, _d, _e, _f, _g;
-        try {
-            this.logger.debug(`LLM end with ID: ${runId}`);
-            const lastResponse = output.generations[output.generations.length - 1][output.generations[output.generations.length - 1].length - 1];
-            const llmUsage = (_a = this.extractUsageMetadata(lastResponse)) !== null && _a !== void 0 ? _a : (_b = output.llmOutput) === null || _b === void 0 ? void 0 : _b["tokenUsage"];
-            const modelName = this.extractModelNameFromMetadata(lastResponse);
-            const usageDetails = {
-                input_tokens: (_c = llmUsage === null || llmUsage === void 0 ? void 0 : llmUsage.input_tokens) !== null && _c !== void 0 ? _c : ("promptTokens" in llmUsage ? llmUsage === null || llmUsage === void 0 ? void 0 : llmUsage.promptTokens : undefined),
-                output_tokens: (_d = llmUsage === null || llmUsage === void 0 ? void 0 : llmUsage.output_tokens) !== null && _d !== void 0 ? _d : ("completionTokens" in llmUsage
-                    ? llmUsage === null || llmUsage === void 0 ? void 0 : llmUsage.completionTokens
-                    : undefined),
-                total_tokens: (_e = llmUsage === null || llmUsage === void 0 ? void 0 : llmUsage.total_tokens) !== null && _e !== void 0 ? _e : ("totalTokens" in llmUsage ? llmUsage === null || llmUsage === void 0 ? void 0 : llmUsage.totalTokens : undefined),
-            };
-            if (llmUsage && "input_token_details" in llmUsage) {
-                for (const [key, val] of Object.entries((_f = llmUsage["input_token_details"]) !== null && _f !== void 0 ? _f : {})) {
-                    if (typeof val === "number") {
-                        usageDetails[`input_${key}`] = val;
-                        const inputTokens = usageDetails["input_tokens"];
-                        if (typeof inputTokens === "number") {
-                            usageDetails["input_tokens"] = Math.max(0, inputTokens - val);
-                        }
-                    }
-                }
-            }
-            if (llmUsage && "output_token_details" in llmUsage) {
-                for (const [key, val] of Object.entries((_g = llmUsage["output_token_details"]) !== null && _g !== void 0 ? _g : {})) {
-                    if (typeof val === "number") {
-                        usageDetails[`output_${key}`] = val;
-                        const outputTokens = usageDetails["output_tokens"];
-                        if (typeof outputTokens === "number") {
-                            usageDetails["output_tokens"] = Math.max(0, outputTokens - val);
-                        }
-                    }
-                }
-            }
-            const val = lastResponse;
-            const llmResponse = val.message ? this.extractChatMessageContent(val.message) : undefined;
-            this.handleOtelSpanEnd({
-                runId,
-                attributes: {
-                    messages: llmResponse ? [llmResponse] : [],
-                    model: modelName,
-                    usageDetails: usageDetails,
-                },
-            });
-        }
-        catch (e) {
-            this.logError('handleLLMEnd', e);
-        }
-    }
-    async handleLLMError(err, runId, parentRunId) {
-        try {
-            const errorMessage = err instanceof Error ? err.message : String(err);
-            this.logger.debug(`LLM error ${errorMessage} with ID: ${runId}`);
-            const azureRefusalError = this.parseAzureRefusalError(err);
-            this.handleOtelSpanEnd({
-                runId,
-                attributes: {
-                    level: "ERROR",
-                    statusMessage: errorMessage + azureRefusalError,
-                },
-            });
-        }
-        catch (e) {
-            this.logError('handleLLMError', e);
-        }
-    }
-    startAndRegisterOtelSpan(params) {
-        const { type, runName, runId, parentRunId, attributes, metadata, tags } = params;
-        this.ensureMapSize();
-        const parentSpan = parentRunId && this.runMap.has(parentRunId)
-            ? this.runMap.get(parentRunId)
-            : undefined;
-        const joinedMetadata = this.joinTagsAndMetaData(tags, metadata);
-        const inputs = { ...attributes };
-        if (joinedMetadata !== undefined) {
-            inputs.metadata = joinedMetadata;
-        }
-        const span = mlflow.startSpan({
-            name: runName,
-            spanType: type,
-            inputs,
-            parent: parentSpan
-        });
-        if (parentRunId) {
-            span.setAttribute("parentRunID", parentRunId);
-        }
-        this.runMap.set(runId, span);
-        return span;
-    }
-    handleOtelSpanEnd(params) {
-        const { runId, attributes = {} } = params;
-        const span = this.runMap.get(runId);
-        if (!span) {
-            this.logger.warn("Span not found in runMap. Skipping operation");
-            return;
-        }
-        span.setOutputs({ ...attributes });
-        span.setStatus(mlflow.SpanStatusCode.OK);
-        span.end();
-        this._lastTraceId = span.traceId;
-        this.runMap.delete(runId);
-    }
-    parseAzureRefusalError(err) {
-        let azureRefusalError = "";
-        if (typeof err == "object" && "error" in err) {
-            try {
-                azureRefusalError =
-                    "\n\nError details:\n" + JSON.stringify(err["error"], null, 2);
-            }
-            catch (serializationError) {
-                this.logger.debug(`Failed to serialize Azure error: ${serializationError}`);
-                azureRefusalError = "\n\nError details: [Unable to serialize error object]";
-            }
-        }
-        return azureRefusalError;
-    }
-    joinTagsAndMetaData(tags, metadata1, metadata2) {
-        const finalDict = {};
-        if (tags && tags.length > 0) {
-            finalDict.tags = tags;
-        }
-        if (metadata1) {
-            Object.assign(finalDict, metadata1);
-        }
-        if (metadata2) {
-            Object.assign(finalDict, metadata2);
-        }
-        return Object.keys(finalDict).length > 0 ? finalDict : undefined;
-    }
-    extractUsageMetadata(generation) {
-        try {
-            const usageMetadata = "message" in generation &&
-                (generation["message"] instanceof messages_1.AIMessage ||
-                    generation["message"] instanceof messages_1.AIMessageChunk)
-                ? generation["message"].usage_metadata
-                : undefined;
-            return usageMetadata;
-        }
-        catch (err) {
-            this.logger.debug(`Error extracting usage metadata: ${err}`);
-            return;
-        }
-    }
-    extractModelNameFromMetadata(generation) {
-        var _a, _b;
-        try {
-            return (_b = (_a = generation.message) === null || _a === void 0 ? void 0 : _a.response_metadata) === null || _b === void 0 ? void 0 : _b.model_name;
-        }
-        catch (error) {
-            this.logError('extractModelNameFromMetadata', error);
-            return undefined;
-        }
-    }
-    extractChatMessageContent(message) {
-        let response = undefined;
-        if (message.getType() == "human") {
-            response = { content: message.content, role: "user" };
-        }
-        else if (message.getType() == "ai") {
-            response = { content: message.content, role: "assistant" };
-        }
-        else if (message.getType() == "system") {
-            response = { content: message.content, role: "system" };
-        }
-        else if (message.getType() == "function") {
-            response = {
-                content: message.content,
-                additional_kwargs: message.additional_kwargs,
-                role: message.name,
-            };
-        }
-        else if (message.getType() == "tool") {
-            response = {
-                content: message.content,
-                additional_kwargs: message.additional_kwargs,
-                role: "tool",
-            };
-        }
-        else if (!message.name) {
-            response = { content: message.content, role: "user" };
-        }
-        else {
-            response = {
-                role: message.name,
-                content: message.content,
-            };
-        }
-        if (message.additional_kwargs.function_call ||
-            message.additional_kwargs.tool_calls) {
-            return { ...response, tool_calls: message.additional_kwargs.tool_calls };
-        }
-        return response;
-    }
-}
-exports.CallbackHandler = CallbackHandler;
-//# sourceMappingURL=CallbackHandler.js.map