@mastra/otel-exporter 0.0.0-unified-sidebar-20251010130811 → 0.0.0-vnext-20251104230439

package/dist/index.cjs

@@ -1,7 +1,7 @@
 'use strict';
 
-var aiTracing = require('@mastra/core/ai-tracing');
-var logger = require('@mastra/core/logger');
+var observability$1 = require('@mastra/core/observability');
+var observability = require('@mastra/observability');
 var api = require('@opentelemetry/api');
 var resources = require('@opentelemetry/resources');
 var sdkTraceBase = require('@opentelemetry/sdk-trace-base');
@@ -209,6 +209,7 @@ var MastraReadableSpan = class {
   name;
   kind;
   spanContext;
+  parentSpanContext;
   parentSpanId;
   startTime;
   endTime;
@@ -276,6 +277,14 @@ var MastraReadableSpan = class {
       traceFlags: api.TraceFlags.SAMPLED,
       isRemote: false
     });
+    if (parentSpanId) {
+      this.parentSpanContext = {
+        traceId: aiSpan.traceId,
+        spanId: parentSpanId,
+        traceFlags: api.TraceFlags.SAMPLED,
+        isRemote: false
+      };
+    }
     this.resource = resource || {};
     this.instrumentationLibrary = instrumentationLibrary || {
       name: "@mastra/otel",
@@ -296,24 +305,14 @@ var MastraReadableSpan = class {
 
 // src/span-converter.ts
 var SPAN_KIND_MAPPING = {
-  // LLM operations are CLIENT spans (calling external AI services)
-  [aiTracing.AISpanType.LLM_GENERATION]: api.SpanKind.CLIENT,
-  [aiTracing.AISpanType.LLM_CHUNK]: api.SpanKind.CLIENT,
-  // Tool calls can be CLIENT (external) or INTERNAL based on context
-  [aiTracing.AISpanType.TOOL_CALL]: api.SpanKind.INTERNAL,
-  [aiTracing.AISpanType.MCP_TOOL_CALL]: api.SpanKind.CLIENT,
+  // Model operations are CLIENT spans (calling external AI services)
+  [observability$1.AISpanType.MODEL_GENERATION]: api.SpanKind.CLIENT,
+  [observability$1.AISpanType.MODEL_CHUNK]: api.SpanKind.CLIENT,
+  // MCP tool calls are CLIENT (external service calls)
+  [observability$1.AISpanType.MCP_TOOL_CALL]: api.SpanKind.CLIENT,
   // Root spans for agent/workflow are SERVER (entry points)
-  [aiTracing.AISpanType.AGENT_RUN]: api.SpanKind.SERVER,
-  [aiTracing.AISpanType.WORKFLOW_RUN]: api.SpanKind.SERVER,
-  // Internal workflow operations
-  [aiTracing.AISpanType.WORKFLOW_STEP]: api.SpanKind.INTERNAL,
-  [aiTracing.AISpanType.WORKFLOW_LOOP]: api.SpanKind.INTERNAL,
-  [aiTracing.AISpanType.WORKFLOW_PARALLEL]: api.SpanKind.INTERNAL,
-  [aiTracing.AISpanType.WORKFLOW_CONDITIONAL]: api.SpanKind.INTERNAL,
-  [aiTracing.AISpanType.WORKFLOW_CONDITIONAL_EVAL]: api.SpanKind.INTERNAL,
-  [aiTracing.AISpanType.WORKFLOW_SLEEP]: api.SpanKind.INTERNAL,
-  [aiTracing.AISpanType.WORKFLOW_WAIT_EVENT]: api.SpanKind.INTERNAL,
-  [aiTracing.AISpanType.GENERIC]: api.SpanKind.INTERNAL
+  [observability$1.AISpanType.AGENT_RUN]: api.SpanKind.SERVER,
+  [observability$1.AISpanType.WORKFLOW_RUN]: api.SpanKind.SERVER
 };
 var SpanConverter = class {
   resource;
@@ -349,7 +348,7 @@ var SpanConverter = class {
    */
   getSpanKind(aiSpan) {
     if (aiSpan.isRootSpan) {
-      if (aiSpan.type === aiTracing.AISpanType.AGENT_RUN || aiSpan.type === aiTracing.AISpanType.WORKFLOW_RUN) {
+      if (aiSpan.type === observability$1.AISpanType.AGENT_RUN || aiSpan.type === observability$1.AISpanType.WORKFLOW_RUN) {
         return api.SpanKind.SERVER;
       }
     }
@@ -360,29 +359,29 @@ var SpanConverter = class {
    */
   buildSpanName(aiSpan) {
     switch (aiSpan.type) {
-      case aiTracing.AISpanType.LLM_GENERATION: {
+      case observability$1.AISpanType.MODEL_GENERATION: {
         const attrs = aiSpan.attributes;
         const operation = attrs?.resultType === "tool_selection" ? "tool_selection" : "chat";
         const model = attrs?.model || "unknown";
         return `${operation} ${model}`;
       }
-      case aiTracing.AISpanType.TOOL_CALL:
-      case aiTracing.AISpanType.MCP_TOOL_CALL: {
+      case observability$1.AISpanType.TOOL_CALL:
+      case observability$1.AISpanType.MCP_TOOL_CALL: {
         const toolAttrs = aiSpan.attributes;
         const toolName = toolAttrs?.toolId || "unknown";
         return `tool.execute ${toolName}`;
       }
-      case aiTracing.AISpanType.AGENT_RUN: {
+      case observability$1.AISpanType.AGENT_RUN: {
         const agentAttrs = aiSpan.attributes;
         const agentId = agentAttrs?.agentId || "unknown";
         return `agent.${agentId}`;
       }
-      case aiTracing.AISpanType.WORKFLOW_RUN: {
+      case observability$1.AISpanType.WORKFLOW_RUN: {
         const workflowAttrs = aiSpan.attributes;
         const workflowId = workflowAttrs?.workflowId || "unknown";
         return `workflow.${workflowId}`;
       }
-      case aiTracing.AISpanType.WORKFLOW_STEP:
+      case observability$1.AISpanType.WORKFLOW_STEP:
         return aiSpan.name;
       default:
         return aiSpan.name;
@@ -405,81 +404,81 @@ var SpanConverter = class {
     if (aiSpan.input !== void 0) {
       const inputStr = typeof aiSpan.input === "string" ? aiSpan.input : JSON.stringify(aiSpan.input);
       attributes["input"] = inputStr;
-      if (aiSpan.type === aiTracing.AISpanType.LLM_GENERATION) {
+      if (aiSpan.type === observability$1.AISpanType.MODEL_GENERATION) {
         attributes["gen_ai.prompt"] = inputStr;
-      } else if (aiSpan.type === aiTracing.AISpanType.TOOL_CALL || aiSpan.type === aiTracing.AISpanType.MCP_TOOL_CALL) {
+      } else if (aiSpan.type === observability$1.AISpanType.TOOL_CALL || aiSpan.type === observability$1.AISpanType.MCP_TOOL_CALL) {
         attributes["gen_ai.tool.input"] = inputStr;
       }
     }
     if (aiSpan.output !== void 0) {
       const outputStr = typeof aiSpan.output === "string" ? aiSpan.output : JSON.stringify(aiSpan.output);
       attributes["output"] = outputStr;
-      if (aiSpan.type === aiTracing.AISpanType.LLM_GENERATION) {
+      if (aiSpan.type === observability$1.AISpanType.MODEL_GENERATION) {
         attributes["gen_ai.completion"] = outputStr;
-      } else if (aiSpan.type === aiTracing.AISpanType.TOOL_CALL || aiSpan.type === aiTracing.AISpanType.MCP_TOOL_CALL) {
+      } else if (aiSpan.type === observability$1.AISpanType.TOOL_CALL || aiSpan.type === observability$1.AISpanType.MCP_TOOL_CALL) {
         attributes["gen_ai.tool.output"] = outputStr;
       }
     }
-    if (aiSpan.type === aiTracing.AISpanType.LLM_GENERATION && aiSpan.attributes) {
-      const llmAttrs = aiSpan.attributes;
-      if (llmAttrs.model) {
-        attributes["gen_ai.request.model"] = llmAttrs.model;
+    if (aiSpan.type === observability$1.AISpanType.MODEL_GENERATION && aiSpan.attributes) {
+      const modelAttrs = aiSpan.attributes;
+      if (modelAttrs.model) {
+        attributes["gen_ai.request.model"] = modelAttrs.model;
       }
-      if (llmAttrs.provider) {
-        attributes["gen_ai.system"] = llmAttrs.provider;
+      if (modelAttrs.provider) {
+        attributes["gen_ai.system"] = modelAttrs.provider;
       }
-      if (llmAttrs.usage) {
-        const inputTokens = llmAttrs.usage.inputTokens ?? llmAttrs.usage.promptTokens;
-        const outputTokens = llmAttrs.usage.outputTokens ?? llmAttrs.usage.completionTokens;
+      if (modelAttrs.usage) {
+        const inputTokens = modelAttrs.usage.inputTokens ?? modelAttrs.usage.promptTokens;
+        const outputTokens = modelAttrs.usage.outputTokens ?? modelAttrs.usage.completionTokens;
         if (inputTokens !== void 0) {
           attributes["gen_ai.usage.input_tokens"] = inputTokens;
         }
         if (outputTokens !== void 0) {
           attributes["gen_ai.usage.output_tokens"] = outputTokens;
         }
-        if (llmAttrs.usage.totalTokens !== void 0) {
-          attributes["gen_ai.usage.total_tokens"] = llmAttrs.usage.totalTokens;
+        if (modelAttrs.usage.totalTokens !== void 0) {
+          attributes["gen_ai.usage.total_tokens"] = modelAttrs.usage.totalTokens;
         }
-        if (llmAttrs.usage.reasoningTokens !== void 0) {
-          attributes["gen_ai.usage.reasoning_tokens"] = llmAttrs.usage.reasoningTokens;
+        if (modelAttrs.usage.reasoningTokens !== void 0) {
+          attributes["gen_ai.usage.reasoning_tokens"] = modelAttrs.usage.reasoningTokens;
         }
-        if (llmAttrs.usage.cachedInputTokens !== void 0) {
-          attributes["gen_ai.usage.cached_input_tokens"] = llmAttrs.usage.cachedInputTokens;
+        if (modelAttrs.usage.cachedInputTokens !== void 0) {
+          attributes["gen_ai.usage.cached_input_tokens"] = modelAttrs.usage.cachedInputTokens;
         }
       }
-      if (llmAttrs.parameters) {
-        if (llmAttrs.parameters.temperature !== void 0) {
-          attributes["gen_ai.request.temperature"] = llmAttrs.parameters.temperature;
+      if (modelAttrs.parameters) {
+        if (modelAttrs.parameters.temperature !== void 0) {
+          attributes["gen_ai.request.temperature"] = modelAttrs.parameters.temperature;
         }
-        if (llmAttrs.parameters.maxOutputTokens !== void 0) {
-          attributes["gen_ai.request.max_tokens"] = llmAttrs.parameters.maxOutputTokens;
+        if (modelAttrs.parameters.maxOutputTokens !== void 0) {
+          attributes["gen_ai.request.max_tokens"] = modelAttrs.parameters.maxOutputTokens;
         }
-        if (llmAttrs.parameters.topP !== void 0) {
-          attributes["gen_ai.request.top_p"] = llmAttrs.parameters.topP;
+        if (modelAttrs.parameters.topP !== void 0) {
+          attributes["gen_ai.request.top_p"] = modelAttrs.parameters.topP;
         }
-        if (llmAttrs.parameters.topK !== void 0) {
-          attributes["gen_ai.request.top_k"] = llmAttrs.parameters.topK;
+        if (modelAttrs.parameters.topK !== void 0) {
+          attributes["gen_ai.request.top_k"] = modelAttrs.parameters.topK;
         }
-        if (llmAttrs.parameters.presencePenalty !== void 0) {
-          attributes["gen_ai.request.presence_penalty"] = llmAttrs.parameters.presencePenalty;
+        if (modelAttrs.parameters.presencePenalty !== void 0) {
+          attributes["gen_ai.request.presence_penalty"] = modelAttrs.parameters.presencePenalty;
         }
-        if (llmAttrs.parameters.frequencyPenalty !== void 0) {
-          attributes["gen_ai.request.frequency_penalty"] = llmAttrs.parameters.frequencyPenalty;
+        if (modelAttrs.parameters.frequencyPenalty !== void 0) {
+          attributes["gen_ai.request.frequency_penalty"] = modelAttrs.parameters.frequencyPenalty;
         }
-        if (llmAttrs.parameters.stopSequences) {
-          attributes["gen_ai.request.stop_sequences"] = JSON.stringify(llmAttrs.parameters.stopSequences);
+        if (modelAttrs.parameters.stopSequences) {
+          attributes["gen_ai.request.stop_sequences"] = JSON.stringify(modelAttrs.parameters.stopSequences);
         }
       }
-      if (llmAttrs.finishReason) {
-        attributes["gen_ai.response.finish_reasons"] = llmAttrs.finishReason;
+      if (modelAttrs.finishReason) {
+        attributes["gen_ai.response.finish_reasons"] = modelAttrs.finishReason;
       }
     }
-    if ((aiSpan.type === aiTracing.AISpanType.TOOL_CALL || aiSpan.type === aiTracing.AISpanType.MCP_TOOL_CALL) && aiSpan.attributes) {
+    if ((aiSpan.type === observability$1.AISpanType.TOOL_CALL || aiSpan.type === observability$1.AISpanType.MCP_TOOL_CALL) && aiSpan.attributes) {
       const toolAttrs = aiSpan.attributes;
       if (toolAttrs.toolId) {
         attributes["gen_ai.tool.name"] = toolAttrs.toolId;
       }
-      if (aiSpan.type === aiTracing.AISpanType.MCP_TOOL_CALL) {
+      if (aiSpan.type === observability$1.AISpanType.MCP_TOOL_CALL) {
         const mcpAttrs = toolAttrs;
         if (mcpAttrs.mcpServer) {
           attributes["mcp.server"] = mcpAttrs.mcpServer;
@@ -496,10 +495,11 @@ var SpanConverter = class {
         attributes["gen_ai.tool.success"] = toolAttrs.success;
       }
     }
-    if (aiSpan.type === aiTracing.AISpanType.AGENT_RUN && aiSpan.attributes) {
+    if (aiSpan.type === observability$1.AISpanType.AGENT_RUN && aiSpan.attributes) {
       const agentAttrs = aiSpan.attributes;
       if (agentAttrs.agentId) {
         attributes["agent.id"] = agentAttrs.agentId;
+        attributes["gen_ai.agent.id"] = agentAttrs.agentId;
       }
       if (agentAttrs.maxSteps) {
         attributes["agent.max_steps"] = agentAttrs.maxSteps;
@@ -508,7 +508,7 @@ var SpanConverter = class {
         attributes["agent.available_tools"] = JSON.stringify(agentAttrs.availableTools);
       }
     }
-    if (aiSpan.type === aiTracing.AISpanType.WORKFLOW_RUN && aiSpan.attributes) {
+    if (aiSpan.type === observability$1.AISpanType.WORKFLOW_RUN && aiSpan.attributes) {
       const workflowAttrs = aiSpan.attributes;
       if (workflowAttrs.workflowId) {
         attributes["workflow.id"] = workflowAttrs.workflowId;
@@ -557,16 +557,16 @@ var SpanConverter = class {
    */
   getOperationName(aiSpan) {
     switch (aiSpan.type) {
-      case aiTracing.AISpanType.LLM_GENERATION: {
+      case observability$1.AISpanType.MODEL_GENERATION: {
         const attrs = aiSpan.attributes;
         return attrs?.resultType === "tool_selection" ? "tool_selection" : "chat";
       }
-      case aiTracing.AISpanType.TOOL_CALL:
-      case aiTracing.AISpanType.MCP_TOOL_CALL:
+      case observability$1.AISpanType.TOOL_CALL:
+      case observability$1.AISpanType.MCP_TOOL_CALL:
         return "tool.execute";
-      case aiTracing.AISpanType.AGENT_RUN:
+      case observability$1.AISpanType.AGENT_RUN:
         return "agent.run";
-      case aiTracing.AISpanType.WORKFLOW_RUN:
+      case observability$1.AISpanType.WORKFLOW_RUN:
         return "workflow.run";
       default:
         return aiSpan.type.replace(/_/g, ".");
@@ -595,20 +595,18 @@ var SpanConverter = class {
 };
 
 // src/ai-tracing.ts
-var OtelExporter = class {
+var OtelExporter = class extends observability.BaseExporter {
   config;
   tracingConfig;
   spanConverter;
   processor;
   exporter;
   isSetup = false;
-  isDisabled = false;
-  logger;
   name = "opentelemetry";
   constructor(config) {
+    super(config);
     this.config = config;
     this.spanConverter = new SpanConverter();
-    this.logger = new logger.ConsoleLogger({ level: config.logLevel ?? "warn" });
     if (config.logLevel === "debug") {
       api.diag.setLogger(new api.DiagConsoleLogger(), api.DiagLogLevel.DEBUG);
     }
@@ -616,11 +614,11 @@ var OtelExporter = class {
   /**
    * Initialize with tracing configuration
    */
-  init(config) {
-    this.tracingConfig = config;
+  init(options) {
+    this.tracingConfig = options.config;
   }
   async setupExporter() {
-    if (this.isSetup) return;
+    if (this.isSetup || this.exporter) return;
     if (!this.config.provider) {
       this.logger.error(
         '[OtelExporter] Provider configuration is required. Use the "custom" provider for generic endpoints.'
@@ -635,6 +633,10 @@ var OtelExporter = class {
       this.isSetup = true;
       return;
     }
+    if (this.config.exporter) {
+      this.exporter = this.config.exporter;
+      return;
+    }
     const endpoint = resolved.endpoint;
     const headers = resolved.headers;
     const protocol = resolved.protocol;
@@ -688,7 +690,10 @@ var OtelExporter = class {
       this.isSetup = true;
       return;
     }
-    const resource = resources.resourceFromAttributes({
+  }
+  async setupProcessor() {
+    if (this.processor || this.isSetup) return;
+    let resource = resources.resourceFromAttributes({
       [semanticConventions.ATTR_SERVICE_NAME]: this.tracingConfig?.serviceName || "mastra-service",
       [semanticConventions.ATTR_SERVICE_VERSION]: "1.0.0",
       // Add telemetry SDK information
@@ -696,6 +701,12 @@ var OtelExporter = class {
       [semanticConventions.ATTR_TELEMETRY_SDK_VERSION]: "1.0.0",
       [semanticConventions.ATTR_TELEMETRY_SDK_LANGUAGE]: "nodejs"
     });
+    if (this.config.resourceAttributes) {
+      resource = resource.merge(
+        // Duplicate attributes from config will override defaults above
+        resources.resourceFromAttributes(this.config.resourceAttributes)
+      );
+    }
     this.spanConverter = new SpanConverter(resource);
     this.processor = new sdkTraceBase.BatchSpanProcessor(this.exporter, {
       maxExportBatchSize: this.config.batchSize || 512,
@@ -710,13 +721,15 @@ var OtelExporter = class {
     this.logger.debug(
       `[OtelExporter] Using BatchSpanProcessor (batch size: ${this.config.batchSize || 512}, delay: 5s)`
     );
+  }
+  async setup() {
+    if (this.isSetup) return;
+    await this.setupExporter();
+    await this.setupProcessor();
     this.isSetup = true;
   }
-  async exportEvent(event) {
-    if (this.isDisabled) {
-      return;
-    }
-    if (event.type !== aiTracing.AITracingEventType.SPAN_ENDED) {
+  async _exportEvent(event) {
+    if (event.type !== observability$1.AITracingEventType.SPAN_ENDED) {
       return;
     }
     const span = event.exportedSpan;
@@ -724,7 +737,7 @@ var OtelExporter = class {
   }
   async exportSpan(span) {
     if (!this.isSetup) {
-      await this.setupExporter();
+      await this.setup();
     }
     if (this.isDisabled || !this.processor) {
       return;