Session State not injected into Team-level tools

[vgandu-veza]

I'm encountering an issue where custom tools attached to an Agno Team don't receive the session state properly, even with enable_agentic_state=True and add_session_state_to_context=True configured.
Setup:

Team with 4 member agents, each with output_schema
Team configured with session_state, enable_agentic_state=True
Custom tool create_show_clause(session_state) that needs to access stored agent outputs

Issue:
The session_state parameter is always None when the custom tool is called, despite:

Built-in update_session_state tool working correctly and showing session state exists
Session state containing all expected agent outputs
Tool being properly registered with the Team

Keep Receiving this error when in debugging output when trying to call a tool with a parameter of session_state as stated in the docs here
https://docs.agno.com/concepts/teams/state
In Agent OS logging I can see the calls to update session state and it shows they are successful and the results show an updated session_state. Somehow my tool call at the end to access values in the session_state receives a NoneType instead of the updated session state.

def create_query(session_state) -> str:
...

ERROR ...... .create_query() argument after
** must be a mapping, not NoneType

Has anyone successfully used custom tools with session state at the Team level? Is this a known limitation or am I missing a configuration step? Docs state: The session_state variable is automatically passed to the tool as an argument. Any updates to it is automatically reflected in the shared state, but I don't get why I am getting an empty session state passed when my logging shows successful calls of update_session_state

Thank You

[pritipsingh]

Hey @vgandu-veza , We have released some more updates to our session_state management logic in v2.0.10. Can you try it out again and let us know if it worked for you?