scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

litellm_fix(test): add acancel_batch to Azure SDK client initialization test (#20143)

Browse source
This commit is contained in:
shin-bot-litellm 2026-01-31 07:34:54 -08:00 • committed by GitHub
parent 1c757dee14
commit 395ad9bdc1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/llms/azure/test_azure_common_utils.py
View file

@ -483,6 +483,7 @@ async def test_ensure_initialize_azure_sdk_client_always_used(call_type):
"input_file_id": "123",
},
"aretrieve_batch": {"batch_id": "123"},
"acancel_batch": {"batch_id": "123"},
"aget_assistants": {"custom_llm_provider": "azure"},
"acreate_assistants": {"custom_llm_provider": "azure"},
"adelete_assistant": {"custom_llm_provider": "azure", "assistant_id": "123"},
@ -537,7 +538,7 @@ async def test_ensure_initialize_azure_sdk_client_always_used(call_type):
patch_target = (
"litellm.rerank_api.main.azure_rerank.initialize_azure_sdk_client"
)
elif call_type == CallTypes.acreate_batch or call_type == CallTypes.aretrieve_batch:
elif call_type == CallTypes.acreate_batch or call_type == CallTypes.aretrieve_batch or call_type == CallTypes.acancel_batch:
patch_target = (
"litellm.batches.main.azure_batches_instance.initialize_azure_sdk_client"
)