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