feat: add compression time to /dcp stats

refactor

refactor: natural time display

fix: format check

fix: stabilize compression timing correlation

refactor: simplify compression timing

fix: restore compression timing durability

fix: attach compression timing to blocks

fix: make compression timing robust

Author: spoons-and-mirrors

index.ts

@@ -13,6 +13,7 @@ import {
     createChatMessageHandler,
     createChatMessageTransformHandler,
     createCommandExecuteHandler,
+    createEventHandler,
     createSystemPromptHandler,
     createTextCompleteHandler,
 } from "./lib/hooks"
@@ -68,6 +69,7 @@ const plugin: Plugin = (async (ctx) => {
         ) as any,
         "chat.message": createChatMessageHandler(state, logger, config, hostPermissions),
         "experimental.text.complete": createTextCompleteHandler(),
+        event: createEventHandler(state, logger),
         "command.execute.before": createCommandExecuteHandler(
             ctx.client,
             state,

lib/commands/compression-targets.ts

@@ -5,6 +5,7 @@ export interface CompressionTarget {
     runId: number
     topic: string
     compressedTokens: number
+    durationMs: number
     grouped: boolean
     blocks: CompressionBlock[]
 }
@@ -26,6 +27,7 @@ function buildTarget(blocks: CompressionBlock[]): CompressionTarget {
         runId: first.runId,
         topic: grouped ? first.batchTopic || first.topic : first.topic,
         compressedTokens: ordered.reduce((total, block) => total + block.compressedTokens, 0),
+        durationMs: ordered.reduce((total, block) => Math.max(total, block.durationMs), 0),
         grouped,
         blocks: ordered,
     }

lib/commands/stats.ts

@@ -9,6 +9,7 @@ import { sendIgnoredMessage } from "../ui/notification"
 import { formatTokenCount } from "../ui/utils"
 import { loadAllSessionStats, type AggregatedStats } from "../state/persistence"
 import { getCurrentParams } from "../token-utils"
+import { getActiveCompressionTargets } from "./compression-targets"
 
 export interface StatsCommandContext {
     client: any
@@ -22,6 +23,7 @@ function formatStatsMessage(
     sessionTokens: number,
     sessionTools: number,
     sessionMessages: number,
+    sessionDurationMs: number,
     allTime: AggregatedStats,
 ): string {
     const lines: string[] = []
@@ -35,6 +37,7 @@ function formatStatsMessage(
     lines.push(`  Tokens pruned:   ~${formatTokenCount(sessionTokens)}`)
     lines.push(`  Tools pruned:     ${sessionTools}`)
     lines.push(`  Messages pruned:  ${sessionMessages}`)
+    lines.push(`  Compression time: ${formatCompressionTime(sessionDurationMs)}`)
     lines.push("")
     lines.push("All-time:")
     lines.push("─".repeat(60))
@@ -46,11 +49,38 @@ function formatStatsMessage(
     return lines.join("\n")
 }
 
+function formatCompressionTime(ms: number): string {
+    const safeMs = Math.max(0, Math.round(ms))
+    if (safeMs < 1000) {
+        return `${safeMs} ms`
+    }
+
+    const totalSeconds = safeMs / 1000
+    if (totalSeconds < 60) {
+        return `${totalSeconds.toFixed(1)} s`
+    }
+
+    const wholeSeconds = Math.floor(totalSeconds)
+    const hours = Math.floor(wholeSeconds / 3600)
+    const minutes = Math.floor((wholeSeconds % 3600) / 60)
+    const seconds = wholeSeconds % 60
+
+    if (hours > 0) {
+        return `${hours}h ${minutes}m ${seconds}s`
+    }
+
+    return `${minutes}m ${seconds}s`
+}
+
 export async function handleStatsCommand(ctx: StatsCommandContext): Promise<void> {
     const { client, state, logger, sessionId, messages } = ctx
 
     // Session stats from in-memory state
     const sessionTokens = state.stats.totalPruneTokens
+    const sessionDurationMs = getActiveCompressionTargets(state.prune.messages).reduce(
+        (total, target) => total + target.durationMs,
+        0,
+    )
 
     const prunedToolIds = new Set<string>(state.prune.tools.keys())
     for (const block of state.prune.messages.blocksById.values()) {
@@ -72,7 +102,13 @@ export async function handleStatsCommand(ctx: StatsCommandContext): Promise<void
     // All-time stats from storage files
     const allTime = await loadAllSessionStats(logger)
 
-    const message = formatStatsMessage(sessionTokens, sessionTools, sessionMessages, allTime)
+    const message = formatStatsMessage(
+        sessionTokens,
+        sessionTools,
+        sessionMessages,
+        sessionDurationMs,
+        allTime,
+    )
 
     const params = getCurrentParams(state, messages, logger)
     await sendIgnoredMessage(client, sessionId, message, params, logger)
@@ -81,6 +117,7 @@ export async function handleStatsCommand(ctx: StatsCommandContext): Promise<void
         sessionTokens,
         sessionTools,
         sessionMessages,
+        sessionDurationMs,
         allTimeTokens: allTime.totalTokens,
         allTimeTools: allTime.totalTools,
         allTimeMessages: allTime.totalMessages,

lib/compress/message.ts

@@ -48,6 +48,10 @@ export function createCompressMessageTool(ctx: ToolContext): ReturnType<typeof t
         async execute(args, toolCtx) {
             const input = args as CompressMessageToolArgs
             validateArgs(input)
+            const callId =
+                typeof (toolCtx as unknown as { callID?: unknown }).callID === "string"
+                    ? (toolCtx as unknown as { callID: string }).callID
+                    : undefined
 
             const { rawMessages, searchContext } = await prepareSession(
                 ctx,
@@ -107,7 +111,9 @@ export function createCompressMessageTool(ctx: ToolContext): ReturnType<typeof t
                         mode: "message",
                         runId,
                         compressMessageId: toolCtx.messageID,
+                        compressCallId: callId,
                         summaryTokens,
+                        durationMs: 0,
                     },
                     plan.selection,
                     plan.anchorMessageId,

lib/compress/range.ts

@@ -59,6 +59,10 @@ export function createCompressRangeTool(ctx: ToolContext): ReturnType<typeof too
         async execute(args, toolCtx) {
             const input = args as CompressRangeToolArgs
             validateArgs(input)
+            const callId =
+                typeof (toolCtx as unknown as { callID?: unknown }).callID === "string"
+                    ? (toolCtx as unknown as { callID: string }).callID
+                    : undefined
 
             const { rawMessages, searchContext } = await prepareSession(
                 ctx,
@@ -148,7 +152,9 @@ export function createCompressRangeTool(ctx: ToolContext): ReturnType<typeof too
                         mode: "range",
                         runId,
                         compressMessageId: toolCtx.messageID,
+                        compressCallId: callId,
                         summaryTokens,
+                        durationMs: 0,
                     },
                     preparedPlan.selection,
                     preparedPlan.anchorMessageId,

lib/compress/state.ts

@@ -26,6 +26,41 @@ export function allocateRunId(state: SessionState): number {
     return next
 }
 
+export function recordCompressionDuration(
+    state: SessionState,
+    callId: string,
+    durationMs: number,
+): void {
+    state.compressionDurations.set(callId, durationMs)
+}
+
+export function attachCompressionDuration(
+    state: SessionState,
+    callId: string,
+    messageId: string,
+): number {
+    const durationMs = state.compressionDurations.get(callId)
+    if (typeof durationMs !== "number" || !Number.isFinite(durationMs)) {
+        return 0
+    }
+
+    state.compressionDurations.delete(callId)
+
+    let updates = 0
+    for (const block of state.prune.messages.blocksById.values()) {
+        const matchesCall = block.compressCallId === callId
+        const matchesMessage = !block.compressCallId && block.compressMessageId === messageId
+        if (!matchesCall && !matchesMessage) {
+            continue
+        }
+
+        block.durationMs = durationMs
+        updates++
+    }
+
+    return updates
+}
+
 export function wrapCompressedSummary(blockId: number, summary: string): string {
     const header = COMPRESSED_BLOCK_HEADER
     const footer = formatMessageIdTag(formatBlockRef(blockId))
@@ -93,13 +128,15 @@ export function applyCompressionState(
         deactivatedByUser: false,
         compressedTokens: 0,
         summaryTokens: input.summaryTokens,
+        durationMs: input.durationMs,
         mode: input.mode,
         topic: input.topic,
         batchTopic: input.batchTopic,
         startId: input.startId,
         endId: input.endId,
         anchorMessageId,
         compressMessageId: input.compressMessageId,
+        compressCallId: input.compressCallId,
         includedBlockIds: included,
         consumedBlockIds: consumed,
         parentBlockIds: [],

lib/compress/types.ts

@@ -102,5 +102,7 @@ export interface CompressionStateInput {
     mode: CompressionMode
     runId: number
     compressMessageId: string
+    compressCallId?: string
     summaryTokens: number
+    durationMs: number
 }

lib/hooks.ts

@@ -16,6 +16,7 @@ import {
 } from "./messages"
 import { renderSystemPrompt, type PromptStore } from "./prompts"
 import { buildProtectedToolsExtension } from "./prompts/extensions/system"
+import { attachCompressionDuration, recordCompressionDuration } from "./compress/state"
 import {
     applyPendingManualTrigger,
     handleContextCommand,
@@ -29,7 +30,7 @@ import {
 } from "./commands"
 import { type HostPermissionSnapshot } from "./host-permissions"
 import { compressPermission, syncCompressPermissionState } from "./compress-permission"
-import { checkSession, ensureSessionInitialized, syncToolCache } from "./state"
+import { checkSession, ensureSessionInitialized, saveSessionState, syncToolCache } from "./state"
 import { cacheSystemPromptTokens } from "./ui/utils"
 
 const INTERNAL_AGENT_SIGNATURES = [
@@ -266,6 +267,135 @@ export function createTextCompleteHandler() {
     }
 }
 
+export function createEventHandler(state: SessionState, logger: Logger) {
+    return async (input: { event: any }) => {
+        const eventTime =
+            typeof input.event?.time === "number" && Number.isFinite(input.event.time)
+                ? input.event.time
+                : typeof input.event?.properties?.time === "number" &&
+                    Number.isFinite(input.event.properties.time)
+                  ? input.event.properties.time
+                  : undefined
+
+        if (input.event.type !== "message.part.updated") {
+            return
+        }
+
+        const part = input.event.properties?.part
+        if (part?.type !== "tool" || part.tool !== "compress") {
+            return
+        }
+
+        if (part.state.status === "pending") {
+            if (typeof part.callID !== "string" || typeof part.messageID !== "string") {
+                return
+            }
+
+            if (state.compressionStarts.has(part.callID)) {
+                return
+            }
+
+            const startedAt = eventTime ?? Date.now()
+            state.compressionStarts.set(part.callID, {
+                messageId: part.messageID,
+                startedAt,
+            })
+            logger.debug("Recorded compression start", {
+                callID: part.callID,
+                messageID: part.messageID,
+                startedAt,
+            })
+            return
+        }
+
+        if (part.state.status === "running") {
+            if (typeof part.callID !== "string") {
+                return
+            }
+
+            const start = state.compressionStarts.get(part.callID)
+            if (!start) {
+                return
+            }
+
+            const runningAt =
+                typeof part.state.time?.start === "number" && Number.isFinite(part.state.time.start)
+                    ? part.state.time.start
+                    : eventTime
+            if (typeof runningAt !== "number") {
+                return
+            }
+
+            state.compressionStarts.delete(part.callID)
+            const durationMs = Math.max(0, runningAt - start.startedAt)
+            recordCompressionDuration(state, part.callID, durationMs)
+
+            logger.info("Recorded compression time", {
+                callID: part.callID,
+                messageID: start.messageId,
+                durationMs,
+            })
+            return
+        }
+
+        if (part.state.status === "completed") {
+            if (typeof part.callID !== "string" || typeof part.messageID !== "string") {
+                return
+            }
+
+            if (!state.compressionDurations.has(part.callID)) {
+                const start = state.compressionStarts.get(part.callID)
+                const runningAt =
+                    typeof part.state.time?.start === "number" &&
+                    Number.isFinite(part.state.time.start)
+                        ? part.state.time.start
+                        : eventTime
+
+                if (start && typeof runningAt === "number") {
+                    state.compressionStarts.delete(part.callID)
+                    const durationMs = Math.max(0, runningAt - start.startedAt)
+                    recordCompressionDuration(state, part.callID, durationMs)
+                } else {
+                    const toolStart = part.state.time?.start
+                    const toolEnd = part.state.time?.end
+                    if (
+                        typeof toolStart === "number" &&
+                        Number.isFinite(toolStart) &&
+                        typeof toolEnd === "number" &&
+                        Number.isFinite(toolEnd)
+                    ) {
+                        const durationMs = Math.max(0, toolEnd - toolStart)
+                        recordCompressionDuration(state, part.callID, durationMs)
+                    }
+                }
+            }
+
+            const updates = attachCompressionDuration(state, part.callID, part.messageID)
+            if (updates === 0) {
+                return
+            }
+
+            logger.info("Attached compression time to blocks", {
+                callID: part.callID,
+                messageID: part.messageID,
+                blocks: updates,
+            })
+
+            saveSessionState(state, logger).catch((error) => {
+                logger.warn("Failed to persist compression time update", {
+                    error: error instanceof Error ? error.message : String(error),
+                })
+            })
+            return
+        }
+
+        if (typeof part.callID === "string") {
+            state.compressionStarts.delete(part.callID)
+            state.compressionDurations.delete(part.callID)
+        }
+    }
+}
+
 export function createChatMessageHandler(
     state: SessionState,
     logger: Logger,

lib/state/state.ts

@@ -1,4 +1,4 @@
-import type { SessionState, ToolParameterEntry, WithParts } from "./types"
+import type { CompressionStart, SessionState, ToolParameterEntry, WithParts } from "./types"
 import type { Logger } from "../logger"
 import { loadSessionState, saveSessionState } from "./persistence"
 import {
@@ -81,6 +81,8 @@ export function createSessionState(): SessionState {
             pruneTokenCounter: 0,
             totalPruneTokens: 0,
         },
+        compressionStarts: new Map<string, CompressionStart>(),
+        compressionDurations: new Map<string, number>(),
         toolParameters: new Map<string, ToolParameterEntry>(),
         subAgentResultCache: new Map<string, string>(),
         toolIdList: [],