mirror of
https://github.com/norandom/OpenUltraSAST.git
synced 2026-10-21 05:59:29 +00:00
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
|
"""Search ceilings carried to every executor; no model or transport dependencies."""
|
||
|
|
|
||
|
|
import math
|
||
|
|
from dataclasses import dataclass
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True)
|
||
|
|
class SearchBudget:
|
||
|
|
spend_usd: float = 0.50
|
||
|
|
max_call_usd: float = 0.05
|
||
|
|
reason_rounds: int = 4
|
||
|
|
intents_per_round: int = 3
|
||
|
|
demonstrations: int = 3
|
||
|
|
tasks: int = 32
|
||
|
|
retries: int = 1
|
||
|
|
memory_bytes: int = 5 * 1024**3
|
||
|
|
disk_bytes: int = 2 * 1024**3
|
||
|
|
task_wall_seconds: float = 900
|
||
|
|
wall_seconds: float = 4 * 3600
|
||
|
|
cleanup_seconds: float = 30
|
||
|
|
|
||
|
|
def __post_init__(self) -> None:
|
||
|
|
for name, value in vars(self).items():
|
||
|
|
if isinstance(value, bool) or not isinstance(value, (int, float)) or not math.isfinite(value):
|
||
|
|
raise ValueError(f"invalid {name}")
|
||
|
|
if value < 0 or (value == 0 and name not in ("retries", "spend_usd")):
|
||
|
|
raise ValueError(f"invalid {name}")
|
||
|
|
if name in (
|
||
|
|
"reason_rounds",
|
||
|
|
"intents_per_round",
|
||
|
|
"demonstrations",
|
||
|
|
"tasks",
|
||
|
|
"retries",
|
||
|
|
"memory_bytes",
|
||
|
|
"disk_bytes",
|
||
|
|
) and not isinstance(value, int):
|
||
|
|
raise ValueError(f"{name} must be an integer")
|