diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py index 41c4bd0b3b5..8bc72941423 100644 --- a/tests/test_litellm/test_utils.py +++ b/tests/test_litellm/test_utils.py @@ -856,7 +856,12 @@ def test_aaamodel_prices_and_context_window_json_is_valid(): "supports_sampling_params": {"type": "boolean"}, "supports_service_tier": {"type": "boolean"}, "supports_preset": {"type": "boolean"}, + "supports_output_config": {"type": "boolean"}, "tool_use_system_prompt_tokens": {"type": "number"}, + "bedrock_output_config_effort_ceiling": { + "type": "string", + "enum": ["low", "medium", "high", "max", "xhigh"], + }, "tpm": {"type": "number"}, "provider_specific_entry": {"type": "object"}, "supported_endpoints": {