If you're just looking for something that's similar in the auto-complete category, I'd suggest adding the Continue plugin/extension (https://docs.continue.dev/quickstart#jetbrains) and pointing it at Claude 3.5.
If you haven't tried it yet I'd also suggest trying https://aider.chat/. It's outside of the IDE but I find it's the right level of 'agent' for where we currently are with models. Even if you use it for nothing else it's nice to be able to run the `/test` command and have it use the test output to fix errors.