From 70f1066beab0d52b628872fa0f4bb63d3029cec6 Mon Sep 17 00:00:00 2001 From: paulaWesselmann Date: Mon, 4 Mar 2019 11:47:28 +0100 Subject: [PATCH] validate both entities and intent --- rasa_core/training/interactive.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/rasa_core/training/interactive.py b/rasa_core/training/interactive.py index 750335f0e88..3820c5d65db 100644 --- a/rasa_core/training/interactive.py +++ b/rasa_core/training/interactive.py @@ -997,6 +997,13 @@ def _validate_user_text(latest_message: Dict[Text, Any], parse_data = latest_message.get("parse_data", {}) text = _as_md_message(parse_data) intent = parse_data.get("intent", {}).get("name") + entities = parse_data.get("entities", []) + if entities: + entity = [] + for e in entities: + entity.append('[' + e['value'] + '](' + e['entity'] + ')') + else: + entity = "'None'" if intent is None: print("The NLU classification for '{}' returned '{}'" @@ -1004,8 +1011,9 @@ def _validate_user_text(latest_message: Dict[Text, Any], return False else: question = questionary.confirm( - "Is the NLU classification for '{}' with intent '{}' correct?" - "".format(text, intent)) + "Is the NLU classification for '{}' with intent '{}' " + "and entities: {} correct?" + "".format(text, intent, entity)) return _ask_or_abort(question, sender_id, endpoint)