From e79d383437d2b7316ee1def8049a13f3eb7c8dfa Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 14 一月 2025 13:38:05 +0800
Subject: [PATCH] Merge branch 'master' of http://192.168.5.5:10010/r/rag-gateway

---
 app/api/excel.py |   19 ++++++++++++++-----
 1 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/app/api/excel.py b/app/api/excel.py
index 490f731..7572fa3 100644
--- a/app/api/excel.py
+++ b/app/api/excel.py
@@ -1,7 +1,7 @@
 import random
 import string
 
-from fastapi import APIRouter, File, UploadFile, Form, BackgroundTasks, Depends
+from fastapi import APIRouter, File, UploadFile, Form, BackgroundTasks, Depends, Request
 from fastapi.responses import JSONResponse, FileResponse
 from sqlalchemy.orm import Session
 from starlette.websockets import WebSocket
@@ -128,8 +128,13 @@
                     if session:
                         session_id = session.id
                         new_message = {
-                            "role": "user",
-                            "download_url": f"./api/document/download/{merge_file}.xlsx"
+                            "role": "assistant",
+                            "content": {
+                                "message": "\u5408\u5e76\u6210\u529f",
+                                "type": "message",
+                                "file_url": f"/api/document/download/{merge_file}.xlsx",
+                                "file_name": f"{merge_file}.xlsx"
+                            }
                         }
                         session_service = SessionService(db)
                         session_service.update_session(session_id, message=new_message)
@@ -146,14 +151,18 @@
 
 
 @router.get("/download/{file_full_name}")
-async def download_file(background_tasks: BackgroundTasks, file_full_name: str):
+async def download_file(file_full_name: str):
     file_name = os.path.basename(file_full_name)
     user_excel = EXCEL_FILES_PATH
     file_path = os.path.join(user_excel, file_full_name)
 
     if not os.path.exists(file_path):
         return JSONResponse(content={"error": "鏂囦欢涓嶅瓨鍦�"}, status_code=404)
-
+    return FileResponse(
+        path=file_path,
+        filename="Excel.xlsx",
+        media_type='application/octet-stream',
+    )
     # def delete_file():
     #     try:
     #         os.unlink(file_path)

--
Gitblit v1.8.0