claude-code-kanban 4.23.1 → 4.23.2

package/lib/parsers.js

@@ -160,6 +160,17 @@ function formatTaskUsage(usage) {
   return parts.length ? ' · ' + parts.join(' · ') : '';
 }
 
+// Raw per-agent cost for the agent modal's chips plus the formatted one-liner for
+// the inline message row. Null when the source carried no numbers.
+function agentUsageEntry(subagentTokens, toolUses, durationMs) {
+  const usage = {};
+  if (subagentTokens != null) usage.tokens = subagentTokens;
+  if (toolUses != null) usage.tools = toolUses;
+  if (durationMs != null) usage.durationMs = durationMs;
+  if (!Object.keys(usage).length) return null;
+  return { usage, usageText: formatTaskUsage({ subagentTokens, toolUses, durationMs }) };
+}
+
 function pushUserMessage(messages, text, timestamp, sysLabel, extras) {
   if (sysLabel === '__skip__') return;
   let safeText = text || '';
@@ -1099,6 +1110,7 @@ function buildSessionDigest(jsonlPath) {
     const taskIdRe = /<task-id>([a-zA-Z0-9_-]+)<\/task-id>/;
     const nameByToolUseId = {};
     const descByToolUseId = {};
+    const usageByToolUseId = {};
     for (const line of content.split('\n')) {
       // Terminated-teammate detection: check first since cheap substring guards
       if (line.includes('teammate-message') &&
@@ -1171,34 +1183,34 @@ function buildSessionDigest(jsonlPath) {
       } else if (line.includes('User rejected tool use') && line.includes('"tool_use_id"')) {
         const m = tmToolIdRe.exec(line);
         if (m) rejectedToolUseIds.add(m[1]);
-      } else if (line.includes('<task-notification>') &&
-                 (line.includes('<status>killed</status>') || line.includes('<status>error</status>'))) {
-        const idMatch = taskIdRe.exec(line);
-        if (idMatch) killedAgentIds.add(idMatch[1]);
+      } else if (line.includes('<task-notification>')) {
+        // The envelope's tags survive JSON escaping, so read them off the raw line.
+        // This covers every delivery shape (user message, queue-operation,
+        // queued_command attachment) without enumerating them.
+        if (line.includes('<status>killed</status>') || line.includes('<status>error</status>')) {
+          const idMatch = taskIdRe.exec(line);
+          if (idMatch) killedAgentIds.add(idMatch[1]);
+        }
+        if (line.includes('<subagent_tokens>')) {
+          const notif = parseTaskNotification(line);
+          const u = notif?.usage;
+          const entry = u && notif.toolUseId ? agentUsageEntry(u.subagentTokens, u.toolUses, u.durationMs) : null;
+          if (entry) usageByToolUseId[notif.toolUseId] = entry;
+        }
       } else if (line.includes('"toolUseResult"') && line.includes('"agentId"') && line.includes('"tool_result"')) {
         try {
           const obj = JSON.parse(line);
           const tur = obj.toolUseResult;
           if (tur?.agentId) {
-            // Foreground agents report their cost on the completion toolUseResult
-            // (totalTokens/totalToolUseCount/totalDurationMs) — the same numbers a
-            // background agent surfaces via its <task-notification> <usage> block. Keep
-            // the raw numbers (for the agent modal's per-stat chips) and a formatted
-            // one-liner via the shared formatter (for the inline message row):
-            // " · 31.5k tok · 5 tools · 34s".
-            const usage = {};
-            if (tur.totalTokens != null) usage.tokens = tur.totalTokens;
-            if (tur.totalToolUseCount != null) usage.tools = tur.totalToolUseCount;
-            if (tur.totalDurationMs != null) usage.durationMs = tur.totalDurationMs;
-            const hasUsage = Object.keys(usage).length > 0;
-            const usageText = hasUsage
-              ? formatTaskUsage({ subagentTokens: usage.tokens, toolUses: usage.tools, durationMs: usage.durationMs })
-              : null;
+            // Foreground agents report their cost on the completion toolUseResult;
+            // background agents report the same numbers in their <task-notification>.
+            const { usage, usageText } = agentUsageEntry(tur.totalTokens, tur.totalToolUseCount, tur.totalDurationMs)
+              || { usage: null, usageText: null };
             const blocks = obj.message?.content;
             if (Array.isArray(blocks)) {
               for (const b of blocks) {
                 if (b.type === 'tool_result' && b.tool_use_id && !map[b.tool_use_id]) {
-                  map[b.tool_use_id] = { agentId: tur.agentId, prompt: tur.prompt || null, usage: hasUsage ? usage : null, usageText };
+                  map[b.tool_use_id] = { agentId: tur.agentId, prompt: tur.prompt || null, usage, usageText };
                 }
               }
             }
@@ -1207,6 +1219,7 @@ function buildSessionDigest(jsonlPath) {
       }
     }
     for (const [key, entry] of Object.entries(map)) {
+      if (!entry.usage && usageByToolUseId[key]) Object.assign(entry, usageByToolUseId[key]);
       if (nameByToolUseId[key]) entry.name = nameByToolUseId[key];
       if (descByToolUseId[key]) entry.description = descByToolUseId[key];
       // Prefer the full prompt from the assistant's tool_use input — Claude Code's

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "claude-code-kanban",
-  "version": "4.23.1",
+  "version": "4.23.2",
   "description": "A web-based Kanban board for viewing Claude Code tasks with agent teams support",
   "main": "server.js",
   "bin": {

package/public/app.js

@@ -2883,10 +2883,8 @@ function showAgentModal(agentId) {
     return `<span class="agent-chip${cls}"${style}${title}>${labelHtml}<span class="agent-chip-val">${value}</span></span>`;
   };
 
-  // Agent cost stats live on the launch row's message (resolved from the completion
-  // toolUseResult), not on the agent-activity record — so read them from there.
   const agentMsg = currentMessages.find((m) => m.tool === 'Agent' && m.agentId === agentId);
-  const usageStats = agentMsg?.agentUsageStats || null;
+  const usageStats = agent.usage || null;
   const fmtTok = fmtTokens;
 
   const chips = [];

package/server.js

@@ -2007,16 +2007,16 @@ app.get('/api/sessions/:sessionId/agents', (req, res) => {
     // (same pattern as agentsNeedingModel below). Each field shares the same resolve flow:
     // look up in progressMap by agentId, fall back to per-field extractor, persist only
     // on actual change.
+    // progressMap is keyed by tool_use_id; re-key by agentId, first value per field wins.
     const byAgentId = {};
-    const nameByAgentId = {};
-    const descByAgentId = {};
     if (meta.jsonlPath) {
       try {
         const progressMap = getProgressMap(meta.jsonlPath);
         for (const entry of Object.values(progressMap)) {
-          if (entry.prompt && !byAgentId[entry.agentId]) byAgentId[entry.agentId] = entry.prompt;
-          if (entry.name && !nameByAgentId[entry.agentId]) nameByAgentId[entry.agentId] = entry.name;
-          if (entry.description && !descByAgentId[entry.agentId]) descByAgentId[entry.agentId] = entry.description;
+          const e = (byAgentId[entry.agentId] ||= {});
+          for (const f of ['prompt', 'name', 'description', 'usage']) {
+            if (entry[f] && !e[f]) e[f] = entry[f];
+          }
         }
       } catch (_) {}
     }
@@ -2025,12 +2025,12 @@ app.get('/api/sessions/:sessionId/agents', (req, res) => {
         field: 'prompt',
         flag: 'promptUnavailable',
         lookup: (a) => {
-          if (byAgentId[a.agentId]) return byAgentId[a.agentId];
+          if (byAgentId[a.agentId]?.prompt) return byAgentId[a.agentId].prompt;
           try { return extractPromptFromTranscript(subagentJsonlForExtraction(meta, a.agentId)); } catch (_) { return null; }
         },
       },
-      { field: 'agentName',   flag: 'agentNameUnavailable',   lookup: (a) => nameByAgentId[a.agentId] || null },
-      { field: 'description', flag: 'descriptionUnavailable', lookup: (a) => descByAgentId[a.agentId] || null },
+      { field: 'agentName',   flag: 'agentNameUnavailable',   lookup: (a) => byAgentId[a.agentId]?.name || null },
+      { field: 'description', flag: 'descriptionUnavailable', lookup: (a) => byAgentId[a.agentId]?.description || null },
     ];
     if (meta.jsonlPath) {
       for (const { field, flag, lookup } of reconcileFields) {
@@ -2133,6 +2133,11 @@ app.get('/api/sessions/:sessionId/agents', (req, res) => {
     }
 
     const waitingForUser = checkWaitingForUser(agentDir, logMtime);
+    // Attached after persistAgent so it stays response-only.
+    for (const agent of visibleAgents) {
+      const usage = byAgentId[agent.agentId]?.usage;
+      if (usage) agent.usage = usage;
+    }
     res.json({ agents: visibleAgents, waitingForUser, teamColors });
   } catch {
     res.json({ agents: [], waitingForUser: null });
@@ -2499,7 +2504,6 @@ app.get('/api/sessions/:sessionId/messages', (req, res) => {
         msg.agentId = entry.agentId;
         if (entry.description) msg.agentDescription = entry.description;
         if (entry.usageText) msg.agentUsage = entry.usageText;
-        if (entry.usage) msg.agentUsageStats = entry.usage;
         if (entry.prompt && !msg.agentPrompt) msg.agentPrompt = entry.prompt;
         try {
           const agentFile = path.join(agentDir, `${entry.agentId}.jsonl`);