From 8bf0f57b5953d769ea5ff7f3776ed934ba39b33a Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 14 Nov 2024 18:13:56 +0530 Subject: [PATCH] test: handle service unavailable error --- tests/local_testing/test_streaming.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/local_testing/test_streaming.py b/tests/local_testing/test_streaming.py index cc220e46e..db84234c7 100644 --- a/tests/local_testing/test_streaming.py +++ b/tests/local_testing/test_streaming.py @@ -3333,7 +3333,7 @@ async def test_acompletion_function_call_with_streaming(model): validate_final_streaming_function_calling_chunk(chunk=chunk) idx += 1 # raise Exception("it worked! ") - except litellm.InternalServerError: + except litellm.InternalServerError or litellm.ServiceUnavailableError: pass except Exception as e: pytest.fail(f"Error occurred: {e}")