3 files changed, 4 insertions, 2 deletions
diff --git a/lib/api/entities/ml/mlflow/experiment.rb b/lib/api/entities/ml/mlflow/experiment.rb
index 54e0fe63985..51650c36d98 100644
--- a/lib/api/entities/ml/mlflow/experiment.rb
+++ b/lib/api/entities/ml/mlflow/experiment.rb
@@ -9,6 +9,7 @@ module API
           expose :name
           expose(:lifecycle_stage) { |experiment| experiment.deleted_on? ? 'deleted' : 'active' }
           expose(:artifact_location) { |experiment| 'not_implemented' }
+          expose :metadata, as: :tags, using: KeyValue
         end
       end
     end
diff --git a/lib/api/entities/ml/mlflow/run_param.rb b/lib/api/entities/ml/mlflow/key_value.rb
index 75fee738f8b..cf2c32f6f44 100644
--- a/lib/api/entities/ml/mlflow/run_param.rb
+++ b/lib/api/entities/ml/mlflow/key_value.rb
@@ -4,7 +4,7 @@ module API
   module Entities
     module Ml
       module Mlflow
-        class RunParam < Grape::Entity
+        class KeyValue < Grape::Entity
           expose :name, as: :key
           expose :value
         end
diff --git a/lib/api/entities/ml/mlflow/run.rb b/lib/api/entities/ml/mlflow/run.rb
index 8b16c67611f..01d85e8862b 100644
--- a/lib/api/entities/ml/mlflow/run.rb
+++ b/lib/api/entities/ml/mlflow/run.rb
@@ -9,7 +9,8 @@ module API
             expose :itself, using: RunInfo, as: :info
             expose :data do
               expose :metrics, using: Metric
-              expose :params, using: RunParam
+              expose :params, using: KeyValue
+              expose :metadata, as: :tags, using: KeyValue
             end
           end
         end