fix(utils.py): fix exception mapping for azure internal server error

This commit is contained in:
Krrish Dholakia 2024-06-06 17:08:05 -07:00
parent 677e0255c8
commit 6e9bca59b0
6 changed files with 6 additions and 3 deletions

View file

@ -9805,8 +9805,7 @@ def exception_type(
elif custom_llm_provider == "azure":
if "Internal server error" in error_str:
exception_mapping_worked = True
raise APIError(
status_code=500,
raise litellm.InternalServerError(
message=f"AzureException Internal server error - {original_exception.message}",
llm_provider="azure",
model=model,