Fix download-binary script to pull latest version of stagehand-server

Author: monadoid

scripts/download-binary.py

@@ -15,11 +15,14 @@
 from __future__ import annotations
 
 import sys
+import os
+import json
 import argparse
 import platform
 import urllib.error
 import urllib.request
 from pathlib import Path
+from typing import Any
 
 
 def get_platform_info() -> tuple[str, str]:
@@ -49,6 +52,71 @@ def get_local_filename(plat: str, arch: str) -> str:
     name = f"stagehand-{plat}-{arch}"
     return name + (".exe" if plat == "win32" else "")
 
+def _parse_server_tag(tag: str) -> tuple[int, int, int] | None:
+    # Expected: stagehand-server/vX.Y.Z
+    if not tag.startswith("stagehand-server/v"):
+        return None
+
+    ver = tag.removeprefix("stagehand-server/v")
+    # Drop any pre-release/build metadata (we only expect stable tags here).
+    ver = ver.split("-", 1)[0].split("+", 1)[0]
+    parts = ver.split(".")
+    if len(parts) != 3:
+        return None
+    try:
+        return int(parts[0]), int(parts[1]), int(parts[2])
+    except ValueError:
+        return None
+
+
+def _http_get_json(url: str) -> Any:
+    headers = {
+        "Accept": "application/vnd.github+json",
+        "User-Agent": "stagehand-python/download-binary",
+    }
+    # Optional, but helps avoid rate limits in CI.
+    token = (os.environ.get("GITHUB_TOKEN") or os.environ.get("GH_TOKEN") or "").strip()
+    if token:
+        headers["Authorization"] = f"Bearer {token}"
+
+    req = urllib.request.Request(url, headers=headers)
+    with urllib.request.urlopen(req, timeout=30) as resp:
+        data = resp.read()
+    return json.loads(data.decode("utf-8"))
+
+
+def resolve_latest_server_tag() -> str:
+    """Resolve the latest stagehand-server/v* tag from GitHub releases."""
+    repo = "browserbase/stagehand"
+    releases_url = f"https://api.github.com/repos/{repo}/releases?per_page=100"
+    try:
+        releases = _http_get_json(releases_url)
+    except urllib.error.HTTPError as e:  # type: ignore[misc]
+        raise RuntimeError(f"Failed to query GitHub releases (HTTP {e.code}): {releases_url}") from e  # type: ignore[union-attr]
+    except Exception as e:
+        raise RuntimeError(f"Failed to query GitHub releases: {releases_url}") from e
+
+    if not isinstance(releases, list):
+        raise RuntimeError(f"Unexpected GitHub API response for releases: {type(releases).__name__}")
+
+    best: tuple[tuple[int, int, int], str] | None = None
+    for r in releases:
+        if not isinstance(r, dict):
+            continue
+        tag = r.get("tag_name")
+        if not isinstance(tag, str):
+            continue
+        parsed = _parse_server_tag(tag)
+        if parsed is None:
+            continue
+        if best is None or parsed > best[0]:
+            best = (parsed, tag)
+
+    if best is None:
+        raise RuntimeError("No stagehand-server/v* GitHub Releases found for browserbase/stagehand")
+
+    return best[1]
+
 
 def download_binary(version: str) -> None:
     """Download the binary for the current platform."""
@@ -123,12 +191,18 @@ def main() -> None:
     )
     parser.add_argument(
         "--version",
-        default="v3.2.0",
-        help="Version to download (default: v3.2.0)",
+        default=None,
+        help="Stagehand server release tag/version to download (e.g. v3.2.0 or stagehand-server/v3.2.0). Defaults to latest stagehand-server/* GitHub Release.",
     )
 
     args = parser.parse_args()
-    download_binary(args.version)
+    version = str(args.version).strip() if args.version is not None else ""
+    if not version:
+        latest_tag = resolve_latest_server_tag()
+        download_binary(latest_tag)
+        return
+
+    download_binary(version)
 
 
 if __name__ == "__main__":