@expiren/opencode-antigravity-auth 1.6.36 → 1.6.37

package/dist/index.js

@@ -3810,7 +3810,7 @@ function deduplicateThinkingText(response, sentBuffer, displayedThinkingHashes)
   }
   return response;
 }
-function transformSseLine(line, signatureStore, thoughtBuffer, sentThinkingBuffer, callbacks, options, debugState2) {
+function transformSseLine(line, signatureStore, thoughtBuffer, sentThinkingBuffer, callbacks, options, debugState2, usageState) {
   if (!line.startsWith("data:")) {
     return line;
   }
@@ -3830,6 +3830,18 @@ function transformSseLine(line, signatureStore, thoughtBuffer, sentThinkingBuffe
           callbacks.onCacheSignature
         );
       }
+      if (usageState) {
+        const resp = parsed.response;
+        const meta = resp.usageMetadata;
+        if (meta && typeof meta === "object") {
+          usageState.lastUsage = {
+            cachedContentTokenCount: typeof meta.cachedContentTokenCount === "number" ? meta.cachedContentTokenCount : 0,
+            promptTokenCount: typeof meta.promptTokenCount === "number" ? meta.promptTokenCount : 0,
+            candidatesTokenCount: typeof meta.candidatesTokenCount === "number" ? meta.candidatesTokenCount : 0,
+            totalTokenCount: typeof meta.totalTokenCount === "number" ? meta.totalTokenCount : 0
+          };
+        }
+      }
       let response = deduplicateThinkingText(
         parsed.response,
         sentThinkingBuffer,
@@ -3906,6 +3918,7 @@ function createStreamingTransformer(signatureStore, callbacks, options = {}) {
   const sentThinkingBuffer = createThoughtBuffer();
   const debugState2 = { injected: false };
   let hasSeenUsageMetadata = false;
+  const usageState = { lastUsage: null };
   return new TransformStream({
     transform(chunk, controller) {
       buffer += decoder.decode(chunk, { stream: true });
@@ -3922,7 +3935,8 @@ function createStreamingTransformer(signatureStore, callbacks, options = {}) {
           sentThinkingBuffer,
           callbacks,
           options,
-          debugState2
+          debugState2,
+          usageState
         );
         controller.enqueue(encoder.encode(transformedLine + "\n"));
       }
@@ -3940,7 +3954,8 @@ function createStreamingTransformer(signatureStore, callbacks, options = {}) {
           sentThinkingBuffer,
           callbacks,
           options,
-          debugState2
+          debugState2,
+          usageState
         );
         controller.enqueue(encoder.encode(transformedLine));
       }
@@ -3959,6 +3974,9 @@ data: ${JSON.stringify(syntheticUsage)}
 
 `));
       }
+      if (usageState.lastUsage && callbacks.onUsageMetadata) {
+        callbacks.onUsageMetadata(usageState.lastUsage);
+      }
     }
   });
 }
@@ -8131,7 +8149,28 @@ async function transformAntigravityResponse(response, streaming, debugContext, r
       {
         onCacheSignature: cacheSignature,
         onInjectDebug: injectDebugThinking,
-        // onInjectSyntheticThinking removed - keep_thinking now uses debugText path
+        onUsageMetadata: (usage) => {
+          if (effectiveModel) {
+            logCacheStats(
+              effectiveModel,
+              usage.cachedContentTokenCount,
+              0,
+              usage.promptTokenCount || usage.totalTokenCount
+            );
+          }
+          if (usage.cachedContentTokenCount !== void 0) {
+            headers.set("x-antigravity-cached-content-token-count", String(usage.cachedContentTokenCount));
+          }
+          if (usage.totalTokenCount !== void 0) {
+            headers.set("x-antigravity-total-token-count", String(usage.totalTokenCount));
+          }
+          if (usage.promptTokenCount !== void 0) {
+            headers.set("x-antigravity-prompt-token-count", String(usage.promptTokenCount));
+          }
+          if (usage.candidatesTokenCount !== void 0) {
+            headers.set("x-antigravity-candidates-token-count", String(usage.candidatesTokenCount));
+          }
+        },
         transformThinkingParts
       },
       {