wuyang
28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
from agent_platform.models import MODEL_SPECS, get_model_spec, openai_model_list
|
|
|
|
|
|
def test_fixed_public_model_catalog_separates_model_and_thinking() -> None:
|
|
assert set(MODEL_SPECS) == {"luna", "terra", "sol", "deepseek-v4-pro"}
|
|
assert get_model_spec("luna").provider_model == "ChatGPT-5.6:Luna"
|
|
assert get_model_spec("terra").provider_model == "ChatGPT-5.6:Terra"
|
|
assert get_model_spec("sol").provider_model == "ChatGPT-5.6:Sol"
|
|
|
|
for model_id in ("luna", "terra", "sol"):
|
|
spec = get_model_spec(model_id)
|
|
assert spec.thinking_enabled is True
|
|
assert spec.thinking_adjustable is False
|
|
assert spec.thinking_label == "开启（不分档）"
|
|
assert spec.reasoning_effort is None
|
|
|
|
deepseek = get_model_spec("deepseek-v4-pro")
|
|
assert deepseek.provider == "deepseek"
|
|
assert deepseek.provider_model == "deepseek-v4-pro"
|
|
assert deepseek.thinking_enabled is True
|
|
assert deepseek.thinking_adjustable is False
|
|
assert deepseek.thinking_label == "极高"
|
|
assert deepseek.reasoning_effort == "max"
|
|
|
|
public_models = openai_model_list()["data"]
|
|
assert {item["id"] for item in public_models} == set(MODEL_SPECS)
|
|
assert all("k1412" in item for item in public_models)
|