From bafc18724813c80e4bf6df6cd4f210a2adc84204 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Fri, 22 May 2026 22:27:14 +0000
Subject: [PATCH] test(gemini realtime): wrap remaining cached session configs
 in setup envelope

The session_configuration_request the proxy caches is always serialized
as {"setup": ...}; three modality-related tests dumped a bare config
dict instead, so transform_session_created_event's
`.get('setup', {})` quietly returned an empty dict and the
responseModalities lookup ran against the default rather than the
fixture. Wrap the remaining tests in the same shape the production
cache uses so any regression in modality forwarding actually trips.
---
 .../test_gemini_realtime_transformation.py     | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/tests/test_litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py b/tests/test_litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py
index d175de7dc47..4806a7e2737 100644
--- a/tests/test_litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py
+++ b/tests/test_litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py
@@ -102,8 +102,10 @@ def test_gemini_realtime_transformation_content_delta():
     assert config is not None
 
     session_configuration_request = {
-        "model": "gemini-1.5-flash",
-        "generationConfig": {"responseModalities": ["TEXT"]},
+        "setup": {
+            "model": "gemini-1.5-flash",
+            "generationConfig": {"responseModalities": ["TEXT"]},
+        }
     }
     session_configuration_request_str = json.dumps(session_configuration_request)
     session_created_message = {
@@ -195,8 +197,10 @@ def test_gemini_realtime_transformation_audio_delta():
     assert config is not None
 
     session_configuration_request = {
-        "model": "gemini-1.5-flash",
-        "generationConfig": {"responseModalities": ["AUDIO"]},
+        "setup": {
+            "model": "gemini-1.5-flash",
+            "generationConfig": {"responseModalities": ["AUDIO"]},
+        }
     }
     session_configuration_request_str = json.dumps(session_configuration_request)
 
@@ -244,8 +248,10 @@ def test_gemini_realtime_transformation_generation_complete():
     assert config is not None
 
     session_configuration_request = {
-        "model": "gemini-1.5-flash",
-        "generationConfig": {"responseModalities": ["AUDIO"]},
+        "setup": {
+            "model": "gemini-1.5-flash",
+            "generationConfig": {"responseModalities": ["AUDIO"]},
+        }
     }
     session_configuration_request_str = json.dumps(session_configuration_request)