diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index e6a24e16f..3bf36dda8 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -995,6 +995,8 @@ async def test_partner_models_httpx(model, sync_mode): pass except litellm.APIConnectionError as e: pass + except litellm.ServiceUnavailableError as e: + pass except Exception as e: if "429 Quota exceeded" in str(e): pass diff --git a/tests/local_testing/test_text_completion.py b/tests/local_testing/test_text_completion.py index a9c15a196..315517d02 100644 --- a/tests/local_testing/test_text_completion.py +++ b/tests/local_testing/test_text_completion.py @@ -4193,6 +4193,9 @@ async def test_completion_codestral_fim_api_stream(model): except litellm.APIConnectionError as e: print(e) pass + except litellm.ServiceUnavailableError as e: + print(e) + pass except Exception as e: pytest.fail(f"Error occurred: {e}")