diff --git a/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py b/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py index 8aefe81..ee19c3b 100644 --- a/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py +++ b/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py @@ -47,6 +47,15 @@ async def create_dataset( result = await RAGFlowService.create_dataset_services(create_dataset_params) return parse_result(result) +# 更新数据集 +@ragflowController.post('/update_dataset/{dataset_id}') +async def update_dataset( + request: Request, + dataset_id: str, + update_dataset_params: CreateDatasetModel, +): + result = await RAGFlowService.update_dataset_services(dataset_id, update_dataset_params) + return parse_result(result) # 列出数据集中文档列表 diff --git a/ruoyi-fastapi-backend/module_admin/service/ragflow_service.py b/ruoyi-fastapi-backend/module_admin/service/ragflow_service.py index fc91fd8..b86dbe8 100644 --- a/ruoyi-fastapi-backend/module_admin/service/ragflow_service.py +++ b/ruoyi-fastapi-backend/module_admin/service/ragflow_service.py @@ -39,6 +39,29 @@ class RAGFlowService: return result + # 更新数据集 + @classmethod + async def update_dataset_services( + cls, + dataset_id: str, + update_dataset_params: CreateDatasetModel, + ): + """更新数据集信息 + + Args: + dataset_id (str): 数据集id + update_dataset_params (CreateDatasetModel): 更新参数 + + Returns: + _type_: _description_ + """ + async with AsyncRAGFlowClient(RAGFlowConfig.RAGFLOW_BASE_URL, RAGFlowConfig.RAGFLOW_API_KEY) as client: + result = await client.update_dataset( + dataset_id=dataset_id, **(update_dataset_params.model_dump()) + ) + + return result + # 获取数据集中文档列表 @classmethod