Steps sometimes render below the LLM response when resuming a chat session

[jbcallaghan]

trafficstars

Describe the bug Steps sometimes render below the LLM response when resuming a chat session

To Reproduce I am using a step to stream thinking, this completes before the LLM response and shows correctly during the chat, but if you resume the chat, the step renders below the LLM response.

Expected behavior Expect steps to always render above LLM response

Additional context Tested with Chainlit 2.8.3 and 2.8.4