update full retrieval intent name

RasaHQ · Jan 18, 2024 · 70f397c · 70f397c
1 parent fc72dd8
commit 70f397c
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/rasa/core/processor.py b/rasa/core/processor.py
@@ -66,7 +66,11 @@
     ENTITIES,
     INTENT,
     INTENT_NAME_KEY,
+    INTENT_RESPONSE_KEY,
     PREDICTED_CONFIDENCE_KEY,
+    FULL_RETRIEVAL_INTENT_NAME_KEY,
+    RESPONSE_SELECTOR,
+    RESPONSE,
     TEXT,
 )
 from rasa.utils.endpoints import EndpointConfig
@@ -721,6 +725,7 @@ async def parse_message(
                 message, tracker, only_output_properties
             )
 
+        self._update_full_retrieval_intent(parse_data)
         structlogger.debug(
             "processor.message.parse",
             parse_data_text=copy.deepcopy(parse_data["text"]),
@@ -732,6 +737,19 @@ async def parse_message(
 
         return parse_data
 
+    def _update_full_retrieval_intent(self, parse_data: Dict[Text, Any]) -> None:
+        """Update the parse data with the full retrieval intent.
+
+        Args:
+            parse_data: Message parse data to update.
+        """
+        intent_name = parse_data.get(INTENT, {}).get(INTENT_NAME_KEY)
+        response_selector = parse_data.get(RESPONSE_SELECTOR, {})
+        all_retrieval_intents = response_selector.get("all_retrieval_intents", [])
+        if intent_name and intent_name in all_retrieval_intents:
+            retrieval_intent = response_selector.get(intent_name, {}).get(RESPONSE, {}).get(INTENT_RESPONSE_KEY)
+            parse_data[INTENT][FULL_RETRIEVAL_INTENT_NAME_KEY] = retrieval_intent
+
     def _parse_message_with_graph(
         self,
         message: UserMessage,