|
|
@ -72,6 +72,9 @@ async def get_excel(question_id: str = Form(...), question_str: str = Form(...),
|
|
|
|
if len(question_id) != 36:
|
|
|
|
if len(question_id) != 36:
|
|
|
|
return {"success": False, "message": "question_id格式错误"}
|
|
|
|
return {"success": False, "message": "question_id格式错误"}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 检查此question_id 是不是已存在,存在的不能再次生成,需要全新创建一个新问题ID
|
|
|
|
|
|
|
|
if await get_question_by_id(db, question_id):
|
|
|
|
|
|
|
|
return {"success": False, "message": "question_id已存在,请重新生成"}
|
|
|
|
common_prompt = '''
|
|
|
|
common_prompt = '''
|
|
|
|
返回的信息要求:
|
|
|
|
返回的信息要求:
|
|
|
|
1、行政区划为NULL 或者是空字符的不参加统计
|
|
|
|
1、行政区划为NULL 或者是空字符的不参加统计
|
|
|
@ -82,9 +85,10 @@ async def get_excel(question_id: str = Form(...), question_str: str = Form(...),
|
|
|
|
# 获取完整 SQL
|
|
|
|
# 获取完整 SQL
|
|
|
|
sql = vn.generate_sql(question)
|
|
|
|
sql = vn.generate_sql(question)
|
|
|
|
print("生成的查询 SQL:\n", sql)
|
|
|
|
print("生成的查询 SQL:\n", sql)
|
|
|
|
# 先删除后插入,防止重复插入
|
|
|
|
|
|
|
|
await delete_question(db, question_id)
|
|
|
|
# 插入数据
|
|
|
|
await insert_question(db, question_id, question)
|
|
|
|
await insert_question(db, question_id, question)
|
|
|
|
|
|
|
|
|
|
|
|
# 检查,如果sql为空,则返回错误信息
|
|
|
|
# 检查,如果sql为空,则返回错误信息
|
|
|
|
if not sql:
|
|
|
|
if not sql:
|
|
|
|
return {"success": False, "message": "无法生成相应的SQL语句!"}
|
|
|
|
return {"success": False, "message": "无法生成相应的SQL语句!"}
|
|
|
|