test: handle predibase api failures

This commit is contained in:
Krrish Dholakia 2024-07-31 19:39:58 -07:00
parent e1696bf426
commit 83f638100e

View file

@ -273,6 +273,8 @@ async def test_completion_predibase():
print(response)
except litellm.Timeout as e:
pass
except litellm.ServiceUnavailableError as e:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")