baro-ai 0.59.1 → 0.60.0

package/dist/cli.mjs

@@ -43312,7 +43312,15 @@ function extractPath(item) {
 }
 
 // ../baro-orchestrator/src/participants/forwarders/token-usage.ts
+var PROGRESS_THROTTLE_MS = 1500;
 var TokenUsageForwarder = class extends BaseObserver {
+  // Per-agent live accumulator for Claude streaming. Claude reports
+  // output_tokens cumulatively *within a message*, restarting each turn — so
+  // we carry committed output across turns and add the current message's count.
+  // We track OUTPUT only: per-message input_tokens includes the whole replayed
+  // context, so summing it across turns would wildly over-count. Input comes
+  // from the authoritative token_usage total at the end instead.
+  live = /* @__PURE__ */ new Map();
   async onExternalEvent(_source, event) {
     if (AgentResult.is(event)) {
       this.handleClaudeResult(event.data);
@@ -43322,6 +43330,38 @@ var TokenUsageForwarder = class extends BaseObserver {
       this.handleCodexTurnEvent(event.data);
       return;
     }
+    if (ClaudeStreamChunk.is(event)) {
+      this.handleStreamChunk(event.data);
+      return;
+    }
+  }
+  handleStreamChunk(item) {
+    const inner = item.raw.event ?? item.raw;
+    const type = typeof inner.type === "string" ? inner.type : "";
+    const num = (v) => typeof v === "number" ? v : 0;
+    const st = this.live.get(item.agentId) ?? { committedOut: 0, curOut: 0, lastEmit: 0 };
+    if (type === "message_start") {
+      st.committedOut += st.curOut;
+      const msg = inner.message ?? {};
+      st.curOut = num((msg.usage ?? {}).output_tokens);
+    } else if (type === "message_delta") {
+      const usage = inner.usage ?? {};
+      st.curOut = num(usage.output_tokens) || st.curOut;
+    } else {
+      this.live.set(item.agentId, st);
+      return;
+    }
+    this.live.set(item.agentId, st);
+    const now = Date.now();
+    if (type !== "message_start" && now - st.lastEmit < PROGRESS_THROTTLE_MS) return;
+    st.lastEmit = now;
+    emit({
+      type: "token_progress",
+      id: item.agentId,
+      input_tokens: 0,
+      // input comes from the authoritative total; see above
+      output_tokens: st.committedOut + st.curOut
+    });
   }
   handleClaudeResult(item) {
     const usage = item.usage;