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

rasa/core/policies/policy.py

@@ -6,45 +6,47 @@ import logging
 from enum import Enum
 from pathlib import Path
 from typing import (
+    TYPE_CHECKING,
     Any,
+    Callable,
+    Dict,
     List,
     Optional,
     Text,
-    Dict,
-    Callable,
     Tuple,
     TypeVar,
-    TYPE_CHECKING,
 )
 
 import numpy as np
 
-from rasa.shared.constants import ROUTE_TO_CALM_SLOT
-from rasa.shared.core.events import Event
-from rasa.engine.graph import GraphComponent, ExecutionContext
-from rasa.engine.storage.resource import Resource
-from rasa.engine.storage.storage import ModelStorage
-from rasa.core.featurizers.precomputation import MessageContainerForCoreFeaturization
-import rasa.utils.common
+import rasa.shared.utils.common
 import rasa.shared.utils.io
-from rasa.shared.exceptions import RasaException, FileIOException
-from rasa.shared.nlu.constants import ENTITIES, INTENT, TEXT, ACTION_TEXT, ACTION_NAME
-from rasa.shared.core.domain import Domain, State
-from rasa.shared.core.trackers import DialogueStateTracker
-from rasa.shared.core.generator import TrackerWithCachedStates
+import rasa.utils.common
 from rasa.core.constants import (
     DEFAULT_POLICY_PRIORITY,
-    POLICY_PRIORITY,
     POLICY_MAX_HISTORY,
+    POLICY_PRIORITY,
 )
-from rasa.shared.core.constants import USER, SLOTS, PREVIOUS_ACTION, ACTIVE_LOOP
-import rasa.shared.utils.common
+from rasa.core.featurizers.precomputation import MessageContainerForCoreFeaturization
+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 ROUTE_TO_CALM_SLOT
+from rasa.shared.core.constants import ACTIVE_LOOP, PREVIOUS_ACTION, SLOTS, USER
+from rasa.shared.core.domain import Domain, State
+from rasa.shared.core.events import Event
+from rasa.shared.core.generator import TrackerWithCachedStates
+from rasa.shared.core.trackers import DialogueStateTracker
+from rasa.shared.exceptions import FileIOException, RasaException
+from rasa.shared.nlu.constants import ACTION_NAME, ACTION_TEXT, ENTITIES, INTENT, TEXT
 
 if TYPE_CHECKING:
-    from rasa.shared.nlu.training_data.features import Features
-    from rasa.core.featurizers.tracker_featurizers import TrackerFeaturizer
-    from rasa.core.featurizers.tracker_featurizers import MaxHistoryTrackerFeaturizer
+    from rasa.core.featurizers.tracker_featurizers import (
+        MaxHistoryTrackerFeaturizer,
+        TrackerFeaturizer,
+    )
     from rasa.dialogue_understanding.stack.frames import DialogueStackFrame
+    from rasa.shared.nlu.training_data.features import Features
 
 logger = logging.getLogger(__name__)
 
@@ -467,8 +469,8 @@ class Policy(GraphComponent):
     ) -> Policy:
         """Loads a trained policy (see parent class for full docstring)."""
         from rasa.core.featurizers.tracker_featurizers import (
-            TrackerFeaturizer,
             FEATURIZER_FILE,
+            TrackerFeaturizer,
         )
 
         featurizer = None

rasa/core/policies/rule_policy.py

@@ -1,60 +1,61 @@
 from __future__ import annotations
+
 import copy
 import functools
+import json
 import logging
-import structlog
-from typing import Any, List, DefaultDict, Dict, Text, Optional, Set, Tuple, cast
+from collections import defaultdict
+from typing import Any, DefaultDict, Dict, List, Optional, Set, Text, Tuple, cast
 
-from tqdm import tqdm
 import numpy as np
-import json
-from collections import defaultdict
+import structlog
+from tqdm import tqdm
 
-from rasa.engine.graph import ExecutionContext
-from rasa.engine.recipes.default_recipe import DefaultV1Recipe
-from rasa.engine.storage.resource import Resource
-from rasa.engine.storage.storage import ModelStorage
-from rasa.shared.constants import DOCS_URL_RULES
-from rasa.shared.exceptions import RasaException
+import rasa.core.test
 import rasa.shared.utils.io
-from rasa.shared.core.events import LoopInterrupted, UserUttered, ActionExecuted
-from rasa.core.featurizers.tracker_featurizers import TrackerFeaturizer
-from rasa.core.policies.memoization import MemoizationPolicy
-from rasa.core.policies.policy import SupportedData, PolicyPrediction
-from rasa.shared.core.trackers import (
-    DialogueStateTracker,
-    get_active_loop_name,
-    is_prev_action_listen_in_state,
-)
-from rasa.shared.core.generator import TrackerWithCachedStates
 from rasa.core.constants import (
     DEFAULT_CORE_FALLBACK_THRESHOLD,
-    RULE_POLICY_PRIORITY,
-    POLICY_PRIORITY,
     POLICY_MAX_HISTORY,
+    POLICY_PRIORITY,
+    RULE_POLICY_PRIORITY,
 )
+from rasa.core.featurizers.tracker_featurizers import TrackerFeaturizer
+from rasa.core.policies.memoization import MemoizationPolicy
+from rasa.core.policies.policy import PolicyPrediction, SupportedData
+from rasa.core.training.training import ActionFingerprint, create_action_fingerprints
+from rasa.engine.graph import ExecutionContext
+from rasa.engine.recipes.default_recipe import DefaultV1Recipe
+from rasa.engine.storage.resource import Resource
+from rasa.engine.storage.storage import ModelStorage
+from rasa.shared.constants import DOCS_URL_RULES
 from rasa.shared.core.constants import (
-    USER_INTENT_RESTART,
-    USER_INTENT_BACK,
-    USER_INTENT_SESSION_START,
+    ACTION_BACK_NAME,
+    ACTION_DEFAULT_FALLBACK_NAME,
     ACTION_LISTEN_NAME,
     ACTION_RESTART_NAME,
     ACTION_SESSION_START_NAME,
-    ACTION_DEFAULT_FALLBACK_NAME,
-    ACTION_BACK_NAME,
+    ACTIVE_LOOP,
+    LOOP_NAME,
+    PREVIOUS_ACTION,
+    RULE_ONLY_LOOPS,
+    RULE_ONLY_SLOTS,
     RULE_SNIPPET_ACTION_NAME,
     SHOULD_NOT_BE_SET,
-    PREVIOUS_ACTION,
-    LOOP_NAME,
     SLOTS,
-    ACTIVE_LOOP,
-    RULE_ONLY_SLOTS,
-    RULE_ONLY_LOOPS,
+    USER_INTENT_BACK,
+    USER_INTENT_RESTART,
+    USER_INTENT_SESSION_START,
 )
-from rasa.shared.core.domain import InvalidDomain, State, Domain
+from rasa.shared.core.domain import Domain, InvalidDomain, State
+from rasa.shared.core.events import ActionExecuted, LoopInterrupted, UserUttered
+from rasa.shared.core.generator import TrackerWithCachedStates
+from rasa.shared.core.trackers import (
+    DialogueStateTracker,
+    get_active_loop_name,
+    is_prev_action_listen_in_state,
+)
+from rasa.shared.exceptions import RasaException
 from rasa.shared.nlu.constants import ACTION_NAME, INTENT_NAME_KEY
-import rasa.core.test
-from rasa.core.training.training import create_action_fingerprints, ActionFingerprint
 
 logger = logging.getLogger(__name__)
 structlogger = structlog.get_logger()

rasa/core/policies/ted_policy.py

@@ -1,131 +1,133 @@
 from __future__ import annotations
 
+import contextlib
 import logging
-from pathlib import Path
 from collections import defaultdict
-import contextlib
-from typing import Any, List, Optional, Text, Dict, Tuple, Union, Type
+from pathlib import Path
+from typing import Any, Dict, List, Optional, Text, Tuple, Type, Union
 
 import numpy as np
 import tensorflow as tf
 
-from rasa.engine.recipes.default_recipe import DefaultV1Recipe
+import rasa.core.actions.action
+import rasa.shared.utils.io
+import rasa.utils.io
+from rasa.core.constants import (
+    DEFAULT_MAX_HISTORY,
+    DEFAULT_POLICY_PRIORITY,
+    DIALOGUE,
+    POLICY_MAX_HISTORY,
+    POLICY_PRIORITY,
+)
+from rasa.core.featurizers.precomputation import MessageContainerForCoreFeaturization
+from rasa.core.featurizers.tracker_featurizers import (
+    MaxHistoryTrackerFeaturizer,
+    TrackerFeaturizer,
+)
+from rasa.core.policies.policy import Policy, PolicyPrediction, SupportedData
 from rasa.engine.graph import ExecutionContext
+from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.exceptions import ModelNotFound
 from rasa.nlu.constants import TOKENS_NAMES
-from rasa.nlu.extractors.extractor import EntityTagSpec, EntityExtractorMixin
-import rasa.core.actions.action
-from rasa.core.featurizers.precomputation import MessageContainerForCoreFeaturization
-from rasa.core.featurizers.tracker_featurizers import TrackerFeaturizer
-from rasa.core.featurizers.tracker_featurizers import MaxHistoryTrackerFeaturizer
+from rasa.nlu.extractors.extractor import EntityExtractorMixin, EntityTagSpec
+from rasa.shared.constants import DIAGNOSTIC_DATA
+from rasa.shared.core.constants import ACTION_LISTEN_NAME, ACTIVE_LOOP, SLOTS
+from rasa.shared.core.domain import Domain
+from rasa.shared.core.events import EntitiesAdded, Event
+from rasa.shared.core.generator import TrackerWithCachedStates
+from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.exceptions import RasaException
 from rasa.shared.nlu.constants import (
-    ACTION_TEXT,
     ACTION_NAME,
-    INTENT,
-    TEXT,
+    ACTION_TEXT,
     ENTITIES,
-    FEATURE_TYPE_SENTENCE,
     ENTITY_ATTRIBUTE_TYPE,
     ENTITY_TAGS,
     EXTRACTOR,
+    FEATURE_TYPE_SENTENCE,
+    INTENT,
     SPLIT_ENTITIES_BY_COMMA,
     SPLIT_ENTITIES_BY_COMMA_DEFAULT_VALUE,
+    TEXT,
 )
-from rasa.core.policies.policy import PolicyPrediction, Policy, SupportedData
-from rasa.core.constants import (
-    DIALOGUE,
-    POLICY_MAX_HISTORY,
-    DEFAULT_MAX_HISTORY,
-    DEFAULT_POLICY_PRIORITY,
-    POLICY_PRIORITY,
-)
-from rasa.shared.constants import DIAGNOSTIC_DATA
-from rasa.shared.core.constants import ACTIVE_LOOP, SLOTS, ACTION_LISTEN_NAME
-from rasa.shared.core.trackers import DialogueStateTracker
-from rasa.shared.core.generator import TrackerWithCachedStates
-from rasa.shared.core.events import EntitiesAdded, Event
-from rasa.shared.core.domain import Domain
-from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.nlu.training_data.features import (
     Features,
-    save_features,
     load_features,
+    save_features,
 )
-import rasa.shared.utils.io
-import rasa.utils.io
+from rasa.shared.nlu.training_data.message import Message
 from rasa.utils import train_utils
-from rasa.utils.tensorflow.feature_array import (
-    FeatureArray,
-    serialize_nested_feature_arrays,
-    deserialize_nested_feature_arrays,
-)
-from rasa.utils.tensorflow.models import RasaModel, TransformerRasaModel
 from rasa.utils.tensorflow import rasa_layers
-from rasa.utils.tensorflow.model_data import RasaModelData, FeatureSignature, Data
-from rasa.utils.tensorflow.model_data_utils import convert_to_data_format
 from rasa.utils.tensorflow.constants import (
-    LABEL,
-    IDS,
-    TRANSFORMER_SIZE,
-    NUM_TRANSFORMER_LAYERS,
-    NUM_HEADS,
+    AUTO,
+    BALANCED,
     BATCH_SIZES,
     BATCH_STRATEGY,
+    BILOU_FLAG,
+    CHECKPOINT_MODEL,
+    CONCAT_DIMENSION,
+    CONNECTION_DENSITY,
+    CONSTRAIN_SIMILARITIES,
+    CROSS_ENTROPY,
+    DENSE_DIMENSION,
+    DENSE_INPUT_DROPOUT,
+    DROP_RATE,
+    DROP_RATE_ATTENTION,
+    DROP_RATE_DIALOGUE,
+    DROP_RATE_LABEL,
+    EMBEDDING_DIMENSION,
+    ENCODING_DIMENSION,
+    ENTITY_RECOGNITION,
+    EPOCH_OVERRIDE,
     EPOCHS,
-    RANDOM_SEED,
+    EVAL_NUM_EPOCHS,
+    EVAL_NUM_EXAMPLES,
+    FEATURIZERS,
+    HIDDEN_LAYERS_SIZES,
+    IDS,
+    KEY_RELATIVE_ATTENTION,
+    LABEL,
     LEARNING_RATE,
-    RANKING_LENGTH,
-    RENORMALIZE_CONFIDENCES,
     LOSS_TYPE,
-    SIMILARITY_TYPE,
-    NUM_NEG,
-    EVAL_NUM_EXAMPLES,
-    EVAL_NUM_EPOCHS,
-    NEGATIVE_MARGIN_SCALE,
-    REGULARIZATION_CONSTANT,
-    SCALE_LOSS,
-    USE_MAX_NEG_SIM,
+    MASK,
+    MASKED_LM,
     MAX_NEG_SIM,
     MAX_POS_SIM,
-    EMBEDDING_DIMENSION,
-    DROP_RATE_DIALOGUE,
-    DROP_RATE_LABEL,
-    DROP_RATE,
-    DROP_RATE_ATTENTION,
-    CONNECTION_DENSITY,
-    KEY_RELATIVE_ATTENTION,
-    VALUE_RELATIVE_ATTENTION,
     MAX_RELATIVE_POSITION,
-    CROSS_ENTROPY,
-    AUTO,
-    BALANCED,
-    TENSORBOARD_LOG_DIR,
-    TENSORBOARD_LOG_LEVEL,
-    CHECKPOINT_MODEL,
-    ENCODING_DIMENSION,
-    UNIDIRECTIONAL_ENCODER,
-    SEQUENCE,
+    MODEL_CONFIDENCE,
+    NEGATIVE_MARGIN_SCALE,
+    NUM_HEADS,
+    NUM_NEG,
+    NUM_TRANSFORMER_LAYERS,
+    RANDOM_SEED,
+    RANKING_LENGTH,
+    REGULARIZATION_CONSTANT,
+    RENORMALIZE_CONFIDENCES,
+    SCALE_LOSS,
     SENTENCE,
+    SEQUENCE,
     SEQUENCE_LENGTH,
-    DENSE_DIMENSION,
-    CONCAT_DIMENSION,
-    SPARSE_INPUT_DROPOUT,
-    DENSE_INPUT_DROPOUT,
-    MASKED_LM,
-    MASK,
-    HIDDEN_LAYERS_SIZES,
-    FEATURIZERS,
-    ENTITY_RECOGNITION,
-    CONSTRAIN_SIMILARITIES,
-    MODEL_CONFIDENCE,
+    SIMILARITY_TYPE,
     SOFTMAX,
-    BILOU_FLAG,
-    EPOCH_OVERRIDE,
+    SPARSE_INPUT_DROPOUT,
+    TENSORBOARD_LOG_DIR,
+    TENSORBOARD_LOG_LEVEL,
+    TRANSFORMER_SIZE,
+    UNIDIRECTIONAL_ENCODER,
     USE_GPU,
+    USE_MAX_NEG_SIM,
+    VALUE_RELATIVE_ATTENTION,
 )
+from rasa.utils.tensorflow.feature_array import (
+    FeatureArray,
+    deserialize_nested_feature_arrays,
+    serialize_nested_feature_arrays,
+)
+from rasa.utils.tensorflow.model_data import Data, FeatureSignature, RasaModelData
+from rasa.utils.tensorflow.model_data_utils import convert_to_data_format
+from rasa.utils.tensorflow.models import RasaModel, TransformerRasaModel
 
 logger = logging.getLogger(__name__)
 

rasa/core/policies/unexpected_intent_policy.py

@@ -1,112 +1,114 @@
 import dataclasses
 import logging
 from pathlib import Path
-from typing import Any, List, Optional, Text, Dict, Type, Union
+from typing import Any, Dict, List, Optional, Text, Type, Union
 
 import numpy as np
 import tensorflow as tf
 
-import rasa.utils.common
-from rasa.engine.graph import ExecutionContext
-from rasa.engine.recipes.default_recipe import DefaultV1Recipe
-from rasa.engine.storage.resource import Resource
-from rasa.engine.storage.storage import ModelStorage
-from rasa.nlu.classifiers import LABEL_RANKING_LENGTH
-from rasa.shared.nlu.training_data.features import Features
-from rasa.shared.core.domain import Domain
-from rasa.shared.core.trackers import DialogueStateTracker
-from rasa.shared.core.constants import SLOTS, ACTIVE_LOOP, ACTION_UNLIKELY_INTENT_NAME
-from rasa.shared.core.events import UserUttered, ActionExecuted
 import rasa.shared.utils.io
-from rasa.shared.nlu.constants import (
-    INTENT,
-    TEXT,
-    ENTITIES,
-    ACTION_NAME,
-    SPLIT_ENTITIES_BY_COMMA,
-    SPLIT_ENTITIES_BY_COMMA_DEFAULT_VALUE,
-)
-from rasa.nlu.extractors.extractor import EntityTagSpec
-from rasa.core.featurizers.precomputation import MessageContainerForCoreFeaturization
-from rasa.core.featurizers.tracker_featurizers import TrackerFeaturizer
-from rasa.core.featurizers.tracker_featurizers import IntentMaxHistoryTrackerFeaturizer
-from rasa.core.featurizers.single_state_featurizer import (
-    IntentTokenizerSingleStateFeaturizer,
-)
-from rasa.shared.core.generator import TrackerWithCachedStates
+import rasa.utils.common
 from rasa.core.constants import (
     DIALOGUE,
     POLICY_MAX_HISTORY,
     POLICY_PRIORITY,
     UNLIKELY_INTENT_POLICY_PRIORITY,
 )
+from rasa.core.exceptions import RasaCoreException
+from rasa.core.featurizers.precomputation import MessageContainerForCoreFeaturization
+from rasa.core.featurizers.single_state_featurizer import (
+    IntentTokenizerSingleStateFeaturizer,
+)
+from rasa.core.featurizers.tracker_featurizers import (
+    IntentMaxHistoryTrackerFeaturizer,
+    TrackerFeaturizer,
+)
 from rasa.core.policies.policy import PolicyPrediction
 from rasa.core.policies.ted_policy import (
     LABEL_KEY,
     LABEL_SUB_KEY,
-    TEDPolicy,
-    TED,
-    SEQUENCE_LENGTH,
-    SEQUENCE,
     PREDICTION_FEATURES,
+    SEQUENCE,
+    SEQUENCE_LENGTH,
+    TED,
+    TEDPolicy,
 )
+from rasa.engine.graph import ExecutionContext
+from rasa.engine.recipes.default_recipe import DefaultV1Recipe
+from rasa.engine.storage.resource import Resource
+from rasa.engine.storage.storage import ModelStorage
+from rasa.nlu.classifiers import LABEL_RANKING_LENGTH
+from rasa.nlu.extractors.extractor import EntityTagSpec
+from rasa.shared.core.constants import ACTION_UNLIKELY_INTENT_NAME, ACTIVE_LOOP, SLOTS
+from rasa.shared.core.domain import Domain
+from rasa.shared.core.events import ActionExecuted, UserUttered
+from rasa.shared.core.generator import TrackerWithCachedStates
+from rasa.shared.core.trackers import DialogueStateTracker
+from rasa.shared.nlu.constants import (
+    ACTION_NAME,
+    ENTITIES,
+    INTENT,
+    SPLIT_ENTITIES_BY_COMMA,
+    SPLIT_ENTITIES_BY_COMMA_DEFAULT_VALUE,
+    TEXT,
+)
+from rasa.shared.nlu.training_data.features import Features
+from rasa.shared.utils import common
 from rasa.utils import train_utils
-from rasa.utils.tensorflow.models import RasaModel
+from rasa.utils.tensorflow import layers
 from rasa.utils.tensorflow.constants import (
-    LABEL,
-    DENSE_DIMENSION,
-    ENCODING_DIMENSION,
-    UNIDIRECTIONAL_ENCODER,
-    TRANSFORMER_SIZE,
-    NUM_TRANSFORMER_LAYERS,
-    NUM_HEADS,
+    BALANCED,
     BATCH_SIZES,
     BATCH_STRATEGY,
+    BILOU_FLAG,
+    CHECKPOINT_MODEL,
+    CONCAT_DIMENSION,
+    CONNECTION_DENSITY,
+    CROSS_ENTROPY,
+    DENSE_DIMENSION,
+    DENSE_INPUT_DROPOUT,
+    DROP_RATE,
+    DROP_RATE_ATTENTION,
+    DROP_RATE_DIALOGUE,
+    DROP_RATE_LABEL,
+    EMBEDDING_DIMENSION,
+    ENCODING_DIMENSION,
+    ENTITY_RECOGNITION,
     EPOCHS,
-    RANDOM_SEED,
-    RANKING_LENGTH,
+    EVAL_NUM_EPOCHS,
+    EVAL_NUM_EXAMPLES,
+    FEATURIZERS,
+    HIDDEN_LAYERS_SIZES,
+    IGNORE_INTENTS_LIST,
+    INNER,
+    KEY_RELATIVE_ATTENTION,
+    LABEL,
+    LABEL_PAD_ID,
+    LEARNING_RATE,
     LOSS_TYPE,
-    SIMILARITY_TYPE,
+    MASKED_LM,
+    MAX_RELATIVE_POSITION,
+    NEGATIVE_SCORES_KEY,
+    NUM_HEADS,
     NUM_NEG,
-    EVAL_NUM_EXAMPLES,
-    EVAL_NUM_EPOCHS,
+    NUM_TRANSFORMER_LAYERS,
+    POSITIVE_SCORES_KEY,
+    RANDOM_SEED,
+    RANKING_LENGTH,
     REGULARIZATION_CONSTANT,
     SCALE_LOSS,
-    EMBEDDING_DIMENSION,
-    DROP_RATE_DIALOGUE,
-    DROP_RATE_LABEL,
-    DROP_RATE,
-    DROP_RATE_ATTENTION,
-    CONNECTION_DENSITY,
-    KEY_RELATIVE_ATTENTION,
-    VALUE_RELATIVE_ATTENTION,
-    MAX_RELATIVE_POSITION,
-    INNER,
-    BALANCED,
+    SIMILARITY_TYPE,
+    SPARSE_INPUT_DROPOUT,
     TENSORBOARD_LOG_DIR,
     TENSORBOARD_LOG_LEVEL,
-    CHECKPOINT_MODEL,
-    FEATURIZERS,
-    ENTITY_RECOGNITION,
-    IGNORE_INTENTS_LIST,
-    BILOU_FLAG,
-    LEARNING_RATE,
-    CROSS_ENTROPY,
-    SPARSE_INPUT_DROPOUT,
-    DENSE_INPUT_DROPOUT,
-    MASKED_LM,
-    HIDDEN_LAYERS_SIZES,
-    CONCAT_DIMENSION,
     TOLERANCE,
-    LABEL_PAD_ID,
-    POSITIVE_SCORES_KEY,
-    NEGATIVE_SCORES_KEY,
+    TRANSFORMER_SIZE,
+    UNIDIRECTIONAL_ENCODER,
     USE_GPU,
+    VALUE_RELATIVE_ATTENTION,
 )
-from rasa.utils.tensorflow import layers
-from rasa.utils.tensorflow.model_data import RasaModelData, FeatureArray, Data
-from rasa.core.exceptions import RasaCoreException
-from rasa.shared.utils import common
+from rasa.utils.tensorflow.model_data import Data, FeatureArray, RasaModelData
+from rasa.utils.tensorflow.models import RasaModel
 
 
 @dataclasses.dataclass