File size: 2,006 Bytes
fc51bb8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | import argparse
import sys
import os
import json
import requests
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
from app.core.config import settings
def check_health(base_url: str):
print(f"[*] Checking system health at {base_url}...")
try:
r = requests.get(f"{base_url}/health", timeout=5)
if r.status_code == 200:
print("[✓] Health Check PASSED:", json.dumps(r.json(), indent=2))
else:
print(f"[✗] Health Check FAILED with status code {r.status_code}")
except Exception as e:
print(f"[✗] Connection Error: {str(e)}")
def check_metrics(base_url: str):
print(f"[*] Fetching Prometheus metrics at {base_url}/metrics...")
try:
r = requests.get(f"{base_url}/metrics", timeout=5)
if r.status_code == 200:
lines = r.text.splitlines()
http_reqs = [l for l in lines if "http_requests_total" in l and not l.startswith("#")]
print("[✓] Prometheus Metrics Endpoint Online.")
print(f" Active Http Request Metrics Snippet ({len(http_reqs)} entries found):")
for line in http_reqs[:5]:
print(f" - {line}")
else:
print(f"[✗] Metrics Endpoint returned status code {r.status_code}")
except Exception as e:
print(f"[✗] Metrics Error: {str(e)}")
def main():
parser = argparse.ArgumentParser(description="12D AI Assessment Engine Ops CLI Utility")
parser.add_argument("command", choices=["health", "metrics", "version"], help="Command to execute")
parser.add_argument("--url", default="http://localhost:8000", help="Base URL of the Assessment Engine API")
args = parser.parse_args()
if args.command == "health":
check_health(args.url)
elif args.command == "metrics":
check_metrics(args.url)
elif args.command == "version":
print("12D AI Assessment Engine Version: 1.0.0-rc1")
if __name__ == "__main__":
main()
|