add option to flip text and intent

m90 · Apr 17, 2019 · d2b9d86 · d2b9d86
1 parent 8f7a779
commit d2b9d86
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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}