You can definitely let the model improvise by defining `weapon` as `Union[Weapon, str]` if that's what you're asking.