From 735a5bfc5b8947b53d4a872072cc78449fcb8b5a Mon Sep 17 00:00:00 2001
From: Mateo Wang <277851410+mateo-berri@users.noreply.github.com>
Date: Fri, 10 Jul 2026 21:10:36 -0700
Subject: [PATCH] test(models): assert capability fields on regional Azure
 gpt-5.6 entries (#32875)

---
 tests/test_litellm/test_gpt_5_6_model_metadata.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/test_litellm/test_gpt_5_6_model_metadata.py b/tests/test_litellm/test_gpt_5_6_model_metadata.py
index af9bb117f78..5a7b621d521 100644
--- a/tests/test_litellm/test_gpt_5_6_model_metadata.py
+++ b/tests/test_litellm/test_gpt_5_6_model_metadata.py
@@ -122,6 +122,8 @@ def test_azure_gpt_5_6_regional_model_info(model):
     assert info is not None, f"{model} not found in model_prices_and_context_window.json"
 
     assert info["litellm_provider"] == "azure"
+    assert info["mode"] == "chat"
+
     input_cost, output_cost, cache_read_cost, _ = STANDARD_PRICING[_tier_key(model)]
 
     assert info["input_cost_per_token"] == pytest.approx(input_cost * 1.1)
@@ -132,6 +134,13 @@ def test_azure_gpt_5_6_regional_model_info(model):
     assert info["input_cost_per_token_priority"] == pytest.approx(input_cost * 2.75)
     assert info["output_cost_per_token_priority"] == pytest.approx(output_cost * 2.75)
 
+    assert info["max_input_tokens"] == 1050000
+    assert info["max_output_tokens"] == 128000
+    assert info["supports_reasoning"] is True
+
+    _, provider, _, _ = get_llm_provider(model=model)
+    assert provider == "azure"
+
 
 def test_gpt_5_6_backup_matches_main():
     """Ensure the bundled model cost map stays in sync with the canonical file."""