test(test_amazing_vertex_completion.py): add retries for 'Content has no parts.' error in vertex test

litellm/tests/test_amazing_vertex_completion.py

@@ -295,6 +295,7 @@ async def test_vertex_ai_anthropic_async_streaming():
 def test_vertex_ai():
     import random
 
+    litellm.num_retries = 3
     load_vertex_ai_credentials()
     test_models = (
         litellm.vertex_chat_models

litellm/tests/test_completion.py

@@ -97,6 +97,8 @@ def test_null_role_response():
     """
     import openai
 
+    litellm.success_callback = ["lunary"]
+
     openai_client = openai.OpenAI()
     with patch.object(
         openai_client.chat.completions, "create", side_effect=_openai_mock_response
@@ -112,6 +114,8 @@ def test_null_role_response():
 
         assert response.choices[0].message.role == "assistant"
 
+    assert False
+
 
 def test_completion_azure_command_r():
     try:

litellm/tests/test_router_retries.py

@@ -67,7 +67,6 @@ async def test_router_retries_errors(sync_mode, error_type):
     - Auth Error -> 0 retries
     - API Error -> 2 retries
     """
-
     _api_key = (
         "bad-key" if error_type == "Authorization Error" else os.getenv("AZURE_API_KEY")
     )