A unifying model for non-adiabatic coupling at metallic surfaces beyond the local harmonic approximation