State remains empty during Langgraph's chat model events ( ag-ui-langgraph )

[Avyakta000]

When using ag-ui-langgraph with the useCoAgent hook on the frontend,
the state remains empty ({}) during chat model execution events:

on_chat_model_start

on_chat_model_stream

on_chat_model_end

However, I do receive the STATE_SNAPSHOT events only after the node transition like after a tool call.

⚠️Problem

This causes the frontend state to be out of sync with the backend graph state,
unlike the behavior in CopilotKit SDK, where state updates are consistently synchronized throughout model execution.

Is this expected behavior in ag-ui-langgraph,
or should we expect the state to update during streaming/model events like in CopilotKit python-sdk?