agent-trace 0.2.6 → 0.2.8

package/agent-trace.cjs

@@ -24225,10 +24225,10 @@ function parseReplay(val) {
     var ev=asRec(e);if(!ev)return null;
     var id=readStr(ev,'id'),type=readStr(ev,'type'),ts=readStr(ev,'timestamp');if(!id||!type||!ts)return null;
     var d=asRec(ev.details),tok=asRec(ev.tokens);
-    return{id:id,type:type,timestamp:ts,promptId:readStr(ev,'promptId'),status:readStr(ev,'status'),costUsd:readNum(ev,'costUsd'),toolName:d?readStr(d,'toolName'):undefined,toolDurationMs:d?readNum(d,'toolDurationMs'):undefined,inputTokens:tok?readNum(tok,'input'):undefined,outputTokens:tok?readNum(tok,'output'):undefined,details:d};
+    return{id:id,type:type,timestamp:ts,promptId:readStr(ev,'promptId'),status:readStr(ev,'status'),costUsd:readNum(ev,'costUsd'),toolName:d?readStr(d,'toolName'):undefined,toolDurationMs:d?readNum(d,'toolDurationMs'):undefined,inputTokens:tok?readNum(tok,'input'):undefined,outputTokens:tok?readNum(tok,'output'):undefined,cacheReadTokens:tok?readNum(tok,'cacheRead'):undefined,cacheWriteTokens:tok?readNum(tok,'cacheWrite'):undefined,details:d};
   }).filter(Boolean);
   return{sessionId:sid,startedAt:sa,endedAt:readStr(r,'endedAt'),gitBranch:gitBranch,
-    metrics:{promptCount:readNum(m,'promptCount')||0,toolCallCount:readNum(m,'toolCallCount')||0,totalCostUsd:readNum(m,'totalCostUsd')||0,totalInputTokens:readNum(m,'totalInputTokens')||0,totalOutputTokens:readNum(m,'totalOutputTokens')||0,linesAdded:readNum(m,'linesAdded')||0,linesRemoved:readNum(m,'linesRemoved')||0,modelsUsed:readArr(m,'modelsUsed').filter(function(x){return typeof x==='string'}),toolsUsed:readArr(m,'toolsUsed').filter(function(x){return typeof x==='string'}),filesTouched:readArr(m,'filesTouched').filter(function(x){return typeof x==='string'})},
+    metrics:{promptCount:readNum(m,'promptCount')||0,toolCallCount:readNum(m,'toolCallCount')||0,totalCostUsd:readNum(m,'totalCostUsd')||0,totalInputTokens:readNum(m,'totalInputTokens')||0,totalOutputTokens:readNum(m,'totalOutputTokens')||0,totalCacheReadTokens:readNum(m,'totalCacheReadTokens')||0,totalCacheWriteTokens:readNum(m,'totalCacheWriteTokens')||0,linesAdded:readNum(m,'linesAdded')||0,linesRemoved:readNum(m,'linesRemoved')||0,modelsUsed:readArr(m,'modelsUsed').filter(function(x){return typeof x==='string'}),toolsUsed:readArr(m,'toolsUsed').filter(function(x){return typeof x==='string'}),filesTouched:readArr(m,'filesTouched').filter(function(x){return typeof x==='string'})},
     commits:commits,pullRequests:prs,timeline:timeline};
 }
 
@@ -24298,7 +24298,7 @@ function buildPromptGroups(timeline, commits) {
     map[ev.promptId].push(ev);
   });
   return order.map(function(pid){
-    var evts = map[pid] || [], promptText, responseText, cost=0,tools=0,inTok=0,outTok=0,dur=0;
+    var evts = map[pid] || [], promptText, responseText, cost=0,tools=0,inTok=0,outTok=0,cacheRTok=0,cacheWTok=0,dur=0;
     var filesR={},filesW={},toolEvts=[];
     evts.forEach(function(ev){
       var d = ev.details;
@@ -24306,7 +24306,7 @@ function buildPromptGroups(timeline, commits) {
       if(ev.type==='assistant_response'||ev.type==='api_call'||ev.type==='api_response'){
         if(d){var rt=readStr(d,'responseText')||readStr(d,'lastAssistantMessage');if(rt)responseText=rt;}
       }
-      cost+=(ev.costUsd||0);inTok+=(ev.inputTokens||0);outTok+=(ev.outputTokens||0);
+      cost+=(ev.costUsd||0);inTok+=(ev.inputTokens||0);outTok+=(ev.outputTokens||0);cacheRTok+=(ev.cacheReadTokens||0);cacheWTok+=(ev.cacheWriteTokens||0);
       if(ev.toolName||(ev.type==='tool_call'||ev.type==='tool_result')){
         toolEvts.push(ev);tools++;dur+=(ev.toolDurationMs||0);
         var dd=ev.details,rti=dd?(dd.toolInput||dd.tool_input):undefined,inp=asRec(rti),tiStr=typeof rti==='string'?rti:undefined;
@@ -24327,7 +24327,7 @@ function buildPromptGroups(timeline, commits) {
       }
       deduped.push(ev);
     });
-    return{promptId:pid,promptText:promptText,responseText:responseText,toolEvents:deduped,commits:byPrompt[pid]||[],totalCostUsd:cost,totalToolCalls:tools,totalInputTokens:inTok,totalOutputTokens:outTok,totalDurationMs:dur,filesRead:Object.keys(filesR),filesWritten:Object.keys(filesW)};
+    return{promptId:pid,promptText:promptText,responseText:responseText,toolEvents:deduped,commits:byPrompt[pid]||[],totalCostUsd:cost,totalToolCalls:tools,totalInputTokens:inTok,totalOutputTokens:outTok,totalCacheReadTokens:cacheRTok,totalCacheWriteTokens:cacheWTok,totalDurationMs:dur,filesRead:Object.keys(filesR),filesWritten:Object.keys(filesW)};
   }).filter(function(g){return g.promptText||g.toolEvents.length>0||g.responseText;});
 }
 
@@ -24450,6 +24450,7 @@ function renderReplay() {
   h += '<div class="tm">';
   h += '<span class="tmi">Cost <span class="badge orange">' + fmt$4(replay.metrics.totalCostUsd) + '</span></span>';
   h += '<span class="tmi">Tokens <span class="badge cyan">' + replay.metrics.totalInputTokens + ' in / ' + replay.metrics.totalOutputTokens + ' out</span></span>';
+  if (replay.metrics.totalCacheReadTokens > 0 || replay.metrics.totalCacheWriteTokens > 0) h += '<span class="tmi">Cache <span class="badge purple">' + replay.metrics.totalCacheReadTokens + ' read / ' + replay.metrics.totalCacheWriteTokens + ' write</span></span>';
   if (replay.metrics.linesAdded > 0 || replay.metrics.linesRemoved > 0) h += '<span class="tmi">Lines <span class="badge green">+' + replay.metrics.linesAdded + '</span> <span class="badge red">-' + replay.metrics.linesRemoved + '</span></span>';
   if (replay.metrics.modelsUsed.length > 0) h += '<span class="tmi">' + esc(replay.metrics.modelsUsed.join(', ')) + '</span>';
   if (replay.metrics.filesTouched.length > 0) h += '<span class="tmi">' + replay.metrics.filesTouched.length + ' files</span>';
@@ -24776,6 +24777,8 @@ async function fetchSessionReplayFromApi(apiBaseUrl, sessionId) {
       totalCostUsd: typeof metrics["totalCostUsd"] === "number" ? metrics["totalCostUsd"] : 0,
       totalInputTokens: typeof metrics["totalInputTokens"] === "number" ? metrics["totalInputTokens"] : 0,
       totalOutputTokens: typeof metrics["totalOutputTokens"] === "number" ? metrics["totalOutputTokens"] : 0,
+      totalCacheReadTokens: typeof metrics["totalCacheReadTokens"] === "number" ? metrics["totalCacheReadTokens"] : 0,
+      totalCacheWriteTokens: typeof metrics["totalCacheWriteTokens"] === "number" ? metrics["totalCacheWriteTokens"] : 0,
       linesAdded: typeof metrics["linesAdded"] === "number" ? metrics["linesAdded"] : 0,
       linesRemoved: typeof metrics["linesRemoved"] === "number" ? metrics["linesRemoved"] : 0,
       modelsUsed,
@@ -25003,6 +25006,8 @@ CREATE TABLE IF NOT EXISTS agent_events (
   cost_usd REAL,
   input_tokens INTEGER,
   output_tokens INTEGER,
+  cache_read_tokens INTEGER,
+  cache_write_tokens INTEGER,
   api_duration_ms REAL,
   lines_added INTEGER,
   lines_removed INTEGER,
@@ -25028,6 +25033,8 @@ CREATE TABLE IF NOT EXISTS session_traces (
   total_cost_usd REAL NOT NULL DEFAULT 0,
   total_input_tokens INTEGER NOT NULL DEFAULT 0,
   total_output_tokens INTEGER NOT NULL DEFAULT 0,
+  total_cache_read_tokens INTEGER NOT NULL DEFAULT 0,
+  total_cache_write_tokens INTEGER NOT NULL DEFAULT 0,
   lines_added INTEGER NOT NULL DEFAULT 0,
   lines_removed INTEGER NOT NULL DEFAULT 0,
   models_used TEXT NOT NULL DEFAULT '[]',
@@ -25102,6 +25109,8 @@ var SqliteClient = class {
     this.db.pragma("synchronous = NORMAL");
     this.migrateDeduplicateEvents();
     this.db.exec(SCHEMA_SQL);
+    this.migrateCacheTokenColumns();
+    this.migrateRebuildBrokenTraces();
   }
   async insertJsonEachRow(request) {
     if (request.rows.length === 0) return;
@@ -25115,10 +25124,11 @@ var SqliteClient = class {
       INSERT INTO session_traces
         (session_id, version, started_at, ended_at, user_id, git_repo, git_branch,
          prompt_count, tool_call_count, api_call_count, total_cost_usd,
-         total_input_tokens, total_output_tokens, lines_added, lines_removed,
+         total_input_tokens, total_output_tokens, total_cache_read_tokens, total_cache_write_tokens,
+         lines_added, lines_removed,
          models_used, tools_used, files_touched, commit_count, updated_at)
       VALUES
-        (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
       ON CONFLICT(session_id) DO UPDATE SET
         version = excluded.version,
         started_at = excluded.started_at,
@@ -25132,6 +25142,8 @@ var SqliteClient = class {
         total_cost_usd = excluded.total_cost_usd,
         total_input_tokens = excluded.total_input_tokens,
         total_output_tokens = excluded.total_output_tokens,
+        total_cache_read_tokens = excluded.total_cache_read_tokens,
+        total_cache_write_tokens = excluded.total_cache_write_tokens,
         lines_added = excluded.lines_added,
         lines_removed = excluded.lines_removed,
         models_used = excluded.models_used,
@@ -25156,6 +25168,8 @@ var SqliteClient = class {
           row.total_cost_usd,
           row.total_input_tokens,
           row.total_output_tokens,
+          row.total_cache_read_tokens,
+          row.total_cache_write_tokens,
           row.lines_added,
           row.lines_removed,
           toJsonArray(row.models_used),
@@ -25265,6 +25279,8 @@ var SqliteClient = class {
       total_cost_usd: raw["total_cost_usd"],
       total_input_tokens: raw["total_input_tokens"],
       total_output_tokens: raw["total_output_tokens"],
+      total_cache_read_tokens: raw["total_cache_read_tokens"] ?? 0,
+      total_cache_write_tokens: raw["total_cache_write_tokens"] ?? 0,
       lines_added: raw["lines_added"],
       lines_removed: raw["lines_removed"],
       models_used: fromJsonArray(raw["models_used"]),
@@ -25278,7 +25294,7 @@ var SqliteClient = class {
     const rows = this.db.prepare(
       `SELECT event_id, event_type, event_timestamp, session_id, prompt_id,
               tool_success, tool_name, tool_duration_ms, model, cost_usd,
-              input_tokens, output_tokens, attributes
+              input_tokens, output_tokens, cache_read_tokens, cache_write_tokens, attributes
        FROM agent_events
        WHERE session_id = ?
        ORDER BY event_timestamp ASC
@@ -25297,6 +25313,8 @@ var SqliteClient = class {
       cost_usd: raw["cost_usd"],
       input_tokens: raw["input_tokens"],
       output_tokens: raw["output_tokens"],
+      cache_read_tokens: raw["cache_read_tokens"] ?? null,
+      cache_write_tokens: raw["cache_write_tokens"] ?? null,
       attributes: fromJsonObject(raw["attributes"])
     }));
   }
@@ -25320,6 +25338,31 @@ var SqliteClient = class {
   close() {
     this.db.close();
   }
+  /**
+   * Migration: add cache_read_tokens / cache_write_tokens columns to existing databases.
+   */
+  migrateCacheTokenColumns() {
+    const addColumnIfMissing = (table, column, definition) => {
+      const cols = this.db.prepare(`PRAGMA table_info('${table}')`).all();
+      if (!cols.some((c) => c.name === column)) {
+        this.db.exec(`ALTER TABLE ${table} ADD COLUMN ${column} ${definition}`);
+      }
+    };
+    const eventsExist = this.db.prepare(
+      "SELECT 1 FROM sqlite_master WHERE type='table' AND name='agent_events'"
+    ).get();
+    if (eventsExist !== void 0) {
+      addColumnIfMissing("agent_events", "cache_read_tokens", "INTEGER");
+      addColumnIfMissing("agent_events", "cache_write_tokens", "INTEGER");
+    }
+    const tracesExist = this.db.prepare(
+      "SELECT 1 FROM sqlite_master WHERE type='table' AND name='session_traces'"
+    ).get();
+    if (tracesExist !== void 0) {
+      addColumnIfMissing("session_traces", "total_cache_read_tokens", "INTEGER NOT NULL DEFAULT 0");
+      addColumnIfMissing("session_traces", "total_cache_write_tokens", "INTEGER NOT NULL DEFAULT 0");
+    }
+  }
   /**
    * Migration: deduplicate agent_events rows from older schemas that lacked a UNIQUE constraint.
    * Runs once — if the old table exists without a unique index, it rebuilds it.
@@ -25375,6 +25418,8 @@ var SqliteClient = class {
         cost_usd REAL,
         input_tokens INTEGER,
         output_tokens INTEGER,
+        cache_read_tokens INTEGER,
+        cache_write_tokens INTEGER,
         api_duration_ms REAL,
         lines_added INTEGER,
         lines_removed INTEGER,
@@ -25395,6 +25440,32 @@ var SqliteClient = class {
     console.log(`[agent-trace] migration complete: ${afterCount.c} events after dedup (removed ${total - afterCount.c} duplicates)`);
     this.rebuildSessionTracesFromEvents();
   }
+  /**
+   * Migration v2: fix databases that ran v0.2.6's broken rebuild (models_used='[]' with data present).
+   * Re-runs the rebuild if session_traces exist but have empty models_used while events have model data.
+   */
+  migrateRebuildBrokenTraces() {
+    const tracesExist = this.db.prepare(
+      "SELECT 1 FROM sqlite_master WHERE type='table' AND name='session_traces'"
+    ).get();
+    if (tracesExist === void 0) {
+      return;
+    }
+    const broken = this.db.prepare(`
+      SELECT 1 FROM session_traces st
+      WHERE st.models_used = '[]'
+        AND EXISTS (
+          SELECT 1 FROM agent_events ae
+          WHERE ae.session_id = st.session_id AND ae.model IS NOT NULL
+        )
+      LIMIT 1
+    `).get();
+    if (broken === void 0) {
+      return;
+    }
+    console.log("[agent-trace] migrating: rebuilding session traces with correct models/tools");
+    this.rebuildSessionTracesFromEvents();
+  }
   /**
    * Rebuild session_traces by aggregating deduplicated agent_events.
    * Called after dedup migration so the dashboard has correct metrics immediately.
@@ -25411,7 +25482,8 @@ var SqliteClient = class {
       INSERT OR REPLACE INTO session_traces
         (session_id, version, started_at, ended_at, user_id, git_repo, git_branch,
          prompt_count, tool_call_count, api_call_count, total_cost_usd,
-         total_input_tokens, total_output_tokens, lines_added, lines_removed,
+         total_input_tokens, total_output_tokens, total_cache_read_tokens, total_cache_write_tokens,
+         lines_added, lines_removed,
          models_used, tools_used, files_touched, commit_count, updated_at)
       SELECT
         session_id,
@@ -25427,6 +25499,8 @@ var SqliteClient = class {
         COALESCE(SUM(cost_usd), 0),
         COALESCE(SUM(input_tokens), 0),
         COALESCE(SUM(output_tokens), 0),
+        COALESCE(SUM(cache_read_tokens), 0),
+        COALESCE(SUM(cache_write_tokens), 0),
         COALESCE(SUM(lines_added), 0),
         COALESCE(SUM(lines_removed), 0),
         '[]',
@@ -25437,6 +25511,32 @@ var SqliteClient = class {
       FROM agent_events
       GROUP BY session_id
     `);
+    const sessionIds = this.db.prepare(
+      "SELECT DISTINCT session_id FROM agent_events"
+    ).all();
+    const updateArrays = this.db.prepare(
+      "UPDATE session_traces SET models_used = ?, tools_used = ?, files_touched = ? WHERE session_id = ?"
+    );
+    const transaction = this.db.transaction((ids) => {
+      for (const { session_id } of ids) {
+        const models = this.db.prepare(
+          "SELECT DISTINCT model FROM agent_events WHERE session_id = ? AND model IS NOT NULL"
+        ).all(session_id);
+        const tools = this.db.prepare(
+          "SELECT DISTINCT tool_name FROM agent_events WHERE session_id = ? AND tool_name IS NOT NULL"
+        ).all(session_id);
+        const files = this.db.prepare(
+          "SELECT DISTINCT commit_sha FROM agent_events WHERE session_id = ? AND commit_sha IS NOT NULL"
+        ).all(session_id);
+        updateArrays.run(
+          JSON.stringify(models.map((r) => r.model)),
+          JSON.stringify(tools.map((r) => r.tool_name)),
+          JSON.stringify(files.map((r) => r.commit_sha)),
+          session_id
+        );
+      }
+    });
+    transaction(sessionIds);
     const rebuilt = this.db.prepare("SELECT COUNT(*) as c FROM session_traces").get();
     console.log(`[agent-trace] rebuilt ${rebuilt.c} session traces from deduplicated events`);
   }
@@ -25445,9 +25545,10 @@ var SqliteClient = class {
       INSERT OR IGNORE INTO agent_events
         (event_id, event_type, event_timestamp, session_id, prompt_id, user_id, source, agent_type,
          tool_name, tool_success, tool_duration_ms, model, cost_usd, input_tokens, output_tokens,
+         cache_read_tokens, cache_write_tokens,
          api_duration_ms, lines_added, lines_removed, files_changed, commit_sha, attributes)
       VALUES
-        (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
     `);
     const transaction = this.db.transaction((eventRows) => {
       for (const row of eventRows) {
@@ -25467,6 +25568,8 @@ var SqliteClient = class {
           row.cost_usd,
           row.input_tokens,
           row.output_tokens,
+          row.cache_read_tokens,
+          row.cache_write_tokens,
           row.api_duration_ms,
           row.lines_added,
           row.lines_removed,
@@ -25681,6 +25784,8 @@ function toClickHouseAgentEventRow(event) {
     cost_usd: pickNumber(payload, "cost_usd", "costUsd") ?? null,
     input_tokens: pickNumber(payload, "input_tokens", "inputTokens") ?? null,
     output_tokens: pickNumber(payload, "output_tokens", "outputTokens") ?? null,
+    cache_read_tokens: pickNumber(payload, "cache_read_tokens", "cacheReadTokens") ?? pickNumber(payload, "cache_read_input_tokens", "cacheReadInputTokens") ?? null,
+    cache_write_tokens: pickNumber(payload, "cache_write_tokens", "cacheWriteTokens") ?? pickNumber(payload, "cache_creation_input_tokens", "cacheCreationInputTokens") ?? null,
     api_duration_ms: pickNumber(payload, "api_duration_ms", "apiDurationMs") ?? null,
     lines_added: pickNumber(payload, "lines_added", "linesAdded") ?? null,
     lines_removed: pickNumber(payload, "lines_removed", "linesRemoved") ?? null,
@@ -25770,6 +25875,8 @@ function toClickHouseSessionTraceRow(trace, version, updatedAt) {
     total_cost_usd: Number.isFinite(trace.metrics.totalCostUsd) ? trace.metrics.totalCostUsd : 0,
     total_input_tokens: toNonNegativeInteger(trace.metrics.totalInputTokens),
     total_output_tokens: toNonNegativeInteger(trace.metrics.totalOutputTokens),
+    total_cache_read_tokens: toNonNegativeInteger(trace.metrics.totalCacheReadTokens),
+    total_cache_write_tokens: toNonNegativeInteger(trace.metrics.totalCacheWriteTokens),
     lines_added: Math.trunc(trace.metrics.linesAdded),
     lines_removed: Math.trunc(trace.metrics.linesRemoved),
     models_used: toUniqueStringArray(trace.metrics.modelsUsed),
@@ -25921,6 +26028,8 @@ var SESSION_TRACE_SELECT_COLUMNS = [
   "total_cost_usd",
   "total_input_tokens",
   "total_output_tokens",
+  "total_cache_read_tokens",
+  "total_cache_write_tokens",
   "lines_added",
   "lines_removed",
   "models_used",
@@ -25992,6 +26101,8 @@ function toAgentSessionTraceFromClickHouseRow(row) {
       totalCostUsd: Number.isFinite(row.total_cost_usd) ? row.total_cost_usd : 0,
       totalInputTokens: toNonNegativeInteger3(row.total_input_tokens),
       totalOutputTokens: toNonNegativeInteger3(row.total_output_tokens),
+      totalCacheReadTokens: toNonNegativeInteger3(row.total_cache_read_tokens),
+      totalCacheWriteTokens: toNonNegativeInteger3(row.total_cache_write_tokens),
       linesAdded: Math.trunc(row.lines_added),
       linesRemoved: Math.trunc(row.lines_removed),
       filesTouched: toUniqueStrings(row.files_touched),
@@ -26019,6 +26130,8 @@ var EVENT_SELECT_COLUMNS = [
   "cost_usd",
   "input_tokens",
   "output_tokens",
+  "cache_read_tokens",
+  "cache_write_tokens",
   "attributes"
 ].join(", ");
 function toNumber(value) {
@@ -26067,6 +26180,8 @@ function toTimelineEventFromClickHouseRow(row) {
   const costUsd = toNumber(row.cost_usd);
   const inputTokens = toNumber(row.input_tokens);
   const outputTokens = toNumber(row.output_tokens);
+  const cacheReadTokens = toNumber(row.cache_read_tokens);
+  const cacheWriteTokens = toNumber(row.cache_write_tokens);
   const status2 = readStatus(row);
   const details = {};
   if (row.tool_name !== null) {
@@ -26133,7 +26248,9 @@ function toTimelineEventFromClickHouseRow(row) {
     ...inputTokens !== void 0 && outputTokens !== void 0 ? {
       tokens: {
         input: inputTokens,
-        output: outputTokens
+        output: outputTokens,
+        ...cacheReadTokens !== void 0 ? { cacheRead: cacheReadTokens } : {},
+        ...cacheWriteTokens !== void 0 ? { cacheWrite: cacheWriteTokens } : {}
       }
     } : {},
     ...Object.keys(details).length > 0 ? { details } : {}
@@ -28305,6 +28422,8 @@ function toTimelineEvent(envelope) {
   const payload = asRecord5(envelope.payload);
   const inputTokens = readNumber3(payload, ["input_tokens", "inputTokens"]);
   const outputTokens = readNumber3(payload, ["output_tokens", "outputTokens"]);
+  const cacheReadTokens = readNumber3(payload, ["cache_read_tokens", "cacheReadTokens", "cache_read_input_tokens"]);
+  const cacheWriteTokens = readNumber3(payload, ["cache_write_tokens", "cacheWriteTokens", "cache_creation_input_tokens"]);
   const costUsd = computeEventCost(payload);
   const details = buildNormalizedDetails(envelope.payload);
   return {
@@ -28316,7 +28435,9 @@ function toTimelineEvent(envelope) {
     ...inputTokens !== void 0 && outputTokens !== void 0 ? {
       tokens: {
         input: inputTokens,
-        output: outputTokens
+        output: outputTokens,
+        ...cacheReadTokens !== void 0 ? { cacheRead: cacheReadTokens } : {},
+        ...cacheWriteTokens !== void 0 ? { cacheWrite: cacheWriteTokens } : {}
       }
     } : {},
     ...details !== void 0 ? { details } : {}
@@ -28349,6 +28470,8 @@ function toBaseTrace(envelope) {
       totalCostUsd: 0,
       totalInputTokens: 0,
       totalOutputTokens: 0,
+      totalCacheReadTokens: 0,
+      totalCacheWriteTokens: 0,
       linesAdded: 0,
       linesRemoved: 0,
       filesTouched: [],
@@ -28390,6 +28513,8 @@ function toUpdatedTrace(existing, envelope) {
   const cost = computeEventCost(payload) ?? 0;
   const inputTokens = readNumber3(payload, ["input_tokens", "inputTokens"]) ?? 0;
   const outputTokens = readNumber3(payload, ["output_tokens", "outputTokens"]) ?? 0;
+  const cacheReadTokens = readNumber3(payload, ["cache_read_tokens", "cacheReadTokens", "cache_read_input_tokens"]) ?? 0;
+  const cacheWriteTokens = readNumber3(payload, ["cache_write_tokens", "cacheWriteTokens", "cache_creation_input_tokens"]) ?? 0;
   const linesAdded = readNumber3(payload, ["lines_added", "linesAdded"]) ?? 0;
   const linesRemoved = readNumber3(payload, ["lines_removed", "linesRemoved"]) ?? 0;
   const model = readString4(payload, ["model"]);
@@ -28443,6 +28568,8 @@ function toUpdatedTrace(existing, envelope) {
       totalCostUsd: Number((existing.metrics.totalCostUsd + cost).toFixed(6)),
       totalInputTokens: existing.metrics.totalInputTokens + inputTokens,
       totalOutputTokens: existing.metrics.totalOutputTokens + outputTokens,
+      totalCacheReadTokens: existing.metrics.totalCacheReadTokens + cacheReadTokens,
+      totalCacheWriteTokens: existing.metrics.totalCacheWriteTokens + cacheWriteTokens,
       linesAdded: existing.metrics.linesAdded + linesAdded,
       linesRemoved: existing.metrics.linesRemoved + linesRemoved,
       filesTouched,
@@ -28489,11 +28616,23 @@ async function close2(server) {
     });
   });
 }
+function enrichEnvelopeWithCost(event) {
+  const payload = event.payload;
+  if (typeof payload["cost_usd"] === "number" && payload["cost_usd"] > 0) {
+    return event;
+  }
+  const cost = computeEventCost(payload);
+  if (cost === void 0 || cost <= 0) {
+    return event;
+  }
+  return { ...event, payload: { ...payload, cost_usd: cost } };
+}
 async function projectAndPersistEvent(event, sessionRepository, persistence) {
   const current = sessionRepository.getBySessionId(event.sessionId);
   const projected = projectEnvelopeToTrace(current, event);
   sessionRepository.upsert(projected);
-  await persistence.persistAcceptedEvent(event, projected);
+  const enriched = enrichEnvelopeWithCost(event);
+  await persistence.persistAcceptedEvent(enriched, projected);
 }
 function createRuntimeOtelSink(runtime) {
   return runtime.collectorService.otelSink;
@@ -28695,7 +28834,9 @@ function hydrateFromSqlite(runtime, sqlite, limit, eventLimit) {
       const recalculated = calculateCostUsd({
         model,
         inputTokens: hydratedTrace.metrics.totalInputTokens,
-        outputTokens: hydratedTrace.metrics.totalOutputTokens
+        outputTokens: hydratedTrace.metrics.totalOutputTokens,
+        cacheReadTokens: hydratedTrace.metrics.totalCacheReadTokens,
+        cacheWriteTokens: hydratedTrace.metrics.totalCacheWriteTokens
       });
       if (recalculated > 0) {
         hydratedTrace = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-trace",
-  "version": "0.2.6",
+  "version": "0.2.8",
   "description": "Self-hosted observability for AI coding agents. One command, zero config.",
   "license": "Apache-2.0",
   "bin": {