I believe Jetbrains' built in models work like that but unfortunately they force you to disable any competing AI chat plugins so you lose out on access to Claude etc. if you use them.