Merge pull request #46 from mahdiwafy/fix/usage-tracking-dashboard

fix: usage tracking for Antigravity/Gemini streaming responses

Author: Vanszs

open-sse/handlers/chatCore/nonStreamingHandler.js

@@ -127,10 +127,15 @@ export function translateNonStreamingResponse(responseBody, targetFormat, source
 
     if (responseBody.usage) {
       result.usage = {
-        prompt_tokens: responseBody.usage.input_tokens || 0,
-        completion_tokens: responseBody.usage.output_tokens || 0,
-        total_tokens: (responseBody.usage.input_tokens || 0) + (responseBody.usage.output_tokens || 0)
+        prompt_tokens: responseBody.usage.prompt_tokens || responseBody.usage.input_tokens || 0,
+        completion_tokens: responseBody.usage.completion_tokens || responseBody.usage.output_tokens || 0,
+        total_tokens: responseBody.usage.total_tokens || 
+          ((responseBody.usage.prompt_tokens || responseBody.usage.input_tokens || 0) + 
+           (responseBody.usage.completion_tokens || responseBody.usage.output_tokens || 0))
       };
+      if (responseBody.usage.completion_tokens_details) {
+        result.usage.completion_tokens_details = responseBody.usage.completion_tokens_details;
+      }
     }
     return result;
   }

open-sse/utils/usageTracking.js

@@ -246,11 +246,11 @@ export function extractUsage(chunk) {
     });
   }
 
-  // Claude format (message_delta event)
+  // Claude format (message_delta event) - also handles OpenAI format in final chunk
   if (chunk.type === "message_delta" && chunk.usage && typeof chunk.usage === "object") {
     return normalizeUsage({
-      prompt_tokens: chunk.usage.input_tokens || 0,
-      completion_tokens: chunk.usage.output_tokens || 0,
+      prompt_tokens: chunk.usage.prompt_tokens || chunk.usage.input_tokens || 0,
+      completion_tokens: chunk.usage.completion_tokens || chunk.usage.output_tokens || 0,
       cache_read_input_tokens: chunk.usage.cache_read_input_tokens,
       cache_creation_input_tokens: chunk.usage.cache_creation_input_tokens
     });