How do i make a mob only take one item then runaway?

so i found an older thread that gave me a pretty good basis on how to make a mob steal an item ( linked here: How to remove random item from a players inventory | MCreator) but i was woundering how i'd go about making the mob steal just one item seeing as my modified version just takes all the items out of your inventory upon contact. Once it has stolen one item (and not the whole inventory) i'd like for it to run in the opposite direction of the player if that's possible.