Initialize SDK using registerOTel()

app/(playground)/p/[agentId]/beta-proto/graph/server-actions.ts

@@ -1,6 +1,5 @@
 "use server";
 
-import { metricReader } from "@/instrumentation.node";
 import { openai } from "@ai-sdk/openai";
 import { streamObject } from "ai";
 import { createStreamableValue } from "ai/rsc";
@@ -59,22 +58,24 @@ export async function generateArtifactStream(
 				});
 				const subscriptionId = await getUserSubscriptionId();
 				const isR06User = await isRoute06User();
+				const environment = process.env.NEXT_PUBLIC_VERCEL_ENV || "not-set";
 
 				console.log("Adding metric:", {
 					totalTokens: result.usage.totalTokens,
 					subscriptionId,
 					isR06User,
 					timestamp: new Date().toISOString(),
+					environment,
 				});
 
 				tokenCounter.add(result.usage.totalTokens, {
 					subscriptionId,
 					isR06User,
+					environment,
 				});
 				generation.end({
 					output: result,
 				});
-				waitUntil(metricReader.forceFlush());
 				await lf.shutdownAsync();
 				console.log("end of onFinish");
 			},

instrumentation.ts

@@ -1,8 +1,42 @@
+import type { ExportResult } from "@opentelemetry/core";
+import type { ResourceMetrics } from "@opentelemetry/sdk-metrics";
+import {
+	ConsoleMetricExporter,
+	PeriodicExportingMetricReader,
+} from "@opentelemetry/sdk-metrics";
 import * as Sentry from "@sentry/nextjs";
+import { registerOTel } from "@vercel/otel";
+
+class DebugConsoleMetricExporter extends ConsoleMetricExporter {
+	override async export(
+		metrics: ResourceMetrics,
+		resultCallback: (result: ExportResult) => void,
+	) {
+		console.log("Export called at:", new Date().toISOString());
+		console.log("Metrics to export:", JSON.stringify(metrics, null, 2));
+
+		return super.export(metrics, (result) => {
+			console.log("Export result:", result);
+			resultCallback(result);
+		});
+	}
+}
+// Configure OTLP metrics exporter
+const otlpMetricsExporter = new DebugConsoleMetricExporter();
+
+export const metricReader = new PeriodicExportingMetricReader({
+	exporter: otlpMetricsExporter,
+	exportIntervalMillis: 1000,
+	exportTimeoutMillis: 500,
+});
 
 export async function register() {
+	registerOTel({
+		serviceName: "serverside",
+		metricReader,
+	});
+
 	if (process.env.NEXT_RUNTIME === "nodejs") {
-		await import("./instrumentation.node");
 		await import("./sentry.server.config");
 	}