diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index 6ae7c27b758..7bec649099a 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -944,6 +944,8 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
                 "code_interpreter_cost_per_session": {"type": "number"},
                 "inference_geo": {"type": "string"},
                 "litellm_provider": {"type": "string"},
+                "max_audio_length_hours": {"type": "number"},
+                "max_audio_per_prompt": {"type": "number"},
                 "max_input_tokens": {"type": "number"},
                 "max_output_tokens": {"type": "number"},
                 "max_tokens": {"type": "number"},