From 5d59fa166c5bb359319268a929365e6edc7fb3ad Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 12 Feb 2024 09:11:21 -0800 Subject: [PATCH] test: api error handling --- litellm/tests/test_completion.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 1d0a74ff3ed..4dcd1119b27 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -1356,6 +1356,8 @@ def test_completion_together_ai_mixtral(): ) except litellm.Timeout as e: pass + except litellm.ServiceUnavailableError as e: + pass except Exception as e: pytest.fail(f"Error occurred: {e}")