MartyNattakit
Reset and redo the project
9613bf1
raw
history blame contribute delete
583 Bytes
"""
pipeline/
Vulnerability classification pipeline.
Components:
classifier.py β€” RoBERTa CWE classifier (text β†’ CWE ID)
code_analyzer.py β€” Qwen code analyzer (code β†’ NL description)
atlas_matcher.py β€” ATLAS pattern matcher (text β†’ ATLAS technique)
router.py β€” Main router (raw input β†’ unified output card)
Usage:
from pipeline.router import route
result = route("def get_user(name): return db.execute('SELECT * FROM users WHERE name=' + name)")
"""
from pipeline.router import route, get_router
__all__ = ["route", "get_router"]