phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(utils.py): watsonx ai exception mapping fix

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-13 17:11:33 -07:00
parent af9489bbfd
commit d7c28509d7
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/utils.py
View file

@ -8513,6 +8513,15 @@ def exception_type(
model=model,
request=original_exception.request,
)
elif custom_llm_provider == "watsonx":
if "token_quota_reached" in error_response:
exception_mapping_worked = True
raise RateLimitError(
message=f"WatsonxException: Rate Limit Errror - {error_str}",
llm_provider="watsonx",
model=model,
response=original_exception.response,
)
elif custom_llm_provider == "bedrock":
if (
"too many tokens" in error_str