I made some enhancements/configurations to the plugin, particularly for Lazyvim integration, which is my use case.

You can take a look at this Gist.

I also included some keymaps and other minor things.
For example with <leader>a:

Other than that, the flow is the same.