From 2f92399e2395980d10de11f438164629c50ad2ea Mon Sep 17 00:00:00 2001
From: diegosouzapw <diegosouza.pw@gmail.com>
Date: Wed, 27 May 2026 23:59:23 -0300
Subject: [PATCH] feat(playground): add streamMetrics pure function

---
 src/lib/playground/streamMetrics.ts | 61 +++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 src/lib/playground/streamMetrics.ts

diff --git a/src/lib/playground/streamMetrics.ts b/src/lib/playground/streamMetrics.ts
new file mode 100644
index 0000000000..4323862ec3
--- /dev/null
+++ b/src/lib/playground/streamMetrics.ts
@@ -0,0 +1,61 @@
+// src/lib/playground/streamMetrics.ts
+// Pure function — no React dependency. Reusable outside of hooks.
+
+import type { StreamMetrics } from "@/shared/schemas/playground";
+
+export interface ComputeMetricsArgs {
+  /** ms timestamp of when the request was dispatched. null = not started. */
+  startedAt: number | null;
+  /** ms timestamp of when the first SSE chunk was received. null = not received yet. */
+  firstChunkAt: number | null;
+  /** ms timestamp of when the stream finished. null = not finished yet. */
+  finishedAt: number | null;
+  /** Number of input/prompt tokens. */
+  tokensIn: number;
+  /** Number of output/completion tokens accumulated. */
+  tokensOut: number;
+  /**
+   * Optional pricing to compute costUsd.
+   * `estimated: true` marks that this is a client-side estimate (D13).
+   */
+  pricing?: {
+    inUsdPer1k: number;
+    outUsdPer1k: number;
+    estimated: true;
+  };
+}
+
+/**
+ * Computes client-perceived streaming metrics from timing + token counts.
+ *
+ * All timing values are "client-perceived" (D12) — measured from the browser's
+ * perspective, not the server's. UI should label these "(client-side estimate)".
+ */
+export function computeMetrics(args: ComputeMetricsArgs): StreamMetrics {
+  const { startedAt, firstChunkAt, finishedAt, tokensIn, tokensOut, pricing } = args;
+
+  const ttftMs =
+    firstChunkAt != null && startedAt != null ? firstChunkAt - startedAt : null;
+
+  const totalMs =
+    finishedAt != null && startedAt != null ? finishedAt - startedAt : null;
+
+  const tps =
+    totalMs != null && totalMs > 0 && tokensOut > 0
+      ? tokensOut / (totalMs / 1000)
+      : null;
+
+  const costUsd =
+    pricing != null
+      ? (tokensIn * pricing.inUsdPer1k + tokensOut * pricing.outUsdPer1k) / 1000
+      : null;
+
+  return {
+    ttftMs,
+    totalMs,
+    tokensIn,
+    tokensOut,
+    tps,
+    costUsd,
+  };
+}