diagram-to-iac 0.6.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

diagram_to_iac/__init__.py

@@ -0,0 +1,10 @@
+try:
+    from diagram_to_iac.tools.sec_utils import load_yaml_secrets
+except Exception:  # noqa: BLE001
+    load_yaml_secrets = None
+
+if load_yaml_secrets:
+    try:
+        load_yaml_secrets()
+    except Exception:  # noqa: BLE001
+        pass

diagram_to_iac/actions/__init__.py

@@ -0,0 +1,7 @@
+# src/diagram_to_iac/actions/__init__.py
+"""
+Actions module for diagram-to-iac CLI entrypoints.
+
+This module contains command-line interfaces and action handlers for various
+agents and tools within the diagram-to-iac system.
+"""

diagram_to_iac/actions/git_entry.py

@@ -0,0 +1,174 @@
+#!/usr/bin/env python
+"""
+CLI entrypoint for GitAgent - Phase 5 Implementation
+
+This module provides a command-line interface for manual execution of the GitAgent,
+following the established patterns in the diagram-to-iac codebase.
+
+Usage:
+    python -m diagram_to_iac.actions.git_entry --repo-url https://github.com/user/repo
+    # Or via console script (after update_deps.py adds it to pyproject.toml):
+    git-agent --repo-url https://github.com/user/repo
+
+Features:
+- Argument parsing for git operations
+- JSON output for programmatic consumption
+- Proper error handling and logging
+- Integration with existing GitAgent implementation
+"""
+
+import argparse
+import json
+import sys
+import logging
+from typing import Optional
+
+from diagram_to_iac.agents.git_langgraph.agent import GitAgent, GitAgentInput
+
+
+def setup_logging(verbose: bool = False) -> None:
+    """Configure logging for the CLI entrypoint."""
+    level = logging.DEBUG if verbose else logging.INFO
+    logging.basicConfig(
+        level=level,
+        format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S'
+    )
+
+
+def create_argument_parser() -> argparse.ArgumentParser:
+    """Create and configure the argument parser following established patterns."""
+    parser = argparse.ArgumentParser(
+        prog='git-agent',
+        description='GitAgent CLI - DevOps automation for git operations',
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Examples:
+  %(prog)s --repo-url https://github.com/user/repo
+  %(prog)s --query "clone https://github.com/user/repo"
+  %(prog)s --query "open issue in repo: bug report" --verbose
+        """
+    )
+    
+    # Primary input options (mutually exclusive)
+    input_group = parser.add_mutually_exclusive_group(required=True)
+    input_group.add_argument(
+        '--repo-url',
+        type=str,
+        help='Git repository URL to clone'
+    )
+    input_group.add_argument(
+        '--query',
+        type=str,
+        help='DevOps query (git clone, GitHub issue, shell command)'
+    )
+    
+    # Optional arguments
+    parser.add_argument(
+        '--thread-id',
+        type=str,
+        help='Optional thread ID for conversation history'
+    )
+    parser.add_argument(
+        '--verbose',
+        action='store_true',
+        help='Enable verbose logging'
+    )
+    parser.add_argument(
+        '--config-path',
+        type=str,
+        help='Path to YAML configuration file'
+    )
+    parser.add_argument(
+        '--memory-type',
+        type=str,
+        choices=['persistent', 'memory', 'langgraph'],
+        default='persistent',
+        help='Type of memory to use (default: persistent)'
+    )
+    
+    return parser
+
+
+def format_output(result: object, indent: int = 2) -> str:
+    """Format output as JSON with proper indentation."""
+    try:
+        if hasattr(result, 'model_dump'):
+            # Pydantic model
+            return json.dumps(result.model_dump(), indent=indent)
+        else:
+            # Regular object
+            return json.dumps(result, indent=indent, default=str)
+    except Exception as e:
+        # Fallback to string representation
+        logging.warning(f"Failed to serialize output as JSON: {e}")
+        return str(result)
+
+
+def construct_query_from_repo_url(repo_url: str) -> str:
+    """Construct a git clone query from a repository URL."""
+    return f"clone repository {repo_url}"
+
+
+def main() -> int:
+    """Main entrypoint function following established CLI patterns."""
+    try:
+        # Parse arguments
+        parser = create_argument_parser()
+        args = parser.parse_args()
+        
+        # Setup logging
+        setup_logging(args.verbose)
+        logger = logging.getLogger(__name__)
+        
+        # Construct query from arguments
+        if args.repo_url:
+            query = construct_query_from_repo_url(args.repo_url)
+            logger.info(f"Constructed query from repo URL: {query}")
+        else:
+            query = args.query
+            logger.info(f"Using provided query: {query}")
+        
+        # Initialize agent with optional configuration
+        agent_kwargs = {}
+        if args.config_path:
+            agent_kwargs['config_path'] = args.config_path
+        if args.memory_type:
+            agent_kwargs['memory_type'] = args.memory_type
+            
+        logger.info(f"Initializing GitAgent with {agent_kwargs}")
+        agent = GitAgent(**agent_kwargs)
+        
+        # Prepare input
+        agent_input = GitAgentInput(
+            query=query,
+            thread_id=args.thread_id
+        )
+        
+        logger.info(f"Running agent with input: {agent_input.query}")
+        
+        # Execute agent
+        result = agent.run(agent_input)
+        
+        # Output results
+        print(format_output(result))
+        
+        # Return appropriate exit code
+        if result.error_message:
+            logger.error(f"Agent execution failed: {result.error_message}")
+            return 1
+        else:
+            logger.info("Agent execution completed successfully")
+            return 0
+            
+    except KeyboardInterrupt:
+        print("\n⚠️  Operation cancelled by user", file=sys.stderr)
+        return 130
+    except Exception as e:
+        print(f"❌ Error: {e}", file=sys.stderr)
+        logging.exception("Unexpected error in main()")
+        return 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())

diagram_to_iac/actions/supervisor_entry.py

@@ -0,0 +1,116 @@
+"""CLI entrypoint for SupervisorAgent."""
+
+import argparse
+import json
+import sys
+import logging
+from datetime import datetime
+from pathlib import Path
+
+from diagram_to_iac.agents.supervisor_langgraph import (
+    SupervisorAgent,
+    SupervisorAgentInput,
+)
+from diagram_to_iac.services import get_log_path, generate_step_summary, reset_log_bus
+
+
+def setup_logging(verbose: bool = False) -> None:
+    level = logging.DEBUG if verbose else logging.INFO
+    logging.basicConfig(
+        level=level,
+        format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+        datefmt="%Y-%m-%d %H:%M:%S",
+    )
+
+
+def create_argument_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(
+        prog="supervisor-agent",
+        description="SupervisorAgent CLI - R2D (Repo-to-Deployment) automation",
+    )
+    parser.add_argument("--repo-url", help="Repository URL to operate on")
+    parser.add_argument("--branch-name", help="Branch name (deprecated - supervisor skips branch creation)")
+    parser.add_argument("--thread-id", help="Optional thread id")
+    parser.add_argument("--verbose", action="store_true", help="Enable verbose logging")
+    parser.add_argument("--no-interactive", action="store_true", help="Skip interactive prompts")
+    parser.add_argument("--dry-run", action="store_true", help="Print issue text instead of creating it")
+    return parser
+
+
+def prompt_for_repo_url() -> str:
+    """Prompt the user for a repository URL."""
+    try:
+        return input("Repository URL: ").strip()
+    except (KeyboardInterrupt, EOFError):
+        print("\n⚠️  Repository URL required")
+        sys.exit(1)
+
+
+def format_output(result: object) -> str:
+    try:
+        if hasattr(result, "model_dump"):
+            return json.dumps(result.model_dump(), indent=2)
+        return json.dumps(result, indent=2, default=str)
+    except Exception as e:
+        logging.warning(f"Failed to serialize output: {e}")
+        return str(result)
+
+
+def main() -> int:
+    parser = create_argument_parser()
+    args = parser.parse_args()
+
+    setup_logging(args.verbose)
+
+    # Handle repo url
+    repo_url = args.repo_url
+    if not repo_url and not args.no_interactive:
+        repo_url = prompt_for_repo_url()
+    elif not repo_url:
+        parser.error("--repo-url is required when --no-interactive is used")
+
+    # Branch name is no longer used since supervisor skips branch creation
+    # All errors are handled via GitHub issues instead
+    branch_name = args.branch_name or "main"  # Placeholder for compatibility
+
+    agent = SupervisorAgent()
+
+
+    while True:
+        reset_log_bus()
+        result = agent.run(
+            SupervisorAgentInput(
+                repo_url=repo_url,
+                branch_name=branch_name,
+                thread_id=args.thread_id,
+                dry_run=args.dry_run,
+            )
+
+        )
+
+        print(format_output(result))
+
+        try:
+            generate_step_summary(get_log_path(), Path("step-summary.md"))
+        except Exception as e:  # noqa: BLE001
+            logging.warning(f"Step summary generation failed: {e}")
+
+        if result.success or args.no_interactive:
+            break
+
+        try:
+            choice = input("Retry workflow? [y/N]: ").strip().lower()
+        except (KeyboardInterrupt, EOFError):
+            choice = ""
+
+        if choice != "y":
+            break
+
+        repo_url = prompt_for_repo_url()
+        # No longer prompt for branch name since supervisor skips branch creation
+
+    return 0 if result.success else 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())

diagram_to_iac/actions/terraform_agent_entry.py

@@ -0,0 +1,207 @@
+#!/usr/bin/env python
+"""
+CLI entrypoint for TerraformAgent - Phase 4 Implementation
+
+This module provides a command-line interface for manual execution of the TerraformAgent,
+following the established patterns in the diagram-to-iac codebase.
+
+Usage:
+    python -m diagram_to_iac.actions.terraform_agent_entry --query "terraform init"
+    python -m diagram_to_iac.actions.terraform_agent_entry --repo-path /workspace --operation init
+    # Or via console script (after update_deps.py adds it to pyproject.toml):
+    terraform-agent --query "terraform apply configuration"
+
+Features:
+- Argument parsing for terraform operations
+- JSON output for programmatic consumption
+- Proper error handling and logging
+- Integration with existing TerraformAgent implementation
+- Support for init, plan, apply operations
+- GitHub issue creation for critical errors
+"""
+
+import argparse
+import json
+import sys
+import logging
+from typing import Optional
+
+from diagram_to_iac.agents.terraform_langgraph.agent import TerraformAgent, TerraformAgentInput
+
+
+def setup_logging(verbose: bool = False) -> None:
+    """Configure logging for the CLI entrypoint."""
+    level = logging.DEBUG if verbose else logging.INFO
+    logging.basicConfig(
+        level=level,
+        format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S'
+    )
+
+
+def create_argument_parser() -> argparse.ArgumentParser:
+    """Create and configure the argument parser following established patterns."""
+    parser = argparse.ArgumentParser(
+        prog='terraform-agent',
+        description='TerraformAgent CLI - DevOps automation for terraform operations',
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Examples:
+  %(prog)s --query "terraform init"
+  %(prog)s --query "terraform plan configuration"
+  %(prog)s --query "terraform apply infrastructure"
+  %(prog)s --repo-path /workspace --operation init --verbose
+  %(prog)s --query "open issue for terraform authentication failure"
+        """
+    )
+    
+    # Primary input options (mutually exclusive)
+    input_group = parser.add_mutually_exclusive_group(required=True)
+    input_group.add_argument(
+        '--query',
+        type=str,
+        help='Terraform DevOps query (init, plan, apply, or error reporting)'
+    )
+    input_group.add_argument(
+        '--repo-path',
+        type=str,
+        help='Repository path for terraform operations (requires --operation)'
+    )
+    
+    # Operation type (required when using --repo-path)
+    parser.add_argument(
+        '--operation',
+        type=str,
+        choices=['init', 'plan', 'apply'],
+        help='Terraform operation type (required with --repo-path)'
+    )
+    
+    # Optional arguments
+    parser.add_argument(
+        '--thread-id',
+        type=str,
+        help='Optional thread ID for conversation history'
+    )
+    parser.add_argument(
+        '--context',
+        type=str,
+        help='Optional context information for error reporting'
+    )
+    parser.add_argument(
+        '--verbose',
+        action='store_true',
+        help='Enable verbose logging'
+    )
+    parser.add_argument(
+        '--config-path',
+        type=str,
+        help='Path to YAML configuration file'
+    )
+    parser.add_argument(
+        '--memory-type',
+        type=str,
+        choices=['persistent', 'memory', 'langgraph'],
+        default='persistent',
+        help='Type of memory to use (default: persistent)'
+    )
+    
+    return parser
+
+
+def format_output(result: object, indent: int = 2) -> str:
+    """Format output as JSON with proper indentation."""
+    try:
+        if hasattr(result, 'model_dump'):
+            # Pydantic model
+            return json.dumps(result.model_dump(), indent=indent)
+        else:
+            # Regular object
+            return json.dumps(result, indent=indent, default=str)
+    except Exception as e:
+        # Fallback to string representation
+        logging.warning(f"Failed to serialize output as JSON: {e}")
+        return str(result)
+
+
+def construct_query_from_repo_path(repo_path: str, operation: str) -> str:
+    """Construct a terraform query from repository path and operation."""
+    operation_templates = {
+        'init': f"terraform init in {repo_path}",
+        'plan': f"terraform plan in {repo_path}",
+        'apply': f"terraform apply in {repo_path}"
+    }
+    return operation_templates.get(operation, f"terraform {operation} in {repo_path}")
+
+
+def validate_arguments(args: argparse.Namespace) -> None:
+    """Validate argument combinations following terraform agent requirements."""
+    if args.repo_path and not args.operation:
+        raise ValueError("--operation is required when using --repo-path")
+
+
+def main() -> int:
+    """Main entrypoint function following established CLI patterns."""
+    try:
+        # Parse arguments
+        parser = create_argument_parser()
+        args = parser.parse_args()
+        
+        # Validate argument combinations
+        validate_arguments(args)
+        
+        # Setup logging
+        setup_logging(args.verbose)
+        logger = logging.getLogger(__name__)
+        
+        # Construct query from arguments
+        if args.repo_path:
+            query = construct_query_from_repo_path(args.repo_path, args.operation)
+            logger.info(f"Constructed query from repo path: {query}")
+        else:
+            query = args.query
+            logger.info(f"Using provided query: {query}")
+        
+        # Initialize agent with optional configuration
+        agent_kwargs = {}
+        if args.config_path:
+            agent_kwargs['config_path'] = args.config_path
+        if args.memory_type:
+            agent_kwargs['memory_type'] = args.memory_type
+            
+        logger.info(f"Initializing TerraformAgent with {agent_kwargs}")
+        agent = TerraformAgent(**agent_kwargs)
+        
+        # Prepare input
+        agent_input = TerraformAgentInput(
+            query=query,
+            thread_id=args.thread_id,
+            context=args.context
+        )
+        
+        logger.info(f"Running agent with input: {agent_input.query}")
+        
+        # Execute agent
+        result = agent.run(agent_input)
+        
+        # Output results
+        print(format_output(result))
+        
+        # Return appropriate exit code
+        if result.error_message:
+            logger.error(f"Agent execution failed: {result.error_message}")
+            return 1
+        else:
+            logger.info("Agent execution completed successfully")
+            return 0
+            
+    except KeyboardInterrupt:
+        print("\n⚠️  Operation cancelled by user", file=sys.stderr)
+        return 130
+    except Exception as e:
+        print(f"❌ Error: {e}", file=sys.stderr)
+        logging.exception("Unexpected error in main()")
+        return 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())

diagram_to_iac/agents/__init__.py

@@ -0,0 +1,26 @@
+# This file makes the 'agents' module a package
+# and can be used to expose specific agents or functionalities.
+
+# Import agents from their new organized locations
+from .hello_langgraph.agent import HelloAgent
+from .git_langgraph.agent import GitAgent
+from .supervisor_langgraph.agent import (
+    SupervisorAgent,
+    SupervisorAgentInput,
+    SupervisorAgentOutput,
+)
+
+# Maintain backward compatibility with old names
+HelloLangGraphAgent = HelloAgent
+GitAgent = GitAgent
+
+# Export all agent classes
+__all__ = [
+    "HelloAgent",
+    "GitAgent",
+    "HelloLangGraphAgent",  # backward compatibility
+    "GitAgent",             # backward compatibility
+    "SupervisorAgent",
+    "SupervisorAgentInput",
+    "SupervisorAgentOutput",
+]

diagram_to_iac/agents/demonstrator_langgraph/__init__.py

@@ -0,0 +1,10 @@
+"""
+DemonstratorAgent for intelligent interactive dry-run demonstrations.
+
+This agent handles all dry-run interactions in an organic, agentic way,
+keeping the SupervisorAgent clean and focused.
+"""
+
+from .agent import DemonstratorAgent, DemonstratorAgentInput, DemonstratorAgentOutput
+
+__all__ = ["DemonstratorAgent", "DemonstratorAgentInput", "DemonstratorAgentOutput"]