r2 transform

This commit is contained in:
jyong 2025-07-16 12:00:26 +08:00
parent 6363ecef97
commit 675ff9bd7d
1 changed files with 2 additions and 0 deletions

View File

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