feat: support native budget simulation parsing

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

Author: asizikov

src/App.tsx

@@ -358,11 +358,6 @@ function App() {
   const handleApplyBudgetSimulation = useCallback(async () => {
     const file = currentFileRef.current
     if (!file) return
-    if (isNativeAiCreditsReport) {
-      setBudgetSimulation(null)
-      setBudgetSimulationError('Budget simulation is not available for native AI Credits reports yet.')
-      return
-    }
 
     const budgetReportUsers = userUsage?.users ?? []
     const hasBudgetOrganizationContext = budgetReportUsers.some((user) => user.organizations.length > 0 || user.costCenters.length > 0)
@@ -427,6 +422,7 @@ function App() {
           },
         },
         resolveIncludedCreditOverrides(seatOverrides),
+        { reportMetadata: reportMetadata ?? undefined },
       )
 
       if (simulationId !== latestSimulationIdRef.current) return
@@ -448,7 +444,7 @@ function App() {
     budgetValues.productCopilot,
     budgetValues.productSpark,
     budgetValues.user,
-    isNativeAiCreditsReport,
+    reportMetadata,
     resolveIncludedCreditOverrides,
     seatOverrides,
     userUsage,

src/utils/budgetSimulation.test.ts

@@ -23,11 +23,35 @@ const HEADER = [
   'aic_gross_amount',
 ].join(',')
 
+const NATIVE_AI_CREDITS_HEADER = [
+  'date',
+  'username',
+  'product',
+  'sku',
+  'model',
+  'quantity',
+  'unit_type',
+  'applied_cost_per_quantity',
+  'gross_amount',
+  'discount_amount',
+  'net_amount',
+  'total_monthly_quota',
+  'organization',
+  'cost_center_name',
+  'aic_quantity',
+  'aic_gross_amount',
+].join(',')
+
 function createCsv(rows: string[][]): File {
   const body = [HEADER, ...rows.map((row) => row.join(','))].join('\n')
   return new File([body], 'usage.csv', { type: 'text/csv' })
 }
 
+function createNativeAiCreditsCsv(rows: string[][]): File {
+  const body = [NATIVE_AI_CREDITS_HEADER, ...rows.map((row) => row.join(','))].join('\n')
+  return new File([body], 'native-usage.csv', { type: 'text/csv' })
+}
+
 function createRecord(overrides: Partial<TokenUsageRecord>): TokenUsageRecord {
   const quantity = overrides.quantity ?? 0
 
@@ -471,4 +495,34 @@ describe('runBudgetSimulation', () => {
       adjustedDailyGrossCostByDate: [{ date: '2026-04-25', amount: 0.5 }],
     })
   })
+
+  it('uses native AI Credits report parsing and policy context when report metadata is provided', async () => {
+    const file = createNativeAiCreditsCsv([
+      ['6/1/26', 'mona', 'copilot', 'copilot_ai_credit', 'GPT-5', '100', 'ai-credits', '0.01', '1.00', '0', '1.00', '0', 'example-org', 'Cost Center A', '', ''],
+    ])
+
+    await expect(runBudgetSimulation(
+      file,
+      { accountBudgetUsd: 0.5 },
+      {},
+      {
+        reportMetadata: {
+          format: 'native-ai-credits',
+          label: 'Native AI Credits report',
+        },
+      },
+    )).resolves.toEqual({
+      totalBill: 0.5,
+      blockedUsers: 1,
+      blockedRequests: 0,
+      blockedIncludedCreditsAic: 0,
+      allowedAicQuantity: 50,
+      budgetExhausted: true,
+      firstUserBlockedDate: null,
+      accountBlockedDate: '2026-06-01',
+      productBlockedDates: {},
+      adjustedDailyNetCostByDate: [{ date: '2026-06-01', amount: 0.5 }],
+      adjustedDailyGrossCostByDate: [{ date: '2026-06-01', amount: 0.5 }],
+    })
+  })
 })

src/utils/budgetSimulation.ts

@@ -1,5 +1,6 @@
 import { calculateAicIncludedCreditsContext, getUsageMonthKey, type AicIncludedCreditsContext, type AicIncludedCreditsOverrides } from '../pipeline/aicIncludedCredits'
-import { getAicUsageMetrics, getUsageMetrics, parseNormalizedTokenUsageRecord, parseTokenUsageHeader, type TokenUsageHeader, type TokenUsageRecord } from '../pipeline/parser'
+import { getAicUsageMetrics, getUsageMetrics, parseNativeAiCreditsUsageRecord, parseNormalizedTokenUsageRecord, parseTokenUsageHeader, type TokenUsageHeader, type TokenUsageRecord } from '../pipeline/parser'
+import type { ReportFormatMetadata } from '../pipeline/reportAdapters'
 import { getProductBudgetName, isNonCopilotCodeReviewUsage, NON_COPILOT_CODE_REVIEW_USER_LABEL, type ProductBudgetName } from '../pipeline/productClassification'
 import { streamLines } from '../pipeline/streamer'
 import type { UserSpendSegmentId } from './userSpendSegments'
@@ -26,6 +27,10 @@ export type BudgetSimulationOptions = {
   productBudgetsUsd?: Partial<Record<ProductBudgetName, number>>
 }
 
+export type BudgetSimulationRunOptions = {
+  reportMetadata?: ReportFormatMetadata
+}
+
 type BudgetSimulationContext = Pick<AicIncludedCreditsContext, 'reportPlanScope' | 'organizationIncludedCreditsPool' | 'individualMonthlyIncludedCredits'>
 type BudgetSimulationState = {
   remainingAccountBudget: number
@@ -378,8 +383,11 @@ export async function runBudgetSimulation(
   file: File,
   options: BudgetSimulationOptions,
   includedCreditsOverrides: AicIncludedCreditsOverrides = {},
+  runOptions: BudgetSimulationRunOptions = {},
 ): Promise<BudgetSimulationResult> {
-  const context = await calculateAicIncludedCreditsContext(file, includedCreditsOverrides)
+  const context = await calculateAicIncludedCreditsContext(file, includedCreditsOverrides, {
+    reportMetadata: runOptions.reportMetadata,
+  })
   const state = createBudgetSimulationState(options, context)
   let header: TokenUsageHeader | null = null
 
@@ -392,7 +400,9 @@ export async function runBudgetSimulation(
       continue
     }
 
-    const record = parseNormalizedTokenUsageRecord(trimmed, header)
+    const record = runOptions.reportMetadata?.format === 'native-ai-credits'
+      ? parseNativeAiCreditsUsageRecord(trimmed, header)
+      : parseNormalizedTokenUsageRecord(trimmed, header)
     if (!record) continue
 
     simulateBudgetRecord(state, record, context)