fix(ui): revert compress notification to match dev

Author: Tarquinen

lib/compress/pipeline.ts

@@ -4,7 +4,7 @@ import { saveSessionState } from "../state/persistence"
 import { assignMessageRefs } from "../message-ids"
 import { isIgnoredUserMessage } from "../messages/utils"
 import { deduplicate, purgeErrors } from "../strategies"
-import { getCurrentParams, getCurrentTokenUsage } from "../strategies/utils"
+import { getCurrentParams } from "../strategies/utils"
 import { sendCompressNotification } from "../ui/notification"
 import type { ToolContext } from "./types"
 import { buildSearchContext, fetchSessionMessages } from "./search"
@@ -86,10 +86,9 @@ export async function finalizeSession(
     await saveSessionState(ctx.state, ctx.logger)
 
     const params = getCurrentParams(ctx.state, rawMessages, ctx.logger)
-    const totalSessionTokens = getCurrentTokenUsage(ctx.state, rawMessages)
-    const sessionMessages = rawMessages.filter(
-        (msg) => !(msg.info.role === "user" && isIgnoredUserMessage(msg)),
-    )
+    const sessionMessageIds = rawMessages
+        .filter((msg) => !isIgnoredUserMessage(msg))
+        .map((msg) => msg.info.id)
 
     await sendCompressNotification(
         ctx.client,
@@ -99,8 +98,7 @@ export async function finalizeSession(
         toolCtx.sessionID,
         entries,
         batchTopic,
-        totalSessionTokens,
-        sessionMessages,
+        sessionMessageIds,
         params,
     )
 }

lib/ui/notification.ts

@@ -1,5 +1,5 @@
 import type { Logger } from "../logger"
-import type { SessionState, WithParts } from "../state"
+import type { SessionState } from "../state"
 import {
     formatPrunedItemsList,
     formatProgressBar,
@@ -9,7 +9,6 @@ import {
 import { ToolParameterEntry } from "../state"
 import { PluginConfig } from "../config"
 import { getActiveSummaryTokenUsage } from "../state/utils"
-import { isMessageCompacted } from "../shared-utils"
 
 export type PruneReason = "completion" | "noise" | "extraction"
 export const PRUNE_REASON_LABELS: Record<PruneReason, string> = {
@@ -170,8 +169,7 @@ export async function sendCompressNotification(
     sessionId: string,
     entries: CompressionNotificationEntry[],
     batchTopic: string | undefined,
-    totalSessionTokens: number,
-    sessionMessages: WithParts[],
+    sessionMessageIds: string[],
     params: any,
 ): Promise<boolean> {
     if (config.pruneNotification === "off") {
@@ -249,12 +247,10 @@ export async function sendCompressNotification(
 
         const pruneTokenCounterStr = `~${formatTokenCount(compressedTokens)}`
 
-        const sessionMessageIds = sessionMessages.map((msg) => msg.info.id)
         const activePrunedMessages = new Map<string, number>()
-        for (const msg of sessionMessages) {
-            if (isMessageCompacted(state, msg)) {
-                const pruneEntry = state.prune.messages.byMessageId.get(msg.info.id)
-                activePrunedMessages.set(msg.info.id, pruneEntry?.tokenCount ?? 0)
+        for (const [messageId, entry] of state.prune.messages.byMessageId) {
+            if (entry.activeBlockIds.length > 0) {
+                activePrunedMessages.set(messageId, entry.tokenCount)
             }
         }
         const progressBar = formatProgressBar(
@@ -263,11 +259,8 @@ export async function sendCompressNotification(
             newlyCompressedMessageIds,
             70,
         )
-        const reduction =
-            totalSessionTokens > 0 ? Math.round((compressedTokens / totalSessionTokens) * 100) : 0
-
         message += `\n\n${progressBar}`
-        message += `\n▣ ${compressionLabel} (${pruneTokenCounterStr} removed, ${reduction}% reduction, ~${formatTokenCount(summaryTokens, true)} summary tokens added)`
+        message += `\n▣ ${compressionLabel} (${pruneTokenCounterStr} removed, ~${formatTokenCount(summaryTokens, true)} summary tokens added)`
         message += `\n→ Topic: ${topic}`
         message += `\n→ Items: ${newlyCompressedMessageIds.length} messages`
         if (newlyCompressedToolIds.length > 0) {

lib/ui/utils.ts

@@ -144,10 +144,7 @@ export function formatStatsHeader(totalTokensSaved: number, pruneTokenCounter: n
 
 export function formatTokenCount(tokens: number, compact?: boolean): string {
     const suffix = compact ? "" : " tokens"
-    if (tokens >= 100_000) {
-        return `${Math.round(tokens / 1000)}K` + suffix
-    }
-    if (tokens >= 10_000 || (compact && tokens >= 1000)) {
+    if (tokens >= 1000) {
         return `${(tokens / 1000).toFixed(1)}K`.replace(".0K", "K") + suffix
     }
     return tokens.toString() + suffix