diff --git a/tests/unit/litellm_core_utils/test_health_check_helpers.py b/tests/unit/litellm_core_utils/test_health_check_helpers.py index 4f62c6fb096..c23749f7a20 100644 --- a/tests/unit/litellm_core_utils/test_health_check_helpers.py +++ b/tests/unit/litellm_core_utils/test_health_check_helpers.py @@ -534,12 +534,12 @@ async def test_ocr_health_check_sends_the_document_kind_the_provider_config_acce assert document[expected_document_type].startswith(expected_uri_prefix) -def test_realtime_health_check_azure_ad_params_drop_reserved_keys(): +async def test_realtime_health_check_azure_ad_params_drop_reserved_keys(): from litellm.realtime_api import main as realtime_main seen = [] with patch.object(realtime_main, "get_azure_ad_token", lambda params: seen.append(params) or "ad-token"): - headers = realtime_main._realtime_health_check_auth_headers( + headers = await realtime_main._realtime_health_check_auth_headers( "azure", None, MappingProxyType({"api_base": "https://x.openai.azure.com", "self": 1, "params": 2, "__class__": 3}),