diff --git a/tests/unit/test_main.py b/tests/unit/test_main.py
index f79dcd695a4..15592fe95ab 100644
--- a/tests/unit/test_main.py
+++ b/tests/unit/test_main.py
@@ -1526,6 +1526,8 @@ def test_responses_api_bridge_check_databricks_gpt_5_4_plus_tools_route_to_respo
         pytest.param("databricks-gpt-5-6-sol", _DATABRICKS_FUNCTION_TOOL, "none", id="explicit-none"),
         pytest.param("databricks-gpt-5-6-sol", None, None, id="no-tools"),
         pytest.param("databricks-gpt-5-2", _DATABRICKS_FUNCTION_TOOL, None, id="pre-5.4"),
+        pytest.param("databricks-gpt-5", _DATABRICKS_FUNCTION_TOOL, None, id="major-only-pre-5.4"),
+        pytest.param("databricks-gpt-5-6-search-api", _DATABRICKS_FUNCTION_TOOL, None, id="search-model"),
         pytest.param("databricks-gpt-oss-120b", _DATABRICKS_FUNCTION_TOOL, None, id="gpt-oss"),
         pytest.param("databricks-claude-sonnet-4-5", _DATABRICKS_FUNCTION_TOOL, None, id="non-gpt"),
     ],
@@ -1550,6 +1552,9 @@ def test_responses_api_bridge_check_databricks_without_rejected_tools_stays_chat
         ("databricks-gpt-5-6-sol", "gpt-5.6-sol"),
         ("databricks/databricks-gpt-5-4", "gpt-5.4"),
         ("databricks-gpt-5-4-mini", "gpt-5.4-mini"),
+        ("databricks-gpt-5", "gpt-5"),
+        ("databricks-gpt-5-mini", "gpt-5-mini"),
+        ("databricks-gpt-6", "gpt-6"),
         ("databricks-gpt-oss-120b", "databricks-gpt-oss-120b"),
         ("databricks-claude-sonnet-4-5", "databricks-claude-sonnet-4-5"),
     ],