fix(utils.py): fix watsonx exception mapping

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

View file

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