add last message to orchestrator history for sentinel (#476)

husseinmozannar · web-flow · commit 00d0a0daa352 · 2025-12-12T13:47:46.000-05:00
diff --git a/src/magentic_ui/teams/orchestrator/_orchestrator.py b/src/magentic_ui/teams/orchestrator/_orchestrator.py
@@ -1636,6 +1636,9 @@ async def _execute_sentinel_step(
                                     "type": "sentinel_error",
                                 },
                             )
+                            # add last message from agent to history
+                            self._state.message_history.append(last_agent_message)
+
                             # inform orchestrator that the step has failed
                             self._state.message_history.append(
                                 TextMessage(
@@ -1687,7 +1690,12 @@ async def _execute_sentinel_step(
                         log_msg,
                         metadata=complete_metadata,
                     )
+
+                    # add last message from agent to history
+                    self._state.message_history.append(last_agent_message)
+
                     # inform orchestrator that the step is completed
+
                     self._state.message_history.append(
                         TextMessage(
                             content=f"Sentinel step '{step.title}' completed successfully. Reason: {reason}",
