mirror of
https://github.com/AgentSeal/codeburn.git
synced 2026-10-27 01:00:36 +00:00
c166e69e46
Consumption-share split in `codeburn models`, reworked per review: - First-party routes only: the split applies to `dsh` (DeepSeek API) and `zcode` (Z.ai plan) calls. The same model behind a gateway never had the vendor discount, and reported-cost providers may already carry it; the gate set stays disjoint from REPORTED_COST_PROVIDERS. - The numbers agree by construction: both legs are shares of the stored list-rate cost (peakUSD + offPeakUSD == costUSD, pinned in tests). The 0.5x off-peak discount stays on the vendors' bills; nothing is repriced here. peakCostMultiplier and OFF_PEAK_MULTIPLIER are gone. - Calendars: unverified 2027 projections dropped; the Z.ai 2026-09-25..10-07 all-day off-peak promotion is honored as a temporary range; deepSeekHolidayCoverageYears() plus a test fail when the current year falls out of the holiday list. - Output shape: JSON carries the five peak keys only on rows with a split, so a report without DeepSeek/Z.ai first-party usage keeps its exact historical shape; the Z.ai quota note is removed. - The `codeburn peak` live-status command is left out (scheduling tool, not usage tracking); it survives on a separate branch for its own PR.
148 lines
6.3 KiB
TypeScript
148 lines
6.3 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
|
|
import {
|
|
classifyPeak,
|
|
deepSeekHolidayCoverageYears,
|
|
isDeepSeekPeak,
|
|
isZaiPeak,
|
|
peakBillingKind,
|
|
} from '../src/peak-hours.js'
|
|
|
|
const at = (iso: string): Date => new Date(iso)
|
|
|
|
describe('isDeepSeekPeak', () => {
|
|
it('is peak on a weekday inside 01:00-04:00 UTC', () => {
|
|
// Monday 2026-09-28.
|
|
expect(isDeepSeekPeak(at('2026-09-28T01:00:00Z'))).toBe(true)
|
|
expect(isDeepSeekPeak(at('2026-09-28T02:30:00Z'))).toBe(true)
|
|
expect(isDeepSeekPeak(at('2026-09-28T03:59:59Z'))).toBe(true)
|
|
})
|
|
|
|
it('is peak on a weekday inside 06:00-10:00 UTC', () => {
|
|
expect(isDeepSeekPeak(at('2026-09-28T06:00:00Z'))).toBe(true)
|
|
expect(isDeepSeekPeak(at('2026-09-28T09:59:59Z'))).toBe(true)
|
|
})
|
|
|
|
it('is off-peak in the gaps between windows', () => {
|
|
expect(isDeepSeekPeak(at('2026-09-28T00:59:59Z'))).toBe(false)
|
|
expect(isDeepSeekPeak(at('2026-09-28T04:00:00Z'))).toBe(false)
|
|
expect(isDeepSeekPeak(at('2026-09-28T05:30:00Z'))).toBe(false)
|
|
expect(isDeepSeekPeak(at('2026-09-28T10:00:00Z'))).toBe(false)
|
|
expect(isDeepSeekPeak(at('2026-09-28T23:00:00Z'))).toBe(false)
|
|
})
|
|
|
|
it('is off-peak on weekends', () => {
|
|
// Saturday 2026-09-26, Sunday 2026-09-27.
|
|
expect(isDeepSeekPeak(at('2026-09-26T02:00:00Z'))).toBe(false)
|
|
expect(isDeepSeekPeak(at('2026-09-27T08:00:00Z'))).toBe(false)
|
|
})
|
|
|
|
it('is off-peak on Chinese public holidays', () => {
|
|
// Mid-Autumn 2026-09-25 is a Friday; National Day 2026-10-01 is a Thursday.
|
|
expect(isDeepSeekPeak(at('2026-09-25T02:00:00Z'))).toBe(false)
|
|
expect(isDeepSeekPeak(at('2026-10-01T08:00:00Z'))).toBe(false)
|
|
})
|
|
})
|
|
|
|
describe('isZaiPeak', () => {
|
|
// Monday 2026-09-21 sits outside the 2026-09-25..2026-10-07 promotion, so
|
|
// these pin the plain weekly window.
|
|
it('is peak Mon-Fri 14:00-18:00 SGT (06:00-10:00 UTC)', () => {
|
|
expect(isZaiPeak(at('2026-09-21T06:00:00Z'))).toBe(true)
|
|
expect(isZaiPeak(at('2026-09-21T09:59:59Z'))).toBe(true)
|
|
})
|
|
|
|
it('is off-peak outside the SGT window', () => {
|
|
expect(isZaiPeak(at('2026-09-21T05:59:59Z'))).toBe(false)
|
|
expect(isZaiPeak(at('2026-09-21T10:00:00Z'))).toBe(false)
|
|
expect(isZaiPeak(at('2026-09-21T02:00:00Z'))).toBe(false)
|
|
})
|
|
|
|
it('is off-peak on weekends (no holiday exception)', () => {
|
|
expect(isZaiPeak(at('2026-09-19T08:00:00Z'))).toBe(false)
|
|
})
|
|
|
|
it('treats the 2026-09-25..10-07 promotion as all-day off-peak', () => {
|
|
// Monday 2026-09-28 and Wednesday 2026-10-07 are weekdays inside the
|
|
// 06:00-10:00 UTC window, yet the promotion makes them off-peak.
|
|
expect(isZaiPeak(at('2026-09-28T08:00:00Z'))).toBe(false)
|
|
expect(isZaiPeak(at('2026-10-07T07:00:00Z'))).toBe(false)
|
|
// The Thursday after the promotion ends goes back to the weekly window.
|
|
expect(isZaiPeak(at('2026-10-08T07:00:00Z'))).toBe(true)
|
|
})
|
|
})
|
|
|
|
describe('deepSeekHolidayCoverageYears', () => {
|
|
it('covers the current year', () => {
|
|
// Fails the moment the calendar goes stale: a new year needs its verified
|
|
// holiday dates added (never projected ahead).
|
|
expect(deepSeekHolidayCoverageYears().has(new Date().getUTCFullYear())).toBe(true)
|
|
})
|
|
|
|
it('does not project future years', () => {
|
|
expect(deepSeekHolidayCoverageYears().has(2027)).toBe(false)
|
|
})
|
|
})
|
|
|
|
describe('peakBillingKind', () => {
|
|
it('maps first-party DeepSeek routes to USD peak billing', () => {
|
|
expect(peakBillingKind('dsh', 'deepseek-chat')).toBe('deepseek-usd')
|
|
expect(peakBillingKind('dsh', 'deepseek/deepseek-chat')).toBe('deepseek-usd')
|
|
expect(peakBillingKind('dsh', 'deepseek-reasoner')).toBe('deepseek-usd')
|
|
expect(peakBillingKind('dsh', 'deepseek-v4-pro')).toBe('deepseek-usd')
|
|
expect(peakBillingKind('dsh', 'deepseek-flash')).toBe('deepseek-usd')
|
|
})
|
|
|
|
it('maps first-party Z.ai routes to credit peak billing', () => {
|
|
expect(peakBillingKind('zcode', 'glm-5.3')).toBe('zai-credits')
|
|
expect(peakBillingKind('zcode', 'GLM-5.2')).toBe('zai-credits')
|
|
expect(peakBillingKind('zcode', 'z-ai/glm-5.2')).toBe('zai-credits')
|
|
expect(peakBillingKind('zcode', 'cliproxy/zcode/glm-5.3-flash')).toBe('zai-credits')
|
|
})
|
|
|
|
it('returns null through gateways, resellers and other providers', () => {
|
|
// The same model behind a gateway pays the gateway's uniform price — no
|
|
// vendor discount to attribute.
|
|
expect(peakBillingKind('openrouter', 'deepseek/deepseek-chat')).toBe(null)
|
|
expect(peakBillingKind('openrouter', 'z-ai/glm-5.2')).toBe(null)
|
|
expect(peakBillingKind('vercel-gateway', 'glm-5.3')).toBe(null)
|
|
expect(peakBillingKind('zerostack', 'deepseek/deepseek-v4-pro')).toBe(null)
|
|
expect(peakBillingKind('claude', 'glm-5.3')).toBe(null)
|
|
})
|
|
|
|
it('returns null when provider and model vendor disagree', () => {
|
|
expect(peakBillingKind('dsh', 'glm-5.3')).toBe(null)
|
|
expect(peakBillingKind('zcode', 'deepseek-chat')).toBe(null)
|
|
})
|
|
|
|
it('returns null for unrelated models and junk', () => {
|
|
expect(peakBillingKind('dsh', 'claude-sonnet-4-6')).toBe(null)
|
|
expect(peakBillingKind('zcode', 'gpt-5.5')).toBe(null)
|
|
expect(peakBillingKind('dsh', '')).toBe(null)
|
|
})
|
|
})
|
|
|
|
describe('classifyPeak', () => {
|
|
it('classifies a first-party DeepSeek call by its timestamp', () => {
|
|
expect(classifyPeak('dsh', 'deepseek-chat', '2026-09-28T02:00:00Z')).toBe('peak')
|
|
expect(classifyPeak('dsh', 'deepseek-chat', '2026-09-28T12:00:00Z')).toBe('off-peak')
|
|
})
|
|
|
|
it('classifies a first-party GLM call by the Z.ai window', () => {
|
|
// 2026-09-21 is outside the Sep 25 - Oct 7 promotion.
|
|
expect(classifyPeak('zcode', 'glm-5.3', '2026-09-21T08:00:00Z')).toBe('peak')
|
|
expect(classifyPeak('zcode', 'glm-5.3', '2026-09-21T12:00:00Z')).toBe('off-peak')
|
|
})
|
|
|
|
it('returns unknown through gateways and provider/model mismatches', () => {
|
|
expect(classifyPeak('openrouter', 'deepseek/deepseek-chat', '2026-09-28T02:00:00Z')).toBe('unknown')
|
|
expect(classifyPeak('vercel-gateway', 'glm-5.3', '2026-09-21T08:00:00Z')).toBe('unknown')
|
|
expect(classifyPeak('dsh', 'glm-5.3', '2026-09-28T02:00:00Z')).toBe('unknown')
|
|
})
|
|
|
|
it('returns unknown without a model match or timestamp', () => {
|
|
expect(classifyPeak('dsh', 'gpt-5.5', '2026-09-28T02:00:00Z')).toBe('unknown')
|
|
expect(classifyPeak('dsh', 'deepseek-chat', undefined)).toBe('unknown')
|
|
expect(classifyPeak('dsh', 'deepseek-chat', 'not-a-date')).toBe('unknown')
|
|
})
|
|
})
|