bchanot
d0fa1001bb
Phases name absolute tiers (best fable>opus>sonnet, big opus>fable>sonnet, work sonnet>opus, cheap haiku>sonnet) resolved to the first available full id; per-model circuit breaker fed by StopFailure kinds (rate_limit, overloaded, billing_error, model_not_found) and PostModelSwitch auto, with episode backoff 15→300 min, cleared by a user /model or /route reload and kept across /clear; fallback chain fable→opus→sonnet→haiku with the effort unchanged; main loop upgrades to a phase's tier by itself under a context cap (fails closed on unknown usage), downgrades only with the switch on, sticky within a turn; derived orchestrate on background dispatches; prompt default rules (plan/reflect, Unicode guards, skipped on slash commands, floor matches and mid-turn). 58 plugin tests.
847 lines
27 KiB
TypeScript
847 lines
27 KiB
TypeScript
import { test, expect, mock } from 'claude-code/testing'
|
|
import type { Engine, MockClock } from 'claude-code/testing'
|
|
import type { On, TurnStepInput } from 'claude-code'
|
|
|
|
const FABLE = 'claude-fable-5-1'
|
|
const OPUS = 'claude-opus-5-5'
|
|
const SONNET = 'claude-sonnet-5-5'
|
|
const HAIKU = 'claude-haiku-4-5-20251001'
|
|
|
|
/**
|
|
* Fires session.start so the mod loads its config and registers /route.
|
|
* The bottom hooks the router reads are in place before the first `$` call:
|
|
* the session model, the classic failure and switch events, a mock clock,
|
|
* a small context (`tokens`; null = usage unanswered, size unknown).
|
|
*/
|
|
async function boot(
|
|
$: Engine,
|
|
on: On,
|
|
model: string = FABLE,
|
|
tokens: number | null = 1000,
|
|
): Promise<MockClock> {
|
|
const clock = mock.clock(on)
|
|
on('session.model', () => ({ value: model }))
|
|
on('classic.StopFailure', () => ({}))
|
|
on('classic.PostModelSwitch', () => ({}))
|
|
if (tokens !== null) usageOf(on, tokens)
|
|
on('session.start', ($, e) => ({ cwd: e.cwd }))
|
|
await $.session.start({ cwd: '/tmp', surface: null, isInteractive: false })
|
|
return clock
|
|
}
|
|
|
|
/** Runs `/route <args>` as the user would type it. */
|
|
async function route($: Engine, args: string): Promise<string> {
|
|
const out = await $.command.run({
|
|
command: 'route',
|
|
args,
|
|
origin: { kind: 'composer' },
|
|
presentation: { isFullscreen: false, columns: 80 },
|
|
})
|
|
return out.text ?? ''
|
|
}
|
|
|
|
/** The `main:` line of `show`: the phases listing holds every id and level. */
|
|
function mainLine(text: string): string {
|
|
return text.split('\n').find(l => l.startsWith('main:')) ?? ''
|
|
}
|
|
|
|
const spawnInput = (model?: string) => ({
|
|
tool_use_id: 't1',
|
|
prompt: 'x',
|
|
description: 'd',
|
|
subagentType: 'Explore',
|
|
provider: { plugin: 'engine', tier: 'core' as const },
|
|
parentModel: 'claude-fable-5-1',
|
|
background: false,
|
|
fork: false,
|
|
...(model === undefined ? {} : { model }),
|
|
})
|
|
|
|
test('Skill(effort-low) is answered without next, route shows low', async (
|
|
$, on) => {
|
|
let reached = false
|
|
on('tool.call', { tool: 'Skill' }, () => {
|
|
reached = true
|
|
return { result: { success: true, commandName: 'bottom' } }
|
|
})
|
|
await boot($, on)
|
|
const out = await $.tool.call({ tool: 'Skill', skill: 'effort-low' })
|
|
expect(out).toMatchObject({
|
|
result: { success: true, commandName: 'effort-low' },
|
|
})
|
|
expect(reached).toBe(false)
|
|
const line = mainLine(await route($, 'show'))
|
|
expect(line).toContain('skill effort-low')
|
|
expect(line).toContain('effort low')
|
|
})
|
|
|
|
test('route tool with phase orchestrate sets medium on main', async (
|
|
$, on) => {
|
|
await boot($, on)
|
|
const out = await $.tool.call({
|
|
tool: 'mcp__model-router__route',
|
|
phase: 'orchestrate',
|
|
})
|
|
expect(out).toHaveProperty('result')
|
|
const line = mainLine(await route($, 'show'))
|
|
expect(line).toContain('model orchestrate')
|
|
expect(line).toContain('effort medium')
|
|
})
|
|
|
|
test('/route clear drops the route', async ($, on) => {
|
|
await boot($, on)
|
|
await $.tool.call({ tool: 'mcp__model-router__route', phase: 'plan' })
|
|
expect(mainLine(await route($, 'show'))).toContain('model plan')
|
|
await route($, 'clear')
|
|
expect(mainLine(await route($, 'show'))).toContain('session defaults')
|
|
})
|
|
|
|
test('/route bogus names the phases', async ($, on) => {
|
|
await boot($, on)
|
|
const text = await route($, 'bogus')
|
|
expect(text).toContain('unknown')
|
|
for (const phase of ['plan', 'judge', 'explore', 'mechanical']) {
|
|
expect(text).toContain(phase)
|
|
}
|
|
})
|
|
|
|
test('ultrathink in a prompt sets a max floor on main', async ($, on) => {
|
|
on('prompt.submit', ($, e) => ({ text: e.text }))
|
|
await boot($, on)
|
|
await $.prompt.submit({
|
|
text: 'ultrathink please',
|
|
wait: false,
|
|
origin: { kind: 'composer' },
|
|
})
|
|
expect(mainLine(await route($, 'show'))).toContain('floor max')
|
|
})
|
|
|
|
test('/route model: alias resolved, id passed, typo refused', async (
|
|
$, on) => {
|
|
await boot($, on)
|
|
const alias = await route($, 'model=sonnet')
|
|
expect(mainLine(alias)).toContain('asked claude-sonnet-5-5, keeps ')
|
|
const id = await route($, 'model=claude-x-9')
|
|
expect(mainLine(id)).toContain('claude-x-9')
|
|
expect(await route($, 'model=sonet')).toContain('unknown')
|
|
})
|
|
|
|
test('Explore spawn gets the sonnet id; an explicit model wins', async (
|
|
$, on) => {
|
|
const seen: (string | undefined)[] = []
|
|
on('agent.spawn', ($, e) => {
|
|
seen.push(e.model)
|
|
return { model: e.model ?? e.parentModel, agentId: 'a1' }
|
|
})
|
|
await boot($, on)
|
|
await $.agent.spawn(spawnInput())
|
|
await $.agent.spawn(spawnInput('opus'))
|
|
expect(seen).toEqual(['claude-sonnet-5-5', 'opus'])
|
|
})
|
|
|
|
test('a sticky /route wins over a model-declared route', async ($, on) => {
|
|
await boot($, on)
|
|
await route($, 'judge')
|
|
const out = await $.tool.call({
|
|
tool: 'mcp__model-router__route',
|
|
phase: 'mechanical',
|
|
})
|
|
expect(JSON.stringify(out)).toContain('sticky')
|
|
expect(mainLine(await route($, 'show'))).toContain('user judge')
|
|
})
|
|
|
|
test('/route off makes the route tool a no-op', async ($, on) => {
|
|
await boot($, on)
|
|
await route($, 'off')
|
|
const out = await $.tool.call({
|
|
tool: 'mcp__model-router__route',
|
|
phase: 'plan',
|
|
})
|
|
expect(JSON.stringify(out)).toContain('off')
|
|
expect(mainLine(await route($, 'show'))).toContain('session defaults')
|
|
})
|
|
|
|
type Seen = { model: string; effort: unknown }
|
|
|
|
/** A bottom turn.step hook that records what reaches the model. */
|
|
function recordSteps(on: On, seen: Seen[]): void {
|
|
on('turn.step', async function* ($, e) {
|
|
seen.push({ model: e.model, effort: e.effort })
|
|
return {
|
|
turnId: e.turnId,
|
|
index: e.index,
|
|
answer: '',
|
|
toolUses: [],
|
|
stopReason: 'end_turn' as const,
|
|
usage: null,
|
|
}
|
|
})
|
|
}
|
|
|
|
const stepInput = (agentId?: string) => ({
|
|
turnId: 'u1',
|
|
index: 0,
|
|
model: 'claude-fable-5-1',
|
|
effort: 'low' as const,
|
|
messageCount: 1,
|
|
...(agentId === undefined ? {} : { agentId }),
|
|
})
|
|
|
|
/** Streams one turn.step to its end; the hooks run as the chunks flow. */
|
|
async function runStep(
|
|
$: Engine,
|
|
input: TurnStepInput,
|
|
): Promise<void> {
|
|
const stream = $.turn.step(input)
|
|
for await (const _chunk of stream) {
|
|
// chunks are not under test
|
|
}
|
|
await stream.result
|
|
}
|
|
|
|
test('the main step carries the routed effort, model untouched', async (
|
|
$, on) => {
|
|
const seen: Seen[] = []
|
|
recordSteps(on, seen)
|
|
await boot($, on)
|
|
await $.tool.call({ tool: 'mcp__model-router__route', phase: 'plan' })
|
|
await runStep($, stepInput())
|
|
expect(seen).toEqual([{ model: 'claude-fable-5-1', effort: 'xhigh' }])
|
|
})
|
|
|
|
test('a tabled agent steps at its table effort', async ($, on) => {
|
|
const seen: Seen[] = []
|
|
recordSteps(on, seen)
|
|
on('agent.spawn', ($, e) => ({
|
|
model: e.model ?? e.parentModel,
|
|
agentId: 'a1',
|
|
}))
|
|
await boot($, on)
|
|
await $.agent.spawn(spawnInput())
|
|
await runStep($, stepInput('a1'))
|
|
expect(seen).toEqual([{ model: 'claude-fable-5-1', effort: 'medium' }])
|
|
})
|
|
|
|
test('/route from a non-composer origin is refused, state kept', async (
|
|
$, on) => {
|
|
await boot($, on)
|
|
await route($, 'judge')
|
|
const out = await $.command.run({
|
|
command: 'route',
|
|
args: 'mechanical',
|
|
origin: { kind: 'plugin', name: 'x' },
|
|
presentation: { isFullscreen: false, columns: 80 },
|
|
})
|
|
expect(out.text).toContain('user-only')
|
|
expect(mainLine(await route($, 'show'))).toContain('user judge')
|
|
})
|
|
|
|
test('an in-agent route sets effort only, the model stays', async ($, on) => {
|
|
const seen: Seen[] = []
|
|
recordSteps(on, seen)
|
|
await boot($, on)
|
|
await $.tool.call({
|
|
tool: 'mcp__model-router__route',
|
|
phase: 'judge',
|
|
agentId: 'a1',
|
|
})
|
|
await runStep($, { ...stepInput('a1'), model: 'claude-sonnet-5-5' })
|
|
expect(seen).toEqual([{ model: 'claude-sonnet-5-5', effort: 'xhigh' }])
|
|
})
|
|
|
|
test('a rule only scans the first 4096 chars of a prompt', async ($, on) => {
|
|
on('prompt.submit', ($, e) => ({ text: e.text }))
|
|
await boot($, on)
|
|
await $.prompt.submit({
|
|
text: 'x'.repeat(5000) + ' ultrathink',
|
|
wait: false,
|
|
origin: { kind: 'composer' },
|
|
})
|
|
expect(mainLine(await route($, 'show'))).toContain('session defaults')
|
|
})
|
|
|
|
// ---- user effort floor -------------------------------------------------
|
|
|
|
const ROUTE_TOOL = 'mcp__model-router__route'
|
|
|
|
/** A main (or agent) step as the engine would make it: engine effort high. */
|
|
const highStep = (agentId?: string): TurnStepInput => ({
|
|
...stepInput(agentId),
|
|
effort: 'high' as const,
|
|
})
|
|
|
|
/** Types `ultrathink` in the composer, idle or over a running turn. */
|
|
async function ultrathink($: Engine, turnId?: string): Promise<void> {
|
|
await $.prompt.submit({
|
|
text: 'ultrathink please',
|
|
wait: turnId !== undefined,
|
|
origin: { kind: 'composer' },
|
|
...(turnId === undefined ? {} : { turnId }),
|
|
})
|
|
}
|
|
|
|
/** Ends a main turn: the floor's life is bounded by this event. */
|
|
async function endTurn($: Engine): Promise<void> {
|
|
await $.turn.complete({
|
|
turnId: 'u1',
|
|
answer: '',
|
|
durationMs: 1,
|
|
isAborted: false,
|
|
reason: 'answer',
|
|
})
|
|
}
|
|
|
|
/** One main step; returns the effort that reached the bottom hook. */
|
|
async function stepEffort($: Engine, seen: Seen[]): Promise<unknown> {
|
|
await runStep($, highStep())
|
|
return seen[seen.length - 1]?.effort
|
|
}
|
|
|
|
/** Boots with a bottom step recorder and prompt/turn hooks in place. */
|
|
async function bootFloor($: Engine, on: On): Promise<Seen[]> {
|
|
const seen: Seen[] = []
|
|
recordSteps(on, seen)
|
|
on('prompt.submit', ($, e) => ({ text: e.text }))
|
|
on('turn.complete', () => ({ text: '' }))
|
|
on('tool.call', { tool: 'Skill' }, () => ({
|
|
result: { success: true, commandName: 'other' },
|
|
}))
|
|
on('agent.spawn', ($, e) => ({
|
|
model: e.model ?? e.parentModel,
|
|
agentId: 'a1',
|
|
}))
|
|
await boot($, on)
|
|
return seen
|
|
}
|
|
|
|
test('floor: ultrathink survives a model route', async ($, on) => {
|
|
const seen = await bootFloor($, on)
|
|
await ultrathink($)
|
|
await $.tool.call({ tool: ROUTE_TOOL, phase: 'orchestrate' })
|
|
expect(await stepEffort($, seen)).toBe('max')
|
|
})
|
|
|
|
test('floor: typed /effort-medium clamps low, lets max pass', async (
|
|
$, on) => {
|
|
const seen = await bootFloor($, on)
|
|
await $.skill.prompt({ skill: 'effort-medium', text: 'x' })
|
|
await $.tool.call({ tool: ROUTE_TOOL, phase: 'mechanical' })
|
|
expect(await stepEffort($, seen)).toBe('medium')
|
|
await $.tool.call({ tool: ROUTE_TOOL, phase: 'escalate' })
|
|
expect(await stepEffort($, seen)).toBe('max')
|
|
})
|
|
|
|
test('floor: typed /effort-low lowers an unrouted turn', async ($, on) => {
|
|
const seen = await bootFloor($, on)
|
|
const out = await $.skill.prompt({ skill: 'effort-low', text: 'x' })
|
|
expect(out.text).toContain('minimum')
|
|
expect(await stepEffort($, seen)).toBe('low')
|
|
})
|
|
|
|
test('floor: survives a skill load', async ($, on) => {
|
|
const seen = await bootFloor($, on)
|
|
await ultrathink($)
|
|
await $.tool.call({ tool: ROUTE_TOOL, phase: 'orchestrate' })
|
|
await $.tool.call({ tool: 'Skill', skill: 'other' })
|
|
expect(await stepEffort($, seen)).toBe('max')
|
|
expect(mainLine(await route($, 'show'))).not.toContain('orchestrate')
|
|
})
|
|
|
|
test('floor: lifts a lower sticky, then ends with the turn', async (
|
|
$, on) => {
|
|
const seen = await bootFloor($, on)
|
|
await route($, 'effort=low')
|
|
await ultrathink($)
|
|
expect(await stepEffort($, seen)).toBe('max')
|
|
await endTurn($)
|
|
expect(await stepEffort($, seen)).toBe('low')
|
|
})
|
|
|
|
test('floor: main only, an agent step is unaffected', async ($, on) => {
|
|
const seen = await bootFloor($, on)
|
|
await $.agent.spawn(spawnInput())
|
|
await ultrathink($)
|
|
await runStep($, highStep('a1'))
|
|
expect(seen[0]?.effort).toBe('medium')
|
|
})
|
|
|
|
test('floor: /route clear removes it', async ($, on) => {
|
|
const seen = await bootFloor($, on)
|
|
await ultrathink($)
|
|
expect(mainLine(await route($, 'show'))).toContain('floor max')
|
|
await route($, 'clear')
|
|
expect(await stepEffort($, seen)).toBe('high')
|
|
expect(mainLine(await route($, 'show'))).not.toContain('floor')
|
|
})
|
|
|
|
test('floor: a mid-turn prompt applies now and is kept for the next', async (
|
|
$, on) => {
|
|
const seen = await bootFloor($, on)
|
|
await ultrathink($, 'u1')
|
|
expect(await stepEffort($, seen)).toBe('max')
|
|
await endTurn($)
|
|
expect(await stepEffort($, seen)).toBe('max')
|
|
await endTurn($)
|
|
expect(await stepEffort($, seen)).toBe('high')
|
|
})
|
|
|
|
test('floor: the route answer names the floor over a sticky', async (
|
|
$, on) => {
|
|
await bootFloor($, on)
|
|
await route($, 'effort=low')
|
|
await ultrathink($)
|
|
const out = await $.tool.call({ tool: ROUTE_TOOL, phase: 'plan' })
|
|
const text = JSON.stringify(out)
|
|
expect(text).toContain('user floor max (prompt rule escalate)')
|
|
expect(text).toContain('/route clear')
|
|
})
|
|
|
|
// `enabled: false` in ~/.claude/model-router.json cannot be reached here (the
|
|
// kit has no fs); the shared off path is covered through `/route off`.
|
|
test('floor: /route off keeps the floor from routing', async ($, on) => {
|
|
const seen = await bootFloor($, on)
|
|
await ultrathink($)
|
|
await route($, 'off')
|
|
expect(await stepEffort($, seen)).toBe('high')
|
|
expect(mainLine(await route($, 'show'))).not.toContain('floor')
|
|
})
|
|
|
|
test('per axis: a model-only sticky keeps the turn route effort', async (
|
|
$, on) => {
|
|
const seen = await bootFloor($, on)
|
|
await route($, 'model=sonnet')
|
|
await $.tool.call({ tool: ROUTE_TOOL, phase: 'orchestrate' })
|
|
expect(await stepEffort($, seen)).toBe('medium')
|
|
})
|
|
|
|
// The config-driven `offConfig` is re-applied in the same rebuild; only the
|
|
// session-only off is reachable here (the kit has no fs).
|
|
test('session.end rebuild drops a session-only /route off', async (
|
|
$, on) => {
|
|
on('session.end', ($, e) => ({ sessionId: e.sessionId }))
|
|
await boot($, on)
|
|
await route($, 'off')
|
|
await $.session.end({
|
|
reason: 'clear',
|
|
sessionId: 's1',
|
|
resume: { id: 's1' },
|
|
})
|
|
expect(await route($, 'show')).toContain('router: on')
|
|
})
|
|
|
|
// The kit has no fs: the reload read fails, so the previous cfg must stay.
|
|
test('reload with an unreadable override keeps the previous config', async (
|
|
$, on) => {
|
|
await boot($, on)
|
|
await route($, 'switch on')
|
|
const out = await route($, 'reload')
|
|
expect(out).toContain('switch: on')
|
|
expect(await route($, 'show')).toContain('switch: on')
|
|
})
|
|
|
|
test('typed marker: a preload in a live agent is ignored', async ($, on) => {
|
|
const seen = await bootFloor($, on)
|
|
await $.agent.spawn(spawnInput())
|
|
const out = await $.skill.prompt({ skill: 'effort-max', text: 'x' })
|
|
expect(out.text?.startsWith('model-router: effort-max preload')).toBe(true)
|
|
expect(mainLine(await route($, 'show'))).not.toContain('floor')
|
|
expect(await stepEffort($, seen)).not.toBe('max')
|
|
})
|
|
|
|
test('typed marker: /effort-max seen at submit writes the floor', async (
|
|
$, on) => {
|
|
await bootFloor($, on)
|
|
await $.agent.spawn(spawnInput())
|
|
await $.prompt.submit({
|
|
text: '/effort-max go',
|
|
wait: false,
|
|
origin: { kind: 'composer' },
|
|
})
|
|
await $.skill.prompt({ skill: 'effort-max', text: 'x' })
|
|
expect(mainLine(await route($, 'show'))).toContain('floor max')
|
|
})
|
|
|
|
test('typed slash with no agent and no marker writes the floor', async (
|
|
$, on) => {
|
|
await bootFloor($, on)
|
|
await $.skill.prompt({ skill: 'effort-max', text: 'x' })
|
|
expect(mainLine(await route($, 'show'))).toContain('floor max')
|
|
})
|
|
|
|
// ---- tiers, breaker, derived phases -------------------------------------
|
|
|
|
type Rig = { seen: Seen[]; clock: MockClock; specs: (string | undefined)[] }
|
|
|
|
type Failure = 'rate_limit' | 'overloaded' | 'invalid_request'
|
|
|
|
/** Bottom hooks for a routed run, then the boot; `specs` records spawns. */
|
|
async function bootRig(
|
|
$: Engine,
|
|
on: On,
|
|
model: string = FABLE,
|
|
tokens: number | null = 1000,
|
|
): Promise<Rig> {
|
|
const seen: Seen[] = []
|
|
const specs: (string | undefined)[] = []
|
|
recordSteps(on, seen)
|
|
on('prompt.submit', ($, e) => ({ text: e.text }))
|
|
on('turn.complete', () => ({ text: '' }))
|
|
on('agent.spawn', ($, e) => {
|
|
specs.push(e.model)
|
|
return { model: e.model ?? e.parentModel, agentId: 'a1' }
|
|
})
|
|
on('tool.call', { tool: 'Agent' }, () => ({
|
|
result: {
|
|
status: 'async_launched' as const,
|
|
agentId: 'a1',
|
|
description: 'd',
|
|
prompt: 'p',
|
|
outputFile: '/tmp/o',
|
|
},
|
|
}))
|
|
const clock = await boot($, on, model, tokens)
|
|
return { seen, clock, specs }
|
|
}
|
|
|
|
/** One main step on `model`; returns what reached the bottom hook. */
|
|
async function stepOn($: Engine, rig: Rig, model: string): Promise<Seen> {
|
|
await runStep($, { ...highStep(), model })
|
|
return rig.seen[rig.seen.length - 1] ?? { model: '', effort: undefined }
|
|
}
|
|
|
|
const failWith = ($: Engine, error: Failure, agentId?: string) =>
|
|
$.classic.StopFailure({
|
|
error,
|
|
...(agentId === undefined ? {} : { agent_id: agentId }),
|
|
})
|
|
|
|
const switchTo = (
|
|
$: Engine,
|
|
from: string,
|
|
to: string,
|
|
source: 'command' | 'auto',
|
|
) => $.classic.PostModelSwitch({
|
|
from_model: from,
|
|
to_model: to,
|
|
requested_model: null,
|
|
source,
|
|
context_tokens: 0,
|
|
prompt_cache_warm: false,
|
|
cache_ttl: '5m',
|
|
estimated_cache_write_usd: 0,
|
|
pricing: 'catalog',
|
|
})
|
|
|
|
const planRoute = ($: Engine) =>
|
|
$.tool.call({ tool: ROUTE_TOOL, phase: 'plan' })
|
|
|
|
/** A bottom `session.usage` answering a context of `tokens`. */
|
|
function usageOf(on: On, tokens: number): void {
|
|
on('session.usage', () => ({
|
|
value: {
|
|
startedAt: 0,
|
|
context: { window: 1000000, tokens },
|
|
rateLimits: [],
|
|
},
|
|
}))
|
|
}
|
|
|
|
test('tier: a plan route upgrades a haiku session to fable xhigh', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on, HAIKU)
|
|
await planRoute($)
|
|
expect(await stepOn($, rig, HAIKU)).toEqual({
|
|
model: FABLE,
|
|
effort: 'xhigh',
|
|
})
|
|
})
|
|
|
|
test('tier: show prints the resolved id of each phase and a down line', async (
|
|
$, on) => {
|
|
await bootRig($, on)
|
|
const text = await route($, 'show')
|
|
expect(text).toContain('plan=best→claude-fable-5-1/xhigh')
|
|
expect(text).toContain('judge=big→claude-opus-5-5/xhigh')
|
|
expect(text).toContain('down: none')
|
|
})
|
|
|
|
test('tier: an upgrade is skipped above the context cap', async ($, on) => {
|
|
const rig = await bootRig($, on, SONNET, 300000)
|
|
await planRoute($)
|
|
expect(await stepOn($, rig, SONNET)).toEqual({
|
|
model: SONNET,
|
|
effort: 'xhigh',
|
|
})
|
|
})
|
|
|
|
test('cap: unknown usage blocks the upgrade', async ($, on) => {
|
|
const rig = await bootRig($, on, HAIKU, null)
|
|
await planRoute($)
|
|
expect((await stepOn($, rig, HAIKU)).model).toBe(HAIKU)
|
|
})
|
|
|
|
test('cap: a down model leaves for a better wanted only under the cap', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on, OPUS, 300000)
|
|
await planRoute($)
|
|
await stepOn($, rig, OPUS)
|
|
await failWith($, 'rate_limit')
|
|
expect((await stepOn($, rig, OPUS)).model).toBe(SONNET)
|
|
})
|
|
|
|
test('breaker: model_not_found on a non-table id marks nothing', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await planRoute($)
|
|
await stepOn($, rig, 'claude-zz-9')
|
|
await $.classic.StopFailure({ error: 'model_not_found' })
|
|
expect(await route($, 'show')).toContain('down: none')
|
|
})
|
|
|
|
test('downgrade: mechanical on fable keeps the model, switch off', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await $.tool.call({ tool: ROUTE_TOOL, phase: 'mechanical' })
|
|
expect(await stepOn($, rig, FABLE)).toEqual({ model: FABLE, effort: 'low' })
|
|
})
|
|
|
|
test('downgrade: the switch on moves main to the cheap tier', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await route($, 'switch on')
|
|
await $.tool.call({ tool: ROUTE_TOOL, phase: 'mechanical' })
|
|
const sent = await stepOn($, rig, FABLE)
|
|
expect(sent.model).toBe(HAIKU)
|
|
expect(sent.effort).toBeUndefined()
|
|
})
|
|
|
|
test('fallback: a rate limit on fable moves plan to opus xhigh', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await planRoute($)
|
|
expect((await stepOn($, rig, FABLE)).model).toBe(FABLE)
|
|
await failWith($, 'rate_limit')
|
|
expect(await stepOn($, rig, FABLE)).toEqual({ model: OPUS, effort: 'xhigh' })
|
|
expect(await route($, 'show')).toContain('claude-fable-5-1 for 15 min')
|
|
await route($, 'reload')
|
|
expect((await stepOn($, rig, FABLE)).model).toBe(FABLE)
|
|
})
|
|
|
|
test('breaker: an invalid_request or a turn error never marks down', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await planRoute($)
|
|
await stepOn($, rig, FABLE)
|
|
await failWith($, 'invalid_request')
|
|
await $.turn.complete({
|
|
turnId: 'u1',
|
|
answer: '',
|
|
durationMs: 1,
|
|
isAborted: false,
|
|
reason: 'error',
|
|
})
|
|
expect((await stepOn($, rig, FABLE)).model).toBe(FABLE)
|
|
expect(await route($, 'show')).toContain('down: none')
|
|
})
|
|
|
|
test('breaker: the hold lapses with the clock, a second one doubles', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await planRoute($)
|
|
await stepOn($, rig, FABLE)
|
|
await failWith($, 'overloaded')
|
|
await rig.clock.advance(15 * 60000)
|
|
expect(await route($, 'show')).toContain('down: none')
|
|
await endTurn($)
|
|
await planRoute($)
|
|
expect((await stepOn($, rig, FABLE)).model).toBe(FABLE)
|
|
await failWith($, 'overloaded')
|
|
expect(await route($, 'show')).toContain('claude-fable-5-1 for 30 min')
|
|
})
|
|
|
|
test('breaker: a /model command clears the mark of its target', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await planRoute($)
|
|
await stepOn($, rig, FABLE)
|
|
await failWith($, 'rate_limit')
|
|
expect(await route($, 'show')).toContain('claude-fable-5-1 for')
|
|
await switchTo($, OPUS, FABLE, 'command')
|
|
expect(await route($, 'show')).toContain('down: none')
|
|
expect((await stepOn($, rig, FABLE)).model).toBe(FABLE)
|
|
})
|
|
|
|
test('breaker: model_not_found holds until the reload', async ($, on) => {
|
|
const rig = await bootRig($, on)
|
|
await planRoute($)
|
|
await stepOn($, rig, FABLE)
|
|
await $.classic.StopFailure({ error: 'model_not_found' })
|
|
await rig.clock.advance(24 * 3600000)
|
|
expect(await route($, 'show')).toContain('until reload (model_not_found)')
|
|
})
|
|
|
|
test('engine fallback: an auto switch marks the model it left, once', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await switchTo($, FABLE, OPUS, 'auto')
|
|
const first = await route($, 'show')
|
|
expect(first).toContain('claude-fable-5-1 for 15 min (engine fallback)')
|
|
await planRoute($)
|
|
expect((await stepOn($, rig, OPUS)).model).toBe(OPUS)
|
|
await rig.clock.advance(60000)
|
|
await switchTo($, FABLE, OPUS, 'auto')
|
|
expect(await route($, 'show')).toContain('claude-fable-5-1 for 14 min')
|
|
})
|
|
|
|
test('unknown: a session model absent from the table is never switched', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await planRoute($)
|
|
expect((await stepOn($, rig, 'claude-zz-9')).model).toBe('claude-zz-9')
|
|
})
|
|
|
|
test('spawn: Explore goes to opus while sonnet is down', async ($, on) => {
|
|
const rig = await bootRig($, on)
|
|
await $.agent.spawn(spawnInput())
|
|
await failWith($, 'overloaded', 'a1')
|
|
await $.agent.spawn(spawnInput())
|
|
expect(rig.specs).toEqual([SONNET, OPUS])
|
|
})
|
|
|
|
/** Types a plain prompt, idle, in the composer. */
|
|
const typed = ($: Engine, text: string) => $.prompt.submit({
|
|
text,
|
|
wait: false,
|
|
origin: { kind: 'composer' },
|
|
})
|
|
|
|
const dispatch = ($: Engine) =>
|
|
$.tool.call({ tool: 'Agent', description: 'd', prompt: 'p' })
|
|
|
|
const agentEnds = ($: Engine) => $.turn.complete({
|
|
turnId: 'a1',
|
|
agentId: 'a1',
|
|
answer: '',
|
|
durationMs: 1,
|
|
isAborted: false,
|
|
reason: 'answer',
|
|
})
|
|
|
|
test('derived: a dispatch pushes orchestrate, the end pops the prompt', async (
|
|
$, on) => {
|
|
await bootRig($, on)
|
|
await typed($, 'planifie la migration')
|
|
expect(mainLine(await route($, 'show'))).toContain('prompt plan')
|
|
await dispatch($)
|
|
expect(mainLine(await route($, 'show'))).toContain('derived orchestrate')
|
|
await agentEnds($)
|
|
expect(mainLine(await route($, 'show'))).toContain('prompt plan')
|
|
})
|
|
|
|
test('derived: a route declared after the dispatch survives the pop', async (
|
|
$, on) => {
|
|
await bootRig($, on)
|
|
await typed($, 'planifie la migration')
|
|
await dispatch($)
|
|
await $.tool.call({ tool: ROUTE_TOOL, phase: 'reflect' })
|
|
await agentEnds($)
|
|
expect(mainLine(await route($, 'show'))).toContain('model reflect')
|
|
})
|
|
|
|
test('default rule: planifie sets plan, a later route overrides', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on, HAIKU)
|
|
await typed($, 'planifie la migration')
|
|
expect(await stepOn($, rig, HAIKU)).toEqual({ model: FABLE, effort: 'xhigh' })
|
|
await $.tool.call({ tool: ROUTE_TOOL, phase: 'implement' })
|
|
const line = mainLine(await route($, 'show'))
|
|
expect(line).toContain('model implement')
|
|
expect(line).not.toContain('prompt')
|
|
})
|
|
|
|
test('default rule: a typed slash command gets no default rule', async (
|
|
$, on) => {
|
|
await bootRig($, on)
|
|
await typed($, '/analyze pourquoi ça plante')
|
|
expect(mainLine(await route($, 'show'))).toContain('session defaults')
|
|
})
|
|
|
|
test('default rule: /effort-low pourquoi sets the floor, no default', async (
|
|
$, on) => {
|
|
await bootRig($, on)
|
|
await typed($, '/effort-low pourquoi ça plante')
|
|
await $.skill.prompt({ skill: 'effort-low', text: 'x' })
|
|
const line = mainLine(await route($, 'show'))
|
|
expect(line).toContain('floor low')
|
|
expect(line).not.toContain('reflect')
|
|
})
|
|
|
|
test('per axis: a model-less sticky never hides a turn route tier', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on, HAIKU)
|
|
await route($, 'effort=low')
|
|
await planRoute($)
|
|
expect(await stepOn($, rig, HAIKU)).toEqual({ model: FABLE, effort: 'low' })
|
|
})
|
|
|
|
// ---- texts follow the decision, from the turn's model ------------------
|
|
|
|
test('text: effort-only route on a down model says fallback', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await stepOn($, rig, FABLE)
|
|
await failWith($, 'rate_limit')
|
|
const out = await $.tool.call({ tool: ROUTE_TOOL, effort: 'low' })
|
|
expect(JSON.stringify(out)).toContain(`model ${OPUS} (fallback)`)
|
|
})
|
|
|
|
test('text: /route effort= on a down model names the fallback', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on)
|
|
await stepOn($, rig, FABLE)
|
|
await failWith($, 'rate_limit')
|
|
await route($, 'effort=low')
|
|
expect(mainLine(await route($, 'show'))).toContain(`${OPUS} (fallback)`)
|
|
})
|
|
|
|
test('text: show names the session model after the turn ended', async (
|
|
$, on) => {
|
|
const rig = await bootRig($, on, HAIKU)
|
|
await planRoute($)
|
|
expect((await stepOn($, rig, HAIKU)).model).toBe(FABLE)
|
|
await endTurn($)
|
|
await route($, 'mechanical')
|
|
const line = mainLine(await route($, 'show'))
|
|
expect(line).toContain(`model ${HAIKU}`)
|
|
expect(line).not.toContain(FABLE)
|
|
})
|
|
|
|
test('text: show reflects a /model command to opus', async ($, on) => {
|
|
const rig = await bootRig($, on)
|
|
await stepOn($, rig, FABLE)
|
|
await switchTo($, FABLE, OPUS, 'command')
|
|
await route($, 'mechanical')
|
|
const line = mainLine(await route($, 'show'))
|
|
expect(line).toContain(OPUS)
|
|
expect(line).not.toContain(FABLE)
|
|
})
|
|
|
|
test('text: an unknown session model prints "session model"', async (
|
|
$, on) => {
|
|
await bootRig($, on, 'mystery-model')
|
|
await route($, 'plan')
|
|
expect(mainLine(await route($, 'show'))).toContain('keeps session model')
|
|
})
|
|
|
|
test('text: show names the model a floor upgrade moves to', async (
|
|
$, on) => {
|
|
await bootRig($, on, HAIKU)
|
|
await $.prompt.submit({
|
|
text: 'ultrathink please',
|
|
wait: false,
|
|
origin: { kind: 'composer' },
|
|
})
|
|
const line = mainLine(await route($, 'show'))
|
|
expect(line).toContain(`${FABLE} (upgrade)`)
|
|
})
|