One way to extract the entity type of the latest user input is by parsing the latest_message
variable of the DialogueStateTracker
class which contains the entire dump of the prediction from the NLU model
prediction = tracker.latest_message
entity_type = prediction['entities'][0]['entity']