zhaoqingang
2025-01-14 e79d383437d2b7316ee1def8049a13f3eb7c8dfa
Merge branch 'master' of http://192.168.5.5:10010/r/rag-gateway
2个文件已修改
13 ■■■■■ 已修改文件
app/api/agent.py 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
app/api/excel.py 11 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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)