소스 검색

合并excel对话记录保存格式与出题组卷保持一致

xuyonghao 1 년 전
부모
커밋
9de1ba1b18
2개의 변경된 파일8개의 추가작업 그리고 5개의 파일을 삭제
  1. 0 2
      app/api/agent.py
  2. 8 3
      app/api/excel.py

+ 0 - 2
app/api/agent.py

@@ -192,8 +192,6 @@ async def session_log(agent_id: str, conversation_id: str, db: Session = Depends
             for i in session.log_to_json().get("message", []):
                 if i.get("role") == "user":
                     tmp_data["question"]=i.get("content")
-                    if i.get("download_url") is not None:
-                        tmp_data["download_url"] = i.get("download_url")
                 elif i.get("role") == "assistant":
 
                     if isinstance(i.get("content"), dict):

+ 8 - 3
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 @@ 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)