phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Union[TranscriptionResponse, Coroutine[Any, Any, TranscriptionResponse]]:

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-18 14:23:14 -07:00
parent 7384d45ef0
commit 55ea2370ba
2 changed files with 16 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/litellm/llms/azure/test_azure_common_utils.py
View file

@ -308,12 +308,10 @@ async def test_ensure_initialize_azure_sdk_client_always_used(call_type):
# Get appropriate input for this call type
input_kwarg = test_inputs.get(call_type.value, {})
patch_target = "litellm.main.azure_chat_completions.initialize_azure_sdk_client"
if call_type == CallTypes.atranscription:
patch_target = (
"litellm.main.azure_audio_transcriptions.initialize_azure_sdk_client"
)
elif call_type == CallTypes.arerank:
patch_target = (
"litellm.llms.azure.common_utils.BaseAzureLLM.initialize_azure_sdk_client"
)
if call_type == CallTypes.arerank:
patch_target = (
"litellm.rerank_api.main.azure_rerank.initialize_azure_sdk_client"
)