From 10ef40faae0cbe6794d30b64518c49bbfd2fc36b Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 5 Mar 2024 19:01:40 -0800 Subject: [PATCH] test(test_completion.py): handle anthropic internal server error --- 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 f5e145769..eca0cd934 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -151,6 +151,8 @@ def test_completion_claude_3_function_call(): assert isinstance( response.choices[0].message.tool_calls[0].function.arguments, str ) + except litellm.ServiceUnavailableError as e: + pass except Exception as e: pytest.fail(f"Error occurred: {e}")