kangda_robot_backend/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py

39 lines
1.6 KiB
Python

from datetime import datetime
from fastapi import APIRouter, Depends, Request
from pydantic_validation_decorator import ValidateFields
from sqlalchemy.ext.asyncio import AsyncSession
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.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
from module_admin.entity.vo.ragflow_vo import RagflowListQueryModel
# from config.env import RAGFlowConfig
ragflowController = APIRouter(prefix="/system/ragflow", dependencies=[Depends(LoginService.get_current_user)])
@ragflowController.post("/dataset_list", response_model=PageResponseModel
# , dependencies=[Depends(CheckUserInterfaceAuth("system:ragflow:list"))]"
)
async def get_system_ragflow_list(
request: Request,
rage_flow_query: RagflowListQueryModel ,
query_db: AsyncSession = Depends(get_db),
):
result = await RAGFlowService.get_ragflow_list_services(query_db, rage_flow_query)
return ResponseUtil.success(data = result)
# 获取分页数据
# ragflow_list, total = await RagflowService.get_ragflow_list(query_db, ragflow_page_query)