Spaces:
Sleeping
Sleeping
File size: 4,298 Bytes
f2871a4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 | from __future__ import annotations
import json
import subprocess
import sys
from pathlib import Path
from typing import List, Tuple
REQUIRED_ENDPOINTS = ["reset", "step", "state"]
REQUIRED_INFERENCE_MARKERS = ["log_start(", "log_step(", "log_end("]
def _run(cmd: List[str], cwd: Path) -> Tuple[bool, str]:
try:
result = subprocess.run(
cmd,
cwd=str(cwd),
check=False,
capture_output=True,
text=True,
)
output = (result.stdout or "") + ("\n" + result.stderr if result.stderr else "")
return result.returncode == 0, output.strip()
except Exception as exc:
return False, str(exc)
def _check_required_files(repo_root: Path) -> Tuple[bool, List[str]]:
required = [
repo_root / "openenv.yaml",
repo_root / "Dockerfile",
repo_root / "inference.py",
repo_root / "app" / "main.py",
repo_root / "tests" / "test_env_contract.py",
]
missing = [str(path.relative_to(repo_root)) for path in required if not path.exists()]
return len(missing) == 0, missing
def _check_openenv_yaml(repo_root: Path) -> Tuple[bool, List[str]]:
path = repo_root / "openenv.yaml"
if not path.exists():
return False, ["openenv.yaml not found"]
text = path.read_text(encoding="utf-8")
issues: List[str] = []
for endpoint in REQUIRED_ENDPOINTS:
marker = f"{endpoint}: /{endpoint}"
if marker not in text:
issues.append(f"Missing endpoint mapping: {marker}")
for field in ["name:", "version:", "tasks:"]:
if field not in text:
issues.append(f"Missing required metadata field: {field}")
return len(issues) == 0, issues
def _check_inference_contract(repo_root: Path) -> Tuple[bool, List[str]]:
path = repo_root / "inference.py"
if not path.exists():
return False, ["inference.py not found"]
text = path.read_text(encoding="utf-8")
missing = [marker for marker in REQUIRED_INFERENCE_MARKERS if marker not in text]
return len(missing) == 0, [f"Missing inference logger marker: {m}" for m in missing]
def main() -> int:
repo_root = Path(__file__).resolve().parents[1]
checks: List[dict] = []
files_ok, files_issues = _check_required_files(repo_root)
checks.append(
{
"check": "Required files",
"status": "pass" if files_ok else "fail",
"details": files_issues if files_issues else ["All required files are present"],
}
)
yaml_ok, yaml_issues = _check_openenv_yaml(repo_root)
checks.append(
{
"check": "openenv.yaml structure",
"status": "pass" if yaml_ok else "fail",
"details": yaml_issues if yaml_issues else ["openenv.yaml contains required core fields"],
}
)
inference_ok, inference_issues = _check_inference_contract(repo_root)
checks.append(
{
"check": "inference.py logging contract",
"status": "pass" if inference_ok else "fail",
"details": inference_issues if inference_issues else ["inference logging markers are present"],
}
)
pytest_ok, pytest_output = _run([sys.executable, "-m", "pytest", "tests", "-q"], repo_root)
checks.append(
{
"check": "Test suite",
"status": "pass" if pytest_ok else "fail",
"details": [pytest_output.splitlines()[-1] if pytest_output else "No output"],
}
)
validate_ok, validate_output = _run([sys.executable, "-m", "openenv.cli", "validate"], repo_root)
checks.append(
{
"check": "OpenEnv validate",
"status": "pass" if validate_ok else "fail",
"details": [validate_output.splitlines()[-1] if validate_output else "No output"],
}
)
passed = sum(1 for c in checks if c["status"] == "pass")
total = len(checks)
overall = "pass" if passed == total else "fail"
report = {
"project": "incident-commander",
"overall": overall,
"passed": passed,
"total": total,
"checks": checks,
}
print(json.dumps(report, indent=2))
return 0 if overall == "pass" else 1
if __name__ == "__main__":
raise SystemExit(main())
|