diff --git a/tests/test_litellm/llms/anthropic/test_anthropic_files_and_batches.py b/tests/test_litellm/llms/anthropic/test_anthropic_files_and_batches.py index 47571220175..0755c189afc 100644 --- a/tests/test_litellm/llms/anthropic/test_anthropic_files_and_batches.py +++ b/tests/test_litellm/llms/anthropic/test_anthropic_files_and_batches.py @@ -336,7 +336,7 @@ class TestAnthropicFilesHandler: "extra_body": None } - with patch.object(handler.anthropic_model_info, "get_api_key", return_value=None): + with patch.object(handler.anthropic_model_info, "get_auth_header", return_value=None): with pytest.raises(ValueError, match="Missing Anthropic API Key"): await handler.afile_content( file_content_request=file_content_request,