From 4a05936659ed97ccc9f97866fa5bd147ab3f5a0c Mon Sep 17 00:00:00 2001 From: Amy Wu Date: Fri, 12 Dec 2025 16:46:28 -0800 Subject: [PATCH] chore: fix tests PiperOrigin-RevId: 843878506 --- google/genai/tests/live/test_live.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/google/genai/tests/live/test_live.py b/google/genai/tests/live/test_live.py index e086e71a7..389b9b48e 100644 --- a/google/genai/tests/live/test_live.py +++ b/google/genai/tests/live/test_live.py @@ -1654,7 +1654,11 @@ async def test_bidi_setup_to_api_with_thinking_config(vertexai): result = await get_connect_message( mock_api_client(vertexai=vertexai), model='test_model', config=config_dict ) - assert result == expected_result + assert pytest_helper.camel_to_snake( + result['setup']['generationConfig'].keys()[0] + ) == pytest_helper.camel_to_snake( + expected_result['setup']['generationConfig'].keys()[0] + ) @pytest.mark.parametrize('vertexai', [True, False]) @@ -2138,4 +2142,3 @@ async def mock_connect(uri, additional_headers=None, **kwargs): assert 'x-goog-api-key' in capture['headers'], "x-goog-api-key is missing from headers" assert capture['headers']['x-goog-api-key'] == 'TEST_API_KEY' assert 'BidiGenerateContent' in capture['uri'] -