Method _exec_with_plugin

src/google/adk/runners.py:1335–1415

Wraps execution with plugin callbacks. Args: invocation_context: The invocation context session: The current session (ignored, kept for backward compatibility) execute_fn: A callable that returns an AsyncGenerator of Events is_live_call: Whether this is a live call

(
      self,
      invocation_context: InvocationContext,
      session: Session,
      execute_fn: Callable[[InvocationContext], AsyncGenerator[Event, None]],
      is_live_call: bool = False,
  )

    return output_event

  async def _exec_with_plugin(
      self,
      invocation_context: InvocationContext,
      session: Session,
      execute_fn: Callable[[InvocationContext], AsyncGenerator[Event, None]],
      is_live_call: bool = False,
  ) -> AsyncGenerator[Event, None]:
    """Wraps execution with plugin callbacks.

    Args:
      invocation_context: The invocation context
      session: The current session (ignored, kept for backward compatibility)
      execute_fn: A callable that returns an AsyncGenerator of Events
      is_live_call: Whether this is a live call

    Yields:
      Events from the execution, including any generated by plugins
    """

    plugin_manager = invocation_context.plugin_manager

    # Step 1: Run the before_run callbacks to see if we should early exit.
    early_exit_result = await plugin_manager.run_before_run_callback(
        invocation_context=invocation_context
    )
    if isinstance(early_exit_result, types.Content):
      early_exit_event = Event(
          invocation_id=invocation_context.invocation_id,
          author='model',
          content=early_exit_result,
      )
      _apply_run_config_custom_metadata(
          early_exit_event, invocation_context.run_config
      )
      if self._should_append_event(early_exit_event, is_live_call):
        await self.session_service.append_event(
            session=invocation_context.session,
            event=early_exit_event,
        )
      yield early_exit_event
    else:
      # Step 2: Otherwise continue with normal execution
      async with aclosing(execute_fn(invocation_context)) as agen:
        async for event in agen:
          _apply_run_config_custom_metadata(
              event, invocation_context.run_config
          )
          # Step 3: Run the on_event callbacks before persisting so callback
          # changes are stored in the session and match the streamed event.
          modified_event = await plugin_manager.run_on_event_callback(
              invocation_context=invocation_context, event=event
          )
          output_event = self._get_output_event(
              original_event=event,
              modified_event=modified_event,
              run_config=invocation_context.run_config,
          )