Fix delegation visualization labels to show correct sender/receiver

examples/01_standalone_sdk/25_agent_delegation.py

@@ -18,9 +18,8 @@
     Tool,
     get_logger,
 )
-from openhands.sdk.conversation import DefaultConversationVisualizer
 from openhands.sdk.tool import register_tool
-from openhands.tools.delegate import DelegateTool
+from openhands.tools.delegate import DelegateTool, DelegationVisualizer
 from openhands.tools.preset.default import get_default_tools
 
 
@@ -51,7 +50,7 @@
 conversation = Conversation(
     agent=main_agent,
     workspace=cwd,
-    visualizer=DefaultConversationVisualizer(name="Delegator"),
+    visualizer=DelegationVisualizer(name="Delegator"),
 )
 
 task_message = (

examples/01_standalone_sdk/26_custom_visualizer.py

@@ -26,15 +26,6 @@
 class MinimalVisualizer(ConversationVisualizerBase):
     """A minimal visualizer that print the raw events as they occur."""
 
-    def __init__(self, name: str | None = None):
-        """Initialize the minimal progress visualizer.
-
-        Args:
-            name: Optional name to identify the agent/conversation.
-        """
-        # Initialize parent - state will be set later via initialize()
-        super().__init__(name=name)
-
     def on_event(self, event: Event) -> None:
         """Handle events for minimal progress visualization."""
         print(f"\n\n[EVENT] {type(event).__name__}: {event.model_dump_json()[:200]}...")

openhands-sdk/openhands/sdk/conversation/base.py

@@ -118,8 +118,17 @@ def state(self) -> ConversationStateProtocol: ...
     def conversation_stats(self) -> ConversationStats: ...
 
     @abstractmethod
-    def send_message(self, message: str | Message) -> None:
-        """Send a message to the agent."""
+    def send_message(self, message: str | Message, sender: str | None = None) -> None:
+        """Send a message to the agent.
+
+        Args:
+            message: Either a string (which will be converted to a user message)
+                    or a Message object
+            sender: Optional identifier of the sender. Can be used to track
+                   message origin in multi-agent scenarios. For example, when
+                   one agent delegates to another, the sender can be set to
+                   identify which agent is sending the message.
+        """
         ...
 
     @abstractmethod

openhands-sdk/openhands/sdk/conversation/impl/local_conversation.py

@@ -185,12 +185,16 @@ def stuck_detector(self) -> StuckDetector | None:
         return self._stuck_detector
 
     @observe(name="conversation.send_message")
-    def send_message(self, message: str | Message) -> None:
+    def send_message(self, message: str | Message, sender: str | None = None) -> None:
         """Send a message to the agent.
 
         Args:
             message: Either a string (which will be converted to a user message)
                     or a Message object
+            sender: Optional identifier of the sender. Can be used to track
+                   message origin in multi-agent scenarios. For example, when
+                   one agent delegates to another, the sender can be set to
+                   identify which agent is sending the message.
         """
         # Convert string to Message if needed
         if isinstance(message, str):
@@ -233,6 +237,7 @@ def send_message(self, message: str | Message) -> None:
                 llm_message=message,
                 activated_skills=activated_skill_names,
                 extended_content=extended_content,
+                sender=sender,
             )
             self._on_event(user_msg_event)
 

openhands-sdk/openhands/sdk/conversation/impl/remote_conversation.py

@@ -553,7 +553,7 @@ def stuck_detector(self):
         )
 
     @observe(name="conversation.send_message")
-    def send_message(self, message: str | Message) -> None:
+    def send_message(self, message: str | Message, sender: str | None = None) -> None:
         if isinstance(message, str):
             message = Message(role="user", content=[TextContent(text=message)])
         assert message.role == "user", (
@@ -564,6 +564,8 @@ def send_message(self, message: str | Message) -> None:
             "content": [c.model_dump() for c in message.content],
             "run": False,  # Mirror local semantics; explicit run() must be called
         }
+        if sender is not None:
+            payload["sender"] = sender
         _send_request(
             self._client, "POST", f"/api/conversations/{self._id}/events", json=payload
         )

openhands-sdk/openhands/sdk/conversation/visualizer/base.py

@@ -18,7 +18,7 @@ class ConversationVisualizerBase(ABC):
 
     The typical usage pattern:
     1. Create a visualizer instance:
-       `viz = MyVisualizer(name="agent1")`
+       `viz = MyVisualizer()`
     2. Pass it to Conversation: `conv = Conversation(agent, visualizer=viz)`
     3. Conversation automatically calls `viz.initialize(state)` to attach the state
 
@@ -28,21 +28,10 @@ class ConversationVisualizerBase(ABC):
     Conversation will then calls `MyVisualizer()` followed by `initialize(state)`
     """
 
-    _name: str | None
     _state: "ConversationStateProtocol | None"
 
-    def __init__(
-        self,
-        name: str | None = None,
-    ):
-        """Initialize the visualizer base.
-
-        Args:
-            name: Optional name to prefix in panel titles to identify
-                                  which agent/conversation is speaking. Will be
-                                  capitalized automatically.
-        """
-        self._name = name.capitalize() if name else None
+    def __init__(self):
+        """Initialize the visualizer base."""
         self._state = None
 
     @final
@@ -60,14 +49,6 @@ def initialize(self, state: "ConversationStateProtocol") -> None:
         """
         self._state = state
 
-    def update_name(self, name: str | None) -> None:
-        """Update the name used for visualization.
-
-        Args:
-            name: New name to use for visualization
-        """
-        self._name = name.capitalize() if name else None
-
     @property
     def conversation_stats(self) -> "ConversationStats | None":
         """Get conversation stats from the state."""

openhands-sdk/openhands/sdk/conversation/visualizer/default.py

@@ -60,25 +60,20 @@ class DefaultConversationVisualizer(ConversationVisualizerBase):
 
     def __init__(
         self,
-        name: str | None = None,
         highlight_regex: dict[str, str] | None = DEFAULT_HIGHLIGHT_REGEX,
         skip_user_messages: bool = False,
     ):
         """Initialize the visualizer.
 
         Args:
-            name: Optional name to prefix in panel titles to identify
-                                  which agent/conversation is speaking.
             highlight_regex: Dictionary mapping regex patterns to Rich color styles
                            for highlighting keywords in the visualizer.
                            For example: {"Reasoning:": "bold blue",
                            "Thought:": "bold green"}
             skip_user_messages: If True, skip displaying user messages. Useful for
                                 scenarios where user input is not relevant to show.
         """
-        super().__init__(
-            name=name,
-        )
+        super().__init__()
         self._console = Console()
         self._skip_user_messages = skip_user_messages
         self._highlight_patterns = highlight_regex or {}
@@ -126,10 +121,7 @@ def _create_event_panel(self, event: Event) -> Panel | None:
 
         # Determine panel styling based on event type
         if isinstance(event, SystemPromptEvent):
-            title = f"[bold {_SYSTEM_COLOR}]"
-            if self._name:
-                title += f"{self._name} "
-            title += f"System Prompt[/bold {_SYSTEM_COLOR}]"
+            title = f"[bold {_SYSTEM_COLOR}]System Prompt[/bold {_SYSTEM_COLOR}]"
             return Panel(
                 content,
                 title=title,
@@ -139,13 +131,13 @@ def _create_event_panel(self, event: Event) -> Panel | None:
             )
         elif isinstance(event, ActionEvent):
             # Check if action is None (non-executable)
-            title = f"[bold {_ACTION_COLOR}]"
-            if self._name:
-                title += f"{self._name} "
             if event.action is None:
-                title += f"Agent Action (Not Executed)[/bold {_ACTION_COLOR}]"
+                title = (
+                    f"[bold {_ACTION_COLOR}]Agent Action (Not Executed)"
+                    f"[/bold {_ACTION_COLOR}]"
+                )
             else:
-                title += f"Agent Action[/bold {_ACTION_COLOR}]"
+                title = f"[bold {_ACTION_COLOR}]Agent Action[/bold {_ACTION_COLOR}]"
             return Panel(
                 content,
                 title=title,
@@ -155,10 +147,9 @@ def _create_event_panel(self, event: Event) -> Panel | None:
                 expand=True,
             )
         elif isinstance(event, ObservationEvent):
-            title = f"[bold {_OBSERVATION_COLOR}]"
-            if self._name:
-                title += f"{self._name} "
-            title += f"Observation[/bold {_OBSERVATION_COLOR}]"
+            title = (
+                f"[bold {_OBSERVATION_COLOR}]Observation[/bold {_OBSERVATION_COLOR}]"
+            )
             return Panel(
                 content,
                 title=title,
@@ -167,10 +158,7 @@ def _create_event_panel(self, event: Event) -> Panel | None:
                 expand=True,
             )
         elif isinstance(event, UserRejectObservation):
-            title = f"[bold {_ERROR_COLOR}]"
-            if self._name:
-                title += f"{self._name} "
-            title += f"User Rejected Action[/bold {_ERROR_COLOR}]"
+            title = f"[bold {_ERROR_COLOR}]User Rejected Action[/bold {_ERROR_COLOR}]"
             return Panel(
                 content,
                 title=title,
@@ -193,20 +181,14 @@ def _create_event_panel(self, event: Event) -> Panel | None:
             }
             role_color = role_colors.get(event.llm_message.role, "white")
 
-            # "User Message To [Name] Agent" for user
-            # "Message from [Name] Agent" for agent
-            agent_name = f"{self._name} " if self._name else ""
-
+            # Simple titles for base visualizer
             if event.llm_message.role == "user":
-                title_text = (
-                    f"[bold {role_color}]User Message to "
-                    f"{agent_name}Agent[/bold {role_color}]"
-                )
+                title_text = f"[bold {role_color}]Message from User[/bold {role_color}]"
             else:
                 title_text = (
-                    f"[bold {role_color}]Message from "
-                    f"{agent_name}Agent[/bold {role_color}]"
+                    f"[bold {role_color}]Message from Agent[/bold {role_color}]"
                 )
+
             return Panel(
                 content,
                 title=title_text,
@@ -216,10 +198,7 @@ def _create_event_panel(self, event: Event) -> Panel | None:
                 expand=True,
             )
         elif isinstance(event, AgentErrorEvent):
-            title = f"[bold {_ERROR_COLOR}]"
-            if self._name:
-                title += f"{self._name} "
-            title += f"Agent Error[/bold {_ERROR_COLOR}]"
+            title = f"[bold {_ERROR_COLOR}]Agent Error[/bold {_ERROR_COLOR}]"
             return Panel(
                 content,
                 title=title,
@@ -229,10 +208,7 @@ def _create_event_panel(self, event: Event) -> Panel | None:
                 expand=True,
             )
         elif isinstance(event, PauseEvent):
-            title = f"[bold {_PAUSE_COLOR}]"
-            if self._name:
-                title += f"{self._name} "
-            title += f"User Paused[/bold {_PAUSE_COLOR}]"
+            title = f"[bold {_PAUSE_COLOR}]User Paused[/bold {_PAUSE_COLOR}]"
             return Panel(
                 content,
                 title=title,
@@ -241,10 +217,7 @@ def _create_event_panel(self, event: Event) -> Panel | None:
                 expand=True,
             )
         elif isinstance(event, Condensation):
-            title = f"[bold {_SYSTEM_COLOR}]"
-            if self._name:
-                title += f"{self._name} "
-            title += f"Condensation[/bold {_SYSTEM_COLOR}]"
+            title = f"[bold {_SYSTEM_COLOR}]Condensation[/bold {_SYSTEM_COLOR}]"
             return Panel(
                 content,
                 title=title,
@@ -254,10 +227,7 @@ def _create_event_panel(self, event: Event) -> Panel | None:
             )
 
         elif isinstance(event, CondensationRequest):
-            title = f"[bold {_SYSTEM_COLOR}]"
-            if self._name:
-                title += f"{self._name} "
-            title += f"Condensation Request[/bold {_SYSTEM_COLOR}]"
+            title = f"[bold {_SYSTEM_COLOR}]Condensation Request[/bold {_SYSTEM_COLOR}]"
             return Panel(
                 content,
                 title=title,
@@ -267,10 +237,10 @@ def _create_event_panel(self, event: Event) -> Panel | None:
             )
         else:
             # Fallback panel for unknown event types
-            title = f"[bold {_ERROR_COLOR}]"
-            if self._name:
-                title += f"{self._name} "
-            title += f"UNKNOWN Event: {event.__class__.__name__}[/bold {_ERROR_COLOR}]"
+            title = (
+                f"[bold {_ERROR_COLOR}]UNKNOWN Event: "
+                f"{event.__class__.__name__}[/bold {_ERROR_COLOR}]"
+            )
             return Panel(
                 content,
                 title=title,

openhands-sdk/openhands/sdk/event/llm_convertible/message.py

@@ -43,6 +43,13 @@ class MessageEvent(LLMConvertibleEvent):
     extended_content: list[TextContent] = Field(
         default_factory=list, description="List of content added by agent context"
     )
+    sender: str | None = Field(
+        default=None,
+        description=(
+            "Optional identifier of the sender. "
+            "Can be used to track message origin in multi-agent scenarios."
+        ),
+    )
 
     @property
     def reasoning_content(self) -> str:

openhands-tools/openhands/tools/delegate/__init__.py

@@ -6,11 +6,13 @@
     DelegateTool,
 )
 from openhands.tools.delegate.impl import DelegateExecutor
+from openhands.tools.delegate.visualizer import DelegationVisualizer
 
 
 __all__ = [
     "DelegateAction",
     "DelegateObservation",
     "DelegateExecutor",
     "DelegateTool",
+    "DelegationVisualizer",
 ]