Don't delete the LLMs response when stream stops

It would be great if it didn’t delete the response just because the stream stopped, ideally it should just have a error at the bottom and give the partial response and maybe a continue button. Thanks