IntelliJ platform (JetBrains IDEs) has this functionality out of the box without "AI" using regular code intelligence. If all your parameters are strings it may not work well I guess but if you're using types it works quite well IME.
The AI models we use are all internally hosted, and any software we use has to go through an extensive security review.
This makes perfect sense. Who could possibly trust a company run from... the Netherlands.
I get that you don't make the rules you're working under, but Jetbrains of all companies seems like a bizarre "risk" factor, given their history and actions.