diff --git a/operations/sagemaker-mlflow-rag-pipeline-automation/sagemaker_pipeline/steps/multi_step_pipeline/requirements.txt b/operations/sagemaker-mlflow-rag-pipeline-automation/sagemaker_pipeline/steps/multi_step_pipeline/requirements.txt
index 3d51451..54f0c4e 100644
--- a/operations/sagemaker-mlflow-rag-pipeline-automation/sagemaker_pipeline/steps/multi_step_pipeline/requirements.txt
+++ b/operations/sagemaker-mlflow-rag-pipeline-automation/sagemaker_pipeline/steps/multi_step_pipeline/requirements.txt
@@ -1,6 +1,6 @@
 datasets
 opensearch-py
-mlflow==2.22.1
+mlflow==3.5.0rc0
 sagemaker-mlflow
 langchain
 langgraph