I'm definitely following book canon. The movie took about the last third and completely changed everything (her character went from being reincarnated lady of the lake to reincarnated Arthur...)

So there was a possible Merlin figure in the book, but it was never confirmed if he was or not. Just a suspicion on Ellie's part. So it would be workable if we bend a few things around.