In JetBrains I limit AI suggestions to just the rest of the line I'm currently writing, tends to work a lot better for me. I also use the older ML models.