scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

Fix greptile review

Browse source
This commit is contained in:
Sameer Kankute 2026-04-27 18:29:31 +05:30
parent 7a989bb8e2
commit 5f6f3e662e
No known key found for this signature in database
2 changed files with 40 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1079,7 +1079,6 @@ class GeminiRealtimeConfig(BaseRealtimeConfig):
logging_session_id,
realtime_response_transform_input["session_configuration_request"],
)
session_configuration_request = json.dumps(transformed_message)
returned_message.append(transformed_message)
elif openai_event == ResponsesAPIStreamEvents.FUNCTION_CALL_ARGUMENTS_DONE:
# Handle toolCall from Gemini

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

@ -49,6 +49,46 @@ def test_gemini_realtime_transformation_session_created():
assert transformed_message["response"][0]["type"] == "session.created"
def test_session_created_does_not_overwrite_session_configuration_request():
config = GeminiRealtimeConfig()
session_configuration_request_str = json.dumps(
{
"setup": {
"model": "models/gemini-2.5-flash-native-audio",
"generationConfig": {"responseModalities": ["AUDIO"]},
}
}
)
setup_complete_message = {"setupComplete": {}}
logging_obj = MagicMock()
logging_obj.litellm_trace_id = "trace_123"
transformed = config.transform_realtime_response(
json.dumps(setup_complete_message),
"gemini-2.5-flash-native-audio",
logging_obj,
realtime_response_transform_input={
"session_configuration_request": session_configuration_request_str,
"current_output_item_id": None,
"current_response_id": None,
"current_conversation_id": None,
"current_delta_chunks": [],
"current_item_chunks": [],
"current_delta_type": None,
},
)
# Must keep original setup payload (with "setup"), not overwrite with session.created event.
assert transformed["session_configuration_request"] == session_configuration_request_str
# Also verify emitted session.created reflects audio modality from setup payload.
session_created = transformed["response"][0]
assert session_created["type"] == "session.created"
assert "audio" in session_created["session"]["modalities"]
def test_gemini_realtime_transformation_content_delta():
config = GeminiRealtimeConfig()
assert config is not None