RAG_Eval / run.sh
Rom89823974978's picture
Updated dockerfile, run.sh and readme file to fit with hugginggface space
ce9fe2d
#!/usr/bin/env bash
# run.sh
set -euo pipefail
# ---- utils ---------------------------------------------------------------
die() { echo "❌ $*" >&2; exit 1; }
exists() { command -v "$1" >/dev/null 2>&1; }
PY=python
[ -n "${VIRTUAL_ENV-}" ] || die "Activate your virtualenv first (python deps)."
CMD=${1:-}
shift || true # remove first arg if present
case "${CMD}" in
single)
CONFIG=${1:-}; QUERIES=${2:-}
[ -f "$CONFIG" ] || die "Config YAML not found: $CONFIG"
[ -f "$QUERIES" ] || die "Queries JSONL not found: $QUERIES"
shift 2
${PY} scripts/run_experiments.py --config "$CONFIG" --queries "$QUERIES" "$@"
;;
grid)
CONFIGS=${1:-}
DATASETS=${2:-}
shift 2
${PY} scripts/run_grid_experiments.py --configs ${CONFIGS} --datasets ${DATASETS} "$@"
;;
dashboard)
if [[ "${1:-}" == "--docker" ]]; then
docker build -t rag-eval . # uses Dockerfile default (Streamlit)
docker run -p 8501:8501 rag-eval
else
streamlit run scripts/dashboard.py
fi
;;
*)
cat <<EOF
Usage: ./run.sh <mode> [args]
Modes
-----
single CONFIG QUERIES [extra run_experiments.py flags]
grid CONFIG_GLOB DATASET_GLOB [extra grid flags]
dashboard [--docker] # without flag runs locally
EOF
[ -n "$CMD" ] && die "Unknown mode: $CMD"
;;
esac