diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index af8942f89eb..1987a7a545f 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/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"] diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index 8d815829d40..de10034ca91 100644 --- a/tests/local_testing/test_completion.py +++ b/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}")