There are nodes which you can use to get the remaining time on an animation, if close to the end you could set a bool “bIsDoingCombo” to false, and make that bool a way to return from the combo / attack state in your state machine to your neutral state.
Anim graph - wait until animation has finished before entering next state