From be963254166a5d826bf8f1f31a88b0a3e81dfba3 Mon Sep 17 00:00:00 2001
From: haotian <2421912570@qq.com>
Date: Fri, 5 Sep 2025 11:44:00 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=9E=E7=8E=B0=E6=9B=B4=E6=96=B0=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E9=9B=86=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/ragflow_controller.py          |  9 ++++++++
 .../module_admin/service/ragflow_service.py   | 23 +++++++++++++++++++
 2 files changed, 32 insertions(+)

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