The implementation code of "PET-SQL: A Prompt-enhanced Two-stage Text-to-SQL Framework with Cross-consistency"

Overview

RUN

docker pull zhishuailii/spider_envs:latest

docker run --ipc=host -itd -v /your-local-path/PETSQL/src:/root/src -v /your-local-path/data:/root/data -v /your-local-path/test_database:/root/database --name spider_db_test_offline the-docker-iamge-ID
# If you want to run it in current directory
mkdir -p src data test_database && docker run --ipc=host -itd -v $(pwd)/src:/root/src -v $(pwd)/data:/root/data -v $(pwd)/test_database:/root/database --name spider_db_test_offline zhishuailii/spider_envs:latest

docker exec -it spider_db_test_offline bash run_all.sh

Citation

@article{li2024pet,
  title={PET-SQL: A Prompt-enhanced Two-stage Text-to-SQL Framework with Cross-consistency},
  author={Li, Zhishuai and Wang, Xiang and Zhao, Jingjing and Yang, Sun and Du, Guoqing and Hu, Xiaoru and Zhang, Bin and Ye, Yuxiao and Li, Ziyue and Zhao, Rui and others},
  journal={arXiv preprint arXiv:2403.09732},
  year={2024}
}