tunacode-cli 0.0.11__py3-none-any.whl → 0.0.13__py3-none-any.whl

tunacode/configuration/settings.py

@@ -1,8 +1,8 @@
 """
 Module: tunacode.configuration.settings
 
-Application settings management for the Sidekick CLI.
-Manages application paths, tool configurations, and runtime settings.
+Application settings management for the TunaCode CLI.
+Handles configuration paths, model registries, and application metadata.
 """
 
 from pathlib import Path

tunacode/constants.py

@@ -7,7 +7,7 @@ Centralizes all magic strings, UI text, error messages, and application constant
 
 # Application info
 APP_NAME = "TunaCode"
-APP_VERSION = "0.0.11"
+APP_VERSION = "0.0.13"
 
 # File patterns
 GUIDE_FILE_PATTERN = "{name}.md"
@@ -35,7 +35,6 @@ CMD_HELP = "/help"
 CMD_CLEAR = "/clear"
 CMD_DUMP = "/dump"
 CMD_YOLO = "/yolo"
-CMD_UNDO = "/undo"
 CMD_COMPACT = "/compact"
 CMD_MODEL = "/model"
 CMD_EXIT = "exit"
@@ -46,7 +45,6 @@ DESC_HELP = "Show this help message"
 DESC_CLEAR = "Clear the conversation history"
 DESC_DUMP = "Show the current conversation history"
 DESC_YOLO = "Toggle confirmation prompts on/off"
-DESC_UNDO = "Undo the last file change"
 DESC_COMPACT = "Summarize the conversation context"
 DESC_MODEL = "List available models"
 DESC_MODEL_SWITCH = "Switch to a specific model"
@@ -56,7 +54,7 @@ DESC_EXIT = "Exit the application"
 # Command Configuration
 COMMAND_PREFIX = "/"
 COMMAND_CATEGORIES = {
-    "state": ["yolo", "undo"],
+    "state": ["yolo"],
     "debug": ["dump", "compact"],
     "ui": ["clear", "help"],
     "config": ["model"],
@@ -110,7 +108,6 @@ ERROR_COMMAND_NOT_FOUND = "Error: Command not found or failed to execute:"
 ERROR_COMMAND_EXECUTION = (
     "Error: Command not found or failed to execute: {command}. Details: {error}"
 )
-ERROR_UNDO_INIT = "Error initializing undo system: {e}"
 
 # Command output messages
 CMD_OUTPUT_NO_OUTPUT = "No output."
@@ -118,11 +115,6 @@ CMD_OUTPUT_NO_ERRORS = "No errors."
 CMD_OUTPUT_FORMAT = "STDOUT:\n{output}\n\nSTDERR:\n{error}"
 CMD_OUTPUT_TRUNCATED = "\n...\n[truncated]\n...\n"
 
-# Undo system messages
-UNDO_DISABLED_HOME = "Undo system disabled, running from home directory"
-UNDO_DISABLED_NO_GIT = "⚠️  Not in a git repository - undo functionality will be limited"
-UNDO_INITIAL_COMMIT = "Initial commit for tunacode undo history"
-UNDO_GIT_TIMEOUT = "Git initialization timed out"
 
 # Log/status messages
 MSG_UPDATE_AVAILABLE = "Update available: v{latest_version}"

tunacode/core/agents/main.py

@@ -1,7 +1,7 @@
-"""Module: sidekick.core.agents.main
+"""Module: tunacode.core.agents.main
 
-Main agent functionality and coordination for the Sidekick CLI.
-Provides agent creation, message processing, and tool call management.
+Main agent functionality and coordination for the TunaCode CLI.
+Handles agent creation, configuration, and request processing.
 """
 
 from datetime import datetime, timezone
@@ -12,6 +12,7 @@ from pydantic_ai.messages import ModelRequest, ToolReturnPart
 
 from tunacode.core.state import StateManager
 from tunacode.services.mcp import get_mcp_servers
+from tunacode.tools.bash import bash
 from tunacode.tools.read_file import read_file
 from tunacode.tools.run_command import run_command
 from tunacode.tools.update_file import update_file
@@ -37,6 +38,7 @@ def get_or_create_agent(model: ModelName, state_manager: StateManager) -> Pydant
         state_manager.session.agents[model] = Agent(
             model=model,
             tools=[
+                Tool(bash, max_retries=max_retries),
                 Tool(read_file, max_retries=max_retries),
                 Tool(run_command, max_retries=max_retries),
                 Tool(update_file, max_retries=max_retries),

tunacode/core/setup/__init__.py

@@ -4,7 +4,6 @@ from .config_setup import ConfigSetup
 from .coordinator import SetupCoordinator
 from .environment_setup import EnvironmentSetup
 from .git_safety_setup import GitSafetySetup
-from .undo_setup import UndoSetup
 
 __all__ = [
     "BaseSetup",
@@ -12,6 +11,5 @@ __all__ = [
     "ConfigSetup",
     "EnvironmentSetup",
     "GitSafetySetup",
-    "UndoSetup",
     "AgentSetup",
 ]

tunacode/core/setup/agent_setup.py

@@ -1,7 +1,7 @@
-"""Module: tinyagent.core.setup.agent_setup
+"""Module: tunacode.core.setup.agent_setup
 
-Agent initialization and configuration for the Sidekick CLI.
-Handles the setup and validation of AI agents with the selected model.
+Agent initialization and configuration for the TunaCode CLI.
+Sets up AI agents with proper model configurations and tools.
 """
 
 from typing import Any, Optional

tunacode/core/setup/base.py

@@ -1,7 +1,7 @@
-"""Module: sidekick.core.setup.base
+"""Module: tunacode.core.setup.base
 
-Base setup step abstraction for the Sidekick CLI initialization process.
-Defines the contract that all setup steps must implement.
+Base setup step abstraction for the TunaCode CLI initialization process.
+Provides common interface and functionality for all setup steps.
 """
 
 from abc import ABC, abstractmethod

tunacode/core/setup/coordinator.py

@@ -1,6 +1,6 @@
-"""Module: tinyagent.core.setup.coordinator
+"""Module: tunacode.core.setup.coordinator
 
-Setup orchestration and coordination for the Sidekick CLI.
+Setup orchestration and coordination for the TunaCode CLI.
 Manages the execution order and validation of all registered setup steps.
 """
 

tunacode/core/setup/environment_setup.py

@@ -1,7 +1,7 @@
 """Module: tunacode.core.setup.environment_setup
 
-Environment detection and configuration for the Sidekick CLI.
-Handles setting up environment variables from user configuration.
+Environment detection and configuration for the TunaCode CLI.
+Validates system requirements and environment variables.
 """
 
 import os

tunacode/core/setup/git_safety_setup.py

@@ -133,8 +133,7 @@ class GitSafetySetup(BaseSetup):
                 await panel(
                     "⚠️  Working Without Safety Branch",
                     "You've chosen to work directly on your current branch.\n"
-                    "TunaCode will modify files in place. Make sure you have backups!\n"
-                    "You can always use /undo to revert changes.",
+                    "TunaCode will modify files in place. Make sure you have backups!",
                     border_style="red"
                 )
                 # Save preference

tunacode/core/state.py

@@ -1,7 +1,7 @@
-"""Module: sidekick.core.state
+"""Module: tunacode.core.state
 
-State management system for session data in Sidekick CLI.
-Provides centralized state tracking for agents, messages, configurations, and session information.
+State management system for session data in TunaCode CLI.
+Handles user preferences, conversation history, and runtime state.
 """
 
 import uuid

tunacode/setup.py

@@ -8,7 +8,7 @@ Provides high-level setup functions for initializing the application and its age
 from typing import Any, Optional
 
 from tunacode.core.setup import (AgentSetup, ConfigSetup, EnvironmentSetup, GitSafetySetup,
-                                 SetupCoordinator, UndoSetup)
+                                 SetupCoordinator)
 from tunacode.core.state import StateManager
 
 
@@ -29,8 +29,7 @@ async def setup(run_setup: bool, state_manager: StateManager, cli_config: dict =
         config_setup.cli_config = cli_config
     coordinator.register_step(config_setup)
     coordinator.register_step(EnvironmentSetup(state_manager))
-    coordinator.register_step(GitSafetySetup(state_manager))  # Run after config/env but before undo
-    coordinator.register_step(UndoSetup(state_manager))
+    coordinator.register_step(GitSafetySetup(state_manager))
 
     # Run all setup steps
     await coordinator.run_setup(force_setup=run_setup)

tunacode/tools/__init__.py

@@ -0,0 +1 @@
+"""TunaCode tools package."""

tunacode/tools/base.py

@@ -1,4 +1,4 @@
-"""Base tool class for all Sidekick tools.
+"""Base tool class for all TunaCode tools.
 
 This module provides a base class that implements common patterns
 for all tools including error handling, UI logging, and ModelRetry support.
@@ -13,7 +13,7 @@ from tunacode.types import FilePath, ToolName, ToolResult, UILogger
 
 
 class BaseTool(ABC):
-    """Base class for all Sidekick tools providing common functionality."""
+    """Base class for all TunaCode tools providing common functionality."""
 
     def __init__(self, ui_logger: UILogger | None = None):
         """Initialize the base tool.
@@ -42,11 +42,6 @@ class BaseTool(ABC):
             if self.ui:
                 await self.ui.info(f"{self.tool_name}({self._format_args(*args, **kwargs)})")
             result = await self._execute(*args, **kwargs)
-            
-            # For file operations, try to create a git commit for undo tracking
-            if isinstance(self, FileBasedTool):
-                await self._commit_for_undo()
-                
             return result
         except ModelRetry as e:
             # Log as warning and re-raise for pydantic-ai
@@ -142,46 +137,12 @@ class FileBasedTool(BaseTool):
     """Base class for tools that work with files.
 
     Provides common file-related functionality like:
-    - Path validation
+    - Path validation  
     - File existence checking
     - Directory creation
     - Encoding handling
-    - Git commit for undo tracking
+    - Enhanced error handling for file operations
     """
-    
-    async def _commit_for_undo(self) -> None:
-        """Create a git commit for undo tracking after file operations.
-        
-        This method gracefully handles cases where git is not available:
-        - No git repository: Warns user about limited undo functionality
-        - Git command fails: Warns but doesn't break the main operation
-        - Any other error: Silently continues (file operation still succeeds)
-        """
-        try:
-            # Import here to avoid circular imports
-            from tunacode.services.undo_service import commit_for_undo, is_in_git_project
-            
-            # Check if we're in a git project first
-            if not is_in_git_project():
-                if self.ui:
-                    await self.ui.muted("⚠️  No git repository - undo functionality limited")
-                return
-            
-            # Try to create commit with tool name as prefix
-            success = commit_for_undo(message_prefix=f"tunacode {self.tool_name.lower()}")
-            if success and self.ui:
-                await self.ui.muted("• Git commit created for undo tracking")
-            elif self.ui:
-                await self.ui.muted("⚠️  Could not create git commit - undo may not work")
-        except Exception:
-            # Silently ignore commit errors - don't break the main file operation
-            # The file operation itself succeeded, we just can't track it for undo
-            if self.ui:
-                try:
-                    await self.ui.muted("⚠️  Git commit failed - undo functionality limited")
-                except:
-                    # Even the warning failed, just continue silently
-                    pass
 
     def _format_args(self, filepath: FilePath, *args, **kwargs) -> str:
         """Format arguments with filepath as first argument."""

tunacode/tools/bash.py

@@ -0,0 +1,252 @@
+"""
+Module: tunacode.tools.bash
+
+Enhanced bash execution tool for agent operations in the TunaCode application.
+Provides advanced shell command execution with working directory support,
+environment variables, timeouts, and improved output handling.
+"""
+
+import asyncio
+import os
+import subprocess
+from typing import Dict, Optional
+
+from pydantic_ai.exceptions import ModelRetry
+
+from tunacode.constants import MAX_COMMAND_OUTPUT
+from tunacode.exceptions import ToolExecutionError
+from tunacode.tools.base import BaseTool
+from tunacode.types import ToolResult
+
+
+class BashTool(BaseTool):
+    """Enhanced shell command execution tool with advanced features."""
+
+    @property
+    def tool_name(self) -> str:
+        return "Bash"
+
+    async def _execute(
+        self,
+        command: str,
+        cwd: Optional[str] = None,
+        env: Optional[Dict[str, str]] = None,
+        timeout: Optional[int] = 30,
+        capture_output: bool = True,
+    ) -> ToolResult:
+        """Execute a bash command with enhanced features.
+
+        Args:
+            command: The bash command to execute
+            cwd: Working directory for the command (defaults to current)
+            env: Additional environment variables to set
+            timeout: Command timeout in seconds (default 30, max 300)
+            capture_output: Whether to capture stdout/stderr (default True)
+
+        Returns:
+            ToolResult: Formatted output with exit code, stdout, and stderr
+
+        Raises:
+            ModelRetry: For guidance on command failures
+            Exception: Any command execution errors
+        """
+        # Validate and sanitize inputs
+        if timeout and (timeout < 1 or timeout > 300):
+            raise ModelRetry(
+                "Timeout must be between 1 and 300 seconds. "
+                "Use shorter timeouts for quick commands, longer for builds/tests."
+            )
+
+        # Validate working directory if specified
+        if cwd and not os.path.isdir(cwd):
+            raise ModelRetry(
+                f"Working directory '{cwd}' does not exist. "
+                "Please verify the path or create the directory first."
+            )
+
+        # Check for potentially destructive commands
+        destructive_patterns = ["rm -rf", "rm -r", "rm /", "dd if=", "mkfs", "fdisk"]
+        if any(pattern in command for pattern in destructive_patterns):
+            raise ModelRetry(
+                f"Command contains potentially destructive operations: {command}\n"
+                "Please confirm this is intentional and safe for your system."
+            )
+
+        # Prepare environment
+        exec_env = os.environ.copy()
+        if env:
+            # Sanitize environment variables
+            for key, value in env.items():
+                if isinstance(key, str) and isinstance(value, str):
+                    exec_env[key] = value
+
+        # Set working directory
+        exec_cwd = cwd or os.getcwd()
+
+        try:
+            # Execute command with timeout
+            process = await asyncio.create_subprocess_shell(
+                command,
+                stdout=subprocess.PIPE if capture_output else None,
+                stderr=subprocess.PIPE if capture_output else None,
+                cwd=exec_cwd,
+                env=exec_env,
+            )
+
+            try:
+                stdout, stderr = await asyncio.wait_for(
+                    process.communicate(), timeout=timeout
+                )
+            except asyncio.TimeoutError:
+                # Kill the process if it times out
+                process.kill()
+                await process.wait()
+                raise ModelRetry(
+                    f"Command timed out after {timeout} seconds: {command}\n"
+                    "Consider using a longer timeout or breaking the command into smaller parts."
+                )
+
+            # Decode output
+            stdout_text = stdout.decode("utf-8", errors="replace").strip() if stdout else ""
+            stderr_text = stderr.decode("utf-8", errors="replace").strip() if stderr else ""
+
+            # Format output
+            result = self._format_output(
+                command=command,
+                exit_code=process.returncode,
+                stdout=stdout_text,
+                stderr=stderr_text,
+                cwd=exec_cwd,
+            )
+
+            # Handle non-zero exit codes as guidance, not failures
+            if process.returncode != 0 and stderr_text:
+                # Provide guidance for common error patterns
+                if "command not found" in stderr_text.lower():
+                    raise ModelRetry(
+                        f"Command '{command}' not found. "
+                        "Check if the command is installed or use the full path."
+                    )
+                elif "permission denied" in stderr_text.lower():
+                    raise ModelRetry(
+                        f"Permission denied for command '{command}'. "
+                        "You may need elevated privileges or different file permissions."
+                    )
+                elif "no such file or directory" in stderr_text.lower():
+                    raise ModelRetry(
+                        f"File or directory not found when running '{command}'. "
+                        "Verify the path exists or create it first."
+                    )
+
+            return result
+
+        except FileNotFoundError:
+            raise ModelRetry(
+                f"Shell not found. Cannot execute command: {command}\n"
+                "This typically indicates a system configuration issue."
+            )
+
+    def _format_output(
+        self,
+        command: str,
+        exit_code: int,
+        stdout: str,
+        stderr: str,
+        cwd: str,
+    ) -> str:
+        """Format command output in a consistent way.
+
+        Args:
+            command: The executed command
+            exit_code: The process exit code
+            stdout: Standard output content
+            stderr: Standard error content
+            cwd: Working directory where command was executed
+
+        Returns:
+            str: Formatted output string
+        """
+        # Build the result
+        lines = [
+            f"Command: {command}",
+            f"Exit Code: {exit_code}",
+            f"Working Directory: {cwd}",
+            "",
+        ]
+
+        # Add stdout if present
+        if stdout:
+            lines.extend(["STDOUT:", stdout, ""])
+        else:
+            lines.extend(["STDOUT:", "(no output)", ""])
+
+        # Add stderr if present
+        if stderr:
+            lines.extend(["STDERR:", stderr])
+        else:
+            lines.extend(["STDERR:", "(no errors)"])
+
+        result = "\n".join(lines)
+
+        # Truncate if too long
+        if len(result) > MAX_COMMAND_OUTPUT:
+            truncate_point = MAX_COMMAND_OUTPUT - 100  # Leave room for truncation message
+            result = result[:truncate_point] + "\n\n[... output truncated ...]"
+
+        return result
+
+    def _format_args(
+        self,
+        command: str,
+        cwd: Optional[str] = None,
+        env: Optional[Dict[str, str]] = None,
+        timeout: Optional[int] = None,
+        **kwargs,
+    ) -> str:
+        """Format arguments for display in UI logging."""
+        args = [repr(command)]
+
+        if cwd:
+            args.append(f"cwd={repr(cwd)}")
+        if timeout:
+            args.append(f"timeout={timeout}")
+        if env:
+            env_summary = f"{len(env)} vars" if len(env) > 3 else str(env)
+            args.append(f"env={env_summary}")
+
+        return ", ".join(args)
+
+    def _get_error_context(self, command: str = None, **kwargs) -> str:
+        """Get error context for bash execution."""
+        if command:
+            return f"executing bash command '{command}'"
+        return super()._get_error_context()
+
+
+# Create the function that maintains the existing interface
+async def bash(
+    command: str,
+    cwd: Optional[str] = None,
+    env: Optional[Dict[str, str]] = None,
+    timeout: Optional[int] = 30,
+    capture_output: bool = True,
+) -> ToolResult:
+    """
+    Execute a bash command with enhanced features.
+
+    Args:
+        command (str): The bash command to execute
+        cwd (Optional[str]): Working directory for the command
+        env (Optional[Dict[str, str]]): Additional environment variables
+        timeout (Optional[int]): Command timeout in seconds (default 30, max 300)
+        capture_output (bool): Whether to capture stdout/stderr
+
+    Returns:
+        ToolResult: Formatted output with exit code, stdout, and stderr
+    """
+    tool = BashTool()
+    try:
+        return await tool.execute(command, cwd=cwd, env=env, timeout=timeout, capture_output=capture_output)
+    except ToolExecutionError as e:
+        # Return error message for pydantic-ai compatibility
+        return str(e)

tunacode/tools/read_file.py

@@ -1,7 +1,7 @@
 """
-Module: sidekick.tools.read_file
+Module: tunacode.tools.read_file
 
-File reading tool for agent operations in the Sidekick application.
+File reading tool for agent operations in the TunaCode application.
 Provides safe file reading with size limits and proper error handling.
 """
 

tunacode/tools/run_command.py

@@ -1,7 +1,7 @@
 """
-Module: sidekick.tools.run_command
+Module: tunacode.tools.run_command
 
-Command execution tool for agent operations in the Sidekick application.
+Command execution tool for agent operations in the TunaCode application.
 Provides controlled shell command execution with output capture and truncation.
 """
 

tunacode/tools/update_file.py

@@ -1,8 +1,8 @@
 """
-Module: sidekick.tools.update_file
+Module: tunacode.tools.update_file
 
-File update tool for agent operations in the Sidekick application.
-Enables safe text replacement in existing files with target/patch semantics.
+File update tool for agent operations in the TunaCode application.
+Provides targeted file content modification with diff-based updates.
 """
 
 import os

tunacode/tools/write_file.py

@@ -1,8 +1,8 @@
 """
-Module: sidekick.tools.write_file
+Module: tunacode.tools.write_file
 
-File writing tool for agent operations in the Sidekick application.
-Creates new files with automatic directory creation and overwrite protection.
+File writing tool for agent operations in the TunaCode application.
+Provides safe file creation with conflict detection and encoding handling.
 """
 
 import os

tunacode/ui/completers.py

@@ -44,7 +44,7 @@ class CommandCompleter(Completer):
             command_names = self.command_registry.get_command_names()
         else:
             # Fallback list of commands
-            command_names = ['/help', '/clear', '/dump', '/yolo', '/undo', 
+            command_names = ['/help', '/clear', '/dump', '/yolo', 
                            '/branch', '/compact', '/model']
         
         # Get the partial command (without /)

tunacode/ui/console.py

@@ -1,7 +1,6 @@
-"""Main console coordination module for Sidekick UI.
+"""Main console coordination module for TunaCode UI.
 
-This module re-exports functions from specialized UI modules to maintain
-backward compatibility while organizing code into focused modules.
+Provides high-level console functions and coordinates between different UI components.
 """
 
 from rich.console import Console as RichConsole

tunacode/ui/constants.py

@@ -1,4 +1,4 @@
-"""UI-specific constants for Sidekick."""
+"""UI-specific constants for TunaCode."""
 
 # UI Layout Constants
 DEFAULT_PANEL_PADDING = {"top": 1, "right": 0, "bottom": 1, "left": 0}

tunacode/ui/decorators.py

@@ -1,7 +1,7 @@
 """
-Module: sidekick.ui.decorators
+Module: tunacode.ui.decorators
 
-Provides decorators for UI functions including sync/async wrapper patterns.
+Decorators for UI functions, particularly for creating sync wrappers of async functions.
 """
 
 import asyncio

tunacode/ui/input.py

@@ -1,4 +1,4 @@
-"""User input handling functions for Sidekick UI."""
+"""User input handling functions for TunaCode UI."""
 
 from typing import Optional
 

tunacode/ui/keybindings.py

@@ -1,4 +1,4 @@
-"""Key binding handlers for Sidekick UI."""
+"""Key binding handlers for TunaCode UI."""
 
 from prompt_toolkit.key_binding import KeyBindings
 

tunacode/ui/output.py

@@ -1,4 +1,4 @@
-"""Output and display functions for Sidekick UI."""
+"""Output and display functions for TunaCode UI."""
 
 from prompt_toolkit.application import run_in_terminal
 from rich.console import Console
@@ -16,9 +16,16 @@ from .decorators import create_sync_wrapper
 console = Console()
 colors = DotDict(UI_COLORS)
 
-BANNER = """[bold #00d7ff]╭─────────────────────────────────────────────────────────────────╮[/bold #00d7ff]
-[bold #00d7ff]│[/bold #00d7ff] [bold white]T U N A C O D E[/bold white] [dim #64748b]• Agentic AI Development Environment[/dim #64748b] [bold #00d7ff]│[/bold #00d7ff]
-[bold #00d7ff]╰─────────────────────────────────────────────────────────────────╯[/bold #00d7ff]"""
+BANNER = """[bold cyan]
+████████╗██╗   ██╗███╗   ██╗ █████╗  ██████╗ ██████╗ ██████╗ ███████╗
+╚══██╔══╝██║   ██║████╗  ██║██╔══██╗██╔════╝██╔═══██╗██╔══██╗██╔════╝
+   ██║   ██║   ██║██╔██╗ ██║███████║██║     ██║   ██║██║  ██║█████╗  
+   ██║   ██║   ██║██║╚██╗██║██╔══██║██║     ██║   ██║██║  ██║██╔══╝  
+   ██║   ╚██████╔╝██║ ╚████║██║  ██║╚██████╗╚██████╔╝██████╔╝███████╗
+   ╚═╝    ╚═════╝ ╚═╝  ╚═══╝╚═╝  ╚═╝ ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝
+[/bold cyan]
+
+● Caution: This tool can modify your codebase - always use git branches"""
 
 
 @create_sync_wrapper