rasa-pro 3.14.0rc4__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
@@ -75,6 +76,8 @@ class MCPBaseAgent(AgentProtocol):
 
     MAX_ITERATIONS = 10
 
+    TOOL_CALL_DEFAULT_TIMEOUT = 10  # seconds
+
     # ============================================================================
     # Initialization & Setup
     # ============================================================================
@@ -288,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,
@@ -303,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
@@ -325,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",
@@ -335,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."""
@@ -624,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(
@@ -637,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:
@@ -655,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

@@ -347,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/copilot_templated_message_provider.py

@@ -2,7 +2,7 @@ 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,

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
 

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/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