gitdone-agent 0.6.6 → 0.6.7

package/index.js

@@ -547,12 +547,15 @@ async function reportCommandResult(cfg, id, status, result) {
 }
 
 // Post a batch of console events (and optional lifecycle status) for an AiRun.
-async function postRunEvents(cfg, runId, events, status, result) {
+// `usage` is sent once on the final (done/error) post so the server can record
+// how many tokens the run cost.
+async function postRunEvents(cfg, runId, events, status, result, usage) {
   await api(cfg, '/api/v1/agent/ai-run/events', {
     runId,
     events,
     ...(status ? { status } : {}),
     ...(result !== undefined ? { result } : {}),
+    ...(usage ? { usage } : {}),
   }).catch((e) => log(`✗ ai-run events post failed: ${e.message}`))
 }
 
@@ -575,7 +578,15 @@ async function postSessionEvents(cfg, sessionId, events, status, claudeSessionId
 // With `--include-partial-messages`, Claude also emits `stream_event` lines
 // carrying incremental text deltas — onDelta gets those so chat sessions can
 // show the reply being written live (gd-302).
-function parseStreamLine(line, push, onInit, onDelta) {
+// "12.3k" / "1.2M" compaction for the console token summary line.
+function fmtTokens(n) {
+  n = Number(n) || 0
+  if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`
+  if (n >= 1_000) return `${(n / 1_000).toFixed(1)}k`
+  return String(n)
+}
+
+function parseStreamLine(line, push, onInit, onDelta, onMeta) {
   let ev
   try { ev = JSON.parse(line) } catch { return }
   if (ev.type === 'system') {
@@ -583,6 +594,8 @@ function parseStreamLine(line, push, onInit, onDelta) {
       push('SYSTEM', `Сесия стартирана${ev.model ? ` (${ev.model})` : ''}.`)
       // Surface Claude's own session id so chat sessions can --resume it.
       if (ev.session_id && typeof onInit === 'function') onInit(ev.session_id)
+      // Remember which model ran, for the token/cost record (gd-334).
+      if (ev.model && typeof onMeta === 'function') onMeta({ model: ev.model })
     }
     return
   }
@@ -607,8 +620,40 @@ function parseStreamLine(line, push, onInit, onDelta) {
     }
     return
   }
-  if (ev.type === 'result' && ev.subtype && ev.subtype !== 'success') {
-    push('SYSTEM', `Резултат: ${ev.subtype}`)
+  if (ev.type === 'result') {
+    if (ev.subtype && ev.subtype !== 'success') push('SYSTEM', `Резултат: ${ev.subtype}`)
+    // claude's final result carries cumulative token usage + its own cost.
+    // Prefer `modelUsage` (summed over every model/subagent turn) which is the
+    // true cumulative; top-level `usage` is often just the last turn. Fall back
+    // to `usage` when modelUsage is absent (older CLI).
+    if (typeof onMeta === 'function' && (ev.modelUsage || ev.usage || typeof ev.total_cost_usd === 'number')) {
+      let usage
+      let model
+      if (ev.modelUsage && typeof ev.modelUsage === 'object') {
+        const acc = { inputTokens: 0, outputTokens: 0, cacheReadTokens: 0, cacheCreateTokens: 0 }
+        for (const [name, mu] of Object.entries(ev.modelUsage)) {
+          if (!model) model = name
+          acc.inputTokens += mu.inputTokens ?? mu.input_tokens ?? 0
+          acc.outputTokens += mu.outputTokens ?? mu.output_tokens ?? 0
+          acc.cacheReadTokens += mu.cacheReadInputTokens ?? mu.cache_read_input_tokens ?? 0
+          acc.cacheCreateTokens += mu.cacheCreationInputTokens ?? mu.cache_creation_input_tokens ?? 0
+        }
+        usage = acc
+      } else if (ev.usage) {
+        const u = ev.usage
+        usage = {
+          inputTokens: u.input_tokens ?? 0,
+          outputTokens: u.output_tokens ?? 0,
+          cacheReadTokens: u.cache_read_input_tokens ?? 0,
+          cacheCreateTokens: u.cache_creation_input_tokens ?? 0,
+        }
+      }
+      onMeta({
+        ...(usage ? { usage } : {}),
+        ...(model ? { model } : {}),
+        costUsd: typeof ev.total_cost_usd === 'number' ? ev.total_cost_usd : undefined,
+      })
+    }
   }
 }
 
@@ -799,6 +844,15 @@ function runAiCommand(cfg, cmd, repoPath) {
   const push = (kind, text) => { if (text != null && String(text) !== '') pending.push({ kind, text: String(text) }) }
   const timer = setInterval(flush, 800)
 
+  // Accumulate model + token usage across the stream (model from init, usage
+  // from the final result event) to report once on exit (gd-334).
+  const meta = { model: undefined, usage: undefined, costUsd: undefined }
+  const onMeta = (m) => {
+    if (m.model) meta.model = m.model
+    if (m.usage) meta.usage = m.usage
+    if (typeof m.costUsd === 'number') meta.costUsd = m.costUsd
+  }
+
   log(`▶ ai_run ${runId} @ ${repoPath} via ${claudePath}`)
   postRunEvents(cfg, runId, [{ kind: 'SYSTEM', text: `Стартиране на Claude Code в ${repoPath}…` }], 'running')
 
@@ -827,21 +881,28 @@ function runAiCommand(cfg, cmd, repoPath) {
     while ((nl = buf.indexOf('\n')) >= 0) {
       const line = buf.slice(0, nl).trim()
       buf = buf.slice(nl + 1)
-      if (line) parseStreamLine(line, push)
+      if (line) parseStreamLine(line, push, undefined, undefined, onMeta)
     }
   })
   child.stderr.on('data', (d) => { const s = d.toString().trim(); if (s) push('SYSTEM', s) })
   child.on('error', (err) => push('SYSTEM', `Процесна грешка: ${err.message}`))
   child.on('close', async (code) => {
     clearInterval(timer)
-    if (buf.trim()) parseStreamLine(buf.trim(), push)
+    if (buf.trim()) parseStreamLine(buf.trim(), push, undefined, undefined, onMeta)
     await flush()
     const ok = code === 0
+    // Build the usage payload + a console summary line from what we saw.
+    const usage = meta.usage
+      ? { model: meta.model, ...meta.usage, ...(typeof meta.costUsd === 'number' ? { costUsd: meta.costUsd } : {}) }
+      : undefined
+    const events = [{ kind: 'SYSTEM', text: ok ? '✓ Готово.' : `✗ Процесът приключи с код ${code}.` }]
+    if (usage) events.push({ kind: 'SYSTEM', text: `📊 Токени: ${fmtTokens(usage.inputTokens)} вход · ${fmtTokens(usage.outputTokens)} изход${usage.cacheReadTokens ? ` · ${fmtTokens(usage.cacheReadTokens)} кеш` : ''}${typeof usage.costUsd === 'number' ? ` · $${usage.costUsd.toFixed(4)}` : ''}` })
     await postRunEvents(
       cfg, runId,
-      [{ kind: 'SYSTEM', text: ok ? '✓ Готово.' : `✗ Процесът приключи с код ${code}.` }],
+      events,
       ok ? 'done' : 'error',
       ok ? 'ok' : `exit ${code}`,
+      usage,
     )
     reportCommandResult(cfg, cmd.id, ok ? 'done' : 'error', `exit ${code}`)
     log(`■ ai_run ${runId} приключи (code ${code})`)

package/package.json

@@ -1,6 +1,6 @@
 {
     "name":  "gitdone-agent",
-    "version":  "0.6.6",
+    "version":  "0.6.7",
     "description":  "Local git agent for gitdone — watches a local repo and sends snapshots to gitdone.eu",
     "type":  "module",
     "bin":  {