@kenkaiiii/gg-ai 5.31.0 → 5.32.0

package/dist/index.cjs

@@ -3376,14 +3376,16 @@ async function* runStream4(options) {
   let thinkingAccum = "";
   let stopReason = "end_turn";
   let inputTokens = 0;
-  let outputTokens = 0;
+  let candidateTokens = 0;
+  let reasoningTokens = 0;
   let cacheRead = 0;
   let toolIndex = 0;
   const handleResponse = function* (chunk) {
     const usage = usageFromResponse(chunk);
     if (usage) {
       inputTokens = usage.promptTokenCount ?? inputTokens;
-      outputTokens = usage.candidatesTokenCount ?? outputTokens;
+      candidateTokens = usage.candidatesTokenCount ?? candidateTokens;
+      reasoningTokens = usage.thoughtsTokenCount ?? reasoningTokens;
       cacheRead = usage.cachedContentTokenCount ?? cacheRead;
     }
     const reason = finishReasonFromResponse(chunk);
@@ -3439,6 +3441,7 @@ async function* runStream4(options) {
   }
   if (pendingToolCalls.length > 0) stopReason = "tool_use";
   const adjustedInputTokens = Math.max(0, inputTokens - cacheRead);
+  const outputTokens = candidateTokens + reasoningTokens;
   const streamResponse = {
     message: {
       role: "assistant",
@@ -3448,6 +3451,7 @@ async function* runStream4(options) {
     usage: {
       inputTokens: adjustedInputTokens,
       outputTokens,
+      ...reasoningTokens > 0 ? { reasoningTokens } : {},
       ...cacheRead > 0 ? { cacheRead } : {}
     }
   };