#!/usr/bin/env bash
# launch_noloop.sh <tb2|swe> [model-id] [num-tasks]
set -euo pipefail
export PATH="/var/lib/agentptb-cache/tb1/prime-rl/.venv/bin:$PATH"
WS="${AGENTPTB_WORKSPACE:-/mnt/pvc/users/simon/agentptb/runs/grok/workspace}"
export PYTHONPATH="$WS/envs/pi_noloop:/var/lib/agentptb-cache/tb1/prime-rl/deps/verifiers${PYTHONPATH:+:$PYTHONPATH}"
export TMPDIR=/var/lib/agentptb-cache/tb1/tmp
export PRIME_API_KEY="$(cat "$AGENTPTB_PRIME_KEY_FILE")"
SUITE="${1:?tb2 or swe}"
DEFAULT_MODEL="$WS/outputs/soup-85-s80/weights"
MODEL="$(readlink -f "${2:-$DEFAULT_MODEL}")"
N="${3:-16}"
OFFSET="${AGENTPTB_PORT_OFFSET:-500}"
PORT=$((8000 + OFFSET + 100))
case "$SUITE" in
  tb2) TS=terminal-bench-2-v1; CFG="$WS/configs/eval-tb2-noloop.toml"; OUT="$WS/outputs/eval-tb2-noloop2" ;;
  swe) TS=swebench-verified-v1; CFG="$WS/configs/eval-swe-noloop.toml"; OUT="$WS/outputs/eval-swe-noloop2" ;;
  *) echo "suite must be tb2 or swe"; exit 1 ;;
esac
mkdir -p "$OUT" "$WS/logs"
cd /var/lib/agentptb-cache/tb1/prime-rl
exec eval "$TS" @ "$CFG" \
  --model "$MODEL" \
  --client.base-url "http://127.0.0.1:${PORT}/v1" \
  -n "$N" -s --no-push --no-rich -o "$OUT" \
  --env.agent.runtime.block-network false