Simplify message mounting flow and remove pending queue

Pass the optional welcome message directly to the container, drop the unused `init_messages` parameter and pending‑mount list, and replace the `enqueue_mount_messages` pattern with immediate scheduling of `mount_messages`. Adjust `append_stream_message_content` to collect new messages and invoke mounting in one step.

Author: wrmsr

ommlds/cli/chat/interfaces/textual/drivers/interface.py

@@ -68,6 +68,7 @@ def __init__(
         self._background_terminal_renderer = background_terminal_renderer
         self._clipboard = clipboard
         self._state_listener = state_listener
+        self._welcome_message = welcome_message
         self._chat_id = chat_id
 
         #
@@ -85,7 +86,6 @@ def __init__(
         #
 
         self._messages_container = MessagesContainer(
-            [welcome_message] if welcome_message is not None else [],
             clipboard=self._clipboard,
             chat_uuid=chat_id.v,
         )
@@ -123,7 +123,8 @@ async def on_mount(self) -> None:
         check.state(self._chat_action_queue_task is None)
         self._chat_action_queue_task = asyncio.create_task(self._chat_action_queue_task_main())
 
-        await self._messages_container.mount_messages()
+        if (wm := self._welcome_message) is not None:
+            await self._messages_container.mount_messages(wm)
 
         self.call_after_refresh(self._chat_driver.start)
 
@@ -194,8 +195,7 @@ async def mount_messages(
             for w in wx:
                 self._suppressed_background_terminal_render_set.add(w)
 
-        await self._messages_container.enqueue_mount_messages(*wx)
-        self.call_later(self._messages_container.mount_messages)
+        self.call_later(self._messages_container.mount_messages, *wx)
 
     ##
     # Chat events
@@ -265,8 +265,7 @@ async def _handle_chat_queue_event(self, ev: ta.Any) -> None:
             )
 
             self._suppressed_background_terminal_render_set.add(tm)
-            await self._messages_container.enqueue_mount_messages(tm)
-            self.call_later(self._messages_container.mount_messages)
+            self.call_later(self._messages_container.mount_messages, tm)
 
         elif isinstance(ev, mc.ToolUseResultEvent):
             pass

ommlds/cli/chat/interfaces/textual/widgets/messages/container.py

@@ -25,8 +25,6 @@ class MessagesContainer(
 
     def __init__(
             self,
-            init_messages: ta.Sequence[Message] | None = None,
-            *,
             clipboard: cpb.Clipboard | None = None,
             chat_uuid: uuid.UUID | None = None,
     ) -> None:
@@ -37,9 +35,6 @@ def __init__(
 
         self._messages_by_uuid: dict[uuid.UUID, Message] = {}
 
-        if init_messages:
-            self._pending_mount_messages = list(init_messages)
-
     @property
     def chat_uuid(self) -> uuid.UUID | None:
         return self._chat_uuid
@@ -68,18 +63,8 @@ def _scroll_messages_to_bottom_and_anchor(self) -> None:
 
     #
 
-    _pending_mount_messages: list[Message] | None = None
-
-    async def enqueue_mount_messages(self, *messages: Message) -> None:
-        if (lst := self._pending_mount_messages) is None:
-            lst = self._pending_mount_messages = []
-
-        lst.extend(messages)
-
-    #
-
     async def append_stream_message_content(self, parts: ta.Sequence[StreamMessagePart]) -> None:
-        mount_messages = False
+        mount_messages: list[Message] = []
         refresh = False
         scroll_to_bottom = False
 
@@ -93,9 +78,7 @@ async def append_stream_message_content(self, parts: ta.Sequence[StreamMessagePa
 
                     self._messages_by_uuid[message_uuid] = aim
 
-                    await self.enqueue_mount_messages(aim)
-
-                    mount_messages = True
+                    mount_messages.append(aim)
                     refresh = True
 
                 else:
@@ -118,7 +101,7 @@ async def append_stream_message_content(self, parts: ta.Sequence[StreamMessagePa
                 raise TypeError(part)
 
         if mount_messages:
-            self.call_later(self.mount_messages)
+            self.call_later(self.mount_messages, *mount_messages)
         if scroll_to_bottom:
             self.call_after_refresh(self._scroll_messages_to_bottom_and_anchor)
         if refresh:
@@ -129,7 +112,7 @@ async def append_stream_message_content(self, parts: ta.Sequence[StreamMessagePa
     async def mount_messages(self, *messages: Message) -> None:
         was_at_bottom = self._is_messages_at_bottom()
 
-        for msg in [*(self._pending_mount_messages or []), *messages]:
+        for msg in messages:
             if (mu := msg.message_uuid) is not None:
                 try:
                     xm = self._messages_by_uuid[mu]
@@ -147,8 +130,6 @@ async def mount_messages(self, *messages: Message) -> None:
             if isinstance(msg, StreamMessage):
                 await msg.start_stream()
 
-        self._pending_mount_messages = None
-
         self.call_after_refresh(self._scroll_messages_to_bottom)
 
         if was_at_bottom: