From 4ec92ba924eb2c7311cce077bb37ede22fddd705 Mon Sep 17 00:00:00 2001 From: Julio Quinteros Pro Date: Wed, 4 Mar 2026 10:34:19 -0300 Subject: [PATCH] fix: add new model_prices properties to validation schema Add cache_read_input_token_cost_per_audio_token, supports_code_execution, and supports_file_search to the JSON schema used by the model prices validation test. Co-Authored-By: Claude Opus 4.6 --- tests/test_litellm/test_utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py index 4a92c7e7fd..92a7b8096e 100644 --- a/tests/test_litellm/test_utils.py +++ b/tests/test_litellm/test_utils.py @@ -605,6 +605,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid(): "cache_read_input_token_cost_above_200k_tokens": {"type": "number"}, "cache_creation_input_token_cost_above_1hr_above_200k_tokens": {"type": "number"}, "cache_read_input_audio_token_cost": {"type": "number"}, + "cache_read_input_token_cost_per_audio_token": {"type": "number"}, "cache_read_input_image_token_cost": {"type": "number"}, "deprecation_date": {"type": "string"}, "input_cost_per_audio_per_second": {"type": "number"}, @@ -715,6 +716,8 @@ def test_aaamodel_prices_and_context_window_json_is_valid(): "supports_audio_input": {"type": "boolean"}, "supports_audio_output": {"type": "boolean"}, "supports_embedding_image_input": {"type": "boolean"}, + "supports_code_execution": {"type": "boolean"}, + "supports_file_search": {"type": "boolean"}, "supports_function_calling": {"type": "boolean"}, "supports_image_input": {"type": "boolean"}, "supports_parallel_function_calling": {"type": "boolean"},