Wait… You have to take a screenshot with the assistant? For me Power + volume down is the key combination that takes a screenshot. I also get a tiny button for scrolling screenshots and one for sharing them.

Funny story, mostly me just being dumb, I tried to take a screenshot of the screenshot UI, but it removes it from the screenshot automatically.

Please do let me know if this works for you though.