wuyang
24 lines
894 B
Python
24 lines
894 B
Python
from agent_platform.models import MODEL_SPECS, get_model_spec, openai_model_list
|
|
|
|
|
|
def test_fixed_public_model_matrix() -> None:
|
|
assert set(MODEL_SPECS) == {
|
|
"chat-light",
|
|
"chat-medium",
|
|
"chat-high",
|
|
"chat-extreme",
|
|
"work-light",
|
|
"work-medium",
|
|
"work-high",
|
|
"work-extreme",
|
|
}
|
|
assert get_model_spec("chat-light").provider_model == "ChatGPT-5.6:Luna"
|
|
assert get_model_spec("work-medium").provider_model == "ChatGPT-5.6:Terra"
|
|
assert get_model_spec("work-high").provider_model == "ChatGPT-5.6:Sol"
|
|
extreme = get_model_spec("work-extreme")
|
|
assert extreme.provider == "deepseek"
|
|
assert extreme.provider_model == "deepseek-v4-pro"
|
|
assert extreme.thinking_enabled is True
|
|
assert extreme.reasoning_effort == "max"
|
|
assert {item["id"] for item in openai_model_list()["data"]} == set(MODEL_SPECS)
|