Piglin Brutes eat golden apples to heal up when they aren't fighting

It's as simple as it sounds. When the player thinks they can hit them and run away, they heal up. (they only have one golden apple but they can collect more) They also will do this with wither skeletons. I don't know if you want to make them drop their golden apples, because of balancing, but if you decide to, maybe it would be kind of okay because other Piglin brutes would just pick-up those golden apples just like how villagers pick up crops. This would add some interesting gameplay because it would be a mob with actual strategy because they wouldn't just eat a golden apple when they are 3 hearts down.