I have used the OnSecurityTokenValidated event for getting the token after authentication like so:
OnSecurityTokenValidated
services .AddAuthentication(shared