Support transcription for single agent live scenario

Author: jinnigu

src/google/adk/runners.py

@@ -716,12 +716,12 @@ def _new_invocation_context_for_live(
       live_request_queue: Optional[LiveRequestQueue] = None,
       run_config: Optional[RunConfig] = None,
   ) -> InvocationContext:
-    """Creates a new invocation context for live multi-agent."""
+    """Creates a new invocation context for live single and multi-agent scenarios."""
     run_config = run_config or RunConfig()
 
     # For live multi-agent, we need model's text transcription as context for
-    # next agent.
-    if self.agent.sub_agents and live_request_queue:
+    # next agent. For single-agent, we need a general transcription support.
+    if live_request_queue:
       if not run_config.response_modalities:
         # default
         run_config.response_modalities = ['AUDIO']
@@ -735,7 +735,8 @@ def _new_invocation_context_for_live(
               types.AudioTranscriptionConfig()
           )
       if not run_config.input_audio_transcription:
-        # need this input transcription for agent transferring in live mode.
+        # need this input transcription for agent transferring in multi-agent live
+        # mode and for general transcription support in single agent live mode.
         run_config.input_audio_transcription = types.AudioTranscriptionConfig()
     return self._new_invocation_context(
         session,

tests/unittests/streaming/test_live_streaming_configs.py

@@ -586,3 +586,153 @@ def test_streaming_with_session_resumption_config():
       llm_request_sent_to_mock.live_connect_config.session_resumption.transparent
       is True
   )
+
+
+def test_single_agent_live_streaming_with_transcription():
+  """Test single-agent streaming with transcription configs."""
+  response1 = LlmResponse(
+      turn_complete=True,
+  )
+
+  mock_model = testing_utils.MockModel.create([response1])
+
+  root_agent = Agent(
+      name='single_agent',
+      model=mock_model,
+      tools=[],
+  )
+
+  runner = testing_utils.InMemoryRunner(
+      root_agent=root_agent, response_modalities=['AUDIO']
+  )
+
+  # Create run config without explicitly setting transcription
+  run_config = RunConfig()
+
+  live_request_queue = LiveRequestQueue()
+  live_request_queue.send_realtime(
+      blob=types.Blob(data=b'\x00\xFF', mime_type='audio/pcm')
+  )
+
+  res_events = runner.run_live(live_request_queue, run_config)
+
+  assert res_events is not None, 'Expected a list of events, got None.'
+  assert (
+      len(res_events) > 0
+  ), 'Expected at least one response, but got an empty list.'
+  assert len(mock_model.requests) == 1
+
+  # Get the request that was captured
+  llm_request_sent_to_mock = mock_model.requests[0]
+
+  # Assert that transcription configs were added
+  assert llm_request_sent_to_mock.live_connect_config is not None
+  assert (
+      llm_request_sent_to_mock.live_connect_config.output_audio_transcription
+      is not None
+  )
+  assert (
+      llm_request_sent_to_mock.live_connect_config.input_audio_transcription
+      is not None
+  )
+
+
+def test_single_agent_live_streaming_respects_explicit_transcription():
+  """Test that single-agent live streaming respects explicitly provided transcription configs."""
+  response1 = LlmResponse(
+      turn_complete=True,
+  )
+
+  mock_model = testing_utils.MockModel.create([response1])
+
+  # Create a single agent (no sub_agents)
+  root_agent = Agent(
+      name='single_agent',
+      model=mock_model,
+      tools=[],
+  )
+
+  runner = testing_utils.InMemoryRunner(
+      root_agent=root_agent, response_modalities=['AUDIO']
+  )
+
+  # Create run config with input and output audio transcription
+  explicit_output_config = types.AudioTranscriptionConfig()
+  explicit_input_config = types.AudioTranscriptionConfig()
+  run_config = RunConfig(
+      output_audio_transcription=explicit_output_config,
+      input_audio_transcription=explicit_input_config,
+  )
+
+  live_request_queue = LiveRequestQueue()
+  live_request_queue.send_realtime(
+      blob=types.Blob(data=b'\x00\xFF', mime_type='audio/pcm')
+  )
+
+  res_events = runner.run_live(live_request_queue, run_config)
+
+  assert res_events is not None, 'Expected a list of events, got None.'
+  assert (
+      len(res_events) > 0
+  ), 'Expected at least one response, but got an empty list.'
+  assert len(mock_model.requests) == 1
+
+  # Get the request that was captured
+  llm_request_sent_to_mock = mock_model.requests[0]
+
+  # Assert that the explicit configs were used
+  assert llm_request_sent_to_mock.live_connect_config is not None
+  assert (
+      llm_request_sent_to_mock.live_connect_config.output_audio_transcription
+      is explicit_output_config
+  )
+  assert (
+      llm_request_sent_to_mock.live_connect_config.input_audio_transcription
+      is explicit_input_config
+  )
+
+
+def test_single_agent_live_streaming_with_text_modality():
+  """Test single-agent live streaming with TEXT in response modalities."""
+  response1 = LlmResponse(
+      turn_complete=True,
+  )
+
+  mock_model = testing_utils.MockModel.create([response1])
+
+  # Create a single agent (no sub_agents)
+  root_agent = Agent(
+      name='single_agent',
+      model=mock_model,
+      tools=[],
+  )
+
+  runner = testing_utils.InMemoryRunner(root_agent=root_agent)
+
+  # Create run config with TEXT in response modalities
+  run_config = RunConfig(response_modalities=['TEXT', 'AUDIO'])
+
+  live_request_queue = LiveRequestQueue()
+  live_request_queue.send_realtime(
+      blob=types.Blob(data=b'\x00\xFF', mime_type='audio/pcm')
+  )
+
+  res_events = runner.run_live(live_request_queue, run_config)
+
+  assert res_events is not None, 'Expected a list of events, got None.'
+  assert (
+      len(res_events) > 0
+  ), 'Expected at least one response, but got an empty list.'
+  assert len(mock_model.requests) == 1
+
+  # Get the request that was captured
+  llm_request_sent_to_mock = mock_model.requests[0]
+
+  # Assert that input transcription is enabled
+  assert llm_request_sent_to_mock.live_connect_config is not None
+  assert (
+      llm_request_sent_to_mock.live_connect_config.input_audio_transcription
+      is not None
+  )
+
+