diff --git a/tests/local_testing/test_anthropic_prompt_caching.py b/tests/local_testing/test_anthropic_prompt_caching.py index df1eed38120..8017a158244 100644 --- a/tests/local_testing/test_anthropic_prompt_caching.py +++ b/tests/local_testing/test_anthropic_prompt_caching.py @@ -235,7 +235,7 @@ async def test_anthropic_vertex_ai_prompt_caching(anthropic_messages, sync_mode) print(mock_post.call_args.kwargs["headers"]) assert "anthropic-beta" not in mock_post.call_args.kwargs["headers"] - +@pytest.mark.flaky(retries=3, delay=1) @pytest.mark.asyncio() async def test_anthropic_api_prompt_caching_basic(): litellm.set_verbose = True