amd-gaia 0.15.0__py3-none-any.whl → 0.15.2__py3-none-any.whl

gaia/agents/code/cli.py

@@ -0,0 +1,377 @@
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
+# SPDX-License-Identifier: MIT
+
+"""CLI for Code Agent."""
+
+import argparse
+import json
+import logging
+import os
+import sys
+from pathlib import Path
+
+from rich.console import Console
+
+logger = logging.getLogger(__name__)
+console = Console()
+
+
+def _run_interactive_mode(agent, project_path, args, log):
+    """Run the interactive REPL loop for the Code Agent."""
+    while True:
+        try:
+            query = input("\ncode> ").strip()
+
+            if query.lower() in ["exit", "quit"]:
+                log.info("Goodbye!")
+                break
+
+            if query.lower() == "help":
+                print("\nAvailable commands:")
+                print("  Generate functions, classes, or tests")
+                print("  Analyze Python files")
+                print("  Validate Python syntax")
+                print("  Lint and format code")
+                print("  Edit files with diffs")
+                print("  Search for code patterns")
+                print("  Type 'exit' or 'quit' to end")
+                continue
+
+            if not query:
+                continue
+
+            # Process the query
+            result = agent.process_query(
+                query,
+                workspace_root=project_path,
+                max_steps=args.max_steps,
+                trace=args.trace,
+            )
+
+            # Display result
+            if not args.silent:
+                if result.get("status") == "success":
+                    log.info(f"\n✅ {result.get('result', 'Task completed')}")
+                else:
+                    log.error(f"\n❌ {result.get('result', 'Task failed')}")
+
+        except KeyboardInterrupt:
+            print("\n\nInterrupted. Type 'exit' to quit.")
+            continue
+        except Exception as e:
+            log.error(f"Error processing query: {e}")
+            if args.debug:
+                import traceback
+
+                traceback.print_exc()
+
+
+def cmd_run(args):
+    """Run the Code Agent with a query."""
+    from gaia.logger import get_logger
+
+    log = get_logger(__name__)
+
+    # Set logging level to DEBUG if --debug flag is used
+    if args.debug:
+        from gaia.logger import log_manager
+
+        # Set root logger level first to ensure all handlers process DEBUG messages
+        root_logger = logging.getLogger()
+        root_logger.setLevel(logging.DEBUG)
+
+        # Update all existing loggers that start with "gaia"
+        for logger_name in list(log_manager.loggers.keys()):
+            if logger_name.startswith("gaia"):
+                log_manager.loggers[logger_name].setLevel(logging.DEBUG)
+
+        # Set default level for future loggers
+        log_manager.set_level("gaia", logging.DEBUG)
+
+        # Also ensure all handlers have DEBUG level
+        for handler in root_logger.handlers:
+            handler.setLevel(logging.DEBUG)
+
+    # Check if code agent is available
+    try:
+        from gaia.agents.code.agent import CodeAgent  # noqa: F401
+
+        CODE_AVAILABLE = True
+    except ImportError:
+        CODE_AVAILABLE = False
+
+    if not CODE_AVAILABLE:
+        log.error("Code agent is not available. Please check your installation.")
+        return 1
+
+    # Get base_url from args or environment
+    base_url = args.base_url
+    if base_url is None:
+        base_url = os.getenv("LEMONADE_BASE_URL", "http://localhost:8000/api/v1")
+
+    # Initialize Lemonade with code agent profile (32768 context)
+    # Skip for remote servers (e.g., devtunnel URLs), external APIs, or --no-lemonade-check
+    is_local = "localhost" in base_url or "127.0.0.1" in base_url
+    skip_lemonade = args.no_lemonade_check
+    if is_local and not skip_lemonade:
+        from gaia.cli import initialize_lemonade_for_agent
+
+        success, _ = initialize_lemonade_for_agent(
+            agent="code",
+            skip_if_external=True,
+            use_claude=args.use_claude,
+            use_chatgpt=args.use_chatgpt,
+        )
+        if not success:
+            return 1
+
+    try:
+        # Import RoutingAgent for intelligent language detection
+        from gaia.agents.routing.agent import RoutingAgent
+
+        # Handle --path argument
+        project_path = args.path if hasattr(args, "path") else None
+        if project_path:
+            project_path = Path(project_path).expanduser().resolve()
+            # Create directory if it doesn't exist
+            project_path.mkdir(parents=True, exist_ok=True)
+            project_path = str(project_path)
+            log.debug(f"Using project path: {project_path}")
+
+        # Get the query to analyze
+        query = args.query if hasattr(args, "query") and args.query else None
+
+        # Use RoutingAgent to determine language and project type
+        if query:
+            # Prepare agent configuration from CLI args
+            agent_config = {
+                "silent_mode": args.silent,
+                "debug": args.debug,
+                "show_prompts": args.show_prompts,
+                "max_steps": args.max_steps,
+                "use_claude": args.use_claude,
+                "use_chatgpt": args.use_chatgpt,
+                "streaming": args.stream,
+                "base_url": args.base_url,
+                "skip_lemonade": args.no_lemonade_check,
+            }
+
+            # Single query mode - use routing with configuration
+            router = RoutingAgent(**agent_config)
+            agent = router.process_query(query)
+        else:
+            # Interactive mode - start with default Python agent
+            # User can still benefit from routing per query
+            agent = CodeAgent(
+                silent_mode=args.silent,
+                debug=args.debug,
+                show_prompts=args.show_prompts,
+                max_steps=args.max_steps,
+                use_claude=args.use_claude,
+                use_chatgpt=args.use_chatgpt,
+                streaming=args.stream,
+                base_url=args.base_url,
+                skip_lemonade=args.no_lemonade_check,
+            )
+
+        # Handle list tools option
+        if args.list_tools:
+            agent.list_tools(verbose=True)
+            return 0
+
+        # Handle interactive mode
+        if args.interactive:
+            log.info("🤖 Code Agent Interactive Mode")
+            log.info("Type 'exit' or 'quit' to end the session")
+            log.info("Type 'help' for available commands\n")
+
+            _run_interactive_mode(agent, project_path, args, log)
+            return 0
+
+        # Single query mode
+        elif query:
+            result = agent.process_query(
+                query,
+                workspace_root=project_path,
+                max_steps=args.max_steps,
+                trace=args.trace,
+                step_through=args.step_through,
+            )
+
+            # Output result
+            if args.silent:
+                # In silent mode, output only JSON
+                print(json.dumps(result, indent=2))
+            else:
+                # Display formatted result
+                agent.display_result("Code Operation Result", result)
+
+            return 0 if result.get("status") == "success" else 1
+
+        else:
+            # Default to interactive mode when no query provided
+            log.info("Starting Code Agent interactive mode (type 'help' for commands)")
+
+            _run_interactive_mode(agent, project_path, args, log)
+            return 0
+
+    except Exception as e:
+        log.error(f"Error initializing Code agent: {e}")
+        if args.debug:
+            import traceback
+
+            traceback.print_exc()
+        return 1
+
+
+def main():
+    """Main CLI entry point."""
+    parser = argparse.ArgumentParser(
+        description="GAIA Code Agent - AI-powered code generation and analysis",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Examples:
+  # Generate code with a query
+  gaia-code "Build me a todo tracking app using typescript"
+
+  # Work in a specific directory
+  gaia-code "Build me a todo app" --path ~/src/todo-app
+
+  # Interactive mode
+  gaia-code --interactive
+  gaia-code -i
+
+  # List available tools
+  gaia-code --list-tools
+
+  # Use external LLM APIs
+  gaia-code "Build an app" --use-claude
+  gaia-code "Build an app" --use-chatgpt
+
+  # Debug mode
+  gaia-code "Build an app" --debug
+        """,
+    )
+
+    # Positional argument - the code query
+    parser.add_argument(
+        "query",
+        nargs="?",
+        help="Code operation query (e.g., 'Build me a todo app')",
+    )
+
+    # Mode flags
+    parser.add_argument(
+        "--interactive",
+        "-i",
+        action="store_true",
+        help="Interactive mode for multiple queries",
+    )
+    parser.add_argument(
+        "--list-tools",
+        action="store_true",
+        help="List all available tools and exit",
+    )
+
+    # Project configuration
+    parser.add_argument(
+        "--path",
+        "-p",
+        type=str,
+        default=None,
+        help="Project directory path. Creates directory if it doesn't exist.",
+    )
+
+    # Debug and output options
+    parser.add_argument(
+        "--debug",
+        action="store_true",
+        help="Enable debug logging",
+    )
+    parser.add_argument(
+        "--silent",
+        "-s",
+        action="store_true",
+        help="Silent mode - suppress console output, return JSON only",
+    )
+    parser.add_argument(
+        "--step-through",
+        action="store_true",
+        help="Enable step-through debugging mode (pause at each agent step)",
+    )
+    parser.add_argument(
+        "--show-prompts",
+        action="store_true",
+        help="Display prompts sent to LLM",
+    )
+    parser.add_argument(
+        "--trace",
+        action="store_true",
+        help="Save conversation trace to JSON file",
+    )
+
+    # LLM backend options
+    parser.add_argument(
+        "--use-claude",
+        action="store_true",
+        help="Use Claude API instead of local Lemonade server",
+    )
+    parser.add_argument(
+        "--use-chatgpt",
+        action="store_true",
+        help="Use ChatGPT/OpenAI API instead of local Lemonade server",
+    )
+    parser.add_argument(
+        "--base-url",
+        default=None,
+        help="Lemonade server URL (default: http://localhost:8000/api/v1)",
+    )
+    parser.add_argument(
+        "--no-lemonade-check",
+        action="store_true",
+        help="Skip Lemonade server initialization check",
+    )
+
+    # Agent configuration
+    parser.add_argument(
+        "--max-steps",
+        type=int,
+        default=100,
+        help="Maximum conversation steps (default: 100)",
+    )
+    parser.add_argument(
+        "--stream",
+        action="store_true",
+        help="Enable streaming responses",
+    )
+
+    # Parse args
+    args = parser.parse_args()
+
+    # Configure logging - WARNING by default, DEBUG with --debug flag
+    if args.debug:
+        logging.basicConfig(
+            level=logging.DEBUG,
+            format="%(asctime)s | %(levelname)s | %(name)s | %(message)s",
+            datefmt="%H:%M:%S",
+        )
+    else:
+        # Suppress logs from gaia modules for cleaner output
+        logging.basicConfig(level=logging.WARNING)
+        for logger_name in ["gaia", "gaia.llm", "gaia.agents"]:
+            logging.getLogger(logger_name).setLevel(logging.WARNING)
+
+    # Run command
+    try:
+        return cmd_run(args)
+    except Exception as e:
+        console.print(f"[red]Error: {e}[/red]")
+        if args.debug:
+            import traceback
+
+            traceback.print_exc()
+        return 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())

gaia/agents/code/models.py

@@ -1,135 +1,135 @@
-#!/usr/bin/env python
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
-# SPDX-License-Identifier: MIT
-"""
-Data models for the Code Agent.
-
-This module contains all data classes and type definitions used across
-the code agent modules.
-"""
-
-import ast
-from dataclasses import dataclass, field
-from typing import Any, Dict, List, Optional
-
-
-@dataclass
-class CodeSymbol:
-    """Represents a code symbol (function, class, variable)."""
-
-    name: str
-    type: str  # 'function', 'class', 'variable', 'import'
-    line: int
-    docstring: Optional[str] = None
-    signature: Optional[str] = None
-
-
-@dataclass
-class ParsedCode:
-    """Result of parsing Python code."""
-
-    ast_tree: Optional[ast.Module] = None
-    symbols: List[CodeSymbol] = field(default_factory=list)
-    imports: List[str] = field(default_factory=list)
-    errors: List[str] = field(default_factory=list)
-    is_valid: bool = False
-
-
-@dataclass
-class ModuleSpec:
-    """Specification for a module in a project."""
-
-    name: str
-    purpose: str
-    classes: List[Dict[str, Any]] = field(default_factory=list)
-    functions: List[Dict[str, Any]] = field(default_factory=list)
-
-
-@dataclass
-class TestSpec:
-    """Specification for a test module."""
-
-    name: str
-    coverage: str
-    test_cases: List[str] = field(default_factory=list)
-
-
-@dataclass
-class ProjectPlan:
-    """Complete project plan with architecture and modules."""
-
-    name: str
-    architecture: Dict[str, Any]
-    modules: List[ModuleSpec]
-    tests: List[TestSpec]
-    description: str = ""
-    project_type: str = "application"
-
-
-@dataclass
-class ValidationResult:
-    """Result of code validation."""
-
-    is_valid: bool
-    errors: List[str] = field(default_factory=list)
-    warnings: List[str] = field(default_factory=list)
-    fixes_applied: List[str] = field(default_factory=list)
-    file_modified: bool = False
-
-
-@dataclass
-class MethodSpec:
-    """Specification for a class method."""
-
-    name: str
-    params: str = "self"
-    docstring: str = "Method description."
-    body: str = "pass"
-    return_type: Optional[str] = None
-
-
-@dataclass
-class ProjectStructure:
-    """Represents a complete project structure."""
-
-    name: str
-    files: Dict[str, str]  # filename -> content
-    structure: Dict[str, Any]  # nested directory structure
-    plan: Optional[ProjectPlan] = None
-
-
-@dataclass
-class WorkflowPlan:
-    """Plan for executing a coding workflow."""
-
-    query: str
-    steps: List[Dict[str, Any]]
-    current_step: int = 0
-    completed_steps: List[int] = field(default_factory=list)
-    status: str = "pending"  # pending, in_progress, completed, failed
-
-
-@dataclass
-class LintIssue:
-    """Represents a linting issue found by pylint."""
-
-    type: str  # error, warning, convention, refactor
-    message: str
-    file: str
-    line: int
-    column: int = 0
-    symbol: Optional[str] = None
-
-
-@dataclass
-class ExecutionResult:
-    """Result of executing Python code."""
-
-    stdout: str
-    stderr: str
-    return_code: int
-    has_errors: bool
-    duration_seconds: float
-    timed_out: bool = False
-    file_path: Optional[str] = None
-    command: Optional[str] = None
+#!/usr/bin/env python
+# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# SPDX-License-Identifier: MIT
+"""
+Data models for the Code Agent.
+
+This module contains all data classes and type definitions used across
+the code agent modules.
+"""
+
+import ast
+from dataclasses import dataclass, field
+from typing import Any, Dict, List, Optional
+
+
+@dataclass
+class CodeSymbol:
+    """Represents a code symbol (function, class, variable)."""
+
+    name: str
+    type: str  # 'function', 'class', 'variable', 'import'
+    line: int
+    docstring: Optional[str] = None
+    signature: Optional[str] = None
+
+
+@dataclass
+class ParsedCode:
+    """Result of parsing Python code."""
+
+    ast_tree: Optional[ast.Module] = None
+    symbols: List[CodeSymbol] = field(default_factory=list)
+    imports: List[str] = field(default_factory=list)
+    errors: List[str] = field(default_factory=list)
+    is_valid: bool = False
+
+
+@dataclass
+class ModuleSpec:
+    """Specification for a module in a project."""
+
+    name: str
+    purpose: str
+    classes: List[Dict[str, Any]] = field(default_factory=list)
+    functions: List[Dict[str, Any]] = field(default_factory=list)
+
+
+@dataclass
+class TestSpec:
+    """Specification for a test module."""
+
+    name: str
+    coverage: str
+    test_cases: List[str] = field(default_factory=list)
+
+
+@dataclass
+class ProjectPlan:
+    """Complete project plan with architecture and modules."""
+
+    name: str
+    architecture: Dict[str, Any]
+    modules: List[ModuleSpec]
+    tests: List[TestSpec]
+    description: str = ""
+    project_type: str = "application"
+
+
+@dataclass
+class ValidationResult:
+    """Result of code validation."""
+
+    is_valid: bool
+    errors: List[str] = field(default_factory=list)
+    warnings: List[str] = field(default_factory=list)
+    fixes_applied: List[str] = field(default_factory=list)
+    file_modified: bool = False
+
+
+@dataclass
+class MethodSpec:
+    """Specification for a class method."""
+
+    name: str
+    params: str = "self"
+    docstring: str = "Method description."
+    body: str = "pass"
+    return_type: Optional[str] = None
+
+
+@dataclass
+class ProjectStructure:
+    """Represents a complete project structure."""
+
+    name: str
+    files: Dict[str, str]  # filename -> content
+    structure: Dict[str, Any]  # nested directory structure
+    plan: Optional[ProjectPlan] = None
+
+
+@dataclass
+class WorkflowPlan:
+    """Plan for executing a coding workflow."""
+
+    query: str
+    steps: List[Dict[str, Any]]
+    current_step: int = 0
+    completed_steps: List[int] = field(default_factory=list)
+    status: str = "pending"  # pending, in_progress, completed, failed
+
+
+@dataclass
+class LintIssue:
+    """Represents a linting issue found by pylint."""
+
+    type: str  # error, warning, convention, refactor
+    message: str
+    file: str
+    line: int
+    column: int = 0
+    symbol: Optional[str] = None
+
+
+@dataclass
+class ExecutionResult:
+    """Result of executing Python code."""
+
+    stdout: str
+    stderr: str
+    return_code: int
+    has_errors: bool
+    duration_seconds: float
+    timed_out: bool = False
+    file_path: Optional[str] = None
+    command: Optional[str] = None

gaia/agents/code/orchestration/__init__.py

@@ -1,24 +1,24 @@
-# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
-# SPDX-License-Identifier: MIT
-"""
-Orchestration framework for Code Agent.
-
-This module provides LLM-driven workflow orchestration using checklist mode.
-The LLM generates a checklist of template invocations, which are then
-executed deterministically with error recovery.
-"""
-
-from .orchestrator import ExecutionResult, Orchestrator
-from .steps.base import BaseStep, ErrorCategory, StepResult, StepStatus, UserContext
-
-__all__ = [
-    # Core
-    "Orchestrator",
-    "ExecutionResult",
-    # Steps
-    "BaseStep",
-    "StepResult",
-    "StepStatus",
-    "ErrorCategory",
-    "UserContext",
-]
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
+# SPDX-License-Identifier: MIT
+"""
+Orchestration framework for Code Agent.
+
+This module provides LLM-driven workflow orchestration using checklist mode.
+The LLM generates a checklist of template invocations, which are then
+executed deterministically with error recovery.
+"""
+
+from .orchestrator import ExecutionResult, Orchestrator
+from .steps.base import BaseStep, ErrorCategory, StepResult, StepStatus, UserContext
+
+__all__ = [
+    # Core
+    "Orchestrator",
+    "ExecutionResult",
+    # Steps
+    "BaseStep",
+    "StepResult",
+    "StepStatus",
+    "ErrorCategory",
+    "UserContext",
+]