File tree Expand file tree Collapse file tree
src/google/adk/flows/llm_flows
tests/unittests/flows/llm_flows Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -517,7 +517,6 @@ async def run_live(
517517 llm_request .live_connect_config .session_resumption .handle = (
518518 invocation_context .live_session_resumption_handle
519519 )
520- llm_request .live_connect_config .session_resumption .transparent = True
521520
522521 logger .info (
523522 'Establishing live connection for agent: %s' ,
Original file line number Diff line number Diff line change @@ -679,6 +679,12 @@ async def mock_receive():
679679
680680 # Verify that send_history was not called because we resumed.
681681 mock_connection .send_history .assert_not_called ()
682+ reconnect_request = mock_connect .call_args .args [0 ]
683+ session_resumption = (
684+ reconnect_request .live_connect_config .session_resumption
685+ )
686+ assert session_resumption .handle == 'test_handle'
687+ assert session_resumption .transparent is not True
682688
683689
684690@pytest .mark .asyncio
You can’t perform that action at this time.
0 commit comments