update access-log (#9)

Author: robmsmt

backend/Dockerfile

@@ -10,4 +10,4 @@ RUN pip install --no-cache-dir -U -r backend/requirements.txt
 ENV PYTHONPATH=/app
 EXPOSE 8080
 
-ENTRYPOINT ["uvicorn", "backend.main:app", "--host", "0.0.0.0", "--port", "8080", "--workers", "4"]
+ENTRYPOINT ["uvicorn", "backend.main:app", "--host", "0.0.0.0", "--port", "8080", "--workers", "4", "--no-access-log"]

backend/main.py

@@ -57,4 +57,4 @@ async def lifespan(app: FastAPI):
 if __name__ == "__main__":
     import uvicorn
 
-    uvicorn.run(app, host="0.0.0.0", port=8080)
+    uvicorn.run(app, host="0.0.0.0", port=8080, access_log=False)

backend/middleware/logging.py

@@ -1,3 +1,4 @@
+import json
 import logging
 import time
 from fastapi import Request
@@ -9,6 +10,14 @@
 class AccessLogMiddleware(BaseHTTPMiddleware):
     async def dispatch(self, request: Request, call_next):
         start = time.time()
+        model = "-"
+        if request.method in ("POST", "PUT", "PATCH"):
+            try:
+                body = await request.body()
+                data = json.loads(body)
+                model = data.get("model", "-")
+            except Exception:
+                pass
         response = await call_next(request)
         duration_ms = (time.time() - start) * 1000
         token = request.headers.get("authorization", "")
@@ -17,10 +26,11 @@ async def dispatch(self, request: Request, call_next):
         else:
             token = "-"
         access_logger.info(
-            "%s %s %s %d %.0fms",
+            "%s %s %s model=%s %d %.0fms",
             token,
             request.method,
             request.url.path,
+            model,
             response.status_code,
             duration_ms,
         )