virtuai-cli 0.6.2__tar.gz → 0.7.0__tar.gz

{virtuai_cli-0.6.2 → virtuai_cli-0.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: virtuai-cli
-Version: 0.6.2
+Version: 0.7.0
 Summary: Run VirtuAI deep agents on your local machine
 Author-email: uCloudStore <lmoreno@ucloudstore.com>
 License: Proprietary
@@ -19,7 +19,7 @@ Requires-Dist: certifi>=2024.0
 Requires-Dist: keyring>=25.0
 Requires-Dist: typer>=0.12
 Requires-Dist: rich>=13.0
-Requires-Dist: textual>=0.50
+Requires-Dist: textual>=0.86
 
 # VirtuAI CLI
 

{virtuai_cli-0.6.2 → virtuai_cli-0.7.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "virtuai-cli"
-version = "0.6.2"
+version = "0.7.0"
 description = "Run VirtuAI deep agents on your local machine"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -27,7 +27,7 @@ dependencies = [
     "keyring>=25.0",
     "typer>=0.12",
     "rich>=13.0",
-    "textual>=0.50",
+    "textual>=0.86",
 ]
 
 [project.urls]

{virtuai_cli-0.6.2 → virtuai_cli-0.7.0}/src/virtuai_cli/__init__.py

@@ -1,2 +1,2 @@
 """VirtuAI local CLI."""
-__version__ = "0.6.2"
+__version__ = "0.7.0"

{virtuai_cli-0.6.2 → virtuai_cli-0.7.0}/src/virtuai_cli/chat/tui.py

@@ -11,12 +11,12 @@ from textual.app import App, ComposeResult
 from textual.binding import Binding
 from textual.containers import Vertical, VerticalScroll
 from textual.reactive import reactive
-from textual.widgets import Footer, Header, Input, Static
+from textual.widgets import Footer, Header, Static, TextArea
 
 from virtuai_cli import runner as ws_runner
 from virtuai_cli.chat.history import list_conversations, load_conversation
 from virtuai_cli.chat.sse import stream_chat
-from virtuai_cli.chat.widgets import AssistantTurn, UserBubble
+from virtuai_cli.chat.widgets import AssistantTurn, ChatInput, TextSegment, UserBubble
 
 
 # Slash command catalog — single source of truth used by /help, the
@@ -25,6 +25,7 @@ SLASH_COMMANDS: list[tuple[str, str]] = [
     ("/help", "show this list"),
     ("/clear", "start a fresh conversation"),
     ("/new", "alias for /clear"),
+    ("/copy", "copy the last assistant response to the clipboard"),
     ("/history", "list this agent's recent conversations"),
     ("/load", "load a past conversation: /load <session_id>"),
     ("/models", "list models available for this agent"),
@@ -63,12 +64,6 @@ class ChatApp(App):
 
     #slash-hints.-visible { display: block; }
 
-    #input {
-        height: 3;
-        border: round $primary;
-        margin: 0 1 1 1;
-    }
-
     #placeholder {
         margin: 1 2;
         color: $text-muted;
@@ -79,6 +74,7 @@ class ChatApp(App):
         Binding("escape", "cancel_stream", "Cancel", show=True),
         Binding("ctrl+c", "quit", "Quit", show=True),
         Binding("ctrl+l", "clear_conversation", "New chat", show=True),
+        Binding("ctrl+y", "copy_last", "Copy last", show=True, priority=True),
         Binding("tab", "complete_slash", "Complete", show=False, priority=True),
     ]
 
@@ -119,11 +115,11 @@ class ChatApp(App):
             yield Static(
                 f"[b]{self.agent_name}[/b] in {self.workspace_name}\n"
                 f"Workdir: {self.workdir}\n"
-                f"[dim]Type a message and press Enter. Esc cancels. /help for commands.[/dim]",
+                f"[dim]Enter sends · Shift+Enter newline · Esc cancels · Ctrl+Y copies · /help for more.[/dim]",
                 id="placeholder",
             )
         yield Static("", id="slash-hints")
-        yield Input(placeholder="Message…", id="input")
+        yield ChatInput(id="input")
         yield Footer()
 
     def _initial_status(self) -> str:
@@ -134,7 +130,7 @@ class ChatApp(App):
     async def on_mount(self) -> None:
         self.title = "VirtuAI"
         self.sub_title = f"{self.agent_name} · {self.workspace_name}"
-        self.query_one(Input).focus()
+        self.query_one(ChatInput).focus()
         self._runner_task = asyncio.create_task(self._run_ws())
 
     async def _run_ws(self) -> None:
@@ -190,25 +186,28 @@ class ChatApp(App):
         hints.update("\n".join(rows))
         hints.add_class("-visible")
 
-    @on(Input.Changed, "#input")
-    def _on_input_changed(self, event: Input.Changed) -> None:
-        self._refresh_slash_hints(event.value)
+    @on(TextArea.Changed, "#input")
+    def _on_input_changed(self, event: TextArea.Changed) -> None:
+        # Slash hints only fire while the user is typing the FIRST line
+        # — long multi-line messages shouldn't trigger the panel.
+        value = event.text_area.text
+        first_line = value.split("\n", 1)[0]
+        self._refresh_slash_hints(first_line if "\n" not in value else "")
 
     def action_complete_slash(self) -> None:
         """Tab — complete the current slash command from the hint list."""
         try:
-            inp = self.query_one("#input", Input)
+            inp = self.query_one("#input", ChatInput)
         except Exception:
             return
-        value = inp.value
-        if not value.startswith("/") or " " in value:
-            return  # not a command-completion context — let Tab default
+        value = inp.text
+        if "\n" in value or not value.startswith("/") or " " in value:
+            return
         matches = self._matching_commands(value)
         if not matches:
             return
         if len(matches) == 1:
-            inp.value = matches[0][0] + " "
-            inp.cursor_position = len(inp.value)
+            inp.text = matches[0][0] + " "
         else:
             # Multiple matches: extend to common prefix
             names = [m[0] for m in matches]
@@ -219,20 +218,21 @@ class ChatApp(App):
                     i += 1
                 common = common[:i]
             if len(common) > len(value):
-                inp.value = common
-                inp.cursor_position = len(common)
+                inp.text = common
 
     # ── Input submission ──────────────────────────────────────────────────
-    @on(Input.Submitted, "#input")
-    async def _on_submit(self, event: Input.Submitted) -> None:
-        text = event.value.strip()
-        if not text:
+    @on(ChatInput.Submitted, "#input")
+    async def _on_submit(self, event: ChatInput.Submitted) -> None:
+        # Don't strip leading/trailing whitespace blindly — newlines inside
+        # the message are intentional. Only reject pure-whitespace inputs.
+        text = event.value
+        if not text.strip():
             return
-        event.input.value = ""
         self._refresh_slash_hints("")  # hide hints after submit
 
-        if text.startswith("/"):
-            await self._handle_slash(text)
+        first_line = text.split("\n", 1)[0].strip()
+        if first_line.startswith("/") and "\n" not in text:
+            await self._handle_slash(first_line)
             return
 
         if self._stream_task and not self._stream_task.done():
@@ -270,17 +270,33 @@ class ChatApp(App):
             await self._show_history()
         elif head == "/load":
             await self._load_session(arg)
+        elif head == "/copy":
+            self.action_copy_last()
         elif head == "/help":
             await self._append(Static(
                 "[b]Commands[/b]\n"
                 "  /help              this list\n"
                 "  /clear, /new       start a fresh conversation\n"
+                "  /copy              copy the last assistant response to clipboard\n"
                 "  /history           list this agent's recent conversations\n"
                 "  /load <id>         load a past conversation by session_id\n"
                 "  /models            list models available for this agent\n"
                 "  /model <id>        switch the model for the next message\n"
                 "  /exit, /quit       close the TUI\n"
+                "\n"
+                "[b]Keys[/b]\n"
+                "  Enter              send message\n"
+                "  Shift/Ctrl+Enter   newline inside message\n"
                 "  Esc                cancel current response\n"
+                "  Ctrl+L             new conversation\n"
+                "  Ctrl+Y             copy last assistant response\n"
+                "  Tab                autocomplete slash command\n"
+                "\n"
+                "[b]Selecting text[/b]\n"
+                "  The TUI captures the mouse, so click-drag doesn't select text.\n"
+                "  Hold [b]Option[/b] (macOS Terminal/iTerm2) or [b]Shift[/b] (most Linux\n"
+                "  terminals) while dragging to bypass the TUI and use your\n"
+                "  terminal's native selection + copy."
             ))
         else:
             await self._append(Static(f"[red]Unknown command: {head}[/red]  (try /help)"))
@@ -416,6 +432,28 @@ class ChatApp(App):
             self._stream_task.cancel()
             self._set_status("response cancelled")
 
+    def action_copy_last(self) -> None:
+        """Copy the most recent assistant turn's text to the system clipboard."""
+        turns = list(self.query(AssistantTurn))
+        if not turns:
+            self._set_status("no assistant response yet")
+            return
+        # Walk newest-first so a /copy right after a streamed reply picks up
+        # the response the user is looking at, even if a turn is empty.
+        for turn in reversed(turns):
+            segments = list(turn.query(TextSegment))
+            text = "\n\n".join(s.content for s in segments if s.content).strip()
+            if text:
+                try:
+                    self.copy_to_clipboard(text)
+                except Exception as exc:
+                    self._set_status(f"clipboard error: {exc}")
+                    return
+                preview = text[:60].replace("\n", " ")
+                self._set_status(f"copied: {preview}{'…' if len(text) > 60 else ''}")
+                return
+        self._set_status("no text to copy in the last response")
+
     def _follow_bottom(self) -> None:
         """Scroll to the end of the conversation.
 

{virtuai_cli-0.6.2 → virtuai_cli-0.7.0}/src/virtuai_cli/chat/widgets.py

@@ -15,8 +15,68 @@ from typing import Any, Optional
 
 from rich.markdown import Markdown
 from rich.text import Text
+from textual.binding import Binding
 from textual.containers import Vertical
-from textual.widgets import Static
+from textual.message import Message
+from textual.widgets import Static, TextArea
+
+
+class ChatInput(TextArea):
+    """Multi-line input — Enter submits, Shift/Ctrl+Enter inserts a newline.
+
+    Height grows with content up to a cap so a long paste doesn't take over
+    the whole screen; once the cap is hit the widget scrolls internally.
+    """
+
+    DEFAULT_CSS = """
+    ChatInput {
+        height: auto;
+        min-height: 3;
+        max-height: 14;
+        border: round $primary;
+        margin: 0 1 1 1;
+        padding: 0 1;
+    }
+    """
+
+    BINDINGS = [
+        Binding("enter", "submit", "Submit", show=False, priority=True),
+        Binding("shift+enter", "insert_newline", "Newline", show=False, priority=True),
+        Binding("ctrl+enter", "insert_newline", "Newline", show=False, priority=True),
+        Binding("ctrl+j", "insert_newline", "Newline", show=False, priority=True),
+    ]
+
+    class Submitted(Message):
+        """Posted when the user hits Enter on a non-empty buffer."""
+        def __init__(self, input_widget: "ChatInput", value: str) -> None:
+            self.input = input_widget
+            self.value = value
+            super().__init__()
+
+        @property
+        def control(self) -> "ChatInput":
+            # Lets `@on(ChatInput.Submitted, "#input")` selectors work.
+            return self.input
+
+    def __init__(self, **kwargs: Any) -> None:
+        super().__init__(**kwargs)
+        self.show_line_numbers = False
+        # We rely on Textual's soft-wrap so long single lines wrap visually
+        # instead of overflowing the input box.
+        try:
+            self.soft_wrap = True  # type: ignore[attr-defined]
+        except Exception:
+            pass
+
+    def action_submit(self) -> None:
+        text = self.text
+        if not text.strip():
+            return
+        self.text = ""
+        self.post_message(self.Submitted(self, text))
+
+    def action_insert_newline(self) -> None:
+        self.insert("\n")
 
 
 # Server-side displays that should render as an EditPreviewCard (path + diff)

{virtuai_cli-0.6.2 → virtuai_cli-0.7.0}/src/virtuai_cli/executor.py

@@ -11,6 +11,37 @@ from typing import Optional
 from virtuai_cli.security import check_command, jail_path, scrub_env
 
 _EXECUTE_TIMEOUT = 300  # seconds
+# Hard cap on the combined stdout+stderr we ship back over the WebSocket.
+# Anything larger gets head/tail-clipped with a marker so a runaway command
+# (e.g. `cat /dev/urandom`, `find /`) can't OOM the runner or blow past the
+# server-side frame-size limit and kill the connection.
+_MAX_OUTPUT_BYTES = 2_000_000  # ~2 MB
+
+
+def _clip_output(stdout: bytes, stderr: bytes) -> str:
+    """Decode + concatenate stdout/stderr, head/tail-clipping if oversized."""
+    total = len(stdout) + len(stderr)
+    if total <= _MAX_OUTPUT_BYTES:
+        parts = []
+        if stdout:
+            parts.append(stdout.decode(errors="replace"))
+        if stderr:
+            parts.append(stderr.decode(errors="replace"))
+        return "\n".join(parts)
+
+    # Keep half the budget at the start and half at the end of the combined
+    # text — that's where users usually look for context vs error tails.
+    combined = (stdout + (b"\n" if stdout and stderr else b"") + stderr)
+    keep = _MAX_OUTPUT_BYTES // 2
+    head = combined[:keep].decode(errors="replace")
+    tail = combined[-keep:].decode(errors="replace")
+    dropped = len(combined) - 2 * keep
+    return (
+        f"{head}\n"
+        f"\n... [output truncated: {dropped:,} bytes dropped, "
+        f"{len(combined):,} bytes total] ...\n\n"
+        f"{tail}"
+    )
 
 
 def _jail_wrap(command: str, workdir: Path) -> str:
@@ -49,12 +80,10 @@ def execute(command: str, workdir: Path, timeout: int = _EXECUTE_TIMEOUT, extra_
             capture_output=True,
             timeout=timeout,
         )
-        parts = []
-        if result.stdout:
-            parts.append(result.stdout.decode(errors="replace"))
-        if result.stderr:
-            parts.append(result.stderr.decode(errors="replace"))
-        return {"output": "\n".join(parts), "exit_code": result.returncode}
+        return {
+            "output": _clip_output(result.stdout or b"", result.stderr or b""),
+            "exit_code": result.returncode,
+        }
     except subprocess.TimeoutExpired:
         return {"output": f"command timed out after {timeout}s", "exit_code": 124}
     except Exception as exc:

{virtuai_cli-0.6.2 → virtuai_cli-0.7.0}/src/virtuai_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: virtuai-cli
-Version: 0.6.2
+Version: 0.7.0
 Summary: Run VirtuAI deep agents on your local machine
 Author-email: uCloudStore <lmoreno@ucloudstore.com>
 License: Proprietary
@@ -19,7 +19,7 @@ Requires-Dist: certifi>=2024.0
 Requires-Dist: keyring>=25.0
 Requires-Dist: typer>=0.12
 Requires-Dist: rich>=13.0
-Requires-Dist: textual>=0.50
+Requires-Dist: textual>=0.86
 
 # VirtuAI CLI
 

{virtuai_cli-0.6.2 → virtuai_cli-0.7.0}/src/virtuai_cli.egg-info/requires.txt

@@ -5,4 +5,4 @@ certifi>=2024.0
 keyring>=25.0
 typer>=0.12
 rich>=13.0
-textual>=0.50
+textual>=0.86