def as_story_string(self, e2e=True):
from rasa.core.events import md_format_message
correct_message = md_format_message(self.text, self.intent,
self.entities)
predicted_message = md_format_message(self.text, self.predicted_intent,
self.predicted_entities)
return ("{}: {} <!-- predicted: {}: {} -->"
"").format(self.intent.get("name"), correct_message,
self.predicted_intent, predicted_message)
def as_story_string(self, e2e: bool = True) -> Text:
from rasa.core.events import md_format_message
correct_message = md_format_message(self.text, self.intent,
self.entities)
predicted_message = md_format_message(self.text, self.predicted_intent,
self.predicted_entities)
return (
f"{self.intent.get('name')}: {correct_message} <!-- predicted: "
f"{self.predicted_intent}: {predicted_message} -->")
def as_story_string(self, e2e: bool = True) -> Text:
from rasa.core.events import md_format_message
correct_message = md_format_message(self.text, self.intent.get("name"),
self.entities)
return (f"{self.intent.get('name')}: {correct_message} "
f"<!-- {self.inline_comment()} -->")
def inline_comment(self) -> Text:
"""A comment attached to this event. Used during dumping."""
from rasa.core.events import md_format_message
predicted_message = md_format_message(self.text, self.predicted_intent,
self.predicted_entities)
return f"predicted: {self.predicted_intent}: {predicted_message}"
def test_md_format_message_using_short_entity_syntax():
formatted = md_format_message(
"I am from Berlin.",
intent="location",
entities=[{"start": 10, "end": 16, "entity": "city", "value": "Berlin"}],
)
assert formatted == """I am from [Berlin](city)."""
def test_md_format_message_using_long_entity_syntax():
formatted = md_format_message(
"I am from Berlin in Germany.",
intent="location",
entities=[
{"start": 10, "end": 16, "entity": "city", "value": "Berlin"},
{
"start": 20,
"end": 27,
"entity": "country",
"value": "Germany",
"role": "destination",
},
],
)
assert (
formatted
== """I am from [Berlin](city) in [Germany]{"entity": "country", "role": "destination"}."""
)
def test_md_format_message_empty():
assert md_format_message("", intent=None, entities=[]) == ""
def test_md_format_message():
assert (md_format_message("Hello there!", intent="greet",
entities=[]) == "Hello there!")
