refactor: use a single agent model path

2026-07-26 18:50:51 +08:00
parent 2e8e81c790
commit d95d06233f
23 changed files with 330 additions and 454 deletions
@@ -1,23 +1,27 @@
 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)
+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)