If the assistant completion has at least one token on the screen, pressing the "Stop" button should keep the completion as-is on the screen.
The current behavior is that the completion in progress gets completely reset.
The user can always retry the completion if they want to try their prompt again.