Do you think the writers had this in mind with Atreus?

I think it would have been a great mission if the Aesir decided to torture Atreus rather than manipulate him. Imagine how Kratos would have felt about rescuing his son.