rasa-pro 3.12.0.dev1__py3-none-any.whl → 3.12.0.dev3__py3-none-any.whl

rasa/dialogue_understanding/commands/repeat_bot_messages_command.py

@@ -1,8 +1,10 @@
 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.patterns.repeat import (
     RepeatBotMessagesPatternFlowStackFrame,
 )
@@ -54,7 +56,17 @@ 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."""
+        return "RepeatLastBotMessages()"
+
+    @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:
+        return r"RepeatLastBotMessages\(\)"

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,17 @@
 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.utils import (
+    clean_extracted_value,
+    get_nullable_slot_value,
+)
 from rasa.dialogue_understanding.patterns.collect_information import (
     CollectInformationPatternFlowStackFrame,
 )
@@ -44,8 +50,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)
@@ -157,4 +163,22 @@ 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."""
+        return f"SetSlot({self.name}, {self.value})"
+
+    @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:
+        return r"""SetSlot\(['"]?([a-zA-Z_][a-zA-Z0-9_-]*)['"]?, ?['"]?(.*)['"]?\)"""

rasa/dialogue_understanding/commands/skip_question_command.py

@@ -1,10 +1,12 @@
 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.patterns.skip_question import (
     SkipQuestionPatternFlowStackFrame,
 )
@@ -69,7 +71,17 @@ 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."""
+        return "SkipQuestion()"
+
+    @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:
+        return r"SkipQuestion\(\)"

rasa/dialogue_understanding/commands/start_flow_command.py

@@ -1,10 +1,12 @@
 from __future__ import annotations
 
+import re
 from dataclasses import dataclass
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Optional
 
 import structlog
-from rasa.dialogue_understanding.commands import Command
+
+from rasa.dialogue_understanding.commands.command import Command
 from rasa.dialogue_understanding.stack.frames.flow_stack_frame import (
     FlowStackFrameType,
     UserFlowStackFrame,
@@ -105,3 +107,16 @@ class StartFlowCommand(Command):
             return False
 
         return other.flow == self.flow
+
+    def to_dsl(self) -> str:
+        """Converts the command to a DSL string."""
+        return f"StartFlow({self.flow})"
+
+    @classmethod
+    def from_dsl(cls, match: re.Match, **kwargs: Any) -> Optional[StartFlowCommand]:
+        """Converts the DSL string to a command."""
+        return StartFlowCommand(flow=str(match.group(1).strip()))
+
+    @staticmethod
+    def regex_pattern() -> str:
+        return r"StartFlow\(['\"]?([a-zA-Z0-9_-]+)['\"]?\)"

rasa/dialogue_understanding/commands/user_silence_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.user_silence import (
     UserSilencePatternFlowStackFrame,
 )
@@ -53,7 +54,4 @@ class UserSilenceCommand(Command):
         return hash(self.command())
 
     def __eq__(self, other: object) -> bool:
-        if not isinstance(other, UserSilenceCommand):
-            return False
-
-        return True
+        return isinstance(other, UserSilenceCommand)

rasa/dialogue_understanding/commands/utils.py

@@ -1,45 +1,64 @@
-from typing import Dict, Type
-
-from rasa.dialogue_understanding.commands import (
-    CancelFlowCommand,
-    CannotHandleCommand,
-    ChitChatAnswerCommand,
-    Command,
-    HumanHandoffCommand,
-    KnowledgeAnswerCommand,
-    SessionStartCommand,
-    SkipQuestionCommand,
-    RestartCommand,
-)
-from rasa.dialogue_understanding.commands.user_silence_command import UserSilenceCommand
-from rasa.dialogue_understanding.patterns.cancel import CancelPatternFlowStackFrame
-from rasa.dialogue_understanding.patterns.cannot_handle import (
-    CannotHandlePatternFlowStackFrame,
-)
-from rasa.dialogue_understanding.patterns.chitchat import ChitchatPatternFlowStackFrame
-from rasa.dialogue_understanding.patterns.human_handoff import (
-    HumanHandoffPatternFlowStackFrame,
-)
-from rasa.dialogue_understanding.patterns.restart import RestartPatternFlowStackFrame
-from rasa.dialogue_understanding.patterns.search import SearchPatternFlowStackFrame
-from rasa.dialogue_understanding.patterns.session_start import (
-    SessionStartPatternFlowStackFrame,
-)
-from rasa.dialogue_understanding.patterns.skip_question import (
-    SkipQuestionPatternFlowStackFrame,
-)
-from rasa.dialogue_understanding.patterns.user_silence import (
-    UserSilencePatternFlowStackFrame,
-)
-
-triggerable_pattern_to_command_class: Dict[str, Type[Command]] = {
-    SessionStartPatternFlowStackFrame.flow_id: SessionStartCommand,
-    UserSilencePatternFlowStackFrame.flow_id: UserSilenceCommand,
-    CancelPatternFlowStackFrame.flow_id: CancelFlowCommand,
-    ChitchatPatternFlowStackFrame.flow_id: ChitChatAnswerCommand,
-    HumanHandoffPatternFlowStackFrame.flow_id: HumanHandoffCommand,
-    SearchPatternFlowStackFrame.flow_id: KnowledgeAnswerCommand,
-    SkipQuestionPatternFlowStackFrame.flow_id: SkipQuestionCommand,
-    CannotHandlePatternFlowStackFrame.flow_id: CannotHandleCommand,
-    RestartPatternFlowStackFrame.flow_id: RestartCommand,
-}
+from typing import TYPE_CHECKING, List, Optional, Union
+
+import structlog
+
+if TYPE_CHECKING:
+    from rasa.dialogue_understanding.commands import StartFlowCommand
+    from rasa.shared.core.flows import FlowsList
+
+structlogger = structlog.get_logger()
+
+
+def start_flow_by_name(
+    flow_name: str, flows: "FlowsList"
+) -> Optional["StartFlowCommand"]:
+    from rasa.dialogue_understanding.commands import StartFlowCommand
+
+    if flow_name in flows.user_flow_ids:
+        return StartFlowCommand(flow=flow_name)
+    else:
+        structlogger.debug(
+            "command_parser.start_flow_by_name.invalid_flow_id", flow=flow_name
+        )
+        return None
+
+
+def extract_cleaned_options(options_str: str) -> List[str]:
+    """Extract and clean options from a string."""
+    return sorted(
+        opt.strip().strip('"').strip("'")
+        for opt in options_str.split(",")
+        if opt.strip()
+    )
+
+
+def is_none_value(value: str) -> bool:
+    """Check if the value is a none value."""
+    if not value:
+        return True
+    return value in {
+        "[missing information]",
+        "[missing]",
+        "None",
+        "undefined",
+        "null",
+    }
+
+
+def clean_extracted_value(value: str) -> str:
+    """Clean up the extracted value from the llm."""
+    # replace any combination of single quotes, double quotes, and spaces
+    # from the beginning and end of the string
+    return value.strip("'\" ")
+
+
+def get_nullable_slot_value(slot_value: str) -> Union[str, None]:
+    """Get the slot value or None if the value is a none value.
+
+    Args:
+        slot_value: the value to coerce
+
+    Returns:
+        The slot value or None if the value is a none value.
+    """
+    return slot_value if not is_none_value(slot_value) else None

rasa/dialogue_understanding/constants.py

@@ -0,0 +1 @@
+RASA_RECORD_COMMANDS_AND_PROMPTS_ENV_VAR_NAME = "RASA_RECORD_COMMANDS_AND_PROMPTS"

rasa/dialogue_understanding/generator/command_generator.py

@@ -5,14 +5,14 @@ import structlog
 
 from rasa.dialogue_understanding.commands import (
     Command,
+    ErrorCommand,
     SetSlotCommand,
     StartFlowCommand,
-    ErrorCommand,
 )
 from rasa.dialogue_understanding.commands.set_slot_command import SetSlotExtractor
 from rasa.shared.constants import (
-    RASA_PATTERN_INTERNAL_ERROR_USER_INPUT_TOO_LONG,
     RASA_PATTERN_INTERNAL_ERROR_USER_INPUT_EMPTY,
+    RASA_PATTERN_INTERNAL_ERROR_USER_INPUT_TOO_LONG,
 )
 from rasa.shared.core.constants import SlotMappingType
 from rasa.shared.core.domain import Domain
@@ -22,10 +22,6 @@ from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.nlu.constants import (
     COMMANDS,
     TEXT,
-    PREDICTED_COMMANDS,
-    PROMPTS,
-    KEY_USER_PROMPT,
-    KEY_SYSTEM_PROMPT,
 )
 from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.utils.llm import DEFAULT_MAX_USER_INPUT_CHARACTERS
@@ -349,116 +345,3 @@ class CommandGenerator:
         ]
 
         return filtered_commands
-
-    @staticmethod
-    def _add_commands_to_message_parse_data(
-        message: Message, component_name: str, commands: List[Command]
-    ) -> None:
-        """Add commands to the message parse data.
-
-        Commands are only added in case the flag 'record_commands_and_prompts' is set.
-        Example of predicted commands in the message parse data:
-            Message(data={
-              PREDICTED_COMMANDS: {
-                "MultiStepLLMCommandGenerator": [
-                    {"command": "set_slot", "name": "slot_name", "value": "slot_value"},
-                ],
-                "NLUCommandAdapter": [
-                    {"command": "start_flow", "name": "test_flow"},
-                ]
-              }
-            })
-        """
-        from rasa.dialogue_understanding.utils import record_commands_and_prompts
-
-        # only set commands if the flag "record_commands_and_prompts" is set to True
-        if not record_commands_and_prompts:
-            return
-
-        commands_as_dict = [command.as_dict() for command in commands]
-
-        if message.get(PREDICTED_COMMANDS) is not None:
-            predicted_commands = message.get(PREDICTED_COMMANDS)
-            if component_name in predicted_commands:
-                predicted_commands[component_name].extend(commands_as_dict)
-            else:
-                predicted_commands[component_name] = commands_as_dict
-        else:
-            predicted_commands = {component_name: commands_as_dict}
-
-        message.set(
-            PREDICTED_COMMANDS,
-            predicted_commands,
-            add_to_output=True,
-        )
-
-    @staticmethod
-    def _add_prompt_to_message_parse_data(
-        message: Message,
-        component_name: str,
-        prompt_name: str,
-        user_prompt: str,
-        system_prompt: Optional[str] = None,
-    ) -> None:
-        """Add prompt to the message parse data.
-
-        Prompt is only added in case the flag 'record_commands_and_prompts' is set.
-        Example of prompts in the message parse data:
-            Message(data={
-              PROMPTS: {
-                "MultiStepLLMCommandGenerator": [
-                    (
-                        "fill_slots_prompt",
-                        {
-                            "user_prompt": <prompt content>",
-                            "system_prompt": <prompt content>"
-                        }
-                    ),
-                    (
-                        "handle_flows_prompt",
-                        {
-                            "user_prompt": <prompt content>",
-                            "system_prompt": <prompt content>"
-                        }
-                    ),
-                ],
-                "SingleStepLLMCommandGenerator": [
-                    (
-                        "prompt_template",
-                        {
-                            "user_prompt": <prompt content>",
-                            "system_prompt": <prompt content>"
-                        }
-                    ),
-                ]
-              }
-            })
-        """
-        from rasa.dialogue_understanding.utils import record_commands_and_prompts
-
-        # only set prompt if the flag "record_commands_and_prompts" is set to True
-        if not record_commands_and_prompts:
-            return
-
-        prompt_tuple = (
-            prompt_name,
-            {
-                KEY_USER_PROMPT: user_prompt,
-                **({KEY_SYSTEM_PROMPT: system_prompt} if system_prompt else {}),
-            },
-        )
-
-        if message.get(PROMPTS) is not None:
-            prompts = message.get(PROMPTS)
-            if component_name in prompts:
-                prompts[component_name].append(prompt_tuple)
-            else:
-                prompts[component_name] = [prompt_tuple]
-        else:
-            prompts = {component_name: [prompt_tuple]}
-
-        message.set(
-            PROMPTS,
-            prompts,
-            add_to_output=True,
-        )