From dc3841ddeb263536585b0f49942c466f6b70e77c Mon Sep 17 00:00:00 2001
From: connermanuel <manuel.conner.g@gmail.com>
Date: Fri, 18 Apr 2025 13:42:56 -0700
Subject: [PATCH] move TrainOnInputs to SFTType, and make it required

---
 openapi.yaml | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

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: