Jetbrains IDE knows that the issue is missing import. You just press Alt+Shift+Enter and it's done in a couple frames.
And yet, someone decided to add annoying "fix with AI" prompt to something that was automated 20 years ago.
Guys, you are the best dev tooling company in the world, you don't need to do this!