@jeffusion/bungee-core 5.3.2 → 5.4.1

package/dist/plugins/token-stats/control.js

@@ -1681,6 +1681,11 @@ var PROVIDER_ADAPTERS = {
   anthropic: createAnthropicAdapter(),
   gemini: createGeminiAdapter()
 };
+// packages/core/src/token-stats-window.ts
+var HOUR = 3600000;
+var TOKEN_STATS_RETENTION_MS = 31 * 24 * HOUR;
+var TOKEN_STATS_RANGES = ["1h", "12h", "24h", "1d", "7d", "30d", "day", "week", "month"];
+
 // plugins/token-stats/server/repository.ts
 class TokenStatsRepositoryError extends Error {
   constructor(message) {
@@ -1796,23 +1801,26 @@ class TokenStatsRepository {
     scheduleDrain();
     return true;
   }
-  async query(range, groupBy, asOfMs = Date.now()) {
-    if (range !== "1h" && range !== "12h" && range !== "24h")
+  async query(range, groupBy, asOfMs = Date.now(), timeZone) {
+    if (!TOKEN_STATS_RANGES.includes(range))
       throw new TokenStatsRepositoryError("invalid token-stats range");
     if (groupBy !== "model" && groupBy !== "time") {
       throw new TokenStatsRepositoryError("invalid token-stats groupBy");
     }
     if (!Number.isSafeInteger(asOfMs) || asOfMs < 0)
       throw new TokenStatsRepositoryError("invalid token-stats asOfMs");
-    const snapshot = await this.metering.queryWindowSnapshot({ asOfMs, range, groupBy });
+    const snapshot = await this.metering.queryWindowSnapshot({ asOfMs, range, groupBy, timeZone });
     const total = metricsToDto(snapshot.all);
     const { inputTokens, outputTokens, ...summary } = total;
     return {
       groupBy,
+      asOfMs,
       ...summary,
       totalInputTokens: inputTokens,
       totalOutputTokens: outputTokens,
       ...snapshot.bucketMs === undefined ? {} : { bucketMs: snapshot.bucketMs },
+      ...snapshot.bucketStarts === undefined ? {} : { bucketStarts: snapshot.bucketStarts },
+      ...snapshot.bucketEndMs === undefined ? {} : { bucketEndMs: snapshot.bucketEndMs },
       data: snapshot.data.map((row) => ({
         dimension: row.dimension,
         ...row.bucketStartMs === undefined ? {} : { bucketStartMs: row.bucketStartMs },
@@ -2115,7 +2123,7 @@ class PriceCatalogManager {
 
 // plugins/token-stats/server/control.ts
 var MAX_RESPONSE_BYTES = 256 * 1024;
-var VALID_RANGES = ["1h", "12h", "24h"];
+var MAX_PAGE_STATS_RESPONSE_BYTES = 4 * 1024 * 1024;
 var VALID_GROUP_BY = ["model", "time"];
 
 class ControlError extends Error {
@@ -2126,9 +2134,9 @@ class ControlError extends Error {
     this.name = "TokenStatsControlError";
   }
 }
-function jsonResponse(value, status = 200) {
+function jsonResponse(value, status = 200, maxBytes = MAX_RESPONSE_BYTES) {
   const body = JSON.stringify(value);
-  if (new TextEncoder().encode(body).byteLength > MAX_RESPONSE_BYTES) {
+  if (new TextEncoder().encode(body).byteLength > maxBytes) {
     return new Response(JSON.stringify({ error: "response_limit" }), {
       status: 500,
       headers: { "content-type": "application/json; charset=utf-8" }
@@ -2267,14 +2275,25 @@ class TokenStatsControl {
         const url = new URL(context.request.url);
         const rawRange = url.searchParams.get("range");
         const rawGroupBy = url.searchParams.get("groupBy");
-        if (url.searchParams.getAll("range").length > 1 || url.searchParams.getAll("groupBy").length > 1 || rawRange !== null && !VALID_RANGES.includes(rawRange) || rawGroupBy !== null && !VALID_GROUP_BY.includes(rawGroupBy)) {
+        const timeZone = url.searchParams.get("timeZone") ?? undefined;
+        if (url.searchParams.getAll("range").length > 1 || url.searchParams.getAll("groupBy").length > 1 || url.searchParams.getAll("timeZone").length > 1 || rawRange !== null && !TOKEN_STATS_RANGES.includes(rawRange) || rawGroupBy !== null && !VALID_GROUP_BY.includes(rawGroupBy)) {
           throw new ControlError("invalid_input");
         }
+        if (timeZone !== undefined) {
+          try {
+            if (!timeZone || timeZone.length > 100)
+              throw new Error("invalid time zone");
+            new Intl.DateTimeFormat("en-US", { timeZone });
+          } catch {
+            throw new ControlError("invalid_input");
+          }
+        }
         const range = rawRange ?? "24h";
         const groupBy = rawGroupBy ?? "model";
-        const payload = await abortable(this.repository.query(range, groupBy), context.requestSignal, context.signal);
+        const payload = await abortable(this.repository.query(range, groupBy, Date.now(), timeZone), context.requestSignal, context.signal);
         this.assertAlive(context.requestSignal);
-        return jsonResponse(payload);
+        const maxBytes = ["1h", "12h", "24h"].includes(range) ? MAX_RESPONSE_BYTES : MAX_PAGE_STATS_RESPONSE_BYTES;
+        return jsonResponse(payload, 200, maxBytes);
       })
     }, {
       path: "/pricing",
@@ -2330,4 +2349,4 @@ export {
   control_default as default
 };
 
-//# debugId=C36CBFA86B22DB2D64756E2164756E21
+//# debugId=CC440B77E27EEC3464756E2164756E21