rasa-pro 3.10.4__py3-none-any.whl → 3.10.6__py3-none-any.whl

rasa/core/policies/intentless_policy.py

@@ -80,6 +80,8 @@ from rasa.utils.ml_utils import (
     persist_faiss_vector_store,
     response_for_template,
 )
+from rasa.dialogue_understanding.patterns.chitchat import FLOW_PATTERN_CHITCHAT
+from rasa.shared.core.constants import ACTION_TRIGGER_CHITCHAT
 from rasa.utils.log_utils import log_llm
 
 if TYPE_CHECKING:
@@ -174,6 +176,21 @@ def filter_responses(responses: Responses, forms: Forms, flows: FlowsList) -> Re
         for name, variants in responses.data.items()
         if name not in combined_responses
     }
+
+    pattern_chitchat = flows.flow_by_id(FLOW_PATTERN_CHITCHAT)
+
+    # The following condition is highly unlikely, but mypy requires the case
+    # of pattern_chitchat == None to be addressed
+    if not pattern_chitchat:
+        return Responses(data=filtered_responses)
+
+    # if action_trigger_chitchat, filter out "utter_free_chitchat_response"
+    has_action_trigger_chitchat = pattern_chitchat.has_action_step(
+        ACTION_TRIGGER_CHITCHAT
+    )
+    if has_action_trigger_chitchat:
+        filtered_responses.pop("utter_free_chitchat_response", None)
+
     return Responses(data=filtered_responses)
 
 
@@ -709,6 +726,7 @@ class IntentlessPolicy(Policy):
             number_of_samples=NUMBER_OF_CONVERSATION_SAMPLES,
             max_number_of_tokens=MAX_NUMBER_OF_TOKENS_FOR_SAMPLES,
         )
+
         extra_ai_responses = self.extract_ai_responses(conversation_samples)
 
         # put conversation responses in front of sampled examples,

rasa/dialogue_understanding/processor/command_processor.py

@@ -1,6 +1,7 @@
 from typing import List, Optional, Type, Set, Dict
 
 import structlog
+from rasa.shared.core.training_data.structures import StoryGraph
 from rasa.dialogue_understanding.commands import (
     CancelFlowCommand,
     ClarifyCommand,
@@ -179,6 +180,7 @@ def execute_commands(
     tracker: DialogueStateTracker,
     all_flows: FlowsList,
     execution_context: ExecutionContext,
+    story_graph: Optional[StoryGraph] = None,
 ) -> List[Event]:
     """Executes a list of commands.
 
@@ -187,6 +189,7 @@ def execute_commands(
         tracker: The tracker to execute the commands on.
         all_flows: All flows.
         execution_context: Information about the single graph run.
+        story_graph: StoryGraph object with stories available for training.
 
     Returns:
         A list of the events that were created.
@@ -194,7 +197,9 @@ def execute_commands(
     commands: List[Command] = get_commands_from_tracker(tracker)
     original_tracker = tracker.copy()
 
-    commands = clean_up_commands(commands, tracker, all_flows, execution_context)
+    commands = clean_up_commands(
+        commands, tracker, all_flows, execution_context, story_graph
+    )
 
     updated_flows = find_updated_flows(tracker, all_flows)
     if updated_flows:
@@ -326,6 +331,7 @@ def clean_up_commands(
     tracker: DialogueStateTracker,
     all_flows: FlowsList,
     execution_context: ExecutionContext,
+    story_graph: Optional[StoryGraph] = None,
 ) -> List[Command]:
     """Clean up a list of commands.
 
@@ -340,6 +346,7 @@ def clean_up_commands(
         tracker: The tracker to clean up the commands for.
         all_flows: All flows.
         execution_context: Information about a single graph run.
+        story_graph: StoryGraph object with stories available for training.
 
     Returns:
     The cleaned up commands.
@@ -386,7 +393,7 @@ def clean_up_commands(
         # handle chitchat command differently from other free-form answer commands
         elif isinstance(command, ChitChatAnswerCommand):
             clean_commands = clean_up_chitchat_command(
-                clean_commands, command, all_flows, execution_context
+                clean_commands, command, all_flows, execution_context, story_graph
             )
 
         elif isinstance(command, FreeFormAnswerCommand):
@@ -487,22 +494,23 @@ def clean_up_slot_command(
     stack = tracker.stack
 
     resulting_commands = commands_so_far[:]
-    if command.name in slots_so_far and command.name != ROUTE_TO_CALM_SLOT:
-        slot = tracker.slots.get(command.name)
-        if slot is None:
-            structlogger.debug(
-                "command_processor.clean_up_slot_command.skip_command_slot_not_in_domain",
-                command=command,
-            )
-            return resulting_commands
 
-        if not should_slot_be_set(slot, command):
-            cannot_handle = CannotHandleCommand(reason=CANNOT_HANDLE_REASON)
-            if cannot_handle not in resulting_commands:
-                resulting_commands.append(cannot_handle)
+    slot = tracker.slots.get(command.name)
+    if slot is None:
+        structlogger.debug(
+            "command_processor.clean_up_slot_command.skip_command_slot_not_in_domain",
+            command=command,
+        )
+        return resulting_commands
+
+    if not should_slot_be_set(slot, command):
+        cannot_handle = CannotHandleCommand(reason=CANNOT_HANDLE_REASON)
+        if cannot_handle not in resulting_commands:
+            resulting_commands.append(cannot_handle)
 
-            return resulting_commands
+        return resulting_commands
 
+    if command.name in slots_so_far and command.name != ROUTE_TO_CALM_SLOT:
         current_collect_info = get_current_collect_step(stack, all_flows)
 
         if current_collect_info and current_collect_info.collect == command.name:
@@ -562,6 +570,7 @@ def clean_up_chitchat_command(
     command: ChitChatAnswerCommand,
     flows: FlowsList,
     execution_context: ExecutionContext,
+    story_graph: Optional[StoryGraph] = None,
 ) -> List[Command]:
     """Clean up a chitchat answer command.
 
@@ -573,6 +582,7 @@ def clean_up_chitchat_command(
         command: The command to clean up.
         flows: All flows.
         execution_context: Information about a single graph run.
+        story_graph: StoryGraph object with stories available for training.
     Returns:
         The cleaned up commands.
     """
@@ -598,7 +608,11 @@ def clean_up_chitchat_command(
     )
     defines_intentless_policy = execution_context.has_node(IntentlessPolicy)
 
-    if has_action_trigger_chitchat and not defines_intentless_policy:
+    has_e2e_stories = True if (story_graph and story_graph.has_e2e_stories()) else False
+
+    if (has_action_trigger_chitchat and not defines_intentless_policy) or (
+        defines_intentless_policy and not has_e2e_stories
+    ):
         resulting_commands.insert(
             0, CannotHandleCommand(RASA_PATTERN_CANNOT_HANDLE_CHITCHAT)
         )

rasa/dialogue_understanding/processor/command_processor_component.py

@@ -9,6 +9,7 @@ from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.core.events import Event
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.trackers import DialogueStateTracker
+from rasa.shared.core.training_data.structures import StoryGraph
 
 
 class CommandProcessorComponent(GraphComponent):
@@ -31,9 +32,12 @@ class CommandProcessorComponent(GraphComponent):
         return cls(execution_context)
 
     def execute_commands(
-        self, tracker: DialogueStateTracker, flows: FlowsList
+        self,
+        tracker: DialogueStateTracker,
+        flows: FlowsList,
+        story_graph: StoryGraph,
     ) -> List[Event]:
         """Execute commands to update tracker state."""
         return rasa.dialogue_understanding.processor.command_processor.execute_commands(
-            tracker, flows, self._execution_context
+            tracker, flows, self._execution_context, story_graph
         )

rasa/engine/recipes/default_recipe.py

@@ -405,7 +405,7 @@ class DefaultV1Recipe(Recipe):
             return {}
 
         def resolver_name_from_parameter(parameter: str) -> str:
-            # we got a couple special cases to handle wher the parameter name
+            # we got a couple special cases to handle where the parameter name
             # doesn't match the provider name
             if "training_trackers" == parameter:
                 return "training_tracker_provider"
@@ -597,7 +597,7 @@ class DefaultV1Recipe(Recipe):
             needs={"importer": "finetuning_validator"},
             uses=StoryGraphProvider,
             constructor_name="create",
-            fn="provide",
+            fn="provide_train",
             config={"exclusion_percentage": cli_parameters.get("exclusion_percentage")},
             is_input=True,
         )
@@ -882,6 +882,14 @@ class DefaultV1Recipe(Recipe):
             config={},
             resource=Resource("domain_provider"),
         )
+        predict_nodes["story_graph_provider"] = SchemaNode(
+            **DEFAULT_PREDICT_KWARGS,
+            needs={},
+            uses=StoryGraphProvider,
+            fn="provide_inference",
+            config={},
+            resource=Resource("story_graph_provider"),
+        )
         predict_nodes["flows_provider"] = SchemaNode(
             **DEFAULT_PREDICT_KWARGS,
             needs={},

rasa/graph_components/providers/story_graph_provider.py

@@ -1,19 +1,37 @@
 from __future__ import annotations
-from typing import Dict, Text, Any
+from typing import Dict, Text, Any, List
 
 from rasa.engine.graph import GraphComponent, ExecutionContext
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
-from rasa.shared.core.training_data.structures import StoryGraph
+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,
+)
+
+
+STORIES_PERSISTENCE_FILE_NAME = "stories.yml"
 
 
 class StoryGraphProvider(GraphComponent):
     """Provides the training data from stories."""
 
-    def __init__(self, config: Dict[Text, Any]) -> None:
+    def __init__(
+        self,
+        config: Dict[Text, Any],
+        model_storage: ModelStorage,
+        resource: Resource,
+        stories: StoryGraph = None,
+    ) -> None:
         """Creates provider from config."""
         self._config = config
+        self._model_storage = model_storage
+        self._resource = resource
+        self._stories = stories
 
     @staticmethod
     def get_default_config() -> Dict[Text, Any]:
@@ -29,9 +47,36 @@ class StoryGraphProvider(GraphComponent):
         execution_context: ExecutionContext,
     ) -> StoryGraphProvider:
         """Creates component (see parent class for full docstring)."""
-        return cls(config)
+        return cls(config, model_storage, resource)
 
-    def provide(self, importer: TrainingDataImporter) -> StoryGraph:
+    @classmethod
+    def load(
+        cls,
+        config: Dict[Text, Any],
+        model_storage: ModelStorage,
+        resource: Resource,
+        execution_context: ExecutionContext,
+        **kwargs: Any,
+    ) -> StoryGraphProvider:
+        """Creates provider using a persisted version of itself."""
+        with model_storage.read_from(resource) as resource_directory:
+            reader = YAMLStoryReader()
+            story_steps = reader.read_from_file(
+                resource_directory / STORIES_PERSISTENCE_FILE_NAME
+            )
+        stories = StoryGraph(story_steps)
+        return cls(config, model_storage, resource, stories)
+
+    def _persist(self, story_steps: List[StoryStep]) -> None:
+        """Persists flows to model storage."""
+        with self._model_storage.write_to(self._resource) as resource_directory:
+            writer = YAMLStoryWriter()
+            writer.dump(
+                resource_directory / STORIES_PERSISTENCE_FILE_NAME,
+                story_steps,
+            )
+
+    def provide_train(self, importer: TrainingDataImporter) -> StoryGraph:
         """Provides the story graph from the training data.
 
         Args:
@@ -40,4 +85,12 @@ class StoryGraphProvider(GraphComponent):
         Returns:
             The story graph containing stories and rules used for training.
         """
-        return importer.get_stories(**self._config)
+        stories = importer.get_stories(**self._config)
+        self._persist(stories.story_steps)
+        return stories
+
+    def provide_inference(self) -> StoryGraph:
+        """Provides the stories configuration during inference."""
+        if self._stories is None:
+            self._stories = StoryGraph([])
+        return self._stories

rasa/nlu/classifiers/logistic_regression_classifier.py

@@ -3,7 +3,9 @@ from typing import Any, Text, Dict, List, Type, Tuple
 import joblib
 import structlog
 from scipy.sparse import hstack, vstack, csr_matrix
+from sklearn.exceptions import NotFittedError
 from sklearn.linear_model import LogisticRegression
+from sklearn.utils.validation import check_is_fitted
 
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
@@ -154,6 +156,17 @@ class LogisticRegressionClassifier(IntentClassifier, GraphComponent):
 
     def process(self, messages: List[Message]) -> List[Message]:
         """Return the most likely intent and its probability for a message."""
+        # Check if the classifier is trained
+        if not self.is_trained():
+            structlogger.warning(
+                "logistic_regression_classifier.not_trained.skip_intent_prediction",
+                event_info=(
+                    f"The '{self.__class__.__name__}' is not trained. "
+                    f"Skipping intent prediction."
+                ),
+            )
+            return messages
+
         X = self._create_X(messages)
         probas = self.clf.predict_proba(X)
         for idx, message in enumerate(messages):
@@ -216,3 +229,12 @@ class LogisticRegressionClassifier(IntentClassifier, GraphComponent):
     def validate_config(cls, config: Dict[Text, Any]) -> None:
         """Validates that the component is configured properly."""
         pass
+
+    def is_trained(self) -> bool:
+        """Checks if the model has been trained."""
+        try:
+            # This will raise a NotFittedError if the classifier isn't fitted
+            check_is_fitted(self.clf)
+            return True
+        except NotFittedError:
+            return False

rasa/nlu/featurizers/sparse_featurizer/count_vectors_featurizer.py

@@ -14,6 +14,7 @@ from rasa.nlu.featurizers.sparse_featurizer.sparse_featurizer import SparseFeatu
 from rasa.nlu.utils.spacy_utils import SpacyModel
 from rasa.shared.constants import DOCS_URL_COMPONENTS
 import rasa.utils.io as io_utils
+from sklearn.exceptions import NotFittedError
 from sklearn.feature_extraction.text import CountVectorizer
 from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.message import Message
@@ -551,36 +552,39 @@ class CountVectorsFeaturizer(SparseFeaturizer, GraphComponent):
         sequence_features: List[Optional[scipy.sparse.spmatrix]] = []
         sentence_features: List[Optional[scipy.sparse.spmatrix]] = []
 
-        for i, tokens in enumerate(all_tokens):
-            if not tokens:
-                # nothing to featurize
-                sequence_features.append(None)
-                sentence_features.append(None)
-                continue
-
-            # vectorizer.transform returns a sparse matrix of size
-            # [n_samples, n_features]
-            # set input to list of tokens if sequence should be returned
-            # otherwise join all tokens to a single string and pass that as a list
-            if not tokens:
-                # attribute is not set (e.g. response not present)
-                sequence_features.append(None)
-                sentence_features.append(None)
-                continue
-
-            seq_vec = self.vectorizers[attribute].transform(tokens)
-            seq_vec.sort_indices()
-
-            sequence_features.append(seq_vec.tocoo())
-
-            if attribute in DENSE_FEATURIZABLE_ATTRIBUTES:
-                tokens_text = [" ".join(tokens)]
-                sentence_vec = self.vectorizers[attribute].transform(tokens_text)
-                sentence_vec.sort_indices()
-
-                sentence_features.append(sentence_vec.tocoo())
-            else:
-                sentence_features.append(None)
+        try:
+            for i, tokens in enumerate(all_tokens):
+                # vectorizer.transform returns a sparse matrix of size
+                # [n_samples, n_features]
+                # set input to list of tokens if sequence should be returned
+                # otherwise join all tokens to a single string and pass that as a list
+                if not tokens:
+                    # attribute is not set (e.g. response not present)
+                    sequence_features.append(None)
+                    sentence_features.append(None)
+                    continue
+
+                seq_vec = self.vectorizers[attribute].transform(tokens)
+                seq_vec.sort_indices()
+
+                sequence_features.append(seq_vec.tocoo())
+
+                if attribute in DENSE_FEATURIZABLE_ATTRIBUTES:
+                    tokens_text = [" ".join(tokens)]
+                    sentence_vec = self.vectorizers[attribute].transform(tokens_text)
+                    sentence_vec.sort_indices()
+
+                    sentence_features.append(sentence_vec.tocoo())
+                else:
+                    sentence_features.append(None)
+        except NotFittedError:
+            logger.warning(
+                f"Unable to train CountVectorizer for message "
+                f"attribute - {attribute}, since the call to sklearn's "
+                f"`.fit()` method failed. Leaving an untrained "
+                f"CountVectorizer for it."
+            )
+            return [None], [None]
 
         return sequence_features, sentence_features
 

rasa/nlu/persistor.py

@@ -64,14 +64,14 @@ def parse_remote_storage(value: str) -> StorageType:
 
         if isinstance(value, str):
             if value == "":
-                raise ValueError(
+                raise RasaException(
                     f"The value can't be an empty string."
                     f" {supported_storages_help_text}"
                 )
 
             return value
 
-        raise ValueError(
+        raise RasaException(
             f"Invalid storage type '{value}'. {supported_storages_help_text}"
         )
 
@@ -82,6 +82,8 @@ 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:
         return AWSPersistor(
             os.environ.get(BUCKET_NAME_ENV), os.environ.get(AWS_ENDPOINT_URL_ENV)

rasa/shared/constants.py

@@ -181,6 +181,7 @@ PROMPT_TEMPLATE_CONFIG_KEY = "prompt_template"
 
 STREAM_CONFIG_KEY = "stream"
 N_REPHRASES_CONFIG_KEY = "n"
+USE_CHAT_COMPLETIONS_ENDPOINT_CONFIG_KEY = "use_chat_completions_endpoint"
 
 AZURE_API_KEY_ENV_VAR = "AZURE_API_KEY"
 AZURE_AD_TOKEN_ENV_VAR = "AZURE_AD_TOKEN"

rasa/shared/core/domain.py

@@ -790,13 +790,18 @@ class Domain:
             }
         else:
             intent_name = next(iter(intent.keys()))
-
-        return (
-            intent_name,
-            cls._transform_intent_properties_for_internal_use(
-                intent, entity_properties
-            ),
-        )
+        try:
+            return (
+                intent_name,
+                cls._transform_intent_properties_for_internal_use(
+                    intent, entity_properties
+                ),
+            )
+        except AttributeError:
+            raise InvalidDomain(
+                f"Detected invalid intent definition: {intent}. "
+                f"Please make sure all intent definitions are valid."
+            )
 
     @classmethod
     def _add_default_intents(

rasa/shared/core/training_data/structures.py

@@ -804,6 +804,26 @@ class StoryGraph:
         """Returns text representation of object."""
         return f"{self.__class__.__name__}: {len(self.story_steps)} story steps"
 
+    def has_e2e_stories(self) -> bool:
+        """
+        Checks if there are end-to-end (E2E) stories present in the story steps.
+
+        An E2E story is determined by checking if any `UserUttered` event has
+        associated text within the story steps.
+
+        Returns:
+            bool: True if any E2E story (i.e., a `UserUttered` event with text)
+            is found, False otherwise.
+        """
+        if not self.story_steps:
+            return False
+        for story_step in self.story_steps:
+            for event in story_step.events:
+                if isinstance(event, UserUttered):
+                    if event.text:
+                        return True
+        return False
+
 
 def generate_id(prefix: Text = "", max_chars: Optional[int] = None) -> Text:
     """Generate a random UUID.

rasa/shared/providers/_configs/self_hosted_llm_client_config.py

@@ -21,6 +21,7 @@ from rasa.shared.constants import (
     PROVIDER_CONFIG_KEY,
     OPENAI_PROVIDER,
     SELF_HOSTED_PROVIDER,
+    USE_CHAT_COMPLETIONS_ENDPOINT_CONFIG_KEY,
 )
 from rasa.shared.providers._configs.utils import (
     raise_deprecation_warnings,
@@ -71,6 +72,7 @@ class SelfHostedLLMClientConfig:
     api_base: str
     api_version: Optional[str] = None
     api_type: Optional[str] = OPENAI_PROVIDER
+    use_chat_completions_endpoint: Optional[bool] = True
     extra_parameters: dict = field(default_factory=dict)
 
     def __post_init__(self) -> None:
@@ -140,6 +142,9 @@ class SelfHostedLLMClientConfig:
             # Optional parameters
             api_type=config.pop(API_TYPE_CONFIG_KEY, OPENAI_PROVIDER),
             api_version=config.pop(API_VERSION_CONFIG_KEY, None),
+            use_chat_completions_endpoint=config.pop(
+                USE_CHAT_COMPLETIONS_ENDPOINT_CONFIG_KEY, True
+            ),
             # The rest of parameters (e.g. model parameters) are considered
             # as extra parameters
             extra_parameters=config,

rasa/shared/providers/llm/_base_litellm_client.py

@@ -3,7 +3,11 @@ from typing import Dict, List, Any, Union
 
 import logging
 import structlog
-from litellm import completion, acompletion, validate_environment
+from litellm import (
+    completion,
+    acompletion,
+    validate_environment,
+)
 
 from rasa.shared.exceptions import (
     ProviderClientAPIException,

rasa/shared/providers/llm/self_hosted_llm_client.py

@@ -1,11 +1,19 @@
-from typing import Any, Dict, Optional
+from typing import Any, Dict, List, Optional, Union
+from litellm import (
+    text_completion,
+    atext_completion,
+)
+import logging
 import structlog
 
 from rasa.shared.constants import OPENAI_PROVIDER
 from rasa.shared.providers._configs.self_hosted_llm_client_config import (
     SelfHostedLLMClientConfig,
 )
+from rasa.shared.exceptions import ProviderClientAPIException
 from rasa.shared.providers.llm._base_litellm_client import _BaseLiteLLMClient
+from rasa.shared.providers.llm.llm_response import LLMResponse, LLMUsage
+from rasa.shared.utils.io import suppress_logs
 
 structlogger = structlog.get_logger()
 
@@ -19,6 +27,8 @@ class SelfHostedLLMClient(_BaseLiteLLMClient):
         api_base (str): The base URL of the API endpoint.
         api_type (Optional[str]): The type of the API endpoint.
         api_version (Optional[str]): The version of the API endpoint.
+        use_chat_completions_endpoint (Optional[bool]): Whether to use the chat
+            completions endpoint for completions. Defaults to True.
         kwargs: Any: Additional configuration parameters that can include, but
             are not limited to model parameters and lite-llm specific
             parameters. These parameters will be passed to the
@@ -36,6 +46,7 @@ class SelfHostedLLMClient(_BaseLiteLLMClient):
         api_base: str,
         api_type: Optional[str] = None,
         api_version: Optional[str] = None,
+        use_chat_completions_endpoint: Optional[bool] = True,
         **kwargs: Any,
     ):
         super().__init__()  # type: ignore
@@ -44,6 +55,7 @@ class SelfHostedLLMClient(_BaseLiteLLMClient):
         self._api_base = api_base
         self._api_type = api_type
         self._api_version = api_version
+        self._use_chat_completions_endpoint = use_chat_completions_endpoint
         self._extra_parameters = kwargs or {}
 
     @classmethod
@@ -66,6 +78,7 @@ class SelfHostedLLMClient(_BaseLiteLLMClient):
             api_base=client_config.api_base,
             api_type=client_config.api_type,
             api_version=client_config.api_version,
+            use_chat_completions_endpoint=client_config.use_chat_completions_endpoint,
             **client_config.extra_parameters,
         )
 
@@ -132,6 +145,7 @@ class SelfHostedLLMClient(_BaseLiteLLMClient):
             api_base=self._api_base,
             api_type=self._api_type,
             api_version=self._api_version,
+            use_chat_completions_endpoint=self._use_chat_completions_endpoint,
             extra_parameters=self._extra_parameters,
         )
         return config.to_dict()
@@ -167,3 +181,101 @@ class SelfHostedLLMClient(_BaseLiteLLMClient):
             }
         )
         return fn_args
+
+    @suppress_logs(log_level=logging.WARNING)
+    def _text_completion(self, prompt: Union[List[str], str]) -> LLMResponse:
+        """
+        Synchronously generate completions for given prompt.
+
+        Args:
+            prompt: Prompt to generate the completion for.
+        Returns:
+            List of message completions.
+        Raises:
+            ProviderClientAPIException: If the API request fails.
+        """
+        try:
+            response = text_completion(prompt=prompt, **self._completion_fn_args)
+            return self._format_text_completion_response(response)
+        except Exception as e:
+            raise ProviderClientAPIException(e)
+
+    @suppress_logs(log_level=logging.WARNING)
+    async def _atext_completion(self, prompt: Union[List[str], str]) -> LLMResponse:
+        """
+        Asynchronously generate completions for given prompt.
+
+        Args:
+            prompt: Prompt to generate the completion for.
+        Returns:
+            List of message completions.
+        Raises:
+            ProviderClientAPIException: If the API request fails.
+        """
+        try:
+            response = await atext_completion(prompt=prompt, **self._completion_fn_args)
+            return self._format_text_completion_response(response)
+        except Exception as e:
+            raise ProviderClientAPIException(e)
+
+    async def acompletion(self, messages: Union[List[str], str]) -> LLMResponse:
+        """Asynchronous completion of the model with the given messages.
+
+        Method overrides the base class method to call the appropriate
+        completion method based on the configuration. If the chat completions
+        endpoint is enabled, the acompletion method is called. Otherwise, the
+        atext_completion method is called.
+
+        Args:
+            messages: The messages to be used for completion.
+
+        Returns:
+            The completion response.
+        """
+        if self._use_chat_completions_endpoint:
+            return await super().acompletion(messages)
+        return await self._atext_completion(messages)
+
+    def completion(self, messages: Union[List[str], str]) -> LLMResponse:
+        """Completion of the model with the given messages.
+
+        Method overrides the base class method to call the appropriate
+        completion method based on the configuration. If the chat completions
+        endpoint is enabled, the completion method is called. Otherwise, the
+        text_completion method is called.
+
+        Args:
+            messages: The messages to be used for completion.
+
+        Returns:
+            The completion response.
+        """
+        if self._use_chat_completions_endpoint:
+            return super().completion(messages)
+        return self._text_completion(messages)
+
+    def _format_text_completion_response(self, response: Any) -> LLMResponse:
+        """Parses the LiteLLM text completion response to Rasa format."""
+        formatted_response = LLMResponse(
+            id=response.id,
+            created=response.created,
+            choices=[choice.text for choice in response.choices],
+            model=response.model,
+        )
+        if (usage := response.usage) is not None:
+            prompt_tokens = (
+                num_tokens
+                if isinstance(num_tokens := usage.prompt_tokens, (int, float))
+                else 0
+            )
+            completion_tokens = (
+                num_tokens
+                if isinstance(num_tokens := usage.completion_tokens, (int, float))
+                else 0
+            )
+            formatted_response.usage = LLMUsage(prompt_tokens, completion_tokens)
+        structlogger.debug(
+            "base_litellm_client.formatted_response",
+            formatted_response=formatted_response.to_dict(),
+        )
+        return formatted_response

rasa/shared/utils/yaml.py

@@ -416,6 +416,47 @@ def validate_raw_yaml_using_schema_file_with_responses(
     )
 
 
+def process_content(content: str) -> str:
+    """
+    Process the content to handle both Windows paths and emojis.
+    Windows paths are processed by escaping backslashes but emojis are left untouched.
+
+    Args:
+        content: yaml content to be processed
+    """
+    # Detect common Windows path patterns: e.g., C:\ or \\
+    UNESCAPED_WINDOWS_PATH_PATTERN = re.compile(
+        r"(?<!\w)[a-zA-Z]:(\\[a-zA-Z0-9_ -]+)*(\\)?(?!\\n)"
+    )
+    ESCAPED_WINDOWS_PATH_PATTERN = re.compile(
+        r"(?<!\w)[a-zA-Z]:(\\\\[a-zA-Z0-9_ -]+)+\\\\?(?!\\n)"
+    )
+
+    # Function to escape backslashes in Windows paths but leave other content as is
+    def escape_windows_paths(match: re.Match) -> str:
+        path = str(match.group(0))
+        return path.replace("\\", "\\\\")  # Escape backslashes only in Windows paths
+
+    def unescape_windows_paths(match: re.Match) -> str:
+        path = str(match.group(0))
+        return path.replace("\\\\", "\\")
+
+    # First, process Windows paths by escaping backslashes
+    content = re.sub(UNESCAPED_WINDOWS_PATH_PATTERN, escape_windows_paths, content)
+
+    # Ensure proper handling of emojis by decoding Unicode sequences
+    content = (
+        content.encode("utf-8")
+        .decode("raw_unicode_escape")
+        .encode("utf-16", "surrogatepass")
+        .decode("utf-16")
+    )
+
+    content = re.sub(ESCAPED_WINDOWS_PATH_PATTERN, unescape_windows_paths, content)
+
+    return content
+
+
 def read_yaml(
     content: str,
     reader_type: Union[str, List[str]] = "safe",
@@ -432,13 +473,7 @@ def read_yaml(
         ruamel.yaml.parser.ParserError: If there was an error when parsing the YAML.
     """
     if _is_ascii(content):
-        # Required to make sure emojis are correctly parsed
-        content = (
-            content.encode("utf-8")
-            .decode("raw_unicode_escape")
-            .encode("utf-16", "surrogatepass")
-            .decode("utf-16")
-        )
+        content = process_content(content)
 
     custom_constructor = kwargs.get("custom_constructor", None)
 

rasa/tracing/config.py

@@ -20,6 +20,7 @@ from rasa.core.actions.custom_action_executor import (
     CustomActionExecutor,
     RetryCustomActionExecutor,
 )
+from rasa.core.actions.grpc_custom_action_executor import GRPCCustomActionExecutor
 from rasa.core.agent import Agent
 from rasa.core.processor import MessageProcessor
 from rasa.core.tracker_store import TrackerStore
@@ -44,11 +45,10 @@ from rasa.tracing.constants import (
     ENDPOINTS_ROOT_CERTIFICATES_KEY,
     ENDPOINTS_METRICS_KEY,
 )
+from rasa.tracing.instrumentation import instrumentation
 from rasa.tracing.metric_instrument_provider import MetricInstrumentProvider
 from rasa.utils.endpoints import EndpointConfig, read_endpoint_config
 
-from rasa.tracing.instrumentation import instrumentation
-
 TRACING_SERVICE_NAME = os.environ.get("TRACING_SERVICE_NAME", "rasa")
 
 logger = logging.getLogger(__name__)
@@ -107,6 +107,7 @@ def configure_tracing(tracer_provider: Optional[TracerProvider]) -> None:
         vector_store_subclasses=vector_store_subclasses,
         nlu_command_adapter_class=NLUCommandAdapter,
         endpoint_config_class=EndpointConfig,
+        grpc_custom_action_executor_class=GRPCCustomActionExecutor,
         single_step_llm_command_generator_class=SingleStepLLMCommandGenerator,
         multi_step_llm_command_generator_class=MultiStepLLMCommandGenerator,
         custom_action_executor_subclasses=custom_action_executor_subclasses,

rasa/tracing/instrumentation/attribute_extractors.py

@@ -42,6 +42,7 @@ from rasa.tracing.constants import (
     PROMPT_TOKEN_LENGTH_ATTRIBUTE_NAME,
     REQUEST_BODY_SIZE_IN_BYTES_ATTRIBUTE_NAME,
 )
+from rasa.shared.core.training_data.structures import StoryGraph
 
 if TYPE_CHECKING:
     from langchain.llms.base import BaseLLM
@@ -374,6 +375,7 @@ def extract_attrs_for_execute_commands(
     tracker: DialogueStateTracker,
     all_flows: FlowsList,
     execution_context: ExecutionContext,
+    story_graph: Optional[StoryGraph] = None,
 ) -> Dict[str, Any]:
     return {
         "number_of_events": len(tracker.events),
@@ -416,6 +418,7 @@ def extract_attrs_for_clean_up_commands(
     tracker: DialogueStateTracker,
     all_flows: FlowsList,
     execution_context: ExecutionContext,
+    story_graph: Optional[StoryGraph] = None,
 ) -> Dict[str, Any]:
     commands_list = []
 

rasa/tracing/instrumentation/instrumentation.py

@@ -20,10 +20,10 @@ from typing import (
 
 from multidict import MultiDict
 from opentelemetry.context import Context
-
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.trace import SpanKind, Tracer
 from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
+
 from rasa.core.actions.action import Action, RemoteAction, CustomActionExecutor
 from rasa.core.actions.custom_action_executor import RetryCustomActionExecutor
 from rasa.core.actions.grpc_custom_action_executor import GRPCCustomActionExecutor
@@ -48,11 +48,13 @@ from rasa.dialogue_understanding.generator import (
 from rasa.dialogue_understanding.generator.nlu_command_adapter import NLUCommandAdapter
 from rasa.engine.graph import GraphNode
 from rasa.engine.training.graph_trainer import GraphTrainer
+from rasa.shared.core.domain import Domain
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.nlu.constants import SET_SLOT_COMMAND
 from rasa.shared.nlu.training_data.message import Message
 from rasa.tracing.constants import REQUEST_BODY_SIZE_IN_BYTES_ATTRIBUTE_NAME
+from rasa.tracing.instrumentation import attribute_extractors
 from rasa.tracing.instrumentation.intentless_policy_instrumentation import (
     _instrument_extract_ai_responses,
     _instrument_generate_answer,
@@ -68,9 +70,6 @@ from rasa.tracing.instrumentation.metrics import (
 )
 from rasa.utils.endpoints import concat_url, EndpointConfig
 
-from rasa.tracing.instrumentation import attribute_extractors
-
-
 # The `TypeVar` representing the return type for a function to be wrapped.
 S = TypeVar("S")
 # The `TypeVar` representing the type of the argument passed to the function to be
@@ -308,6 +307,7 @@ def instrument(
     vector_store_subclasses: Optional[List[Type[InformationRetrievalType]]] = None,
     nlu_command_adapter_class: Optional[Type[NLUCommandAdapterType]] = None,
     endpoint_config_class: Optional[Type[EndpointConfigType]] = None,
+    grpc_custom_action_executor_class: Optional[Type[GRPCCustomActionExecutor]] = None,
     single_step_llm_command_generator_class: Optional[
         Type[SingleStepLLMCommandGeneratorType]
     ] = None,
@@ -354,6 +354,9 @@ def instrument(
         `None` is given, no `NLUCommandAdapter` will be instrumented.
     :param endpoint_config_class: The `EndpointConfig` to be instrumented. If
         `None` is given, no `EndpointConfig` will be instrumented.
+    :param grpc_custom_action_executor_class: The `GRPCCustomActionExecution` to be
+        instrumented. If `None` is given, no `GRPCCustomActionExecution`
+        will be instrumented.
     :param single_step_llm_command_generator_class: The `SingleStepLLMCommandGenerator`
         to be instrumented. If `None` is given, no `SingleStepLLMCommandGenerator` will
         be instrumented.
@@ -581,6 +584,14 @@ def instrument(
             endpoint_config_class,
         )
 
+    if grpc_custom_action_executor_class is not None and not class_is_instrumented(
+        grpc_custom_action_executor_class
+    ):
+        _instrument_grpc_custom_action_executor(
+            tracer_provider.get_tracer(grpc_custom_action_executor_class.__module__),
+            grpc_custom_action_executor_class,
+        )
+
     if custom_action_executor_subclasses:
         for custom_action_executor_subclass in custom_action_executor_subclasses:
             if (
@@ -1096,6 +1107,38 @@ def _instrument_endpoint_config(
     logger.debug(f"Instrumented '{endpoint_config_class.__name__}.request'.")
 
 
+def _instrument_grpc_custom_action_executor(
+    tracer: Tracer, grpc_custom_action_executor_class: Type[GRPCCustomActionExecutor]
+) -> None:
+    """Instrument the `run` method of the `GRPCCustomActionExecutor` class.
+
+    Args:
+        tracer: The `Tracer` that shall be used for tracing.
+        grpc_custom_action_executor_class: The `GRPCCustomActionExecutor` to
+            be instrumented.
+    """
+
+    def tracing_grpc_custom_action_executor_wrapper(fn: Callable) -> Callable:
+        @functools.wraps(fn)
+        async def wrapper(
+            self: Type[GRPCCustomActionExecutor],
+            tracker: Type[DialogueStateTracker],
+            domain: Type[Domain],
+            include_domain: bool = False,
+        ) -> bool:
+            TraceContextTextMapPropagator().inject(self.action_endpoint.headers)
+            result = await fn(self, tracker, domain, include_domain)
+            return result
+
+        return wrapper
+
+    grpc_custom_action_executor_class.run = tracing_grpc_custom_action_executor_wrapper(  # type: ignore[assignment]
+        grpc_custom_action_executor_class.run
+    )
+
+    logger.debug(f"Instrumented '{grpc_custom_action_executor_class.__name__}.run.")
+
+
 def _mangled_instrumented_boolean_attribute_name(instrumented_class: Type) -> Text:
     # see https://peps.python.org/pep-0008/#method-names-and-instance-variables
     # and https://stackoverflow.com/a/50401073

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.4"
+__version__ = "3.10.6"

{rasa_pro-3.10.4.dist-info → rasa_pro-3.10.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rasa-pro
-Version: 3.10.4
+Version: 3.10.6
 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
@@ -33,7 +33,7 @@ Requires-Dist: apscheduler (>=3.10,<3.11)
 Requires-Dist: attrs (>=23.1,<23.2)
 Requires-Dist: azure-storage-blob (>=12.16.0,<12.17.0)
 Requires-Dist: boto3 (>=1.35.5,<1.36.0)
-Requires-Dist: certifi (>=2024.2.2)
+Requires-Dist: certifi (>=2024.07.04)
 Requires-Dist: cloudpickle (>=2.2.1,<3.1)
 Requires-Dist: colorama (>=0.4.6,<0.5.0) ; sys_platform == "win32"
 Requires-Dist: colorclass (>=2.2,<2.3)
@@ -49,6 +49,7 @@ Requires-Dist: faiss-cpu (>=1.7.4,<2.0.0)
 Requires-Dist: faker (>=26.0.0,<27.0.0)
 Requires-Dist: fbmessenger (>=6.0.0,<6.1.0)
 Requires-Dist: github3.py (>=3.2.0,<3.3.0) ; extra == "gh-release-notes"
+Requires-Dist: gitpython (>=3.1.41,<3.2.0) ; extra == "full"
 Requires-Dist: google-auth (>=2.23.4,<3)
 Requires-Dist: google-cloud-storage (>=2.14.0,<3.0.0)
 Requires-Dist: hvac (>=1.2.1,<2.0.0)
@@ -63,13 +64,13 @@ Requires-Dist: jsonschema (>=4.22)
 Requires-Dist: keras (==2.14.0)
 Requires-Dist: langchain (>=0.2.0,<0.3.0)
 Requires-Dist: langchain-community (>=0.2.0,<0.3.0)
-Requires-Dist: litellm (>=1.44.7,<1.45.0)
+Requires-Dist: litellm (>=1.45.0,<1.46.0)
 Requires-Dist: matplotlib (>=3.7,<3.8)
 Requires-Dist: mattermostwrapper (>=2.2,<2.3)
 Requires-Dist: mlflow (>=2.15.1,<3.0.0) ; extra == "mlflow"
 Requires-Dist: networkx (>=3.1,<3.2)
 Requires-Dist: numpy (>=1.23.5,<1.25.0) ; python_version >= "3.9" and python_version < "3.11"
-Requires-Dist: openai (>=1.40.0,<1.41.0)
+Requires-Dist: openai (>=1.47.0,<1.48.0)
 Requires-Dist: openpyxl (>=3.1.5,<4.0.0)
 Requires-Dist: opentelemetry-api (>=1.16.0,<1.17.0)
 Requires-Dist: opentelemetry-exporter-jaeger (>=1.16.0,<1.17.0)

{rasa_pro-3.10.4.dist-info → rasa_pro-3.10.6.dist-info}/RECORD

@@ -302,7 +302,7 @@ rasa/core/policies/flows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 rasa/core/policies/flows/flow_exceptions.py,sha256=_FQuN-cerQDM1pivce9bz4zylh5UYkljvYS1gjDukHI,1527
 rasa/core/policies/flows/flow_executor.py,sha256=okOxDzBekd9sMOLuxxVs6KLQj_xVqKgxO0sxKAA7gfg,25482
 rasa/core/policies/flows/flow_step_result.py,sha256=agjPrD6lahGSe2ViO5peBeoMdI9ngVGRSgtytgxmJmg,1360
-rasa/core/policies/intentless_policy.py,sha256=5NtOu4pepTfjTTQptqGcvLKAcmVNpj0ymQovlbSmYf4,34417
+rasa/core/policies/intentless_policy.py,sha256=M2pUHtveJ7Oh6cDRX2NiFCnfRUP6HkW4PmOv_T-9Jv0,35121
 rasa/core/policies/intentless_prompt_template.jinja2,sha256=KhIL3cruMmkxhrs5oVbqgSvK6ZiN_6TQ_jXrgtEB-ZY,677
 rasa/core/policies/memoization.py,sha256=XoRxUdYUGRfO47tAEyc5k5pUgt38a4fipO336EU5Vdc,19466
 rasa/core/policies/policy.py,sha256=HeVtIaV0dA1QcAG3vjdn-4g7-oUEJPL4u01ETJt78YA,27464
@@ -385,8 +385,8 @@ rasa/dialogue_understanding/patterns/search.py,sha256=X7HaMyFwS8gPprKFSOJvCoC6Ng
 rasa/dialogue_understanding/patterns/session_start.py,sha256=yglhIEkkquRf0YppZ4Cuv2eHlA5qscGoVXr0d-2bV-E,1153
 rasa/dialogue_understanding/patterns/skip_question.py,sha256=rvZuVUxulikwUhP01MAIgkcHZ4Si7mzxNedH6QBPdX4,1214
 rasa/dialogue_understanding/processor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rasa/dialogue_understanding/processor/command_processor.py,sha256=tzwSiQfVwJqTf5VINpU6pPUAbef1nd9Um_FnRzzk1Mg,24473
-rasa/dialogue_understanding/processor/command_processor_component.py,sha256=KVDW0H2K-CGXljVww1vn61g-TqCWGzbZ-wwvQ9THMV0,1398
+rasa/dialogue_understanding/processor/command_processor.py,sha256=_kLtp3jV3MDKvakyyVp5O7ypp8btesPTfCxlTQ_QD-M,25058
+rasa/dialogue_understanding/processor/command_processor_component.py,sha256=oyz6lq64CRmnugTrIJJ_Q7RLGYD_xbZJPVJ47xNbyvw,1526
 rasa/dialogue_understanding/stack/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/dialogue_understanding/stack/dialogue_stack.py,sha256=j8MnLCyv6cAZVpKRaUVM-Z5HqgWP-scrnaiQXzLNBwY,5243
 rasa/dialogue_understanding/stack/frames/__init__.py,sha256=CXLs8I_eeJ-d2tQmS19V26OM6CHy3VN5whH5uHBodj4,656
@@ -425,7 +425,7 @@ rasa/engine/loader.py,sha256=ommFdjddJWkb-KHpemxRhWYlMYRAUs8rcmtJx_DzqnM,1384
 rasa/engine/recipes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/engine/recipes/config_files/default_config.yml,sha256=M1ehmJScNbVCQD_wkBktmf0YqXn-O4UzJVbB7tnAtqs,1219
 rasa/engine/recipes/default_components.py,sha256=XRg51DZ5CKmDjghXCUq-rTv8rkSOYUTkLNulLHnwWHc,3950
-rasa/engine/recipes/default_recipe.py,sha256=LSdhag8xjMtEIk1-lIuWgATKLGv8XsuVi9INO7xoulU,45630
+rasa/engine/recipes/default_recipe.py,sha256=mxNyU8vxcsspGEBv_ne_Ibw_Wcj1m09eYGOcPvYWrdQ,45918
 rasa/engine/recipes/graph_recipe.py,sha256=Kd2vKIRCENzWe9W24HwNedk_b6lIXL4aYqVIZsCJ4ts,3301
 rasa/engine/recipes/recipe.py,sha256=v-t5jwntbXcLWVzfKhDuluz__ouEgl_Q5VFTZieqYgs,3346
 rasa/engine/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -454,7 +454,7 @@ rasa/graph_components/providers/forms_provider.py,sha256=BU6BDzIFjt2SZOmZ8xY8m_s
 rasa/graph_components/providers/nlu_training_data_provider.py,sha256=ZoCRcVzAuoqIxfynWkwRf5AG3q7XUk6HMthd8LR1Uh0,2119
 rasa/graph_components/providers/responses_provider.py,sha256=yTtVABjLN1uTtVjAeqPjiEnm0egnA21CwWCjMWekqJY,1354
 rasa/graph_components/providers/rule_only_provider.py,sha256=mTzf5_PulQB7Y_voLnljm2Ob7shYPAAx_2cKN-MRovg,1540
-rasa/graph_components/providers/story_graph_provider.py,sha256=7kLBteTxWGJRFjkdwgpsEemsKSCkffsDKk3pdX0L6Vg,1466
+rasa/graph_components/providers/story_graph_provider.py,sha256=tvTCq5dtAHa3VFOYQqcd15gA0cFJWqgPF3K0vQXCm_4,3300
 rasa/graph_components/providers/training_tracker_provider.py,sha256=nCHyLsiC8q3B0CIgVCbhUBCAil24ja43UKasoVFJ3DM,1965
 rasa/graph_components/validators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/graph_components/validators/default_recipe_validator.py,sha256=BHrF6NTfJz42xG1LfVjkP5CdQef4NTcmiiC8xtMemaI,24457
@@ -489,7 +489,7 @@ rasa/nlu/classifiers/classifier.py,sha256=9fm1mORuFf1vowYIXmqE9yLRKdSC4nGQW7UqNZ
 rasa/nlu/classifiers/diet_classifier.py,sha256=C2mKZ2GP7Uptpag240fFkAEZf6P1NuU_2TrnSsR3IA0,71936
 rasa/nlu/classifiers/fallback_classifier.py,sha256=FYOgM7bLG3HlasVWRozanz-MmDozygTlTIFcPHJWJoo,7150
 rasa/nlu/classifiers/keyword_intent_classifier.py,sha256=dxDzCK7YzYKslZiXYkBD1Al1y_yZWdZYkBBl7FLyPm8,7581
-rasa/nlu/classifiers/logistic_regression_classifier.py,sha256=tK8esDssBf74IyGOqHSefIQd0-vGu9Yn6MvlrbJ4cX8,8317
+rasa/nlu/classifiers/logistic_regression_classifier.py,sha256=Qga66-PjW4I4D2uIMoX2aW8ywdufq9ISmt12rP3rj9g,9124
 rasa/nlu/classifiers/mitie_intent_classifier.py,sha256=_hf0aKWjcjZ8NdH61gbutgY5vAjMmpYDhCpO3dwIrDk,5559
 rasa/nlu/classifiers/regex_message_handler.py,sha256=r6Z-uFJvqFZjpI1rUeaZZnAOUL9lxuBxGK7W6WZIPOw,1989
 rasa/nlu/classifiers/sklearn_intent_classifier.py,sha256=zPLr1GNCEAG8xW5SEPLgc2lsenXavTG9KDby8JUDX3o,11923
@@ -518,12 +518,12 @@ rasa/nlu/featurizers/dense_featurizer/mitie_featurizer.py,sha256=xE-dOmdBqCJ4NEm
 rasa/nlu/featurizers/dense_featurizer/spacy_featurizer.py,sha256=tJzDeX8wkOO1iUNmx13FSIeMHNC0U0RB5ZF9pPo8nqQ,4888
 rasa/nlu/featurizers/featurizer.py,sha256=cV2v4f1V2DWDqJY1-oGAZsytv0L827nsCtUY6KjEChg,3348
 rasa/nlu/featurizers/sparse_featurizer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rasa/nlu/featurizers/sparse_featurizer/count_vectors_featurizer.py,sha256=twFecYgWF4uyfV_VxHIDBCykllypSSndfszpAnniaY8,33295
+rasa/nlu/featurizers/sparse_featurizer/count_vectors_featurizer.py,sha256=275NcC7W9_n7V0AyVXm8jtYd9fcVHXZRQMgr5MVZAvA,33600
 rasa/nlu/featurizers/sparse_featurizer/lexical_syntactic_featurizer.py,sha256=awydhZZTRmff35L1838bbghNbutEf5xty301OyRIgvI,21848
 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=SXTmuNBM9o2uReePRdohVkQQusnZ9YDBJHw3okkJc5U,13698
+rasa/nlu/persistor.py,sha256=Sc0NH2VSK9efOYSD0INYd3za3esQvgNHa4FwClJVH-c,13788
 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
@@ -545,12 +545,12 @@ 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=MDCiIeygQL5ZwPcRVgynW1WFWR4uQLbPSp422tuwCc4,9093
+rasa/shared/constants.py,sha256=FL_lmM6w87X3tFUqDkXwIisoOODkomLorLPyG-TZpqQ,9168
 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
 rasa/shared/core/conversation.py,sha256=tw1fD2XB3gOdQjDI8hHo5TAAmE2JYNogQGWe3rE929w,1385
-rasa/shared/core/domain.py,sha256=L5QpwKckMUifGttbp-MPJiKUNyg9gCJJPspTRVRC5gg,79804
+rasa/shared/core/domain.py,sha256=zHnwLneOmRRQQp7ZGAt0RSzH_jyhzRgEPRXCa_b70wo,80054
 rasa/shared/core/events.py,sha256=HXBy-DfulOhcHxrfVtP-uUBVnCx40wkRwqDx80mOFX4,82791
 rasa/shared/core/flows/__init__.py,sha256=HszhIvEARpmyxABFc1MKYvj8oy04WiZW1xmCdToakbs,181
 rasa/shared/core/flows/flow.py,sha256=cxf3AVbweOezV8SuCQSog2eYI6nKw7R0SQt8XJiG7zU,21208
@@ -588,7 +588,7 @@ rasa/shared/core/training_data/story_reader/yaml_story_reader.py,sha256=Yn4dpLCn
 rasa/shared/core/training_data/story_writer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/shared/core/training_data/story_writer/story_writer.py,sha256=cR4VVhhS2UzFYhhqL9iojR88Fknc63psCJ4YN7RPeKo,2544
 rasa/shared/core/training_data/story_writer/yaml_story_writer.py,sha256=0proT_BgtzCK3LO3nRA8blVabz3LElORQ-izF_7DdjE,15792
-rasa/shared/core/training_data/structures.py,sha256=GTGOdYo3M5u8yS5pPTN2mL0gWyCwtPU14FGdk7-IZFg,29341
+rasa/shared/core/training_data/structures.py,sha256=diNYa5aCFz0TA9jM1H4-Q3I1vdsXtExk98FCuC0pDYg,30039
 rasa/shared/core/training_data/visualization.html,sha256=viHR5xPh9FxWGXH9b_4ZXzj-ZoG_S3k_fDKe2ZACVAg,3499
 rasa/shared/core/training_data/visualization.py,sha256=th3lzvbmJJh8OHdI7Bs-bHoisQRuhj8fy7xKVmBzgsI,20413
 rasa/shared/data.py,sha256=gIcyPU_ZG4nk3owSml_WkivxIASXlTzV5x3YhfnyZn8,6873
@@ -631,7 +631,7 @@ rasa/shared/providers/_configs/client_config.py,sha256=Tvnv9CCpzoHB053lvuqth_Jwj
 rasa/shared/providers/_configs/default_litellm_client_config.py,sha256=ywD0EYWCKqNpx0fPyOavIwwS2BTd3q9QtNeAyIvOd8E,4318
 rasa/shared/providers/_configs/huggingface_local_embedding_client_config.py,sha256=laahM8xQJ-r0nAL8X2SnBM5kZAOiPNo-w0hv7BudAdk,8166
 rasa/shared/providers/_configs/openai_client_config.py,sha256=bkSQyL0o3MYVSrKWe76QPo7tXtsZPhg3tfA7QQ2EmLo,5805
-rasa/shared/providers/_configs/self_hosted_llm_client_config.py,sha256=HKSYPuD5uGKTjKsOZwPhECplcX53ijIAl_-aZmAmohs,5694
+rasa/shared/providers/_configs/self_hosted_llm_client_config.py,sha256=fUEvqCrhEIX-y2WLP2UlG8zMjNTWBZb1LUIEAUy6bYQ,5929
 rasa/shared/providers/_configs/utils.py,sha256=vxtWmrnpnH2ryHWiP7hpLbSBAq3x-3hsGxGWSEgp4Lk,3334
 rasa/shared/providers/_ssl_verification_utils.py,sha256=4tujCOjg0KKX2_DzOb7lZTdsUXtzRB4UkfhkC3W0jO0,4166
 rasa/shared/providers/embedding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -644,13 +644,13 @@ rasa/shared/providers/embedding/embedding_response.py,sha256=H55mSAL3LfVvDlBklaC
 rasa/shared/providers/embedding/huggingface_local_embedding_client.py,sha256=Zo3gyj49h4LxXV7bx39TXpIPKlernG-5xzqXczTCbig,6913
 rasa/shared/providers/embedding/openai_embedding_client.py,sha256=XNRGE7apo2v3kWRrtgxE-Gq4rvNko3IiXtvgC4krDYE,5429
 rasa/shared/providers/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rasa/shared/providers/llm/_base_litellm_client.py,sha256=qf_jsQqTmANKwYY316t1TrgNnFxOrMOGoV8-D4xkMD8,8375
+rasa/shared/providers/llm/_base_litellm_client.py,sha256=q24u62HNVPKdEvr5fHy_5AWzi3EwexuuKOMqn5nP_U0,8392
 rasa/shared/providers/llm/azure_openai_llm_client.py,sha256=jwEntKsBHKOXUhNEQM-kTYIfRCJ6qEka58ZPnKsSsb8,11836
 rasa/shared/providers/llm/default_litellm_llm_client.py,sha256=yvqd4ARoGSi9iqfE2uFvVEYRU6rICePBnEEKTduCc9w,2777
 rasa/shared/providers/llm/llm_client.py,sha256=6-gMsEJqquhUPGXzNiq_ybM_McLWxAJ_QhbmWcLnb_Q,2358
 rasa/shared/providers/llm/llm_response.py,sha256=Ltmc8yk9cAqtK8QgwfZZywudM5ZQsT4y_AKAQ3q05hA,1490
 rasa/shared/providers/llm/openai_llm_client.py,sha256=uDdcugBcO3sfxbduc00eqaZdrJP0VFX5dkBd2Dem47M,4844
-rasa/shared/providers/llm/self_hosted_llm_client.py,sha256=oSegJImwrdT7tlYf-ehF8ceCwmhsglxgwzkEneLgOV8,5229
+rasa/shared/providers/llm/self_hosted_llm_client.py,sha256=BeFndaz5gknobEFvNHM4GLFs_Kx2LnaXaPG_GGTqEpk,9765
 rasa/shared/providers/mappings.py,sha256=xg-KpfiqotGScmLywg7Ywu10orAPhOTQ4FZPv_JqhEo,2988
 rasa/shared/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/shared/utils/cli.py,sha256=bJpkf0VzzmtpmBnDnIl7SgvrntnBuaJQMHBXHm2WxcA,2916
@@ -665,7 +665,7 @@ rasa/shared/utils/schemas/domain.yml,sha256=b2k4ZYSV-QL3hGjDaRg8rfoqaTh4hbhDc_hB
 rasa/shared/utils/schemas/events.py,sha256=nm7WotE1UCZpkMMLNW2vs-vKslwngfG7VvjFp0T8ADM,6819
 rasa/shared/utils/schemas/model_config.yml,sha256=GU1lL_apXjJ3Xbd9Fj5jKm2h1HeB6V6TNqrhK5hOrGY,998
 rasa/shared/utils/schemas/stories.yml,sha256=DV3wAFnv1leD7kV-FH-GQihF1QX5oKHc8Eb24mxjizc,4737
-rasa/shared/utils/yaml.py,sha256=r5s7jjsGWU7a5X0WuOvB3jKx37xySvOHe_ydL5vqroY,31362
+rasa/shared/utils/yaml.py,sha256=XJoKLe_vOmCcTI_Vg5o-iJzu2uumen7j1rrO6T95QXM,32592
 rasa/studio/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rasa/studio/auth.py,sha256=yPbDHJQAttWyDcjurJywgFlcSfBs6j0FfA4mWO7vKEA,9535
 rasa/studio/config.py,sha256=wqjvg_hARKG-6cV3JrhnP_ptZIq0VSFbdv-aWrH0u_Q,4091
@@ -677,11 +677,11 @@ rasa/studio/train.py,sha256=gfPtirITzBDo9gV4hqDNSwPYtVp_22cq8OWI6YIBgyk,4243
 rasa/studio/upload.py,sha256=3XqYyIZE1L3ohJtRcpf039IRc73A_rZ4KVWC32_dBo4,14027
 rasa/telemetry.py,sha256=Q6MQnDhOY6cKRVs3PayiM6WYWb8QJ_Hb3_eOm12n0tI,61093
 rasa/tracing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rasa/tracing/config.py,sha256=2UvJFmeH2fP-Te3hIP4WHNN9piJJRMBltpmfsBqbBCM,12586
+rasa/tracing/config.py,sha256=O4iHkE4kF6r4uVAt3JUb--TniK7pWD4j3d08Vf_GuYY,12736
 rasa/tracing/constants.py,sha256=N_MJLStE3IkmPKQCQv42epd3jdBMJ4Ith1dVO65N5ho,2425
 rasa/tracing/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rasa/tracing/instrumentation/attribute_extractors.py,sha256=mKeI-aliz3U9MfwveUqyQWdDEzXZynlDcGqkEJX7VXw,23079
-rasa/tracing/instrumentation/instrumentation.py,sha256=wYSY9MftTX06fmMwphm8iL-ULIlVr1gpNb-zFmcJ_zg,43987
+rasa/tracing/instrumentation/attribute_extractors.py,sha256=_AbL_8zhvK8MKh0TVP0Kv9SR66c0acYGmPVLa18-Es0,23236
+rasa/tracing/instrumentation/instrumentation.py,sha256=buoETr3rUHGirkaSY95SiTRqMZI1FCXjphcGXEH_e1A,45821
 rasa/tracing/instrumentation/intentless_policy_instrumentation.py,sha256=8AdMOy_2mlKnlmt-muV8-eoT8jA52GXDzM0avejfg8A,4821
 rasa/tracing/instrumentation/metrics.py,sha256=ByfKshoxNOqjKZwKTulqL71s5b3WugqLfjha3So0OEU,10534
 rasa/tracing/metric_instrument_provider.py,sha256=VtqT1XGQiDODJxbJDXA89AZP5dZMpVRAJuKkblsI2Kg,9989
@@ -719,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=HM0ZIWjo3JRt2FMIfgNI_s932OABOSXkflm-rFTNkvY,62608
-rasa/version.py,sha256=td7_iTF0LQr0S8id0WK1wCSUkhVbSJTmfFOtzTCXj-8,117
-rasa_pro-3.10.4.dist-info/METADATA,sha256=66kFNQG7Fv8mRsJgIl1HMpLAiOlEyQZ0r3Td-rx0enA,28148
-rasa_pro-3.10.4.dist-info/NOTICE,sha256=7HlBoMHJY9CL2GlYSfTQ-PZsVmLmVkYmMiPlTjhuCqA,218
-rasa_pro-3.10.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-rasa_pro-3.10.4.dist-info/entry_points.txt,sha256=ckJ2SfEyTPgBqj_I6vm_tqY9dZF_LAPJZA335Xp0Q9U,43
-rasa_pro-3.10.4.dist-info/RECORD,,
+rasa/version.py,sha256=GYr3HissJmOn-bRoxLYW3Vm7fq_m-bG_jo6kK71I-Oo,117
+rasa_pro-3.10.6.dist-info/METADATA,sha256=xvdyCk1zQU8Aoa-H7UdsTOX2ZpJcbVz9aLCT9xz4icg,28211
+rasa_pro-3.10.6.dist-info/NOTICE,sha256=7HlBoMHJY9CL2GlYSfTQ-PZsVmLmVkYmMiPlTjhuCqA,218
+rasa_pro-3.10.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+rasa_pro-3.10.6.dist-info/entry_points.txt,sha256=ckJ2SfEyTPgBqj_I6vm_tqY9dZF_LAPJZA335Xp0Q9U,43
+rasa_pro-3.10.6.dist-info/RECORD,,