bingocode 1.1.200-beta.2 → 1.1.200-beta.21

package/src/utils/tokens.ts

@@ -52,6 +52,29 @@ export function getTokenCountFromUsage(usage: Usage): number {
   )
 }
 
+/**
+ * Calculate total tokens across all agent messages, matching the ProgressTracker
+ * accumulation logic in LocalAgentTask. In Claude's API:
+ * - input_tokens (and cache tokens) are cumulative per turn — each call already
+ *   includes all previous context, so we keep only the LATEST value.
+ * - output_tokens are per-turn independent, so we SUM them across all turns.
+ */
+export function getTokenCountFromMessages(messages: Message[]): number {
+  let latestInputTokens = 0
+  let cumulativeOutputTokens = 0
+  for (const message of messages) {
+    const usage = getTokenUsage(message)
+    if (usage) {
+      latestInputTokens =
+        usage.input_tokens +
+        (usage.cache_creation_input_tokens ?? 0) +
+        (usage.cache_read_input_tokens ?? 0)
+      cumulativeOutputTokens += usage.output_tokens
+    }
+  }
+  return latestInputTokens + cumulativeOutputTokens
+}
+
 export function tokenCountFromLastAPIResponse(messages: Message[]): number {
   let i = messages.length - 1
   while (i >= 0) {