Extract project name from mastra componentName attr

Author: cephalization

js/packages/openinference-mastra/src/OpenInferenceTraceExporter.ts

@@ -2,6 +2,7 @@ import type { ReadableSpan } from "@opentelemetry/sdk-trace-base";
 import type { ExportResult } from "@opentelemetry/core";
 import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-proto";
 import { addOpenInferenceAttributesToSpan } from "@arizeai/openinference-vercel/utils";
+import { addOpenInferenceAttributesToMastraSpan } from "./utils.js";
 
 type ConstructorArgs = {
   /**
@@ -63,7 +64,10 @@ export class OpenInferenceOTLPTraceExporter extends OTLPTraceExporter {
     resultCallback: (result: ExportResult) => void,
   ) {
     let filteredItems = items.map((i) => {
+      // add OpenInference attributes to the span based on Vercel span attributes
       addOpenInferenceAttributesToSpan(i);
+      // add OpenInference attributes to the span based on Mastra span attributes
+      addOpenInferenceAttributesToMastraSpan(i);
       return i;
     });
     if (this.spanFilter) {

js/packages/openinference-mastra/src/utils.ts

@@ -0,0 +1,16 @@
+import { SEMRESATTRS_PROJECT_NAME } from "@arizeai/openinference-semantic-conventions";
+import { ReadableSpan } from "@opentelemetry/sdk-trace-base";
+
+/**
+ * Augments a Mastra span with OpenInference attributes.
+ *
+ * This function will add additional attributes to the span, based on the Mastra span's attributes.
+ *
+ * @param span - The Mastra span to augment.
+ */
+export const addOpenInferenceAttributesToMastraSpan = (span: ReadableSpan) => {
+  const attributes = span.attributes;
+  if (attributes["componentName"]) {
+    span.attributes[SEMRESATTRS_PROJECT_NAME] = attributes["componentName"];
+  }
+};