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

fix(gemini realtime): default response.done modalities to AUDIO and correct audio-done test

Browse source
This commit is contained in:
mateo-berri 2026-05-22 18:34:43 +00:00
parent 2f0d8080d1
commit b721714783
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/gemini/realtime/transformation.py
View file

@ -904,7 +904,7 @@ class GeminiRealtimeConfig(BaseRealtimeConfig):
)
temperature = generation_config.get("temperature")
max_output_tokens = generation_config.get("max_output_tokens")
gemini_modalities = generation_config.get("responseModalities", ["TEXT"])
gemini_modalities = generation_config.get("responseModalities", ["AUDIO"])
_modalities = [
modality.lower() for modality in cast(List[str], gemini_modalities)
]

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

@ -265,9 +265,9 @@ def test_gemini_realtime_transformation_generation_complete():
contains_audio_done_event = False
for response in responses:
if response["type"] == OpenAIRealtimeEventTypes.RESPONSE_AUDIO_DONE.value:
contains_audio_delta = True
contains_audio_done_event = True
break
assert contains_audio_delta, "Expected audio delta event"
assert contains_audio_done_event, "Expected audio done event"
def test_gemini_3_1_flash_live_preview_model_cost_map_entry():