fix: remove orphaned prune origin tracking from dev-only commits

Author: Tarquinen

lib/commands/sweep.ts

@@ -219,21 +219,11 @@ export async function handleSweepCommand(ctx: SweepCommandContext): Promise<void
     }
 
     const tokensSaved = getTotalToolTokens(state, newToolIds)
-    const originMessageId = getLastUserMessage(messages)?.info.id || ""
-    if (!originMessageId) {
-        logger.warn("Sweep prune origin unavailable - missing user message")
-    }
 
     // Add to prune list
     for (const id of newToolIds) {
         const entry = state.toolParameters.get(id)
         state.prune.tools.set(id, entry?.tokenCount ?? 0)
-        if (originMessageId) {
-            state.prune.origins.set(id, {
-                source: "sweep",
-                originMessageId,
-            })
-        }
     }
     state.stats.pruneTokenCounter += tokensSaved
     state.stats.totalPruneTokens += state.stats.pruneTokenCounter

lib/state/utils.ts

@@ -255,27 +255,6 @@ export function collectTurnNudgeAnchors(messages: WithParts[]): Set<string> {
     return anchors
 }
 
-export function loadPruneOriginMap(obj?: Record<string, PruneOrigin>): Map<string, PruneOrigin> {
-    if (!obj || typeof obj !== "object") {
-        return new Map()
-    }
-
-    const entries: [string, PruneOrigin][] = []
-    for (const [toolId, origin] of Object.entries(obj)) {
-        if (
-            origin &&
-            typeof origin === "object" &&
-            typeof origin.source === "string" &&
-            typeof origin.originMessageId === "string" &&
-            origin.originMessageId.length > 0
-        ) {
-            entries.push([toolId, origin])
-        }
-    }
-
-    return new Map(entries)
-}
-
 export function resetOnCompaction(state: SessionState): void {
     state.toolParameters.clear()
     state.prune.tools = new Map<string, number>()

lib/strategies/deduplication.ts

@@ -83,21 +83,11 @@ export const deduplicate = (
     }
 
     state.stats.totalPruneTokens += getTotalToolTokens(state, newPruneIds)
-    const decisionMessageId = getLastUserMessage(messages)?.info.id || ""
 
     if (newPruneIds.length > 0) {
-        if (!decisionMessageId) {
-            logger.warn("Deduplication prune origin unavailable - missing user message")
-        }
         for (const id of newPruneIds) {
             const entry = state.toolParameters.get(id)
             state.prune.tools.set(id, entry?.tokenCount ?? 0)
-            if (decisionMessageId) {
-                state.prune.origins.set(id, {
-                    source: "deduplication",
-                    originMessageId: decisionMessageId,
-                })
-            }
         }
         logger.debug(`Marked ${newPruneIds.length} duplicate tool calls for pruning`)
     }

lib/strategies/purge-errors.ts

@@ -76,20 +76,10 @@ export const purgeErrors = (
     }
 
     if (newPruneIds.length > 0) {
-        const decisionMessageId = getLastUserMessage(messages)?.info.id || ""
-        if (!decisionMessageId) {
-            logger.warn("Purge errors prune origin unavailable - missing user message")
-        }
         state.stats.totalPruneTokens += getTotalToolTokens(state, newPruneIds)
         for (const id of newPruneIds) {
             const entry = state.toolParameters.get(id)
             state.prune.tools.set(id, entry?.tokenCount ?? 0)
-            if (decisionMessageId) {
-                state.prune.origins.set(id, {
-                    source: "purgeErrors",
-                    originMessageId: decisionMessageId,
-                })
-            }
         }
         logger.debug(
             `Marked ${newPruneIds.length} error tool calls for pruning (older than ${turnThreshold} turns)`,

lib/strategies/supersede-writes.ts

@@ -105,20 +105,10 @@ export const supersedeWrites = (
     }
 
     if (newPruneIds.length > 0) {
-        const decisionMessageId = getLastUserMessage(messages)?.info.id || ""
-        if (!decisionMessageId) {
-            logger.warn("Supersede writes prune origin unavailable - missing user message")
-        }
         state.stats.totalPruneTokens += getTotalToolTokens(state, newPruneIds)
         for (const id of newPruneIds) {
             const entry = state.toolParameters.get(id)
             state.prune.tools.set(id, entry?.tokenCount ?? 0)
-            if (decisionMessageId) {
-                state.prune.origins.set(id, {
-                    source: "supersedeWrites",
-                    originMessageId: decisionMessageId,
-                })
-            }
         }
         logger.debug(`Marked ${newPruneIds.length} superseded write tool calls for pruning`)
     }