feat: cap per-user AI token spend

2026-07-30 23:32:13 +02:00
parent 09fc2b03f7
commit 792814b04e
5 changed files with 18 additions and 9 deletions
@@ -22,7 +22,7 @@ public sealed class AiUsageController : ControllerBase
    }

    public sealed record UsagePeriodDto(int Calls, long InputCharacters, long OutputCharacters, long EstimatedTokens);
-    public sealed record UsageDto(UsagePeriodDto CurrentMonth, UsagePeriodDto AllTime, string Plan, int MonthlyCallLimit);
+    public sealed record UsageDto(UsagePeriodDto CurrentMonth, UsagePeriodDto AllTime, string Plan, int MonthlyCallLimit, long MonthlyTokenLimit);

    [HttpGet]
    public async Task<ActionResult<UsageDto>> Get(CancellationToken cancellationToken)
@@ -37,7 +37,8 @@ public sealed class AiUsageController : ControllerBase
            await SumAsync(_db.AiInteractions.Where(x => x.OwnerUserId == user.Id && x.CreatedAtUtc >= monthStart), cancellationToken),
            await SumAsync(_db.AiInteractions.Where(x => x.OwnerUserId == user.Id), cancellationToken),
            entitlements.AdvancedAi ? "premium" : "free",
-            entitlements.MonthlyAiCalls));
+            entitlements.MonthlyAiCalls,
+            entitlements.MonthlyAiTokens));
    }

    private static async Task<UsagePeriodDto> SumAsync(IQueryable<AiInteraction> query, CancellationToken cancellationToken)