martynattakit/CodeSentinel-CWE_Classification
1
1"""2pipeline/3Vulnerability classification pipeline.4 5Components:6 classifier.py — RoBERTa CWE classifier (text → CWE ID)7 code_analyzer.py — Qwen code analyzer (code → NL description)8 atlas_matcher.py — ATLAS pattern matcher (text → ATLAS technique)9 router.py — Main router (raw input → unified output card)10 11Usage:12 from pipeline.router import route13 result = route("def get_user(name): return db.execute('SELECT * FROM users WHERE name=' + name)")14"""15 16from pipeline.router import route, get_router17 18__all__ = ["route", "get_router"]19 