fix: persist session usage metadata on suppressed replies

src/auto-reply/reply/session-usage.ts

@@ -0,0 +1,92 @@
+import { setCliSessionId } from "../../agents/cli-session.js";
+import { hasNonzeroUsage, type NormalizedUsage } from "../../agents/usage.js";
+import {
+  type SessionSystemPromptReport,
+  type SessionEntry,
+  updateSessionStoreEntry,
+} from "../../config/sessions.js";
+import { logVerbose } from "../../globals.js";
+
+export async function persistSessionUsageUpdate(params: {
+  storePath?: string;
+  sessionKey?: string;
+  usage?: NormalizedUsage;
+  modelUsed?: string;
+  providerUsed?: string;
+  contextTokensUsed?: number;
+  systemPromptReport?: SessionSystemPromptReport;
+  cliSessionId?: string;
+  logLabel?: string;
+}): Promise<void> {
+  const { storePath, sessionKey } = params;
+  if (!storePath || !sessionKey) return;
+
+  const label = params.logLabel ? `${params.logLabel} ` : "";
+  if (hasNonzeroUsage(params.usage)) {
+    try {
+      await updateSessionStoreEntry({
+        storePath,
+        sessionKey,
+        update: async (entry) => {
+          const input = params.usage?.input ?? 0;
+          const output = params.usage?.output ?? 0;
+          const promptTokens =
+            input + (params.usage?.cacheRead ?? 0) + (params.usage?.cacheWrite ?? 0);
+          const patch: Partial<SessionEntry> = {
+            inputTokens: input,
+            outputTokens: output,
+            totalTokens: promptTokens > 0 ? promptTokens : (params.usage?.total ?? input),
+            modelProvider: params.providerUsed ?? entry.modelProvider,
+            model: params.modelUsed ?? entry.model,
+            contextTokens: params.contextTokensUsed ?? entry.contextTokens,
+            systemPromptReport: params.systemPromptReport ?? entry.systemPromptReport,
+            updatedAt: Date.now(),
+          };
+          if (params.cliSessionId) {
+            const nextEntry = { ...entry, ...patch };
+            setCliSessionId(nextEntry, params.providerUsed, params.cliSessionId);
+            return {
+              ...patch,
+              cliSessionIds: nextEntry.cliSessionIds,
+              claudeCliSessionId: nextEntry.claudeCliSessionId,
+            };
+          }
+          return patch;
+        },
+      });
+    } catch (err) {
+      logVerbose(`failed to persist ${label}usage update: ${String(err)}`);
+    }
+    return;
+  }
+
+  if (params.modelUsed || params.contextTokensUsed) {
+    try {
+      await updateSessionStoreEntry({
+        storePath,
+        sessionKey,
+        update: async (entry) => {
+          const patch: Partial<SessionEntry> = {
+            modelProvider: params.providerUsed ?? entry.modelProvider,
+            model: params.modelUsed ?? entry.model,
+            contextTokens: params.contextTokensUsed ?? entry.contextTokens,
+            systemPromptReport: params.systemPromptReport ?? entry.systemPromptReport,
+            updatedAt: Date.now(),
+          };
+          if (params.cliSessionId) {
+            const nextEntry = { ...entry, ...patch };
+            setCliSessionId(nextEntry, params.providerUsed, params.cliSessionId);
+            return {
+              ...patch,
+              cliSessionIds: nextEntry.cliSessionIds,
+              claudeCliSessionId: nextEntry.claudeCliSessionId,
+            };
+          }
+          return patch;
+        },
+      });
+    } catch (err) {
+      logVerbose(`failed to persist ${label}model/context update: ${String(err)}`);
+    }
+  }
+}