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)