python cache_item.py --dataset redial --batch_size 1000 --api_key your_api_key