Merge pull request #73 from PureWeen/dev/alex/requests

Add premium quota display to session title bar and fix SDK numeric type casting

Author: dalexsoto

PolyPilot/Components/ChatMessageList.razor

@@ -10,7 +10,7 @@
     }
     else
     {
-        @foreach (var msg in Messages)
+        @foreach (var msg in Messages.ToList())
         {
             <ChatMessageItem Message="msg" Compact="Compact" UserAvatarUrl="@UserAvatarUrl" />
         }

PolyPilot/Components/ExpandedSessionView.razor

@@ -29,6 +29,11 @@
             {
                 <span class="context-badge branch-badge"> @Session.GitBranch</span>
             }
+            @if (UsageInfo?.PremiumQuota != null)
+            {
+                var q = UsageInfo.PremiumQuota;
+                <span class="context-badge premium-badge" title="@(q.IsUnlimited ? "Unlimited premium requests" : $"{q.EntitlementRequests - q.UsedRequests} of {q.EntitlementRequests} remaining (resets {q.ResetDate?[..10]})")">⚡ @(q.IsUnlimited ? "∞" : $"{q.EntitlementRequests - q.UsedRequests}/{q.EntitlementRequests}") premium</span>
+            }
             <button class="collapse-card-btn" @onclick="OnCollapse" title="Back to grid (Esc / ⌘E)">⊟ Grid</button>
         </div>
     </div>

PolyPilot/Components/ExpandedSessionView.razor.css

@@ -25,6 +25,7 @@
 .chat-header-context { display: flex; align-items: center; gap: 0.5rem; flex-wrap: wrap; }
 .context-badge { font-size: var(--type-caption1, 0.6rem); color: var(--text-dim); white-space: nowrap; overflow: hidden; text-overflow: ellipsis; max-width: 300px; }
 .branch-badge { color: var(--accent-primary); font-weight: 500; }
+.premium-badge { color: #e0a040; font-weight: 500; }
 .model-badge { padding: 0.2rem 0.5rem; background: var(--hover-bg); border: 1px solid var(--border-accent); border-radius: 4px; font-size: var(--type-body); color: var(--accent-primary); line-height: 1.2; }
 .session-id-badge { padding: 0.2rem 0.5rem; background: rgba(251,191,36,0.2); border: 1px solid rgba(251,191,36,0.3); border-radius: 4px; font-size: var(--type-body); color: #fbbf24; font-family: monospace; cursor: pointer; }
 .session-id-badge:hover { background: rgba(251,191,36,0.3); }

PolyPilot/Components/Pages/Dashboard.razor

@@ -644,7 +644,8 @@
                 info.CurrentTokens ?? current.CurrentTokens,
                 info.TokenLimit ?? current.TokenLimit,
                 info.InputTokens ?? current.InputTokens,
-                info.OutputTokens ?? current.OutputTokens);
+                info.OutputTokens ?? current.OutputTokens,
+                info.PremiumQuota ?? current.PremiumQuota);
         }
         else
         {

PolyPilot/Services/CopilotService.Events.cs

@@ -165,26 +165,42 @@ void Invoke(Action action)
 
             case SessionUsageInfoEvent usageInfo:
                 var uData = usageInfo.Data;
-                var uModel = uData?.GetType().GetProperty("Model")?.GetValue(uData)?.ToString();
-                var uCurrentTokens = uData?.GetType().GetProperty("CurrentTokens")?.GetValue(uData) as int?;
-                var uTokenLimit = uData?.GetType().GetProperty("TokenLimit")?.GetValue(uData) as int?;
-                var uInputTokens = uData?.GetType().GetProperty("InputTokens")?.GetValue(uData) as int?;
-                var uOutputTokens = uData?.GetType().GetProperty("OutputTokens")?.GetValue(uData) as int?;
-                if (!string.IsNullOrEmpty(uModel))
-                    state.Info.Model = uModel;
-                Invoke(() => OnUsageInfoChanged?.Invoke(sessionName, new SessionUsageInfo(uModel, uCurrentTokens, uTokenLimit, uInputTokens, uOutputTokens)));
+                var uCurrentTokensRaw = uData?.GetType().GetProperty("CurrentTokens")?.GetValue(uData);
+                var uTokenLimitRaw = uData?.GetType().GetProperty("TokenLimit")?.GetValue(uData);
+                var uCurrentTokens = uCurrentTokensRaw != null ? (int?)Convert.ToInt32(uCurrentTokensRaw) : null;
+                var uTokenLimit = uTokenLimitRaw != null ? (int?)Convert.ToInt32(uTokenLimitRaw) : null;
+                Invoke(() => OnUsageInfoChanged?.Invoke(sessionName, new SessionUsageInfo(null, uCurrentTokens, uTokenLimit, null, null)));
                 break;
 
             case AssistantUsageEvent assistantUsage:
                 var aData = assistantUsage.Data;
                 var aModel = aData?.GetType().GetProperty("Model")?.GetValue(aData)?.ToString();
-                var aInput = aData?.GetType().GetProperty("InputTokens")?.GetValue(aData) as int?;
-                var aOutput = aData?.GetType().GetProperty("OutputTokens")?.GetValue(aData) as int?;
+                var aInputRaw = aData?.GetType().GetProperty("InputTokens")?.GetValue(aData);
+                var aOutputRaw = aData?.GetType().GetProperty("OutputTokens")?.GetValue(aData);
+                var aInput = aInputRaw != null ? (int?)Convert.ToInt32(aInputRaw) : null;
+                var aOutput = aOutputRaw != null ? (int?)Convert.ToInt32(aOutputRaw) : null;
+                QuotaInfo? aPremiumQuota = null;
+                try
+                {
+                    var quotaSnapshots = aData?.GetType().GetProperty("QuotaSnapshots")?.GetValue(aData);
+                    if (quotaSnapshots is Dictionary<string, object> qs &&
+                        qs.TryGetValue("premium_interactions", out var premiumObj) &&
+                        premiumObj is System.Text.Json.JsonElement je)
+                    {
+                        var isUnlimited = je.TryGetProperty("isUnlimitedEntitlement", out var u) && u.GetBoolean();
+                        var entitlement = je.TryGetProperty("entitlementRequests", out var e) ? e.GetInt32() : -1;
+                        var used = je.TryGetProperty("usedRequests", out var ur) ? ur.GetInt32() : 0;
+                        var remaining = je.TryGetProperty("remainingPercentage", out var rp) ? rp.GetInt32() : 100;
+                        var resetDate = je.TryGetProperty("resetDate", out var rd) ? rd.GetString() : null;
+                        aPremiumQuota = new QuotaInfo(isUnlimited, entitlement, used, remaining, resetDate);
+                    }
+                }
+                catch { }
                 if (!string.IsNullOrEmpty(aModel))
                     state.Info.Model = aModel;
-                if (aInput.HasValue || aOutput.HasValue)
+                if (aInput.HasValue || aOutput.HasValue || aPremiumQuota != null)
                 {
-                    Invoke(() => OnUsageInfoChanged?.Invoke(sessionName, new SessionUsageInfo(aModel, null, null, aInput, aOutput)));
+                    Invoke(() => OnUsageInfoChanged?.Invoke(sessionName, new SessionUsageInfo(aModel, null, null, aInput, aOutput, aPremiumQuota)));
                 }
                 break;
 

PolyPilot/Services/CopilotService.cs

@@ -1040,5 +1040,14 @@ public record SessionUsageInfo(
     int? CurrentTokens,
     int? TokenLimit,
     int? InputTokens,
-    int? OutputTokens
+    int? OutputTokens,
+    QuotaInfo? PremiumQuota = null
+);
+
+public record QuotaInfo(
+    bool IsUnlimited,
+    int EntitlementRequests,
+    int UsedRequests,
+    int RemainingPercentage,
+    string? ResetDate
 );