I don’t think it works to reveal Leia knew of the vision, but nothing else. Her scenes with Rey at the beginning already hint that she could sense something was wrong with her so it’s not really a big surprise, and it’s got little to do with the big reveal about her being Palpatine’s granddaughter. It feels kinda muddled to go from talking about her bloodline to talking about the vision, and then back to bloodline. Yes the vision is related to her bloodline, but when Luke says “because you’re a Palpatine” he shifts the focus of the conversation, so it’s kinda awkward for him to go “You’re a Palpatine, and Leia also knew…”, “What? that I was a Palpatine?!”, “Oh no, she had no idea about that, she knew about the visions though.”
Also, in the movie Rey only has the one vision of the throne, she tells Finn “I had a vision…” not “I’ve been having visions”, and she tells Luke “I saw myself on the dark throne”, no “I’ve been seeing myself…” etc. so it wouldn’t make much sense to imply Leia still trained Rey despite a vision she had yesterday. Unless she’s been having visions about other, unrelated stuff that we never hear of.