virtuai-cli 0.5.2__tar.gz → 0.5.3__tar.gz

{virtuai_cli-0.5.2 → virtuai_cli-0.5.3}/PKG-INFO

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

{virtuai_cli-0.5.2 → virtuai_cli-0.5.3}/pyproject.toml

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

{virtuai_cli-0.5.2 → virtuai_cli-0.5.3}/src/virtuai_cli/__init__.py

@@ -1,2 +1,2 @@
 """VirtuAI local CLI."""
-__version__ = "0.5.2"
+__version__ = "0.5.3"

{virtuai_cli-0.5.2 → virtuai_cli-0.5.3}/src/virtuai_cli/chat/ask.py

@@ -107,6 +107,8 @@ async def _run_ask(
             server_url, token, agent_id, message,
             session_id=session_id, model_id=model_id,
         ):
+            if not isinstance(event, dict):
+                continue
             etype = event.get("type")
 
             if mode == "json":

{virtuai_cli-0.5.2 → virtuai_cli-0.5.3}/src/virtuai_cli/chat/sse.py

@@ -44,8 +44,20 @@ async def stream_chat(
             async for sse in event_source.aiter_sse():
                 if not sse.data:
                     continue
+                # Classic-agent path emits `data: [DONE]` after the stream;
+                # treat that as a done event for symmetry with the deep path.
+                if sse.data.strip() == "[DONE]":
+                    yield {"type": "done"}
+                    continue
                 try:
-                    yield json.loads(sse.data)
+                    parsed = json.loads(sse.data)
                 except json.JSONDecodeError:
-                    # Server occasionally emits non-JSON keepalive lines; skip.
                     continue
+                if isinstance(parsed, dict):
+                    yield parsed
+                elif isinstance(parsed, str):
+                    # Classic agents stream tokens as bare JSON strings
+                    # (`data: "hello "`). Wrap them so downstream handlers
+                    # don't need to know about the legacy protocol.
+                    yield {"type": "token", "content": parsed}
+                # Other shapes (lists, ints, null) are dropped silently.

{virtuai_cli-0.5.2 → virtuai_cli-0.5.3}/src/virtuai_cli/chat/tui.py

@@ -462,6 +462,10 @@ class ChatApp(App):
                 pass
 
     async def _handle_event(self, event: dict, turn: AssistantTurn) -> None:
+        # Defensive: a malformed server frame (string/list/None) used to crash
+        # the stream with "'str' object has no attribute 'get'".
+        if not isinstance(event, dict):
+            return
         etype = event.get("type")
         if etype == "start":
             sid = event.get("session_id")

{virtuai_cli-0.5.2 → virtuai_cli-0.5.3}/src/virtuai_cli.egg-info/PKG-INFO

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