62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
import importlib.util
|
|
from pathlib import Path
|
|
|
|
|
|
SCRIPT_PATH = Path(__file__).resolve().parents[1] / "scripts" / "sync_zen_models.py"
|
|
SPEC = importlib.util.spec_from_file_location("sync_zen_models", SCRIPT_PATH)
|
|
assert SPEC is not None and SPEC.loader is not None
|
|
MODULE = importlib.util.module_from_spec(SPEC)
|
|
SPEC.loader.exec_module(MODULE)
|
|
|
|
|
|
def test_convert_model_applies_family_defaults_for_gpt_5_4():
|
|
converted = MODULE.convert_model({"id": "gpt-5.4"}, curated_models={})
|
|
|
|
assert converted is not None
|
|
assert converted["model_name"] == "gpt-5.4"
|
|
assert converted["context_window"] == 1050000
|
|
assert converted["max_output_tokens"] == 128000
|
|
assert converted["supports_extended_thinking"] is True
|
|
assert converted["supports_images"] is True
|
|
assert converted["use_openai_response_api"] is True
|
|
|
|
|
|
def test_convert_model_prefers_curated_metadata_when_available():
|
|
converted = MODULE.convert_model(
|
|
{"id": "claude-opus-4-5"},
|
|
curated_models={
|
|
"claude-opus-4-5": {
|
|
"context_window": 200000,
|
|
"max_output_tokens": 64000,
|
|
"supports_extended_thinking": False,
|
|
"description": "Curated Opus 4.5",
|
|
"intelligence_score": 18,
|
|
"allow_code_generation": True,
|
|
}
|
|
},
|
|
)
|
|
|
|
assert converted is not None
|
|
assert converted["model_name"] == "claude-opus-4-5"
|
|
assert converted["context_window"] == 200000
|
|
assert converted["max_output_tokens"] == 64000
|
|
assert converted["supports_extended_thinking"] is False
|
|
assert converted["description"] == "Curated Opus 4.5"
|
|
assert converted["allow_code_generation"] is True
|
|
|
|
|
|
def test_build_output_document_sorts_model_ids():
|
|
document = MODULE.build_output_document(
|
|
{
|
|
"data": [
|
|
{"id": "gpt-5.4-pro"},
|
|
{"id": "claude-opus-4-6"},
|
|
]
|
|
},
|
|
"https://opencode.ai/zen/v1/models",
|
|
curated_models={},
|
|
)
|
|
|
|
model_names = [item["model_name"] for item in document["models"]]
|
|
assert model_names == sorted(model_names)
|