'commit'

AI/Text2Sql/app.py

@@ -72,6 +72,9 @@ async def get_excel(question_id: str = Form(...), question_str: str = Form(...),
     if len(question_id) != 36:
         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 = '''
         返回的信息要求：
         1、行政区划为NULL 或者是空字符的不参加统计
@@ -82,9 +85,10 @@ async def get_excel(question_id: str = Form(...), question_str: str = Form(...),
     # 获取完整 SQL
     sql = vn.generate_sql(question)
     print("生成的查询 SQL:\n", sql)
-    # 先删除后插入,防止重复插入
-    await delete_question(db, question_id)
+
+    # 插入数据
     await insert_question(db, question_id, question)
+
     # 检查，如果sql为空，则返回错误信息
     if not sql:
         return {"success": False, "message": "无法生成相应的SQL语句！"}