From 8d6618487fd20fc2f3a6e1c9202036437084cc25 Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Fri, 16 Aug 2024 21:31:43 +0300 Subject: [PATCH] fix(middleware): --- agentic_security/app.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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}"