test(proxy): cursor variant matrix covers ultra and max levels
Some checks failed
LiteLLM Rust / rustfmt, clippy, test (push) Has been cancelled

This commit is contained in:
Tin Chi Lo 2026-08-21 14:10:36 -07:00
parent 1f8b64f51f
commit 9f14c228bf
2 changed files with 12 additions and 7 deletions

View file

@ -1352,7 +1352,9 @@ class TestParseCursorModelVariant:
("gemini-3.0-pro-thinking-low", "gemini-3.0-pro", "low"),
("claude-opus-5-fast", "claude-opus-5", None),
("gpt-5.6-sol", "gpt-5.6-sol", None),
("foo-thinking-ultra-fast", "foo-thinking-ultra", None),
("gpt-5.6-thinking-ultra-fast", "gpt-5.6", "ultra"),
("gpt-5.6-thinking-max", "gpt-5.6", "max"),
("foo-thinking-mega-fast", "foo-thinking-mega", None),
("-thinking-high", "-thinking-high", None),
],
)

View file

@ -19,12 +19,15 @@ interface AvailableModel {
supported_reasoning_efforts?: string[] | null;
}
const toModelGroup = (item: AvailableModel): ModelGroup => ({
model_group: item.model_group || item.id || item.model_name || "",
...(item.mode && { mode: item.mode }),
...(item.supports_reasoning === true && { supports_reasoning: true }),
...(item.supported_reasoning_efforts && { supported_reasoning_efforts: item.supported_reasoning_efforts }),
});
const toModelGroup = (item: AvailableModel): ModelGroup => {
const groupName = (item.model_group || item.id || item.model_name) ?? "";
return {
model_group: groupName,
...(item.mode && { mode: item.mode }),
...(item.supports_reasoning === true && { supports_reasoning: true }),
...(item.supported_reasoning_efforts && { supported_reasoning_efforts: item.supported_reasoning_efforts }),
};
};
/**
* /models carries no capability metadata, so the team-allowed names are joined against