From 6b1d2551d1016ee6f993877bc9c2897bd50dfa65 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 26 Mar 2024 13:53:18 -0700 Subject: [PATCH] test(test_batch_completions.py): handle overloaded anthropic error --- litellm/tests/test_batch_completions.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_batch_completions.py b/litellm/tests/test_batch_completions.py index 485f51757b9..cd5eb251e85 100644 --- a/litellm/tests/test_batch_completions.py +++ b/litellm/tests/test_batch_completions.py @@ -74,6 +74,8 @@ def test_batch_completion_models_all_responses(): assert len(responses) == 2 except Timeout as e: pass + except litellm.APIError as e: + pass except Exception as e: pytest.fail(f"An error occurred: {e}")