diff --git a/openapi.yaml b/openapi.yaml
index 029a20a..82f68a2 100644
--- a/openapi.yaml
+++ b/openapi.yaml
@@ -713,6 +713,7 @@ paths:
                   type: boolean
                   default: auto
                   description: Whether to mask the user messages in conversational data or prompts in instruction data.
+                  deprecated: true
                 training_method:
                   type: object
                   oneOf:
@@ -2865,15 +2866,25 @@ components:
         - type
         - lora_r
         - lora_alpha
+
     TrainingMethodSFT:
       type: object
       properties:
         method:
           type: string
           enum: ["sft"]
+        train_on_inputs:
+          oneOf:
+            - type: boolean
+            - type: string
+              enum:
+                - auto
+          type: boolean
+          default: auto
+          description: Whether to mask the user messages in conversational data or prompts in instruction data.
       required:
         - method
-
+        - train_on_inputs
     TrainingMethodDPO:
       type: object
       properties: