app/interceptor.py

class PromptInterceptor:
    def __init__(self):
        from app.detectors.jailbreak import JailbreakDetector
        from app.detectors.toxicity import ToxicityDetector
        from app.detectors.faiss_injection import FAISSInjectionDetector

        self.jailbreak = JailbreakDetector()
        self.toxicity = ToxicityDetector()
        self.injection = FAISSInjectionDetector()

    def run_all(self, prompt: str) -> dict:
        results = {}

        results['detect_jailbreak'] = self.jailbreak.detect(prompt)
        results['detect_toxicity'] = self.toxicity.detect(prompt)
        results['detect_injection_vector'] = self.injection.detect(prompt)

        return results