info router

app/api/info.py

@@ -0,0 +1,13 @@
+from fastapi import APIRouter
+import subprocess
+
+router = APIRouter()
+
+@router.get("/info")
+async def get_build_number():
+    try:
+        # Run the "git rev-parse HEAD" command to get the latest commit hash
+        commit_hash = subprocess.check_output(["git", "rev-parse", "HEAD"]).decode().strip()
+        return {"build_number": commit_hash}
+    except Exception as e:
+        return {"error": str(e)}

app/main.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI
-from app.api import upload, process  # Import your endpoint modules
+from app.api import upload, process, info  # Import your endpoint modules
 
 from pydantic import BaseSettings
 
@@ -10,6 +10,7 @@
 # Include your API endpoint routers here
 app.include_router(upload.router, prefix="", tags=["dataset"])
 app.include_router(process.router, prefix="", tags=["process"])
+app.include_router(info.router, prefix="", tags=["info"])
 
 for route in app.routes:
     print(f"Route: {route.path} | Methods: {route.methods}")