diff --git a/tests/local_testing/test_text_completion.py b/tests/local_testing/test_text_completion.py index 315517d02..5d94820dc 100644 --- a/tests/local_testing/test_text_completion.py +++ b/tests/local_testing/test_text_completion.py @@ -4145,6 +4145,9 @@ async def test_completion_codestral_fim_api(model): # cost = litellm.completion_cost(completion_response=response) # print("cost to make mistral completion=", cost) # assert cost > 0.0 + except litellm.ServiceUnavailableError: + print("got ServiceUnavailableError") + pass except Exception as e: pytest.fail(f"Error occurred: {e}")