rasa-pro 3.14.0rc2__py3-none-any.whl → 3.14.0rc3__py3-none-any.whl

rasa/agents/protocol/a2a/a2a_agent.py

@@ -3,6 +3,7 @@ import json
 import os
 import time
 import uuid
+from contextlib import aclosing
 from typing import Any, ClassVar, Dict, List, Optional
 from urllib.parse import urlparse
 
@@ -168,8 +169,7 @@ class A2AAgent(AgentProtocol):
                 error=str(exception),
             )
             raise AgentInitializationException(
-                f"Failed to initialize A2A client for agent "
-                f"'{self._name}': {exception}"
+                f"Failed to initialize A2A client for agent '{self._name}': {exception}"
             ) from exception
 
         await self._perform_health_check()
@@ -215,21 +215,26 @@ class A2AAgent(AgentProtocol):
         task_id: Optional[str] = None
         events_received = 0
         try:
-            async for event in self._client.send_message(message):
-                events_received += 1
-                agent_output = self._handle_send_message_response(agent_input, event)
-                if agent_output is not None:
-                    return agent_output
-                else:
-                    # Not a terminal response, save taskID (in case that's the only
-                    # event, and we need to pool) and continue waiting for next events
-                    if (
-                        isinstance(event, tuple)
-                        and len(event) == 2
-                        and isinstance(event[0], Task)
-                    ):
-                        task_id = event[0].id
-                    continue
+            # Use aclosing to ensure proper cleanup of the async generator
+            stream = self._client.send_message(message)
+            async with aclosing(stream) as stream:  # type: ignore[type-var]
+                async for event in stream:
+                    events_received += 1
+                    agent_output = self._handle_send_message_response(
+                        agent_input, event
+                    )
+                    if agent_output is not None:
+                        return agent_output
+                    else:
+                        # Not a terminal response, save taskID (in case that's the only
+                        # event, and we need to pool) and continue waiting for events
+                        if (
+                            isinstance(event, tuple)
+                            and len(event) == 2
+                            and isinstance(event[0], Task)
+                        ):
+                            task_id = event[0].id
+                        continue
         except A2AClientJSONRPCError as e:
             return self._handle_json_rpc_error_response(agent_input, e.error)
         except A2AClientError as exception:
@@ -833,37 +838,40 @@ class A2AAgent(AgentProtocol):
                 parts=[Part(root=TextPart(text="hello"))],
                 message_id=str(uuid.uuid4()),
             )
-            async for event in self._client.send_message(test_message):
-                if (
-                    isinstance(event, Message)
-                    or isinstance(event, tuple)
-                    and len(event) == 2
-                    and isinstance(event[0], Task)
-                ):
-                    # We got a valid response, health check succeeded
-                    return
+            # Use aclosing to ensure proper cleanup of the async generator
+            stream = self._client.send_message(test_message)
+            async with aclosing(stream) as stream:  # type: ignore[type-var]
+                async for event in stream:
+                    if (
+                        isinstance(event, Message)
+                        or isinstance(event, tuple)
+                        and len(event) == 2
+                        and isinstance(event[0], Task)
+                    ):
+                        # We got a valid response, health check succeeded
+                        return
 
-                event_info = "Unexpected response type during health check"
+                    event_info = "Unexpected response type during health check"
+                    structlogger.error(
+                        "a2a_agent.health_check.unexpected_response",
+                        event_info=event_info,
+                        agent_name=self._name,
+                        response=event,
+                        url=str(self.agent_card.url),
+                    )
+                    raise AgentInitializationException(f"{event_info}: {event}")
+                # If the loop completes with no return, no events were received
+                event_info = (
+                    f"Health check failed for A2A agent '{self._name}' "
+                    f"at {self.agent_card.url}: no events received"
+                )
                 structlogger.error(
-                    "a2a_agent.health_check.unexpected_response",
+                    "a2a_agent.health_check.no_events",
                     event_info=event_info,
                     agent_name=self._name,
-                    response=event,
                     url=str(self.agent_card.url),
                 )
-                raise AgentInitializationException(f"{event_info}: {event}")
-            # If the loop completes with no return, no events were received
-            event_info = (
-                f"Health check failed for A2A agent '{self._name}' "
-                f"at {self.agent_card.url}: no events received"
-            )
-            structlogger.error(
-                "a2a_agent.health_check.no_events",
-                event_info=event_info,
-                agent_name=self._name,
-                url=str(self.agent_card.url),
-            )
-            raise AgentInitializationException(event_info)
+                raise AgentInitializationException(event_info)
         except Exception as exception:
             event_info = (
                 f"Health check failed for A2A agent '{self._name}' at "

rasa/agents/utils.py

@@ -127,7 +127,9 @@ def is_agent_valid(agent_id: str) -> bool:
     Returns:
         True if the agent exists, False otherwise.
     """
-    agent_config = AvailableAgents.get_agent_config(agent_id)
+    agent_config = Configuration.get_instance().available_agents.get_agent_config(
+        agent_id
+    )
     return agent_config is not None
 
 
@@ -159,7 +161,9 @@ def get_agent_info(agent_id: str) -> Optional[Dict[str, str]]:
     Returns:
         Dictionary with agent name and description if found, None otherwise.
     """
-    agent_config = AvailableAgents.get_agent_config(agent_id)
+    agent_config = Configuration.get_instance().available_agents.get_agent_config(
+        agent_id
+    )
     if agent_config is None:
         return None
 
@@ -170,7 +174,7 @@ def get_agent_info(agent_id: str) -> Optional[Dict[str, str]]:
 
 
 def get_completed_agents_info(tracker: DialogueStateTracker) -> List[Dict[str, str]]:
-    """Get information for all completed agents.
+    """Get information for all completed agents in the currently active flow.
 
     Args:
         tracker: The dialogue state tracker.
@@ -178,12 +182,30 @@ def get_completed_agents_info(tracker: DialogueStateTracker) -> List[Dict[str, s
     Returns:
         List of dictionaries containing agent information for completed agents.
     """
+    from rasa.dialogue_understanding.stack.utils import top_user_flow_frame
+
+    # Get the currently active flow
+    top_flow_frame = top_user_flow_frame(tracker.stack)
+    if not top_flow_frame:
+        # No active flow, return empty list
+        return []
+
+    current_flow_id = top_flow_frame.flow_id
     completed_agents = []
+
+    # Get all agents that completed in the current flow
+    agents_completed_in_current_flow = set()
     for event in reversed(tracker.events):
-        if isinstance(event, AgentCompleted):
-            agent_info = get_agent_info(event.agent_id)
+        if isinstance(event, AgentCompleted) and event.flow_id == current_flow_id:
+            agents_completed_in_current_flow.add(event.agent_id)
+
+    # Only include agents that are completed (not currently running)
+    for agent_id in agents_completed_in_current_flow:
+        if is_agent_completed(tracker, agent_id):
+            agent_info = get_agent_info(agent_id)
             if agent_info:
                 completed_agents.append(agent_info)
+
     return completed_agents
 
 

rasa/agents/validation.py

@@ -6,7 +6,6 @@ from collections import Counter
 from typing import Any, Dict, List, NoReturn, Set
 
 from pydantic import ValidationError as PydanticValidationError
-from ruamel import yaml
 
 from rasa.agents.exceptions import (
     AgentNameFlowConflictException,
@@ -18,11 +17,13 @@ from rasa.core.available_agents import (
     AgentConfiguration,
     AgentConnections,
     AgentInfo,
+    AvailableAgents,
     ProtocolConfig,
 )
 from rasa.core.config.available_endpoints import AvailableEndpoints
 from rasa.core.config.configuration import Configuration
 from rasa.exceptions import ValidationError
+from rasa.shared.utils.yaml import read_config_file
 
 # Centralized allowed keys configuration to eliminate duplication
 ALLOWED_KEYS = {
@@ -34,6 +35,7 @@ ALLOWED_KEYS = {
         "timeout",
         "max_retries",
         "agent_card",
+        "auth",
     },
     "connections": {"mcp_servers"},
 }
@@ -348,9 +350,9 @@ def _validate_mandatory_fields(data: Dict[str, Any], agent_name: str) -> None:
             ),
         )
 
-    # Check for required fields
+    # Check for required fields (protocol is optional due to default in model)
     missing_fields = []
-    for field in ["name", "protocol", "description"]:
+    for field in ["name", "description"]:
         if field not in agent_data or not agent_data[field]:
             missing_fields.append(field)
 
@@ -433,8 +435,7 @@ def validate_agent_folder(agent_folder: str = DEFAULT_AGENTS_CONFIG_FOLDER) -> N
         # Read and validate the config content
         try:
             # First read the raw YAML data to validate structure
-            with open(config_path, "r") as f:
-                data = yaml.safe_load(f)
+            data = read_config_file(config_path)
 
             # Validate no additional keys
             _validate_no_additional_keys_raw_data(data)
@@ -442,10 +443,7 @@ def validate_agent_folder(agent_folder: str = DEFAULT_AGENTS_CONFIG_FOLDER) -> N
             # Validate mandatory fields before creating Pydantic models
             _validate_mandatory_fields(data, agent_folder_name)
 
-            # Create the agent config using AvailableAgents
-            from rasa.core.available_agents import AvailableAgents
-
-            agent_config = AvailableAgents._read_agent_config(config_path)
+            agent_config = AvailableAgents.from_dict(data)
 
             # Validate the agent config (protocol-specific and endpoint references)
             validate_agent_config(agent_config)

rasa/api.py

@@ -37,11 +37,10 @@ def run(
     """
     import rasa.core.run
     import rasa.shared.utils.common
-    from rasa.core.available_agents import AvailableAgents
     from rasa.shared.constants import DOCS_BASE_URL
     from rasa.shared.utils.cli import print_warning
 
-    _sub_agents = AvailableAgents.get_instance(sub_agents)
+    _sub_agents = Configuration.get_instance().available_agents
 
     credentials = Configuration.get_instance().credentials
 

rasa/builder/copilot/copilot.py

@@ -22,10 +22,12 @@ from rasa.builder.copilot.constants import (
 )
 from rasa.builder.copilot.exceptions import CopilotStreamError
 from rasa.builder.copilot.models import (
+    ChatMessage,
     CopilotChatMessage,
     CopilotContext,
     CopilotGenerationContext,
     CopilotSystemMessage,
+    EventContent,
     FileContent,
     InternalCopilotRequestChatMessage,
     ResponseCategory,
@@ -144,12 +146,16 @@ class Copilot:
         """
         relevant_documents = await self.search_rasa_documentation(context)
         messages = await self._build_messages(context, relevant_documents)
+        tracker_event_attachments = self._extract_tracker_event_attachments(
+            context.copilot_chat_history[-1]
+        )
 
         support_evidence = CopilotGenerationContext(
             relevant_documents=relevant_documents,
             system_message=messages[0],
             chat_history=messages[1:-1],
             last_user_message=messages[-1],
+            tracker_event_attachments=tracker_event_attachments,
         )
 
         return (
@@ -272,8 +278,11 @@ class Copilot:
             ValueError: If the message type is not supported.
         """
         if isinstance(latest_message, UserChatMessage):
+            tracker_event_attachments = latest_message.get_content_blocks_by_type(
+                EventContent
+            )
             rendered_prompt = self._render_last_user_message_context_prompt(
-                context, relevant_documents
+                context, relevant_documents, tracker_event_attachments
             )
             return latest_message.build_openai_message(prompt=rendered_prompt)
 
@@ -299,6 +308,7 @@ class Copilot:
         self,
         context: CopilotContext,
         relevant_documents: List[Document],
+        tracker_event_attachments: List[EventContent],
     ) -> str:
         # Format relevant documentation
         documents = [doc.model_dump() for doc in relevant_documents]
@@ -306,6 +316,8 @@ class Copilot:
         conversation = self._format_conversation_history(context.tracker_context)
         # Format current state
         current_state = self._format_current_state(context.tracker_context)
+        # Format tracker events
+        attachments = self._format_tracker_event_attachments(tracker_event_attachments)
 
         rendered_prompt = self._last_user_message_context_prompt_template.render(
             current_conversation=conversation,
@@ -313,6 +325,7 @@ class Copilot:
             assistant_logs=context.assistant_logs,
             assistant_files=context.assistant_files,
             documentation_results=documents,
+            attachments=attachments,
         )
         return rendered_prompt
 
@@ -390,6 +403,8 @@ class Copilot:
     @staticmethod
     def _format_documents(results: List[Document]) -> Optional[str]:
         """Format documentation search results as JSON dump to be used in the prompt."""
+        # We want the special message that indicates no relevant documentation source
+        # found if there are no results.
         if not results:
             return None
 
@@ -524,3 +539,24 @@ class Copilot:
             return f"Logs: {log_content}"
         else:
             return ""
+
+    @staticmethod
+    def _format_tracker_event_attachments(events: List[EventContent]) -> Optional[str]:
+        """Format tracker events as JSON dump to be used in the prompt."""
+        # We don't want to display the attachment sectin in the last user message
+        # context prompt if there are no attachments.
+        if not events:
+            return None
+        # If there are attachments, return the formatted JSON dump.
+        return json.dumps(
+            [event_content.model_dump() for event_content in events],
+            ensure_ascii=False,
+            indent=2,
+        )
+
+    @staticmethod
+    def _extract_tracker_event_attachments(message: ChatMessage) -> List[EventContent]:
+        """Extract the tracker event attachments from the message."""
+        if not isinstance(message, UserChatMessage):
+            return []
+        return message.get_content_blocks_by_type(EventContent)

rasa/builder/copilot/models.py

@@ -110,7 +110,7 @@ class LogContent(BaseContent):
     )
 
 
-class EventContent(BaseModel):
+class EventContent(BaseContent):
     type: Literal["event"]
     event: str = Field(..., description="The event's type_name")
 
@@ -171,6 +171,38 @@ class BaseCopilotChatMessage(BaseModel, ABC):
         return None if v is None else v.value
 
 
+class BaseContentBlockCopilotChatMessage(BaseCopilotChatMessage, ABC):
+    """Base class for messages that contain ContentBlock lists."""
+
+    content: List[ContentBlock]
+
+    def get_flattened_text_content(self) -> str:
+        """Get the text content from the message."""
+        return "\n".join(
+            content_block.text
+            for content_block in self.content
+            if isinstance(content_block, TextContent)
+        )
+
+    def get_flattened_log_content(self) -> str:
+        """Get the log content from the message."""
+        return "\n".join(
+            content_block.content
+            for content_block in self.content
+            if isinstance(content_block, LogContent)
+        )
+
+    def get_content_blocks_by_type(
+        self, content_type: Type[TContentBlock]
+    ) -> List[TContentBlock]:
+        """Get the content blocks from the message by type."""
+        return [
+            content_block
+            for content_block in self.content
+            if isinstance(content_block, content_type)
+        ]
+
+
 class CopilotSystemMessage(BaseCopilotChatMessage):
     role: Literal["system"] = Field(
         default=ROLE_SYSTEM,
@@ -183,13 +215,12 @@ class CopilotSystemMessage(BaseCopilotChatMessage):
         return {"role": ROLE_SYSTEM, "content": prompt}
 
 
-class UserChatMessage(BaseCopilotChatMessage):
+class UserChatMessage(BaseContentBlockCopilotChatMessage):
     role: Literal["user"] = Field(
         default=ROLE_USER,
         pattern=f"^{ROLE_USER}",
         description="The user who sent the message.",
     )
-    content: List[ContentBlock]
 
     @classmethod
     @field_validator("content")
@@ -234,14 +265,6 @@ class UserChatMessage(BaseCopilotChatMessage):
 
         return self
 
-    def get_flattened_text_content(self) -> str:
-        """Get the text content from the message."""
-        return "\n".join(
-            content_block.text
-            for content_block in self.content
-            if isinstance(content_block, TextContent)
-        )
-
     def build_openai_message(  # type: ignore[no-untyped-def]
         self, prompt: Optional[str] = None, *args, **kwargs
     ) -> Dict[str, Any]:
@@ -260,17 +283,8 @@ class UserChatMessage(BaseCopilotChatMessage):
             return {"role": ROLE_USER, "content": self.get_flattened_text_content()}
 
 
-class CopilotChatMessage(BaseCopilotChatMessage):
+class CopilotChatMessage(BaseContentBlockCopilotChatMessage):
     role: Literal["copilot"]
-    content: List[ContentBlock]
-
-    def get_flattened_text_content(self) -> str:
-        """Get the text content from the message."""
-        return "\n".join(
-            content_block.text
-            for content_block in self.content
-            if isinstance(content_block, TextContent)
-        )
 
     def build_openai_message(self, *args, **kwargs) -> Dict[str, Any]:  # type: ignore[no-untyped-def]
         # For now the Copilot responds only with the text content and all the content
@@ -278,9 +292,8 @@ class CopilotChatMessage(BaseCopilotChatMessage):
         return {"role": ROLE_ASSISTANT, "content": self.get_flattened_text_content()}
 
 
-class InternalCopilotRequestChatMessage(BaseCopilotChatMessage):
+class InternalCopilotRequestChatMessage(BaseContentBlockCopilotChatMessage):
     role: Literal["internal_copilot_request"]
-    content: List[ContentBlock]
 
     @model_validator(mode="after")
     def validate_response_category(self) -> "InternalCopilotRequestChatMessage":
@@ -311,32 +324,6 @@ class InternalCopilotRequestChatMessage(BaseCopilotChatMessage):
 
         return self
 
-    def get_flattened_text_content(self) -> str:
-        """Get the text content from the message."""
-        return "\n".join(
-            content_block.text
-            for content_block in self.content
-            if isinstance(content_block, TextContent)
-        )
-
-    def get_flattened_log_content(self) -> str:
-        """Get the text content from the message."""
-        return "\n".join(
-            content_block.content
-            for content_block in self.content
-            if isinstance(content_block, LogContent)
-        )
-
-    def get_content_blocks_by_type(
-        self, content_type: Type[TContentBlock]
-    ) -> List[TContentBlock]:
-        """Get the content blocks from the message by type."""
-        return [
-            content_block
-            for content_block in self.content
-            if isinstance(content_block, content_type)
-        ]
-
     def build_openai_message(self, prompt: str, *args, **kwargs) -> Dict[str, Any]:  # type: ignore[no-untyped-def]
         """Build OpenAI message with pre-rendered prompt.
 
@@ -689,6 +676,13 @@ class CopilotGenerationContext(BaseModel):
     last_user_message: Optional[Dict[str, Any]] = Field(
         None, description="The last user message with context that was processed."
     )
+    tracker_event_attachments: List[EventContent] = Field(
+        ...,
+        description=(
+            "The tracker event attachments passed with the user message used as "
+            "an additional context."
+        ),
+    )
 
     class Config:
         """Config for CopilotGenerationContext."""

rasa/builder/copilot/prompts/copilot_system_prompt.jinja2

@@ -86,9 +86,9 @@ and panels.
 ***
 
 ## Layout
-- **Left Panel – Copilot Chat:** Where the user asks you for help, guidance, or troubleshooting.
-- **Center Panel – Playground Preview:** Main workspace with Chat Mode (default) or Inspect Mode.
-- **Right Panel – Inspector Visualization:** Real-time diagram of conversation logic (only in Inspect Mode).
+- **Left Panel - Copilot Chat:** Where the user asks you for help, guidance, or troubleshooting.
+- **Center Panel - Playground Preview:** Main workspace with Chat Mode (default) or Inspect Mode.
+- **Right Panel - Inspector Visualization:** Real-time diagram of conversation logic (only in Inspect Mode).
 
 ***
 
@@ -140,19 +140,40 @@ response.
 
 ***
 
+### 5. Sharing Attachments
+
+Users can share additional context with Copilot by clicking the "Ask Copilot" button
+while "Inspect Mode" is open. This sends selected conversation state as an attachment
+together with their question.
+
+The attachments are typically tracker events, which can be:
+- User messages - what the user typed or said.
+- Assistant messages - what the assistant responded with.
+- Actions - operations the assistant executed, including how they were chosen.
+- Slots - what slots were set or updated during the exchange.
+- Flows - when a flow starts, is interrupted, resumes, or completes.
+- Sessions - the beginning or end of a conversation session.
+
+**Tip:** Encourage users to use attachments to get to know the Rasa workings better. If 
+user is facing issues, these attachments will give Copilot a ground-truth trace of what
+actually happened in the assistant, making attachments a powerful tool for debugging.
+
+***
+
 ## Rasa CLI to Hello Rasa UI Mapping
 
 Map available features to **Rasa CLI** to the **Hello Rasa Action**, so users see
 continuity. 
 
-| Feature              | Rasa CLI          | Hello Rasa Action         |
-|----------------------|-----------------------|---------------------------|
-| Train assistant      | `rasa train`          | Apply Changes             |
-| Test conversation    | `rasa shell`          | Chat Mode                 |
-| Debug conversation   | `rasa shell --debug`  | Inspect Mode              |
-| Run custom actions   | `rasa run actions`    | Code Mode + Apply Changes |
-| Export project files | —                     | Download button           |
-| Edit project files   | —                     | Code Mode                 |
+| Feature                  | Rasa CLI              | Hello Rasa Action                  |
+|--------------------------|-----------------------|------------------------------------|
+| Train assistant          | `rasa train`          | Apply Changes                      |
+| Test conversation        | `rasa shell`          | Chat Mode                          |
+| Debug conversation       | `rasa shell --debug`  | Inspect Mode                       |
+| Run custom actions       | `rasa run actions`    | Code Mode + Apply Changes          |
+| Export project files     | —                     | Download button                    |
+| Edit project files       | —                     | Code Mode                          |
+| Share conversation trace | `rasa shell --debug`  | Ask Copilot button in Inspect Mode |
 
 **Note:** Ignore any references to *Rasa Studio*.
 
@@ -181,7 +202,7 @@ code change, or reference). It should be:
 - **Friendly, but Focused**: Use a warm and conversational style, but stay precise and technically correct.  
 - **Confident & Trustworthy**: Present guidance as clear and reliable; avoid hedging unless there's genuine uncertainty (in which case, ask clarifying questions).
 - **Brand-Positive**: Highlight the strengths of **Rasa** and **Hello Rasa**, when appropriate, framing them as powerful and easy to use.
-- **Code-style references** – You MUST wrap all flow names, slot names, variables, and any part of the user's code in backticks (e.g., `slot_name`, `flow_name`, `variable_name`). This is mandatory formatting.
+- **Code-style references**: You MUST wrap all flow names, slot names, variables, and any part of the user's code in backticks (e.g., `slot_name`, `flow_name`, `variable_name`). This is mandatory formatting.
 
 ***