Function ai_analyze

cli/main.py:662–713

Run scans and get AI-powered analysis.

(
    target: str = typer.Argument(..., help="Target to analyze"),
    provider: str = typer.Option("anthropic", "--provider", "-p", help="LLM provider (anthropic/openai/ollama/qwen/llama3/mistral)"),
    model: Optional[str] = typer.Option(None, "--model", "-m", help="Model name"),
    base_url: Optional[str] = typer.Option(None, "--base-url", "-b", help="Base URL for ollama or custom API"),
    quick: bool = typer.Option(False, "--quick", "-q", help="Quick scan (fewer modules)"),
)

    console.print()
    console.print("[bold]Usage Examples:[/bold]")
    console.print("  secsuite ai analyze example.com -p ollama -m qwen2.5")
    console.print("  secsuite ai analyze example.com -p ollama/llama3.2")
    console.print("  secsuite ai analyze example.com -p qwen")
    console.print("  secsuite ai analyze example.com -p mistral")
    console.print("  secsuite ai ask 'What is XSS?' -p ollama -m mistral")


@ai_app.command("analyze")
def ai_analyze(
    target: str = typer.Argument(..., help="Target to analyze"),
    provider: str = typer.Option("anthropic", "--provider", "-p", help="LLM provider (anthropic/openai/ollama/qwen/llama3/mistral)"),
    model: str | None = typer.Option(None, "--model", "-m", help="Model name"),
    base_url: str | None = typer.Option(None, "--base-url", "-b", help="Base URL for ollama or custom API"),
    quick: bool = typer.Option(False, "--quick", "-q", help="Quick scan (fewer modules)"),
):
    """Run scans and get AI-powered analysis."""
    setup_logging()

    console.print(f"[bold]AI Security Analysis:[/bold] {target}")
    console.print()

    # Determine which modules to run
    if quick:
        modules = ["dns", "headers", "tech", "ssl"]
    else:
        modules = ["dns", "whois", "headers", "tech", "ports", "ssl", "dirs"]

    console.print("[cyan]Running security scans...[/cyan]")
    results = _collect_scan_results(target, modules)

    if not results:
        console.print("[red]No scan results collected[/red]")
        return

    console.print(f"[green]Collected {len(results)} scan results[/green]")
    console.print()

    # Run AI analysis
    console.print("[cyan]Running AI analysis...[/cyan]")
    try:
        from modules.ai import SecurityCopilot

        copilot = SecurityCopilot(provider=provider, model=model, base_url=base_url)
        copilot.load_scan_results(results)

        analysis = run_async(copilot.analyze())

        console.print()
        console.print(Panel(analysis, title="[bold magenta]AI Security Analysis[/bold magenta]", border_style="magenta"))

    except ValueError as e:
        console.print(f"[red]Configuration error: {e}[/red]")
        console.print("[dim]For cloud: Set SECSUITE_ANTHROPIC_API_KEY or SECSUITE_OPENAI_API_KEY[/dim]")
        console.print("[dim]For local: Use -p ollama -m <model> (requires Ollama running)[/dim]")