The regular JetBrains IDEs have a setting to disable the AI-based inline completion, you can then just assign it to a hotkey and call it when needed.

I found that it makes the AI experience so much better.