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 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 test_wit_response(): from rasa_nlu.emulators.wit import WitEmulator em = WitEmulator() data = { "text": "I want italian food", "intent": "inform", "entities": { "cuisine": "italian" } } norm = em.normalise_response_json(data) assert norm == [{ 'entities': { 'cuisine': { 'confidence': None, 'type': 'value', 'value': 'italian' } }, 'confidence': None, 'intent': 'inform', '_text': 'I want italian food' }]
def __create_emulator(self): mode = self.config['emulate'] if mode is None: from rasa_nlu.emulators import NoEmulator 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() == 'api': from rasa_nlu.emulators.api import ApiEmulator return ApiEmulator() else: raise ValueError("unknown mode : {0}".format(mode))
def __create_emulator(self): """Sets which NLU webservice to emulate among those supported by Rasa""" mode = self.config['emulate'] if mode is None: from rasa_nlu.emulators import NoEmulator 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() == 'api': from rasa_nlu.emulators.api import ApiEmulator return ApiEmulator() else: raise ValueError("unknown mode : {0}".format(mode))
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({"q": ["arb text"]}) assert norm == {"text": "arb text", "project": "default", "time": None}