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

fix(utils.py): fix azure exception mapping

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-01 19:05:20 -08:00
parent 241f0aad5e
commit 245ec2430e
2 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/utils.py
View file

@ -6964,6 +6964,21 @@ def exception_type(
llm_provider="azure",
response=original_exception.response,
)
elif original_exception.status_code == 503:
exception_mapping_worked = True
raise ServiceUnavailableError(
message=f"AzureException - {original_exception.message}",
model=model,
llm_provider="azure",
response=original_exception.response,
)
elif original_exception.status_code == 504: # gateway timeout error
exception_mapping_worked = True
raise Timeout(
message=f"AzureException - {original_exception.message}",
model=model,
llm_provider="azure",
)
else:
exception_mapping_worked = True
raise APIError(