simplify compress notifications

Author: Tarquinen

lib/tools/compress-message.ts

@@ -20,7 +20,7 @@ import { assignMessageRefs } from "../message-ids"
 import { getCurrentParams, getCurrentTokenUsage, countTokens } from "../strategies/utils"
 import { deduplicate, purgeErrors } from "../strategies"
 import { saveSessionState } from "../state/persistence"
-import { sendCompressBatchNotification } from "../ui/notification"
+import { sendCompressNotification } from "../ui/notification"
 
 // Non-primitive arrays are hidden in the TUI, so keep the schema simple and explicit.
 function buildSchema() {
@@ -156,7 +156,7 @@ export function createCompressMessageTool(ctx: ToolContext): ReturnType<typeof t
                 .filter((msg) => !(msg.info.role === "user" && isIgnoredUserMessage(msg)))
                 .map((msg) => msg.info.id)
 
-            await sendCompressBatchNotification(
+            await sendCompressNotification(
                 ctx.client,
                 ctx.logger,
                 ctx.config,

lib/tools/compress-range.ts

@@ -204,16 +204,22 @@ export function createCompressRangeTool(ctx: ToolContext): ReturnType<typeof too
             const sessionMessageIds = rawMessages
                 .filter((msg) => !(msg.info.role === "user" && isIgnoredUserMessage(msg)))
                 .map((msg) => msg.info.id)
+            const notificationEntries = [
+                {
+                    blockId,
+                    summary: compressRangeArgs.content.summary,
+                    summaryTokens,
+                },
+            ]
 
             await sendCompressNotification(
                 ctx.client,
                 ctx.logger,
                 ctx.config,
                 ctx.state,
                 toolCtx.sessionID,
-                blockId,
-                compressRangeArgs.content.summary,
-                summaryTokens,
+                notificationEntries,
+                undefined,
                 totalSessionTokens,
                 sessionMessageIds,
                 params,

lib/ui/notification.ts

@@ -133,39 +133,6 @@ export async function sendUnifiedNotification(
     return true
 }
 
-export async function sendCompressNotification(
-    client: any,
-    logger: Logger,
-    config: PluginConfig,
-    state: SessionState,
-    sessionId: string,
-    compressionId: number,
-    summary: string,
-    summaryTokens: number,
-    totalSessionTokens: number,
-    sessionMessageIds: string[],
-    params: any,
-): Promise<boolean> {
-    return sendCompressBatchNotification(
-        client,
-        logger,
-        config,
-        state,
-        sessionId,
-        [
-            {
-                blockId: compressionId,
-                summary,
-                summaryTokens,
-            },
-        ],
-        undefined,
-        totalSessionTokens,
-        sessionMessageIds,
-        params,
-    )
-}
-
 function buildCompressionSummary(
     entries: CompressionNotificationEntry[],
     state: SessionState,
@@ -196,7 +163,7 @@ function getCompressionLabel(entries: CompressionNotificationEntry[]): string {
     return `Compression #${firstBlockId}`
 }
 
-export async function sendCompressBatchNotification(
+export async function sendCompressNotification(
     client: any,
     logger: Logger,
     config: PluginConfig,