1 files changed, 6 insertions, 2 deletions
diff --git a/backend/middleware.py b/backend/middleware.py
index c1aa731..cf46dc6 100644
--- a/backend/middleware.py
+++ b/backend/middleware.py
@@ -1,12 +1,16 @@
-from starlette.middleware.base import BaseHTTPMiddleware
+import typing as t
+
 import pymongo
 import ssl
+from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.requests import Request
+from starlette.responses import Response
 
 from backend.constants import DATABASE_URL, MONGO_DATABASE
 
 
 class DatabaseMiddleware(BaseHTTPMiddleware):
-    async def dispatch(self, request, call_next):
+    async def dispatch(self, request: Request, call_next: t.Callable) -> Response:
         client = pymongo.MongoClient(
             DATABASE_URL,
             ssl_cert_reqs=ssl.CERT_NONE