Function ai_ask

cli/main.py:717–749

Ask the AI copilot a security question.

(
    question: str = typer.Argument(..., help="Question to ask about security"),
    target: Optional[str] = typer.Option(None, "--target", "-t", help="Target to scan first"),
    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"),
)

        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]")
    except ImportError as e:
        console.print(f"[red]Missing package: {e}[/red]")
    except Exception as e:
        console.print(f"[red]Error: {e}[/red]")
        if "ollama" in provider.lower():
            console.print("[dim]Make sure Ollama is running: ollama serve[/dim]")


@ai_app.command("ask")
def ai_ask(
    question: str = typer.Argument(..., help="Question to ask about security"),
    target: str | None = typer.Option(None, "--target", "-t", help="Target to scan first"),
    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"),
):
    """Ask the AI copilot a security question."""
    setup_logging()

    try:
        from modules.ai import SecurityCopilot

        copilot = SecurityCopilot(provider=provider, model=model, base_url=base_url)

        # If target provided, run quick scan first
        if target:
            console.print(f"[dim]Scanning {target}...[/dim]")
            results = _collect_scan_results(target, ["dns", "headers", "tech"])
            copilot.load_scan_results(results)

        console.print(f"[cyan]Question:[/cyan] {question}")
        console.print()

        response = run_async(copilot.ask(question))
        console.print(Panel(response, title="[bold magenta]AI Response[/bold magenta]", border_style="magenta"))