Added bundling logic for wheels

Author: monadoid

.github/workflows/publish-pypi.yml

@@ -23,7 +23,7 @@ jobs:
           - os: ubuntu-latest
             binary_name: stagehand-linux-x64
             output_path: src/stagehand/_sea/stagehand-linux-x64
-            wheel_platform_tag: linux_x86_64
+            wheel_platform_tag: manylinux2014_x86_64
           - os: macos-latest
             binary_name: stagehand-darwin-arm64
             output_path: src/stagehand/_sea/stagehand-darwin-arm64

hatch_build.py

@@ -0,0 +1,58 @@
+from __future__ import annotations
+
+import os
+from pathlib import Path
+
+from hatchling.builders.hooks.plugin.interface import BuildHookInterface
+
+
+def _infer_platform_tag() -> str:
+    from packaging.tags import sys_tags
+
+    # Linux tag is after many/musl; skip those to get the generic platform tag.
+    tag = next(iter(t for t in sys_tags() if "manylinux" not in t.platform and "musllinux" not in t.platform))
+    return tag.platform
+
+
+def _has_embedded_sea_binaries() -> bool:
+    sea_dir = Path(__file__).resolve().parent / "src" / "stagehand" / "_sea"
+    if not sea_dir.exists():
+        return False
+
+    for path in sea_dir.iterdir():
+        if not path.is_file():
+            continue
+        if path.name in {".keep"}:
+            continue
+        if path.name.startswith("."):
+            continue
+        return True
+
+    return False
+
+
+class CustomBuildHook(BuildHookInterface):
+    def initialize(self, _version: str, build_data: dict) -> None:
+        if not _has_embedded_sea_binaries():
+            return
+
+        # We are bundling a platform-specific executable, so this must not be a
+        # "pure python" wheel.
+        build_data["pure_python"] = False
+
+        # CI sets this so we get deterministic wheel tags that match the SEA
+        # artifact we're embedding (e.g. "py3-none-macosx_11_0_arm64").
+        wheel_tag = os.environ.get("STAGEHAND_WHEEL_TAG", "").strip()
+        if wheel_tag:
+            if wheel_tag.count("-") != 2:
+                raise ValueError(
+                    "Invalid STAGEHAND_WHEEL_TAG. Expected a full wheel tag like 'py3-none-macosx_11_0_arm64'."
+                )
+            build_data["tag"] = wheel_tag
+            build_data["infer_tag"] = False
+        else:
+            # For local builds, infer just the platform portion so the wheel
+            # remains Python-version agnostic (our embedded server binary is not
+            # tied to a specific Python ABI).
+            build_data["tag"] = f"py3-none-{_infer_platform_tag()}"
+            build_data["infer_tag"] = False

pyproject.toml

@@ -77,14 +77,17 @@ pydantic-v2 = [
 ]
 
 [build-system]
-requires = ["hatchling==1.26.3", "hatch-fancy-pypi-readme"]
+requires = ["hatchling==1.26.3", "hatch-fancy-pypi-readme", "packaging"]
 build-backend = "hatchling.build"
 
 [tool.hatch.build]
 include = [
   "src/*"
 ]
 
+[tool.hatch.build.hooks.custom]
+path = "hatch_build.py"
+
 [tool.hatch.build.targets.wheel]
 packages = ["src/stagehand"]
 

src/stagehand/_sea/.keep

@@ -106,5 +106,16 @@ def resolve_binary_path(
     here = Path(__file__).resolve()
     repo_root = here.parents[3]  # stagehand-python/
     candidate = repo_root / "bin" / "sea" / filename
+
+    if not candidate.exists():
+        raise FileNotFoundError(
+            f"Stagehand SEA binary not found at {candidate}.\n"
+            f"For local development, download the binary using:\n"
+            f"  uv run python scripts/download-binary.py\n"
+            f"Or set the STAGEHAND_SEA_BINARY environment variable to point to your binary.\n"
+            f"For production use, install a platform-specific wheel from PyPI.\n"
+            f"See: https://github.com/browserbase/stagehand-python#local-development"
+        )
+
     _ensure_executable(candidate)
     return candidate