From f54cdd47cf062c4e771b7e69f4bb81b452ca9cbe Mon Sep 17 00:00:00 2001 From: HuangHai <10402852@qq.com> Date: Mon, 17 Mar 2025 09:48:30 +0800 Subject: [PATCH] 'commit' --- AI/Text2Sql/__pycache__/app.cpython-310.pyc | Bin 6561 -> 6768 bytes AI/Text2Sql/app.py | 13 +++++++++---- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/AI/Text2Sql/__pycache__/app.cpython-310.pyc b/AI/Text2Sql/__pycache__/app.cpython-310.pyc index 49de5cb82d76bcc90ca099de795b9d46bb2c5818..840a68c3a37d618b0736d09c60d317b7619736d2 100644 GIT binary patch delta 1262 zcmb7@&u<$=6vy|?u6Nfz*7>pa+Fr+Q(>6bxUk+`dr9k*06g3>mkHTecQQo*p*lo%t zK?1qBmBRrEOH`wXE4^1-{1;S3LL3mcB3XhP;=qL}9JrM7cD9kMk_&6i+c)pM@8`|V zjPIWN^L&2WaWWb`E1&&*tvYijKLOv~nV;+EK`J!42^wl>1XgHsTlo;Ad75GnWO#<& zsiZE<@~q06G|%x|=y2yhS}VwhBYZ@~?O-$<<73KC2jlO@ebRxe-kG4#35&d_lG(ru zOT47~Tu=@xyaJjqyBha}zg0y?*qFzW9*`aF>6EKbZbBG1Cd?MYadh{MF7TQ#MGXr* z;64`7+t+wq_}`qs(p?ZaEO(D=R@A%D>gZSzC()--73rO`DK9}L#OWd849+UzjG{D#DV_g;nZtR7nN{Dj zaP0Ra%@>E7FJV)u8cMbNjH)xd=3&`$cvi{gNH&k>NS0K>(|uxi9-mhg3##IUV=6A- zMOCpVmhi<7*;f#KATnD`yo4{|%lHbe;H&vwicTD^s?m5&e#+_>VPLJU@9e(*&RPQo z*{i$jyPF%Un}OJAut=vHJh0vrH^g19e-8~U~(*J?|2vCz>8v{7ok5adE zxX?eeT?20SE$5*Q2mN11e*-}I-T0Otla3yPoOnE-v2#UUc9*KGt##QN-PX(Wzo^jt zuV^=QG|_77@(XvvF`{`Yj^2*VqWsI9_P&qBqh#1E+cLoh6EdtUmkZ0F%cff@9o8v% z8+Nre!Q0T)6TJMqu;%CquAAT<7G|Aa$UQU_ZrqgV;#5MBtHmquyF4g10pxnYt;K6T z^vNi))`f(66A;=U|0rIvS+q#$fiE+jFW>N{a*tw3g3im2yy=7l59B>>;i%e^_dQQO z@+RSL>6Z3ZT67}oH#U<4aG%bWC`O0ViVr7>H!!Nkovgi&H#fGU>rpBW#f{|s(ogcB z{AJ-DO>E7k-N~t6?8#>3y|+v-9y4H17-Rn!dhW4ZG?__T_)Ii?eAEZq^dV!F!Fcj7 DPjV}Z delta 1072 zcmaJx&kR=)+phQhvQHgjO4VI zfc*ocqsBcAq5~=fk`YKC(NH1LB4BQTC`gEvX7<~iZ@!tiy)Sbcdo$P5X-A{A)_8M8 zmhWduaQgn#WD`O&Fi}H2v?4pOBOY+j=xoqA5hNlfaMalhlTj*2DQt!5C=+B9w!@w% z8)OycVeiE_mxtTbo(OYMKFF(>6S|QXcnT-OLR1Wjpb5L#(gI)jD}9&}DNJLg4di>; zIt7aq>lX%QgixTGA5crr!cU=nxg#U05y$uivEVTZ3E(Tgq)v+kH zqfgc+hE|5F7WQLl-6Wk6krAV!xdR8pQ0ap(%e;Wi^Fz!4lJLr(L#hfJO z6z3MT8qGYdZuZg5emtOVCU%J#Jb0DegW9MtS7z}L9>ybh6p!KYwn?uA7nDt$ zknh=`!@BnT;%fcm*#%kBOUVoM#rpEn{BkH(J9GLAfPTr$E>t@2%)2`5>s(6Y3}|+0 z>DM~Q&O+}efXi|=x6;SrKk@g{X*bB8lC93mPHrYgy`tf3X!_c1h8EflCcovDxL%tg zQtgZk^JCt#n4cxb?sCfl4i-2pD<9=6<*o8Dj!T>(`W; zCvUiJvK!Olb{r-jxMlaBQu4Wb4nE33Z+4uRn+phSmWn&8Yc1_IeZPh#?|6p+K=U-{>UjPDHC55`j8d42@A9h(E3 LJK*pl7(adjUsviH diff --git a/AI/Text2Sql/app.py b/AI/Text2Sql/app.py index 63d82c3e..854f1581 100644 --- a/AI/Text2Sql/app.py +++ b/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 = ''' @@ -154,7 +154,7 @@ async def get_docx_stream( chunk_content = chunk.choices[0].delta.content # 逐字拆分并返回 for char in chunk_content: - #print(char, end="", flush=True) # 逐字输出到控制台 + # print(char, end="", flush=True) # 逐字输出到控制台 yield char.encode("utf-8") # 将字符编码为 UTF-8 字节 summary += char # 将内容拼接到 summary 中 @@ -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("客户端断开连接") @@ -172,7 +173,7 @@ async def get_docx_stream( "success": False, "message": f"生成Word文件失败: {str(e)}" }) - #print(error_response) # 输出错误信息到控制台 + # print(error_response) # 输出错误信息到控制台 yield error_response.encode("utf-8") # 将错误信息编码为 UTF-8 字节 finally: @@ -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)