diff --git a/examples/mlflow_plugin/requirements.txt b/examples/mlflow_plugin/requirements.txt
index da24d689b..514e62c72 100644
--- a/examples/mlflow_plugin/requirements.txt
+++ b/examples/mlflow_plugin/requirements.txt
@@ -10,6 +10,8 @@ absl-py==1.4.0
     #   tensorflow
 alembic==1.11.1
     # via mlflow
+aniso8601==9.0.1
+    # via graphene
 arrow==1.2.3
     # via jinja2-time
 astunparse==1.6.3
@@ -19,7 +21,9 @@ binaryornot==0.4.4
 blinker==1.6.2
     # via flask
 cachetools==5.3.1
-    # via google-auth
+    # via
+    #   google-auth
+    #   mlflow
 certifi==2023.5.7
     # via
     #   kubernetes
@@ -33,7 +37,6 @@ charset-normalizer==3.1.0
 click==8.1.3
     # via
     #   cookiecutter
-    #   databricks-cli
     #   flask
     #   flytekit
     #   mlflow
@@ -48,17 +51,19 @@ cookiecutter==2.1.1
 croniter==1.4.1
     # via flytekit
 cryptography==41.0.1
-    # via pyopenssl
+    # via
+    #   pyopenssl
+    #   secretstorage
 cycler==0.11.0
     # via matplotlib
-databricks-cli==0.17.7
-    # via mlflow
 dataclasses-json==0.5.9
     # via flytekit
 decorator==5.1.1
     # via retry
 deprecated==1.2.14
-    # via flytekit
+    # via
+    #   flytekit
+    #   opentelemetry-api
 diskcache==5.6.1
     # via flytekit
 docker==6.1.3
@@ -104,6 +109,14 @@ googleapis-common-protos==1.56.4
     # via
     #   flyteidl
     #   grpcio-status
+graphene==3.3
+    # via mlflow
+graphql-core==3.2.3
+    # via
+    #   graphene
+    #   graphql-relay
+graphql-relay==3.2.0
+    # via graphene
 greenlet==2.0.2
     # via sqlalchemy
 grpcio==1.48.2
@@ -128,6 +141,7 @@ importlib-metadata==6.7.0
     #   keyring
     #   markdown
     #   mlflow
+    #   opentelemetry-api
 importlib-resources==5.12.0
     # via
     #   alembic
@@ -137,6 +151,10 @@ itsdangerous==2.1.2
     # via flask
 jaraco-classes==3.2.3
     # via keyring
+jeepney==0.8.0
+    # via
+    #   keyring
+    #   secretstorage
 jinja2==3.1.2
     # via
     #   cookiecutter
@@ -183,7 +201,7 @@ marshmallow-jsonschema==0.13.0
     # via flytekit
 matplotlib==3.7.2
     # via mlflow
-mlflow==2.4.1
+mlflow==2.13.2
     # via flytekitplugins-mlflow
 more-itertools==9.1.0
     # via jaraco-classes
@@ -207,9 +225,16 @@ numpy==1.23.5
     #   tensorboard
     #   tensorflow
 oauthlib==3.2.2
+    # via requests-oauthlib
+opentelemetry-api==1.25.0
     # via
-    #   databricks-cli
-    #   requests-oauthlib
+    #   mlflow
+    #   opentelemetry-sdk
+    #   opentelemetry-semantic-conventions
+opentelemetry-sdk==1.25.0
+    # via mlflow
+opentelemetry-semantic-conventions==0.46b0
+    # via opentelemetry-sdk
 opt-einsum==3.3.0
     # via tensorflow
 packaging==23.1
@@ -255,8 +280,6 @@ pyasn1-modules==0.3.0
     # via google-auth
 pycparser==2.21
     # via cffi
-pyjwt==2.7.0
-    # via databricks-cli
 pyopenssl==23.2.0
     # via flytekit
 pyparsing==3.0.9
@@ -294,7 +317,6 @@ regex==2023.6.3
 requests==2.31.0
     # via
     #   cookiecutter
-    #   databricks-cli
     #   docker
     #   flytekit
     #   kubernetes
@@ -318,10 +340,11 @@ scipy==1.10.1
     # via
     #   mlflow
     #   scikit-learn
+secretstorage==3.3.3
+    # via keyring
 six==1.16.0
     # via
     #   astunparse
-    #   databricks-cli
     #   google-auth
     #   google-pasta
     #   grpcio
@@ -342,8 +365,6 @@ sqlparse==0.4.4
     # via mlflow
 statsd==3.3.0
     # via flytekit
-tabulate==0.9.0
-    # via databricks-cli
 tenacity==8.2.2
     # via plotly
 tensorboard==2.9.0
@@ -370,6 +391,7 @@ typing-extensions==4.7.1
     # via
     #   alembic
     #   flytekit
+    #   opentelemetry-sdk
     #   sqlalchemy
     #   tensorflow
     #   typing-inspect
@@ -377,7 +399,6 @@ typing-inspect==0.9.0
     # via dataclasses-json
 urllib3==1.26.16
     # via
-    #   databricks-cli
     #   docker
     #   flytekit
     #   google-auth