Merge pull request #129 from Opencode-DCP/fix/persist-token-stats-after-update

fix: save session state after updating token stats

Author: Tarquinen

lib/strategies/prune-tool.ts

@@ -73,9 +73,6 @@ export function createPruneTool(
             const pruneToolIds: string[] = getPruneToolIds(numericToolIds, toolIdList)
             state.prune.toolIds.push(...pruneToolIds)
 
-            saveSessionState(state, logger)
-                .catch(err => logger.error("Failed to persist state", { error: err.message }))
-
             const toolMetadata = new Map<string, ToolParameterEntry>()
             for (const id of pruneToolIds) {
                 const toolParameters = state.toolParameters.get(id)
@@ -104,6 +101,9 @@ export function createPruneTool(
             state.stats.pruneTokenCounter = 0
             state.nudgeCounter = 0
 
+            saveSessionState(state, logger)
+                .catch(err => logger.error("Failed to persist state", { error: err.message }))
+
             return formatPruningResultForTool(
                 pruneToolIds,
                 toolMetadata,