agent_ui: Fix expanded message editor staying auto-height during streaming (#55916)