mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-29 06:33:28 +00:00
test(realtime): await the now async azure health-check auth header helper
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
c008f25b55
commit
4848674037
1 changed files with 2 additions and 2 deletions
|
|
@ -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}),
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue