agy-statusblocks 0.1.4 → 0.1.5

package/dist/segments/usage.js

@@ -89,7 +89,7 @@ function bucketPool(key) {
         return 'gemini';
     return 'other';
 }
-function isActiveModelBucket(key, data) {
+function isActiveModelBucket(key, data, hasExplicitArgonBucket = false) {
     const k = key.toLowerCase();
     const mName = (data.model?.display_name ?? '').toLowerCase();
     const mId = (data.model?.id ?? '').toLowerCase();
@@ -103,8 +103,8 @@ function isActiveModelBucket(key, data) {
     if ((mName.includes('claude') || mName.includes('sonnet') || mName.includes('opus') || mName.includes('gpt')) && bucketPool(key) === 'claude') {
         return true;
     }
-    // Native Gemini model uses the Gemini pool (but not the separate Argon pool)
-    if (mName.includes('gemini') && !mName.includes('argon') && bucketPool(key) === 'gemini') {
+    // Native Gemini model (including Gemini 4 Argon when no dedicated argon bucket exists) uses the Gemini pool
+    if (mName.includes('gemini') && (!mName.includes('argon') || !hasExplicitArgonBucket) && bucketPool(key) === 'gemini') {
         return true;
     }
     return false;
@@ -115,10 +115,11 @@ function extractLimits(data) {
     if (data.quota && typeof data.quota === 'object') {
         const entries = Object.entries(data.quota)
             .filter(([, bucket]) => bucket && typeof bucket === 'object' && !bucket.disabled);
+        const hasExplicitArgonBucket = entries.some(([k]) => bucketPool(k) === 'argon');
         // Sort active model's bucket first, then prefer showing distinct pools (Gemini/Argon + Claude), then by highest utilization
         entries.sort(([keyA, a], [keyB, b]) => {
-            const actA = isActiveModelBucket(keyA, data) ? 1 : 0;
-            const actB = isActiveModelBucket(keyB, data) ? 1 : 0;
+            const actA = isActiveModelBucket(keyA, data, hasExplicitArgonBucket) ? 1 : 0;
+            const actB = isActiveModelBucket(keyB, data, hasExplicitArgonBucket) ? 1 : 0;
             if (actA !== actB)
                 return actB - actA;
             if (actA && actB) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agy-statusblocks",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "Opinionated, adaptive block-based status line for Antigravity CLI (agy) with bin-packing layout, rate limit tracking, and live agent telemetry",
   "type": "module",
   "bin": {