Replace jest with vitest in mastra package

Author: cephalization

js/packages/openinference-mastra/README.md

@@ -18,6 +18,12 @@ A typical Mastra project will already have OpenTelemetry and related packages in
 
 To process your Mastra spans add an `OpenInferenceOTLPTraceExporter` to your `telemetry` configuration within your `Mastra` instance.
 
+```shell
+# Set the Phoenix collector endpoint and API key in your environment
+export PHOENIX_COLLECTOR_ENDPOINT="https://localhost:6006/v1/traces"
+export PHOENIX_API_KEY="your-api-key"
+```
+
 ```typescript
 import { Mastra } from "@mastra/core";
 import { OpenInferenceOTLPTraceExporter } from "@arizeai/openinference-mastra";
@@ -42,4 +48,4 @@ For general details on Mastra's OpenTelemetry support see the [Mastra Observabil
 
 ## Examples
 
-TODO
+TODO: Add examples

js/packages/openinference-mastra/jest.config.js

@@ -10,7 +10,7 @@
     "build": "tsc --build tsconfig.esm.json && tsc-alias -p tsconfig.esm.json",
     "postbuild": "echo '{\"type\": \"module\"}' > ./dist/esm/package.json && rimraf dist/test",
     "type:check": "tsc --noEmit",
-    "test": "jest"
+    "test": "vitest"
   },
   "exports": {
     ".": {
@@ -43,12 +43,12 @@
     "@arizeai/openinference-semantic-conventions": "workspace:*",
     "@arizeai/openinference-vercel": "workspace:*",
     "@opentelemetry/core": "^1.25.1",
-    "@opentelemetry/exporter-trace-otlp-proto": "^0.50.0"
+    "@opentelemetry/exporter-trace-otlp-proto": "^0.50.0",
+    "@opentelemetry/semantic-conventions": "^1.33.0"
   },
   "devDependencies": {
-    "@types/jest": "^29.5.12",
-    "jest": "^29.7.0",
+    "@opentelemetry/api": ">=1.0.0 <1.9.0",
     "@opentelemetry/sdk-trace-base": "^1.19.0",
-    "@opentelemetry/api": ">=1.0.0 <1.9.0"
+    "vitest": "^3.1.3"
   }
 }

js/packages/openinference-mastra/package.json

@@ -2,7 +2,10 @@ 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";
+import {
+  addOpenInferenceAttributesToMastraSpan,
+  addOpenInferenceResourceAttributesToMastraSpan,
+} from "./utils.js";
 
 type ConstructorArgs = {
   /**
@@ -42,6 +45,7 @@ type ConstructorArgs = {
 
 export class OpenInferenceOTLPTraceExporter extends OTLPTraceExporter {
   private readonly spanFilter?: (span: ReadableSpan) => boolean;
+
   constructor({
     apiKey,
     collectorEndpoint,
@@ -64,6 +68,8 @@ export class OpenInferenceOTLPTraceExporter extends OTLPTraceExporter {
     resultCallback: (result: ExportResult) => void,
   ) {
     let filteredItems = items.map((i) => {
+      // add OpenInference resource attributes to the span based on Mastra span attributes
+      addOpenInferenceResourceAttributesToMastraSpan(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

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

@@ -1,5 +1,17 @@
 import { SEMRESATTRS_PROJECT_NAME } from "@arizeai/openinference-semantic-conventions";
-import { ReadableSpan } from "@opentelemetry/sdk-trace-base";
+import type { ReadableSpan } from "@opentelemetry/sdk-trace-base";
+import { ATTR_SERVICE_NAME } from "@opentelemetry/semantic-conventions";
+
+export const addOpenInferenceResourceAttributesToMastraSpan = (
+  span: ReadableSpan,
+) => {
+  const attributes = span.resource.attributes;
+  if (ATTR_SERVICE_NAME in attributes) {
+    attributes[SEMRESATTRS_PROJECT_NAME] = attributes[ATTR_SERVICE_NAME];
+  }
+  // eslint-disable-next-line no-console
+  console.log("attributes", attributes);
+};
 
 /**
  * Augments a Mastra span with OpenInference attributes.
@@ -8,9 +20,7 @@ import { ReadableSpan } from "@opentelemetry/sdk-trace-base";
  *
  * @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"];
-  }
-};
+export const addOpenInferenceAttributesToMastraSpan = (
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  span: ReadableSpan,
+) => {};