"""Prompt templates for PentestAgent agents."""

from pathlib import Path

from jinja2 import Template

PROMPTS_DIR = Path(__file__).parent


def load_prompt(name: str) -> Template:
    """Load a prompt template by name.

    Args:
        name: Prompt name without extension (e.g., 'pa_agent', 'pa_assist')

    Returns:
        Jinja2 Template object
    """
    path = PROMPTS_DIR / f"{name}.jinja"
    return Template(path.read_text(encoding="utf-8"))


# Pre-loaded templates for convenience
pa_agent = load_prompt("pa_agent")
pa_assist = load_prompt("pa_assist")
pa_crew = load_prompt("pa_crew")
pa_interact = load_prompt("pa_interact")
pa_mcp = load_prompt("pa_mcp")