From 675ff9bd7da10fe221cd411bd1785fd2b9b1ad8e Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Wed, 16 Jul 2025 12:00:26 +0800
Subject: [PATCH] r2 transform

---
 api/services/rag_pipeline/rag_pipeline_transform_service.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/api/services/rag_pipeline/rag_pipeline_transform_service.py b/api/services/rag_pipeline/rag_pipeline_transform_service.py
index f87f2e994a..009495e690 100644
--- a/api/services/rag_pipeline/rag_pipeline_transform_service.py
+++ b/api/services/rag_pipeline/rag_pipeline_transform_service.py
@@ -147,6 +147,8 @@ class RagPipelineTransformService:
             if indexing_technique == "economy":
                 retrieval_setting.search_method = "keyword_search"
             knowledge_configuration.retrieval_model = retrieval_setting
+        else:
+            dataset.retrieval_model = knowledge_configuration.retrieval_model.model_dump()
 
         return knowledge_configuration.model_dump()