From 7646c91dff6cc292dd75a08a98a299050b6ee0aa Mon Sep 17 00:00:00 2001 From: Rod Boev Date: Thu, 16 Jul 2026 22:07:15 -0400 Subject: [PATCH] fix(nv_build): cover reported model metadata Signed-off-by: Rod Boev --- .../providers/nv_build/model_registry.yaml | 9 +++++++++ tests/unit/test_providers.py | 19 +++++++++++++++++++ 2 files changed, 28 insertions(+) diff --git a/src/skillspector/providers/nv_build/model_registry.yaml b/src/skillspector/providers/nv_build/model_registry.yaml index aeba04e1..226fddd9 100644 --- a/src/skillspector/providers/nv_build/model_registry.yaml +++ b/src/skillspector/providers/nv_build/model_registry.yaml @@ -15,6 +15,15 @@ models: # NVIDIA-curated NIMs on build.nvidia.com. + "z-ai/glm-5.2": + context_length: 1000000 + + "z-ai/glm-5.1": + context_length: 205000 + + "moonshotai/kimi-k2.6": + context_length: 256000 + "deepseek-ai/deepseek-v4-pro": context_length: 1000000 max_output_tokens: 128000 diff --git a/tests/unit/test_providers.py b/tests/unit/test_providers.py index fae2572f..4ca7c09f 100644 --- a/tests/unit/test_providers.py +++ b/tests/unit/test_providers.py @@ -127,6 +127,25 @@ def _clean_provider_env(monkeypatch: pytest.MonkeyPatch): class TestNvBuildProvider: """build.nvidia.com provider — credentials + bundled YAML metadata.""" + @pytest.mark.parametrize( + ("model", "context_length"), + [ + ("z-ai/glm-5.2", 1_000_000), + ("z-ai/glm-5.1", 205_000), + ("moonshotai/kimi-k2.6", 256_000), + ], + ) + def test_nv_build_reported_model_metadata(self, model: str, context_length: int) -> None: + provider = NvBuildProvider() + assert provider.get_context_length(model) == context_length + assert provider.get_max_output_tokens(model) is None + + @pytest.mark.parametrize("model", ["glm-5.2", "z-ai/glm-5.2 "]) + def test_nv_build_model_near_match_stays_unresolved(self, model: str) -> None: + provider = NvBuildProvider() + assert provider.get_context_length(model) is None + assert provider.get_max_output_tokens(model) is None + def test_returns_none_without_env_var(self) -> None: assert NvBuildProvider().resolve_credentials() is None