'commit'

AI/Text2Sql/app.py

@@ -112,7 +112,7 @@ async def get_docx_stream(
         return {"success": False, "message": "缺少问题ID参数"}
 
     # 根据问题ID获取查询sql
-    sql = (await db.fetch("SELECT * FROM t_bi_question WHERE id = $1", question_id))[0]['sql']
+    sql = (await get_question_by_id(db, question_id))[0]['sql']
 
     # 生成word报告
     prompt = '''
@@ -162,7 +162,8 @@ async def get_docx_stream(
             markdown_to_docx(summary, output_file=filename)
 
             # 记录到数据库
-            await db.execute("UPDATE t_bi_question SET docx_file_name = $1 WHERE id = $2", filename, question_id)
+            await update_question_by_id(db, question_id, docx_file_name=filename)
+            #await db.execute("UPDATE t_bi_question SET docx_file_name = $1 WHERE id = $2", filename, question_id)
         except asyncio.CancelledError:
             # 客户端提前断开连接，无需处理
             print("客户端断开连接")
@@ -228,6 +229,7 @@ async def set_system_recommend(
     # 提示保存成功
     return {"success": True, "message": "保存成功"}
 
+
 @app.post("/questions/set_user_collect")
 async def set_user_collect(
         question_id: str = Form(...),
@@ -238,6 +240,7 @@ async def set_user_collect(
     # 提示保存成功
     return {"success": True, "message": "保存成功"}
 
+
 @app.get("/questions/get_system_recommend")
 async def get_system_recommend(db: asyncpg.Connection = Depends(get_db)):  # 添加 db 参数
     # 查询所有系统推荐问题
@@ -245,6 +248,7 @@ async def get_system_recommend(db: asyncpg.Connection = Depends(get_db)):  # 添
     # 返回查询结果
     return {"success": True, "data": system_recommend_questions}
 
+
 @app.get("/questions/get_user_collect")
 async def get_user_collect(db: asyncpg.Connection = Depends(get_db)):  # 添加 db 参数
     # 查询所有用户收藏问题
@@ -252,6 +256,7 @@ async def get_user_collect(db: asyncpg.Connection = Depends(get_db)):  # 添加
     # 返回查询结果
     return {"success": True, "data": user_collect_questions}
 
+
 # 启动 FastAPI
 if __name__ == "__main__":
     uvicorn.run("app:app", host="0.0.0.0", port=8000, reload=True, workers=4)