Choice of Models for Rewrite

I would love the option to choose what models Rewrite can use as well, not just Write. I don’t want the AI to write for me, but I would love for it to help me shift the tone, make some words flow better, and see what other models could do in different situations. Or if it could also attach to the same choices as Write.