batch compress tool inputs

Author: Tarquinen

lib/config.ts

@@ -23,7 +23,6 @@ export interface CompressConfig {
     nudgeFrequency: number
     iterationNudgeThreshold: number
     nudgeForce: "strong" | "soft"
-    flatSchema: boolean
     protectedTools: string[]
     protectUserMessages: boolean
 }
@@ -38,10 +37,6 @@ export interface ManualModeConfig {
     automaticStrategies: boolean
 }
 
-export interface SupersedeWrites {
-    enabled: boolean
-}
-
 export interface PurgeErrors {
     enabled: boolean
     turns: number
@@ -71,7 +66,6 @@ export interface PluginConfig {
     compress: CompressConfig
     strategies: {
         deduplication: Deduplication
-        supersedeWrites: SupersedeWrites
         purgeErrors: PurgeErrors
     }
 }
@@ -124,15 +118,12 @@ export const VALID_CONFIG_KEYS = new Set([
     "compress.nudgeFrequency",
     "compress.iterationNudgeThreshold",
     "compress.nudgeForce",
-    "compress.flatSchema",
     "compress.protectedTools",
     "compress.protectUserMessages",
     "strategies",
     "strategies.deduplication",
     "strategies.deduplication.enabled",
     "strategies.deduplication.protectedTools",
-    "strategies.supersedeWrites",
-    "strategies.supersedeWrites.enabled",
     "strategies.purgeErrors",
     "strategies.purgeErrors.enabled",
     "strategies.purgeErrors.turns",
@@ -404,14 +395,6 @@ export function validateConfigTypes(config: Record<string, any>): ValidationErro
                 })
             }
 
-            if (compress.flatSchema !== undefined && typeof compress.flatSchema !== "boolean") {
-                errors.push({
-                    key: "compress.flatSchema",
-                    expected: "boolean",
-                    actual: typeof compress.flatSchema,
-                })
-            }
-
             if (compress.protectedTools !== undefined && !Array.isArray(compress.protectedTools)) {
                 errors.push({
                     key: "compress.protectedTools",
@@ -547,19 +530,6 @@ export function validateConfigTypes(config: Record<string, any>): ValidationErro
             })
         }
 
-        if (strategies.supersedeWrites) {
-            if (
-                strategies.supersedeWrites.enabled !== undefined &&
-                typeof strategies.supersedeWrites.enabled !== "boolean"
-            ) {
-                errors.push({
-                    key: "strategies.supersedeWrites.enabled",
-                    expected: "boolean",
-                    actual: typeof strategies.supersedeWrites.enabled,
-                })
-            }
-        }
-
         if (strategies.purgeErrors) {
             if (
                 strategies.purgeErrors.enabled !== undefined &&
@@ -685,7 +655,6 @@ const defaultConfig: PluginConfig = {
         nudgeFrequency: 5,
         iterationNudgeThreshold: 15,
         nudgeForce: "soft",
-        flatSchema: false,
         protectedTools: [...COMPRESS_DEFAULT_PROTECTED_TOOLS],
         protectUserMessages: false,
     },
@@ -694,9 +663,6 @@ const defaultConfig: PluginConfig = {
             enabled: true,
             protectedTools: [],
         },
-        supersedeWrites: {
-            enabled: true,
-        },
         purgeErrors: {
             enabled: true,
             turns: 4,
@@ -821,9 +787,6 @@ function mergeStrategies(
                 ]),
             ],
         },
-        supersedeWrites: {
-            enabled: override.supersedeWrites?.enabled ?? base.supersedeWrites.enabled,
-        },
         purgeErrors: {
             enabled: override.purgeErrors?.enabled ?? base.purgeErrors.enabled,
             turns: override.purgeErrors?.turns ?? base.purgeErrors.turns,
@@ -856,7 +819,6 @@ function mergeCompress(
         nudgeFrequency: override.nudgeFrequency ?? base.nudgeFrequency,
         iterationNudgeThreshold: override.iterationNudgeThreshold ?? base.iterationNudgeThreshold,
         nudgeForce: override.nudgeForce ?? base.nudgeForce,
-        flatSchema: override.flatSchema ?? base.flatSchema,
         protectedTools: [...new Set([...base.protectedTools, ...(override.protectedTools ?? [])])],
         protectUserMessages: override.protectUserMessages ?? base.protectUserMessages,
     }
@@ -925,7 +887,6 @@ function deepCloneConfig(config: PluginConfig): PluginConfig {
                 ...config.strategies.deduplication,
                 protectedTools: [...config.strategies.deduplication.protectedTools],
             },
-            supersedeWrites: { ...config.strategies.supersedeWrites },
             purgeErrors: {
                 ...config.strategies.purgeErrors,
                 protectedTools: [...config.strategies.purgeErrors.protectedTools],

lib/prompts/compress-message.ts

@@ -48,21 +48,6 @@ BATCHING
 Do not call the tool once per message. Select MANY messages in a single tool call when they are independently safe to compress.
 Each entry should summarize exactly one message, and the tool can receive as many entries as needed in one batch.
 
-THE FORMAT OF MESSAGE COMPRESS
-
-~~~json
-{
-  "topic": "overall batch label",
-  "content": [
-    {
-      "messageId": "m0001",
-      "topic": "short message label",
-      "summary": "Complete technical summary replacing that one message"
-    }
-  ]
-}
-~~~
-
 Because each message is compressed independently:
 
 - Do not describe ranges

lib/prompts/compress-range.ts

@@ -79,6 +79,6 @@ Rules:
 - Prefer boundaries that produce short, closed ranges.
 - Do not invent IDs. Use only IDs that are present in context.
 
-PARALLEL COMPRESS EXECUTION
-When multiple independent ranges are ready and their boundaries do not overlap, launch MULTIPLE \`compress\` calls in parallel in a single response. This is the PREFERRED pattern over a single large-range compression when the work can be safely split. Run compression sequentially only when ranges overlap or when a later range depends on the result of an earlier compression.
+BATCHING
+Do not call the tool once per range. When multiple independent ranges are ready and their boundaries do not overlap, include all of them as separate entries in the \`content\` array of a single tool call. Each entry should have its own \`startId\`, \`endId\`, and \`summary\`.
 `

lib/prompts/internal-overlays.ts

@@ -18,28 +18,34 @@ All subsequent messages in the session will have IDs.
 </dcp-system-reminder>
 `
 
-export const NESTED_FORMAT_OVERLAY = `
+export const RANGE_FORMAT_OVERLAY = `
 THE FORMAT OF COMPRESS
 
 \`\`\`
 {
   topic: string,           // Short label (3-5 words) - e.g., "Auth System Exploration"
-  content: {
-    startId: string,       // Boundary ID at range start: mNNNN or bN
-    endId: string,         // Boundary ID at range end: mNNNN or bN
-    summary: string        // Complete technical summary replacing all content in range
-  }
+  content: [               // One or more ranges to compress
+    {
+      startId: string,     // Boundary ID at range start: mNNNN or bN
+      endId: string,       // Boundary ID at range end: mNNNN or bN
+      summary: string      // Complete technical summary replacing all content in range
+    }
+  ]
 }
 \`\`\``
 
-export const FLAT_FORMAT_OVERLAY = `
+export const MESSAGE_FORMAT_OVERLAY = `
 THE FORMAT OF COMPRESS
 
 \`\`\`
 {
-  topic: string,           // Short label (3-5 words) - e.g., "Auth System Exploration"
-  startId: string,         // Boundary ID at range start: mNNNN or bN
-  endId: string,           // Boundary ID at range end: mNNNN or bN
-  summary: string          // Complete technical summary replacing all content in range
+  topic: string,           // Short label (3-5 words) for the overall batch
+  content: [               // One or more messages to compress independently
+    {
+      messageId: string,   // Raw message ID only: mNNNN
+      topic: string,       // Short label (3-5 words) for this one message summary
+      summary: string      // Complete technical summary replacing that one message
+    }
+  ]
 }
 \`\`\``

lib/tools/compress-message.ts

@@ -10,7 +10,6 @@ import {
     fetchSessionMessages,
     formatCompressMessageIssues,
     formatCompressMessageResult,
-    COMPRESSED_BLOCK_HEADER,
     resolveMessageCompressions,
     validateCompressMessageArgs,
     type CompressMessageToolArgs,
@@ -21,8 +20,8 @@ import { getCurrentParams, getCurrentTokenUsage, countTokens } from "../strategi
 import { deduplicate, purgeErrors } from "../strategies"
 import { saveSessionState } from "../state/persistence"
 import { sendCompressNotification } from "../ui/notification"
+import { MESSAGE_FORMAT_OVERLAY } from "../prompts/internal-overlays"
 
-// Non-primitive arrays are hidden in the TUI, so keep the schema simple and explicit.
 function buildSchema() {
     return {
         topic: tool.schema
@@ -53,7 +52,7 @@ export function createCompressMessageTool(ctx: ToolContext): ReturnType<typeof t
     const runtimePrompts = ctx.prompts.getRuntimePrompts()
 
     return tool({
-        description: runtimePrompts.compressMessage,
+        description: runtimePrompts.compressMessage + MESSAGE_FORMAT_OVERLAY,
         args: buildSchema(),
         async execute(args, toolCtx) {
             if (ctx.state.manualMode && ctx.state.manualMode !== "compress-pending") {
@@ -109,6 +108,11 @@ export function createCompressMessageTool(ctx: ToolContext): ReturnType<typeof t
                 summaryTokens: number
             }> = []
 
+            const preparedPlans: Array<{
+                plan: (typeof plans)[number]
+                summaryWithProtectedTools: string
+            }> = []
+
             for (const plan of plans) {
                 const summaryWithProtectedTools = await appendProtectedTools(
                     ctx.client,
@@ -121,6 +125,13 @@ export function createCompressMessageTool(ctx: ToolContext): ReturnType<typeof t
                     ctx.config.protectedFilePatterns,
                 )
 
+                preparedPlans.push({
+                    plan,
+                    summaryWithProtectedTools,
+                })
+            }
+
+            for (const { plan, summaryWithProtectedTools } of preparedPlans) {
                 const blockId = allocateBlockId(ctx.state)
                 const storedSummary = wrapCompressedSummary(blockId, summaryWithProtectedTools)
                 const summaryTokens = countTokens(storedSummary)