@mastra/arize 1.0.0-beta.4 → 1.0.0-beta.5

package/CHANGELOG.md

@@ -1,5 +1,15 @@
 # @mastra/arize
 
+## 1.0.0-beta.5
+
+### Patch Changes
+
+- Fixed CachedToken tracking in all Observability Exporters. Also fixed TimeToFirstToken in Langfuse, Braintrust, PostHog exporters. Fixed trace formatting in Posthog Exporter. ([#11029](https://github.com/mastra-ai/mastra/pull/11029))
+
+- Updated dependencies [[`edb07e4`](https://github.com/mastra-ai/mastra/commit/edb07e49283e0c28bd094a60e03439bf6ecf0221), [`b7e17d3`](https://github.com/mastra-ai/mastra/commit/b7e17d3f5390bb5a71efc112204413656fcdc18d), [`261473a`](https://github.com/mastra-ai/mastra/commit/261473ac637e633064a22076671e2e02b002214d), [`5d7000f`](https://github.com/mastra-ai/mastra/commit/5d7000f757cd65ea9dc5b05e662fd83dfd44e932), [`4f0331a`](https://github.com/mastra-ai/mastra/commit/4f0331a79bf6eb5ee598a5086e55de4b5a0ada03), [`8a000da`](https://github.com/mastra-ai/mastra/commit/8a000da0c09c679a2312f6b3aa05b2ca78ca7393)]:
+  - @mastra/core@1.0.0-beta.10
+  - @mastra/otel-exporter@1.0.0-beta.5
+
 ## 1.0.0-beta.4
 
 ### Patch Changes

package/dist/index.cjs

@@ -8,8 +8,48 @@ var exporterTraceOtlpProto = require('@opentelemetry/exporter-trace-otlp-proto')
 var incubating = require('@opentelemetry/semantic-conventions/incubating');
 
 // src/tracing.ts
+var GEN_AI_USAGE_REASONING_TOKENS = "gen_ai.usage.reasoning_tokens";
+var GEN_AI_USAGE_CACHED_INPUT_TOKENS = "gen_ai.usage.cached_input_tokens";
+var GEN_AI_USAGE_CACHE_WRITE_TOKENS = "gen_ai.usage.cache_write_tokens";
+var GEN_AI_USAGE_AUDIO_INPUT_TOKENS = "gen_ai.usage.audio_input_tokens";
+var GEN_AI_USAGE_AUDIO_OUTPUT_TOKENS = "gen_ai.usage.audio_output_tokens";
 var MASTRA_GENERAL_PREFIX = "mastra.";
 var MASTRA_METADATA_PREFIX = "mastra.metadata.";
+function convertUsageMetricsToOpenInference(attributes) {
+  const result = {};
+  const inputTokens = attributes[incubating.ATTR_GEN_AI_USAGE_INPUT_TOKENS];
+  const outputTokens = attributes[incubating.ATTR_GEN_AI_USAGE_OUTPUT_TOKENS];
+  if (inputTokens !== void 0) {
+    result[openinferenceSemanticConventions.LLM_TOKEN_COUNT_PROMPT] = inputTokens;
+  }
+  if (outputTokens !== void 0) {
+    result[openinferenceSemanticConventions.LLM_TOKEN_COUNT_COMPLETION] = outputTokens;
+  }
+  if (inputTokens !== void 0 && outputTokens !== void 0) {
+    result[openinferenceSemanticConventions.LLM_TOKEN_COUNT_TOTAL] = inputTokens + outputTokens;
+  }
+  const cachedInputTokens = attributes[GEN_AI_USAGE_CACHED_INPUT_TOKENS];
+  if (cachedInputTokens !== void 0) {
+    result[openinferenceSemanticConventions.LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ] = cachedInputTokens;
+  }
+  const cacheWriteTokens = attributes[GEN_AI_USAGE_CACHE_WRITE_TOKENS];
+  if (cacheWriteTokens !== void 0) {
+    result[openinferenceSemanticConventions.LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_WRITE] = cacheWriteTokens;
+  }
+  const reasoningTokens = attributes[GEN_AI_USAGE_REASONING_TOKENS];
+  if (reasoningTokens !== void 0) {
+    result[openinferenceSemanticConventions.LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING] = reasoningTokens;
+  }
+  const audioInputTokens = attributes[GEN_AI_USAGE_AUDIO_INPUT_TOKENS];
+  if (audioInputTokens !== void 0) {
+    result[openinferenceSemanticConventions.LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO] = audioInputTokens;
+  }
+  const audioOutputTokens = attributes[GEN_AI_USAGE_AUDIO_OUTPUT_TOKENS];
+  if (audioOutputTokens !== void 0) {
+    result[openinferenceSemanticConventions.LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO] = audioOutputTokens;
+  }
+  return result;
+}
 function splitMastraAttributes(attributes) {
   return Object.entries(attributes).reduce(
     (acc, [key, value]) => {
@@ -67,6 +107,8 @@ var OpenInferenceOTLPTraceExporter = class extends exporterTraceOtlpProto.OTLPTr
           processedAttributes[openinferenceSemanticConventions.OUTPUT_MIME_TYPE] = "application/json";
           processedAttributes[openinferenceSemanticConventions.OUTPUT_VALUE] = outputMessages;
         }
+        const usageMetrics = convertUsageMetricsToOpenInference(attributes);
+        Object.assign(processedAttributes, usageMetrics);
         mutableSpan.attributes = { ...processedAttributes, ...mastraOther };
       }
       return mutableSpan;

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/openInferenceOTLPExporter.ts","../src/tracing.ts"],"names":["OTLPTraceExporter","convertGenAISpanAttributesToOpenInferenceSpanAttributes","SESSION_ID","TAG_TAGS","USER_ID","METADATA","ATTR_GEN_AI_INPUT_MESSAGES","INPUT_MIME_TYPE","INPUT_VALUE","ATTR_GEN_AI_OUTPUT_MESSAGES","OUTPUT_MIME_TYPE","OUTPUT_VALUE","OtelExporter","logger","ConsoleLogger","SEMRESATTRS_PROJECT_NAME"],"mappings":";;;;;;;;;;AAoBA,IAAM,qBAAA,GAAwB,SAAA;AAC9B,IAAM,sBAAA,GAAyB,kBAAA;AAS/B,SAAS,sBAAsB,UAAA,EAG7B;AACA,EAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,UAAU,CAAA,CAAE,MAAA;AAAA,IAChC,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,KAAK,CAAA,KAAM;AACrB,MAAA,IAAI,GAAA,CAAI,UAAA,CAAW,qBAAqB,CAAA,EAAG;AACzC,QAAA,IAAI,GAAA,CAAI,UAAA,CAAW,sBAAsB,CAAA,EAAG;AAC1C,UAAA,MAAM,WAAA,GAAc,GAAA,CAAI,KAAA,CAAM,sBAAA,CAAuB,MAAM,CAAA;AAC3D,UAAA,GAAA,CAAI,cAAA,CAAe,WAAW,CAAA,GAAI,KAAA;AAAA,QACpC,CAAA,MAAO;AACL,UAAA,GAAA,CAAI,WAAA,CAAY,GAAG,CAAA,GAAI,KAAA;AAAA,QACzB;AAAA,MACF;AACA,MAAA,OAAO,GAAA;AAAA,IACT,CAAA;AAAA,IACA;AAAA,MACE,gBAAgB,EAAC;AAAA,MACjB,aAAa;AAAC;AAChB,GACF;AACF;AAEO,IAAM,8BAAA,GAAN,cAA6CA,wCAAA,CAAkB;AAAA,EACpE,MAAA,CAAO,OAAuB,cAAA,EAAgD;AAC5E,IAAA,MAAM,cAAA,GAAiB,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AACvC,MAAA,MAAM,aAAa,EAAE,GAAI,IAAA,CAAK,UAAA,IAAc,EAAC,EAAG;AAChD,MAAA,MAAM,WAAA,GAAc,IAAA;AAEpB,MAAA,MAAM,EAAE,cAAA,EAAgB,WAAA,EAAY,GAAI,sBAAsB,UAAU,CAAA;AACxE,MAAA,MAAM,mBAAA,GAAsBC,2EAAwD,UAAU,CAAA;AAG9F,MAAA,IAAI,mBAAA,EAAqB;AACvB,QAAA,MAAM,QAAA,GAAW,eAAe,UAAU,CAAA;AAC1C,QAAA,IAAI,QAAA,EAAU;AACZ,UAAA,OAAO,eAAe,UAAU,CAAA;AAChC,UAAA,mBAAA,CAAoBC,2CAAU,CAAA,GAAI,QAAA;AAAA,QACpC;AAGA,QAAA,IAAI,WAAA,CAAY,aAAa,CAAA,EAAG;AAC9B,UAAA,mBAAA,CAAoBC,yCAAQ,CAAA,GAAI,WAAA,CAAY,aAAa,CAAA;AACzD,UAAA,OAAO,YAAY,aAAa,CAAA;AAAA,QAClC;AAEA,QAAA,MAAM,MAAA,GAAS,eAAe,QAAQ,CAAA;AACtC,QAAA,IAAI,MAAA,EAAQ;AACV,UAAA,OAAO,eAAe,QAAQ,CAAA;AAC9B,UAAA,mBAAA,CAAoBC,wCAAO,CAAA,GAAI,MAAA;AAAA,QACjC;AAGA,QAAA,IAAI,MAAA,CAAO,IAAA,CAAK,cAAc,CAAA,CAAE,SAAS,CAAA,EAAG;AAC1C,UAAA,IAAI;AACF,YAAA,mBAAA,CAAoBC,yCAAQ,CAAA,GAAI,IAAA,CAAK,SAAA,CAAU,cAAc,CAAA;AAAA,UAC/D,CAAA,CAAA,MAAQ;AAAA,UAER;AAAA,QACF;AAEA,QAAA,MAAM,aAAA,GAAgB,WAAWC,qCAA0B,CAAA;AAC3D,QAAA,IAAI,aAAA,EAAe;AACjB,UAAA,mBAAA,CAAoBC,gDAAe,CAAA,GAAI,kBAAA;AACvC,UAAA,mBAAA,CAAoBC,4CAAW,CAAA,GAAI,aAAA;AAAA,QACrC;AACA,QAAA,MAAM,cAAA,GAAiB,WAAWC,sCAA2B,CAAA;AAC7D,QAAA,IAAI,cAAA,EAAgB;AAClB,UAAA,mBAAA,CAAoBC,iDAAgB,CAAA,GAAI,kBAAA;AACxC,UAAA,mBAAA,CAAoBC,6CAAY,CAAA,GAAI,cAAA;AAAA,QACtC;AAEA,QAAA,WAAA,CAAY,UAAA,GAAa,EAAE,GAAG,mBAAA,EAAqB,GAAG,WAAA,EAAY;AAAA,MACpE;AAEA,MAAA,OAAO,WAAA;AAAA,IACT,CAAC,CAAA;AAED,IAAA,KAAA,CAAM,MAAA,CAAO,gBAAgB,cAAc,CAAA;AAAA,EAC7C;AACF,CAAA;;;ACvGA,IAAM,UAAA,GAAa,iBAAA;AAEZ,IAAM,iBAAA,GAAoB;AA6B1B,IAAM,aAAA,GAAN,cAA4BC,yBAAA,CAAa;AAAA,EAC9C,IAAA,GAAO,OAAA;AAAA,EAEP,YAAY,MAAA,EAA6B;AACvC,IAAA,MAAMC,QAAA,GAAS,IAAIC,oBAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AACrE,IAAA,IAAI,WAA+B,MAAA,CAAO,QAAA;AAC1C,IAAA,MAAM,OAAA,GAAkC;AAAA,MACtC,GAAG,MAAA,CAAO;AAAA,KACZ;AACA,IAAA,IAAI,OAAO,OAAA,EAAS;AAElB,MAAA,OAAA,CAAQ,UAAU,IAAI,MAAA,CAAO,OAAA;AAC7B,MAAA,OAAA,CAAQ,SAAS,CAAA,GAAI,MAAA,CAAO,MAAA,IAAU,EAAA;AACtC,MAAA,QAAA,GAAW,OAAO,QAAA,IAAY,iBAAA;AAAA,IAChC,CAAA,MAAA,IAAW,OAAO,MAAA,EAAQ;AAExB,MAAA,OAAA,CAAQ,eAAe,CAAA,GAAI,CAAA,OAAA,EAAU,MAAA,CAAO,MAAM,CAAA,CAAA;AAAA,IACpD;AACA,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAAD,QAAA,CAAO,KAAA,CAAM,CAAA,EAAG,UAAU,CAAA,2DAAA,CAA6D,CAAA;AACvF,MAAA;AAAA,IACF;AACA,IAAA,KAAA,CAAM;AAAA,MACJ,QAAA,EAAU,IAAI,8BAAA,CAA+B;AAAA,QAC3C,GAAA,EAAK,QAAA;AAAA,QACL;AAAA,OACD,CAAA;AAAA,MACD,GAAG,MAAA;AAAA,MACH,kBAAA,EAAoB;AAAA,QAClB,CAACE,yDAAwB,GAAG,MAAA,CAAO,WAAA;AAAA,QACnC,GAAG,MAAA,CAAO;AAAA,OACZ;AAAA,MACA,QAAA,EAAU;AAAA,QACR,MAAA,EAAQ;AAAA,UACN,QAAA;AAAA,UACA,OAAA;AAAA,UACA,QAAA,EAAU;AAAA;AACZ;AACF,KAC4B,CAAA;AAAA,EAChC;AACF","file":"index.cjs","sourcesContent":["import { convertGenAISpanAttributesToOpenInferenceSpanAttributes } from '@arizeai/openinference-genai';\nimport type { Mutable } from '@arizeai/openinference-genai/types';\nimport {\n  INPUT_MIME_TYPE,\n  INPUT_VALUE,\n  METADATA,\n  OUTPUT_MIME_TYPE,\n  OUTPUT_VALUE,\n  SESSION_ID,\n  TAG_TAGS,\n  USER_ID,\n} from '@arizeai/openinference-semantic-conventions';\nimport type { ExportResult } from '@opentelemetry/core';\nimport { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-proto';\nimport type { ReadableSpan } from '@opentelemetry/sdk-trace-base';\nimport {\n  ATTR_GEN_AI_INPUT_MESSAGES,\n  ATTR_GEN_AI_OUTPUT_MESSAGES,\n} from '@opentelemetry/semantic-conventions/incubating';\n\nconst MASTRA_GENERAL_PREFIX = 'mastra.';\nconst MASTRA_METADATA_PREFIX = 'mastra.metadata.';\n\n/**\n * Splits Mastra span attributes into two groups:\n * - `metadata`: keys starting with \"mastra.metadata.\" (prefix removed)\n * - `other`: all remaining keys starting with \"mastra.\"\n *\n * Any attributes not starting with \"mastra.\" are ignored entirely.\n */\nfunction splitMastraAttributes(attributes: Record<string, any>): {\n  mastraMetadata: Record<string, any>;\n  mastraOther: Record<string, any>;\n} {\n  return Object.entries(attributes).reduce(\n    (acc, [key, value]) => {\n      if (key.startsWith(MASTRA_GENERAL_PREFIX)) {\n        if (key.startsWith(MASTRA_METADATA_PREFIX)) {\n          const strippedKey = key.slice(MASTRA_METADATA_PREFIX.length);\n          acc.mastraMetadata[strippedKey] = value;\n        } else {\n          acc.mastraOther[key] = value;\n        }\n      }\n      return acc;\n    },\n    {\n      mastraMetadata: {} as Record<string, any>,\n      mastraOther: {} as Record<string, any>,\n    },\n  );\n}\n\nexport class OpenInferenceOTLPTraceExporter extends OTLPTraceExporter {\n  export(spans: ReadableSpan[], resultCallback: (result: ExportResult) => void) {\n    const processedSpans = spans.map(span => {\n      const attributes = { ...(span.attributes ?? {}) };\n      const mutableSpan = span as Mutable<ReadableSpan>;\n\n      const { mastraMetadata, mastraOther } = splitMastraAttributes(attributes);\n      const processedAttributes = convertGenAISpanAttributesToOpenInferenceSpanAttributes(attributes);\n\n      // only add processed attributes if conversion was successful\n      if (processedAttributes) {\n        const threadId = mastraMetadata['threadId'];\n        if (threadId) {\n          delete mastraMetadata['threadId'];\n          processedAttributes[SESSION_ID] = threadId;\n        }\n\n        // Map mastra.tags to OpenInference native tag.tags convention (tags are only on root spans)\n        if (mastraOther['mastra.tags']) {\n          processedAttributes[TAG_TAGS] = mastraOther['mastra.tags'];\n          delete mastraOther['mastra.tags'];\n        }\n\n        const userId = mastraMetadata['userId'];\n        if (userId) {\n          delete mastraMetadata['userId'];\n          processedAttributes[USER_ID] = userId;\n        }\n\n        // Gather custom metadata into OpenInference metadata (flat best-effort)\n        if (Object.keys(mastraMetadata).length > 0) {\n          try {\n            processedAttributes[METADATA] = JSON.stringify(mastraMetadata);\n          } catch {\n            // best-effort only\n          }\n        }\n\n        const inputMessages = attributes[ATTR_GEN_AI_INPUT_MESSAGES];\n        if (inputMessages) {\n          processedAttributes[INPUT_MIME_TYPE] = 'application/json';\n          processedAttributes[INPUT_VALUE] = inputMessages;\n        }\n        const outputMessages = attributes[ATTR_GEN_AI_OUTPUT_MESSAGES];\n        if (outputMessages) {\n          processedAttributes[OUTPUT_MIME_TYPE] = 'application/json';\n          processedAttributes[OUTPUT_VALUE] = outputMessages;\n        }\n\n        mutableSpan.attributes = { ...processedAttributes, ...mastraOther };\n      }\n\n      return mutableSpan;\n    });\n\n    super.export(processedSpans, resultCallback);\n  }\n}\n","import { SEMRESATTRS_PROJECT_NAME } from '@arizeai/openinference-semantic-conventions';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { OtelExporter } from '@mastra/otel-exporter';\nimport type { OtelExporterConfig } from '@mastra/otel-exporter';\n\nimport { OpenInferenceOTLPTraceExporter } from './openInferenceOTLPExporter.js';\n\nconst LOG_PREFIX = '[ArizeExporter]';\n\nexport const ARIZE_AX_ENDPOINT = 'https://otlp.arize.com/v1/traces';\n\nexport type ArizeExporterConfig = Omit<OtelExporterConfig, 'provider'> & {\n  /**\n   * Required if sending traces to Arize AX\n   */\n  spaceId?: string;\n  /**\n   * Required if sending traces to Arize AX, or to any other collector that\n   * requires an Authorization header\n   */\n  apiKey?: string;\n  /**\n   * Collector endpoint destination for trace exports.\n   * Required when sending traces to Phoenix, Phoenix Cloud, or other collectors.\n   * Optional when sending traces to Arize AX.\n   */\n  endpoint?: string;\n  /**\n   * Optional project name to be added as a resource attribute using\n   * OpenInference Semantic Conventions\n   */\n  projectName?: string;\n  /**\n   * Optional headers to be added to each OTLP request\n   */\n  headers?: Record<string, string>;\n};\n\nexport class ArizeExporter extends OtelExporter {\n  name = 'arize';\n\n  constructor(config: ArizeExporterConfig) {\n    const logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n    let endpoint: string | undefined = config.endpoint;\n    const headers: Record<string, string> = {\n      ...config.headers,\n    };\n    if (config.spaceId) {\n      // arize ax header configuration\n      headers['space_id'] = config.spaceId;\n      headers['api_key'] = config.apiKey ?? '';\n      endpoint = config.endpoint || ARIZE_AX_ENDPOINT;\n    } else if (config.apiKey) {\n      // standard otel header configuration\n      headers['Authorization'] = `Bearer ${config.apiKey}`;\n    }\n    if (!endpoint) {\n      logger.error(`${LOG_PREFIX} Endpoint is required in configuration. Disabling exporter.`);\n      return;\n    }\n    super({\n      exporter: new OpenInferenceOTLPTraceExporter({\n        url: endpoint,\n        headers,\n      }),\n      ...config,\n      resourceAttributes: {\n        [SEMRESATTRS_PROJECT_NAME]: config.projectName,\n        ...config.resourceAttributes,\n      },\n      provider: {\n        custom: {\n          endpoint,\n          headers,\n          protocol: 'http/protobuf',\n        },\n      } satisfies OtelExporterConfig['provider'],\n    } satisfies OtelExporterConfig);\n  }\n}\n"]}
+{"version":3,"sources":["../src/openInferenceOTLPExporter.ts","../src/tracing.ts"],"names":["ATTR_GEN_AI_USAGE_INPUT_TOKENS","ATTR_GEN_AI_USAGE_OUTPUT_TOKENS","LLM_TOKEN_COUNT_PROMPT","LLM_TOKEN_COUNT_COMPLETION","LLM_TOKEN_COUNT_TOTAL","LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ","LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_WRITE","LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING","LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO","LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO","OTLPTraceExporter","convertGenAISpanAttributesToOpenInferenceSpanAttributes","SESSION_ID","TAG_TAGS","USER_ID","METADATA","ATTR_GEN_AI_INPUT_MESSAGES","INPUT_MIME_TYPE","INPUT_VALUE","ATTR_GEN_AI_OUTPUT_MESSAGES","OUTPUT_MIME_TYPE","OUTPUT_VALUE","OtelExporter","logger","ConsoleLogger","SEMRESATTRS_PROJECT_NAME"],"mappings":";;;;;;;;;;AA+BA,IAAM,6BAAA,GAAgC,+BAAA;AACtC,IAAM,gCAAA,GAAmC,kCAAA;AACzC,IAAM,+BAAA,GAAkC,iCAAA;AACxC,IAAM,+BAAA,GAAkC,iCAAA;AACxC,IAAM,gCAAA,GAAmC,kCAAA;AAEzC,IAAM,qBAAA,GAAwB,SAAA;AAC9B,IAAM,sBAAA,GAAyB,kBAAA;AAS/B,SAAS,mCAAmC,UAAA,EAAsD;AAChG,EAAA,MAAM,SAA8B,EAAC;AAErC,EAAA,MAAM,WAAA,GAAc,WAAWA,yCAA8B,CAAA;AAC7D,EAAA,MAAM,YAAA,GAAe,WAAWC,0CAA+B,CAAA;AAG/D,EAAA,IAAI,gBAAgB,MAAA,EAAW;AAC7B,IAAA,MAAA,CAAOC,uDAAsB,CAAA,GAAI,WAAA;AAAA,EACnC;AACA,EAAA,IAAI,iBAAiB,MAAA,EAAW;AAC9B,IAAA,MAAA,CAAOC,2DAA0B,CAAA,GAAI,YAAA;AAAA,EACvC;AAGA,EAAA,IAAI,WAAA,KAAgB,MAAA,IAAa,YAAA,KAAiB,MAAA,EAAW;AAC3D,IAAA,MAAA,CAAOC,sDAAqB,IAAI,WAAA,GAAc,YAAA;AAAA,EAChD;AAGA,EAAA,MAAM,iBAAA,GAAoB,WAAW,gCAAgC,CAAA;AACrE,EAAA,IAAI,sBAAsB,MAAA,EAAW;AACnC,IAAA,MAAA,CAAOC,0EAAyC,CAAA,GAAI,iBAAA;AAAA,EACtD;AAEA,EAAA,MAAM,gBAAA,GAAmB,WAAW,+BAA+B,CAAA;AACnE,EAAA,IAAI,qBAAqB,MAAA,EAAW;AAClC,IAAA,MAAA,CAAOC,2EAA0C,CAAA,GAAI,gBAAA;AAAA,EACvD;AAGA,EAAA,MAAM,eAAA,GAAkB,WAAW,6BAA6B,CAAA;AAChE,EAAA,IAAI,oBAAoB,MAAA,EAAW;AACjC,IAAA,MAAA,CAAOC,6EAA4C,CAAA,GAAI,eAAA;AAAA,EACzD;AAGA,EAAA,MAAM,gBAAA,GAAmB,WAAW,+BAA+B,CAAA;AACnE,EAAA,IAAI,qBAAqB,MAAA,EAAW;AAClC,IAAA,MAAA,CAAOC,qEAAoC,CAAA,GAAI,gBAAA;AAAA,EACjD;AAEA,EAAA,MAAM,iBAAA,GAAoB,WAAW,gCAAgC,CAAA;AACrE,EAAA,IAAI,sBAAsB,MAAA,EAAW;AACnC,IAAA,MAAA,CAAOC,yEAAwC,CAAA,GAAI,iBAAA;AAAA,EACrD;AAEA,EAAA,OAAO,MAAA;AACT;AASA,SAAS,sBAAsB,UAAA,EAG7B;AACA,EAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,UAAU,CAAA,CAAE,MAAA;AAAA,IAChC,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,KAAK,CAAA,KAAM;AACrB,MAAA,IAAI,GAAA,CAAI,UAAA,CAAW,qBAAqB,CAAA,EAAG;AACzC,QAAA,IAAI,GAAA,CAAI,UAAA,CAAW,sBAAsB,CAAA,EAAG;AAC1C,UAAA,MAAM,WAAA,GAAc,GAAA,CAAI,KAAA,CAAM,sBAAA,CAAuB,MAAM,CAAA;AAC3D,UAAA,GAAA,CAAI,cAAA,CAAe,WAAW,CAAA,GAAI,KAAA;AAAA,QACpC,CAAA,MAAO;AACL,UAAA,GAAA,CAAI,WAAA,CAAY,GAAG,CAAA,GAAI,KAAA;AAAA,QACzB;AAAA,MACF;AACA,MAAA,OAAO,GAAA;AAAA,IACT,CAAA;AAAA,IACA;AAAA,MACE,gBAAgB,EAAC;AAAA,MACjB,aAAa;AAAC;AAChB,GACF;AACF;AAEO,IAAM,8BAAA,GAAN,cAA6CC,wCAAA,CAAkB;AAAA,EACpE,MAAA,CAAO,OAAuB,cAAA,EAAgD;AAC5E,IAAA,MAAM,cAAA,GAAiB,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AACvC,MAAA,MAAM,aAAa,EAAE,GAAI,IAAA,CAAK,UAAA,IAAc,EAAC,EAAG;AAChD,MAAA,MAAM,WAAA,GAAc,IAAA;AAEpB,MAAA,MAAM,EAAE,cAAA,EAAgB,WAAA,EAAY,GAAI,sBAAsB,UAAU,CAAA;AACxE,MAAA,MAAM,mBAAA,GAAsBC,2EAAwD,UAAU,CAAA;AAG9F,MAAA,IAAI,mBAAA,EAAqB;AACvB,QAAA,MAAM,QAAA,GAAW,eAAe,UAAU,CAAA;AAC1C,QAAA,IAAI,QAAA,EAAU;AACZ,UAAA,OAAO,eAAe,UAAU,CAAA;AAChC,UAAA,mBAAA,CAAoBC,2CAAU,CAAA,GAAI,QAAA;AAAA,QACpC;AAGA,QAAA,IAAI,WAAA,CAAY,aAAa,CAAA,EAAG;AAC9B,UAAA,mBAAA,CAAoBC,yCAAQ,CAAA,GAAI,WAAA,CAAY,aAAa,CAAA;AACzD,UAAA,OAAO,YAAY,aAAa,CAAA;AAAA,QAClC;AAEA,QAAA,MAAM,MAAA,GAAS,eAAe,QAAQ,CAAA;AACtC,QAAA,IAAI,MAAA,EAAQ;AACV,UAAA,OAAO,eAAe,QAAQ,CAAA;AAC9B,UAAA,mBAAA,CAAoBC,wCAAO,CAAA,GAAI,MAAA;AAAA,QACjC;AAGA,QAAA,IAAI,MAAA,CAAO,IAAA,CAAK,cAAc,CAAA,CAAE,SAAS,CAAA,EAAG;AAC1C,UAAA,IAAI;AACF,YAAA,mBAAA,CAAoBC,yCAAQ,CAAA,GAAI,IAAA,CAAK,SAAA,CAAU,cAAc,CAAA;AAAA,UAC/D,CAAA,CAAA,MAAQ;AAAA,UAER;AAAA,QACF;AAEA,QAAA,MAAM,aAAA,GAAgB,WAAWC,qCAA0B,CAAA;AAC3D,QAAA,IAAI,aAAA,EAAe;AACjB,UAAA,mBAAA,CAAoBC,gDAAe,CAAA,GAAI,kBAAA;AACvC,UAAA,mBAAA,CAAoBC,4CAAW,CAAA,GAAI,aAAA;AAAA,QACrC;AACA,QAAA,MAAM,cAAA,GAAiB,WAAWC,sCAA2B,CAAA;AAC7D,QAAA,IAAI,cAAA,EAAgB;AAClB,UAAA,mBAAA,CAAoBC,iDAAgB,CAAA,GAAI,kBAAA;AACxC,UAAA,mBAAA,CAAoBC,6CAAY,CAAA,GAAI,cAAA;AAAA,QACtC;AAGA,QAAA,MAAM,YAAA,GAAe,mCAAmC,UAAU,CAAA;AAClE,QAAA,MAAA,CAAO,MAAA,CAAO,qBAAqB,YAAY,CAAA;AAE/C,QAAA,WAAA,CAAY,UAAA,GAAa,EAAE,GAAG,mBAAA,EAAqB,GAAG,WAAA,EAAY;AAAA,MACpE;AAEA,MAAA,OAAO,WAAA;AAAA,IACT,CAAC,CAAA;AAED,IAAA,KAAA,CAAM,MAAA,CAAO,gBAAgB,cAAc,CAAA;AAAA,EAC7C;AACF,CAAA;;;ACrLA,IAAM,UAAA,GAAa,iBAAA;AAEZ,IAAM,iBAAA,GAAoB;AA6B1B,IAAM,aAAA,GAAN,cAA4BC,yBAAA,CAAa;AAAA,EAC9C,IAAA,GAAO,OAAA;AAAA,EAEP,YAAY,MAAA,EAA6B;AACvC,IAAA,MAAMC,QAAA,GAAS,IAAIC,oBAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AACrE,IAAA,IAAI,WAA+B,MAAA,CAAO,QAAA;AAC1C,IAAA,MAAM,OAAA,GAAkC;AAAA,MACtC,GAAG,MAAA,CAAO;AAAA,KACZ;AACA,IAAA,IAAI,OAAO,OAAA,EAAS;AAElB,MAAA,OAAA,CAAQ,UAAU,IAAI,MAAA,CAAO,OAAA;AAC7B,MAAA,OAAA,CAAQ,SAAS,CAAA,GAAI,MAAA,CAAO,MAAA,IAAU,EAAA;AACtC,MAAA,QAAA,GAAW,OAAO,QAAA,IAAY,iBAAA;AAAA,IAChC,CAAA,MAAA,IAAW,OAAO,MAAA,EAAQ;AAExB,MAAA,OAAA,CAAQ,eAAe,CAAA,GAAI,CAAA,OAAA,EAAU,MAAA,CAAO,MAAM,CAAA,CAAA;AAAA,IACpD;AACA,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAAD,QAAA,CAAO,KAAA,CAAM,CAAA,EAAG,UAAU,CAAA,2DAAA,CAA6D,CAAA;AACvF,MAAA;AAAA,IACF;AACA,IAAA,KAAA,CAAM;AAAA,MACJ,QAAA,EAAU,IAAI,8BAAA,CAA+B;AAAA,QAC3C,GAAA,EAAK,QAAA;AAAA,QACL;AAAA,OACD,CAAA;AAAA,MACD,GAAG,MAAA;AAAA,MACH,kBAAA,EAAoB;AAAA,QAClB,CAACE,yDAAwB,GAAG,MAAA,CAAO,WAAA;AAAA,QACnC,GAAG,MAAA,CAAO;AAAA,OACZ;AAAA,MACA,QAAA,EAAU;AAAA,QACR,MAAA,EAAQ;AAAA,UACN,QAAA;AAAA,UACA,OAAA;AAAA,UACA,QAAA,EAAU;AAAA;AACZ;AACF,KAC4B,CAAA;AAAA,EAChC;AACF","file":"index.cjs","sourcesContent":["import { convertGenAISpanAttributesToOpenInferenceSpanAttributes } from '@arizeai/openinference-genai';\nimport type { Mutable } from '@arizeai/openinference-genai/types';\nimport {\n  INPUT_MIME_TYPE,\n  INPUT_VALUE,\n  LLM_TOKEN_COUNT_COMPLETION,\n  LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO,\n  LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING,\n  LLM_TOKEN_COUNT_PROMPT,\n  LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO,\n  LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ,\n  LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_WRITE,\n  LLM_TOKEN_COUNT_TOTAL,\n  METADATA,\n  OUTPUT_MIME_TYPE,\n  OUTPUT_VALUE,\n  SESSION_ID,\n  TAG_TAGS,\n  USER_ID,\n} from '@arizeai/openinference-semantic-conventions';\nimport type { ExportResult } from '@opentelemetry/core';\nimport { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-proto';\nimport type { ReadableSpan } from '@opentelemetry/sdk-trace-base';\nimport {\n  ATTR_GEN_AI_INPUT_MESSAGES,\n  ATTR_GEN_AI_OUTPUT_MESSAGES,\n  ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n  ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n} from '@opentelemetry/semantic-conventions/incubating';\n\n// GenAI usage attribute keys (not all are in @opentelemetry/semantic-conventions yet)\nconst GEN_AI_USAGE_REASONING_TOKENS = 'gen_ai.usage.reasoning_tokens';\nconst GEN_AI_USAGE_CACHED_INPUT_TOKENS = 'gen_ai.usage.cached_input_tokens';\nconst GEN_AI_USAGE_CACHE_WRITE_TOKENS = 'gen_ai.usage.cache_write_tokens';\nconst GEN_AI_USAGE_AUDIO_INPUT_TOKENS = 'gen_ai.usage.audio_input_tokens';\nconst GEN_AI_USAGE_AUDIO_OUTPUT_TOKENS = 'gen_ai.usage.audio_output_tokens';\n\nconst MASTRA_GENERAL_PREFIX = 'mastra.';\nconst MASTRA_METADATA_PREFIX = 'mastra.metadata.';\n\n/**\n * Converts GenAI usage metrics to OpenInference LLM token count attributes.\n * Maps from OTEL GenAI semantic conventions to OpenInference semantic conventions.\n *\n * @param attributes - The span attributes containing GenAI usage metrics\n * @returns OpenInference token count attributes\n */\nfunction convertUsageMetricsToOpenInference(attributes: Record<string, any>): Record<string, any> {\n  const result: Record<string, any> = {};\n\n  const inputTokens = attributes[ATTR_GEN_AI_USAGE_INPUT_TOKENS];\n  const outputTokens = attributes[ATTR_GEN_AI_USAGE_OUTPUT_TOKENS];\n\n  // Core token counts\n  if (inputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_PROMPT] = inputTokens;\n  }\n  if (outputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_COMPLETION] = outputTokens;\n  }\n\n  // Total tokens (compute if we have both input and output)\n  if (inputTokens !== undefined && outputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_TOTAL] = inputTokens + outputTokens;\n  }\n\n  // Cache tokens (prompt details)\n  const cachedInputTokens = attributes[GEN_AI_USAGE_CACHED_INPUT_TOKENS];\n  if (cachedInputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ] = cachedInputTokens;\n  }\n\n  const cacheWriteTokens = attributes[GEN_AI_USAGE_CACHE_WRITE_TOKENS];\n  if (cacheWriteTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_WRITE] = cacheWriteTokens;\n  }\n\n  // Reasoning tokens (completion details)\n  const reasoningTokens = attributes[GEN_AI_USAGE_REASONING_TOKENS];\n  if (reasoningTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING] = reasoningTokens;\n  }\n\n  // Audio tokens\n  const audioInputTokens = attributes[GEN_AI_USAGE_AUDIO_INPUT_TOKENS];\n  if (audioInputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO] = audioInputTokens;\n  }\n\n  const audioOutputTokens = attributes[GEN_AI_USAGE_AUDIO_OUTPUT_TOKENS];\n  if (audioOutputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO] = audioOutputTokens;\n  }\n\n  return result;\n}\n\n/**\n * Splits Mastra span attributes into two groups:\n * - `metadata`: keys starting with \"mastra.metadata.\" (prefix removed)\n * - `other`: all remaining keys starting with \"mastra.\"\n *\n * Any attributes not starting with \"mastra.\" are ignored entirely.\n */\nfunction splitMastraAttributes(attributes: Record<string, any>): {\n  mastraMetadata: Record<string, any>;\n  mastraOther: Record<string, any>;\n} {\n  return Object.entries(attributes).reduce(\n    (acc, [key, value]) => {\n      if (key.startsWith(MASTRA_GENERAL_PREFIX)) {\n        if (key.startsWith(MASTRA_METADATA_PREFIX)) {\n          const strippedKey = key.slice(MASTRA_METADATA_PREFIX.length);\n          acc.mastraMetadata[strippedKey] = value;\n        } else {\n          acc.mastraOther[key] = value;\n        }\n      }\n      return acc;\n    },\n    {\n      mastraMetadata: {} as Record<string, any>,\n      mastraOther: {} as Record<string, any>,\n    },\n  );\n}\n\nexport class OpenInferenceOTLPTraceExporter extends OTLPTraceExporter {\n  export(spans: ReadableSpan[], resultCallback: (result: ExportResult) => void) {\n    const processedSpans = spans.map(span => {\n      const attributes = { ...(span.attributes ?? {}) };\n      const mutableSpan = span as Mutable<ReadableSpan>;\n\n      const { mastraMetadata, mastraOther } = splitMastraAttributes(attributes);\n      const processedAttributes = convertGenAISpanAttributesToOpenInferenceSpanAttributes(attributes);\n\n      // only add processed attributes if conversion was successful\n      if (processedAttributes) {\n        const threadId = mastraMetadata['threadId'];\n        if (threadId) {\n          delete mastraMetadata['threadId'];\n          processedAttributes[SESSION_ID] = threadId;\n        }\n\n        // Map mastra.tags to OpenInference native tag.tags convention (tags are only on root spans)\n        if (mastraOther['mastra.tags']) {\n          processedAttributes[TAG_TAGS] = mastraOther['mastra.tags'];\n          delete mastraOther['mastra.tags'];\n        }\n\n        const userId = mastraMetadata['userId'];\n        if (userId) {\n          delete mastraMetadata['userId'];\n          processedAttributes[USER_ID] = userId;\n        }\n\n        // Gather custom metadata into OpenInference metadata (flat best-effort)\n        if (Object.keys(mastraMetadata).length > 0) {\n          try {\n            processedAttributes[METADATA] = JSON.stringify(mastraMetadata);\n          } catch {\n            // best-effort only\n          }\n        }\n\n        const inputMessages = attributes[ATTR_GEN_AI_INPUT_MESSAGES];\n        if (inputMessages) {\n          processedAttributes[INPUT_MIME_TYPE] = 'application/json';\n          processedAttributes[INPUT_VALUE] = inputMessages;\n        }\n        const outputMessages = attributes[ATTR_GEN_AI_OUTPUT_MESSAGES];\n        if (outputMessages) {\n          processedAttributes[OUTPUT_MIME_TYPE] = 'application/json';\n          processedAttributes[OUTPUT_VALUE] = outputMessages;\n        }\n\n        // Convert GenAI usage metrics to OpenInference token count attributes\n        const usageMetrics = convertUsageMetricsToOpenInference(attributes);\n        Object.assign(processedAttributes, usageMetrics);\n\n        mutableSpan.attributes = { ...processedAttributes, ...mastraOther };\n      }\n\n      return mutableSpan;\n    });\n\n    super.export(processedSpans, resultCallback);\n  }\n}\n","import { SEMRESATTRS_PROJECT_NAME } from '@arizeai/openinference-semantic-conventions';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { OtelExporter } from '@mastra/otel-exporter';\nimport type { OtelExporterConfig } from '@mastra/otel-exporter';\n\nimport { OpenInferenceOTLPTraceExporter } from './openInferenceOTLPExporter.js';\n\nconst LOG_PREFIX = '[ArizeExporter]';\n\nexport const ARIZE_AX_ENDPOINT = 'https://otlp.arize.com/v1/traces';\n\nexport type ArizeExporterConfig = Omit<OtelExporterConfig, 'provider'> & {\n  /**\n   * Required if sending traces to Arize AX\n   */\n  spaceId?: string;\n  /**\n   * Required if sending traces to Arize AX, or to any other collector that\n   * requires an Authorization header\n   */\n  apiKey?: string;\n  /**\n   * Collector endpoint destination for trace exports.\n   * Required when sending traces to Phoenix, Phoenix Cloud, or other collectors.\n   * Optional when sending traces to Arize AX.\n   */\n  endpoint?: string;\n  /**\n   * Optional project name to be added as a resource attribute using\n   * OpenInference Semantic Conventions\n   */\n  projectName?: string;\n  /**\n   * Optional headers to be added to each OTLP request\n   */\n  headers?: Record<string, string>;\n};\n\nexport class ArizeExporter extends OtelExporter {\n  name = 'arize';\n\n  constructor(config: ArizeExporterConfig) {\n    const logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n    let endpoint: string | undefined = config.endpoint;\n    const headers: Record<string, string> = {\n      ...config.headers,\n    };\n    if (config.spaceId) {\n      // arize ax header configuration\n      headers['space_id'] = config.spaceId;\n      headers['api_key'] = config.apiKey ?? '';\n      endpoint = config.endpoint || ARIZE_AX_ENDPOINT;\n    } else if (config.apiKey) {\n      // standard otel header configuration\n      headers['Authorization'] = `Bearer ${config.apiKey}`;\n    }\n    if (!endpoint) {\n      logger.error(`${LOG_PREFIX} Endpoint is required in configuration. Disabling exporter.`);\n      return;\n    }\n    super({\n      exporter: new OpenInferenceOTLPTraceExporter({\n        url: endpoint,\n        headers,\n      }),\n      ...config,\n      resourceAttributes: {\n        [SEMRESATTRS_PROJECT_NAME]: config.projectName,\n        ...config.resourceAttributes,\n      },\n      provider: {\n        custom: {\n          endpoint,\n          headers,\n          protocol: 'http/protobuf',\n        },\n      } satisfies OtelExporterConfig['provider'],\n    } satisfies OtelExporterConfig);\n  }\n}\n"]}

package/dist/index.js

@@ -1,13 +1,53 @@
-import { SEMRESATTRS_PROJECT_NAME, SESSION_ID, TAG_TAGS, USER_ID, METADATA, INPUT_MIME_TYPE, INPUT_VALUE, OUTPUT_MIME_TYPE, OUTPUT_VALUE } from '@arizeai/openinference-semantic-conventions';
+import { SEMRESATTRS_PROJECT_NAME, SESSION_ID, TAG_TAGS, USER_ID, METADATA, INPUT_MIME_TYPE, INPUT_VALUE, OUTPUT_MIME_TYPE, OUTPUT_VALUE, LLM_TOKEN_COUNT_PROMPT, LLM_TOKEN_COUNT_COMPLETION, LLM_TOKEN_COUNT_TOTAL, LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ, LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_WRITE, LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING, LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO, LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO } from '@arizeai/openinference-semantic-conventions';
 import { ConsoleLogger } from '@mastra/core/logger';
 import { OtelExporter } from '@mastra/otel-exporter';
 import { convertGenAISpanAttributesToOpenInferenceSpanAttributes } from '@arizeai/openinference-genai';
 import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-proto';
-import { ATTR_GEN_AI_INPUT_MESSAGES, ATTR_GEN_AI_OUTPUT_MESSAGES } from '@opentelemetry/semantic-conventions/incubating';
+import { ATTR_GEN_AI_INPUT_MESSAGES, ATTR_GEN_AI_OUTPUT_MESSAGES, ATTR_GEN_AI_USAGE_INPUT_TOKENS, ATTR_GEN_AI_USAGE_OUTPUT_TOKENS } from '@opentelemetry/semantic-conventions/incubating';
 
 // src/tracing.ts
+var GEN_AI_USAGE_REASONING_TOKENS = "gen_ai.usage.reasoning_tokens";
+var GEN_AI_USAGE_CACHED_INPUT_TOKENS = "gen_ai.usage.cached_input_tokens";
+var GEN_AI_USAGE_CACHE_WRITE_TOKENS = "gen_ai.usage.cache_write_tokens";
+var GEN_AI_USAGE_AUDIO_INPUT_TOKENS = "gen_ai.usage.audio_input_tokens";
+var GEN_AI_USAGE_AUDIO_OUTPUT_TOKENS = "gen_ai.usage.audio_output_tokens";
 var MASTRA_GENERAL_PREFIX = "mastra.";
 var MASTRA_METADATA_PREFIX = "mastra.metadata.";
+function convertUsageMetricsToOpenInference(attributes) {
+  const result = {};
+  const inputTokens = attributes[ATTR_GEN_AI_USAGE_INPUT_TOKENS];
+  const outputTokens = attributes[ATTR_GEN_AI_USAGE_OUTPUT_TOKENS];
+  if (inputTokens !== void 0) {
+    result[LLM_TOKEN_COUNT_PROMPT] = inputTokens;
+  }
+  if (outputTokens !== void 0) {
+    result[LLM_TOKEN_COUNT_COMPLETION] = outputTokens;
+  }
+  if (inputTokens !== void 0 && outputTokens !== void 0) {
+    result[LLM_TOKEN_COUNT_TOTAL] = inputTokens + outputTokens;
+  }
+  const cachedInputTokens = attributes[GEN_AI_USAGE_CACHED_INPUT_TOKENS];
+  if (cachedInputTokens !== void 0) {
+    result[LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ] = cachedInputTokens;
+  }
+  const cacheWriteTokens = attributes[GEN_AI_USAGE_CACHE_WRITE_TOKENS];
+  if (cacheWriteTokens !== void 0) {
+    result[LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_WRITE] = cacheWriteTokens;
+  }
+  const reasoningTokens = attributes[GEN_AI_USAGE_REASONING_TOKENS];
+  if (reasoningTokens !== void 0) {
+    result[LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING] = reasoningTokens;
+  }
+  const audioInputTokens = attributes[GEN_AI_USAGE_AUDIO_INPUT_TOKENS];
+  if (audioInputTokens !== void 0) {
+    result[LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO] = audioInputTokens;
+  }
+  const audioOutputTokens = attributes[GEN_AI_USAGE_AUDIO_OUTPUT_TOKENS];
+  if (audioOutputTokens !== void 0) {
+    result[LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO] = audioOutputTokens;
+  }
+  return result;
+}
 function splitMastraAttributes(attributes) {
   return Object.entries(attributes).reduce(
     (acc, [key, value]) => {
@@ -65,6 +105,8 @@ var OpenInferenceOTLPTraceExporter = class extends OTLPTraceExporter {
           processedAttributes[OUTPUT_MIME_TYPE] = "application/json";
           processedAttributes[OUTPUT_VALUE] = outputMessages;
         }
+        const usageMetrics = convertUsageMetricsToOpenInference(attributes);
+        Object.assign(processedAttributes, usageMetrics);
         mutableSpan.attributes = { ...processedAttributes, ...mastraOther };
       }
       return mutableSpan;

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/openInferenceOTLPExporter.ts","../src/tracing.ts"],"names":[],"mappings":";;;;;;;;AAoBA,IAAM,qBAAA,GAAwB,SAAA;AAC9B,IAAM,sBAAA,GAAyB,kBAAA;AAS/B,SAAS,sBAAsB,UAAA,EAG7B;AACA,EAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,UAAU,CAAA,CAAE,MAAA;AAAA,IAChC,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,KAAK,CAAA,KAAM;AACrB,MAAA,IAAI,GAAA,CAAI,UAAA,CAAW,qBAAqB,CAAA,EAAG;AACzC,QAAA,IAAI,GAAA,CAAI,UAAA,CAAW,sBAAsB,CAAA,EAAG;AAC1C,UAAA,MAAM,WAAA,GAAc,GAAA,CAAI,KAAA,CAAM,sBAAA,CAAuB,MAAM,CAAA;AAC3D,UAAA,GAAA,CAAI,cAAA,CAAe,WAAW,CAAA,GAAI,KAAA;AAAA,QACpC,CAAA,MAAO;AACL,UAAA,GAAA,CAAI,WAAA,CAAY,GAAG,CAAA,GAAI,KAAA;AAAA,QACzB;AAAA,MACF;AACA,MAAA,OAAO,GAAA;AAAA,IACT,CAAA;AAAA,IACA;AAAA,MACE,gBAAgB,EAAC;AAAA,MACjB,aAAa;AAAC;AAChB,GACF;AACF;AAEO,IAAM,8BAAA,GAAN,cAA6C,iBAAA,CAAkB;AAAA,EACpE,MAAA,CAAO,OAAuB,cAAA,EAAgD;AAC5E,IAAA,MAAM,cAAA,GAAiB,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AACvC,MAAA,MAAM,aAAa,EAAE,GAAI,IAAA,CAAK,UAAA,IAAc,EAAC,EAAG;AAChD,MAAA,MAAM,WAAA,GAAc,IAAA;AAEpB,MAAA,MAAM,EAAE,cAAA,EAAgB,WAAA,EAAY,GAAI,sBAAsB,UAAU,CAAA;AACxE,MAAA,MAAM,mBAAA,GAAsB,wDAAwD,UAAU,CAAA;AAG9F,MAAA,IAAI,mBAAA,EAAqB;AACvB,QAAA,MAAM,QAAA,GAAW,eAAe,UAAU,CAAA;AAC1C,QAAA,IAAI,QAAA,EAAU;AACZ,UAAA,OAAO,eAAe,UAAU,CAAA;AAChC,UAAA,mBAAA,CAAoB,UAAU,CAAA,GAAI,QAAA;AAAA,QACpC;AAGA,QAAA,IAAI,WAAA,CAAY,aAAa,CAAA,EAAG;AAC9B,UAAA,mBAAA,CAAoB,QAAQ,CAAA,GAAI,WAAA,CAAY,aAAa,CAAA;AACzD,UAAA,OAAO,YAAY,aAAa,CAAA;AAAA,QAClC;AAEA,QAAA,MAAM,MAAA,GAAS,eAAe,QAAQ,CAAA;AACtC,QAAA,IAAI,MAAA,EAAQ;AACV,UAAA,OAAO,eAAe,QAAQ,CAAA;AAC9B,UAAA,mBAAA,CAAoB,OAAO,CAAA,GAAI,MAAA;AAAA,QACjC;AAGA,QAAA,IAAI,MAAA,CAAO,IAAA,CAAK,cAAc,CAAA,CAAE,SAAS,CAAA,EAAG;AAC1C,UAAA,IAAI;AACF,YAAA,mBAAA,CAAoB,QAAQ,CAAA,GAAI,IAAA,CAAK,SAAA,CAAU,cAAc,CAAA;AAAA,UAC/D,CAAA,CAAA,MAAQ;AAAA,UAER;AAAA,QACF;AAEA,QAAA,MAAM,aAAA,GAAgB,WAAW,0BAA0B,CAAA;AAC3D,QAAA,IAAI,aAAA,EAAe;AACjB,UAAA,mBAAA,CAAoB,eAAe,CAAA,GAAI,kBAAA;AACvC,UAAA,mBAAA,CAAoB,WAAW,CAAA,GAAI,aAAA;AAAA,QACrC;AACA,QAAA,MAAM,cAAA,GAAiB,WAAW,2BAA2B,CAAA;AAC7D,QAAA,IAAI,cAAA,EAAgB;AAClB,UAAA,mBAAA,CAAoB,gBAAgB,CAAA,GAAI,kBAAA;AACxC,UAAA,mBAAA,CAAoB,YAAY,CAAA,GAAI,cAAA;AAAA,QACtC;AAEA,QAAA,WAAA,CAAY,UAAA,GAAa,EAAE,GAAG,mBAAA,EAAqB,GAAG,WAAA,EAAY;AAAA,MACpE;AAEA,MAAA,OAAO,WAAA;AAAA,IACT,CAAC,CAAA;AAED,IAAA,KAAA,CAAM,MAAA,CAAO,gBAAgB,cAAc,CAAA;AAAA,EAC7C;AACF,CAAA;;;ACvGA,IAAM,UAAA,GAAa,iBAAA;AAEZ,IAAM,iBAAA,GAAoB;AA6B1B,IAAM,aAAA,GAAN,cAA4B,YAAA,CAAa;AAAA,EAC9C,IAAA,GAAO,OAAA;AAAA,EAEP,YAAY,MAAA,EAA6B;AACvC,IAAA,MAAM,MAAA,GAAS,IAAI,aAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AACrE,IAAA,IAAI,WAA+B,MAAA,CAAO,QAAA;AAC1C,IAAA,MAAM,OAAA,GAAkC;AAAA,MACtC,GAAG,MAAA,CAAO;AAAA,KACZ;AACA,IAAA,IAAI,OAAO,OAAA,EAAS;AAElB,MAAA,OAAA,CAAQ,UAAU,IAAI,MAAA,CAAO,OAAA;AAC7B,MAAA,OAAA,CAAQ,SAAS,CAAA,GAAI,MAAA,CAAO,MAAA,IAAU,EAAA;AACtC,MAAA,QAAA,GAAW,OAAO,QAAA,IAAY,iBAAA;AAAA,IAChC,CAAA,MAAA,IAAW,OAAO,MAAA,EAAQ;AAExB,MAAA,OAAA,CAAQ,eAAe,CAAA,GAAI,CAAA,OAAA,EAAU,MAAA,CAAO,MAAM,CAAA,CAAA;AAAA,IACpD;AACA,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,MAAA,CAAO,KAAA,CAAM,CAAA,EAAG,UAAU,CAAA,2DAAA,CAA6D,CAAA;AACvF,MAAA;AAAA,IACF;AACA,IAAA,KAAA,CAAM;AAAA,MACJ,QAAA,EAAU,IAAI,8BAAA,CAA+B;AAAA,QAC3C,GAAA,EAAK,QAAA;AAAA,QACL;AAAA,OACD,CAAA;AAAA,MACD,GAAG,MAAA;AAAA,MACH,kBAAA,EAAoB;AAAA,QAClB,CAAC,wBAAwB,GAAG,MAAA,CAAO,WAAA;AAAA,QACnC,GAAG,MAAA,CAAO;AAAA,OACZ;AAAA,MACA,QAAA,EAAU;AAAA,QACR,MAAA,EAAQ;AAAA,UACN,QAAA;AAAA,UACA,OAAA;AAAA,UACA,QAAA,EAAU;AAAA;AACZ;AACF,KAC4B,CAAA;AAAA,EAChC;AACF","file":"index.js","sourcesContent":["import { convertGenAISpanAttributesToOpenInferenceSpanAttributes } from '@arizeai/openinference-genai';\nimport type { Mutable } from '@arizeai/openinference-genai/types';\nimport {\n  INPUT_MIME_TYPE,\n  INPUT_VALUE,\n  METADATA,\n  OUTPUT_MIME_TYPE,\n  OUTPUT_VALUE,\n  SESSION_ID,\n  TAG_TAGS,\n  USER_ID,\n} from '@arizeai/openinference-semantic-conventions';\nimport type { ExportResult } from '@opentelemetry/core';\nimport { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-proto';\nimport type { ReadableSpan } from '@opentelemetry/sdk-trace-base';\nimport {\n  ATTR_GEN_AI_INPUT_MESSAGES,\n  ATTR_GEN_AI_OUTPUT_MESSAGES,\n} from '@opentelemetry/semantic-conventions/incubating';\n\nconst MASTRA_GENERAL_PREFIX = 'mastra.';\nconst MASTRA_METADATA_PREFIX = 'mastra.metadata.';\n\n/**\n * Splits Mastra span attributes into two groups:\n * - `metadata`: keys starting with \"mastra.metadata.\" (prefix removed)\n * - `other`: all remaining keys starting with \"mastra.\"\n *\n * Any attributes not starting with \"mastra.\" are ignored entirely.\n */\nfunction splitMastraAttributes(attributes: Record<string, any>): {\n  mastraMetadata: Record<string, any>;\n  mastraOther: Record<string, any>;\n} {\n  return Object.entries(attributes).reduce(\n    (acc, [key, value]) => {\n      if (key.startsWith(MASTRA_GENERAL_PREFIX)) {\n        if (key.startsWith(MASTRA_METADATA_PREFIX)) {\n          const strippedKey = key.slice(MASTRA_METADATA_PREFIX.length);\n          acc.mastraMetadata[strippedKey] = value;\n        } else {\n          acc.mastraOther[key] = value;\n        }\n      }\n      return acc;\n    },\n    {\n      mastraMetadata: {} as Record<string, any>,\n      mastraOther: {} as Record<string, any>,\n    },\n  );\n}\n\nexport class OpenInferenceOTLPTraceExporter extends OTLPTraceExporter {\n  export(spans: ReadableSpan[], resultCallback: (result: ExportResult) => void) {\n    const processedSpans = spans.map(span => {\n      const attributes = { ...(span.attributes ?? {}) };\n      const mutableSpan = span as Mutable<ReadableSpan>;\n\n      const { mastraMetadata, mastraOther } = splitMastraAttributes(attributes);\n      const processedAttributes = convertGenAISpanAttributesToOpenInferenceSpanAttributes(attributes);\n\n      // only add processed attributes if conversion was successful\n      if (processedAttributes) {\n        const threadId = mastraMetadata['threadId'];\n        if (threadId) {\n          delete mastraMetadata['threadId'];\n          processedAttributes[SESSION_ID] = threadId;\n        }\n\n        // Map mastra.tags to OpenInference native tag.tags convention (tags are only on root spans)\n        if (mastraOther['mastra.tags']) {\n          processedAttributes[TAG_TAGS] = mastraOther['mastra.tags'];\n          delete mastraOther['mastra.tags'];\n        }\n\n        const userId = mastraMetadata['userId'];\n        if (userId) {\n          delete mastraMetadata['userId'];\n          processedAttributes[USER_ID] = userId;\n        }\n\n        // Gather custom metadata into OpenInference metadata (flat best-effort)\n        if (Object.keys(mastraMetadata).length > 0) {\n          try {\n            processedAttributes[METADATA] = JSON.stringify(mastraMetadata);\n          } catch {\n            // best-effort only\n          }\n        }\n\n        const inputMessages = attributes[ATTR_GEN_AI_INPUT_MESSAGES];\n        if (inputMessages) {\n          processedAttributes[INPUT_MIME_TYPE] = 'application/json';\n          processedAttributes[INPUT_VALUE] = inputMessages;\n        }\n        const outputMessages = attributes[ATTR_GEN_AI_OUTPUT_MESSAGES];\n        if (outputMessages) {\n          processedAttributes[OUTPUT_MIME_TYPE] = 'application/json';\n          processedAttributes[OUTPUT_VALUE] = outputMessages;\n        }\n\n        mutableSpan.attributes = { ...processedAttributes, ...mastraOther };\n      }\n\n      return mutableSpan;\n    });\n\n    super.export(processedSpans, resultCallback);\n  }\n}\n","import { SEMRESATTRS_PROJECT_NAME } from '@arizeai/openinference-semantic-conventions';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { OtelExporter } from '@mastra/otel-exporter';\nimport type { OtelExporterConfig } from '@mastra/otel-exporter';\n\nimport { OpenInferenceOTLPTraceExporter } from './openInferenceOTLPExporter.js';\n\nconst LOG_PREFIX = '[ArizeExporter]';\n\nexport const ARIZE_AX_ENDPOINT = 'https://otlp.arize.com/v1/traces';\n\nexport type ArizeExporterConfig = Omit<OtelExporterConfig, 'provider'> & {\n  /**\n   * Required if sending traces to Arize AX\n   */\n  spaceId?: string;\n  /**\n   * Required if sending traces to Arize AX, or to any other collector that\n   * requires an Authorization header\n   */\n  apiKey?: string;\n  /**\n   * Collector endpoint destination for trace exports.\n   * Required when sending traces to Phoenix, Phoenix Cloud, or other collectors.\n   * Optional when sending traces to Arize AX.\n   */\n  endpoint?: string;\n  /**\n   * Optional project name to be added as a resource attribute using\n   * OpenInference Semantic Conventions\n   */\n  projectName?: string;\n  /**\n   * Optional headers to be added to each OTLP request\n   */\n  headers?: Record<string, string>;\n};\n\nexport class ArizeExporter extends OtelExporter {\n  name = 'arize';\n\n  constructor(config: ArizeExporterConfig) {\n    const logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n    let endpoint: string | undefined = config.endpoint;\n    const headers: Record<string, string> = {\n      ...config.headers,\n    };\n    if (config.spaceId) {\n      // arize ax header configuration\n      headers['space_id'] = config.spaceId;\n      headers['api_key'] = config.apiKey ?? '';\n      endpoint = config.endpoint || ARIZE_AX_ENDPOINT;\n    } else if (config.apiKey) {\n      // standard otel header configuration\n      headers['Authorization'] = `Bearer ${config.apiKey}`;\n    }\n    if (!endpoint) {\n      logger.error(`${LOG_PREFIX} Endpoint is required in configuration. Disabling exporter.`);\n      return;\n    }\n    super({\n      exporter: new OpenInferenceOTLPTraceExporter({\n        url: endpoint,\n        headers,\n      }),\n      ...config,\n      resourceAttributes: {\n        [SEMRESATTRS_PROJECT_NAME]: config.projectName,\n        ...config.resourceAttributes,\n      },\n      provider: {\n        custom: {\n          endpoint,\n          headers,\n          protocol: 'http/protobuf',\n        },\n      } satisfies OtelExporterConfig['provider'],\n    } satisfies OtelExporterConfig);\n  }\n}\n"]}
+{"version":3,"sources":["../src/openInferenceOTLPExporter.ts","../src/tracing.ts"],"names":[],"mappings":";;;;;;;;AA+BA,IAAM,6BAAA,GAAgC,+BAAA;AACtC,IAAM,gCAAA,GAAmC,kCAAA;AACzC,IAAM,+BAAA,GAAkC,iCAAA;AACxC,IAAM,+BAAA,GAAkC,iCAAA;AACxC,IAAM,gCAAA,GAAmC,kCAAA;AAEzC,IAAM,qBAAA,GAAwB,SAAA;AAC9B,IAAM,sBAAA,GAAyB,kBAAA;AAS/B,SAAS,mCAAmC,UAAA,EAAsD;AAChG,EAAA,MAAM,SAA8B,EAAC;AAErC,EAAA,MAAM,WAAA,GAAc,WAAW,8BAA8B,CAAA;AAC7D,EAAA,MAAM,YAAA,GAAe,WAAW,+BAA+B,CAAA;AAG/D,EAAA,IAAI,gBAAgB,MAAA,EAAW;AAC7B,IAAA,MAAA,CAAO,sBAAsB,CAAA,GAAI,WAAA;AAAA,EACnC;AACA,EAAA,IAAI,iBAAiB,MAAA,EAAW;AAC9B,IAAA,MAAA,CAAO,0BAA0B,CAAA,GAAI,YAAA;AAAA,EACvC;AAGA,EAAA,IAAI,WAAA,KAAgB,MAAA,IAAa,YAAA,KAAiB,MAAA,EAAW;AAC3D,IAAA,MAAA,CAAO,qBAAqB,IAAI,WAAA,GAAc,YAAA;AAAA,EAChD;AAGA,EAAA,MAAM,iBAAA,GAAoB,WAAW,gCAAgC,CAAA;AACrE,EAAA,IAAI,sBAAsB,MAAA,EAAW;AACnC,IAAA,MAAA,CAAO,yCAAyC,CAAA,GAAI,iBAAA;AAAA,EACtD;AAEA,EAAA,MAAM,gBAAA,GAAmB,WAAW,+BAA+B,CAAA;AACnE,EAAA,IAAI,qBAAqB,MAAA,EAAW;AAClC,IAAA,MAAA,CAAO,0CAA0C,CAAA,GAAI,gBAAA;AAAA,EACvD;AAGA,EAAA,MAAM,eAAA,GAAkB,WAAW,6BAA6B,CAAA;AAChE,EAAA,IAAI,oBAAoB,MAAA,EAAW;AACjC,IAAA,MAAA,CAAO,4CAA4C,CAAA,GAAI,eAAA;AAAA,EACzD;AAGA,EAAA,MAAM,gBAAA,GAAmB,WAAW,+BAA+B,CAAA;AACnE,EAAA,IAAI,qBAAqB,MAAA,EAAW;AAClC,IAAA,MAAA,CAAO,oCAAoC,CAAA,GAAI,gBAAA;AAAA,EACjD;AAEA,EAAA,MAAM,iBAAA,GAAoB,WAAW,gCAAgC,CAAA;AACrE,EAAA,IAAI,sBAAsB,MAAA,EAAW;AACnC,IAAA,MAAA,CAAO,wCAAwC,CAAA,GAAI,iBAAA;AAAA,EACrD;AAEA,EAAA,OAAO,MAAA;AACT;AASA,SAAS,sBAAsB,UAAA,EAG7B;AACA,EAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,UAAU,CAAA,CAAE,MAAA;AAAA,IAChC,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,KAAK,CAAA,KAAM;AACrB,MAAA,IAAI,GAAA,CAAI,UAAA,CAAW,qBAAqB,CAAA,EAAG;AACzC,QAAA,IAAI,GAAA,CAAI,UAAA,CAAW,sBAAsB,CAAA,EAAG;AAC1C,UAAA,MAAM,WAAA,GAAc,GAAA,CAAI,KAAA,CAAM,sBAAA,CAAuB,MAAM,CAAA;AAC3D,UAAA,GAAA,CAAI,cAAA,CAAe,WAAW,CAAA,GAAI,KAAA;AAAA,QACpC,CAAA,MAAO;AACL,UAAA,GAAA,CAAI,WAAA,CAAY,GAAG,CAAA,GAAI,KAAA;AAAA,QACzB;AAAA,MACF;AACA,MAAA,OAAO,GAAA;AAAA,IACT,CAAA;AAAA,IACA;AAAA,MACE,gBAAgB,EAAC;AAAA,MACjB,aAAa;AAAC;AAChB,GACF;AACF;AAEO,IAAM,8BAAA,GAAN,cAA6C,iBAAA,CAAkB;AAAA,EACpE,MAAA,CAAO,OAAuB,cAAA,EAAgD;AAC5E,IAAA,MAAM,cAAA,GAAiB,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AACvC,MAAA,MAAM,aAAa,EAAE,GAAI,IAAA,CAAK,UAAA,IAAc,EAAC,EAAG;AAChD,MAAA,MAAM,WAAA,GAAc,IAAA;AAEpB,MAAA,MAAM,EAAE,cAAA,EAAgB,WAAA,EAAY,GAAI,sBAAsB,UAAU,CAAA;AACxE,MAAA,MAAM,mBAAA,GAAsB,wDAAwD,UAAU,CAAA;AAG9F,MAAA,IAAI,mBAAA,EAAqB;AACvB,QAAA,MAAM,QAAA,GAAW,eAAe,UAAU,CAAA;AAC1C,QAAA,IAAI,QAAA,EAAU;AACZ,UAAA,OAAO,eAAe,UAAU,CAAA;AAChC,UAAA,mBAAA,CAAoB,UAAU,CAAA,GAAI,QAAA;AAAA,QACpC;AAGA,QAAA,IAAI,WAAA,CAAY,aAAa,CAAA,EAAG;AAC9B,UAAA,mBAAA,CAAoB,QAAQ,CAAA,GAAI,WAAA,CAAY,aAAa,CAAA;AACzD,UAAA,OAAO,YAAY,aAAa,CAAA;AAAA,QAClC;AAEA,QAAA,MAAM,MAAA,GAAS,eAAe,QAAQ,CAAA;AACtC,QAAA,IAAI,MAAA,EAAQ;AACV,UAAA,OAAO,eAAe,QAAQ,CAAA;AAC9B,UAAA,mBAAA,CAAoB,OAAO,CAAA,GAAI,MAAA;AAAA,QACjC;AAGA,QAAA,IAAI,MAAA,CAAO,IAAA,CAAK,cAAc,CAAA,CAAE,SAAS,CAAA,EAAG;AAC1C,UAAA,IAAI;AACF,YAAA,mBAAA,CAAoB,QAAQ,CAAA,GAAI,IAAA,CAAK,SAAA,CAAU,cAAc,CAAA;AAAA,UAC/D,CAAA,CAAA,MAAQ;AAAA,UAER;AAAA,QACF;AAEA,QAAA,MAAM,aAAA,GAAgB,WAAW,0BAA0B,CAAA;AAC3D,QAAA,IAAI,aAAA,EAAe;AACjB,UAAA,mBAAA,CAAoB,eAAe,CAAA,GAAI,kBAAA;AACvC,UAAA,mBAAA,CAAoB,WAAW,CAAA,GAAI,aAAA;AAAA,QACrC;AACA,QAAA,MAAM,cAAA,GAAiB,WAAW,2BAA2B,CAAA;AAC7D,QAAA,IAAI,cAAA,EAAgB;AAClB,UAAA,mBAAA,CAAoB,gBAAgB,CAAA,GAAI,kBAAA;AACxC,UAAA,mBAAA,CAAoB,YAAY,CAAA,GAAI,cAAA;AAAA,QACtC;AAGA,QAAA,MAAM,YAAA,GAAe,mCAAmC,UAAU,CAAA;AAClE,QAAA,MAAA,CAAO,MAAA,CAAO,qBAAqB,YAAY,CAAA;AAE/C,QAAA,WAAA,CAAY,UAAA,GAAa,EAAE,GAAG,mBAAA,EAAqB,GAAG,WAAA,EAAY;AAAA,MACpE;AAEA,MAAA,OAAO,WAAA;AAAA,IACT,CAAC,CAAA;AAED,IAAA,KAAA,CAAM,MAAA,CAAO,gBAAgB,cAAc,CAAA;AAAA,EAC7C;AACF,CAAA;;;ACrLA,IAAM,UAAA,GAAa,iBAAA;AAEZ,IAAM,iBAAA,GAAoB;AA6B1B,IAAM,aAAA,GAAN,cAA4B,YAAA,CAAa;AAAA,EAC9C,IAAA,GAAO,OAAA;AAAA,EAEP,YAAY,MAAA,EAA6B;AACvC,IAAA,MAAM,MAAA,GAAS,IAAI,aAAA,CAAc,EAAE,OAAO,MAAA,CAAO,QAAA,IAAY,QAAQ,CAAA;AACrE,IAAA,IAAI,WAA+B,MAAA,CAAO,QAAA;AAC1C,IAAA,MAAM,OAAA,GAAkC;AAAA,MACtC,GAAG,MAAA,CAAO;AAAA,KACZ;AACA,IAAA,IAAI,OAAO,OAAA,EAAS;AAElB,MAAA,OAAA,CAAQ,UAAU,IAAI,MAAA,CAAO,OAAA;AAC7B,MAAA,OAAA,CAAQ,SAAS,CAAA,GAAI,MAAA,CAAO,MAAA,IAAU,EAAA;AACtC,MAAA,QAAA,GAAW,OAAO,QAAA,IAAY,iBAAA;AAAA,IAChC,CAAA,MAAA,IAAW,OAAO,MAAA,EAAQ;AAExB,MAAA,OAAA,CAAQ,eAAe,CAAA,GAAI,CAAA,OAAA,EAAU,MAAA,CAAO,MAAM,CAAA,CAAA;AAAA,IACpD;AACA,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,MAAA,CAAO,KAAA,CAAM,CAAA,EAAG,UAAU,CAAA,2DAAA,CAA6D,CAAA;AACvF,MAAA;AAAA,IACF;AACA,IAAA,KAAA,CAAM;AAAA,MACJ,QAAA,EAAU,IAAI,8BAAA,CAA+B;AAAA,QAC3C,GAAA,EAAK,QAAA;AAAA,QACL;AAAA,OACD,CAAA;AAAA,MACD,GAAG,MAAA;AAAA,MACH,kBAAA,EAAoB;AAAA,QAClB,CAAC,wBAAwB,GAAG,MAAA,CAAO,WAAA;AAAA,QACnC,GAAG,MAAA,CAAO;AAAA,OACZ;AAAA,MACA,QAAA,EAAU;AAAA,QACR,MAAA,EAAQ;AAAA,UACN,QAAA;AAAA,UACA,OAAA;AAAA,UACA,QAAA,EAAU;AAAA;AACZ;AACF,KAC4B,CAAA;AAAA,EAChC;AACF","file":"index.js","sourcesContent":["import { convertGenAISpanAttributesToOpenInferenceSpanAttributes } from '@arizeai/openinference-genai';\nimport type { Mutable } from '@arizeai/openinference-genai/types';\nimport {\n  INPUT_MIME_TYPE,\n  INPUT_VALUE,\n  LLM_TOKEN_COUNT_COMPLETION,\n  LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO,\n  LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING,\n  LLM_TOKEN_COUNT_PROMPT,\n  LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO,\n  LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ,\n  LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_WRITE,\n  LLM_TOKEN_COUNT_TOTAL,\n  METADATA,\n  OUTPUT_MIME_TYPE,\n  OUTPUT_VALUE,\n  SESSION_ID,\n  TAG_TAGS,\n  USER_ID,\n} from '@arizeai/openinference-semantic-conventions';\nimport type { ExportResult } from '@opentelemetry/core';\nimport { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-proto';\nimport type { ReadableSpan } from '@opentelemetry/sdk-trace-base';\nimport {\n  ATTR_GEN_AI_INPUT_MESSAGES,\n  ATTR_GEN_AI_OUTPUT_MESSAGES,\n  ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n  ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n} from '@opentelemetry/semantic-conventions/incubating';\n\n// GenAI usage attribute keys (not all are in @opentelemetry/semantic-conventions yet)\nconst GEN_AI_USAGE_REASONING_TOKENS = 'gen_ai.usage.reasoning_tokens';\nconst GEN_AI_USAGE_CACHED_INPUT_TOKENS = 'gen_ai.usage.cached_input_tokens';\nconst GEN_AI_USAGE_CACHE_WRITE_TOKENS = 'gen_ai.usage.cache_write_tokens';\nconst GEN_AI_USAGE_AUDIO_INPUT_TOKENS = 'gen_ai.usage.audio_input_tokens';\nconst GEN_AI_USAGE_AUDIO_OUTPUT_TOKENS = 'gen_ai.usage.audio_output_tokens';\n\nconst MASTRA_GENERAL_PREFIX = 'mastra.';\nconst MASTRA_METADATA_PREFIX = 'mastra.metadata.';\n\n/**\n * Converts GenAI usage metrics to OpenInference LLM token count attributes.\n * Maps from OTEL GenAI semantic conventions to OpenInference semantic conventions.\n *\n * @param attributes - The span attributes containing GenAI usage metrics\n * @returns OpenInference token count attributes\n */\nfunction convertUsageMetricsToOpenInference(attributes: Record<string, any>): Record<string, any> {\n  const result: Record<string, any> = {};\n\n  const inputTokens = attributes[ATTR_GEN_AI_USAGE_INPUT_TOKENS];\n  const outputTokens = attributes[ATTR_GEN_AI_USAGE_OUTPUT_TOKENS];\n\n  // Core token counts\n  if (inputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_PROMPT] = inputTokens;\n  }\n  if (outputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_COMPLETION] = outputTokens;\n  }\n\n  // Total tokens (compute if we have both input and output)\n  if (inputTokens !== undefined && outputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_TOTAL] = inputTokens + outputTokens;\n  }\n\n  // Cache tokens (prompt details)\n  const cachedInputTokens = attributes[GEN_AI_USAGE_CACHED_INPUT_TOKENS];\n  if (cachedInputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ] = cachedInputTokens;\n  }\n\n  const cacheWriteTokens = attributes[GEN_AI_USAGE_CACHE_WRITE_TOKENS];\n  if (cacheWriteTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_WRITE] = cacheWriteTokens;\n  }\n\n  // Reasoning tokens (completion details)\n  const reasoningTokens = attributes[GEN_AI_USAGE_REASONING_TOKENS];\n  if (reasoningTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING] = reasoningTokens;\n  }\n\n  // Audio tokens\n  const audioInputTokens = attributes[GEN_AI_USAGE_AUDIO_INPUT_TOKENS];\n  if (audioInputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_PROMPT_DETAILS_AUDIO] = audioInputTokens;\n  }\n\n  const audioOutputTokens = attributes[GEN_AI_USAGE_AUDIO_OUTPUT_TOKENS];\n  if (audioOutputTokens !== undefined) {\n    result[LLM_TOKEN_COUNT_COMPLETION_DETAILS_AUDIO] = audioOutputTokens;\n  }\n\n  return result;\n}\n\n/**\n * Splits Mastra span attributes into two groups:\n * - `metadata`: keys starting with \"mastra.metadata.\" (prefix removed)\n * - `other`: all remaining keys starting with \"mastra.\"\n *\n * Any attributes not starting with \"mastra.\" are ignored entirely.\n */\nfunction splitMastraAttributes(attributes: Record<string, any>): {\n  mastraMetadata: Record<string, any>;\n  mastraOther: Record<string, any>;\n} {\n  return Object.entries(attributes).reduce(\n    (acc, [key, value]) => {\n      if (key.startsWith(MASTRA_GENERAL_PREFIX)) {\n        if (key.startsWith(MASTRA_METADATA_PREFIX)) {\n          const strippedKey = key.slice(MASTRA_METADATA_PREFIX.length);\n          acc.mastraMetadata[strippedKey] = value;\n        } else {\n          acc.mastraOther[key] = value;\n        }\n      }\n      return acc;\n    },\n    {\n      mastraMetadata: {} as Record<string, any>,\n      mastraOther: {} as Record<string, any>,\n    },\n  );\n}\n\nexport class OpenInferenceOTLPTraceExporter extends OTLPTraceExporter {\n  export(spans: ReadableSpan[], resultCallback: (result: ExportResult) => void) {\n    const processedSpans = spans.map(span => {\n      const attributes = { ...(span.attributes ?? {}) };\n      const mutableSpan = span as Mutable<ReadableSpan>;\n\n      const { mastraMetadata, mastraOther } = splitMastraAttributes(attributes);\n      const processedAttributes = convertGenAISpanAttributesToOpenInferenceSpanAttributes(attributes);\n\n      // only add processed attributes if conversion was successful\n      if (processedAttributes) {\n        const threadId = mastraMetadata['threadId'];\n        if (threadId) {\n          delete mastraMetadata['threadId'];\n          processedAttributes[SESSION_ID] = threadId;\n        }\n\n        // Map mastra.tags to OpenInference native tag.tags convention (tags are only on root spans)\n        if (mastraOther['mastra.tags']) {\n          processedAttributes[TAG_TAGS] = mastraOther['mastra.tags'];\n          delete mastraOther['mastra.tags'];\n        }\n\n        const userId = mastraMetadata['userId'];\n        if (userId) {\n          delete mastraMetadata['userId'];\n          processedAttributes[USER_ID] = userId;\n        }\n\n        // Gather custom metadata into OpenInference metadata (flat best-effort)\n        if (Object.keys(mastraMetadata).length > 0) {\n          try {\n            processedAttributes[METADATA] = JSON.stringify(mastraMetadata);\n          } catch {\n            // best-effort only\n          }\n        }\n\n        const inputMessages = attributes[ATTR_GEN_AI_INPUT_MESSAGES];\n        if (inputMessages) {\n          processedAttributes[INPUT_MIME_TYPE] = 'application/json';\n          processedAttributes[INPUT_VALUE] = inputMessages;\n        }\n        const outputMessages = attributes[ATTR_GEN_AI_OUTPUT_MESSAGES];\n        if (outputMessages) {\n          processedAttributes[OUTPUT_MIME_TYPE] = 'application/json';\n          processedAttributes[OUTPUT_VALUE] = outputMessages;\n        }\n\n        // Convert GenAI usage metrics to OpenInference token count attributes\n        const usageMetrics = convertUsageMetricsToOpenInference(attributes);\n        Object.assign(processedAttributes, usageMetrics);\n\n        mutableSpan.attributes = { ...processedAttributes, ...mastraOther };\n      }\n\n      return mutableSpan;\n    });\n\n    super.export(processedSpans, resultCallback);\n  }\n}\n","import { SEMRESATTRS_PROJECT_NAME } from '@arizeai/openinference-semantic-conventions';\nimport { ConsoleLogger } from '@mastra/core/logger';\nimport { OtelExporter } from '@mastra/otel-exporter';\nimport type { OtelExporterConfig } from '@mastra/otel-exporter';\n\nimport { OpenInferenceOTLPTraceExporter } from './openInferenceOTLPExporter.js';\n\nconst LOG_PREFIX = '[ArizeExporter]';\n\nexport const ARIZE_AX_ENDPOINT = 'https://otlp.arize.com/v1/traces';\n\nexport type ArizeExporterConfig = Omit<OtelExporterConfig, 'provider'> & {\n  /**\n   * Required if sending traces to Arize AX\n   */\n  spaceId?: string;\n  /**\n   * Required if sending traces to Arize AX, or to any other collector that\n   * requires an Authorization header\n   */\n  apiKey?: string;\n  /**\n   * Collector endpoint destination for trace exports.\n   * Required when sending traces to Phoenix, Phoenix Cloud, or other collectors.\n   * Optional when sending traces to Arize AX.\n   */\n  endpoint?: string;\n  /**\n   * Optional project name to be added as a resource attribute using\n   * OpenInference Semantic Conventions\n   */\n  projectName?: string;\n  /**\n   * Optional headers to be added to each OTLP request\n   */\n  headers?: Record<string, string>;\n};\n\nexport class ArizeExporter extends OtelExporter {\n  name = 'arize';\n\n  constructor(config: ArizeExporterConfig) {\n    const logger = new ConsoleLogger({ level: config.logLevel ?? 'warn' });\n    let endpoint: string | undefined = config.endpoint;\n    const headers: Record<string, string> = {\n      ...config.headers,\n    };\n    if (config.spaceId) {\n      // arize ax header configuration\n      headers['space_id'] = config.spaceId;\n      headers['api_key'] = config.apiKey ?? '';\n      endpoint = config.endpoint || ARIZE_AX_ENDPOINT;\n    } else if (config.apiKey) {\n      // standard otel header configuration\n      headers['Authorization'] = `Bearer ${config.apiKey}`;\n    }\n    if (!endpoint) {\n      logger.error(`${LOG_PREFIX} Endpoint is required in configuration. Disabling exporter.`);\n      return;\n    }\n    super({\n      exporter: new OpenInferenceOTLPTraceExporter({\n        url: endpoint,\n        headers,\n      }),\n      ...config,\n      resourceAttributes: {\n        [SEMRESATTRS_PROJECT_NAME]: config.projectName,\n        ...config.resourceAttributes,\n      },\n      provider: {\n        custom: {\n          endpoint,\n          headers,\n          protocol: 'http/protobuf',\n        },\n      } satisfies OtelExporterConfig['provider'],\n    } satisfies OtelExporterConfig);\n  }\n}\n"]}

package/dist/openInferenceOTLPExporter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"openInferenceOTLPExporter.d.ts","sourceRoot":"","sources":["../src/openInferenceOTLPExporter.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,0CAA0C,CAAC;AAC7E,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAuClE,qBAAa,8BAA+B,SAAQ,iBAAiB;IACnE,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,cAAc,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,IAAI;CAwD7E"}
+{"version":3,"file":"openInferenceOTLPExporter.d.ts","sourceRoot":"","sources":["../src/openInferenceOTLPExporter.ts"],"names":[],"mappings":"AAoBA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,0CAA0C,CAAC;AAC7E,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAyGlE,qBAAa,8BAA+B,SAAQ,iBAAiB;IACnE,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,cAAc,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,IAAI;CA4D7E"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/arize",
-  "version": "1.0.0-beta.4",
+  "version": "1.0.0-beta.5",
   "description": "Arize observability provider for Mastra - includes tracing and future observability features",
   "type": "module",
   "main": "dist/index.js",
@@ -31,7 +31,7 @@
     "@opentelemetry/resources": "^2.1.0",
     "@opentelemetry/sdk-trace-base": "^2.1.0",
     "@opentelemetry/semantic-conventions": "1.38.0",
-    "@mastra/otel-exporter": "1.0.0-beta.4"
+    "@mastra/otel-exporter": "1.0.0-beta.5"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.8",
@@ -42,9 +42,9 @@
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
     "vitest": "4.0.12",
-    "@internal/types-builder": "0.0.28",
     "@internal/lint": "0.0.53",
-    "@mastra/core": "1.0.0-beta.9"
+    "@internal/types-builder": "0.0.28",
+    "@mastra/core": "1.0.0-beta.10"
   },
   "peerDependencies": {
     "@mastra/core": ">=1.0.0-0 <2.0.0-0"