Skip to content

Commit 5182362

Browse files
feat: add compression summary tokens amount to dcp stats
1 parent b55b57c commit 5182362

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

lib/commands/stats.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ function formatStatsMessage(
2323
sessionTokens: number,
2424
sessionTools: number,
2525
sessionMessages: number,
26+
sessionSummaryTokens: number,
2627
sessionDurationMs: number,
2728
allTime: AggregatedStats,
2829
): string {
@@ -37,6 +38,7 @@ function formatStatsMessage(
3738
lines.push(` Tokens pruned: ~${formatTokenCount(sessionTokens)}`)
3839
lines.push(` Tools pruned: ${sessionTools}`)
3940
lines.push(` Messages pruned: ${sessionMessages}`)
41+
lines.push(` Summary tokens: ~${formatTokenCount(sessionSummaryTokens)}`)
4042
lines.push(` Compression time: ${formatCompressionTime(sessionDurationMs)}`)
4143
lines.push("")
4244
lines.push("All-time:")
@@ -77,6 +79,10 @@ export async function handleStatsCommand(ctx: StatsCommandContext): Promise<void
7779

7880
// Session stats from in-memory state
7981
const sessionTokens = state.stats.totalPruneTokens
82+
const sessionSummaryTokens = Array.from(state.prune.messages.blocksById.values()).reduce(
83+
(total, block) => (block.active ? total + block.summaryTokens : total),
84+
0,
85+
)
8086
const sessionDurationMs = getActiveCompressionTargets(state.prune.messages).reduce(
8187
(total, target) => total + target.durationMs,
8288
0,
@@ -106,6 +112,7 @@ export async function handleStatsCommand(ctx: StatsCommandContext): Promise<void
106112
sessionTokens,
107113
sessionTools,
108114
sessionMessages,
115+
sessionSummaryTokens,
109116
sessionDurationMs,
110117
allTime,
111118
)
@@ -117,6 +124,7 @@ export async function handleStatsCommand(ctx: StatsCommandContext): Promise<void
117124
sessionTokens,
118125
sessionTools,
119126
sessionMessages,
127+
sessionSummaryTokens,
120128
sessionDurationMs,
121129
allTimeTokens: allTime.totalTokens,
122130
allTimeTools: allTime.totalTools,

0 commit comments

Comments
 (0)