tenant.openagents/omega

1from __future__ import annotations
2
3from harbor.models.verifier.result import VerifierResult
4from harbor.utils.env import resolve_env_vars
5from harbor.verifier.verifier import Verifier
6
7from .judge_proxy import JUDGE_PROXY_ENSURE_SCRIPT, JUDGE_PROXY_SCRIPT
8
9
10class ZedJudgeProxyVerifier(Verifier):
11    async def verify(self) -> VerifierResult:
12        await self._ensure_judge_proxy()
13        return await super().verify()
14
15    def _merged_verifier_env(self) -> dict[str, str]:
16        return {
17            **self.task.config.verifier.env,
18            **(self.verifier_env or {}),
19            **self.override_env,
20        }
21
22    async def _ensure_judge_proxy(self) -> None:
23        merged_env = self._merged_verifier_env()
24        if not merged_env.get("ZED_JUDGE_UPSTREAM"):
25            return
26
27        env = resolve_env_vars(merged_env)
28        result = await self.environment.exec(
29            command=(
30                "set -e\n"
31                "mkdir -p /usr/local/lib /usr/local/bin\n"
32                "cat > /usr/local/lib/zed_judge_proxy.py <<'ZED_PROXY_EOF'\n"
33                f"{JUDGE_PROXY_SCRIPT}"
34                "ZED_PROXY_EOF\n"
35                "cat > /usr/local/bin/ensure-judge-proxy <<'ZED_ENSURE_EOF'\n"
36                f"{JUDGE_PROXY_ENSURE_SCRIPT}"
37                "ZED_ENSURE_EOF\n"
38                "chmod +x /usr/local/bin/ensure-judge-proxy\n"
39                "ensure-judge-proxy\n"
40                "python3 - <<'ZED_PROXY_CHECK_EOF'\n"
41                "import os, socket\n"
42                "port = int(os.environ.get('ZED_JUDGE_PROXY_PORT', '8089'))\n"
43                "with socket.create_connection(('127.0.0.1', port), timeout=5):\n"
44                "    pass\n"
45                "ZED_PROXY_CHECK_EOF\n"
46                "if [ -f /tmp/zed-judge-proxy.log ]; then "
47                "tail -2 /tmp/zed-judge-proxy.log; fi\n"
48            ),
49            user="root",
50            env=env,
51        )
52        if result.return_code != 0:
53            output = "\n".join(
54                part for part in (result.stdout, result.stderr) if part
55            ).strip()
56            detail = f": {output}" if output else ""
57            raise RuntimeError(
58                f"Failed to install or start judge proxy (exit {result.return_code}){detail}"
59            )
60