scripts/check_space_requirements.py

#!/usr/bin/env python3
"""Gagal jika requirements.txt Space tidak memuat wheel llama-cpp (cegah build timeout)."""
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]
REQ = ROOT / "requirements.txt"
WHEEL_MARKERS = ("llama-cpp-python @", "llama_cpp_python", "llama-cpp-python-wheels")


def main() -> int:
    if not REQ.is_file():
        print("❌ requirements.txt tidak ditemukan")
        return 1
    isi = REQ.read_text(encoding="utf-8")
    if any(marker in isi for marker in WHEEL_MARKERS):
        print("✅ requirements.txt tampak siap untuk Space (wheel llama.cpp)")
        return 0
    print(
        "❌ requirements.txt tidak memuat wheel llama-cpp-python.\n"
        "   Jalankan: ./scripts/prepare_space_requirements.sh sebelum push ke HF."
    )
    return 1


if __name__ == "__main__":
    sys.exit(main())