def test_wit_response(): from rasa.nlu.emulators.wit import WitEmulator em = WitEmulator() data = { "text": "I want italian food", "intent": { "name": "inform", "confidence": 0.4794813722432127 }, "entities": [{ "entity": "cuisine", "value": "italian", "start": 7, "end": 14 }], } norm = em.normalise_response_json(data) assert norm == [{ "entities": { "cuisine": { "confidence": None, "type": "value", "value": "italian", "start": 7, "end": 14, } }, "intent": "inform", "_text": "I want italian food", "confidence": 0.4794813722432127, }]
def _create_emulator(mode: Optional[Text]) -> NoEmulator: """Create emulator for specified mode. If no emulator is specified, we will use the Rasa NLU format.""" if mode is None: return NoEmulator() elif mode.lower() == "wit": from rasa.nlu.emulators.wit import WitEmulator return WitEmulator() elif mode.lower() == "luis": from rasa.nlu.emulators.luis import LUISEmulator return LUISEmulator() elif mode.lower() == "dialogflow": from rasa.nlu.emulators.dialogflow import DialogflowEmulator return DialogflowEmulator() else: raise ErrorResponse( 400, "BadRequest", "Invalid parameter value for 'emulation_mode'. " "Should be one of 'WIT', 'LUIS', 'DIALOGFLOW'.", { "parameter": "emulation_mode", "in": "query" }, )
def _create_emulator(mode: Optional[Text]) -> NoEmulator: """Create emulator for specified mode. If no emulator is specified, we will use the Rasa NLU format.""" if mode is None: return NoEmulator() elif mode.lower() == 'wit': from rasa.nlu.emulators.wit import WitEmulator return WitEmulator() elif mode.lower() == 'luis': from rasa.nlu.emulators.luis import LUISEmulator return LUISEmulator() elif mode.lower() == 'dialogflow': from rasa.nlu.emulators.dialogflow import DialogflowEmulator return DialogflowEmulator() elif mode.lower() == 'lite': from litemind.nlu.emulators.lite import LiteEmulator return LiteEmulator() elif mode.lower() == 'coref': from litemind.nlu.emulators.coref import CorefEmulator return CorefEmulator() elif mode.lower() == 'entity': from litemind.nlu.emulators.entity import EntityEmulator return EntityEmulator() elif mode.lower() == 'link': from litemind.nlu.emulators.link import LinkEmulator return LinkEmulator() elif mode.lower() == 'relation': from litemind.nlu.emulators.relation import RelationEmulator return RelationEmulator() else: raise ValueError("unknown emulator mode : {0}".format(mode))
def test_wit_response(): from rasa.nlu.emulators.wit import WitEmulator em = WitEmulator() data = { "text": "I want italian food", "intent": {"name": "inform", "confidence": 0.4794813722432127}, "entities": [{"entity": "cuisine", "value": "italian", "start": 7, "end": 14}]} norm = em.normalise_response_json(data) assert norm == [{ 'entities': { 'cuisine': { 'confidence': None, 'type': 'value', 'value': 'italian', 'start': 7, 'end': 14 } }, 'intent': 'inform', '_text': 'I want italian food', 'confidence': 0.4794813722432127, }]
def _create_emulator(mode: Optional[Text]) -> NoEmulator: """Create emulator for specified mode. If no emulator is specified, we will use the Rasa NLU format.""" if mode is None: return NoEmulator() elif mode.lower() == 'wit': from rasa.nlu.emulators.wit import WitEmulator return WitEmulator() elif mode.lower() == 'luis': from rasa.nlu.emulators.luis import LUISEmulator return LUISEmulator() elif mode.lower() == 'dialogflow': from rasa.nlu.emulators.dialogflow import DialogflowEmulator return DialogflowEmulator() else: raise ValueError("unknown mode : {0}".format(mode))
def test_wit_request(): from rasa.nlu.emulators.wit import WitEmulator em = WitEmulator() norm = em.normalise_request_json({"text": ["arb text"]}) assert norm == {"text": "arb text", "time": None}
def test_wit_response(): from rasa.nlu.emulators.wit import WitEmulator em = WitEmulator() data = { "text": "I want italian food", "intent": { "name": "inform", "confidence": 0.4794813722432127 }, "entities": [ { "entity": "cuisine", "value": "italian", "start": 7, "end": 14, "confidence_entity": 0.1234, }, { "entity": "cuisine", "value": "italian", "role": "desert", "start": 7, "end": 14, "confidence_entity": 0.1234, }, ], } norm = em.normalise_response_json(data) expected = { "text": "I want italian food", "intents": [{ "name": "inform", "confidence": 0.4794813722432127 }], "entities": { "cuisine:cuisine": [{ "name": "cuisine", "role": "cuisine", "start": 7, "end": 14, "body": "italian", "value": "italian", "confidence": 0.1234, "entities": [], }], "cuisine:desert": [{ "name": "cuisine", "role": "desert", "start": 7, "end": 14, "body": "italian", "value": "italian", "confidence": 0.1234, "entities": [], }], }, } assert norm == expected