rasa-pro 3.13.0.dev9__py3-none-any.whl → 3.13.0.dev11__py3-none-any.whl

rasa/cli/export.py

@@ -184,6 +184,7 @@ async def _export_trackers(args: argparse.Namespace) -> None:
     event_broker = await _get_event_broker(endpoints)
     _prepare_event_broker(event_broker)
     requested_conversation_ids = _get_requested_conversation_ids(args.conversation_ids)
+    is_pii_enabled = True if endpoints.privacy else False
 
     from rasa.core.exporter import Exporter
 
@@ -191,6 +192,7 @@ async def _export_trackers(args: argparse.Namespace) -> None:
         tracker_store,
         event_broker,
         args.endpoints,
+        is_pii_enabled,
         requested_conversation_ids,
         args.minimum_timestamp,
         args.maximum_timestamp,

rasa/cli/studio/download.py

@@ -4,11 +4,12 @@ from typing import List
 from rasa.cli import SubParsersAction
 from rasa.cli.arguments.default_arguments import (
     add_config_param,
+    add_data_param,
+    add_domain_param,
     add_endpoint_param,
 )
-from rasa.cli.arguments.train import add_data_param, add_domain_param
 from rasa.shared.constants import DEFAULT_CONFIG_PATH, DEFAULT_ENDPOINTS_PATH
-from rasa.studio.download.download import handle_download
+from rasa.studio.download import handle_download
 
 
 def add_subparser(
@@ -49,13 +50,6 @@ def set_studio_download_arguments(parser: argparse.ArgumentParser) -> None:
     parser.add_argument(
         "assistant_name",
         default=None,
-        nargs=1,
         type=str,
         help="Name of the assistant on Rasa Studio",
     )
-
-    parser.add_argument(
-        "--overwrite",
-        action="store_true",
-        help="Overwrite local data with data from Rasa Studio",
-    )

rasa/cli/studio/link.py

@@ -2,7 +2,7 @@ import argparse
 from typing import List, Text
 
 from rasa.cli import SubParsersAction
-from rasa.cli.studio.upload import (
+from rasa.cli.arguments.default_arguments import (
     add_config_param,
     add_data_param,
     add_domain_param,
@@ -37,7 +37,6 @@ def add_subparser(
 
     link_parser.add_argument(
         "assistant_name",
-        nargs=1,
         type=str,
         help="Name of the assistant in Rasa Studio.",
     )

rasa/cli/studio/pull.py

@@ -4,15 +4,16 @@ from typing import List, Text
 from rasa.cli import SubParsersAction
 from rasa.cli.arguments.default_arguments import (
     add_config_param,
+    add_data_param,
+    add_domain_param,
     add_endpoint_param,
 )
-from rasa.cli.studio.upload import add_data_param, add_domain_param
 from rasa.shared.constants import (
     DEFAULT_CONFIG_PATH,
     DEFAULT_DOMAIN_PATH,
     DEFAULT_ENDPOINTS_PATH,
 )
-from rasa.studio.pull import (
+from rasa.studio.pull.pull import (
     handle_pull,
     handle_pull_config,
     handle_pull_endpoints,

rasa/cli/studio/push.py

@@ -2,7 +2,7 @@ import argparse
 from typing import List, Text
 
 from rasa.cli import SubParsersAction
-from rasa.cli.studio.upload import (
+from rasa.cli.arguments.default_arguments import (
     add_config_param,
     add_data_param,
     add_domain_param,

rasa/cli/studio/train.py

@@ -36,7 +36,6 @@ def set_studio_train_arguments(parser: argparse.ArgumentParser) -> None:
     parser.add_argument(
         "assistant_name",
         default=None,
-        nargs=1,
         type=str,
         help="Name of the assistant on Rasa Studio",
     )

rasa/core/exporter.py

@@ -16,6 +16,11 @@ from rasa.exceptions import (
     NoEventsToMigrateError,
     PublishingError,
 )
+from rasa.shared.core.events import (
+    BotUttered,
+    SlotSet,
+    UserUttered,
+)
 from rasa.shared.core.trackers import EventVerbosity
 
 logger = logging.getLogger(__name__)
@@ -43,6 +48,7 @@ class Exporter:
         tracker_store: TrackerStore,
         event_broker: EventBroker,
         endpoints_path: Text,
+        is_pii_enabled: bool = False,
         requested_conversation_ids: Optional[Text] = None,
         minimum_timestamp: Optional[float] = None,
         maximum_timestamp: Optional[float] = None,
@@ -52,6 +58,7 @@ class Exporter:
         self.tracker_store = tracker_store
 
         self.event_broker = event_broker
+        self.is_pii_enabled = is_pii_enabled
         self.requested_conversation_ids = requested_conversation_ids
         self.minimum_timestamp = minimum_timestamp
         self.maximum_timestamp = maximum_timestamp
@@ -72,10 +79,12 @@ class Exporter:
         current_timestamp = None
 
         headers = self._get_message_headers()
+        warned_sender_ids: Set[Text] = set()
 
         async for event in self._fetch_events_within_time_range():
             # noinspection PyBroadException
             try:
+                self._check_anonymization_status(event, warned_sender_ids)
                 self._publish_with_message_headers(event, headers)
                 published_events += 1
                 current_timestamp = event["timestamp"]
@@ -282,3 +291,30 @@ class Exporter:
             events_with_conversation_id.append(event)
 
         return events_with_conversation_id
+
+    def _check_anonymization_status(
+        self, event: Dict[Text, Any], warned_sender_ids: Set[Text]
+    ) -> None:
+        """Check if the tracker store contains unanonymized events.
+
+        If it does, print a warning that these events will be published as is.
+
+        Args:
+            event: The event to check for anonymization status
+            warned_sender_ids: Set of sender IDs that have already been warned about
+        """
+        sender_id = event["sender_id"]
+        if (
+            self.is_pii_enabled
+            and sender_id not in warned_sender_ids
+            and event["event"]
+            in (UserUttered.type_name, BotUttered.type_name, SlotSet.type_name)
+            and not event.get("anonymized_at", None)
+        ):
+            rasa.shared.utils.cli.print_warning(
+                f"Retrieved un-anonymized event for sender_id {sender_id}. "
+                f"All events after this timestamp {event['timestamp']} "
+                "are not anonymized for this tracker. Proceeding with "
+                "publishing plaintext values for all events following this.",
+            )
+            warned_sender_ids.add(sender_id)