Refactor get_role_from_context method