Method _load_agent_state

src/google/adk/agents/base_agent.py:178–195

Loads the agent state from the invocation context. Args: ctx: The invocation context. state_type: The type of the agent state. Returns: The current state if exists; otherwise, None.

(
      self,
      ctx: InvocationContext,
      state_type: Type[AgentState],
  )

  """

  def _load_agent_state(
      self,
      ctx: InvocationContext,
      state_type: Type[AgentState],
  ) -> Optional[AgentState]:
    """Loads the agent state from the invocation context.

    Args:
      ctx: The invocation context.
      state_type: The type of the agent state.

    Returns:
        The current state if exists; otherwise, None.
    """
    if ctx.agent_states is None or self.name not in ctx.agent_states:
      return None
    else:
      return state_type.model_validate(ctx.agent_states.get(self.name))

  def _create_agent_state_event(
      self,