diff --git a/rasa_dialogflow_interpreter/interpreter.py b/rasa_dialogflow_interpreter/interpreter.py index bcaadc2..6bd36a3 100644 --- a/rasa_dialogflow_interpreter/interpreter.py +++ b/rasa_dialogflow_interpreter/interpreter.py @@ -48,10 +48,12 @@ def __init__( service_account_json=None, language_code="en", raise_on_error=False, + flip_text_and_intent=False, ): self.project_id = project_id self.language_code = language_code self.raise_on_error = raise_on_error + self.flip_text_and_intent = flip_text_and_intent credentials = None if service_account_json: @@ -94,7 +96,9 @@ def parse(self, text): return build_response( response.query_result.query_text, intent={ - "name": response.query_result.intent.display_name, + "name": response.query_result.intent.display_name + if not self.flip_text_and_intent + else response.query_result.fulfillment_text, "confidence": response.query_result.intent_detection_confidence, }, entities=response.query_result.parameters, diff --git a/rasa_dialogflow_interpreter/tests/test_interpreter.py b/rasa_dialogflow_interpreter/tests/test_interpreter.py index 89859f0..ee46e0e 100644 --- a/rasa_dialogflow_interpreter/tests/test_interpreter.py +++ b/rasa_dialogflow_interpreter/tests/test_interpreter.py @@ -6,12 +6,11 @@ build_response, build_entity, DialogflowInterpreter, - UpstreamError + UpstreamError, ) from google.api_core.exceptions import GoogleAPICallError - class TestBuildEntity(unittest.TestCase): def test_default(self): expected = {"entity": "things", "value": "many", "start": 0, "end": 0}