add final assistant response

Author: mayinghan

eval_protocol/mcp/execution/manager.py

@@ -337,13 +337,15 @@ async def _execute_rollout(
                                 "tool_calls": [f"{tool_call.tool_name}({tool_call.arguments})"],
                                 "num_tool_calls": 1,
                             }
+                            print(f"🔍 control_plane_step: {control_plane_step}")
                             conversation_history[-1]["control_plane_step"] = control_plane_step
                             trajectory.control_plane_steps.append(control_plane_step)
 
                             # Log conversation state for playback if in recording mode
                             if recording_mode:
                                 policy.log_conversation_state_for_playback(rollout_idx, step - 1, conversation_history)
 
+                        # tool indicates rollout should be terminated, call policy one last time to get the final response
                         if rollout_end:
                             trajectory.terminated = True
                             trajectory.termination_reason = TerminationReason.CONTROL_PLANE_SIGNAL
@@ -390,6 +392,9 @@ async def _execute_rollout(
                     trajectory.terminated = True
                     trajectory.termination_reason = TerminationReason.CONTROL_PLANE_SIGNAL
 
+                    _, usage_stats = await policy(tool_schema, rollout_idx, conversation_history)
+                    usage_stats_list.append(usage_stats)
+
                     # Add final control plane summary
                     trajectory.control_plane_summary.update(
                         {