I have an agent that updates the session state with the following call:

tool_context.state["test"] = "foo"

This agent should then hand control over to a second agent (it uses sub_agents).
However, when the LLM invokes both tools within the same call, the state update does not appear to persist.

Is this expected behavior, or am I overlooking something?