run_infer.sh 1002 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. #!/bin/bash
  2. set -eo pipefail
  3. source "evaluation/utils/version_control.sh"
  4. # configure miniwob website, change URL to yours
  5. export MINIWOB_URL="file:///home/fangzhex/miniwob-plusplus/miniwob/html/miniwob/"
  6. # configure browsing agent
  7. export USE_NAV="false"
  8. export USE_CONCISE_ANSWER="true"
  9. MODEL_CONFIG=$1
  10. COMMIT_HASH=$2
  11. AGENT=$3
  12. NOTE=$4
  13. EVAL_LIMIT=$5
  14. checkout_eval_branch
  15. if [ -z "$AGENT" ]; then
  16. echo "Agent not specified, use default BrowsingAgent"
  17. AGENT="BrowsingAgent"
  18. fi
  19. get_agent_version
  20. echo "AGENT: $AGENT"
  21. echo "AGENT_VERSION: $AGENT_VERSION"
  22. echo "MODEL_CONFIG: $MODEL_CONFIG"
  23. EVAL_NOTE="${AGENT_VERSION}_${NOTE}"
  24. COMMAND="poetry run python evaluation/miniwob/run_infer.py \
  25. --agent-cls $AGENT \
  26. --llm-config $MODEL_CONFIG \
  27. --max-iterations 10 \
  28. --max-chars 10000000 \
  29. --eval-note $EVAL_NOTE"
  30. if [ -n "$EVAL_LIMIT" ]; then
  31. echo "EVAL_LIMIT: $EVAL_LIMIT"
  32. COMMAND="$COMMAND --eval-n-limit $EVAL_LIMIT"
  33. fi
  34. # Run the command
  35. eval $COMMAND
  36. checkout_original_branch