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!")