From 48681cb31bc904383a62b62cb592154ad3b54b87 Mon Sep 17 00:00:00 2001 From: Google Team Member Date: Wed, 5 Nov 2025 15:59:25 -0800 Subject: [PATCH] chore: Returns agent state regardless if ctx.is_resumable This allows state to be passing across agents PiperOrigin-RevId: 828665491 --- src/google/adk/agents/base_agent.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/google/adk/agents/base_agent.py b/src/google/adk/agents/base_agent.py index b13001c9..09eef908 100644 --- a/src/google/adk/agents/base_agent.py +++ b/src/google/adk/agents/base_agent.py @@ -164,16 +164,19 @@ class BaseAgent(BaseModel): ctx: InvocationContext, state_type: Type[AgentState], ) -> Optional[AgentState]: - """Loads the agent state from the invocation context. + """Loads the agent state from the invocation context, handling resumption. Args: ctx: The invocation context. state_type: The type of the agent state. Returns: - The current state if exists, otherwise None. + The current state if resuming, otherwise None. """ - if ctx.agent_states is None or self.name not in ctx.agent_states: + if not ctx.is_resumable: + return None + + if self.name not in ctx.agent_states: return None else: return state_type.model_validate(ctx.agent_states.get(self.name))