@mastra/datadog 1.2.3 → 1.2.4-alpha.0

package/dist/index.js

@@ -124,8 +124,8 @@ function toDatadogMessages(messages) {
       role: m.role,
       content: m.content == null ? "" : toMessageContent(m.content)
     };
-    if (m.toolCalls) {
-      message.toolCalls = m.toolCalls;
+    if (Array.isArray(m.toolCalls) && m.toolCalls.length > 0) {
+      message.toolCalls = toDatadogToolCalls(m.toolCalls);
     }
     return message;
   }).filter((m) => !(m.role === "user" && m.content.trim().length === 0));
@@ -148,7 +148,7 @@ function parseToolArguments(args) {
 function toDatadogToolCalls(toolCalls) {
   return toolCalls.map((c) => ({
     name: c?.toolName ?? c?.name ?? c?.function?.name ?? "unknown",
-    arguments: parseToolArguments(c?.args ?? c?.input ?? c?.function?.arguments),
+    arguments: parseToolArguments(c?.args ?? c?.input ?? c?.arguments ?? c?.function?.arguments),
     toolId: c?.toolCallId ?? c?.toolId ?? c?.id,
     type: c?.type === "function" ? c.type : "function"
   }));
@@ -271,7 +271,14 @@ var DatadogBridge = class extends BaseExporter {
       this.config = config;
       return;
     }
-    this.config = { ...config, mlApp, site, apiKey, agentless, env };
+    this.config = {
+      ...config,
+      mlApp,
+      site,
+      apiKey,
+      agentless,
+      env
+    };
     ensureTracer({
       mlApp,
       site,
@@ -343,6 +350,37 @@ var DatadogBridge = class extends BaseExporter {
   executeInContextSync(spanId, fn) {
     return this.executeWithSpanContext(spanId, fn);
   }
+  async onScoreEvent(event) {
+    if (this.isDisabled || !tracer3.llmobs?.submitEvaluation) return;
+    const { score } = event;
+    if (!score.traceId || !score.spanId) {
+      this.logger.warn("Datadog bridge: dropping score with no traceId/spanId", {
+        scorerId: score.scorerId
+      });
+      return;
+    }
+    try {
+      tracer3.llmobs.submitEvaluation(
+        { traceId: score.traceId, spanId: toDatadogSpanId(score.spanId) },
+        {
+          label: score.scorerName ?? score.scorerId,
+          value: score.score,
+          metricType: "score",
+          mlApp: this.config.mlApp,
+          timestampMs: score.timestamp instanceof Date ? score.timestamp.getTime() : Date.now(),
+          ...score.reason ? { reasoning: score.reason } : {},
+          ...score.metadata ? { metadata: score.metadata } : {}
+        }
+      );
+    } catch (error) {
+      this.logger.error("Datadog bridge: Failed to submit evaluation", {
+        error,
+        traceId: score.traceId,
+        spanId: score.spanId,
+        scorerId: score.scorerId
+      });
+    }
+  }
   executeWithSpanContext(spanId, fn) {
     const ddSpan = this.ddSpanMap.get(spanId);
     if (ddSpan) {
@@ -514,7 +552,7 @@ var DatadogBridge = class extends BaseExporter {
         annotations.metrics = metrics;
       }
     }
-    const knownFields = ["usage", "model", "provider", "parameters"];
+    const knownFields = ["usage", "model", "provider"];
     const otherAttributes = omitKeys(span.attributes ?? {}, knownFields);
     const contextKeySet = new Set(this.config.requestContextKeys ?? []);
     const flatContextTags = {};
@@ -646,6 +684,12 @@ function fillRandomBytes(bytes) {
     bytes[i] = Math.floor(Math.random() * 256);
   }
 }
+function toDatadogSpanId(spanId) {
+  if (/^[0-9a-f]{16}$/i.test(spanId)) {
+    return BigInt(`0x${spanId}`).toString(10);
+  }
+  return spanId;
+}
 function generateSpanId() {
   const bytes = new Uint8Array(8);
   fillRandomBytes(bytes);
@@ -781,7 +825,7 @@ var DatadogExporter = class extends BaseExporter {
         annotations.metrics = metrics;
       }
     }
-    const knownFields = ["usage", "model", "provider", "parameters"];
+    const knownFields = ["usage", "model", "provider"];
     const otherAttributes = omitKeys(span.attributes ?? {}, knownFields);
     const contextKeySet = new Set(this.config.requestContextKeys ?? []);
     const flatContextTags = {};