From b729622bf5073185ab7bc12a291fdca013592edc Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Thu, 22 Jan 2026 19:00:37 +0530 Subject: [PATCH] Fix: generationConfig removal from tests --- tests/local_testing/test_amazing_vertex_completion.py | 4 +--- tests/local_testing/test_completion.py | 4 ---- 2 files changed, 1 insertion(+), 7 deletions(-) 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}")