Added db wake up endpoint

main.py

@@ -28,6 +28,3 @@ app.include_router(holdings.router)
 async def root():
     return {"message": "Welcome to FinSight-AI"}
 
-@app.get("/health")
-async def health_check():
-    return {"status": "healthy"}

routers/user.py

@@ -1,6 +1,7 @@
 from fastapi import APIRouter, Depends, HTTPException
 from fastapi.security import OAuth2PasswordRequestForm
 from sqlalchemy.orm import Session
+from sqlalchemy import select, literal
 from models import User
 from database.connection import get_db
 from auth.password import hash_password, verify_password
@@ -35,4 +36,9 @@ async def login_for_access_token(form_data: OAuth2PasswordRequestForm = Depends(
 
 @router.get("/me", response_model=schemas.UserResponse)
 async def get_logged_in_user(current_user: User = Depends(get_current_user)):
-    return current_user
+    return current_user
+
+@router.get("/health",)
+async def health_check(db: Session = Depends(get_db)):
+    result = db.execute(select(literal(1))).scalar()
+    return {"status": "healthy"}