From 2995f3adc299956ab9c4093ade43bf6f9b0c43dc Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 15 Nov 2024 00:43:56 +0530 Subject: [PATCH] test: handle anthropic overloaded error --- tests/local_testing/test_completion.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index 5b9a534fa..5685231fc 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -434,6 +434,8 @@ def test_completion_claude_3(): ) # Add any assertions, here to check response args print(response) + except litellm.InternalServerError as e: + pytest.skip(f"InternalServerError - {str(e)}") except Exception as e: pytest.fail(f"Error occurred: {e}")