├── .env ├── .gitignore ├── LICENSE ├── README.md ├── alphasql ├── algorithm │ ├── mcts │ │ ├── __pycache__ │ │ │ ├── mcts.cpython-311.pyc │ │ │ ├── mcts_action.cpython-311.pyc │ │ │ ├── mcts_node.cpython-311.pyc │ │ │ └── reward.cpython-311.pyc │ │ ├── mcts.py │ │ ├── mcts_action.py │ │ ├── mcts_node.py │ │ └── reward.py │ └── selection │ │ ├── __pycache__ │ │ └── utils.cpython-311.pyc │ │ └── utils.py ├── config │ ├── __pycache__ │ │ └── mcts_config.cpython-311.pyc │ └── mcts_config.py ├── database │ ├── __pycache__ │ │ ├── schema.cpython-311.pyc │ │ ├── sql_execution.cpython-310.pyc │ │ ├── sql_execution.cpython-311.pyc │ │ └── utils.cpython-311.pyc │ ├── database_manager.py │ ├── lsh_index.py │ ├── schema.py │ ├── sql_execution.py │ ├── sql_parse.py │ └── utils.py ├── llm_call │ ├── __pycache__ │ │ ├── cost_recoder.cpython-311.pyc │ │ ├── openai_llm.cpython-311.pyc │ │ └── prompt_factory.cpython-311.pyc │ ├── cost_recoder.py │ ├── embedding_utils.py │ ├── openai_llm.py │ └── prompt_factory.py ├── runner │ ├── __pycache__ │ │ ├── mcts_runner.cpython-311.pyc │ │ └── task.cpython-311.pyc │ ├── mcts_runner.py │ ├── preprocessor.py │ ├── selection_runner.py │ ├── sql_selection.py │ └── task.py └── templates │ ├── identify_column_functions.txt │ ├── identify_column_values.txt │ ├── keywords_extraction.txt │ ├── raphrase_question.txt │ ├── schema_selection.txt │ ├── sql_generation.txt │ └── sql_revision.txt ├── assets ├── Alpha-SQL-overview.png └── intro-figure.png ├── config ├── qwen32b_bird_dev.yaml └── qwen7b_sds_exp.yaml ├── requirements.txt ├── results └── dev_pred_sqls.json └── script ├── preprocess.sh ├── qwen32b_bird_dev_exp.sh ├── qwen7b_sds_exp.sh └── sql_selection.sh /.env: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/.env -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/README.md -------------------------------------------------------------------------------- /alphasql/algorithm/mcts/__pycache__/mcts.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/algorithm/mcts/__pycache__/mcts.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/algorithm/mcts/__pycache__/mcts_action.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/algorithm/mcts/__pycache__/mcts_action.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/algorithm/mcts/__pycache__/mcts_node.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/algorithm/mcts/__pycache__/mcts_node.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/algorithm/mcts/__pycache__/reward.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/algorithm/mcts/__pycache__/reward.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/algorithm/mcts/mcts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/algorithm/mcts/mcts.py -------------------------------------------------------------------------------- /alphasql/algorithm/mcts/mcts_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/algorithm/mcts/mcts_action.py -------------------------------------------------------------------------------- /alphasql/algorithm/mcts/mcts_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/algorithm/mcts/mcts_node.py -------------------------------------------------------------------------------- /alphasql/algorithm/mcts/reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/algorithm/mcts/reward.py -------------------------------------------------------------------------------- /alphasql/algorithm/selection/__pycache__/utils.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/algorithm/selection/__pycache__/utils.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/algorithm/selection/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/algorithm/selection/utils.py -------------------------------------------------------------------------------- /alphasql/config/__pycache__/mcts_config.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/config/__pycache__/mcts_config.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/config/mcts_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/config/mcts_config.py -------------------------------------------------------------------------------- /alphasql/database/__pycache__/schema.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/database/__pycache__/schema.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/database/__pycache__/sql_execution.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/database/__pycache__/sql_execution.cpython-310.pyc -------------------------------------------------------------------------------- /alphasql/database/__pycache__/sql_execution.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/database/__pycache__/sql_execution.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/database/__pycache__/utils.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/database/__pycache__/utils.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/database/database_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/database/database_manager.py -------------------------------------------------------------------------------- /alphasql/database/lsh_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/database/lsh_index.py -------------------------------------------------------------------------------- /alphasql/database/schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/database/schema.py -------------------------------------------------------------------------------- /alphasql/database/sql_execution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/database/sql_execution.py -------------------------------------------------------------------------------- /alphasql/database/sql_parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/database/sql_parse.py -------------------------------------------------------------------------------- /alphasql/database/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/database/utils.py -------------------------------------------------------------------------------- /alphasql/llm_call/__pycache__/cost_recoder.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/llm_call/__pycache__/cost_recoder.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/llm_call/__pycache__/openai_llm.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/llm_call/__pycache__/openai_llm.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/llm_call/__pycache__/prompt_factory.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/llm_call/__pycache__/prompt_factory.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/llm_call/cost_recoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/llm_call/cost_recoder.py -------------------------------------------------------------------------------- /alphasql/llm_call/embedding_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/llm_call/embedding_utils.py -------------------------------------------------------------------------------- /alphasql/llm_call/openai_llm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/llm_call/openai_llm.py -------------------------------------------------------------------------------- /alphasql/llm_call/prompt_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/llm_call/prompt_factory.py -------------------------------------------------------------------------------- /alphasql/runner/__pycache__/mcts_runner.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/runner/__pycache__/mcts_runner.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/runner/__pycache__/task.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/runner/__pycache__/task.cpython-311.pyc -------------------------------------------------------------------------------- /alphasql/runner/mcts_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/runner/mcts_runner.py -------------------------------------------------------------------------------- /alphasql/runner/preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/runner/preprocessor.py -------------------------------------------------------------------------------- /alphasql/runner/selection_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/runner/selection_runner.py -------------------------------------------------------------------------------- /alphasql/runner/sql_selection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/runner/sql_selection.py -------------------------------------------------------------------------------- /alphasql/runner/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/runner/task.py -------------------------------------------------------------------------------- /alphasql/templates/identify_column_functions.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/templates/identify_column_functions.txt -------------------------------------------------------------------------------- /alphasql/templates/identify_column_values.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/templates/identify_column_values.txt -------------------------------------------------------------------------------- /alphasql/templates/keywords_extraction.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/templates/keywords_extraction.txt -------------------------------------------------------------------------------- /alphasql/templates/raphrase_question.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/templates/raphrase_question.txt -------------------------------------------------------------------------------- /alphasql/templates/schema_selection.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/templates/schema_selection.txt -------------------------------------------------------------------------------- /alphasql/templates/sql_generation.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/templates/sql_generation.txt -------------------------------------------------------------------------------- /alphasql/templates/sql_revision.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/alphasql/templates/sql_revision.txt -------------------------------------------------------------------------------- /assets/Alpha-SQL-overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/assets/Alpha-SQL-overview.png -------------------------------------------------------------------------------- /assets/intro-figure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/assets/intro-figure.png -------------------------------------------------------------------------------- /config/qwen32b_bird_dev.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/config/qwen32b_bird_dev.yaml -------------------------------------------------------------------------------- /config/qwen7b_sds_exp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/config/qwen7b_sds_exp.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/dev_pred_sqls.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/results/dev_pred_sqls.json -------------------------------------------------------------------------------- /script/preprocess.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/script/preprocess.sh -------------------------------------------------------------------------------- /script/qwen32b_bird_dev_exp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/script/qwen32b_bird_dev_exp.sh -------------------------------------------------------------------------------- /script/qwen7b_sds_exp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/script/qwen7b_sds_exp.sh -------------------------------------------------------------------------------- /script/sql_selection.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUSTDial/Alpha-SQL/HEAD/script/sql_selection.sh --------------------------------------------------------------------------------