virtuai-cli 0.7.5__tar.gz → 0.7.6__tar.gz

{virtuai_cli-0.7.5 → virtuai_cli-0.7.6}/PKG-INFO

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

{virtuai_cli-0.7.5 → virtuai_cli-0.7.6}/pyproject.toml

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

{virtuai_cli-0.7.5 → virtuai_cli-0.7.6}/src/virtuai_cli/__init__.py

@@ -1,2 +1,2 @@
 """VirtuAI local CLI."""
-__version__ = "0.7.5"
+__version__ = "0.7.6"

{virtuai_cli-0.7.5 → virtuai_cli-0.7.6}/src/virtuai_cli/chat/tui.py

@@ -176,10 +176,13 @@ class ChatApp(App):
         self._runner_task = asyncio.create_task(self._run_ws())
 
     def _tick_status(self) -> None:
+        # Only repaint when something is actually animating — otherwise we'd
+        # force a Textual reflow 10x/sec across the whole scrollback, which
+        # makes long conversations laggy.
         moving = self._streaming or self._connection_state in ("connecting", "reconnecting")
         if moving:
             self._spinner_idx = (self._spinner_idx + 1) % len(self._SPINNER)
-        self._refresh_status_bar()
+            self._refresh_status_bar()
 
     def _refresh_status_bar(self) -> None:
         try:

{virtuai_cli-0.7.5 → virtuai_cli-0.7.6}/src/virtuai_cli/chat/widgets.py

@@ -200,19 +200,52 @@ class UserBubble(Static):
 
 
 class TextSegment(Static):
-    """A streaming markdown segment of assistant text."""
+    """A streaming markdown segment of assistant text.
+
+    Renders are throttled to ~20 Hz: the markdown AST is rebuilt from the
+    full buffer on every paint (Rich/mistletoe parse from scratch), so
+    rendering once per token on a long response is O(n²) and dominates CPU.
+    With throttling, bursts of tokens coalesce into a single render and any
+    final pending render is flushed via mark_final()/flush().
+    """
 
     DEFAULT_CSS = """
     TextSegment { margin: 0 0 0 2; }
     """
 
+    _RENDER_INTERVAL = 0.05  # seconds (= 20 fps)
+
     def __init__(self) -> None:
         super().__init__("")
         self._buf = ""
+        self._last_render_t: float = 0.0
+        self._render_pending: bool = False
 
     def append(self, chunk: str) -> None:
         self._buf += chunk
-        self.update(Markdown(normalize_artifacts(self._buf)))
+        if self._render_pending:
+            # A flush is already scheduled and will pick up the new content.
+            return
+        now = time.monotonic()
+        elapsed = now - self._last_render_t
+        if elapsed >= self._RENDER_INTERVAL:
+            self._flush()
+        else:
+            self._render_pending = True
+            self.set_timer(self._RENDER_INTERVAL - elapsed, self._flush)
+
+    def flush(self) -> None:
+        """Force a render now — used to finalize the segment at end-of-stream."""
+        self._flush()
+
+    def _flush(self) -> None:
+        self._render_pending = False
+        self._last_render_t = time.monotonic()
+        try:
+            self.update(Markdown(normalize_artifacts(self._buf)))
+        except Exception:
+            # Widget might be unmounted by the time a scheduled flush fires.
+            pass
 
     @property
     def content(self) -> str:
@@ -585,4 +618,11 @@ class AssistantTurn(Vertical):
 
     async def mark_final(self) -> None:
         await self._stop_thinking()
+        # Force any throttled TextSegment to do its final render so the user
+        # doesn't see the last sub-50ms of tokens missing after stream-end.
+        for seg in self.query(TextSegment):
+            try:
+                seg.flush()
+            except Exception:
+                pass
         self._final = True

{virtuai_cli-0.7.5 → virtuai_cli-0.7.6}/src/virtuai_cli.egg-info/PKG-INFO

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