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

rasa/core/tracker_store.py

@@ -1,28 +1,29 @@
 from __future__ import annotations
+
 import contextlib
 import itertools
 import json
-import logging
 import os
 from inspect import isawaitable, iscoroutinefunction
-
 from time import sleep
 from typing import (
+    TYPE_CHECKING,
     Any,
     Callable,
     Dict,
+    Generator,
+    Generic,
     Iterable,
     Iterator,
     List,
     Optional,
     Text,
-    Union,
-    TYPE_CHECKING,
-    Generator,
     TypeVar,
-    Generic,
+    Union,
 )
 
+import sqlalchemy as sa
+import structlog
 from boto3.dynamodb.conditions import Key
 from pymongo.collection import Collection
 
@@ -30,17 +31,18 @@ import rasa.shared.utils.cli
 import rasa.shared.utils.common
 import rasa.shared.utils.io
 import rasa.utils.json_utils
-from rasa.plugin import plugin_manager
-from rasa.shared.core.constants import ACTION_LISTEN_NAME
+from rasa.constants import DEFAULT_SANIC_WORKERS, ENV_SANIC_WORKERS
 from rasa.core.brokers.broker import EventBroker
 from rasa.core.constants import (
-    POSTGRESQL_SCHEMA,
     POSTGRESQL_MAX_OVERFLOW,
     POSTGRESQL_POOL_SIZE,
+    POSTGRESQL_SCHEMA,
 )
+from rasa.plugin import plugin_manager
+from rasa.shared.core.constants import ACTION_LISTEN_NAME
 from rasa.shared.core.conversation import Dialogue
 from rasa.shared.core.domain import Domain
-from rasa.shared.core.events import SessionStarted, Event
+from rasa.shared.core.events import Event, SessionStarted
 from rasa.shared.core.trackers import (
     ActionExecuted,
     DialogueStateTracker,
@@ -50,16 +52,15 @@ from rasa.shared.core.trackers import (
 from rasa.shared.exceptions import ConnectionException, RasaException
 from rasa.shared.nlu.constants import INTENT_NAME_KEY
 from rasa.utils.endpoints import EndpointConfig
-import sqlalchemy as sa
 
 if TYPE_CHECKING:
     import boto3.resources.factory.dynamodb.Table
-    from sqlalchemy.engine.url import URL
-    from sqlalchemy.engine.base import Engine
-    from sqlalchemy.orm import Session, Query
     from sqlalchemy import Sequence
+    from sqlalchemy.engine.base import Engine
+    from sqlalchemy.engine.url import URL
+    from sqlalchemy.orm import Query, Session
 
-logger = logging.getLogger(__name__)
+structlogger = structlog.get_logger(__name__)
 
 # default values of PostgreSQL pool size and max overflow
 POSTGRESQL_DEFAULT_MAX_OVERFLOW = 100
@@ -158,9 +159,9 @@ class TrackerStore:
         if isinstance(obj, TrackerStore):
             return obj
 
-        from botocore.exceptions import BotoCoreError
         import pymongo.errors
         import sqlalchemy.exc
+        from botocore.exceptions import BotoCoreError
 
         try:
             _tracker_store = plugin_manager().hook.create_tracker_store(
@@ -315,7 +316,10 @@ class TrackerStore:
     async def stream_events(self, tracker: DialogueStateTracker) -> None:
         """Streams events to a message broker."""
         if self.event_broker is None:
-            logger.debug("No event broker configured. Skipping streaming events.")
+            structlogger.debug(
+                "tracker_store.stream_events.no_broker_configured",
+                event_info="No event broker configured. Skipping streaming events.",
+            )
             return None
 
         old_tracker = await self.retrieve(tracker.sender_id)
@@ -437,13 +441,22 @@ class InMemoryTrackerStore(TrackerStore, SerializedTrackerAsText):
             fetch_all_sessions: Whether to fetch all sessions or only the last one.
         """
         if sender_id not in self.store:
-            logger.debug(f"Could not find tracker for conversation ID '{sender_id}'.")
+            structlogger.debug(
+                "in_memory_tracker_store.retrieve.no_tracker_for_sender_id",
+                event_info=f"Could not find tracker for conversation ID '{sender_id}'.",
+            )
             return None
 
         tracker = self.deserialise_tracker(sender_id, self.store[sender_id])
 
         if not tracker:
-            logger.debug(f"Could not find tracker for conversation ID '{sender_id}'.")
+            structlogger.debug(
+                "in_memory_tracker_store.retrieve.failed_to_deserialize_tracker",
+                event_info=(
+                    f"Could not deserialize tracker "
+                    f"for conversation ID '{sender_id}'.",
+                ),
+            )
             return None
 
         if fetch_all_sessions:
@@ -499,7 +512,10 @@ class RedisTrackerStore(TrackerStore, SerializedTrackerAsText):
 
         self.key_prefix = DEFAULT_REDIS_TRACKER_STORE_KEY_PREFIX
         if key_prefix:
-            logger.debug(f"Setting non-default redis key prefix: '{key_prefix}'.")
+            structlogger.debug(
+                "redis_tracker_store.init.custom_key_prefix",
+                event_info=f"Setting non-default redis key prefix: '{key_prefix}'.",
+            )
             self._set_key_prefix(key_prefix)
 
         super().__init__(domain, event_broker, **kwargs)
@@ -508,9 +524,13 @@ class RedisTrackerStore(TrackerStore, SerializedTrackerAsText):
         if isinstance(key_prefix, str) and key_prefix.isalnum():
             self.key_prefix = key_prefix + ":" + DEFAULT_REDIS_TRACKER_STORE_KEY_PREFIX
         else:
-            logger.warning(
-                f"Omitting provided non-alphanumeric redis key prefix: '{key_prefix}'. "
-                f"Using default '{self.key_prefix}' instead."
+            structlogger.warning(
+                "redis_tracker_store.init.invalid_key_prefix",
+                event_info=(
+                    f"Omitting provided non-alphanumeric "
+                    f"redis key prefix: '{key_prefix}'. "
+                    f"Using default '{self.key_prefix}' instead."
+                ),
             )
 
     def _get_key_prefix(self) -> Text:
@@ -576,7 +596,10 @@ class RedisTrackerStore(TrackerStore, SerializedTrackerAsText):
         """
         stored = self.red.get(self.key_prefix + sender_id)
         if stored is None:
-            logger.debug(f"Could not find tracker for conversation ID '{sender_id}'.")
+            structlogger.debug(
+                "redis_tracker_store.retrieve.no_tracker_for_sender_id",
+                event_info=f"Could not find tracker for conversation ID '{sender_id}'.",
+            )
             return None
 
         tracker = self.deserialise_tracker(sender_id, stored)
@@ -674,6 +697,31 @@ class DynamoTrackerStore(TrackerStore, SerializedTrackerAsDict):
         try:
             self.client.describe_table(TableName=table_name)
         except self.client.exceptions.ResourceNotFoundException:
+            sanic_workers_count = int(
+                os.environ.get(ENV_SANIC_WORKERS, DEFAULT_SANIC_WORKERS)
+            )
+
+            if sanic_workers_count > 1:
+                structlogger.error(
+                    "dynamo_tracker_store.table_creation_not_supported_in_multi_worker_mode",
+                    event_info=(
+                        "DynamoDB table creation is not "
+                        "supported in multi-worker mode. "
+                        "Table should already exist.",
+                    ),
+                )
+                raise RasaException(
+                    "DynamoDB table creation is not supported in "
+                    "case of multiple sanic workers. To create the table either "
+                    "run Rasa with a single worker or create the table manually."
+                    "Here are the defaults which can be used to "
+                    "create the table manually: "
+                    f"Table name: {table_name}, Primary key: sender_id, "
+                    f"key type `HASH`, attribute type `S` (String), "
+                    "Provisioned throughput: Read capacity units: 5, "
+                    "Write capacity units: 5"
+                )
+
             table = dynamo.create_table(
                 TableName=self.table_name,
                 KeySchema=[{"AttributeName": "sender_id", "KeyType": "HASH"}],
@@ -799,8 +847,8 @@ class MongoTrackerStore(TrackerStore, SerializedTrackerAsText):
         event_broker: Optional[EventBroker] = None,
         **kwargs: Dict[Text, Any],
     ) -> None:
-        from pymongo.database import Database
         from pymongo import MongoClient
+        from pymongo.database import Database
 
         self.client: MongoClient = MongoClient(
             host,
@@ -1001,7 +1049,10 @@ def create_engine_kwargs(url: Union[Text, "URL"]) -> Dict[Text, Any]:
     schema_name = os.environ.get(POSTGRESQL_SCHEMA)
 
     if schema_name:
-        logger.debug(f"Using PostgreSQL schema '{schema_name}'.")
+        structlogger.debug(
+            "postgresql_tracker_store.schema_name",
+            event_inf=f"Using PostgreSQL schema '{schema_name}'.",
+        )
         kwargs["connect_args"] = {"options": f"-csearch_path={schema_name}"}
 
     # pool_size and max_overflow can be set to control the number of
@@ -1114,7 +1165,10 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
 
         self.engine = sa.create_engine(engine_url, **create_engine_kwargs(engine_url))
 
-        logger.debug(f"Attempting to connect to database via '{self.engine.url!r}'.")
+        structlogger.debug(
+            "sql_tracker_store.connect_to_sql_database",
+            event_info=f"Attempting to connect to database via '{self.engine.url!r}'.",
+        )
 
         # Database might take a while to come up
         while True:
@@ -1133,7 +1187,11 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
                     # Several Rasa services started in parallel may attempt to
                     # create tables at the same time. That is okay so long as
                     # the first services finishes the table creation.
-                    logger.error(f"Could not create tables: {e}")
+                    structlogger.error(
+                        "sql_tracker_store.create_tables_failed",
+                        event_info="Could not create tables",
+                        exec_info=e,
+                    )
 
                 self.sessionmaker = sa.orm.session.sessionmaker(bind=self.engine)
                 break
@@ -1141,10 +1199,17 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
                 sqlalchemy.exc.OperationalError,
                 sqlalchemy.exc.IntegrityError,
             ) as error:
-                logger.warning(error)
+                structlogger.warning(
+                    "sql_tracker_store.initialisation_error",
+                    event_info="Failed to establish a connection to the SQL database. ",
+                    exc_info=error,
+                )
                 sleep(5)
 
-        logger.debug(f"Connection to SQL database '{db}' successful.")
+        structlogger.debug(
+            "sql_tracker_store.connected_to_sql_database",
+            event_info=f"Connection to SQL database '{db}' successful.",
+        )
 
         super().__init__(domain, event_broker, **kwargs)
 
@@ -1212,7 +1277,7 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
         """Creates database `db` and updates engine accordingly."""
         from sqlalchemy import create_engine
 
-        if not self.engine.dialect.name == "postgresql":
+        if self.engine.dialect.name != "postgresql":
             rasa.shared.utils.io.raise_warning(
                 "The parameter 'login_db' can only be used with a postgres database."
             )
@@ -1252,7 +1317,11 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
                     sqlalchemy.exc.ProgrammingError,
                     sqlalchemy.exc.IntegrityError,
                 ) as e:
-                    logger.error(f"Could not create database '{database_name}': {e}")
+                    structlogger.error(
+                        "sql_tracker_store.create_database_failed",
+                        event_info=f"Could not create database '{database_name}'",
+                        exec_info=e,
+                    )
 
     @contextlib.contextmanager
     def session_scope(self) -> Generator["Session", None, None]:
@@ -1316,15 +1385,21 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
             events = [json.loads(event.data) for event in serialised_events]
 
             if self.domain and len(events) > 0:
-                logger.debug(f"Recreating tracker from sender id '{sender_id}'")
+                structlogger.debug(
+                    "sql_tracker_store.recreating_tracker",
+                    event_info=f"Recreating tracker from sender id '{sender_id}'",
+                )
                 return DialogueStateTracker.from_dict(
                     sender_id, events, self.domain.slots
                 )
             else:
-                logger.debug(
-                    f"Can't retrieve tracker matching "
-                    f"sender id '{sender_id}' from SQL storage. "
-                    f"Returning `None` instead."
+                structlogger.debug(
+                    "sql_tracker_store._retrieve.no_tracker_for_sender_id",
+                    event_info=(
+                        f"Can't retrieve tracker matching "
+                        f"sender id '{sender_id}' from SQL storage. "
+                        f"Returning `None` instead.",
+                    ),
                 )
                 return None
 
@@ -1401,7 +1476,12 @@ class SQLTrackerStore(TrackerStore, SerializedTrackerAsText):
                 )
             session.commit()
 
-        logger.debug(f"Tracker with sender_id '{tracker.sender_id}' stored to database")
+        structlogger.debug(
+            "sql_tracker_store.save_tracker",
+            event_info=(
+                f"Tracker with sender_id " f"'{tracker.sender_id}' stored to database",
+            ),
+        )
 
     def _additional_events(
         self, session: "Session", tracker: DialogueStateTracker
@@ -1469,11 +1549,14 @@ class FailSafeTrackerStore(TrackerStore):
         if self._on_tracker_store_error:
             self._on_tracker_store_error(error)
         else:
-            logger.error(
-                f"Error happened when trying to save conversation tracker to "
-                f"'{self._tracker_store.__class__.__name__}'. Falling back to use "
-                f"the '{InMemoryTrackerStore.__name__}'. Please "
-                f"investigate the following error: {error}."
+            structlogger.error(
+                "fail_safe_tracker_store.tracker_store_error",
+                event_info=(
+                    f"Error happened when trying to save conversation tracker to "
+                    f"'{self._tracker_store.__class__.__name__}'. Falling back to use "
+                    f"the '{InMemoryTrackerStore.__name__}'. Please "
+                    f"investigate the following error: {error}."
+                ),
             )
 
     async def retrieve(self, sender_id: Text) -> Optional[DialogueStateTracker]:
@@ -1525,11 +1608,14 @@ class FailSafeTrackerStore(TrackerStore):
         if self._on_tracker_store_error:
             self._on_tracker_store_error(error)
         else:
-            logger.error(
-                f"Error happened when trying to retrieve conversation tracker from "
-                f"'{self._tracker_store.__class__.__name__}'. Falling back to use "
-                f"the '{InMemoryTrackerStore.__name__}'. Please "
-                f"investigate the following error: {error}."
+            structlogger.error(
+                "fail_safe_tracker_store.tracker_store_retrieve_error",
+                event_info=(
+                    f"Error happened when trying to retrieve conversation tracker from "
+                    f"'{self._tracker_store.__class__.__name__}'. Falling back to use "
+                    f"the '{InMemoryTrackerStore.__name__}'."
+                ),
+                exec_info=error,
             )
 
 
@@ -1574,7 +1660,10 @@ def _create_from_endpoint_config(
             domain, endpoint_config, event_broker
         )
 
-    logger.debug(f"Connected to {tracker_store.__class__.__name__}.")
+    structlogger.debug(
+        "tracker_store.create_tracker_store_from_endpoint_config",
+        eventi_info=f"Connected to {tracker_store.__class__.__name__}.",
+    )
 
     return tracker_store
 

rasa/core/train.py

@@ -2,7 +2,7 @@ import argparse
 import logging
 import os
 from pathlib import Path
-from typing import Dict, Optional, Text, List
+from typing import Dict, List, Optional, Text
 
 import rasa.shared.utils.io
 import rasa.utils.io

rasa/core/training/__init__.py

@@ -1,4 +1,4 @@
-from typing import Text, List, Optional, Union, TYPE_CHECKING
+from typing import TYPE_CHECKING, List, Optional, Text, Union
 
 if TYPE_CHECKING:
     from rasa.shared.core.domain import Domain
@@ -21,8 +21,8 @@ def extract_story_graph(
     Returns:
         The loaded training data as graph.
     """
-    from rasa.shared.core.training_data.structures import StoryGraph
     import rasa.shared.core.training_data.loading as core_loading
+    from rasa.shared.core.training_data.structures import StoryGraph
 
     story_steps = core_loading.load_data_from_resource(
         resource_name, domain, exclusion_percentage=exclusion_percentage

rasa/core/training/converters/responses_prefix_converter.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 from typing import Text
 
+from rasa.shared.constants import UTTER_PREFIX
 from rasa.shared.core.domain import Domain, InvalidDomain
 from rasa.shared.core.events import ActionExecuted
 from rasa.shared.core.training_data.story_reader.yaml_story_reader import (
@@ -9,10 +10,8 @@ from rasa.shared.core.training_data.story_reader.yaml_story_reader import (
 from rasa.shared.core.training_data.story_writer.yaml_story_writer import (
     YAMLStoryWriter,
 )
-from rasa.shared.constants import UTTER_PREFIX
 from rasa.utils.converter import TrainingDataConverter
 
-
 OBSOLETE_RESPOND_PREFIX = "respond_"
 
 

rasa/core/training/interactive.py

@@ -13,10 +13,10 @@ from typing import (
     Dict,
     List,
     Optional,
+    Set,
     Text,
     Tuple,
     Union,
-    Set,
     cast,
 )
 
@@ -50,28 +50,28 @@ from rasa.core import run, utils
 from rasa.core.constants import DEFAULT_SERVER_FORMAT, DEFAULT_SERVER_PORT
 from rasa.core.utils import AvailableEndpoints
 from rasa.shared.constants import (
-    INTENT_MESSAGE_PREFIX,
     DEFAULT_SENDER_ID,
-    UTTER_PREFIX,
     DOCS_URL_NLU_BASED_POLICIES,
+    INTENT_MESSAGE_PREFIX,
+    UTTER_PREFIX,
 )
 from rasa.shared.core.constants import (
-    USER_INTENT_RESTART,
     ACTION_LISTEN_NAME,
-    LOOP_NAME,
+    ACTION_UNLIKELY_INTENT_NAME,
     ACTIVE_LOOP,
+    LOOP_INTERRUPTED,
+    LOOP_NAME,
     LOOP_REJECTED,
     REQUESTED_SLOT,
-    LOOP_INTERRUPTED,
-    ACTION_UNLIKELY_INTENT_NAME,
+    USER_INTENT_RESTART,
 )
 from rasa.shared.core.domain import (
-    Domain,
-    KEY_INTENTS,
+    KEY_ACTIONS,
     KEY_ENTITIES,
+    KEY_INTENTS,
     KEY_RESPONSES,
-    KEY_ACTIONS,
     KEY_RESPONSES_TEXT,
+    Domain,
 )
 from rasa.shared.core.events import (
     ActionExecuted,
@@ -79,11 +79,11 @@ from rasa.shared.core.events import (
     BotUttered,
     Event,
     Restarted,
-    UserUttered,
     UserUtteranceReverted,
+    UserUttered,
 )
 from rasa.shared.core.generator import TrackerWithCachedStates
-from rasa.shared.core.trackers import EventVerbosity, DialogueStateTracker
+from rasa.shared.core.trackers import DialogueStateTracker, EventVerbosity
 from rasa.shared.core.training_data import visualization
 from rasa.shared.core.training_data.visualization import (
     VISUALIZATION_TEMPLATE_PATH,
@@ -91,7 +91,7 @@ from rasa.shared.core.training_data.visualization import (
 )
 from rasa.shared.exceptions import InvalidConfigException
 from rasa.shared.importers.rasa import TrainingDataImporter
-from rasa.shared.nlu.constants import TEXT, INTENT_NAME_KEY
+from rasa.shared.nlu.constants import INTENT_NAME_KEY, TEXT
 
 # noinspection PyProtectedMember
 from rasa.shared.nlu.training_data import loading

rasa/core/training/story_conflict.py

@@ -1,20 +1,19 @@
-from collections import defaultdict
-import logging
 import json
+import logging
+from collections import defaultdict
 from typing import DefaultDict, Dict, Generator, List, NamedTuple, Optional, Text, Tuple
 
 from rasa.core.featurizers.tracker_featurizers import MaxHistoryTrackerFeaturizer
+from rasa.nlu.tokenizers.tokenizer import Tokenizer
 from rasa.shared.core.constants import (
     ACTION_LISTEN_NAME,
-    PREVIOUS_ACTION,
     ACTION_UNLIKELY_INTENT_NAME,
+    PREVIOUS_ACTION,
     USER,
 )
 from rasa.shared.core.domain import Domain, State
 from rasa.shared.core.events import ActionExecuted, Event
 from rasa.shared.core.generator import TrackerWithCachedStates
-
-from rasa.nlu.tokenizers.tokenizer import Tokenizer
 from rasa.shared.nlu.constants import TEXT
 from rasa.shared.nlu.training_data.message import Message
 

rasa/core/training/training.py

@@ -1,15 +1,13 @@
 import dataclasses
-
-from typing import DefaultDict, Dict, List, Optional, Set, Text, TYPE_CHECKING
 from collections import defaultdict
+from typing import TYPE_CHECKING, DefaultDict, Dict, List, Optional, Set, Text
 
-from rasa.shared.core.events import ActionExecuted, UserUttered
-from rasa.shared.core.events import SlotSet, ActiveLoop
+from rasa.shared.core.events import ActionExecuted, ActiveLoop, SlotSet, UserUttered
 
 if TYPE_CHECKING:
     from rasa.shared.core.domain import Domain
-    from rasa.shared.core.trackers import DialogueStateTracker
     from rasa.shared.core.events import Event
+    from rasa.shared.core.trackers import DialogueStateTracker
 
 
 @dataclasses.dataclass

rasa/core/utils.py

@@ -1,23 +1,23 @@
-import structlog
 import logging
 import os
 from pathlib import Path
 from socket import SOCK_DGRAM, SOCK_STREAM
-from typing import Any, Dict, Optional, List, Set, TYPE_CHECKING, Text, Tuple, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Set, Text, Tuple, Union
 
 import numpy as np
+import structlog
 from sanic import Sanic
 
 import rasa.cli.utils as cli_utils
 import rasa.shared.utils.io
 from rasa.constants import DEFAULT_SANIC_WORKERS, ENV_SANIC_WORKERS
 from rasa.core.constants import (
-    DOMAIN_GROUND_TRUTH_METADATA_KEY,
-    UTTER_SOURCE_METADATA_KEY,
     ACTIVE_FLOW_METADATA_KEY,
+    DOMAIN_GROUND_TRUTH_METADATA_KEY,
     STEP_ID_METADATA_KEY,
+    UTTER_SOURCE_METADATA_KEY,
 )
-from rasa.core.lock_store import LockStore, RedisLockStore, InMemoryLockStore
+from rasa.core.lock_store import InMemoryLockStore, LockStore, RedisLockStore
 from rasa.shared.constants import DEFAULT_ENDPOINTS_PATH, TCP_PROTOCOL
 from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.utils.endpoints import (

rasa/core/visualize.py

@@ -3,9 +3,9 @@ import os
 from typing import Text
 
 from rasa import telemetry
+from rasa.shared.core.domain import Domain, InvalidDomain
 from rasa.shared.core.training_data import loading
 from rasa.shared.utils.cli import print_error
-from rasa.shared.core.domain import InvalidDomain, Domain
 
 logger = logging.getLogger(__name__)
 

rasa/dialogue_understanding/coexistence/intent_based_router.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 from typing import Any, Dict, List, Optional
 
 import structlog
@@ -6,8 +7,8 @@ import structlog
 from rasa.dialogue_understanding.coexistence.constants import (
     CALM_ENTRY,
     NLU_ENTRY,
-    STICKY,
     NON_STICKY,
+    STICKY,
 )
 from rasa.dialogue_understanding.commands import Command, SetSlotCommand
 from rasa.dialogue_understanding.commands.noop_command import NoopCommand
@@ -27,7 +28,6 @@ from rasa.shared.nlu.constants import COMMANDS, INTENT, INTENT_NAME_KEY
 from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.nlu.training_data.training_data import TrainingData
 
-
 structlogger = structlog.get_logger()
 
 

rasa/dialogue_understanding/coexistence/llm_based_router.py

@@ -10,8 +10,8 @@ import rasa.shared.utils.io
 from rasa.dialogue_understanding.coexistence.constants import (
     CALM_ENTRY,
     NLU_ENTRY,
-    STICKY,
     NON_STICKY,
+    STICKY,
 )
 from rasa.dialogue_understanding.commands import Command, SetSlotCommand
 from rasa.dialogue_understanding.commands.noop_command import NoopCommand
@@ -23,15 +23,15 @@ 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,
-    PROMPT_CONFIG_KEY,
-    PROVIDER_CONFIG_KEY,
     MODEL_CONFIG_KEY,
     OPENAI_PROVIDER,
+    PROMPT_CONFIG_KEY,
+    PROVIDER_CONFIG_KEY,
+    ROUTE_TO_CALM_SLOT,
     TIMEOUT_CONFIG_KEY,
 )
 from rasa.shared.core.trackers import DialogueStateTracker
-from rasa.shared.exceptions import InvalidConfigException, FileIOException
+from rasa.shared.exceptions import FileIOException, InvalidConfigException
 from rasa.shared.nlu.constants import COMMANDS, TEXT
 from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.nlu.training_data.training_data import TrainingData