diff --git a/src/main.py b/src/main.py index ca47123..684b389 100644 --- a/src/main.py +++ b/src/main.py @@ -31,7 +31,7 @@ from src.rate_limiter import limiter from src.api import api_router from src.database.core import engine, async_session, collector_async_session from src.exceptions import handle_exception - +from src.middleware import RequestValidationMiddleware log = logging.getLogger(__name__) @@ -58,6 +58,8 @@ _request_id_ctx_var: ContextVar[Optional[str]] = ContextVar( def get_request_id() -> Optional[str]: return _request_id_ctx_var.get() +app.add_middleware(RequestValidationMiddleware) + @app.middleware("http") async def db_session_middleware(request: Request, call_next):