Update status_check.py

status_check.py

@@ -5,6 +5,9 @@ import os
 import requests
 from urllib.parse import urlparse
 
+from auth_helpers import hf_headers  # or same module where you defined it
+
+
     
 def resolve_endpoint() -> Optional[str]:
     uri = (os.environ.get("HF_ENDPOINT_URI") or "").strip()
@@ -17,11 +20,36 @@ def resolve_endpoint() -> Optional[str]:
     return uri
     
 
-def is_endpoint_healthy(uri: str, timeout: float = 5.0) -> Tuple[bool, str]:
+
+def is_endpoint_healthy(uri: Optional[str], timeout: float = 5.0) -> Tuple[bool, str]:
     if not uri:
-        return False, "No endpoint URI configured."
+        return False, "no URI configured"
+
+    headers = hf_headers()
+
+    # Try /health first (some endpoints support it)
+    try:
+        r = requests.get(f"{uri.rstrip('/')}/health", headers=headers, timeout=timeout)
+        if r.ok:
+            return True, "OK"
+        try:
+            msg = (r.json().get("error") or r.json().get("message"))  # type: ignore
+        except Exception:
+            msg = (r.text or "").strip()
+        if r.status_code in (401, 403):
+            return False, f"HTTP {r.status_code} – Unauthorized (check HF_TOKEN). {msg}"
+    except requests.RequestException:
+        pass
+
+    # Fallback: POST /
     try:
-        r = requests.post(uri, json={"inputs": "ping"}, timeout=timeout)
-        return (True, "OK") if r.ok else (False, f"HTTP {r.status_code}")
-    except requests.exceptions.RequestException as e:
-        return False, f"{type(e).__name__}"
+        r = requests.post(uri, headers=headers, json={"inputs": "ping"}, timeout=timeout)
+        if r.ok:
+            return True, "OK"
+        try:
+            msg = (r.json().get("error") or r.json().get("message"))  # type: ignore
+        except Exception:
+            msg = (r.text or "").strip()
+        return False, f"HTTP {r.status_code}" + (f" – {msg}" if msg else "")
+    except requests.RequestException as e:
+        return False, type(e).__name__