If I had to use AI with neovim I'd probably use https://github.com/ThePrimeagen/99 or one with a similar workflow.