Tweak AI Assistant Unit Test command

I would like to tweak or create my own Generate Unit Tests command for the AI Assistant. I know I can add my own prompts, but the default one will create the file in the project already instead of the chat, I like that. 

I would like to give it some extra context, like use Mockito, annotate the tests with @DisplayName, use arrange act assert comments, etc. I do this manual now each time, but it would nice if I could tweak the default command. Any way to do this?