89 lines
3.2 KiB
Python
89 lines
3.2 KiB
Python
from datetime import datetime
|
|
from typing import List
|
|
from fastapi import APIRouter, Depends, Request, UploadFile, File, Form
|
|
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, ListDocumentsQueryModel, UpdateFileModel
|
|
# 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_dastset_query: RagflowListQueryModel ,
|
|
# query_db: AsyncSession = Depends(get_db),
|
|
):
|
|
|
|
result = await RAGFlowService.get_ragflow_dataset_list_services(None, rage_flow_dastset_query)
|
|
|
|
return ResponseUtil.success(data = result)
|
|
|
|
# 获取分页数据
|
|
# ragflow_list, total = await RagflowService.get_ragflow_list(query_db, ragflow_page_query)
|
|
|
|
# 列出数据集中文档列表
|
|
@ragflowController.get("/list_documents/{dataset_id}")
|
|
async def list_documents_by_dataset_id(
|
|
request: Request,
|
|
dataset_id: str,
|
|
list_documents_query: ListDocumentsQueryModel = Depends(ListDocumentsQueryModel.as_query),
|
|
|
|
# query_db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""
|
|
列出数据集中文档列表
|
|
"""
|
|
print(list_documents_query)
|
|
result = await RAGFlowService.list_documents_services(None, dataset_id, list_documents_query)
|
|
|
|
return ResponseUtil.success(data = result)
|
|
|
|
# 上传文件到数据集
|
|
@ragflowController.post("/upload_file/{dataset_id}")
|
|
async def upload_file_dataset(
|
|
dataset_id: str,
|
|
files: List[UploadFile] = File(...),
|
|
# query_db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""
|
|
上传文件到数据集
|
|
"""
|
|
# print(file)
|
|
result = await RAGFlowService.upload_file_dataset_services(None, dataset_id ,files)
|
|
|
|
return ResponseUtil.success(data = result)
|
|
|
|
# 更新文档
|
|
@ragflowController.post("/update_file/{dataset_id}/{document_id}")
|
|
async def update_file_dataset(
|
|
dataset_id: str,
|
|
document_id: str,
|
|
update_params: UpdateFileModel,
|
|
# query_db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""
|
|
更新文件到数据集
|
|
"""
|
|
# print(file)
|
|
result = await RAGFlowService.update_file_dataset_services(dataset_id ,document_id, update_params)
|
|
|
|
return ResponseUtil.success(result) |