ayechat-dev 0.36.8.20260126091750__tar.gz → 0.36.9.20260201141756__tar.gz

{ayechat_dev-0.36.8.20260126091750 → ayechat_dev-0.36.9.20260201141756}/.github/workflows/test-windows-installer.yml

@@ -45,7 +45,7 @@ jobs:
           Write-Host "Using workflow run ID: $runId"
 
       - name: Download installer artifact
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v7
         with:
           pattern: aye-chat-installer-*
           run-id: ${{ steps.get-run-id.outputs.RUN_ID }}

{ayechat_dev-0.36.8.20260126091750/src/ayechat_dev.egg-info → ayechat_dev-0.36.9.20260201141756}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ayechat-dev
-Version: 0.36.8.20260126091750
+Version: 0.36.9.20260201141756
 Summary: Aye Chat: Terminal-first AI Code Generator
 Author-email: "Acrotron, Inc." <info@acrotron.com>
 License: MIT

{ayechat_dev-0.36.8.20260126091750 → ayechat_dev-0.36.9.20260201141756}/pyproject.toml

@@ -8,7 +8,7 @@ description = "Aye Chat: Terminal-first AI Code Generator"
 readme = "README.md"
 requires-python = ">=3.10, <3.14"
 dependencies = [ "typer>=0.20.0", "httpx>=0.28.1", "keyring>=25.7.0", "prompt-toolkit>=3.0.52", "pathspec>=0.12.1", "chromadb>=1.3.5", "rapidfuzz",]
-version = "0.36.8.20260126091750"
+version = "0.36.9.20260201141756"
 [[project.authors]]
 name = "Acrotron, Inc."
 email = "info@acrotron.com"

{ayechat_dev-0.36.8.20260126091750 → ayechat_dev-0.36.9.20260201141756}/src/aye/controller/repl.py

@@ -1,7 +1,7 @@
 import os
 import json
 from pathlib import Path
-from typing import Optional, Any
+from typing import Optional, Any, List
 import shlex
 import threading
 import glob
@@ -252,6 +252,34 @@ def create_prompt_session(completer: Any, completion_style: str = "readline") ->
     )
 
 
+def _execute_forced_shell_command(command: str, args: List[str], conf: Any) -> None:
+    """Execute a shell command with force flag (bypasses command validation).
+    
+    Used when user prefixes input with '!' to force shell execution.
+    
+    Args:
+        command: The command to execute
+        args: List of arguments to pass to the command
+        conf: Configuration object with plugin_manager
+    """
+    telemetry.record_command(command, has_args=len(args) > 0, prefix=_CMD_PREFIX)
+    shell_response = conf.plugin_manager.handle_command(
+        "execute_shell_command", 
+        {"command": command, "args": args, "force": True}
+    )
+    if shell_response is not None:
+        if "stdout" in shell_response or "stderr" in shell_response:
+            if shell_response.get("stdout", "").strip():
+                rprint(shell_response["stdout"])
+            if shell_response.get("stderr", "").strip():
+                rprint(f"[yellow]{shell_response['stderr']}[/]")
+            if "error" in shell_response:
+                rprint(f"[red]Error:[/] {shell_response['error']}")
+        elif "message" in shell_response:
+            rprint(shell_response["message"])
+    else:
+        rprint(f"[red]Error:[/] Failed to execute shell command")
+
 
 def chat_repl(conf: Any) -> None:
     is_first_run = run_first_time_tutorial_if_needed()
@@ -327,6 +355,14 @@ def chat_repl(conf: Any) -> None:
                         chat_id = new_chat_id
                     continue
 
+                # Check for '!' prefix - force shell execution
+                force_shell = False
+                if prompt.strip().startswith('!'):
+                    force_shell = True
+                    prompt = prompt.strip()[1:]  # Remove the '!'
+                    if not prompt.strip():
+                        continue  # Nothing after the '!', skip
+
                 if not prompt.strip():
                     continue
                 tokens = shlex.split(prompt.strip(), posix=False)
@@ -340,6 +376,11 @@ def chat_repl(conf: Any) -> None:
 
             original_first, lowered_first = tokens[0], tokens[0].lower()
 
+            # If force_shell is True, execute as shell command directly and skip all other checks
+            if force_shell:
+                _execute_forced_shell_command(original_first, tokens[1:], conf)
+                continue
+
             # Normalize slash-prefixed commands: /restore -> restore, /model -> model, etc.
             if lowered_first.startswith('/'):
                 lowered_first = lowered_first[1:]  # Remove leading slash

{ayechat_dev-0.36.8.20260126091750 → ayechat_dev-0.36.9.20260201141756}/src/aye/plugins/shell_executor.py

@@ -10,7 +10,7 @@ from rich import print as rprint
 
 class ShellExecutorPlugin(Plugin):
     name = "shell_executor"
-    version = "1.0.2"  # Fixed Windows localization bug in command detection
+    version = "1.0.3"  # Added force parameter for ! prefix execution
     premium = "free"
 
     # Known interactive commands that require a TTY (add more as needed)
@@ -143,15 +143,17 @@ class ShellExecutorPlugin(Plugin):
                 "returncode": e.returncode
             }
         except FileNotFoundError:
-            return None  # Command not found
+            return {"error": f"Command not found: {command}", "stdout": "", "stderr": "", "returncode": 127}
 
     def on_command(self, command_name: str, params: Dict[str, Any]) -> Optional[Dict[str, Any]]:
         """Handle shell command execution through plugin system."""
         if command_name == "execute_shell_command":
             command = params.get("command", "")
             args = params.get("args", [])
+            force = params.get("force", False)
             
-            if not self._is_valid_command(command):
+            # If force is False, validate the command exists first
+            if not force and not self._is_valid_command(command):
                 return None  # Command not found or not executable
             
             full_cmd_str = self._build_full_cmd(command, args)

{ayechat_dev-0.36.8.20260126091750 → ayechat_dev-0.36.9.20260201141756}/src/aye/presenter/repl_ui.py

@@ -87,7 +87,7 @@ def print_help_message():
     commands = [
         # Some commands are intentionally undocumented: keep them as such.
         ("@filename", "Include a file in your prompt inline (e.g., \"explain @main.py\"). Supports wildcards (e.g., @*.py, @src/*.js)."),
-        ("blog <intent>", "Generate a technical deep-dive blog post derived from the current chat session and write it to blog.md."),
+        ("!command", "Force shell execution (e.g., \"!echo hello\")."),
         ("model", "Select a different model. Selection will persist between sessions."),
         (r"verbose \[on|off]", "Toggle verbose mode to increase or decrease chattiness (on/off, persists between sessions)"),
         (r"completion \[readline|multi]", "Switch auto-completion style (readline or multi, persists between sessions)"),

{ayechat_dev-0.36.8.20260126091750 → ayechat_dev-0.36.9.20260201141756/src/ayechat_dev.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ayechat-dev
-Version: 0.36.8.20260126091750
+Version: 0.36.9.20260201141756
 Summary: Aye Chat: Terminal-first AI Code Generator
 Author-email: "Acrotron, Inc." <info@acrotron.com>
 License: MIT

{ayechat_dev-0.36.8.20260126091750 → ayechat_dev-0.36.9.20260201141756}/tests/test_repl.py

@@ -332,6 +332,241 @@ class TestRepl(TestCase):
             mock_process.assert_called_once()
 
 
+class TestExecuteForcedShellCommand(TestCase):
+    """Tests for _execute_forced_shell_command function."""
+
+    def setUp(self):
+        repl.telemetry.reset()
+        repl.telemetry.set_enabled(False)
+
+    @patch("aye.controller.repl.rprint")
+    @patch("aye.controller.repl.telemetry.record_command")
+    def test_executes_with_force_flag_and_prints_stdout(self, mock_record, mock_rprint):
+        """Test that command executes with force=True and stdout is printed."""
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.return_value = {
+            "stdout": "file1.txt\nfile2.txt",
+            "stderr": "",
+        }
+        conf = SimpleNamespace(plugin_manager=plugin_manager)
+
+        repl._execute_forced_shell_command("ls", ["-la"], conf)
+
+        plugin_manager.handle_command.assert_called_once_with(
+            "execute_shell_command",
+            {"command": "ls", "args": ["-la"], "force": True}
+        )
+        mock_rprint.assert_called_once_with("file1.txt\nfile2.txt")
+
+    @patch("aye.controller.repl.rprint")
+    @patch("aye.controller.repl.telemetry.record_command")
+    def test_prints_stderr_with_yellow_formatting(self, mock_record, mock_rprint):
+        """Test that stderr is printed with yellow formatting."""
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.return_value = {
+            "stdout": "",
+            "stderr": "warning: something happened",
+        }
+        conf = SimpleNamespace(plugin_manager=plugin_manager)
+
+        repl._execute_forced_shell_command("cmd", [], conf)
+
+        mock_rprint.assert_called_once_with("[yellow]warning: something happened[/]")
+
+    @patch("aye.controller.repl.rprint")
+    @patch("aye.controller.repl.telemetry.record_command")
+    def test_prints_both_stdout_and_stderr(self, mock_record, mock_rprint):
+        """Test that both stdout and stderr are printed when present."""
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.return_value = {
+            "stdout": "output here",
+            "stderr": "warning here",
+        }
+        conf = SimpleNamespace(plugin_manager=plugin_manager)
+
+        repl._execute_forced_shell_command("cmd", ["arg1"], conf)
+
+        assert mock_rprint.call_count == 2
+        mock_rprint.assert_any_call("output here")
+        mock_rprint.assert_any_call("[yellow]warning here[/]")
+
+    @patch("aye.controller.repl.rprint")
+    @patch("aye.controller.repl.telemetry.record_command")
+    def test_prints_error_with_red_formatting(self, mock_record, mock_rprint):
+        """Test that error in response is printed with red formatting."""
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.return_value = {
+            "stdout": "partial output",
+            "stderr": "",
+            "error": "Command failed with exit code 1",
+        }
+        conf = SimpleNamespace(plugin_manager=plugin_manager)
+
+        repl._execute_forced_shell_command("failing_cmd", [], conf)
+
+        assert mock_rprint.call_count == 2
+        mock_rprint.assert_any_call("partial output")
+        mock_rprint.assert_any_call("[red]Error:[/] Command failed with exit code 1")
+
+    @patch("aye.controller.repl.rprint")
+    @patch("aye.controller.repl.telemetry.record_command")
+    def test_prints_message_for_interactive_commands(self, mock_record, mock_rprint):
+        """Test that message-only responses (e.g., interactive commands) are printed."""
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.return_value = {
+            "message": "Interactive command 'vim' completed (exit code: 0)."
+        }
+        conf = SimpleNamespace(plugin_manager=plugin_manager)
+
+        repl._execute_forced_shell_command("vim", ["file.txt"], conf)
+
+        mock_rprint.assert_called_once_with(
+            "Interactive command 'vim' completed (exit code: 0)."
+        )
+
+    @patch("aye.controller.repl.rprint")
+    @patch("aye.controller.repl.telemetry.record_command")
+    def test_prints_error_when_plugin_returns_none(self, mock_record, mock_rprint):
+        """Test that error is printed when plugin returns None."""
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.return_value = None
+        conf = SimpleNamespace(plugin_manager=plugin_manager)
+
+        repl._execute_forced_shell_command("nonexistent", [], conf)
+
+        mock_rprint.assert_called_once_with("[red]Error:[/] Failed to execute shell command")
+
+    @patch("aye.controller.repl.rprint")
+    @patch("aye.controller.repl.telemetry.record_command")
+    def test_records_telemetry_with_args(self, mock_record, mock_rprint):
+        """Test that telemetry is recorded with has_args=True when args present."""
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.return_value = {"stdout": "ok", "stderr": ""}
+        conf = SimpleNamespace(plugin_manager=plugin_manager)
+
+        repl._execute_forced_shell_command("git", ["status", "-s"], conf)
+
+        mock_record.assert_called_once_with("git", has_args=True, prefix="cmd:")
+
+    @patch("aye.controller.repl.rprint")
+    @patch("aye.controller.repl.telemetry.record_command")
+    def test_records_telemetry_without_args(self, mock_record, mock_rprint):
+        """Test that telemetry is recorded with has_args=False when no args."""
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.return_value = {"stdout": "ok", "stderr": ""}
+        conf = SimpleNamespace(plugin_manager=plugin_manager)
+
+        repl._execute_forced_shell_command("pwd", [], conf)
+
+        mock_record.assert_called_once_with("pwd", has_args=False, prefix="cmd:")
+
+    @patch("aye.controller.repl.rprint")
+    @patch("aye.controller.repl.telemetry.record_command")
+    def test_does_not_print_empty_stdout_or_stderr(self, mock_record, mock_rprint):
+        """Test that empty stdout/stderr strings don't result in prints."""
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.return_value = {
+            "stdout": "   ",  # whitespace only
+            "stderr": "",
+        }
+        conf = SimpleNamespace(plugin_manager=plugin_manager)
+
+        repl._execute_forced_shell_command("cmd", [], conf)
+
+        mock_rprint.assert_not_called()
+
+    @patch("aye.controller.repl.rprint")
+    @patch("aye.controller.repl.telemetry.record_command")
+    def test_handles_stdout_stderr_and_error_together(self, mock_record, mock_rprint):
+        """Test all three outputs printed when all present."""
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.return_value = {
+            "stdout": "some output",
+            "stderr": "some warning",
+            "error": "but it failed",
+        }
+        conf = SimpleNamespace(plugin_manager=plugin_manager)
+
+        repl._execute_forced_shell_command("cmd", ["arg"], conf)
+
+        assert mock_rprint.call_count == 3
+        mock_rprint.assert_any_call("some output")
+        mock_rprint.assert_any_call("[yellow]some warning[/]")
+        mock_rprint.assert_any_call("[red]Error:[/] but it failed")
+
+
+def test_chat_repl_force_shell_prefix_executes_via_forced_function():
+    """Test that ! prefix triggers _execute_forced_shell_command in REPL."""
+    with (
+        patch("aye.controller.repl.PromptSession") as mock_session_cls,
+        patch("aye.controller.repl.run_first_time_tutorial_if_needed", return_value=False),
+        patch("aye.controller.repl._prompt_for_telemetry_consent_if_needed", return_value=False),
+        patch("aye.controller.repl.print_startup_header"),
+        patch("aye.controller.repl.print_prompt", return_value="> "),
+        patch("aye.controller.repl.Path") as mock_path,
+        patch("aye.controller.repl._execute_forced_shell_command") as mock_forced,
+        patch("aye.controller.repl.collect_and_send_feedback"),
+    ):
+        session = MagicMock()
+        session.prompt.side_effect = ["!mycommand arg1 arg2", "exit"]
+        mock_session_cls.return_value = session
+
+        _setup_mock_chat_id_path(mock_path)
+
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.side_effect = [{"completer": None}]
+
+        conf = SimpleNamespace(
+            root=Path.cwd(),
+            file_mask="*.py",
+            plugin_manager=plugin_manager,
+            index_manager=None,
+            verbose=False,
+            selected_model="model",
+            use_rag=True,
+        )
+
+        repl.chat_repl(conf)
+
+        mock_forced.assert_called_once_with("mycommand", ["arg1", "arg2"], conf)
+
+
+def test_chat_repl_force_shell_empty_after_bang_skips():
+    """Test that '!' alone is skipped without error."""
+    with (
+        patch("aye.controller.repl.PromptSession") as mock_session_cls,
+        patch("aye.controller.repl.run_first_time_tutorial_if_needed", return_value=False),
+        patch("aye.controller.repl._prompt_for_telemetry_consent_if_needed", return_value=False),
+        patch("aye.controller.repl.print_startup_header"),
+        patch("aye.controller.repl.print_prompt", return_value="> "),
+        patch("aye.controller.repl.Path") as mock_path,
+        patch("aye.controller.repl._execute_forced_shell_command") as mock_forced,
+        patch("aye.controller.repl.collect_and_send_feedback"),
+    ):
+        session = MagicMock()
+        session.prompt.side_effect = ["!", "!   ", "exit"]
+        mock_session_cls.return_value = session
+
+        _setup_mock_chat_id_path(mock_path)
+
+        plugin_manager = MagicMock()
+        plugin_manager.handle_command.side_effect = [{"completer": None}]
+
+        conf = SimpleNamespace(
+            root=Path.cwd(),
+            file_mask="*.py",
+            plugin_manager=plugin_manager,
+            index_manager=None,
+            verbose=False,
+            selected_model="model",
+            use_rag=True,
+        )
+
+        repl.chat_repl(conf)
+
+        mock_forced.assert_not_called()
+
+
 def test_create_key_bindings_registers_two_enter_bindings_and_handlers_work():
     bindings = repl.create_key_bindings()
     enter_bindings = bindings.get_bindings_for_keys((Keys.Enter,))

{ayechat_dev-0.36.8.20260126091750 → ayechat_dev-0.36.9.20260201141756}/tests/test_shell_executor_plugin.py

@@ -153,7 +153,7 @@ class TestShellExecutorPlugin(TestCase):
     @patch('subprocess.run', side_effect=FileNotFoundError)
     def test_execute_non_interactive_file_not_found(self, mock_run):
         result = self.plugin._execute_non_interactive('badcmd', [])
-        self.assertIsNone(result)
+        self.assertEqual(result, {'error': 'Command not found: badcmd', 'stdout': '', 'stderr': '', 'returncode': 127})
 
     @patch.object(ShellExecutorPlugin, '_is_valid_command', return_value=True)
     @patch.object(ShellExecutorPlugin, '_execute_non_interactive')