From 74b8a1dbdf627783acde48c20610052ee6de7c6c Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Thu, 23 Oct 2025 08:47:08 -0700 Subject: [PATCH] test_aaamodel_prices_and_context_window_json_is_valid --- tests/test_litellm/test_utils.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) 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, }, },