chore: import upstream snapshot with attribution

2026-07-13 12:26:02 +08:00
commit a0a48b5e45
437 changed files with 100767 additions and 0 deletions
@@ -0,0 +1,112 @@
+import { describe, it, expect } from 'vitest'
+
+import { aggregateAudit } from '../src/audit-report.js'
+import type {
+  ProjectSummary,
+  SessionSummary,
+  ClassifiedTurn,
+  ParsedApiCall,
+  TokenUsage,
+  TaskCategory,
+} from '../src/types.js'
+
+function emptyTokens(): TokenUsage {
+  return {
+    inputTokens: 0,
+    outputTokens: 0,
+    cacheCreationInputTokens: 0,
+    cacheReadInputTokens: 0,
+    cachedInputTokens: 0,
+    reasoningTokens: 0,
+    webSearchRequests: 0,
+  }
+}
+
+function makeCall(usage: Partial<TokenUsage>, costUSD: number, model = 'unknown-model-xyz', provider = 'claude'): ParsedApiCall {
+  return {
+    provider,
+    model,
+    usage: { ...emptyTokens(), ...usage },
+    costUSD,
+    tools: [],
+    mcpTools: [],
+    skills: [],
+    hasAgentSpawn: false,
+    hasPlanMode: false,
+    speed: 'standard',
+    timestamp: '2026-05-09T00:00:00.000Z',
+    bashCommands: [],
+    deduplicationKey: `${provider}-${model}-${costUSD}-${usage.inputTokens ?? 0}-${usage.cachedInputTokens ?? 0}`,
+  }
+}
+
+function makeProject(calls: ParsedApiCall[]): ProjectSummary {
+  const turn: ClassifiedTurn = {
+    userMessage: 't',
+    assistantCalls: calls,
+    timestamp: '2026-05-09T00:00:00.000Z',
+    sessionId: 's1',
+    category: 'feature' as TaskCategory,
+    retries: 0,
+    hasEdits: false,
+  }
+  const session: SessionSummary = {
+    sessionId: 's1',
+    project: 'p',
+    firstTimestamp: '2026-05-09T00:00:00.000Z',
+    lastTimestamp: '2026-05-09T00:00:00.000Z',
+    totalCostUSD: 0,
+    totalInputTokens: 0,
+    totalOutputTokens: 0,
+    totalCacheReadTokens: 0,
+    totalCacheWriteTokens: 0,
+    apiCalls: 0,
+    turns: [turn],
+    modelBreakdown: {},
+    toolBreakdown: {},
+    mcpBreakdown: {},
+    bashBreakdown: {},
+    categoryBreakdown: {} as SessionSummary['categoryBreakdown'],
+    skillBreakdown: {},
+  }
+  return { project: 'p', projectPath: 'p', sessions: [session], totalCostUSD: 0, totalApiCalls: 0 }
+}
+
+describe('aggregateAudit', () => {
+  it('keeps raw fields and exposes codeburn normalizations', async () => {
+    const anthropicCall = makeCall({ inputTokens: 100, outputTokens: 50, reasoningTokens: 10, cacheReadInputTokens: 200 }, 0.5)
+    const openaiCall = makeCall({ inputTokens: 100, outputTokens: 50, cachedInputTokens: 300 }, 0.5)
+    const rows = await aggregateAudit([makeProject([anthropicCall, openaiCall])])
+
+    expect(rows).toHaveLength(1)
+    const r = rows[0]!
+    // raw fields are summed untouched
+    expect(r.raw.inputTokens).toBe(200)
+    expect(r.raw.outputTokens).toBe(100)
+    expect(r.raw.reasoningTokens).toBe(10)
+    expect(r.raw.cacheReadInputTokens).toBe(200)
+    expect(r.raw.cachedInputTokens).toBe(300)
+    // reasoning folds into output for pricing
+    expect(r.displayed.outputTokens).toBe(110)
+    // cache read is the SUM of per-call max(anthropic, openai), not max of sums
+    expect(r.displayed.cacheReadTokens).toBe(500)
+    // attributed cost is preserved exactly
+    expect(r.attributedCostUSD).toBeCloseTo(1.0)
+  })
+
+  it('returns null rates and zero component cost for an unpriced model', async () => {
+    const rows = await aggregateAudit([makeProject([makeCall({ inputTokens: 1000 }, 0, 'definitely-not-a-real-model-zzz')])])
+    expect(rows).toHaveLength(1)
+    expect(rows[0]!.rates).toBeNull()
+    expect(rows[0]!.cost.recomputedTotalUSD).toBe(0)
+  })
+
+  it('splits buckets by (provider, model)', async () => {
+    const rows = await aggregateAudit([makeProject([
+      makeCall({ inputTokens: 10 }, 0.1, 'model-a', 'claude'),
+      makeCall({ inputTokens: 20 }, 0.2, 'model-b', 'claude'),
+      makeCall({ inputTokens: 30 }, 0.3, 'model-a', 'codex'),
+    ])])
+    expect(rows).toHaveLength(3)
+  })
+})