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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: virtuai-cli
-Version: 0.8.1
+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.1 → virtuai_cli-0.8.2}/pyproject.toml

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

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

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

{virtuai_cli-0.8.1 → 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.1 → 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.1
+Version: 0.8.2
 Summary: Run VirtuAI deep agents on your local machine
 Author-email: uCloudStore <lmoreno@ucloudstore.com>
 License: Proprietary