rasa-pro 3.13.0.dev5__py3-none-any.whl → 3.13.0.dev7__py3-none-any.whl

rasa/shared/core/events.py

@@ -107,6 +107,8 @@ if TYPE_CHECKING:
 logger = logging.getLogger(__name__)
 structlogger = structlog.get_logger()
 
+INVALID_DATETIME_ERROR_MESSAGE = "`anonymized_at` must be a datetime object."
+
 
 def deserialise_events(serialized_events: List[Dict[Text, Any]]) -> List["Event"]:
     """Convert a list of dictionaries to a list of corresponding events.
@@ -498,6 +500,23 @@ class UserUttered(Event):
         if parse_data:
             self.parse_data.update(**parse_data)
 
+        self._anonymized_at: Optional[datetime] = None
+
+    @property
+    def anonymized_at(self) -> Optional[datetime]:
+        """Returns the time when the event was anonymized in the tracker store.
+
+        If the event was not anonymized, it returns None.
+        """
+        return self._anonymized_at
+
+    @anonymized_at.setter
+    def anonymized_at(self, value: Optional[datetime]) -> None:
+        """Sets the time when the event was anonymized in the tracker store."""
+        if value is not None and not isinstance(value, datetime):
+            raise ValueError(INVALID_DATETIME_ERROR_MESSAGE)
+        self._anonymized_at = value
+
     @staticmethod
     def _from_parse_data(
         text: Text,
@@ -614,6 +633,7 @@ class UserUttered(Event):
                 "input_channel": getattr(self, "input_channel", None),
                 "message_id": getattr(self, "message_id", None),
                 "metadata": self.metadata,
+                "anonymized_at": self.anonymized_at,
             }
         )
         return _dict
@@ -911,8 +931,24 @@ class BotUttered(SkipEventInMDStoryMixin):
         """
         self.text = text
         self.data = data or {}
+        self._anonymized_at: Optional[datetime] = None
         super().__init__(timestamp, metadata)
 
+    @property
+    def anonymized_at(self) -> Optional[datetime]:
+        """Returns the time when the event was anonymized in the tracker store.
+
+        If the event was not anonymized, it returns None.
+        """
+        return self._anonymized_at
+
+    @anonymized_at.setter
+    def anonymized_at(self, value: Optional[datetime]) -> None:
+        """Sets the time when the event was anonymized in the tracker store."""
+        if value is not None and not isinstance(value, datetime):
+            raise ValueError(INVALID_DATETIME_ERROR_MESSAGE)
+        self._anonymized_at = value
+
     def __members(self) -> Tuple[Optional[Text], Text, Text]:
         data_no_nones = {k: v for k, v in self.data.items() if v is not None}
         meta_no_nones = {k: v for k, v in self.metadata.items() if v is not None}
@@ -999,7 +1035,14 @@ class BotUttered(SkipEventInMDStoryMixin):
     def as_dict(self) -> Dict[Text, Any]:
         """Returns serialized event."""
         d = super().as_dict()
-        d.update({"text": self.text, "data": self.data, "metadata": self.metadata})
+        d.update(
+            {
+                "text": self.text,
+                "data": self.data,
+                "metadata": self.metadata,
+                "anonymized_at": self.anonymized_at,
+            }
+        )
         return d
 
     @classmethod
@@ -1046,8 +1089,24 @@ class SlotSet(Event):
         self.key = key
         self.value = value
         self._filled_by = filled_by
+        self._anonymized_at: Optional[datetime] = None
         super().__init__(timestamp, metadata)
 
+    @property
+    def anonymized_at(self) -> Optional[datetime]:
+        """Returns the time when the event was anonymized in the tracker store.
+
+        If the event was not anonymized, it returns None.
+        """
+        return self._anonymized_at
+
+    @anonymized_at.setter
+    def anonymized_at(self, value: Optional[datetime]) -> None:
+        """Sets the time when the event was anonymized in the tracker store."""
+        if value is not None and not isinstance(value, datetime):
+            raise ValueError(INVALID_DATETIME_ERROR_MESSAGE)
+        self._anonymized_at = value
+
     def __repr__(self) -> Text:
         """Returns text representation of event."""
         return f"SlotSet(key: {self.key}, value: {self.value})"
@@ -1092,7 +1151,14 @@ class SlotSet(Event):
     def as_dict(self) -> Dict[Text, Any]:
         """Returns serialized event."""
         d = super().as_dict()
-        d.update({"name": self.key, "value": self.value, "filled_by": self.filled_by})
+        d.update(
+            {
+                "name": self.key,
+                "value": self.value,
+                "filled_by": self.filled_by,
+                "anonymized_at": self.anonymized_at,
+            }
+        )
         return d
 
     @classmethod

rasa/shared/providers/_configs/azure_openai_client_config.py

@@ -23,6 +23,8 @@ from rasa.shared.constants import (
     DEPLOYMENT_NAME_CONFIG_KEY,
     ENGINE_CONFIG_KEY,
     LANGCHAIN_TYPE_CONFIG_KEY,
+    MAX_COMPLETION_TOKENS_CONFIG_KEY,
+    MAX_TOKENS_CONFIG_KEY,
     MODEL_CONFIG_KEY,
     MODEL_NAME_CONFIG_KEY,
     N_REPHRASES_CONFIG_KEY,
@@ -71,6 +73,8 @@ DEPRECATED_ALIASES_TO_STANDARD_KEY_MAPPING = {
     MODEL_NAME_CONFIG_KEY: MODEL_CONFIG_KEY,
     # Timeout aliases
     REQUEST_TIMEOUT_CONFIG_KEY: TIMEOUT_CONFIG_KEY,
+    # Max tokens aliases
+    MAX_TOKENS_CONFIG_KEY: MAX_COMPLETION_TOKENS_CONFIG_KEY,
 }
 
 REQUIRED_KEYS = [DEPLOYMENT_CONFIG_KEY]

rasa/shared/providers/_configs/openai_client_config.py

@@ -10,6 +10,8 @@ from rasa.shared.constants import (
     API_TYPE_CONFIG_KEY,
     API_VERSION_CONFIG_KEY,
     LANGCHAIN_TYPE_CONFIG_KEY,
+    MAX_COMPLETION_TOKENS_CONFIG_KEY,
+    MAX_TOKENS_CONFIG_KEY,
     MODEL_CONFIG_KEY,
     MODEL_NAME_CONFIG_KEY,
     N_REPHRASES_CONFIG_KEY,
@@ -48,6 +50,8 @@ DEPRECATED_ALIASES_TO_STANDARD_KEY_MAPPING = {
     OPENAI_API_VERSION_CONFIG_KEY: API_VERSION_CONFIG_KEY,
     # Timeout aliases
     REQUEST_TIMEOUT_CONFIG_KEY: TIMEOUT_CONFIG_KEY,
+    # Max tokens aliases
+    MAX_TOKENS_CONFIG_KEY: MAX_COMPLETION_TOKENS_CONFIG_KEY,
 }
 
 REQUIRED_KEYS = [MODEL_CONFIG_KEY]

rasa/shared/providers/embedding/_base_litellm_embedding_client.py

@@ -70,7 +70,10 @@ class _BaseLiteLLMEmbeddingClient:
     def _embedding_fn_args(self) -> Dict[str, Any]:
         """Returns the arguments to be passed to the embedding function."""
         return {
+            # Parameters set through config, can override drop_params
             **self._litellm_extra_parameters,
+            # Model name is constructed in the LiteLLM format from the provided config
+            # Non-overridable to ensure consistency
             "model": self._litellm_model_name,
         }
 

rasa/shared/providers/llm/_base_litellm_client.py

@@ -84,12 +84,15 @@ class _BaseLiteLLMClient:
     @property
     def _completion_fn_args(self) -> dict:
         return {
-            **self._litellm_extra_parameters,
-            "model": self._litellm_model_name,
             # Since all providers covered by LiteLLM use the OpenAI format, but
             # not all support every OpenAI parameter, raise an exception if
             # provider/model uses unsupported parameter
             "drop_params": False,
+            # All other parameters set through config, can override drop_params
+            **self._litellm_extra_parameters,
+            # Model name is constructed in the LiteLLM format from the provided config
+            # Non-overridable to ensure consistency
+            "model": self._litellm_model_name,
         }
 
     def validate_client_setup(self) -> None:

rasa/telemetry.py

@@ -64,7 +64,7 @@ if typing.TYPE_CHECKING:
     from rasa.core.agent import Agent
     from rasa.core.brokers.broker import EventBroker
     from rasa.core.channels.channel import InputChannel
-    from rasa.core.tracker_store import TrackerStore
+    from rasa.core.tracker_stores.tracker_store import TrackerStore
     from rasa.core.utils import AvailableEndpoints
     from rasa.e2e_test.e2e_test_case import Fixture, Metadata, TestCase
     from rasa.shared.importers.importer import TrainingDataImporter
@@ -915,7 +915,7 @@ def initialize_error_reporting() -> None:
             OSError,
         ],
         in_app_include=["rasa"],  # only submit errors in this package
-        with_locals=False,  # don't submit local variables
+        include_local_variables=False,  # don't submit local variables
         release=f"rasa-{rasa.__version__}",
         default_integrations=False,
         environment="development" if in_continuous_integration() else "production",

rasa/tracing/config.py

@@ -23,7 +23,7 @@ from rasa.core.actions.custom_action_executor import (
 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
+from rasa.core.tracker_stores.tracker_store import TrackerStore
 from rasa.dialogue_understanding.commands import (
     Command,
     FreeFormAnswerCommand,

rasa/tracing/instrumentation/attribute_extractors.py

@@ -16,7 +16,7 @@ from rasa.core.channels import UserMessage
 from rasa.core.lock_store import LOCK_LIFETIME, LockStore
 from rasa.core.nlg.contextual_response_rephraser import ContextualResponseRephraser
 from rasa.core.processor import MessageProcessor
-from rasa.core.tracker_store import TrackerStore
+from rasa.core.tracker_stores.tracker_store import TrackerStore
 from rasa.dialogue_understanding.commands import Command
 from rasa.dialogue_understanding.generator import LLMBasedCommandGenerator
 from rasa.dialogue_understanding.generator.constants import FLOW_RETRIEVAL_KEY

rasa/tracing/instrumentation/instrumentation.py

@@ -38,7 +38,7 @@ from rasa.core.nlg import NaturalLanguageGenerator
 from rasa.core.policies.flows.flow_step_result import FlowActionPrediction
 from rasa.core.policies.policy import Policy, PolicyPrediction
 from rasa.core.processor import MessageProcessor
-from rasa.core.tracker_store import TrackerStore
+from rasa.core.tracker_stores.tracker_store import TrackerStore
 from rasa.dialogue_understanding.commands import Command
 from rasa.dialogue_understanding.generator import (
     CompactLLMCommandGenerator,

rasa/utils/licensing.py

@@ -18,8 +18,7 @@ from rasa import telemetry
 from rasa.shared.utils.cli import print_error_and_exit
 
 if typing.TYPE_CHECKING:
-    from rasa.core.tracker_store import TrackerStore
-
+    from rasa.core.tracker_stores.tracker_store import TrackerStore
 
 LICENSE_ENV_VAR = "RASA_PRO_LICENSE"
 ALGORITHM = "RS256"

rasa/version.py

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

{rasa_pro-3.13.0.dev5.dist-info → rasa_pro-3.13.0.dev7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rasa-pro
-Version: 3.13.0.dev5
+Version: 3.13.0.dev7
 Summary: State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development.
 Keywords: nlp,machine-learning,machine-learning-library,bot,bots,botkit,rasa conversational-agents,conversational-ai,chatbot,chatbot-framework,bot-framework
 Author: Rasa Technologies GmbH
@@ -63,12 +63,12 @@ Requires-Dist: keras (==2.14.0)
 Requires-Dist: langchain (>=0.2.17,<0.3.0)
 Requires-Dist: langchain-community (>=0.2.19,<0.3.0)
 Requires-Dist: langcodes (>=3.5.0,<4.0.0)
-Requires-Dist: litellm (==1.52.6)
+Requires-Dist: litellm (>=1.69.0,<1.70.0)
 Requires-Dist: matplotlib (>=3.7,<3.8)
 Requires-Dist: mattermostwrapper (>=2.2,<2.3)
 Requires-Dist: networkx (>=3.1,<3.2)
 Requires-Dist: numpy (>=1.26.4,<1.27.0)
-Requires-Dist: openai (>=1.55.3,<1.56.0)
+Requires-Dist: openai (>=1.68.2,<1.69.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)
@@ -115,7 +115,7 @@ Requires-Dist: sanic-routing (>=22.8.0,<23.0.0)
 Requires-Dist: scikit-learn (>=1.5.1,<1.6.0)
 Requires-Dist: scipy (>=1.13.1,<1.14.0)
 Requires-Dist: sentencepiece[sentencepiece] (>=0.1.99,<0.2.0) ; extra == "transformers" or extra == "full"
-Requires-Dist: sentry-sdk (>=1.14.0,<1.15.0)
+Requires-Dist: sentry-sdk (>=2.8.0,<3)
 Requires-Dist: setuptools (>=70.0.0,<70.1.0)
 Requires-Dist: sklearn-crfsuite (>=0.3.6,<0.4.0)
 Requires-Dist: skops (>=0.10.0,<0.11.0)