It would be great if one could copy the code while the response is still being displayed. It's a bit frustrating to click on the icon button without getting the expected behavior (code copied).
Currently, one needs to wait for the message to be fully rendered before the icon button works.
To copy the code in the clipboard before the end of the response