|
|
@@ -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 @@ async def ws_excel(websocket: WebSocket,
|
|
|
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)
|