rasa-pro 3.8.18__py3-none-any.whl → 3.9.15__py3-none-any.whl

rasa/core/tracker_store.py

@@ -276,10 +276,10 @@ class TrackerStore:
     async def retrieve_full_tracker(
         self, conversation_id: Text
     ) -> Optional[DialogueStateTracker]:
-        """Retrieve method for fetching all tracker events across conversation sessions\
-        that may be overridden by specific tracker.
+        """Retrieve method for fetching all tracker events.
 
-        The default implementation uses `self.retrieve()`.
+        Fetches events across conversation sessions. The default implementation
+        uses `self.retrieve()`.
 
         Args:
             conversation_id: The conversation ID to retrieve the tracker for.
@@ -339,6 +339,28 @@ class TrackerStore:
         """Returns the set of values for the tracker store's primary key."""
         raise NotImplementedError()
 
+    async def count_conversations(self, after_timestamp: float = 0.0) -> int:
+        """Returns the number of conversations that have occurred after a timestamp.
+
+        By default, this method returns the number of conversations that
+        have occurred after the Unix epoch (i.e. timestamp 0). A conversation
+        is considered to have occurred after a timestamp if at least one event
+        happened after that timestamp.
+        """
+        tracker_keys = await self.keys()
+
+        conversation_count = 0
+        for key in tracker_keys:
+            tracker = await self.retrieve(key)
+            if tracker is None or not tracker.events:
+                continue
+
+            last_event = tracker.events[-1]
+            if last_event.timestamp >= after_timestamp:
+                conversation_count += 1
+
+        return conversation_count
+
     def deserialise_tracker(
         self, sender_id: Text, serialised_tracker: Union[Text, bytes]
     ) -> Optional[DialogueStateTracker]:
@@ -930,7 +952,7 @@ def _create_sequence(table_name: Text) -> "Sequence":
     """Creates a sequence object for a specific table name.
 
     If using Oracle you will need to create a sequence in your database,
-    as described here: https://rasa.com/docs/rasa/tracker-stores#sqltrackerstore
+    as described here: https://rasa.com/docs/rasa-pro/production/tracker-stores#sqltrackerstore
     Args:
         table_name: The name of the table, which gets a Sequence assigned
 
@@ -1045,6 +1067,8 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
     from sqlalchemy.orm import DeclarativeBase
 
     class Base(DeclarativeBase):
+        """Base class for all tracker store tables."""
+
         pass
 
     class SQLEvent(Base):
@@ -1113,7 +1137,6 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
                 sqlalchemy.exc.OperationalError,
                 sqlalchemy.exc.IntegrityError,
             ) as error:
-
                 logger.warning(error)
                 sleep(5)
 
@@ -1132,8 +1155,10 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
         login_db: Optional[Text] = None,
         query: Optional[Dict] = None,
     ) -> Union[Text, "URL"]:
-        """Build an SQLAlchemy `URL` object representing the parameters needed
-        to connect to an SQL database.
+        """Build an SQLAlchemy `URL` object.
+
+        The URL object represents the parameters needed to connect to an
+        SQL database.
 
         Args:
             dialect: SQL database type.
@@ -1260,11 +1285,24 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
             conversation_id, fetch_events_from_all_sessions=True
         )
 
+    async def count_conversations(self, after_timestamp: float = 0.0) -> int:
+        """Returns the number of conversations that have occurred after a timestamp.
+
+        By default, this method returns the number of conversations that
+        have occurred after the Unix epoch (i.e. timestamp 0).
+        """
+        with self.session_scope() as session:
+            query = (
+                session.query(self.SQLEvent.sender_id)
+                .distinct()
+                .filter(self.SQLEvent.timestamp >= after_timestamp)
+            )
+            return query.count()
+
     async def _retrieve(
         self, sender_id: Text, fetch_events_from_all_sessions: bool
     ) -> Optional[DialogueStateTracker]:
         with self.session_scope() as session:
-
             serialised_events = self._event_query(
                 session,
                 sender_id,
@@ -1290,6 +1328,7 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
         self, session: "Session", sender_id: Text, fetch_events_from_all_sessions: bool
     ) -> "Query":
         """Provide the query to retrieve the conversation events for a specific sender.
+
         The events are ordered by ID to ensure correct sequence of events.
         As `timestamp` is not guaranteed to be unique and low-precision (float), it
         cannot be used to order the events.
@@ -1637,9 +1676,7 @@ class AwaitableTrackerStore(TrackerStore):
         """Wrapper to call `retrieve` method of primary tracker store."""
         result = self._tracker_store.retrieve(sender_id)
         return (
-            await result
-            if isawaitable(result)
-            else result  # type: ignore[return-value]
+            await result if isawaitable(result) else result  # type: ignore[return-value, misc]
         )
 
     async def keys(self) -> Iterable[Text]:
@@ -1658,7 +1695,5 @@ class AwaitableTrackerStore(TrackerStore):
         """Wrapper to call `retrieve_full_tracker` method of primary tracker store."""
         result = self._tracker_store.retrieve_full_tracker(conversation_id)
         return (
-            await result
-            if isawaitable(result)
-            else result  # type: ignore[return-value]
+            await result if isawaitable(result) else result  # type: ignore[return-value, misc]
         )

rasa/core/train.py

@@ -34,9 +34,7 @@ async def train_comparison_models(
             for policy_config in policy_configs:
                 config_name = os.path.splitext(os.path.basename(policy_config))[0]
                 logging.info(
-                    "Starting to train {} round {}/{}"
-                    " with {}% exclusion"
-                    "".format(
+                    "Starting to train {} round {}/{} with {}% exclusion".format(
                         config_name, current_run, len(exclusion_percentages), percentage
                     )
                 )

rasa/core/training/interactive.py

@@ -3,6 +3,7 @@ import logging
 import os
 import textwrap
 import uuid
+import warnings
 from functools import partial
 from multiprocessing import Process
 from typing import (
@@ -336,7 +337,7 @@ async def _ask_questions(
 
 
 def _selection_choices_from_intent_prediction(
-    predictions: List[Dict[Text, Any]]
+    predictions: List[Dict[Text, Any]],
 ) -> List[Dict[Text, Any]]:
     """Given a list of ML predictions create a UI choice list."""
     sorted_intents = sorted(
@@ -762,7 +763,7 @@ async def _request_export_info() -> Tuple[Text, Text, Text]:
 
 
 def _split_conversation_at_restarts(
-    events: List[Dict[Text, Any]]
+    events: List[Dict[Text, Any]],
 ) -> List[List[Dict[Text, Any]]]:
     """Split a conversation at restart events.
 
@@ -1600,6 +1601,7 @@ def _serve_application(
     """Start a core server and attach the interactive learning IO."""
     endpoint = EndpointConfig(url=DEFAULT_SERVER_FORMAT.format("http", port))
 
+    @app.after_server_start
     async def run_interactive_io(running_app: Sanic) -> None:
         """Small wrapper to shut down the server once cmd io is done."""
         await record_messages(
@@ -1613,11 +1615,9 @@ def _serve_application(
 
         running_app.stop()  # kill the sanic server
 
-    app.add_task(run_interactive_io)
-
     update_sanic_log_level()
 
-    app.run(host="0.0.0.0", port=port)
+    app.run(host="0.0.0.0", port=port, legacy=True)
 
     return app
 
@@ -1626,6 +1626,9 @@ def start_visualization(image_path: Text, port: int) -> None:
     """Add routes to serve the conversation visualization files."""
     app = Sanic("rasa_interactive")
 
+    # Reset Sanic warnings filter that allows the triggering of Sanic warnings
+    warnings.filterwarnings("ignore", category=DeprecationWarning, module=r"sanic.*")
+
     # noinspection PyUnusedLocal
     @app.exception(NotFound)
     async def ignore_404s(request: Request, exception: Exception) -> HTTPResponse:
@@ -1647,7 +1650,7 @@ def start_visualization(image_path: Text, port: int) -> None:
 
     update_sanic_log_level()
 
-    app.run(host="0.0.0.0", port=port, access_log=False)
+    app.run(host="0.0.0.0", port=port, access_log=False, legacy=True)
 
 
 def run_interactive_learning(

rasa/core/utils.py

@@ -73,8 +73,9 @@ def one_hot(hot_idx: int, length: int, dtype: Optional[Text] = None) -> np.ndarr
     """
     if hot_idx >= length:
         raise ValueError(
-            "Can't create one hot. Index '{}' is out "
-            "of range (length '{}')".format(hot_idx, length)
+            "Can't create one hot. Index '{}' is out of range (length '{}')".format(
+                hot_idx, length
+            )
         )
     r = np.zeros(length, dtype)
     r[hot_idx] = 1
@@ -159,12 +160,6 @@ def is_limit_reached(num_messages: int, limit: Optional[int]) -> bool:
     return limit is not None and num_messages >= limit
 
 
-def file_as_bytes(path: Text) -> bytes:
-    """Read in a file as a byte array."""
-    with open(path, "rb") as f:
-        return f.read()
-
-
 class AvailableEndpoints:
     """Collection of configured endpoints."""
 
@@ -216,7 +211,7 @@ class AvailableEndpoints:
 
 
 def read_endpoints_from_path(
-    endpoints_path: Optional[Union[Path, Text]] = None
+    endpoints_path: Optional[Union[Path, Text]] = None,
 ) -> AvailableEndpoints:
     """Get `AvailableEndpoints` object from specified path.
 
@@ -283,7 +278,7 @@ def replace_decimals_with_floats(obj: Any) -> Any:
 
 
 def _lock_store_is_multi_worker_compatible(
-    lock_store: Union[EndpointConfig, LockStore, None]
+    lock_store: Union[EndpointConfig, LockStore, None],
 ) -> bool:
     if isinstance(lock_store, InMemoryLockStore):
         return False

rasa/dialogue_understanding/coexistence/intent_based_router.py

@@ -19,6 +19,7 @@ from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.constants import ROUTE_TO_CALM_SLOT
+from rasa.shared.core.domain import Domain
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.exceptions import InvalidConfigException
@@ -111,6 +112,7 @@ class IntentBasedRouter(GraphComponent):
         messages: List[Message],
         flows: FlowsList,
         tracker: Optional[DialogueStateTracker] = None,
+        domain: Optional[Domain] = None,
     ) -> List[Message]:
         """Process a list of messages."""
         if tracker is None:
@@ -118,7 +120,7 @@ class IntentBasedRouter(GraphComponent):
             return messages
 
         for message in messages:
-            commands = await self.predict_commands(message, flows, tracker)
+            commands = await self.predict_commands(message, flows, tracker, domain)
             commands_dicts = [command.as_dict() for command in commands]
             message.set(COMMANDS, commands_dicts, add_to_output=True)
 
@@ -129,6 +131,7 @@ class IntentBasedRouter(GraphComponent):
         message: Message,
         flows: FlowsList,
         tracker: DialogueStateTracker,
+        domain: Optional[Domain] = None,
     ) -> List[Command]:
         if not tracker.has_coexistence_routing_slot:
             raise InvalidConfigException(
@@ -144,8 +147,8 @@ class IntentBasedRouter(GraphComponent):
             )
             return commands
         elif route_session_to_calm is True:
-            # don't set any commands so that the `LLMCommandGenerator` is triggered
-            # and can predict the actual commands.
+            # don't set any commands so that a `LLMBasedCommandGenerator` is
+            # triggered and can predict the actual commands.
             return []
         else:
             # If the session is assigned to DM1 add a `NoopCommand` to silence
@@ -156,7 +159,11 @@ class IntentBasedRouter(GraphComponent):
         self, message: Message, tracker: DialogueStateTracker, flows: FlowsList
     ) -> bool:
         """Check if the intent is part of a nlu trigger."""
-        commands = NLUCommandAdapter.convert_nlu_to_commands(message, tracker, flows)
+        commands = NLUCommandAdapter.convert_nlu_to_commands(
+            message,
+            tracker,
+            flows,
+        )
         return len(commands) > 0
 
     def _generate_command_using_intent(

rasa/dialogue_understanding/coexistence/llm_based_router.py

@@ -13,7 +13,7 @@ from rasa.dialogue_understanding.coexistence.constants import (
 )
 from rasa.dialogue_understanding.commands import Command, SetSlotCommand
 from rasa.dialogue_understanding.commands.noop_command import NoopCommand
-from rasa.dialogue_understanding.generator.llm_command_generator import LLM_CONFIG_KEY
+from rasa.dialogue_understanding.generator.constants import LLM_CONFIG_KEY
 from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
@@ -196,7 +196,7 @@ class LLMBasedRouter(GraphComponent):
             structlogger.info("llm_based_router.predicated_commands", commands=commands)
             return commands
         elif route_session_to_calm is True:
-            # don't set any commands so that the `LLMCommandGenerator` is triggered
+            # don't set any commands so that a `LLMBasedCommandGenerator` is triggered
             # and can predict the actual commands.
             return []
         else:
@@ -231,7 +231,6 @@ class LLMBasedRouter(GraphComponent):
             return [SetSlotCommand(ROUTE_TO_CALM_SLOT, False)]
 
     def render_template(self, message: Message) -> str:
-
         inputs = {
             "user_message": message.get(TEXT),
             f"{CALM_ENTRY}_{STICKY}": self.config[CALM_ENTRY][STICKY],

rasa/dialogue_understanding/commands/__init__.py

@@ -26,6 +26,8 @@ from rasa.dialogue_understanding.commands.correct_slots_command import (
     CorrectSlotsCommand,
     CorrectedSlot,
 )
+from rasa.dialogue_understanding.commands.noop_command import NoopCommand
+from rasa.dialogue_understanding.commands.change_flow_command import ChangeFlowCommand
 
 __all__ = [
     "Command",
@@ -42,4 +44,6 @@ __all__ = [
     "CorrectSlotsCommand",
     "CorrectedSlot",
     "ErrorCommand",
+    "NoopCommand",
+    "ChangeFlowCommand",
 ]

rasa/dialogue_understanding/commands/can_not_handle_command.py

@@ -59,3 +59,12 @@ class CannotHandleCommand(Command):
         else:
             stack.push(CannotHandlePatternFlowStackFrame())
         return tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        return hash(self.reason)
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, CannotHandleCommand):
+            return False
+
+        return other.reason == self.reason

rasa/dialogue_understanding/commands/cancel_flow_command.py

@@ -114,3 +114,12 @@ class CancelFlowCommand(Command):
             applied_events.append(FlowCancelled(user_frame.flow_id, user_frame.step_id))
 
         return applied_events + tracker.create_stack_updated_events(stack)
+
+    def __hash__(self) -> int:
+        return hash(self.command())
+
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, CancelFlowCommand):
+            return False
+
+        return True

rasa/dialogue_understanding/commands/change_flow_command.py

@@ -0,0 +1,38 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, Dict, List
+from rasa.dialogue_understanding.commands import Command
+from rasa.shared.core.events import Event
+from rasa.shared.core.flows import FlowsList
+from rasa.shared.core.trackers import DialogueStateTracker
+
+
+@dataclass
+class ChangeFlowCommand(Command):
+    """A command to indicate a change of flows was requested by the command
+    generator."""
+
+    @classmethod
+    def command(cls) -> str:
+        """Returns the command type."""
+        return "change_flow"
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> ChangeFlowCommand:
+        """Converts the dictionary to a command.
+
+        Returns:
+            The converted dictionary.
+        """
+        return ChangeFlowCommand()
+
+    def run_command_on_tracker(
+        self,
+        tracker: DialogueStateTracker,
+        all_flows: FlowsList,
+        original_tracker: DialogueStateTracker,
+    ) -> List[Event]:
+        # the change flow command is not actually pushing anything to the tracker,
+        # but it is predicted by the MultiStepLLMCommandGenerator and used internally
+        return []

rasa/dialogue_understanding/commands/chit_chat_answer_command.py

@@ -46,3 +46,12 @@ class ChitChatAnswerCommand(FreeFormAnswerCommand):
         stack = tracker.stack
         stack.push(ChitchatPatternFlowStackFrame())
         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, ChitChatAnswerCommand):
+            return False
+
+        return True

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,15 @@ 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
+            in {
+                SetSlotExtractor.LLM.value,
+                SetSlotExtractor.COMMAND_PAYLOAD_READER.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 +149,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",
+]