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

rasa/core/policies/enterprise_search_policy.py

@@ -2,6 +2,7 @@ import importlib.resources
 import json
 import re
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text
+
 import dotenv
 import structlog
 from jinja2 import Template
@@ -16,8 +17,8 @@ from rasa.core.constants import (
 )
 from rasa.core.information_retrieval import (
     InformationRetrieval,
-    SearchResult,
     InformationRetrievalException,
+    SearchResult,
     create_from_endpoint_config,
 )
 from rasa.core.information_retrieval.faiss import FAISS_Store
@@ -34,9 +35,9 @@ from rasa.dialogue_understanding.patterns.internal_error import (
 )
 from rasa.dialogue_understanding.stack.frames import (
     DialogueStackFrame,
+    PatternFlowStackFrame,
     SearchStackFrame,
 )
-from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
 from rasa.engine.graph import ExecutionContext
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
@@ -46,12 +47,12 @@ from rasa.graph_components.providers.responses_provider import Responses
 from rasa.shared.constants import (
     EMBEDDINGS_CONFIG_KEY,
     MODEL_CONFIG_KEY,
+    MODEL_GROUP_ID_CONFIG_KEY,
+    MODEL_NAME_CONFIG_KEY,
+    OPENAI_PROVIDER,
     PROMPT_CONFIG_KEY,
     PROVIDER_CONFIG_KEY,
-    OPENAI_PROVIDER,
     TIMEOUT_CONFIG_KEY,
-    MODEL_NAME_CONFIG_KEY,
-    MODEL_GROUP_ID_CONFIG_KEY,
 )
 from rasa.shared.core.constants import (
     ACTION_CANCEL_FLOW,
@@ -59,15 +60,23 @@ from rasa.shared.core.constants import (
     DEFAULT_SLOT_NAMES,
 )
 from rasa.shared.core.domain import Domain
-from rasa.shared.core.events import Event, UserUttered, BotUttered
+from rasa.shared.core.events import BotUttered, Event, UserUttered
 from rasa.shared.core.generator import TrackerWithCachedStates
 from rasa.shared.core.trackers import DialogueStateTracker, EventVerbosity
-from rasa.shared.exceptions import RasaException, FileIOException
+from rasa.shared.exceptions import FileIOException, RasaException
+from rasa.shared.nlu.constants import (
+    KEY_COMPONENT_NAME,
+    KEY_LLM_RESPONSE_METADATA,
+    KEY_PROMPT_NAME,
+    KEY_USER_PROMPT,
+    PROMPTS,
+)
 from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.providers.embedding._langchain_embedding_client_adapter import (
     _LangchainEmbeddingClientAdapter,
 )
 from rasa.shared.providers.llm.llm_client import LLMClient
+from rasa.shared.providers.llm.llm_response import LLMResponse, measure_llm_latency
 from rasa.shared.utils.cli import print_error_and_exit
 from rasa.shared.utils.health_check.embeddings_health_check_mixin import (
     EmbeddingsHealthCheckMixin,
@@ -80,9 +89,9 @@ from rasa.shared.utils.llm import (
     embedder_factory,
     get_prompt_template,
     llm_factory,
+    resolve_model_client_config,
     sanitize_message_for_prompt,
     tracker_as_readable_transcript,
-    resolve_model_client_config,
 )
 from rasa.telemetry import (
     track_enterprise_search_policy_predict,
@@ -92,6 +101,7 @@ from rasa.telemetry import (
 
 if TYPE_CHECKING:
     from langchain.schema.embeddings import Embeddings
+
     from rasa.core.featurizers.tracker_featurizers import TrackerFeaturizer
 
 from rasa.utils.log_utils import log_llm
@@ -272,6 +282,43 @@ class EnterpriseSearchPolicy(LLMHealthCheckMixin, EmbeddingsHealthCheckMixin, Po
         # Wrap the embedding client in the adapter
         return _LangchainEmbeddingClientAdapter(client)
 
+    @classmethod
+    def _add_prompt_and_llm_response_to_latest_message(
+        cls,
+        tracker: DialogueStateTracker,
+        prompt_name: str,
+        user_prompt: str,
+        llm_response: Optional[LLMResponse] = None,
+    ) -> None:
+        """Stores the prompt and LLMResponse metadata in the tracker.
+
+        Args:
+            tracker: The DialogueStateTracker containing the current conversation state.
+            prompt_name: A name identifying prompt usage.
+            user_prompt: The user prompt that was sent to the LLM.
+            llm_response: The response object from the LLM (None if no response).
+        """
+        from rasa.dialogue_understanding.utils import record_commands_and_prompts
+
+        if not record_commands_and_prompts:
+            return
+
+        if not tracker.latest_message:
+            return
+
+        parse_data = tracker.latest_message.parse_data
+        if PROMPTS not in parse_data:
+            parse_data[PROMPTS] = []  # type: ignore[literal-required]
+
+        prompt_data: Dict[Text, Any] = {
+            KEY_COMPONENT_NAME: cls.__name__,
+            KEY_PROMPT_NAME: prompt_name,
+            KEY_USER_PROMPT: user_prompt,
+            KEY_LLM_RESPONSE_METADATA: llm_response.to_dict() if llm_response else None,
+        }
+
+        parse_data[PROMPTS].append(prompt_data)  # type: ignore[literal-required]
+
     def train(  # type: ignore[override]
         self,
         training_trackers: List[TrackerWithCachedStates],
@@ -498,13 +545,27 @@ class EnterpriseSearchPolicy(LLMHealthCheckMixin, EmbeddingsHealthCheckMixin, Po
 
         if self.use_llm:
             prompt = self._render_prompt(tracker, documents.results)
-            llm_answer = await self._generate_llm_answer(llm, prompt)
+            llm_response = await self._generate_llm_answer(llm, prompt)
+            llm_response = LLMResponse.ensure_llm_response(llm_response)
+
+            self._add_prompt_and_llm_response_to_latest_message(
+                tracker=tracker,
+                prompt_name="enterprise_search_prompt",
+                user_prompt=prompt,
+                llm_response=llm_response,
+            )
+
+            if llm_response is None or not llm_response.choices:
+                logger.debug(f"{logger_key}.no_llm_response")
+                response = None
+            else:
+                llm_answer = llm_response.choices[0]
 
-            if self.citation_enabled:
-                llm_answer = self.post_process_citations(llm_answer)
+                if self.citation_enabled:
+                    llm_answer = self.post_process_citations(llm_answer)
 
-            logger.debug(f"{logger_key}.llm_answer", llm_answer=llm_answer)
-            response = llm_answer
+                logger.debug(f"{logger_key}.llm_answer", llm_answer=llm_answer)
+                response = llm_answer
         else:
             response = documents.results[0].metadata.get("answer", None)
             if not response:
@@ -516,7 +577,6 @@ class EnterpriseSearchPolicy(LLMHealthCheckMixin, EmbeddingsHealthCheckMixin, Po
                 "enterprise_search_policy.predict_action_probabilities.no_llm",
                 search_results=documents,
             )
-
         if response is None:
             return self._create_prediction_internal_error(domain, tracker)
 
@@ -579,12 +639,21 @@ class EnterpriseSearchPolicy(LLMHealthCheckMixin, EmbeddingsHealthCheckMixin, Po
         )
         return prompt
 
+    @measure_llm_latency
     async def _generate_llm_answer(
         self, llm: LLMClient, prompt: Text
-    ) -> Optional[Text]:
+    ) -> Optional[LLMResponse]:
+        """Fetches an LLM completion for the provided prompt.
+
+        Args:
+            llm: The LLM client used to get the completion.
+            prompt: The prompt text to send to the model.
+
+        Returns:
+            An LLMResponse object, or None if the call fails.
+        """
         try:
-            llm_response = await llm.acompletion(prompt)
-            llm_answer = llm_response.choices[0]
+            return await llm.acompletion(prompt)
         except Exception as e:
             # unfortunately, langchain does not wrap LLM exceptions which means
             # we have to catch all exceptions here
@@ -592,9 +661,7 @@ class EnterpriseSearchPolicy(LLMHealthCheckMixin, EmbeddingsHealthCheckMixin, Po
                 "enterprise_search_policy._generate_llm_answer.llm_error",
                 error=e,
             )
-            llm_answer = None
-
-        return llm_answer
+            return None
 
     def _create_prediction(
         self,
@@ -692,7 +759,6 @@ class EnterpriseSearchPolicy(LLMHealthCheckMixin, EmbeddingsHealthCheckMixin, Po
         **kwargs: Any,
     ) -> "EnterpriseSearchPolicy":
         """Loads a trained policy (see parent class for full docstring)."""
-
         # Perform health checks for both LLM and embeddings client configs
         cls._perform_health_checks(config, "enterprise_search_policy.load")
 

rasa/core/policies/enterprise_search_prompt_with_citation_template.jinja2

@@ -4,7 +4,7 @@ If the answer is not known or cannot be determined from the provided documents o
 Use the following documents to answer the question:
 {% for doc in docs %}
 {{ loop.cycle("*")}}. {{ doc.metadata }}
-{{ doc.page_content }}
+{{ doc.text }}
 {% endfor %}
 
 {% if citation_enabled %}

rasa/core/policies/flow_policy.py

@@ -1,9 +1,18 @@
 from __future__ import annotations
 
-from typing import Any, Dict, Text, List, Optional
-from rasa.core.policies.flows import flow_executor
+from typing import Any, Dict, List, Optional, Text
+
+import structlog
 
+from rasa.core.constants import (
+    FLOW_POLICY_PRIORITY,
+    POLICY_MAX_HISTORY,
+    POLICY_PRIORITY,
+)
+from rasa.core.featurizers.tracker_featurizers import TrackerFeaturizer
+from rasa.core.policies.flows import flow_executor
 from rasa.core.policies.flows.flow_exceptions import FlowCircuitBreakerTrippedException
+from rasa.core.policies.policy import Policy, PolicyPrediction
 from rasa.dialogue_understanding.patterns.internal_error import (
     InternalErrorPatternFlowStackFrame,
 )
@@ -14,27 +23,17 @@ from rasa.dialogue_understanding.stack.frames import (
 from rasa.dialogue_understanding.stack.utils import (
     end_top_user_flow,
 )
-
-from rasa.core.constants import (
-    FLOW_POLICY_PRIORITY,
-    POLICY_MAX_HISTORY,
-    POLICY_PRIORITY,
-)
-
-from rasa.shared.core.events import Event
-from rasa.shared.core.flows import FlowsList
-from rasa.core.featurizers.tracker_featurizers import TrackerFeaturizer
-from rasa.core.policies.policy import Policy, PolicyPrediction
 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.core.domain import Domain
+from rasa.shared.core.events import Event
+from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.generator import TrackerWithCachedStates
 from rasa.shared.core.trackers import (
     DialogueStateTracker,
 )
-import structlog
 
 structlogger = structlog.get_logger()
 

rasa/core/policies/flows/flow_executor.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any, Dict, Text, List, Optional
+from typing import Any, Dict, List, Optional, Text
 
 import structlog
 from jinja2 import Template
@@ -9,20 +9,21 @@ from structlog.contextvars import (
     bound_contextvars,
 )
 
-from rasa.core.constants import STEP_ID_METADATA_KEY, ACTIVE_FLOW_METADATA_KEY
+from rasa.core.constants import ACTIVE_FLOW_METADATA_KEY, STEP_ID_METADATA_KEY
 from rasa.core.policies.flows.flow_exceptions import (
     FlowCircuitBreakerTrippedException,
     FlowException,
     NoNextStepInFlowException,
 )
 from rasa.core.policies.flows.flow_step_result import (
-    FlowActionPrediction,
     ContinueFlowWithNextStep,
+    FlowActionPrediction,
     FlowStepResult,
     PauseFlowReturnPrediction,
 )
 from rasa.dialogue_understanding.commands import CancelFlowCommand
 from rasa.dialogue_understanding.patterns.cancel import CancelPatternFlowStackFrame
+from rasa.dialogue_understanding.patterns.clarify import ClarifyPatternFlowStackFrame
 from rasa.dialogue_understanding.patterns.collect_information import (
     CollectInformationPatternFlowStackFrame,
 )
@@ -50,9 +51,12 @@ from rasa.dialogue_understanding.stack.frames.flow_stack_frame import (
 )
 from rasa.dialogue_understanding.stack.utils import (
     top_user_flow_frame,
+    user_flows_on_the_stack,
 )
 from rasa.shared.constants import RASA_PATTERN_HUMAN_HANDOFF
-from rasa.shared.core.constants import ACTION_LISTEN_NAME, SlotMappingType
+from rasa.shared.core.constants import (
+    ACTION_LISTEN_NAME,
+)
 from rasa.shared.core.events import (
     Event,
     FlowCompleted,
@@ -67,22 +71,22 @@ from rasa.shared.core.flows.flow import (
     FlowStep,
 )
 from rasa.shared.core.flows.flow_step_links import (
-    StaticFlowStepLink,
-    IfFlowStepLink,
     ElseFlowStepLink,
+    IfFlowStepLink,
+    StaticFlowStepLink,
 )
 from rasa.shared.core.flows.steps import (
     ActionFlowStep,
-    SetSlotsFlowStep,
-    LinkFlowStep,
-    ContinueFlowStep,
-    EndFlowStep,
     CallFlowStep,
     CollectInformationFlowStep,
+    ContinueFlowStep,
+    EndFlowStep,
+    LinkFlowStep,
     NoOperationFlowStep,
+    SetSlotsFlowStep,
 )
-from rasa.shared.core.flows.steps.collect import SlotRejection
-from rasa.shared.core.slots import Slot
+from rasa.shared.core.flows.steps.constants import START_STEP
+from rasa.shared.core.slots import Slot, SlotRejection
 from rasa.shared.core.trackers import (
     DialogueStateTracker,
 )
@@ -238,7 +242,10 @@ def events_for_collect_step_execution(
 
 
 def trigger_pattern_continue_interrupted(
-    current_frame: DialogueStackFrame, stack: DialogueStack, flows: FlowsList
+    current_frame: DialogueStackFrame,
+    stack: DialogueStack,
+    flows: FlowsList,
+    tracker: DialogueStateTracker,
 ) -> List[Event]:
     """Trigger the pattern to continue an interrupted flow if needed."""
     events: List[Event] = []
@@ -261,7 +268,9 @@ def trigger_pattern_continue_interrupted(
     ):
         stack.push(
             ContinueInterruptedPatternFlowStackFrame(
-                previous_flow_name=interrupted_user_flow.readable_name(),
+                previous_flow_name=interrupted_user_flow.readable_name(
+                    language=tracker.current_language
+                ),
             )
         )
         events.append(
@@ -271,6 +280,33 @@ def trigger_pattern_continue_interrupted(
     return events
 
 
+def trigger_pattern_clarification(
+    current_frame: DialogueStackFrame, stack: DialogueStack, flows: FlowsList
+) -> None:
+    """Trigger the pattern to clarify which topic to continue if needed."""
+    if not isinstance(current_frame, UserFlowStackFrame):
+        return None
+
+    if current_frame.frame_type in [
+        FlowStackFrameType.CALL,
+        FlowStackFrameType.INTERRUPT,
+    ]:
+        # we want to return to the flow that called
+        # the current flow or the flow that was interrupted
+        # by the current flow
+        return None
+
+    pending_flows = [
+        flows.flow_by_id(frame.flow_id)
+        for frame in stack.frames
+        if isinstance(frame, UserFlowStackFrame)
+        and frame.flow_id != current_frame.flow_id
+    ]
+
+    flow_names = [flow.readable_name() for flow in pending_flows if flow is not None]
+    stack.push(ClarifyPatternFlowStackFrame(names=flow_names))
+
+
 def trigger_pattern_completed(
     current_frame: DialogueStackFrame, stack: DialogueStack, flows: FlowsList
 ) -> None:
@@ -282,6 +318,9 @@ def trigger_pattern_completed(
         or isinstance(current_frame, SearchPatternFlowStackFrame)
     ):
         completed_flow = current_frame.flow(flows)
+        if not completed_flow.run_pattern_completed:
+            return
+
         completed_flow_name = completed_flow.readable_name() if completed_flow else None
         stack.push(
             CompletedPatternFlowStackFrame(
@@ -316,7 +355,7 @@ def reset_scoped_slots(
     def _reset_slot(slot_name: Text, dialogue_tracker: DialogueStateTracker) -> None:
         slot = dialogue_tracker.slots.get(slot_name, None)
         initial_value = slot.initial_value if slot else None
-        events.append(SlotSet(slot_name, initial_value))
+        events.append(SlotSet(slot_name, initial_value, metadata={"reset": True}))
 
     if (
         isinstance(current_frame, UserFlowStackFrame)
@@ -448,6 +487,7 @@ def advance_flows_until_next_action(
                     tracker,
                     available_actions,
                     flows,
+                    previous_step_id,
                 )
                 new_events = step_result.events
                 if (
@@ -464,6 +504,9 @@ def advance_flows_until_next_action(
                     new_events.insert(
                         idx, FlowCompleted(active_frame.flow_id, previous_step_id)
                     )
+                attach_stack_metadata_to_events(
+                    next_step.id, current_flow.id, new_events
+                )
                 tracker.update_stack(step_stack)
                 tracker.update_with_events(new_events)
 
@@ -535,36 +578,15 @@ def cancel_flow_and_push_internal_error(stack: DialogueStack, flow_name: str) ->
     stack.push(InternalErrorPatternFlowStackFrame())
 
 
-def validate_custom_slot_mappings(
-    step: CollectInformationFlowStep,
-    stack: DialogueStack,
-    tracker: DialogueStateTracker,
-    available_actions: List[str],
-    flow_name: str,
-) -> bool:
-    """Validate a slot with custom mappings.
-
-    If invalid, trigger pattern_internal_error and return False.
-    """
-    slot = tracker.slots.get(step.collect, None)
-    slot_mappings = slot.mappings if slot else []
-    for mapping in slot_mappings:
-        if (
-            mapping.get("type") == SlotMappingType.CUSTOM.value
-            and mapping.get("action") is None
-        ):
-            # this is a slot that must be filled by a custom action
-            # check if collect_action exists
-            if step.collect_action not in available_actions:
-                structlogger.error(
-                    "flow.step.run.collect_action_not_found_for_custom_slot_mapping",
-                    action=step.collect_action,
-                    collect=step.collect,
-                )
-                cancel_flow_and_push_internal_error(stack, flow_name)
-                return False
-
-    return True
+def attach_stack_metadata_to_events(
+    step_id: str,
+    flow_id: str,
+    events: List[Event],
+) -> None:
+    """Attach the stack metadata to the events."""
+    for event in events:
+        event.metadata[STEP_ID_METADATA_KEY] = step_id
+        event.metadata[ACTIVE_FLOW_METADATA_KEY] = flow_id
 
 
 def run_step(
@@ -574,6 +596,7 @@ def run_step(
     tracker: DialogueStateTracker,
     available_actions: List[str],
     flows: FlowsList,
+    previous_step_id: str,
 ) -> FlowStepResult:
     """Run a single step of a flow.
 
@@ -591,12 +614,19 @@ def run_step(
         tracker: The tracker to run the step on.
         available_actions: The actions that are available in the domain.
         flows: All flows.
+        previous_step_id: The ID of the previous step.
 
     Returns:
     A result of running the step describing where to transition to.
     """
     initial_events: List[Event] = []
-    if step == flow.first_step_in_flow():
+    if previous_step_id == START_STEP:
+        # if the previous step id is the start step, we need to add a flow
+        # started event to the initial events.
+        # we can't use the current step to check this, as the current step is the
+        # first step in the flow -> other steps might link to this flow, so the
+        # only reliable way to check if we are starting a new flow is checking for
+        # the START_STEP meta step
         initial_events.append(FlowStarted(flow.id, metadata=stack.current_context()))
 
     if isinstance(step, CollectInformationFlowStep):
@@ -645,7 +675,15 @@ def _run_end_step(
     structlogger.debug("flow.step.run.flow_end")
     current_frame = stack.pop()
     trigger_pattern_completed(current_frame, stack, flows)
-    resumed_events = trigger_pattern_continue_interrupted(current_frame, stack, flows)
+    resumed_events = []
+    if len(user_flows_on_the_stack(stack)) > 1:
+        # if there are more user flows on the stack,
+        # we need to trigger the pattern clarify
+        trigger_pattern_clarification(current_frame, stack, flows)
+    else:
+        resumed_events = trigger_pattern_continue_interrupted(
+            current_frame, stack, flows, tracker
+        )
     reset_events: List[Event] = reset_scoped_slots(current_frame, flow, tracker)
     return ContinueFlowWithNextStep(
         events=initial_events + reset_events + resumed_events, has_flow_ended=True
@@ -736,14 +774,6 @@ def _run_collect_information_step(
         # if we return any other FlowStepResult, the assistant will stay silent
         # instead of triggering the internal error pattern
         return ContinueFlowWithNextStep(events=initial_events)
-    is_mapping_valid = validate_custom_slot_mappings(
-        step, stack, tracker, available_actions, flow_name
-    )
-
-    if not is_mapping_valid:
-        # if we return any other FlowStepResult, the assistant will stay silent
-        # instead of triggering the internal error pattern
-        return ContinueFlowWithNextStep(events=initial_events)
 
     structlogger.debug("flow.step.run.collect")
     trigger_pattern_ask_collect_information(

rasa/core/policies/intentless_policy.py

@@ -1,7 +1,7 @@
 import importlib.resources
 import math
 from dataclasses import dataclass, field
-from typing import Any, Dict, List, Optional, Set, TYPE_CHECKING, Text, Tuple
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Set, Text, Tuple
 
 import structlog
 import tiktoken
@@ -30,19 +30,18 @@ from rasa.engine.storage.storage import ModelStorage
 from rasa.graph_components.providers.forms_provider import Forms
 from rasa.graph_components.providers.responses_provider import Responses
 from rasa.shared.constants import (
-    REQUIRED_SLOTS_KEY,
     EMBEDDINGS_CONFIG_KEY,
     LLM_CONFIG_KEY,
     MODEL_CONFIG_KEY,
+    MODEL_GROUP_ID_CONFIG_KEY,
     MODEL_NAME_CONFIG_KEY,
+    OPENAI_PROVIDER,
     PROMPT_CONFIG_KEY,
     PROVIDER_CONFIG_KEY,
-    OPENAI_PROVIDER,
+    REQUIRED_SLOTS_KEY,
     TIMEOUT_CONFIG_KEY,
-    MODEL_GROUP_ID_CONFIG_KEY,
 )
-from rasa.shared.core.constants import ACTION_LISTEN_NAME
-from rasa.shared.core.constants import ACTION_TRIGGER_CHITCHAT
+from rasa.shared.core.constants import ACTION_LISTEN_NAME, ACTION_TRIGGER_CHITCHAT
 from rasa.shared.core.domain import KEY_RESPONSES_TEXT, Domain
 from rasa.shared.core.events import (
     ActionExecuted,
@@ -75,9 +74,9 @@ from rasa.shared.utils.llm import (
     embedder_factory,
     get_prompt_template,
     llm_factory,
+    resolve_model_client_config,
     sanitize_message_for_prompt,
     tracker_as_readable_transcript,
-    resolve_model_client_config,
 )
 from rasa.utils.log_utils import log_llm
 from rasa.utils.ml_utils import (

rasa/core/policies/memoization.py

@@ -1,39 +1,41 @@
 from __future__ import annotations
-import copy
-import zlib
 
 import base64
+import copy
 import json
 import logging
-import structlog
+import zlib
+from pathlib import Path
+from typing import Any, Dict, List, Optional, Text
 
+import structlog
 from tqdm import tqdm
-from typing import Optional, Any, Dict, List, Text
-from pathlib import Path
 
-import rasa.utils.io
 import rasa.shared.utils.io
+import rasa.utils.io
+from rasa.core.constants import (
+    DEFAULT_MAX_HISTORY,
+    MEMOIZATION_POLICY_PRIORITY,
+    POLICY_MAX_HISTORY,
+    POLICY_PRIORITY,
+)
+from rasa.core.featurizers.tracker_featurizers import (
+    FEATURIZER_FILE,
+    MaxHistoryTrackerFeaturizer,
+    TrackerFeaturizer,
+)
+from rasa.core.policies.policy import Policy, PolicyPrediction, SupportedData
 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.core.domain import State, Domain
+from rasa.shared.core.constants import ACTION_LISTEN_NAME
+from rasa.shared.core.domain import Domain, State
 from rasa.shared.core.events import ActionExecuted
-from rasa.core.featurizers.tracker_featurizers import TrackerFeaturizer
-from rasa.core.featurizers.tracker_featurizers import MaxHistoryTrackerFeaturizer
-from rasa.core.featurizers.tracker_featurizers import FEATURIZER_FILE
-from rasa.shared.exceptions import FileIOException
-from rasa.core.policies.policy import PolicyPrediction, Policy, SupportedData
-from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.core.generator import TrackerWithCachedStates
+from rasa.shared.core.trackers import DialogueStateTracker
+from rasa.shared.exceptions import FileIOException
 from rasa.shared.utils.io import is_logging_disabled
-from rasa.core.constants import (
-    MEMOIZATION_POLICY_PRIORITY,
-    DEFAULT_MAX_HISTORY,
-    POLICY_MAX_HISTORY,
-    POLICY_PRIORITY,
-)
-from rasa.shared.core.constants import ACTION_LISTEN_NAME
 
 logger = logging.getLogger(__name__)
 structlogger = structlog.get_logger()