diff --git a/serverless-training/api-reference/openapi.json b/serverless-training/api-reference/openapi.json
index 59c543502a..6af6f90fb3 100644
--- a/serverless-training/api-reference/openapi.json
+++ b/serverless-training/api-reference/openapi.json
@@ -1754,6 +1754,43 @@
             ],
             "title": "Kl Penalty Coef"
           },
+          "kl_penalty_reference_step": {
+            "anyOf": [
+              {
+                "type": "integer"
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "title": "Kl Penalty Reference Step"
+          },
+          "kl_penalty_source": {
+            "anyOf": [
+              {
+                "type": "string",
+                "enum": [
+                  "current_learner",
+                  "sample"
+                ]
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "title": "Kl Penalty Source"
+          },
+          "kl_penalty_step_lag": {
+            "anyOf": [
+              {
+                "type": "integer"
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "title": "Kl Penalty Step Lag"
+          },
           "kl_ref_adapter_path": {
             "anyOf": [
               {
@@ -2387,6 +2424,18 @@
               }
             ],
             "title": "Learning Rate"
+          },
+          "metric_logging": {
+            "anyOf": [
+              {
+                "additionalProperties": true,
+                "type": "object"
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "title": "Metric Logging"
           }
         },
         "type": "object",