Initial curated RAGcore source from c16667f0c9397e2bfc8dd0255b114fba23c8d05e

scripts/check_app_health.py

@@ -0,0 +1,42 @@
+"""Wait for app liveness and readiness from a CI or Compose network."""
+
+from __future__ import annotations
+
+import argparse
+import time
+import urllib.request
+from urllib.parse import urlsplit
+
+
+def wait_for_health(base_url: str, timeout: float) -> None:
+    parsed = urlsplit(base_url)
+    if parsed.scheme not in {"http", "https"} or parsed.hostname is None:
+        raise ValueError("base URL must be an absolute HTTP(S) URL")
+
+    urls = (f"{base_url.rstrip('/')}/health/live", f"{base_url.rstrip('/')}/health/ready")
+    deadline = time.monotonic() + timeout
+    last_error = "unknown health failure"
+    while time.monotonic() < deadline:
+        try:
+            for url in urls:
+                with urllib.request.urlopen(url, timeout=1) as response:  # noqa: S310
+                    if response.status != 200:
+                        raise OSError(f"{url} returned HTTP {response.status}")
+            return
+        except OSError as exc:
+            last_error = str(exc)
+            time.sleep(0.5)
+    raise TimeoutError(f"Timed out waiting for app health: {last_error}")
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--base-url", default="http://127.0.0.1:8080")
+    parser.add_argument("--timeout", type=float, default=30)
+    args = parser.parse_args()
+    wait_for_health(args.base_url, args.timeout)
+    print("App liveness and readiness ready")
+
+
+if __name__ == "__main__":
+    main()