test: fix watsonx api error

This commit is contained in:
Krrish Dholakia
2024-05-13 19:01:19 -07:00
parent 724d880a45
commit 071a70c5fc
+2
View File
@@ -3363,6 +3363,8 @@ def test_completion_stream_watsonx():
print(chunk)
except litellm.APIError as e:
pass
except litellm.RateLimitError as e:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")