wuyang
17 lines
615 B
Python
17 lines
615 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",
|
|
"work-light",
|
|
"work-medium",
|
|
"work-high",
|
|
}
|
|
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"
|
|
assert {item["id"] for item in openai_model_list()["data"]} == set(MODEL_SPECS)
|