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

fix(utils.py): fix watsonx exception mapping

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-13 18:13:13 -07:00
parent 3694b5e7c0
commit 29449aa5c1
2 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_completion.py
View file

@ -3342,6 +3342,8 @@ def test_completion_watsonx():
print(response) print(response)
except litellm.APIError as e: except litellm.APIError as e:
pass pass
except litellm.RateLimitError as e:
pass
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")

2
litellm/utils.py
View file

@ -8514,7 +8514,7 @@ def exception_type(
request=original_exception.request, request=original_exception.request,
) )
elif custom_llm_provider == "watsonx": elif custom_llm_provider == "watsonx":
if "token_quota_reached" in error_response: if "token_quota_reached" in error_str:
exception_mapping_worked = True exception_mapping_worked = True
raise RateLimitError( raise RateLimitError(
message=f"WatsonxException: Rate Limit Errror - {error_str}", message=f"WatsonxException: Rate Limit Errror - {error_str}",