实现更新数据集接口

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)
 
 
 # 列出数据集中文档列表

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