With a engine replacement or rebuild on the original engine at 2 months in, I would request meeting with someone high enough up at the dealership demanding a good trade in option. If they swap the engine its going to hurt the value even more. Your only option is to keep it w/replaced, rebuilt engine from the start or deal with the original dealership at a better deal on a new truck, another dealership wont touch it.
I had it happen to me on a 2009 silverado, it drank oil from the start and by 43,000 and 15+ service vists engine had a bottom end done at dealership under warranty.
For me I wasn't going to make payments on a truck with rebuilt engine. I approached the #2 at the dealership and hashed out a good trade in value no other dealership would have given.