Refactor session runtime persistence

src/agent_runtime.py

@@ -569,6 +569,7 @@ class LocalCodingAgent:
             user_context=stored_session.user_context,
             system_context=stored_session.system_context,
             messages=stored_session.messages,
+            display_messages=stored_session.display_messages,
         )
         self._append_file_history_replay_if_needed(
             session,
@@ -916,6 +917,7 @@ class LocalCodingAgent:
                                         'continuation_index': continuation_count,
                                     },
                                     message_id=f'continuation_{turn_index}',
+                                    display=False,
                                 )
                                 stream_events.append(
                                     {
@@ -1073,6 +1075,7 @@ class LocalCodingAgent:
                             'continuation_index': continuation_count,
                         },
                         message_id=f'continuation_{turn_index}',
+                        display=False,
                     )
                     stream_events.append(
                         {
@@ -1444,6 +1447,7 @@ class LocalCodingAgent:
                             'plugin_preflight_count': len(plugin_preflight_messages),
                         },
                         message_id=f'plugin_tool_runtime_{tool_call.id}',
+                        display=False,
                     )
                     stream_events.append(
                         {
@@ -3862,6 +3866,7 @@ class LocalCodingAgent:
                 'file_history_snapshot_count': snapshot_count,
             },
             message_id=f'file_history_replay_{replay_count}',
+            display=False,
         )
 
     def _render_file_history_replay(
@@ -3977,6 +3982,7 @@ class LocalCodingAgent:
             message_id=(
                 f'compaction_replay_{len(compact_messages)}_{len(snipped_messages)}'
             ),
+            display=False,
         )
 
     def _render_compaction_replay(
@@ -4194,6 +4200,7 @@ class LocalCodingAgent:
                         'message_count': len(persist_messages),
                     },
                     message_id=f'plugin_persist_{result.session_id}',
+                    display=False,
                 )
                 persist_events.append(
                     {
@@ -4239,7 +4246,8 @@ class LocalCodingAgent:
             system_prompt_parts=session.system_prompt_parts,
             user_context=dict(session.user_context),
             system_context=dict(session.system_context),
-            messages=session.transcript(),
+            messages=session.model_transcript(),
+            display_messages=session.display_transcript(),
             turns=previous_turns + result.turns,
             tool_calls=previous_tool_calls + result.tool_calls,
             usage=result.usage.to_dict(),