fix: handle unlabeled review usage budgets

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: asizikov

src/utils/enterpriseBudgetSimulation.test.ts

@@ -143,6 +143,28 @@ describe('simulateBudgetFromRecords', () => {
     })
   })
 
+  it('uses the non-copilot code review label for empty-username user budgets', () => {
+    const result = simulateBudgetFromRecords([
+      createRecord({ username: '', model: 'code review', quantity: 50, aic_quantity: 50, aic_gross_amount: 5 }),
+      createRecord({ username: '', model: 'code review', quantity: 50, aic_quantity: 50, aic_gross_amount: 5 }),
+    ], { userBudgetUsd: 5 }, {
+      ...pooledContext,
+      organizationEntitlementPool: 100,
+    })
+
+    expect(result).toEqual({
+      totalBill: 0,
+      blockedUsers: 1,
+      blockedRequests: 50,
+      blockedEntitlementsAic: 50,
+      budgetExhausted: false,
+      firstUserBlockedDate: '2026-06-01',
+      enterpriseBlockedDate: null,
+      productBlockedDates: {},
+      adjustedDailyNetCostByDate: [],
+    })
+  })
+
   it('does not count entitlements that later get consumed by other usage', () => {
     const result = simulateBudgetFromRecords([
       createRecord({ username: 'mona', quantity: 50, aic_quantity: 50, aic_gross_amount: 5 }),

src/utils/enterpriseBudgetSimulation.ts

@@ -1,6 +1,6 @@
 import { calculateAicEntitlementContext, type AicEntitlementContext, type AicEntitlementOverrides } from '../pipeline/aicEntitlements'
 import { getAicUsageMetrics, getUsageMetrics, parseTokenUsageHeader, parseTokenUsageRecord, type TokenUsageHeader, type TokenUsageRecord } from '../pipeline/parser'
-import { getProductBudgetName, type ProductBudgetName } from '../pipeline/productClassification'
+import { getProductBudgetName, isNonCopilotCodeReviewUsage, NON_COPILOT_CODE_REVIEW_USER_LABEL, type ProductBudgetName } from '../pipeline/productClassification'
 import { streamLines } from '../pipeline/streamer'
 
 export type BudgetSimulationResult = {
@@ -61,6 +61,19 @@ function getIndividualEntitlementKey(record: TokenUsageRecord): string | null {
   return `${username}\u0000${weekStart}`
 }
 
+function getBudgetSubject(record: TokenUsageRecord): string | null {
+  const username = record.username.trim()
+  if (username) {
+    return username
+  }
+
+  if (isNonCopilotCodeReviewUsage(record)) {
+    return NON_COPILOT_CODE_REVIEW_USER_LABEL
+  }
+
+  return null
+}
+
 function getRemainingEntitlements(
   record: TokenUsageRecord,
   context: BudgetSimulationContext,
@@ -123,7 +136,7 @@ export function simulateBudgetFromRecords(
   const seenIndividualEntitlementKeys = new Set<string>()
 
   for (const record of records) {
-    const username = record.username.trim()
+    const budgetSubject = getBudgetSubject(record)
     const productBudgetName = getProductBudgetName(record)
     const { requests } = getUsageMetrics(record)
     const { aicQuantity, aicGrossAmount } = getAicUsageMetrics(record)
@@ -145,7 +158,7 @@ export function simulateBudgetFromRecords(
 
     const remainingUserBudget = userBudgetCap === Number.POSITIVE_INFINITY
       ? Number.POSITIVE_INFINITY
-      : (remainingUserBudgetByUser.get(username) ?? userBudgetCap)
+      : (budgetSubject ? (remainingUserBudgetByUser.get(budgetSubject) ?? userBudgetCap) : Number.POSITIVE_INFINITY)
     const remainingProductBudget = remainingProductBudgetByName.get(productBudgetName) ?? Number.POSITIVE_INFINITY
     const remainingEntitlements = getRemainingEntitlements(
       record,
@@ -177,8 +190,8 @@ export function simulateBudgetFromRecords(
 
     if (allowedRatio < 1) {
       blockedRequests += requests * (1 - allowedRatio)
-      if (username) {
-        blockedUsers.add(username)
+      if (budgetSubject) {
+        blockedUsers.add(budgetSubject)
       }
       if (userBudgetLimited && firstUserBlockedDate === null) {
         firstUserBlockedDate = record.date || null
@@ -228,8 +241,8 @@ export function simulateBudgetFromRecords(
       remainingEnterpriseBudget = nextRemainingEnterpriseBudget
     }
 
-    if (remainingUserBudget !== Number.POSITIVE_INFINITY) {
-      remainingUserBudgetByUser.set(username, Math.max(remainingUserBudget - allowedGrossAmount, 0))
+    if (budgetSubject && remainingUserBudget !== Number.POSITIVE_INFINITY) {
+      remainingUserBudgetByUser.set(budgetSubject, Math.max(remainingUserBudget - allowedGrossAmount, 0))
     }
     if (remainingProductBudget !== Number.POSITIVE_INFINITY) {
       const nextRemainingProductBudget = Math.max(remainingProductBudget - overageAmount, 0)