Vehviläinen edusti Suomea MM-kilpailuissa vuonna 2019, jolloin hän voitti maailmanmestaruuden. Vehviläinen pelasi turnauksessa yhden ottelun, jatkoaikatappioon päättyneen alkulohkon pelin Yhdysvaltoja vastaan Yhteensä hän on pelannut viisi A-maaottelua. Vehviläinen teki A-maajoukkuedebyyttinsä Karjala-turnauksen toisessa ottelussa 10. marraskuuta 2018 Helsingin Hartwall Arenalla Tšekkiä vastaan. Hän torjui heti nollapelin ja sai Suomen 3–0-voittoon päättyneessä ottelussa tililleen 18 torjuntaa. Vehviläinen torjui nollapelin myös toisessa ottelussaan 11. marraskuuta 2018 Ruotsia vastaan. Hän torjui Suomen 3–0-voittoon päättyneessä ottelussa 22 kertaa. Vehviläinen palkittiin joukkueensa ottelun parhaana pelaajana.