I have the same issue. Enabling "Stay awake (Screen will never sleep while charging.)" from Developer options seems to only work with cabled chargers. My use case is more of nice to have: While charging wirelessly at my work desk, listening to music, and having the music app focused, I can pause music quickly within 1 touch. Not having this feature anymore means I need to activate the screen and then pause the music.
Hopefully this gets addressed soon.