diff --git a/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py b/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py index 81f67d1..cc8986a 100644 --- a/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py +++ b/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py @@ -6,11 +6,12 @@ from fastapi import APIRouter, Depends, Request, UploadFile, File, Form # from config.enums import BusinessType # from config.get_db import get_db from module_admin.annotation.log_annotation import Log -# from module_admin.aspect.interface_auth import CheckUserInterfaceAuth +from module_admin.aspect.interface_auth import CheckUserInterfaceAuth, CheckRoleInterfaceAuth # from module_admin.entity.vo.notice_vo import DeleteNoticeModel, NoticeModel, NoticePageQueryModel # from module_admin.entity.vo.user_vo import CurrentUserModel from module_admin.service.login_service import LoginService from module_admin.service.ragflow_service import RAGFlowService + from utils.log_util import logger # from utils.page_util import PageResponseModel from utils.response_util import ResponseUtil @@ -24,6 +25,7 @@ import asyncio ragflowController = APIRouter(prefix="/system/ragflow", dependencies=[Depends(LoginService.get_current_user)]) + # 查看数据集列表 @ragflowController.post("/dataset_list" # , response_model=PageResponseModel @@ -208,6 +210,19 @@ async def converse_with_chat_assistant( # return parse_result(result) + +# 获取用户权限 +@ragflowController.get('/get_user_permission', dependencies=[Depends(CheckRoleInterfaceAuth('pad'))]) +async def get_user_permission(current_user = Depends(LoginService.get_current_user)): + """ + 获取用户权限 + """ + + user_auth_list = current_user.permissions + print(user_auth_list) + + + return ResponseUtil.success(data=user_auth_list) def parse_result(result): code = result.get('code', 0)