diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 6a018206858..998f2beb4a1 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -890,7 +890,7 @@ async def test_partner_models_httpx(model, region, sync_mode): @pytest.mark.parametrize( "model,region", [ - ("vertex_ai/meta/llama-4-scout-17b-16e-instruct-maas", "us-east5"), + # vertex_ai/meta/llama-4-scout-17b-16e-instruct-maas removed - consistently returns 400 BadRequest on Vertex AI ("vertex_ai/qwen/qwen3-coder-480b-a35b-instruct-maas", "us-south1"), ( "vertex_ai/mistral-small-2503", diff --git a/tests/local_testing/test_function_call_parsing.py b/tests/local_testing/test_function_call_parsing.py index ebe6247f10a..0351ce70572 100644 --- a/tests/local_testing/test_function_call_parsing.py +++ b/tests/local_testing/test_function_call_parsing.py @@ -144,6 +144,7 @@ def trade(model_name: str) -> List[Trade]: # type: ignore @pytest.mark.parametrize( "model", ["claude-3-haiku-20240307", "anthropic.claude-3-haiku-20240307-v1:0"] ) +@pytest.mark.flaky(retries=6, delay=10) def test_function_call_parsing(model): trades = trade(model) print([trade.order for trade in trades if trade is not None]) diff --git a/tests/local_testing/test_streaming.py b/tests/local_testing/test_streaming.py index b3f13e8a4b1..f0f3b884709 100644 --- a/tests/local_testing/test_streaming.py +++ b/tests/local_testing/test_streaming.py @@ -1930,6 +1930,7 @@ def test_openai_chat_completion_complete_response_call(): [True, False], ) @pytest.mark.asyncio +@pytest.mark.flaky(retries=6, delay=10) async def test_openai_stream_options_call(model, sync): litellm.enable_preview_features = True litellm.set_verbose = True