Add configurable pruning summary modes and fix input minimization for state-modifying tools

Author: Tarquinen

index.ts

@@ -56,7 +56,7 @@ const plugin: Plugin = (async (ctx) => {
     const stateManager = new StateManager()
     const toolParametersCache = new Map<string, any>() // callID -> parameters
     const modelCache = new Map<string, { providerID: string; modelID: string }>() // sessionID -> model info
-    const janitor = new Janitor(ctx.client, stateManager, logger, toolParametersCache, config.protectedTools, modelCache, config.model, config.showModelErrorToasts, config.pruningMode)
+    const janitor = new Janitor(ctx.client, stateManager, logger, toolParametersCache, config.protectedTools, modelCache, config.model, config.showModelErrorToasts, config.pruningMode, config.pruning_summary)
 
     const cacheToolParameters = (messages: any[], component: string) => {
         for (const message of messages) {
@@ -244,6 +244,7 @@ const plugin: Plugin = (async (ctx) => {
         protectedTools: config.protectedTools,
         model: config.model,
         pruningMode: config.pruningMode,
+        pruning_summary: config.pruning_summary,
         globalConfigFile: join(homedir(), ".config", "opencode", "dcp.jsonc"),
         projectConfigFile: ctx.directory ? join(ctx.directory, ".opencode", "dcp.jsonc") : "N/A",
         logDirectory: join(homedir(), ".config", "opencode", "logs", "dcp"),

lib/config.ts

@@ -13,14 +13,16 @@ export interface PluginConfig {
     model?: string // Format: "provider/model" (e.g., "anthropic/claude-haiku-4-5")
     showModelErrorToasts?: boolean // Show toast notifications when model selection fails
     pruningMode: "auto" | "smart" // Pruning strategy: auto (deduplication only) or smart (deduplication + LLM analysis)
+    pruning_summary: "off" | "minimal" | "detailed" // UI summary display mode
 }
 
 const defaultConfig: PluginConfig = {
     enabled: true, // Plugin is enabled by default
     debug: false, // Disable debug logging by default
     protectedTools: ['task', 'todowrite', 'todoread'], // Tools that should never be pruned (including stateful tools)
     showModelErrorToasts: true, // Show model error toasts by default
-    pruningMode: 'smart' // Default to smart mode (deduplication + LLM analysis)
+    pruningMode: 'smart', // Default to smart mode (deduplication + LLM analysis)
+    pruning_summary: 'detailed' // Default to detailed summary
 }
 
 const GLOBAL_CONFIG_DIR = join(homedir(), '.config', 'opencode')
@@ -109,6 +111,12 @@ function createDefaultConfig(): void {
   // "smart": Deduplication + AI analysis for intelligent pruning (recommended)
   "pruningMode": "smart",
 
+  // Pruning summary display mode:
+  // "off": No UI summary (silent pruning)
+  // "minimal": Show tokens saved and count (e.g., "Saved ~2.5K tokens (6 tools pruned)")
+  // "detailed": Show full breakdown by tool type and pruning method (default)
+  "pruning_summary": "detailed",
+
   // List of tools that should never be pruned from context
   // "task": Each subagent invocation is intentional
   // "todowrite"/"todoread": Stateful tools where each call matters
@@ -161,7 +169,8 @@ export function getConfig(ctx?: PluginInput): PluginConfig {
                 protectedTools: globalConfig.protectedTools ?? config.protectedTools,
                 model: globalConfig.model ?? config.model,
                 showModelErrorToasts: globalConfig.showModelErrorToasts ?? config.showModelErrorToasts,
-                pruningMode: globalConfig.pruningMode ?? config.pruningMode
+                pruningMode: globalConfig.pruningMode ?? config.pruningMode,
+                pruning_summary: globalConfig.pruning_summary ?? config.pruning_summary
             }
             logger.info('config', 'Loaded global config', { path: configPaths.global })
         }
@@ -181,7 +190,8 @@ export function getConfig(ctx?: PluginInput): PluginConfig {
                 protectedTools: projectConfig.protectedTools ?? config.protectedTools,
                 model: projectConfig.model ?? config.model,
                 showModelErrorToasts: projectConfig.showModelErrorToasts ?? config.showModelErrorToasts,
-                pruningMode: projectConfig.pruningMode ?? config.pruningMode
+                pruningMode: projectConfig.pruningMode ?? config.pruningMode,
+                pruning_summary: projectConfig.pruning_summary ?? config.pruning_summary
             }
             logger.info('config', 'Loaded project config (overrides global)', { path: configPaths.project })
         }

lib/janitor.ts

@@ -16,7 +16,8 @@ export class Janitor {
         private modelCache: Map<string, { providerID: string; modelID: string }>,
         private configModel?: string, // Format: "provider/model"
         private showModelErrorToasts: boolean = true, // Whether to show toast for model errors
-        private pruningMode: "auto" | "smart" = "smart" // Pruning strategy
+        private pruningMode: "auto" | "smart" = "smart", // Pruning strategy
+        private pruningSummary: "off" | "minimal" | "detailed" = "detailed" // UI summary display mode
     ) { }
 
     /**
@@ -583,6 +584,26 @@ export class Janitor {
         return toolsSummary
     }
 
+    /**
+     * Send minimal summary notification (just tokens saved and count)
+     */
+    private async sendMinimalNotification(
+        sessionID: string,
+        totalPruned: number,
+        toolOutputs: Map<string, string>,
+        prunedIds: string[]
+    ) {
+        if (totalPruned === 0) return
+
+        const tokensSaved = await this.calculateTokensSaved(prunedIds, toolOutputs)
+        const tokensFormatted = formatTokenCount(tokensSaved)
+        const toolText = totalPruned === 1 ? 'tool' : 'tools'
+
+        const message = `🧹 DCP: Saved ~${tokensFormatted} tokens (${totalPruned} ${toolText} pruned)`
+
+        await this.sendIgnoredMessage(sessionID, message)
+    }
+
     /**
      * Auto mode notification - shows only deduplication results
      */
@@ -594,6 +615,16 @@ export class Janitor {
     ) {
         if (deduplicatedIds.length === 0) return
 
+        // Check if notifications are disabled
+        if (this.pruningSummary === 'off') return
+
+        // Send minimal notification if configured
+        if (this.pruningSummary === 'minimal') {
+            await this.sendMinimalNotification(sessionID, deduplicatedIds.length, toolOutputs, deduplicatedIds)
+            return
+        }
+
+        // Otherwise send detailed notification
         // Calculate token savings
         const tokensSaved = await this.calculateTokensSaved(deduplicatedIds, toolOutputs)
         const tokensFormatted = formatTokenCount(tokensSaved)
@@ -647,6 +678,17 @@ export class Janitor {
         const totalPruned = deduplicatedIds.length + llmPrunedIds.length
         if (totalPruned === 0) return
 
+        // Check if notifications are disabled
+        if (this.pruningSummary === 'off') return
+
+        // Send minimal notification if configured
+        if (this.pruningSummary === 'minimal') {
+            const allPrunedIds = [...deduplicatedIds, ...llmPrunedIds]
+            await this.sendMinimalNotification(sessionID, totalPruned, toolOutputs, allPrunedIds)
+            return
+        }
+
+        // Otherwise send detailed notification
         // Calculate token savings
         const allPrunedIds = [...deduplicatedIds, ...llmPrunedIds]
         const tokensSaved = await this.calculateTokensSaved(allPrunedIds, toolOutputs)

lib/logger.ts

@@ -89,9 +89,9 @@ export class Logger {
                 ? idsMatch[1].split(',').map(id => id.trim())
                 : []
 
-            // Extract session history (between "Session history:\n" and "\n\nYou MUST respond")
+            // Extract session history (between "Session history" and "\n\nYou MUST respond")
             // The captured text has literal newlines, so we need to escape them back to \n for valid JSON
-            const historyMatch = prompt.match(/Session history:\s*\n([\s\S]*?)\n\nYou MUST respond/)
+            const historyMatch = prompt.match(/Session history[^\n]*:\s*\n([\s\S]*?)\n\nYou MUST respond/)
             let sessionHistory: any[] = []
 
             if (historyMatch) {
@@ -113,7 +113,8 @@ export class Logger {
 
             // Extract response schema (after "You MUST respond with valid JSON matching this exact schema:")
             // Note: The schema contains "..." placeholders which aren't valid JSON, so we save it as a string
-            const schemaMatch = prompt.match(/matching this exact schema:\s*\n(\{[\s\S]*?\})\s*\n\nReturn ONLY/)
+            // Now matches until end of prompt since we removed the "Return ONLY..." line
+            const schemaMatch = prompt.match(/matching this exact schema:\s*\n(\{[\s\S]*?\})\s*$/)
             const responseSchema = schemaMatch 
                 ? schemaMatch[1] // Keep as string since it has "..." placeholders
                 : null

lib/prompt.ts

@@ -65,8 +65,13 @@ function minimizeMessages(messages: any[], alreadyPrunedIds?: string[], protecte
                         if (part.state?.input) {
                             const input = part.state.input
 
-                            // For file operations, just keep the file path
-                            if (input.filePath) {
+                            // For write/edit tools, keep file path AND content (what was changed matters)
+                            // These tools: write, edit, multiedit, patch
+                            if (input.filePath && (part.tool === 'write' || part.tool === 'edit' || part.tool === 'multiedit' || part.tool === 'patch')) {
+                                toolPart.input = input // Keep full input (content, oldString, newString, etc.)
+                            }
+                            // For read-only file operations, just keep the file path
+                            else if (input.filePath) {
                                 toolPart.input = { filePath: input.filePath }
                             }
                             // For batch operations, summarize instead of full array