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

tunacode/core/agents/tinyagent_main.py

@@ -1,194 +0,0 @@
-"""TinyAgent-based agent implementation."""
-
-import os
-from contextlib import asynccontextmanager
-from datetime import datetime, timezone
-from pathlib import Path
-from typing import Any, Dict, Optional
-
-from tinyagent import ReactAgent
-
-from tunacode.core.state import StateManager
-from tunacode.tools.tinyagent_tools import read_file, run_command, update_file, write_file
-from tunacode.types import ModelName, ToolCallback
-
-# Set up tinyagent configuration
-# First check if config exists in the package directory
-_package_dir = Path(__file__).parent.parent.parent.parent  # Navigate to tunacode root
-_config_candidates = [
-    _package_dir / "tunacode_config.yml",  # In package root
-    Path.home() / ".config" / "tunacode_config.yml",  # In user config dir
-    Path.cwd() / "tunacode_config.yml",  # In current directory
-]
-
-# Find the first existing config file
-for config_path in _config_candidates:
-    if config_path.exists():
-        os.environ["TINYAGENT_CONFIG"] = str(config_path)
-        break
-else:
-    # If no config found, we'll rely on tinyagent to handle it
-    # This prevents the immediate error on import
-    pass
-
-
-def get_or_create_react_agent(model: ModelName, state_manager: StateManager) -> ReactAgent:
-    """
-    Get or create a ReactAgent for the specified model.
-
-    Args:
-        model: The model name (e.g., "openai:gpt-4o", "openrouter:openai/gpt-4.1")
-        state_manager: The state manager instance
-
-    Returns:
-        ReactAgent instance configured for the model
-    """
-    agents = state_manager.session.agents
-
-    if model not in agents:
-        # Parse model string to determine provider and actual model name
-        # Format: "provider:model" or "openrouter:provider/model"
-        if model.startswith("openrouter:"):
-            # OpenRouter model - extract the actual model name
-            actual_model = model.replace("openrouter:", "")
-            # Set environment to use OpenRouter base URL
-            import os
-
-            os.environ["OPENAI_BASE_URL"] = "https://openrouter.ai/api/v1"
-            # Use OpenRouter API key if available
-            if state_manager.session.user_config["env"].get("OPENROUTER_API_KEY"):
-                os.environ["OPENAI_API_KEY"] = state_manager.session.user_config["env"][
-                    "OPENROUTER_API_KEY"
-                ]
-        else:
-            # Direct provider (openai, anthropic, google-gla)
-            provider, actual_model = model.split(":", 1)
-            # Reset to default base URL for direct providers
-            import os
-
-            if provider == "openai":
-                os.environ["OPENAI_BASE_URL"] = "https://api.openai.com/v1"
-            # Set appropriate API key
-            provider_key_map = {
-                "openai": "OPENAI_API_KEY",
-                "anthropic": "ANTHROPIC_API_KEY",
-                "google-gla": "GEMINI_API_KEY",
-            }
-            if provider in provider_key_map:
-                key_name = provider_key_map[provider]
-                if state_manager.session.user_config["env"].get(key_name):
-                    os.environ[key_name] = state_manager.session.user_config["env"][key_name]
-
-        # Create new ReactAgent with tools
-        # Note: tinyAgent gets model from environment variables, not constructor
-        agent = ReactAgent(tools=[read_file, write_file, update_file, run_command])
-
-        # Add MCP compatibility method
-        @asynccontextmanager
-        async def run_mcp_servers():
-            # TinyAgent doesn't have built-in MCP support yet
-            # This is a placeholder for compatibility
-            yield
-
-        agent.run_mcp_servers = run_mcp_servers
-
-        # Cache the agent
-        agents[model] = agent
-
-    return agents[model]
-
-
-async def process_request_with_tinyagent(
-    model: ModelName,
-    message: str,
-    state_manager: StateManager,
-    tool_callback: Optional[ToolCallback] = None,
-) -> Dict[str, Any]:
-    """
-    Process a request using TinyAgent's ReactAgent.
-
-    Args:
-        model: The model to use
-        message: The user message
-        state_manager: State manager instance
-        tool_callback: Optional callback for tool execution (for UI updates)
-
-    Returns:
-        Dict containing the result and any metadata
-    """
-    agent = get_or_create_react_agent(model, state_manager)
-
-    # Convert message history to format expected by tinyAgent
-    # Note: tinyAgent handles message history differently than pydantic-ai
-    # We'll need to adapt based on tinyAgent's actual API
-
-    try:
-        # Run the agent with the message
-        # The new API's run() method might be synchronous based on the examples
-        import asyncio
-        if asyncio.iscoroutinefunction(agent.run):
-            result = await agent.run(message)
-        else:
-            result = agent.run(message)
-
-        # Update message history in state_manager
-        # This will need to be adapted based on how tinyAgent returns messages
-        state_manager.session.messages.append(
-            {
-                "role": "user",
-                "content": message,
-                "timestamp": datetime.now(timezone.utc).isoformat(),
-            }
-        )
-
-        state_manager.session.messages.append(
-            {
-                "role": "assistant",
-                "content": result,
-                "timestamp": datetime.now(timezone.utc).isoformat(),
-            }
-        )
-
-        return {"result": result, "success": True, "model": model}
-
-    except Exception as e:
-        # Handle errors
-        error_result = {
-            "result": f"Error: {str(e)}",
-            "success": False,
-            "model": model,
-            "error": str(e),
-        }
-
-        # Still update message history with the error
-        state_manager.session.messages.append(
-            {
-                "role": "user",
-                "content": message,
-                "timestamp": datetime.now(timezone.utc).isoformat(),
-            }
-        )
-
-        state_manager.session.messages.append(
-            {
-                "role": "assistant",
-                "content": f"Error occurred: {str(e)}",
-                "timestamp": datetime.now(timezone.utc).isoformat(),
-                "error": True,
-            }
-        )
-
-        return error_result
-
-
-def patch_tool_messages(
-    error_message: str = "Tool operation failed",
-    state_manager: StateManager = None,
-):
-    """
-    Compatibility function for patching tool messages.
-    With tinyAgent, this may not be needed as it handles tool errors differently.
-    """
-    # TinyAgent handles tool retries and errors internally
-    # This function is kept for compatibility but may be simplified
-    pass

tunacode/core/setup/optimized_coordinator.py

@@ -1,73 +0,0 @@
-"""Optimized setup coordinator with deferred loading."""
-
-import asyncio
-from typing import List, Set
-
-from tunacode.core.setup.base import BaseSetup
-from tunacode.core.state import StateManager
-from tunacode.ui import console as ui
-
-
-class OptimizedSetupCoordinator:
-    """Optimized coordinator that defers non-critical setup steps."""
-
-    def __init__(self, state_manager: StateManager):
-        self.state_manager = state_manager
-        self.critical_steps: List[BaseSetup] = []
-        self.deferred_steps: List[BaseSetup] = []
-        self._deferred_task = None
-
-        # Define critical steps that must run at startup
-        self.critical_step_names: Set[str] = {
-            "Configuration",  # Need config to know which model to use
-            "Environment Variables",  # Need API keys
-        }
-
-    def register_step(self, step: BaseSetup) -> None:
-        """Register a setup step, separating critical from deferred."""
-        if step.name in self.critical_step_names:
-            self.critical_steps.append(step)
-        else:
-            self.deferred_steps.append(step)
-
-    async def run_setup(self, force_setup: bool = False) -> None:
-        """Run critical setup immediately, defer the rest."""
-        # Run critical steps synchronously
-        for step in self.critical_steps:
-            try:
-                if await step.should_run(force_setup):
-                    await step.execute(force_setup)
-                    if not await step.validate():
-                        await ui.error(f"Setup validation failed: {step.name}")
-                        raise RuntimeError(f"Setup step '{step.name}' failed validation")
-            except Exception as e:
-                await ui.error(f"Setup failed at step '{step.name}': {str(e)}")
-                raise
-
-        # Schedule deferred steps to run in background
-        if self.deferred_steps and not self._deferred_task:
-            self._deferred_task = asyncio.create_task(self._run_deferred_steps(force_setup))
-
-    async def _run_deferred_steps(self, force_setup: bool) -> None:
-        """Run deferred steps in the background."""
-        # Wait a moment to let the main UI start
-        await asyncio.sleep(0.1)
-
-        for step in self.deferred_steps:
-            try:
-                if await step.should_run(force_setup):
-                    await step.execute(force_setup)
-                    # Don't validate deferred steps - they're non-critical
-            except Exception:
-                # Log but don't fail on deferred steps
-                pass
-
-    async def ensure_deferred_complete(self) -> None:
-        """Ensure deferred steps are complete before certain operations."""
-        if self._deferred_task and not self._deferred_task.done():
-            await self._deferred_task
-
-    def clear_steps(self) -> None:
-        """Clear all registered setup steps."""
-        self.critical_steps.clear()
-        self.deferred_steps.clear()

tunacode/services/enhanced_undo_service.py

@@ -1,322 +0,0 @@
-"""
-Enhanced undo service with multiple failsafe mechanisms.
-Provides three layers of protection for file operations.
-"""
-
-import json
-import shutil
-import subprocess
-from datetime import datetime
-from pathlib import Path
-from typing import Dict, List, Optional, Tuple
-
-from tunacode.core.state import StateManager
-from tunacode.utils.system import get_session_dir
-
-
-class EnhancedUndoService:
-    """Multi-layer undo system with failsafes."""
-
-    def __init__(self, state_manager: StateManager):
-        self.state_manager = state_manager
-        self.session_dir = get_session_dir(state_manager)
-
-        # Layer 1: File backups directory
-        self.backup_dir = self.session_dir / "backups"
-        self.backup_dir.mkdir(parents=True, exist_ok=True)
-
-        # Layer 2: Operation log
-        self.op_log_file = self.session_dir / "operations.jsonl"
-
-        # Layer 3: Git (existing system)
-        self.git_dir = self.session_dir / ".git"
-
-        self._init_systems()
-
-    def _init_systems(self):
-        """Initialize all undo systems."""
-        # Ensure operation log exists
-        if not self.op_log_file.exists():
-            self.op_log_file.touch()
-
-    # ===== LAYER 1: File Backups =====
-
-    def backup_file(self, filepath: Path) -> Optional[Path]:
-        """
-        Create a timestamped backup of a file before modification.
-
-        Returns:
-            Path to backup file, or None if backup failed
-        """
-        # Note: We'll create empty backup even if file doesn't exist yet
-        # This helps track file creation operations
-
-        try:
-            # Create unique backup name with timestamp
-            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
-            backup_name = f"{filepath.name}.{timestamp}.bak"
-            backup_path = self.backup_dir / backup_name
-
-            # Copy the file
-            shutil.copy2(filepath, backup_path)
-
-            # Log the backup
-            self._log_operation(
-                {
-                    "type": "backup",
-                    "original": str(filepath),
-                    "backup": str(backup_path),
-                    "timestamp": timestamp,
-                }
-            )
-
-            return backup_path
-        except Exception as e:
-            print(f"⚠️  Failed to create backup: {e}")
-            return None
-
-    def restore_from_backup(self, original_path: Path) -> Tuple[bool, str]:
-        """
-        Restore a file from its most recent backup.
-
-        Returns:
-            (success, message) tuple
-        """
-        try:
-            # Find most recent backup for this file
-            backups = sorted(
-                [f for f in self.backup_dir.glob(f"{original_path.name}.*.bak")], reverse=True
-            )
-
-            if not backups:
-                return False, f"No backups found for {original_path.name}"
-
-            latest_backup = backups[0]
-
-            # Restore the file
-            shutil.copy2(latest_backup, original_path)
-
-            # Log the restore
-            self._log_operation(
-                {
-                    "type": "restore",
-                    "file": str(original_path),
-                    "from_backup": str(latest_backup),
-                    "timestamp": datetime.now().isoformat(),
-                }
-            )
-
-            return True, f"Restored {original_path.name} from backup"
-
-        except Exception as e:
-            return False, f"Failed to restore from backup: {e}"
-
-    # ===== LAYER 2: Operation Log =====
-
-    def _log_operation(self, operation: Dict):
-        """Log an operation to the operations file."""
-        try:
-            with open(self.op_log_file, "a") as f:
-                json.dump(operation, f)
-                f.write("\n")
-        except Exception:
-            pass  # Silent fail for logging
-
-    def log_file_operation(
-        self,
-        op_type: str,
-        filepath: Path,
-        old_content: Optional[str] = None,
-        new_content: Optional[str] = None,
-    ):
-        """
-        Log a file operation with content for potential recovery.
-        """
-        operation = {
-            "type": "file_operation",
-            "operation": op_type,
-            "file": str(filepath),
-            "timestamp": datetime.now().isoformat(),
-        }
-
-        # For safety, only log content for smaller files
-        if old_content and len(old_content) < 100_000:  # 100KB limit
-            operation["old_content"] = old_content
-        if new_content and len(new_content) < 100_000:
-            operation["new_content"] = new_content
-
-        self._log_operation(operation)
-
-    def get_recent_operations(self, limit: int = 10) -> List[Dict]:
-        """Get recent operations from the log."""
-        operations = []
-        try:
-            with open(self.op_log_file, "r") as f:
-                for line in f:
-                    if line.strip():
-                        operations.append(json.loads(line))
-        except Exception:
-            pass
-
-        return operations[-limit:]
-
-    def undo_from_log(self) -> Tuple[bool, str]:
-        """
-        Attempt to undo the last operation using the operation log.
-        """
-        operations = self.get_recent_operations(20)
-
-        # Find the last file operation
-        for op in reversed(operations):
-            if op.get("type") == "file_operation" and "old_content" in op:
-                filepath = Path(op["file"])
-
-                try:
-                    # Restore old content
-                    with open(filepath, "w") as f:
-                        f.write(op["old_content"])
-
-                    self._log_operation(
-                        {
-                            "type": "undo_from_log",
-                            "restored_file": str(filepath),
-                            "timestamp": datetime.now().isoformat(),
-                        }
-                    )
-
-                    return True, f"Restored {filepath.name} from operation log"
-                except Exception as e:
-                    return False, f"Failed to restore from log: {e}"
-
-        return False, "No recoverable operations found in log"
-
-    # ===== LAYER 3: Git Integration =====
-
-    def git_commit(self, message: str = "Auto-save") -> bool:
-        """Create a git commit if available."""
-        if not self.git_dir.exists():
-            return False
-
-        try:
-            git_dir_arg = f"--git-dir={self.git_dir}"
-            subprocess.run(["git", git_dir_arg, "add", "."], capture_output=True, timeout=5)
-
-            subprocess.run(
-                ["git", git_dir_arg, "commit", "-m", message], capture_output=True, timeout=5
-            )
-            return True
-        except Exception:
-            return False
-
-    # ===== Unified Undo Interface =====
-
-    def perform_undo(self) -> Tuple[bool, str]:
-        """
-        Perform undo with automatic failover:
-        1. Try Git first (if available)
-        2. Fall back to operation log
-        3. Fall back to file backups
-        """
-        # Layer 3: Try Git first
-        if self.git_dir.exists():
-            try:
-                from tunacode.services.undo_service import perform_undo
-
-                success, message = perform_undo(self.state_manager)
-                if success:
-                    return success, f"[Git] {message}"
-            except Exception:
-                pass
-
-        # Layer 2: Try operation log
-        success, message = self.undo_from_log()
-        if success:
-            return success, f"[OpLog] {message}"
-
-        # Layer 1: Show available backups
-        backups = list(self.backup_dir.glob("*.bak"))
-        if backups:
-            recent_backups = sorted(backups, reverse=True)[:5]
-            backup_list = "\n".join([f"  - {b.name}" for b in recent_backups])
-            return False, f"Manual restore available from backups:\n{backup_list}"
-
-        return False, "No undo information available"
-
-    def cleanup_old_backups(self, keep_recent: int = 50):
-        """Clean up old backup files, keeping the most recent ones."""
-        backups = sorted(self.backup_dir.glob("*.bak"))
-
-        if len(backups) > keep_recent:
-            for backup in backups[:-keep_recent]:
-                try:
-                    backup.unlink()
-                except Exception:
-                    pass
-
-
-# ===== Safe File Operations Wrapper =====
-
-
-class SafeFileOperations:
-    """Wrapper for file operations with automatic backup."""
-
-    def __init__(self, undo_service: EnhancedUndoService):
-        self.undo = undo_service
-
-    async def safe_write(self, filepath: Path, content: str) -> Tuple[bool, str]:
-        """Write file with automatic backup."""
-        filepath = Path(filepath)
-
-        # Get old content if file exists
-        old_content = None
-        if filepath.exists():
-            try:
-                old_content = filepath.read_text()
-            except Exception:
-                pass
-
-        # Always create backup before write (even for new files)
-        self.undo.backup_file(filepath)
-
-        # Write new content
-        try:
-            filepath.write_text(content)
-
-            # Log the operation
-            self.undo.log_file_operation("write", filepath, old_content, content)
-
-            # Commit to git if available
-            self.undo.git_commit(f"Modified {filepath.name}")
-
-            return True, "File written successfully"
-
-        except Exception as e:
-            return False, f"Failed to write file: {e}"
-
-    async def safe_delete(self, filepath: Path) -> Tuple[bool, str]:
-        """Delete file with automatic backup."""
-        filepath = Path(filepath)
-
-        if not filepath.exists():
-            return False, "File does not exist"
-
-        try:
-            # Create backup first
-            backup_path = self.undo.backup_file(filepath)
-
-            # Get content for operation log
-            content = filepath.read_text()
-
-            # Delete the file
-            filepath.unlink()
-
-            # Log the operation
-            self.undo.log_file_operation("delete", filepath, old_content=content)
-
-            # Commit to git
-            self.undo.git_commit(f"Deleted {filepath.name}")
-
-            return True, f"File deleted (backup at {backup_path.name})"
-
-        except Exception as e:
-            return False, f"Failed to delete file: {e}"