rasa-pro 3.10.10__py3-none-any.whl → 3.10.11__py3-none-any.whl

rasa/cli/arguments/train.py

@@ -3,12 +3,12 @@ from typing import Union
 
 from rasa.cli.arguments.default_arguments import (
     add_config_param,
-    add_stories_param,
-    add_nlu_data_param,
-    add_out_param,
     add_domain_param,
     add_endpoint_param,
+    add_nlu_data_param,
+    add_out_param,
     add_remote_storage_param,
+    add_stories_param,
 )
 from rasa.graph_components.providers.training_tracker_provider import (
     TrainingTrackerProvider,
@@ -40,6 +40,12 @@ def set_train_arguments(parser: argparse.ArgumentParser) -> None:
         parser, help_text="Configuration file for the connectors as a yml file."
     )
     add_remote_storage_param(parser)
+    parser.add_argument(
+        "--remote-bot-config-path",
+        help="Path to the bot configuration file in the remote storage.",
+        required=False,
+        type=str,
+    )
 
 
 def set_train_core_arguments(parser: argparse.ArgumentParser) -> None:

rasa/cli/train.py

@@ -1,18 +1,22 @@
 import argparse
 import asyncio
+import os
 import sys
 from pathlib import Path
 from typing import Dict, List, Optional, Text, Union
 
 import structlog
+from tarsafe import TarSafe
 
 import rasa.cli.arguments.train as train_arguments
 import rasa.cli.utils
 import rasa.core.utils
 import rasa.utils.common
+from rasa.api import train as train_all
 from rasa.cli import SubParsersAction
 from rasa.core.nlg.generator import NaturalLanguageGenerator
 from rasa.core.train import do_compare_training
+from rasa.nlu.persistor import get_persistor
 from rasa.shared.constants import (
     CONFIG_MANDATORY_KEYS,
     CONFIG_MANDATORY_KEYS_CORE,
@@ -20,6 +24,7 @@ from rasa.shared.constants import (
     DEFAULT_DATA_PATH,
     DEFAULT_DOMAIN_PATHS,
 )
+from rasa.shared.exceptions import RasaException
 from rasa.shared.importers.importer import TrainingDataImporter
 
 structlogger = structlog.getLogger(__name__)
@@ -36,7 +41,7 @@ def add_subparser(
     """
     train_parser = subparsers.add_parser(
         "train",
-        help="Trains a Rasa model using your NLU data and stories.",
+        help="Trains a Rasa model using your CALM flows, NLU data and stories.",
         parents=parents,
         formatter_class=argparse.ArgumentDefaultsHelpFormatter,
     )
@@ -83,6 +88,37 @@ def _check_nlg_endpoint_validity(endpoint: Union[Path, str]) -> None:
         sys.exit(1)
 
 
+def retrieve_and_unpack_bot_config_from_remote_storage(
+    args: argparse.Namespace,
+) -> None:
+    """Retrieve and unpack bot config from remote storage.
+
+    Bot config is retrieved from remote storage and unpacked
+    to the current working directory.
+    """
+    persistor = get_persistor(args.remote_storage)
+    if persistor is None:
+        raise RasaException(
+            f"Could not find a persistor for "
+            f"the storage type '{args.remote_storage}'."
+        )
+
+    current_working_directory = os.getcwd()
+
+    persistor.retrieve(args.remote_bot_config_path, current_working_directory)
+
+    remote_bot_config_tar_file_name = os.path.basename(args.remote_bot_config_path)
+
+    with TarSafe.open(remote_bot_config_tar_file_name, "r:gz") as tar:
+        tar.extractall(path=current_working_directory)
+
+    structlogger.debug(
+        "rasa.train.retrieve_and_unpack_bot_config.remove_downloaded_archive",
+        training_data_path=args.remote_bot_config_path,
+    )
+    os.remove(Path(current_working_directory).joinpath(remote_bot_config_tar_file_name))
+
+
 def run_training(args: argparse.Namespace, can_exit: bool = False) -> Optional[Text]:
     """Trains a model.
 
@@ -94,7 +130,9 @@ def run_training(args: argparse.Namespace, can_exit: bool = False) -> Optional[T
     Returns:
         Path to a trained model or `None` if training was not successful.
     """
-    from rasa.api import train as train_all
+    # retrieve and unpack bot_config from remote storage
+    if hasattr(args, "remote_bot_config_path") and args.remote_bot_config_path:
+        retrieve_and_unpack_bot_config_from_remote_storage(args)
 
     domain = rasa.cli.utils.get_validated_path(
         args.domain, "domain", DEFAULT_DOMAIN_PATHS, none_is_valid=True

rasa/cli/utils.py

@@ -1,30 +1,32 @@
-import json
 import argparse
-import structlog
 import importlib
+import json
 import os
 import sys
 import time
 from pathlib import Path
 from types import FrameType
-from typing import Any, Dict, List, Optional, TYPE_CHECKING, Text, Union, overload
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text, Union, overload
+
 import randomname
+import structlog
 
 import rasa.shared.utils.cli
 import rasa.shared.utils.io
-from rasa.shared.importers.importer import TrainingDataImporter
+from rasa import telemetry
 from rasa.shared.constants import (
     ASSISTANT_ID_DEFAULT_VALUE,
     ASSISTANT_ID_KEY,
     DEFAULT_CONFIG_PATH,
 )
-from rasa import telemetry
+from rasa.shared.importers.importer import TrainingDataImporter
 from rasa.shared.utils.yaml import read_config_file
 from rasa.utils.io import write_yaml
 
 if TYPE_CHECKING:
     from questionary import Question
     from typing_extensions import Literal
+
     from rasa.validator import Validator
 
 structlogger = structlog.get_logger()

rasa/dialogue_understanding/generator/multi_step/multi_step_llm_command_generator.py

@@ -229,9 +229,9 @@ class MultiStepLLMCommandGenerator(LLMBasedCommandGenerator):
         commands: List[Command] = []
 
         slot_set_re = re.compile(
-            r"""SetSlot\((\"?[a-zA-Z_][a-zA-Z0-9_-]*?\"?), ?(.*)\)"""
+            r"""SetSlot\(['"]?([a-zA-Z_][a-zA-Z0-9_-]*)['"]?, ?['"]?(.*)['"]?\)"""
         )
-        start_flow_re = re.compile(r"StartFlow\(([a-zA-Z0-9_-]+?)\)")
+        start_flow_re = re.compile(r"StartFlow\(['\"]?([a-zA-Z0-9_-]+)['\"]?\)")
         change_flow_re = re.compile(r"ChangeFlow\(\)")
         cancel_flow_re = re.compile(r"CancelFlow\(\)")
         chitchat_re = re.compile(r"ChitChat\(\)")
@@ -280,9 +280,19 @@ class MultiStepLLMCommandGenerator(LLMBasedCommandGenerator):
                 commands.append(HumanHandoffCommand())
             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 options
+                    for flow in cleaned_options
                     if flow in flows.user_flow_ids
                     and flow not in user_flows_on_the_stack(tracker.stack)
                 ]
@@ -293,6 +303,13 @@ class MultiStepLLMCommandGenerator(LLMBasedCommandGenerator):
             elif change_flow_re.search(action):
                 commands.append(ChangeFlowCommand())
 
+        if not commands:
+            structlogger.debug(
+                "multi_step_llm_command_generator.parse_commands",
+                message="No commands were parsed from the LLM actions.",
+                actions=actions,
+            )
+
         return commands
 
     ### Helper methods

rasa/dialogue_understanding/generator/single_step/single_step_llm_command_generator.py

@@ -185,6 +185,12 @@ class SingleStepLLMCommandGenerator(LLMBasedCommandGenerator):
 
         if not commands:
             # no commands are parsed or there's an invalid command
+            structlogger.warning(
+                "single_step_llm_command_generator.predict_commands",
+                message="No commands were predicted as the LLM response could "
+                "not be parsed or the LLM responded with an invalid command."
+                "Returning a CannotHandleCommand instead.",
+            )
             commands = [CannotHandleCommand()]
 
         if tracker.has_coexistence_routing_slot:
@@ -285,14 +291,16 @@ class SingleStepLLMCommandGenerator(LLMBasedCommandGenerator):
 
         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_-]+?)\)")
+        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_, ]+)\)")
+        clarify_re = re.compile(r"Clarify\(([\"\'a-zA-Z0-9_, ]+)\)")
 
         for action in actions.strip().splitlines():
             if match := slot_set_re.search(action):
@@ -321,14 +329,31 @@ class SingleStepLLMCommandGenerator(LLMBasedCommandGenerator):
                 commands.append(HumanHandoffCommand())
             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 options if flow in flows.user_flow_ids
+                    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))
 
+        if not commands:
+            structlogger.debug(
+                "single_step_llm_command_generator.parse_commands",
+                message="No commands were parsed from the LLM actions.",
+                actions=actions,
+            )
+
         return commands
 
     @classmethod

rasa/e2e_test/e2e_test_runner.py

@@ -190,11 +190,11 @@ class E2ETestRunner:
                     error=f"Message handling timed out for user message '{step.text}'.",
                     exc_info=True,
                 )
-            except Exception:
+            except Exception as exc:
                 structlogger.error(
                     "e2e_test_runner.run_prediction_loop",
                     error=f"An exception occurred while handling "
-                    f"user message '{step.text}'.",
+                    f"user message '{step.text}'. Error: {exc}",
                 )
             tracker = await self.agent.tracker_store.retrieve(sender_id)  # type: ignore[assignment]
             turns[position], event_cursor = self.get_actual_step_output(

rasa/engine/storage/local_model_storage.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import logging
 import shutil
 import sys
 import tempfile
@@ -8,19 +7,21 @@ import uuid
 from contextlib import contextmanager
 from datetime import datetime
 from pathlib import Path
-from tarsafe import TarSafe
 from typing import Generator, Optional, Text, Tuple, Union
 
-import rasa.utils.common
+import structlog
+from tarsafe import TarSafe
+
+import rasa.model
 import rasa.shared.utils.io
-from rasa.engine.storage.storage import ModelMetadata, ModelStorage
+import rasa.utils.common
 from rasa.engine.graph import GraphModelConfiguration
 from rasa.engine.storage.resource import Resource
+from rasa.engine.storage.storage import ModelMetadata, ModelStorage
 from rasa.exceptions import UnsupportedModelVersionError
 from rasa.shared.core.domain import Domain
-import rasa.model
 
-logger = logging.getLogger(__name__)
+structlogger = structlog.get_logger()
 
 # Paths within model archive
 MODEL_ARCHIVE_COMPONENTS_DIR = "components"
@@ -86,7 +87,14 @@ class LocalModelStorage(ModelStorage):
             cls._extract_archive_to_directory(
                 model_archive_path, temporary_directory_path
             )
-            logger.debug(f"Extracted model to '{temporary_directory_path}'.")
+            structlogger.debug(
+                "local_model_storage.from_model_archive",
+                event_info=(
+                    f"Extracted model '{model_archive_path}' to "
+                    f"'{temporary_directory_path}'."
+                ),
+            )
+
             cls._initialize_model_storage_from_model_archive(
                 temporary_directory_path, storage_path
             )
@@ -142,6 +150,10 @@ class LocalModelStorage(ModelStorage):
         temporary_directory: Path, storage_path: Path
     ) -> None:
         for path in (temporary_directory / MODEL_ARCHIVE_COMPONENTS_DIR).glob("*"):
+            structlogger.debug(
+                "local_model_storage._initialize_model_storage_from_model_archive",
+                event_info=f"Moving '{path}' to '{storage_path}'.",
+            )
             shutil.move(str(path), str(storage_path))
 
     @staticmethod
@@ -155,7 +167,10 @@ class LocalModelStorage(ModelStorage):
     @contextmanager
     def write_to(self, resource: Resource) -> Generator[Path, None, None]:
         """Persists data for a resource (see parent class for full docstring)."""
-        logger.debug(f"Resource '{resource.name}' was requested for writing.")
+        structlogger.debug(
+            "local_model_storage.write_to.resource_write_requested",
+            event_info=f"Resource '{resource.name}' was requested for writing.",
+        )
         directory = self._directory_for_resource(resource)
 
         if not directory.exists():
@@ -163,7 +178,10 @@ class LocalModelStorage(ModelStorage):
 
         yield directory
 
-        logger.debug(f"Resource '{resource.name}' was persisted.")
+        structlogger.debug(
+            "local_model_storage.write_to.resource_persisted",
+            event_info=f"Resource '{resource.name}' was persisted.",
+        )
 
     def _directory_for_resource(self, resource: Resource) -> Path:
         return self._storage_path / resource.name
@@ -171,7 +189,10 @@ class LocalModelStorage(ModelStorage):
     @contextmanager
     def read_from(self, resource: Resource) -> Generator[Path, None, None]:
         """Provides the data of a `Resource` (see parent class for full docstring)."""
-        logger.debug(f"Resource '{resource.name}' was requested for reading.")
+        structlogger.debug(
+            "local_model_storage.read_from",
+            event_info=f"Resource '{resource.name}' was requested for reading.",
+        )
         directory = self._directory_for_resource(resource)
 
         if not directory.exists():
@@ -193,7 +214,12 @@ class LocalModelStorage(ModelStorage):
         domain: Domain,
     ) -> ModelMetadata:
         """Creates model package (see parent class for full docstring)."""
-        logger.debug(f"Start to created model package for path '{model_archive_path}'.")
+        structlogger.debug(
+            "local_model_storage.create_model_package.started",
+            event_info=(
+                f"Start to created model " f"package for path '{model_archive_path}'.",
+            ),
+        )
 
         with windows_safe_temporary_directory() as temp_dir:
             temporary_directory = Path(temp_dir)
@@ -214,7 +240,10 @@ class LocalModelStorage(ModelStorage):
             with TarSafe.open(model_archive_path, "w:gz") as tar:
                 tar.add(temporary_directory, arcname="")
 
-        logger.debug(f"Model package created in path '{model_archive_path}'.")
+        structlogger.debug(
+            "local_model_storage.create_model_package.finished",
+            event_info=f"Model package created in path '{model_archive_path}'.",
+        )
 
         return model_metadata
 

rasa/model_training.py

@@ -22,7 +22,7 @@ from rasa.engine.storage.local_model_storage import LocalModelStorage
 from rasa.engine.storage.storage import ModelStorage
 from rasa.engine.training.components import FingerprintStatus
 from rasa.engine.training.graph_trainer import GraphTrainer
-from rasa.nlu.persistor import StorageType
+from rasa.nlu.persistor import RemoteStorageType, StorageType
 from rasa.shared.core.domain import Domain
 from rasa.shared.core.events import SlotSet
 from rasa.shared.core.training_data.structures import StoryGraph
@@ -350,18 +350,25 @@ async def _train_graph(
             if remote_storage:
                 push_model_to_remote_storage(full_model_path, remote_storage)
                 full_model_path.unlink()
+                remote_storage_string = (
+                    remote_storage.value
+                    if isinstance(remote_storage, RemoteStorageType)
+                    else remote_storage
+                )
                 structlogger.info(
                     "model_training.train.finished_training",
                     event_info=(
                         f"Your Rasa model {model_name} is trained "
-                        f"and saved at remote storage provider '{remote_storage}'."
+                        f"and saved at remote storage provider "
+                        f"'{remote_storage_string}'."
                     ),
                 )
             else:
                 structlogger.info(
                     "model_training.train.finished_training",
                     event_info=(
-                        f"Your Rasa model is trained and saved at '{full_model_path}'."
+                        f"Your Rasa model is trained and saved at "
+                        f"'{full_model_path}'."
                     ),
                 )
 

rasa/nlu/persistor.py

@@ -82,16 +82,14 @@ def get_persistor(storage: StorageType) -> Optional[Persistor]:
     Currently, `aws`, `gcs`, `azure` and providing module paths are supported remote
     storages.
     """
-    storage = storage.value if isinstance(storage, RemoteStorageType) else storage
-
-    if storage == RemoteStorageType.AWS.value:
+    if storage == RemoteStorageType.AWS:
         return AWSPersistor(
             os.environ.get(BUCKET_NAME_ENV), os.environ.get(AWS_ENDPOINT_URL_ENV)
         )
-    if storage == RemoteStorageType.GCS.value:
+    if storage == RemoteStorageType.GCS:
         return GCSPersistor(os.environ.get(BUCKET_NAME_ENV))
 
-    if storage == RemoteStorageType.AZURE.value:
+    if storage == RemoteStorageType.AZURE:
         return AzurePersistor(
             os.environ.get(AZURE_CONTAINER_ENV),
             os.environ.get(AZURE_ACCOUNT_NAME_ENV),
@@ -125,24 +123,36 @@ class Persistor(abc.ABC):
         """Downloads a model that has been persisted to cloud storage.
 
         Downloaded model will be saved to the `target_path`.
-        If `target_path` is a directory, the model will be saved to that directory.
-        If `target_path` is a file, the model will be saved to that file.
+        If `target_path` is a directory, the model will be downloaded to that directory.
+        If `target_path` is a file, the model will be downloaded to that file.
 
         Args:
             model_name: The name of the model to retrieve.
-            target_path: The path to which the model should be saved.
+            target_path: The path to which the model should be downloaded.
         """
         tar_name = model_name
         if not model_name.endswith(MODEL_ARCHIVE_EXTENSION):
             # ensure backward compatibility
             tar_name = self._tar_name(model_name)
-        tar_name = self._create_file_key(tar_name)
-        self._retrieve_tar(tar_name)
-        self._copy(os.path.basename(tar_name), target_path)
+        remote_object_path = self._create_file_key(tar_name)
+        self._retrieve_tar(remote_object_path)
 
+        target_tar_file_name = os.path.basename(tar_name)
         if os.path.isdir(target_path):
-            return os.path.join(target_path, model_name)
+            target_path = os.path.join(target_path, target_tar_file_name)
+
+        if not os.path.exists(target_path):
+            structlogger.debug(
+                "persistor.retrieve.copy_model",
+                event_info=f"Copying model '{target_tar_file_name}' to "
+                f"'{target_path}'.",
+            )
+            self._copy(target_tar_file_name, target_path)
 
+        structlogger.debug(
+            "persistor.retrieve.model_retrieved",
+            event_info=f"Model retrieved and saved to '{target_path}'.",
+        )
         return target_path
 
     @abc.abstractmethod
@@ -262,6 +272,11 @@ class AWSPersistor(Persistor):
 
     def _persist_tar(self, file_key: Text, tar_path: Text) -> None:
         """Uploads a model persisted in the `target_dir` to s3."""
+        structlogger.debug(
+            "aws_persistor.persist_tar.uploading_model",
+            event_info=f"Uploading tar archive {file_key} to "
+            f"s3 bucket '{self.bucket_name}'.",
+        )
         with open(tar_path, "rb") as f:
             self.s3.Object(self.bucket_name, file_key).put(Body=f)
 
@@ -329,7 +344,7 @@ class GCSPersistor(Persistor):
     def _retrieve_tar(self, target_filename: Text) -> None:
         """Downloads a model that has previously been persisted to GCS."""
         blob = self.bucket.blob(target_filename)
-        blob.download_to_filename(target_filename)
+        blob.download_to_filename(os.path.basename(target_filename))
 
 
 class AzurePersistor(Persistor):
@@ -370,12 +385,19 @@ class AzurePersistor(Persistor):
     def _persist_tar(self, file_key: Text, tar_path: Text) -> None:
         """Uploads a model persisted in the `target_dir` to Azure."""
         with open(tar_path, "rb") as data:
-            self._container_client().upload_blob(name=file_key, data=data)
+            self._container_client().upload_blob(
+                name=file_key,
+                data=data,
+                # overwrite is set to True to keep in line with
+                # how GCS and AWS APIs work this enables easy
+                # updating of models in the cloud
+                overwrite=True,
+            )
 
     def _retrieve_tar(self, target_filename: Text) -> None:
         """Downloads a model that has previously been persisted to Azure."""
         blob_client = self._container_client().get_blob_client(target_filename)
 
-        with open(target_filename, "wb") as blob:
+        with open(os.path.basename(target_filename), "wb") as blob:
             download_stream = blob_client.download_blob()
             blob.write(download_stream.readall())

rasa/shared/constants.py

@@ -111,7 +111,10 @@ CONFIG_KEYS_NLU = ["language", "pipeline"] + CONFIG_MANDATORY_COMMON_KEYS
 CONFIG_KEYS = CONFIG_KEYS_CORE + CONFIG_KEYS_NLU
 CONFIG_MANDATORY_KEYS_CORE: List[Text] = [] + CONFIG_MANDATORY_COMMON_KEYS
 CONFIG_MANDATORY_KEYS_NLU = ["language"] + CONFIG_MANDATORY_COMMON_KEYS
-CONFIG_MANDATORY_KEYS = CONFIG_MANDATORY_KEYS_CORE + CONFIG_MANDATORY_KEYS_NLU
+# we need the list to contain unique values
+CONFIG_MANDATORY_KEYS = list(
+    set(CONFIG_MANDATORY_KEYS_CORE + CONFIG_MANDATORY_KEYS_NLU)
+)
 
 # Keys related to Forms (in the Domain)
 REQUIRED_SLOTS_KEY = "required_slots"

rasa/shared/importers/importer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from abc import ABC, abstractmethod
 from functools import reduce
@@ -114,7 +116,7 @@ class TrainingDataImporter(ABC):
         domain_path: Optional[Text] = None,
         training_data_paths: Optional[List[Text]] = None,
         args: Optional[Dict[Text, Any]] = {},
-    ) -> "TrainingDataImporter":
+    ) -> TrainingDataImporter:
         """Loads a `TrainingDataImporter` instance from a configuration file."""
         config = read_config_file(config_path)
         return TrainingDataImporter.load_from_dict(
@@ -127,7 +129,7 @@ class TrainingDataImporter(ABC):
         domain_path: Optional[Text] = None,
         training_data_paths: Optional[List[Text]] = None,
         args: Optional[Dict[Text, Any]] = {},
-    ) -> "TrainingDataImporter":
+    ) -> TrainingDataImporter:
         """Loads core `TrainingDataImporter` instance.
 
         Instance loaded from configuration file will only read Core training data.
@@ -143,7 +145,7 @@ class TrainingDataImporter(ABC):
         domain_path: Optional[Text] = None,
         training_data_paths: Optional[List[Text]] = None,
         args: Optional[Dict[Text, Any]] = {},
-    ) -> "TrainingDataImporter":
+    ) -> TrainingDataImporter:
         """Loads nlu `TrainingDataImporter` instance.
 
         Instance loaded from configuration file will only read NLU training data.
@@ -166,7 +168,7 @@ class TrainingDataImporter(ABC):
         domain_path: Optional[Text] = None,
         training_data_paths: Optional[List[Text]] = None,
         args: Optional[Dict[Text, Any]] = None,
-    ) -> "TrainingDataImporter":
+    ) -> TrainingDataImporter:
         """Loads a `TrainingDataImporter` instance from a dictionary."""
         from rasa.shared.importers.rasa import RasaFileImporter
 
@@ -195,18 +197,15 @@ class TrainingDataImporter(ABC):
         domain_path: Optional[Text] = None,
         training_data_paths: Optional[List[Text]] = None,
         args: Optional[Dict[Text, Any]] = None,
-    ) -> Optional["TrainingDataImporter"]:
+    ) -> Optional[TrainingDataImporter]:
         from rasa.shared.importers.multi_project import MultiProjectImporter
         from rasa.shared.importers.rasa import RasaFileImporter
-        from rasa.shared.importers.remote_importer import RemoteTrainingDataImporter
 
         module_path = importer_config.pop("name", None)
         if module_path == RasaFileImporter.__name__:
             importer_class: Type[TrainingDataImporter] = RasaFileImporter
         elif module_path == MultiProjectImporter.__name__:
             importer_class = MultiProjectImporter
-        elif module_path == RemoteTrainingDataImporter.__name__:
-            importer_class = RemoteTrainingDataImporter
         else:
             try:
                 importer_class = rasa.shared.utils.common.class_from_module_path(

rasa/version.py

@@ -1,3 +1,3 @@
 # this file will automatically be changed,
 # do not add anything but the version number here!
-__version__ = "3.10.10"
+__version__ = "3.10.11"

{rasa_pro-3.10.10.dist-info → rasa_pro-3.10.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rasa-pro
-Version: 3.10.10
+Version: 3.10.11
 Summary: State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development.
 Home-page: https://rasa.com
 Keywords: nlp,machine-learning,machine-learning-library,bot,bots,botkit,rasa conversational-agents,conversational-ai,chatbot,chatbot-framework,bot-framework
@@ -103,7 +103,7 @@ Requires-Dist: pyyaml (>=6.0)
 Requires-Dist: qdrant-client (>=1.9.0,<2.0.0)
 Requires-Dist: questionary (>=1.10.0,<2.1.0)
 Requires-Dist: randomname (>=0.2.1,<0.3.0)
-Requires-Dist: rasa-sdk (==3.10.0)
+Requires-Dist: rasa-sdk (>=3.10.0,<3.11.0)
 Requires-Dist: redis (>=4.6.0,<6.0)
 Requires-Dist: regex (>=2022.10.31,<2022.11)
 Requires-Dist: requests (>=2.31.0,<2.32.0)

{rasa_pro-3.10.10.dist-info → rasa_pro-3.10.11.dist-info}/RECORD

@@ -19,7 +19,7 @@ rasa/cli/arguments/interactive.py,sha256=S-cY9XUOg7vzgVh_1mlQWObTmIO9pDQ0OdWVVCx
 rasa/cli/arguments/run.py,sha256=G7Ldj5pEsfhXMhhHOqTO1lgRG0939VZ6SRVpwBYaP_M,6340
 rasa/cli/arguments/shell.py,sha256=Vyt3XizJPxKAMo4NIcpdSOs73rD-q-yjWH1Qzc15xCs,367
 rasa/cli/arguments/test.py,sha256=2g6OvwkTlwCK85_nxYvKiFUMICBUMjfttd6Qif2Hme8,6854
-rasa/cli/arguments/train.py,sha256=gyqJysj0nydSdMQt8EWrjgmqDiSd898aFMT1Rh9A34k,8356
+rasa/cli/arguments/train.py,sha256=CMf6h9oCDYS29UXEdVDuSYCkLJTobURhplmvMW4wlNQ,8539
 rasa/cli/arguments/visualize.py,sha256=Su0qyXv4bEx5mrteRqEyf-K3JGQ1t2WCXOYlCpGYfAk,861
 rasa/cli/arguments/x.py,sha256=FQkarKvNBtzZ5xrPBhHWk-ZKPgEHvgE5ItwRL1TNR3I,1027
 rasa/cli/data.py,sha256=M33oHv3PCqDx4Gc5ifGmcZoEc6h2_uP8pRIygimjT8w,12695
@@ -85,8 +85,8 @@ rasa/cli/studio/train.py,sha256=ruXA5UkaRbO3femk8w3I2cXKs06-34FYtB_9MKdP6hw,1572
 rasa/cli/studio/upload.py,sha256=kdHqrVGsEbbqH5fz_HusWwJEycB31SHaPlXer8lXAE0,2069
 rasa/cli/telemetry.py,sha256=ZywhlOpp0l2Yz9oEcOGA2ej3SEkSTisKPpBhn_fS7tc,3538
 rasa/cli/test.py,sha256=Ub7Cm9rFQ_tkB310jPYzVwU0Di88Z7IE0nLi1o-aYbA,8901
-rasa/cli/train.py,sha256=X4q-ub66Jto8K2vW6g_AOk06SdC-DXC_Mnf6gMiR7lc,8514
-rasa/cli/utils.py,sha256=9VKC04qX9hJiMvQG9BWWJCH1Sb4jVNO0N_zE7oyUz1Y,15660
+rasa/cli/train.py,sha256=a8KB-zc9mFZrWYztpiC7g5Ab3O86KJcTlo_CBLyB7Tk,9934
+rasa/cli/utils.py,sha256=Q4WFdSYrqQvMY2nZY4i2P-vBimUh_cS08KEN-PGkJlg,15662
 rasa/cli/visualize.py,sha256=YmRAATAfxHpgE8_PknGyM-oIujwICNzVftTzz6iLNNc,1256
 rasa/cli/x.py,sha256=1w-H6kb_3OG3zVPJ1isX67BTb_T-x2MJo4OGffCD4Vc,6827
 rasa/constants.py,sha256=BuDQ59uM3GhxmShZR-1IuDHh5VHfEneA9P9HUQagZ9M,1311
@@ -364,11 +364,11 @@ rasa/dialogue_understanding/generator/llm_command_generator.py,sha256=P1LA0eJxf3
 rasa/dialogue_understanding/generator/multi_step/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/dialogue_understanding/generator/multi_step/fill_slots_prompt.jinja2,sha256=Y0m673tAML3cFPaLM-urMXDsBYUUcXIw9YUpkAhGUuA,2933
 rasa/dialogue_understanding/generator/multi_step/handle_flows_prompt.jinja2,sha256=8l93_QBKBYnqLICVdiTu5ejZDE8F36BU8-qwba0px44,1927
-rasa/dialogue_understanding/generator/multi_step/multi_step_llm_command_generator.py,sha256=JjNvD3Zp5_riFYdUuk4oerT9WDSm8uPKky8qeklJyyw,30818
+rasa/dialogue_understanding/generator/multi_step/multi_step_llm_command_generator.py,sha256=133WGlwv49PF4_DKM53u0J_pWrDs88dgUYc5fq1m6NQ,31568
 rasa/dialogue_understanding/generator/nlu_command_adapter.py,sha256=-oc3lxAdQEcX25f4IalxOhEw9XhA_6HxGpItlY1RUvY,8412
 rasa/dialogue_understanding/generator/single_step/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/dialogue_understanding/generator/single_step/command_prompt_template.jinja2,sha256=qVTuas5XgAv2M_hsihyXl-wAnBDEpg_uhVvNrR5m-h0,3751
-rasa/dialogue_understanding/generator/single_step/single_step_llm_command_generator.py,sha256=aCE3CP2jSW8Tn4iQcwoY8fZJjzgU9am2MO-ddjot4vo,14912
+rasa/dialogue_understanding/generator/single_step/single_step_llm_command_generator.py,sha256=lzSGEqsKtRPWK9hP3GmaZazN39icII-MZp_0sykvpNQ,16031
 rasa/dialogue_understanding/patterns/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/dialogue_understanding/patterns/cancel.py,sha256=IQ4GVHNnNCqwKRLlAqBtLsgolcbPPnHsHdb3aOAFhEs,3868
 rasa/dialogue_understanding/patterns/cannot_handle.py,sha256=pg0zJHl-hDBnl6y9IyxZzW57yuMdfD8xI8eiK6EVrG8,1406
@@ -410,7 +410,7 @@ rasa/e2e_test/e2e_test_converter.py,sha256=VxIx7uk36HzLIyEumJiR6G6-CyyqkV_lYoX-X
 rasa/e2e_test/e2e_test_converter_prompt.jinja2,sha256=EMy-aCd7jLARHmwAuZUGT5ABnNHjR872_pexRIMGA7c,2791
 rasa/e2e_test/e2e_test_coverage_report.py,sha256=Cv5bMtoOC240231YMNHz10ibSqm_UD1-eskQVdjPUsw,11326
 rasa/e2e_test/e2e_test_result.py,sha256=9LlH6vIQeK_dxDwMQ5RzlNHoxCNXpWC9S527-ch8kUA,1649
-rasa/e2e_test/e2e_test_runner.py,sha256=h5r5IQMCAjHILn75TQV4fr3jp-rpVQie50oF4o3DmZE,44001
+rasa/e2e_test/e2e_test_runner.py,sha256=o-vCDTMV3xoElNV-Cj2_vnuDwhTxnQBXdMjRE5z1cko,44021
 rasa/e2e_test/e2e_test_schema.yml,sha256=0deWjuKRHNo6e_LSCnUoiw9NLIYf6dj1-zFPl_AqLYA,5632
 rasa/e2e_test/pykwalify_extensions.py,sha256=OGYKIKYJXd2S0NrWknoQuijyBQaE-oMLkfV_eMRkGSM,1331
 rasa/e2e_test/stub_custom_action.py,sha256=teq8c5I6IuUsFX4lPdeBLY3j0SLSMCC95KmKx7GrE8I,2369
@@ -434,7 +434,7 @@ rasa/engine/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 rasa/engine/runner/dask.py,sha256=npWZXpYtYjdbQOH4OVCTsA2F0TNK43g8201y2wCS6U8,9540
 rasa/engine/runner/interface.py,sha256=4mbJSMbXPwnITDl5maYZjq8sUcbWw1ToGjYVQtnWCkc,1678
 rasa/engine/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rasa/engine/storage/local_model_storage.py,sha256=m8VpHEBoBy4_oNpCnR0O2cRJ1Tui1lMsU58hgzXg2Fc,9578
+rasa/engine/storage/local_model_storage.py,sha256=gybPSv4HuG6JdbU6Blxs3PEPTeY-VkOFyzWUJSGcCSM,10517
 rasa/engine/storage/resource.py,sha256=1ecgZbDw7y6CLLFLdi5cYRfdk0yTmD7V1seWEzdtzpU,3931
 rasa/engine/storage/storage.py,sha256=Qx1kp8LwX6Oe0W6rfTd0qzo9jWhNVttXWhVXRy_qW4w,6923
 rasa/engine/training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -483,7 +483,7 @@ rasa/markers/upload.py,sha256=Ot1s_O-CEIB9c4CKUlfOldiJo92pdqxFUHOPCU7E_NU,2518
 rasa/markers/validate.py,sha256=YypXKRS87xxrMMEz9HpAQzYJUwg0OPbczMXBRNjlJq4,709
 rasa/model.py,sha256=GH1-N6Po3gL3nwfa9eGoN2bMRNMrn4f3mi17-osW3T0,3491
 rasa/model_testing.py,sha256=h0QUpJu6p_TDse3aHjCfYwI6OGH47b3Iuo5Ot0HQADM,14959
-rasa/model_training.py,sha256=5NNOr5IJ6WMTx5ok0tL8EocByXpfaXUHJOXyxlm_TXQ,20339
+rasa/model_training.py,sha256=1LoW9TMgzpXeXedLtzxmGHxOl1NBTbNDaqLzu2XayxE,20631
 rasa/nlu/__init__.py,sha256=D0IYuTK_ZQ_F_9xsy0bXxVCAtU62Fzvp8S7J9tmfI_c,123
 rasa/nlu/classifiers/__init__.py,sha256=Qvrf7_rfiMxm2Vt2fClb56R3QFExf7WPdFdL-AOvgsk,118
 rasa/nlu/classifiers/classifier.py,sha256=9fm1mORuFf1vowYIXmqE9yLRKdSC4nGQW7UqNZQipKY,133
@@ -524,7 +524,7 @@ rasa/nlu/featurizers/sparse_featurizer/lexical_syntactic_featurizer.py,sha256=aw
 rasa/nlu/featurizers/sparse_featurizer/regex_featurizer.py,sha256=PhzJ17lNv3I5h8WrCvjzjjcUvbu_MJBxY6k3pQTDCac,10289
 rasa/nlu/featurizers/sparse_featurizer/sparse_featurizer.py,sha256=m6qpixorfTDFWSfGVmLImTOHM6zKdgydPaP_wVxCQ-w,220
 rasa/nlu/model.py,sha256=r6StZb4Dmum_3dRoocxZWo2M5KVNV20_yKNvYZNvpOc,557
-rasa/nlu/persistor.py,sha256=Sc0NH2VSK9efOYSD0INYd3za3esQvgNHa4FwClJVH-c,13788
+rasa/nlu/persistor.py,sha256=QniAoBRjk9CPzFJOdkJIpgc_eXXan1cKC61L_xWQjGk,14702
 rasa/nlu/run.py,sha256=WumXqNn2PEyab463geNnOu3IPwgaCtBai-x685BYCNw,803
 rasa/nlu/selectors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/nlu/selectors/response_selector.py,sha256=gwffu9zLniMseOzX-SuqaZ2CQiGi4JUbDcwUe-BsThI,39021
@@ -546,7 +546,7 @@ rasa/nlu/utils/spacy_utils.py,sha256=pBvsCVKVuZ3b2Pjn-XuOVZ6lzZu9Voc2R4N1VczwtCM
 rasa/plugin.py,sha256=H_OZcHy_U3eAK-JHr43TSxcPqS0JEGcZkFvmumeeJEs,2670
 rasa/server.py,sha256=pUdhi5dkhzEj4bngj2bTUKJohepjpY-aZ4MGKHYZRH0,57775
 rasa/shared/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rasa/shared/constants.py,sha256=WCq5MZibbolWulcnfV4w_yjbtx7lGe-Hhef01sf6wRk,9266
+rasa/shared/constants.py,sha256=MfO5vD2BRZMKWt_V3XiT7Yap1XhFWf5k13epuMG-P2s,9327
 rasa/shared/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/shared/core/command_payload_reader.py,sha256=Vhiop9LWFawaEruRifBBrVmoEJ-fj1Tli1wBvsYu2_I,3563
 rasa/shared/core/constants.py,sha256=U1q0-oGMQ7aDsP5p6H4CRaBi-H-KZpm2IXpTzIbaEv8,5169
@@ -597,10 +597,9 @@ rasa/shared/engine/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 rasa/shared/engine/caching.py,sha256=PiBV4vQ5c5z5sopxK1XpzmmaF4V42QQxP1_MX0Yw9PI,743
 rasa/shared/exceptions.py,sha256=bUTjjIH4PDykWJqApxYXdpPvfo03EVQ6xo1sOc3yjH4,5118
 rasa/shared/importers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rasa/shared/importers/importer.py,sha256=gZUOkSdjZcj5TigMcAd2g8UfeM1Q5fgGIhZpRUQMnbs,27370
+rasa/shared/importers/importer.py,sha256=wBHncehjhIdNpzdVaekKz2e7OJTJOWgWIdD2Vpqqv6o,27190
 rasa/shared/importers/multi_project.py,sha256=b5LypQf0wdiDaZyAihqItQXSuzDjMibypbUySDXff7g,7863
 rasa/shared/importers/rasa.py,sha256=9Ie86LgcS9ZtGWR5-NLylxSfK_I329SLvb_UTC71vu4,3853
-rasa/shared/importers/remote_importer.py,sha256=SCmwWDKb7GczVEJhAdH8_s2dq55P4f31qDwQ8y18ndk,7694
 rasa/shared/importers/utils.py,sha256=-EzZ0yH_YnJCSaC4JVx7xrRrXyr8CV6tAKlbASeMtJg,1242
 rasa/shared/nlu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/shared/nlu/constants.py,sha256=rf628BT4r6hnvN6QWyh_t2UFKOD7PR5APspi6igmeCU,1643
@@ -720,9 +719,9 @@ rasa/utils/train_utils.py,sha256=f1NWpp5y6al0dzoQyyio4hc4Nf73DRoRSHDzEK6-C4E,212
 rasa/utils/url_tools.py,sha256=JQcHL2aLqLHu82k7_d9imUoETCm2bmlHaDpOJ-dKqBc,1218
 rasa/utils/yaml.py,sha256=KjbZq5C94ZP7Jdsw8bYYF7HASI6K4-C_kdHfrnPLpSI,2000
 rasa/validator.py,sha256=ToRaa4dS859CJO3H2VGqS943O5qWOg45ypbDfFMKECU,62699
-rasa/version.py,sha256=XXyTkiKqj_NtVo-pisivHTuGq0D3-uIm_AB7GwyQE6g,118
-rasa_pro-3.10.10.dist-info/METADATA,sha256=NmyzTOEcUHg6Lm9zHbjQStpWyuChaZSCnhmjbQaD0R8,30893
-rasa_pro-3.10.10.dist-info/NOTICE,sha256=7HlBoMHJY9CL2GlYSfTQ-PZsVmLmVkYmMiPlTjhuCqA,218
-rasa_pro-3.10.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-rasa_pro-3.10.10.dist-info/entry_points.txt,sha256=ckJ2SfEyTPgBqj_I6vm_tqY9dZF_LAPJZA335Xp0Q9U,43
-rasa_pro-3.10.10.dist-info/RECORD,,
+rasa/version.py,sha256=0PycS7JWF6zF5RV14lJkREbKvYdYXHB_CbCKkQiYfP0,118
+rasa_pro-3.10.11.dist-info/METADATA,sha256=DoqzzlFIc1_OXg5TQimVBnOhWchKq-_c6vblqlOOeH0,30901
+rasa_pro-3.10.11.dist-info/NOTICE,sha256=7HlBoMHJY9CL2GlYSfTQ-PZsVmLmVkYmMiPlTjhuCqA,218
+rasa_pro-3.10.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+rasa_pro-3.10.11.dist-info/entry_points.txt,sha256=ckJ2SfEyTPgBqj_I6vm_tqY9dZF_LAPJZA335Xp0Q9U,43
+rasa_pro-3.10.11.dist-info/RECORD,,

rasa/shared/importers/remote_importer.py

@@ -1,196 +0,0 @@
-import os
-from typing import Dict, List, Optional, Text, Union
-
-import structlog
-from tarsafe import TarSafe
-
-import rasa.shared.core.flows.yaml_flows_io
-import rasa.shared.data
-import rasa.shared.utils.common
-import rasa.shared.utils.io
-from rasa.nlu.persistor import StorageType
-from rasa.shared.core.domain import Domain, InvalidDomain
-from rasa.shared.core.flows import FlowsList
-from rasa.shared.core.training_data.story_reader.yaml_story_reader import (
-    YAMLStoryReader,
-)
-from rasa.shared.core.training_data.structures import StoryGraph
-from rasa.shared.exceptions import RasaException
-from rasa.shared.importers import utils
-from rasa.shared.importers.importer import TrainingDataImporter
-from rasa.shared.nlu.training_data.training_data import TrainingData
-from rasa.shared.utils.yaml import read_model_configuration
-
-structlogger = structlog.get_logger()
-
-TRAINING_DATA_ARCHIVE = "training_data.tar.gz"
-
-
-class RemoteTrainingDataImporter(TrainingDataImporter):
-    """Remote `TrainingFileImporter` implementation.
-
-    Fetches training data from a remote storage and extracts it to a local directory.
-    Extracted training data is then used to load flows, NLU, stories,
-        domain, and config files.
-    """
-
-    def __init__(
-        self,
-        config_file: Optional[Text] = None,
-        domain_path: Optional[Text] = None,
-        training_data_paths: Optional[Union[List[Text], Text]] = None,
-        project_directory: Optional[Text] = None,
-        remote_storage: Optional[StorageType] = None,
-        training_data_path: Optional[Text] = None,
-    ):
-        """Initializes `RemoteTrainingDataImporter`.
-
-        Args:
-            config_file: Path to the model configuration file.
-            domain_path: Path to the domain file.
-            training_data_paths: List of paths to the training data files.
-            project_directory: Path to the project directory.
-            remote_storage: Storage to use to load the training data.
-            training_data_path: Path to the training data.
-        """
-        self.remote_storage = remote_storage
-        self.training_data_path = training_data_path
-
-        self.extracted_path = self._fetch_and_extract_training_archive(
-            TRAINING_DATA_ARCHIVE, self.training_data_path
-        )
-
-        self._nlu_files = rasa.shared.data.get_data_files(
-            self.extracted_path, rasa.shared.data.is_nlu_file
-        )
-        self._story_files = rasa.shared.data.get_data_files(
-            self.extracted_path, YAMLStoryReader.is_stories_file
-        )
-        self._flow_files = rasa.shared.data.get_data_files(
-            self.extracted_path, rasa.shared.core.flows.yaml_flows_io.is_flows_file
-        )
-        self._conversation_test_files = rasa.shared.data.get_data_files(
-            self.extracted_path, YAMLStoryReader.is_test_stories_file
-        )
-
-        self.config_file = config_file
-
-    def _fetch_training_archive(
-        self, training_file: str, training_data_path: Optional[str] = None
-    ) -> str:
-        """Fetches training files from remote storage."""
-        from rasa.nlu.persistor import get_persistor
-
-        persistor = get_persistor(self.remote_storage)
-        if persistor is None:
-            raise RasaException(
-                f"Could not find a persistor for "
-                f"the storage type '{self.remote_storage}'."
-            )
-
-        return persistor.retrieve(training_file, training_data_path)
-
-    def _fetch_and_extract_training_archive(
-        self, training_file: str, training_data_path: Optional[Text] = None
-    ) -> Optional[str]:
-        """Fetches and extracts training files from remote storage.
-
-        If the `training_data_path` is not provided, the training
-        data is extracted to the current working directory.
-
-        Args:
-            training_file: Name of the training data archive file.
-            training_data_path: Path to the training data.
-
-        Returns:
-            Path to the extracted training data.
-        """
-
-        if training_data_path is None:
-            training_data_path = os.path.join(os.getcwd(), "data")
-
-        if os.path.isfile(training_data_path):
-            raise ValueError(
-                f"Training data path '{training_data_path}' is a file. "
-                f"Please provide a directory path."
-            )
-
-        structlogger.debug(
-            "rasa.importers.remote_training_data_importer.fetch_training_archive",
-            training_data_path=training_data_path,
-        )
-        training_archive_file_path = self._fetch_training_archive(
-            training_file, training_data_path
-        )
-
-        if not os.path.isfile(training_archive_file_path):
-            raise FileNotFoundError(
-                f"Training data archive '{training_archive_file_path}' not found. "
-                f"Please make sure to provide the correct path."
-            )
-
-        structlogger.debug(
-            "rasa.importers.remote_training_data_importer.extract_training_archive",
-            training_archive_file_path=training_archive_file_path,
-            training_data_path=training_data_path,
-        )
-        with TarSafe.open(training_archive_file_path, "r:gz") as tar:
-            tar.extractall(path=training_data_path)
-
-        structlogger.debug(
-            "rasa.importers.remote_training_data_importer.remove_downloaded_archive",
-            training_data_path=training_data_path,
-        )
-        os.remove(training_archive_file_path)
-        return training_data_path
-
-    def get_config(self) -> Dict:
-        """Retrieves model config (see parent class for full docstring)."""
-        if not self.config_file or not os.path.exists(self.config_file):
-            structlogger.debug(
-                "rasa.importers.remote_training_data_importer.no_config_file",
-                message="No configuration file was provided to the RasaFileImporter.",
-            )
-            return {}
-
-        config = read_model_configuration(self.config_file)
-        return config
-
-    @rasa.shared.utils.common.cached_method
-    def get_config_file_for_auto_config(self) -> Optional[Text]:
-        """Returns config file path for auto-config only if there is a single one."""
-        return self.config_file
-
-    def get_stories(self, exclusion_percentage: Optional[int] = None) -> StoryGraph:
-        """Retrieves training stories / rules (see parent class for full docstring)."""
-        return utils.story_graph_from_paths(
-            self._story_files, self.get_domain(), exclusion_percentage
-        )
-
-    def get_flows(self) -> FlowsList:
-        """Retrieves training stories / rules (see parent class for full docstring)."""
-        return utils.flows_from_paths(self._flow_files)
-
-    def get_conversation_tests(self) -> StoryGraph:
-        """Retrieves conversation test stories (see parent class for full docstring)."""
-        return utils.story_graph_from_paths(
-            self._conversation_test_files, self.get_domain()
-        )
-
-    def get_nlu_data(self, language: Optional[Text] = "en") -> TrainingData:
-        """Retrieves NLU training data (see parent class for full docstring)."""
-        return utils.training_data_from_paths(self._nlu_files, language)
-
-    def get_domain(self) -> Domain:
-        """Retrieves model domain (see parent class for full docstring)."""
-        domain = Domain.empty()
-        domain_path = f"{self.extracted_path}"
-        try:
-            domain = Domain.load(domain_path)
-        except InvalidDomain as e:
-            rasa.shared.utils.io.raise_warning(
-                f"Loading domain from '{domain_path}' failed. Using "
-                f"empty domain. Error: '{e}'"
-            )
-
-        return domain