feat(version): expose /version, include version in /healthz and startup logs; add --version flag

Author: Caripson

src/api.py

@@ -57,8 +57,12 @@ def Field(default=None, **_):  # type: ignore
 from src.metrics import decisions as decisions_store
 from src.policy.engine import PolicyEngine as _PE, Rule as _PRule, Event as _PEvent
 from scripts.setup_xevents import render_xevents_sql
+try:
+    from src.version import __version__
+except Exception:
+    __version__ = "0.0.0"
 
-app = FastAPI(title="SQLumAI Policy API", version="0.1.0")
+app = FastAPI(title="SQLumAI Policy API", version=__version__)
 
 RULES_PATH = os.getenv("RULES_PATH", "config/rules.json")
 PROPOSED_RULES_PATH = os.getenv("PROPOSED_RULES_PATH", "config/rules_proposed.json")
@@ -120,7 +124,12 @@ def delete_rule(rule_id: str):
 
 @app.get("/healthz")
 def healthz():
-    return {"status": "ok"}
+    return {"status": "ok", "version": __version__}
+
+
+@app.get("/version")
+def version():
+    return {"version": __version__}
 
 
 @app.get("/metrics")

src/main.py

@@ -2,6 +2,11 @@
 import os
 import signal
 from dotenv import load_dotenv
+import sys
+try:
+    from src.version import __version__
+except Exception:
+    __version__ = "0.0.0"
 
 from src.proxy.tds_proxy import run_proxy
 from src.proxy.tds_tls import run_tls_terminating_proxy
@@ -11,6 +16,8 @@
 
 async def main() -> None:
     load_dotenv()
+    # Startup log with version for visibility
+    print(f"[sqlumai] starting version {__version__}")
     listen_host = os.getenv("PROXY_LISTEN_ADDR", "0.0.0.0")
     listen_port = int(os.getenv("PROXY_LISTEN_PORT", "61433"))
     sql_host = os.getenv("SQL_HOST", "localhost")
@@ -53,6 +60,9 @@ def _handle_sig(*_):
 
 
 if __name__ == "__main__":
+    if "--version" in sys.argv or "-V" in sys.argv:
+        print(__version__)
+        raise SystemExit(0)
     try:
         asyncio.run(main())
     except KeyboardInterrupt:

src/runtime/api_runner.py

@@ -4,6 +4,11 @@
 
 
 async def run_api(stop_event: asyncio.Event):
+    try:
+        from src.version import __version__
+    except Exception:
+        __version__ = "0.0.0"
+    print(f"[sqlumai] API server starting (version {__version__})")
     config = uvicorn.Config("src.api:app", host=os.getenv("API_HOST", "0.0.0.0"), port=int(os.getenv("API_PORT", "8080")), log_level="info")
     server = uvicorn.Server(config)
 
@@ -15,4 +20,3 @@ async def _serve():
     if server.should_exit is False:
         server.should_exit = True
     await task
-