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()