Please restore the ability to manage messages while the model is generating a response

This feature disappeared several updates ago and still hasn't been fixed. When I'm waiting for the model to reply, I can't fork the chat or copy my message – I can only do so after the model has finished generating its response.