# PETSQL

**Repository Path**: llxyw/PETSQL

## Basic Information

- **Project Name**: PETSQL
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-07-03
- **Last Updated**: 2024-07-03

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

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

## Overview
![The proposed PETSQL framework](./figs/overview.png)

## RUN
```bash
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
```bibtex
@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}
}
```