diff --git a/agentic_security/app.py b/agentic_security/app.py index d2bd126..b4e4a92 100644 --- a/agentic_security/app.py +++ b/agentic_security/app.py @@ -226,7 +226,11 @@ config.dictConfig( class LogNon200ResponsesMiddleware(BaseHTTPMiddleware): async def dispatch(self, request: Request, call_next): - response = await call_next(request) + try: + response = await call_next(request) + except Exception as e: + logger.exception("Yikes") + raise e if response.status_code != 200: logger.error( f"{request.method} {request.url} - Status code: {response.status_code}"