This was a known bug: https://github.com/langchain-ai/langgraph/issues/6318
For this code to work I had to upgrade to langchain==1.0.5
!pip install langgraph langchain==1.0.5 langchain_openai==1.0.0