From f887bbedab0aa1ad32031cc686dd70155ae5be2a Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Fri, 25 Jul 2025 17:06:29 +0800
Subject: [PATCH] r2 transform

---
 .../rag_pipeline_transform_service.py         | 20 +++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/api/services/rag_pipeline/rag_pipeline_transform_service.py b/api/services/rag_pipeline/rag_pipeline_transform_service.py
index 6b0a7b36bb..b85101a445 100644
--- a/api/services/rag_pipeline/rag_pipeline_transform_service.py
+++ b/api/services/rag_pipeline/rag_pipeline_transform_service.py
@@ -25,19 +25,22 @@ class RagPipelineTransformService:
         if not dataset:
             raise ValueError("Dataset not found")
         if dataset.pipeline_id and dataset.runtime_mode == "rag_pipeline":
-            return
+            return {
+            "pipeline_id": dataset.pipeline_id,
+            "dataset_id": dataset_id,
+            "status": "success",
+        }
         if dataset.provider != "vendor":
             raise ValueError("External dataset is not supported")
         datasource_type = dataset.data_source_type
         indexing_technique = dataset.indexing_technique
 
         if not datasource_type and not indexing_technique:
-            self._transfrom_to_empty_pipeline(dataset)
-            return
+            return self._transfrom_to_empty_pipeline(dataset)
+
         doc_form = dataset.doc_form
         if not doc_form:
-            self._transfrom_to_empty_pipeline(dataset)
-            return
+            return self._transfrom_to_empty_pipeline(dataset)
         retrieval_model = dataset.retrieval_model
         pipeline_yaml = self._get_transform_yaml(doc_form, datasource_type, indexing_technique)
         # deal dependencies
@@ -281,4 +284,9 @@ class RagPipelineTransformService:
         dataset.updated_by = current_user.id
         dataset.updated_at = datetime.now(UTC).replace(tzinfo=None)
         db.session.add(dataset)
-        db.session.commit()
\ No newline at end of file
+        db.session.commit()
+        return {
+            "pipeline_id": pipeline.id,
+            "dataset_id": dataset.id,
+            "status": "success",
+        }