I would say them both, though the first one seems more formal to me as the second one isn't technically correct (in the fact that I am already awake when I drink my coffee).
New words, one handy idiom, and a 2-minute quiz — delivered to your inbox to keep your streak alive.