Sadly, this is not possible yet, neither in the frontend nor the agent code.
Tags are really only usable for scopes as of today.
A similar feature request was asked a couple weeks ago by another customer: Passing Back Tags In JWT Data Token
I will notify the product team, as it would make sense