feat: add status check

Author: gusye1234

src/server/api/api.py

@@ -106,6 +106,11 @@ def custom_openapi():
     "/healthcheck", tags=["chore"], openapi_extra=API_X_CODE_DOCS["GET /healthcheck"]
 )(api_layer.chore.healthcheck)
 
+router.get(
+    "/admin/status_check",
+    tags=["admin"],
+    # openapi_extra=API_X_CODE_DOCS["GET /admin/status_check"],
+)(api_layer.chore.root_running_status_check)
 
 router.post(
     "/project/profile_config",

src/server/api/memobase_server/api_layer/chore.py

@@ -1,7 +1,11 @@
+import traceback
+from fastapi import HTTPException, Request
 from ..env import LOG
 from ..models.response import BaseResponse, CODE
+from ..models.database import DEFAULT_PROJECT_ID
 from ..connectors import db_health_check, redis_health_check
-from fastapi import HTTPException
+from ..llms.embeddings import check_embedding_sanity
+from ..llms import llm_sanity_check
 
 
 async def healthcheck() -> BaseResponse:
@@ -18,3 +22,32 @@ async def healthcheck() -> BaseResponse:
             detail="Redis not available",
         )
     return BaseResponse()
+
+
+async def root_running_status_check(request: Request) -> BaseResponse:
+    """Check if your memobase is set up correctly"""
+    project_id = request.state.memobase_project_id
+    if project_id != DEFAULT_PROJECT_ID:
+        return BaseResponse(
+            errno=CODE.METHOD_NOT_ALLOWED, errmsg="Only Root can access this"
+        )
+    try:
+        if not db_health_check():
+            return BaseResponse(
+                errno=CODE.INTERNAL_SERVER_ERROR,
+                errmsg="Database not available",
+            )
+        if not await redis_health_check():
+            return BaseResponse(
+                errno=CODE.INTERNAL_SERVER_ERROR,
+                errmsg="Redis not available",
+            )
+        await check_embedding_sanity()
+        await llm_sanity_check()
+    except Exception as e:
+        return BaseResponse(
+            errno=CODE.INTERNAL_SERVER_ERROR,
+            errmsg=f"Occuring Error when status checking: {e}\n{traceback.format_exc()}",
+        )
+
+    return BaseResponse()

src/server/api/memobase_server/api_layer/middleware.py

@@ -20,6 +20,8 @@
 
 
 PATH_MAPPINGS = [
+    "/api/v1/admin/status_check",
+    "/api/v1/users/blobs",
     "/api/v1/users/blobs",
     "/api/v1/users/profile",
     "/api/v1/users/buffer",