@jeffusion/bungee-core 5.3.1 → 5.4.0

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

@@ -16612,6 +16612,11 @@ function createTokenAccountingSession(input, options = {}) {
     }
   };
 }
+// 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) {
@@ -16756,23 +16761,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 },
@@ -17688,4 +17696,4 @@ export {
   server_default as default
 };
 
-//# debugId=DFB468D2A011105464756E2164756E21
+//# debugId=5769627DAE62677664756E2164756E21