tunacode-cli 0.0.17__py3-none-any.whl → 0.0.18__py3-none-any.whl

tunacode/core/setup/git_safety_setup.py

@@ -13,15 +13,12 @@ from tunacode.ui.panels import panel
 async def yes_no_prompt(question: str, default: bool = True) -> bool:
     """Simple yes/no prompt."""
     default_text = "[Y/n]" if default else "[y/N]"
-    response = await prompt_input(
-        session_key="yes_no",
-        pretext=f"{question} {default_text}: "
-    )
-    
+    response = await prompt_input(session_key="yes_no", pretext=f"{question} {default_text}: ")
+
     if not response.strip():
         return default
-    
-    return response.lower().strip() in ['y', 'yes']
+
+    return response.lower().strip() in ["y", "yes"]
 
 
 class GitSafetySetup(BaseSetup):
@@ -29,7 +26,7 @@ class GitSafetySetup(BaseSetup):
 
     def __init__(self, state_manager: StateManager):
         super().__init__(state_manager)
-    
+
     @property
     def name(self) -> str:
         """Return the name of this setup step."""
@@ -45,18 +42,15 @@ class GitSafetySetup(BaseSetup):
         try:
             # Check if git is installed
             result = subprocess.run(
-                ["git", "--version"],
-                capture_output=True,
-                text=True,
-                check=False
+                ["git", "--version"], capture_output=True, text=True, check=False
             )
-            
+
             if result.returncode != 0:
                 await panel(
                     "⚠️  Git Not Found",
                     "Git is not installed or not in PATH. TunaCode will modify files directly.\n"
                     "It's strongly recommended to install Git for safety.",
-                    border_style="yellow"
+                    border_style="yellow",
                 )
                 return
 
@@ -66,33 +60,30 @@ class GitSafetySetup(BaseSetup):
                 capture_output=True,
                 text=True,
                 check=False,
-                cwd=Path.cwd()
+                cwd=Path.cwd(),
             )
-            
+
             if result.returncode != 0:
                 await panel(
                     "⚠️  Not a Git Repository",
                     "This directory is not a Git repository. TunaCode will modify files directly.\n"
                     "Consider initializing a Git repository for safety: git init",
-                    border_style="yellow"
+                    border_style="yellow",
                 )
                 return
 
             # Get current branch name
             result = subprocess.run(
-                ["git", "branch", "--show-current"],
-                capture_output=True,
-                text=True,
-                check=True
+                ["git", "branch", "--show-current"], capture_output=True, text=True, check=True
             )
             current_branch = result.stdout.strip()
-            
+
             if not current_branch:
                 # Detached HEAD state
                 await panel(
                     "⚠️  Detached HEAD State",
                     "You're in a detached HEAD state. TunaCode will continue without creating a branch.",
-                    border_style="yellow"
+                    border_style="yellow",
                 )
                 return
 
@@ -103,57 +94,52 @@ class GitSafetySetup(BaseSetup):
 
             # Propose new branch name
             new_branch = f"{current_branch}-tunacode"
-            
+
             # Check if there are uncommitted changes
             result = subprocess.run(
-                ["git", "status", "--porcelain"],
-                capture_output=True,
-                text=True,
-                check=True
+                ["git", "status", "--porcelain"], capture_output=True, text=True, check=True
             )
-            
+
             has_changes = bool(result.stdout.strip())
-            
+
             # Ask user if they want to create a safety branch
             message = (
                 f"For safety, TunaCode can create a new branch '{new_branch}' based on '{current_branch}'.\n"
                 f"This helps protect your work from unintended changes.\n"
             )
-            
+
             if has_changes:
-                message += "\n⚠️  You have uncommitted changes that will be brought to the new branch."
-            
-            create_branch = await yes_no_prompt(
-                f"{message}\n\nCreate safety branch?",
-                default=True
-            )
-            
+                message += (
+                    "\n⚠️  You have uncommitted changes that will be brought to the new branch."
+                )
+
+            create_branch = await yes_no_prompt(f"{message}\n\nCreate safety branch?", default=True)
+
             if not create_branch:
                 # User declined - show warning
                 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!",
-                    border_style="red"
+                    border_style="red",
                 )
                 # Save preference
                 self.state_manager.session.user_config["skip_git_safety"] = True
                 return
-            
+
             # Create and checkout the new branch
             try:
                 # Check if branch already exists
                 result = subprocess.run(
                     ["git", "show-ref", "--verify", f"refs/heads/{new_branch}"],
                     capture_output=True,
-                    check=False
+                    check=False,
                 )
-                
+
                 if result.returncode == 0:
                     # Branch exists, ask to use it
                     use_existing = await yes_no_prompt(
-                        f"Branch '{new_branch}' already exists. Switch to it?",
-                        default=True
+                        f"Branch '{new_branch}' already exists. Switch to it?", default=True
                     )
                     if use_existing:
                         subprocess.run(["git", "checkout", new_branch], check=True)
@@ -164,24 +150,24 @@ class GitSafetySetup(BaseSetup):
                     # Create new branch
                     subprocess.run(["git", "checkout", "-b", new_branch], check=True)
                     await ui.success(f"Created and switched to new branch: {new_branch}")
-                    
+
             except subprocess.CalledProcessError as e:
                 await panel(
                     "❌ Failed to Create Branch",
                     f"Could not create branch '{new_branch}': {str(e)}\n"
                     "Continuing on current branch.",
-                    border_style="red"
+                    border_style="red",
                 )
-                
+
         except Exception as e:
             # Non-fatal error - just warn the user
             await panel(
                 "⚠️  Git Safety Setup Failed",
                 f"Could not set up Git safety: {str(e)}\n"
                 "TunaCode will continue without branch protection.",
-                border_style="yellow"
+                border_style="yellow",
             )
 
     async def validate(self) -> bool:
         """Validate git safety setup - always returns True as this is optional."""
-        return True
+        return True

tunacode/core/state.py

@@ -8,15 +8,8 @@ import uuid
 from dataclasses import dataclass, field
 from typing import Any, Optional
 
-from tunacode.types import (
-    DeviceId,
-    InputSessions,
-    MessageHistory,
-    ModelName,
-    SessionId,
-    ToolName,
-    UserConfig,
-)
+from tunacode.types import (DeviceId, InputSessions, MessageHistory, ModelName, SessionId, ToolName,
+                            UserConfig)
 
 
 @dataclass

tunacode/exceptions.py

@@ -77,8 +77,6 @@ class MCPError(ServiceError):
         super().__init__(f"MCP server '{server_name}' error: {message}")
 
 
-
-
 class GitOperationError(ServiceError):
     """Raised when Git operations fail."""
 

tunacode/tools/__init__.py

@@ -1 +1,10 @@
-"""TunaCode tools package."""
+"""TunaCode tools package. Implements lazy loading of submodules for faster startup."""
+
+import importlib
+
+
+def __getattr__(name):
+    try:
+        return importlib.import_module(f".{name}", __name__)
+    except ImportError as e:
+        raise AttributeError(f"module {{__name__}} has no attribute '{{name}}'") from e

tunacode/tools/base.py

@@ -137,7 +137,7 @@ 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

tunacode/tools/bash.py

@@ -94,9 +94,7 @@ class BashTool(BaseTool):
             )
 
             try:
-                stdout, stderr = await asyncio.wait_for(
-                    process.communicate(), timeout=timeout
-                )
+                stdout, stderr = await asyncio.wait_for(process.communicate(), timeout=timeout)
             except asyncio.TimeoutError:
                 # Kill the process if it times out
                 process.kill()
@@ -246,7 +244,9 @@ async def bash(
     """
     tool = BashTool()
     try:
-        return await tool.execute(command, cwd=cwd, env=env, timeout=timeout, capture_output=capture_output)
+        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)
+        return str(e)