rasa-pro 3.12.0.dev2__py3-none-any.whl → 3.12.0.dev4__py3-none-any.whl

rasa/dialogue_understanding/generator/nlu_command_adapter.py

@@ -1,18 +1,19 @@
-from typing import Dict, Text, Any, Optional, List
+from typing import Any, Dict, List, Optional, Text
 
 import structlog
 
 from rasa.dialogue_understanding.commands import (
     Command,
-    StartFlowCommand,
     SetSlotCommand,
+    StartFlowCommand,
 )
 from rasa.dialogue_understanding.commands.set_slot_command import SetSlotExtractor
-from rasa.dialogue_understanding.commands.utils import (
+from rasa.dialogue_understanding.generator import CommandGenerator
+from rasa.dialogue_understanding.generator.utils import (
     triggerable_pattern_to_command_class,
 )
-from rasa.dialogue_understanding.generator import CommandGenerator
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from rasa.dialogue_understanding.utils import add_commands_to_message_parse_data
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
@@ -20,10 +21,7 @@ from rasa.shared.constants import ROUTE_TO_CALM_SLOT
 from rasa.shared.core.domain import Domain
 from rasa.shared.core.flows.flows_list import FlowsList
 from rasa.shared.core.flows.steps import CollectInformationFlowStep
-from rasa.shared.core.slot_mappings import (
-    SlotFillingManager,
-    extract_slot_value,
-)
+from rasa.shared.core.slot_mappings import SlotFillingManager, extract_slot_value
 from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.nlu.constants import ENTITIES, INTENT
 from rasa.shared.nlu.training_data.message import Message
@@ -205,7 +203,7 @@ class NLUCommandAdapter(GraphComponent, CommandGenerator):
             commands=commands,
         )
 
-        CommandGenerator._add_commands_to_message_parse_data(
+        add_commands_to_message_parse_data(
             message, NLUCommandAdapter.__name__, commands
         )
         return commands

rasa/dialogue_understanding/generator/single_step/single_step_llm_command_generator.py

@@ -1,59 +1,57 @@
 import importlib.resources
-import re
-from typing import Dict, Any, List, Optional, Text
+from typing import Any, Dict, List, Optional, Text
 
 import structlog
 
 import rasa.shared.utils.io
 from rasa.dialogue_understanding.commands import (
+    CannotHandleCommand,
     Command,
     ErrorCommand,
     SetSlotCommand,
-    CancelFlowCommand,
-    HumanHandoffCommand,
-    ChitChatAnswerCommand,
-    SkipQuestionCommand,
-    KnowledgeAnswerCommand,
-    ClarifyCommand,
-    CannotHandleCommand,
-    RepeatBotMessagesCommand,
+)
+from rasa.dialogue_understanding.generator.command_parser import (
+    parse_commands as parse_commands_using_command_parsers,
 )
 from rasa.dialogue_understanding.generator.constants import (
+    DEFAULT_LLM_CONFIG,
+    FLOW_RETRIEVAL_KEY,
     LLM_CONFIG_KEY,
     USER_INPUT_CONFIG_KEY,
-    FLOW_RETRIEVAL_KEY,
-    DEFAULT_LLM_CONFIG,
-)
-from rasa.dialogue_understanding.generator.flow_retrieval import (
-    FlowRetrieval,
 )
+from rasa.dialogue_understanding.generator.flow_retrieval import FlowRetrieval
 from rasa.dialogue_understanding.generator.llm_based_command_generator import (
     LLMBasedCommandGenerator,
 )
 from rasa.dialogue_understanding.stack.utils import top_flow_frame
+from rasa.dialogue_understanding.utils import (
+    add_commands_to_message_parse_data,
+    add_prompt_to_message_parse_data,
+)
 from rasa.engine.graph import ExecutionContext
 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,
+    EMBEDDINGS_CONFIG_KEY,
     PROMPT_CONFIG_KEY,
     PROMPT_TEMPLATE_CONFIG_KEY,
-    EMBEDDINGS_CONFIG_KEY,
+    ROUTE_TO_CALM_SLOT,
 )
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.exceptions import ProviderClientAPIException
-from rasa.shared.nlu.constants import TEXT, LLM_COMMANDS, LLM_PROMPT
+from rasa.shared.nlu.constants import LLM_COMMANDS, LLM_PROMPT, TEXT
 from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.providers.llm.llm_response import LLMResponse
 from rasa.shared.utils.io import deep_container_fingerprint
 from rasa.shared.utils.llm import (
     get_prompt_template,
-    tracker_as_readable_transcript,
-    sanitize_message_for_prompt,
     resolve_model_client_config,
+    sanitize_message_for_prompt,
+    tracker_as_readable_transcript,
 )
-from rasa.utils.beta import ensure_beta_feature_is_enabled, BetaNotEnabledException
+from rasa.utils.beta import BetaNotEnabledException, ensure_beta_feature_is_enabled
 from rasa.utils.log_utils import log_llm
 
 COMMAND_PROMPT_FILE_NAME = "command_prompt.jinja2"
@@ -264,13 +262,16 @@ class SingleStepLLMCommandGenerator(LLMBasedCommandGenerator):
             prompt=flow_prompt,
         )
 
-        action_list = await self.invoke_llm(flow_prompt)
+        response = await self.invoke_llm(flow_prompt)
+        llm_response = LLMResponse.ensure_llm_response(response)
         # The check for 'None' maintains compatibility with older versions
         # of LLMCommandGenerator. In previous implementations, 'invoke_llm'
         # might return 'None' to indicate a failure to generate actions.
-        if action_list is None:
+        if llm_response is None or not llm_response.choices:
             return [ErrorCommand()]
 
+        action_list = llm_response.choices[0]
+
         log_llm(
             logger=structlogger,
             log_module="SingleStepLLMCommandGenerator",
@@ -281,14 +282,15 @@ class SingleStepLLMCommandGenerator(LLMBasedCommandGenerator):
         commands = self.parse_commands(action_list, tracker, flows)
 
         self._update_message_parse_data_for_fine_tuning(message, commands, flow_prompt)
-        self._add_commands_to_message_parse_data(
+        add_commands_to_message_parse_data(
             message, SingleStepLLMCommandGenerator.__name__, commands
         )
-        self._add_prompt_to_message_parse_data(
-            message,
-            SingleStepLLMCommandGenerator.__name__,
-            "command_generator_prompt",
-            flow_prompt,
+        add_prompt_to_message_parse_data(
+            message=message,
+            component_name=SingleStepLLMCommandGenerator.__name__,
+            prompt_name="command_generator_prompt",
+            user_prompt=flow_prompt,
+            llm_response=llm_response,
         )
 
         return commands
@@ -323,70 +325,7 @@ class SingleStepLLMCommandGenerator(LLMBasedCommandGenerator):
         Returns:
             The parsed commands.
         """
-        if actions is None:
-            return []
-
-        commands: List[Command] = []
-
-        slot_set_re = re.compile(
-            r"""SetSlot\(['"]?([a-zA-Z_][a-zA-Z0-9_-]*)['"]?, ?['"]?(.*)['"]?\)"""
-        )
-        start_flow_re = re.compile(r"StartFlow\(['\"]?([a-zA-Z0-9_-]+)['\"]?\)")
-        cancel_flow_re = re.compile(r"CancelFlow\(\)")
-        chitchat_re = re.compile(r"ChitChat\(\)")
-        skip_question_re = re.compile(r"SkipQuestion\(\)")
-        knowledge_re = re.compile(r"SearchAndReply\(\)")
-        humand_handoff_re = re.compile(r"HumanHandoff\(\)")
-        clarify_re = re.compile(r"Clarify\(([\"\'a-zA-Z0-9_, ]+)\)")
-        repeat_re = re.compile(r"RepeatLastBotMessages\(\)")
-
-        for action in actions.strip().splitlines():
-            if match := slot_set_re.search(action):
-                slot_name = match.group(1).strip()
-                slot_value = cls.clean_extracted_value(match.group(2))
-                # error case where the llm tries to start a flow using a slot set
-                if slot_name == "flow_name":
-                    commands.extend(cls.start_flow_by_name(slot_value, flows))
-                else:
-                    typed_slot_value = cls.get_nullable_slot_value(slot_value)
-                    commands.append(
-                        SetSlotCommand(name=slot_name, value=typed_slot_value)
-                    )
-            elif match := start_flow_re.search(action):
-                flow_name = match.group(1).strip()
-                commands.extend(cls.start_flow_by_name(flow_name, flows))
-            elif cancel_flow_re.search(action):
-                commands.append(CancelFlowCommand())
-            elif chitchat_re.search(action):
-                commands.append(ChitChatAnswerCommand())
-            elif skip_question_re.search(action):
-                commands.append(SkipQuestionCommand())
-            elif knowledge_re.search(action):
-                commands.append(KnowledgeAnswerCommand())
-            elif humand_handoff_re.search(action):
-                commands.append(HumanHandoffCommand())
-            elif repeat_re.search(action):
-                commands.append(RepeatBotMessagesCommand())
-            elif match := clarify_re.search(action):
-                options = sorted([opt.strip() for opt in match.group(1).split(",")])
-                # Remove surrounding quotes if present
-                cleaned_options = []
-                for flow in options:
-                    if (flow.startswith('"') and flow.endswith('"')) or (
-                        flow.startswith("'") and flow.endswith("'")
-                    ):
-                        cleaned_options.append(flow[1:-1])
-                    else:
-                        cleaned_options.append(flow)
-                # check if flow is valid
-                valid_options = [
-                    flow for flow in cleaned_options if flow in flows.user_flow_ids
-                ]
-                if len(set(valid_options)) == 1:
-                    commands.extend(cls.start_flow_by_name(valid_options[0], flows))
-                elif len(valid_options) > 1:
-                    commands.append(ClarifyCommand(valid_options))
-
+        commands = parse_commands_using_command_parsers(actions, flows)
         if not commands:
             structlogger.debug(
                 "single_step_llm_command_generator.parse_commands",

rasa/dialogue_understanding/generator/utils.py

@@ -0,0 +1,45 @@
+from typing import Dict, Type
+
+from rasa.dialogue_understanding.commands import (
+    CancelFlowCommand,
+    CannotHandleCommand,
+    ChitChatAnswerCommand,
+    Command,
+    HumanHandoffCommand,
+    KnowledgeAnswerCommand,
+    RestartCommand,
+    SessionStartCommand,
+    SkipQuestionCommand,
+)
+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,
+}

rasa/dialogue_understanding/patterns/cancel.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
+
 from dataclasses import dataclass, field
 from typing import Any, Dict, List, Optional
 
 import structlog
+
 from rasa.core.actions import action
 from rasa.core.channels.channel import OutputChannel
 from rasa.core.nlg.generator import NaturalLanguageGenerator
@@ -18,7 +20,6 @@ from rasa.shared.core.flows.steps.constants import END_STEP
 from rasa.shared.core.flows.steps.continuation import ContinueFlowStep
 from rasa.shared.core.trackers import DialogueStateTracker
 
-
 structlogger = structlog.get_logger()
 
 FLOW_PATTERN_CANCEL = RASA_DEFAULT_FLOW_PATTERN_PREFIX + "cancel_flow"

rasa/dialogue_understanding/patterns/cannot_handle.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict, Optional, Text
 

rasa/dialogue_understanding/patterns/chitchat.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict
 
 from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
 from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
 
-
 FLOW_PATTERN_CHITCHAT = RASA_DEFAULT_FLOW_PATTERN_PREFIX + "chitchat"
 
 

rasa/dialogue_understanding/patterns/clarify.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
+
 from dataclasses import dataclass, field
 from typing import Any, Dict, List, Optional
 
 import structlog
+
 from rasa.core.actions import action
 from rasa.core.channels.channel import OutputChannel
 from rasa.core.nlg.generator import NaturalLanguageGenerator
@@ -13,7 +15,6 @@ from rasa.shared.core.domain import Domain
 from rasa.shared.core.events import Event
 from rasa.shared.core.trackers import DialogueStateTracker
 
-
 structlogger = structlog.get_logger()
 
 FLOW_PATTERN_CLARIFICATION = RASA_DEFAULT_FLOW_PATTERN_PREFIX + "clarification"

rasa/dialogue_understanding/patterns/code_change.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict
 
 import structlog
+
 from rasa.dialogue_understanding.stack.frames import (
     PatternFlowStackFrame,
 )

rasa/dialogue_understanding/patterns/collect_information.py

@@ -1,12 +1,15 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict, List, Optional
-from rasa.dialogue_understanding.stack.frames import DialogueStackFrame
-from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
+
+from rasa.dialogue_understanding.stack.frames import (
+    DialogueStackFrame,
+    PatternFlowStackFrame,
+)
 from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
 from rasa.shared.core.flows.steps.collect import SlotRejection
 
-
 FLOW_PATTERN_COLLECT_INFORMATION = (
     RASA_DEFAULT_FLOW_PATTERN_PREFIX + "collect_information"
 )

rasa/dialogue_understanding/patterns/completed.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict
 
 from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
 from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
 
-
 FLOW_PATTERN_COMPLETED = RASA_DEFAULT_FLOW_PATTERN_PREFIX + "completed"
 
 

rasa/dialogue_understanding/patterns/continue_interrupted.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict
 
 from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
 from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
 
-
 FLOW_PATTERN_CONTINUE_INTERRUPTED = (
     RASA_DEFAULT_FLOW_PATTERN_PREFIX + "continue_interrupted"
 )
@@ -40,3 +40,12 @@ class ContinueInterruptedPatternFlowStackFrame(PatternFlowStackFrame):
             step_id=data["step_id"],
             previous_flow_name=data["previous_flow_name"],
         )
+
+    def __eq__(self, other: Any) -> bool:
+        if not isinstance(other, ContinueInterruptedPatternFlowStackFrame):
+            return False
+        return (
+            self.flow_id == other.flow_id
+            and self.step_id == other.step_id
+            and self.previous_flow_name == other.previous_flow_name
+        )

rasa/dialogue_understanding/patterns/correction.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
+
 from dataclasses import dataclass, field
-from typing import Any, Dict, Text, List, Optional
+from typing import Any, Dict, List, Optional, Text
 
 import structlog
+
 from rasa.core.actions import action
 from rasa.core.channels import OutputChannel
 from rasa.core.nlg import NaturalLanguageGenerator
@@ -26,7 +28,7 @@ from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
 from rasa.shared.core.constants import ACTION_CORRECT_FLOW_SLOT
 from rasa.shared.core.domain import Domain
 from rasa.shared.core.events import Event, SlotSet
-from rasa.shared.core.flows.flow import ContinueFlowStep, END_STEP, START_STEP
+from rasa.shared.core.flows.flow import END_STEP, START_STEP, ContinueFlowStep
 from rasa.shared.core.trackers import DialogueStateTracker
 
 structlogger = structlog.get_logger()

rasa/dialogue_understanding/patterns/human_handoff.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict
 
 from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
 from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
 
-
 FLOW_PATTERN_HUMAN_HANDOFF = RASA_DEFAULT_FLOW_PATTERN_PREFIX + "human_handoff"
 
 

rasa/dialogue_understanding/patterns/internal_error.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 from dataclasses import dataclass, field
 from typing import Any, Dict, Optional, Text
 

rasa/dialogue_understanding/patterns/search.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict
 
 from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
 from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
 
-
 FLOW_PATTERN_SEARCH = RASA_DEFAULT_FLOW_PATTERN_PREFIX + "search"
 
 

rasa/dialogue_understanding/patterns/session_start.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict
 
 from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
 from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
 
-
 FLOW_PATTERN_SESSION_START = RASA_DEFAULT_FLOW_PATTERN_PREFIX + "session_start"
 
 

rasa/dialogue_understanding/patterns/skip_question.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict
 

rasa/dialogue_understanding/patterns/user_silence.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict
 
 from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
 from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
 
-
 FLOW_PATTERN_USER_SILENCE = RASA_DEFAULT_FLOW_PATTERN_PREFIX + "user_silence"
 
 

rasa/dialogue_understanding/processor/command_processor.py

@@ -1,19 +1,19 @@
-from typing import List, Optional, Type, Set, Dict
+from typing import Dict, List, Optional, Set, Type
 
 import structlog
-from rasa.shared.core.training_data.structures import StoryGraph
+
 from rasa.dialogue_understanding.commands import (
     CancelFlowCommand,
+    CannotHandleCommand,
+    ChitChatAnswerCommand,
     ClarifyCommand,
     Command,
-    CorrectSlotsCommand,
     CorrectedSlot,
+    CorrectSlotsCommand,
+    FreeFormAnswerCommand,
     RepeatBotMessagesCommand,
     SetSlotCommand,
     StartFlowCommand,
-    FreeFormAnswerCommand,
-    ChitChatAnswerCommand,
-    CannotHandleCommand,
 )
 from rasa.dialogue_understanding.commands.handle_code_change_command import (
     HandleCodeChangeCommand,
@@ -36,16 +36,20 @@ from rasa.dialogue_understanding.stack.utils import (
 )
 from rasa.engine.graph import ExecutionContext
 from rasa.shared.constants import (
-    ROUTE_TO_CALM_SLOT,
     RASA_PATTERN_CANNOT_HANDLE_CHITCHAT,
+    ROUTE_TO_CALM_SLOT,
+)
+from rasa.shared.core.constants import (
+    ACTION_TRIGGER_CHITCHAT,
+    FLOW_HASHES_SLOT,
+    SlotMappingType,
 )
-from rasa.shared.core.constants import ACTION_TRIGGER_CHITCHAT, SlotMappingType
-from rasa.shared.core.constants import FLOW_HASHES_SLOT
 from rasa.shared.core.events import Event, SlotSet
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.flows.steps.collect import CollectInformationFlowStep
 from rasa.shared.core.slots import Slot
 from rasa.shared.core.trackers import DialogueStateTracker
+from rasa.shared.core.training_data.structures import StoryGraph
 from rasa.shared.nlu.constants import COMMANDS
 
 structlogger = structlog.get_logger()

rasa/dialogue_understanding/processor/command_processor_component.py

@@ -1,8 +1,8 @@
 from __future__ import annotations
 
 from typing import Any, Dict, List, Text
-import rasa.dialogue_understanding.processor.command_processor
 
+import rasa.dialogue_understanding.processor.command_processor
 from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage

rasa/dialogue_understanding/stack/dialogue_stack.py

@@ -1,11 +1,12 @@
 from __future__ import annotations
-import copy
 
+import copy
+import typing
 from dataclasses import dataclass
 from typing import Any, Callable, Dict, List, Optional
-import structlog
-import typing
+
 import jsonpatch
+import structlog
 
 if typing.TYPE_CHECKING:
     from rasa.dialogue_understanding.stack.frames import DialogueStackFrame

rasa/dialogue_understanding/stack/frames/__init__.py

@@ -1,13 +1,13 @@
+from rasa.dialogue_understanding.stack.frames.chit_chat_frame import ChitChatStackFrame
 from rasa.dialogue_understanding.stack.frames.dialogue_stack_frame import (
     DialogueStackFrame,
 )
 from rasa.dialogue_understanding.stack.frames.flow_stack_frame import (
-    UserFlowStackFrame,
     BaseFlowStackFrame,
+    UserFlowStackFrame,
 )
 from rasa.dialogue_understanding.stack.frames.pattern_frame import PatternFlowStackFrame
 from rasa.dialogue_understanding.stack.frames.search_frame import SearchStackFrame
-from rasa.dialogue_understanding.stack.frames.chit_chat_frame import ChitChatStackFrame
 
 __all__ = [
     "DialogueStackFrame",

rasa/dialogue_understanding/stack/frames/chit_chat_frame.py

@@ -2,7 +2,10 @@ from __future__ import annotations
 
 from dataclasses import dataclass
 from typing import Any, Dict
-from rasa.dialogue_understanding.stack.frames import DialogueStackFrame
+
+from rasa.dialogue_understanding.stack.frames.dialogue_stack_frame import (
+    DialogueStackFrame,
+)
 
 
 @dataclass

rasa/dialogue_understanding/stack/frames/dialogue_stack_frame.py

@@ -1,19 +1,18 @@
 from __future__ import annotations
 
+import dataclasses
 import inspect
 from dataclasses import dataclass, field
-import dataclasses
 from enum import Enum
 from functools import lru_cache
 from typing import Any, Dict, List, Tuple, Type
 
 import structlog
-from rasa.shared.exceptions import RasaException
 
 import rasa.shared.utils.common
+from rasa.shared.exceptions import RasaException
 from rasa.shared.utils.io import random_string
 
-
 structlogger = structlog.get_logger()
 
 

rasa/dialogue_understanding/stack/frames/flow_stack_frame.py

@@ -1,11 +1,14 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from enum import Enum
 from typing import Any, Dict, Optional
 
-from rasa.dialogue_understanding.stack.frames import DialogueStackFrame
-from rasa.shared.core.flows.steps.constants import START_STEP
+from rasa.dialogue_understanding.stack.frames.dialogue_stack_frame import (
+    DialogueStackFrame,
+)
 from rasa.shared.core.flows import Flow, FlowsList, FlowStep
+from rasa.shared.core.flows.steps.constants import START_STEP
 from rasa.shared.exceptions import RasaException
 
 

rasa/dialogue_understanding/stack/frames/search_frame.py

@@ -1,8 +1,11 @@
 from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict
 
-from rasa.dialogue_understanding.stack.frames import DialogueStackFrame
+from rasa.dialogue_understanding.stack.frames.dialogue_stack_frame import (
+    DialogueStackFrame,
+)
 
 
 @dataclass