Merge pull request #555 from UiPath/fix/chat_aggregations

fix: chat aggregations

Author: cristipufu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.1.37"
+version = "2.1.38"
 description = "Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools."
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath/_cli/_dev/_terminal/__init__.py

@@ -295,7 +295,7 @@ def _handle_chat_event(
         if not updated_chat_message:
             return
         details_panel = self.app.query_one("#details-panel", RunDetailsPanel)
-        details_panel.add_chat_message(updated_chat_message, execution_id)
+        details_panel.add_chat_message(event, updated_chat_message, execution_id)
 
     def _add_info_log(self, run: ExecutionRun, message: str):
         """Add info log to run."""

src/uipath/_cli/_dev/_terminal/_components/_chat.py

@@ -6,6 +6,7 @@
 from textual.widgets import Input, Markdown
 
 from uipath.agent.conversation import (
+    UiPathConversationEvent,
     UiPathConversationMessage,
     UiPathExternalValue,
     UiPathInlineValue,
@@ -17,11 +18,11 @@ class Prompt(Markdown):
 
 
 class Response(Markdown):
-    BORDER_TITLE = "ai"
+    BORDER_TITLE = "🤖 ai"
 
 
 class Tool(Markdown):
-    BORDER_TITLE = "tool"
+    BORDER_TITLE = "🛠️  tool"
 
 
 class ChatPanel(Container):
@@ -43,38 +44,40 @@ def compose(self) -> ComposeResult:
                 id="chat-input",
             )
 
-    def add_chat_message(self, chat_msg: UiPathConversationMessage) -> None:
+    def add_chat_message(
+        self, event: UiPathConversationEvent, chat_msg: UiPathConversationMessage
+    ) -> None:
         """Add or update a chat message bubble."""
         chat_view = self.query_one("#chat-view")
 
         widget_cls: Union[type[Prompt], type[Response], type[Tool]]
         if chat_msg.role == "user":
             widget_cls = Prompt
-            prefix = "👤"
         elif chat_msg.role == "assistant":
             widget_cls = Response
-            prefix = "🤖"
         else:
             widget_cls = Response
-            prefix = "⚒️"
 
         parts: List[str] = []
         if chat_msg.content_parts:
             for part in chat_msg.content_parts:
-                if part.mime_type.startswith("text/"):
+                if (
+                    part.mime_type.startswith("text/")
+                    or part.mime_type == "application/json"
+                ):
                     if isinstance(part.data, UiPathInlineValue):
                         parts.append(part.data.inline or "")
                     elif isinstance(part.data, UiPathExternalValue):
-                        # maybe fetch from URL or just show a placeholder
                         parts.append(f"[external: {part.data.url}]")
 
         text_block = "\n".join(parts).strip()
-        content_lines = [f"{prefix} {text_block}"] if text_block else []
+        content_lines = [f"{text_block}"] if text_block else []
 
         if chat_msg.tool_calls:
+            widget_cls = Tool
             for call in chat_msg.tool_calls:
-                args = call.arguments
-                content_lines.append(f"🛠 **{call.name}**\n{args}")
+                status_icon = "✓" if call.result else "⚙"
+                content_lines.append(f" {status_icon} **{call.name}**")
 
         if not content_lines:
             return
@@ -86,7 +89,12 @@ def add_chat_message(self, chat_msg: UiPathConversationMessage) -> None:
         last_update = self._last_update_time.get(chat_msg.message_id, 0.0)
 
         if existing:
-            if now - last_update > 0.15:
+            should_update = (
+                event.exchange
+                and event.exchange.message
+                and event.exchange.message.end is not None
+            )
+            if should_update or now - last_update > 0.15:
                 existing.update(content)
                 self._last_update_time[chat_msg.message_id] = now
                 chat_view.scroll_end(animate=False)

src/uipath/_cli/_dev/_terminal/_components/_details.py

@@ -6,7 +6,7 @@
 from textual.widgets import Input, RichLog, TabbedContent, TabPane, Tree
 from textual.widgets.tree import TreeNode
 
-from uipath.agent.conversation import UiPathConversationMessage
+from uipath.agent.conversation import UiPathConversationEvent, UiPathConversationMessage
 
 from .._models._execution import ExecutionRun
 from .._models._messages import LogMessage, TraceMessage
@@ -430,13 +430,16 @@ def add_log(self, log_msg: LogMessage):
             logs_log.write(log_msg.message)
 
     def add_chat_message(
-        self, chat_msg: UiPathConversationMessage, run_id: str
+        self,
+        event: UiPathConversationEvent,
+        chat_msg: UiPathConversationMessage,
+        run_id: str,
     ) -> None:
         """Add a chat message to the display."""
         if not self.current_run or run_id != self.current_run.id:
             return
         chat_panel = self.query_one("#chat-panel", ChatPanel)
-        chat_panel.add_chat_message(chat_msg)
+        chat_panel.add_chat_message(event, chat_msg)
 
     def clear_display(self):
         """Clear both traces and logs display."""

src/uipath/_cli/_dev/_terminal/_styles/terminal.tcss

@@ -226,15 +226,16 @@ TextArea.invalid {
 
 
 Prompt {
-    background: $primary 5%;
+    border: wide $primary-background;
+    background: $surface;
     color: $text;
     margin-right: 8;
     margin-left: 1;
     padding: 1 1 0 1;
 }
 Response, Tool {
-    border: wide $success;
-    background: $success 5%;
+    border: wide $primary-background;
+    background: $surface;
     color: $text;
     margin: 1;
     margin-left: 8;

src/uipath/_cli/_dev/_terminal/_utils/_chat.py

@@ -74,9 +74,10 @@ def _process_message_event(
             )
             self.messages[ev.message_id] = msg
 
-        # --- Handle content parts ---
+        # --- Handle content parts (text, JSON, etc.) ---
         if ev.content_part:
             cp_event = ev.content_part
+
             existing = next(
                 (
                     cp
@@ -86,27 +87,41 @@ def _process_message_event(
                 None,
             )
 
-            if cp_event.start:
+            # Start of a new content part
+            if cp_event.start and not existing:
+                new_cp = UiPathConversationContentPart(
+                    content_part_id=cp_event.content_part_id,
+                    mime_type=cp_event.start.mime_type,
+                    data=UiPathInlineValue(inline=""),
+                    citations=[],
+                    is_transcript=None,
+                    is_incomplete=True,
+                )
+                if msg.content_parts is None:
+                    msg.content_parts = []
+                msg.content_parts.append(new_cp)
+                existing = new_cp
+
+            # Chunk for an existing part (or backfill if start missing)
+            if cp_event.chunk:
                 if not existing:
+                    new_cp = UiPathConversationContentPart(
+                        content_part_id=cp_event.content_part_id,
+                        mime_type="text/plain",  # fallback if start missing
+                        data=UiPathInlineValue(inline=""),
+                        citations=[],
+                        is_transcript=None,
+                        is_incomplete=True,
+                    )
                     if msg.content_parts is None:
                         msg.content_parts = []
-                    msg.content_parts.append(
-                        UiPathConversationContentPart(
-                            content_part_id=cp_event.content_part_id,
-                            mime_type=cp_event.start.mime_type,
-                            data=UiPathInlineValue(
-                                inline=cp_event.chunk.data
-                                if cp_event.chunk and cp_event.chunk.data
-                                else ""
-                            ),
-                            citations=[],
-                            is_transcript=None,
-                            is_incomplete=None,
-                        )
-                    )
-            if cp_event.chunk and existing:
+                    msg.content_parts.append(new_cp)
+                    existing = new_cp
+
                 if isinstance(existing.data, UiPathInlineValue):
                     existing.data.inline += cp_event.chunk.data or ""
+
+            # End of a part
             if cp_event.end and existing:
                 existing.is_incomplete = bool(cp_event.end.interrupted)
 
@@ -122,43 +137,31 @@ def _process_message_event(
                 None,
             )
 
+            # Start of a tool call
             if tc_event.start:
                 if not existing_tool_call:
-                    # create new tool call
                     new_tc = UiPathConversationToolCall(
                         tool_call_id=tc_event.tool_call_id,
                         name=tc_event.start.tool_name,
-                        arguments=tc_event.start.arguments,
+                        arguments=None,  # args will arrive as JSON content part
                         timestamp=tc_event.start.timestamp,
                         result=None,
                     )
                     if msg.tool_calls is None:
                         msg.tool_calls = []
                     msg.tool_calls.append(new_tc)
+                    existing_tool_call = new_tc
                 else:
-                    # merge/update existing
                     existing_tool_call.name = (
                         tc_event.start.tool_name or existing_tool_call.name
                     )
-                    if tc_event.start.arguments:
-                        # If Inline, append more data
-                        if isinstance(
-                            tc_event.start.arguments, UiPathInlineValue
-                        ) and isinstance(
-                            existing_tool_call.arguments, UiPathInlineValue
-                        ):
-                            existing_tool_call.arguments.inline += (
-                                tc_event.start.arguments.inline or ""
-                            )
-                        else:
-                            existing_tool_call.arguments = tc_event.start.arguments
                     existing_tool_call.timestamp = (
                         tc_event.start.timestamp or existing_tool_call.timestamp
                     )
 
+            # End of a tool call
             if tc_event.end:
                 if not existing_tool_call:
-                    # Shouldn't happen, but initialize if end comes first
                     existing_tool_call = UiPathConversationToolCall(
                         tool_call_id=tc_event.tool_call_id,
                         name="",  # unknown until start seen
@@ -177,7 +180,6 @@ def _process_message_event(
 
         # --- Update timestamps ---
         msg.updated_at = self._timestamp(ev)
-
         return msg
 
     def _timestamp(self, ev: UiPathConversationMessageEvent) -> str: