diff --git a/assets/training/aoai/proxy_components/environments/context/requirements.txt b/assets/training/aoai/proxy_components/environments/context/requirements.txt
index 5a97cc9382..73136902fe 100644
--- a/assets/training/aoai/proxy_components/environments/context/requirements.txt
+++ b/assets/training/aoai/proxy_components/environments/context/requirements.txt
@@ -1,6 +1,6 @@
 azure-ai-ml==1.14.0
 openai==1.28.0
-mlflow==2.12.2
+mlflow==2.16.0
 pandas==2.2.0
 azureml-mlflow==1.56.0
 azure-mgmt-cognitiveservices==13.5.0
@@ -9,4 +9,4 @@ jsonlines==4.0.0
 azureml-telemetry==1.56.0
 pydantic==2.7.0
 idna>=3.7
-azure-keyvault-secrets==4.8.0
\ No newline at end of file
+azure-keyvault-secrets==4.8.0