rasa-pro 3.11.4__py3-none-any.whl → 3.12.0__py3-none-any.whl

rasa/dialogue_understanding/commands/handle_digressions_command.py

@@ -0,0 +1,144 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, Dict, List
+
+import structlog
+
+from rasa.dialogue_understanding.commands.command import Command
+from rasa.dialogue_understanding.patterns.cannot_handle import (
+    CannotHandlePatternFlowStackFrame,
+)
+from rasa.dialogue_understanding.patterns.handle_digressions import (
+    HandleDigressionsPatternFlowStackFrame,
+)
+from rasa.dialogue_understanding.stack.utils import (
+    top_flow_frame,
+)
+from rasa.shared.core.events import Event
+from rasa.shared.core.flows import FlowsList
+from rasa.shared.core.flows.steps import CollectInformationFlowStep
+from rasa.shared.core.flows.utils import ALL_LABEL
+from rasa.shared.core.trackers import DialogueStateTracker
+from rasa.shared.nlu.constants import HANDLE_DIGRESSIONS_COMMAND
+
+structlogger = structlog.get_logger()
+
+
+@dataclass
+class HandleDigressionsCommand(Command):
+    """A command to handle digressions during an active flow."""
+
+    flow: str
+    """The interrupting flow."""
+
+    @classmethod
+    def command(cls) -> str:
+        """Returns the command type."""
+        return HANDLE_DIGRESSIONS_COMMAND
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> HandleDigressionsCommand:
+        """Converts the dictionary to a command.
+
+        Returns:
+            The converted dictionary.
+        """
+        try:
+            return HandleDigressionsCommand(flow=data["flow"])
+        except KeyError as e:
+            raise ValueError(
+                f"Missing parameter '{e}' while parsing HandleDigressionsCommand."
+            ) from e
+
+    def run_command_on_tracker(
+        self,
+        tracker: DialogueStateTracker,
+        all_flows: FlowsList,
+        original_tracker: DialogueStateTracker,
+    ) -> List[Event]:
+        """Runs the command on the tracker.
+
+        Args:
+            tracker: The tracker to run the command on.
+            all_flows: All flows in the assistant.
+            original_tracker: The tracker before any command was executed.
+
+        Returns:
+            The events to apply to the tracker.
+        """
+        stack = tracker.stack
+        original_stack = original_tracker.stack
+
+        if self.flow not in all_flows.flow_ids:
+            structlogger.debug(
+                "command_executor.push_cannot_handle.start_invalid_flow_id",
+                command=self,
+            )
+            stack.push(CannotHandlePatternFlowStackFrame())
+            return tracker.create_stack_updated_events(stack)
+
+        # this allows to include called user flows in the stack search
+        latest_user_frame = top_flow_frame(original_stack, ignore_call_frames=False)
+
+        if latest_user_frame is None:
+            structlogger.debug(
+                "command_executor.skip_command.no_top_flow", command=self
+            )
+            return []
+
+        original_top_flow = latest_user_frame.flow(all_flows)
+        current_step = original_top_flow.step_by_id(latest_user_frame.step_id)
+        if not isinstance(current_step, CollectInformationFlowStep):
+            structlogger.debug(
+                "command_executor.skip_command.not_at_a_collect_step", command=self
+            )
+            return []
+
+        ask_confirm_digressions = set(
+            current_step.ask_confirm_digressions
+            + original_top_flow.ask_confirm_digressions
+        )
+
+        block_digressions = set(
+            current_step.block_digressions + original_top_flow.block_digressions
+        )
+
+        if block_digressions:
+            if ALL_LABEL in block_digressions:
+                block_digressions.remove(ALL_LABEL)
+                block_digressions.add(self.flow)
+
+        if ask_confirm_digressions:
+            if ALL_LABEL in ask_confirm_digressions:
+                ask_confirm_digressions.remove(ALL_LABEL)
+                ask_confirm_digressions.add(self.flow)
+
+        structlogger.debug(
+            "command_executor.push_handle_digressions",
+            interrupting_flow_id=self.flow,
+            interrupted_flow_id=original_top_flow.id,
+            interrupted_step_id=current_step.id,
+            ask_confirm_digressions=ask_confirm_digressions,
+            block_digressions=block_digressions,
+        )
+        stack.push(
+            HandleDigressionsPatternFlowStackFrame(
+                interrupting_flow_id=self.flow,
+                interrupted_flow_id=original_top_flow.id,
+                interrupted_step_id=current_step.id,
+                ask_confirm_digressions=ask_confirm_digressions,
+                block_digressions=block_digressions,
+            )
+        )
+
+        return tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        return hash(self.flow)
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, HandleDigressionsCommand):
+            return False
+
+        return other.flow == self.flow

rasa/dialogue_understanding/commands/human_handoff_command.py

@@ -1,10 +1,16 @@
 from __future__ import annotations
 
+import re
 from dataclasses import dataclass
 from typing import Any, Dict, List
 
 import structlog
-from rasa.dialogue_understanding.commands import Command
+
+from rasa.dialogue_understanding.commands.command import Command
+from rasa.dialogue_understanding.commands.command_syntax_manager import (
+    CommandSyntaxManager,
+    CommandSyntaxVersion,
+)
 from rasa.dialogue_understanding.patterns.human_handoff import (
     HumanHandoffPatternFlowStackFrame,
 )
@@ -60,7 +66,31 @@ class HumanHandoffCommand(Command):
         return hash(self.command())
 
     def __eq__(self, other: object) -> bool:
-        if not isinstance(other, HumanHandoffCommand):
-            return False
+        return isinstance(other, HumanHandoffCommand)
+
+    def to_dsl(self) -> str:
+        """Converts the command to a DSL string."""
+        mapper = {
+            CommandSyntaxVersion.v1: "HumanHandoff()",
+            CommandSyntaxVersion.v2: "hand over",
+        }
+        return mapper.get(
+            CommandSyntaxManager.get_syntax_version(),
+            mapper[CommandSyntaxManager.get_default_syntax_version()],
+        )
 
-        return True
+    @classmethod
+    def from_dsl(cls, match: re.Match, **kwargs: Any) -> HumanHandoffCommand:
+        """Converts the DSL string to a command."""
+        return HumanHandoffCommand()
+
+    @staticmethod
+    def regex_pattern() -> str:
+        mapper = {
+            CommandSyntaxVersion.v1: r"HumanHandoff\(\)",
+            CommandSyntaxVersion.v2: r"""^[\s\W\d]*hand over['"`]*$""",
+        }
+        return mapper.get(
+            CommandSyntaxManager.get_syntax_version(),
+            mapper[CommandSyntaxManager.get_default_syntax_version()],
+        )

rasa/dialogue_understanding/commands/knowledge_answer_command.py

@@ -1,8 +1,16 @@
 from __future__ import annotations
 
+import re
 from dataclasses import dataclass
 from typing import Any, Dict, List
-from rasa.dialogue_understanding.commands import FreeFormAnswerCommand
+
+from rasa.dialogue_understanding.commands.command_syntax_manager import (
+    CommandSyntaxManager,
+    CommandSyntaxVersion,
+)
+from rasa.dialogue_understanding.commands.free_form_answer_command import (
+    FreeFormAnswerCommand,
+)
 from rasa.dialogue_understanding.patterns.search import SearchPatternFlowStackFrame
 from rasa.shared.core.events import Event
 from rasa.shared.core.flows import FlowsList
@@ -51,7 +59,31 @@ class KnowledgeAnswerCommand(FreeFormAnswerCommand):
         return hash(self.command())
 
     def __eq__(self, other: object) -> bool:
-        if not isinstance(other, KnowledgeAnswerCommand):
-            return False
+        return isinstance(other, KnowledgeAnswerCommand)
+
+    def to_dsl(self) -> str:
+        """Converts the command to a DSL string."""
+        mapper = {
+            CommandSyntaxVersion.v1: "SearchAndReply()",
+            CommandSyntaxVersion.v2: "provide info",
+        }
+        return mapper.get(
+            CommandSyntaxManager.get_syntax_version(),
+            mapper[CommandSyntaxManager.get_default_syntax_version()],
+        )
+
+    @classmethod
+    def from_dsl(cls, match: re.Match, **kwargs: Any) -> KnowledgeAnswerCommand:
+        """Converts the DSL string to a command."""
+        return KnowledgeAnswerCommand()
 
-        return True
+    @staticmethod
+    def regex_pattern() -> str:
+        mapper = {
+            CommandSyntaxVersion.v1: r"SearchAndReply\(\)",
+            CommandSyntaxVersion.v2: r"""^[\s\W\d]*provide info['"`]*$""",
+        }
+        return mapper.get(
+            CommandSyntaxManager.get_syntax_version(),
+            mapper[CommandSyntaxManager.get_default_syntax_version()],
+        )

rasa/dialogue_understanding/commands/noop_command.py

@@ -2,7 +2,8 @@ from __future__ import annotations
 
 from dataclasses import dataclass
 from typing import Any, Dict, List
-from rasa.dialogue_understanding.commands import Command
+
+from rasa.dialogue_understanding.commands.command import Command
 from rasa.shared.core.events import Event
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.trackers import DialogueStateTracker

rasa/dialogue_understanding/commands/prompt_command.py

@@ -0,0 +1,94 @@
+from __future__ import annotations
+
+import re
+from typing import Any, Dict, List, Optional, Protocol, runtime_checkable
+
+from rasa.shared.core.events import Event
+from rasa.shared.core.flows import FlowsList
+from rasa.shared.core.trackers import DialogueStateTracker
+
+
+@runtime_checkable
+class PromptCommand(Protocol):
+    """
+    A protocol for commands predicted by the LLM model and incorporated into the prompt.
+    """
+
+    @classmethod
+    def command(cls) -> str:
+        """
+        Returns the command name.
+
+        This class method should be implemented to return the name of the command.
+        """
+        ...
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> PromptCommand:
+        """
+        Converts the dictionary to a command.
+
+        This class method should be implemented to create a command instance from the
+        given dictionary.
+        """
+        ...
+
+    def run_command_on_tracker(
+        self,
+        tracker: DialogueStateTracker,
+        all_flows: FlowsList,
+        original_tracker: DialogueStateTracker,
+    ) -> List[Event]:
+        """
+        Runs the command on the tracker.
+
+        This method should be implemented to execute the command on the given tracker.
+        """
+        ...
+
+    def __hash__(self) -> int:
+        """
+        Returns the hash of the command.
+
+        This method should be implemented to return the hash of the command.
+        Useful for comparing commands and storing them in sets.
+        """
+        ...
+
+    def __eq__(self, other: object) -> bool:
+        """
+        Compares the command with another object.
+
+        This method should be implemented to compare the command with another object.
+        """
+        ...
+
+    def to_dsl(self) -> str:
+        """
+        Converts the command to a DSL string.
+
+        This method should be implemented to convert the command to a DSL string.
+        A DSL string is a string representation of the command that is used in the
+        prompt.
+        """
+        ...
+
+    @classmethod
+    def from_dsl(cls, match: re.Match, **kwargs: Any) -> Optional[PromptCommand]:
+        """
+        Converts the regex match to a command.
+
+        This class method should be implemented to create a command instance from the
+        given DSL string.
+        """
+        ...
+
+    @staticmethod
+    def regex_pattern() -> str:
+        """
+        Returns the regex pattern for the command.
+
+        This method should be implemented to return the regex pattern that matches the
+        command in the prompt.
+        """
+        ...

rasa/dialogue_understanding/commands/repeat_bot_messages_command.py

@@ -1,8 +1,14 @@
 from __future__ import annotations
 
+import re
 from dataclasses import dataclass
 from typing import Any, Dict, List
-from rasa.dialogue_understanding.commands import Command
+
+from rasa.dialogue_understanding.commands.command import Command
+from rasa.dialogue_understanding.commands.command_syntax_manager import (
+    CommandSyntaxManager,
+    CommandSyntaxVersion,
+)
 from rasa.dialogue_understanding.patterns.repeat import (
     RepeatBotMessagesPatternFlowStackFrame,
 )
@@ -54,7 +60,31 @@ class RepeatBotMessagesCommand(Command):
         return hash(self.command())
 
     def __eq__(self, other: object) -> bool:
-        if not isinstance(other, RepeatBotMessagesCommand):
-            return False
+        return isinstance(other, RepeatBotMessagesCommand)
+
+    def to_dsl(self) -> str:
+        """Converts the command to a DSL string."""
+        mapper = {
+            CommandSyntaxVersion.v1: "RepeatLastBotMessages()",
+            CommandSyntaxVersion.v2: "repeat message",
+        }
+        return mapper.get(
+            CommandSyntaxManager.get_syntax_version(),
+            mapper[CommandSyntaxManager.get_default_syntax_version()],
+        )
+
+    @classmethod
+    def from_dsl(cls, match: re.Match, **kwargs: Any) -> RepeatBotMessagesCommand:
+        """Converts the DSL string to a command."""
+        return RepeatBotMessagesCommand()
 
-        return True
+    @staticmethod
+    def regex_pattern() -> str:
+        mapper = {
+            CommandSyntaxVersion.v1: r"RepeatLastBotMessages\(\)",
+            CommandSyntaxVersion.v2: r"""^[\s\W\d]*repeat message['"`]*$""",
+        }
+        return mapper.get(
+            CommandSyntaxManager.get_syntax_version(),
+            mapper[CommandSyntaxManager.get_default_syntax_version()],
+        )

rasa/dialogue_understanding/commands/restart_command.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from dataclasses import dataclass
 from typing import Any, Dict, List
 
-from rasa.dialogue_understanding.commands import Command
+from rasa.dialogue_understanding.commands.command import Command
 from rasa.dialogue_understanding.patterns.restart import RestartPatternFlowStackFrame
 from rasa.shared.core.events import Event
 from rasa.shared.core.flows import FlowsList
@@ -52,7 +52,4 @@ class RestartCommand(Command):
         return hash(self.command())
 
     def __eq__(self, other: object) -> bool:
-        if not isinstance(other, RestartCommand):
-            return False
-
-        return True
+        return isinstance(other, RestartCommand)

rasa/dialogue_understanding/commands/session_end_command.py

@@ -2,7 +2,8 @@ from __future__ import annotations
 
 from dataclasses import dataclass
 from typing import Any, Dict, List
-from rasa.dialogue_understanding.commands import Command
+
+from rasa.dialogue_understanding.commands.command import Command
 from rasa.shared.core.events import Event, SessionEnded
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.trackers import DialogueStateTracker
@@ -55,7 +56,4 @@ class SessionEndCommand(Command):
         return hash(self.command())
 
     def __eq__(self, other: object) -> bool:
-        if not isinstance(other, SessionEndCommand):
-            return False
-
-        return True
+        return isinstance(other, SessionEndCommand)

rasa/dialogue_understanding/commands/session_start_command.py

@@ -2,7 +2,8 @@ from __future__ import annotations
 
 from dataclasses import dataclass
 from typing import Any, Dict, List
-from rasa.dialogue_understanding.commands import Command
+
+from rasa.dialogue_understanding.commands.command import Command
 from rasa.dialogue_understanding.patterns.session_start import (
     SessionStartPatternFlowStackFrame,
 )
@@ -53,7 +54,4 @@ class SessionStartCommand(Command):
         return hash(self.command())
 
     def __eq__(self, other: object) -> bool:
-        if not isinstance(other, SessionStartCommand):
-            return False
-
-        return True
+        return isinstance(other, SessionStartCommand)

rasa/dialogue_understanding/commands/set_slot_command.py

@@ -1,11 +1,20 @@
 from __future__ import annotations
 
+import re
 from dataclasses import dataclass
-from enum import Enum
 from typing import Any, Dict, List
 
 import structlog
-from rasa.dialogue_understanding.commands import Command
+
+from rasa.dialogue_understanding.commands.command import Command
+from rasa.dialogue_understanding.commands.command_syntax_manager import (
+    CommandSyntaxManager,
+    CommandSyntaxVersion,
+)
+from rasa.dialogue_understanding.commands.utils import (
+    clean_extracted_value,
+    get_nullable_slot_value,
+)
 from rasa.dialogue_understanding.patterns.collect_information import (
     CollectInformationPatternFlowStackFrame,
 )
@@ -13,6 +22,7 @@ from rasa.dialogue_understanding.stack.utils import (
     get_collect_steps_excluding_ask_before_filling_for_active_flow,
 )
 from rasa.shared.constants import ROUTE_TO_CALM_SLOT
+from rasa.shared.core.constants import SetSlotExtractor
 from rasa.shared.core.events import Event, SlotSet
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.trackers import DialogueStateTracker
@@ -21,17 +31,6 @@ from rasa.shared.nlu.constants import SET_SLOT_COMMAND
 structlogger = structlog.get_logger()
 
 
-class SetSlotExtractor(Enum):
-    """The extractors that can set a slot."""
-
-    LLM = "LLM"
-    COMMAND_PAYLOAD_READER = "CommandPayloadReader"
-    NLU = "NLU"
-
-    def __str__(self) -> str:
-        return self.value
-
-
 def get_flows_predicted_to_start_from_tracker(
     tracker: DialogueStateTracker,
 ) -> List[str]:
@@ -44,8 +43,8 @@ def get_flows_predicted_to_start_from_tracker(
         The flows that are predicted to start from the current state.
     """
     from rasa.dialogue_understanding.processor.command_processor import (
-        get_commands_from_tracker,
         filter_start_flow_commands,
+        get_commands_from_tracker,
     )
 
     commands = get_commands_from_tracker(tracker)
@@ -127,10 +126,12 @@ class SetSlotCommand(Command):
         if (
             self.name not in slots_of_active_flow
             and self.name != ROUTE_TO_CALM_SLOT
+            and not slot.is_builtin
             and self.extractor
             in {
                 SetSlotExtractor.LLM.value,
                 SetSlotExtractor.COMMAND_PAYLOAD_READER.value,
+                SetSlotExtractor.NLU.value,
             }
         ):
             # Get the other predicted flows from the most recent message on the tracker.
@@ -148,7 +149,9 @@ class SetSlotCommand(Command):
                 return []
 
         structlogger.debug("command_executor.set_slot", command=self)
-        return [SlotSet(self.name, slot.coerce_value(self.value))]
+        return [
+            SlotSet(self.name, slot.coerce_value(self.value), filled_by=self.extractor)
+        ]
 
     def __hash__(self) -> int:
         return hash(self.value) + hash(self.name)
@@ -157,4 +160,40 @@ class SetSlotCommand(Command):
         if not isinstance(other, SetSlotCommand):
             return False
 
-        return other.value == self.value and other.name == self.name
+        return (
+            str(other.value).lower() == str(self.value).lower()
+            and other.name == self.name
+        )
+
+    def to_dsl(self) -> str:
+        """Converts the command to a DSL string."""
+        mapper = {
+            CommandSyntaxVersion.v1: f"SetSlot({self.name}, {self.value})",
+            CommandSyntaxVersion.v2: f"set slot {self.name} {self.value}",
+        }
+        return mapper.get(
+            CommandSyntaxManager.get_syntax_version(),
+            mapper[CommandSyntaxManager.get_default_syntax_version()],
+        )
+
+    @classmethod
+    def from_dsl(cls, match: re.Match, **kwargs: Any) -> SetSlotCommand:
+        """Converts the DSL string to a command."""
+        slot_name = str(match.group(1).strip())
+        slot_value = clean_extracted_value(match.group(2))
+        return SetSlotCommand(name=slot_name, value=get_nullable_slot_value(slot_value))
+
+    @staticmethod
+    def regex_pattern() -> str:
+        mapper = {
+            CommandSyntaxVersion.v1: (
+                r"""SetSlot\(['"]?([a-zA-Z_][a-zA-Z0-9_-]*)['"]?, ?['"]?(.*)['"]?\)"""
+            ),
+            CommandSyntaxVersion.v2: (
+                r"""^[\s\W\d]*set slot ['"`]?([a-zA-Z_][a-zA-Z0-9_-]*)['"`]? ['"`]?(.+?)['"`]*$"""  # noqa: E501
+            ),
+        }
+        return mapper.get(
+            CommandSyntaxManager.get_syntax_version(),
+            mapper[CommandSyntaxManager.get_default_syntax_version()],
+        )

rasa/dialogue_understanding/commands/skip_question_command.py

@@ -1,10 +1,16 @@
 from __future__ import annotations
 
+import re
 from dataclasses import dataclass
 from typing import Any, Dict, List
 
 import structlog
-from rasa.dialogue_understanding.commands import Command
+
+from rasa.dialogue_understanding.commands.command import Command
+from rasa.dialogue_understanding.commands.command_syntax_manager import (
+    CommandSyntaxManager,
+    CommandSyntaxVersion,
+)
 from rasa.dialogue_understanding.patterns.skip_question import (
     SkipQuestionPatternFlowStackFrame,
 )
@@ -69,7 +75,31 @@ class SkipQuestionCommand(Command):
         return hash(self.command())
 
     def __eq__(self, other: object) -> bool:
-        if not isinstance(other, SkipQuestionCommand):
-            return False
+        return isinstance(other, SkipQuestionCommand)
+
+    def to_dsl(self) -> str:
+        """Converts the command to a DSL string."""
+        mapper = {
+            CommandSyntaxVersion.v1: "SkipQuestion()",
+            CommandSyntaxVersion.v2: "skip question",
+        }
+        return mapper.get(
+            CommandSyntaxManager.get_syntax_version(),
+            mapper[CommandSyntaxManager.get_default_syntax_version()],
+        )
+
+    @classmethod
+    def from_dsl(cls, match: re.Match, **kwargs: Any) -> SkipQuestionCommand:
+        """Converts the DSL string to a command."""
+        return SkipQuestionCommand()
 
-        return True
+    @staticmethod
+    def regex_pattern() -> str:
+        mapper = {
+            CommandSyntaxVersion.v1: r"SkipQuestion\(\)",
+            CommandSyntaxVersion.v2: r"""^[\s\W\d]*skip question['"`]*$""",
+        }
+        return mapper.get(
+            CommandSyntaxManager.get_syntax_version(),
+            mapper[CommandSyntaxManager.get_default_syntax_version()],
+        )