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

xuyonghao

2025-01-14 9de1ba1b182cc82582da5825660ad5671bda0df2

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

 app/api/agent.py

@@ -192,8 +192,6 @@
            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):

 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)