Running into same issue. Not using create_react_agent() to create the agent though. Passed the state_schema to the StateGraph(). But still getting a KeyError.
Spending a lot of time debugging instead of coding features!