rasa-pro 3.13.0.dev20250613__py3-none-any.whl → 3.13.0rc2__py3-none-any.whl

rasa/dialogue_understanding/generator/command_generator.py

@@ -8,6 +8,7 @@ from rasa.dialogue_understanding.commands import (
     Command,
     CorrectSlotsCommand,
     ErrorCommand,
+    HandleCodeChangeCommand,
     SetSlotCommand,
     StartFlowCommand,
 )
@@ -398,15 +399,24 @@ class CommandGenerator:
             The filtered commands.
         """
         from rasa.dialogue_understanding.processor.command_processor import (
+            find_updated_flows,
             get_current_collect_step,
         )
 
         if tracker is None:
-            structlogger.error(
+            structlogger.debug(
                 "command_generator.filter_commands_during_force_slot_filling.tracker_not_found",
             )
             return commands
 
+        updated_flows = find_updated_flows(tracker, available_flows)
+        if updated_flows:
+            structlogger.debug(
+                "command_generator.filter_commands_during_force_slot_filling.running_flows_were_updated",
+                updated_flow_ids=updated_flows,
+            )
+            return [HandleCodeChangeCommand()]
+
         stack = tracker.stack
         step = get_current_collect_step(stack, available_flows)
 

rasa/dialogue_understanding/generator/multi_step/multi_step_llm_command_generator.py

@@ -125,8 +125,9 @@ class MultiStepLLMCommandGenerator(LLMBasedCommandGenerator):
         raise_deprecation_warning(
             message=(
                 "Support for `MultiStepLLMCommandGenerator` will be removed in Rasa "
-                "`4.0.0`. Please modify your assistant's configuration to use other "
-                "LLM command generators like the `SingleStepLLMCommandGenerator`."
+                "`4.0.0`. Please modify your assistant's configuration to use the "
+                "`CompactLLMCommandGenerator` or `SearchReadyLLMCommandGenerator` "
+                "instead."
             )
         )
 

rasa/dialogue_understanding/generator/nlu_command_adapter.py

@@ -190,9 +190,9 @@ class NLUCommandAdapter(GraphComponent, CommandGenerator):
         if len(commands) > 1:
             structlogger.warning(
                 "nlu_command_adapter.predict_commands",
-                messag=f"Two many flows found that are triggered by the "
+                message=f"Too many flows found that are triggered by the "
                 f"intent '{message.get(INTENT)['name']}'. Take the first one.",
-                commands=commands,
+                commands=[command.__class__.__name__ for command in commands],
             )
             commands = [commands[0]]
 

rasa/dialogue_understanding/generator/prompt_templates/command_prompt_template.jinja2

@@ -41,9 +41,7 @@ Based on this information generate a list of actions you want to take. Your job
 * Responding to knowledge-oriented user messages, described by "SearchAndReply()"
 * Responding to a casual, non-task-oriented user message, described by "ChitChat()".
 * Handing off to a human, in case the user seems frustrated or explicitly asks to speak to one, described by "HumanHandoff()".
-{% if is_repeat_command_enabled %}
 * Repeat the last bot messages, described by "RepeatLastBotMessages()". This is useful when the user asks to repeat the last bot messages.
-{% endif %}
 
 ===
 Write out the actions you want to take, one per line, in the order they should take place.

rasa/dialogue_understanding/generator/prompt_templates/command_prompt_v2_claude_3_5_sonnet_20240620_template.jinja2

@@ -11,6 +11,7 @@ Your task is to analyze the current conversation context and generate a list of
 * `provide info`: Responding to the user's questions by supplying relevant information, such as answering FAQs or explaining services.
 * `offtopic reply`: Responding to casual or social user messages that are unrelated to any flows, engaging in friendly conversation and addressing off-topic remarks.
 * `hand over`: Handing over to a human, in case the user seems frustrated or explicitly asks to speak to one.
+* `repeat message`: Repeating the last bot message.
 
 --
 

rasa/dialogue_understanding/generator/prompt_templates/command_prompt_v2_gpt_4o_2024_11_20_template.jinja2

@@ -19,6 +19,7 @@ Use the following structured data:
 * `provide info`: Responding to the user's questions by supplying relevant information, such as answering FAQs or explaining services.
 * `offtopic reply`: Responding to casual or social user messages that are unrelated to any flows, engaging in friendly conversation and addressing off-topic remarks.
 * `hand over`: Handing over to a human, in case the user seems frustrated or explicitly asks to speak to one.
+* `repeat message`: Repeating the last bot message.
 
 ---
 

rasa/dialogue_understanding/generator/prompt_templates/command_prompt_v3_claude_3_5_sonnet_20240620_template.jinja2

@@ -0,0 +1,79 @@
+## Task Description
+Your task is to analyze the current conversation context and generate a list of actions to start new business processes that we call flows, to extract slots, or respond to off-topic and knowledge requests.
+
+---
+
+## Available Actions:
+* `start flow flow_name`: Start a flow. For example, `start flow transfer_money` or `start flow list_contacts`.
+* `set slot slot_name slot_value`: Set a slot for the active flow. For example, `set slot transfer_money_recipient Freddy`. Can be used to correct and change previously set values.
+* `disambiguate flows flow_name1 flow_name2 ... flow_name_n`: When a message could refer to multiple flows, list the possible flows as options to clarify. Example: `disambiguate flows list_contacts add_contact remove_contact`.
+* `search and reply`: Provide a response from the knowledge base to address the user's inquiry when no flows fit, including domain knowledge, FAQs, and all off-topic or social messages.
+* `cancel flow`: Cancel the current flow if the user requests it.
+* `repeat message`: Repeat the last bot message.
+
+---
+
+## General Instructions
+### Start Flow
+* Only start a flow if the user's message is clear and fully addressed by that flow's description and purpose.
+* Pay close attention to exact wording and scope in the flow description — do not assume or “stretch” the intended use of a flow.
+### Set Slot
+* Do not fill slots with abstract values or placeholders.
+* For categorical slots, try to match the user message with allowed slot values. Use "other" if you cannot match it.
+* Set the boolean slots based on the user's response. Map positive responses to `True`, and negative to `False`.
+* Extract text slot values exactly as provided by the user. Avoid assumptions, format changes, or partial extractions.
+### Disambiguate Flows
+* Use `disambiguate flows` when the user's message matches multiple flows and you cannot decide which flow is most appropriate.
+* If the user message is short and not precise enough to start a flow or `search and reply`, disambiguate.
+* If a single flow is a strong/plausible fit, prefer starting that flow directly.
+* If a user's message unambiguously and distinctly matches multiple flows, start all relevant flows at once (rather than disambiguating).
+### Search and Reply
+* Only start `search and reply` if the user intent is clear.
+* Flow Priority: If you are unsure between starting a flow or `search and reply`, always prioritize starting a flow.
+### Cancel Flow
+* Do not cancel any flow unless the user explicitly requests it.
+* Multiple flows can be started without cancelling the previous, if the user wants to pursue multiple processes.
+### General Tips
+* Only use information provided by the user.
+* Strictly adhere to the provided action format.
+* Focus on the last message and take it one step at a time.
+* Use the previous conversation steps only to aid understanding.
+
+---
+
+## Decision Rule Table
+| Condition                                             | Action             |
+|-------------------------------------------------------|--------------------|
+| Flow perfectly matches user's message                 | start flow         |
+| Multiple flows are equally strong, relevant matches   | disambiguate flows |
+| User's message is unclear or imprecise                | disambiguate flows |
+| No flow fits at all, but knowledge base may help      | search and reply   |
+
+---
+
+## Available Flows and Slots
+Use the following structured data:
+```json
+{"flows":[{% for flow in available_flows %}{"name":"{{ flow.name }}","description":{{ flow.description | to_json_escaped_string }}{% if flow.slots %},"slots":[{% for slot in flow.slots %}{"name":"{{ slot.name }}"{% if slot.description %},"description":{{ slot.description | to_json_escaped_string }}{% endif %}{% if slot.allowed_values %},"allowed_values":{{ slot.allowed_values }}{% endif %}}{% if not loop.last %},{% endif %}{% endfor %}]{% endif %}}{% if not loop.last %},{% endif %}{% endfor %}]}
+```
+
+---
+
+## Current State
+{% if current_flow != None %}Use the following structured data:
+```json
+{"active_flow":"{{ current_flow }}","current_step":{"requested_slot":"{{ current_slot }}","requested_slot_description":{{ current_slot_description | to_json_escaped_string }}},"slots":[{% for slot in flow_slots %}{"name":"{{ slot.name }}","value":"{{ slot.value }}","type":"{{ slot.type }}"{% if slot.description %},"description":{{ slot.description | to_json_escaped_string }}{% endif %}{% if slot.allowed_values %},"allowed_values":"{{ slot.allowed_values }}"{% endif %}}{% if not loop.last %},{% endif %}{% endfor %}]}
+```{% else %}
+You are currently not inside any flow.{% endif %}
+
+---
+
+## Conversation History
+{{ current_conversation }}
+
+---
+
+## Task
+Create an action list with one action per line in response to the user's last message: """{{ user_message }}""".
+
+Your action list:

rasa/dialogue_understanding/generator/prompt_templates/command_prompt_v3_gpt_4o_2024_11_20_template.jinja2

@@ -17,6 +17,7 @@ Use the following structured data:
 * `disambiguate flows flow_name1 flow_name2 ... flow_name_n`: When a message could refer to multiple flows, list the possible flows as options to clarify. Example: `disambiguate flows list_contacts add_contact remove_contact`.
 * `search and reply`: Provide a response from the knowledge base to address the user’s inquiry when no flows fit, including domain knowledge, FAQs, and all off-topic or social messages.
 * `cancel flow`: Cancel the current flow if the user requests it.
+* `repeat message`: Repeat the last bot message.
 
 ---
 

rasa/dialogue_understanding/generator/single_step/search_ready_llm_command_generator.py

@@ -59,10 +59,10 @@ MODEL_PROMPT_MAPPER = {
     ),
     f"{AWS_BEDROCK_PROVIDER}/anthropic."
     f"{MODEL_NAME_CLAUDE_3_5_SONNET_20240620}-v1:0": (
-        "command_prompt_v2_claude_3_5_sonnet_20240620_template.jinja2"
+        "command_prompt_v3_claude_3_5_sonnet_20240620_template.jinja2"
     ),
     f"{ANTHROPIC_PROVIDER}/{MODEL_NAME_CLAUDE_3_5_SONNET_20240620}": (
-        "command_prompt_v2_claude_3_5_sonnet_20240620_template.jinja2"
+        "command_prompt_v3_claude_3_5_sonnet_20240620_template.jinja2"
     ),
 }
 

rasa/dialogue_understanding/generator/single_step/single_step_based_llm_command_generator.py

@@ -61,7 +61,6 @@ from rasa.shared.utils.llm import (
     sanitize_message_for_prompt,
     tracker_as_readable_transcript,
 )
-from rasa.utils.beta import BetaNotEnabledException, ensure_beta_feature_is_enabled
 from rasa.utils.log_utils import log_llm
 
 structlogger = structlog.get_logger()
@@ -106,14 +105,13 @@ class SingleStepBasedLLMCommandGenerator(LLMBasedCommandGenerator, ABC):
         )
 
         self.trace_prompt_tokens = self.config.get("trace_prompt_tokens", False)
-        self.repeat_command_enabled = self.is_repeat_command_enabled()
 
     ### Implementations of LLMBasedCommandGenerator parent
     @staticmethod
     def get_default_config() -> Dict[str, Any]:
         """The component's default config (see parent class for full docstring)."""
         return {
-            PROMPT_TEMPLATE_CONFIG_KEY: None,
+            PROMPT_TEMPLATE_CONFIG_KEY: None,  # TODO: remove in Rasa 4.0.0
             USER_INPUT_CONFIG_KEY: None,
             LLM_CONFIG_KEY: None,
             FLOW_RETRIEVAL_KEY: FlowRetrieval.get_default_config(),
@@ -212,7 +210,7 @@ class SingleStepBasedLLMCommandGenerator(LLMBasedCommandGenerator, ABC):
                 event_info=(
                     "ProviderClientAPIException occurred while predicting commands."
                 ),
-                commands=commands,
+                commands=commands,  # no PII
             )
 
         if not commands and not prior_commands:
@@ -414,24 +412,10 @@ class SingleStepBasedLLMCommandGenerator(LLMBasedCommandGenerator, ABC):
             "current_slot_type": current_slot_type,
             "current_slot_allowed_values": current_slot_allowed_values,
             "user_message": latest_user_message,
-            "is_repeat_command_enabled": self.repeat_command_enabled,
         }
 
         return self.compile_template(self.prompt_template).render(**inputs)
 
-    def is_repeat_command_enabled(self) -> bool:
-        """Check for feature flag"""
-        RASA_PRO_BETA_REPEAT_COMMAND_ENV_VAR_NAME = "RASA_PRO_BETA_REPEAT_COMMAND"
-        try:
-            ensure_beta_feature_is_enabled(
-                "Repeat Command",
-                env_flag=RASA_PRO_BETA_REPEAT_COMMAND_ENV_VAR_NAME,
-            )
-        except BetaNotEnabledException:
-            return False
-
-        return True
-
     @classmethod
     def fingerprint_addon(cls: Any, config: Dict[str, Any]) -> Optional[str]:
         """Add a fingerprint for the graph."""

rasa/dialogue_understanding/generator/single_step/single_step_llm_command_generator.py

@@ -16,7 +16,11 @@ from rasa.shared.constants import (
     PROMPT_CONFIG_KEY,
     PROMPT_TEMPLATE_CONFIG_KEY,
 )
-from rasa.shared.utils.llm import get_prompt_template
+from rasa.shared.utils.io import raise_deprecation_warning
+from rasa.shared.utils.llm import (
+    check_prompt_config_keys_and_warn_if_deprecated,
+    get_prompt_template,
+)
 
 DEFAULT_COMMAND_PROMPT_TEMPLATE = importlib.resources.read_text(
     "rasa.dialogue_understanding.generator.prompt_templates",
@@ -44,6 +48,14 @@ class SingleStepLLMCommandGenerator(SingleStepBasedLLMCommandGenerator):
         prompt_template: Optional[Text] = None,
         **kwargs: Any,
     ) -> None:
+        raise_deprecation_warning(
+            message=(
+                "Support for `SingleStepLLMCommandGenerator` will be removed in Rasa "
+                "`4.0.0`. Please modify your assistant's configuration to use the "
+                "`CompactLLMCommandGenerator` or `SearchReadyLLMCommandGenerator` "
+                "instead."
+            )
+        )
         super().__init__(
             config,
             model_storage,
@@ -52,16 +64,10 @@ class SingleStepLLMCommandGenerator(SingleStepBasedLLMCommandGenerator):
             **kwargs,
         )
 
-        # Set the prompt template
-        if config.get(PROMPT_CONFIG_KEY):
-            structlogger.warning(
-                "single_step_llm_command_generator.init",
-                event_info=(
-                    "The config parameter 'prompt' is deprecated "
-                    "and will be removed in Rasa 4.0.0. "
-                    "Please use the config parameter 'prompt_template' instead. "
-                ),
-            )
+        # Warn if the prompt config key is used to set the prompt template
+        check_prompt_config_keys_and_warn_if_deprecated(
+            config, "single_step_llm_command_generator"
+        )
 
     @staticmethod
     def get_component_command_syntax_version() -> CommandSyntaxVersion:

rasa/dialogue_understanding/patterns/cancel.py

@@ -90,7 +90,7 @@ class ActionCancelFlow(action.Action):
             return []
 
         if not isinstance(top, CancelPatternFlowStackFrame):
-            structlogger.warning("action.cancel_flow.no_cancel_frame", top=top)
+            structlogger.warning("action.cancel_flow.no_cancel_frame")
             return []
 
         for canceled_frame_id in top.canceled_frames:
@@ -105,7 +105,6 @@ class ActionCancelFlow(action.Action):
             else:
                 structlogger.warning(
                     "action.cancel_flow.frame_not_found",
-                    dialogue_stack=stack,
                     frame_id=canceled_frame_id,
                 )
 

rasa/dialogue_understanding/patterns/clarify.py

@@ -89,7 +89,7 @@ class ActionClarifyFlows(action.Action):
             return []
 
         if not isinstance(top, ClarifyPatternFlowStackFrame):
-            structlogger.warning("action.clarify_flows.no_clarification_frame", top=top)
+            structlogger.warning("action.clarify_flows.no_clarification_frame")
             return []
 
         options_string = self.assemble_options_string(top.names)

rasa/dialogue_understanding/patterns/correction.py

@@ -114,7 +114,8 @@ class ActionCorrectFlowSlot(action.Action):
 
         if not isinstance(top, CorrectionPatternFlowStackFrame):
             structlogger.warning(
-                "action.correct_flow_slot.no_correction_frame", top=top
+                "action.correct_flow_slot.no_correction_frame",
+                top=top,  # no PII
             )
             return []
 
@@ -246,7 +247,6 @@ def reset_stack_on_tracker_to_prior_state(
             "action.correct_flow_slot.no_target_frame_found",
             reset_step_id=reset_step_id,
             reset_flow_id=reset_flow_id,
-            stack_to_reset_to=stack_to_reset_to,
         )
         return tracker.stack
 

rasa/dialogue_understanding/processor/command_processor.py

@@ -214,18 +214,18 @@ 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, story_graph, domain
-    )
-
     updated_flows = find_updated_flows(tracker, all_flows)
     if updated_flows:
-        # Override commands
+        # if there are updated flows, we need to handle the code change
         structlogger.debug(
             "command_processor.execute_commands.running_flows_were_updated",
             updated_flow_ids=updated_flows,
         )
         commands = [HandleCodeChangeCommand()]
+    else:
+        commands = clean_up_commands(
+            commands, tracker, all_flows, execution_context, story_graph, domain
+        )
 
     # store current flow hashes if they changed
     new_hashes = calculate_flow_fingerprints(all_flows)
@@ -348,7 +348,6 @@ def get_current_collect_step(
         # but no flow that triggered it. this should never happen.
         structlogger.warning(
             "command_processor.get_current_collect_step.no_flow_on_stack",
-            stack=dialogue_stack,
         )
         return None
 
@@ -358,7 +357,7 @@ def get_current_collect_step(
         # step from it
         structlogger.warning(
             "command_processor.get_current_collect_step.no_step_for_frame",
-            frame=frame_that_triggered_collect_infos,
+            frame=frame_that_triggered_collect_infos.frame_id,
         )
         return None
 
@@ -640,9 +639,9 @@ def clean_up_slot_command(
             resulting_commands.append(command)
             return resulting_commands
 
-        if (slot := tracker.slots.get(command.name)) is not None and slot.value == str(
-            command.value
-        ):
+        if (slot := tracker.slots.get(command.name)) is not None and str(
+            slot.value
+        ) == str(command.value):
             # the slot is already set, we don't need to set it again
             structlogger.debug(
                 "command_processor.clean_up_slot_command.skip_command_slot_already_set",
@@ -724,7 +723,7 @@ def clean_up_chitchat_command(
         )
         structlogger.warn(
             "command_processor.clean_up_chitchat_command.pattern_chitchat_not_found",
-            command=resulting_commands[0],
+            command=resulting_commands[0],  # no PII
         )
         return resulting_commands
 
@@ -742,7 +741,7 @@ def clean_up_chitchat_command(
         )
         structlogger.warn(
             "command_processor.clean_up_chitchat_command.replace_chitchat_answer_with_cannot_handle",
-            command=resulting_commands[0],
+            command=resulting_commands[0],  # no PII
             pattern_chitchat_uses_action_trigger_chitchat=has_action_trigger_chitchat,
             defined_intentless_policy_in_config=defines_intentless_policy,
         )

rasa/dialogue_understanding/stack/utils.py

@@ -209,7 +209,9 @@ def get_collect_steps_excluding_ask_before_filling_for_active_flow(
         All collect steps that are part of the current active flow,
         excluding the collect steps that have to be asked before filling.
     """
-    active_frame = top_user_flow_frame(dialogue_stack)
+    active_frame = top_user_flow_frame(
+        dialogue_stack, ignore_call_and_link_frames=False
+    )
     if active_frame is None:
         return set()
     active_flow = active_frame.flow(all_flows)

rasa/e2e_test/constants.py

@@ -40,7 +40,7 @@ DEFAULT_GROUNDEDNESS_PROMPT_TEMPLATE_FILE_NAME = "groundedness_prompt_template.j
 DEFAULT_ANSWER_RELEVANCE_PROMPT_TEMPLATE_FILE_NAME = (
     "answer_relevance_prompt_template.jinja2"
 )
-DEFAULT_E2E_TESTING_MODEL = "gpt-4o-mini"
+DEFAULT_E2E_TESTING_MODEL = "gpt-4.1-mini-2025-04-14"
 KEY_SCORE = "score"
 KEY_JUSTIFICATION = "justification"
 KEY_EXTRA_PARAMETERS = "extra_parameters"

rasa/e2e_test/e2e_test_coverage_report.py

@@ -21,7 +21,7 @@ from rasa.shared.core.flows.flow_path import FlowPath, FlowPathsList, PathNode
 FLOW_NAME_COL_NAME = "Flow Name"
 NUM_STEPS_COL_NAME = "Num Steps"
 MISSING_STEPS_COL_NAME = "Missing Steps"
-LINE_NUMBERS_COL_NAME = "Line Numbers"
+LINE_NUMBERS_COL_NAME = "Line Numbers for Missing Steps"
 COVERAGE_COL_NAME = "Coverage"
 
 FLOWS_KEY = "flows"

rasa/engine/graph.py

@@ -500,9 +500,9 @@ class GraphNode:
                 structlogger.warning(
                     "graph.node.input_not_resolved",
                     node_name=self._node_name,
-                    input_name=i,
+                    input_name=i,  # no PII
                     event_info=(
-                        "Node input was not resolved, there is no putput. "
+                        "Node input was not resolved, there is no output. "
                         "Another component should have provided this as an output."
                     ),
                 )

rasa/llm_fine_tuning/paraphrasing/conversation_rephraser.py

@@ -22,11 +22,7 @@ from rasa.shared.constants import (
 from rasa.shared.exceptions import ProviderClientAPIException
 from rasa.shared.providers.mappings import OPENAI_PROVIDER
 from rasa.shared.utils.constants import LOG_COMPONENT_SOURCE_METHOD_INIT
-from rasa.shared.utils.llm import (
-    USER,
-    get_prompt_template,
-    llm_factory,
-)
+from rasa.shared.utils.llm import USER, get_prompt_template, llm_factory
 
 SEPARATOR = "\n\n"
 BACKUP_SEPARATOR = "\nUSER:"
@@ -39,7 +35,7 @@ DEFAULT_REPHRASING_PROMPT_TEMPLATE = importlib.resources.read_text(
 
 DEFAULT_LLM_CONFIG = {
     PROVIDER_CONFIG_KEY: OPENAI_PROVIDER,
-    MODEL_CONFIG_KEY: "gpt-4o-mini",
+    MODEL_CONFIG_KEY: "gpt-4.1-mini-2025-04-14",
     TIMEOUT_CONFIG_KEY: 7,
     TEMPERATURE_CONFIG_KEY: 0.0,
     MAX_COMPLETION_TOKENS_CONFIG_KEY: 4096,

rasa/model_manager/runner_service.py

@@ -2,7 +2,8 @@ import os
 import shutil
 import subprocess
 from enum import Enum
-from typing import Dict, Optional
+from pathlib import Path
+from typing import Dict, Optional, Union
 
 import aiohttp
 import structlog
@@ -18,6 +19,7 @@ from rasa.model_manager.utils import (
     write_encoded_data_to_file,
 )
 from rasa.model_manager.warm_rasa_process import start_rasa_process
+from rasa.studio.prompts import handle_prompts
 
 structlogger = structlog.get_logger()
 
@@ -121,11 +123,25 @@ def get_open_port() -> int:
 
 
 def write_encoded_config_data_to_files(
-    encoded_configs: Dict[str, bytes], base_path: str
+    encoded_configs: Dict[str, Union[bytes, Dict[str, str]]], base_path: str
 ) -> None:
     """Write the encoded config data to files."""
-    for key, value in encoded_configs.items():
-        write_encoded_data_to_file(value, subpath(base_path, f"{key}.yml"))
+    endpoints_encoded = encoded_configs.get("endpoints")
+    if endpoints_encoded:
+        write_encoded_data_to_file(
+            endpoints_encoded, subpath(base_path, "endpoints.yml")
+        )
+    config_encoded = encoded_configs.get("config")
+    if config_encoded:
+        write_encoded_data_to_file(config_encoded, subpath(base_path, "config.yml"))
+    credentials_encoded = encoded_configs.get("credentials")
+    if credentials_encoded:
+        write_encoded_data_to_file(
+            credentials_encoded, subpath(base_path, "credentials.yml")
+        )
+
+    if prompts := encoded_configs.get("prompts"):
+        handle_prompts(prompts, Path(base_path))
 
 
 def prepare_bot_directory(

rasa/model_manager/trainer_service.py

@@ -2,6 +2,7 @@ import os
 import shutil
 import subprocess
 from enum import Enum
+from pathlib import Path
 from typing import Any, Dict, Optional
 
 import structlog
@@ -20,6 +21,7 @@ from rasa.model_manager.warm_rasa_process import (
     start_rasa_process,
 )
 from rasa.model_training import generate_random_model_name
+from rasa.studio.prompts import handle_prompts
 
 structlogger = structlog.get_logger()
 
@@ -208,6 +210,7 @@ def write_training_data_to_files(
         "stories": "base64 encoded stories.yml",
         "rules": "base64 encoded rules.yml",
         "nlu": "base64 encoded nlu.yml"
+        "prompts": "dictionary with the prompts",
     }
     ```
     """
@@ -230,6 +233,9 @@ def write_training_data_to_files(
             subpath(training_base_path + "/" + parent_path, file_name),
         )
 
+    if prompts := encoded_training_data.get("prompts"):
+        handle_prompts(prompts, Path(training_base_path))
+
 
 def prepare_training_directory(
     training_base_path: str, assistant_id: str, encoded_training_data: Dict[str, Any]

rasa/privacy/privacy_manager.py

@@ -12,7 +12,7 @@ import structlog
 from apscheduler.schedulers.background import BackgroundScheduler
 
 import rasa.shared.core.trackers
-from rasa.core.tracker_stores.tracker_store import TrackerStore
+from rasa.core.tracker_stores.tracker_store import FailSafeTrackerStore, TrackerStore
 from rasa.privacy.constants import (
     TEXT_KEY,
     USER_CHAT_INACTIVITY_IN_MINUTES_ENV_VAR_NAME,
@@ -63,6 +63,7 @@ class BackgroundPrivacyManager:
         self,
         endpoints: Optional["AvailableEndpoints"],
         event_loop: Optional["AbstractEventLoop"] = None,
+        in_memory_tracker_store: Optional[TrackerStore] = None,
     ):
         self.config = (
             PrivacyConfig.from_dict(endpoints.privacy)
@@ -76,15 +77,28 @@ class BackgroundPrivacyManager:
             os.getenv(USER_CHAT_INACTIVITY_IN_MINUTES_ENV_VAR_NAME, 30)
         )
 
-        # we recreate the tracker store here to ensure
-        # that this instance has no event brokers
-        # that could publish events during the tracker store
-        # background jobs
-        self.tracker_store = (
-            TrackerStore.create(endpoints.tracker_store)
-            if endpoints
-            else TrackerStore.create(None)
-        )
+        if in_memory_tracker_store is not None:
+            # if an in-memory tracker store is provided,
+            # we need to keep the reference to it
+            # so that the background jobs can access it.
+            # We also set the event broker to None
+            # to prevent it from publishing events
+            # during the tracker store background jobs
+            in_memory_tracker_store.event_broker = None
+            tracker_store = in_memory_tracker_store
+        else:
+            # we recreate the tracker store here to ensure
+            # that this instance has no event brokers
+            # that could publish events during the tracker store
+            # background jobs
+            tracker_store = (
+                TrackerStore.create(endpoints.tracker_store)
+                if endpoints
+                else TrackerStore.create(None)
+            )
+
+        self.tracker_store = FailSafeTrackerStore(tracker_store)
+
         self.event_brokers: List["EventBroker"] = []
         self.event_loop = event_loop
 
@@ -124,9 +138,10 @@ class BackgroundPrivacyManager:
         cls,
         endpoints: Optional["AvailableEndpoints"],
         event_loop: Optional["AbstractEventLoop"] = None,
+        in_memory_tracker_store: Optional[TrackerStore] = None,
     ) -> BackgroundPrivacyManager:
         """Create an instance of BackgroundPrivacyManager."""
-        instance = cls(endpoints, event_loop)
+        instance = cls(endpoints, event_loop, in_memory_tracker_store)
         return await instance.initialize(endpoints)
 
     def stop(self) -> None: