@mastra/posthog 0.0.0-bundle-studio-cloud-20251222034739

package/dist/index.js

@@ -0,0 +1,347 @@
+import { SpanType } from '@mastra/core/observability';
+import { BaseExporter } from '@mastra/observability';
+import { PostHog } from 'posthog-node';
+
+// src/tracing.ts
+function formatUsageMetrics(usage) {
+  if (!usage) return {};
+  const props = {};
+  if (usage.inputTokens !== void 0) props.$ai_input_tokens = usage.inputTokens;
+  if (usage.outputTokens !== void 0) props.$ai_output_tokens = usage.outputTokens;
+  if (usage.inputDetails?.cacheRead !== void 0) props.$ai_cache_read_input_tokens = usage.inputDetails.cacheRead;
+  if (usage.inputDetails?.cacheWrite !== void 0)
+    props.$ai_cache_creation_input_tokens = usage.inputDetails.cacheWrite;
+  return props;
+}
+var PosthogExporter = class _PosthogExporter extends BaseExporter {
+  name = "posthog";
+  client;
+  config;
+  traceMap = /* @__PURE__ */ new Map();
+  static SERVERLESS_FLUSH_AT = 10;
+  static SERVERLESS_FLUSH_INTERVAL = 2e3;
+  static DEFAULT_FLUSH_AT = 20;
+  static DEFAULT_FLUSH_INTERVAL = 1e4;
+  constructor(config) {
+    super(config);
+    this.config = config;
+    if (!config.apiKey) {
+      this.setDisabled("Missing required API key");
+      this.client = null;
+      return;
+    }
+    const clientConfig = this.buildClientConfig(config);
+    this.client = new PostHog(config.apiKey, clientConfig);
+    this.logInitialization(config.serverless ?? false, clientConfig);
+  }
+  buildClientConfig(config) {
+    const isServerless = config.serverless ?? false;
+    const flushAt = config.flushAt ?? (isServerless ? _PosthogExporter.SERVERLESS_FLUSH_AT : _PosthogExporter.DEFAULT_FLUSH_AT);
+    const flushInterval = config.flushInterval ?? (isServerless ? _PosthogExporter.SERVERLESS_FLUSH_INTERVAL : _PosthogExporter.DEFAULT_FLUSH_INTERVAL);
+    const host = config.host || process.env.POSTHOG_HOST || "https://us.i.posthog.com";
+    if (!config.host && !process.env.POSTHOG_HOST) {
+      this.logger.warn(
+        'No PostHog host specified, using US default (https://us.i.posthog.com). For EU region, set `host: "https://eu.i.posthog.com"` in config or POSTHOG_HOST env var. For self-hosted, provide your instance URL.'
+      );
+    }
+    return {
+      host,
+      flushAt,
+      flushInterval,
+      privacyMode: config.enablePrivacyMode
+    };
+  }
+  logInitialization(isServerless, config) {
+    const message = isServerless ? "PostHog exporter initialized in serverless mode" : "PostHog exporter initialized";
+    this.logger.debug(message, config);
+  }
+  async _exportTracingEvent(event) {
+    if (!this.client) {
+      return;
+    }
+    try {
+      if (event.exportedSpan.isEvent) {
+        if (event.type === "span_started") {
+          await this.captureEventSpan(event.exportedSpan);
+        }
+        return;
+      }
+      switch (event.type) {
+        case "span_started":
+          await this.handleSpanStarted(event.exportedSpan);
+          break;
+        case "span_updated":
+          break;
+        case "span_ended":
+          await this.handleSpanEnded(event.exportedSpan);
+          break;
+      }
+    } catch (error) {
+      this.logger.error("PostHog exporter error", { error, event });
+    }
+  }
+  async handleSpanStarted(span) {
+    let traceData = this.traceMap.get(span.traceId);
+    if (!traceData) {
+      traceData = {
+        spans: /* @__PURE__ */ new Map(),
+        distinctId: void 0
+      };
+      this.traceMap.set(span.traceId, traceData);
+    }
+    traceData.spans.set(span.id, {
+      startTime: this.toDate(span.startTime),
+      type: span.type,
+      isRootSpan: span.isRootSpan
+    });
+    if (!traceData.distinctId) {
+      const userId = span.metadata?.userId;
+      if (userId) {
+        traceData.distinctId = String(userId);
+      }
+    }
+  }
+  async handleSpanEnded(span) {
+    const traceData = this.traceMap.get(span.traceId);
+    if (!traceData) {
+      return;
+    }
+    const cachedSpan = traceData.spans.get(span.id);
+    if (!cachedSpan) {
+      return;
+    }
+    const startTime = cachedSpan.startTime.getTime();
+    const endTime = span.endTime ? this.toDate(span.endTime).getTime() : Date.now();
+    const latency = (endTime - startTime) / 1e3;
+    const distinctId = this.getDistinctId(span, traceData);
+    if (span.isRootSpan) {
+      this.captureTraceEvent(span, distinctId, endTime);
+    } else {
+      const eventName = this.mapToPostHogEvent(span.type);
+      const parentIsRootSpan = this.isParentRootSpan(span, traceData);
+      const properties = this.buildEventProperties(span, latency, parentIsRootSpan);
+      this.client.capture({
+        distinctId,
+        event: eventName,
+        properties,
+        timestamp: new Date(endTime)
+      });
+    }
+    traceData.spans.delete(span.id);
+    if (traceData.spans.size === 0) {
+      this.traceMap.delete(span.traceId);
+    }
+  }
+  /**
+   * Capture an explicit $ai_trace event for root spans.
+   * This gives us control over trace-level metadata like name and tags,
+   * rather than relying on PostHog's pseudo-trace auto-creation.
+   */
+  captureTraceEvent(span, distinctId, endTime) {
+    const traceProperties = {
+      $ai_trace_id: span.traceId,
+      $ai_span_name: span.name,
+      $ai_is_error: !!span.errorInfo
+    };
+    if (span.metadata?.sessionId) {
+      traceProperties.$ai_session_id = span.metadata.sessionId;
+    }
+    if (span.input) {
+      traceProperties.$ai_input_state = span.input;
+    }
+    if (span.output) {
+      traceProperties.$ai_output_state = span.output;
+    }
+    if (span.errorInfo) {
+      traceProperties.$ai_error = {
+        message: span.errorInfo.message,
+        ...span.errorInfo.id && { id: span.errorInfo.id },
+        ...span.errorInfo.category && { category: span.errorInfo.category }
+      };
+    }
+    if (span.tags?.length) {
+      for (const tag of span.tags) {
+        traceProperties[tag] = true;
+      }
+    }
+    const { userId, sessionId, ...customMetadata } = span.metadata ?? {};
+    Object.assign(traceProperties, customMetadata);
+    this.client.capture({
+      distinctId,
+      event: "$ai_trace",
+      properties: traceProperties,
+      timestamp: new Date(endTime)
+    });
+  }
+  async captureEventSpan(span) {
+    const eventName = this.mapToPostHogEvent(span.type);
+    const traceData = this.traceMap.get(span.traceId);
+    const distinctId = this.getDistinctId(span, traceData);
+    const properties = this.buildEventProperties(span, 0);
+    this.client.capture({
+      distinctId,
+      event: eventName,
+      properties,
+      timestamp: span.endTime ? new Date(span.endTime) : /* @__PURE__ */ new Date()
+    });
+  }
+  async shutdown() {
+    if (this.client) {
+      await this.client.shutdown();
+    }
+    this.traceMap.clear();
+    await super.shutdown();
+    this.logger.info("PostHog exporter shutdown complete");
+  }
+  toDate(timestamp) {
+    return timestamp instanceof Date ? timestamp : new Date(timestamp);
+  }
+  mapToPostHogEvent(spanType) {
+    if (spanType == SpanType.MODEL_GENERATION) {
+      return "$ai_generation";
+    }
+    return "$ai_span";
+  }
+  getDistinctId(span, traceData) {
+    if (span.metadata?.userId) {
+      return String(span.metadata.userId);
+    }
+    if (traceData?.distinctId) {
+      return traceData.distinctId;
+    }
+    if (this.config.defaultDistinctId) {
+      return this.config.defaultDistinctId;
+    }
+    return "anonymous";
+  }
+  /**
+   * Check if the parent of this span is the root span.
+   * We need this because we don't create $ai_span for root spans,
+   * so children of root spans should use $ai_trace_id as their $ai_parent_id.
+   */
+  isParentRootSpan(span, traceData) {
+    if (!span.parentSpanId) {
+      return false;
+    }
+    const parentCache = traceData.spans.get(span.parentSpanId);
+    if (parentCache) {
+      return parentCache.isRootSpan;
+    }
+    return false;
+  }
+  buildEventProperties(span, latency, parentIsRootSpan = false) {
+    const baseProperties = {
+      $ai_trace_id: span.traceId,
+      $ai_latency: latency,
+      $ai_is_error: !!span.errorInfo
+    };
+    if (span.parentSpanId) {
+      baseProperties.$ai_parent_id = parentIsRootSpan ? span.traceId : span.parentSpanId;
+    }
+    if (span.metadata?.sessionId) {
+      baseProperties.$ai_session_id = span.metadata.sessionId;
+    }
+    if (span.isRootSpan && span.tags?.length) {
+      for (const tag of span.tags) {
+        baseProperties[tag] = true;
+      }
+    }
+    if (span.type === SpanType.MODEL_GENERATION) {
+      baseProperties.$ai_generation_id = span.id;
+      return { ...baseProperties, ...this.buildGenerationProperties(span) };
+    } else {
+      baseProperties.$ai_span_id = span.id;
+      baseProperties.$ai_span_name = span.name;
+      return { ...baseProperties, ...this.buildSpanProperties(span) };
+    }
+  }
+  extractErrorProperties(span) {
+    if (!span.errorInfo) {
+      return {};
+    }
+    const props = {
+      error_message: span.errorInfo.message
+    };
+    if (span.errorInfo.id) {
+      props.error_id = span.errorInfo.id;
+    }
+    if (span.errorInfo.category) {
+      props.error_category = span.errorInfo.category;
+    }
+    return props;
+  }
+  extractCustomMetadata(span) {
+    const { userId, sessionId, ...customMetadata } = span.metadata ?? {};
+    return customMetadata;
+  }
+  buildGenerationProperties(span) {
+    const props = {};
+    const attrs = span.attributes ?? {};
+    props.$ai_model = attrs.model || "unknown-model";
+    props.$ai_provider = attrs.provider || "unknown-provider";
+    if (span.input) props.$ai_input = this.formatMessages(span.input, "user");
+    if (span.output) props.$ai_output_choices = this.formatMessages(span.output, "assistant");
+    Object.assign(props, formatUsageMetrics(attrs.usage));
+    if (attrs.parameters) {
+      if (attrs.parameters.temperature !== void 0) props.$ai_temperature = attrs.parameters.temperature;
+      if (attrs.parameters.maxOutputTokens !== void 0) props.$ai_max_tokens = attrs.parameters.maxOutputTokens;
+    }
+    if (attrs.streaming !== void 0) props.$ai_stream = attrs.streaming;
+    return { ...props, ...this.extractErrorProperties(span), ...this.extractCustomMetadata(span) };
+  }
+  buildSpanProperties(span) {
+    const props = {};
+    if (span.input) props.$ai_input_state = span.input;
+    if (span.output) props.$ai_output_state = span.output;
+    if (span.type === SpanType.MODEL_CHUNK) {
+      const attrs = span.attributes;
+      if (attrs?.chunkType) props.chunk_type = attrs.chunkType;
+      if (attrs?.sequenceNumber !== void 0) props.chunk_sequence_number = attrs.sequenceNumber;
+    }
+    if (span.attributes) {
+      Object.assign(props, span.attributes);
+    }
+    return { ...props, ...this.extractErrorProperties(span), ...this.extractCustomMetadata(span) };
+  }
+  formatMessages(data, defaultRole = "user") {
+    if (this.isMessageArray(data)) {
+      return data.map((msg) => this.normalizeMessage(msg));
+    }
+    if (typeof data === "string") {
+      return [{ role: defaultRole, content: [{ type: "text", text: data }] }];
+    }
+    return [{ role: defaultRole, content: [{ type: "text", text: this.safeStringify(data) }] }];
+  }
+  isMessageArray(data) {
+    if (!Array.isArray(data) || data.length === 0) {
+      return false;
+    }
+    return data.every((item) => typeof item === "object" && item !== null && "role" in item && "content" in item);
+  }
+  normalizeMessage(msg) {
+    if (typeof msg.content === "string") {
+      return {
+        role: msg.role,
+        content: [{ type: "text", text: msg.content }]
+      };
+    }
+    return {
+      role: msg.role,
+      content: msg.content
+    };
+  }
+  safeStringify(data) {
+    try {
+      return JSON.stringify(data);
+    } catch {
+      if (typeof data === "object" && data !== null) {
+        return `[Non-serializable ${data.constructor?.name || "Object"}]`;
+      }
+      return String(data);
+    }
+  }
+};
+
+export { PosthogExporter, formatUsageMetrics };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/tracing.ts"],"names":[],"mappings":";;;;;AAuBO,SAAS,mBAAmB,KAAA,EAAyC;AAC1E,EAAA,IAAI,CAAC,KAAA,EAAO,OAAO,EAAC;AAEpB,EAAA,MAAM,QAA6B,EAAC;AAEpC,EAAA,IAAI,KAAA,CAAM,WAAA,KAAgB,MAAA,EAAW,KAAA,CAAM,mBAAmB,KAAA,CAAM,WAAA;AACpE,EAAA,IAAI,KAAA,CAAM,YAAA,KAAiB,MAAA,EAAW,KAAA,CAAM,oBAAoB,KAAA,CAAM,YAAA;AAGtE,EAAA,IAAI,MAAM,YAAA,EAAc,SAAA,KAAc,QAAW,KAAA,CAAM,2BAAA,GAA8B,MAAM,YAAA,CAAa,SAAA;AAGxG,EAAA,IAAI,KAAA,CAAM,cAAc,UAAA,KAAe,MAAA;AACrC,IAAA,KAAA,CAAM,+BAAA,GAAkC,MAAM,YAAA,CAAa,UAAA;AAE7D,EAAA,OAAO,KAAA;AACT;AA+CO,IAAM,eAAA,GAAN,MAAM,gBAAA,SAAwB,YAAA,CAAa;AAAA,EAChD,IAAA,GAAO,SAAA;AAAA,EACC,MAAA;AAAA,EACA,MAAA;AAAA,EACA,QAAA,uBAAe,GAAA,EAA2B;AAAA,EAElD,OAAwB,mBAAA,GAAsB,EAAA;AAAA,EAC9C,OAAwB,yBAAA,GAA4B,GAAA;AAAA,EACpD,OAAwB,gBAAA,GAAmB,EAAA;AAAA,EAC3C,OAAwB,sBAAA,GAAyB,GAAA;AAAA,EAEjD,YAAY,MAAA,EAA+B;AACzC,IAAA,KAAA,CAAM,MAAM,CAAA;AACZ,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAEd,IAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,MAAA,IAAA,CAAK,YAAY,0BAA0B,CAAA;AAC3C,MAAA,IAAA,CAAK,MAAA,GAAS,IAAA;AACd,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,YAAA,GAAe,IAAA,CAAK,iBAAA,CAAkB,MAAM,CAAA;AAClD,IAAA,IAAA,CAAK,MAAA,GAAS,IAAI,OAAA,CAAQ,MAAA,CAAO,QAAQ,YAAY,CAAA;AACrD,IAAA,IAAA,CAAK,iBAAA,CAAkB,MAAA,CAAO,UAAA,IAAc,KAAA,EAAO,YAAY,CAAA;AAAA,EACjE;AAAA,EAEQ,kBAAkB,MAAA,EAA+B;AACvD,IAAA,MAAM,YAAA,GAAe,OAAO,UAAA,IAAc,KAAA;AAC1C,IAAA,MAAM,UACJ,MAAA,CAAO,OAAA,KAAY,YAAA,GAAe,gBAAA,CAAgB,sBAAsB,gBAAA,CAAgB,gBAAA,CAAA;AAC1F,IAAA,MAAM,gBACJ,MAAA,CAAO,aAAA,KACN,YAAA,GAAe,gBAAA,CAAgB,4BAA4B,gBAAA,CAAgB,sBAAA,CAAA;AAE9E,IAAA,MAAM,IAAA,GAAO,MAAA,CAAO,IAAA,IAAQ,OAAA,CAAQ,IAAI,YAAA,IAAgB,0BAAA;AAExD,IAAA,IAAI,CAAC,MAAA,CAAO,IAAA,IAAQ,CAAC,OAAA,CAAQ,IAAI,YAAA,EAAc;AAC7C,MAAA,IAAA,CAAK,MAAA,CAAO,IAAA;AAAA,QACV;AAAA,OAGF;AAAA,IACF;AAEA,IAAA,OAAO;AAAA,MACL,IAAA;AAAA,MACA,OAAA;AAAA,MACA,aAAA;AAAA,MACA,aAAa,MAAA,CAAO;AAAA,KACtB;AAAA,EACF;AAAA,EAEQ,iBAAA,CACN,cACA,MAAA,EACM;AACN,IAAA,MAAM,OAAA,GAAU,eAAe,iDAAA,GAAoD,8BAAA;AACnF,IAAA,IAAA,CAAK,MAAA,CAAO,KAAA,CAAM,OAAA,EAAS,MAAM,CAAA;AAAA,EACnC;AAAA,EAEA,MAAgB,oBAAoB,KAAA,EAAoC;AACtE,IAAA,IAAI,CAAC,KAAK,MAAA,EAAQ;AAChB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI;AACF,MAAA,IAAI,KAAA,CAAM,aAAa,OAAA,EAAS;AAC9B,QAAA,IAAI,KAAA,CAAM,SAAS,cAAA,EAAgB;AACjC,UAAA,MAAM,IAAA,CAAK,gBAAA,CAAiB,KAAA,CAAM,YAAY,CAAA;AAAA,QAChD;AACA,QAAA;AAAA,MACF;AAEA,MAAA,QAAQ,MAAM,IAAA;AAAM,QAClB,KAAK,cAAA;AACH,UAAA,MAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,CAAM,YAAY,CAAA;AAC/C,UAAA;AAAA,QACF,KAAK,cAAA;AACH,UAAA;AAAA,QACF,KAAK,YAAA;AACH,UAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAA,CAAM,YAAY,CAAA;AAC7C,UAAA;AAAA;AACJ,IACF,SAAS,KAAA,EAAO;AACd,MAAA,IAAA,CAAK,OAAO,KAAA,CAAM,wBAAA,EAA0B,EAAE,KAAA,EAAO,OAAO,CAAA;AAAA,IAC9D;AAAA,EACF;AAAA,EAEA,MAAc,kBAAkB,IAAA,EAAsC;AACpE,IAAA,IAAI,SAAA,GAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,KAAK,OAAO,CAAA;AAE9C,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,SAAA,GAAY;AAAA,QACV,KAAA,sBAAW,GAAA,EAAI;AAAA,QACf,UAAA,EAAY;AAAA,OACd;AACA,MAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,IAAA,CAAK,OAAA,EAAS,SAAS,CAAA;AAAA,IAC3C;AAEA,IAAA,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI;AAAA,MAC3B,SAAA,EAAW,IAAA,CAAK,MAAA,CAAO,IAAA,CAAK,SAAS,CAAA;AAAA,MACrC,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,YAAY,IAAA,CAAK;AAAA,KAClB,CAAA;AAED,IAAA,IAAI,CAAC,UAAU,UAAA,EAAY;AACzB,MAAA,MAAM,MAAA,GAAS,KAAK,QAAA,EAAU,MAAA;AAC9B,MAAA,IAAI,MAAA,EAAQ;AACV,QAAA,SAAA,CAAU,UAAA,GAAa,OAAO,MAAM,CAAA;AAAA,MACtC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBAAgB,IAAA,EAAsC;AAClE,IAAA,MAAM,SAAA,GAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,KAAK,OAAO,CAAA;AAEhD,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,UAAA,GAAa,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,KAAK,EAAE,CAAA;AAC9C,IAAA,IAAI,CAAC,UAAA,EAAY;AACf,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,SAAA,GAAY,UAAA,CAAW,SAAA,CAAU,OAAA,EAAQ;AAC/C,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,OAAA,GAAU,IAAA,CAAK,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA,CAAE,OAAA,EAAQ,GAAI,IAAA,CAAK,GAAA,EAAI;AAC9E,IAAA,MAAM,OAAA,GAAA,CAAW,UAAU,SAAA,IAAa,GAAA;AAExC,IAAA,MAAM,UAAA,GAAa,IAAA,CAAK,aAAA,CAAc,IAAA,EAAM,SAAS,CAAA;AAIrD,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,IAAA,CAAK,iBAAA,CAAkB,IAAA,EAAM,UAAA,EAAY,OAAO,CAAA;AAAA,IAClD,CAAA,MAAO;AACL,MAAA,MAAM,SAAA,GAAY,IAAA,CAAK,iBAAA,CAAkB,IAAA,CAAK,IAAI,CAAA;AAIlD,MAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,gBAAA,CAAiB,IAAA,EAAM,SAAS,CAAA;AAC9D,MAAA,MAAM,UAAA,GAAa,IAAA,CAAK,oBAAA,CAAqB,IAAA,EAAM,SAAS,gBAAgB,CAAA;AAE5E,MAAA,IAAA,CAAK,OAAO,OAAA,CAAQ;AAAA,QAClB,UAAA;AAAA,QACA,KAAA,EAAO,SAAA;AAAA,QACP,UAAA;AAAA,QACA,SAAA,EAAW,IAAI,IAAA,CAAK,OAAO;AAAA,OAC5B,CAAA;AAAA,IACH;AAEA,IAAA,SAAA,CAAU,KAAA,CAAM,MAAA,CAAO,IAAA,CAAK,EAAE,CAAA;AAC9B,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,IAAA,KAAS,CAAA,EAAG;AAC9B,MAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA;AAAA,IACnC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOQ,iBAAA,CAAkB,IAAA,EAAuB,UAAA,EAAoB,OAAA,EAAuB;AAG1F,IAAA,MAAM,eAAA,GAAuC;AAAA,MAC3C,cAAc,IAAA,CAAK,OAAA;AAAA,MACnB,eAAe,IAAA,CAAK,IAAA;AAAA,MACpB,YAAA,EAAc,CAAC,CAAC,IAAA,CAAK;AAAA,KACvB;AAEA,IAAA,IAAI,IAAA,CAAK,UAAU,SAAA,EAAW;AAC5B,MAAA,eAAA,CAAgB,cAAA,GAAiB,KAAK,QAAA,CAAS,SAAA;AAAA,IACjD;AAEA,IAAA,IAAI,KAAK,KAAA,EAAO;AACd,MAAA,eAAA,CAAgB,kBAAkB,IAAA,CAAK,KAAA;AAAA,IACzC;AAEA,IAAA,IAAI,KAAK,MAAA,EAAQ;AACf,MAAA,eAAA,CAAgB,mBAAmB,IAAA,CAAK,MAAA;AAAA,IAC1C;AAEA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,eAAA,CAAgB,SAAA,GAAY;AAAA,QAC1B,OAAA,EAAS,KAAK,SAAA,CAAU,OAAA;AAAA,QACxB,GAAI,KAAK,SAAA,CAAU,EAAA,IAAM,EAAE,EAAA,EAAI,IAAA,CAAK,UAAU,EAAA,EAAG;AAAA,QACjD,GAAI,KAAK,SAAA,CAAU,QAAA,IAAY,EAAE,QAAA,EAAU,IAAA,CAAK,UAAU,QAAA;AAAS,OACrE;AAAA,IACF;AAGA,IAAA,IAAI,IAAA,CAAK,MAAM,MAAA,EAAQ;AACrB,MAAA,KAAA,MAAW,GAAA,IAAO,KAAK,IAAA,EAAM;AAC3B,QAAA,eAAA,CAAgB,GAAG,CAAA,GAAI,IAAA;AAAA,MACzB;AAAA,IACF;AAGA,IAAA,MAAM,EAAE,QAAQ,SAAA,EAAW,GAAG,gBAAe,GAAI,IAAA,CAAK,YAAY,EAAC;AACnE,IAAA,MAAA,CAAO,MAAA,CAAO,iBAAiB,cAAc,CAAA;AAE7C,IAAA,IAAA,CAAK,OAAO,OAAA,CAAQ;AAAA,MAClB,UAAA;AAAA,MACA,KAAA,EAAO,WAAA;AAAA,MACP,UAAA,EAAY,eAAA;AAAA,MACZ,SAAA,EAAW,IAAI,IAAA,CAAK,OAAO;AAAA,KAC5B,CAAA;AAAA,EACH;AAAA,EAEA,MAAc,iBAAiB,IAAA,EAAsC;AACnE,IAAA,MAAM,SAAA,GAAY,IAAA,CAAK,iBAAA,CAAkB,IAAA,CAAK,IAAI,CAAA;AAClD,IAAA,MAAM,SAAA,GAAY,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,KAAK,OAAO,CAAA;AAEhD,IAAA,MAAM,UAAA,GAAa,IAAA,CAAK,aAAA,CAAc,IAAA,EAAM,SAAS,CAAA;AACrD,IAAA,MAAM,UAAA,GAAa,IAAA,CAAK,oBAAA,CAAqB,IAAA,EAAM,CAAC,CAAA;AAEpD,IAAA,IAAA,CAAK,OAAO,OAAA,CAAQ;AAAA,MAClB,UAAA;AAAA,MACA,KAAA,EAAO,SAAA;AAAA,MACP,UAAA;AAAA,MACA,SAAA,EAAW,KAAK,OAAA,GAAU,IAAI,KAAK,IAAA,CAAK,OAAO,CAAA,mBAAI,IAAI,IAAA;AAAK,KAC7D,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,QAAA,GAA0B;AAC9B,IAAA,IAAI,KAAK,MAAA,EAAQ;AACf,MAAA,MAAM,IAAA,CAAK,OAAO,QAAA,EAAS;AAAA,IAC7B;AACA,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AACpB,IAAA,MAAM,MAAM,QAAA,EAAS;AACrB,IAAA,IAAA,CAAK,MAAA,CAAO,KAAK,oCAAoC,CAAA;AAAA,EACvD;AAAA,EAEQ,OAAO,SAAA,EAAgC;AAC7C,IAAA,OAAO,SAAA,YAAqB,IAAA,GAAO,SAAA,GAAY,IAAI,KAAK,SAAS,CAAA;AAAA,EACnE;AAAA,EAEQ,kBAAkB,QAAA,EAA4B;AACpD,IAAA,IAAI,QAAA,IAAY,SAAS,gBAAA,EAAkB;AACzC,MAAA,OAAO,gBAAA;AAAA,IACT;AACA,IAAA,OAAO,UAAA;AAAA,EACT;AAAA,EAEQ,aAAA,CAAc,MAAuB,SAAA,EAAmC;AAC9E,IAAA,IAAI,IAAA,CAAK,UAAU,MAAA,EAAQ;AACzB,MAAA,OAAO,MAAA,CAAO,IAAA,CAAK,QAAA,CAAS,MAAM,CAAA;AAAA,IACpC;AAEA,IAAA,IAAI,WAAW,UAAA,EAAY;AACzB,MAAA,OAAO,SAAA,CAAU,UAAA;AAAA,IACnB;AAEA,IAAA,IAAI,IAAA,CAAK,OAAO,iBAAA,EAAmB;AACjC,MAAA,OAAO,KAAK,MAAA,CAAO,iBAAA;AAAA,IACrB;AAEA,IAAA,OAAO,WAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOQ,gBAAA,CAAiB,MAAuB,SAAA,EAAmC;AACjF,IAAA,IAAI,CAAC,KAAK,YAAA,EAAc;AACtB,MAAA,OAAO,KAAA;AAAA,IACT;AAGA,IAAA,MAAM,WAAA,GAAc,SAAA,CAAU,KAAA,CAAM,GAAA,CAAI,KAAK,YAAY,CAAA;AACzD,IAAA,IAAI,WAAA,EAAa;AACf,MAAA,OAAO,WAAA,CAAY,UAAA;AAAA,IACrB;AAGA,IAAA,OAAO,KAAA;AAAA,EACT;AAAA,EAEQ,oBAAA,CACN,IAAA,EACA,OAAA,EACA,gBAAA,GAA4B,KAAA,EACP;AACrB,IAAA,MAAM,cAAA,GAAsC;AAAA,MAC1C,cAAc,IAAA,CAAK,OAAA;AAAA,MACnB,WAAA,EAAa,OAAA;AAAA,MACb,YAAA,EAAc,CAAC,CAAC,IAAA,CAAK;AAAA,KACvB;AAEA,IAAA,IAAI,KAAK,YAAA,EAAc;AAGrB,MAAA,cAAA,CAAe,aAAA,GAAgB,gBAAA,GAAmB,IAAA,CAAK,OAAA,GAAU,IAAA,CAAK,YAAA;AAAA,IACxE;AAEA,IAAA,IAAI,IAAA,CAAK,UAAU,SAAA,EAAW;AAC5B,MAAA,cAAA,CAAe,cAAA,GAAiB,KAAK,QAAA,CAAS,SAAA;AAAA,IAChD;AAKA,IAAA,IAAI,IAAA,CAAK,UAAA,IAAc,IAAA,CAAK,IAAA,EAAM,MAAA,EAAQ;AACxC,MAAA,KAAA,MAAW,GAAA,IAAO,KAAK,IAAA,EAAM;AAC3B,QAAA,cAAA,CAAe,GAAG,CAAA,GAAI,IAAA;AAAA,MACxB;AAAA,IACF;AAEA,IAAA,IAAI,IAAA,CAAK,IAAA,KAAS,QAAA,CAAS,gBAAA,EAAkB;AAC3C,MAAA,cAAA,CAAe,oBAAoB,IAAA,CAAK,EAAA;AACxC,MAAA,OAAO,EAAE,GAAG,cAAA,EAAgB,GAAG,IAAA,CAAK,yBAAA,CAA0B,IAAI,CAAA,EAAE;AAAA,IACtE,CAAA,MAAO;AACL,MAAA,cAAA,CAAe,cAAc,IAAA,CAAK,EAAA;AAClC,MAAA,cAAA,CAAe,gBAAgB,IAAA,CAAK,IAAA;AACpC,MAAA,OAAO,EAAE,GAAG,cAAA,EAAgB,GAAG,IAAA,CAAK,mBAAA,CAAoB,IAAI,CAAA,EAAE;AAAA,IAChE;AAAA,EACF;AAAA,EAEQ,uBAAuB,IAAA,EAA4C;AACzE,IAAA,IAAI,CAAC,KAAK,SAAA,EAAW;AACnB,MAAA,OAAO,EAAC;AAAA,IACV;AAEA,IAAA,MAAM,KAAA,GAAgC;AAAA,MACpC,aAAA,EAAe,KAAK,SAAA,CAAU;AAAA,KAChC;AAEA,IAAA,IAAI,IAAA,CAAK,UAAU,EAAA,EAAI;AACrB,MAAA,KAAA,CAAM,QAAA,GAAW,KAAK,SAAA,CAAU,EAAA;AAAA,IAClC;AAEA,IAAA,IAAI,IAAA,CAAK,UAAU,QAAA,EAAU;AAC3B,MAAA,KAAA,CAAM,cAAA,GAAiB,KAAK,SAAA,CAAU,QAAA;AAAA,IACxC;AAEA,IAAA,OAAO,KAAA;AAAA,EACT;AAAA,EAEQ,sBAAsB,IAAA,EAA4C;AACxE,IAAA,MAAM,EAAE,QAAQ,SAAA,EAAW,GAAG,gBAAe,GAAI,IAAA,CAAK,YAAY,EAAC;AACnE,IAAA,OAAO,cAAA;AAAA,EACT;AAAA,EAEQ,0BAA0B,IAAA,EAA4C;AAC5E,IAAA,MAAM,QAA6B,EAAC;AACpC,IAAA,MAAM,KAAA,GAAS,IAAA,CAAK,UAAA,IAAc,EAAC;AAEnC,IAAA,KAAA,CAAM,SAAA,GAAY,MAAM,KAAA,IAAS,eAAA;AACjC,IAAA,KAAA,CAAM,YAAA,GAAe,MAAM,QAAA,IAAY,kBAAA;AAEvC,IAAA,IAAI,IAAA,CAAK,OAAO,KAAA,CAAM,SAAA,GAAY,KAAK,cAAA,CAAe,IAAA,CAAK,OAAO,MAAM,CAAA;AACxE,IAAA,IAAI,IAAA,CAAK,QAAQ,KAAA,CAAM,kBAAA,GAAqB,KAAK,cAAA,CAAe,IAAA,CAAK,QAAQ,WAAW,CAAA;AAGxF,IAAA,MAAA,CAAO,MAAA,CAAO,KAAA,EAAO,kBAAA,CAAmB,KAAA,CAAM,KAAK,CAAC,CAAA;AAEpD,IAAA,IAAI,MAAM,UAAA,EAAY;AACpB,MAAA,IAAI,MAAM,UAAA,CAAW,WAAA,KAAgB,QAAW,KAAA,CAAM,eAAA,GAAkB,MAAM,UAAA,CAAW,WAAA;AACzF,MAAA,IAAI,MAAM,UAAA,CAAW,eAAA,KAAoB,QAAW,KAAA,CAAM,cAAA,GAAiB,MAAM,UAAA,CAAW,eAAA;AAAA,IAC9F;AACA,IAAA,IAAI,KAAA,CAAM,SAAA,KAAc,MAAA,EAAW,KAAA,CAAM,aAAa,KAAA,CAAM,SAAA;AAE5D,IAAA,OAAO,EAAE,GAAG,KAAA,EAAO,GAAG,IAAA,CAAK,sBAAA,CAAuB,IAAI,CAAA,EAAG,GAAG,IAAA,CAAK,qBAAA,CAAsB,IAAI,CAAA,EAAE;AAAA,EAC/F;AAAA,EAEQ,oBAAoB,IAAA,EAA4C;AACtE,IAAA,MAAM,QAA6B,EAAC;AAEpC,IAAA,IAAI,IAAA,CAAK,KAAA,EAAO,KAAA,CAAM,eAAA,GAAkB,IAAA,CAAK,KAAA;AAC7C,IAAA,IAAI,IAAA,CAAK,MAAA,EAAQ,KAAA,CAAM,gBAAA,GAAmB,IAAA,CAAK,MAAA;AAE/C,IAAA,IAAI,IAAA,CAAK,IAAA,KAAS,QAAA,CAAS,WAAA,EAAa;AACtC,MAAA,MAAM,QAAQ,IAAA,CAAK,UAAA;AACnB,MAAA,IAAI,KAAA,EAAO,SAAA,EAAW,KAAA,CAAM,UAAA,GAAa,KAAA,CAAM,SAAA;AAC/C,MAAA,IAAI,KAAA,EAAO,cAAA,KAAmB,MAAA,EAAW,KAAA,CAAM,wBAAwB,KAAA,CAAM,cAAA;AAAA,IAC/E;AAEA,IAAA,IAAI,KAAK,UAAA,EAAY;AACnB,MAAA,MAAA,CAAO,MAAA,CAAO,KAAA,EAAO,IAAA,CAAK,UAAU,CAAA;AAAA,IACtC;AAEA,IAAA,OAAO,EAAE,GAAG,KAAA,EAAO,GAAG,IAAA,CAAK,sBAAA,CAAuB,IAAI,CAAA,EAAG,GAAG,IAAA,CAAK,qBAAA,CAAsB,IAAI,CAAA,EAAE;AAAA,EAC/F;AAAA,EAEQ,cAAA,CAAe,IAAA,EAAgB,WAAA,GAAoC,MAAA,EAA0B;AACnG,IAAA,IAAI,IAAA,CAAK,cAAA,CAAe,IAAI,CAAA,EAAG;AAC7B,MAAA,OAAO,KAAK,GAAA,CAAI,CAAA,GAAA,KAAO,IAAA,CAAK,gBAAA,CAAiB,GAAG,CAAC,CAAA;AAAA,IACnD;AAEA,IAAA,IAAI,OAAO,SAAS,QAAA,EAAU;AAC5B,MAAA,OAAO,CAAC,EAAE,IAAA,EAAM,WAAA,EAAa,OAAA,EAAS,CAAC,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,IAAA,EAAM,GAAG,CAAA;AAAA,IACxE;AAEA,IAAA,OAAO,CAAC,EAAE,IAAA,EAAM,WAAA,EAAa,OAAA,EAAS,CAAC,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,KAAK,aAAA,CAAc,IAAI,CAAA,EAAG,GAAG,CAAA;AAAA,EAC5F;AAAA,EAEQ,eAAe,IAAA,EAAwC;AAC7D,IAAA,IAAI,CAAC,KAAA,CAAM,OAAA,CAAQ,IAAI,CAAA,IAAK,IAAA,CAAK,WAAW,CAAA,EAAG;AAC7C,MAAA,OAAO,KAAA;AAAA,IACT;AAEA,IAAA,OAAO,IAAA,CAAK,KAAA,CAAM,CAAA,IAAA,KAAQ,OAAO,IAAA,KAAS,QAAA,IAAY,IAAA,KAAS,IAAA,IAAQ,MAAA,IAAU,IAAA,IAAQ,SAAA,IAAa,IAAI,CAAA;AAAA,EAC5G;AAAA,EAEQ,iBAAiB,GAAA,EAAoC;AAC3D,IAAA,IAAI,OAAO,GAAA,CAAI,OAAA,KAAY,QAAA,EAAU;AACnC,MAAA,OAAO;AAAA,QACL,MAAM,GAAA,CAAI,IAAA;AAAA,QACV,OAAA,EAAS,CAAC,EAAE,IAAA,EAAM,QAAQ,IAAA,EAAM,GAAA,CAAI,SAAS;AAAA,OAC/C;AAAA,IACF;AAEA,IAAA,OAAO;AAAA,MACL,MAAM,GAAA,CAAI,IAAA;AAAA,MACV,SAAS,GAAA,CAAI;AAAA,KACf;AAAA,EACF;AAAA,EAEQ,cAAc,IAAA,EAAuB;AAC3C,IAAA,IAAI;AACF,MAAA,OAAO,IAAA,CAAK,UAAU,IAAI,CAAA;AAAA,IAC5B,CAAA,CAAA,MAAQ;AACN,MAAA,IAAI,OAAO,IAAA,KAAS,QAAA,IAAY,IAAA,KAAS,IAAA,EAAM;AAC7C,QAAA,OAAO,CAAA,kBAAA,EAAqB,IAAA,CAAK,WAAA,EAAa,IAAA,IAAQ,QAAQ,CAAA,CAAA,CAAA;AAAA,MAChE;AACA,MAAA,OAAO,OAAO,IAAI,CAAA;AAAA,IACpB;AAAA,EACF;AACF","file":"index.js","sourcesContent":["import type { TracingEvent, AnyExportedSpan, ModelGenerationAttributes, UsageStats } from '@mastra/core/observability';\nimport { SpanType } from '@mastra/core/observability';\nimport type { BaseExporterConfig } from '@mastra/observability';\nimport { BaseExporter } from '@mastra/observability';\nimport { PostHog } from 'posthog-node';\n\n/**\n * Token usage format compatible with PostHog.\n * @see https://posthog.com/docs/llm-analytics/generations#event-properties\n */\nexport interface PostHogUsageMetrics {\n  $ai_input_tokens?: number;\n  $ai_output_tokens?: number;\n  $ai_cache_read_input_tokens?: number;\n  $ai_cache_creation_input_tokens?: number;\n}\n\n/**\n * Formats UsageStats to PostHog's expected property format.\n *\n * @param usage - The UsageStats from span attributes\n * @returns PostHog-formatted usage properties\n */\nexport function formatUsageMetrics(usage?: UsageStats): PostHogUsageMetrics {\n  if (!usage) return {};\n\n  const props: PostHogUsageMetrics = {};\n\n  if (usage.inputTokens !== undefined) props.$ai_input_tokens = usage.inputTokens;\n  if (usage.outputTokens !== undefined) props.$ai_output_tokens = usage.outputTokens;\n\n  // Cache read tokens from inputDetails\n  if (usage.inputDetails?.cacheRead !== undefined) props.$ai_cache_read_input_tokens = usage.inputDetails.cacheRead;\n\n  // Cache write tokens from inputDetails\n  if (usage.inputDetails?.cacheWrite !== undefined)\n    props.$ai_cache_creation_input_tokens = usage.inputDetails.cacheWrite;\n\n  return props;\n}\n\ninterface PostHogMessage {\n  role: 'user' | 'assistant' | 'system' | 'tool';\n  content: PostHogContent[];\n}\n\ninterface PostHogContent {\n  type: string;\n  text?: string;\n  [key: string]: unknown;\n}\n\ninterface MastraMessage {\n  role: string;\n  content: string | MastraContent[];\n}\n\ninterface MastraContent {\n  type: string;\n  text?: string;\n  [key: string]: unknown;\n}\n\ntype SpanData = string | MastraMessage[] | Record<string, unknown> | unknown;\n\nexport interface PosthogExporterConfig extends BaseExporterConfig {\n  apiKey: string;\n  host?: string;\n  flushAt?: number;\n  flushInterval?: number;\n  serverless?: boolean;\n  defaultDistinctId?: string;\n  enablePrivacyMode?: boolean;\n}\n\ntype SpanCache = {\n  startTime: Date;\n  type: SpanType;\n  isRootSpan: boolean;\n};\n\ntype TraceMetadata = {\n  spans: Map<string, SpanCache>;\n  distinctId?: string;\n};\n\nexport class PosthogExporter extends BaseExporter {\n  name = 'posthog';\n  private client: PostHog;\n  private config: PosthogExporterConfig;\n  private traceMap = new Map<string, TraceMetadata>();\n\n  private static readonly SERVERLESS_FLUSH_AT = 10;\n  private static readonly SERVERLESS_FLUSH_INTERVAL = 2000;\n  private static readonly DEFAULT_FLUSH_AT = 20;\n  private static readonly DEFAULT_FLUSH_INTERVAL = 10000;\n\n  constructor(config: PosthogExporterConfig) {\n    super(config);\n    this.config = config;\n\n    if (!config.apiKey) {\n      this.setDisabled('Missing required API key');\n      this.client = null as any;\n      return;\n    }\n\n    const clientConfig = this.buildClientConfig(config);\n    this.client = new PostHog(config.apiKey, clientConfig);\n    this.logInitialization(config.serverless ?? false, clientConfig);\n  }\n\n  private buildClientConfig(config: PosthogExporterConfig) {\n    const isServerless = config.serverless ?? false;\n    const flushAt =\n      config.flushAt ?? (isServerless ? PosthogExporter.SERVERLESS_FLUSH_AT : PosthogExporter.DEFAULT_FLUSH_AT);\n    const flushInterval =\n      config.flushInterval ??\n      (isServerless ? PosthogExporter.SERVERLESS_FLUSH_INTERVAL : PosthogExporter.DEFAULT_FLUSH_INTERVAL);\n\n    const host = config.host || process.env.POSTHOG_HOST || 'https://us.i.posthog.com';\n\n    if (!config.host && !process.env.POSTHOG_HOST) {\n      this.logger.warn(\n        'No PostHog host specified, using US default (https://us.i.posthog.com). ' +\n          'For EU region, set `host: \"https://eu.i.posthog.com\"` in config or POSTHOG_HOST env var. ' +\n          'For self-hosted, provide your instance URL.',\n      );\n    }\n\n    return {\n      host,\n      flushAt,\n      flushInterval,\n      privacyMode: config.enablePrivacyMode,\n    };\n  }\n\n  private logInitialization(\n    isServerless: boolean,\n    config: { host: string; flushAt: number; flushInterval: number; privacyMode?: boolean },\n  ): void {\n    const message = isServerless ? 'PostHog exporter initialized in serverless mode' : 'PostHog exporter initialized';\n    this.logger.debug(message, config);\n  }\n\n  protected async _exportTracingEvent(event: TracingEvent): Promise<void> {\n    if (!this.client) {\n      return;\n    }\n\n    try {\n      if (event.exportedSpan.isEvent) {\n        if (event.type === 'span_started') {\n          await this.captureEventSpan(event.exportedSpan);\n        }\n        return;\n      }\n\n      switch (event.type) {\n        case 'span_started':\n          await this.handleSpanStarted(event.exportedSpan);\n          break;\n        case 'span_updated':\n          break;\n        case 'span_ended':\n          await this.handleSpanEnded(event.exportedSpan);\n          break;\n      }\n    } catch (error) {\n      this.logger.error('PostHog exporter error', { error, event });\n    }\n  }\n\n  private async handleSpanStarted(span: AnyExportedSpan): Promise<void> {\n    let traceData = this.traceMap.get(span.traceId);\n\n    if (!traceData) {\n      traceData = {\n        spans: new Map(),\n        distinctId: undefined,\n      };\n      this.traceMap.set(span.traceId, traceData);\n    }\n\n    traceData.spans.set(span.id, {\n      startTime: this.toDate(span.startTime),\n      type: span.type,\n      isRootSpan: span.isRootSpan,\n    });\n\n    if (!traceData.distinctId) {\n      const userId = span.metadata?.userId;\n      if (userId) {\n        traceData.distinctId = String(userId);\n      }\n    }\n  }\n\n  private async handleSpanEnded(span: AnyExportedSpan): Promise<void> {\n    const traceData = this.traceMap.get(span.traceId);\n\n    if (!traceData) {\n      return;\n    }\n\n    const cachedSpan = traceData.spans.get(span.id);\n    if (!cachedSpan) {\n      return;\n    }\n\n    const startTime = cachedSpan.startTime.getTime();\n    const endTime = span.endTime ? this.toDate(span.endTime).getTime() : Date.now();\n    const latency = (endTime - startTime) / 1000;\n\n    const distinctId = this.getDistinctId(span, traceData);\n\n    // For root spans, only send $ai_trace (not $ai_span) to avoid duplicate entries\n    // For non-root spans, send $ai_span or $ai_generation as normal\n    if (span.isRootSpan) {\n      this.captureTraceEvent(span, distinctId, endTime);\n    } else {\n      const eventName = this.mapToPostHogEvent(span.type);\n\n      // Check if parent is the root span - if so, use traceId as parent_id\n      // since we don't create an $ai_span for root spans\n      const parentIsRootSpan = this.isParentRootSpan(span, traceData);\n      const properties = this.buildEventProperties(span, latency, parentIsRootSpan);\n\n      this.client.capture({\n        distinctId,\n        event: eventName,\n        properties,\n        timestamp: new Date(endTime),\n      });\n    }\n\n    traceData.spans.delete(span.id);\n    if (traceData.spans.size === 0) {\n      this.traceMap.delete(span.traceId);\n    }\n  }\n\n  /**\n   * Capture an explicit $ai_trace event for root spans.\n   * This gives us control over trace-level metadata like name and tags,\n   * rather than relying on PostHog's pseudo-trace auto-creation.\n   */\n  private captureTraceEvent(span: AnyExportedSpan, distinctId: string, endTime: number): void {\n    // Note: We don't set $ai_latency on $ai_trace events because PostHog\n    // aggregates latency from child events. Setting it here causes double-counting.\n    const traceProperties: Record<string, any> = {\n      $ai_trace_id: span.traceId,\n      $ai_span_name: span.name,\n      $ai_is_error: !!span.errorInfo,\n    };\n\n    if (span.metadata?.sessionId) {\n      traceProperties.$ai_session_id = span.metadata.sessionId;\n    }\n\n    if (span.input) {\n      traceProperties.$ai_input_state = span.input;\n    }\n\n    if (span.output) {\n      traceProperties.$ai_output_state = span.output;\n    }\n\n    if (span.errorInfo) {\n      traceProperties.$ai_error = {\n        message: span.errorInfo.message,\n        ...(span.errorInfo.id && { id: span.errorInfo.id }),\n        ...(span.errorInfo.category && { category: span.errorInfo.category }),\n      };\n    }\n\n    // Add tags as custom properties (PostHog doesn't have native tag support on traces)\n    if (span.tags?.length) {\n      for (const tag of span.tags) {\n        traceProperties[tag] = true;\n      }\n    }\n\n    // Add custom metadata (excluding userId and sessionId which are handled separately)\n    const { userId, sessionId, ...customMetadata } = span.metadata ?? {};\n    Object.assign(traceProperties, customMetadata);\n\n    this.client.capture({\n      distinctId,\n      event: '$ai_trace',\n      properties: traceProperties,\n      timestamp: new Date(endTime),\n    });\n  }\n\n  private async captureEventSpan(span: AnyExportedSpan): Promise<void> {\n    const eventName = this.mapToPostHogEvent(span.type);\n    const traceData = this.traceMap.get(span.traceId);\n\n    const distinctId = this.getDistinctId(span, traceData);\n    const properties = this.buildEventProperties(span, 0);\n\n    this.client.capture({\n      distinctId,\n      event: eventName,\n      properties,\n      timestamp: span.endTime ? new Date(span.endTime) : new Date(),\n    });\n  }\n\n  async shutdown(): Promise<void> {\n    if (this.client) {\n      await this.client.shutdown();\n    }\n    this.traceMap.clear();\n    await super.shutdown();\n    this.logger.info('PostHog exporter shutdown complete');\n  }\n\n  private toDate(timestamp: Date | number): Date {\n    return timestamp instanceof Date ? timestamp : new Date(timestamp);\n  }\n\n  private mapToPostHogEvent(spanType: SpanType): string {\n    if (spanType == SpanType.MODEL_GENERATION) {\n      return '$ai_generation';\n    }\n    return '$ai_span';\n  }\n\n  private getDistinctId(span: AnyExportedSpan, traceData?: TraceMetadata): string {\n    if (span.metadata?.userId) {\n      return String(span.metadata.userId);\n    }\n\n    if (traceData?.distinctId) {\n      return traceData.distinctId;\n    }\n\n    if (this.config.defaultDistinctId) {\n      return this.config.defaultDistinctId;\n    }\n\n    return 'anonymous';\n  }\n\n  /**\n   * Check if the parent of this span is the root span.\n   * We need this because we don't create $ai_span for root spans,\n   * so children of root spans should use $ai_trace_id as their $ai_parent_id.\n   */\n  private isParentRootSpan(span: AnyExportedSpan, traceData: TraceMetadata): boolean {\n    if (!span.parentSpanId) {\n      return false;\n    }\n\n    // Look up the parent span in our cache to check if it's a root span\n    const parentCache = traceData.spans.get(span.parentSpanId);\n    if (parentCache) {\n      return parentCache.isRootSpan;\n    }\n\n    // Parent not found in cache - shouldn't happen normally, but default to false\n    return false;\n  }\n\n  private buildEventProperties(\n    span: AnyExportedSpan,\n    latency: number,\n    parentIsRootSpan: boolean = false,\n  ): Record<string, any> {\n    const baseProperties: Record<string, any> = {\n      $ai_trace_id: span.traceId,\n      $ai_latency: latency,\n      $ai_is_error: !!span.errorInfo,\n    };\n\n    if (span.parentSpanId) {\n      // If parent is the root span, use trace_id as parent_id since we don't\n      // create an $ai_span for root spans (only $ai_trace)\n      baseProperties.$ai_parent_id = parentIsRootSpan ? span.traceId : span.parentSpanId;\n    }\n\n    if (span.metadata?.sessionId) {\n      baseProperties.$ai_session_id = span.metadata.sessionId;\n    }\n\n    // Include tags for root spans (tags are only set on root spans by design)\n    // PostHog doesn't allow setting tags directly, so we iterate through each tag\n    // and set it as a property with value true\n    if (span.isRootSpan && span.tags?.length) {\n      for (const tag of span.tags) {\n        baseProperties[tag] = true;\n      }\n    }\n\n    if (span.type === SpanType.MODEL_GENERATION) {\n      baseProperties.$ai_generation_id = span.id;\n      return { ...baseProperties, ...this.buildGenerationProperties(span) };\n    } else {\n      baseProperties.$ai_span_id = span.id;\n      baseProperties.$ai_span_name = span.name;\n      return { ...baseProperties, ...this.buildSpanProperties(span) };\n    }\n  }\n\n  private extractErrorProperties(span: AnyExportedSpan): Record<string, any> {\n    if (!span.errorInfo) {\n      return {};\n    }\n\n    const props: Record<string, string> = {\n      error_message: span.errorInfo.message,\n    };\n\n    if (span.errorInfo.id) {\n      props.error_id = span.errorInfo.id;\n    }\n\n    if (span.errorInfo.category) {\n      props.error_category = span.errorInfo.category;\n    }\n\n    return props;\n  }\n\n  private extractCustomMetadata(span: AnyExportedSpan): Record<string, any> {\n    const { userId, sessionId, ...customMetadata } = span.metadata ?? {};\n    return customMetadata;\n  }\n\n  private buildGenerationProperties(span: AnyExportedSpan): Record<string, any> {\n    const props: Record<string, any> = {};\n    const attrs = (span.attributes ?? {}) as ModelGenerationAttributes;\n\n    props.$ai_model = attrs.model || 'unknown-model';\n    props.$ai_provider = attrs.provider || 'unknown-provider';\n\n    if (span.input) props.$ai_input = this.formatMessages(span.input, 'user');\n    if (span.output) props.$ai_output_choices = this.formatMessages(span.output, 'assistant');\n\n    // Extract usage properties using the shared utility\n    Object.assign(props, formatUsageMetrics(attrs.usage));\n\n    if (attrs.parameters) {\n      if (attrs.parameters.temperature !== undefined) props.$ai_temperature = attrs.parameters.temperature;\n      if (attrs.parameters.maxOutputTokens !== undefined) props.$ai_max_tokens = attrs.parameters.maxOutputTokens;\n    }\n    if (attrs.streaming !== undefined) props.$ai_stream = attrs.streaming;\n\n    return { ...props, ...this.extractErrorProperties(span), ...this.extractCustomMetadata(span) };\n  }\n\n  private buildSpanProperties(span: AnyExportedSpan): Record<string, any> {\n    const props: Record<string, any> = {};\n\n    if (span.input) props.$ai_input_state = span.input;\n    if (span.output) props.$ai_output_state = span.output;\n\n    if (span.type === SpanType.MODEL_CHUNK) {\n      const attrs = span.attributes as any;\n      if (attrs?.chunkType) props.chunk_type = attrs.chunkType;\n      if (attrs?.sequenceNumber !== undefined) props.chunk_sequence_number = attrs.sequenceNumber;\n    }\n\n    if (span.attributes) {\n      Object.assign(props, span.attributes);\n    }\n\n    return { ...props, ...this.extractErrorProperties(span), ...this.extractCustomMetadata(span) };\n  }\n\n  private formatMessages(data: SpanData, defaultRole: 'user' | 'assistant' = 'user'): PostHogMessage[] {\n    if (this.isMessageArray(data)) {\n      return data.map(msg => this.normalizeMessage(msg));\n    }\n\n    if (typeof data === 'string') {\n      return [{ role: defaultRole, content: [{ type: 'text', text: data }] }];\n    }\n\n    return [{ role: defaultRole, content: [{ type: 'text', text: this.safeStringify(data) }] }];\n  }\n\n  private isMessageArray(data: unknown): data is MastraMessage[] {\n    if (!Array.isArray(data) || data.length === 0) {\n      return false;\n    }\n\n    return data.every(item => typeof item === 'object' && item !== null && 'role' in item && 'content' in item);\n  }\n\n  private normalizeMessage(msg: MastraMessage): PostHogMessage {\n    if (typeof msg.content === 'string') {\n      return {\n        role: msg.role as PostHogMessage['role'],\n        content: [{ type: 'text', text: msg.content }],\n      };\n    }\n\n    return {\n      role: msg.role as PostHogMessage['role'],\n      content: msg.content as PostHogContent[],\n    };\n  }\n\n  private safeStringify(data: unknown): string {\n    try {\n      return JSON.stringify(data);\n    } catch {\n      if (typeof data === 'object' && data !== null) {\n        return `[Non-serializable ${data.constructor?.name || 'Object'}]`;\n      }\n      return String(data);\n    }\n  }\n}\n"]}

package/dist/tracing.d.ts

@@ -0,0 +1,72 @@
+import type { TracingEvent, UsageStats } from '@mastra/core/observability';
+import type { BaseExporterConfig } from '@mastra/observability';
+import { BaseExporter } from '@mastra/observability';
+/**
+ * Token usage format compatible with PostHog.
+ * @see https://posthog.com/docs/llm-analytics/generations#event-properties
+ */
+export interface PostHogUsageMetrics {
+    $ai_input_tokens?: number;
+    $ai_output_tokens?: number;
+    $ai_cache_read_input_tokens?: number;
+    $ai_cache_creation_input_tokens?: number;
+}
+/**
+ * Formats UsageStats to PostHog's expected property format.
+ *
+ * @param usage - The UsageStats from span attributes
+ * @returns PostHog-formatted usage properties
+ */
+export declare function formatUsageMetrics(usage?: UsageStats): PostHogUsageMetrics;
+export interface PosthogExporterConfig extends BaseExporterConfig {
+    apiKey: string;
+    host?: string;
+    flushAt?: number;
+    flushInterval?: number;
+    serverless?: boolean;
+    defaultDistinctId?: string;
+    enablePrivacyMode?: boolean;
+}
+export declare class PosthogExporter extends BaseExporter {
+    name: string;
+    private client;
+    private config;
+    private traceMap;
+    private static readonly SERVERLESS_FLUSH_AT;
+    private static readonly SERVERLESS_FLUSH_INTERVAL;
+    private static readonly DEFAULT_FLUSH_AT;
+    private static readonly DEFAULT_FLUSH_INTERVAL;
+    constructor(config: PosthogExporterConfig);
+    private buildClientConfig;
+    private logInitialization;
+    protected _exportTracingEvent(event: TracingEvent): Promise<void>;
+    private handleSpanStarted;
+    private handleSpanEnded;
+    /**
+     * Capture an explicit $ai_trace event for root spans.
+     * This gives us control over trace-level metadata like name and tags,
+     * rather than relying on PostHog's pseudo-trace auto-creation.
+     */
+    private captureTraceEvent;
+    private captureEventSpan;
+    shutdown(): Promise<void>;
+    private toDate;
+    private mapToPostHogEvent;
+    private getDistinctId;
+    /**
+     * Check if the parent of this span is the root span.
+     * We need this because we don't create $ai_span for root spans,
+     * so children of root spans should use $ai_trace_id as their $ai_parent_id.
+     */
+    private isParentRootSpan;
+    private buildEventProperties;
+    private extractErrorProperties;
+    private extractCustomMetadata;
+    private buildGenerationProperties;
+    private buildSpanProperties;
+    private formatMessages;
+    private isMessageArray;
+    private normalizeMessage;
+    private safeStringify;
+}
+//# sourceMappingURL=tracing.d.ts.map

package/dist/tracing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracing.d.ts","sourceRoot":"","sources":["../src/tracing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAA8C,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAEvH,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAGrD;;;GAGG;AACH,MAAM,WAAW,mBAAmB;IAClC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,2BAA2B,CAAC,EAAE,MAAM,CAAC;IACrC,+BAA+B,CAAC,EAAE,MAAM,CAAC;CAC1C;AAED;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,CAAC,EAAE,UAAU,GAAG,mBAAmB,CAgB1E;AA0BD,MAAM,WAAW,qBAAsB,SAAQ,kBAAkB;IAC/D,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAaD,qBAAa,eAAgB,SAAQ,YAAY;IAC/C,IAAI,SAAa;IACjB,OAAO,CAAC,MAAM,CAAU;IACxB,OAAO,CAAC,MAAM,CAAwB;IACtC,OAAO,CAAC,QAAQ,CAAoC;IAEpD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAM;IACjD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAQ;IACzD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAM;IAC9C,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAS;gBAE3C,MAAM,EAAE,qBAAqB;IAezC,OAAO,CAAC,iBAAiB;IA0BzB,OAAO,CAAC,iBAAiB;cAQT,mBAAmB,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;YA4BzD,iBAAiB;YAyBjB,eAAe;IA4C7B;;;;OAIG;IACH,OAAO,CAAC,iBAAiB;YAgDX,gBAAgB;IAexB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAS/B,OAAO,CAAC,MAAM;IAId,OAAO,CAAC,iBAAiB;IAOzB,OAAO,CAAC,aAAa;IAgBrB;;;;OAIG;IACH,OAAO,CAAC,gBAAgB;IAexB,OAAO,CAAC,oBAAoB;IAwC5B,OAAO,CAAC,sBAAsB;IAoB9B,OAAO,CAAC,qBAAqB;IAK7B,OAAO,CAAC,yBAAyB;IAsBjC,OAAO,CAAC,mBAAmB;IAmB3B,OAAO,CAAC,cAAc;IAYtB,OAAO,CAAC,cAAc;IAQtB,OAAO,CAAC,gBAAgB;IAcxB,OAAO,CAAC,aAAa;CAUtB"}

package/package.json

@@ -0,0 +1,62 @@
+{
+  "name": "@mastra/posthog",
+  "version": "0.0.0-bundle-studio-cloud-20251222034739",
+  "description": "PostHog observability provider for Mastra",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "CHANGELOG.md"
+  ],
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "require": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.cjs"
+      }
+    },
+    "./package.json": "./package.json"
+  },
+  "license": "Apache-2.0",
+  "dependencies": {
+    "posthog-node": "^4.0.1",
+    "@mastra/observability": "0.0.0-bundle-studio-cloud-20251222034739"
+  },
+  "devDependencies": {
+    "@types/node": "^20.19.0",
+    "eslint": "^9.37.0",
+    "tsup": "^8.5.0",
+    "typescript": "^5.8.3",
+    "vitest": "^3.2.4",
+    "@mastra/core": "0.0.0-bundle-studio-cloud-20251222034739",
+    "@internal/types-builder": "0.0.0-bundle-studio-cloud-20251222034739",
+    "@internal/lint": "0.0.0-bundle-studio-cloud-20251222034739"
+  },
+  "peerDependencies": {
+    "@mastra/core": "0.0.0-bundle-studio-cloud-20251222034739"
+  },
+  "homepage": "https://mastra.ai",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/mastra-ai/mastra.git",
+    "directory": "observability/posthog"
+  },
+  "bugs": {
+    "url": "https://github.com/mastra-ai/mastra/issues"
+  },
+  "engines": {
+    "node": ">=22.13.0"
+  },
+  "scripts": {
+    "build": "tsup --silent --config tsup.config.ts",
+    "build:watch": "pnpm build --watch",
+    "test": "vitest run",
+    "test:watch": "vitest watch",
+    "lint": "eslint ."
+  }
+}