init ++

Author: saga4

codeflash/cli_cmds/cli.py

@@ -26,23 +26,23 @@ def parse_args() -> Namespace:
     # Integrate command - native integrations (replaces setup)
     integrate_parser = subparsers.add_parser("integrate", help="Integrate codeflash with development tools")
     integrate_subparsers = integrate_parser.add_subparsers(dest="integration_target", help="Integration targets")
-    
+
     # Claude Code integration
     claude_parser = integrate_subparsers.add_parser("claude", help="Integrate with Claude Code")
-    claude_parser.add_argument("--project", action="store_true", 
-                              help="Also create project-specific subagent")
-    claude_parser.add_argument("--force", action="store_true",
-                              help="Force integration even if Claude Code not detected")
-    claude_parser.add_argument("--remove", action="store_true",
-                              help="Remove Claude Code integration")
-    
+    claude_parser.add_argument("--project", action="store_true", help="Also create project-specific subagent")
+    claude_parser.add_argument(
+        "--force", action="store_true", help="Force integration even if Claude Code not detected"
+    )
+    claude_parser.add_argument("--remove", action="store_true", help="Remove Claude Code integration")
+
     # Status command
     status_parser = integrate_subparsers.add_parser("status", help="Show integration status")
-    
+
     def integration_handler():
         from codeflash.cli_cmds.cmd_integrate import handle_integration
+
         handle_integration()
-    
+
     integrate_parser.set_defaults(func=integration_handler)
 
     trace_optimize = subparsers.add_parser("optimize", help="Trace and optimize a Python project.")

codeflash/cli_cmds/cmd_integrate.py

@@ -8,11 +8,9 @@
 
 def handle_integration():
     """Handle integration subcommands."""
-    import argparse
-    
     # Get the arguments after 'codeflash integrate'
     args = sys.argv[2:]  # Skip 'codeflash integrate'
-    
+
     if not args or args[0] == "status":
         show_integration_status()
     elif args[0] == "claude":
@@ -26,66 +24,63 @@ def handle_integration():
 def handle_claude_integration(args):
     """Handle Claude Code integration."""
     import argparse
-    
+
     parser = argparse.ArgumentParser(prog="codeflash integrate claude")
-    parser.add_argument("--project", action="store_true", 
-                        help="Also create project-specific subagent")
-    parser.add_argument("--force", action="store_true",
-                        help="Force integration even if Claude Code not detected")
-    parser.add_argument("--remove", action="store_true",
-                        help="Remove Claude Code integration")
-    
+    parser.add_argument("--project", action="store_true", help="Also create project-specific subagent")
+    parser.add_argument("--force", action="store_true", help="Force integration even if Claude Code not detected")
+    parser.add_argument("--remove", action="store_true", help="Remove Claude Code integration")
+
     parsed_args = parser.parse_args(args)
-    
+
     integration = ClaudeCodeIntegration()
-    
+
     if parsed_args.remove:
         success, message = integration.remove_integration()
         print(message)
         sys.exit(0 if success else 1)
-    
+
     # Perform integration
     project_path = str(Path.cwd()) if parsed_args.project else None
     success = setup_claude_integration(
-        include_project=parsed_args.project,
-        project_path=project_path,
-        force=parsed_args.force
+        include_project=parsed_args.project, project_path=project_path, force=parsed_args.force
     )
-    
+
     sys.exit(0 if success else 1)
 
 
 def show_integration_status():
     """Show current integration status."""
     print("🔍 Codeflash Integration Status")
     print("=" * 40)
-    
+
     # Claude Code status
     integration = ClaudeCodeIntegration()
     status = integration.get_integration_status()
-    
+
     print("\n📋 Claude Code Integration:")
     print(f"  Claude Available: {'✅' if status['claude_available'] else '❌'}")
-    
-    if status['config_path']:
+
+    if status["config_path"]:
         print(f"  Config File: {status['config_path']}")
     else:
         print("  Config File: ❌ Not found")
-    
+
     print(f"  Agents Directory: {status['agents_directory']}")
     print(f"  Subagents Installed: {len(status['installed_subagents'])}/3")
-    for subagent in status['installed_subagents']:
+    for subagent in status["installed_subagents"]:
         print(f"    ✅ {subagent}")
-    
-    missing_subagents = set(["codeflash-optimizer", "codeflash-profiler", "codeflash-reviewer"]) - set(status['installed_subagents'])
+
+    missing_subagents = set(["codeflash-optimizer", "codeflash-profiler", "codeflash-reviewer"]) - set(
+        status["installed_subagents"]
+    )
     for subagent in missing_subagents:
         print(f"    ❌ {subagent}")
-    
+
     print(f"  MCP Tools: {'✅' if status['mcp_tools_configured'] else '❌'}")
     print(f"  System: {status['system']}")
-    
+
     # Overall status
-    if status['claude_available'] and status['all_subagents_installed']:
+    if status["claude_available"] and status["all_subagents_installed"]:
         print("\n🎉 Complete Claude Code integration is active!")
         print("\nAvailable subagents:")
         print("  @codeflash-optimizer - AI-powered code optimization")
@@ -95,17 +90,17 @@ def show_integration_status():
         print("  '@codeflash-optimizer optimize my slow function'")
         print("  '@codeflash-profiler find bottlenecks in my script'")
         print("  '@codeflash-reviewer review this code for performance'")
-    elif status['claude_available'] and status['installed_subagents']:
+    elif status["claude_available"] and status["installed_subagents"]:
         print(f"\n⚠️  Partial integration ({len(status['installed_subagents'])}/3 subagents)")
         print("  Run: codeflash integrate claude  # to install all subagents")
-    elif status['claude_available']:
+    elif status["claude_available"]:
         print("\n⚠️  Claude Code available but subagents not installed")
         print("  Run: codeflash integrate claude")
     else:
         print("\n📖 Install Claude Code to enable integration:")
         print("  Visit: https://docs.anthropic.com/en/docs/claude-code")
         print("  Then run: codeflash integrate claude")
-    
+
     # Check for project-specific integration
     project_agent = Path.cwd() / ".claude" / "agents" / "codeflash-optimizer.md"
     if project_agent.exists():
@@ -116,4 +111,4 @@ def show_integration_status():
 
 
 if __name__ == "__main__":
-    handle_integration()
+    handle_integration()

codeflash/cli_cmds/cmd_setup.py

@@ -1,11 +1,9 @@
 """Setup command for codeflash integrations and configurations."""
 
 import sys
-from typing import Optional
 
 import click
 
-from codeflash.cli_cmds.console import logger
 from codeflash.mcp.integration import ClaudeCodeIntegration
 
 
@@ -15,7 +13,7 @@
 @click.option("--remove", is_flag=True, help="Remove the integration")
 def setup_integrations(integration: str, force: bool = False, remove: bool = False):
     """Set up codeflash integrations with various tools and editors.
-    
+
     Available integrations:
     - claude-code: Set up as Claude Code MCP server subagent
     - status: Show current integration status
@@ -32,60 +30,60 @@ def setup_integrations(integration: str, force: bool = False, remove: bool = Fal
 def setup_claude_code(force: bool = False, remove: bool = False):
     """Set up Claude Code integration."""
     integration = ClaudeCodeIntegration()
-    
+
     if remove:
         success, message = integration.remove_integration()
         click.echo(message)
         sys.exit(0 if success else 1)
-    
+
     success, message = integration.setup_integration(force=force)
     click.echo(message)
-    
+
     if success:
         click.echo("\n🎉 Codeflash is now available as a Claude Code subagent!")
         click.echo("\nUsage examples in Claude Code:")
         click.echo("• 'Optimize the bubble_sort function using codeflash'")
-        click.echo("• 'Use codeflash to trace and optimize my main.py script'") 
+        click.echo("• 'Use codeflash to trace and optimize my main.py script'")
         click.echo("• 'Initialize codeflash in this project'")
         click.echo("• 'Run codeflash benchmarks to measure performance'")
         click.echo("\n📝 Restart Claude Code to load the new MCP server.")
-        
+
         # Show quick verification steps
         click.echo("\n🔍 Verification steps:")
         click.echo("1. Restart Claude Code")
         click.echo("2. In Claude Code, type: 'Show me codeflash help'")
         click.echo("3. Claude Code should respond with codeflash functionality")
-    
+
     sys.exit(0 if success else 1)
 
 
 def show_integration_status():
     """Show current integration status for all supported tools."""
     integration = ClaudeCodeIntegration()
     status = integration.get_status()
-    
+
     click.echo("🔍 Codeflash Integration Status\n")
-    
+
     # Claude Code status
     click.echo("📋 Claude Code:")
     if status["claude_code_installed"]:
         click.echo("  ✅ Claude Code is installed")
     else:
         click.echo("  ❌ Claude Code not found")
-    
+
     if status["config_file"]:
         click.echo(f"  📁 Config file: {status['config_file']}")
     else:
         click.echo("  📁 No config file found")
-    
+
     if status["codeflash_configured"]:
         click.echo("  ✅ Codeflash MCP server is configured")
     else:
         click.echo("  ❌ Codeflash MCP server not configured")
-    
+
     click.echo(f"  🔧 MCP executable: {status['mcp_executable']}")
     click.echo(f"  💻 System: {status['system']}")
-    
+
     # Overall status
     if status["claude_code_installed"] and status["codeflash_configured"]:
         click.echo("\n🎉 All integrations are working!")
@@ -99,10 +97,11 @@ def show_integration_status():
 
 # Function that can be called from main CLI
 def setup_claude_code_integration(force: bool = False):
-    """Setup function that can be called from main CLI.""" 
+    """Setup function that can be called from main CLI."""
     from codeflash.mcp.integration import setup_claude_code_integration as _setup
+
     return _setup(force=force)
 
 
 if __name__ == "__main__":
-    setup_integrations()
+    setup_integrations()

codeflash/install_hooks.py

@@ -1,36 +1,35 @@
 #!/usr/bin/env python3
 """Post-install hooks for codeflash to set up integrations automatically."""
 
-import os
 import sys
-from pathlib import Path
+
 
 def post_install_hook():
     """Run after codeflash installation to set up integrations."""
     try:
         # Only auto-setup in interactive environments
         if not sys.stdin.isatty():
             return
-            
+
         from codeflash.integrations.claude_code import ClaudeCodeIntegration
-        
+
         integration = ClaudeCodeIntegration()
-        
+
         # Check if Claude Code is available
         if not integration.is_claude_code_available():
             print("\n📖 Claude Code integration available!")
             print("   Install Claude Code: https://docs.anthropic.com/en/docs/claude-code")
             print("   Then run: codeflash integrate claude")
             return
-        
+
         # Check if already configured
         status = integration.get_integration_status()
         if status["subagent_installed"]:
             print("\n✅ Codeflash Claude Code integration is already active!")
             return
-        
+
         print("\n🚀 Claude Code detected! Setting up codeflash subagent...")
-        
+
         success, message = integration.setup_complete_integration()
         if success:
             print("✅ Codeflash subagent installed!")
@@ -39,15 +38,15 @@ def post_install_hook():
             print("  Or let Claude auto-invoke for performance tasks")
             print("\n📝 Restart Claude Code to activate the subagent")
         else:
-            print(f"⚠️  Auto-setup encountered issues:")
+            print("⚠️  Auto-setup encountered issues:")
             print(message)
             print("\n   Run manually: codeflash integrate claude")
-    
+
     except Exception as e:
         # Silently fail to avoid breaking installation
         print(f"\n⚠️  Auto-integration encountered an issue: {e}")
         print("   Run manually: codeflash integrate claude")
 
 
 if __name__ == "__main__":
-    post_install_hook()
+    post_install_hook()

codeflash/integrations/__init__.py

@@ -1,3 +1,3 @@
 """Claude Code and other integrations for codeflash."""
 
-__version__ = "1.0.0"
+__version__ = "1.0.0"