INNER CODE UNIT · Python

classify_vuln_class

openhackai/OpenHack · openhack/agents/specialists/__init__.py:45

def classify_vuln_class(category_or_notes: str) -> str:
    """Map a vuln tag/category/free-text hint to one of the specialist keys."""
    if not category_or_notes:
        return "injection"
    text = str(category_or_notes).lower()
    for needles, key in _CLASSIFY_RULES:
        if any(n in text for n in needles):
            return key
    return "injection"


def build_specialist(
    vuln_class: str,
    target_dir,
    session,
    model=None,
    base_url: str = "",
):

View source record →