vrr/codeburn
2
0
Fork 0
mirror of https://github.com/AgentSeal/codeburn.git synced 2026-08-22 15:05:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix(models): alias bare MiMo 2.5 ids to the LiteLLM Xiaomi rows

Browse source 
Hermes and token-plan sessions store mimo-v2.5-pro. The snapshot
row is xiaomi/mimo-v2.5-pro. Same class as the existing
mimo-v2-flash alias. No invented rate.

Looking up the display name on the stripped leaf before following a
pricing alias, so cline-pass/mimo-v2.5-pro cannot recurse
strip → alias → last-segment forever.
This commit is contained in:
Aditya Vikram Singh 2026-08-19 20:44:18 +05:30
parent bb1974dab8
commit d3f86f5d16
2 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/models.ts
View file

@ -297,6 +297,10 @@ const BUILTIN_ALIASES: Record<string, string> = {
'k3-agent': 'kimi-k3',
'k2d6-agent': 'kimi-k2p6',
'mimo-v2-flash': 'xiaomi/mimo-v2-flash',
// Hermes / Xiaomi token-plan sessions store the bare id. LiteLLM's row is
// namespaced. Same class as mimo-v2-flash above — do not invent a rate.
'mimo-v2.5-pro': 'xiaomi/mimo-v2.5-pro',
'mimo-v2.5': 'xiaomi/mimo-v2.5',
'kat-coder-pro-v1': 'kwaipilot/kat-coder-pro',
// Cursor emits dot-version tier-last names plus tier/reasoning suffixes
// that LiteLLM does not index (`-high`, `-low`, `-medium`, `-thinking`,
@ -983,13 +987,17 @@ function deriveClaudeShortName(canonical: string): string | undefined {
export function getShortModelName(model: string): string {
if (autoModelNames[model]) return autoModelNames[model]
const canonical = resolveAlias(getCanonicalName(model))
const stripped = getCanonicalName(model)
// Pricing aliases may re-namespace a leaf (mimo-v2.5-pro → xiaomi/…).
// Display names live on the leaf. Look that up before following the alias
// or we recurse forever: strip → alias → last-segment → strip.
for (const [key, name] of SORTED_SHORT_NAMES) {
if (stripped === key || stripped.startsWith(key + '-')) return name
}
const canonical = resolveAlias(stripped)
const claude = deriveClaudeShortName(canonical)
if (claude) return claude
for (const [key, name] of SORTED_SHORT_NAMES) {
// Match on a version boundary, not a bare prefix: an unlisted future minor
// (e.g. gpt-5.6) must NOT collapse into the base "gpt-5" entry — it should
// fall through to its raw id rather than show a wrong name/tier.
if (canonical === key || canonical.startsWith(key + '-')) return name
}
// getCanonicalName only strips the leading provider prefix, so a raw

2
tests/models.test.ts
View file

@ -737,6 +737,8 @@ describe('provider pricing suffix variants', () => {
describe('observed provider model aliases', () => {
const cases: Array<[string, string]> = [
['MiMo-V2-Flash', 'xiaomi/mimo-v2-flash'],
['mimo-v2.5-pro', 'xiaomi/mimo-v2.5-pro'],
['MiMo-v2.5-Pro', 'xiaomi/mimo-v2.5-pro'],
['KAT-Coder-Pro-V1', 'kwaipilot/kat-coder-pro'],
// Kimi Code wires report bare `k3` in llm.request.model; it must price
// through the kimi-k3 table entry, not fall through to $0.