rasa-pro 3.14.0rc3__py3-none-any.whl → 3.14.1__py3-none-any.whl

rasa/builder/jobs.py

@@ -4,6 +4,13 @@ import structlog
 from sanic import Sanic
 
 from rasa.builder import config
+from rasa.builder.copilot.constants import (
+    PROMPT_TO_BOT_KEY,
+)
+from rasa.builder.copilot.copilot_templated_message_provider import (
+    load_copilot_handler_default_responses,
+    load_copilot_welcome_messages,
+)
 from rasa.builder.copilot.models import (
     CopilotContext,
     FileContent,
@@ -80,11 +87,20 @@ async def run_prompt_to_bot_job(
         update_agent(agent, app)
         await push_job_status_event(job, JobStatus.train_success)
 
+        # 3. Create copilot welcome message job
+        copilot_welcome_job = job_manager.create_job()
+        app.add_task(run_copilot_welcome_message_job(app, copilot_welcome_job))
+
         structlogger.info(
             "bot_builder_service.prompt_to_bot.success",
             files_generated=list(bot_files.keys()),
+            copilot_welcome_job_id=copilot_welcome_job.id,
+        )
+        await push_job_status_event(
+            job=job,
+            status=JobStatus.done,
+            payload={"copilot_welcome_job_id": copilot_welcome_job.id},
         )
-        await push_job_status_event(job, JobStatus.done)
         job_manager.mark_done(job)
 
     except TrainingError as exc:
@@ -165,12 +181,23 @@ async def run_template_to_bot_job(
         update_agent(agent, app)
         await push_job_status_event(job, JobStatus.train_success)
 
-        # 3) Done
+        # 3) Create copilot welcome message job
+        copilot_welcome_job = job_manager.create_job()
+        app.add_task(
+            run_copilot_welcome_message_job(app, copilot_welcome_job, template_name)
+        )
+
+        # 4) Done - include welcome job ID in payload
         structlogger.info(
             "bot_builder_service.template_to_bot.success",
             files_generated=list(bot_files.keys()),
+            copilot_welcome_job_id=copilot_welcome_job.id,
+        )
+        await push_job_status_event(
+            job=job,
+            status=JobStatus.done,
+            payload={"copilot_welcome_job_id": copilot_welcome_job.id},
         )
-        await push_job_status_event(job, JobStatus.done)
         job_manager.mark_done(job)
 
     except TrainingError as exc:
@@ -256,8 +283,19 @@ async def run_replace_all_files_job(
         update_agent(agent, app)
         await push_job_status_event(job, JobStatus.train_success)
 
-        # Send final done event
-        await push_job_status_event(job, JobStatus.done)
+        # Send final done event with copilot training success response job ID
+        copilot_training_success_job = job_manager.create_job()
+        app.add_task(
+            run_copilot_training_success_job(app, copilot_training_success_job)
+        )
+
+        await push_job_status_event(
+            job=job,
+            status=JobStatus.done,
+            payload={
+                "copilot_training_success_job_id": copilot_training_success_job.id
+            },
+        )
         job_manager.mark_done(job)
 
     except ValidationError as exc:
@@ -450,3 +488,122 @@ async def run_copilot_training_error_analysis_job(
             job, JobStatus.copilot_analysis_error, message=str(exc)
         )
         job_manager.mark_done(job, error=str(exc))
+
+
+async def run_copilot_welcome_message_job(
+    app: "Sanic",
+    job: JobInfo,
+    template_name: Optional[ProjectTemplateName] = None,
+) -> None:
+    """Run the welcome message job in the background.
+
+    This job sends a welcome message to the user after successful bot creation.
+    For template-based bots, it sends a predefined message.
+    For prompt-based bots, it can be extended to stream generated messages.
+
+    Args:
+        app: The Sanic application instance.
+        job: The job information instance.
+        template_name: The template name for template-based bots, None for prompt-based.
+    """
+    try:
+        # Load welcome messages from YAML
+        welcome_messages = load_copilot_welcome_messages()
+
+        # Get the appropriate welcome message
+        if template_name:
+            welcome_message = welcome_messages.get(
+                template_name.value,
+                welcome_messages.get(PROMPT_TO_BOT_KEY),
+            )
+        else:
+            welcome_message = welcome_messages.get(PROMPT_TO_BOT_KEY)
+
+        # Send the welcome message as a single event
+        await push_job_status_event(
+            job,
+            JobStatus.copilot_welcome_message,
+            payload={
+                "content": welcome_message,
+                "response_category": "copilot",
+                "completeness": "complete",
+            },
+        )
+
+        # Mark job as done
+        await push_job_status_event(job, JobStatus.done)
+        job_manager.mark_done(job)
+
+        structlogger.info(
+            "copilot_welcome_message_job.success",
+            job_id=job.id,
+            template=template_name.value if template_name else PROMPT_TO_BOT_KEY,
+        )
+
+    except Exception as exc:
+        structlogger.exception(
+            "welcome_message_job.error",
+            job_id=job.id,
+            error=str(exc),
+        )
+        await push_job_status_event(job, JobStatus.error, message=str(exc))
+        job_manager.mark_done(job, error=str(exc))
+
+
+async def run_copilot_training_success_job(
+    app: "Sanic",
+    job: JobInfo,
+) -> None:
+    """Run the training success job in the background.
+
+    This job sends a training success message to the user after successful bot training.
+
+    Args:
+        app: The Sanic application instance.
+        job: The job information instance.
+    """
+    try:
+        # Load copilot default messages from YAML
+        internal_messages = load_copilot_handler_default_responses()
+
+        # Get the appropriate training success message
+        training_success_message = internal_messages.get("training_success_response")
+
+        # Send the training success message
+        await push_job_status_event(
+            job,
+            JobStatus.train_success_message,
+            payload={
+                "content": training_success_message,
+                "response_category": "copilot",
+                "completeness": "complete",
+            },
+        )
+
+        # Send the training success category
+        await push_job_status_event(
+            job,
+            JobStatus.train_success_message,
+            payload={
+                "response_category": "copilot_training_success",
+                "completeness": "complete",
+            },
+        )
+
+        # Mark job as done
+        await push_job_status_event(job, JobStatus.done)
+        job_manager.mark_done(job)
+
+        structlogger.info(
+            "copilot_training_success_job.success",
+            job_id=job.id,
+        )
+
+    except Exception as exc:
+        structlogger.exception(
+            "copilot_training_success_job.error",
+            job_id=job.id,
+            error=str(exc),
+        )
+        await push_job_status_event(job, JobStatus.error, message=str(exc))
+        job_manager.mark_done(job, error=str(exc))

rasa/builder/models.py

@@ -191,6 +191,7 @@ class JobStatus(str, Enum):
 
     training = "training"
     train_success = "train_success"
+    train_success_message = "train_success_message"
     train_error = "train_error"
 
     validating = "validating"
@@ -202,6 +203,8 @@ class JobStatus(str, Enum):
     copilot_analysis_success = "copilot_analysis_success"
     copilot_analysis_error = "copilot_analysis_error"
 
+    copilot_welcome_message = "copilot_welcome_message"
+
 
 class JobCreateResponse(BaseModel):
     job_id: str = Field(...)

rasa/builder/validation_service.py

@@ -47,6 +47,10 @@ async def validate_project(importer: TrainingDataImporter) -> Optional[str]:
     with capture_validation_logs() as captured_logs:
         try:
             with _mock_sys_exit() as exit_tracker:
+                from rasa.core.config.configuration import Configuration
+
+                Configuration.initialise_empty()
+
                 validate_files(
                     fail_on_warnings=config.VALIDATION_FAIL_ON_WARNINGS,
                     max_history=config.VALIDATION_MAX_HISTORY,

rasa/cli/arguments/data.py

@@ -4,8 +4,10 @@ from typing import Text
 from rasa.cli.arguments.default_arguments import (
     add_data_param,
     add_domain_param,
+    add_endpoint_param,
     add_nlu_data_param,
     add_out_param,
+    add_sub_agents_param,
 )
 from rasa.shared.constants import DEFAULT_CONVERTED_DATA_PATH
 
@@ -91,6 +93,13 @@ def set_validator_arguments(parser: argparse.ArgumentParser) -> None:
     )
     add_domain_param(parser)
     add_data_param(parser)
+    add_sub_agents_param(parser)
+    # Endpoints are optional for `data validate` command
+    add_endpoint_param(
+        parser,
+        help_text="Configuration file for the connectors as a yml file.",
+        default=None,
+    )
 
 
 def set_migrate_arguments(parser: argparse.ArgumentParser) -> None:

rasa/cli/data.py

@@ -1,7 +1,7 @@
 import argparse
 import logging
 import pathlib
-from typing import List
+from typing import List, Optional
 
 import rasa.cli.utils
 import rasa.shared.core.domain
@@ -17,6 +17,7 @@ from rasa.cli.arguments import data as arguments
 from rasa.cli.arguments import default_arguments
 from rasa.cli.validation.bot_config import validate_files
 from rasa.cli.validation.config_path_validation import get_validated_path
+from rasa.core.config.configuration import Configuration
 from rasa.e2e_test.e2e_config import create_llm_e2e_test_converter_config
 from rasa.e2e_test.e2e_test_converter import E2ETestConverter
 from rasa.e2e_test.utils.e2e_yaml_utils import E2ETestYAMLWriter
@@ -139,8 +140,12 @@ def _add_data_validate_parsers(
     )
     _append_story_structure_arguments(validate_parser)
     validate_parser.set_defaults(
-        func=lambda args: validate_files(
-            args.fail_on_warnings, args.max_history, _build_training_data_importer(args)
+        func=lambda args: _validate_files_with_configuration(
+            args.fail_on_warnings,
+            args.max_history,
+            _build_training_data_importer(args),
+            sub_agents=args.sub_agents,
+            endpoints=args.endpoints,
         )
     )
     arguments.set_validator_arguments(validate_parser)
@@ -155,10 +160,12 @@ def _add_data_validate_parsers(
     _append_story_structure_arguments(story_structure_parser)
 
     story_structure_parser.set_defaults(
-        func=lambda args: validate_files(
+        func=lambda args: _validate_files_with_configuration(
             args.fail_on_warnings,
             args.max_history,
             _build_training_data_importer(args),
+            sub_agents=args.sub_agents,
+            endpoints=args.endpoints,
             stories_only=True,
         )
     )
@@ -171,10 +178,12 @@ def _add_data_validate_parsers(
         help="Checks for inconsistencies in the flows files.",
     )
     flows_structure_parser.set_defaults(
-        func=lambda args: validate_files(
+        func=lambda args: _validate_files_with_configuration(
             args.fail_on_warnings,
             args.max_history,
             _build_training_data_importer(args),
+            sub_agents=args.sub_agents,
+            endpoints=args.endpoints,
             flows_only=True,
         )
     )
@@ -187,10 +196,12 @@ def _add_data_validate_parsers(
         help="Checks for inconsistencies of the flow and response translation.",
     )
     translations_structure_parser.set_defaults(
-        func=lambda args: validate_files(
+        func=lambda args: _validate_files_with_configuration(
             args.fail_on_warnings,
             args.max_history,
             _build_training_data_importer(args),
+            sub_agents=args.sub_agents,
+            endpoints=args.endpoints,
             translations_only=True,
         )
     )
@@ -376,3 +387,58 @@ def convert_data_to_e2e_tests(args: argparse.Namespace) -> None:
         rasa.shared.utils.cli.print_error_and_exit(
             f"Failed to convert the data into E2E tests. Error: {exc}"
         )
+
+
+def _initialize_configuration_for_validation(
+    sub_agents: Optional[str] = None, endpoints: Optional[str] = None
+) -> None:
+    """Initialize Configuration before validation.
+
+    Args:
+        sub_agents: Path to sub-agents directory for validation.
+        endpoints: Path to the endpoints configuration file.
+    """
+    if endpoints:
+        Configuration.initialise_endpoints(endpoints_path=pathlib.Path(endpoints))
+    if sub_agents:
+        Configuration.initialise_sub_agents(pathlib.Path(sub_agents))
+    if not endpoints and not sub_agents:
+        Configuration.initialise_empty()
+
+
+def _validate_files_with_configuration(
+    fail_on_warnings: bool,
+    max_history: int,
+    importer: TrainingDataImporter,
+    sub_agents: Optional[str] = None,
+    endpoints: Optional[str] = None,
+    stories_only: bool = False,
+    flows_only: bool = False,
+    translations_only: bool = False,
+) -> None:
+    """Wrapper for validate_files that ensures Configuration is initialized first.
+
+    Args:
+        fail_on_warnings: `True` if the process should exit with a non-zero status
+        max_history: The max history to use when validating the story structure.
+        importer: The `TrainingDataImporter` to use to load the training data.
+        sub_agents: Path to sub-agents directory for validation.
+        endpoints: Path to the endpoints configuration file.
+        stories_only: If `True`, only the story structure is validated.
+        flows_only: If `True`, only the flows are validated.
+        translations_only: If `True`, only the translations data is validated.
+    """
+    # Initialize Configuration before calling validate_files
+    _initialize_configuration_for_validation(sub_agents=sub_agents, endpoints=endpoints)
+
+    # Call the original validate_files function
+    validate_files(
+        fail_on_warnings=fail_on_warnings,
+        max_history=max_history,
+        importer=importer,
+        stories_only=stories_only,
+        flows_only=flows_only,
+        translations_only=translations_only,
+        sub_agents=sub_agents,
+        endpoints=endpoints,
+    )

rasa/cli/interactive.py

@@ -12,6 +12,7 @@ from rasa import model
 from rasa.cli import SubParsersAction
 from rasa.cli.arguments import interactive as arguments
 from rasa.cli.validation.config_path_validation import get_validated_path
+from rasa.core.config.configuration import Configuration
 from rasa.core.constants import DEFAULT_SUB_AGENTS
 from rasa.engine.storage.local_model_storage import LocalModelStorage
 from rasa.shared.constants import (
@@ -140,6 +141,8 @@ def perform_interactive_learning(
         args.endpoints, "endpoints", DEFAULT_ENDPOINTS_PATH, True
     )
 
+    Configuration.initialise_endpoints(endpoints_path=Path(args.endpoints))
+
     do_interactive_learning(args, file_importer)
 
 

rasa/cli/llm_fine_tuning.py

@@ -414,6 +414,7 @@ def get_valid_endpoints(endpoints_file: str) -> AvailableEndpoints:
 
 def set_up_e2e_test_runner(args: argparse.Namespace) -> E2ETestRunner:
     endpoints = get_valid_endpoints(args.endpoints)
+    Configuration.initialise_sub_agents(args.sub_agents)
 
     if endpoints.model is None:
         args.model = validate_model_path(args.model, "model", DEFAULT_MODELS_PATH)

rasa/cli/project_templates/defaults.py

@@ -138,6 +138,7 @@ def get_rasa_defaults(config_yaml: Text, endpoints_yaml: Text) -> RasaDefaults:
         A RasaDefaults object containing the default values for the project.
     """
     config = read_yaml(config_yaml)
+    # Does not read from file, it converts the YAML content to a dictionary.
     endpoints = read_yaml(endpoints_yaml)
 
     prompts = get_system_default_prompts(config, endpoints)

rasa/cli/validation/bot_config.py

@@ -149,6 +149,7 @@ def validate_files(
     flows_only: bool = False,
     translations_only: bool = False,
     sub_agents: Optional[str] = None,
+    endpoints: Optional[str] = None,
 ) -> None:
     """Validates either the story structure or the entire project.
 
@@ -160,6 +161,7 @@ def validate_files(
         flows_only: If `True`, only the flows are validated.
         translations_only: If `True`, only the translations data is validated.
         sub_agents: Path to sub-agents directory for validation.
+        endpoints: Path to the endpoints configuration file.
     """
     from rasa.validator import Validator
 
@@ -189,6 +191,7 @@ def validate_files(
             validator, max_history, fail_on_warnings
         )
         valid_flows = validator.verify_flows()
+        valid_prompt_templates = validator.verify_prompt_templates()
         if validator.config:
             valid_translations = validator.verify_translations(summary_mode=True)
         else:
@@ -207,6 +210,7 @@ def validate_files(
             and valid_nlu
             and valid_stories
             and valid_flows
+            and valid_prompt_templates
             and valid_translations
             and valid_CALM_slot_mappings
             and valid_sub_agents

rasa/constants.py

@@ -18,7 +18,7 @@ CONFIG_TELEMETRY_ID = "rasa_user_id"
 CONFIG_TELEMETRY_ENABLED = "enabled"
 CONFIG_TELEMETRY_DATE = "date"
 
-MINIMUM_COMPATIBLE_VERSION = "3.14.0.dev10"
+MINIMUM_COMPATIBLE_VERSION = "3.14.0rc3"
 
 GLOBAL_USER_CONFIG_PATH = os.path.expanduser("~/.config/rasa/global.yml")
 

rasa/core/actions/action_exceptions.py

@@ -12,7 +12,7 @@ class ActionExecutionRejection(RasaException):
         self.message = message or "Custom action '{}' rejected to run".format(
             action_name
         )
-        super(ActionExecutionRejection, self).__init__()
+        super(ActionExecutionRejection, self).__init__(self.message)
 
     def __str__(self) -> str:
         return self.message

rasa/core/agent.py

@@ -297,7 +297,10 @@ async def load_agent(
         return agent
 
     except AgentInitializationException as e:
-        raise e
+        if e.suppress_stack_trace:
+            raise e from None
+        else:
+            raise e
     except Exception as e:
         logger.error(f"Could not load model due to {e}.", exc_info=True)
         return agent

rasa/core/available_agents.py

@@ -114,7 +114,7 @@ class AvailableAgents:
             else:
                 # We are using the default folder, it may not be created yet
                 # Init with an empty agents in this case
-                structlogger.info(
+                structlogger.debug(
                     f"Default agents config folder '{sub_agents_folder}' does not "
                     f"exist. Agent configurations won't be loaded."
                 )

rasa/core/config/available_endpoints.py

@@ -7,6 +7,7 @@ from typing import Any, Dict, List, Optional, Union
 from pydantic import BaseModel, ConfigDict, Field, model_validator
 
 from rasa.core.constants import MCP_SERVERS_KEY
+from rasa.shared.agents.auth.utils import validate_secrets_in_params
 from rasa.shared.core.constants import (
     GLOBAL_SILENCE_TIMEOUT_DEFAULT_VALUE,
     GLOBAL_SILENCE_TIMEOUT_KEY,
@@ -77,6 +78,11 @@ class MCPServerConfig(BaseModel):
         # validate that name and url are not empty
         if not self.name or not self.url:
             raise ValueError("Name and URL cannot be empty")
+        # validate secrets in additional_params
+        if self.additional_params:
+            validate_secrets_in_params(
+                self.additional_params, f"MCP server - '{self.name}'"
+            )
         return self
 
     @model_validator(mode="before")

rasa/core/exceptions.py

@@ -14,7 +14,7 @@ class AgentNotReady(RasaCoreException):
     def __init__(self, message: Text) -> None:
         """Initialize message attribute."""
         self.message = message
-        super(AgentNotReady, self).__init__()
+        super(AgentNotReady, self).__init__(message)
 
 
 class ChannelConfigError(RasaCoreException):

rasa/core/featurizers/tracker_featurizers.py

@@ -58,7 +58,7 @@ class InvalidStory(RasaException):
           message: a custom exception message.
         """
         self.message = message
-        super(InvalidStory, self).__init__()
+        super(InvalidStory, self).__init__(message)
 
     def __str__(self) -> Text:
         return self.message
@@ -174,6 +174,7 @@ class TrackerFeaturizer:
 
         Args:
             trackers_as_actions: A list of tracker labels.
+            domain: The domain containing action names.
 
         Returns:
             Label IDs for each tracker
@@ -854,7 +855,7 @@ class MaxHistoryTrackerFeaturizer(TrackerFeaturizer):
         """Creates an iterator over training examples from a tracker.
 
         Args:
-            trackers: The tracker from which to extract training examples.
+            tracker: The tracker from which to extract training examples.
             domain: The domain of the training data.
             omit_unset_slots: If `True` do not include the initial values of slots.
             ignore_action_unlikely_intent: Whether to remove `action_unlikely_intent`

rasa/core/persistor.py

@@ -314,7 +314,7 @@ class AWSPersistor(Persistor):
             obj = self.s3.Object(self.bucket_name, model_path)
             return obj.content_length
         except Exception:
-            raise ModelNotFound()
+            raise ModelNotFound("Model not found")
 
     def _retrieve_tar(
         self, target_filename: str, target_path: Optional[str] = None
@@ -349,7 +349,7 @@ class AWSPersistor(Persistor):
                     target_filename=target_filename,
                     event_info=log,
                 )
-                raise ModelNotFound() from exc
+                raise ModelNotFound("Model not found") from exc
         except exceptions.BotoCoreError as exc:
             structlogger.error(
                 "aws_persistor.retrieve_tar.model_download_error",
@@ -357,7 +357,7 @@ class AWSPersistor(Persistor):
                 target_filename=target_filename,
                 event_info=log,
             )
-            raise ModelNotFound() from exc
+            raise ModelNotFound("Model not found") from exc
 
 
 class GCSPersistor(Persistor):
@@ -447,7 +447,7 @@ class GCSPersistor(Persistor):
             blob = self.bucket.blob(target_filename)
             return blob.size
         except Exception:
-            raise ModelNotFound()
+            raise ModelNotFound("Model not found")
 
     def _retrieve_tar(
         self, target_filename: str, target_path: Optional[str] = None
@@ -481,7 +481,7 @@ class GCSPersistor(Persistor):
                 target_filename=target_filename,
                 event_info=log,
             )
-            raise ModelNotFound() from exc
+            raise ModelNotFound("Model not found") from exc
 
 
 class AzurePersistor(Persistor):
@@ -534,7 +534,7 @@ class AzurePersistor(Persistor):
             properties = blob_client.get_blob_properties()
             return properties.size
         except Exception:
-            raise ModelNotFound()
+            raise ModelNotFound("Model not found")
 
     def _retrieve_tar(
         self, target_filename: Text, target_path: Optional[str] = None
@@ -570,4 +570,4 @@ class AzurePersistor(Persistor):
                 event_info=log,
                 exception=exc,
             )
-            raise ModelNotFound() from exc
+            raise ModelNotFound("Model not found") from exc