dh-cli 0.4.1__tar.gz → 0.4.2__tar.gz

{dh_cli-0.4.1 → dh_cli-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dh-cli
-Version: 0.4.1
+Version: 0.4.2
 Summary: Dayhoff Labs developer CLI
 Author-email: Dayhoff Labs <dev@dayhofflabs.com>
 License: # PolyForm Noncommercial License 1.0.0

{dh_cli-0.4.1 → dh_cli-0.4.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dh-cli"
-version = "0.4.1"
+version = "0.4.2"
 description = "Dayhoff Labs developer CLI"
 requires-python = ">=3.11"
 readme = "README.md"

{dh_cli-0.4.1 → dh_cli-0.4.2}/src/dh_cli/hz/__init__.py

@@ -10,6 +10,48 @@ hz_app = typer.Typer(
     context_settings={"help_option_names": ["-h", "--help"]},
 )
 
+API_URLS = {
+    "prod": "https://api.horizyn.dayhofflabs.com",
+    "dev": "https://horizyn-api-dev.dayhofflabs.com",
+}
+
+
+@hz_app.command()
+def health(
+    detailed: bool = typer.Option(False, "-d", "--detailed", help="Include GPU and EFS details."),
+):
+    """Check API health for both prod and dev environments."""
+    import json
+    from concurrent.futures import ThreadPoolExecutor
+    from urllib.request import urlopen
+    from urllib.error import URLError
+
+    path = "/healthz/detailed" if detailed else "/healthz"
+
+    def _fetch(env: str) -> tuple[str, dict | str]:
+        try:
+            with urlopen(f"{API_URLS[env]}{path}", timeout=10) as resp:
+                return env, json.loads(resp.read())
+        except (URLError, OSError) as exc:
+            return env, f"unreachable ({exc})"
+        except json.JSONDecodeError:
+            return env, "invalid response"
+
+    with ThreadPoolExecutor(max_workers=2) as pool:
+        results = dict(pool.map(_fetch, API_URLS))
+
+    for env in ("prod", "dev"):
+        label = f"[{env}] {API_URLS[env]}"
+        data = results[env]
+        if isinstance(data, str):
+            typer.secho(f"{label}  ✗ {data}", fg=typer.colors.RED)
+        else:
+            status = data.get("status", "unknown")
+            version = data.get("version", "?")
+            color = typer.colors.GREEN if status == "healthy" else typer.colors.YELLOW
+            typer.secho(f"{label}  v{version}  {status}", fg=color)
+            typer.echo(json.dumps(data, indent=2))
+
 
 def _find_workspace_root() -> Path:
     root = os.environ.get("WORKSPACE_ROOT")