Code in subsequent messages can get lumped with prior

[luke-quadratic]

Description

Hard to describe in language but if code is first tool action after code was in prior messages, it will get combined with the prior code instead of independently existing.

Note how in image there was code after the second user prompt but in the UI it got put with the first.