Add health checker

app/health/check_status.py

@@ -0,0 +1,32 @@
+# app/health/check_status.py
+from fastapi import APIRouter
+from app.db import db
+from bson.json_util import dumps
+import logging
+
+router = APIRouter()
+logger = logging.getLogger("book-query")
+
+@router.get("")
+async def get_status():
+    try:
+        docs = await db.documents.find().sort("_id", -1).limit(5).to_list(length=5)
+        doc_count = await db.documents.count_documents({})
+        embed_count = await db.embeddings.count_documents({})
+
+        return {
+            "status": "ok",
+            "documents_total": doc_count,
+            "embeddings_total": embed_count,
+            "recent_documents": [
+                {
+                    "id": doc.get("_id"),
+                    "title": doc.get("title"),
+                    "status": doc.get("status"),
+                }
+                for doc in docs
+            ]
+        }
+    except Exception as e:
+        logger.exception("❌ Health check failed")
+        return {"status": "error", "error": str(e)}

app/main.py

@@ -20,6 +20,7 @@
 # app/main.py
 from fastapi import FastAPI, WebSocket
 from app.routers import search, import_doc
+from app.health import check_status
 
 # Debugger
 import logging
@@ -37,6 +38,7 @@ app = FastAPI()
 
 app.include_router(search.router, prefix="/search")
 app.include_router(import_doc.router, prefix="/import")
+app.include_router(check_status.router, prefix="/health")
 
 @app.websocket("/ws/documents/{document_id}")
 async def websocket_endpoint(websocket: WebSocket, document_id: str):