From 0c47042df3230fe54f7a4f3f284efd2bebb5837f Mon Sep 17 00:00:00 2001
From: Vigilans <vigilans@foxmail.com>
Date: Mon, 20 Apr 2026 15:43:27 +0800
Subject: [PATCH] test(model-registry): verify gpt-5.4 supports_web_search in
 model_info

---
 .../test_gpt54_web_search_model_info.py       | 36 +++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 tests/test_litellm/llms/openai/test_gpt54_web_search_model_info.py

diff --git a/tests/test_litellm/llms/openai/test_gpt54_web_search_model_info.py b/tests/test_litellm/llms/openai/test_gpt54_web_search_model_info.py
new file mode 100644
index 00000000000..8d3040043d7
--- /dev/null
+++ b/tests/test_litellm/llms/openai/test_gpt54_web_search_model_info.py
@@ -0,0 +1,36 @@
+"""Verify gpt-5.4 models expose supports_web_search in model_info."""
+
+import pytest
+
+import litellm
+from litellm.litellm_core_utils.get_model_cost_map import get_model_cost_map
+from litellm.utils import get_model_info
+
+
+@pytest.fixture(autouse=True)
+def use_local_model_cost_map(monkeypatch: pytest.MonkeyPatch):
+    monkeypatch.setenv("LITELLM_LOCAL_MODEL_COST_MAP", "True")
+    monkeypatch.setattr(
+        litellm, "model_cost", get_model_cost_map(url=litellm.model_cost_map_url)
+    )
+    litellm.add_known_models(model_cost_map=litellm.model_cost)
+
+
+class TestGPT54WebSearch:
+    """gpt-5.4 variants must have supports_web_search=True."""
+
+    @pytest.mark.parametrize(
+        "model",
+        [
+            "gpt-5.4",
+            "gpt-5.4-2026-03-05",
+            "azure/gpt-5.4",
+            "azure/gpt-5.4-2026-03-05",
+        ],
+    )
+    def test_supports_web_search(self, model: str):
+        info = get_model_info(model)
+        assert info is not None, f"get_model_info returned None for {model}"
+        assert (
+            info["supports_web_search"] is True
+        ), f"{model} should have supports_web_search=True"