#! /bin/bash

# YOUR SETTINGS
API_KEY="YOUR_API_KEY"
# API model
# MODEL_NAME="gpt-4o-mini"
# BASE_URL="https://api.openai.com/v1"

# vLLM
# MODEL_NAME="Qwen/Qwen2.5-VL-72B-Instruct-AWQ"
# BASE_URL="http://YOUR_VLLM_SERVER_URL/v1"

# vLLM (ours)
MODEL_NAME="YOUR_MODEL_NAME"
BASE_URL="YOUR_BASE_URL"

# save model name
SAVE_MODEL_NAME="YOUR_SAVE_MODEL_NAME"

## Experiment Config
PROMPT_TYPE="YOUR_PROMPT_TYPE"
TEMPERATURE=0.0
NUM_GENERATE=1
NUM_WORKERS=10

## Others
DATASET_NAME="WPRM/webarena_checklist_raw"
SAVE_DIR="checklist_output"

python src/checklist_generation.py \
    --dataset_name $DATASET_NAME \
    --model_name $MODEL_NAME \
    --base_url $BASE_URL \
    --api_key $API_KEY \
    --temperature $TEMPERATURE \
    --num_generate $NUM_GENERATE \
    --num_workers $NUM_WORKERS \
    --prompt_type $PROMPT_TYPE \
    --save_model_name $SAVE_MODEL_NAME \
    --save_dir $SAVE_DIR