rasa-pro 3.14.0.dev10__py3-none-any.whl → 3.14.0.dev12__py3-none-any.whl

rasa/engine/recipes/default_components.py

@@ -11,38 +11,21 @@ from rasa.dialogue_understanding.generator import (
     LLMCommandGenerator,
 )
 from rasa.dialogue_understanding.generator.nlu_command_adapter import NLUCommandAdapter
-from rasa.engine.graph import GraphComponent
 from rasa.nlu.classifiers.fallback_classifier import FallbackClassifier
 from rasa.nlu.classifiers.keyword_intent_classifier import KeywordIntentClassifier
-from rasa.nlu.classifiers.logistic_regression_classifier import (
-    LogisticRegressionClassifier,
-)
-from rasa.nlu.classifiers.mitie_intent_classifier import MitieIntentClassifier
-from rasa.nlu.classifiers.sklearn_intent_classifier import SklearnIntentClassifier
-from rasa.nlu.extractors.crf_entity_extractor import CRFEntityExtractor
 from rasa.nlu.extractors.duckling_entity_extractor import DucklingEntityExtractor
 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.extractors.spacy_entity_extractor import SpacyEntityExtractor
-from rasa.nlu.featurizers.dense_featurizer.mitie_featurizer import MitieFeaturizer
-from rasa.nlu.featurizers.dense_featurizer.spacy_featurizer import SpacyFeaturizer
 from rasa.nlu.featurizers.sparse_featurizer.count_vectors_featurizer import (
     CountVectorsFeaturizer,
 )
-from rasa.nlu.featurizers.sparse_featurizer.lexical_syntactic_featurizer import (
-    LexicalSyntacticFeaturizer,
-)
 from rasa.nlu.featurizers.sparse_featurizer.regex_featurizer import RegexFeaturizer
-from rasa.nlu.tokenizers.jieba_tokenizer import JiebaTokenizer
-from rasa.nlu.tokenizers.mitie_tokenizer import MitieTokenizer
-from rasa.nlu.tokenizers.spacy_tokenizer import SpacyTokenizer
 from rasa.nlu.tokenizers.whitespace_tokenizer import WhitespaceTokenizer
-from rasa.nlu.utils.mitie_utils import MitieNLP
-from rasa.nlu.utils.spacy_utils import SpacyNLP
 from rasa.shared.utils.common import conditional_import
 
-# Conditional imports for TensorFlow-dependent components
+# Conditional imports for components with external dependencies
+
+# components dependent on tensorflow
 TEDPolicy, TED_POLICY_AVAILABLE = conditional_import(
     "rasa.core.policies.ted_policy", "TEDPolicy"
 )
@@ -62,40 +45,82 @@ ResponseSelector, RESPONSE_SELECTOR_AVAILABLE = conditional_import(
     "rasa.nlu.selectors.response_selector", "ResponseSelector"
 )
 
+# components dependent on skops
+LogisticRegressionClassifier, LOGISTIC_REGRESSION_CLASSIFIER_AVAILABLE = (
+    conditional_import(
+        "rasa.nlu.classifiers.logistic_regression_classifier",
+        "LogisticRegressionClassifier",
+    )
+)
+SklearnIntentClassifier, SKLEARN_INTENT_CLASSIFIER_AVAILABLE = conditional_import(
+    "rasa.nlu.classifiers.sklearn_intent_classifier", "SklearnIntentClassifier"
+)
+
+# components dependent on spacy
+LexicalSyntacticFeaturizer, LEXICAL_SYNTACTIC_FEATURIZER_AVAILABLE = conditional_import(
+    "rasa.nlu.featurizers.sparse_featurizer.lexical_syntactic_featurizer",
+    "LexicalSyntacticFeaturizer",
+)
+SpacyFeaturizer, SPACY_FEATURIZER_AVAILABLE = conditional_import(
+    "rasa.nlu.featurizers.dense_featurizer.spacy_featurizer", "SpacyFeaturizer"
+)
+SpacyTokenizer, SPACY_TOKENIZER_AVAILABLE = conditional_import(
+    "rasa.nlu.tokenizers.spacy_tokenizer", "SpacyTokenizer"
+)
+SpacyEntityExtractor, SPACY_ENTITY_EXTRACTOR_AVAILABLE = conditional_import(
+    "rasa.nlu.extractors.spacy_entity_extractor", "SpacyEntityExtractor"
+)
+SpacyNLP, SPACY_NLP_AVAILABLE = conditional_import(
+    "rasa.nlu.utils.spacy_utils", "SpacyNLP"
+)
+
+# components dependent on sklearn_crfsuite
+CRFEntityExtractor, CRF_ENTITY_EXTRACTOR_AVAILABLE = conditional_import(
+    "rasa.nlu.extractors.crf_entity_extractor", "CRFEntityExtractor"
+)
+
+# components dependent on mitie
+MitieFeaturizer, MITIE_FEATURIZER_AVAILABLE = conditional_import(
+    "rasa.nlu.featurizers.dense_featurizer.mitie_featurizer", "MitieFeaturizer"
+)
+MitieTokenizer, MITIE_TOKENIZER_AVAILABLE = conditional_import(
+    "rasa.nlu.tokenizers.mitie_tokenizer", "MitieTokenizer"
+)
+MitieIntentClassifier, MITIE_INTENT_CLASSIFIER_AVAILABLE = conditional_import(
+    "rasa.nlu.classifiers.mitie_intent_classifier", "MitieIntentClassifier"
+)
+MitieEntityExtractor, MITIE_ENTITY_EXTRACTOR_AVAILABLE = conditional_import(
+    "rasa.nlu.extractors.mitie_entity_extractor", "MitieEntityExtractor"
+)
+MitieNLP, MITIE_NLP_AVAILABLE = conditional_import(
+    "rasa.nlu.utils.mitie_utils", "MitieNLP"
+)
+
+# components dependent on jieba
+JiebaTokenizer, JIEBA_TOKENIZER_AVAILABLE = conditional_import(
+    "rasa.nlu.tokenizers.jieba_tokenizer", "JiebaTokenizer"
+)
+
+# Base components that are always available (no external dependencies)
 DEFAULT_COMPONENTS = [
-    # Message Classifiers
+    # Classifiers
     FallbackClassifier,
     KeywordIntentClassifier,
-    MitieIntentClassifier,
-    SklearnIntentClassifier,
-    LogisticRegressionClassifier,
     NLUCommandAdapter,
     LLMCommandGenerator,
     LLMBasedRouter,
     IntentBasedRouter,
-    # Response Selectors
-    # Message Entity Extractors
-    CRFEntityExtractor,
+    # Entity Extractors
     DucklingEntityExtractor,
     EntitySynonymMapper,
-    MitieEntityExtractor,
-    SpacyEntityExtractor,
     RegexEntityExtractor,
-    # Message Feauturizers
-    LexicalSyntacticFeaturizer,
-    MitieFeaturizer,
-    SpacyFeaturizer,
+    # Featurizers
     CountVectorsFeaturizer,
     RegexFeaturizer,
     # Tokenizers
-    JiebaTokenizer,
-    MitieTokenizer,
-    SpacyTokenizer,
     WhitespaceTokenizer,
     # Language Model Providers
-    MitieNLP,
-    SpacyNLP,
-    # Dialogue Management Policies
+    # Policies
     RulePolicy,
     MemoizationPolicy,
     AugmentedMemoizationPolicy,
@@ -104,52 +129,56 @@ DEFAULT_COMPONENTS = [
     IntentlessPolicy,
 ]
 
+# Conditionally add components based on dependencies
 
-class ComponentInserter:
-    """Manages insertion of components at specific positions with index adjustment."""
-
-    def __init__(self, insertion_points: dict[str, int]):
-        self.insertion_points = insertion_points
-        self.insertion_counts = {category: 0 for category in insertion_points}
+# components dependent on tensorflow
+if DIET_CLASSIFIER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(DIETClassifier)
+if CONVERT_FEATURIZER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(ConveRTFeaturizer)
+if LANGUAGE_MODEL_FEATURIZER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(LanguageModelFeaturizer)
+if RESPONSE_SELECTOR_AVAILABLE:
+    DEFAULT_COMPONENTS.append(ResponseSelector)
+if TED_POLICY_AVAILABLE:
+    DEFAULT_COMPONENTS.append(TEDPolicy)
+if UNEXPECTED_INTENT_POLICY_AVAILABLE:
+    DEFAULT_COMPONENTS.append(UnexpecTEDIntentPolicy)
 
-    def insert_if_available(
-        self, component: GraphComponent, is_available: bool, category: str
-    ) -> None:
-        """Insert a component at the appropriate position if available."""
-        if is_available and component is not None:
-            # Calculate the adjusted index based on insertions in earlier categories
-            adjusted_index = self.insertion_points[category]
-            for cat, count in self.insertion_counts.items():
-                if self.insertion_points[cat] < self.insertion_points[category]:
-                    adjusted_index += count
+# components dependent on skops
+if LOGISTIC_REGRESSION_CLASSIFIER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(LogisticRegressionClassifier)
+if SKLEARN_INTENT_CLASSIFIER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(SklearnIntentClassifier)
 
-            DEFAULT_COMPONENTS.insert(adjusted_index, component)
-            self.insertion_counts[category] += 1
+# components dependent on spacy
+if LEXICAL_SYNTACTIC_FEATURIZER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(LexicalSyntacticFeaturizer)
+if SPACY_FEATURIZER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(SpacyFeaturizer)
+if SPACY_TOKENIZER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(SpacyTokenizer)
+if SPACY_ENTITY_EXTRACTOR_AVAILABLE:
+    DEFAULT_COMPONENTS.append(SpacyEntityExtractor)
+if SPACY_NLP_AVAILABLE:
+    DEFAULT_COMPONENTS.append(SpacyNLP)
 
+# components dependent on mitie
+if MITIE_FEATURIZER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(MitieFeaturizer)
+if MITIE_TOKENIZER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(MitieTokenizer)
+if MITIE_INTENT_CLASSIFIER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(MitieIntentClassifier)
+if MITIE_ENTITY_EXTRACTOR_AVAILABLE:
+    DEFAULT_COMPONENTS.append(MitieEntityExtractor)
+if MITIE_NLP_AVAILABLE:
+    DEFAULT_COMPONENTS.append(MitieNLP)
 
-# Define insertion points by component category to preserve order
-INSERTION_POINTS = {
-    "classifiers": 1,
-    "response_selectors": 11,
-    "featurizers": 20,
-    "policies": 35,
-}
+# components dependent on jieba
+if JIEBA_TOKENIZER_AVAILABLE:
+    DEFAULT_COMPONENTS.append(JiebaTokenizer)
 
-# Create inserter instance
-inserter = ComponentInserter(INSERTION_POINTS)
-
-# Conditionally add TensorFlow-dependent components
-inserter.insert_if_available(DIETClassifier, DIET_CLASSIFIER_AVAILABLE, "classifiers")
-inserter.insert_if_available(
-    ResponseSelector, RESPONSE_SELECTOR_AVAILABLE, "response_selectors"
-)
-inserter.insert_if_available(
-    ConveRTFeaturizer, CONVERT_FEATURIZER_AVAILABLE, "featurizers"
-)
-inserter.insert_if_available(
-    LanguageModelFeaturizer, LANGUAGE_MODEL_FEATURIZER_AVAILABLE, "featurizers"
-)
-inserter.insert_if_available(TEDPolicy, TED_POLICY_AVAILABLE, "policies")
-inserter.insert_if_available(
-    UnexpecTEDIntentPolicy, UNEXPECTED_INTENT_POLICY_AVAILABLE, "policies"
-)
+# components dependent on sklearn_crfsuite
+if CRF_ENTITY_EXTRACTOR_AVAILABLE:
+    DEFAULT_COMPONENTS.append(CRFEntityExtractor)

rasa/engine/recipes/default_recipe.py

@@ -183,9 +183,23 @@ class DefaultV1Recipe(Recipe):
         # Importing all the default Rasa components will automatically register them
         from rasa.engine.recipes.default_components import (
             CONVERT_FEATURIZER_AVAILABLE,
+            CRF_ENTITY_EXTRACTOR_AVAILABLE,
             DIET_CLASSIFIER_AVAILABLE,
+            JIEBA_TOKENIZER_AVAILABLE,
             LANGUAGE_MODEL_FEATURIZER_AVAILABLE,
+            LEXICAL_SYNTACTIC_FEATURIZER_AVAILABLE,
+            LOGISTIC_REGRESSION_CLASSIFIER_AVAILABLE,
+            MITIE_ENTITY_EXTRACTOR_AVAILABLE,
+            MITIE_FEATURIZER_AVAILABLE,
+            MITIE_INTENT_CLASSIFIER_AVAILABLE,
+            MITIE_NLP_AVAILABLE,
+            MITIE_TOKENIZER_AVAILABLE,
             RESPONSE_SELECTOR_AVAILABLE,
+            SKLEARN_INTENT_CLASSIFIER_AVAILABLE,
+            SPACY_ENTITY_EXTRACTOR_AVAILABLE,
+            SPACY_FEATURIZER_AVAILABLE,
+            SPACY_NLP_AVAILABLE,
+            SPACY_TOKENIZER_AVAILABLE,
             TED_POLICY_AVAILABLE,
             UNEXPECTED_INTENT_POLICY_AVAILABLE,
         )
@@ -207,27 +221,45 @@ class DefaultV1Recipe(Recipe):
                 "Please use Python 3.11 or earlier for TensorFlow-based components. "
             )
 
-        # Check for TensorFlow components that failed to load
+        # Check for nlu components that failed to load
         # because conditional module import of those components did not succeed
-        tensorflow_component_availability = {
+        nlu_components = {
+            # tensorflow-dependent components
             "DIETClassifier": DIET_CLASSIFIER_AVAILABLE,
             "TEDPolicy": TED_POLICY_AVAILABLE,
             "UnexpecTEDIntentPolicy": UNEXPECTED_INTENT_POLICY_AVAILABLE,
             "ResponseSelector": RESPONSE_SELECTOR_AVAILABLE,
             "ConveRTFeaturizer": CONVERT_FEATURIZER_AVAILABLE,
             "LanguageModelFeaturizer": LANGUAGE_MODEL_FEATURIZER_AVAILABLE,
+            # nlu components dependent on other dependencies than tensorflow
+            "LogisticRegressionClassifier": LOGISTIC_REGRESSION_CLASSIFIER_AVAILABLE,
+            "SklearnIntentClassifier": SKLEARN_INTENT_CLASSIFIER_AVAILABLE,
+            "LexicalSyntacticFeaturizer": LEXICAL_SYNTACTIC_FEATURIZER_AVAILABLE,
+            "MitieFeaturizer": MITIE_FEATURIZER_AVAILABLE,
+            "SpacyFeaturizer": SPACY_FEATURIZER_AVAILABLE,
+            "JiebaTokenizer": JIEBA_TOKENIZER_AVAILABLE,
+            "MitieTokenizer": MITIE_TOKENIZER_AVAILABLE,
+            "SpacyTokenizer": SPACY_TOKENIZER_AVAILABLE,
+            "MitieIntentClassifier": MITIE_INTENT_CLASSIFIER_AVAILABLE,
+            "MitieEntityExtractor": MITIE_ENTITY_EXTRACTOR_AVAILABLE,
+            "SpacyEntityExtractor": SPACY_ENTITY_EXTRACTOR_AVAILABLE,
+            "MitieNLP": MITIE_NLP_AVAILABLE,
+            "SpacyNLP": SPACY_NLP_AVAILABLE,
+            "CRFEntityExtractor": CRF_ENTITY_EXTRACTOR_AVAILABLE,
         }
 
-        if name in tensorflow_component_availability:
-            # TensorFlow-dependent component failed to load
-            # if TensorFlow is not available
-            if not tensorflow_component_availability[name]:
-                raise InvalidConfigException(
-                    f"{name} requires TensorFlow. "
-                    "Please install it with: "
-                    "`pip install rasa[tensorflow]`. "
-                    "If using poetry, install it with "
-                    "`poetry install --extras tensorflow`."
+        if name in nlu_components:
+            # nlu-dependent component failed to load
+            # because nlu dependency group not installed
+            if not nlu_components[name]:
+                from rasa.exceptions import MissingDependencyException
+
+                raise MissingDependencyException(
+                    f"The {name} component requires additional dependencies "
+                    f"which are not installed. "
+                    f"Please install the required extra by running: "
+                    f"pip install 'rasa-pro[nlu]' OR "
+                    f"poetry add 'rasa-pro[nlu]'"
                 )
 
         raise InvalidConfigException(
@@ -525,17 +557,42 @@ class DefaultV1Recipe(Recipe):
         component: Type[GraphComponent],
         component_config: Dict[Text, Any],
     ) -> Dict[Text, Any]:
-        from rasa.nlu.classifiers.mitie_intent_classifier import MitieIntentClassifier
-        from rasa.nlu.classifiers.sklearn_intent_classifier import (
-            SklearnIntentClassifier,
-        )
-        from rasa.nlu.extractors.mitie_entity_extractor import MitieEntityExtractor
+        # Try to import MitieIntentClassifier, but handle ImportError gracefully
+        # since the dependencies for it may not be installed
+        try:
+            from rasa.nlu.classifiers.mitie_intent_classifier import (
+                MitieIntentClassifier,
+            )
+        except ImportError:
+            MitieIntentClassifier = None  # type: ignore
 
-        cli_args_mapping: Dict[Type[GraphComponent], List[Text]] = {
-            MitieIntentClassifier: ["num_threads"],
-            MitieEntityExtractor: ["num_threads"],
-            SklearnIntentClassifier: ["num_threads"],
-        }
+        # Try to import MitieEntityExtractor, but handle ImportError gracefully
+        # since the dependencies for it may not be installed
+        try:
+            from rasa.nlu.extractors.mitie_entity_extractor import MitieEntityExtractor
+        except ImportError:
+            MitieEntityExtractor = None  # type: ignore
+
+        # Try to import SklearnIntentClassifier, but handle ImportError gracefully
+        # since the dependencies for it may not be installed
+        try:
+            from rasa.nlu.classifiers.sklearn_intent_classifier import (
+                SklearnIntentClassifier,
+            )
+        except ImportError:
+            SklearnIntentClassifier = None  # type: ignore
+
+        cli_args_mapping: Dict[Type[GraphComponent], List[Text]] = {}
+
+        # Only add components if they were successfully imported
+        if MitieIntentClassifier is not None:
+            cli_args_mapping[MitieIntentClassifier] = ["num_threads"]
+
+        if MitieEntityExtractor is not None:
+            cli_args_mapping[MitieEntityExtractor] = ["num_threads"]
+
+        if SklearnIntentClassifier is not None:
+            cli_args_mapping[SklearnIntentClassifier] = ["num_threads"]
 
         config_from_cli = {
             param: cli_parameters[param]

rasa/engine/runner/dask.py

@@ -1,18 +1,18 @@
 from __future__ import annotations
 
 import asyncio
-import logging
 from typing import Any, Dict, List, Optional, Text
 
 import dask.core
 import dask.local
+import structlog
 
 from rasa.engine.exceptions import GraphRunError
 from rasa.engine.graph import ExecutionContext, GraphNode, GraphNodeHook, GraphSchema
 from rasa.engine.runner.interface import GraphRunner
 from rasa.engine.storage.storage import ModelStorage
 
-logger = logging.getLogger(__name__)
+structlogger = structlog.get_logger()
 
 
 class DaskGraphRunner(GraphRunner):
@@ -94,9 +94,12 @@ class DaskGraphRunner(GraphRunner):
         if inputs:
             self._add_inputs_to_graph(inputs, run_graph)
 
-        logger.debug(
-            f"Running graph with inputs: {inputs}, targets: {targets} "
-            f"and {self._execution_context}."
+        structlogger.debug(
+            "rasa.engine.runner.dask.run",
+            event_info="Running graph with inputs, targets and execution context.",
+            inputs=inputs,
+            targets=targets,
+            execution_context=self._execution_context,
         )
 
         try:

rasa/graph_components/validators/default_recipe_validator.py

@@ -13,12 +13,7 @@ from rasa.core.policies.rule_policy import RulePolicy
 from rasa.engine.graph import ExecutionContext, GraphComponent, GraphSchema, SchemaNode
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
-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
@@ -57,8 +52,23 @@ ResponseSelector, RESPONSE_SELECTOR_AVAILABLE = conditional_import(
     "rasa.nlu.selectors.response_selector", "ResponseSelector"
 )
 
-# TODO: Can we replace this with the registered types from the regitry?
-TRAINABLE_EXTRACTORS = [MitieEntityExtractor, CRFEntityExtractor]
+# Conditional imports for nlu components requiring other dependencies than tensorflow
+MitieEntityExtractor, MITIE_ENTITY_EXTRACTOR_AVAILABLE = conditional_import(
+    "rasa.nlu.extractors.mitie_entity_extractor", "MitieEntityExtractor"
+)
+CRFEntityExtractor, CRF_ENTITY_EXTRACTOR_AVAILABLE = conditional_import(
+    "rasa.nlu.extractors.crf_entity_extractor", "CRFEntityExtractor"
+)
+CRFEntityExtractorOptions, _ = conditional_import(
+    "rasa.nlu.extractors.crf_entity_extractor", "CRFEntityExtractorOptions"
+)
+
+# TODO: Can we replace this with the registered types from the registry?
+TRAINABLE_EXTRACTORS = []
+if CRF_ENTITY_EXTRACTOR_AVAILABLE:
+    TRAINABLE_EXTRACTORS.append(CRFEntityExtractor)
+if MITIE_ENTITY_EXTRACTOR_AVAILABLE:
+    TRAINABLE_EXTRACTORS.append(MitieEntityExtractor)
 if DIET_CLASSIFIER_AVAILABLE:
     TRAINABLE_EXTRACTORS.append(DIETClassifier)
 # TODO: replace these once the Recipe is merged (used in tests)
@@ -173,18 +183,27 @@ class DefaultV1RecipeValidator(GraphComponent):
                 docs=DOCS_URL_COMPONENTS,
             )
 
+        available_extractors = {DIETClassifier}
+        if CRF_ENTITY_EXTRACTOR_AVAILABLE:
+            available_extractors.add(CRFEntityExtractor)
+
         if training_data.entity_examples and self._component_types.isdisjoint(
-            {DIETClassifier, CRFEntityExtractor}
+            available_extractors
         ):
             if training_data.entity_roles_groups_used():
+                crf_name = (
+                    CRFEntityExtractor.__name__
+                    if CRFEntityExtractor
+                    else "CRFEntityExtractor"
+                )
                 rasa.shared.utils.io.raise_warning(
                     f"You have defined training data with entities that "
                     f"have roles/groups, but your NLU configuration does not "
                     f"include a '{DIETClassifier.__name__}' "
-                    f"or a '{CRFEntityExtractor.__name__}'. "
+                    f"or a '{crf_name}'. "
                     f"To train entities that have roles/groups, "
                     f"add either '{DIETClassifier.__name__}' "
-                    f"or '{CRFEntityExtractor.__name__}' to your "
+                    f"or '{crf_name}' to your "
                     f"configuration.",
                     docs=DOCS_URL_COMPONENTS,
                 )
@@ -218,7 +237,16 @@ class DefaultV1RecipeValidator(GraphComponent):
             )
 
         if training_data.lookup_tables:
-            if self._component_types.isdisjoint([CRFEntityExtractor, DIETClassifier]):
+            lookup_extractors = [DIETClassifier]
+            if CRF_ENTITY_EXTRACTOR_AVAILABLE:
+                lookup_extractors.append(CRFEntityExtractor)
+
+            if self._component_types.isdisjoint(lookup_extractors):
+                crf_name = (
+                    CRFEntityExtractor.__name__
+                    if CRFEntityExtractor
+                    else "CRFEntityExtractor"
+                )
                 rasa.shared.utils.io.raise_warning(
                     f"You have defined training data consisting of lookup tables, but "
                     f"your NLU configuration does not include any components "
@@ -226,33 +254,45 @@ class DefaultV1RecipeValidator(GraphComponent):
                     f"To make use of the features that are created with the "
                     f"help of the lookup tables, "
                     f"add a '{DIETClassifier.__name__}' or a "
-                    f"'{CRFEntityExtractor.__name__}' "
+                    f"'{crf_name}' "
                     f"with the 'pattern' feature "
                     f"to your configuration.",
                     docs=DOCS_URL_COMPONENTS,
                 )
 
-            elif CRFEntityExtractor in self._component_types:
+            elif (
+                CRF_ENTITY_EXTRACTOR_AVAILABLE
+                and CRFEntityExtractor in self._component_types
+            ):
                 crf_schema_nodes = [
                     schema_node
                     for schema_node in self._graph_schema.nodes.values()
                     if schema_node.uses == CRFEntityExtractor
                 ]
-                has_pattern_feature = any(
-                    CRFEntityExtractorOptions.PATTERN in feature_list
-                    for crf in crf_schema_nodes
-                    for feature_list in crf.config.get("features", [])
+                has_pattern_feature = (
+                    any(
+                        CRFEntityExtractorOptions.PATTERN in feature_list
+                        for crf in crf_schema_nodes
+                        for feature_list in crf.config.get("features", [])
+                    )
+                    if CRFEntityExtractorOptions
+                    else False
                 )
 
                 if not has_pattern_feature:
+                    crf_name = (
+                        CRFEntityExtractor.__name__
+                        if CRFEntityExtractor
+                        else "CRFEntityExtractor"
+                    )
                     rasa.shared.utils.io.raise_warning(
                         f"You have defined training data consisting of "
                         f"lookup tables, but your NLU configuration's "
-                        f"'{CRFEntityExtractor.__name__}' "
+                        f"'{crf_name}' "
                         f"does not include the "
                         f"'pattern' feature. To featurize lookup tables, "
                         f"add the 'pattern' feature to the "
-                        f"'{CRFEntityExtractor.__name__}' "
+                        f"'{crf_name}' "
                         "in your configuration.",
                         docs=DOCS_URL_COMPONENTS,
                     )

rasa/model_manager/warm_rasa_process.py

@@ -1,6 +1,7 @@
 import os
 import shlex
 import subprocess
+import time
 import uuid
 from dataclasses import dataclass
 from typing import List
@@ -149,19 +150,28 @@ def warmup() -> None:
     We import them now, so that the training / deployment can later
     directly start.
     """
+    start_time = time.time()
     try:
         import langchain  # noqa: F401
         import litellm  # noqa: F401
         import matplotlib  # noqa: F401
         import numpy  # noqa: F401
         import pandas  # noqa: F401
-        import presidio_analyzer  # noqa: F401
-        import spacy  # noqa: F401
         import tensorflow  # noqa: F401
 
         import rasa.validator  # noqa: F401
-    except ImportError:
+    except ImportError as e:
+        structlogger.warning(
+            "model_trainer.warmup_error",
+            error=str(e),
+        )
         pass
+    finally:
+        end_time = time.time()
+        structlogger.debug(
+            "model_trainer.warmup_time",
+            time_in_seconds=end_time - start_time,
+        )
 
 
 def warm_rasa_main() -> None:

rasa/model_training.py

@@ -201,20 +201,6 @@ async def train(
             config, domain, training_files, core_additional_arguments
         )
 
-    # Check TensorFlow availability and warn if not available
-    import importlib.util
-
-    if importlib.util.find_spec("tensorflow") is None:
-        structlogger.warning(
-            "model_training.train.tensorflow_not_available",
-            event_info=(
-                "TensorFlow is not installed. Components requiring TensorFlow "
-                "(DIETClassifier, TEDPolicy, UnexpecTEDIntentPolicy, ResponseSelector, "
-                "ConveRTFeaturizer, LanguageModelFeaturizer) "
-                "will not be available."
-            ),
-        )
-
     stories = file_importer.get_stories()
     flows = file_importer.get_flows()
     nlu_data = file_importer.get_nlu_data()

rasa/nlu/classifiers/logistic_regression_classifier.py

@@ -1,5 +1,6 @@
 from typing import Any, Dict, List, Text, Tuple, Type
 
+import skops.io as sio
 import structlog
 from scipy.sparse import csr_matrix, hstack, vstack
 from sklearn.exceptions import NotFittedError
@@ -21,21 +22,6 @@ from rasa.utils.tensorflow.constants import RANKING_LENGTH
 structlogger = structlog.get_logger()
 
 
-def _check_skops_dependency() -> None:
-    """Check if skops is available and raise error if not."""
-    import importlib.util
-
-    if importlib.util.find_spec("skops") is None:
-        from rasa.exceptions import MissingDependencyException
-
-        raise MissingDependencyException(
-            "The LogisticRegressionClassifier requires the 'skops' package "
-            "to save and load trained models. "
-            "To install it, run: `pip install rasa[skops]` or `pip install skops`. "
-            "If using poetry, run `poetry install --extras skops`. "
-        )
-
-
 @DefaultV1Recipe.register(
     DefaultV1Recipe.ComponentType.INTENT_CLASSIFIER, is_trainable=True
 )
@@ -166,7 +152,6 @@ class LogisticRegressionClassifier(IntentClassifier, GraphComponent):
         execution_context: ExecutionContext,
     ) -> "LogisticRegressionClassifier":
         """Creates a new untrained component (see parent class for full docstring)."""
-        _check_skops_dependency()
         return cls(config, execution_context.node_name, model_storage, resource)
 
     def process(self, messages: List[Message]) -> List[Message]:
@@ -199,9 +184,6 @@ class LogisticRegressionClassifier(IntentClassifier, GraphComponent):
 
     def persist(self) -> None:
         """Persist this model into the passed directory."""
-        _check_skops_dependency()
-        import skops.io as sio
-
         with self._model_storage.write_to(self._resource) as model_dir:
             path = model_dir / f"{self._resource.name}.skops"
             sio.dump(self.clf, path)
@@ -220,9 +202,6 @@ class LogisticRegressionClassifier(IntentClassifier, GraphComponent):
         **kwargs: Any,
     ) -> "LogisticRegressionClassifier":
         """Loads trained component (see parent class for full docstring)."""
-        _check_skops_dependency()
-        import skops.io as sio
-
         try:
             with model_storage.read_from(resource) as model_dir:
                 classifier_file = model_dir / f"{resource.name}.skops"