From 76fffdbc1de2e374959150592b0df19e442cd61d Mon Sep 17 00:00:00 2001
From: haotian <2421912570@qq.com>
Date: Fri, 5 Sep 2025 10:10:51 +0800
Subject: [PATCH] =?UTF-8?q?1.=E4=BF=AE=E6=94=B9service=E8=BF=94=E5=9B=9E?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E6=A0=BC=E5=BC=8F\n2.=E5=B0=81=E8=A3=85?=
 =?UTF-8?q?=E8=A7=A3=E6=9E=90=E8=BF=94=E5=9B=9E=E7=BB=93=E6=9E=9C=E6=96=B9?=
 =?UTF-8?q?=E6=B3=95,=E7=8E=B0=E5=8F=AF=E6=A0=B9=E6=8D=AE=E8=BF=94?=
 =?UTF-8?q?=E5=9B=9Ecode=E8=BF=94=E5=9B=9E=E4=B8=8D=E5=90=8C=E7=9A=84succe?=
 =?UTF-8?q?s/error?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/ragflow_controller.py          | 23 +++++++++++++------
 .../module_admin/service/ragflow_service.py   |  8 +++----
 2 files changed, 20 insertions(+), 11 deletions(-)

diff --git a/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py b/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py
index 21fe39d..439666a 100644
--- a/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py
+++ b/ruoyi-fastapi-backend/module_admin/controller/ragflow_controller.py
@@ -35,10 +35,8 @@ async def get_system_ragflow_list(
     
     result = await RAGFlowService.get_ragflow_dataset_list_services(None, rage_flow_dastset_query)
     
-    return ResponseUtil.success(data = result)
+    return parse_result(result)
     
-    # 获取分页数据
-    # ragflow_list, total = await RagflowService.get_ragflow_list(query_db, ragflow_page_query)
     
 # 列出数据集中文档列表
 @ragflowController.get("/list_documents/{dataset_id}")
@@ -55,7 +53,8 @@ async def list_documents_by_dataset_id(
     print(list_documents_query)
     result = await RAGFlowService.list_documents_services(None, dataset_id, list_documents_query)
     
-    return ResponseUtil.success(data = result)
+    return parse_result(result)
+    
 
 # 上传文件到数据集
 @ragflowController.post("/upload_file/{dataset_id}")
@@ -70,8 +69,8 @@ async def upload_file_dataset(
     # print(file)
     result = await RAGFlowService.upload_file_dataset_services(None, dataset_id ,files)
     
-    return ResponseUtil.success(data = result)
-
+    return parse_result(result)
+    
 # 更新文档
 @ragflowController.post("/update_file/{dataset_id}/{document_id}")
 async def update_file_dataset(
@@ -86,4 +85,14 @@ async def update_file_dataset(
     # print(file)
     result = await RAGFlowService.update_file_dataset_services(dataset_id ,document_id, update_params)
     
-    return ResponseUtil.success(result)
\ No newline at end of file
+    return parse_result(result)
+    
+
+
+
+
+def parse_result(result):
+    code = result.get('code', 0)
+    if code != 0:
+        return ResponseUtil.error(result.get('data', None))
+    return ResponseUtil.success(result.get('data', None))
\ No newline at end of file
diff --git a/ruoyi-fastapi-backend/module_admin/service/ragflow_service.py b/ruoyi-fastapi-backend/module_admin/service/ragflow_service.py
index df48f11..5ef00b4 100644
--- a/ruoyi-fastapi-backend/module_admin/service/ragflow_service.py
+++ b/ruoyi-fastapi-backend/module_admin/service/ragflow_service.py
@@ -19,7 +19,7 @@ class RAGFlowService:
             result = await client.list_datasets(**(rage_flow_query.model_dump()))
         
         # 获取分页数据
-        return result.get('data', None)
+        return result
     
     # 获取数据集中文档列表
     @classmethod
@@ -32,7 +32,7 @@ class RAGFlowService:
         async with AsyncRAGFlowClient(RAGFlowConfig.RAGFLOW_BASE_URL, RAGFlowConfig.RAGFLOW_API_KEY) as client:
             result = await client.list_documents(dataset_id=dataset_id, **(list_documents_query.model_dump()))
              
-        return result.get('data', None)
+        return result
     
     # 上传文档到数据集
     async def upload_file_dataset_services(
@@ -43,7 +43,7 @@ class RAGFlowService:
         async with AsyncRAGFlowClient(RAGFlowConfig.RAGFLOW_BASE_URL, RAGFlowConfig.RAGFLOW_API_KEY) as client:
             result = await client.upload_documents_bytes(dataset_id=dataset_id, file_bytes=files)
              
-        return result.get('data', None)
+        return result
     
     # 更新文档内容
     @classmethod
@@ -56,4 +56,4 @@ class RAGFlowService:
         async with AsyncRAGFlowClient(RAGFlowConfig.RAGFLOW_BASE_URL, RAGFlowConfig.RAGFLOW_API_KEY) as client:
             result = await client.update_document(dataset_id=dataset_id, document_id=document_id, **(update_params.model_dump()))
              
-        return result.get('data', None)
\ No newline at end of file
+        return result
\ No newline at end of file