The primary function of a flywheel in water pumps is to smooth out the power delivery from the engine to the pump mechanism. This smoothing effect helps reduce vibrations and fluctuations in pressure, which can lead to more efficient operation. The flywheel absorbs excess energy during high-speed operations and releases it when needed, ensuring that the pump operates smoothly even under varying loads.