README.md 575 Bytes
Newer Older
mashun1's avatar
omnisql  
mashun1 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
# 复杂性感知型 SQL 查询生成

## 第 1 步:SQL 查询生成

利用数据库架构、数据库值、查询复杂性和 SQLite 支持的函数生成 SQL 查询

```bash
# 创建用于生成 SQL 查询的提示。

mkdir prompts
python3 generate_sql_synthesis_prompts.py
```

```bash
# 运行以使用 LLM 生成 SQL 查询
python3 synthesize_sql.py --model model_name --base_url vllm_serve_url(http://x.x.x.x:8000/v1)
```

## 第 2 步:后处理

优化生成的 SQL 查询以确保质量并删除无效或冗余的查询

```bash
python3 post_process_sqls.py
```