fix(status): preserve legacy context totals

ashishpatel26 · ashishpatel26 · commit 8ca9ac38ef2b · 2026-06-13T18:24:41.000+05:30
diff --git a/src/auto-reply/status.test.ts b/src/auto-reply/status.test.ts
@@ -222,6 +222,30 @@ describe("buildStatusMessage", () => {
     expect(normalized).not.toContain("Context: 3.8m/1.0m");
   });
 
+  it("preserves legacy unknown-freshness totalTokens as context usage", () => {
+    const text = buildStatusMessage({
+      agent: {
+        model: "anthropic/test:opus",
+        contextTokens: 1_000_000,
+      },
+      sessionEntry: {
+        sessionId: "abc",
+        updatedAt: 0,
+        totalTokens: 25_000,
+        contextTokens: 1_000_000,
+      },
+      sessionKey: "agent:main:main",
+      sessionScope: "per-sender",
+      queue: { mode: "collect", depth: 0 },
+      modelAuth: "api-key",
+      now: 10 * 60_000,
+    });
+    const normalized = normalizeTestText(text);
+
+    expect(normalized).toContain("Context: 25k/1.0m");
+    expect(normalized).not.toContain("Context: ?/1.0m");
+  });
+
   it("uses estimated context budget status when fresh totalTokens are unavailable", () => {
     const text = buildStatusMessage({
       agent: {
diff --git a/src/status/status-message.ts b/src/status/status-message.ts
@@ -632,13 +632,14 @@ export function buildStatusMessage(args: StatusArgs): string {
   let outputTokens = entry?.outputTokens;
   let cacheRead = entry?.cacheRead;
   let cacheWrite = entry?.cacheWrite;
-  const freshTotalTokens =
-    entry?.totalTokensFresh === true ? resolveFreshSessionTotalTokens(entry) : undefined;
+  const freshTotalTokens = resolveFreshSessionTotalTokens(entry);
+  // Undefined freshness is legacy, not stale: keep persisted totals for /status,
+  // but let a fresh transcript prompt snapshot replace them when available.
   const allowTranscriptContextUsage = entry?.totalTokensFresh !== false;
   let totalTokens = freshTotalTokens;
 
-  // Context percentages need a fresh prompt snapshot; cumulative session/cache
-  // usage can still hydrate Tokens/Cache lines but must not become Context.
+  // Explicitly stale session/cache usage can still hydrate Tokens/Cache lines
+  // but must not become Context.
   if (args.includeTranscriptUsage) {
     const logUsage = readUsageFromSessionLog(
       entry?.sessionId,
@@ -655,7 +656,10 @@ export function buildStatusMessage(args: StatusArgs): string {
         allowTranscriptContextUsage &&
         candidate !== undefined &&
         candidate > 0 &&
-        (!totalTokens || totalTokens === 0 || candidate > totalTokens)
+        (entry?.totalTokensFresh !== true ||
+          !totalTokens ||
+          totalTokens === 0 ||
+          candidate > totalTokens)
       ) {
         totalTokens = candidate;
       }
