Skip to content

Commit 0d9fb23

Browse files
committed
fix: show combined AI + GC tokens in notification headline
1 parent 4800cf7 commit 0d9fb23

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

lib/ui/notification.ts

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -79,15 +79,16 @@ function buildMinimalMessage(data: NotificationData): string {
7979
const hasGcActivity = data.gcPending && data.gcPending.toolsDeduped > 0
8080

8181
if (hasAiPruning) {
82-
const tokensSaved = formatTokenCount(data.aiTokensSaved)
82+
const gcTokens = hasGcActivity ? data.gcPending!.tokensCollected : 0
83+
const totalSaved = formatTokenCount(data.aiTokensSaved + gcTokens)
8384
const toolText = data.aiPrunedCount === 1 ? 'tool' : 'tools'
8485

8586
let cycleStats = `${data.aiPrunedCount} ${toolText}`
8687
if (hasGcActivity) {
8788
cycleStats += `, ♻️ ~${formatTokenCount(data.gcPending!.tokensCollected)}`
8889
}
8990

90-
let message = `🧹 DCP: ~${tokensSaved} saved (${cycleStats})`
91+
let message = `🧹 DCP: ~${totalSaved} saved (${cycleStats})`
9192
message += buildSessionSuffix(data.sessionStats, data.aiPrunedCount)
9293

9394
return message
@@ -108,15 +109,16 @@ function buildDetailedMessage(data: NotificationData, workingDirectory?: string)
108109
let message: string
109110

110111
if (hasAiPruning) {
111-
const tokensSaved = formatTokenCount(data.aiTokensSaved)
112+
const gcTokens = hasGcActivity ? data.gcPending!.tokensCollected : 0
113+
const totalSaved = formatTokenCount(data.aiTokensSaved + gcTokens)
112114
const toolText = data.aiPrunedCount === 1 ? 'tool' : 'tools'
113115

114116
let cycleStats = `${data.aiPrunedCount} ${toolText}`
115117
if (hasGcActivity) {
116118
cycleStats += `, ♻️ ~${formatTokenCount(data.gcPending!.tokensCollected)}`
117119
}
118120

119-
message = `🧹 DCP: ~${tokensSaved} saved (${cycleStats})`
121+
message = `🧹 DCP: ~${totalSaved} saved (${cycleStats})`
120122
message += buildSessionSuffix(data.sessionStats, data.aiPrunedCount)
121123
message += '\n'
122124

0 commit comments

Comments
 (0)