add- timing

src/server/main.py

@@ -453,6 +453,7 @@ class ChatRequest(BaseModel):
             raise ValueError(f"Unsupported language code: {v}. Supported codes: {', '.join(SUPPORTED_LANGUAGES)}")
         return v
 
+
 class ChatResponse(BaseModel):
     response: str
 
@@ -531,6 +532,7 @@ app = FastAPI(
     lifespan=lifespan
 )
 
+# Add CORS Middleware
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -539,6 +541,17 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Add Timing Middleware
+@app.middleware("http")
+async def add_request_timing(request: Request, call_next):
+    start_time = time()
+    response = await call_next(request)
+    end_time = time()
+    duration = end_time - start_time
+    logger.info(f"Request to {request.url.path} took {duration:.3f} seconds")
+    response.headers["X-Response-Time"] = f"{duration:.3f}"
+    return response
+
 limiter = Limiter(key_func=get_remote_address)
 app.state.limiter = limiter