code加密

zhaoqingang

2025-04-01 b2e47e75a231baf5a7beca476256ab3d94e76c46

 app/api/agent.py

@@ -18,7 +18,7 @@
from app.service.dialog import get_session_history
from app.service.ragflow import RagflowService
from app.service.service_token import get_ragflow_token, get_bisheng_token
from app.task.fetch_agent import initialize_agents
# from app.task.fetch_agent import initialize_agents

router = APIRouter()

@@ -45,10 +45,10 @@
    if agent_type == AgentType.RAGFLOW:
        ragflow_service = RagflowService(base_url=settings.fwr_base_url)
        try:
            token = await get_ragflow_token(db, current_user.id)
            result = await ragflow_service.get_chat_sessions(token, agent_id)
            result = await get_session_history(db, current_user.id, agent_id, page, limit)
            if not result:
                result = await get_session_history(db, current_user.id, agent_id, page, limit)
                token = await get_ragflow_token(db, current_user.id)
                result = await ragflow_service.get_chat_sessions(token, agent_id)
        except Exception as e:
            print(e)
            raise HTTPException(status_code=500, detail=str(e))