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

rasa/core/nlg/generator.py

@@ -1,12 +1,12 @@
 import logging
-from typing import List, Optional, Union, Text, Any, Dict
+from typing import Any, Dict, List, Optional, Text, Union
 
 import rasa.shared.utils.common
 import rasa.shared.utils.io
 from rasa.shared.constants import CHANNEL, RESPONSE_CONDITION
 from rasa.shared.core.domain import Domain
-from rasa.utils.endpoints import EndpointConfig
 from rasa.shared.core.trackers import DialogueStateTracker
+from rasa.utils.endpoints import EndpointConfig
 
 logger = logging.getLogger(__name__)
 

rasa/core/nlg/interpolator.py

@@ -1,10 +1,11 @@
 import copy
-import re
 import logging
-from jinja2 import Template
+import re
+from typing import Any, Dict, List, Text, Union
+
 import jinja2
 import structlog
-from typing import Text, Dict, Union, Any, List
+from jinja2 import Template
 
 from rasa.core.constants import JINJA2_TEMPLATE_ENGINE, RASA_FORMAT_TEMPLATE_ENGINE
 

rasa/core/nlg/response.py

@@ -1,14 +1,13 @@
 import copy
 import logging
-from rasa.core.constants import DEFAULT_TEMPLATE_ENGINE, TEMPLATE_ENGINE_CONFIG_KEY
-
-from rasa.shared.core.trackers import DialogueStateTracker
-from typing import Text, Any, Dict, Optional, List
+from typing import Any, Dict, List, Optional, Text
 
+from rasa.core.constants import DEFAULT_TEMPLATE_ENGINE, TEMPLATE_ENGINE_CONFIG_KEY
 from rasa.core.nlg import interpolator
 from rasa.core.nlg.generator import NaturalLanguageGenerator, ResponseVariationFilter
 from rasa.shared.constants import RESPONSE_CONDITION
 from rasa.shared.core.domain import RESPONSE_KEYS_TO_INTERPOLATE
+from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.nlu.constants import METADATA
 
 logger = logging.getLogger(__name__)

rasa/core/nlg/summarize.py

@@ -2,6 +2,7 @@ from typing import Optional
 
 import structlog
 from jinja2 import Template
+
 from rasa.core.tracker_store import DialogueStateTracker
 from rasa.shared.providers.llm.llm_client import LLMClient
 from rasa.shared.utils.llm import (

rasa/core/persistor.py

@@ -9,14 +9,13 @@ from typing import TYPE_CHECKING, List, Optional, Text, Tuple, Union
 
 import structlog
 
-from rasa.exceptions import ModelNotFound
 import rasa.shared.utils.common
 import rasa.utils.common
 from rasa.constants import (
+    DEFAULT_BUCKET_NAME,
     HTTP_STATUS_FORBIDDEN,
     HTTP_STATUS_NOT_FOUND,
     MODEL_ARCHIVE_EXTENSION,
-    DEFAULT_BUCKET_NAME,
 )
 from rasa.env import (
     AWS_ENDPOINT_URL_ENV,
@@ -26,6 +25,7 @@ from rasa.env import (
     BUCKET_NAME_ENV,
     REMOTE_STORAGE_PATH_ENV,
 )
+from rasa.exceptions import ModelNotFound
 from rasa.shared.exceptions import RasaException
 from rasa.shared.utils.io import raise_warning
 
@@ -367,8 +367,8 @@ class GCSPersistor(Persistor):
         self.bucket = self.storage_client.bucket(bucket_name)
 
     def _ensure_bucket_exists(self, bucket_name: Text) -> None:
-        from google.cloud import exceptions
         from google.auth import exceptions as auth_exceptions
+        from google.cloud import exceptions
 
         try:
             self.storage_client.get_bucket(bucket_name)

rasa/core/policies/ensemble.py

@@ -1,22 +1,23 @@
 from __future__ import annotations
-from abc import abstractmethod, ABC
-from typing import Optional, Text, List, Dict, Any
+
 import logging
+from abc import ABC, abstractmethod
+from typing import Any, Dict, List, Optional, Text
 
+from rasa.core.policies.policy import PolicyPrediction
 from rasa.engine.graph import GraphComponent
-from rasa.engine.storage.storage import ModelStorage
-from rasa.engine.storage.resource import Resource
 from rasa.engine.runner.interface import ExecutionContext
-from rasa.core.policies.policy import PolicyPrediction
-from rasa.shared.exceptions import RasaException, InvalidConfigException
+from rasa.engine.storage.resource import Resource
+from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.core.constants import ACTION_LISTEN_NAME
 from rasa.shared.core.domain import Domain
 from rasa.shared.core.events import (
-    ActionExecutionRejected,
     ActionExecuted,
+    ActionExecutionRejected,
     DefinePrevUserUtteredFeaturization,
 )
 from rasa.shared.core.trackers import DialogueStateTracker
+from rasa.shared.exceptions import InvalidConfigException, RasaException
 
 logger = logging.getLogger(__name__)
 
@@ -34,11 +35,11 @@ def is_not_in_training_data(
         `False` if and only if an action was predicted (i.e. `max_confidence` > 0) by
         a `MemoizationPolicy`
     """
-    from rasa.core.policies.rule_policy import RulePolicy
     from rasa.core.policies.memoization import (
-        MemoizationPolicy,
         AugmentedMemoizationPolicy,
+        MemoizationPolicy,
     )
+    from rasa.core.policies.rule_policy import RulePolicy
 
     if not policy_name:
         return True

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
 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)
 
@@ -581,10 +641,18 @@ class EnterpriseSearchPolicy(LLMHealthCheckMixin, EmbeddingsHealthCheckMixin, Po
 
     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 +660,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 +758,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/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,15 +9,15 @@ 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,
 )
@@ -67,21 +67,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.flows.steps.constants import START_STEP
 from rasa.shared.core.slots import Slot
 from rasa.shared.core.trackers import (
     DialogueStateTracker,
@@ -316,7 +317,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 +449,7 @@ def advance_flows_until_next_action(
                     tracker,
                     available_actions,
                     flows,
+                    previous_step_id,
                 )
                 new_events = step_result.events
                 if (
@@ -464,6 +466,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)
 
@@ -567,6 +572,17 @@ def validate_custom_slot_mappings(
     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(
     step: FlowStep,
     flow: Flow,
@@ -574,6 +590,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 +608,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):

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()