AI for selected text

I often want to improve a specific sentence within a paragraph. In this case, the ideal would be to select the words where I want to receive some suggestions for alternatives and improvements from this selected text only. However, it would only work properly by passing the surrounding context and then receiving the alternatives for the selected text. In Notion, it works more or less like this.
I think that could be an interesting feature here. Don't just apply AI to a block.