def test_md_format_message_no_text(): formatted = format_message( "", intent="location", entities=[], ) assert formatted == ""
def test_md_format_message_using_short_entity_syntax(): formatted = 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 inline_comment(self) -> Text: """A comment attached to this event. Used during dumping.""" from rasa.shared.core.events import format_message predicted_message = format_message(self.text, self.predicted_intent, self.predicted_entities) return f"predicted: {self.predicted_intent}: {predicted_message}"
def as_story_string(self, e2e: bool = True) -> Text: """Returns text representation of event.""" from rasa.shared.core.events import format_message correct_message = format_message(self.text, self.intent.get("name"), self.entities) return (f"{self.intent.get('name')}: {correct_message} " f"<!-- {self.inline_comment()} -->")
def test_md_format_message_using_short_entity_syntax_no_start_end_or_text(): formatted = format_message("", intent="location", entities=[{ "entity": "city", "value": "Berlin" }]) assert formatted == ""
def test_md_format_message_using_long_entity_syntax_no_start_end(): formatted = format_message( "I am from Berlin.", intent="location", entities=[ {"start": 10, "end": 16, "entity": "city", "value": "Berlin"}, {"entity": "country", "value": "Germany", "role": "destination"}, ], ) assert formatted == "I am from [Berlin](city)."
def inline_comment(self) -> Optional[Text]: """A comment attached to this event. Used during dumping.""" from rasa.shared.core.events import format_message if self.predicted_intent != self.intent["name"]: predicted_message = format_message(self.text, self.predicted_intent, self.predicted_entities) return f"predicted: {self.predicted_intent}: {predicted_message}" else: return None
def test_md_format_message_using_long_entity_syntax(): formatted = 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 format_message("", intent=None, entities=[]) == ""
def test_md_format_message(): assert format_message("Hello there!", intent="greet", entities=[]) == "Hello there!"