From 50fe6a639dea6be85fe2c317d56e38684943f434 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 15 Nov 2024 00:13:22 +0530 Subject: [PATCH] test: handle internal server error --- tests/local_testing/test_streaming.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/local_testing/test_streaming.py b/tests/local_testing/test_streaming.py index ee7fde9d0..bc4827d92 100644 --- a/tests/local_testing/test_streaming.py +++ b/tests/local_testing/test_streaming.py @@ -3333,6 +3333,8 @@ async def test_acompletion_claude_3_function_call_with_streaming(model): validate_final_streaming_function_calling_chunk(chunk=chunk) idx += 1 # raise Exception("it worked! ") + except litellm.InternalServerError as e: + pytest.skip(f"InternalServerError - {str(e)}") except Exception as e: pytest.fail(f"Error occurred: {e}")