rasa-pro 3.8.17__py3-none-any.whl → 3.9.14__py3-none-any.whl

rasa/dialogue_understanding/commands/clarify_command.py

@@ -75,3 +75,12 @@ class ClarifyCommand(Command):
         names = [flow.readable_name() for flow in relevant_flows if flow is not None]
         stack.push(ClarifyPatternFlowStackFrame(names=names))
         return tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        return hash(tuple(self.options))
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, ClarifyCommand):
+            return False
+
+        return other.options == self.options

rasa/dialogue_understanding/commands/correct_slots_command.py

@@ -286,3 +286,12 @@ class CorrectSlotsCommand(Command):
 
         stack.push(correction_frame, index=insertion_index)
         return tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        return hash(self.command())
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, CorrectSlotsCommand):
+            return False
+
+        return True

rasa/dialogue_understanding/commands/error_command.py

@@ -65,3 +65,15 @@ class ErrorCommand(Command):
             )
         )
         return tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        hashed = hash(self.error_type)
+        if self.info:
+            hashed += hash(str(self.info))
+        return hashed
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, ErrorCommand):
+            return False
+
+        return other.error_type == self.error_type and other.info == self.info

rasa/dialogue_understanding/commands/handle_code_change_command.py

@@ -62,3 +62,12 @@ class HandleCodeChangeCommand(Command):
 
         stack.push(CodeChangeFlowStackFrame())
         return tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        return hash(self.command())
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, HandleCodeChangeCommand):
+            return False
+
+        return True

rasa/dialogue_understanding/commands/human_handoff_command.py

@@ -55,3 +55,12 @@ class HumanHandoffCommand(Command):
             "command_executor.human_handoff.pushed_to_stack", command=self
         )
         return tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        return hash(self.command())
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, HumanHandoffCommand):
+            return False
+
+        return True

rasa/dialogue_understanding/commands/knowledge_answer_command.py

@@ -46,3 +46,12 @@ class KnowledgeAnswerCommand(FreeFormAnswerCommand):
         stack = tracker.stack
         stack.push(SearchPatternFlowStackFrame())
         return tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        return hash(self.command())
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, KnowledgeAnswerCommand):
+            return False
+
+        return True

rasa/dialogue_understanding/commands/noop_command.py

@@ -43,3 +43,12 @@ class NoopCommand(Command):
             The events to apply to the tracker.
         """
         return []
+
+    def __hash__(self) -> int:
+        return hash(self.command())
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, NoopCommand):
+            return False
+
+        return True

rasa/dialogue_understanding/commands/set_slot_command.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass
+from enum import Enum
 from typing import Any, Dict, List
 
 import structlog
@@ -15,10 +16,22 @@ from rasa.shared.constants import ROUTE_TO_CALM_SLOT
 from rasa.shared.core.events import Event, SlotSet
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.trackers import DialogueStateTracker
+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]:
@@ -45,11 +58,12 @@ class SetSlotCommand(Command):
 
     name: str
     value: Any
+    extractor: str = SetSlotExtractor.LLM.value
 
     @classmethod
     def command(cls) -> str:
         """Returns the command type."""
-        return "set slot"
+        return SET_SLOT_COMMAND
 
     @classmethod
     def from_dict(cls, data: Dict[str, Any]) -> SetSlotCommand:
@@ -59,7 +73,11 @@ class SetSlotCommand(Command):
             The converted dictionary.
         """
         try:
-            return SetSlotCommand(name=data["name"], value=data["value"])
+            return SetSlotCommand(
+                name=data["name"],
+                value=data["value"],
+                extractor=data.get("extractor", SetSlotExtractor.LLM.value),
+            )
         except KeyError as e:
             raise ValueError(f"Missing key when parsing SetSlotCommand: {e}") from e
 
@@ -106,7 +124,11 @@ class SetSlotCommand(Command):
         if isinstance(top_frame, CollectInformationPatternFlowStackFrame):
             slots_of_active_flow.add(top_frame.collect)
 
-        if self.name not in slots_of_active_flow and self.name != ROUTE_TO_CALM_SLOT:
+        if (
+            self.name not in slots_of_active_flow
+            and self.name != ROUTE_TO_CALM_SLOT
+            and self.extractor == SetSlotExtractor.LLM.value
+        ):
             # Get the other predicted flows from the most recent message on the tracker.
             predicted_flows = get_flows_predicted_to_start_from_tracker(tracker)
             use_slot_fill = any(
@@ -123,3 +145,12 @@ class SetSlotCommand(Command):
 
         structlogger.debug("command_executor.set_slot", command=self)
         return [SlotSet(self.name, slot.coerce_value(self.value))]
+
+    def __hash__(self) -> int:
+        return hash(self.value) + hash(self.name)
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, SetSlotCommand):
+            return False
+
+        return other.value == self.value and other.name == self.name

rasa/dialogue_understanding/commands/skip_question_command.py

@@ -64,3 +64,12 @@ class SkipQuestionCommand(Command):
 
         stack.push(SkipQuestionPatternFlowStackFrame())
         return tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        return hash(self.command())
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, SkipQuestionCommand):
+            return False
+
+        return True

rasa/dialogue_understanding/commands/start_flow_command.py

@@ -96,3 +96,12 @@ class StartFlowCommand(Command):
         structlogger.debug("command_executor.start_flow", command=self)
         stack.push(UserFlowStackFrame(flow_id=self.flow, frame_type=frame_type))
         return applied_events + tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        return hash(self.flow)
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, StartFlowCommand):
+            return False
+
+        return other.flow == self.flow

rasa/dialogue_understanding/generator/__init__.py

@@ -1,6 +1,21 @@
 from rasa.dialogue_understanding.generator.command_generator import CommandGenerator
+from rasa.dialogue_understanding.generator.llm_based_command_generator import (
+    LLMBasedCommandGenerator,
+)
 from rasa.dialogue_understanding.generator.llm_command_generator import (
     LLMCommandGenerator,
 )
+from rasa.dialogue_understanding.generator.multi_step.multi_step_llm_command_generator import (  # noqa: E501
+    MultiStepLLMCommandGenerator,
+)
+from rasa.dialogue_understanding.generator.single_step.single_step_llm_command_generator import (  # noqa: E501
+    SingleStepLLMCommandGenerator,
+)
 
-__all__ = ["CommandGenerator", "LLMCommandGenerator"]
+__all__ = [
+    "CommandGenerator",
+    "LLMBasedCommandGenerator",
+    "LLMCommandGenerator",
+    "MultiStepLLMCommandGenerator",
+    "SingleStepLLMCommandGenerator",
+]

rasa/dialogue_understanding/generator/command_generator.py

@@ -3,8 +3,17 @@ from typing import Any, Dict, List, Optional, Text
 
 import structlog
 
-from rasa.dialogue_understanding.commands import Command, StartFlowCommand, ErrorCommand
+from rasa.dialogue_understanding.commands import (
+    Command,
+    SetSlotCommand,
+    StartFlowCommand,
+    ErrorCommand,
+)
+from rasa.dialogue_understanding.commands.set_slot_command import SetSlotExtractor
+from rasa.shared.core.constants import SlotMappingType
+from rasa.shared.core.domain import Domain
 from rasa.shared.core.flows import FlowsList
+from rasa.shared.core.slot_mappings import SlotFillingManager
 from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.nlu.constants import COMMANDS, TEXT
@@ -53,6 +62,7 @@ class CommandGenerator:
         messages: List[Message],
         flows: FlowsList,
         tracker: Optional[DialogueStateTracker] = None,
+        domain: Optional[Domain] = None,
     ) -> List[Message]:
         """Process a list of messages. For each message predict commands.
 
@@ -63,6 +73,7 @@ class CommandGenerator:
             messages: The messages to process.
             tracker: The tracker containing the conversation history up to now.
             flows: The flows to use for command prediction.
+            domain: The domain.
 
         Returns:
         The processed messages (usually this is just one during prediction).
@@ -78,14 +89,13 @@ class CommandGenerator:
         )
 
         for message in messages:
-
             if message.get(COMMANDS):
                 # do not overwrite commands if they are already present
                 # i.e. another command generator already predicted commands
                 continue
 
             commands = await self._evaluate_and_predict(
-                message, available_flows, tracker
+                message, available_flows, tracker, domain
             )
             # Double check commands for guarded flows. Unlikely but the llm could
             # have predicted a command for a flow that is not in the startable
@@ -93,6 +103,9 @@ class CommandGenerator:
             commands = self._check_commands_against_startable_flows(
                 commands, startable_flows
             )
+            commands = self._check_commands_against_slot_mappings(
+                commands, tracker, domain
+            )
             commands_dicts = [command.as_dict() for command in commands]
             message.set(COMMANDS, commands_dicts, add_to_output=True)
 
@@ -139,6 +152,7 @@ class CommandGenerator:
         message: Message,
         startable_flows: FlowsList,
         tracker: Optional[DialogueStateTracker] = None,
+        domain: Optional[Domain] = None,
     ) -> List[Command]:
         """Evaluates message for errors and predicts commands if no errors are found.
 
@@ -156,7 +170,9 @@ class CommandGenerator:
 
         # if no errors, try predicting commands
         try:
-            return await self.predict_commands(message, startable_flows, tracker)
+            return await self.predict_commands(
+                message, startable_flows, tracker, domain=domain
+            )
         except NotImplementedError:
             raise
         except Exception as e:
@@ -168,6 +184,7 @@ class CommandGenerator:
         message: Message,
         flows: FlowsList,
         tracker: Optional[DialogueStateTracker] = None,
+        **kwargs: Any,
     ) -> List[Command]:
         """Predict commands for a single message.
 
@@ -175,7 +192,7 @@ class CommandGenerator:
             message: The message to predict commands for.
             flows: The flows to use for command prediction.
             tracker: The tracker containing the conversation history up to now.
-
+            **kwargs: Keyword arguments for forward compatibility.
         Returns:
         The predicted commands.
         """
@@ -188,7 +205,7 @@ class CommandGenerator:
 
         Args:
             commands: The commands to check.
-            startable_flows: The flows which have their starting conditions statisfied.
+            startable_flows: The flows which have their starting conditions satisfied.
 
         Returns:
             The commands that are startable.
@@ -255,3 +272,72 @@ class CommandGenerator:
     def check_if_message_is_empty(self, message: Message) -> bool:
         """Checks if the given message is empty or whitespace-only."""
         return len(message.get(TEXT, "").strip()) == 0
+
+    @staticmethod
+    def _check_commands_against_slot_mappings(
+        commands: List[Command],
+        tracker: DialogueStateTracker,
+        domain: Optional[Domain] = None,
+    ) -> List[Command]:
+        """Check if the LLM-issued slot commands are fillable.
+
+        The LLM-issued slot commands are fillable if the slot
+        mappings are satisfied.
+        """
+        if not domain:
+            return commands
+
+        llm_fillable_slot_names = [
+            command.name
+            for command in commands
+            if isinstance(command, SetSlotCommand)
+            and command.extractor == SetSlotExtractor.LLM.value
+        ]
+
+        if not llm_fillable_slot_names:
+            return commands
+
+        llm_fillable_slots = [
+            slot for slot in domain.slots if slot.name in llm_fillable_slot_names
+        ]
+
+        slot_filling_manager = SlotFillingManager(domain, tracker)
+        slots_to_be_removed = []
+
+        structlogger.debug(
+            "command_processor.check_commands_against_slot_mappings.active_flow",
+            active_flow=tracker.active_flow,
+        )
+
+        for slot in llm_fillable_slots:
+            should_fill_slot = False
+            for mapping in slot.mappings:
+                mapping_type = SlotMappingType(mapping.get("type"))
+
+                should_fill_slot = slot_filling_manager.should_fill_slot(
+                    slot.name, mapping_type, mapping
+                )
+
+                if should_fill_slot:
+                    break
+
+            if not should_fill_slot:
+                structlogger.debug(
+                    "command_processor.check_commands_against_slot_mappings.slot_not_fillable",
+                    slot_name=slot.name,
+                )
+                slots_to_be_removed.append(slot.name)
+
+        if not slots_to_be_removed:
+            return commands
+
+        filtered_commands = [
+            command
+            for command in commands
+            if not (
+                isinstance(command, SetSlotCommand)
+                and command.name in slots_to_be_removed
+            )
+        ]
+
+        return filtered_commands

rasa/dialogue_understanding/generator/constants.py

@@ -0,0 +1,18 @@
+from rasa.shared.utils.llm import (
+    DEFAULT_OPENAI_CHAT_MODEL_NAME_ADVANCED,
+    DEFAULT_OPENAI_MAX_GENERATED_TOKENS,
+)
+
+DEFAULT_LLM_CONFIG = {
+    "_type": "openai",
+    "request_timeout": 7,
+    "temperature": 0.0,
+    "model_name": DEFAULT_OPENAI_CHAT_MODEL_NAME_ADVANCED,
+    "max_tokens": DEFAULT_OPENAI_MAX_GENERATED_TOKENS,
+}
+
+LLM_CONFIG_KEY = "llm"
+USER_INPUT_CONFIG_KEY = "user_input"
+
+FLOW_RETRIEVAL_KEY = "flow_retrieval"
+FLOW_RETRIEVAL_ACTIVE_KEY = "active"

rasa/dialogue_understanding/generator/flow_retrieval.py

@@ -34,6 +34,7 @@ from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.nlu.constants import TEXT, FLOWS_FROM_SEMANTIC_SEARCH
 from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.exceptions import ProviderClientAPIException
 from rasa.shared.utils.llm import (
     tracker_as_readable_transcript,
     embedder_factory,
@@ -93,11 +94,10 @@ class FlowRetrieval:
 
     @classmethod
     def validate_config(cls, config: Dict[Text, Any]) -> Dict[Text, Any]:
-
         if config[MAX_FLOWS_FROM_SEMANTIC_SEARCH_KEY] < 0:
-            config[
-                MAX_FLOWS_FROM_SEMANTIC_SEARCH_KEY
-            ] = DEFAULT_MAX_FLOWS_FROM_SEMANTIC_SEARCH
+            config[MAX_FLOWS_FROM_SEMANTIC_SEARCH_KEY] = (
+                DEFAULT_MAX_FLOWS_FROM_SEMANTIC_SEARCH
+            )
             structlogger.error(
                 f"flow_retrieval.validate_config.{MAX_FLOWS_FROM_SEMANTIC_SEARCH_KEY}.set_as_negative",
                 event_info=(
@@ -407,4 +407,6 @@ class FlowRetrieval:
                 error=e,
                 query=query,
             )
-            raise
+            raise ProviderClientAPIException(
+                message="Cannot fetch flows from vector store", original_exception=e
+            )