How do I make my AI character pause when taking damage?

I have an AI character that patrols, chases player, attacks player, dies etc… But when I melee attack it and it takes damage, I want it to stop attacking me, hence pause (and play a take damage animation), then after 1 second continue to attack me (assuming I don’t keep attacking it).

How do I do this within the behaviour tree or another way?

Okay so managed to figure this out. I made a bool called CanMelee and made a branch in front of the event melee attack and set that to true. In the event recieve damage I set CanMelee to off, update the health, pause for 2 seconds, then set CanMelee to on, which enables the AI to continue attacking