@versot/vaguspi 0.1.2 → 0.1.3

package/dist/bin.js

@@ -448,6 +448,10 @@ function messageToText(message) {
 
 // packages/host/engine/dist/usage-stats.js
 import { SessionManager } from "@earendil-works/pi-coding-agent";
+function dayAt(ts) {
+  const d = new Date(ts);
+  return new Date(d.getFullYear(), d.getMonth(), d.getDate()).getTime();
+}
 async function aggregateUsageStats() {
   const infos = await SessionManager.listAll();
   let totalTokens = 0;
@@ -461,11 +465,10 @@ async function aggregateUsageStats() {
   const activeDates = /* @__PURE__ */ new Set();
   const DAY = 864e5;
   const dayMap = /* @__PURE__ */ new Map();
-  const dayAt = (ts) => Math.floor(ts / DAY);
   const ensureDay = (day) => {
     let p = dayMap.get(day);
     if (!p) {
-      p = { tokens: 0, messages: 0, sessions: 0, byModel: /* @__PURE__ */ new Map() };
+      p = { tokens: 0, messages: 0, sessions: 0, cost: 0, byModel: /* @__PURE__ */ new Map() };
       dayMap.set(day, p);
     }
     return p;
@@ -489,12 +492,13 @@ async function aggregateUsageStats() {
         if (m.role !== "assistant" || m.usage === void 0)
           continue;
         const u = m.usage;
-        const tokens = typeof u.totalTokens === "number" ? u.totalTokens : Number(u.input ?? 0) + Number(u.output ?? 0) + Number(u.cacheRead ?? 0) + Number(u.cacheWrite ?? 0);
+        const tokens = Number(u.input ?? 0) + Number(u.output ?? 0);
         const cost = typeof u.cost?.total === "number" ? u.cost.total : 0;
         totalTokens += tokens;
         totalCost += cost;
         sessionTokens += tokens;
         dayPoint.tokens += tokens;
+        dayPoint.cost += cost;
         const key = `${String(m.provider ?? "unknown")}/${String(m.model ?? "unknown")}`;
         const cur = byModel.get(key) ?? { tokens: 0, cost: 0 };
         byModel.set(key, { tokens: cur.tokens + tokens, cost: cur.cost + cost });
@@ -513,8 +517,9 @@ async function aggregateUsageStats() {
     }
   }
   const daily = [...dayMap.entries()].toSorted((a, b) => a[0] - b[0]).map(([day, p]) => ({
-    ts: day * DAY,
+    ts: day,
     tokens: p.tokens,
+    cost: p.cost,
     messages: p.messages,
     sessions: p.sessions,
     byModel: Object.fromEntries(p.byModel)
@@ -524,14 +529,14 @@ async function aggregateUsageStats() {
   let longestStreak = 0;
   let run = 0;
   for (let i = 0; i < dates.length; i++) {
-    run = i === 0 || dates[i] === (dates[i - 1] ?? 0) + 1 ? run + 1 : 1;
+    run = i === 0 || dates[i] === (dates[i - 1] ?? 0) + DAY ? run + 1 : 1;
     if (run > longestStreak)
       longestStreak = run;
   }
-  const today = Math.floor(Date.now() / DAY);
+  const today = new Date((/* @__PURE__ */ new Date()).getFullYear(), (/* @__PURE__ */ new Date()).getMonth(), (/* @__PURE__ */ new Date()).getDate()).getTime();
   let currentStreak = 0;
   for (let i = dates.length - 1; i >= 0; i--) {
-    if (dates[i] === today - currentStreak) {
+    if (dates[i] === today - currentStreak * DAY) {
       currentStreak++;
     } else {
       break;