Best way to restrict weapon use?

I’m using Dynamic Combat System integrated with RPG Inventory and Interaction system.

What would be the best way to have it so my mages can only use magic or melee staffs, warriors can only use swords and knights use sword and shields?

You can use either tags or enums or both.