From f9a307e86b771f20bd2dc043a875b2ee86cc5d50 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期二, 12 十一月 2024 18:42:12 +0800 Subject: [PATCH] Merge branch 'master' of http://192.168.5.5:10010/r/rag-gateway --- app/api/agent.py | 12 ++++++++++++ 1 files changed, 12 insertions(+), 0 deletions(-) diff --git a/app/api/agent.py b/app/api/agent.py index 2879506..028bded 100644 --- a/app/api/agent.py +++ b/app/api/agent.py @@ -1,6 +1,7 @@ import uuid from fastapi import Depends, APIRouter, Query, HTTPException +from fastapi.responses import JSONResponse from pydantic import BaseModel from sqlalchemy.orm import Session @@ -51,6 +52,17 @@ return ResponseList(code=200, msg="Unsupported agent type") +@router.get("/{conversation_id}/session_log") +async def session_log(conversation_id: str, db: Session = Depends(get_db), current_user: UserModel = Depends(get_current_user)): + ragflow_service = RagflowService(base_url=settings.fwr_base_url) + try: + token = get_ragflow_token(db, current_user.id) + result = await ragflow_service.get_session_log(token, conversation_id) + except Exception as e: + raise HTTPException(status_code=500, detail=str(e)) + return JSONResponse(status_code=200, content={"code": 200, "log": result}) + + @router.get("/get-chat-id/{agent_id}", response_model=Response) async def get_chat_id(agent_id: str, db: Session = Depends(get_db)): agent = db.query(AgentModel).filter(AgentModel.id == agent_id).first() -- Gitblit v1.8.0