diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index 906ab424c3..e073af0eed 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -576,6 +576,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                         "rerank",
                         "responses",
                         "ocr",
+                        "search",
                     ],
                 },
                 "output_cost_per_audio_token": {"type": "number"},
@@ -682,8 +683,14 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                             "output_cost_per_token": {"type": "number"},
                             "cache_read_input_token_cost": {"type": "number"},
                             "output_cost_per_reasoning_token": {"type": "number"},
+                            "max_results_range": {
+                                "type": "array",
+                                "items": {"type": "number"},
+                                "minItems": 2,
+                                "maxItems": 2,
+                            },
+                            "input_cost_per_query": {"type": "number"},
                         },
-                        "required": ["range"],
                         "additionalProperties": False,
                     },
                 },