refactor: use context manager for request logging

Author: arthurhenrique

AGENTS.md

@@ -0,0 +1,2 @@
+- Always run `pytest {{cookiecutter.project_slug}}/tests` and ensure all tests pass after any code changes.
+- Run tests even if not explicitly requested.

{{cookiecutter.project_slug}}/.env.example

@@ -3,3 +3,4 @@ DEBUG=True
 MODEL_PATH={{cookiecutter.machine_learn_model_path}}
 MODEL_NAME={{cookiecutter.machine_learn_model_name}}
 MEMOIZATION_FLAG=False
+DATABASE_URL=postgresql://postgres:postgres@db:5432/app

{{cookiecutter.project_slug}}/app/api/routes/predictor.py

@@ -5,6 +5,9 @@
 from core.config import INPUT_EXAMPLE
 from fastapi import APIRouter, HTTPException
 from fastapi.concurrency import run_in_threadpool
+from loguru import logger
+from db import SessionLocal
+from models.log import RequestLog
 from models.prediction import (
     HealthResponse,
     MachineLearningDataInput,
@@ -44,10 +47,24 @@ async def predict(data_input: MachineLearningDataInput):
     except Exception as err:
         raise HTTPException(status_code=500, detail=f"Exception: {err}") from err
 
-    return MachineLearningResponse(
+    response = MachineLearningResponse(
         prediction=prediction, prediction_label=prediction_label
     )
 
+    try:
+        with SessionLocal() as db:
+            db.add(
+                RequestLog(
+                    request=json.dumps(data_input.model_dump()),
+                    response=json.dumps(response.model_dump()),
+                )
+            )
+            db.commit()
+    except Exception:
+        logger.exception("failed to log request")
+
+    return response
+
 
 @router.get(
     "/health",

{{cookiecutter.project_slug}}/app/core/config.py

@@ -15,6 +15,7 @@
 MIN_CONNECTIONS_COUNT: int = config("MIN_CONNECTIONS_COUNT", cast=int, default=10)
 SECRET_KEY: Secret = config("SECRET_KEY", cast=Secret, default="")
 MEMOIZATION_FLAG: bool = config("MEMOIZATION_FLAG", cast=bool, default=True)
+DATABASE_URL: str = config("DATABASE_URL", default="sqlite:///./app.db")
 
 PROJECT_NAME: str = config("PROJECT_NAME", default="{{cookiecutter.project_name}}")
 

{{cookiecutter.project_slug}}/app/core/events.py

@@ -2,6 +2,8 @@
 
 import joblib
 from fastapi import FastAPI
+from core.config import MEMOIZATION_FLAG
+from db import Base, engine
 
 
 def preload_model():
@@ -15,6 +17,8 @@ def preload_model():
 
 def create_start_app_handler(app: FastAPI) -> Callable:
     def start_app() -> None:
-        preload_model()
+        if MEMOIZATION_FLAG:
+            preload_model()
+        Base.metadata.create_all(bind=engine)
 
     return start_app

{{cookiecutter.project_slug}}/app/db.py

@@ -0,0 +1,8 @@
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker, declarative_base
+
+from core.config import DATABASE_URL
+
+engine = create_engine(DATABASE_URL)
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+Base = declarative_base()

{{cookiecutter.project_slug}}/app/main.py

@@ -7,8 +7,7 @@
 def get_application() -> FastAPI:
     application = FastAPI(title=PROJECT_NAME, debug=DEBUG, version=VERSION)
     application.include_router(api_router, prefix=API_PREFIX)
-    if MEMOIZATION_FLAG:
-        application.add_event_handler("startup", create_start_app_handler(application))
+    application.add_event_handler("startup", create_start_app_handler(application))
     return application
 
 

{{cookiecutter.project_slug}}/app/models/log.py

@@ -0,0 +1,11 @@
+from sqlalchemy import Column, Integer, Text
+
+from db import Base
+
+
+class RequestLog(Base):
+    __tablename__ = "request_logs"
+
+    id = Column(Integer, primary_key=True, index=True)
+    request = Column(Text, nullable=False)
+    response = Column(Text, nullable=False)

{{cookiecutter.project_slug}}/docker-compose.yml

@@ -13,4 +13,19 @@ services:
     command: uvicorn main:app --reload --host 0.0.0.0 --port 8080
     volumes:
       - ./app:/app/
-      - ./ml/model/:/app/ml/model/
+      - ./ml/model/:/app/ml/model/
+    depends_on:
+      - db
+  db:
+    image: postgres:16
+    environment:
+      POSTGRES_USER: postgres
+      POSTGRES_PASSWORD: postgres
+      POSTGRES_DB: app
+    ports:
+      - "5432:5432"
+    volumes:
+      - postgres_data:/var/lib/postgresql/data
+
+volumes:
+  postgres_data:

{{cookiecutter.project_slug}}/pyproject.toml

@@ -15,7 +15,9 @@ dependencies = [
     "joblib>=1.2.0",
     "scikit-learn>=1.1.3",
     "pandas>=2.2.3",
-    "httpx>=0.27.0"
+    "httpx>=0.27.0",
+    "sqlalchemy>=2.0.0",
+    "psycopg2-binary>=2.9.0"
 ]
 
 [project.optional-dependencies]