You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.
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 ' )
