mirror of
https://github.com/wu736139669/hapi.git
synced 2026-08-05 06:24:37 +00:00
17 lines
677 B
TypeScript
17 lines
677 B
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import { getContextBudgetTokens } from './modelConfig'
|
|
|
|
describe('getContextBudgetTokens', () => {
|
|
it('uses the existing 200k budget for default Claude modes', () => {
|
|
expect(getContextBudgetTokens(undefined)).toBe(190_000)
|
|
expect(getContextBudgetTokens('default')).toBe(190_000)
|
|
expect(getContextBudgetTokens('sonnet')).toBe(190_000)
|
|
expect(getContextBudgetTokens('opus')).toBe(190_000)
|
|
})
|
|
|
|
it('uses the 1m budget for Claude 1m modes', () => {
|
|
expect(getContextBudgetTokens('sonnet[1m]')).toBe(990_000)
|
|
expect(getContextBudgetTokens('opus[1m]')).toBe(990_000)
|
|
})
|
|
})
|