Fix: generationConfig removal from tests

tests/local_testing/test_amazing_vertex_completion.py

@@ -2352,8 +2352,7 @@ async def test_completion_fine_tuned_model():
     expected_payload = {
         "contents": [
             {"role": "user", "parts": [{"text": "Write a short poem about the sky"}]}
-        ],
-        "generationConfig": {},
+        ]
     }
 
     with patch(
@@ -2833,7 +2832,6 @@ def test_gemini_function_call_parameter_in_messages():
                 }
             ],
             "toolConfig": {"functionCallingConfig": {"mode": "AUTO"}},
-            "generationConfig": {},
         } == mock_client.call_args.kwargs["json"]
 
 

tests/local_testing/test_completion.py

@@ -2330,10 +2330,6 @@ async def test_completion_functions_param():
                 "litellm_param_is_function_call"
                 not in mock_client.call_args.kwargs["json"]
             )
-            assert (
-                "litellm_param_is_function_call"
-                not in mock_client.call_args.kwargs["json"]["generationConfig"]
-            )
             assert response.choices[0].message.function_call is not None
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")