virtuai-cli 0.8.0__tar.gz → 0.8.2__tar.gz

{virtuai_cli-0.8.0 → virtuai_cli-0.8.2}/PKG-INFO

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

{virtuai_cli-0.8.0 → virtuai_cli-0.8.2}/pyproject.toml

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

{virtuai_cli-0.8.0 → virtuai_cli-0.8.2}/src/virtuai_cli/__init__.py

@@ -1,2 +1,2 @@
 """VirtuAI local CLI."""
-__version__ = "0.8.0"
+__version__ = "0.8.2"

{virtuai_cli-0.8.0 → virtuai_cli-0.8.2}/src/virtuai_cli/chat/tui.py

@@ -718,6 +718,7 @@ class ChatApp(App):
     async def _stream_response(self, message: str, turn: AssistantTurn) -> None:
         self._streaming = True
         self._refresh_status_bar()
+        cancelled = False
         # Snapshot + clear pending attachments BEFORE the request so a new
         # /attach during streaming doesn't bleed into this turn or the next.
         attachments = list(self._pending_attachments)
@@ -742,7 +743,9 @@ class ChatApp(App):
                 await self._handle_event(event, turn)
                 self._follow_bottom()
         except asyncio.CancelledError:
-            # App may be shutting down — don't touch widgets here.
+            # User hit Esc (or app is shutting down). Skip the expensive
+            # mark_final flush — see widgets.AssistantTurn.mark_final.
+            cancelled = True
             raise
         except Exception as exc:
             try:
@@ -754,7 +757,7 @@ class ChatApp(App):
             self._streaming = False
             self._refresh_status_bar()
             try:
-                await turn.mark_final()
+                await turn.mark_final(cancelled=cancelled)
             except Exception:
                 pass
             try:

{virtuai_cli-0.8.0 → virtuai_cli-0.8.2}/src/virtuai_cli/chat/widgets.py

@@ -220,6 +220,7 @@ class TextSegment(Static):
         self._buf = ""
         self._last_render_t: float = 0.0
         self._render_pending: bool = False
+        self._render_timer: Any = None
 
     def append(self, chunk: str) -> None:
         self._buf += chunk
@@ -232,14 +233,28 @@ class TextSegment(Static):
             self._flush()
         else:
             self._render_pending = True
-            self.set_timer(self._RENDER_INTERVAL - elapsed, self._flush)
+            self._render_timer = 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 cancel_pending(self) -> None:
+        """Drop any scheduled render — used on stream cancel so the cancel
+        doesn't trigger another expensive Markdown reparse a moment later."""
+        if self._render_timer is not None:
+            try:
+                self._render_timer.stop()
+            except Exception:
+                pass
+            self._render_timer = None
+        self._render_pending = False
+
     def _flush(self) -> None:
         self._render_pending = False
+        self._render_timer = None
         self._last_render_t = time.monotonic()
         try:
             self.update(Markdown(normalize_artifacts(self._buf)))
@@ -616,13 +631,20 @@ class AssistantTurn(Vertical):
             await self.mount(self._todo)
         self._todo.set_items(items)
 
-    async def mark_final(self) -> None:
+    async def mark_final(self, *, cancelled: bool = False) -> 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.
+        # On graceful completion: do a final synchronous flush so the last
+        # sub-50ms of tokens definitely render. On cancel: SKIP the flush —
+        # building the markdown AST from a multi-KB buffer is a 100-500ms
+        # blocking spike on the event loop right when the user wants to
+        # type again, and trying to "polish" the last fragment of an
+        # interrupted response isn't worth the lag.
         for seg in self.query(TextSegment):
             try:
-                seg.flush()
+                if cancelled:
+                    seg.cancel_pending()
+                else:
+                    seg.flush()
             except Exception:
                 pass
         self._final = True

{virtuai_cli-0.8.0 → virtuai_cli-0.8.2}/src/virtuai_cli/runner.py

@@ -40,6 +40,7 @@ def _audit(workdir: Path, command: str, exit_code: int, elapsed: float) -> None:
         pass
 
 _RECONNECT_DELAYS = [1, 2, 4, 8, 16, 30]  # backoff steps
+_RECV_TIMEOUT = 60.0  # seconds of silence before declaring the connection stale
 
 
 def _default_on_status(level: str, msg: str) -> None:
@@ -139,7 +140,14 @@ async def run_session(
     # JSON pings every 20 s to keep the connection alive (cli_api/router.py).
     ssl_ctx = _ssl_context() if ws_url.startswith("wss://") else None
     async with websockets.connect(url, ping_interval=None, ssl=ssl_ctx) as ws:
-        async for raw in ws:
+        while True:
+            try:
+                raw = await asyncio.wait_for(ws.recv(), timeout=_RECV_TIMEOUT)
+            except asyncio.TimeoutError:
+                # No frame received within the timeout window — the server
+                # sends a JSON ping every ~20 s, so 60 s of silence means the
+                # TCP connection is stale without a clean close.
+                raise ConnectionError("No data received for 60 s — stale connection")
             try:
                 frame = json.loads(raw)
                 await _handle_frame(frame, workdir, ws, on_status)

{virtuai_cli-0.8.0 → virtuai_cli-0.8.2}/src/virtuai_cli.egg-info/PKG-INFO

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