37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from Text2Sql.Util.VannaUtil import *
|
||
|
||
'''
|
||
经验：
|
||
1、尽量使用宽表，少用关联，越少越好
|
||
2、应该有一些固定的组合用法预置出来，给出范例，让用户可以简单修改后就能使用
|
||
3、应该有类似于 保存为用例，查询历史等功能，让用户方便利旧。
|
||
'''
|
||
|
||
if __name__ == "__main__":
|
||
vn = VannaUtil()
|
||
|
||
# 开始训练
|
||
print("开始训练...")
|
||
# 打开AreaSchoolLesson.sql文件内容
|
||
with open("Sql/AreaSchoolLessonDDL.sql", "r", encoding="utf-8") as file:
|
||
ddl = file.read()
|
||
# 训练数据
|
||
vn.train(
|
||
ddl=ddl
|
||
)
|
||
# 添加有关业务术语或定义的文档
|
||
# vn.train(documentation="Sql/AreaSchoolLesson.md")
|
||
|
||
# 使用 SQL 进行训练
|
||
with open('Sql/AreaSchoolLessonGenerate.sql', 'r', encoding='utf-8') as file:
|
||
sql_content = file.read()
|
||
# 使用正则表达式提取注释和 SQL 语句
|
||
sql_pattern = r'/\*(.*?)\*/(.*?);'
|
||
sql_snippets = re.findall(sql_pattern, sql_content, re.DOTALL)
|
||
|
||
# 打印提取的注释和 SQL 语句
|
||
for i, (comment, sql) in enumerate(sql_snippets, 1):
|
||
vn.train(sql=comment.strip() + '\n' + sql.strip() + '\n')
|
||
|
||
|