encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

chore: Returns agent state regardless if ctx.is_resumable

Browse Source 
This allows state to be passing across agents

Co-authored-by: Shangjie Chen <deanchen@google.com>
PiperOrigin-RevId: 828557989
This commit is contained in:
Shangjie Chen
2025-11-05 11:31:26 -08:00
committed by Copybara-Service
parent 744f94f0c8
commit d6b928bdf7
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/agents/base_agent.py
+3 -6
View File
@@ -164,19 +164,16 @@ class BaseAgent(BaseModel):
ctx: InvocationContext,
state_type: Type[AgentState],
) -> Optional[AgentState]:
"""Loads the agent state from the invocation context, handling resumption.
"""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 resuming, otherwise None.
The current state if exists, otherwise None.
"""
if not ctx.is_resumable:
return None
if self.name not in ctx.agent_states:
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))