Skip to content

Commit 5e84f1a

Browse files
committed
fix: include GC tokens in session total and improve icon placement
1 parent 42ba48f commit 5e84f1a

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

lib/ui/notification.ts

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ function buildMinimalMessage(data: NotificationData): string {
8585

8686
let cycleStats = `${data.aiPrunedCount} ${toolText}`
8787
if (hasGcActivity) {
88-
cycleStats += `, 🗑️ ~${formatTokenCount(data.gcPending!.tokensCollected)}`
88+
cycleStats += `, ~${formatTokenCount(data.gcPending!.tokensCollected)} 🗑️`
8989
}
9090

9191
let message = `🧹 DCP: ~${totalSaved} saved (${cycleStats})`
@@ -115,7 +115,7 @@ function buildDetailedMessage(data: NotificationData, workingDirectory?: string)
115115

116116
let cycleStats = `${data.aiPrunedCount} ${toolText}`
117117
if (hasGcActivity) {
118-
cycleStats += `, 🗑️ ~${formatTokenCount(data.gcPending!.tokensCollected)}`
118+
cycleStats += `, ~${formatTokenCount(data.gcPending!.tokensCollected)} 🗑️`
119119
}
120120

121121
message = `🧹 DCP: ~${totalSaved} saved (${cycleStats})`
@@ -163,10 +163,11 @@ function buildSessionSuffix(sessionStats: SessionStats | null, currentAiPruned:
163163
return ''
164164
}
165165

166-
let suffix = ` │ Session: ~${formatTokenCount(sessionStats.totalTokensSaved)} (${sessionStats.totalToolsPruned} tools`
166+
const totalSaved = sessionStats.totalTokensSaved + sessionStats.totalGCTokens
167+
let suffix = ` │ Session: ~${formatTokenCount(totalSaved)} (${sessionStats.totalToolsPruned} tools`
167168

168169
if (sessionStats.totalGCTokens > 0) {
169-
suffix += `, 🗑️ ~${formatTokenCount(sessionStats.totalGCTokens)}`
170+
suffix += `, ~${formatTokenCount(sessionStats.totalGCTokens)} 🗑️`
170171
}
171172

172173
suffix += ')'

0 commit comments

Comments
 (0)