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

rasa/engine/recipes/recipe.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
 
 import abc
-from typing import Text, Dict, Any, Optional, Tuple, Set
+from typing import Any, Dict, Optional, Set, Text, Tuple
 
 import rasa.shared.utils.io
 from rasa.engine.graph import GraphModelConfiguration
-from rasa.shared.exceptions import RasaException
 from rasa.shared.data import TrainingType
+from rasa.shared.exceptions import RasaException
 
 
 class InvalidRecipeException(RasaException):

rasa/engine/runner/dask.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
 
+import asyncio
 import logging
 from typing import Any, Dict, List, Optional, Text
 
-import dask.local
 import dask.core
-import asyncio
+import dask.local
 
 from rasa.engine.exceptions import GraphRunError
 from rasa.engine.graph import ExecutionContext, GraphNode, GraphNodeHook, GraphSchema

rasa/engine/runner/interface.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional, Text
 

rasa/engine/storage/local_model_storage.py

@@ -8,17 +8,18 @@ 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
+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__)
 

rasa/engine/storage/resource.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
+
 import logging
 import typing
+import uuid
 from dataclasses import dataclass, field
 from pathlib import Path
 from typing import Text
-import uuid
 
 import rasa.utils.common
 import rasa.utils.io

rasa/engine/storage/storage.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 import abc
 import logging
 import typing
@@ -6,17 +7,18 @@ from contextlib import contextmanager
 from dataclasses import dataclass
 from datetime import datetime
 from pathlib import Path
-from typing import Tuple, Union, Text, Generator, Dict, Any, Optional
+from typing import Any, Dict, Generator, Optional, Text, Tuple, Union
+
 from packaging import version
 
 from rasa.constants import MINIMUM_COMPATIBLE_VERSION
-from rasa.exceptions import UnsupportedModelVersionError
 from rasa.engine.storage.resource import Resource
+from rasa.exceptions import UnsupportedModelVersionError
 from rasa.shared.core.domain import Domain
 from rasa.shared.data import TrainingType
 
 if typing.TYPE_CHECKING:
-    from rasa.engine.graph import GraphSchema, GraphModelConfiguration
+    from rasa.engine.graph import GraphModelConfiguration, GraphSchema
 
 logger = logging.getLogger(__name__)
 

rasa/engine/training/components.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
-from typing import Any, Dict, Optional, Text, Type
+
 import dataclasses
 import uuid
+from typing import Any, Dict, Optional, Text, Type
 
 from rasa.engine.caching import Cacheable, TrainingCache
 from rasa.engine.graph import ExecutionContext, GraphComponent, SchemaNode

rasa/engine/training/fingerprinting.py

@@ -1,10 +1,12 @@
 import inspect
 import logging
 from typing import Any, Dict, Text, Type
-from typing_extensions import Protocol, runtime_checkable
+
 import importlib_metadata
-import rasa.utils.common
+from typing_extensions import Protocol, runtime_checkable
+
 import rasa.shared.utils.io
+import rasa.utils.common
 from rasa.engine.graph import GraphComponent
 
 logger = logging.getLogger(__name__)

rasa/engine/training/graph_trainer.py

@@ -4,16 +4,16 @@ from pathlib import Path
 from typing import Any, Dict, Text, Type, Union
 
 from rasa.engine.caching import TrainingCache
-from rasa.engine.graph import ExecutionContext, GraphSchema, GraphModelConfiguration
 from rasa.engine.constants import PLACEHOLDER_IMPORTER
+from rasa.engine.graph import ExecutionContext, GraphModelConfiguration, GraphSchema
 from rasa.engine.runner.interface import GraphRunner
-from rasa.engine.storage.storage import ModelStorage, ModelMetadata
+from rasa.engine.storage.storage import ModelMetadata, ModelStorage
 from rasa.engine.training.components import (
-    PrecomputedValueProvider,
     FingerprintComponent,
     FingerprintStatus,
+    PrecomputedValueProvider,
 )
-from rasa.engine.training.hooks import TrainingHook, LoggingHook
+from rasa.engine.training.hooks import LoggingHook, TrainingHook
 from rasa.shared.importers.importer import TrainingDataImporter
 
 logger = logging.getLogger(__name__)

rasa/engine/training/hooks.py

@@ -1,12 +1,12 @@
 import logging
 from typing import Any, Dict, Text, Type
 
+import rasa.shared.utils.io
 from rasa.engine.caching import TrainingCache
 from rasa.engine.graph import ExecutionContext, GraphNodeHook, GraphSchema, SchemaNode
 from rasa.engine.storage.storage import ModelStorage
-from rasa.engine.training.components import PrecomputedValueProvider
-import rasa.shared.utils.io
 from rasa.engine.training import fingerprinting
+from rasa.engine.training.components import PrecomputedValueProvider
 
 logger = logging.getLogger(__name__)
 

rasa/engine/validation.py

@@ -5,31 +5,32 @@ import re
 import sys
 import typing
 from typing import (
-    Optional,
+    Any,
     Callable,
+    Dict,
+    List,
+    Literal,
+    Optional,
+    Set,
     Text,
     Tuple,
-    Dict,
     Type,
-    Any,
-    Set,
-    Union,
     TypeVar,
-    List,
-    Literal,
+    Union,
 )
 
-import rasa.utils.common
 import structlog
 import typing_utils
-from rasa.core import IntentlessPolicy, ContextualResponseRephraser
+
+import rasa.utils.common
+from rasa.core import ContextualResponseRephraser, IntentlessPolicy
 from rasa.core.policies.policy import PolicyPrediction
 from rasa.core.utils import AvailableEndpoints
 from rasa.dialogue_understanding.coexistence.constants import (
-    NLU_ENTRY,
     CALM_ENTRY,
-    STICKY,
+    NLU_ENTRY,
     NON_STICKY,
+    STICKY,
 )
 from rasa.dialogue_understanding.coexistence.intent_based_router import (
     IntentBasedRouter,
@@ -39,52 +40,52 @@ from rasa.dialogue_understanding.generator import (
     LLMBasedCommandGenerator,
 )
 from rasa.dialogue_understanding.generator.constants import (
-    LLM_CONFIG_KEY,
     FLOW_RETRIEVAL_KEY,
+    LLM_CONFIG_KEY,
 )
 from rasa.dialogue_understanding.patterns.chitchat import FLOW_PATTERN_CHITCHAT
 from rasa.engine.constants import RESERVED_PLACEHOLDERS
 from rasa.engine.exceptions import GraphSchemaValidationException
 from rasa.engine.graph import (
-    GraphSchema,
-    GraphComponent,
-    SchemaNode,
     ExecutionContext,
+    GraphComponent,
     GraphModelConfiguration,
+    GraphSchema,
+    SchemaNode,
 )
 from rasa.engine.storage.resource import Resource
-from rasa.engine.storage.storage import ModelStorage, ModelMetadata
+from rasa.engine.storage.storage import ModelMetadata, ModelStorage
 from rasa.engine.training.fingerprinting import Fingerprintable
 from rasa.shared.constants import (
-    DOCS_URL_GRAPH_COMPONENTS,
-    ROUTE_TO_CALM_SLOT,
-    EMBEDDINGS_CONFIG_KEY,
     API_BASE_CONFIG_KEY,
-    DEPLOYMENT_CONFIG_KEY,
-    API_VERSION_CONFIG_KEY,
     API_KEY,
+    API_TYPE_CONFIG_KEY,
+    API_VERSION_CONFIG_KEY,
+    AWS_ACCESS_KEY_ID_CONFIG_KEY,
     AWS_REGION_NAME_CONFIG_KEY,
+    AWS_SECRET_ACCESS_KEY_CONFIG_KEY,
+    AWS_SESSION_TOKEN_CONFIG_KEY,
+    DEPLOYMENT_CONFIG_KEY,
+    DOCS_URL_GRAPH_COMPONENTS,
+    EMBEDDINGS_CONFIG_KEY,
+    MODEL_GROUP_CONFIG_KEY,
     MODEL_GROUP_ID_CONFIG_KEY,
-    ROUTER_CONFIG_KEY,
     MODELS_CONFIG_KEY,
-    MODEL_GROUP_CONFIG_KEY,
-    ROUTING_STRATEGY_CONFIG_KEY,
-    VALID_ROUTING_STRATEGIES,
-    ROUTING_STRATEGIES_REQUIRING_REDIS_CACHE,
-    ROUTING_STRATEGIES_NOT_REQUIRING_CACHE,
+    PROVIDER_CONFIG_KEY,
     REDIS_HOST_CONFIG_KEY,
-    AWS_ACCESS_KEY_ID_CONFIG_KEY,
-    AWS_SECRET_ACCESS_KEY_CONFIG_KEY,
-    AWS_SESSION_TOKEN_CONFIG_KEY,
+    ROUTE_TO_CALM_SLOT,
+    ROUTER_CONFIG_KEY,
+    ROUTING_STRATEGIES_NOT_REQUIRING_CACHE,
+    ROUTING_STRATEGIES_REQUIRING_REDIS_CACHE,
+    ROUTING_STRATEGY_CONFIG_KEY,
     SENSITIVE_DATA,
     USE_CHAT_COMPLETIONS_ENDPOINT_CONFIG_KEY,
-    API_TYPE_CONFIG_KEY,
     VALID_PROVIDERS_FOR_API_TYPE_CONFIG_KEY,
-    PROVIDER_CONFIG_KEY,
+    VALID_ROUTING_STRATEGIES,
 )
 from rasa.shared.core.constants import ACTION_RESET_ROUTING, ACTION_TRIGGER_CHITCHAT
 from rasa.shared.core.domain import Domain
-from rasa.shared.core.flows import FlowsList, Flow
+from rasa.shared.core.flows import Flow, FlowsList
 from rasa.shared.core.slots import Slot
 from rasa.shared.exceptions import RasaException
 from rasa.shared.nlu.training_data.message import Message

rasa/exceptions.py

@@ -1,9 +1,10 @@
+from dataclasses import dataclass
 from typing import Text
+
 from packaging import version
-from dataclasses import dataclass
 
-from rasa.shared.exceptions import RasaException
 from rasa.constants import MINIMUM_COMPATIBLE_VERSION
+from rasa.shared.exceptions import RasaException
 
 
 @dataclass

rasa/graph_components/converters/nlu_message_converter.py

@@ -1,9 +1,9 @@
 from __future__ import annotations
-from typing import Dict, Text, Any, List
 
-from rasa.core.channels.channel import UserMessage
+from typing import Any, Dict, List, Text
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from rasa.core.channels.channel import UserMessage
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.nlu.constants import TEXT, TEXT_TOKENS

rasa/graph_components/providers/domain_for_core_training_provider.py

@@ -1,13 +1,13 @@
 from __future__ import annotations
 
 import copy
-from typing import Dict, Text, Any
+from typing import Any, Dict, Text
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.constants import REQUIRED_SLOTS_KEY
-from rasa.shared.core.domain import KEY_RESPONSES, Domain, SESSION_CONFIG_KEY, KEY_FORMS
+from rasa.shared.core.domain import KEY_FORMS, KEY_RESPONSES, SESSION_CONFIG_KEY, Domain
 
 
 class DomainForCoreTrainingProvider(GraphComponent):

rasa/graph_components/providers/domain_provider.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
-from typing import Dict, Text, Any, Optional
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from typing import Any, Dict, Optional, Text
+
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.core.domain import Domain

rasa/graph_components/providers/flows_provider.py

@@ -5,10 +5,9 @@ from typing import Any, Dict, Text
 from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
-from rasa.shared.importers.importer import TrainingDataImporter
-from rasa.shared.core.flows.yaml_flows_io import YAMLFlowsReader, YamlFlowsWriter
-
 from rasa.shared.core.flows import FlowsList
+from rasa.shared.core.flows.yaml_flows_io import YAMLFlowsReader, YamlFlowsWriter
+from rasa.shared.importers.importer import TrainingDataImporter
 
 FLOWS_PERSISTENCE_FILE_NAME = "flows.yml"
 

rasa/graph_components/providers/forms_provider.py

@@ -1,13 +1,13 @@
 from __future__ import annotations
-import dataclasses
 
-from typing import Dict, Text, Any
+import dataclasses
+from typing import Any, Dict, Text
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+import rasa.shared.utils.io
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.core.domain import Domain
-import rasa.shared.utils.io
 
 
 @dataclasses.dataclass

rasa/graph_components/providers/nlu_training_data_provider.py

@@ -1,12 +1,14 @@
 from __future__ import annotations
-from typing import Dict, Text, Any
-from rasa.engine.graph import GraphComponent, ExecutionContext
+
+from typing import Any, Dict, Text
+
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.importers.importer import TrainingDataImporter
 from rasa.shared.nlu.training_data.training_data import (
-    TrainingData,
     DEFAULT_TRAINING_DATA_OUTPUT_PATH,
+    TrainingData,
 )
 
 

rasa/graph_components/providers/responses_provider.py

@@ -1,13 +1,13 @@
 from __future__ import annotations
-import dataclasses
 
-from typing import Dict, List, Text, Any
+import dataclasses
+from typing import Any, Dict, List, Text
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+import rasa.shared.utils.io
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.core.domain import Domain
-import rasa.shared.utils.io
 
 
 @dataclasses.dataclass

rasa/graph_components/providers/rule_only_provider.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
+
 import dataclasses
 import logging
-from typing import Dict, Text, Any
+from typing import Any, Dict, Text
 
 import rasa.shared.utils.io
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 

rasa/graph_components/providers/story_graph_provider.py

@@ -1,18 +1,18 @@
 from __future__ import annotations
-from typing import Dict, Text, Any, List
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from typing import Any, Dict, List, Text
+
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
-from rasa.shared.core.training_data.structures import StoryGraph, StoryStep
-from rasa.shared.importers.importer import TrainingDataImporter
-from rasa.shared.core.training_data.story_writer.yaml_story_writer import (
-    YAMLStoryWriter,
-)
 from rasa.shared.core.training_data.story_reader.yaml_story_reader import (
     YAMLStoryReader,
 )
-
+from rasa.shared.core.training_data.story_writer.yaml_story_writer import (
+    YAMLStoryWriter,
+)
+from rasa.shared.core.training_data.structures import StoryGraph, StoryStep
+from rasa.shared.importers.importer import TrainingDataImporter
 
 STORIES_PERSISTENCE_FILE_NAME = "stories.yml"
 

rasa/graph_components/providers/training_tracker_provider.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
-from typing import Dict, Text, Any, List
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from typing import Any, Dict, List, Text
+
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.core.domain import Domain

rasa/graph_components/validators/default_recipe_validator.py

@@ -1,31 +1,32 @@
 from __future__ import annotations
+
 from collections import defaultdict
-from typing import Iterable, List, Dict, Text, Any, Set, Type, cast
+from typing import Any, Dict, Iterable, List, Set, Text, Type, cast
 
+import rasa.shared.utils.io
+from rasa.core.constants import POLICY_PRIORITY
 from rasa.core.featurizers.precomputation import CoreFeaturizationInputConverter
+from rasa.core.policies.flow_policy import FlowPolicy
+from rasa.core.policies.memoization import AugmentedMemoizationPolicy, MemoizationPolicy
+from rasa.core.policies.policy import Policy, SupportedData
+from rasa.core.policies.rule_policy import RulePolicy
+from rasa.core.policies.ted_policy import TEDPolicy
 from rasa.core.policies.unexpected_intent_policy import UnexpecTEDIntentPolicy
 from rasa.engine.graph import ExecutionContext, GraphComponent, GraphSchema, SchemaNode
-from rasa.engine.storage.storage import ModelStorage
 from rasa.engine.storage.resource import Resource
-from rasa.nlu.featurizers.featurizer import Featurizer
-from rasa.nlu.extractors.mitie_entity_extractor import MitieEntityExtractor
-from rasa.nlu.extractors.regex_entity_extractor import RegexEntityExtractor
+from rasa.engine.storage.storage import ModelStorage
+from rasa.nlu.classifiers.diet_classifier import DIETClassifier
 from rasa.nlu.extractors.crf_entity_extractor import (
     CRFEntityExtractor,
     CRFEntityExtractorOptions,
 )
 from rasa.nlu.extractors.entity_synonyms import EntitySynonymMapper
+from rasa.nlu.extractors.mitie_entity_extractor import MitieEntityExtractor
+from rasa.nlu.extractors.regex_entity_extractor import RegexEntityExtractor
+from rasa.nlu.featurizers.featurizer import Featurizer
 from rasa.nlu.featurizers.sparse_featurizer.regex_featurizer import RegexFeaturizer
-from rasa.nlu.classifiers.diet_classifier import DIETClassifier
 from rasa.nlu.selectors.response_selector import ResponseSelector
 from rasa.nlu.tokenizers.tokenizer import Tokenizer
-from rasa.core.policies.rule_policy import RulePolicy
-from rasa.core.policies.policy import Policy, SupportedData
-from rasa.core.policies.memoization import AugmentedMemoizationPolicy, MemoizationPolicy
-from rasa.core.policies.ted_policy import TEDPolicy
-from rasa.core.policies.flow_policy import FlowPolicy
-from rasa.core.constants import POLICY_PRIORITY
-from rasa.shared.core.training_data.structures import RuleStep, StoryGraph
 from rasa.shared.constants import (
     DEFAULT_CONFIG_PATH,
     DOCS_URL_COMPONENTS,
@@ -33,18 +34,17 @@ from rasa.shared.constants import (
     DOCS_URL_NLU_BASED_POLICIES,
     DOCS_URL_RULES,
 )
-from rasa.shared.core.domain import Domain, InvalidDomain
 from rasa.shared.core.constants import (
     ACTION_BACK_NAME,
     ACTION_RESTART_NAME,
     USER_INTENT_BACK,
     USER_INTENT_RESTART,
 )
+from rasa.shared.core.domain import Domain, InvalidDomain
+from rasa.shared.core.training_data.structures import RuleStep, StoryGraph
 from rasa.shared.exceptions import InvalidConfigException
 from rasa.shared.importers.importer import TrainingDataImporter
 from rasa.shared.nlu.training_data.training_data import TrainingData
-import rasa.shared.utils.io
-
 
 # TODO: Can we replace this with the registered types from the regitry?
 TRAINABLE_EXTRACTORS = [MitieEntityExtractor, CRFEntityExtractor, DIETClassifier]

rasa/graph_components/validators/finetuning_validator.py

@@ -1,21 +1,23 @@
 from __future__ import annotations
-from typing import Dict, Text, Any, Optional
+
 import copy
 import logging
+from typing import Any, Dict, Optional, Text
 
 from packaging import version
+
+import rasa.shared.utils.io
 from rasa.constants import MINIMUM_COMPATIBLE_VERSION
-from rasa.engine.graph import GraphComponent, ExecutionContext
-from rasa.engine.storage.storage import ModelStorage
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.storage.resource import Resource
-from rasa.shared.exceptions import InvalidConfigException
-from rasa.shared.core.domain import Domain
-from rasa.shared.importers.importer import TrainingDataImporter
-import rasa.shared.utils.io
-from rasa.utils.tensorflow.constants import EPOCHS
+from rasa.engine.storage.storage import ModelStorage
 from rasa.graph_components.providers.domain_for_core_training_provider import (
     DomainForCoreTrainingProvider,
 )
+from rasa.shared.core.domain import Domain
+from rasa.shared.exceptions import InvalidConfigException
+from rasa.shared.importers.importer import TrainingDataImporter
+from rasa.utils.tensorflow.constants import EPOCHS
 
 FINGERPRINT_CONFIG = "fingerprint-config"
 FINGERPRINT_CORE = "fingerprint-core"

rasa/hooks.py

@@ -1,21 +1,20 @@
 import argparse
 import logging
-from typing import Optional, TYPE_CHECKING, List, Text, Union
+from typing import TYPE_CHECKING, List, Optional, Text, Union
 
 import pluggy
 
-from rasa.cli import SubParsersAction
-from rasa.cli import x as rasa_x
-from rasa.core.auth_retry_tracker_store import AuthRetryTrackerStore
-from rasa.core.secrets_manager.factory import load_secret_manager
-from rasa.tracing import config
-from rasa.utils.endpoints import EndpointConfig
+# IMPORTANT: do not import anything from rasa here - use scoped imports
+#  this avoids circular imports, as the hooks are used in different places
+#  across the codebase.
 
 if TYPE_CHECKING:
+    from rasa.anonymization.anonymization_pipeline import AnonymizationPipeline
+    from rasa.cli import SubParsersAction
     from rasa.core.brokers.broker import EventBroker
     from rasa.core.tracker_store import TrackerStore
     from rasa.shared.core.domain import Domain
-    from rasa.anonymization.anonymization_pipeline import AnonymizationPipeline
+    from rasa.utils.endpoints import EndpointConfig
 
 hookimpl = pluggy.HookimplMarker("rasa")
 logger = logging.getLogger(__name__)
@@ -23,13 +22,12 @@ logger = logging.getLogger(__name__)
 
 @hookimpl  # type: ignore[misc]
 def refine_cli(
-    subparsers: SubParsersAction,
+    subparsers: "SubParsersAction",
     parent_parsers: List[argparse.ArgumentParser],
 ) -> None:
-    from rasa.cli import e2e_test, inspect, markers, dialogue_understanding_test
-    from rasa.cli.studio import studio
-
+    from rasa.cli import dialogue_understanding_test, e2e_test, inspect, markers
     from rasa.cli import license as license_cli
+    from rasa.cli.studio import studio
 
     e2e_test.add_subparser(subparsers, parent_parsers)
     dialogue_understanding_test.add_subparser(subparsers, parent_parsers)
@@ -42,6 +40,9 @@ def refine_cli(
 
 @hookimpl  # type: ignore[misc]
 def configure_commandline(cmdline_arguments: argparse.Namespace) -> Optional[Text]:
+    from rasa.cli import x as rasa_x
+    from rasa.tracing import config
+
     endpoints_file = None
 
     if cmdline_arguments.func.__name__ == "rasa_x":
@@ -68,6 +69,8 @@ def init_telemetry(endpoints_file: Optional[Text]) -> None:
 
 @hookimpl  # type: ignore[misc]
 def init_managers(endpoints_file: Optional[Text]) -> None:
+    from rasa.core.secrets_manager.factory import load_secret_manager
+
     load_secret_manager(endpoints_file)
 
 
@@ -77,6 +80,9 @@ def create_tracker_store(
     domain: "Domain",
     event_broker: Optional["EventBroker"],
 ) -> "TrackerStore":
+    from rasa.core.auth_retry_tracker_store import AuthRetryTrackerStore
+    from rasa.utils.endpoints import EndpointConfig
+
     if isinstance(endpoint_config, EndpointConfig):
         return AuthRetryTrackerStore(
             endpoint_config=endpoint_config, domain=domain, event_broker=event_broker

rasa/jupyter.py

@@ -3,10 +3,10 @@ import pprint as pretty_print
 import typing
 from typing import Any, Dict, Optional, Text
 
-from rasa.shared.exceptions import RasaException
-from rasa.shared.utils.cli import print_success
 import rasa.core.agent
 import rasa.utils.common
+from rasa.shared.exceptions import RasaException
+from rasa.shared.utils.cli import print_success
 
 if typing.TYPE_CHECKING:
     from rasa.core.agent import Agent