groknroll 2.0.0__py3-none-any.whl

groknroll/agents/plan_agent.py

@@ -0,0 +1,215 @@
+"""
+Plan Agent
+
+Read-only agent for planning and analysis without making changes.
+This agent can read files and analyze code but cannot modify anything.
+"""
+
+from pathlib import Path
+from typing import Dict, Any, Optional
+import time
+
+from groknroll.agents.base_agent import (
+    BaseAgent, AgentConfig, AgentCapability, AgentResponse
+)
+from groknroll.core.rlm_integration import RLMIntegration, RLMConfig
+from groknroll.operations.file_ops import FileOperations
+
+
+class PlanAgent(BaseAgent):
+    """
+    Plan Agent - Read-Only
+
+    Capabilities:
+    - Read files
+    - Analyze code
+    - Search code
+
+    Use for:
+    - Planning implementations
+    - Code reviews
+    - Architecture analysis
+    - Understanding codebases
+    - Generating documentation
+
+    CANNOT:
+    - Write/edit/delete files
+    - Execute bash commands
+    - Make git commits
+    """
+
+    def __init__(
+        self,
+        project_path: Path,
+        model: str = "gpt-4o-mini",
+        max_cost: float = 5.0,
+        timeout: int = 300
+    ):
+        """
+        Initialize plan agent
+
+        Args:
+            project_path: Project root path
+            model: LLM model to use
+            max_cost: Maximum cost per execution
+            timeout: Timeout in seconds
+        """
+        # Plan agent has read-only capabilities
+        config = AgentConfig(
+            name="plan",
+            description="Read-only agent for planning and analysis",
+            capabilities=[
+                AgentCapability.READ_FILES,
+                AgentCapability.ANALYZE_CODE,
+                AgentCapability.SEARCH_CODE,
+            ],
+            model=model,
+            max_cost=max_cost,
+            timeout=timeout
+        )
+
+        super().__init__(config, project_path)
+
+        # Initialize read-only file operations
+        self.file_ops = FileOperations(project_path, create_backups=False)
+
+        # Initialize RLM
+        rlm_config = RLMConfig(
+            model=model,
+            max_cost=max_cost,
+            timeout_seconds=timeout
+        )
+        self.rlm = RLMIntegration(rlm_config)
+
+    def execute(self, task: str, context: Optional[Dict[str, Any]] = None) -> AgentResponse:
+        """
+        Execute planning task
+
+        Args:
+            task: Task description
+            context: Additional context
+
+        Returns:
+            AgentResponse
+        """
+        start_time = time.time()
+
+        try:
+            # Prepare context with available operations
+            exec_context = {
+                "agent": "plan",
+                "capabilities": self.get_capabilities(),
+                "project_path": str(self.project_path),
+                "operations": {
+                    "file": "Read-only: can read files, cannot write/edit/delete",
+                    "bash": "Not available (read-only agent)",
+                    "git": "Not available (read-only agent)"
+                },
+                "mode": "READ-ONLY",
+                **(context or {})
+            }
+
+            # Add instructions for planning mode
+            enhanced_task = f"""Task: {task}
+
+You are the PLAN agent with READ-ONLY access:
+- Can read files
+- Can analyze code
+- Can search code
+- CANNOT modify files
+- CANNOT execute bash
+- CANNOT use git
+
+Your goal is to:
+1. Understand the codebase
+2. Analyze architecture
+3. Plan implementations
+4. Provide recommendations
+5. Generate documentation
+
+Available in context:
+- file_ops: FileOperations instance (read-only)
+
+Example usage in RLM code:
+```python
+# Read file
+result = file_ops.read_file(Path("example.py"))
+if result.success:
+    content = result.message
+    # Analyze content...
+
+# List files
+result = file_ops.list_files(pattern="*.py", recursive=True)
+if result.success:
+    files = result.message.split("\\n")
+    # Process files...
+```
+
+Analyze and plan for the task. DO NOT attempt to modify files.
+"""
+
+            # Execute with RLM
+            result = self.rlm.complete(
+                task=enhanced_task,
+                context=exec_context
+            )
+
+            elapsed_time = time.time() - start_time
+
+            response = AgentResponse(
+                success=result.success,
+                message=result.response if result.success else result.error or "Unknown error",
+                agent_name=self.config.name,
+                task=task,
+                cost=result.total_cost,
+                time=elapsed_time,
+                metadata={
+                    "iterations": result.iterations,
+                    "rlm_success": result.success,
+                    "mode": "read-only"
+                }
+            )
+
+            self._log_execution(response)
+            return response
+
+        except Exception as e:
+            elapsed_time = time.time() - start_time
+
+            response = AgentResponse(
+                success=False,
+                message=f"Plan agent error: {e}",
+                agent_name=self.config.name,
+                task=task,
+                time=elapsed_time
+            )
+
+            self._log_execution(response)
+            return response
+
+    # =========================================================================
+    # Convenience Methods (Read-Only Operations)
+    # =========================================================================
+
+    def read_file(self, path: Path) -> str:
+        """Read file (convenience method)"""
+        self.require(AgentCapability.READ_FILES)
+        result = self.file_ops.read_file(path)
+        if not result.success:
+            raise RuntimeError(result.message)
+        return result.message
+
+    def list_files(self, pattern: str = "*.py", recursive: bool = True) -> list[str]:
+        """List files matching pattern (convenience method)"""
+        self.require(AgentCapability.READ_FILES)
+        result = self.file_ops.list_files(pattern=pattern, recursive=recursive)
+        if not result.success:
+            raise RuntimeError(result.message)
+        return result.message.split("\n") if result.message else []
+
+    def analyze(self, task: str) -> str:
+        """Analyze and plan (convenience method)"""
+        response = self.execute(task)
+        if not response.success:
+            raise RuntimeError(response.message)
+        return response.message

groknroll/cli/__init__.py

@@ -0,0 +1,7 @@
+"""
+groknroll CLI - Beautiful terminal interface
+"""
+
+from groknroll.cli.main import main
+
+__all__ = ["main"]

groknroll/cli/enhanced_cli.py

@@ -0,0 +1,372 @@
+"""
+Enhanced CLI with multi-agent support and new operations
+
+Adds build/plan agent switching, file operations, bash execution, and git integration.
+"""
+
+import sys
+from pathlib import Path
+from typing import Optional
+
+import click
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+from rich.markdown import Markdown
+from rich.syntax import Syntax
+
+from groknroll.agents.agent_manager import AgentManager
+from groknroll.operations.file_ops import FileOperations, FileEdit
+from groknroll.operations.bash_ops import BashOperations
+from groknroll.operations.git_ops import GitOperations
+from groknroll import __version__
+
+console = Console()
+
+
+@click.group(invoke_without_command=True)
+@click.option('--version', is_flag=True, help='Show version')
+@click.pass_context
+def main(ctx, version):
+    """
+    🎸 groknroll 2.0 - The Ultimate CLI Coding Agent
+
+    Now with multi-agent support, file operations, bash execution, and git integration!
+    """
+    if version:
+        console.print(f"[bold cyan]groknroll[/bold cyan] version {__version__}")
+        return
+
+    if ctx.invoked_subcommand is None:
+        console.print(ctx.get_help())
+
+
+# ============================================================================
+# Agent Commands
+# ============================================================================
+
+@main.command()
+@click.argument('task', nargs=-1, required=True)
+@click.option('--path', type=click.Path(exists=True), help='Project path')
+@click.option('--agent', type=click.Choice(['build', 'plan']), default='build',
+              help='Agent to use (build=full access, plan=read-only)')
+def build(task: tuple, path: Optional[str], agent: str):
+    """Execute task with build or plan agent"""
+    project_path = Path(path) if path else Path.cwd()
+    task_str = " ".join(task)
+
+    try:
+        with console.status(f"[bold cyan]{agent.title()} agent working...") as status:
+            manager = AgentManager(project_path)
+            response = manager.execute(task_str, agent=agent)
+
+        # Display response
+        console.print()
+        if response.success:
+            console.print(Panel(
+                Markdown(response.message),
+                title=f"[bold green]✓ {agent.title()} Agent[/bold green]",
+                border_style="green"
+            ))
+
+            # Show metrics
+            console.print(f"\n[dim]Completed in {response.time:.1f}s "
+                         f"(cost: ${response.cost:.4f})[/dim]")
+        else:
+            console.print(Panel(
+                response.message,
+                title=f"[bold red]✗ {agent.title()} Agent Failed[/bold red]",
+                border_style="red"
+            ))
+
+    except Exception as e:
+        console.print(f"[bold red]Error:[/bold red] {e}")
+        sys.exit(1)
+
+
+@main.command()
+@click.option('--path', type=click.Path(exists=True), help='Project path')
+def agents(path: Optional[str]):
+    """List available agents"""
+    project_path = Path(path) if path else Path.cwd()
+
+    try:
+        manager = AgentManager(project_path)
+        agent_list = manager.list_agents()
+
+        # Create table
+        table = Table(title="Available Agents", show_header=True)
+        table.add_column("Name", style="cyan")
+        table.add_column("Description", style="white")
+        table.add_column("Active", style="green")
+        table.add_column("Capabilities", style="dim")
+
+        for agent_info in agent_list:
+            table.add_row(
+                agent_info.name,
+                agent_info.description,
+                "✓" if agent_info.is_active else "",
+                ", ".join(agent_info.capabilities[:3]) + "..."
+            )
+
+        console.print()
+        console.print(table)
+
+    except Exception as e:
+        console.print(f"[bold red]Error:[/bold red] {e}")
+        sys.exit(1)
+
+
+# ============================================================================
+# File Operations Commands
+# ============================================================================
+
+@main.command()
+@click.argument('file_path', type=click.Path(exists=True))
+@click.option('--syntax', is_flag=True, help='Syntax highlighting')
+def read(file_path: str, syntax: bool):
+    """Read file contents"""
+    path = Path(file_path)
+
+    try:
+        file_ops = FileOperations(Path.cwd())
+        result = file_ops.read_file(path)
+
+        if result.success:
+            console.print()
+            if syntax:
+                # Detect language from extension
+                extension = path.suffix.lstrip('.')
+                syntax_obj = Syntax(result.message, extension or "text",
+                                   theme="monokai", line_numbers=True)
+                console.print(syntax_obj)
+            else:
+                console.print(Panel(
+                    result.message,
+                    title=f"[cyan]{path.name}[/cyan]",
+                    border_style="cyan"
+                ))
+        else:
+            console.print(f"[bold red]Error:[/bold red] {result.message}")
+            sys.exit(1)
+
+    except Exception as e:
+        console.print(f"[bold red]Error:[/bold red] {e}")
+        sys.exit(1)
+
+
+@main.command()
+@click.argument('file_path', type=click.Path())
+@click.argument('content')
+@click.option('--overwrite', is_flag=True, help='Overwrite if exists')
+def write(file_path: str, content: str, overwrite: bool):
+    """Write content to file"""
+    path = Path(file_path)
+
+    try:
+        file_ops = FileOperations(Path.cwd())
+        result = file_ops.write_file(path, content, overwrite=overwrite)
+
+        if result.success:
+            console.print(f"[bold green]✓[/bold green] {result.message}")
+            if result.backup_path:
+                console.print(f"[dim]Backup: {result.backup_path.name}[/dim]")
+        else:
+            console.print(f"[bold red]Error:[/bold red] {result.message}")
+            sys.exit(1)
+
+    except Exception as e:
+        console.print(f"[bold red]Error:[/bold red] {e}")
+        sys.exit(1)
+
+
+@main.command()
+@click.argument('file_path', type=click.Path(exists=True))
+@click.option('--backup/--no-backup', default=True, help='Create backup')
+def delete(file_path: str, backup: bool):
+    """Delete file"""
+    path = Path(file_path)
+
+    try:
+        file_ops = FileOperations(Path.cwd())
+        result = file_ops.delete_file(path, backup=backup)
+
+        if result.success:
+            console.print(f"[bold green]✓[/bold green] {result.message}")
+            if result.backup_path:
+                console.print(f"[dim]Backup: {result.backup_path.name}[/dim]")
+        else:
+            console.print(f"[bold red]Error:[/bold red] {result.message}")
+            sys.exit(1)
+
+    except Exception as e:
+        console.print(f"[bold red]Error:[/bold red] {e}")
+        sys.exit(1)
+
+
+# ============================================================================
+# Bash Operations Commands
+# ============================================================================
+
+@main.command()
+@click.argument('command')
+@click.option('--cwd', type=click.Path(exists=True), help='Working directory')
+@click.option('--timeout', type=int, default=300, help='Timeout in seconds')
+def run(command: str, cwd: Optional[str], timeout: int):
+    """Execute bash command"""
+    work_dir = Path(cwd) if cwd else Path.cwd()
+
+    try:
+        bash_ops = BashOperations(Path.cwd())
+        result = bash_ops.execute(command, cwd=work_dir, timeout=timeout)
+
+        console.print()
+        if result.success:
+            console.print(Panel(
+                result.stdout or "[dim]No output[/dim]",
+                title=f"[bold green]✓ Command: {command}[/bold green]",
+                border_style="green"
+            ))
+        else:
+            console.print(Panel(
+                result.stderr or result.stdout,
+                title=f"[bold red]✗ Command Failed (exit {result.exit_code})[/bold red]",
+                border_style="red"
+            ))
+            sys.exit(result.exit_code)
+
+    except Exception as e:
+        console.print(f"[bold red]Error:[/bold red] {e}")
+        sys.exit(1)
+
+
+# ============================================================================
+# Git Operations Commands
+# ============================================================================
+
+@main.command()
+@click.option('--path', type=click.Path(exists=True), help='Repository path')
+def git_status(path: Optional[str]):
+    """Show git status"""
+    repo_path = Path(path) if path else Path.cwd()
+
+    try:
+        git_ops = GitOperations(repo_path)
+        status = git_ops.status()
+
+        # Create status display
+        console.print()
+        console.print(Panel.fit(
+            f"[bold cyan]Branch:[/bold cyan] {status.branch}\n"
+            f"[cyan]Ahead:[/cyan] {status.ahead}  [cyan]Behind:[/cyan] {status.behind}\n"
+            f"[green]Staged:[/green] {len(status.staged)}  "
+            f"[yellow]Unstaged:[/yellow] {len(status.unstaged)}  "
+            f"[red]Untracked:[/red] {len(status.untracked)}",
+            title="[bold]Git Status[/bold]",
+            border_style="cyan"
+        ))
+
+        # Show file lists
+        if status.staged:
+            console.print("\n[bold green]Staged files:[/bold green]")
+            for file in status.staged[:10]:
+                console.print(f"  [green]+[/green] {file}")
+
+        if status.unstaged:
+            console.print("\n[bold yellow]Unstaged files:[/bold yellow]")
+            for file in status.unstaged[:10]:
+                console.print(f"  [yellow]M[/yellow] {file}")
+
+        if status.untracked:
+            console.print("\n[bold red]Untracked files:[/bold red]")
+            for file in status.untracked[:10]:
+                console.print(f"  [red]?[/red] {file}")
+
+    except Exception as e:
+        console.print(f"[bold red]Error:[/bold red] {e}")
+        sys.exit(1)
+
+
+@main.command()
+@click.argument('message')
+@click.option('--path', type=click.Path(exists=True), help='Repository path')
+@click.option('--all', 'all_files', is_flag=True, help='Stage all files')
+def git_commit(message: str, path: Optional[str], all_files: bool):
+    """Create git commit"""
+    repo_path = Path(path) if path else Path.cwd()
+
+    try:
+        git_ops = GitOperations(repo_path)
+
+        # Stage all if requested
+        if all_files:
+            add_result = git_ops.add(all_files=True)
+            if not add_result.success:
+                console.print(f"[bold red]Error staging files:[/bold red] {add_result.message}")
+                sys.exit(1)
+
+        # Commit
+        result = git_ops.commit(message)
+
+        if result.success:
+            console.print(f"[bold green]✓[/bold green] {result.message}")
+        else:
+            console.print(f"[bold red]Error:[/bold red] {result.message}")
+            sys.exit(1)
+
+    except Exception as e:
+        console.print(f"[bold red]Error:[/bold red] {e}")
+        sys.exit(1)
+
+
+@main.command()
+@click.option('--path', type=click.Path(exists=True), help='Repository path')
+@click.option('--remote', default='origin', help='Remote name')
+@click.option('--force', is_flag=True, help='Force push')
+def git_push(path: Optional[str], remote: str, force: bool):
+    """Push to remote"""
+    repo_path = Path(path) if path else Path.cwd()
+
+    try:
+        git_ops = GitOperations(repo_path)
+        result = git_ops.push(remote=remote, force=force)
+
+        if result.success:
+            console.print(f"[bold green]✓[/bold green] {result.message}")
+        else:
+            console.print(f"[bold red]Error:[/bold red] {result.message}")
+            sys.exit(1)
+
+    except Exception as e:
+        console.print(f"[bold red]Error:[/bold red] {e}")
+        sys.exit(1)
+
+
+@main.command()
+@click.argument('title')
+@click.option('--body', help='PR description')
+@click.option('--base', default='main', help='Base branch')
+@click.option('--draft', is_flag=True, help='Create as draft')
+@click.option('--path', type=click.Path(exists=True), help='Repository path')
+def create_pr(title: str, body: Optional[str], base: str, draft: bool, path: Optional[str]):
+    """Create pull request"""
+    repo_path = Path(path) if path else Path.cwd()
+
+    try:
+        git_ops = GitOperations(repo_path)
+        result = git_ops.create_pr(title=title, body=body, base=base, draft=draft)
+
+        if result.success:
+            console.print(f"[bold green]✓ PR created![/bold green]")
+            console.print(f"\n{result.output}")
+        else:
+            console.print(f"[bold red]Error:[/bold red] {result.message}")
+            sys.exit(1)
+
+    except Exception as e:
+        console.print(f"[bold red]Error:[/bold red] {e}")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()