@expiren/opencode-antigravity-auth 1.6.36 → 1.6.38

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,16 @@ 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
+            );
+          }
+        },
         transformThinkingParts
       },
       {
@@ -12610,6 +12637,7 @@ var createAntigravityPlugin = (providerId) => async ({ client, directory }) => {
                   );
                 }
               };
+              let apiRequestCount = 0;
               let shouldSwitchAccount = false;
               let headerStyle = preferredHeaderStyle;
               pushDebug(`headerStyle=${headerStyle} explicit=${explicitQuota}`);
@@ -12667,7 +12695,6 @@ var createAntigravityPlugin = (providerId) => async ({ client, directory }) => {
                 let tokenConsumed = false;
                 let capacityRetryCount = 0;
                 let lastEndpointIndex = -1;
-                let apiRequestCount = 0;
                 for (let i = 0; i < ANTIGRAVITY_ENDPOINT_FALLBACKS.length; i++) {
                   if (i !== lastEndpointIndex) {
                     capacityRetryCount = 0;