Steps to reproduce:
Write something that makes the assistant generate code
Expected behavior:
This button shows up while the assistant is typing (even if after it's completed the code block that's fine)
Current behavior: It only shows up after the assistant is done typing entirely.
Debug info:
Firefox
