[FE] Migrate validity-check to V2

Please see spike for more info > #555417 (comment 2654348406)

Edited by Samantha Ming