rasa-pro 3.14.0.dev2__py3-none-any.whl → 3.14.0.dev4__py3-none-any.whl

rasa/agents/agent_manager.py

@@ -77,6 +77,8 @@ class AgentManager(metaclass=Singleton):
         Raises:
             ConnectionError: If the agent connection fails.
         """
+        # Add the agent to the manager
+        agent_identifier = make_agent_identifier(agent_name, protocol_type)
         try:
             # Create the agent client
             client = AgentFactory.create_client(protocol_type, config)
@@ -84,8 +86,6 @@ class AgentManager(metaclass=Singleton):
             # Connect the agent client
             await client.connect()
 
-            # Add the agent to the manager
-            agent_identifier = make_agent_identifier(agent_name, protocol_type)
             self._add_agent(agent_identifier, client)
             structlogger.info(
                 "agent_manager.connect_agent.success",

rasa/agents/constants.py

@@ -1,3 +1,8 @@
+# Agent configuration constants
+AGENT_DEFAULT_MAX_RETRIES = 3
+AGENT_DEFAULT_TIMEOUT_SECONDS = 30
+MAX_AGENT_RETRY_DELAY_SECONDS = 5
+
 # MCP Tool related constants (primary)
 TOOL_TYPE_KEY = "type"
 TOOL_NAME_KEY = "name"
@@ -5,6 +10,7 @@ TOOL_DESCRIPTION_KEY = "description"
 TOOL_PARAMETERS_KEY = "parameters"
 TOOL_STRICT_KEY = "strict"
 TOOL_TYPE_FUNCTION_KEY = "function"
+TOOL_EXECUTOR_KEY = "tool_executor"
 
 # MCP Tool related constants (secondary)
 TOOL_ADDITIONAL_PROPERTIES_KEY = "additionalProperties"
@@ -29,3 +35,6 @@ KEY_TOOL_CALLS = "tool_calls"
 AGENT_METADATA_AGENT_RESPONSE_KEY = "agent_response"
 AGENT_METADATA_TOOL_RESULTS_KEY = "tool_results"
 AGENT_METADATA_EXIT_IF_KEY = "exit_if"
+# A2A-specific constants
+A2A_AGENT_CONTEXT_ID_KEY = "context_id"
+A2A_AGENT_TASK_ID_KEY = "task_id"

rasa/agents/core/agent_protocol.py

@@ -18,8 +18,7 @@ class AgentProtocol(Protocol):
 
     @classmethod
     def from_config(cls, config: AgentConfig) -> AgentProtocol:
-        """
-        Initialize the Agent with the given configuration.
+        """Initialize the Agent with the given configuration.
 
         This class method should be implemented to parse the given
         configuration and create an instance of an agent.