scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 133

test(gemini realtime): wrap remaining cached session configs in setup envelope

Browse source 
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.
This commit is contained in:
mateo-berri 2026-05-22 22:27:14 +00:00
parent e5ffd021f7
commit bafc187248
No known key found for this signature in database
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py
View file

@ -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)