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

rasa/agents/agent_manager.py

@@ -15,7 +15,7 @@ structlogger = structlog.get_logger()
 
 
 class AgentManager(metaclass=Singleton):
-    """High-level agent management with protocol abstraction"""
+    """High-level agent management with protocol abstraction."""
 
     agents: ClassVar[Dict[AgentIdentifier, AgentProtocol]] = {}
 
@@ -66,9 +66,11 @@ class AgentManager(metaclass=Singleton):
     async def connect_agent(
         self, agent_name: str, protocol_type: ProtocolType, config: AgentConfig
     ) -> None:
-        """Connect to agent using specified protocol, load the default resources
-        and persist the agent to the manager in a ready-to-use state so that
-        it can be used immediately to send messages to the agent.
+        """Connect to agent using specified protocol.
+
+        Also, load the default resources and persist the agent to the manager
+        in a ready-to-use state so that it can be used immediately
+        to send messages to the agent.
 
         Args:
             agent_name: The name of the agent.
@@ -109,7 +111,7 @@ class AgentManager(metaclass=Singleton):
                 agent_id=str(agent_identifier),
                 event_info=event_info,
             )
-            raise AgentInitializationException(e) from e
+            raise AgentInitializationException(e, suppress_stack_trace=True) from e
 
     async def run_agent(
         self, agent_name: str, protocol_type: ProtocolType, context: AgentInput

rasa/agents/protocol/a2a/a2a_agent.py

@@ -65,7 +65,7 @@ structlogger = structlog.get_logger()
 
 
 class A2AAgent(AgentProtocol):
-    """A2A client implementation"""
+    """A2A client implementation."""
 
     __SUPPORTED_OUTPUT_MODES: ClassVar[list[str]] = [
         "text",
@@ -169,7 +169,8 @@ class A2AAgent(AgentProtocol):
                 error=str(exception),
             )
             raise AgentInitializationException(
-                f"Failed to initialize A2A client for agent '{self._name}': {exception}"
+                f"Failed to initialize A2A client "
+                f"for agent '{self._name}': {exception}",
             ) from exception
 
         await self._perform_health_check()
@@ -180,7 +181,7 @@ class A2AAgent(AgentProtocol):
         )
 
     async def disconnect(self) -> None:
-        """We don't need to explicitly disconnect the A2A client"""
+        """We don't need to explicitly disconnect the A2A client."""
         return
 
     # ============================================================================
@@ -297,7 +298,7 @@ class A2AAgent(AgentProtocol):
     def _handle_send_message_response(
         self, agent_input: AgentInput, response: ClientEvent | Message
     ) -> Optional[AgentOutput]:
-        """Handle possible response types from the A2A client:
+        """Handle possible response types from the A2A client.
 
         In case of streaming, the response can be either exactly *one* Message,
         or a *series* of tuples of (Task, Optional[TaskUpdateEvent]).
@@ -410,8 +411,8 @@ class A2AAgent(AgentProtocol):
         agent_input: AgentInput,
         task: Task,
     ) -> Optional[AgentOutput]:
-        """If the task status is terminal (i.e., completed, failed, etc.),
-        return an AgentOutput.
+        """If task status is terminal (e.g. completed, failed) return AgentOutput.
+
         If the task is still in progress (i.e., submitted, working), return None,
         so that the streaming or pooling agent can continue to wait for updates.
         """
@@ -655,6 +656,7 @@ class A2AAgent(AgentProtocol):
     @staticmethod
     def _generate_completed_response_message(task: Task) -> str:
         """Generate a response message for a completed task.
+
         In case of completed tasks, the final message might be in
         the task status message or in the artifacts (or both).
         """
@@ -728,19 +730,19 @@ class A2AAgent(AgentProtocol):
 
         except FileNotFoundError as e:
             raise AgentInitializationException(
-                f"Agent card file not found: {agent_card_path}"
+                f"Agent card file not found: {agent_card_path}",
             ) from e
         except (IOError, PermissionError) as e:
             raise AgentInitializationException(
-                f"Error reading agent card file {agent_card_path}: {e}"
+                f"Error reading agent card file {agent_card_path}: {e}",
             ) from e
         except json.JSONDecodeError as e:
             raise AgentInitializationException(
-                f"Invalid JSON in agent card file {agent_card_path}: {e}"
+                f"Invalid JSON in agent card file {agent_card_path}: {e}",
             ) from e
         except ValidationError as e:
             raise AgentInitializationException(
-                f"Failed to load agent card from {agent_card_path}: {e}"
+                f"Failed to load agent card from {agent_card_path}: {e}",
             ) from e
 
     @staticmethod
@@ -798,7 +800,7 @@ class A2AAgent(AgentProtocol):
 
         raise AgentInitializationException(
             f"Failed to resolve agent card from {agent_card_path} after "
-            f"{max_retries} attempts."
+            f"{max_retries} attempts.",
         )
 
     # ============================================================================

rasa/agents/protocol/mcp/mcp_base_agent.py

@@ -1,9 +1,10 @@
 import json
 from abc import abstractmethod
-from datetime import datetime
+from datetime import datetime, timedelta
 from inspect import isawaitable
 from typing import Any, Dict, List, Optional, Tuple
 
+import anyio
 import structlog
 from jinja2 import Template
 from mcp import ListToolsResult
@@ -71,10 +72,12 @@ structlogger = structlog.get_logger()
 
 
 class MCPBaseAgent(AgentProtocol):
-    """MCP protocol implementation"""
+    """MCP protocol implementation."""
 
     MAX_ITERATIONS = 10
 
+    TOOL_CALL_DEFAULT_TIMEOUT = 10  # seconds
+
     # ============================================================================
     # Initialization & Setup
     # ============================================================================
@@ -135,6 +138,20 @@ class MCPBaseAgent(AgentProtocol):
     @classmethod
     def from_config(cls, config: AgentConfig) -> "MCPBaseAgent":
         """Initialize the MCP Open Agent with the given configuration."""
+        # Warn if configuration.timeout is set for MCP agents
+        if config.configuration and config.configuration.timeout is not None:
+            structlogger.warning(
+                "mcp_agent.configuration.timeout.not_implemented",
+                event_info="configuration.timeout is not implemented for MCP agents. "
+                "MCP agents do not establish external connections, "
+                "so agent-level timeout is not used. "
+                "To set timeout for LLM requests, "
+                "configure 'timeout' in the model_group "
+                "in endpoints.yml and reference it via configuration.llm.model_group.",
+                agent_name=config.agent.name,
+                timeout_value=config.configuration.timeout,
+            )
+
         return cls(
             name=config.agent.name,
             description=config.agent.description,
@@ -274,14 +291,16 @@ class MCPBaseAgent(AgentProtocol):
                         event_info="All connection attempts failed.",
                     )
                     raise AgentInitializationException(
-                        f"Failed to connect to MCP servers after {self._max_retries} "
-                        f"attempts. Agent `{self._name}` failed to initialize."
+                        f"Agent `{self._name}` failed to initialize. Failed to connect "
+                        f"to MCP servers after {self._max_retries} attempts. {ce!s}"
                     ) from ce
             except (Exception, AuthenticationError) as e:
                 if isinstance(e, AuthenticationError):
-                    event_info = "Authentication error during agent initialization."
+                    event_info = (
+                        f"Authentication error during agent initialization. {e!s}"
+                    )
                 else:
-                    event_info = "Unexpected error during agent initialization."
+                    event_info = f"Unexpected error during agent initialization. {e!s}"
                 structlogger.error(
                     "mcp_agent.connect.unexpected_exception",
                     event_info=event_info,
@@ -289,7 +308,7 @@ class MCPBaseAgent(AgentProtocol):
                     agent_name=self._name,
                     agent_id=str(make_agent_identifier(self._name, self.protocol_type)),
                 )
-                raise AgentInitializationException(e) from e
+                raise AgentInitializationException(event_info) from e
 
     async def connect_to_server(self, server_config: AgentMCPServerConfig) -> None:
         server_name = server_config.name
@@ -311,7 +330,7 @@ class MCPBaseAgent(AgentProtocol):
         except Exception as e:
             event_info = (
                 f"Agent `{self._name}` failed to connect to MCP server - "
-                f"`{server_name}` @ `{server_config.url}`: {e!s}"
+                f"`{server_name}` @ `{server_config.url}`"
             )
             structlogger.error(
                 "mcp_agent.connect.failed_to_connect",
@@ -321,7 +340,9 @@ class MCPBaseAgent(AgentProtocol):
                 agent_name=self._name,
                 agent_id=str(make_agent_identifier(self._name, self.protocol_type)),
             )
-            raise e
+
+            # Wrap exceptions with extra info and raise the same type of exception.
+            raise type(e)(f"{event_info} : {e!s}") from e
 
     async def connect_to_servers(self) -> None:
         """Connect to MCP servers."""
@@ -610,7 +631,11 @@ class MCPBaseAgent(AgentProtocol):
         connection = self._server_connections[server_id]
         try:
             session = await connection.ensure_active_session()
-            result = await session.call_tool(tool_name, arguments)
+            result = await session.call_tool(
+                tool_name,
+                arguments,
+                read_timeout_seconds=timedelta(seconds=self.TOOL_CALL_DEFAULT_TIMEOUT),
+            )
             return AgentToolResult.from_mcp_tool_result(tool_name, result)
         except Exception as e:
             return AgentToolResult(
@@ -623,6 +648,21 @@ class MCPBaseAgent(AgentProtocol):
                 ),
             )
 
+    async def _run_custom_tool(
+        self, custom_tool: CustomToolSchema, arguments: Dict[str, Any]
+    ) -> AgentToolResult:
+        """Run a custom tool and return the result.
+
+        Args:
+            custom_tool: The custom tool schema containing the tool executor.
+            arguments: The arguments to pass to the tool executor.
+
+        Returns:
+            The result of the tool execution as an AgentToolResult.
+        """
+        result = custom_tool.tool_executor(arguments)
+        return await result if isawaitable(result) else result
+
     async def _execute_tool_call(
         self, tool_name: str, arguments: Dict[str, Any]
     ) -> AgentToolResult:
@@ -641,8 +681,20 @@ class MCPBaseAgent(AgentProtocol):
         try:
             for custom_tool in self._custom_tools:
                 if custom_tool.tool_name == tool_name:
-                    result = custom_tool.tool_executor(arguments)
-                    return await result if isawaitable(result) else result
+                    try:
+                        with anyio.fail_after(self.TOOL_CALL_DEFAULT_TIMEOUT):
+                            return await self._run_custom_tool(custom_tool, arguments)
+
+                    except TimeoutError:
+                        return AgentToolResult(
+                            tool_name=tool_name,
+                            result=None,
+                            is_error=True,
+                            error_message=(
+                                f"Built-in tool `{tool_name}` timed out after "
+                                f"{self.TOOL_CALL_DEFAULT_TIMEOUT} seconds."
+                            ),
+                        )
         except Exception as e:
             return AgentToolResult(
                 tool_name=tool_name,

rasa/agents/validation.py

@@ -5,6 +5,8 @@ import urllib.parse
 from collections import Counter
 from typing import Any, Dict, List, NoReturn, Set
 
+import jinja2.exceptions
+import structlog
 from pydantic import ValidationError as PydanticValidationError
 
 from rasa.agents.exceptions import (
@@ -23,8 +25,13 @@ from rasa.core.available_agents import (
 from rasa.core.config.available_endpoints import AvailableEndpoints
 from rasa.core.config.configuration import Configuration
 from rasa.exceptions import ValidationError
+from rasa.shared.agents.auth.utils import validate_secrets_in_params
+from rasa.shared.utils.llm import get_prompt_template, validate_jinja2_template
 from rasa.shared.utils.yaml import read_config_file
 
+# Initialize logger
+structlogger = structlog.get_logger()
+
 # Centralized allowed keys configuration to eliminate duplication
 ALLOWED_KEYS = {
     "agent": {"name", "protocol", "description"},
@@ -151,6 +158,41 @@ def _validate_a2a_config(agent_config: AgentConfig) -> None:
                 f"exist: {agent_card}",
             )
 
+    # Validate right use of secrets in auth configuration
+    if agent_config.configuration.auth:
+        validate_secrets_in_params(
+            agent_config.configuration.auth, f"agent '{agent_name}'"
+        )
+
+
+def _validate_prompt_template_syntax(prompt_path: str, agent_name: str) -> None:
+    """Validate Jinja2 syntax of a prompt template file."""
+    try:
+        # Use a simple default template, as we're assuming
+        # that the default templates are valid
+        default_template = "{{ content }}"
+        template_content = get_prompt_template(
+            prompt_path,
+            default_template,
+            log_source_component=f"agent.validation.{agent_name}",
+            log_source_method="init",
+        )
+        validate_jinja2_template(template_content)
+
+    except jinja2.exceptions.TemplateSyntaxError as e:
+        raise ValidationError(
+            code="agent.validation.prompt_template_syntax_error",
+            event_info=(
+                f"Agent '{agent_name}' has invalid Jinja2 template syntax at line "
+                f"{e.lineno}: {e.message}"
+            ),
+        ) from e
+    except Exception as e:
+        raise ValidationError(
+            code="agent.validation.optional.prompt_template_error",
+            event_info=(f"Agent '{agent_name}' has error reading prompt template: {e}"),
+        ) from e
+
 
 def _validate_optional_keys(agent_config: AgentConfig) -> None:
     """Validate optional keys in agent configuration."""
@@ -160,11 +202,22 @@ def _validate_optional_keys(agent_config: AgentConfig) -> None:
     if agent_config.configuration and agent_config.configuration.prompt_template:
         prompt_path = agent_config.configuration.prompt_template
         if not os.path.exists(prompt_path):
-            raise ValidationError(
-                code="agent.validation.optional.prompt_template_not_found",
-                event_info=f"Agent '{agent_name}' has prompt template that "
-                f"does not exist: {prompt_path}",
+            # If reading the custom prompt fails,
+            # allow fallback to default prompt template
+            structlogger.warning(
+                "agent.validation.optional.prompt_template_file_not_found",
+                agent_name=agent_name,
+                prompt_path=prompt_path,
+                event_info=(
+                    f"Prompt template file not found: {prompt_path}. "
+                    f"Agent will use default template."
+                ),
             )
+            # Don't raise ValidationError, allow fallback to default template
+            return
+
+        # Validate Jinja2 syntax
+        _validate_prompt_template_syntax(prompt_path, agent_name)
 
     # Validate module if present
     if agent_config.configuration and agent_config.configuration.module:
@@ -294,9 +347,11 @@ def _handle_pydantic_validation_error(
 def _validate_endpoint_references(agent_config: AgentConfig) -> None:
     """Validate that LLM and MCP server references in agent config are valid."""
     agent_name = agent_config.agent.name
-
-    # Get available endpoints
     endpoints = Configuration.get_instance().endpoints
+    if not endpoints.config_file_path:
+        # If no endpoints were loaded (e.g., `data validate` without --endpoints), skip
+        # endpoint reference checks
+        return
 
     # Validate LLM configuration references
     if agent_config.configuration and agent_config.configuration.llm:

rasa/builder/copilot/constants.py

@@ -14,6 +14,7 @@ COPILOT_TRAINING_ERROR_HANDLER_PROMPT_FILE = (
 COPILOT_MESSAGE_TEMPLATES_DIR = "builder.copilot.templated_messages"
 RASA_INTERNAL_MESSAGES_TEMPLATES_FILE = "copilot_internal_messages_templates.yml"
 COPILOT_HANDLER_RESPONSES_FILE = "copilot_templated_responses.yml"
+COPILOT_WELCOME_MESSAGES_FILE = "copilot_welcome_messages.yml"
 
 # OpenAI roles copilot utilizes - Use literal types to avoid type errors with OpenAI
 ROLE_USER: Literal["user"] = "user"
@@ -32,3 +33,6 @@ COPILOT_SEGMENT_WRITE_KEY_ENV_VAR = "COPILOT_SEGMENT_WRITE_KEY"
 
 # Copilot signing
 SIGNATURE_VERSION_V1 = "v1"
+
+# Welcome message key for prompt-to-bot (not a template enum)
+PROMPT_TO_BOT_KEY = "prompt_to_bot"

rasa/builder/copilot/copilot_templated_message_provider.py

@@ -2,11 +2,12 @@ import importlib.resources
 from typing import Dict
 
 import structlog
-import yaml  # type: ignore
+import yaml  # type: ignore[import-untyped]
 
 from rasa.builder.copilot.constants import (
     COPILOT_HANDLER_RESPONSES_FILE,
     COPILOT_MESSAGE_TEMPLATES_DIR,
+    COPILOT_WELCOME_MESSAGES_FILE,
     RASA_INTERNAL_MESSAGES_TEMPLATES_FILE,
 )
 from rasa.shared.constants import PACKAGE_NAME
@@ -56,3 +57,25 @@ def load_copilot_handler_default_responses() -> Dict[str, str]:
             error=e,
         )
         return dict()
+
+
+def load_copilot_welcome_messages() -> Dict[str, str]:
+    """Load welcome message templates from the YAML configuration file.
+
+    Returns:
+        Dictionary mapping template names to welcome message text.
+    """
+    try:
+        config = yaml.safe_load(
+            importlib.resources.read_text(
+                f"{PACKAGE_NAME}.{COPILOT_MESSAGE_TEMPLATES_DIR}",
+                COPILOT_WELCOME_MESSAGES_FILE,
+            )
+        )
+        return config.get("welcome_messages", {})
+    except Exception as e:
+        structlogger.error(
+            "copilot_templated_message_provider.failed_to_load_welcome_messages",
+            error=e,
+        )
+        return dict()

rasa/builder/copilot/templated_messages/copilot_templated_responses.yml

@@ -36,3 +36,6 @@ responses:
   guardrail_blocked_project_response: |
     Your Copilot access has been temporarily blocked due to repeated policy violations.
     If you believe this is an error or would like to request a review, please reach out to support@rasa.com.
+
+  training_success_response: |
+    Your changes have been saved successfully.

rasa/builder/copilot/templated_messages/copilot_welcome_messages.yml

@@ -0,0 +1,56 @@
+welcome_messages:
+  copilot_intro: |
+    👋 Welcome to Hello Rasa!
+
+    I'm your **Copilot** — here to help you explore Rasa and start customizing your agent.
+    You can ask me how your agent works, how to add new skills, or how to connect integrations.
+
+  finance: |
+    👋 Welcome to Hello Rasa!
+
+    I'm your **Copilot** — here to help you explore Rasa and start customizing your agent.
+    You can ask me how your agent works, how to add new skills, or how to connect integrations.
+
+    Your **Banking Agent template** is now live.
+
+    ### ▶️ **First step: try it out**
+    Ask your agent (in the chat preview on the right):
+    - *What's my current balance?*
+    - *Send 100 dollars to James.*
+
+  telco: |
+    👋 Welcome to Hello Rasa!
+
+    I'm your **Copilot** — here to help you explore Rasa and start customizing your agent.
+    You can ask me how your agent works, how to add new skills, or how to connect integrations.
+
+    Your **Telecom Support Agent template** is now live.
+
+    ### ▶️ **First step: try it out**
+    Ask your agent (in the chat preview on the right):
+    - *Why is my internet slow?*
+    - *How do i reboot my router?*
+
+  basic: |
+    👋 Welcome to Hello Rasa!
+
+    I'm your **Copilot** — here to help you explore Rasa and start customizing your agent.
+    You can ask me how your agent works, how to add new skills, or how to connect integrations.
+
+    Your **Starter Agent template** is now live.
+
+    ### ▶️ **First step: try it out**
+    Ask your agent (in the chat preview on the right):
+    - *What can you do?*
+    - *Can I talk to a human?*
+
+  prompt_to_bot: |
+    👋 Welcome to Hello Rasa!
+
+    I'm your **Copilot** — here to help you explore Rasa and start customizing your agent.
+    You can ask me how your agent works, how to add new skills, or how to connect integrations.
+
+    Your custom agent has been created and trained successfully.
+
+    ### ▶️ **First step: try it out**
+    Test your agent in the chat preview on the right to see how it responds to your users.