virtuai-cli 0.7.1__tar.gz → 0.7.2__tar.gz

{virtuai_cli-0.7.1 → virtuai_cli-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: virtuai-cli
-Version: 0.7.1
+Version: 0.7.2
 Summary: Run VirtuAI deep agents on your local machine
 Author-email: uCloudStore <lmoreno@ucloudstore.com>
 License: Proprietary

{virtuai_cli-0.7.1 → virtuai_cli-0.7.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "virtuai-cli"
-version = "0.7.1"
+version = "0.7.2"
 description = "Run VirtuAI deep agents on your local machine"
 readme = "README.md"
 requires-python = ">=3.11"

{virtuai_cli-0.7.1 → virtuai_cli-0.7.2}/src/virtuai_cli/__init__.py

@@ -1,2 +1,2 @@
 """VirtuAI local CLI."""
-__version__ = "0.7.1"
+__version__ = "0.7.2"

{virtuai_cli-0.7.1 → virtuai_cli-0.7.2}/src/virtuai_cli/chat/tui.py

@@ -379,17 +379,53 @@ class ChatApp(App):
         )
         await scroll.mount(Static(header))
 
+        loaded_session = self.session_id
         for msg in messages:
             mtype = (msg.get("message_type") or "").lower()
             content = msg.get("content") or ""
+            steps = msg.get("steps") or []
             if mtype in ("user", "human"):
                 await scroll.mount(UserBubble(content))
-            else:
-                turn = AssistantTurn(show_thinking=False)
-                await scroll.mount(turn)
-                if content:
-                    await turn.append_token(content)
-                await turn.mark_final()
+                continue
+
+            turn = AssistantTurn(show_thinking=False)
+            await scroll.mount(turn)
+
+            replayed_visible = False
+            if isinstance(steps, list) and steps:
+                # Replay the original event stream through the same handler
+                # the live stream uses, so tool cards, file diffs, todos,
+                # and subagent traces all re-render identically to when
+                # the turn first ran. Skip lifecycle events that mutate
+                # app state (session_id) or close the stream.
+                for step in steps:
+                    if not isinstance(step, dict):
+                        continue
+                    etype = step.get("type")
+                    if etype in ("start", "done", "complete"):
+                        continue
+                    try:
+                        await self._handle_event(step, turn)
+                        if etype in (
+                            "token", "text_segment", "tool_start", "tool_end",
+                            "todos", "subagent_start", "subagent_end",
+                            "memory_updated", "skill_loaded",
+                        ):
+                            replayed_visible = True
+                    except Exception:
+                        # One bad step shouldn't break the whole replay.
+                        continue
+
+            # Fall back to the persisted final text if there were no steps
+            # (older messages saved before steps tracking) or the steps
+            # rendered nothing visible.
+            if not replayed_visible and content:
+                await turn.append_token(content)
+            await turn.mark_final()
+
+        # Replay's `start` events were skipped, but in case anything else
+        # touched session_id, restore it to the conversation we loaded.
+        self.session_id = loaded_session
         scroll.scroll_end(animate=False)
         self._set_status(f"loaded {self.session_id}")
 

{virtuai_cli-0.7.1 → virtuai_cli-0.7.2}/src/virtuai_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: virtuai-cli
-Version: 0.7.1
+Version: 0.7.2
 Summary: Run VirtuAI deep agents on your local machine
 Author-email: uCloudStore <lmoreno@ucloudstore.com>
 License: Proprietary