next-ai-draw-io/instrumentation.ts

import { LangfuseSpanProcessor } from '@langfuse/otel';
import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node';

export function register() {
  // Skip telemetry if Langfuse env vars are not configured
  if (!process.env.LANGFUSE_PUBLIC_KEY || !process.env.LANGFUSE_SECRET_KEY) {
    console.warn('[Langfuse] Environment variables not configured - telemetry disabled');
    return;
  }

  const langfuseSpanProcessor = new LangfuseSpanProcessor({
    publicKey: process.env.LANGFUSE_PUBLIC_KEY,
    secretKey: process.env.LANGFUSE_SECRET_KEY,
    baseUrl: process.env.LANGFUSE_BASEURL,
    // Filter out Next.js HTTP request spans so AI SDK spans become root traces
    shouldExportSpan: ({ otelSpan }) => {
      const spanName = otelSpan.name;
      // Skip Next.js HTTP infrastructure spans
      if (spanName.startsWith('POST /') ||
          spanName.startsWith('GET /') ||
          spanName.includes('BaseServer') ||
          spanName.includes('handleRequest')) {
        return false;
      }
      return true;
    },
  });

  const tracerProvider = new NodeTracerProvider({
    spanProcessors: [langfuseSpanProcessor],
  });

  // Register globally so AI SDK's telemetry also uses this processor
  tracerProvider.register();
}