Yeah, I'm already used to this kind of mechanic from playing platformers, but I was honestly scratching my head a bit when I tried to implement it myself, haha!
Just measure the time the character is not on the ground and if it is under 100-200 milliseconds, then still allow the jump :)
