mirror of
https://github.com/deepseek-ai/deepseek-harness
synced 2026-08-15 21:04:50 +00:00
61 lines
2.6 KiB
Python
61 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import platform
|
|
import stat
|
|
from pathlib import Path
|
|
|
|
from hatchling.builders.hooks.plugin.interface import BuildHookInterface
|
|
|
|
|
|
_PLATFORMS = {
|
|
"linux-x64": ("manylinux_2_28_x86_64", "dsh-jsonrpc-agent-pkg-linux-x64"),
|
|
"linux-arm64": ("manylinux_2_28_aarch64", "dsh-jsonrpc-agent-pkg-linux-arm64"),
|
|
"macos-arm64": ("macosx_11_0_arm64", "dsh-jsonrpc-agent-pkg-macos-arm64"),
|
|
}
|
|
|
|
|
|
def _host_platform_tag() -> str:
|
|
machine = platform.machine().lower()
|
|
arch = "arm64" if machine in {"arm64", "aarch64"} else "x64" if machine in {"x86_64", "amd64"} else machine
|
|
system = platform.system().lower()
|
|
key = f"macos-{arch}" if system == "darwin" else f"linux-{arch}" if system == "linux" else system
|
|
try:
|
|
return _PLATFORMS[key][0]
|
|
except KeyError as exc:
|
|
raise RuntimeError(f"unsupported deepseek-harness-runtime-bin build platform: {key}") from exc
|
|
|
|
|
|
class RuntimeBuildHook(BuildHookInterface):
|
|
"""Assign the native wheel tag and reject incomplete or mixed-platform payloads."""
|
|
|
|
def initialize(self, version: str, build_data: dict[str, object]) -> None:
|
|
if version == "editable":
|
|
return
|
|
if self.target_name == "sdist":
|
|
raise RuntimeError(
|
|
"deepseek-harness-runtime-bin is wheel-only; build and publish platform wheels only."
|
|
)
|
|
|
|
platform_tag = os.environ.get("DSH_RUNTIME_PLATFORM_TAG") or _host_platform_tag()
|
|
matches = [value for value in _PLATFORMS.values() if value[0] == platform_tag]
|
|
if len(matches) != 1:
|
|
supported = ", ".join(value[0] for value in _PLATFORMS.values())
|
|
raise RuntimeError(
|
|
f"unsupported DSH_RUNTIME_PLATFORM_TAG {platform_tag!r}; expected one of {supported}"
|
|
)
|
|
expected_executable = matches[0][1]
|
|
runtime_dir = Path(self.root) / "src" / "deepseek_harness_runtime" / "runtime"
|
|
executables = sorted(runtime_dir.glob("dsh-jsonrpc-agent-pkg-*") if runtime_dir.is_dir() else [])
|
|
if [path.name for path in executables] != [expected_executable]:
|
|
found = ", ".join(path.name for path in executables) or "none"
|
|
raise RuntimeError(
|
|
f"runtime wheel {platform_tag} must contain only {expected_executable}; found {found}"
|
|
)
|
|
if executables[0].stat().st_mode & stat.S_IXUSR == 0:
|
|
raise RuntimeError(f"runtime executable is not executable: {executables[0]}")
|
|
|
|
build_data["pure_python"] = False
|
|
build_data["infer_tag"] = False
|
|
build_data["tag"] = f"py3-none-{platform_tag}"
|