Merge pull request #22 from MiroMindAI/frontend-dev-pengxiang

feat(agent): update UI for miroflow in alignment with MiroThinker online demo

Author: BinWang28

web_app/api/routes/tasks.py

@@ -121,6 +121,8 @@ async def get_task_status(
         raise HTTPException(status_code=404, detail="Task not found")
 
     progress: dict = {}
+    stored_messages: list = []
+
     if task.status == "running":
         progress = task_executor.get_task_progress(task_id)
         # Update session with progress
@@ -131,9 +133,15 @@ async def get_task_status(
                 "step_count": progress.get("step_count", 0),
             },
         )
-
-    # Convert messages to Message objects
-    messages = [Message(**m) for m in progress.get("messages", [])]
+    else:
+        # For completed/failed/cancelled tasks, get stored messages from session
+        session_data = session_manager._read_session(task_id)
+        if session_data:
+            stored_messages = session_data.get("messages", [])
+
+    # Convert messages to Message objects - use progress messages for running, stored for completed
+    raw_messages = progress.get("messages", []) if progress else stored_messages
+    messages = [Message(**m) for m in raw_messages]
 
     return TaskStatusUpdate(
         id=task.id,

web_app/core/task_executor.py

@@ -138,13 +138,17 @@ async def _run_task(
             # Run agent
             result = await agent.run(ctx)
 
-            # Update session with results
+            # Get final message history before cleanup
+            final_messages = self._get_all_messages_from_tracer(tracer)
+
+            # Update session with results and full message history
             self.session_manager.update_task(
                 task_id,
                 {
                     "status": "completed",
                     "final_answer": result.get("final_boxed_answer", ""),
                     "summary": result.get("summary", ""),
+                    "messages": final_messages,
                 },
             )
 
@@ -169,6 +173,24 @@ async def _run_task(
             if task_id in self._task_tracers:
                 del self._task_tracers[task_id]
 
+    def _get_all_messages_from_tracer(self, tracer: Any) -> list[dict]:
+        """Extract all messages from tracer for persistence."""
+        try:
+            with tracer._data_lock:
+                for key, log_file in tracer._active_tasks.items():
+                    agent_states = log_file.agent_states
+                    for agent_name, state in agent_states.items():
+                        state_data = (
+                            state.state
+                            if hasattr(state, "state")
+                            else state.get("state", {})
+                        )
+                        message_history = state_data.get("message_history", [])
+                        return self._format_messages(message_history)
+        except Exception:
+            pass
+        return []
+
     def get_task_progress(self, task_id: str) -> dict[str, Any]:
         """Get current progress from tracer."""
         tracer = self._task_tracers.get(task_id)
@@ -199,8 +221,8 @@ def get_task_progress(self, task_id: str) -> dict[str, Any]:
                         current_turn = max(
                             current_turn, (len(message_history) + 1) // 2
                         )
-                        # Get recent messages for display
-                        messages = self._format_messages(message_history[-6:])
+                        # Get ALL messages for display (full history)
+                        messages = self._format_messages(message_history)
 
                     # Filter and format logs to show tool calls
                     recent_logs = (
@@ -285,7 +307,12 @@ def _format_recent_logs(self, logs: list[dict]) -> list[dict]:
         return formatted[-15:]
 
     def _format_messages(self, messages: list[dict]) -> list[dict]:
-        """Format message history for display."""
+        """Format message history for display.
+
+        Note: We do NOT truncate message content here to preserve:
+        - Full thinking/reasoning content for proper display
+        - Complete tool results (e.g., search results JSON) for parsing
+        """
         formatted = []
         for msg in messages:
             role = msg.get("role", "unknown")
@@ -300,9 +327,12 @@ def _format_messages(self, messages: list[dict]) -> list[dict]:
                         if item.get("type") == "text":
                             text_parts.append(item.get("text", ""))
                         elif item.get("type") == "tool_result":
-                            text_parts.append(
-                                f"[Tool Result: {self._truncate_output(item.get('content', ''))}]"
-                            )
+                            # Don't truncate tool results - need full JSON for parsing
+                            tool_content = item.get("content", "")
+                            if isinstance(tool_content, str):
+                                text_parts.append(tool_content)
+                            else:
+                                text_parts.append(str(tool_content))
                         elif item.get("type") == "tool_use":
                             text_parts.append(f"[Tool Call: {item.get('name', '')}]")
                     elif isinstance(item, str):
@@ -311,10 +341,11 @@ def _format_messages(self, messages: list[dict]) -> list[dict]:
             elif not isinstance(content, str):
                 content = str(content)
 
+            # Don't truncate - preserve full content for thinking and tool results
             formatted.append(
                 {
                     "role": role,
-                    "content": self._truncate_output(content),
+                    "content": content,
                 }
             )